//
// Author: Michele Floris APR2011
// Derived by root's the statusBar.C tutorial
// This macro allows to browse the OADB of the spectra


#include <TApplication.h>
#include <TGClient.h>
#include <TGButton.h>
#include <TGFrame.h>
#include <TFrame.h>
#include <TRootEmbeddedCanvas.h>
#include <TGStatusBar.h>
#include <TGComboBox.h>
#include <TCanvas.h>
#include <TF1.h>
#include <TRandom.h>
#include <TGraph.h>
#include <TAxis.h>
#include <AliOADBPWG2Spectra.h>
#include "TSystem.h"
#include "AliOADBContainer.h"
#include <TGTextEntry.h>
#include "TFile.h"
#include <iostream>
#include "TH1D.h"
#include "TInterpreter.h"
#include "AliBWFunc.h"
#include "TDatabasePDG.h"

using namespace std;



class MyMainFrame : public TGMainFrame {

private:
  TRootEmbeddedCanvas  *fEcan; // embedded canvas
  TGStatusBar          *fStatusBar;// status bar
  TGComboBox           *fComboDetector;//detectors
  TGComboBox           *fComboParticle;//particle
  TGComboBox           *fComboCharge;//charge
  TGComboBox           *fComboPID;//PID
  TGComboBox           *fComboContainer;//container
  TGTextEntry          *fRunNumber; // run number 
  TGComboBox           *fComboCentr; // centr tag
  TGTextEntry          *fTextCentrBin; // centrality bin
  TGTextEntry          *fTextDrawOpt; // draw opt
  TGTextEntry          *fTextSaveName; // save name
  AliOADBPWG2Spectra   *fOADBSpectra; // spectra OADB
  AliOADBContainer     *fOADBContainer; // OADB container
   
  Double_t fMass[3]; //masses
  AliBWFunc * fFuncMan; // functions
  static const char * fkContainers[];
  static const Int_t fkNContainers ;
  static const char * fkCentrTags[];
  static const Int_t fkNCentrTags ;

public:
  MyMainFrame(const TGWindow *p, UInt_t w, UInt_t h);
  virtual ~MyMainFrame();
  void DoExit();
  void DoDraw();
  void DoRatio();
  void DoFit();
  void DoLegend();
  void DoLoad();
  void DoClear();
  void DoPrint();
  void DoSelectedDetector(Int_t id);
  void SetStatusText(const char *txt, Int_t pi);
  void EventInfo(Int_t event, Int_t px, Int_t py, TObject *selected);
   
  ClassDef(MyMainFrame, 0)
};

const char * MyMainFrame::fkContainers[] = {"Corrected", "Raw"};
const Int_t MyMainFrame::fkNContainers = 2;
const char * MyMainFrame::fkCentrTags[] = {"V0M", "MB", "Ntracks", "SPD2"};
const Int_t MyMainFrame::fkNCentrTags = 4;

void MyMainFrame::DoSelectedDetector(Int_t id) {
  // Change the analysis type based on the detector
  cout << "ID " << id << endl;
  if (id == AliOADBPWG2Spectra::kTOFTPC) {
    fComboPID->Select(AliOADBPWG2Spectra::kNSigma);
  }
  else {
    fComboPID->Select(AliOADBPWG2Spectra::kGaussFit);
  }

}

void MyMainFrame::DoFit()
{
  // print to file
  static TF1 * func = 0;
  if(func) delete func;
  func = fFuncMan->GetLevi(fMass[fComboParticle->GetSelected()], 0.12, 7, 1.5);  
  TCanvas *c1 = fEcan->GetCanvas();
  TH1D * h = (TH1D*) c1->GetListOfPrimitives()->At(c1->GetListOfPrimitives()->GetEntries()-1);
  h->Fit(func,"","same");
  c1->Update();
  c1->Modified();
  c1->Update(); 

}

void MyMainFrame::DoPrint()
{
  // print to file
  TCanvas *c1 = fEcan->GetCanvas();
  c1->Print(fTextSaveName->GetText());
}
void MyMainFrame::DoClear()
{
  // clear canvas
  TCanvas *c1 = fEcan->GetCanvas();
  c1->Clear();
  c1->Update();
  c1->Modified();
  c1->Update(); 

}

