Possibility to compile the macro
[u/mrichter/AliRoot.git] / STEER / CheckESD.C
CommitLineData
5e00415f 1#if !defined( __CINT__) || defined(__MAKECINT__)
cbc28ac7 2#include <TROOT.h>
5e00415f 3#include <TFile.h>
4#include <TError.h>
5#include <TH1.h>
6#include <TH2.h>
7#include <TF1.h>
8#include <TCanvas.h>
9#include <TVector3.h>
10#include <TPDGCode.h>
8b462fd8 11#include <TParticle.h>
5e00415f 12
13#include "AliRunLoader.h"
14#include "AliLoader.h"
af885e0f 15#include "AliESDEvent.h"
a50baf66 16#include "AliESDv0.h"
17#include "AliESDcascade.h"
18#include "AliESDMuonTrack.h"
19#include "AliESDCaloCluster.h"
5e00415f 20#include "AliRun.h"
21#include "AliStack.h"
22#include "AliHeader.h"
23#include "AliGenEventHeader.h"
5fed899a 24#include "AliPID.h"
a50baf66 25#endif
f0cfab93 26const Int_t kXiMinus = 3312;
27const Int_t kOmegaMinus = 3334;
5e00415f 28
29
30TH1F* CreateHisto(const char* name, const char* title,
31 Int_t nBins, Double_t xMin, Double_t xMax,
32 const char* xLabel = NULL, const char* yLabel = NULL)
33{
34// create a histogram
35
36 TH1F* result = new TH1F(name, title, nBins, xMin, xMax);
37 result->SetOption("E");
38 if (xLabel) result->GetXaxis()->SetTitle(xLabel);
39 if (yLabel) result->GetYaxis()->SetTitle(yLabel);
40 result->SetMarkerStyle(kFullCircle);
41 return result;
42}
43
44TH1F* CreateEffHisto(TH1F* hGen, TH1F* hRec)
45{
46// create an efficiency histogram
47
48 Int_t nBins = hGen->GetNbinsX();
49 TH1F* hEff = (TH1F*) hGen->Clone("hEff");
50 hEff->SetTitle("");
51 hEff->SetStats(kFALSE);
52 hEff->SetMinimum(0.);
53 hEff->SetMaximum(110.);
54 hEff->GetYaxis()->SetTitle("#epsilon [%]");
55
56 for (Int_t iBin = 0; iBin <= nBins; iBin++) {
57 Double_t nGen = hGen->GetBinContent(iBin);
58 Double_t nRec = hRec->GetBinContent(iBin);
59 if (nGen > 0) {
60 Double_t eff = nRec/nGen;
61 hEff->SetBinContent(iBin, 100. * eff);
62 Double_t error = sqrt(eff*(1.-eff) / nGen);
63 if (error == 0) error = 0.0001;
64 hEff->SetBinError(iBin, 100. * error);
65 } else {
66 hEff->SetBinContent(iBin, -100.);
67 hEff->SetBinError(iBin, 0);
68 }
69 }
70
71 return hEff;
72}
73
74Bool_t FitHisto(TH1* histo, Double_t& res, Double_t& resError)
75{
76// fit a gaussian to a histogram
77
78 static TF1* fitFunc = new TF1("fitFunc", "gaus");
79 fitFunc->SetLineWidth(2);
80 fitFunc->SetFillStyle(0);
81 Double_t maxFitRange = 2;
82
83 if (histo->Integral() > 50) {
84 Float_t mean = histo->GetMean();
85 Float_t rms = histo->GetRMS();
86 fitFunc->SetRange(mean - maxFitRange*rms, mean + maxFitRange*rms);
87 fitFunc->SetParameters(mean, rms);
88 histo->Fit(fitFunc, "QRI0");
89 histo->GetFunction("fitFunc")->ResetBit(1<<9);
90 res = TMath::Abs(fitFunc->GetParameter(2));
91 resError = TMath::Abs(fitFunc->GetParError(2));
92 return kTRUE;
93 }
94
95 return kFALSE;
96}
97
98
99Bool_t CheckESD(const char* gAliceFileName = "galice.root",
100 const char* esdFileName = "AliESDs.root")
101{
102// check the content of the ESD
103
104 // check values
105 Int_t checkNGenLow = 1;
106
107 Double_t checkEffLow = 0.5;
108 Double_t checkEffSigma = 3;
109 Double_t checkFakeHigh = 0.5;
110 Double_t checkFakeSigma = 3;
111
112 Double_t checkResPtInvHigh = 5;
113 Double_t checkResPtInvSigma = 3;
114 Double_t checkResPhiHigh = 10;
115 Double_t checkResPhiSigma = 3;
116 Double_t checkResThetaHigh = 10;
117 Double_t checkResThetaSigma = 3;
118
119 Double_t checkPIDEffLow = 0.5;
120 Double_t checkPIDEffSigma = 3;
121 Double_t checkResTOFHigh = 500;
122 Double_t checkResTOFSigma = 3;
123
124 Double_t checkPHOSNLow = 5;
125 Double_t checkPHOSEnergyLow = 0.3;
126 Double_t checkPHOSEnergyHigh = 1.0;
127 Double_t checkEMCALNLow = 50;
128 Double_t checkEMCALEnergyLow = 0.05;
129 Double_t checkEMCALEnergyHigh = 1.0;
130
131 Double_t checkMUONNLow = 1;
132 Double_t checkMUONPtLow = 0.5;
133 Double_t checkMUONPtHigh = 10.;
134
135 Double_t cutPtV0 = 0.3;
136 Double_t checkV0EffLow = 0.02;
137 Double_t checkV0EffSigma = 3;
138 Double_t cutPtCascade = 0.5;
139 Double_t checkCascadeEffLow = 0.01;
140 Double_t checkCascadeEffSigma = 3;
141
142 // open run loader and load gAlice, kinematics and header
143 AliRunLoader* runLoader = AliRunLoader::Open(gAliceFileName);
144 if (!runLoader) {
145 Error("CheckESD", "getting run loader from file %s failed",
146 gAliceFileName);
147 return kFALSE;
148 }
149 runLoader->LoadgAlice();
150 gAlice = runLoader->GetAliRun();
151 if (!gAlice) {
152 Error("CheckESD", "no galice object found");
153 return kFALSE;
154 }
155 runLoader->LoadKinematics();
156 runLoader->LoadHeader();
157
158 // open the ESD file
159 TFile* esdFile = TFile::Open(esdFileName);
160 if (!esdFile || !esdFile->IsOpen()) {
161 Error("CheckESD", "opening ESD file %s failed", esdFileName);
162 return kFALSE;
163 }
af885e0f 164 AliESDEvent * esd = new AliESDEvent;
8b462fd8 165 TTree* tree = (TTree*) esdFile->Get("esdTree");
166 if (!tree) {
167 Error("CheckESD", "no ESD tree found");
168 return kFALSE;
169 }
cbc28ac7 170 esd->ReadFromTree(tree);
5e00415f 171
af885e0f 172 // efficiency and resolution histograms
5e00415f 173 Int_t nBinsPt = 15;
174 Float_t minPt = 0.1;
175 Float_t maxPt = 3.1;
176 TH1F* hGen = CreateHisto("hGen", "generated tracks",
177 nBinsPt, minPt, maxPt, "p_{t} [GeV/c]", "N");
178 TH1F* hRec = CreateHisto("hRec", "reconstructed tracks",
179 nBinsPt, minPt, maxPt, "p_{t} [GeV/c]", "N");
180 Int_t nGen = 0;
181 Int_t nRec = 0;
182 Int_t nFake = 0;
183
184 TH1F* hResPtInv = CreateHisto("hResPtInv", "", 100, -10, 10,
185 "(p_{t,rec}^{-1}-p_{t,sim}^{-1}) / p_{t,sim}^{-1} [%]", "N");
186 TH1F* hResPhi = CreateHisto("hResPhi", "", 100, -20, 20,
187 "#phi_{rec}-#phi_{sim} [mrad]", "N");
188 TH1F* hResTheta = CreateHisto("hResTheta", "", 100, -20, 20,
189 "#theta_{rec}-#theta_{sim} [mrad]", "N");
190
191 // PID
5fed899a 192 Int_t partCode[AliPID::kSPECIES] =
5e00415f 193 {kElectron, kMuonMinus, kPiPlus, kKPlus, kProton};
5fed899a 194 const char* partName[AliPID::kSPECIES+1] =
5e00415f 195 {"electron", "muon", "pion", "kaon", "proton", "other"};
5fed899a 196 Double_t partFrac[AliPID::kSPECIES] =
5e00415f 197 {0.01, 0.01, 0.85, 0.10, 0.05};
5fed899a 198 Int_t identified[AliPID::kSPECIES+1][AliPID::kSPECIES];
199 for (Int_t iGen = 0; iGen < AliPID::kSPECIES+1; iGen++) {
200 for (Int_t iRec = 0; iRec < AliPID::kSPECIES; iRec++) {
5e00415f 201 identified[iGen][iRec] = 0;
202 }
203 }
204 Int_t nIdentified = 0;
205
206 // dE/dx and TOF
207 TH2F* hDEdxRight = new TH2F("hDEdxRight", "", 300, 0, 3, 100, 0, 400);
208 hDEdxRight->SetStats(kFALSE);
209 hDEdxRight->GetXaxis()->SetTitle("p [GeV/c]");
210 hDEdxRight->GetYaxis()->SetTitle("dE/dx_{TPC}");
211 hDEdxRight->SetMarkerStyle(kFullCircle);
212 hDEdxRight->SetMarkerSize(0.4);
213 TH2F* hDEdxWrong = new TH2F("hDEdxWrong", "", 300, 0, 3, 100, 0, 400);
214 hDEdxWrong->SetStats(kFALSE);
215 hDEdxWrong->GetXaxis()->SetTitle("p [GeV/c]");
216 hDEdxWrong->GetYaxis()->SetTitle("dE/dx_{TPC}");
217 hDEdxWrong->SetMarkerStyle(kFullCircle);
218 hDEdxWrong->SetMarkerSize(0.4);
219 hDEdxWrong->SetMarkerColor(kRed);
220 TH1F* hResTOFRight = CreateHisto("hResTOFRight", "", 100, -1000, 1000,
221 "t_{TOF}-t_{track} [ps]", "N");
222 TH1F* hResTOFWrong = CreateHisto("hResTOFWrong", "", 100, -1000, 1000,
223 "t_{TOF}-t_{track} [ps]", "N");
224 hResTOFWrong->SetLineColor(kRed);
225
226 // calorimeters
227 TH1F* hEPHOS = CreateHisto("hEPHOS", "PHOS", 100, 0, 5, "E [GeV]", "N");
f04585c3 228 TH1F* hEEMCAL = CreateHisto("hEEMCAL", "EMCAL", 100, 0, 50, "E [GeV]", "N");
5e00415f 229
230 // muons
231 TH1F* hPtMUON = CreateHisto("hPtMUON", "MUON", 100, 0, 20,
232 "p_{t} [GeV/c]", "N");
233
234 // V0s and cascades
235 TH1F* hMassK0 = CreateHisto("hMassK0", "K^{0}", 100, 0.4, 0.6,
236 "M(#pi^{+}#pi^{-}) [GeV/c^{2}]", "N");
237 TH1F* hMassLambda = CreateHisto("hMassLambda", "#Lambda", 100, 1.0, 1.2,
238 "M(p#pi^{-}) [GeV/c^{2}]", "N");
239 TH1F* hMassLambdaBar = CreateHisto("hMassLambdaBar", "#bar{#Lambda}",
240 100, 1.0, 1.2,
241 "M(#bar{p}#pi^{+}) [GeV/c^{2}]", "N");
242 Int_t nGenV0s = 0;
243 Int_t nRecV0s = 0;
244 TH1F* hMassXi = CreateHisto("hMassXi", "#Xi", 100, 1.2, 1.5,
245 "M(#Lambda#pi) [GeV/c^{2}]", "N");
246 TH1F* hMassOmega = CreateHisto("hMassOmega", "#Omega", 100, 1.5, 1.8,
247 "M(#LambdaK) [GeV/c^{2}]", "N");
248 Int_t nGenCascades = 0;
249 Int_t nRecCascades = 0;
250
251 // loop over events
252 for (Int_t iEvent = 0; iEvent < runLoader->GetNumberOfEvents(); iEvent++) {
253 runLoader->GetEvent(iEvent);
254
255 // select simulated primary particles, V0s and cascades
256 AliStack* stack = gAlice->Stack();
257 Int_t nParticles = stack->GetNtrack();
258 TArrayF vertex(3);
259 runLoader->GetHeader()->GenEventHeader()->PrimaryVertex(vertex);
260 TObjArray selParticles;
261 TObjArray selV0s;
262 TObjArray selCascades;
263 for (Int_t iParticle = 0; iParticle < nParticles; iParticle++) {
264 TParticle* particle = stack->Particle(iParticle);
265 if (!particle) continue;
266 if (particle->Pt() < 0.001) continue;
267 if (TMath::Abs(particle->Eta()) > 0.9) continue;
268 TVector3 dVertex(particle->Vx() - vertex[0],
269 particle->Vy() - vertex[1],
270 particle->Vz() - vertex[2]);
271 if (dVertex.Mag() > 0.0001) continue;
272
273 switch (TMath::Abs(particle->GetPdgCode())) {
274 case kElectron:
275 case kMuonMinus:
276 case kPiPlus:
277 case kKPlus:
278 case kProton: {
279 if (particle->Pt() > minPt) {
280 selParticles.Add(particle);
281 nGen++;
282 hGen->Fill(particle->Pt());
283 }
284 break;
285 }
286 case kK0Short:
287 case kLambda0: {
288 if (particle->Pt() > cutPtV0) {
289 nGenV0s++;
290 selV0s.Add(particle);
291 }
292 break;
293 }
294 case kXiMinus:
295 case kOmegaMinus: {
296 if (particle->Pt() > cutPtCascade) {
297 nGenCascades++;
298 selCascades.Add(particle);
299 }
300 break;
301 }
302 default: break;
303 }
304 }
305
306 // get the event summary data
8b462fd8 307 tree->GetEvent(iEvent);
5e00415f 308 if (!esd) {
309 Error("CheckESD", "no ESD object found for event %d", iEvent);
310 return kFALSE;
311 }
312
313 // loop over tracks
314 for (Int_t iTrack = 0; iTrack < esd->GetNumberOfTracks(); iTrack++) {
315 AliESDtrack* track = esd->GetTrack(iTrack);
316
317 // select tracks of selected particles
318 Int_t label = TMath::Abs(track->GetLabel());
319 if (label > stack->GetNtrack()) continue; // background
320 TParticle* particle = stack->Particle(label);
321 if (!selParticles.Contains(particle)) continue;
322 if ((track->GetStatus() & AliESDtrack::kITSrefit) == 0) continue;
323 if (track->GetConstrainedChi2() > 1e9) continue;
324 selParticles.Remove(particle); // don't count multiple tracks
325
326 nRec++;
327 hRec->Fill(particle->Pt());
328 if (track->GetLabel() < 0) nFake++;
329
330 // resolutions
331 Double_t p[3];
332 track->GetConstrainedPxPyPz(p);
333 TVector3 pTrack(p);
334 hResPtInv->Fill(100. * (1./pTrack.Pt() - 1./particle->Pt()) *
335 particle->Pt());
336 hResPhi->Fill(1000. * (pTrack.Phi() - particle->Phi()));
337 hResTheta->Fill(1000. * (pTrack.Theta() - particle->Theta()));
338
339 // PID
340 if ((track->GetStatus() & AliESDtrack::kESDpid) == 0) continue;
341 Int_t iGen = 5;
5fed899a 342 for (Int_t i = 0; i < AliPID::kSPECIES; i++) {
5e00415f 343 if (TMath::Abs(particle->GetPdgCode()) == partCode[i]) iGen = i;
344 }
5fed899a 345 Double_t probability[AliPID::kSPECIES];
5e00415f 346 track->GetESDpid(probability);
347 Double_t pMax = 0;
348 Int_t iRec = 0;
5fed899a 349 for (Int_t i = 0; i < AliPID::kSPECIES; i++) {
5e00415f 350 probability[i] *= partFrac[i];
351 if (probability[i] > pMax) {
352 pMax = probability[i];
353 iRec = i;
354 }
355 }
356 identified[iGen][iRec]++;
357 if (iGen == iRec) nIdentified++;
358
359 // dE/dx and TOF
5fed899a 360 Double_t time[AliPID::kSPECIES];
5e00415f 361 track->GetIntegratedTimes(time);
362 if (iGen == iRec) {
363 hDEdxRight->Fill(pTrack.Mag(), track->GetTPCsignal());
364 if ((track->GetStatus() & AliESDtrack::kTOFpid) != 0) {
365 hResTOFRight->Fill(track->GetTOFsignal() - time[iRec]);
366 }
367 } else {
368 hDEdxWrong->Fill(pTrack.Mag(), track->GetTPCsignal());
369 if ((track->GetStatus() & AliESDtrack::kTOFpid) != 0) {
370 hResTOFWrong->Fill(track->GetTOFsignal() - time[iRec]);
371 }
372 }
373 }
374
5e00415f 375 // loop over muon tracks
f0cfab93 376 {
5e00415f 377 for (Int_t iTrack = 0; iTrack < esd->GetNumberOfMuonTracks(); iTrack++) {
f0cfab93 378 AliESDMuonTrack* muonTrack = esd->GetMuonTrack(iTrack);
379 Double_t ptInv = TMath::Abs(muonTrack->GetInverseBendingMomentum());
5e00415f 380 if (ptInv > 0.001) {
381 hPtMUON->Fill(1./ptInv);
382 }
383 }
f0cfab93 384 }
5e00415f 385
386 // loop over V0s
387 for (Int_t iV0 = 0; iV0 < esd->GetNumberOfV0s(); iV0++) {
388 AliESDv0* v0 = esd->GetV0(iV0);
d6a49f20 389 if (v0->GetOnFlyStatus()) continue;
f980f8fb 390 v0->ChangeMassHypothesis(kK0Short);
391 hMassK0->Fill(v0->GetEffMass());
392 v0->ChangeMassHypothesis(kLambda0);
393 hMassLambda->Fill(v0->GetEffMass());
394 v0->ChangeMassHypothesis(kLambda0Bar);
395 hMassLambdaBar->Fill(v0->GetEffMass());
5e00415f 396
397 Int_t negLabel = TMath::Abs(esd->GetTrack(v0->GetNindex())->GetLabel());
398 if (negLabel > stack->GetNtrack()) continue; // background
399 Int_t negMother = stack->Particle(negLabel)->GetMother(0);
400 if (negMother < 0) continue;
f980f8fb 401 Int_t posLabel = TMath::Abs(esd->GetTrack(v0->GetPindex())->GetLabel());
5e00415f 402 if (posLabel > stack->GetNtrack()) continue; // background
403 Int_t posMother = stack->Particle(posLabel)->GetMother(0);
404 if (negMother != posMother) continue;
405 TParticle* particle = stack->Particle(negMother);
406 if (!selV0s.Contains(particle)) continue;
407 selV0s.Remove(particle);
408 nRecV0s++;
409 }
410
411 // loop over Cascades
412 for (Int_t iCascade = 0; iCascade < esd->GetNumberOfCascades();
413 iCascade++) {
414 AliESDcascade* cascade = esd->GetCascade(iCascade);
f980f8fb 415 Double_t v0q;
416 cascade->ChangeMassHypothesis(v0q,kXiMinus);
417 hMassXi->Fill(cascade->GetEffMass());
418 cascade->ChangeMassHypothesis(v0q,kOmegaMinus);
419 hMassOmega->Fill(cascade->GetEffMass());
5e00415f 420
421 Int_t negLabel = TMath::Abs(esd->GetTrack(cascade->GetNindex())
422 ->GetLabel());
423 if (negLabel > stack->GetNtrack()) continue; // background
424 Int_t negMother = stack->Particle(negLabel)->GetMother(0);
425 if (negMother < 0) continue;
f980f8fb 426 Int_t posLabel = TMath::Abs(esd->GetTrack(cascade->GetPindex())
5e00415f 427 ->GetLabel());
428 if (posLabel > stack->GetNtrack()) continue; // background
429 Int_t posMother = stack->Particle(posLabel)->GetMother(0);
430 if (negMother != posMother) continue;
431 Int_t v0Mother = stack->Particle(negMother)->GetMother(0);
432 if (v0Mother < 0) continue;
433 Int_t bacLabel = TMath::Abs(esd->GetTrack(cascade->GetBindex())
434 ->GetLabel());
435 if (bacLabel > stack->GetNtrack()) continue; // background
436 Int_t bacMother = stack->Particle(bacLabel)->GetMother(0);
437 if (v0Mother != bacMother) continue;
438 TParticle* particle = stack->Particle(v0Mother);
439 if (!selCascades.Contains(particle)) continue;
440 selCascades.Remove(particle);
441 nRecCascades++;
442 }
f04585c3 443
444 // loop over the PHOS clusters
445 {
446 Int_t firstPHOSCluster = esd->GetFirstPHOSCluster();
447 Int_t lastPHOSCluster = firstPHOSCluster + esd->GetNumberOfPHOSClusters();
448 for (Int_t iCluster=firstPHOSCluster; iCluster<lastPHOSCluster; iCluster++)
cbc28ac7 449 hEPHOS->Fill(esd->GetCaloCluster(iCluster)->E());
f04585c3 450 }
451
452 // loop over the EMCAL clusters
453 {
454 Int_t firstEMCALCluster = esd->GetFirstEMCALCluster();
455 Int_t lastEMCALCluster = firstEMCALCluster + esd->GetNumberOfEMCALClusters();
456 for (Int_t iCluster=firstEMCALCluster; iCluster<lastEMCALCluster; iCluster++)
cbc28ac7 457 hEEMCAL->Fill(esd->GetCaloCluster(iCluster)->E());
f04585c3 458 }
5e00415f 459 }
460
461 // perform checks
462 if (nGen < checkNGenLow) {
463 Warning("CheckESD", "low number of generated particles: %d", Int_t(nGen));
464 }
465
466 TH1F* hEff = CreateEffHisto(hGen, hRec);
467
468 Info("CheckESD", "%d out of %d tracks reconstructed including %d "
469 "fake tracks", nRec, nGen, nFake);
470 if (nGen > 0) {
471 // efficiency
472 Double_t eff = nRec*1./nGen;
473 Double_t effError = TMath::Sqrt(eff*(1.-eff) / nGen);
474 Double_t fake = nFake*1./nGen;
475 Double_t fakeError = TMath::Sqrt(fake*(1.-fake) / nGen);
476 Info("CheckESD", "eff = (%.1f +- %.1f) %% fake = (%.1f +- %.1f) %%",
477 100.*eff, 100.*effError, 100.*fake, 100.*fakeError);
478
479 if (eff < checkEffLow - checkEffSigma*effError) {
480 Warning("CheckESD", "low efficiency: (%.1f +- %.1f) %%",
481 100.*eff, 100.*effError);
482 }
483 if (fake > checkFakeHigh + checkFakeSigma*fakeError) {
484 Warning("CheckESD", "high fake: (%.1f +- %.1f) %%",
485 100.*fake, 100.*fakeError);
486 }
487
488 // resolutions
489 Double_t res, resError;
490 if (FitHisto(hResPtInv, res, resError)) {
491 Info("CheckESD", "relative inverse pt resolution = (%.1f +- %.1f) %%",
492 res, resError);
493 if (res > checkResPtInvHigh + checkResPtInvSigma*resError) {
494 Warning("CheckESD", "bad pt resolution: (%.1f +- %.1f) %%",
495 res, resError);
496 }
497 }
498
499 if (FitHisto(hResPhi, res, resError)) {
500 Info("CheckESD", "phi resolution = (%.1f +- %.1f) mrad", res, resError);
501 if (res > checkResPhiHigh + checkResPhiSigma*resError) {
502 Warning("CheckESD", "bad phi resolution: (%.1f +- %.1f) mrad",
503 res, resError);
504 }
505 }
506
507 if (FitHisto(hResTheta, res, resError)) {
508 Info("CheckESD", "theta resolution = (%.1f +- %.1f) mrad",
509 res, resError);
510 if (res > checkResThetaHigh + checkResThetaSigma*resError) {
511 Warning("CheckESD", "bad theta resolution: (%.1f +- %.1f) mrad",
512 res, resError);
513 }
514 }
515
516 // PID
517 if (nRec > 0) {
518 Double_t eff = nIdentified*1./nRec;
519 Double_t effError = TMath::Sqrt(eff*(1.-eff) / nRec);
520 Info("CheckESD", "PID eff = (%.1f +- %.1f) %%",
521 100.*eff, 100.*effError);
522 if (eff < checkPIDEffLow - checkPIDEffSigma*effError) {
523 Warning("CheckESD", "low PID efficiency: (%.1f +- %.1f) %%",
524 100.*eff, 100.*effError);
525 }
526 }
527
528 printf("%9s:", "gen\\rec");
5fed899a 529 for (Int_t iRec = 0; iRec < AliPID::kSPECIES; iRec++) {
5e00415f 530 printf("%9s", partName[iRec]);
531 }
532 printf("\n");
5fed899a 533 for (Int_t iGen = 0; iGen < AliPID::kSPECIES+1; iGen++) {
5e00415f 534 printf("%9s:", partName[iGen]);
5fed899a 535 for (Int_t iRec = 0; iRec < AliPID::kSPECIES; iRec++) {
5e00415f 536 printf("%9d", identified[iGen][iRec]);
537 }
538 printf("\n");
539 }
540
541 if (FitHisto(hResTOFRight, res, resError)) {
542 Info("CheckESD", "TOF resolution = (%.1f +- %.1f) ps", res, resError);
543 if (res > checkResTOFHigh + checkResTOFSigma*resError) {
544 Warning("CheckESD", "bad TOF resolution: (%.1f +- %.1f) ps",
545 res, resError);
546 }
547 }
548
549 // calorimeters
550 if (hEPHOS->Integral() < checkPHOSNLow) {
551 Warning("CheckESD", "low number of PHOS particles: %d",
552 Int_t(hEPHOS->Integral()));
553 } else {
554 Double_t mean = hEPHOS->GetMean();
555 if (mean < checkPHOSEnergyLow) {
556 Warning("CheckESD", "low mean PHOS energy: %.1f GeV", mean);
557 } else if (mean > checkPHOSEnergyHigh) {
558 Warning("CheckESD", "high mean PHOS energy: %.1f GeV", mean);
559 }
560 }
561
562 if (hEEMCAL->Integral() < checkEMCALNLow) {
563 Warning("CheckESD", "low number of EMCAL particles: %d",
564 Int_t(hEEMCAL->Integral()));
565 } else {
566 Double_t mean = hEEMCAL->GetMean();
567 if (mean < checkEMCALEnergyLow) {
568 Warning("CheckESD", "low mean EMCAL energy: %.1f GeV", mean);
569 } else if (mean > checkEMCALEnergyHigh) {
570 Warning("CheckESD", "high mean EMCAL energy: %.1f GeV", mean);
571 }
572 }
573
574 // muons
575 if (hPtMUON->Integral() < checkMUONNLow) {
576 Warning("CheckESD", "low number of MUON particles: %d",
577 Int_t(hPtMUON->Integral()));
578 } else {
579 Double_t mean = hPtMUON->GetMean();
580 if (mean < checkMUONPtLow) {
581 Warning("CheckESD", "low mean MUON pt: %.1f GeV/c", mean);
582 } else if (mean > checkMUONPtHigh) {
583 Warning("CheckESD", "high mean MUON pt: %.1f GeV/c", mean);
584 }
585 }
586
587 // V0s
588 if (nGenV0s > 0) {
589 Double_t eff = nRecV0s*1./nGenV0s;
590 Double_t effError = TMath::Sqrt(eff*(1.-eff) / nGenV0s);
591 if (effError == 0) effError = checkV0EffLow / TMath::Sqrt(1.*nGenV0s);
592 Info("CheckESD", "V0 eff = (%.1f +- %.1f) %%",
593 100.*eff, 100.*effError);
594 if (eff < checkV0EffLow - checkV0EffSigma*effError) {
595 Warning("CheckESD", "low V0 efficiency: (%.1f +- %.1f) %%",
596 100.*eff, 100.*effError);
597 }
598 }
599
600 // Cascades
601 if (nGenCascades > 0) {
602 Double_t eff = nRecCascades*1./nGenCascades;
603 Double_t effError = TMath::Sqrt(eff*(1.-eff) / nGenCascades);
604 if (effError == 0) effError = checkV0EffLow /
605 TMath::Sqrt(1.*nGenCascades);
606 Info("CheckESD", "Cascade eff = (%.1f +- %.1f) %%",
607 100.*eff, 100.*effError);
608 if (eff < checkCascadeEffLow - checkCascadeEffSigma*effError) {
609 Warning("CheckESD", "low Cascade efficiency: (%.1f +- %.1f) %%",
610 100.*eff, 100.*effError);
611 }
612 }
613 }
614
615 // draw the histograms if not in batch mode
616 if (!gROOT->IsBatch()) {
617 new TCanvas;
618 hEff->DrawCopy();
619 new TCanvas;
620 hResPtInv->DrawCopy("E");
621 new TCanvas;
622 hResPhi->DrawCopy("E");
623 new TCanvas;
624 hResTheta->DrawCopy("E");
625 new TCanvas;
626 hDEdxRight->DrawCopy();
627 hDEdxWrong->DrawCopy("SAME");
628 new TCanvas;
629 hResTOFRight->DrawCopy("E");
630 hResTOFWrong->DrawCopy("SAME");
631 new TCanvas;
632 hEPHOS->DrawCopy("E");
633 new TCanvas;
634 hEEMCAL->DrawCopy("E");
635 new TCanvas;
636 hPtMUON->DrawCopy("E");
637 new TCanvas;
638 hMassK0->DrawCopy("E");
639 new TCanvas;
640 hMassLambda->DrawCopy("E");
641 new TCanvas;
642 hMassLambdaBar->DrawCopy("E");
643 new TCanvas;
644 hMassXi->DrawCopy("E");
645 new TCanvas;
646 hMassOmega->DrawCopy("E");
647 }
648
649 // write the output histograms to a file
650 TFile* outputFile = TFile::Open("check.root", "recreate");
651 if (!outputFile || !outputFile->IsOpen()) {
652 Error("CheckESD", "opening output file check.root failed");
653 return kFALSE;
654 }
655 hEff->Write();
656 hResPtInv->Write();
657 hResPhi->Write();
658 hResTheta->Write();
659 hDEdxRight->Write();
660 hDEdxWrong->Write();
661 hResTOFRight->Write();
662 hResTOFWrong->Write();
663 hEPHOS->Write();
664 hEEMCAL->Write();
665 hPtMUON->Write();
666 hMassK0->Write();
667 hMassLambda->Write();
668 hMassLambdaBar->Write();
669 hMassXi->Write();
670 hMassOmega->Write();
671 outputFile->Close();
672 delete outputFile;
673
674 // clean up
675 delete hGen;
676 delete hRec;
677 delete hEff;
678 delete hResPtInv;
679 delete hResPhi;
680 delete hResTheta;
681 delete hDEdxRight;
682 delete hDEdxWrong;
683 delete hResTOFRight;
684 delete hResTOFWrong;
685 delete hEPHOS;
686 delete hEEMCAL;
687 delete hPtMUON;
688 delete hMassK0;
689 delete hMassLambda;
690 delete hMassLambdaBar;
691 delete hMassXi;
692 delete hMassOmega;
693
8b462fd8 694 delete esd;
5e00415f 695 esdFile->Close();
696 delete esdFile;
697
698 runLoader->UnloadHeader();
699 runLoader->UnloadKinematics();
700 delete runLoader;
701
702 // result of check
703 Info("CheckESD", "check of ESD was successfull");
704 return kTRUE;
705}