dynamic update canvas

Headers/draw.h

@@ -11,6 +11,54 @@
 
 #include "../Headers/setup.h"
 
+class Image
+{
+private:
+    /* data */
+public:
+    const Setup& config;
+    TCanvas* canvas;
+    TH2D* histo;
+    void initDrawImage();
+    // gridlines
+    const int Nl = 19; // Number of drawn latitudes
+    const int NL = 19; // Number of drawn longitudes
+    int M  = 30;
+    std::vector<TGraph*> latitudes;
+    std::vector<TGraph*> longitudes;
+    void drawGridlines();
+    // source marker
+    TEllipse* sourceMarker;
+    void drawSource();
+    void addCone(const std::vector<Cone> cones);
+    Image(const Setup& config_): config(config_) {
+        initDrawImage();
+        drawGridlines();
+        drawSource();
+    }
+    ~Image(){
+        if (canvas)
+            delete canvas;
+        if (histo)
+            delete histo;
+        for (int i = 0; i < Nl; i++)
+        {
+            if (latitudes[i])
+                delete latitudes[i];
+        }
+        for (int i = 0; i < NL; i++)
+        {
+            if (longitudes[i])
+                delete longitudes[i];
+        }
+        if (!sourceMarker)
+        {
+            delete sourceMarker;
+        }
+        
+    }
+};
+
 int Backprojection(const Setup& config, const std::vector<Cone>& cones, 
                    std::vector<std::vector<double>>& image);
 

Sources/draw.cpp

 #include "../Headers/draw.h"
 
