plotting macro

Author: DanielSamitz

Tutorials/PWGEM/Cocktail/plotLFCocktail.C

@@ -0,0 +1,62 @@
+const int nHists = 7;
+TH1F *mee[nHists];
+TH1F *mee_orig[nHists];
+TString histNames[nHists] = {"",     "_Pi0",   "_Eta", "_EtaP",
+                             "_Rho", "_Omega", "_Phi"};
+TString histLegends[nHists] = {
+    "Cocktail sum",
+    "#pi^{0}#rightarrow#gammae^{+}e^{-}",
+    "#eta#rightarrow#gammae^{+}e^{-}",
+    "#eta'#rightarrow#gammae^{+}e^{-}, #eta'#rightarrow#omegae^{+}e^{-}",
+    "#rho#rightarrowe^{+}e^{-}",
+    "#omega#rightarrow#pi^{0}e^{+}e^{-}, #omega#rightarrowe^{+}e^{-}",
+    "#phi#rightarrow#etae^{+}e^{-}, #phi#rightarrow#pi^{0}e^{+}e^{-}, "
+    "#phi#rightarrowe^{+}e^{-}"};
+
+void loadHistos(TFile *file, TH1F *hists[], TString name_extra, int rebin,
+                int nEvents) {
+  for (int i = 0; i < nHists; i++) {
+    hists[i] = (TH1F *)file->GetDirectory("em-lmee-lf-cocktail")
+                   ->Get(TString("mee") + name_extra + histNames[i]);
+    hists[i]->Rebin(rebin);
+    hists[i]->Scale(1. / nEvents);
+    hists[i]->SetTitle(histLegends[i]);
+  }
+}
+
+void plotLFCocktail(TString filename = "AnalysisResults.root", int rebin = 1) {
+
+  TFile *file = TFile::Open(filename.Data());
+
+  TH1I *hNEvents =
+      (TH1I *)file->GetDirectory("em-lmee-lf-cocktail")->Get("NEvents");
+  int nEvents = hNEvents->GetBinContent(1);
+
+  loadHistos(file, mee, "", rebin, nEvents);
+  loadHistos(file, mee_orig, "_orig", rebin, nEvents);
+
+  auto canvas = new TCanvas("LF Cocktail", "LF Cocktail", 1500, 500);
+  canvas->Divide(2, 1);
+
+  mee[0]->SetStats(0);
+  mee[0]->GetXaxis()->SetRangeUser(0.0, 1.1);
+  mee_orig[0]->SetStats(0);
+  mee_orig[0]->GetXaxis()->SetRangeUser(0.0, 1.1);
+
+  canvas->cd(1);
+  gPad->SetLogy();
+  for (int i = 0; i < nHists; i++) {
+    mee_orig[i]->SetLineColor(i + 1);
+    mee_orig[i]->Draw("same");
+  }
+  gPad->BuildLegend(0.62, 0.9, 0.9, 0.6);
+  mee_orig[0]->SetTitle("before resolution and acceptance");
+  canvas->cd(2);
+  gPad->SetLogy();
+  for (int i = 0; i < nHists; i++) {
+    mee[i]->SetLineColor(i + 1);
+    mee[i]->Draw("same");
+  }
+  gPad->BuildLegend(0.62, 0.9, 0.9, 0.6);
+  mee[0]->SetTitle("after resolution and acceptance");
+}