#include <iostream>
#include <TH1F.h>
#include <TH2F.h>
#include <TFile.h>
#include <TCanvas.h>
#include <TPostScript.h>
#include <TGraph.h>
#include <TMultiGraph.h>
#include <TStyle.h>
#include <TLegend.h>

float lumi = 146.12;
float lumi_qcd = 125.05;
//int rebin = 1;
int max_value = 100;

void electron(void)
{
  gROOT->SetStyle("Plain");
  gROOT->ForceStyle();
  gStyle->SetOptTitle(0);
  gStyle->SetOptStat(0);

  //TDatime *time = new TDatime();
  //time->Set();
  //int now = time->Get();
  char rootfilename[255] ="NoTRIG_tautype1.root";
  //sprintf(&rootfilename,"NoTRIG_tautype1_%d.root",now);
  
  //TFile *hfile = gROOT->FindObject(rootfilename); if (hfile) hfile->Close();
  //TFile *hfile = new TFile(rootfilename,"RECREATE","Demo ROOT file with histograms");

  //25-SEP-03: replaced EM1with EM2
  //replace: NoTRIG with EM on 9/16/03
  //replace: EMMX with NoTRIG on 9/16/03
  //replace: NoTrigA with EMMX on 9/16/03

  TFile p1305("output/p1305_EM2_1_hist.root");
  TFile p1306("output/p1306_EM2_1_hist.root");
  TFile r1306("output/r1306_EM2_1_hist.root");

  TFile wjete("output/wjete_EM2_1_hist.root");
  TFile wjetmu("output/wjetmu_EM2_1_hist.root");
  TFile wjettau("output/wjettau_EM2_1_hist.root");
  
  TFile wgame("output/wgam_e_EM2_1_hist.root");
  TFile wgammu("output/wgam_mu_EM2_1_hist.root");
  TFile wgamtau("output/wgam_tau_EM2_1_hist.root");
  
  TFile zee("output/zee_NEW_EM2_1_hist.root");
  TFile ztt("output/ztt_EM2_1_hist.root");

  TFile ww("output/ww_EM2_1_hist.root");
  TFile wz("output/wz_EM2_1_hist.root");
  TFile zz("output/zz_EM2_1_hist.root");
  TFile tt("output/ttbar_EM2_1_hist.root");
  
  //QCD files
  TFile p1305_qcd("output/p1305_QCD_EM2_1_hist.root");
  TFile p1306_qcd("output/p1306_QCD_EM2_1_hist.root");
  TFile r1306_qcd("output/r1306_QCD_EM2_1_hist.root");

  TFile wjete_qcd("output/wjete_QCD_EM2_1_hist.root");
  TFile wjetmu_qcd("output/wjetmu_QCD_EM2_1_hist.root");
  TFile wjettau_qcd("output/wjettau_QCD_EM2_1_hist.root");
  
  TFile wgame_qcd("output/wgam_e_QCD_EM2_1_hist.root");
  TFile wgammu_qcd("output/wgam_mu_QCD_EM2_1_hist.root");
  TFile wgamtau_qcd("output/wgam_tau_QCD_EM2_1_hist.root");
  
  TFile zee_qcd("output/zee_QCD_EM2_1_hist.root");
  TFile ztt_qcd("output/ztt_QCD_EM2_1_hist.root");

  TFile ww_qcd("output/ww_QCD_EM2_1_hist.root");
  TFile wz_qcd("output/wz_QCD_EM2_1_hist.root");
  TFile zz_qcd("output/zz_QCD_EM2_1_hist.root");
  TFile tt_qcd("output/ttbar_QCD_EM2_1_hist.root");


  //enter histogram to display here:
  //TString *histogram = new TString("ett1_eta");
  TString *a1 = new TString("ett1_eta");
  TString *a2 = new TString("ett1_phi");
  TString *a3 = new TString("ett1_pt");
  TString *a4 = new TString("ett2_eta");
  TString *a5 = new TString("ett2_phi");
  TString *a6 = new TString("ett2_pt");
 

  TString *histogram[6] = {a1,a2,a3,a4,a5,a6};
  int max_val[6] = {500,500,1500,1500,1000,2500};
  int rebin[6] = {2,3,1,2,3,1};
  int logscale[6]={0,0,0,0,0,0};

  TCanvas *c1[6];
  for (int k=0;k<6;++k) {
    c1[k] = new TCanvas(histogram[k]->Data(),histogram[k]->Data(),600,400);  
  }

  for (int i=0;i<6;++i) {
    
    c1[i]->cd();

    TH1F *hist_p1305 = (TH1F *) p1305.Get(*histogram[i]);
    TH1F *hist_p1306 = (TH1F *) p1306.Get(*histogram[i]);
    TH1F *hist_r1306 = (TH1F *) r1306.Get(*histogram[i]);

    TH1F *hist_wjete   = (TH1F *) wjete.Get(*histogram[i]);
    TH1F *hist_wjetmu  = (TH1F *) wjetmu.Get(*histogram[i]);
    TH1F *hist_wjettau = (TH1F *) wjettau.Get(*histogram[i]);

    TH1F *hist_wgame   = (TH1F *) wgame.Get(*histogram[i]);
    TH1F *hist_wgammu  = (TH1F *) wgammu.Get(*histogram[i]);
    TH1F *hist_wgamtau = (TH1F *) wgamtau.Get(*histogram[i]);

    TH1F *hist_zee = (TH1F *) zee.Get(*histogram[i]);
    TH1F *hist_ztt = (TH1F *) ztt.Get(*histogram[i]);

    TH1F *hist_ww = (TH1F *) ww.Get(*histogram[i]);
    TH1F *hist_wz = (TH1F *) wz.Get(*histogram[i]);
    TH1F *hist_zz = (TH1F *) zz.Get(*histogram[i]);
    TH1F *hist_tt = (TH1F *) tt.Get(*histogram[i]);

    TH1F *qcd_p1305 = (TH1F *) p1305_qcd.Get(*histogram[i]);
    TH1F *qcd_p1306 = (TH1F *) p1306_qcd.Get(*histogram[i]);
    TH1F *qcd_r1306 = (TH1F *) r1306_qcd.Get(*histogram[i]);

    TH1F *qcd_wjete   = (TH1F *) wjete_qcd.Get(*histogram[i]);
    TH1F *qcd_wjetmu  = (TH1F *) wjetmu_qcd.Get(*histogram[i]);
    TH1F *qcd_wjettau = (TH1F *) wjettau_qcd.Get(*histogram[i]);

    TH1F *qcd_wgame   = (TH1F *) wgame_qcd.Get(*histogram[i]);
    TH1F *qcd_wgammu  = (TH1F *) wgammu_qcd.Get(*histogram[i]);
    TH1F *qcd_wgamtau = (TH1F *) wgamtau_qcd.Get(*histogram[i]);

    TH1F *qcd_zee = (TH1F *) zee_qcd.Get(*histogram[i]);
    TH1F *qcd_ztt = (TH1F *) ztt_qcd.Get(*histogram[i]);

    TH1F *qcd_ww = (TH1F *) ww_qcd.Get(*histogram[i]);
    TH1F *qcd_wz = (TH1F *) wz_qcd.Get(*histogram[i]);
    TH1F *qcd_zz = (TH1F *) zz_qcd.Get(*histogram[i]);
    TH1F *qcd_tt = (TH1F *) tt_qcd.Get(*histogram[i]);

    //TPostScript *ps = new TPostScript("WM.ps",112);
  
    // ======================
    // Prepare QCD background
    // ======================
    qcd_p1305->Add(qcd_p1306,1);
    qcd_p1305->Add(qcd_r1306,1);
    qcd_p1305->Rebin(rebin[i]);
  
    qcd_zee->Scale(1044.*lumi_qcd/304819);
    qcd_zee->Rebin(rebin[i]);
    qcd_ztt->Scale(265.*lumi_qcd/53396.);
    qcd_ztt->Rebin(rebin[i]);
  
    qcd_wjettau->Scale(662.*lumi_qcd/121500.);
    qcd_wjettau->Rebin(rebin[i]); 
    qcd_wjetmu->Scale(662.*lumi_qcd/132925.);
    qcd_wjetmu->Rebin(rebin[i]); 
    qcd_wjete->Scale(662.*lumi_qcd/107500.);
    qcd_wjete->Rebin(rebin[i]);

    qcd_wgame->Scale(4.7*lumi_qcd/19423.);
    qcd_wgame->Rebin(rebin[i]);
    qcd_wgammu->Scale(4.7*lumi_qcd/19500.);
    qcd_wgammu->Rebin(rebin[i]);
    qcd_wgamtau->Scale(4.7*lumi_qcd/19500.);
    qcd_wgamtau->Rebin(rebin[i]);

    qcd_ww->Scale(0.906*lumi_qcd/23000.);
    qcd_ww->Rebin(rebin[i]);
    qcd_wz->Scale(0.081*lumi_qcd/10500.);
    qcd_wz->Rebin(rebin[i]);
    qcd_zz->Scale(0.018*lumi_qcd/26790.);
    qcd_zz->Rebin(rebin[i]);

    qcd_tt->Scale(3.*lumi_qcd/21250.);
    qcd_tt->Rebin(rebin[i]);
  
    qcd_zee->Add(qcd_ztt,1);
    qcd_zee->Add(qcd_wjete,1);
    qcd_zee->Add(qcd_wjetmu,1);
    qcd_zee->Add(qcd_wjettau,1);
    qcd_zee->Add(qcd_wgame,1);
    qcd_zee->Add(qcd_wgammu,1);
    qcd_zee->Add(qcd_wgamtau,1);
    qcd_zee->Add(qcd_ww,1);
    qcd_zee->Add(qcd_wz,1);
    qcd_zee->Add(qcd_zz,1);
    qcd_zee->Add(qcd_tt,1);
    qcd_p1305->Add(qcd_zee,-1);


    // ===============
    // DATA histograms
    // ===============
    //data histogram
    hist_p1305->Add(hist_p1306,1);
    hist_p1305->Add(hist_r1306,1);
    hist_p1305->Rebin(rebin[i]);
    cout << "data entries: " << hist_p1305->Integral() << endl;

    //scale MC histograms
    hist_zee->Scale(1044.*lumi/304819.);
    hist_zee->Rebin(rebin[i]);
    hist_ztt->Scale(265.*lumi/53396.);
    hist_ztt->Rebin(rebin[i]);

    hist_wjettau->Scale(662.*lumi/121500.);
    hist_wjettau->Rebin(rebin[i]); 
    hist_wjetmu->Scale(662.*lumi/132925.);
    hist_wjetmu->Rebin(rebin[i]); 
    hist_wjete->Scale(662.*lumi/107500.);
    hist_wjete->Rebin(rebin[i]);
  
    hist_wgame->Scale(4.7*lumi/19423.);
    hist_wgame->Rebin(rebin[i]);
    hist_wgammu->Scale(4.7*lumi/19500.);
    hist_wgammu->Rebin(rebin[i]);
    hist_wgamtau->Scale(4.7*lumi/19500.);
    hist_wgamtau->Rebin(rebin[i]);

    hist_ww->Scale(0.906*lumi/23000.);
    hist_ww->Rebin(rebin[i]);
    hist_wz->Scale(0.081*lumi/10500.);
    hist_wz->Rebin(rebin[i]);
    hist_zz->Scale(0.018*lumi/26790.);
    hist_zz->Rebin(rebin[i]);

    hist_tt->Scale(2.*lumi/21250.);
    hist_tt->Rebin(rebin[i]);


    //draw histograms
    if (logscale[i]==1) {
      c1[i]->SetLogy(1);
    }

    hist_p1305->SetMarkerSize(0.7);
    hist_p1305->SetMarkerStyle(20);
    hist_p1305->Draw("E"); 
    hist_p1305->SetMaximum(max_val[i]);

    hist_tt->Add(hist_wgame,1);
    hist_tt->Add(hist_wgammu,1);
    hist_tt->Add(hist_wgamtau,1);
    hist_tt->Add(hist_wjetmu,1);
    hist_tt->Add(hist_wjettau,1);
    hist_tt->Add(hist_ww,1);
    hist_tt->Add(hist_wz,1);
    hist_tt->Add(hist_zz,1);
    hist_tt->Add(hist_ztt,1);
    hist_tt->Add(hist_zee,1);
    hist_tt->Add(hist_wjete,1);
    hist_tt->Add(qcd_p1305,1);
    hist_tt->SetFillColor(50);
    hist_tt->Draw("same");
    cout << "bgnd entries: " << hist_tt->Integral() << endl;
 
    hist_wgame->Add(hist_wgammu,1);
    hist_wgame->Add(hist_wgamtau,1);
    hist_wgame->Add(hist_wjetmu,1);
    hist_wgame->Add(hist_wjettau,1);
    hist_wgame->Add(hist_ww,1);
    hist_wgame->Add(hist_wz,1);
    hist_wgame->Add(hist_zz,1);
    hist_wgame->Add(hist_ztt,1);
    hist_wgame->Add(hist_zee,1);
    hist_wgame->Add(hist_wjete,1);
    hist_wgame->Add(qcd_p1305,1);
    hist_wgame->SetFillColor(38);
    hist_wgame->Draw("same");

    hist_wjetmu->Add(hist_wjettau,1);
    hist_wjetmu->Add(hist_ww,1);
    hist_wjetmu->Add(hist_wz,1);
    hist_wjetmu->Add(hist_zz,1);
    hist_wjetmu->Add(hist_ztt,1);
    hist_wjetmu->Add(hist_zee,1);
    hist_wjetmu->Add(hist_wjete,1);
    hist_wjetmu->Add(qcd_p1305,1);
    hist_wjetmu->SetFillColor(30);
    hist_wjetmu->Draw("same");

    hist_wjettau->Add(hist_ww,1);
    hist_wjettau->Add(hist_wz,1);
    hist_wjettau->Add(hist_zz,1);
    hist_wjettau->Add(hist_ztt,1);
    hist_wjettau->Add(hist_zee,1);
    hist_wjettau->Add(hist_wjete,1);
    hist_wjettau->Add(qcd_p1305,1);
    hist_wjettau->SetFillColor(16);
    hist_wjettau->Draw("same");

    hist_ww->Add(hist_wz,1);
    hist_ww->Add(hist_zz,1);
    hist_ww->Add(hist_ztt,1);
    hist_ww->Add(hist_zee,1);
    hist_ww->Add(hist_wjete,1);
    hist_ww->Add(qcd_p1305,1);
    hist_ww->SetFillColor(9);
    hist_ww->Draw("same");

    hist_ztt->Add(hist_zee,1);
    hist_ztt->Add(hist_wjete,1);
    hist_ztt->Add(qcd_p1305,1);
    hist_ztt->SetFillColor(7);
    hist_ztt->Draw("same");

    hist_zee->Add(hist_wjete,1);
    hist_zee->Add(qcd_p1305,1);
    hist_zee->SetFillColor(3);
    hist_zee->Draw("same");

    hist_wjete->SetFillColor(4);
    hist_wjete->Add(qcd_p1305,1);
    hist_wjete->Draw("same"); 

    qcd_p1305->SetFillColor(12);
    qcd_p1305->Draw("same");

    hist_p1305->Draw("E same");  

    TLegend *leg = new TLegend(0.6,0.7,0.95,0.95);
    //leg->SetHeader("Transverse Mass");
    leg->AddEntry(hist_zee,"Z #rightarrow ee","f");
    leg->AddEntry(hist_ztt,"Z #rightarrow #tau#tau","f");
    leg->AddEntry(hist_wjete,"W+jet, W #rightarrow e","f");
    leg->AddEntry(hist_wjettau,"W+jet, W #rightarrow #tau","f");
    leg->AddEntry(hist_wjetmu,"W+jet, W #rightarrow #mu","f");
    leg->AddEntry(hist_ww,"WW/WZ/ZZ","f");
    leg->AddEntry(hist_wgame,"W+#gamma","f");
    leg->AddEntry(hist_tt,"tt(bar)","f");
    leg->AddEntry(qcd_p1305,"QCD","f");
    leg->Draw();

    c1[i]->Update();
    //c1->Print("WM.root","root");
    //c1->Print("WM3.eps","eps");
    //c1->Print("ett1_eta.ps","ps");
    
    //histogram[i]->Append(".root");
    //histogram[i]->Prepend("tau1_plots/");
    //c1->Print(*histogram[i],"root");

    //c1->Print("file.ps");
    //ps->Close();

  }

  TFile *test = new TFile("plots_25SEP/em_trig2.root","RECREATE");
  
  for (int k=0; k<6; ++k) {
    c1[k]->Write();
  }

  delete test;

  //c1->Print("file.ps]");
  //hfile->Write();
}