#include <iostream>
#include <cmath>

#include "Rtypes.h"

#include "AtlasUtils.h"
// #ifndef __CINT__
#include "AtlasStyle.C"
// #endif

#include "TCanvas.h"
#include "TFile.h"
#include "TROOT.h"
#include "TH1F.h"
#include "TRandom.h"
#include "TGraphErrors.h"
using namespace std;

Double_t fermi(Double_t *pt,Double_t *par)
{
Double_t arg = 0;
if (par[2] != 0)
  Double_t fitval = par[0] * 1/(1+TMath::Exp((pt[0]-par[1])/par[2]));
return fitval;
}


void plot()
{
#ifdef __CINT__
  //gROOT->LoadMacro("AtlasUtils.C");
#endif
  SetAtlasStyle();

  TFile * file = TFile::Open("TPout_EFMU10_EFMU10_DATA2010_CombComb_staco.root");
  TFile * fileMC = TFile::Open("TPoutMCZ_EFMU10_EFMU10_2011_CombComb_staco.root");
  
  // TFile *outfile= new TFile("efficienzapt.root", "RECREATE");
   
 
  ///////////////DATA
  TH1F     * h_probes_pass_all  =(TH1F*)    file->Get("hptmsselall");
  TH1F     * h_probes_all  =(TH1F*)    file->Get("hptmsall");

  TH1F     * h_probes_pass  =(TH1F*)    file->Get("hptmsselb");
  TH1F     * h_probes  =(TH1F*)    file->Get("hptmsb");

  TH1F     * h_probes_pass_ec  =(TH1F*)    file->Get("hptmssele");
  TH1F     * h_probes_ec  =(TH1F*)    file->Get("hptmse");

  
  TGraphAsymmErrors* heff_all = new TGraphAsymmErrors();
  heff_all->BayesDivide(h_probes_pass_all,h_probes_all);
  TGraphAsymmErrors* heff = new TGraphAsymmErrors();
  heff->BayesDivide(h_probes_pass,h_probes);
  TGraphAsymmErrors* heffE = new TGraphAsymmErrors();
  heffE->BayesDivide(h_probes_pass_ec,h_probes_ec);
  
  ////////////////MC

   TH1F     * h_probes_pass_allMC  =(TH1F*)    fileMC->Get("hptmsselall");
  TH1F     * h_probes_allMC  =(TH1F*)    fileMC->Get("hptmsall");

  TH1F     * h_probes_passMC  =(TH1F*)    fileMC->Get("hptmsselb");
  TH1F     * h_probesMC  =(TH1F*)    fileMC->Get("hptmsb");

  TH1F     * h_probes_pass_ecMC  =(TH1F*)    fileMC->Get("hptmssele");
  TH1F     * h_probes_ecMC  =(TH1F*)    fileMC->Get("hptmse");

  
  TGraphAsymmErrors* heff_allMC = new TGraphAsymmErrors();
  heff_allMC->BayesDivide(h_probes_pass_allMC,h_probes_allMC);
  TGraphAsymmErrors* heffMC = new TGraphAsymmErrors();
  heffMC->BayesDivide(h_probes_passMC,h_probesMC);
  TGraphAsymmErrors* heffEMC = new TGraphAsymmErrors();
  heffEMC->BayesDivide(h_probes_pass_ecMC,h_probes_ecMC);
  
//   /////////////////////
   TCanvas *c3 = new TCanvas("c3","",10,10,700,500);
  TH1F *hr = c3->DrawFrame(0.,0.,50.,1.1);
  
  hr->GetXaxis()->SetTitle("#font[42]{p_{T} (GeV)}");
  hr->GetYaxis()->SetTitle("EFMU10 efficiency ALL");
  //c1->Divide(1,2);
  // c1->cd(1);
  // file->cd();
  heff_all->SetLineColor(1);
  heff_all->SetMarkerColor(1);
  heff_all->SetFillColor(3);
  heff_all->SetMarkerStyle(20);
  heff_all->SetFillStyle(3001);
  heff_all->Draw("P");
 TF1 *func = new TF1("fit",fermi,0,50,3);
 func->SetParameters(1,1,1);

     heff_all->Fit("fit");
  
      heff_allMC->SetLineColor(2);
  heff_allMC->SetMarkerColor(2);
  heff_allMC->SetFillColor(3);
  heff_allMC->SetMarkerStyle(22);
  heff_allMC->SetFillStyle(3001);
  heff_allMC->Draw("P");
  /*TF1 *func = new TF1("fit",fermi,0,50,3);
 func->SetParameters(1,1,1);

     heff_allMC->Fit("fit");

  //outfile->cd();
  //c3->Write("heffA");

   
 
  myMarkerText(0.7, 0.40,1,20, "MC h150",1.0);
  myMarkerText(0.7, 0.35,2,22, "MC Zmumu ",1.0);
  
  //c3->Print("efficienzaplot/efficienzapt_ALL_Data-MC.pdf");


//   //////////
*/

  TCanvas *c1 = new TCanvas("c1","",10,10,700,500);
  TH1F *hr = c1->DrawFrame(0.,0.,50.,1.1);
  hr->GetXaxis()->SetTitle("#font[42]{p_{T} (GeV)}");
  hr->GetYaxis()->SetTitle("EFMU10 efficiency BARREL");
  heff->SetLineColor(1);
  heff->SetMarkerColor(1);
  heff->SetFillColor(3);
  heff->SetMarkerStyle(20);
  heff->SetFillStyle(3001);
  heff->Draw("P");
 TF1 *func = new TF1("fit",fermi,0,50,3);
 func->SetParameters(1,1,1);

     heff->Fit("fit");

     
  //fileMC->cd();
  heffMC->SetLineColor(2);
  heffMC->SetMarkerColor(2);
  heffMC->SetFillColor(3);
  heffMC->SetMarkerStyle(22);
  heffMC->SetFillStyle(3001);
  heffMC->Draw("P");
  /*TF1 *func = new TF1("fit",fermi,0,50,3);
 func->SetParameters(1,1,1);

     heffMC->Fit("fit");

  //c1->Write("heffB");

  myMarkerText(0.7, 0.40,1,20, "MC h150",1.0);
  myMarkerText(0.7, 0.35,2,22, "MC Zmumu ",1.0);
  
  
  //c1->Print("efficienzaplot/efficienzapt_barrel_Data-MC.pdf");

 
  /////ENDCAP
  */
 TCanvas *c2 = new TCanvas("c2","",10,10,700,500);
 TH1F *hr = c2->DrawFrame(0.,0.,50.,1.1);
 hr->GetXaxis()->SetTitle("#font[42]{p_{T} (GeV)}");
 hr->GetYaxis()->SetTitle("EFMU10 efficiency ENDCAP");
 
 //file->cd();
 heffE->SetLineColor(1);
 heffE->SetMarkerColor(1);
 heffE->SetFillColor(3);
 heffE->SetMarkerStyle(20);
 heffE->SetFillStyle(3001);
 heffE->Draw("P");

  TF1 *func = new TF1("fit",fermi,0,50,3);
 func->SetParameters(1,1,1);

     heffE->Fit("fit");
    //fileMC->cd();
 heffEMC->SetLineColor(2);
 heffEMC->SetMarkerColor(2);
 heffEMC->SetFillColor(3);
 heffEMC->SetMarkerStyle(22);
 heffEMC->SetFillStyle(3001);
 heffEMC->Draw("P");
 /*TF1 *func = new TF1("fit",fermi,0,50,3);
 func->SetParameters(1,1,1);

     heffEMC->Fit("fit");

 //c2->Write("heffE");

  myMarkerText(0.7, 0.40,1,20, "MC h150",1.0);
  myMarkerText(0.7, 0.35,2,22, "MC Zmumu ",1.0);
  
 
  //c2->Print("efficienzaplot/efficienzapt_endcap_Data-MC.pdf");

 
  //outfile->Write();
  //outfile->Close();  
  */
}