#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 readUnfoldingBreakdown2(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_Yboost_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_bjet_DY20_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/UnfoldingBreakdownNumbers2.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];
  Double_t MT_NT[nrT], MTm_NTm[nrT], MRm_NRm[nrT], MR_NR[nrT], Mbbb[nrT], Nbbb[nrT], Mbbb_Nbbb[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;
	    ss>>ax1>>ax2>>ax3>>ax4>>ax5>>ax6>>ax7;
	    //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];
	    MT_NT[nr]=ax1;
	    MTm_NTm[nr]=ax2;
	    MRm_NRm[nr]=ax3;
	    MR_NR[nr]=ax4;
	    Mbbb[nr]=ax5;
	    Nbbb[nr]=ax6;
	    Mbbb_Nbbb[nr]=ax7; 
	    //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("UnfoldingChecks2.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_Yboost_binned")
      {
      int hoff = 14;
      const int myn = 9 ;
      Double_t mynewBinsPlot   [myn] = {0,0.2,0.4,0.6,0.8,1.2,1.6,2.0,2.5};
      }
    if (hname=="j_bjet_ZPt_binned")
      {
      int hoff =48;
      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 = 56;
      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 =22;
      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 =30;
      const int myn = 10 ;
      Double_t mynewBinsPlot   [myn] = {0,1,1.5,2,2.5,3,3.5,4,4.5,6};   
      }	
    if (hname=="j_bjet_DY20_binned")
      {
      int hoff =39;
      const int myn = 10 ;
      Double_t mynewBinsPlot   [myn] = {0,0.2,0.4,0.6,0.8,1.2,1.6,2.0,3.,5.};   
      }	

    TH1F* h_MT_NT     	= new TH1F(("h_MT_NT"    +hname).c_str()     ,""   ,myn-1,mynewBinsPlot);
    TH1F* h_MTm_NTm     = new TH1F(("h_MTm_NTm"  +hname).c_str()     ,""   ,myn-1,mynewBinsPlot);
    TH1F* h_MRm_NRm     = new TH1F(("h_MRm_NRm"  +hname).c_str()     ,""   ,myn-1,mynewBinsPlot);
    TH1F* h_MR_NR       = new TH1F(("h_MR_NR"    +hname).c_str()     ,""   ,myn-1,mynewBinsPlot);
    TH1F* h_Mbbb        = new TH1F(("h_Mbbb"     +hname).c_str()     ,""   ,myn-1,mynewBinsPlot);
    TH1F* h_Nbbb        = new TH1F(("h_Nbbb"     +hname).c_str()     ,""   ,myn-1,mynewBinsPlot);
    TH1F* h_Mbbb_Nbbb   = new TH1F(("h_Mbbb_Nbbb"+hname).c_str()     ,""   ,myn-1,mynewBinsPlot); 

    for (int j=1; j<myn; ++j) 
      {
	int index = hoff+j-1;

	h_MT_NT     ->SetBinContent(j, MT_NT    [index]);
	h_MTm_NTm   ->SetBinContent(j, MTm_NTm  [index]);
	h_MRm_NRm   ->SetBinContent(j, MRm_NRm  [index]);
	h_MR_NR     ->SetBinContent(j, MR_NR    [index]);
	h_Mbbb      ->SetBinContent(j, Mbbb     [index]);
	h_Nbbb      ->SetBinContent(j, Nbbb     [index]);
	h_Mbbb_Nbbb ->SetBinContent(j, Mbbb_Nbbb[index]);
	// 
	std::cout<<"at index "<<index<<" Reco MT_NT, MTm_NTm, MRm_NRm, MR_NR "
		 <<MT_NT[hoff+j-1]<<" "<<MTm_NTm[hoff+j-1]<<" "<<MRm_NRm[hoff+j-1]<<" "<<MR_NR[hoff+j-1]<<" "
		 <<" Mbbb, Nbbb, Mbbb_Nbbb "<<Mbbb[hoff+j-1]<<" "<<Nbbb[hoff+j-1]<<" "<<Mbbb_Nbbb[hoff+j-1]<<std::endl;
      }


 
  TCanvas* c0 = new TCanvas("Truth and Reco","Truth and Reco");
  //h_Data0MR->GetYaxis()->SetRangeUser(0.5,1.5);
  h_MT_NT->GetXaxis()->SetTitle(hname.c_str());
  h_MT_NT->GetYaxis()->SetRangeUser(0.98,1.02);
  
  h_MT_NT->SetLineColor(1);
  h_MT_NT->SetMarkerColor(1);
  h_MT_NT->SetMarkerStyle(20);
  h_MT_NT->Draw("CP");
  h_MTm_NTm->SetLineColor(4);
  h_MTm_NTm->SetMarkerColor(4);
  h_MTm_NTm->SetMarkerStyle(20);
  h_MTm_NTm->Draw("CPSAME");
  h_MRm_NRm->SetLineColor(2);
  h_MRm_NRm->SetMarkerColor(2);
  h_MRm_NRm->SetMarkerStyle(23);
  h_MRm_NRm->Draw("CPSAME");
  h_MR_NR->SetLineColor(6);
  h_MR_NR->SetMarkerColor(6);
  h_MR_NR->SetMarkerStyle(23);
  h_MR_NR->Draw("CPSAME");

  TLegend* myLegend = new TLegend(0.6,0.7,0.95,0.90);
  myLegend->SetShadowColor(1);
  myLegend->SetBorderSize(0);
  myLegend->SetFillStyle(0);
  myLegend->SetTextSize(0.03);  
  myLegend->AddEntry(h_MT_NT,        "Truth - an.1/an.2","EPL");
  myLegend->AddEntry(h_MTm_NTm,      "Matched Truth - an.1/an.2","EPL");
  myLegend->AddEntry(h_MRm_NRm,      "Matched Reco  - an.1/an.2","EPL");
  myLegend->AddEntry(h_MR_NR,        "Reco - an.1/an.2","EPL");
  myLegend->Draw();

  c0->SaveAs(std::string("temporary/UnfBreak_"+hname+".eps").c_str());

    return;
}