void MyMainFrame::DoLoad()
{
  // Load file
  cout << "Getting " << fkContainers[fComboContainer->GetSelected()] << endl;
  cout << fRunNumber->GetText() << endl;
  
  TString fileName = AliOADBPWG2Spectra::GetOADBPWG2SpectraFileName();
  TFile * f = new TFile (fileName);
  fOADBContainer = (AliOADBContainer*) f->Get(fkContainers[fComboContainer->GetSelected()]);
  if(!fOADBContainer) return;
  f->Close();
  fOADBSpectra = (AliOADBPWG2Spectra*) fOADBContainer->GetObject(atoi(fRunNumber->GetText() ));
  if(!fOADBSpectra) fOADBContainer->List();
  fOADBSpectra->Print();
}

void MyMainFrame::DoLegend()
{
  // legend
  gInterpreter->ProcessLine("NewLegend(\"\", \"lpf\",0,1,0);");
  TCanvas *c1 = fEcan->GetCanvas();
  c1->Update();
  c1->Modified();
  c1->Update(); 
}
void MyMainFrame::DoRatio() {
  // Divide 2 histos on canvas
  gInterpreter->ProcessLine("Divide2HistosOnCanvas();");
}


void MyMainFrame::DoDraw()
{
   // Draw something in the canvas

   Printf("Slot DoDraw()");
   cout << "DET " << fComboDetector->GetSelected() << endl;
   if(!fOADBSpectra) {
     cout << "spectra not loaded" << endl;
     DoLoad();
     if(!fOADBSpectra)      {
       cout << "ERROR: Cannot load spectra" << endl;       
       return;
     }
   }

   TCanvas *c1 = fEcan->GetCanvas();
   c1->cd();
   TH1D * h = 0;
   if(strcmp(fkCentrTags[fComboCentr->GetSelected()],"")){
       h = fOADBSpectra->GetHisto(fComboDetector->GetSelected(), 
				  fComboPID->GetSelected(), 
				  fComboParticle->GetSelected(), 
				  fComboCharge->GetSelected(), 
				  fkCentrTags[fComboCentr->GetSelected()],
				  atoi(fTextCentrBin->GetText()));

     }
     else {
       h = fOADBSpectra->GetHisto(fComboDetector->GetSelected(), 
				  fComboPID->GetSelected(), 
				  fComboParticle->GetSelected(), 
				  fComboCharge->GetSelected());

     }
   // Draw the selected histogram

   if(!h) {
     cout << "Cannot get pointer to histo" << endl;
     
     cout << fkCentrTags[fComboCentr->GetSelected()] << " " << 
   
     fOADBSpectra->GetHistoName(fComboDetector->GetSelected(), 
				fComboPID->GetSelected(), 
				fComboParticle->GetSelected(), 
				fComboCharge->GetSelected(), 
				fkCentrTags[fComboCentr->GetSelected()],
				atoi(fTextCentrBin->GetText()))
	<< endl;

     return;
   }
   TString opt = fTextDrawOpt->GetText();
   if(opt=="auto") {
     c1->GetListOfPrimitives()->Print();
     if (c1->GetListOfPrimitives()->GetEntries()>0) opt = "same";
     else opt = "";
   }
   
   h->SetXTitle("p_{T} (GeV/c)");
   //   h->SetXTitle("dN/dp_{T}");
   h->Draw(opt);
   
   // TCanvas::Update() draws the frame, after which it can be changed
   c1->Update();
   c1->Modified();
   c1->Update();
}

void MyMainFrame::DoExit()
{
   printf("Exit application...");
   gApplication->Terminate(0);
}

void MyMainFrame::SetStatusText(const char *txt, Int_t pi)
{
   // Set text in status bar.
   fStatusBar->SetText(txt,pi);
}

void MyMainFrame::EventInfo(Int_t event, Int_t px, Int_t py, TObject *selected)
{
//  Writes the event status in the status bar parts

   const char *text0, *text1, *text3;
   char text2[50];
   text0 = selected->GetTitle();
   SetStatusText(text0,0);
   text1 = selected->GetName();
   SetStatusText(text1,1);
   if (event == kKeyPress)
      sprintf(text2, "%c", (char) px);
   else
      sprintf(text2, "%d,%d", px, py);
   SetStatusText(text2,2);
   text3 = selected->GetObjectInfo(px,py);
   SetStatusText(text3,3);
}

