#include <stdlib.h>
#include <string>
#include <iostream>
#include <cstring>
#include <sstream>
#include <vector>

#include "TFile.h"
#include "TCut.h"
#include "TLine.h"
#include "TFitResultPtr.h"
#include "TTree.h"
#include "TStyle.h"
#include "TCanvas.h"
#include "TH1.h"
#include "TGraph.h"
#include "THStack.h"
#include "TAttLine.h"
#include "TH1.h"
#include "TH2.h"
#include "TF1.h"
#include "TProfile.h"
#include "TChain.h"
#include "TMath.h"
#include "TLegend.h"
#include "TROOT.h"
#include "TStyle.h"
#include "TPaveStats.h"
#include "TPave.h"
#include <iostream>
#include <fstream>

TH1F** initializeEpsilonHistograms(const char *name, const char *title, int size );
void  deleteEpsilonPlot(TH1F **h, int size);

void FillMassPlot(TString File_path ){

    using namespace std;
    cout<<"I start with the TChain."<<endl;
    TFile outfile("/tmp/ALL_2012B_epsilonPlots.root","RECREATE");
    outfile.cd();
    TChain chEB("Pi0Info_EB","EB");
    TChain chEE("Pi0Info_EE","EE");
    TString path = "root://eoscms//eos/cms/store/group/alca_ecalcalib/lpernie/ALL_2012B_TTree_01/iter_0/2012B_EcalNtp_*.root";
    chEB.Add(path.Data());
    chEE.Add(path.Data());

    Long64_t nentries_EB = chEB.GetEntries();
    Long64_t nentries_EE = chEE.GetEntries();
    cout<<"Running on "<<nentries_EB << " + "<<nentries_EE<<" entries"<<endl;

    Float_t mpi0_EB, weight_EB;
    UShort_t index_EB; 
    Float_t mpi0_EE, weight_EE;
    UShort_t index_EE;
    chEB.SetBranchAddress( "mpi0_EB", &mpi0_EB);
    chEB.SetBranchAddress( "weight_EB", &weight_EB);
    chEB.SetBranchAddress( "index_EB", &index_EB);
    chEE.SetBranchAddress( "mpi0_EE", &mpi0_EE);
    chEE.SetBranchAddress( "weight_EE", &weight_EE);
    chEE.SetBranchAddress( "index_EE", &index_EE);

    //Init Histos
    int N_EB(61200), N_EE(14648);
    cout<<"Now I inizialize the histos..."<<endl;

    TH1F** epsilon_EB_h = initializeEpsilonHistograms("epsilon_EB_iR_","#pi^{0} Mass distribution EB - iR ", N_EB );
    TH1F** epsilon_EE_h = initializeEpsilonHistograms("epsilon_EE_iR_","#pi^{0} Mass distribution EE - iR ", N_EE );


    //Starting with EB
    cout<<"Now Loop on the EB"<<endl;
    for(Long64_t iEntry=0; iEntry<nentries_EB; iEntry++){
    //for(Long64_t iEntry=0; iEntry<1000; iEntry++){
	  chEB.GetEntry(iEntry);
	  epsilon_EB_h[(int)index_EB]->Fill( mpi0_EB, weight_EB );
    }
    //And then EE
    cout<<"...and now on the EE"<<endl;
    for(Long64_t iEntry=0; iEntry<nentries_EE; iEntry++){
    //for(Long64_t iEntry=0; iEntry<1000; iEntry++){
	  chEE.GetEntry(iEntry);
	  epsilon_EE_h[(int)index_EE]->Fill( mpi0_EE, weight_EE );
    }

    cout<<"Finally I write the histos..."<<endl;
    //Writing histos
    outfile.mkdir("Barrel");
    outfile.cd("Barrel");
    for(int jR=0; jR<N_EB; jR++){
      epsilon_EB_h[jR]->Write();
    }
    outfile.cd();

    outfile.mkdir("Endcap");
    outfile.cd("Endcap");
    for(int jR=0; jR<N_EE; jR++){
      epsilon_EE_h[jR]->Write();
    }
    outfile.cd();
    outfile.Write();
    cout<<"Closing file:"<<endl;
    outfile.Close();
    //In any case I deallocate the memory when I close root
    //cout<<"...and I delete them!"<<endl;
    //deleteEpsilonPlot(epsilon_EB_h, N_EB );
    //deleteEpsilonPlot(epsilon_EE_h, N_EE );
    cout<<"END of the TTree <-> Histos conversion!!"<<endl;
    
}

TH1F** initializeEpsilonHistograms(const char *name, const char *title, int size )
{
    TH1F **h = new TH1F*[size];
    char name_c[100];
    char title_c[200];
    TH1::AddDirectory(kFALSE);
    for(int jR=0; jR<size; jR++)
    {
	  sprintf(name_c, "%s%d", name, jR);
	  sprintf(title_c, "%s%d", title, jR);

	  h[jR] = new TH1F(name_c, title_c, 120,0.,0.5);
	  h[jR]->GetXaxis()->SetTitle("Mass(#gamma#gamma)");
	  //	  h[jR]->SetDirectory(0);
    }
    return h;
}

void  deleteEpsilonPlot(TH1F **h, int size)
{
    for(int jR=0; jR<size; jR++){
	  delete h[jR];
	  cout<<"delete "<<jR<<endl;
    }

    delete h;
}