#include <iostream>
#include <sstream>
#include <iomanip>
#include "TCanvas.h"
#include "TH1F.h"
#include "AtlasUtils.h"
#include "AtlasStyle.C"
#include "AtlasLabels.C"


#define nrT 100

void readUnfoldingBreakdown(std::string hname, bool saving=false)
{



    if (hname=="XXX"){
    std::cout<<"No histogram selected - possible choices are: \n";
    std::cout<<"\"j_bjet_Pt_binned\""<<std::endl;
    std::cout<<"\"j_bjet_Y_binned\""<<std::endl;
    std::cout<<"\"j_bjet_ZPt_binned\""<<std::endl;
    std::cout<<"\"j_bjet_ZY_binned\""<<std::endl;
    std::cout<<"\"j_bjet_ZYf_binned\""<<std::endl;
    std::cout<<"\"j_bjet_ZdR620_binned\""<<std::endl;
    std::cout<<"\"j_bjet_ZdPhiF20_binned\""<<std::endl;
    std::cout<<"\"j_2bjet_ZPt_binned\""<<std::endl;
    std::cout<<"\"j_2bjet_dR_binned\""<<std::endl; 
    std::cout<<"\"j_2bjet_Mbb_binned\""<<std::endl;

    return;
  }

     
  gROOT->LoadMacro("AtlasUtils.C");
  SetAtlasStyle(); 

  gStyle->SetOptStat(0);
  gStyle->SetOptTitle(0);



  std::string filen="../data/UnfoldingBreakdownNumbers.txt";
  
  
  Double_t MR[nrT], NR[nrT], MmatchR[nrT], NmatchR[nrT], Mfake[nrT], Nfake[nrT], MT[nrT], NT[nrT], MmatchT[nrT], NmatchT[nrT], Meff[nrT], Neff[nrT];

  int m_debug;
    //Int_t nr = 0;
    if (m_debug>1) std::cout<<" file to be read = <"<<filen<<">"<<std::endl;
    
    int nr=0;
    std::string STRING;
    stringstream ss;
    ifstream infile(filen.c_str());
    //Double_t* X1,X2,X3,X4;
    if (infile.is_open())                     //if the file is open
    {
	while(!infile.eof()) // To get you all the lines.
	{
	    ss.clear();
	    getline(infile,STRING); // Saves the line in STRING.
	    if (STRING.substr(0,1)=="*" || STRING.size()==0) {
	      if (STRING.substr(0,1)=="*") std::cout<<" ----------------OFFSET "<<nr<<std::endl;  
	      continue; // skip comments
	    }
	    if (m_debug==0) std::cout<<" now read --- "<<STRING<<std::endl; // Prints our STRING.
	    ss << STRING;
	    //	    std::cout<<"size of line read ="<<STRING.size()<<std::endl;
	    // 
	    Double_t ax1,ax2,ax3,ax4,ax5,ax6,ax7,ax8,ax9,ax10,ax11,ax12;
	    ss>>ax1>>ax2>>ax3>>ax4>>ax5>>ax6>>ax7>>ax8>>ax9>>ax10>>ax11>>ax12;
	    //if (m_debug==0) std::cout<<"==== at index "<<nr<<" aX1-7 are "<<ax1<<" "<<ax2<<" "<<ax3<<" "<<ax4<<" "<<ax5<<" "<<ax6<<" "<<ax7<<std::endl;	
	    //Double_t X0[nr], X0stat[nr], Double_t JERup[nr], Double_t JERdw[nr], Double_t JES[nr], Double_t bTAGup[nr], Double_t bTAGdw[nr];
	    MR[nr]=ax1;
	    NR[nr]=ax2;
	    MmatchR[nr]=ax3;
	    NmatchR[nr]=ax4;
	    Mfake[nr]=ax5;
	    Nfake[nr]=ax6;
	    MT[nr]=ax7;
	    NT[nr]=ax8;
	    MmatchT[nr]=ax9;
	    NmatchT[nr]=ax10;
	    Meff[nr]=ax11;
	    Neff[nr]=ax12;
	    //std::cout<<nr<<" in order: "<<MR[nr]<<" "<<NR[nr]<<" "<<MmatchR[nr]<<" "<<NmatchR[nr]<<" "<<Mfake[nr]<<" "<<Nfake[nr]<<std::endl;
	    ++nr;
	    if (nr==100) break;
	}
    }
    else 
    {
	std::cout<<"Cannot open file "<<std::endl;
	return;	
    }
    
    infile.close();

    std::cout<<" number of lines "<<std::endl;

    TFile* newFile ;
    if (saving) newFile  = new TFile("UnfoldingChecks.root","recreate");

    

    TH1F *h_Data0MR       ;
    TH1F *h_Data0NR      ;
    TH1F *h_Data0MmatchR      ;
    TH1F *h_Data0NmatchR        ;
    TH1F *h_Data0Mfake     ;
    TH1F *h_Data0Nfake     ;
    TH1F *h_Data0MT       ;
    TH1F *h_Data0NT      ;
    TH1F *h_Data0MmatchT      ;
    TH1F *h_Data0NmatchT        ;
    TH1F *h_Data0Meff     ;
    TH1F *h_Data0Neff     ;

    
    if (hname=="j_bjet_Pt_binned")
      {
      const int myn = 7 ;
      Double_t mynewBinsPlot   [myn] = {20,30,50,75,110,200,500};
      int hoff = 0;
      }
    if (hname=="j_bjet_Y_binned")
      {
      int hoff = 6;
      const int myn = 9 ;
      Double_t mynewBinsPlot   [myn] = {0,0.2,0.4,0.6,0.8,1.2,1.6,2.0,2.4};
      }
    if (hname=="j_bjet_ZPt_binned")
      {
      int hoff =31;
      const int myn = 9 ;
      Double_t mynewBinsPlot   [myn] = {1,20,30,40,60,80,110,200,500};
      }
    if (hname=="j_bjet_ZY_binned")
      {
      int hoff = 39;
      const int myn = 9 ;
      Double_t mynewBinsPlot   [myn] = {0.0,0.2,0.4,0.6,0.8,1.2,1.6,2.0,2.5};
      }
    if (hname=="j_bjet_ZdPhiF20_binned")
      {
      int hoff =14;
      const int myn = 9 ;
      Double_t mynewBinsPlot   [myn] = {0,0.5,1,1.5,2.0,2.4,2.8,3,TMath::Pi()};
      }
    if (hname=="j_bjet_ZdR620_binned")
      {
      int hoff =22;
      const int myn = 10 ;
      Double_t mynewBinsPlot   [myn] = {0,1,1.5,2,2.5,3,3.5,4,4.5,6};   
      }	

    h_Data0MR            = new TH1F(("h_Data0MR_"+hname).c_str()     ,""   ,myn-1,mynewBinsPlot);
    h_Data0NR            = new TH1F(("h_Data0NR_"+hname).c_str()     ,""   ,myn-1,mynewBinsPlot);
    h_Data0MmatchR       = new TH1F(("h_Data0MmatchR_"+hname).c_str(),""   ,myn-1,mynewBinsPlot);
    h_Data0NmatchR       = new TH1F(("h_Data0NmatchR_"+hname).c_str(),""   ,myn-1,mynewBinsPlot);
    h_Data0Mfake         = new TH1F(("h_Data0Mfake_"+hname).c_str()  ,""   ,myn-1,mynewBinsPlot);
    h_Data0Nfake         = new TH1F(("h_Data0Nfake_"+hname).c_str()  ,""   ,myn-1,mynewBinsPlot);
    h_Data0MT            = new TH1F(("h_Data0MR_"+hname).c_str()     ,""   ,myn-1,mynewBinsPlot);
    h_Data0NT            = new TH1F(("h_Data0NR_"+hname).c_str()     ,""   ,myn-1,mynewBinsPlot);
    h_Data0MmatchT       = new TH1F(("h_Data0MmatchR_"+hname).c_str(),""   ,myn-1,mynewBinsPlot);
    h_Data0NmatchT       = new TH1F(("h_Data0NmatchR_"+hname).c_str(),""   ,myn-1,mynewBinsPlot);
    h_Data0Meff          = new TH1F(("h_Data0Meff_"+hname).c_str()   ,""   ,myn-1,mynewBinsPlot);
    h_Data0Neff          = new TH1F(("h_Data0Neff_"+hname).c_str()   ,""   ,myn-1,mynewBinsPlot);
    for (int j=1; j<myn; ++j) 
      {
	int index = hoff+j-1;
	h_Data0MR     ->SetBinContent(j, MR      [index]);
	h_Data0NR     ->SetBinContent(j, NR      [index]);
	h_Data0MmatchR->SetBinContent(j, MmatchR [index]);
	h_Data0NmatchR->SetBinContent(j, NmatchR [index]);
	h_Data0Mfake  ->SetBinContent(j, Mfake   [index]);
	h_Data0Nfake  ->SetBinContent(j, Nfake   [index]);
	h_Data0MT     ->SetBinContent(j, MT      [index]);
	h_Data0NT     ->SetBinContent(j, NT      [index]);
	h_Data0MmatchT->SetBinContent(j, MmatchT [index]);
	h_Data0NmatchT->SetBinContent(j, NmatchT [index]);
	h_Data0Meff   ->SetBinContent(j, Meff    [index]);
	h_Data0Neff   ->SetBinContent(j, Neff    [index]);
	// 
	std::cout<<"at index "<<index<<" Reco M and N "<<MR      [hoff+j-1]<<" "<<NR      [hoff+j-1]<<"         matched to Reco M N "<<MmatchR [hoff+j-1]<<" "<<NmatchR [hoff+j-1]<<std::endl;
      }


 
  TCanvas* c0 = new TCanvas("Reco","reco");
  //h_Data0MR->GetYaxis()->SetRangeUser(0.5,1.5);
  h_Data0NR->GetXaxis()->SetTitle(hname.c_str());
  h_Data0NR->Draw("EP");
  h_Data0MR->Draw("SAME");
  h_Data0MmatchR->SetLineColor(2);
  h_Data0MmatchR->SetLineStyle(2);
  h_Data0NmatchR->SetMarkerColor(2);
  h_Data0NmatchR->SetMarkerStyle(24);
  h_Data0NmatchR->SetMarkerSize(2.);
  h_Data0MmatchR->Draw("SAME");
  h_Data0NmatchR->Draw("SAMEP");
  TLegend* myLegend7 = new TLegend(0.7,0.7,0.95,0.95);
  myLegend7->AddEntry(h_Data0NR, "Reco N","P");
  myLegend7->AddEntry(h_Data0MR, "Reco M","L");
  myLegend7->AddEntry(h_Data0NmatchR, "matched-Reco N","P");
  myLegend7->AddEntry(h_Data0MmatchR, "matched-Reco M","L");
  myLegend7->SetShadowColor(1);
  myLegend7->SetBorderSize(0);
  myLegend7->SetFillStyle(0);
  myLegend7->Draw();
  TCanvas* c1 = new TCanvas("Fakes","fakes");
  h_Data0Nfake->GetXaxis()->SetTitle(hname.c_str());
  h_Data0Mfake->Draw();
  h_Data0Nfake->Draw("SAMEP");
  TLegend* myLegend8 = new TLegend(0.7,0.7,0.95,0.95);
  myLegend8->AddEntry(h_Data0Nfake, "Fakes N","P");
  myLegend8->AddEntry(h_Data0Mfake, "Fakes M","L");
  myLegend8->SetShadowColor(1);
  myLegend8->SetBorderSize(0);
  myLegend8->SetFillStyle(0);
  myLegend8->Draw();


  TCanvas* c2 = new TCanvas("Truth","truth");
  //h_Data0MR->GetYaxis()->SetRangeUser(0.5,1.5);
  h_Data0NT->GetXaxis()->SetTitle(hname.c_str());
  h_Data0NT->Draw("EP");
  h_Data0MT->Draw("SAME");
  h_Data0MmatchT->SetLineColor(2);
  h_Data0MmatchT->SetLineStyle(2);
  h_Data0NmatchT->SetMarkerStyle(24);
  h_Data0NmatchT->SetMarkerSize(2.);
  h_Data0NmatchT->SetMarkerColor(2);
  h_Data0MmatchT->Draw("SAME");
  h_Data0NmatchT->Draw("SAMEP");
  TLegend* myLegend6 = new TLegend(0.7,0.7,0.95,0.95);
  myLegend6->AddEntry(h_Data0NT, "Truth N","P");
  myLegend6->AddEntry(h_Data0MT, "Truth M","L");
  myLegend6->AddEntry(h_Data0NmatchT, "matched-Truth N","P");
  myLegend6->AddEntry(h_Data0MmatchT, "matched-Truth M","L");
  myLegend6->SetShadowColor(1);
  myLegend6->SetBorderSize(0);
  myLegend6->SetFillStyle(0);
  myLegend6->Draw();
  TCanvas* c3 = new TCanvas("Efficiency","eff");
  h_Data0Neff->GetXaxis()->SetTitle(hname.c_str());
  h_Data0Neff->Draw();
  h_Data0Meff->Draw("SAMEP");
  TLegend* myLegend5 = new TLegend(0.7,0.7,0.95,0.95);
  myLegend5->AddEntry(h_Data0Neff, "Efficiency N","P");
  myLegend5->AddEntry(h_Data0Meff, "Efficiency M","L");
  myLegend5->SetShadowColor(1);
  myLegend5->SetBorderSize(0);
  myLegend5->SetFillStyle(0);
  myLegend5->Draw();


  TH1F *Rratio = h_Data0NR->Clone(("hratioR_"+hname).c_str());
  Rratio->Divide(h_Data0MR);
  TH1F *RMratio = h_Data0NmatchR->Clone(("hratioRmatch_"+hname).c_str());
  RMratio->Divide(h_Data0MmatchR);
  TH1F *Tratio = h_Data0NT->Clone(("hratioT_"+hname).c_str());
  Tratio->Divide(h_Data0MT);
  TH1F *TMratio = h_Data0NmatchT->Clone(("hratioTmatch_"+hname).c_str());
  TMratio->Divide(h_Data0MmatchT);
  TH1F *Effratio = h_Data0Neff->Clone(("hratioRff_"+hname).c_str());
  Effratio->Divide(h_Data0Meff);
  TH1F *Fakesratio = h_Data0Nfake->Clone(("hratioFakes_"+hname).c_str());
  Fakesratio->Divide(h_Data0Mfake);


  TCanvas* cr0 = new TCanvas("RatioReco","RatioReco");
  Rratio->GetXaxis()->SetTitle(hname.c_str());
  Rratio->SetMinimum(0.7);
  Rratio->SetMaximum(1.3);
  Rratio->Draw("P");
  RMratio->SetMarkerStyle(23);
  RMratio->Draw("PSAME");
  TLegend* myLegend4 = new TLegend(0.7,0.7,0.95,0.95);
  myLegend4->AddEntry(Rratio, "Reco N/M","PL");
  myLegend4->AddEntry(RMratio,"matched-Reco N/M","PL");
  myLegend4->SetShadowColor(1);
  myLegend4->SetBorderSize(0);
  myLegend4->SetFillStyle(0);
  myLegend4->Draw();
  TCanvas* cr1 = new TCanvas("RatioTruth","RatioTruth");
  Tratio->GetXaxis()->SetTitle(hname.c_str());
  Tratio->SetMinimum(0.7);
  Tratio->SetMaximum(1.3);
  Tratio->Draw("P");
  TMratio->SetMarkerStyle(23);
  TMratio->Draw("PSAME");
  TLegend* myLegend3 = new TLegend(0.7,0.7,0.95,0.95);
  myLegend3->AddEntry(Tratio, "Truth N/M","PL");
  myLegend3->AddEntry(TMratio,"matched-Truth N/M","PL");
  myLegend3->SetShadowColor(1);
  myLegend3->SetBorderSize(0);
  myLegend3->SetFillStyle(0);
  myLegend3->Draw();
  TCanvas* cr2 = new TCanvas("RatioEff","RatioEff");
  Effratio->GetXaxis()->SetTitle(hname.c_str());
  Effratio->SetMinimum(0.7);
  Effratio->SetMaximum(1.3);
  Effratio->Draw("P");
  TLegend* myLegend2 = new TLegend(0.7,0.7,0.95,0.95);
  myLegend2->AddEntry(Effratio,"Efficiency N/M","P");
  myLegend2->SetShadowColor(1);
  myLegend2->SetBorderSize(0);
  myLegend2->SetFillStyle(0);
  myLegend2->Draw();
  TCanvas* cr3 = new TCanvas("RatioFakes","RatioFakes");
  Fakesratio->GetXaxis()->SetTitle(hname.c_str());
  Fakesratio->SetMinimum(0.7);
  Fakesratio->SetMaximum(1.3);
  Fakesratio->Draw("P");
  TLegend* myLegend1 = new TLegend(0.7,0.7,0.95,0.95);
  myLegend1->AddEntry(Fakesratio,"Fake rate N/M","P");
  myLegend1->SetShadowColor(1);
  myLegend1->SetBorderSize(0);
  myLegend1->SetFillStyle(0);
  myLegend1->Draw();
  TCanvas* cr4 = new TCanvas("Ratios","Ratios");
  Fakesratio->SetMinimum(0.7);
  Fakesratio->SetMaximum(1.3);
  Effratio->SetMarkerColor(2);
  Rratio->SetMarkerColor(4);
  Tratio->SetMarkerColor(3);
  Fakesratio->Draw("CP");
  Effratio  ->Draw("CPSAME");
  Rratio    ->Draw("CPSAME");
  Tratio    ->Draw("CPSAME");
  TLegend* myLegend = new TLegend(0.7,0.7,0.95,0.95);
  myLegend->AddEntry(Fakesratio,"Fake rate","P");
  myLegend->AddEntry(Effratio,  "Efficiency","P");
  myLegend->AddEntry(Rratio,    "Reco yield","P");
  myLegend->AddEntry(Tratio,    "Truth yield","P");
  myLegend->SetShadowColor(1);
  myLegend->SetBorderSize(0);
  myLegend->SetFillStyle(0);
  myLegend->Draw();
  

  if (saving){
    newFile->Write();
    newFile->Close();
  }


  c0->SaveAs(("temporary/"+std::string(c0->GetName())+hname+".eps").c_str());
  c1->SaveAs(("temporary/"+std::string(c1->GetName())+hname+".eps").c_str());
  c2->SaveAs(("temporary/"+std::string(c2->GetName())+hname+".eps").c_str());
  c3->SaveAs(("temporary/"+std::string(c3->GetName())+hname+".eps").c_str());
  cr0->SaveAs(("temporary/"+std::string(cr0->GetName())+hname+".eps").c_str());
  cr1->SaveAs(("temporary/"+std::string(cr1->GetName())+hname+".eps").c_str());
  cr2->SaveAs(("temporary/"+std::string(cr2->GetName())+hname+".eps").c_str());
  cr3->SaveAs(("temporary/"+std::string(cr3->GetName())+hname+".eps").c_str());
  cr4->SaveAs(("temporary/"+std::string(cr4->GetName())+hname+".eps").c_str());


    return;
}