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