Reader: Add option to remove or not event with primary vertex not reconstructed

[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/TOTALhistosscaled-LHC09b2-0.root",\r
           "data/scaled25Oct09/histosscaledLHC09b4AODc.root",\r
	   "data/scaled25Oct09/histosWboson.root");\r
\r
  gStyle->SetOptStat(0);\r
  drawAnnualYields();\r
  drawPtCutRates();\r
  drawHadEleRatios();\r
\r
}\r
\r
void drawAnnualYields() {\r
\r
  TCanvas* crates = new TCanvas();\r
  crates->SetFillColor(0);\r
  crates->SetBorderMode(0);\r
  crates->SetBorderSize(2);\r
  crates->SetFrameBorderMode(0);\r
  crates->SetFrameBorderMode(0);\r
\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
  bMC->Draw("same");\r
  cMC->Draw("same");  \r
  cbMC->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->SetFillColor(0);\r
  leg->SetTextSize(leg->GetTextSize()*1.2);\r
  leg->AddEntry(allMC,"All MC electrons","l");\r
  leg->AddEntry(bMC,"Bottom e","l");\r
  leg->AddEntry(cMC,"Charm e","l");\r
  leg->AddEntry(cbMC,"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
  TCanvas* crates2 = new TCanvas();\r
  crates2->Divide(2,4);\r
  crates2->cd(1); gPad->SetLogy(); allMC->Draw();\r
  crates2->cd(2); gPad->SetLogy(); bMC->Draw();\r
  crates2->cd(3); gPad->SetLogy(); cMC->Draw();\r
  crates2->cd(4); gPad->SetLogy(); cbMC->Draw();\r
  crates2->cd(5); gPad->SetLogy(); convMC->Draw();\r
  crates2->cd(6); gPad->SetLogy(); dalMC->Draw();\r
  crates2->cd(7); gPad->SetLogy(); wzMC->Draw();\r
  crates2->cd(8); gPad->SetLogy(); mchad->Draw();\r
\r
}\r
\r
void drawPtCutRates() {\r
\r
  TCanvas* cptcut = new TCanvas();\r
  cptcut->SetFillColor(0);\r
  cptcut->SetBorderMode(0);\r
  cptcut->SetBorderSize(2);\r
  cptcut->SetFrameBorderMode(0);\r
  cptcut->SetFrameBorderMode(0);\r
\r
  cptcut->cd();\r
  gPad->SetLogy();\r
  TH1F* alleptcut = GetPtCutHisto(allMC);\r
  TH1F* beleptcut = GetPtCutHisto(bMC);\r
  TH1F* celeptcut = GetPtCutHisto(cMC);\r
  TH1F* cbeleptcut = GetPtCutHisto(cbMC);\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->SetFillColor(0);\r
  ceh->SetBorderMode(0);\r
  ceh->SetBorderSize(2);\r
  ceh->SetFrameBorderMode(0);\r
  ceh->SetFrameBorderMode(0);\r
\r
  ceh->cd();\r
  gPad->SetLogy();\r
  gStyle->SetOptStat(0);\r
  allheratio->Rebin(2); allheratio->Scale(1./2.);\r
  behratio->Rebin(2); behratio->Scale(1./2.);\r
  allheratio->SetLineWidth(2);\r
  allheratio->GetYaxis()->SetRangeUser(10,2e3);\r
  allheratio->GetXaxis()->SetRangeUser(10.,49.);\r
  behratio->GetXaxis()->SetRangeUser(10.,49.);\r
  allheratio->SetMarkerStyle(20);\r
  behratio->SetMarkerStyle(24);\r
  behratio->SetMarkerColor(1);\r
  allheratio->Draw();\r
  behratio->Draw("psame");\r
\r
  TLegend *heleg = new TLegend(0.15,0.15,0.5,0.35);\r
  heleg->SetFillColor(0);\r
  heleg->SetTextSize(heleg->GetTextSize()*1.5);\r
  heleg->AddEntry(allheratio,"All electrons","l");\r
  heleg->AddEntry(behratio,"Bottom electrons","p");\r
  heleg->Draw();\r
  ceh->Print("MCRates_heratio.pdf");\r
}\r
\r