#include "../Headers/gui.h"

MyMainFrame::MyMainFrame(const Setup* config_, const TGWindow *p,UInt_t w,UInt_t h): config(config_) {
   // use hierarchical cleaning
   fMain = new TGMainFrame(p,w,h);
   fMain->SetCleanup(kDeepCleanup);

   fMain->Connect("CloseWindow()", "MyMainFrame", this, "CloseWindow()");
   // create layout
   Init();
   canvas = fEcanvas->GetCanvas(); 
   canvas->Update();
   // popup window
   fMain->SetWindowName("Back projection reconstruction");
   fMain->MapSubwindows();
   fMain->Resize(fMain->GetDefaultSize());
   fMain->Resize(w, h);
   fMain->MapWindow();
}

MyMainFrame::~MyMainFrame() {
    // // Clean up used widgets: frames, buttons, layout hints
    // fMain->Cleanup();
    //buttons
    delete fF3;
    delete fL3;
    delete fStopB;
    delete fClearB;
    delete fStartB;
    // canvas
    delete fF5;
    delete fL4;
    delete fEcanvas;
    delete fMain;
    delete histo;
    for (int i = 0; i < Nl; i++)
    {
        delete latitudes[i];
    }
    for (int i = 0; i < NL; i++)
    {
        delete longitudes[i];
    }
    delete sourceMarker;
}

void MyMainFrame::CloseWindow() {
   // Got close message for this MainFrame. Terminates the application.
   gApplication->Terminate(0);
}

int MyMainFrame::Init(){
    // buttons
    fF3 = new TGCompositeFrame(fMain, 60, 20, kHorizontalFrame);
    fStartB = new TGTextButton(fF3, "Start");
    fStartB->Connect("Clicked()", "MyMainFrame", this, "Start()");
    fClearB = new TGTextButton(fF3, "Clear");
    fClearB->Connect("Clicked()", "MyMainFrame", this, "Clear()");
    fStopB = new TGTextButton(fF3, "Stop");
    fStopB->Connect("Clicked()", "MyMainFrame", this, "Stop()");
    fL3 = new TGLayoutHints(kLHintsTop | kLHintsLeft, 
                        5, 5, 5, 5);
    fF3->AddFrame(fStartB, fL3);
    fF3->AddFrame(fClearB, fL3);
    fF3->AddFrame(fStopB, fL3);

    // embedded canvas
    fF5 = new TGCompositeFrame(fMain, 60, 60, kHorizontalFrame);
    fL4 = new TGLayoutHints(kLHintsTop | kLHintsLeft 
                            | kLHintsExpandX | kLHintsExpandY, 5, 5, 5, 5);
    fEcanvas = new TRootEmbeddedCanvas("ec1", fF5, 100, 100);
    fF5->AddFrame(fEcanvas, fL4);
    fMain->AddFrame(fF3, fL3);
    fMain->AddFrame(fF5, fL4);

    initHist();
    drawGridlines();
    drawSource();
    // end embeded canvas
    return 0;
}

void MyMainFrame::initHist() {
    histo = new TH2D("ROI", " ; Azimuth; Elevation", 
                            config->phiBins, -180, 180, 
                            config->thetaBins, -90, 90);
    // canvas = new TCanvas("THCanvas","Sinocanvas", 1000, 500);
    // canvas->Connect("Closed()", "Image", this, "closed()");
    // 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 MyMainFrame::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 MyMainFrame::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 MyMainFrame::Start()
{
//    printf("\"Start updating\" Button Pressed!\n");
   stop = false;
}

void MyMainFrame::Stop()
{
//    printf("\"Stop updating\" Button Pressed!\n");
   stop = true;
}

void MyMainFrame::Clear()
{
//    printf("\"Clear image\" Button Pressed!\n");
   counts = 0;
   histo->Reset();
   redraw();
}

void MyMainFrame::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));
            }
        }
    }
    // redraw
    redraw();
}

void MyMainFrame::redraw() {
    canvas->Modified();
    canvas->Update();
}

int aitoff2xy(const double& l, const double& b, double &Al, double &Ab)
{
   Double_t x, y;

   Double_t alpha2 = (l/2)*TMath::DegToRad();
   Double_t delta  = b*TMath::DegToRad();
   Double_t r2     = TMath::Sqrt(2.);
   Double_t f      = 2*r2/TMath::Pi();
   Double_t cdec   = TMath::Cos(delta);
   Double_t denom  = TMath::Sqrt(1. + cdec*TMath::Cos(alpha2));
   x      = cdec*TMath::Sin(alpha2)*2.*r2/denom;
   y      = TMath::Sin(delta)*r2/denom;
   x     *= TMath::RadToDeg()/f;
   y     *= TMath::RadToDeg()/f;
   //  x *= -1.; // for a skymap swap left<->right
   Al = x;
   Ab = y;

   return 0;
}