+void Image::initDrawImage() {
+    histo = new TH2D("ROI", " ; Azimuth; Elevation", 
+                            config.phiBins, -180, 180, 
+                            config.thetaBins, -90, 90);
+    canvas = new TCanvas("THCanvas","Sinocanvas", 1000, 500);
+    // init image
+    for (int i = 0; i < config.phiBins; i++)
+    {
+        for (int j = 0; j < config.thetaBins; j++)
+        {
+            histo->SetBinContent(i+1, j+1, 0);
+        }
+    }
+    gStyle->SetOptStat(0);
+    histo->GetZaxis()->SetLabelSize(0.02);
+    histo->Draw("z aitoff");
+}
+
+void Image::drawGridlines() {
+    float conv=M_PI/180;
+    float la, lo, x, y, z;
+    for (int j=0;j<Nl;++j) {
+        // latitudes[j]=new TGraph();
+        latitudes.push_back(new TGraph());
+        la = -90+180/(Nl-1)*j;
+        for (int i=0;i<M+1;++i) {
+            lo = -180+360/M*i;
+            z  = sqrt(1+cos(la*conv)*cos(lo*conv/2));
+            x  = 180*cos(la*conv)*sin(lo*conv/2)/z;
+            y  = 90*sin(la*conv)/z;
+            latitudes[j]->SetPoint(i,x,y);
+        }
+    } 
+    
+    for (int j=0;j<NL;++j) {
+        // longitudes[j]=new TGraph();
+        longitudes.push_back(new TGraph());
+        lo = -180+360/(NL-1)*j;
+        for (int i=0;i<M+1;++i) {
+            la = -90+180/M*i;
+            z  = sqrt(1+cos(la*conv)*cos(lo*conv/2));
+            x  = 180*cos(la*conv)*sin(lo*conv/2)/z;
+            y  = 90*sin(la*conv)/z;
+            longitudes[j]->SetPoint(i,x,y);
+        }
+    }
+    // Draw the grid	
+    TPad *pad2 = new TPad("pad2","",0,0,1,1);
+    pad2->SetFillStyle(4000);
+    pad2->SetFillColor(0);
+    pad2->SetBorderSize(0);
+    pad2->SetFrameBorderMode(0);
+    pad2->SetFrameLineColor(0); 
+    pad2->SetFrameBorderMode(0);
+    pad2->Draw();
+    pad2->cd();
+    Double_t ymin = -89.5;
+    Double_t ymax = 89.5;
+    Double_t dy = (ymax-ymin)/0.8; //10 per cent margins top and bottom
+    Double_t xmin = -180;
+    Double_t xmax = 180;
+    Double_t dx = (xmax-xmin)/0.8; //10 per cent margins left and right
+
+    pad2->Range(xmin-0.1*dx,ymin-0.1*dy,xmax+0.1*dx,ymax+0.1*dy);
+
+    for (int j=0;j<Nl;++j) latitudes[j]->Draw("l");
+    for (int j=0;j<NL;++j) longitudes[j]->Draw("l");
+}
+
+void Image::drawSource() {
+    // draw source marker
+    float conv=M_PI/180;
+    double sigma = 1;
+    double Ax, Ay;
+    aitoff2xy(config.truePhi / conv, config.trueTheta / conv, Ax, Ay);
+    sourceMarker = new TEllipse(Ax, Ay, sigma, sigma);
+    sourceMarker->SetFillColor(0);
+    sourceMarker->SetFillStyle(0);
+    sourceMarker->SetLineWidth(2);
+    sourceMarker->SetLineColor(kRed);
+    sourceMarker->Draw("SAME");
+}
+
+void Image::addCone(const std::vector<Cone> cones){
+    // project cones onto the spherical surface
+    // std::cout << "Projecting cones onto the designated spherical surface..."<<std::endl;
+    double alpha(0);
+    double beta(0);
+    for (const Cone& event : cones)
+    {
+        // update the image
+        //TODO: add the uncertainty
+        alpha = event.cosHalfAngle;
+        for (int i = 0; i < config.thetaBins; i++)
+        {
+            for (int j = 0; j < config.phiBins; j++)
+            {
+                beta = getCosAngle(event.apex - config.xbs[i][j], event.axis);
+                histo->SetBinContent(j+1, i+1, histo->GetBinContent(j+1, i+1) + std::exp(-(beta-alpha)*(beta-alpha) / 0.002));
+                // histo->SetBinContent(j+1, i+1, std::exp(-(beta-alpha)*(beta-alpha) / 0.002));
+            }
+        }
+    }
+}
+
 int Backprojection(const Setup& config, const std::vector<Cone>& cones, 
                    std::vector<std::vector<double>>& image){
     // project cones onto the spherical surface

Sources/main.cpp

@@ -3,6 +3,7 @@
 #include <cstdlib>
 #include <ctime>
 
+#include "TSystem.h"
 #include "TApplication.h"
 
 #include "../Headers/setup.h"
@@ -10,22 +11,43 @@
 #include "../Headers/draw.h"
 
 int main(int argc, char** argv) {
+    TApplication *myapp = new TApplication("myApp", 0, 0);
     // init
     const Setup config(10);
-    // simulate some cones
+    // simulate some cones on worker thread
     std::vector<Cone> cones;
     std::srand(std::time(nullptr));
-    for (int i = 0; i < config.nCones; i++)
+    for (int i = 0; i < 10; i++)
     {
         cones.push_back(createCone(config));
     }
-    // generate a projection image
+    // generate a projection image on master thread
     // init image
-    std::vector<std::vector<double>> image(config.thetaBins, std::vector<double>(config.phiBins, 0));
-    // draw    
-    TApplication *myapp = new TApplication("myApp", 0, 0);
-    Backprojection(config, cones, image);
-    drawImage(config, image);
+    // std::vector<std::vector<double>> image(config.thetaBins, std::vector<double>(config.phiBins, 0));
+    // draw
+    Image image(config);
+    image.canvas->Draw();
+    std::cout << "init draw" << '\n';
+    int counter(1);
+    while (true)
+    {
+        // update image
+        for (int i = 0; i < 5; i++)
+        {
+            cones[i] = createCone(config);
+        }
+        image.addCone(cones);
+        std::cout << "draw " << counter << '\n';
+        counter++;
+        // redraw
+        image.canvas->Modified();
+        image.canvas->Update();
+        gSystem->Sleep(100);
+        gSystem->ProcessEvents();
+    }
+    
+    // Backprojection(config, cones, image);
+    // drawImage(config, image);
     myapp->Run();
     return 0;
 }
\ No newline at end of file