PWG4/macros/electrons/plotMCRates.C

   1 /////////////////////////////////////////////////\r
   2 //\r
   3 // Macro for plotting MC rates of electrons\r
   4 // for the EMCAL PPR\r
   5 //\r
   6 // J.L. Klay (Cal Poly)\r
   7 //\r
   8 /////////////////////////////////////////////////\r
   9 \r
  10 TLegend* leg;\r
  11 \r
  12 void plotMCRates() {\r
  13 \r
  14   gROOT->LoadMacro("makeCombinedData.C");\r
  15   makeData("data/scaled25Oct09/TOTALhistosscaled-LHC09b2-0.root",\r
  16            "data/scaled25Oct09/histosscaledLHC09b4AODc.root",\r
  17            "data/scaled25Oct09/histosWboson.root");\r
  18 \r
  19   gStyle->SetOptStat(0);\r
  20   drawAnnualYields();\r
  21   drawPtCutRates();\r
  22   drawHadEleRatios();\r
  23 \r
  24 }\r
  25 \r
  26 void drawAnnualYields() {\r
  27 \r
  28   TCanvas* crates = new TCanvas();\r
  29   crates->SetFillColor(0);\r
  30   crates->SetBorderMode(0);\r
  31   crates->SetBorderSize(2);\r
  32   crates->SetFrameBorderMode(0);\r
  33   crates->SetFrameBorderMode(0);\r
  34 \r
  35   crates->cd();\r
  36   gPad->SetLogy();\r
  37   allMC->SetXTitle("p_{T} (GeV/c)");\r
  38   allMC->SetTitle("MC electrons in Pb+Pb, 5.5 TeV");\r
  39   allMC->SetYTitle("Annual yield in EMCAL dN/dp_{T} (GeV/c)^{-1}");\r
  40   allMC->GetYaxis()->SetRangeUser(1,2.E6);\r
  41   allMC->GetXaxis()->SetRangeUser(10.,50.);\r
  42   allMC->Draw();\r
  43   bMC->Draw("same");\r
  44   cMC->Draw("same");  \r
  45   cbMC->Draw("same");  \r
  46   convMC->Draw("same");  \r
  47   dalMC->Draw("same");  \r
  48   wzMC->Draw("same");\r
  49 \r
  50   leg = new TLegend(0.6,0.6,0.9,0.9);\r
  51   leg->SetFillColor(0);\r
  52   leg->SetTextSize(leg->GetTextSize()*1.2);\r
  53   leg->AddEntry(allMC,"All MC electrons","l");\r
  54   leg->AddEntry(bMC,"Bottom e","l");\r
  55   leg->AddEntry(cMC,"Charm e","l");\r
  56   leg->AddEntry(cbMC,"B-->C e","l");\r
  57   leg->AddEntry(dalMC,"Dalitz e","l");\r
  58   leg->AddEntry(convMC,"Conversion e","l");\r
  59   leg->AddEntry(wzMC,"W Boson e","l");\r
  60   leg->Draw();\r
  61   crates->Print("MCRates_all.pdf");\r
  62 \r
  63   TCanvas* crates2 = new TCanvas();\r
  64   crates2->Divide(2,4);\r
  65   crates2->cd(1); gPad->SetLogy(); allMC->Draw();\r
  66   crates2->cd(2); gPad->SetLogy(); bMC->Draw();\r
  67   crates2->cd(3); gPad->SetLogy(); cMC->Draw();\r
  68   crates2->cd(4); gPad->SetLogy(); cbMC->Draw();\r
  69   crates2->cd(5); gPad->SetLogy(); convMC->Draw();\r
  70   crates2->cd(6); gPad->SetLogy(); dalMC->Draw();\r
  71   crates2->cd(7); gPad->SetLogy(); wzMC->Draw();\r
  72   crates2->cd(8); gPad->SetLogy(); mchad->Draw();\r
  73 \r
  74 }\r
  75 \r
  76 void drawPtCutRates() {\r
  77 \r
  78   TCanvas* cptcut = new TCanvas();\r
  79   cptcut->SetFillColor(0);\r
  80   cptcut->SetBorderMode(0);\r
  81   cptcut->SetBorderSize(2);\r
  82   cptcut->SetFrameBorderMode(0);\r
  83   cptcut->SetFrameBorderMode(0);\r
  84 \r
  85   cptcut->cd();\r
  86   gPad->SetLogy();\r
  87   TH1F* alleptcut = GetPtCutHisto(allMC);\r
  88   TH1F* beleptcut = GetPtCutHisto(bMC);\r
  89   TH1F* celeptcut = GetPtCutHisto(cMC);\r
  90   TH1F* cbeleptcut = GetPtCutHisto(cbMC);\r
  91   TH1F* dalitzptcut = GetPtCutHisto(dalMC);\r
  92   TH1F* convptcut = GetPtCutHisto(convMC);\r
  93   TH1F* wzptcut = GetPtCutHisto(wzMC);\r
  94   alleptcut->GetXaxis()->SetRangeUser(10,50);\r
  95   alleptcut->GetYaxis()->SetRangeUser(10,2.e6);\r
  96   alleptcut->SetXTitle("p_{T}^{cut} (GeV/c)");\r
  97   alleptcut->SetYTitle("Annual Yield in EMCAL for p_{T}>p_{T}^{cut}");\r
  98   alleptcut->SetTitle("MC electrons in Pb+Pb, 5.5 TeV");\r
  99   alleptcut->Draw();\r
 100   beleptcut->Draw("same");\r
 101   celeptcut->Draw("same");\r
 102   cbeleptcut->Draw("same");\r
 103   dalitzptcut->Draw("same");\r
 104   convptcut->Draw("same");\r
 105   wzptcut->Draw("same");\r
 106   leg->Draw();\r
 107   cptcut->Print("MCRates_ptcut_all.pdf");\r
 108 \r
 109 }\r
 110 \r
 111 void drawHadEleRatios() {\r
 112 \r
 113   TCanvas* ceh = new TCanvas();\r
 114   ceh->SetFillColor(0);\r
 115   ceh->SetBorderMode(0);\r
 116   ceh->SetBorderSize(2);\r
 117   ceh->SetFrameBorderMode(0);\r
 118   ceh->SetFrameBorderMode(0);\r
 119 \r
 120   ceh->cd();\r
 121   gPad->SetLogy();\r
 122   gStyle->SetOptStat(0);\r
 123   allheratio->Rebin(2); allheratio->Scale(1./2.);\r
 124   behratio->Rebin(2); behratio->Scale(1./2.);\r
 125   allheratio->SetLineWidth(2);\r
 126   allheratio->GetYaxis()->SetRangeUser(10,2e3);\r
 127   allheratio->GetXaxis()->SetRangeUser(10.,49.);\r
 128   behratio->GetXaxis()->SetRangeUser(10.,49.);\r
 129   allheratio->SetMarkerStyle(20);\r
 130   behratio->SetMarkerStyle(24);\r
 131   behratio->SetMarkerColor(1);\r
 132   allheratio->Draw();\r
 133   behratio->Draw("psame");\r
 134 \r
 135   TLegend *heleg = new TLegend(0.15,0.15,0.5,0.35);\r
 136   heleg->SetFillColor(0);\r
 137   heleg->SetTextSize(heleg->GetTextSize()*1.5);\r
 138   heleg->AddEntry(allheratio,"All electrons","l");\r
 139   heleg->AddEntry(behratio,"Bottom electrons","p");\r
 140   heleg->Draw();\r
 141   ceh->Print("MCRates_heratio.pdf");\r
 142 }\r
 143 \r