STEER/CheckESD.C

   1 #if !defined( __CINT__) || defined(__MAKECINT__)
   2 #include <TROOT.h>
   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>
  11 #include <TParticle.h>
  12
  13 #include "AliRunLoader.h"
  14 #include "AliLoader.h"
  15 #include "AliESDEvent.h"
  16 #include "AliESDv0.h"
  17 #include "AliESDcascade.h"
  18 #include "AliESDMuonTrack.h"
  19 #include "AliESDCaloCluster.h"
  20 #include "AliRun.h"
  21 #include "AliStack.h"
  22 #include "AliHeader.h"
  23 #include "AliGenEventHeader.h"
  24 #include "AliPID.h"
  25 #endif
  26 const Int_t kXiMinus = 3312;
  27 const Int_t kOmegaMinus = 3334;
  28
  29
  30 TH1F* 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
  44 TH1F* 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
  74 Bool_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
  99 Bool_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   }
 164   AliESDEvent * esd = new AliESDEvent;
 165   TTree* tree = (TTree*) esdFile->Get("esdTree");
 166   if (!tree) {
 167     Error("CheckESD", "no ESD tree found");
 168     return kFALSE;
 169   }
 170   esd->ReadFromTree(tree);
 171
 172   // efficiency and resolution histograms
 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
 192   Int_t partCode[AliPID::kSPECIES] =
 193     {kElectron, kMuonMinus, kPiPlus, kKPlus, kProton};
 194   const char* partName[AliPID::kSPECIES+1] =
 195     {"electron", "muon", "pion", "kaon", "proton", "other"};
 196   Double_t partFrac[AliPID::kSPECIES] =
 197     {0.01, 0.01, 0.85, 0.10, 0.05};
 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++) {
 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");
 228   TH1F* hEEMCAL = CreateHisto("hEEMCAL", "EMCAL", 100, 0, 50, "E [GeV]", "N");
 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
 307     tree->GetEvent(iEvent);
 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;
 342       for (Int_t i = 0; i < AliPID::kSPECIES; i++) {
 343         if (TMath::Abs(particle->GetPdgCode()) == partCode[i]) iGen = i;
 344       }
 345       Double_t probability[AliPID::kSPECIES];
 346       track->GetESDpid(probability);
 347       Double_t pMax = 0;
 348       Int_t iRec = 0;
 349       for (Int_t i = 0; i < AliPID::kSPECIES; i++) {
 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
 360       Double_t time[AliPID::kSPECIES];
 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
 375     // loop over muon tracks
 376     {
 377     for (Int_t iTrack = 0; iTrack < esd->GetNumberOfMuonTracks(); iTrack++) {
 378       AliESDMuonTrack* muonTrack = esd->GetMuonTrack(iTrack);
 379       Double_t ptInv = TMath::Abs(muonTrack->GetInverseBendingMomentum());
 380       if (ptInv > 0.001) {
 381         hPtMUON->Fill(1./ptInv);
 382       }
 383     }
 384     }
 385
 386     // loop over V0s
 387     for (Int_t iV0 = 0; iV0 < esd->GetNumberOfV0s(); iV0++) {
 388       AliESDv0* v0 = esd->GetV0(iV0);
 389       if (v0->GetOnFlyStatus()) continue;
 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());
 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;
 401       Int_t posLabel = TMath::Abs(esd->GetTrack(v0->GetPindex())->GetLabel());
 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);
 415       Double_t v0q;
 416       cascade->ChangeMassHypothesis(v0q,kXiMinus);
 417       hMassXi->Fill(cascade->GetEffMass());
 418       cascade->ChangeMassHypothesis(v0q,kOmegaMinus);
 419       hMassOmega->Fill(cascade->GetEffMass());
 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;
 426       Int_t posLabel = TMath::Abs(esd->GetTrack(cascade->GetPindex())
 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     }
 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++)
 449       hEPHOS->Fill(esd->GetCaloCluster(iCluster)->E());
 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++)
 457       hEEMCAL->Fill(esd->GetCaloCluster(iCluster)->E());
 458     }
 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");
 529     for (Int_t iRec = 0; iRec < AliPID::kSPECIES; iRec++) {
 530       printf("%9s", partName[iRec]);
 531     }
 532     printf("\n");
 533     for (Int_t iGen = 0; iGen < AliPID::kSPECIES+1; iGen++) {
 534       printf("%9s:", partName[iGen]);
 535       for (Int_t iRec = 0; iRec < AliPID::kSPECIES; iRec++) {
 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
 694   delete esd;
 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 }