MyMainFrame::MyMainFrame(const TGWindow *p, UInt_t w, UInt_t h) :
  TGMainFrame(p, w, h), fOADBContainer(0), fOADBSpectra(0)
{
  // Create a horizontal frame (canvas and status bar on the left, buttons on the right)
   TGHorizontalFrame *hframeMain = new TGHorizontalFrame(this, 200, 40);

   // Create a vertical frame for the canvas and for the status bar
   TGVerticalFrame *vframeCanvas = new TGVerticalFrame(this, 200, 40);

   // Create the embedded canvas
   fEcan = new TRootEmbeddedCanvas(0,this,500,400);
   Int_t wid = fEcan->GetCanvasWindowId();
   TCanvas *myc = new TCanvas("MyCanvas", 10,10,wid);
   cout << myc << endl;
   
   fEcan->AdoptCanvas(myc);
   myc->Connect("ProcessedEvent(Int_t,Int_t,Int_t,TObject*)","MyMainFrame",this, 
               "EventInfo(Int_t,Int_t,Int_t,TObject*)");

   vframeCanvas->AddFrame(fEcan, new TGLayoutHints(kLHintsTop | kLHintsLeft | kLHintsExpandX  | kLHintsExpandY,0,0,1,1));
   // status bar
   Int_t parts[] = {45, 15, 10, 30};
   fStatusBar = new TGStatusBar(this, 50, 10, kVerticalFrame);
   fStatusBar->SetParts(parts, 4);
   fStatusBar->Draw3DCorner(kFALSE);
   vframeCanvas->AddFrame(fStatusBar, new TGLayoutHints(kLHintsExpandX, 0, 0, 10, 0));
   
   hframeMain->AddFrame(vframeCanvas, new TGLayoutHints(kLHintsTop | kLHintsLeft | kLHintsExpandX  | kLHintsExpandY, 2, 2, 2, 2));

   // Create a vertical frame containing  buttons and controls
   TGVerticalFrame *vframeButtons = new TGVerticalFrame(this, 200, 40);
  
   // Container
   fComboContainer = new TGComboBox(vframeButtons);
   vframeButtons->AddFrame(fComboContainer, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));
   for(Int_t icont = 0; icont < fkNContainers; icont++){
     fComboContainer->AddEntry(fkContainers[icont],icont);
   }
   fComboContainer->Select(0);
   fComboContainer->Resize(120,20);
   fRunNumber = new TGTextEntry (vframeButtons, "116562");
   vframeButtons->AddFrame(fRunNumber, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));
   

   TGTextButton *load = new TGTextButton(vframeButtons, "&Load");
   load->Connect("Clicked()", "MyMainFrame", this, "DoLoad()");
   vframeButtons->AddFrame(load, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));

   // Combos
   fComboDetector = new TGComboBox(vframeButtons);
   vframeButtons->AddFrame(fComboDetector, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));   
   for(Int_t idet = 0; idet < AliOADBPWG2Spectra::kNDetectors; idet++){
     fComboDetector->AddEntry(AliOADBPWG2Spectra::GetDetectorName(idet),idet);
   }
   fComboDetector->Resize(120,20);
   fComboDetector->Select(1);
   fComboDetector->Connect("Selected(Int_t)", "MyMainFrame", this, "DoSelectedDetector(Int_t)");

   fComboPID = new TGComboBox(vframeButtons);
   vframeButtons->AddFrame(fComboPID, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));   
   for(Int_t idet = 0; idet < AliOADBPWG2Spectra::kNPIDTypes; idet++){
     fComboPID->AddEntry(AliOADBPWG2Spectra::GetPIDName(idet),idet);
   }
   fComboPID->Resize(120,20);
   fComboPID->Select(0);

   fComboCharge = new TGComboBox(vframeButtons);
   vframeButtons->AddFrame(fComboCharge, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));   
   for(Int_t idet = 0; idet < AliOADBPWG2Spectra::kNCharge; idet++){
     fComboCharge->AddEntry(AliOADBPWG2Spectra::GetChargeName(idet),idet);
   }
   fComboCharge->Resize(120,20);
   fComboCharge->Select(0);

   fComboParticle = new TGComboBox(vframeButtons);
   vframeButtons->AddFrame(fComboParticle, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));   
   for(Int_t idet = 0; idet < AliOADBPWG2Spectra::kNParticle; idet++){
     fComboParticle->AddEntry(AliOADBPWG2Spectra::GetParticleName(idet),idet);
   }
   fComboParticle->Resize(120,20);
   fComboParticle->Select(0);

   fComboCentr = new TGComboBox(vframeButtons);
   vframeButtons->AddFrame(fComboCentr, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));   
   for(Int_t icentr = 0; icentr < fkNCentrTags; icentr++){
     fComboCentr->AddEntry(fkCentrTags[icentr],icentr);
   }
   fComboCentr->Resize(120,20);
   fComboCentr->Select(0);

   // Text fields
   fTextCentrBin = new TGTextEntry (vframeButtons, "1");
   fTextCentrBin->Resize(120,20);
   vframeButtons->AddFrame(fTextCentrBin, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));   
   fTextDrawOpt = new TGTextEntry (vframeButtons, "auto");
   vframeButtons->AddFrame(fTextDrawOpt, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));   
   fTextDrawOpt->Resize(120,20);
   fTextSaveName = new TGTextEntry (vframeButtons, "spectra.png");
   vframeButtons->AddFrame(fTextSaveName, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));   
   fTextSaveName->Resize(120,20);

   // buttons
   TGTextButton *draw = new TGTextButton(vframeButtons, "&Draw");
   draw->Connect("Clicked()", "MyMainFrame", this, "DoDraw()");
   vframeButtons->AddFrame(draw, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));
   TGTextButton *legend = new TGTextButton(vframeButtons, "&Legend");
   legend->Connect("Clicked()", "MyMainFrame", this, "DoLegend()");
   vframeButtons->AddFrame(legend, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));
   TGTextButton *ratio = new TGTextButton(vframeButtons, "&Ratio");
   ratio->Connect("Clicked()", "MyMainFrame", this, "DoRatio()");
   vframeButtons->AddFrame(ratio, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));
   TGTextButton *clear = new TGTextButton(vframeButtons, "&Clear");
   clear->Connect("Clicked()", "MyMainFrame", this, "DoClear()");
   vframeButtons->AddFrame(clear, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));
   TGTextButton *print = new TGTextButton(vframeButtons, "&Print");
   print->Connect("Clicked()", "MyMainFrame", this, "DoPrint()");
   vframeButtons->AddFrame(print, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));
   TGTextButton *fit = new TGTextButton(vframeButtons, "&Fit");
   fit->Connect("Clicked()", "MyMainFrame", this, "DoFit()");
   vframeButtons->AddFrame(fit, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));
   TGTextButton *exit = new TGTextButton(vframeButtons, "&Exit ");
   exit->Connect("Pressed()", "MyMainFrame", this, "DoExit()");
   vframeButtons->AddFrame(exit, new TGLayoutHints(kLHintsCenterX, 5, 5, 3, 4));




   hframeMain->AddFrame(vframeButtons, new TGLayoutHints(kLHintsRight, 2, 2, 2, 2));

   AddFrame(hframeMain,new TGLayoutHints(kLHintsTop | kLHintsLeft | kLHintsExpandX  | kLHintsExpandY, 2, 2, 2, 2));

   // Set a name to the main frame   
   SetWindowName("Spectra OADB Browser");
   MapSubwindows();

   // Initialize the layout algorithm via Resize()
   Resize(GetDefaultSize());

   // Map main frame
   MapWindow();

   // Misc init
   fMass[AliOADBPWG2Spectra::kPion]   = TDatabasePDG::Instance()->GetParticle("pi+")->Mass();
   fMass[AliOADBPWG2Spectra::kKaon]   = TDatabasePDG::Instance()->GetParticle("K+")->Mass();
   fMass[AliOADBPWG2Spectra::kProton] = TDatabasePDG::Instance()->GetParticle("proton")->Mass();

   fFuncMan = new AliBWFunc;
   fFuncMan->SetVarType(AliBWFunc::kdNdpt);

}


MyMainFrame::~MyMainFrame()
{
   // Clean up main frame...
   Cleanup();
   delete fEcan;
}


void spectraOADBGUI()
{
  // Popup the GUI...
  
  new MyMainFrame(gClient->GetRoot(), 200, 200);
}