major updates to plotting macros

[u/mrichter/AliRoot.git] / PWG4 / macros / electrons / plotMCRates.C

/////////////////////////////////////////////////\r
//\r
// Macro for plotting MC rates of electrons\r
// for the EMCAL PPR\r
//\r
// J.L. Klay (Cal Poly)\r
//\r
/////////////////////////////////////////////////\r
\r
TLegend* leg;\r
\r
void plotMCRates() {\r
\r
  gROOT->LoadMacro("makeCombinedData.C");\r
  makeData("data/scaled25Oct09/histosLHC08d6.root",\r
           "data/scaled25Oct09/TOTALhistosscaled-LHC09b2-0.root",\r
           "data/scaled25Oct09/histosscaledLHC09b4AODc.root",\r
           "data/scaled25Oct09/histosWboson.root");\r
\r
  gStyle->SetOptStat(0);\r
  //drawXSRates();\r
  drawAnnualYields();\r
  drawPtCutRates();\r
  drawHadEleRatios();\r
  drawSigBkg();\r
\r
}\r
\r
void drawAnnualYields() {\r
\r
  TCanvas* crates = new TCanvas();\r
  crates->cd();\r
  gPad->SetLogy();\r
  allmc->SetXTitle("p_{T} (GeV/c)");\r
  allmc->SetTitle("MC electrons in Pb+Pb, 5.5 TeV");\r
  allmc->SetYTitle("Annual yield in EMCAL dN/dp_{T} (GeV/c)^{-1}");\r
  allmc->GetYaxis()->SetRangeUser(1,2.E6);\r
  allmc->GetXaxis()->SetRangeUser(10.,50.);\r
  allmc->Draw();\r
  belemc->Draw("same");  \r
  celemc->Draw("same");  \r
  candbmc->Draw("same");  \r
  convmc->Draw("same");  \r
  dalmc->Draw("same");  \r
  wzmc->Draw("same");\r
\r
  leg = new TLegend(0.6,0.6,0.9,0.9);\r
  leg->SetTextSize(leg->GetTextSize()*1.2);\r
  leg->AddEntry(allmc,"All MC electrons","l");\r
  leg->AddEntry(belemc,"Bottom e","l");\r
  leg->AddEntry(celemc,"Charm e","l");\r
  leg->AddEntry(candbmc,"B-->C e","l");\r
  leg->AddEntry(dalmc,"Dalitz e","l");\r
  leg->AddEntry(convmc,"Conversion e","l");\r
  leg->AddEntry(wzmc,"W Boson e","l");\r
  leg->Draw();\r
  crates->Print("MCRates_all.pdf");\r
\r
}\r
\r
void drawSigBkg() {\r
\r
  TCanvas* csigbkg = new TCanvas();\r
  csigbkg->cd();\r
  gPad->SetLogy();\r
  allmc->SetXTitle("p_{T} (GeV/c)");\r
  allmc->SetTitle("MC electrons in Pb+Pb, 5.5 TeV");\r
  allmc->SetYTitle("Annual Yield in EMCAL dN/dp_{T} (GeV/c)^{-1}");\r
  allmc->GetYaxis()->SetRangeUser(1.,6.E8);\r
  allmc->GetXaxis()->SetRangeUser(0.,50.);\r
  allmc->Draw();\r
  sigemc->Draw("same");  \r
  bkgemc->Draw("same");  \r
  hijemc->Draw("same");  \r
  wallemc->Draw("same");\r
\r
  TLegend* leg1 = new TLegend(0.6,0.6,0.9,0.9);\r
  leg1->SetTextSize(leg->GetTextSize()*1.2);\r
  leg1->AddEntry(allmc,"All MC electrons","l");\r
  leg1->AddEntry(sigemc,"B-Jet Events","l");\r
  leg1->AddEntry(hijemc,"Pb+Pb Underlying Event","l");\r
  leg1->AddEntry(bkgemc,"Jet-Jet Events","l");\r
  leg1->AddEntry(wallemc,"W-decay Events","l");\r
  leg1->Draw();\r
  csigbkg->Print("MCRates_byEventSource.pdf");\r
\r
}\r
\r
void drawPtCutRates() {\r
\r
  TCanvas* cptcut = new TCanvas();\r
  cptcut->cd();\r
  gPad->SetLogy();\r
  TH1F* alleptcut = GetPtCutHisto(allmc);\r
  TH1F* beleptcut = GetPtCutHisto(belemc);\r
  TH1F* celeptcut = GetPtCutHisto(celemc);\r
  TH1F* cbeleptcut = GetPtCutHisto(candbmc);\r
  TH1F* dalitzptcut = GetPtCutHisto(dalmc);\r
  TH1F* convptcut = GetPtCutHisto(convmc);\r
  TH1F* wzptcut = GetPtCutHisto(wzmc);\r
  alleptcut->GetXaxis()->SetRangeUser(10,50);\r
  alleptcut->GetYaxis()->SetRangeUser(10,2.e6);\r
  alleptcut->SetXTitle("p_{T}^{cut} (GeV/c)");\r
  alleptcut->SetYTitle("Annual Yield in EMCAL for p_{T}>p_{T}^{cut}");\r
  alleptcut->SetTitle("MC electrons in Pb+Pb, 5.5 TeV");\r
  alleptcut->Draw();\r
  beleptcut->Draw("same");\r
  celeptcut->Draw("same");\r
  cbeleptcut->Draw("same");\r
  dalitzptcut->Draw("same");\r
  convptcut->Draw("same");\r
  wzptcut->Draw("same");\r
  leg->Draw();\r
  cptcut->Print("MCRates_ptcut_all.pdf");\r
\r
}\r
\r
void drawHadEleRatios() {\r
\r
  TCanvas* ceh = new TCanvas();\r
  ceh->cd();\r
  gPad->SetLogy();\r
  gStyle->SetOptStat(0);\r
  TH1F* allratio = (TH1F*)allmc->Clone();\r
  TH1F* behratio = (TH1F*)belemc->Clone();\r
  allratio->SetTitle("MC hadrons and electrons in Pb+Pb, 5.5 TeV");\r
  allratio->SetXTitle("p_{T} (GeV/c)");\r
  allratio->SetYTitle("Hadrons/Electrons");\r
  for(Int_t i = 1; i < allmc->GetNbinsX(); i++) {\r
    Double_t vale = allmc->GetBinContent(i);\r
    Double_t valb = belemc->GetBinContent(i);\r
    Double_t valh = mchad->GetBinContent(i);\r
    //printf("pT %.2f, Hadron %.1f, Electron %.1f, B-electron %.1f\n",all->GetBinCenter(i),valh,vale,valb);\r
    if(vale>0) allratio->SetBinContent(i,valh/vale);\r
    else allratio->SetBinContent(i,0.);\r
\r
    if(valb>0) behratio->SetBinContent(i,valh/valb);\r
    else behratio->SetBinContent(i,0.);\r
\r
    allratio->SetBinError(i,0.);\r
    behratio->SetBinError(i,0.);\r
  }\r
  allratio->Rebin(5); allratio->Scale(1./5.);\r
  behratio->Rebin(5); behratio->Scale(1./5.);\r
  allratio->GetYaxis()->SetRangeUser(50,1e4);\r
  allratio->GetXaxis()->SetRangeUser(10.,49.);\r
  behratio->GetXaxis()->SetRangeUser(10.,49.);\r
  allratio->SetMarkerStyle(20);\r
  behratio->SetMarkerStyle(24);\r
  allratio->Fit("pol0");\r
  allratio->Draw();\r
  behratio->Draw("psame");\r
\r
  TLegend *heleg = new TLegend(0.4,0.75,0.75,0.9);\r
  heleg->SetTextSize(heleg->GetTextSize()*1.5);\r
  heleg->AddEntry(allratio,"All electrons","l");\r
  heleg->AddEntry(behratio,"Bottom electrons","p");\r
  heleg->Draw();\r
  ceh->Print("MCRates_heratio.pdf");\r
}\r
\r