3 #if !defined(__CINT__) || defined(__MAKECINT__)
5 #include "AliPWG0Helper.h"
6 #include "dNdEtaAnalysis.h"
7 #include "AlidNdEtaCorrection.h"
21 extern TSystem* gSystem;
25 gSystem->Load("libANALYSIS");
26 gSystem->Load("libPWG0base");
29 void SetRanges(TAxis* axis)
31 if (strcmp(axis->GetTitle(), "#eta") == 0)
32 axis->SetRangeUser(-1.7999, 1.7999);
33 if (strcmp(axis->GetTitle(), "p_{T} [GeV/c]") == 0)
34 axis->SetRangeUser(0, 9.9999);
35 if (strcmp(axis->GetTitle(), "vtx z [cm]") == 0)
36 axis->SetRangeUser(-15, 14.9999);
37 if (strcmp(axis->GetTitle(), "Ntracks") == 0)
38 axis->SetRangeUser(0, 99.9999);
41 void SetRanges(TH1* hist)
43 SetRanges(hist->GetXaxis());
44 SetRanges(hist->GetYaxis());
45 SetRanges(hist->GetZaxis());
49 void Prepare3DPlot(TH3* hist)
51 hist->GetXaxis()->SetTitleOffset(1.5);
52 hist->GetYaxis()->SetTitleOffset(1.5);
53 hist->GetZaxis()->SetTitleOffset(1.5);
55 hist->SetStats(kFALSE);
58 void Prepare2DPlot(TH2* hist)
60 hist->SetStats(kFALSE);
61 hist->GetYaxis()->SetTitleOffset(1.4);
64 hist->SetMaximum(gMax);
69 void Prepare1DPlot(TH1* hist)
71 hist->SetLineWidth(2);
72 hist->SetStats(kFALSE);
74 hist->GetXaxis()->SetLabelOffset(0.02);
75 hist->GetXaxis()->SetTitleOffset(1.3);
76 hist->GetYaxis()->SetTitleOffset(1.3);
83 gPad->Range(0, 0, 1, 1);
84 gPad->SetLeftMargin(0.15);
85 //gPad->SetRightMargin(0.05);
86 //gPad->SetTopMargin(0.13);
87 gPad->SetBottomMargin(0.12);
95 gPad->Range(0, 0, 1, 1);
96 gPad->SetRightMargin(0.15);
97 gPad->SetLeftMargin(0.12);
103 // --- end of helpers --- begin functions ---
105 void DrawOverview(const char* fileName = "correction_map.root", const char* dirName = "dndeta_correction")
108 if (!TFile::Open(fileName))
111 AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection(dirName, dirName);
112 if (!dNdEtaCorrection->LoadHistograms())
115 dNdEtaCorrection->Finish();
117 dNdEtaCorrection->DrawOverview();
120 void ComparedNdEta(const char* ESDfolder = "dndeta", const char* MCfolder = "dndeta", const char* esdFile = "analysis_esd.root", const char* mcFile = "analysis_mc.root")
122 gSystem->Load("libPWG0base");
124 TFile::Open(esdFile);
125 dNdEtaAnalysis* fdNdEtaAnalysisESD = new dNdEtaAnalysis(ESDfolder, ESDfolder);
126 fdNdEtaAnalysisESD->LoadHistograms();
129 dNdEtaAnalysis* fdNdEtaAnalysisMC = new dNdEtaAnalysis(MCfolder, MCfolder);
130 fdNdEtaAnalysisMC->LoadHistograms();
131 //fdNdEtaAnalysisMC->Finish(0, 0.3, AlidNdEtaCorrection::kNone);
133 for (Int_t i=0; i<dNdEtaAnalysis::kVertexBinning; ++i)
134 fdNdEtaAnalysisESD->GetdNdEtaPtCutOffCorrectedHistogram(i)->Divide(fdNdEtaAnalysisMC->GetdNdEtaPtCutOffCorrectedHistogram(i));
136 fdNdEtaAnalysisESD->DrawHistograms();
139 void CompareVertexDist(Int_t plot = 0, const char* correctionMapFile = "correction_map.root", const char* correctionMapFolder = "dndeta_correction")
141 gSystem->Load("libPWG0base");
143 const char* ESDfolder = 0;
145 if (plot == 0) // all
146 ESDfolder = "dndeta";
147 else if (plot == 1) // mb
148 ESDfolder = "dndeta_mb";
149 else if (plot == 2) // mb vtx
150 ESDfolder = "dndeta_mbvtx";
152 TFile::Open("analysis_esd.root");
153 dNdEtaAnalysis* fdNdEtaAnalysisESD = new dNdEtaAnalysis(ESDfolder, ESDfolder);
154 fdNdEtaAnalysisESD->LoadHistograms();
156 AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection(correctionMapFolder, correctionMapFolder);
157 dNdEtaCorrection->LoadHistograms(correctionMapFile, correctionMapFolder);
161 if (plot == 0) // all
162 hist = dNdEtaCorrection->GetTriggerBiasCorrection()->GetGeneratedHistogram();
163 else if (plot == 1) // mb
164 hist = dNdEtaCorrection->GetTriggerBiasCorrection()->GetMeasuredHistogram();
165 else if (plot == 2) // mb vtx
166 hist = dNdEtaCorrection->GetVertexRecoCorrection()->GetMeasuredHistogram();
168 TH1* proj = hist->ProjectionX();
170 TH1* vertex = fdNdEtaAnalysisESD->GetVtxHistogram();
171 for (Int_t i=1; i<=vertex->GetNbinsX(); ++i)
173 Float_t value = proj->GetBinContent(proj->FindBin(vertex->GetBinCenter(i)));
176 printf("vtx = %f, esd = %f, corr = %f, ratio = %f\n", vertex->GetBinCenter(i), vertex->GetBinContent(i), value, vertex->GetBinContent(i) / value);
177 vertex->SetBinContent(i, vertex->GetBinContent(i) / value);
185 void CompareTrack2ParticleWithAnalysisData(const char* correctionMapFile = "correction_map.root", const char* correctionMapFolder = "dndeta_correction")
187 gSystem->Load("libPWG0base");
189 TFile::Open("analysis_esd.root");
190 dNdEtaAnalysis* fdNdEtaAnalysisESD = new dNdEtaAnalysis("dndeta_mbvtx", "dndeta_mbvtx");
191 fdNdEtaAnalysisESD->LoadHistograms();
193 AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection(correctionMapFolder, correctionMapFolder);
194 dNdEtaCorrection->LoadHistograms(correctionMapFile, correctionMapFolder);
196 //TH1* histESD = fdNdEtaAnalysisESD->GetUncorrectedHistogram();
197 //TH1* histCorr = dNdEtaCorrection->GetTrack2ParticleCorrection()->GetMeasuredHistogram();
199 TH1* histESD = fdNdEtaAnalysisESD->GetHistogram();
200 TH1* histCorr = dNdEtaCorrection->GetTrack2ParticleCorrection()->GetGeneratedHistogram();
202 TH1F* diff = new TH1F("diff", "diff", 100, 0.98, 1.02);
210 for (Int_t x=1; x<=histESD->GetNbinsX(); ++x)
211 for (Int_t y=1; y<=histESD->GetNbinsY(); ++y)
212 for (Int_t z=1; z<=histESD->GetNbinsZ(); ++z)
214 Float_t value1 = histESD->GetBinContent(x, y, z);
215 Float_t value2 = histCorr->GetBinContent(histCorr->FindBin(histESD->GetXaxis()->GetBinCenter(x), histESD->GetYaxis()->GetBinCenter(y), histESD->GetZaxis()->GetBinCenter(z)));
217 if (value2 > 0 && value1 > 0)
219 printf("%f %f %f\n", value1, value2, value1 / value2);
220 diff->Fill(value1 / value2);
228 Double_t PrintIntegratedDeviation(TH1* histMC, TH1* histESD, const char* desc = "")
232 for (Int_t bin = histMC->FindBin(-0.7999); bin <= histMC->FindBin(0.7999); bin++)
234 avgMC += histMC->GetBinContent(bin);
235 avgESD += histESD->GetBinContent(bin);
237 Int_t nBins = histMC->FindBin(0.7999) - histMC->FindBin(-0.7999) + 1;
242 // deviation when integrate in |eta| < 0.8 between mc and esd
243 Double_t diffFullRange = (avgMC - avgESD) / avgMC;
245 Printf("%s: Integrated deviation in |eta| < 0.8 is %.2f %%", desc, diffFullRange * 100);
247 return diffFullRange;
250 void dNdEta(Bool_t onlyESD = kFALSE, Bool_t save = kTRUE)
252 TFile* file = TFile::Open("analysis_esd.root");
253 TH1* histESD = (TH1*) file->Get("dndeta/dNdEta_corrected");
254 TH1* histESDNoPt = (TH1*) file->Get("dndeta/dNdEta");
255 TH1* histESDMB = (TH1*) file->Get("dndetaTr/dNdEta_corrected");
256 TH1* histESDMBNoPt = (TH1*) file->Get("dndetaTr/dNdEta");
257 TH1* histESDMBVtx = (TH1*) file->Get("dndetaTrVtx/dNdEta_corrected");
258 TH1* histESDMBVtxNoPt = (TH1*) file->Get("dndetaTrVtx/dNdEta");
259 TH1* histESDMBTracksNoPt = (TH1*) file->Get("dndetaTracks/dNdEta");
261 Prepare1DPlot(histESD);
262 Prepare1DPlot(histESDMB);
263 Prepare1DPlot(histESDMBVtx);
265 Prepare1DPlot(histESDNoPt);
266 Prepare1DPlot(histESDMBNoPt);
267 Prepare1DPlot(histESDMBVtxNoPt);
268 Prepare1DPlot(histESDMBTracksNoPt);
270 histESD->SetLineWidth(0);
271 histESDMB->SetLineWidth(0);
272 histESDMBVtx->SetLineWidth(0);
274 histESDNoPt->SetLineWidth(0);
275 histESDMBNoPt->SetLineWidth(0);
276 histESDMBVtxNoPt->SetLineWidth(0);
278 histESD->SetMarkerColor(1);
279 histESDMB->SetMarkerColor(2);
280 histESDMBVtx->SetMarkerColor(3);
282 histESD->SetLineColor(1);
283 histESDMB->SetLineColor(2);
284 histESDMBVtx->SetLineColor(3);
286 histESDNoPt->SetMarkerColor(1);
287 histESDMBNoPt->SetMarkerColor(2);
288 histESDMBVtxNoPt->SetMarkerColor(3);
289 histESDMBTracksNoPt->SetMarkerColor(4);
291 histESD->SetMarkerStyle(20);
292 histESDMB->SetMarkerStyle(21);
293 histESDMBVtx->SetMarkerStyle(22);
295 histESDNoPt->SetMarkerStyle(20);
296 histESDMBNoPt->SetMarkerStyle(21);
297 histESDMBVtxNoPt->SetMarkerStyle(22);
298 histESDMBTracksNoPt->SetMarkerStyle(23);
300 TH2F* dummy = new TH2F("dummy", "", 100, -1.5, 1.5, 1000, 0, histESDMBVtx->GetMaximum() * 1.1);
301 Prepare1DPlot(dummy);
302 dummy->SetStats(kFALSE);
303 dummy->SetXTitle("#eta");
304 dummy->SetYTitle("dN_{ch}/d#eta");
305 dummy->GetYaxis()->SetTitleOffset(1);
307 Float_t etaLimit = 1.2999;
309 histESDMBVtx->GetXaxis()->SetRangeUser(-etaLimit, etaLimit);
310 histESDMB->GetXaxis()->SetRangeUser(-etaLimit, etaLimit);
311 histESD->GetXaxis()->SetRangeUser(-etaLimit, etaLimit);
313 histESDNoPt->GetXaxis()->SetRangeUser(-etaLimit, etaLimit);
314 histESDMBNoPt->GetXaxis()->SetRangeUser(-etaLimit, etaLimit);
315 histESDMBVtxNoPt->GetXaxis()->SetRangeUser(-etaLimit, etaLimit);
316 histESDMBTracksNoPt->GetXaxis()->SetRangeUser(-etaLimit, etaLimit);
318 TCanvas* canvas = new TCanvas("dNdEta1", "dNdEta1", 500, 500);
321 histESDMBVtx->Draw("SAME");
322 histESDMB->Draw("SAME");
323 histESD->Draw("SAME");
327 canvas->SaveAs("dNdEta1.gif");
328 canvas->SaveAs("dNdEta1.eps");
336 TFile* file2 = TFile::Open("analysis_mc.root");
337 TH1* histMC = (TH1*) file2->Get("dndeta/dNdEta_corrected")->Clone("cloned");
338 TH1* histMCTr = (TH1*) file2->Get("dndetaTr/dNdEta_corrected")->Clone("cloned2");
339 TH1* histMCTrVtx = (TH1*) file2->Get("dndetaTrVtx/dNdEta_corrected")->Clone("cloned3");
341 dNdEtaAnalysis* fdNdEtaAnalysis = new dNdEtaAnalysis("dndeta", "dndeta");
342 fdNdEtaAnalysis->LoadHistograms();
343 dNdEtaAnalysis* fdNdEtaAnalysis2 = (dNdEtaAnalysis*) fdNdEtaAnalysis->Clone();
345 fdNdEtaAnalysis->Finish(0, 0.3, AlidNdEtaCorrection::kNone);
346 TH1* histMCPtCut = (TH1*) fdNdEtaAnalysis->GetdNdEtaHistogram(0)->Clone("histMCPtCut");
348 fdNdEtaAnalysis = new dNdEtaAnalysis("dndetaTr", "dndetaTr");
349 fdNdEtaAnalysis->LoadHistograms();
350 fdNdEtaAnalysis->Finish(0, 0.3, AlidNdEtaCorrection::kNone);
351 TH1* histMCTrPtCut = fdNdEtaAnalysis->GetdNdEtaHistogram(0);
353 fdNdEtaAnalysis = new dNdEtaAnalysis("dndetaTrVtx", "dndetaTrVtx");
354 fdNdEtaAnalysis->LoadHistograms();
355 fdNdEtaAnalysis->Finish(0, 0.3, AlidNdEtaCorrection::kNone);
356 TH1* histMCTrVtxPtCut = fdNdEtaAnalysis->GetdNdEtaHistogram(0);
358 fdNdEtaAnalysis = new dNdEtaAnalysis("dndetaTracks", "dndetaTracks");
359 fdNdEtaAnalysis->LoadHistograms();
360 fdNdEtaAnalysis->Finish(0, 0.3, AlidNdEtaCorrection::kNone);
361 TH1* histMCTracksPtCut = fdNdEtaAnalysis->GetdNdEtaHistogram(0);
363 TCanvas* canvas2 = new TCanvas("dNdEta2", "dNdEta2", 500, 500);
365 Prepare1DPlot(histMC);
366 Prepare1DPlot(histMCTr);
367 Prepare1DPlot(histMCTrVtx);
369 Prepare1DPlot(histMCPtCut);
370 Prepare1DPlot(histMCTrPtCut);
371 Prepare1DPlot(histMCTrVtxPtCut);
372 if (histMCTracksPtCut)
373 Prepare1DPlot(histMCTracksPtCut);
375 histMC->SetLineColor(1);
376 histMCTr->SetLineColor(2);
377 histMCTrVtx->SetLineColor(3);
379 histMCPtCut->SetLineColor(1);
380 histMCTrPtCut->SetLineColor(2);
381 histMCTrVtxPtCut->SetLineColor(3);
382 if (histMCTracksPtCut)
383 histMCTracksPtCut->SetLineColor(4);
385 TH2* dummy2 = (TH2F*) dummy->Clone("dummy2");
386 Prepare1DPlot(dummy2);
387 dummy2->GetYaxis()->SetRangeUser(0, histESDMBVtx->GetMaximum() * 1.1);
390 histMC->Draw("SAME");
391 histMCTr->Draw("SAME");
392 histMCTrVtx->Draw("SAME");
393 histESD->Draw("SAME");
394 histESDMB->Draw("SAME");
395 histESDMBVtx->Draw("SAME");
396 histESDNoPt->Draw("SAME");
397 histESDMBNoPt->Draw("SAME");
398 histESDMBVtxNoPt->Draw("SAME");
399 histESDMBTracksNoPt->Draw("SAME");
400 histMCPtCut->Draw("SAME");
401 histMCTrPtCut->Draw("SAME");
402 histMCTrVtxPtCut->Draw("SAME");
403 if (histMCTracksPtCut)
404 histMCTracksPtCut->Draw("SAME");
408 canvas2->SaveAs("dNdEta2.gif");
409 canvas2->SaveAs("dNdEta2.eps");
412 TH1* ratio = (TH1*) histMC->Clone("ratio");
413 TH1* ratioNoPt = (TH1*) histMCPtCut->Clone("ratioNoPt");
415 ratio->Divide(histESD);
416 ratioNoPt->Divide(histESDNoPt);
418 ratio->GetXaxis()->SetRangeUser(-etaLimit, etaLimit);
420 ratio->SetLineColor(1);
421 ratioNoPt->SetLineColor(2);
423 Double_t average = 0; // average deviation from 1 in ratio (depends on the number of bins if statistical)
424 for (Int_t bin = ratio->FindBin(-0.7999); bin <= ratio->FindBin(0.7999); bin++)
425 average += TMath::Abs(ratio->GetBinContent(bin) - 1);
426 Int_t nBins = ratio->FindBin(0.7999) - ratio->FindBin(-0.7999) + 1;
428 Printf("Average deviation in |eta| < 0.8 is %.2f %%", average * 100);
430 PrintIntegratedDeviation(histMC, histESD, "all events");
431 PrintIntegratedDeviation(histMCTr, histESDMB, "triggered");
432 PrintIntegratedDeviation(histMCTrVtx, histESDMBVtx, "trigger, vertex");
433 PrintIntegratedDeviation(histMCPtCut, histESDNoPt, "all events (no pt corr)");
434 PrintIntegratedDeviation(histMCTrPtCut, histESDMBNoPt, "triggered (no pt corr)");
435 PrintIntegratedDeviation(histMCTrVtxPtCut, histESDMBVtxNoPt, "trigger, vertex (no pt corr)");
437 TCanvas* canvas3 = new TCanvas("dNdEta", "dNdEta", 700, 600);
438 canvas3->Range(0, 0, 1, 1);
439 //canvas3->Divide(1, 2, 0, 0);
442 TPad* pad1 = new TPad("dNdEta_1", "", 0, 0.5, 0.98, 0.98);
445 TPad* pad2 = new TPad("dNdEta_2", "", 0, 0.02, 0.98, 0.5);
448 pad1->SetRightMargin(0.05);
449 pad2->SetRightMargin(0.05);
451 // no border between them
452 pad1->SetBottomMargin(0);
453 pad2->SetTopMargin(0);
457 TLegend* legend = new TLegend(0.4, 0.05, 0.65, 0.3);
458 legend->SetFillColor(0);
459 legend->AddEntry(histESDMBVtx, "triggered, vertex");
460 legend->AddEntry(histESDMB, "triggered");
461 legend->AddEntry(histESD, "all events");
462 legend->AddEntry(histMC, "MC prediction");
464 dummy->GetXaxis()->SetLabelSize(0.06);
465 dummy->GetYaxis()->SetLabelSize(0.06);
466 dummy->GetXaxis()->SetTitleSize(0.06);
467 dummy->GetYaxis()->SetTitleSize(0.06);
468 dummy->GetYaxis()->SetTitleOffset(0.7);
470 histESDMBVtx->Draw("SAME");
471 histESDMB->Draw("SAME");
472 histESD->Draw("SAME");
473 histMC->Draw("SAME");
478 pad2->SetBottomMargin(0.15);
480 Float_t min = TMath::Min(0.961, ratio->GetMinimum() * 0.95);
481 Float_t max = TMath::Max(1.049, ratio->GetMaximum() * 1.05);
483 TH1F dummy3("dummy3", ";#eta;Ratio: MC / ESD", 1, -1.5, 1.5);
484 dummy3.SetStats(kFALSE);
485 dummy3.SetBinContent(1, 1);
486 dummy3.GetYaxis()->SetRangeUser(min, max);
487 dummy3.SetLineWidth(2);
488 dummy3.GetXaxis()->SetLabelSize(0.06);
489 dummy3.GetYaxis()->SetLabelSize(0.06);
490 dummy3.GetXaxis()->SetTitleSize(0.06);
491 dummy3.GetYaxis()->SetTitleSize(0.06);
492 dummy3.GetYaxis()->SetTitleOffset(0.7);
503 canvas3->SaveAs("dNdEta.gif");
504 canvas3->SaveAs("dNdEta.eps");
507 TCanvas* canvas4 = new TCanvas("ratio", "ratio", 700, 500);
510 ratioNoPt->Draw("SAME");
512 TLegend* legend = new TLegend(0.6, 0.7, 0.95, 0.9);
513 legend->SetFillColor(0);
514 legend->AddEntry(ratio, "mc/esd");
515 legend->AddEntry(ratioNoPt, "mc/esd, not pt cut off corrected");
521 TFile* file = TFile::Open("analysis_esd.root");
522 TH1* histESD = (TH1*) file->Get("dndeta/dndeta_pt");
524 TFile* file2 = TFile::Open("analysis_mc.root");
525 TH1* histMC = (TH1*) file2->Get("dndeta/dndeta_pt");
527 TCanvas* canvas = new TCanvas("ptSpectrum", "ptSpectrum", 500, 500);
531 Prepare1DPlot(histMC);
532 Prepare1DPlot(histESD);
534 histESD->SetTitle("");
535 histESD->GetXaxis()->SetTitle("p_{T} [GeV/c]");
536 histESD->GetYaxis()->SetTitle("#frac{dN}{d#eta dp_{T}} [c/GeV]");
538 histMC->SetLineColor(kBlue);
539 histESD->SetLineColor(kRed);
541 histESD->GetYaxis()->SetTitleOffset(1.5);
542 histESD->GetXaxis()->SetRangeUser(0, 4.9999);
544 histESD->SetMaximum(TMath::Max(histESD->GetMaximum(), histMC->GetMaximum()) * 2);
547 histMC->Draw("SAME");
549 canvas->SaveAs("ptSpectrum.gif");
550 canvas->SaveAs("ptSpectrum.eps");
555 gSystem->Load("libPWG0base");
557 TFile::Open("correction_map.root");
558 AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection("dndeta_correction", "dndeta_correction");
559 dNdEtaCorrection->LoadHistograms();
561 dNdEtaCorrection->GetMeasuredFraction(AlidNdEtaCorrection::kINEL, 0.3, -100, kTRUE);
563 TH1* hist = dynamic_cast<TH1*> (gROOT->FindObject("generated_pt")->Clone("ptcutoff"));
565 hist->GetXaxis()->SetRangeUser(0, 0.9999);
568 hist->SetTitle("Generated Particles");
571 TCanvas* canvas = new TCanvas("ptCutoff", "ptCutoff", 700, 500);
574 TLine* line = new TLine(0.3, 0 - hist->GetMaximum() * 0, 0.3, hist->GetMaximum() * 1.1);
575 line->SetLineWidth(3);
576 line->SetLineColor(kRed);
579 canvas->SaveAs("ptCutoff.gif");
580 canvas->SaveAs("ptCutoff.eps");
582 TH1F* factor = new TH1F("factor", ";#eta;correction factor", 20, -1, 1.000001);
583 factor->SetLineWidth(2);
584 for (Float_t eta = -0.95; eta<1; eta += 0.1)
585 factor->Fill(eta, 1.0 / dNdEtaCorrection->GetMeasuredFraction(AlidNdEtaCorrection::kINEL, 0.3, eta, kFALSE));
587 TCanvas* canvas = new TCanvas("ptCutoff_factor", "ptCutoff_factor", 700, 500);
590 Prepare1DPlot(factor);
591 factor->GetYaxis()->SetRangeUser(1, 2);
592 factor->GetYaxis()->SetTitleOffset(1);
595 canvas->SaveAs("ptCutoff_factor.eps");
598 void TriggerBiasVtxRecon(const char* fileName = "correction_map.root", const char* folder = "dndeta_correction")
600 gSystem->Load("libPWG0base");
602 TFile::Open(fileName);
603 AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection("dndeta_correction", "dndeta_correction");
604 dNdEtaCorrection->LoadHistograms();
606 TH2* corrTrigger = dNdEtaCorrection->GetTriggerBiasCorrectionINEL()->GetEventCorrection()->GetCorrectionHistogram();
607 TH2* corrVtx = dNdEtaCorrection->GetVertexRecoCorrection()->GetEventCorrection()->GetCorrectionHistogram();
609 Prepare2DPlot(corrTrigger);
610 corrTrigger->SetTitle("b) Trigger bias correction");
612 Prepare2DPlot(corrVtx);
613 corrVtx->SetTitle("a) Vertex reconstruction correction");
615 corrTrigger->GetYaxis()->SetTitle("Multiplicity");
616 corrVtx->GetYaxis()->SetTitle("Multiplicity");
618 TCanvas* canvas = new TCanvas("TriggerBiasVtxRecon", "TriggerBiasVtxRecon", 1000, 500);
619 canvas->Divide(2, 1);
623 corrVtx->DrawCopy("COLZ");
627 corrTrigger->DrawCopy("COLZ");
629 canvas->SaveAs(Form("TriggerBiasVtxRecon_%d.gif", gMax));
630 canvas->SaveAs(Form("TriggerBiasVtxRecon_%d.eps", gMax));
632 canvas = new TCanvas("TriggerBiasVtxReconZoom", "TriggerBiasVtxReconZoom", 1000, 500);
633 canvas->Divide(2, 1);
635 corrTrigger->GetYaxis()->SetRangeUser(0, 5);
636 corrVtx->GetYaxis()->SetRangeUser(0, 5);
640 corrVtx->DrawCopy("COLZ");
644 corrTrigger->DrawCopy("COLZ");
646 canvas->SaveAs(Form("TriggerBiasVtxReconZoom_%d.gif", gMax));
647 canvas->SaveAs(Form("TriggerBiasVtxReconZoom_%d.eps", gMax));
650 void TriggerBias(const char* fileName = "correction_map.root")
652 TFile* file = TFile::Open(fileName);
654 TH2* corr = dynamic_cast<TH2*> (file->Get("dndeta_correction/corr_dndeta_correction_trigger"));
657 corr->SetTitle("Trigger bias correction");
659 TCanvas* canvas = new TCanvas("TriggerBias", "TriggerBias", 500, 500);
661 corr->DrawCopy("COLZ");
663 canvas->SaveAs(Form("TriggerBias_%d.gif", gMax));
664 canvas->SaveAs(Form("TriggerBias_%d.eps", gMax));
666 corr->GetYaxis()->SetRangeUser(0, 5);
668 canvas = new TCanvas("TriggerBiasZoom", "TriggerBiasZoom", 500, 500);
670 corr->DrawCopy("COLZ");
672 canvas->SaveAs(Form("TriggerBiasZoom_%d.gif", gMax));
673 canvas->SaveAs(Form("TriggerBiasZoom_%d.eps", gMax));
676 void TriggerBias1D(const char* fileName = "correction_map.root", const char* folderName = "dndeta_correction")
678 gSystem->Load("libPWG0base");
680 TFile* file = TFile::Open(fileName);
681 AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection(folderName, folderName);
682 dNdEtaCorrection->LoadHistograms();
684 TH1* hist = dNdEtaCorrection->GetTriggerBiasCorrectionINEL()->GetEventCorrection()->Get1DCorrection("x");
685 TH1* hist2 = dNdEtaCorrection->GetTriggerBiasCorrectionINEL()->GetEventCorrection()->Get1DCorrection("y", -10, 10);
687 TCanvas* canvas = new TCanvas("TriggerBias1D", "TriggerBias1D", 1000, 500);
688 canvas->Divide(2, 1);
695 hist->GetYaxis()->SetTitle("correction factor");
696 hist->GetYaxis()->SetRangeUser(1, 1.5);
697 hist->GetYaxis()->SetTitleOffset(1.6);
703 Prepare1DPlot(hist2);
705 hist2->GetYaxis()->SetTitle("correction factor");
706 hist2->GetXaxis()->SetRangeUser(0, 5);
707 hist2->GetYaxis()->SetTitleOffset(1.6);
708 hist2->GetXaxis()->SetTitle("multiplicity");
711 TPaveText* pave = new TPaveText(0.6, 0.8, 0.8, 0.85, "NDC");
712 pave->SetFillColor(0);
713 pave->AddText("|z| < 10 cm");
716 canvas->SaveAs("TriggerBias1D.eps");
721 TFile* file = TFile::Open("correction_map.root");
723 TH2* corr = dynamic_cast<TH2*> (file->Get("dndeta_correction/corr_dndeta_correction_vtxReco"));
726 corr->SetTitle("Vertex reconstruction correction");
728 TCanvas* canvas = new TCanvas("VtxRecon", "VtxRecon", 500, 500);
730 corr->DrawCopy("COLZ");
732 canvas->SaveAs(Form("VtxRecon_%d.eps", gMax));
733 canvas->SaveAs(Form("VtxRecon_%d.eps", gMax));
735 corr->GetYaxis()->SetRangeUser(0, 5);
737 canvas = new TCanvas("VtxReconZoom", "VtxReconZoom", 500, 500);
739 corr->DrawCopy("COLZ");
741 canvas->SaveAs(Form("VtxReconZoom_%d.gif", gMax));
742 canvas->SaveAs(Form("VtxReconZoom_%d.eps", gMax));
745 void VtxRecon1D(const char* fileName = "correction_map.root", const char* folderName = "dndeta_correction")
747 gSystem->Load("libPWG0base");
749 TFile* file = TFile::Open(fileName);
750 AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection(folderName, folderName);
751 dNdEtaCorrection->LoadHistograms();
753 TH1* hist = dNdEtaCorrection->GetVertexRecoCorrection()->GetEventCorrection()->Get1DCorrection("x");
754 TH1* hist2 = dNdEtaCorrection->GetVertexRecoCorrection()->GetEventCorrection()->Get1DCorrection("y", -10, 10);
756 TCanvas* canvas = new TCanvas("VtxRecon1D", "VtxRecon1D", 1000, 500);
757 canvas->Divide(2, 1);
764 hist->GetYaxis()->SetTitle("correction factor");
765 hist->GetYaxis()->SetRangeUser(1, 1.8);
766 hist->GetYaxis()->SetTitleOffset(1.6);
772 Prepare1DPlot(hist2);
774 hist2->GetYaxis()->SetTitle("correction factor");
775 hist2->GetXaxis()->SetRangeUser(0, 20);
776 hist2->GetYaxis()->SetTitleOffset(1.6);
777 hist2->GetXaxis()->SetTitle("multiplicity");
780 TPaveText* pave = new TPaveText(0.6, 0.8, 0.8, 0.85, "NDC");
781 pave->SetFillColor(0);
782 pave->AddText("|z| < 10 cm");
785 canvas->SaveAs("VtxRecon1D.eps");
787 Correction1DCreatePlots(fileName, folderName, 9.9, 2);
789 TH1* corrX = dynamic_cast<TH1*> (gROOT->FindObject("generated_x_div_measured_x"));
790 TH1* corrZ = dynamic_cast<TH1*> (gROOT->FindObject("generated_z_div_measured_z"));
792 Prepare1DPlot(corrX);
793 Prepare1DPlot(corrZ);
795 corrX->GetYaxis()->SetTitleOffset(1.5);
796 corrZ->GetYaxis()->SetTitleOffset(1.5);
798 corrX->SetTitle("a) z projection");
799 corrZ->SetTitle("b) p_{T} projection");
801 corrX->GetYaxis()->SetTitle("correction factor");
802 corrZ->GetYaxis()->SetTitle("correction factor");
804 corrZ->GetXaxis()->SetRangeUser(0.11, 9.9);
807 canvasName.Form("VtxRecon1D_Track");
808 TCanvas* canvas = new TCanvas(canvasName, canvasName, 800, 400);
809 canvas->Divide(2, 1);
820 canvas->SaveAs("VtxRecon1D_Track.eps");
821 canvas->SaveAs("VtxRecon1D_Track.gif");
824 void Track2ParticleAsNumber(const char* fileName = "correction_map.root")
826 gSystem->Load("libPWG0base");
828 TFile::Open(fileName);
829 AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection("dndeta_correction", "dndeta_correction");
830 dNdEtaCorrection->LoadHistograms(fileName, "dndeta_correction");
832 TH3F* gene = dNdEtaCorrection->GetTrack2ParticleCorrection()->GetGeneratedHistogram();
833 TH3F* meas = dNdEtaCorrection->GetTrack2ParticleCorrection()->GetMeasuredHistogram();
835 TH3F* gene = dNdEtaCorrection->GetTrack2ParticleCorrection()->GetGeneratedHistogram();
836 TH3F* meas = dNdEtaCorrection->GetTrack2ParticleCorrection()->GetMeasuredHistogram();
838 gene->GetYaxis()->SetRangeUser(-0.8, 0.8);
839 meas->GetYaxis()->SetRangeUser(-0.8, 0.8);
840 gene->GetXaxis()->SetRangeUser(-10, 10);
841 meas->GetXaxis()->SetRangeUser(-10, 10);
843 Float_t eff1 = gene->Integral() / meas->Integral();
844 Float_t error1 = TMath::Sqrt(gene->Integral()) / meas->Integral();
846 printf("Correction without pT cut: %f +- %f\n", eff1, error1);
848 gene->GetZaxis()->SetRangeUser(0.3, 10);
849 meas->GetZaxis()->SetRangeUser(0.3, 10);
851 Float_t eff2 = gene->Integral() / meas->Integral();
852 Float_t error2 = TMath::Sqrt(gene->Integral()) / meas->Integral();
854 printf("Correction with pT cut: %f +- %f\n", eff2, error2);
856 gene->GetZaxis()->SetRangeUser(0.3, 1);
857 meas->GetZaxis()->SetRangeUser(0.3, 1);
859 Float_t eff3 = gene->Integral() / meas->Integral();
860 Float_t error3 = TMath::Sqrt(gene->Integral()) / meas->Integral();
862 printf("Correction with 0.3 < pT < 0.5: %f +- %f\n", eff3, error3);
865 void Correction1DCreatePlots(const char* fileName = "correction_map.root", const char* folderName = "dndeta_correction", Float_t upperPtLimit = 9.9, Int_t correctionType = 0)
867 TFile::Open(fileName);
868 AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection(folderName, folderName);
869 dNdEtaCorrection->LoadHistograms();
871 TH3F* gene = dNdEtaCorrection->GetCorrection(correctionType)->GetTrackCorrection()->GetGeneratedHistogram();
872 TH3F* meas = dNdEtaCorrection->GetCorrection(correctionType)->GetTrackCorrection()->GetMeasuredHistogram();
874 gene->GetZaxis()->SetRangeUser(0.3, upperPtLimit);
875 meas->GetZaxis()->SetRangeUser(0.3, upperPtLimit);
876 gene->GetYaxis()->SetRangeUser(-0.8, 0.8);
877 meas->GetYaxis()->SetRangeUser(-0.8, 0.8);
878 AliPWG0Helper::CreateDividedProjections(gene, meas, "x", kTRUE);
879 gene->GetYaxis()->SetRange(0, 0);
880 meas->GetYaxis()->SetRange(0, 0);
882 gene->GetXaxis()->SetRangeUser(-10, 10);
883 meas->GetXaxis()->SetRangeUser(-10, 10);
884 AliPWG0Helper::CreateDividedProjections(gene, meas, "y", kTRUE);
885 gene->GetZaxis()->SetRange(0, 0);
886 meas->GetZaxis()->SetRange(0, 0);
888 gene->GetYaxis()->SetRangeUser(-0.8, 0.8);
889 meas->GetYaxis()->SetRangeUser(-0.8, 0.8);
890 AliPWG0Helper::CreateDividedProjections(gene, meas, "z", kTRUE);
893 void Correction1D(Int_t correctionType = 0, const char* fileName = "correction_map.root", const char* folder = "dndeta_correction", Float_t upperPtLimit = 9.9)
895 gSystem->Load("libPWG0base");
897 Correction1DCreatePlots(fileName, folder, upperPtLimit, correctionType);
899 TH1* corrX = dynamic_cast<TH1*> (gROOT->FindObject(Form("generated_x_div_measured_x", folder, folder)));
900 TH1* corrY = dynamic_cast<TH1*> (gROOT->FindObject(Form("generated_y_div_measured_y", folder, folder)));
901 TH1* corrZ = dynamic_cast<TH1*> (gROOT->FindObject(Form("generated_z_div_measured_z", folder, folder)));
903 Prepare1DPlot(corrX);
904 Prepare1DPlot(corrY);
905 Prepare1DPlot(corrZ);
907 corrX->SetTitle("a) z projection");
908 corrY->SetTitle("b) #eta projection");
909 corrZ->SetTitle("c) p_{T} projection");
911 corrX->GetYaxis()->SetTitle("correction factor");
912 corrY->GetYaxis()->SetTitle("correction factor");
913 corrZ->GetYaxis()->SetTitle("correction factor");
915 corrZ->GetXaxis()->SetRangeUser(0, upperPtLimit);
918 canvasName.Form("Correction1D_%s", folder);
919 TCanvas* canvas = new TCanvas(canvasName, canvasName, 1200, 400);
920 canvas->Divide(3, 1);
934 canvas->SaveAs(Form("Correction1D_%d_%s_%f.gif", correctionType, fileName, upperPtLimit));
935 canvas->SaveAs(Form("Correction1D_%d_%s_%f.eps", correctionType, fileName, upperPtLimit));
938 void Track2Particle1D(const char* fileName = "correction_map.root", const char* folder = "dndeta_correction", Float_t upperPtLimit = 9.9)
940 gSystem->Load("libPWG0base");
942 Correction1DCreatePlots(fileName, folder, upperPtLimit, AlidNdEtaCorrection::kTrack2Particle);
944 TH1* corrX = dynamic_cast<TH1*> (gROOT->FindObject(Form("generated_x_div_measured_x", folder, folder)));
945 TH1* corrY = dynamic_cast<TH1*> (gROOT->FindObject(Form("generated_y_div_measured_y", folder, folder)));
946 TH1* corrZ = dynamic_cast<TH1*> (gROOT->FindObject(Form("generated_z_div_measured_z", folder, folder)));
948 Prepare1DPlot(corrX);
949 Prepare1DPlot(corrY);
950 Prepare1DPlot(corrZ);
952 corrX->SetTitle("a) z projection");
953 corrY->SetTitle("a) #eta projection");
954 corrZ->SetTitle("b) p_{T} projection");
956 corrY->GetYaxis()->SetTitle("correction factor");
957 corrZ->GetYaxis()->SetTitle("correction factor");
959 corrZ->GetXaxis()->SetRangeUser(0, upperPtLimit);
962 canvasName.Form("Track2Particle1D_%s", folder);
963 TCanvas* canvas = new TCanvas(canvasName, canvasName, 1200, 400);
964 canvas->Divide(3, 1);
978 canvas->SaveAs(Form("Track2Particle1D_%s_%f.gif", fileName, upperPtLimit));
979 canvas->SaveAs(Form("Track2Particle1D_%s_%f.eps", fileName, upperPtLimit));
981 //TPaveText* pave = new TPaveText(-0.4, 1.35, 0.4, 1.45);
983 canvasName.Form("Track2Particle1D_%s_etapt", folder);
984 TCanvas* canvas = new TCanvas(canvasName, canvasName, 1000, 500);
985 canvas->Divide(2, 1);
989 corrY->GetXaxis()->SetRangeUser(-0.99, 0.99);
990 corrY->GetYaxis()->SetRangeUser(1, 1.5);
991 corrY->GetYaxis()->SetTitleOffset(1.5);
993 TPaveText* pave = new TPaveText(0.3, 0.7, 0.7, 0.8, "NDC");
994 pave->AddText("|z| < 10 cm");
995 pave->AddText("0.3 GeV/c < p_{T} < 10 GeV/c");
1001 corrZ->GetYaxis()->SetRangeUser(1, 2.5);
1002 corrZ->GetXaxis()->SetRangeUser(0.101, upperPtLimit);
1003 corrZ->GetYaxis()->SetTitleOffset(1.5);
1005 pave = new TPaveText(0.5, 0.7, 0.8, 0.8, "NDC");
1006 pave->AddText("|z| < 10 cm");
1007 pave->AddText("|#eta| < 0.8");
1010 canvas->SaveAs(Form("Track2Particle1D_etapt_%s_%f.eps", fileName, upperPtLimit));
1011 canvas->SaveAs(Form("Track2Particle1D_etapt_%s_%f.gif", fileName, upperPtLimit));
1014 void CompareTrack2Particle1D(Float_t upperPtLimit = 9.9)
1016 gSystem->Load("libPWG0base");
1019 for (Int_t particle=0; particle<4; ++particle)
1022 dirName.Form("correction_%d", particle);
1023 Track2Particle1DCreatePlots("systematics-detail-only-positive.root", dirName, upperPtLimit);
1025 TString tmpx, tmpy, tmpz;
1026 tmpx.Form("gene_%s_nTrackToNPart_x_div_meas_%s_nTrackToNPart_x", dirName.Data(), dirName.Data());
1027 tmpy.Form("gene_%s_nTrackToNPart_y_div_meas_%s_nTrackToNPart_y", dirName.Data(), dirName.Data());
1028 tmpz.Form("gene_%s_nTrackToNPart_z_div_meas_%s_nTrackToNPart_z", dirName.Data(), dirName.Data());
1030 TH1* posX = dynamic_cast<TH1*> (gROOT->FindObject(tmpx)->Clone("pos_x"));
1031 TH1* posY = dynamic_cast<TH1*> (gROOT->FindObject(tmpy)->Clone("pos_y"));
1032 TH1* posZ = dynamic_cast<TH1*> (gROOT->FindObject(tmpz)->Clone("pos_z"));
1034 Track2Particle1DCreatePlots("systematics-detail-only-negative.root", dirName, upperPtLimit);
1036 TH1* negX = dynamic_cast<TH1*> (gROOT->FindObject(tmpx)->Clone("neg_x"));
1037 TH1* negY = dynamic_cast<TH1*> (gROOT->FindObject(tmpy)->Clone("neg_y"));
1038 TH1* negZ = dynamic_cast<TH1*> (gROOT->FindObject(tmpz)->Clone("neg_z"));
1044 Prepare1DPlot(posX);
1045 Prepare1DPlot(posY);
1046 Prepare1DPlot(posZ);
1051 posX->SetMinimum(min);
1052 posX->SetMaximum(max);
1053 posY->SetMinimum(min);
1054 posY->SetMaximum(max);
1055 posZ->SetMinimum(min);
1056 posZ->SetMaximum(max);
1058 posZ->GetXaxis()->SetRangeUser(0, upperPtLimit);
1060 posX->GetYaxis()->SetTitleOffset(1.7);
1061 posX->GetYaxis()->SetTitle("C_{+} / C_{-}");
1062 posY->GetYaxis()->SetTitleOffset(1.7);
1063 posY->GetYaxis()->SetTitle("C_{+} / C_{-}");
1064 posZ->GetYaxis()->SetTitleOffset(1.7);
1065 posZ->GetYaxis()->SetTitle("C_{+} / C_{-}");
1067 posZ->GetXaxis()->SetRangeUser(0, 1);
1070 canvasName.Form("PosNegRatios_%s_%f", ((particle == 0) ? "Pi" : ((particle == 1) ? "K" : ((particle == 2) ? "p" : "other"))), upperPtLimit);
1072 TCanvas* canvas = new TCanvas(canvasName, canvasName, 1200, 400);
1073 canvas->Divide(3, 1);
1087 canvas->SaveAs(Form("%s.gif", canvas->GetName()));
1088 canvas->SaveAs(Form("%s.eps", canvas->GetName()));
1092 void Track2Particle2DCreatePlots(const char* fileName = "correction_map.root")
1094 TFile::Open(fileName);
1095 AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection("dndeta_correction", "dndeta_correction");
1096 dNdEtaCorrection->LoadHistograms();
1098 TH3F* gene = dNdEtaCorrection->GetTrack2ParticleCorrection()->GetTrackCorrection()->GetGeneratedHistogram();
1099 TH3F* meas = dNdEtaCorrection->GetTrack2ParticleCorrection()->GetTrackCorrection()->GetMeasuredHistogram();
1101 gene->GetZaxis()->SetRangeUser(0.3, 10);
1102 meas->GetZaxis()->SetRangeUser(0.3, 10);
1103 AliPWG0Helper::CreateDividedProjections(gene, meas, "yx");
1104 gene->GetZaxis()->SetRange(0, 0);
1105 meas->GetZaxis()->SetRange(0, 0);
1107 gene->GetYaxis()->SetRangeUser(-0.8, 0.8);
1108 meas->GetYaxis()->SetRangeUser(-0.8, 0.8);
1109 AliPWG0Helper::CreateDividedProjections(gene, meas, "zx");
1110 gene->GetYaxis()->SetRange(0, 0);
1111 meas->GetYaxis()->SetRange(0, 0);
1113 gene->GetXaxis()->SetRangeUser(-10, 10);
1114 meas->GetXaxis()->SetRangeUser(-10, 10);
1115 AliPWG0Helper::CreateDividedProjections(gene, meas, "zy");
1116 gene->GetXaxis()->SetRange(0, 0);
1117 meas->GetXaxis()->SetRange(0, 0);
1120 void Track2Particle2D(const char* fileName = "correction_map.root", const char* folder = "dndeta_correction")
1122 gSystem->Load("libPWG0base");
1124 Track2Particle2DCreatePlots(fileName);
1126 TH2* corrYX = dynamic_cast<TH2*> (gROOT->FindObject("generated_yx_div_measured_yx"));
1127 TH2* corrZX = dynamic_cast<TH2*> (gROOT->FindObject("generated_zx_div_measured_zx"));
1128 TH2* corrZY = dynamic_cast<TH2*> (gROOT->FindObject("generated_zy_div_measured_zy"));
1130 /* this reads them from the file
1131 TH2* corrYX = dynamic_cast<TH2*> (file->Get("dndeta_correction/gene_nTrackToNPart_yx_div_meas_nTrackToNPart_yx"));
1132 TH2* corrZX = dynamic_cast<TH2*> (file->Get("dndeta_correction/gene_nTrackToNPart_zx_div_meas_nTrackToNPart_zx"));
1133 TH2* corrZY = dynamic_cast<TH2*> (file->Get("dndeta_correction/gene_nTrackToNPart_zy_div_meas_nTrackToNPart_zy"));*/
1135 Prepare2DPlot(corrYX);
1136 Prepare2DPlot(corrZX);
1137 Prepare2DPlot(corrZY);
1139 const char* title = "";
1140 corrYX->SetTitle(title);
1141 corrZX->SetTitle(title);
1142 corrZY->SetTitle(title);
1144 TCanvas* canvas = new TCanvas("Track2Particle2D", "Track2Particle2D", 1200, 400);
1145 canvas->Divide(3, 1);
1149 corrYX->Draw("COLZ");
1153 corrZX->Draw("COLZ");
1157 corrZY->Draw("COLZ");
1159 canvas->SaveAs(Form("Track2Particle2D_%d.gif", gMax));
1160 canvas->SaveAs(Form("Track2Particle2D_%d.eps", gMax));
1163 void CompareTrack2Particle2D()
1165 gSystem->Load("libPWG0base");
1167 Track2Particle2DCreatePlots("correction_maponly-positive.root");
1169 TH2* posYX = dynamic_cast<TH2*> (gROOT->FindObject("gene_nTrackToNPart_yx_div_meas_nTrackToNPart_yx")->Clone("pos_yx"));
1170 TH2* posZX = dynamic_cast<TH2*> (gROOT->FindObject("gene_nTrackToNPart_zx_div_meas_nTrackToNPart_zx")->Clone("pos_zx"));
1171 TH2* posZY = dynamic_cast<TH2*> (gROOT->FindObject("gene_nTrackToNPart_zy_div_meas_nTrackToNPart_zy")->Clone("pos_zy"));
1173 Track2Particle2DCreatePlots("correction_maponly-negative.root");
1175 TH2* negYX = dynamic_cast<TH2*> (gROOT->FindObject("gene_nTrackToNPart_yx_div_meas_nTrackToNPart_yx")->Clone("neg_yx"));
1176 TH2* negZX = dynamic_cast<TH2*> (gROOT->FindObject("gene_nTrackToNPart_zx_div_meas_nTrackToNPart_zx")->Clone("neg_zx"));
1177 TH2* negZY = dynamic_cast<TH2*> (gROOT->FindObject("gene_nTrackToNPart_zy_div_meas_nTrackToNPart_zy")->Clone("neg_zy"));
1179 posYX->Divide(negYX);
1180 posZX->Divide(negZX);
1181 posZY->Divide(negZY);
1183 Prepare2DPlot(posYX);
1184 Prepare2DPlot(posZX);
1185 Prepare2DPlot(posZY);
1190 posYX->SetMinimum(min);
1191 posYX->SetMaximum(max);
1192 posZX->SetMinimum(min);
1193 posZX->SetMaximum(max);
1194 posZY->SetMinimum(min);
1195 posZY->SetMaximum(max);
1197 TCanvas* canvas = new TCanvas("CompareTrack2Particle2D", "CompareTrack2Particle2D", 1200, 400);
1198 canvas->Divide(3, 1);
1202 posYX->Draw("COLZ");
1206 posZX->Draw("COLZ");
1210 posZY->Draw("COLZ");
1212 canvas->SaveAs("CompareTrack2Particle2D.gif");
1213 canvas->SaveAs("CompareTrack2Particle2D.eps");
1216 void Track2Particle3D()
1218 // get left margin proper
1220 TFile* file = TFile::Open("correction_map.root");
1222 TH3* corr = dynamic_cast<TH3*> (file->Get("dndeta_correction/corr_nTrackToNPart"));
1224 corr->SetTitle("Correction Factor");
1225 SetRanges(corr->GetZaxis());
1227 Prepare3DPlot(corr);
1229 TCanvas* canvas = new TCanvas("Track2Particle3D", "Track2Particle3D", 500, 500);
1230 canvas->SetTheta(29.428);
1231 canvas->SetPhi(16.5726);
1235 canvas->SaveAs("Track2Particle3D.gif");
1236 canvas->SaveAs("Track2Particle3D.eps");
1239 void Track2Particle3DAll()
1241 TFile* file = TFile::Open("correction_map.root");
1243 TH3* gene = dynamic_cast<TH3*> (file->Get("dndeta_correction/gene_nTrackToNPart"));
1244 TH3* meas = dynamic_cast<TH3*> (file->Get("dndeta_correction/meas_nTrackToNPart"));
1245 TH3* corr = dynamic_cast<TH3*> (file->Get("dndeta_correction/corr_nTrackToNPart"));
1247 gene->SetTitle("Generated Particles");
1248 meas->SetTitle("Measured Tracks");
1249 corr->SetTitle("Correction Factor");
1251 Prepare3DPlot(gene);
1252 Prepare3DPlot(meas);
1253 Prepare3DPlot(corr);
1255 TCanvas* canvas = new TCanvas("Track2Particle3DAll", "Track2Particle3DAll", 1200, 400);
1256 canvas->Divide(3, 1);
1268 canvas->SaveAs("Track2Particle3DAll.gif");
1269 canvas->SaveAs("Track2Particle3DAll.eps");
1272 void MultiplicityMC(Int_t xRangeMax = 50)
1274 TFile* file = TFile::Open("multiplicityMC.root");
1278 printf("multiplicityMC.root could not be opened.\n");
1282 TH1F* fMultiplicityESD = dynamic_cast<TH1F*> (file->Get("fMultiplicityESD"));
1283 TH1F* fMultiplicityMC = dynamic_cast<TH1F*> (file->Get("fMultiplicityMC"));
1284 TH2F* fCorrelation = dynamic_cast<TH2F*> (file->Get("fCorrelation"));
1286 TH1F* correction = new TH1F("MultiplicityMC_correction", "MultiplicityMC_correction;Ntracks;Npart", 76, -0.5, 75.5);
1287 TH1F* correctionWidth = new TH1F("MultiplicityMC_correctionwidth", "MultiplicityMC_correctionwidth;Ntracks;Npart", 76, -0.5, 75.5);
1288 //fMultiplicityMC->GetNbinsX(), fMultiplicityMC->GetXaxis()->GetXmin(), fMultiplicityMC->GetXaxis()->GetXmax());
1289 for (Int_t i=1; i<=correction->GetNbinsX(); ++i)
1291 TH1D* proj = fCorrelation->ProjectionX("_px", i, i+1);
1292 proj->Fit("gaus", "0");
1293 correction->SetBinContent(i, proj->GetFunction("gaus")->GetParameter(1));
1294 correctionWidth->SetBinContent(i, proj->GetFunction("gaus")->GetParameter(2));
1298 // draw for debugging
1301 proj->GetFunction("gaus")->DrawCopy("SAME");
1304 TH1F* fMultiplicityESDCorrected = new TH1F("fMultiplicityESDCorrected", "fMultiplicityESDCorrected", 2010, -0.5, 200.5);
1306 for (Int_t i=1; i<=correction->GetNbinsX(); ++i)
1308 Float_t mean = correction->GetBinContent(i);
1309 Float_t width = correctionWidth->GetBinContent(i);
1311 Int_t fillBegin = fMultiplicityESDCorrected->FindBin(mean - width * 3);
1312 Int_t fillEnd = fMultiplicityESDCorrected->FindBin(mean + width * 3);
1313 printf("bin %d mean %f width %f, filling from %d to %d\n", i, mean, width, fillBegin, fillEnd);
1315 for (Int_t j=fillBegin; j <= fillEnd; ++j)
1317 fMultiplicityESDCorrected->AddBinContent(j, TMath::Gaus(fMultiplicityESDCorrected->GetXaxis()->GetBinCenter(j), mean, width, kTRUE) * fMultiplicityESD->GetBinContent(i));
1321 TH1F* fMultiplicityESDCorrectedRebinned = dynamic_cast<TH1F*> (fMultiplicityESDCorrected->Clone("fMultiplicityESDCorrectedRebinned"));
1322 fMultiplicityESDCorrectedRebinned->Rebin(10);
1323 fMultiplicityESDCorrectedRebinned->Scale(0.1);
1325 TH1F* ratio = dynamic_cast<TH1F*> (fMultiplicityESD->Clone("multiplicity_ratio"));
1326 ratio->SetTitle("ratio;Ntracks;Nreco/Ngene");
1327 ratio->Divide(fMultiplicityMC);
1329 TH1F* ratio2 = dynamic_cast<TH1F*> (fMultiplicityESDCorrectedRebinned->Clone("multiplicity_ratio_corrected"));
1330 ratio2->Divide(fMultiplicityMC);
1332 TCanvas* canvas = new TCanvas("MultiplicityMC", "MultiplicityMC", 1500, 1000);
1333 canvas->Divide(3, 2);
1335 fMultiplicityESD->GetXaxis()->SetRangeUser(0, xRangeMax);
1336 ratio->GetXaxis()->SetRangeUser(0, xRangeMax);
1337 fCorrelation->GetXaxis()->SetRangeUser(0, xRangeMax);
1338 fCorrelation->GetYaxis()->SetRangeUser(0, xRangeMax);
1339 correction->GetXaxis()->SetRangeUser(0, xRangeMax);
1340 fMultiplicityESDCorrected->GetXaxis()->SetRangeUser(0, xRangeMax);
1341 fMultiplicityESDCorrectedRebinned->GetXaxis()->SetRangeUser(0, xRangeMax);
1343 canvas->cd(1); //InitPad();
1344 fMultiplicityESD->Draw();
1345 fMultiplicityMC->SetLineColor(2);
1346 fMultiplicityMC->Draw("SAME");
1348 TLegend* legend = new TLegend(0.6, 0.7, 0.85, 0.85);
1349 legend->AddEntry(fMultiplicityESD, "ESD");
1350 legend->AddEntry(fMultiplicityMC, "MC");
1354 fCorrelation->Draw("COLZ");
1358 //correction->Fit("pol1");
1359 correctionWidth->SetLineColor(2);
1360 correctionWidth->Draw("SAME");
1362 legend = new TLegend(0.2, 0.7, 0.45, 0.85);
1363 legend->AddEntry(correction, "#bar{x}");
1364 legend->AddEntry(correctionWidth, "#sigma");
1370 ratio2->SetLineColor(2);
1371 ratio2->Draw("SAME");
1373 legend = new TLegend(0.6, 0.7, 0.85, 0.85);
1374 legend->AddEntry(ratio, "uncorrected");
1375 legend->AddEntry(ratio2, "corrected");
1379 fMultiplicityESDCorrected->SetLineColor(kBlue);
1380 fMultiplicityESDCorrected->Draw();
1381 fMultiplicityMC->Draw("SAME");
1382 fMultiplicityESD->Draw("SAME");
1384 legend = new TLegend(0.6, 0.7, 0.85, 0.85);
1385 legend->AddEntry(fMultiplicityESDCorrected, "ESD corrected");
1386 legend->AddEntry(fMultiplicityMC, "MC");
1387 legend->AddEntry(fMultiplicityESD, "ESD");
1391 fMultiplicityESDCorrectedRebinned->SetLineColor(kBlue);
1392 fMultiplicityESDCorrectedRebinned->Draw();
1393 fMultiplicityMC->Draw("SAME");
1395 legend = new TLegend(0.6, 0.7, 0.85, 0.85);
1396 legend->AddEntry(fMultiplicityESDCorrectedRebinned, "ESD corrected");
1397 legend->AddEntry(fMultiplicityMC, "MC");
1400 canvas->SaveAs("MultiplicityMC.gif");
1403 void MultiplicityESD()
1405 TFile* file = TFile::Open("multiplicityESD.root");
1409 printf("multiplicityESD.root could not be opened.\n");
1413 TH1F* fMultiplicityESD = dynamic_cast<TH1F*> (file->Get("fMultiplicity"));
1415 TCanvas* canvas = new TCanvas("MultiplicityESD", "MultiplicityESD", 500, 500);
1417 fMultiplicityESD->Draw();
1420 void drawPlots(Int_t max)
1439 void CompareCorrection2Measured(const char* dataInput = "analysis_esd_raw.root", const char* correctionMapFile = "correction_map.root", const char* correctionMapFolder = "dndeta_correction")
1443 AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection(correctionMapFolder, correctionMapFolder);
1444 TFile::Open(correctionMapFile);
1445 dNdEtaCorrection->LoadHistograms();
1447 TFile* file = TFile::Open(dataInput);
1451 cout << "Error. File not found" << endl;
1455 dNdEtaAnalysis* fdNdEtaAnalysis = new dNdEtaAnalysis("dndeta", "dndeta");
1456 fdNdEtaAnalysis->LoadHistograms("fdNdEtaAnalysisESD");
1460 TH3* hist1 = (TH3*) dNdEtaCorrection->GetTrack2ParticleCorrection()->GetTrackCorrection()->GetMeasuredHistogram()->Clone("mc");
1461 hist1->SetTitle("mc");
1462 Printf("mc contains %f entries", hist1->Integral());
1463 Printf("mc contains %f entries in |vtx-z| < 10, pt > 0.3", hist1->Integral(hist1->GetXaxis()->FindBin(-9.9), hist1->GetXaxis()->FindBin(9.9), 1, hist1->GetNbinsY(), hist1->GetZaxis()->FindBin(0.301), hist1->GetNbinsZ()));
1465 TH3* hist2 = (TH3*) fdNdEtaAnalysis->GetData()->GetTrackCorrection()->GetMeasuredHistogram()->Clone("esd");
1466 hist2->SetTitle("esd");
1467 Printf("esd contains %f entries", hist2->Integral());
1468 Printf("esd contains %f entries in |vtx-z| < 10, pt > 0.3", hist2->Integral(hist2->GetXaxis()->FindBin(-9.9), hist2->GetXaxis()->FindBin(9.9), 1, hist2->GetNbinsY(), hist2->GetZaxis()->FindBin(0.301), hist2->GetNbinsZ()));
1470 AliPWG0Helper::CreateDividedProjections(hist1, hist2);
1472 new TCanvas; gROOT->FindObject("mc_yx_div_esd_yx")->Draw("COLZ");
1473 new TCanvas; gROOT->FindObject("mc_zx_div_esd_zx")->Draw("COLZ");
1474 new TCanvas; gROOT->FindObject("mc_zy_div_esd_zy")->Draw("COLZ");
1477 void CompareMeasured2Measured(const char* dataInput = "analysis_esd_raw.root", const char* dataInput2 = "analysis_esd_raw.root")
1481 TFile* file = TFile::Open(dataInput);
1485 cout << "Error. File not found" << endl;
1489 dNdEtaAnalysis* fdNdEtaAnalysis = new dNdEtaAnalysis("dndeta", "dndeta");
1490 fdNdEtaAnalysis->LoadHistograms("fdNdEtaAnalysisESD");
1492 TFile* file = TFile::Open(dataInput2);
1496 cout << "Error. File not found" << endl;
1500 dNdEtaAnalysis* fdNdEtaAnalysis2 = new dNdEtaAnalysis("dndeta2", "dndeta2");
1501 fdNdEtaAnalysis2->LoadHistograms("fdNdEtaAnalysisESD");
1505 TH3* hist1 = (TH3*) fdNdEtaAnalysis->GetData()->GetTrackCorrection()->GetMeasuredHistogram()->Clone("esd1");
1506 hist1->SetTitle("esd1");
1507 Printf("esd1 contains %f entries", hist1->GetEntries());
1508 Printf("esd1 contains %f entries in |vtx-z| < 10, pt > 0.3", hist1->Integral(hist1->GetXaxis()->FindBin(-9.9), hist1->GetXaxis()->FindBin(9.9), 1, hist1->GetNbinsY(), hist1->GetZaxis()->FindBin(0.301), hist1->GetNbinsZ()));
1510 TH3* hist2 = (TH3*) fdNdEtaAnalysis2->GetData()->GetTrackCorrection()->GetMeasuredHistogram()->Clone("esd2");
1511 hist2->SetTitle("esd2");
1512 Printf("esd2 contains %f entries", hist2->GetEntries());
1513 Printf("esd2 contains %f entries in |vtx-z| < 10, pt > 0.3", hist2->Integral(hist2->GetXaxis()->FindBin(-9.9), hist2->GetXaxis()->FindBin(9.9), 1, hist2->GetNbinsY(), hist2->GetZaxis()->FindBin(0.301), hist2->GetNbinsZ()));
1515 AliPWG0Helper::CreateDividedProjections(hist1, hist2);
1517 new TCanvas; gROOT->FindObject("esd1_yx_div_esd2_yx")->Draw("COLZ");
1518 new TCanvas; gROOT->FindObject("esd1_zx_div_esd2_zx")->Draw("COLZ");
1519 new TCanvas; gROOT->FindObject("esd1_zy_div_esd2_zy")->Draw("COLZ");
1521 TH2* event1 = (TH2*) fdNdEtaAnalysis->GetData()->GetEventCorrection()->GetMeasuredHistogram()->Clone("event1");
1522 TH2* event2 = (TH2*) fdNdEtaAnalysis2->GetData()->GetEventCorrection()->GetMeasuredHistogram()->Clone("event2");
1524 Printf("event1 contains %f entries", event1->GetEntries());
1525 Printf("event2 contains %f entries", event2->GetEntries());
1526 Printf("event1 integral is %f", event1->Integral());
1527 Printf("event2 integral is %f", event2->Integral());
1528 Printf("event1 contains %f entries in |vtx-z| < 10", event1->Integral(event1->GetXaxis()->FindBin(-9.9), event1->GetXaxis()->FindBin(9.9), 1, event1->GetNbinsY()));
1529 Printf("event2 contains %f entries in |vtx-z| < 10", event2->Integral(event2->GetXaxis()->FindBin(-9.9), event2->GetXaxis()->FindBin(9.9), 1, event2->GetNbinsY()));
1531 projx1 = event1->ProjectionX();
1532 projx2 = event2->ProjectionX();
1534 new TCanvas; projx1->DrawCopy(); projx2->SetLineColor(2); projx2->DrawCopy("SAME");
1536 projx1->Divide(projx2);
1537 new TCanvas; projx1->Draw();
1539 event1->Divide(event2);
1540 new TCanvas; event1->Draw("COLZ");