[u/mrichter/AliRoot.git] / PWGHF / hfe / macros / PerformanceSpectrumUncorr.C

const Double_t ptmin = 0.2;
const Double_t ptmax = 5.;

void DefineTPChisto(TH2 *h2, const char *yname);
void ALICEWorkInProgress(TCanvas *c, TString date = "today");
void PerformanceSpectrumUncorr(const Char_t *fname = "HFEtask.root"){

  gROOT->SetStyle("Plain");
  gStyle->SetTitleFillColor(0);
  gStyle->SetTitleBorderSize(0);
  gStyle->SetTitleX(0.1);
  gStyle->SetTitleY(0.96);

  TFile *in = TFile::Open(fname);
  TList *res = (TList *)in->Get("HFE_Results");
  TList *qa = (TList *)in->Get("HFE_QA");
  gROOT->cd();
  AliHFEcontainer *tcont = dynamic_cast<AliHFEcontainer *>(res->FindObject("trackContainer"));
  AliCFContainer *c = tcont->GetCFContainer("recTrackContReco");
  AliCFContainer *cb = tcont->GetCFContainer("hadronicBackground");

  TH1 *spec = c->Project(c->GetNStep() - 1, 0);
  spec->GetXaxis()->SetTitle("p_{T} / GeV/c");
  spec->GetYaxis()->SetTitle("#frac{dN}{dp_{T}} / (GeV/c)^{-1}");
  spec->GetXaxis()->SetRangeUser(ptmin, ptmax);
  spec->GetYaxis()->SetTitleOffset(1.1);
  spec->SetTitle();
  spec->SetStats(kFALSE);

  spec->SetLineColor(kBlue);
  spec->SetLineWidth(1);
  spec->SetMarkerColor(kBlue);
  spec->SetMarkerStyle(22);

  // Produce background subtracted spectrum
  AliCFDataGrid tracks("tracks", "track grid", *c, c->GetNStep() - 1);
  AliCFDataGrid background("background", "background grid", *cb, 1);
  tracks.ApplyBGCorrection(background);
  TH1 *spec_subtracted = tracks.Project(0);
  spec_subtracted->GetXaxis()->SetTitle("p_{T} / GeV/c");
  spec_subtracted->GetYaxis()->SetTitle("#frac{dN}{dp_{T}} / (GeV/c)^{-1}");
  spec_subtracted->GetXaxis()->SetRangeUser(ptmin, ptmax);
  spec_subtracted->GetYaxis()->SetTitleOffset(1.1);
  spec_subtracted->SetTitle();
  spec_subtracted->SetStats(kFALSE);
  spec_subtracted->SetLineColor(kRed);
  spec_subtracted->SetLineWidth(1);
  spec_subtracted->SetMarkerColor(kRed);
  spec_subtracted->SetMarkerStyle(22);

  TLegend *leg = new TLegend(0.2, 0.25, 0.4, 0.35);
  leg->SetBorderSize(0);
  leg->SetFillStyle(0);
  leg->AddEntry(spec, "Raw Spectrum", "p");
  leg->AddEntry(spec_subtracted, "Spectrum after background subtraction", "p");
 
  TCanvas *c1 = new TCanvas("cspec", "Single-inclusive electron spectrum", 1200, 750);
  c1->cd();
  c1->SetLogy();
  c1->SetGridx(kFALSE);
  c1->SetGridy(kFALSE);
  spec->Draw("ep");
  spec_subtracted->Draw("epsame");
  leg->Draw();
  ALICEWorkInProgress(c1, "today");

  // PID
  TList *pidqa = (TList *)qa->FindObject("HFEpidQA");
  AliHFEtpcPIDqa *tpcqa = (AliHFEtpcPIDqa *)pidqa->FindObject("TPCQA");
  AliHFEtofPIDqa *tofqa = (AliHFEtofPIDqa *)pidqa->FindObject("TOFQA");

  // Make Plots for TPC
  // Create histograms by projecting the THnSparse
  TH2 *hTPCall = tpcqa->MakeSpectrumdEdx(AliHFEdetPIDqa::kBeforePID);
  TH2 *hTPCselected = tpcqa->MakeSpectrumdEdx(AliHFEdetPIDqa::kAfterPID);
  TH2 *hTPCsigmaAll = tpcqa->MakeSpectrumNSigma(AliHFEdetPIDqa::kBeforePID);
  TH2* hTPCsigmaSelected = tpcqa->MakeSpectrumNSigma(AliHFEdetPIDqa::kAfterPID);
  // Make Plots for TOF
  TH2 *hTOFsigmaAll = tofqa->MakeSpectrumNSigma(AliHFEdetPIDqa::kBeforePID);
  TH2 *hTOFsigmaSelected = tofqa->MakeSpectrumNSigma(AliHFEdetPIDqa::kAfterPID);

  hTPCsigmaAll->SetTitle("TPC n#sigma around the electron line");
  hTPCsigmaSelected->SetTitle("TPC n#sigma around the electron line for selected tracks");
  hTOFsigmaAll->SetTitle("TOF n#sigma around the electron line");
  hTOFsigmaSelected->SetTitle("TOF n#sigma around the electron line for selected tracks");
  DefineTPChisto(hTPCall, "TPC Signal / a.u");
  DefineTPChisto(hTPCselected, "TPC Signal / a.u.");
  DefineTPChisto(hTPCsigmaAll, "TPC Sigma");
  DefineTPChisto(hTPCsigmaSelected, "TPC Sigma");

  // Also make nice histograms for TOF
  DefineTPChisto(hTOFsigmaAll, "TOF Sigma");
  DefineTPChisto(hTOFsigmaSelected, "TOF Sigma");

  // Plot them
  TCanvas *c2 = new TCanvas("cTPCall", "TPC Signal for all tracks", 640, 480);
  c2->cd();
  c2->SetGridx(kFALSE);
  c2->SetGridy(kFALSE);
  c2->SetLogx();
  c2->SetLogz();
  hTPCall->GetYaxis()->SetRangeUser(40., 100.);
  hTPCall->Draw("colz");
  ALICEWorkInProgress(c2, "today");

  TCanvas *c3 = new TCanvas("cTPCsel", "TPC Signal for selected tracks", 640, 480);
  c3->cd();
  c3->SetGridx(kFALSE);
  c3->SetGridy(kFALSE);
  c3->SetLogx();
  c3->SetLogz();
  hTPCselected->GetYaxis()->SetRangeUser(40., 100.);
  hTPCselected->Draw("colz");
  ALICEWorkInProgress(c3, "today");

  TCanvas *c4 = new TCanvas("cTPCsigAll", "TPC Sigma for all tracks", 640, 480);
  c4->cd();
  c4->SetGridx(kFALSE);
  c4->SetGridy(kFALSE);
  c4->SetLogx();
  c4->SetLogz();
  //hTPCsigmaAll->GetYaxis()->SetRangeUser(-3.5, 5.);
  hTPCsigmaAll->Draw("colz");
  ALICEWorkInProgress(c4, "today");

  TCanvas *c5 = new TCanvas("cTPCsigSel", "TPC Sigma for selected tracks", 640, 480);
  c5->cd();
  c5->SetGridx(kFALSE);
  c5->SetGridy(kFALSE);
  c5->SetLogx();
  c5->SetLogz();
  hTPCsigmaSelected->GetYaxis()->SetRangeUser(-3.5, 5.);
  hTPCsigmaSelected->Draw("colz");
  ALICEWorkInProgress(c5, "today");

  TCanvas *c6 = new TCanvas("cTOFsigAll", "TOF Sigma for all tracks", 640, 480);
  c6->cd();
  c6->SetGridx(kFALSE);
  c6->SetGridy(kFALSE);
  c6->SetLogx();
  c6->SetLogz();
  hTOFsigmaAll->Draw("colz");
  ALICEWorkInProgress(c6, "today");

  TCanvas *c7 = new TCanvas("cTOFsigSel", "TOF Sigma for selected tracks", 640, 480);
  c7->cd();
  c7->SetGridx(kFALSE);
  c7->SetGridy(kFALSE);
  c7->SetLogx();
  c7->SetLogz();
  //hTOFsigmaSelected->GetYaxis()->SetRangeUser(-3, 3);
  hTOFsigmaSelected->Draw("colz");
  ALICEWorkInProgress(c7, "today");

  TFile *output = new TFile("Performance.root", "RECREATE");
  output->cd();
  spec->Write();
  hTPCall->Write();
  hTPCselected->Write();
  hTPCsigmaAll->Write();
  hTPCsigmaSelected->Write();
  c1->Write();
  c2->Write();
  c3->Write();
  c4->Write();
  c5->Write();
  c6->Write();
  c7->Write();
  output->Close();
  delete output;
}

