1 // Author: Ionut-Cristian Arsene
2 // Contacts: iarsene@cern.ch
3 // Make QA pictures for the J/psi->ee analysis
8 // -- to add it in automatic script
9 // -- changes Naming conventions
10 // -- saving objects instead of canvas
13 //______________________________________________________________________________
14 void processJpsi2eeQAplots(const Char_t* filename="jpsi_Default.root",
15 TString suffix = "eps",
16 const Char_t* outfile="Jpsi2eeQAplots_output.root") {
20 TFile* file = TFile::Open(filename);
22 // event wise histograms
23 TH1F* zDistrib = (TH1F*)GetHistogram(file, "default", "Event", "Z");
25 // electron candidate histograms
26 TH2F* tpcDedx = (TH2F*)GetHistogram(file, "default", "Track_ev1+", "P_InnerParam_TPC_signal");
27 tpcDedx->Add((TH2F*)GetHistogram(file, "default", "Track_ev1-", "P_InnerParam_TPC_signal"));
28 TH2F* tpcNsigmaEle = (TH2F*)GetHistogram(file, "default", "Track_ev1+", "P_InnerParam_TPC_nSigma_Electrons");
29 tpcNsigmaEle->Add((TH2F*)GetHistogram(file, "default", "Track_ev1-", "P_InnerParam_TPC_nSigma_Electrons"));
31 // pair candidate histograms
32 TH1F* invmass_pp = (TH1F*)GetHistogram(file, "default", "Pair_ev1+_ev1+", "pM");
33 TH1F* invmass_pm = (TH1F*)GetHistogram(file, "default", "Pair_ev1+_ev1-", "pM");
34 TH1F* invmass_mm = (TH1F*)GetHistogram(file, "default", "Pair_ev1-_ev1-", "pM");
37 TLatex* latex=new TLatex();
39 latex->SetTextSize(0.04);
40 latex->SetTextFont(42);
41 TCanvas* c1=new TCanvas("fig_dq_tpcDedx", "");
43 tpcDedx->SetStats(kFALSE);
44 tpcDedx->GetXaxis()->SetRangeUser(0.0,10.0);
45 tpcDedx->GetYaxis()->SetRangeUser(40.0,120.0);
46 tpcDedx->SetTitle("");
47 tpcDedx->Draw("colz");
48 latex->DrawLatex(0.12, 0.83, "J/#psi electron candidates");
51 TCanvas* c2=new TCanvas("fig_dq_tpcNsigmaElectron", "");
53 tpcNsigmaEle->SetStats(kFALSE);
54 tpcNsigmaEle->GetYaxis()->SetRangeUser(-5.0,5.0);
55 tpcNsigmaEle->GetXaxis()->SetRangeUser(0.0,10.0);
56 tpcNsigmaEle->SetTitle("");
57 tpcNsigmaEle->Draw("colz");
58 latex->DrawLatex(0.12, 0.83, "J/#psi electron candidates");
61 TCanvas* c3=new TCanvas("fig_dq_eeInvMass", "");
63 invmass_pm->SetStats(kFALSE);
64 invmass_pm->SetTitle("");
65 invmass_pm->SetLineColor(1);
66 invmass_pm->GetYaxis()->SetTitle(Form("Pairs per %.0f MeV/c^{2}", 1000.0*invmass_pm->GetXaxis()->GetBinWidth(1)));
67 invmass_pm->GetXaxis()->SetTitle("m_{e^{+}e^{-}} (GeV/c^{2})");
71 invmass_pp->SetLineColor(2);
72 invmass_pp->Draw((invmass_pm ? "same" : ""));
75 invmass_mm->SetLineColor(4);
76 invmass_mm->Draw((invmass_mm ? "same" : ""));
78 if(invmass_pm || invmass_mm || invmass_pp)
79 latex->DrawLatex(0.12, 0.85, "J/#psi candidates");
80 if(invmass_pm && zDistrib && zDistrib->Integral()>1.);
81 latex->DrawLatex(0.12, 0.80, Form("candidates / event = %.3f", invmass_pm->Integral() / zDistrib->Integral()));
82 TLegend* legend=new TLegend(0.7,0.7,0.89,0.89);
83 legend->SetTextFont(42);
84 legend->SetBorderSize(0);
85 legend->SetFillColor(0);
86 if(invmass_pm) legend->AddEntry(invmass_pm, "+- pairs", "l");
87 if(invmass_pp) legend->AddEntry(invmass_pp, "++ pairs", "l");
88 if(invmass_mm) legend->AddEntry(invmass_mm, "-- pairs", "l");
89 if(invmass_mm || invmass_pm || invmass_pp) legend->Draw();
91 c1->SaveAs(Form("fig_dq_tpcDedx.%s",suffix.Data()));
92 c2->SaveAs(Form("fig_dq_tpcNsigmaElectron.%s",suffix.Data()));
93 c3->SaveAs(Form("fig_dq_eeInvMass.%s",suffix.Data()));
96 // TFile* save=new TFile(outputName, "RECREATE");
105 TFile *fout = TFile::Open(outfile,"UPDATE");
108 TDirectoryFile *cdd = NULL;
109 cdd = (TDirectoryFile*)fout->Get("DQ");
111 Printf("Warning: DQ <dir> doesn't exist, creating a new one");
112 cdd = (TDirectoryFile*)fout->mkdir("DQ");
120 invmass_pp->SetName(Form("fig_dq_%s_pp", invmass_pp->GetName()));
125 invmass_pm->SetName(Form("fig_dq_%s_pm", invmass_pm->GetName()));
129 invmass_mm->SetName(Form("fig_dq_%s_mm", invmass_mm->GetName()));
133 tpcNsigmaEle->SetName(Form("fig_dq_%s", tpcNsigmaEle->GetName()));
134 tpcNsigmaEle->Write();
136 tpcDedx->SetName(Form("fig_dq_%s", tpcDedx->GetName()));
145 //______________________________________________________________________________
146 TObject* GetHistogram(TFile* file,
147 const Char_t* cutClass,
148 const Char_t* histList,
149 const Char_t* histName) {
151 // get a histogram from the J/psi output
153 TKey* qaKey = (TKey*)file->GetListOfKeys()->At(0);
154 TList* qaList = (TList*)qaKey->ReadObj();
156 THashList* qaCutClass = (THashList*)qaList->FindObject(cutClass);
157 THashList* qaHistList = (THashList*)qaCutClass->FindObject(histList);
158 return qaHistList->FindObject(histName);