using namespace std;
#include <TH2.h>
#include "TNtupleD.h"
#include "TProfile.h"
#include <TStyle.h>
#include <vector>
#include <string.h>
#include <iostream>
#include <fstream>
#include <sstream>
#include "TMath.h"
#include "TLorentzVector.h"
#include "TTree.h"
#include "TTreeReader.h"
#include "TLeaf.h"
#include <TROOT.h>
#include <TChain.h>
#include <TFile.h>
#include <TSelector.h>
#include <TH1F.h>
#include <TSystem.h>
#include <cmath>


void MakeTemplates(TString Region, TString SampleType, TString LeptonType="elec", TString SliceType="1", TString RegionABCD= "A"){

cout<<"Run with region: "<<Region<<" With sample: "<<SampleType<<" With lepton: "<<LeptonType<<" With slice: "<<SliceType<<" And ABCD region: "<<RegionABCD<<endl;


 TString dir="";
 TString fileToGet="";
 TString nameHistoCF="";

 dir="/nfs/dust/atlas/user/ychinn/llgammaMxAOD/h024VBStest/mc16a/";
 //dir="/nfs/dust/atlas/user/ychinn/llgammaMxAOD/h024VBStestPflow/mc16a/";

 if(LeptonType == "elec"){

 if(SampleType == "VBS"){
   fileToGet="user.yyap.user.yyap.20191206.mc16a.363267.h024VBStestPflow_MxAOD.root/merge.root";
   nameHistoCF="CutFlow_Madgraph_EWKZeegamma_weighted";
 }

 if(SampleType == "QCD_aMCAtNLO"){
   if(SliceType == "1"){
     fileToGet="user.yyap.mc16a.345775.aMcAtNloPythia8_NNPDF30NLO_FxFx_eegamma_HT0_125.p3565.h024VBStest2_MxAOD.root/merge.root";
     //fileToGet="user.yyap.mc16a.345775.aMcAtNloPythia8_NNPDF30NLO_FxFx_eegamma_HT0_125.p3795.h024VBStestPflow_MxAOD.root/merge.root";
     nameHistoCF="CutFlow_Madgraph_345775_weighted";
   }
   if(SliceType == "2"){
     //fileToGet="user.yyap.mc16a.345776.aMcAtNloPythia8_NNPDF30NLO_FxFx_eegamma_HTGT125_MjjLT150.p3565.h024VBStest2_MxAOD.root/merge.root";
     fileToGet="user.yyap.mc16a.345776.aMcAtNloPythia8_NNPDF30NLO_FxFx_eegamma_HTGT125_MjjLT150.p3795.h024VBStestPflow_MxAOD.root/merge.root";
     nameHistoCF="CutFlow_Madgraph_345776_weighted";
   }
   if(SliceType == "3"){
     //fileToGet="user.yyap.mc16a.345777.aMcAtNloPythia8_NNPDF30NLO_FxFx_eegamma_HTGT125_MjjLT500.p3565.h024VBStest2_MxAOD.root/merge.root";   
     fileToGet="user.yyap.mc16a.345777.aMcAtNloPythia8_NNPDF30NLO_FxFx_eegamma_HTGT125_MjjLT500.p3795.h024VBStestPflow_MxAOD.root/merge.root";   
     nameHistoCF="CutFlow_Madgraph_345777_weighted";
   }
   if(SliceType == "4"){
     //fileToGet="user.yyap.mc16a.345778.aMcAtNloPythia8_NNPDF30NLO_FxFx_eegamma_HTGT125_MjjGT500.p3565.h024VBStest2_MxAOD.root/merge.root";   
     fileToGet="user.yyap.mc16a.345778.aMcAtNloPythia8_NNPDF30NLO_FxFx_eegamma_HTGT125_MjjGT500.p3795.h024VBStestPflow_MxAOD.root/merge.root";   
     nameHistoCF="CutFlow_Madgraph_345778_weighted";
   }
 }

 if(SampleType == "QCD_S222"){
   if(SliceType == "1"){
     fileToGet="user.yyap.mc16a.364500.Sherpa_222_NNPDF30NNLO_eegamma_pty_7_15.p3565.h024VBStest2_MxAOD.root/merge.root";
     nameHistoCF="CutFlow_Sherpa2_eegamma_pty_7_15_weighted";
   }
   if(SliceType == "2"){
     fileToGet="user.yyap.mc16a.364501.Sherpa_222_NNPDF30NNLO_eegamma_pty_15_35.p3565.h024VBStest2_MxAOD.root/merge.root";
     nameHistoCF="CutFlow_Sherpa2_eegamma_pty_15_35_weighted";
   }
   if(SliceType == "3"){
     fileToGet="user.yyap.mc16a.364502.Sherpa_222_NNPDF30NNLO_eegamma_pty_35_70.p3565.h024VBStest2_MxAOD.root/merge.root";
     nameHistoCF="CutFlow_Sherpa2_eegamma_pty_35_70_weighted";
   }
   if(SliceType == "4"){
     fileToGet="user.yyap.mc16a.364503.Sherpa_222_NNPDF30NNLO_eegamma_pty_70_140.p3565.h024VBStest2_MxAOD.root/merge.root";
     nameHistoCF="CutFlow_Sherpa2_eegamma_pty_70_140_weighted";
   }
   if(SliceType == "5"){
     fileToGet="user.yyap.mc16a.364504.Sherpa_222_NNPDF30NNLO_eegamma_pty_140_E_CMS.p3565.h024VBStest2_MxAOD.root/merge.root";
     nameHistoCF="CutFlow_Sherpa2_eegamma_pty_140_weighted";
   }
 }

 if(SampleType == "QCD_S224"){
   if(SliceType == "1"){
     fileToGet="user.yyap.mc16a.366140.Sh_224_NN30NNLO_eegamma_LO_pty_7_15.p3705.h024VBStest2_MxAOD.root/merge.root";
     nameHistoCF="CutFlow_Sherpa_366140_weighted";
   }
   if(SliceType == "2"){
     fileToGet="user.yyap.mc16a.366141.Sh_224_NN30NNLO_eegamma_LO_pty_15_35.p3705.h024VBStest2_MxAOD.root/merge.root";
     nameHistoCF="CutFlow_Sherpa_366141_weighted";
   }
   if(SliceType == "3"){
     fileToGet="user.yyap.mc16a.366142.Sh_224_NN30NNLO_eegamma_LO_pty_35_70.p3705.h024VBStest2_MxAOD.root/merge.root";
     nameHistoCF="CutFlow_Sherpa_366142_weighted";
   }
   if(SliceType == "4"){
     fileToGet="user.yyap.mc16a.366143.Sh_224_NN30NNLO_eegamma_LO_pty_70_140.p3705.h024VBStest2_MxAOD.root/merge.root";
     nameHistoCF="CutFlow_Sherpa_366143_weighted";
   }
   if(SliceType == "5"){
     fileToGet="user.yyap.mc16a.366144.Sh_224_NN30NNLO_eegamma_LO_pty_140_E_CMS.p3705.h024VBStest2_MxAOD.root/merge.root";
     nameHistoCF="CutFlow_Sherpa_366144_weighted";
   }
 }
 }

 if(LeptonType == "muon"){

   if(SampleType == "VBS"){
     fileToGet="user.yyap.user.yyap.20191206.mc16a.363268.h024VBStestPflow_MxAOD.root/merge.root";
     nameHistoCF="CutFlow_Madgraph_EWKZmumugamma_weighted";
   }


   if(SampleType == "QCD_aMCAtNLO"){
     if(SliceType == "1"){
       //fileToGet="user.yyap.mc16a.345779.aMcAtNloPythia8_NNPDF30NLO_FxFx_mumugamma_HT0_125.p3565.h024VBStest2_MxAOD.root/merge.root";
       fileToGet="user.yyap.mc16a.345779.aMcAtNloPythia8_NNPDF30NLO_FxFx_mumugamma_HT0_125.p3795.h024VBStestPflow_MxAOD.root/merge.root";
       nameHistoCF="CutFlow_Madgraph_345779_weighted";
     }
     if(SliceType == "2"){
       //fileToGet="user.yyap.mc16a.345780.aMcAtNloPythia8_NNPDF30NLO_FxFx_mumugamma_HTGT125_MjjLT150.p3565.h024VBStest2_MxAOD.root/merge.root";
       fileToGet="user.yyap.mc16a.345780.aMcAtNloPythia8_NNPDF30NLO_FxFx_mumugamma_HTGT125_MjjLT150.p3795.h024VBStestPflow_MxAOD.root/merge.root";
       nameHistoCF="CutFlow_Madgraph_345780_weighted";
     }
     if(SliceType == "3"){
       //fileToGet="user.yyap.mc16a.345781.aMcAtNloPythia8_NNPDF30NLO_FxFx_mumugamma_HTGT125_MjjLT500.p3565.h024VBStest2_MxAOD.root/merge.root";
       fileToGet="user.yyap.mc16a.345781.aMcAtNloPythia8_NNPDF30NLO_FxFx_mumugamma_HTGT125_MjjLT500.p3795.h024VBStestPflow_MxAOD.root/merge.root";
       nameHistoCF="CutFlow_Madgraph_345781_weighted";
     }
     if(SliceType == "4"){
       //fileToGet="user.yyap.mc16a.345782.aMcAtNloPythia8_NNPDF30NLO_FxFx_mumugamma_HTGT125_MjjGT500.p3565.h024VBStest2_MxAOD.root/merge.root";
       fileToGet="user.yyap.mc16a.345782.aMcAtNloPythia8_NNPDF30NLO_FxFx_mumugamma_HTGT125_MjjGT500.p3795.h024VBStestPflow_MxAOD.root/merge.root";
       nameHistoCF="CutFlow_Madgraph_345782_weighted";
     }
   }
       
   if(SampleType == "QCD_S222"){
     if(SliceType == "1"){
       fileToGet="user.yyap.mc16a.364505.Sherpa_222_NNPDF30NNLO_mumugamma_pty_7_15.p3565.h024VBStest2_MxAOD.root/merge.root";
       nameHistoCF="CutFlow_Sherpa2_mumugamma_pty_7_15_weighted";
     }
     if(SliceType == "2"){
       fileToGet="user.yyap.mc16a.364506.Sherpa_222_NNPDF30NNLO_mumugamma_pty_15_35.p3565.h024VBStest2_MxAOD.root/merge.root";
       nameHistoCF="CutFlow_Sherpa2_mumugamma_pty_15_35_weighted";
     }
     if(SliceType == "3"){
       fileToGet="user.yyap.mc16a.364507.Sherpa_222_NNPDF30NNLO_mumugamma_pty_35_70.p3565.h024VBStest2_MxAOD.root/merge.root";
       nameHistoCF="CutFlow_Sherpa2_mumugamma_pty_35_70_weighted";
     }
     if(SliceType == "4"){
       fileToGet="user.yyap.mc16a.364508.Sherpa_222_NNPDF30NNLO_mumugamma_pty_70_140.p3565.h024VBStest2_MxAOD.root/merge.root";
       nameHistoCF="CutFlow_Sherpa2_mumugamma_pty_70_140_weighted";
     }
     if(SliceType == "5"){
       fileToGet="user.yyap.mc16a.364509.Sherpa_222_NNPDF30NNLO_mumugamma_pty_140_E_CMS.p3565.h024VBStest2_MxAOD.root/merge.root";       
       nameHistoCF="CutFlow_Sherpa2_mumugamma_pty_140_weighted";
     }
   }

   if(SampleType == "QCD_S224"){
     if(SliceType == "1"){
       fileToGet="user.yyap.mc16a.366145.Sh_224_NN30NNLO_mumugamma_LO_pty_7_15.p3705.h024VBStest2_MxAOD.root/merge.root";
       nameHistoCF="CutFlow_Sherpa_366145_weighted";
     }
     if(SliceType == "2"){
       fileToGet="user.yyap.mc16a.366146.Sh_224_NN30NNLO_mumugamma_LO_pty_15_35.p3705.h024VBStest2_MxAOD.root/merge.root";
       nameHistoCF="CutFlow_Sherpa_366146_weighted";
     }
     if(SliceType == "3"){
       fileToGet="user.yyap.mc16a.366147.Sh_224_NN30NNLO_mumugamma_LO_pty_35_70.p3705.h024VBStest2_MxAOD.root/merge.root";
       nameHistoCF="CutFlow_Sherpa_366147_weighted";
     }
     if(SliceType == "4"){
       fileToGet="user.yyap.mc16a.366148.Sh_224_NN30NNLO_mumugamma_LO_pty_70_140.p3705.h024VBStest2_MxAOD.root/merge.root";
       nameHistoCF="CutFlow_Sherpa_366148_weighted";
     }
     if(SliceType == "5"){
       fileToGet="user.yyap.mc16a.366149.Sh_224_NN30NNLO_mumugamma_LO_pty_140_E_CMS.p3705.h024VBStest2_MxAOD.root/merge.root";
       nameHistoCF="CutFlow_Sherpa_366149_weighted";
     }
   }
 }

 TString all="";
 all=dir+fileToGet;
 TFile *f0=TFile::Open(all);
 cout<<all<<endl;

 float weight2,weightXS, Zgcentr, ptcone20g, mll, mlly, ptg, ptj1;
 int ngj, nBJets,  isIso, isTight, isPass, Ne, Nmu;
 double m_countw=0.;    
 double m_countw1=0.;    
 double m_countw2=0.;    


 TNtupleD *t0 = (TNtupleD*)f0->Get("CollectionTree");
 TH1F*hCF=(TH1F*)f0->Get(nameHistoCF);    
 hCF->Sumw2();
 cout<<"1st bin content "<<hCF->GetBinContent(1)<<endl;

 int nev0 = t0->GetEntries(); 
 cout<<"nev0 "<<nev0<<endl;

 std::vector<Float_t> v_photonptcone20;
 std::vector<Char_t> v_tight;
 std::vector<Char_t> v_iso;
 std::vector<Float_t> v_photonpt;
 std::vector<Float_t> v_jetpt;

 TTreeReader reader("CollectionTree", f0);

 TTreeReaderValue<Float_t> weight(reader, "HGamEventInfoAuxDyn.weight");
 TTreeReaderValue<Float_t> centrality(reader, "HGamEventInfoAuxDyn.Zy_centrality");
 TTreeReaderValue<Int_t> N_j_btag(reader, "HGamEventInfoAuxDyn.N_j_btag30");
 TTreeReaderValue<Int_t> N_j(reader, "HGamEventInfoAuxDyn.N_j");
 TTreeReaderValue<std::vector<Float_t>> photonptcone20(reader, "HGamPhotonsAuxDyn.ptcone20");
 TTreeReaderValue<std::vector<Float_t>> photonpt(reader, "HGamPhotonsAuxDyn.pt");
 TTreeReaderValue<std::vector<Char_t>> tight(reader, "HGamPhotonsAuxDyn.isTight");
 TTreeReaderValue<std::vector<Char_t>> iso(reader, "HGamPhotonsAuxDyn.isIsoFixedCutLoose");
 TTreeReaderValue<Int_t> N_e(reader, "HGamEventInfoAuxDyn.N_e");
 TTreeReaderValue<Int_t> N_mu(reader, "HGamEventInfoAuxDyn.N_mu");
 TTreeReaderValue<Float_t> m_ll(reader, "HGamEventInfoAuxDyn.m_ll");
 TTreeReaderValue<Float_t> m_lly(reader, "HGamEventInfoAuxDyn.m_lly");
 TTreeReaderValue<Char_t> pass(reader, "HGamEventInfoAuxDyn.isPassedZyVBSPreSel");
 TTreeReaderValue< vector<float> > jetpt(reader, "HGamAntiKt4EMTopoJetsAuxDyn.pt");

 TString XSreader="";
 XSreader="HGamEventInfoAuxDyn.crossSectionBRfilterEff";
 TTreeReaderValue<Float_t> weightCrossSection(reader, XSreader);



 TH1F *h_ngj=new TH1F("h_ngj", "", 10, 0, 10);
 TH1F *h_mj=new TH1F("h_mj", "", 300, 0, 3000);
 TH1F *h_ptj1=new TH1F("h_ptj1", "",500, 0, 1000);
 TH1F *h_ptj2=new TH1F("h_ptj2", "",500, 0, 1000);
 TH1F *h_ptg=new TH1F("h_ptg", "",500, 0, 1000);
 TH1F *h_etag=new TH1F("h_etag", "",60, -3, -3);
 TH1F *h_phig=new TH1F("h_phig", "",64, -3.2, 3.2);
 TH1F *h_etaj1=new TH1F("h_etaj1", "",100, -5, 5);
 TH1F *h_etaj2=new TH1F("h_etaj2", "",100, -5, 5);
 TH1F *h_Zgcentr=new TH1F("h_Zgcentr", "",200, 0, 100);

 h_ngj->Sumw2();
 h_mj->Sumw2();
 h_ptj1->Sumw2();
 h_ptj2->Sumw2();
 h_ptg->Sumw2();
 h_etag->Sumw2();
 h_phig->Sumw2();
 h_etaj1->Sumw2();
 h_etaj2->Sumw2();
 h_Zgcentr->Sumw2();


 while (reader.Next()) {

   v_tight=*tight;
   v_iso=*iso;  
   v_photonptcone20= *photonptcone20;
   v_photonpt = *photonpt; //passing the value
   isTight=v_tight[0];
   isIso=v_iso[0];
   ptcone20g = v_photonptcone20[0]*1e-3;
   ptg = v_photonpt[0]*1e-3;
   v_jetpt = *jetpt; //passing the value
   ptj1 = v_jetpt[0]*1e-3;

   isPass=*pass;
   Ne=*N_e;
   Nmu=*N_mu;
   mll=*m_ll*1e-3;      
   mlly=*m_lly*1e-3;      

   ngj=*N_j;
   nBJets=*N_j_btag;   
   Zgcentr=*centrality;

   weight2=*weight;      
   weightXS=*weightCrossSection;   

   //cout<<"isPass "<<isPass<<endl;
   //cout<<"Ne+Nmu "<<Ne+Nmu<<endl;
   //cout<<"mll+mlly "<<mll+mlly<<endl;
   //cout<<"ptcone20g "<<ptcone20g<<endl;
   //cout<<"(0.05*ptg) "<<(0.05*ptg)<<endl;
   //cout<<"isTight "<<isTight<<endl;
   //cout<<"isIso "<<isIso<<endl;

   if(isPass && (Ne+Nmu)==2 && (mll+mlly)>182 && (ptcone20g<(0.05*ptg)) && isTight && isIso){     
     //cout<<"Pass first step"<<endl;



     if(Region=="BDT"){
     if(ngj>=2 && nBJets==0 && fabs(Zgcentr)<=5){
       //cout<<"Pass second step"<<endl;
       //m_countw +=(weight2*weightXS);  

       m_countw1 +=weight2;
       m_countw2 +=weightXS;
       m_countw +=(weight2*weightXS);  

     }
     }
   }
 }

 cout<<"m_countw1 "<<m_countw1<<endl;
 cout<<"m_countw2 "<<m_countw2<<endl;
 cout<<"m_countw before "<<m_countw<<endl;
 m_countw = m_countw*(36000/hCF->GetBinContent(1));
 cout<<"m_countw "<<m_countw<<endl;

 f0->Close();



}