void DefineTPChisto(TH2 *h, const char *yaxis){
  h->SetStats(kFALSE);
  h->GetXaxis()->SetRangeUser(ptmin, ptmax);
  h->GetXaxis()->SetMoreLogLabels();
  h->GetXaxis()->SetNoExponent();
  h->GetXaxis()->SetTitle("p / GeV/c");
  h->GetYaxis()->SetTitle(yaxis);
}

void ALICEWorkInProgress(TCanvas *c,TString today){
 //date must be in the form: 04/05/2010
 if(today=="today"){
   TDatime startt;                                                                                                                                                        
   int date=startt.GetDate();

   int y=date/10000;
   int m=(date%10000)/100;
   int d=date%100;


   today="";
   today+=d;
   if(m<10)
     today.Append("/0");
   else today.Append("/");
   today+=m;
   today.Append("/");
   today+=y;  

 }
 TPad *myPadLogo = new TPad("myPadLogo", "Pad for ALICE Logo",0.67,0.65,0.82,0.89);
 //  myPadLogo->SetFillColor(2); 
 myPadLogo->SetBorderMode(0);
 myPadLogo->SetBorderSize(2);
 myPadLogo->SetFrameBorderMode(0);
 myPadLogo->SetLeftMargin(0.0);
 myPadLogo->SetTopMargin(0.0);
 myPadLogo->SetBottomMargin(0.0);
 myPadLogo->SetRightMargin(0.0);
 myPadLogo->Draw();
 myPadLogo->cd();
 TASImage *myAliceLogo = new TASImage("/u/mfasel/work/electron/Spectrum/alice_logo.png");
 myAliceLogo->Draw();
 c->cd();
 TPaveText* t1=new TPaveText(0.59,0.59,0.89,0.66,"NDC");
 t1->SetFillStyle(0);
 t1->SetBorderSize(0);
 t1->AddText(0.,0.,"ALICE Performance");
 t1->SetTextColor(kRed);
 t1->SetTextFont(42);
 t1->Draw();
 TPaveText* t2=new TPaveText(0.59,0.54,0.89,0.60,"NDC");
 t2->SetFillStyle(0);
 t2->SetBorderSize(0);
 t2->SetTextColor(kRed);
 t2->SetTextFont(52);
 t2->AddText(0.,0.,today.Data());
 t2->Draw();
}
Commit	Line	Data
8c1c76e9	1	const Double_t ptmin = 0.2;
	2	const Double_t ptmax = 5.;
	3
	4	void DefineTPChisto(TH2 h2, const char yname);
	5	void ALICEWorkInProgress(TCanvas *c, TString date = "today");
	6	void PerformanceSpectrumUncorr(const Char_t *fname = "HFEtask.root"){
	7
	8	gROOT->SetStyle("Plain");
	9	gStyle->SetTitleFillColor(0);
	10	gStyle->SetTitleBorderSize(0);
	11	gStyle->SetTitleX(0.1);
	12	gStyle->SetTitleY(0.96);
	13
	14	TFile *in = TFile::Open(fname);
	15	TList res = (TList )in->Get("HFE_Results");
	16	TList qa = (TList )in->Get("HFE_QA");
	17	gROOT->cd();
	18	AliHFEcontainer tcont = dynamic_cast<AliHFEcontainer >(res->FindObject("trackContainer"));
	19	AliCFContainer *c = tcont->GetCFContainer("recTrackContReco");
	20	AliCFContainer *cb = tcont->GetCFContainer("hadronicBackground");
	21
	22	TH1 *spec = c->Project(c->GetNStep() - 1, 0);
	23	spec->GetXaxis()->SetTitle("p_{T} / GeV/c");
	24	spec->GetYaxis()->SetTitle("#frac{dN}{dp_{T}} / (GeV/c)^{-1}");
	25	spec->GetXaxis()->SetRangeUser(ptmin, ptmax);
	26	spec->GetYaxis()->SetTitleOffset(1.1);
	27	spec->SetTitle();
	28	spec->SetStats(kFALSE);
	29
	30	spec->SetLineColor(kBlue);
	31	spec->SetLineWidth(1);
	32	spec->SetMarkerColor(kBlue);
	33	spec->SetMarkerStyle(22);
	34
	35	// Produce background subtracted spectrum
	36	AliCFDataGrid tracks("tracks", "track grid", *c, c->GetNStep() - 1);
	37	AliCFDataGrid background("background", "background grid", *cb, 1);
	38	tracks.ApplyBGCorrection(background);
	39	TH1 *spec_subtracted = tracks.Project(0);
	40	spec_subtracted->GetXaxis()->SetTitle("p_{T} / GeV/c");
	41	spec_subtracted->GetYaxis()->SetTitle("#frac{dN}{dp_{T}} / (GeV/c)^{-1}");
	42	spec_subtracted->GetXaxis()->SetRangeUser(ptmin, ptmax);
	43	spec_subtracted->GetYaxis()->SetTitleOffset(1.1);
	44	spec_subtracted->SetTitle();
	45	spec_subtracted->SetStats(kFALSE);
	46	spec_subtracted->SetLineColor(kRed);
	47	spec_subtracted->SetLineWidth(1);
	48	spec_subtracted->SetMarkerColor(kRed);
	49	spec_subtracted->SetMarkerStyle(22);
	50
	51	TLegend *leg = new TLegend(0.2, 0.25, 0.4, 0.35);
	52	leg->SetBorderSize(0);
	53	leg->SetFillStyle(0);
	54	leg->AddEntry(spec, "Raw Spectrum", "p");
	55	leg->AddEntry(spec_subtracted, "Spectrum after background subtraction", "p");
	56
	57	TCanvas *c1 = new TCanvas("cspec", "Single-inclusive electron spectrum", 1200, 750);
	58	c1->cd();
	59	c1->SetLogy();
	60	c1->SetGridx(kFALSE);
	61	c1->SetGridy(kFALSE);
	62	spec->Draw("ep");
	63	spec_subtracted->Draw("epsame");
	64	leg->Draw();
65	ALICEWorkInProgress(c1, "today");
66
67	// PID
68	TList pidqa = (TList )qa->FindObject("HFEpidQA");
69	AliHFEtpcPIDqa tpcqa = (AliHFEtpcPIDqa )pidqa->FindObject("TPCQA");
70	AliHFEtofPIDqa tofqa = (AliHFEtofPIDqa )pidqa->FindObject("TOFQA");
71
72	// Make Plots for TPC
73	// Create histograms by projecting the THnSparse
74	TH2 *hTPCall = tpcqa->MakeSpectrumdEdx(AliHFEdetPIDqa::kBeforePID);
75	TH2 *hTPCselected = tpcqa->MakeSpectrumdEdx(AliHFEdetPIDqa::kAfterPID);
76	TH2 *hTPCsigmaAll = tpcqa->MakeSpectrumNSigma(AliHFEdetPIDqa::kBeforePID);
77	TH2* hTPCsigmaSelected = tpcqa->MakeSpectrumNSigma(AliHFEdetPIDqa::kAfterPID);
78	// Make Plots for TOF
79	TH2 *hTOFsigmaAll = tofqa->MakeSpectrumNSigma(AliHFEdetPIDqa::kBeforePID);
80	TH2 *hTOFsigmaSelected = tofqa->MakeSpectrumNSigma(AliHFEdetPIDqa::kAfterPID);
81
82	hTPCsigmaAll->SetTitle("TPC n#sigma around the electron line");
83	hTPCsigmaSelected->SetTitle("TPC n#sigma around the electron line for selected tracks");
84	hTOFsigmaAll->SetTitle("TOF n#sigma around the electron line");
85	hTOFsigmaSelected->SetTitle("TOF n#sigma around the electron line for selected tracks");
86	DefineTPChisto(hTPCall, "TPC Signal / a.u");
87	DefineTPChisto(hTPCselected, "TPC Signal / a.u.");
88	DefineTPChisto(hTPCsigmaAll, "TPC Sigma");
89	DefineTPChisto(hTPCsigmaSelected, "TPC Sigma");
90
91	// Also make nice histograms for TOF
92	DefineTPChisto(hTOFsigmaAll, "TOF Sigma");
93	DefineTPChisto(hTOFsigmaSelected, "TOF Sigma");
94
95	// Plot them
96	TCanvas *c2 = new TCanvas("cTPCall", "TPC Signal for all tracks", 640, 480);
97	c2->cd();
98	c2->SetGridx(kFALSE);
99	c2->SetGridy(kFALSE);
100	c2->SetLogx();
101	c2->SetLogz();
102	hTPCall->GetYaxis()->SetRangeUser(40., 100.);
103	hTPCall->Draw("colz");
104	ALICEWorkInProgress(c2, "today");
105
106	TCanvas *c3 = new TCanvas("cTPCsel", "TPC Signal for selected tracks", 640, 480);
107	c3->cd();
108	c3->SetGridx(kFALSE);
109	c3->SetGridy(kFALSE);
110	c3->SetLogx();
111	c3->SetLogz();
112	hTPCselected->GetYaxis()->SetRangeUser(40., 100.);
113	hTPCselected->Draw("colz");
114	ALICEWorkInProgress(c3, "today");
115
116	TCanvas *c4 = new TCanvas("cTPCsigAll", "TPC Sigma for all tracks", 640, 480);
117	c4->cd();
118	c4->SetGridx(kFALSE);
119	c4->SetGridy(kFALSE);
120	c4->SetLogx();
121	c4->SetLogz();
122	//hTPCsigmaAll->GetYaxis()->SetRangeUser(-3.5, 5.);
123	hTPCsigmaAll->Draw("colz");
124	ALICEWorkInProgress(c4, "today");
125
126	TCanvas *c5 = new TCanvas("cTPCsigSel", "TPC Sigma for selected tracks", 640, 480);
127	c5->cd();
128	c5->SetGridx(kFALSE);
129	c5->SetGridy(kFALSE);
130	c5->SetLogx();
131	c5->SetLogz();
132	hTPCsigmaSelected->GetYaxis()->SetRangeUser(-3.5, 5.);
133	hTPCsigmaSelected->Draw("colz");
134	ALICEWorkInProgress(c5, "today");
135
136	TCanvas *c6 = new TCanvas("cTOFsigAll", "TOF Sigma for all tracks", 640, 480);
137	c6->cd();
138	c6->SetGridx(kFALSE);
139	c6->SetGridy(kFALSE);
140	c6->SetLogx();
141	c6->SetLogz();
142	hTOFsigmaAll->Draw("colz");
143	ALICEWorkInProgress(c6, "today");
144
145	TCanvas *c7 = new TCanvas("cTOFsigSel", "TOF Sigma for selected tracks", 640, 480);
146	c7->cd();
147	c7->SetGridx(kFALSE);
148	c7->SetGridy(kFALSE);
149	c7->SetLogx();
150	c7->SetLogz();
151	//hTOFsigmaSelected->GetYaxis()->SetRangeUser(-3, 3);
152	hTOFsigmaSelected->Draw("colz");
153	ALICEWorkInProgress(c7, "today");
154
155	TFile *output = new TFile("Performance.root", "RECREATE");
156	output->cd();
157	spec->Write();
158	hTPCall->Write();
159	hTPCselected->Write();
160	hTPCsigmaAll->Write();
161	hTPCsigmaSelected->Write();
162	c1->Write();
163	c2->Write();
164	c3->Write();
165	c4->Write();
166	c5->Write();
167	c6->Write();
168	c7->Write();
169	output->Close();
170	delete output;
171	}
172
173	void DefineTPChisto(TH2 h, const char yaxis){
174	h->SetStats(kFALSE);
175	h->GetXaxis()->SetRangeUser(ptmin, ptmax);
176	h->GetXaxis()->SetMoreLogLabels();
177	h->GetXaxis()->SetNoExponent();
178	h->GetXaxis()->SetTitle("p / GeV/c");
179	h->GetYaxis()->SetTitle(yaxis);
180	}
181
182	void ALICEWorkInProgress(TCanvas *c,TString today){
183	//date must be in the form: 04/05/2010
184	if(today=="today"){
185	TDatime startt;
186	int date=startt.GetDate();
187
188	int y=date/10000;
189	int m=(date%10000)/100;
190	int d=date%100;
191
192
193	today="";
194	today+=d;
195	if(m<10)
196	today.Append("/0");
197	else today.Append("/");
198	today+=m;
199	today.Append("/");
200	today+=y;
201
202	}
203	TPad *myPadLogo = new TPad("myPadLogo", "Pad for ALICE Logo",0.67,0.65,0.82,0.89);
204	// myPadLogo->SetFillColor(2);
205	myPadLogo->SetBorderMode(0);
206	myPadLogo->SetBorderSize(2);
207	myPadLogo->SetFrameBorderMode(0);
208	myPadLogo->SetLeftMargin(0.0);
209	myPadLogo->SetTopMargin(0.0);
210	myPadLogo->SetBottomMargin(0.0);
211	myPadLogo->SetRightMargin(0.0);
212	myPadLogo->Draw();
213	myPadLogo->cd();
214	TASImage *myAliceLogo = new TASImage("/u/mfasel/work/electron/Spectrum/alice_logo.png");
215	myAliceLogo->Draw();
216	c->cd();
217	TPaveText* t1=new TPaveText(0.59,0.59,0.89,0.66,"NDC");
218	t1->SetFillStyle(0);
219	t1->SetBorderSize(0);
220	t1->AddText(0.,0.,"ALICE Performance");
221	t1->SetTextColor(kRed);
222	t1->SetTextFont(42);
223	t1->Draw();
224	TPaveText* t2=new TPaveText(0.59,0.54,0.89,0.60,"NDC");
225	t2->SetFillStyle(0);
226	t2->SetBorderSize(0);
227	t2->SetTextColor(kRed);
228	t2->SetTextFont(52);
229	t2->AddText(0.,0.,today.Data());
230	t2->Draw();
231	}
232