#define Monitor_raw_cxx

#include "Monitor_raw.h"
#include <TH2.h>
#include <TH1.h>
#include <TStyle.h>
#include <TCanvas.h>
#include <TMath.h>
#include <vector>
#include <TStopwatch.h>

//############################################ User setting

int rawEnergyRange[2] = {500,  6000}; // in ch, {min, max}
int    energyRange[3] = {2, 40,  2000}; // in keV, {resol, min, max}

TString e_corr = "correction_e.dat";

Long64_t maxEvent = -1;

//############################################ end of user setting

ULong64_t NumEntries = 0;
ULong64_t ProcessedEntries = 0;
Float_t Frac = 0.1; ///Progress bar
TStopwatch StpWatch;

vector<vector<double>> eCorr;

//############################################ histogram declaration

TH1F * he[NCRYSTAL];
TH1F * heCal[NCRYSTAL];

TH2F * heCalvID;

//############################################ BEGIN
void Monitor_raw::Begin(TTree * tree){
   TString option = GetOption();
   
   NumEntries = tree->GetEntries();
   
   printf("======================== Creating histograms\n");

   for( int i = 0 ; i < NCRYSTAL ; i++){
      he[i]    = new TH1F (Form("he%02d", i),            Form("e%02d", i),            rawEnergyRange[1]-rawEnergyRange[0], rawEnergyRange[0], rawEnergyRange[1]);  
      heCal[i] = new TH1F (Form("heCal%02d", i), Form("e%02d (Cali.)", i), (energyRange[2]-energyRange[1])/energyRange[0],    energyRange[1],    energyRange[2]);  
      
      switch (i%4){
         case 0: he[i]->SetLineColor(2);break;
         case 1: he[i]->SetLineColor(kYellow+3);break;
         case 2: he[i]->SetLineColor(kGreen+2);break;
         case 3: he[i]->SetLineColor(4);break;
      }
      switch (i%4){
         case 0: heCal[i]->SetLineColor(2);break;
         case 1: heCal[i]->SetLineColor(kYellow+3);break;
         case 2: heCal[i]->SetLineColor(kGreen+2);break;
         case 3: heCal[i]->SetLineColor(4);break;
      }
   }
   
   heCalvID = new TH2F("heCalvID", "ID vs Energy (Cali.); ID; Energy",   NCRYSTAL, 0, NCRYSTAL, (energyRange[2]-energyRange[1])/energyRange[0],    energyRange[1],    energyRange[2]);

   
   printf("======================== end of histograms creation.\n");
   
   printf("======================== Load parameters.\n");
   eCorr = LoadCorrectionParameters(e_corr); 
   
   StpWatch.Start();
   printf("======================== Start processing....\n");
   
}

//############################################ PROCESS
Bool_t Monitor_raw::Process(Long64_t entry){

   ProcessedEntries++;
   
   /*********** Progress Bar ******************************************/ 
   if (ProcessedEntries>NumEntries*Frac-1) {
      TString msg; msg.Form("%llu", NumEntries/1000);
      int len = msg.Sizeof();
      printf(" %3.0f%% (%*llu/%llu k) processed in %6.1f sec | expect %6.1f sec\n",
               Frac*100, len, ProcessedEntries/1000,NumEntries/1000,StpWatch.RealTime(), StpWatch.RealTime()/Frac);
      StpWatch.Start(kFALSE);
      Frac+=0.1;
   }

   if ( maxEvent > 0 && entry > maxEvent ) {
      Abort(Form("user abort, > %lld \n", maxEvent));
      Terminate();
   }
   
   b_ID->GetEntry(entry);
   b_energy->GetEntry(entry);
   b_timestamp->GetEntry(entry);
   
   int detID = mapping[ID];
   
   if( 0 <= detID && detID < 100 ){
      he[detID]->Fill(e);
      
      //double eCal = ApplyCorrection(eCorr, detID, e);
      double eCal = eCorr[detID][0]+eCorr[detID][1]*e;
      heCal[detID]->Fill(eCal);
      
      heCalvID->Fill(detID, eCal);
   }
   

   return kTRUE;
}

//############################################ TERMINATE
void Monitor_raw::Terminate(){
   
   printf("============================== finishing.\n");
   gROOT->cd();
   
   int nCrystalPerClover = 4;
   int nClover = NCRYSTAL / nCrystalPerClover;

   TCanvas * cc = new TCanvas("cc", "cc", 2000, 2000);
   if( cc->GetShowEventStatus() == 0 ) cc->ToggleEventStatus();
   cc->Divide(1, 9, 0);
   
    for (Int_t i = 0; i < nClover; i++) {
      int canvasID =  i + 1;
      cc->cd(canvasID); 
      cc->cd(canvasID)->SetGrid();       
      cc->cd(canvasID)->SetTickx(2);   
      cc->cd(canvasID)->SetTicky(2);   
      cc->cd(canvasID)->SetBottomMargin(0.06);
      cc->cd(canvasID)->SetLogy();

      for( Int_t j = 0; j < nCrystalPerClover; j++){
         int hID = nCrystalPerClover*i+ j ;
         heCal[hID]->Draw("same");         
         //he[hID]->Draw("same");         
      }
   }
   cc->SetCrosshair(1);
   
   TCanvas * c1 = new TCanvas("c1", "c1", 1000, 1000);
   if( c1->GetShowEventStatus() == 0 ) c1->ToggleEventStatus();
   c1->SetLogz();
   c1->SetGridx();
   heCalvID->SetNdivisions(-409, "X");
   heCalvID->Draw("colz");
   
}