-#if !defined( __CINT__) || defined(__MAKECINT__)\r
-#include <TFile.h>\r
-#include <TError.h>\r
-#include <TH1.h>\r
-#include <TH2.h>\r
-#include <TF1.h>\r
-#include <TCanvas.h>\r
-#include <TVector3.h>\r
-#include <TPDGCode.h>\r
-#include <TParticle.h>\r
-\r
-#include "AliRunLoader.h"\r
-#include "AliLoader.h"\r
-#include "AliESDEvent.h"\r
-#include "AliRun.h"\r
-#include "AliStack.h"\r
-#include "AliHeader.h"\r
-#include "AliGenEventHeader.h"\r
-#include "AliPID.h"\r
-#else\r
-const Int_t kXiMinus = 3312;\r
-const Int_t kOmegaMinus = 3334;\r
-#endif\r
-\r
-\r
-TH1F* CreateHisto(const char* name, const char* title, \r
- Int_t nBins, Double_t xMin, Double_t xMax,\r
- const char* xLabel = NULL, const char* yLabel = NULL)\r
-{\r
-// create a histogram\r
-\r
- TH1F* result = new TH1F(name, title, nBins, xMin, xMax);\r
- result->SetOption("E");\r
- if (xLabel) result->GetXaxis()->SetTitle(xLabel);\r
- if (yLabel) result->GetYaxis()->SetTitle(yLabel);\r
- result->SetMarkerStyle(kFullCircle);\r
- return result;\r
-}\r
-\r
-TH1F* CreateEffHisto(TH1F* hGen, TH1F* hRec)\r
-{\r
-// create an efficiency histogram\r
-\r
- Int_t nBins = hGen->GetNbinsX();\r
- TH1F* hEff = (TH1F*) hGen->Clone("hEff");\r
- hEff->SetTitle("");\r
- hEff->SetStats(kFALSE);\r
- hEff->SetMinimum(0.);\r
- hEff->SetMaximum(110.);\r
- hEff->GetYaxis()->SetTitle("#epsilon [%]");\r
-\r
- for (Int_t iBin = 0; iBin <= nBins; iBin++) {\r
- Double_t nGen = hGen->GetBinContent(iBin);\r
- Double_t nRec = hRec->GetBinContent(iBin);\r
- if (nGen > 0) {\r
- Double_t eff = nRec/nGen;\r
- hEff->SetBinContent(iBin, 100. * eff);\r
- Double_t error = sqrt(eff*(1.-eff) / nGen);\r
- if (error == 0) error = 0.0001;\r
- hEff->SetBinError(iBin, 100. * error);\r
- } else {\r
- hEff->SetBinContent(iBin, -100.);\r
- hEff->SetBinError(iBin, 0);\r
- }\r
- }\r
-\r
- return hEff;\r
-}\r
-\r
-Bool_t FitHisto(TH1* histo, Double_t& res, Double_t& resError)\r
-{\r
-// fit a gaussian to a histogram\r
-\r
- static TF1* fitFunc = new TF1("fitFunc", "gaus");\r
- fitFunc->SetLineWidth(2);\r
- fitFunc->SetFillStyle(0);\r
- Double_t maxFitRange = 2;\r
-\r
- if (histo->Integral() > 50) {\r
- Float_t mean = histo->GetMean();\r
- Float_t rms = histo->GetRMS();\r
- fitFunc->SetRange(mean - maxFitRange*rms, mean + maxFitRange*rms);\r
- fitFunc->SetParameters(mean, rms);\r
- histo->Fit(fitFunc, "QRI0");\r
- histo->GetFunction("fitFunc")->ResetBit(1<<9);\r
- res = TMath::Abs(fitFunc->GetParameter(2));\r
- resError = TMath::Abs(fitFunc->GetParError(2));\r
- return kTRUE;\r
- }\r
-\r
- return kFALSE;\r
-}\r
-\r
-\r
-Bool_t CheckESD(const char* gAliceFileName = "galice.root", \r
- const char* esdFileName = "AliESDs.root")\r
-{\r
-// check the content of the ESD\r
- \r
- // check values\r
- Int_t checkNGenLow = 1;\r
-\r
- Double_t checkEffLow = 0.5;\r
- Double_t checkEffSigma = 3;\r
- Double_t checkFakeHigh = 0.5;\r
- Double_t checkFakeSigma = 3;\r
-\r
- Double_t checkResPtInvHigh = 5;\r
- Double_t checkResPtInvSigma = 3;\r
- Double_t checkResPhiHigh = 10;\r
- Double_t checkResPhiSigma = 3;\r
- Double_t checkResThetaHigh = 10;\r
- Double_t checkResThetaSigma = 3;\r
-\r
- Double_t checkPIDEffLow = 0.5;\r
- Double_t checkPIDEffSigma = 3;\r
- Double_t checkResTOFHigh = 500;\r
- Double_t checkResTOFSigma = 3;\r
-\r
- Double_t checkPHOSNLow = 5;\r
- Double_t checkPHOSEnergyLow = 0.3;\r
- Double_t checkPHOSEnergyHigh = 1.0;\r
- Double_t checkEMCALNLow = 50;\r
- Double_t checkEMCALEnergyLow = 0.05;\r
- Double_t checkEMCALEnergyHigh = 1.0;\r
-\r
- Double_t checkMUONNLow = 1;\r
- Double_t checkMUONPtLow = 0.5;\r
- Double_t checkMUONPtHigh = 10.;\r
-\r
- Double_t cutPtV0 = 0.3;\r
- Double_t checkV0EffLow = 0.02;\r
- Double_t checkV0EffSigma = 3;\r
- Double_t cutPtCascade = 0.5;\r
- Double_t checkCascadeEffLow = 0.01;\r
- Double_t checkCascadeEffSigma = 3;\r
-\r
- // open run loader and load gAlice, kinematics and header\r
- AliRunLoader* runLoader = AliRunLoader::Open(gAliceFileName);\r
- if (!runLoader) {\r
- Error("CheckESD", "getting run loader from file %s failed", \r
- gAliceFileName);\r
- return kFALSE;\r
- }\r
- runLoader->LoadgAlice();\r
- gAlice = runLoader->GetAliRun();\r
- if (!gAlice) {\r
- Error("CheckESD", "no galice object found");\r
- return kFALSE;\r
- }\r
- runLoader->LoadKinematics();\r
- runLoader->LoadHeader();\r
-\r
- // open the ESD file\r
- TFile* esdFile = TFile::Open(esdFileName);\r
- if (!esdFile || !esdFile->IsOpen()) {\r
- Error("CheckESD", "opening ESD file %s failed", esdFileName);\r
- return kFALSE;\r
- }\r
- AliESDEvent* esd = new AliESDEvent;\r
- TTree* tree = (TTree*) esdFile->Get("esdTree");\r
- if (!tree) {\r
- Error("CheckESD", "no ESD tree found");\r
- return kFALSE;\r
- }\r
- esd->ReadFromTree(tree);\r
-\r
- // efficiency and resolution histograms\r
- Int_t nBinsPt = 15;\r
- Float_t minPt = 0.1;\r
- Float_t maxPt = 3.1;\r
- TH1F* hGen = CreateHisto("hGen", "generated tracks", \r
- nBinsPt, minPt, maxPt, "p_{t} [GeV/c]", "N");\r
- TH1F* hRec = CreateHisto("hRec", "reconstructed tracks", \r
- nBinsPt, minPt, maxPt, "p_{t} [GeV/c]", "N");\r
- Int_t nGen = 0;\r
- Int_t nRec = 0;\r
- Int_t nFake = 0;\r
-\r
- TH1F* hResPtInv = CreateHisto("hResPtInv", "", 100, -10, 10, \r
- "(p_{t,rec}^{-1}-p_{t,sim}^{-1}) / p_{t,sim}^{-1} [%]", "N");\r
- TH1F* hResPhi = CreateHisto("hResPhi", "", 100, -20, 20, \r
- "#phi_{rec}-#phi_{sim} [mrad]", "N");\r
- TH1F* hResTheta = CreateHisto("hResTheta", "", 100, -20, 20, \r
- "#theta_{rec}-#theta_{sim} [mrad]", "N");\r
-\r
- // PID\r
- Int_t partCode[] = \r
- {kElectron, kMuonMinus, kPiPlus, kKPlus, kProton};\r
- const char* partName[] = \r
- {"electron", "muon", "pion", "kaon", "proton", "other"};\r
- Double_t partFrac[] = \r
- {0.01, 0.01, 0.85, 0.10, 0.05};\r
- Int_t identified[6][5];\r
- for (Int_t iGen = 0; iGen < 6; iGen++) {\r
- for (Int_t iRec = 0; iRec < 5; iRec++) {\r
- identified[iGen][iRec] = 0;\r
- }\r
- }\r
- Int_t nIdentified = 0;\r
-\r
- // dE/dx and TOF\r
- TH2F* hDEdxRight = new TH2F("hDEdxRight", "", 300, 0, 3, 100, 0, 400);\r
- hDEdxRight->SetStats(kFALSE);\r
- hDEdxRight->GetXaxis()->SetTitle("p [GeV/c]");\r
- hDEdxRight->GetYaxis()->SetTitle("dE/dx_{TPC}");\r
- hDEdxRight->SetMarkerStyle(kFullCircle);\r
- hDEdxRight->SetMarkerSize(0.4);\r
- TH2F* hDEdxWrong = new TH2F("hDEdxWrong", "", 300, 0, 3, 100, 0, 400);\r
- hDEdxWrong->SetStats(kFALSE);\r
- hDEdxWrong->GetXaxis()->SetTitle("p [GeV/c]");\r
- hDEdxWrong->GetYaxis()->SetTitle("dE/dx_{TPC}");\r
- hDEdxWrong->SetMarkerStyle(kFullCircle);\r
- hDEdxWrong->SetMarkerSize(0.4);\r
- hDEdxWrong->SetMarkerColor(kRed);\r
- TH1F* hResTOFRight = CreateHisto("hResTOFRight", "", 100, -1000, 1000, \r
- "t_{TOF}-t_{track} [ps]", "N");\r
- TH1F* hResTOFWrong = CreateHisto("hResTOFWrong", "", 100, -1000, 1000, \r
- "t_{TOF}-t_{track} [ps]", "N");\r
- hResTOFWrong->SetLineColor(kRed);\r
-\r
- // calorimeters\r
- TH1F* hEPHOS = CreateHisto("hEPHOS", "PHOS", 100, 0, 5, "E [GeV]", "N");\r
- TH1F* hEEMCAL = CreateHisto("hEEMCAL", "EMCAL", 100, 0, 50, "E [GeV]", "N");\r
-\r
- // muons\r
- TH1F* hPtMUON = CreateHisto("hPtMUON", "MUON", 100, 0, 20, \r
- "p_{t} [GeV/c]", "N");\r
-\r
- // V0s and cascades\r
- TH1F* hMassK0 = CreateHisto("hMassK0", "K^{0}", 100, 0.4, 0.6, \r
- "M(#pi^{+}#pi^{-}) [GeV/c^{2}]", "N");\r
- TH1F* hMassLambda = CreateHisto("hMassLambda", "#Lambda", 100, 1.0, 1.2, \r
- "M(p#pi^{-}) [GeV/c^{2}]", "N");\r
- TH1F* hMassLambdaBar = CreateHisto("hMassLambdaBar", "#bar{#Lambda}", \r
- 100, 1.0, 1.2, \r
- "M(#bar{p}#pi^{+}) [GeV/c^{2}]", "N");\r
- Int_t nGenV0s = 0;\r
- Int_t nRecV0s = 0;\r
- TH1F* hMassXi = CreateHisto("hMassXi", "#Xi", 100, 1.2, 1.5, \r
- "M(#Lambda#pi) [GeV/c^{2}]", "N");\r
- TH1F* hMassOmega = CreateHisto("hMassOmega", "#Omega", 100, 1.5, 1.8, \r
- "M(#LambdaK) [GeV/c^{2}]", "N");\r
- Int_t nGenCascades = 0;\r
- Int_t nRecCascades = 0;\r
-\r
- // loop over events\r
- for (Int_t iEvent = 0; iEvent < runLoader->GetNumberOfEvents(); iEvent++) {\r
- runLoader->GetEvent(iEvent);\r
-\r
- // select simulated primary particles, V0s and cascades\r
- AliStack* stack = gAlice->Stack();\r
- Int_t nParticles = stack->GetNtrack();\r
- TArrayF vertex(3);\r
- runLoader->GetHeader()->GenEventHeader()->PrimaryVertex(vertex);\r
- TObjArray selParticles;\r
- TObjArray selV0s;\r
- TObjArray selCascades;\r
- for (Int_t iParticle = 0; iParticle < nParticles; iParticle++) {\r
- TParticle* particle = stack->Particle(iParticle);\r
- if (!particle) continue;\r
- if (particle->Pt() < 0.001) continue;\r
- if (TMath::Abs(particle->Eta()) > 0.9) continue;\r
- TVector3 dVertex(particle->Vx() - vertex[0], \r
- particle->Vy() - vertex[1],\r
- particle->Vz() - vertex[2]);\r
- if (dVertex.Mag() > 0.0001) continue;\r
-\r
- switch (TMath::Abs(particle->GetPdgCode())) {\r
- case kElectron:\r
- case kMuonMinus:\r
- case kPiPlus:\r
- case kKPlus:\r
- case kProton: {\r
- if (particle->Pt() > minPt) {\r
- selParticles.Add(particle);\r
- nGen++;\r
- hGen->Fill(particle->Pt());\r
- }\r
- break;\r
- }\r
- case kK0Short:\r
- case kLambda0: {\r
- if (particle->Pt() > cutPtV0) {\r
- nGenV0s++;\r
- selV0s.Add(particle);\r
- }\r
- break;\r
- }\r
- case kXiMinus:\r
- case kOmegaMinus: {\r
- if (particle->Pt() > cutPtCascade) {\r
- nGenCascades++;\r
- selCascades.Add(particle);\r
- }\r
- break;\r
- }\r
- default: break;\r
- }\r
- }\r
-\r
- // get the event summary data\r
- tree->GetEvent(iEvent);\r
- if (!esd) {\r
- Error("CheckESD", "no ESD object found for event %d", iEvent);\r
- return kFALSE;\r
- }\r
-\r
- // loop over tracks\r
- for (Int_t iTrack = 0; iTrack < esd->GetNumberOfTracks(); iTrack++) {\r
- AliESDtrack* track = esd->GetTrack(iTrack);\r
-\r
- // select tracks of selected particles\r
- Int_t label = TMath::Abs(track->GetLabel());\r
- if (label > stack->GetNtrack()) continue; // background\r
- TParticle* particle = stack->Particle(label);\r
- if (!selParticles.Contains(particle)) continue;\r
- if ((track->GetStatus() & AliESDtrack::kITSrefit) == 0) continue;\r
- if (track->GetConstrainedChi2() > 1e9) continue;\r
- selParticles.Remove(particle); // don't count multiple tracks\r
-\r
- nRec++;\r
- hRec->Fill(particle->Pt());\r
- if (track->GetLabel() < 0) nFake++;\r
-\r
- // resolutions\r
- Double_t p[3];\r
- track->GetConstrainedPxPyPz(p);\r
- TVector3 pTrack(p);\r
- hResPtInv->Fill(100. * (1./pTrack.Pt() - 1./particle->Pt()) * \r
- particle->Pt());\r
- hResPhi->Fill(1000. * (pTrack.Phi() - particle->Phi()));\r
- hResTheta->Fill(1000. * (pTrack.Theta() - particle->Theta()));\r
-\r
- // PID\r
- if ((track->GetStatus() & AliESDtrack::kESDpid) == 0) continue;\r
- Int_t iGen = 5;\r
- for (Int_t i = 0; i < 5; i++) {\r
- if (TMath::Abs(particle->GetPdgCode()) == partCode[i]) iGen = i;\r
- }\r
- Double_t probability[5];\r
- track->GetESDpid(probability);\r
- Double_t pMax = 0;\r
- Int_t iRec = 0;\r
- for (Int_t i = 0; i < 5; i++) {\r
- probability[i] *= partFrac[i];\r
- if (probability[i] > pMax) {\r
- pMax = probability[i];\r
- iRec = i;\r
- }\r
- }\r
- identified[iGen][iRec]++;\r
- if (iGen == iRec) nIdentified++;\r
-\r
- // dE/dx and TOF\r
- Double_t time[5];\r
- track->GetIntegratedTimes(time);\r
- if (iGen == iRec) {\r
- hDEdxRight->Fill(pTrack.Mag(), track->GetTPCsignal());\r
- if ((track->GetStatus() & AliESDtrack::kTOFpid) != 0) {\r
- hResTOFRight->Fill(track->GetTOFsignal() - time[iRec]);\r
- }\r
- } else {\r
- hDEdxWrong->Fill(pTrack.Mag(), track->GetTPCsignal());\r
- if ((track->GetStatus() & AliESDtrack::kTOFpid) != 0) {\r
- hResTOFWrong->Fill(track->GetTOFsignal() - time[iRec]);\r
- }\r
- }\r
- }\r
-\r
- // loop over muon tracks\r
- {\r
- for (Int_t iTrack = 0; iTrack < esd->GetNumberOfMuonTracks(); iTrack++) {\r
- AliESDMuonTrack* muonTrack = esd->GetMuonTrack(iTrack);\r
- Double_t ptInv = TMath::Abs(muonTrack->GetInverseBendingMomentum());\r
- if (ptInv > 0.001) {\r
- hPtMUON->Fill(1./ptInv);\r
- }\r
- }\r
- }\r
-\r
- // loop over V0s\r
- for (Int_t iV0 = 0; iV0 < esd->GetNumberOfV0s(); iV0++) {\r
- AliESDv0* v0 = esd->GetV0(iV0);\r
- if (v0->GetOnFlyStatus()) continue;\r
- v0->ChangeMassHypothesis(kK0Short);\r
- hMassK0->Fill(v0->GetEffMass());\r
- v0->ChangeMassHypothesis(kLambda0);\r
- hMassLambda->Fill(v0->GetEffMass());\r
- v0->ChangeMassHypothesis(kLambda0Bar);\r
- hMassLambdaBar->Fill(v0->GetEffMass());\r
-\r
- Int_t negLabel = TMath::Abs(esd->GetTrack(v0->GetNindex())->GetLabel());\r
- if (negLabel > stack->GetNtrack()) continue; // background\r
- Int_t negMother = stack->Particle(negLabel)->GetMother(0);\r
- if (negMother < 0) continue;\r
- Int_t posLabel = TMath::Abs(esd->GetTrack(v0->GetPindex())->GetLabel());\r
- if (posLabel > stack->GetNtrack()) continue; // background\r
- Int_t posMother = stack->Particle(posLabel)->GetMother(0);\r
- if (negMother != posMother) continue;\r
- TParticle* particle = stack->Particle(negMother);\r
- if (!selV0s.Contains(particle)) continue;\r
- selV0s.Remove(particle);\r
- nRecV0s++;\r
- }\r
-\r
- // loop over Cascades\r
- for (Int_t iCascade = 0; iCascade < esd->GetNumberOfCascades(); \r
- iCascade++) {\r
- AliESDcascade* cascade = esd->GetCascade(iCascade);\r
- Double_t v0q;\r
- cascade->ChangeMassHypothesis(v0q,kXiMinus);\r
- hMassXi->Fill(cascade->GetEffMass());\r
- cascade->ChangeMassHypothesis(v0q,kOmegaMinus);\r
- hMassOmega->Fill(cascade->GetEffMass());\r
-\r
- Int_t negLabel = TMath::Abs(esd->GetTrack(cascade->GetNindex())\r
- ->GetLabel());\r
- if (negLabel > stack->GetNtrack()) continue; // background\r
- Int_t negMother = stack->Particle(negLabel)->GetMother(0);\r
- if (negMother < 0) continue;\r
- Int_t posLabel = TMath::Abs(esd->GetTrack(cascade->GetPindex())\r
- ->GetLabel());\r
- if (posLabel > stack->GetNtrack()) continue; // background\r
- Int_t posMother = stack->Particle(posLabel)->GetMother(0);\r
- if (negMother != posMother) continue;\r
- Int_t v0Mother = stack->Particle(negMother)->GetMother(0);\r
- if (v0Mother < 0) continue;\r
- Int_t bacLabel = TMath::Abs(esd->GetTrack(cascade->GetBindex())\r
- ->GetLabel());\r
- if (bacLabel > stack->GetNtrack()) continue; // background\r
- Int_t bacMother = stack->Particle(bacLabel)->GetMother(0);\r
- if (v0Mother != bacMother) continue;\r
- TParticle* particle = stack->Particle(v0Mother);\r
- if (!selCascades.Contains(particle)) continue;\r
- selCascades.Remove(particle);\r
- nRecCascades++;\r
- }\r
-\r
- // loop over the calorimeters clusters\r
- {\r
- Int_t nCaloCluster = esd->GetNumberOfCaloClusters();\r
- cout<<"CaloClusters "<<nCaloCluster<<endl;\r
- for (Int_t iCluster=0; iCluster<nCaloCluster; iCluster++){\r
- if(esd->GetCaloCluster(iCluster)->IsPHOS()) \r
- hEPHOS->Fill(esd->GetCaloCluster(iCluster)->E());\r
- if(esd->GetCaloCluster(iCluster)->GetClusterType()==AliESDCaloCluster::kEMCALClusterv1) {\r
- hEEMCAL->Fill(esd->GetCaloCluster(iCluster)->E());\r
- //cout<<esd->GetCaloCluster(iCluster)->E()<<endl;\r
- }\r
- }\r
- }\r
- }\r
-\r
- // perform checks\r
- if (nGen < checkNGenLow) {\r
- Warning("CheckESD", "low number of generated particles: %d", Int_t(nGen));\r
- }\r
-\r
- TH1F* hEff = CreateEffHisto(hGen, hRec);\r
-\r
- Info("CheckESD", "%d out of %d tracks reconstructed including %d "\r
- "fake tracks", nRec, nGen, nFake);\r
- if (nGen > 0) {\r
- // efficiency\r
- Double_t eff = nRec*1./nGen;\r
- Double_t effError = TMath::Sqrt(eff*(1.-eff) / nGen);\r
- Double_t fake = nFake*1./nGen;\r
- Double_t fakeError = TMath::Sqrt(fake*(1.-fake) / nGen);\r
- Info("CheckESD", "eff = (%.1f +- %.1f) %% fake = (%.1f +- %.1f) %%",\r
- 100.*eff, 100.*effError, 100.*fake, 100.*fakeError);\r
-\r
- if (eff < checkEffLow - checkEffSigma*effError) {\r
- Warning("CheckESD", "low efficiency: (%.1f +- %.1f) %%", \r
- 100.*eff, 100.*effError);\r
- }\r
- if (fake > checkFakeHigh + checkFakeSigma*fakeError) {\r
- Warning("CheckESD", "high fake: (%.1f +- %.1f) %%", \r
- 100.*fake, 100.*fakeError);\r
- }\r
-\r
- // resolutions\r
- Double_t res, resError;\r
- if (FitHisto(hResPtInv, res, resError)) {\r
- Info("CheckESD", "relative inverse pt resolution = (%.1f +- %.1f) %%",\r
- res, resError);\r
- if (res > checkResPtInvHigh + checkResPtInvSigma*resError) {\r
- Warning("CheckESD", "bad pt resolution: (%.1f +- %.1f) %%", \r
- res, resError);\r
- }\r
- }\r
-\r
- if (FitHisto(hResPhi, res, resError)) {\r
- Info("CheckESD", "phi resolution = (%.1f +- %.1f) mrad", res, resError);\r
- if (res > checkResPhiHigh + checkResPhiSigma*resError) {\r
- Warning("CheckESD", "bad phi resolution: (%.1f +- %.1f) mrad", \r
- res, resError);\r
- }\r
- }\r
-\r
- if (FitHisto(hResTheta, res, resError)) {\r
- Info("CheckESD", "theta resolution = (%.1f +- %.1f) mrad", \r
- res, resError);\r
- if (res > checkResThetaHigh + checkResThetaSigma*resError) {\r
- Warning("CheckESD", "bad theta resolution: (%.1f +- %.1f) mrad", \r
- res, resError);\r
- }\r
- }\r
-\r
- // PID\r
- if (nRec > 0) {\r
- Double_t eff = nIdentified*1./nRec;\r
- Double_t effError = TMath::Sqrt(eff*(1.-eff) / nRec);\r
- Info("CheckESD", "PID eff = (%.1f +- %.1f) %%", \r
- 100.*eff, 100.*effError);\r
- if (eff < checkPIDEffLow - checkPIDEffSigma*effError) {\r
- Warning("CheckESD", "low PID efficiency: (%.1f +- %.1f) %%", \r
- 100.*eff, 100.*effError);\r
- }\r
- }\r
-\r
- printf("%9s:", "gen\\rec");\r
- for (Int_t iRec = 0; iRec < AliPID::kSPECIES; iRec++) {\r
- printf("%9s", partName[iRec]);\r
- }\r
- printf("\n");\r
- for (Int_t iGen = 0; iGen < AliPID::kSPECIES+1; iGen++) {\r
- printf("%9s:", partName[iGen]);\r
- for (Int_t iRec = 0; iRec < AliPID::kSPECIES; iRec++) {\r
- printf("%9d", identified[iGen][iRec]);\r
- }\r
- printf("\n");\r
- }\r
-\r
- if (FitHisto(hResTOFRight, res, resError)) {\r
- Info("CheckESD", "TOF resolution = (%.1f +- %.1f) ps", res, resError);\r
- if (res > checkResTOFHigh + checkResTOFSigma*resError) {\r
- Warning("CheckESD", "bad TOF resolution: (%.1f +- %.1f) ps", \r
- res, resError);\r
- }\r
- }\r
-\r
- // calorimeters\r
- if (hEPHOS->Integral() < checkPHOSNLow) {\r
- Warning("CheckESD", "low number of PHOS particles: %d", \r
- Int_t(hEPHOS->Integral()));\r
- } else {\r
- Double_t mean = hEPHOS->GetMean();\r
- if (mean < checkPHOSEnergyLow) {\r
- Warning("CheckESD", "low mean PHOS energy: %.1f GeV", mean);\r
- } else if (mean > checkPHOSEnergyHigh) {\r
- Warning("CheckESD", "high mean PHOS energy: %.1f GeV", mean);\r
- }\r
- }\r
-\r
- if (hEEMCAL->Integral() < checkEMCALNLow) {\r
- Warning("CheckESD", "low number of EMCAL particles: %d", \r
- Int_t(hEEMCAL->Integral()));\r
- } else {\r
- Double_t mean = hEEMCAL->GetMean();\r
- if (mean < checkEMCALEnergyLow) {\r
- Warning("CheckESD", "low mean EMCAL energy: %.1f GeV", mean);\r
- } else if (mean > checkEMCALEnergyHigh) {\r
- Warning("CheckESD", "high mean EMCAL energy: %.1f GeV", mean);\r
- }\r
- }\r
-\r
- // muons\r
- if (hPtMUON->Integral() < checkMUONNLow) {\r
- Warning("CheckESD", "low number of MUON particles: %d", \r
- Int_t(hPtMUON->Integral()));\r
- } else {\r
- Double_t mean = hPtMUON->GetMean();\r
- if (mean < checkMUONPtLow) {\r
- Warning("CheckESD", "low mean MUON pt: %.1f GeV/c", mean);\r
- } else if (mean > checkMUONPtHigh) {\r
- Warning("CheckESD", "high mean MUON pt: %.1f GeV/c", mean);\r
- }\r
- }\r
-\r
- // V0s\r
- if (nGenV0s > 0) {\r
- Double_t eff = nRecV0s*1./nGenV0s;\r
- Double_t effError = TMath::Sqrt(eff*(1.-eff) / nGenV0s);\r
- if (effError == 0) effError = checkV0EffLow / TMath::Sqrt(1.*nGenV0s);\r
- Info("CheckESD", "V0 eff = (%.1f +- %.1f) %%", \r
- 100.*eff, 100.*effError);\r
- if (eff < checkV0EffLow - checkV0EffSigma*effError) {\r
- Warning("CheckESD", "low V0 efficiency: (%.1f +- %.1f) %%", \r
- 100.*eff, 100.*effError);\r
- }\r
- }\r
-\r
- // Cascades\r
- if (nGenCascades > 0) {\r
- Double_t eff = nRecCascades*1./nGenCascades;\r
- Double_t effError = TMath::Sqrt(eff*(1.-eff) / nGenCascades);\r
- if (effError == 0) effError = checkV0EffLow / \r
- TMath::Sqrt(1.*nGenCascades);\r
- Info("CheckESD", "Cascade eff = (%.1f +- %.1f) %%", \r
- 100.*eff, 100.*effError);\r
- if (eff < checkCascadeEffLow - checkCascadeEffSigma*effError) {\r
- Warning("CheckESD", "low Cascade efficiency: (%.1f +- %.1f) %%", \r
- 100.*eff, 100.*effError);\r
- }\r
- }\r
- }\r
-\r
- // draw the histograms if not in batch mode\r
- if (!gROOT->IsBatch()) {\r
- new TCanvas;\r
- hEff->DrawCopy();\r
- new TCanvas;\r
- hResPtInv->DrawCopy("E");\r
- new TCanvas;\r
- hResPhi->DrawCopy("E");\r
- new TCanvas;\r
- hResTheta->DrawCopy("E");\r
- new TCanvas;\r
- hDEdxRight->DrawCopy();\r
- hDEdxWrong->DrawCopy("SAME");\r
- new TCanvas;\r
- hResTOFRight->DrawCopy("E");\r
- hResTOFWrong->DrawCopy("SAME");\r
- new TCanvas;\r
- hEPHOS->DrawCopy("E");\r
- new TCanvas;\r
- hEEMCAL->DrawCopy("E");\r
- new TCanvas;\r
- hPtMUON->DrawCopy("E");\r
- new TCanvas;\r
- hMassK0->DrawCopy("E");\r
- new TCanvas;\r
- hMassLambda->DrawCopy("E");\r
- new TCanvas;\r
- hMassLambdaBar->DrawCopy("E");\r
- new TCanvas;\r
- hMassXi->DrawCopy("E");\r
- new TCanvas;\r
- hMassOmega->DrawCopy("E");\r
- }\r
-\r
- // write the output histograms to a file\r
- TFile* outputFile = TFile::Open("check.root", "recreate");\r
- if (!outputFile || !outputFile->IsOpen()) {\r
- Error("CheckESD", "opening output file check.root failed");\r
- return kFALSE;\r
- }\r
- hEff->Write();\r
- hResPtInv->Write();\r
- hResPhi->Write();\r
- hResTheta->Write();\r
- hDEdxRight->Write();\r
- hDEdxWrong->Write();\r
- hResTOFRight->Write();\r
- hResTOFWrong->Write();\r
- hEPHOS->Write();\r
- hEEMCAL->Write();\r
- hPtMUON->Write();\r
- hMassK0->Write();\r
- hMassLambda->Write();\r
- hMassLambdaBar->Write();\r
- hMassXi->Write();\r
- hMassOmega->Write();\r
- outputFile->Close();\r
- delete outputFile;\r
-\r
- // clean up\r
- delete hGen;\r
- delete hRec;\r
- delete hEff;\r
- delete hResPtInv;\r
- delete hResPhi;\r
- delete hResTheta;\r
- delete hDEdxRight;\r
- delete hDEdxWrong;\r
- delete hResTOFRight;\r
- delete hResTOFWrong;\r
- delete hEPHOS;\r
- delete hEEMCAL;\r
- delete hPtMUON;\r
- delete hMassK0;\r
- delete hMassLambda;\r
- delete hMassLambdaBar;\r
- delete hMassXi;\r
- delete hMassOmega;\r
-\r
- delete esd;\r
- esdFile->Close();\r
- delete esdFile;\r
-\r
- runLoader->UnloadHeader();\r
- runLoader->UnloadKinematics();\r
- delete runLoader;\r
-\r
- // result of check\r
- Info("CheckESD", "check of ESD was successfull");\r
- return kTRUE;\r
-}\r
+#if !defined( __CINT__) || defined(__MAKECINT__)
+#include <TFile.h>
+#include <TError.h>
+#include <TH1.h>
+#include <TH2.h>
+#include <TF1.h>
+#include <TCanvas.h>
+#include <TVector3.h>
+#include <TPDGCode.h>
+#include <TParticle.h>
+
+#include "AliRunLoader.h"
+#include "AliLoader.h"
+#include "AliESDEvent.h"
+#include "AliRun.h"
+#include "AliStack.h"
+#include "AliHeader.h"
+#include "AliGenEventHeader.h"
+#include "AliPID.h"
+#else
+const Int_t kXiMinus = 3312;
+const Int_t kOmegaMinus = 3334;
+#endif
+
+
+TH1F* CreateHisto(const char* name, const char* title,
+ Int_t nBins, Double_t xMin, Double_t xMax,
+ const char* xLabel = NULL, const char* yLabel = NULL)
+{
+// create a histogram
+
+ TH1F* result = new TH1F(name, title, nBins, xMin, xMax);
+ result->SetOption("E");
+ if (xLabel) result->GetXaxis()->SetTitle(xLabel);
+ if (yLabel) result->GetYaxis()->SetTitle(yLabel);
+ result->SetMarkerStyle(kFullCircle);
+ return result;
+}
+
+TH1F* CreateEffHisto(TH1F* hGen, TH1F* hRec)
+{
+// create an efficiency histogram
+
+ Int_t nBins = hGen->GetNbinsX();
+ TH1F* hEff = (TH1F*) hGen->Clone("hEff");
+ hEff->SetTitle("");
+ hEff->SetStats(kFALSE);
+ hEff->SetMinimum(0.);
+ hEff->SetMaximum(110.);
+ hEff->GetYaxis()->SetTitle("#epsilon [%]");
+
+ for (Int_t iBin = 0; iBin <= nBins; iBin++) {
+ Double_t nGen = hGen->GetBinContent(iBin);
+ Double_t nRec = hRec->GetBinContent(iBin);
+ if (nGen > 0) {
+ Double_t eff = nRec/nGen;
+ hEff->SetBinContent(iBin, 100. * eff);
+ Double_t error = sqrt(eff*(1.-eff) / nGen);
+ if (error == 0) error = 0.0001;
+ hEff->SetBinError(iBin, 100. * error);
+ } else {
+ hEff->SetBinContent(iBin, -100.);
+ hEff->SetBinError(iBin, 0);
+ }
+ }
+
+ return hEff;
+}
+
+Bool_t FitHisto(TH1* histo, Double_t& res, Double_t& resError)
+{
+// fit a gaussian to a histogram
+
+ static TF1* fitFunc = new TF1("fitFunc", "gaus");
+ fitFunc->SetLineWidth(2);
+ fitFunc->SetFillStyle(0);
+ Double_t maxFitRange = 2;
+
+ if (histo->Integral() > 50) {
+ Float_t mean = histo->GetMean();
+ Float_t rms = histo->GetRMS();
+ fitFunc->SetRange(mean - maxFitRange*rms, mean + maxFitRange*rms);
+ fitFunc->SetParameters(mean, rms);
+ histo->Fit(fitFunc, "QRI0");
+ histo->GetFunction("fitFunc")->ResetBit(1<<9);
+ res = TMath::Abs(fitFunc->GetParameter(2));
+ resError = TMath::Abs(fitFunc->GetParError(2));
+ return kTRUE;
+ }
+
+ return kFALSE;
+}
+
+
+Bool_t CheckESD(const char* gAliceFileName = "galice.root",
+ const char* esdFileName = "AliESDs.root")
+{
+// check the content of the ESD
+
+ // check values
+ Int_t checkNGenLow = 1;
+
+ Double_t checkEffLow = 0.5;
+ Double_t checkEffSigma = 3;
+ Double_t checkFakeHigh = 0.5;
+ Double_t checkFakeSigma = 3;
+
+ Double_t checkResPtInvHigh = 5;
+ Double_t checkResPtInvSigma = 3;
+ Double_t checkResPhiHigh = 10;
+ Double_t checkResPhiSigma = 3;
+ Double_t checkResThetaHigh = 10;
+ Double_t checkResThetaSigma = 3;
+
+ Double_t checkPIDEffLow = 0.5;
+ Double_t checkPIDEffSigma = 3;
+ Double_t checkResTOFHigh = 500;
+ Double_t checkResTOFSigma = 3;
+
+ Double_t checkPHOSNLow = 5;
+ Double_t checkPHOSEnergyLow = 0.3;
+ Double_t checkPHOSEnergyHigh = 1.0;
+ Double_t checkEMCALNLow = 50;
+ Double_t checkEMCALEnergyLow = 0.05;
+ Double_t checkEMCALEnergyHigh = 1.0;
+
+ Double_t checkMUONNLow = 1;
+ Double_t checkMUONPtLow = 0.5;
+ Double_t checkMUONPtHigh = 10.;
+
+ Double_t cutPtV0 = 0.3;
+ Double_t checkV0EffLow = 0.02;
+ Double_t checkV0EffSigma = 3;
+ Double_t cutPtCascade = 0.5;
+ Double_t checkCascadeEffLow = 0.01;
+ Double_t checkCascadeEffSigma = 3;
+
+ // open run loader and load gAlice, kinematics and header
+ AliRunLoader* runLoader = AliRunLoader::Open(gAliceFileName);
+ if (!runLoader) {
+ Error("CheckESD", "getting run loader from file %s failed",
+ gAliceFileName);
+ return kFALSE;
+ }
+ runLoader->LoadgAlice();
+ gAlice = runLoader->GetAliRun();
+ if (!gAlice) {
+ Error("CheckESD", "no galice object found");
+ return kFALSE;
+ }
+ runLoader->LoadKinematics();
+ runLoader->LoadHeader();
+
+ // open the ESD file
+ TFile* esdFile = TFile::Open(esdFileName);
+ if (!esdFile || !esdFile->IsOpen()) {
+ Error("CheckESD", "opening ESD file %s failed", esdFileName);
+ return kFALSE;
+ }
+ AliESDEvent* esd = new AliESDEvent;
+ TTree* tree = (TTree*) esdFile->Get("esdTree");
+ if (!tree) {
+ Error("CheckESD", "no ESD tree found");
+ return kFALSE;
+ }
+ esd->ReadFromTree(tree);
+
+ // efficiency and resolution histograms
+ Int_t nBinsPt = 15;
+ Float_t minPt = 0.1;
+ Float_t maxPt = 3.1;
+ TH1F* hGen = CreateHisto("hGen", "generated tracks",
+ nBinsPt, minPt, maxPt, "p_{t} [GeV/c]", "N");
+ TH1F* hRec = CreateHisto("hRec", "reconstructed tracks",
+ nBinsPt, minPt, maxPt, "p_{t} [GeV/c]", "N");
+ Int_t nGen = 0;
+ Int_t nRec = 0;
+ Int_t nFake = 0;
+
+ TH1F* hResPtInv = CreateHisto("hResPtInv", "", 100, -10, 10,
+ "(p_{t,rec}^{-1}-p_{t,sim}^{-1}) / p_{t,sim}^{-1} [%]", "N");
+ TH1F* hResPhi = CreateHisto("hResPhi", "", 100, -20, 20,
+ "#phi_{rec}-#phi_{sim} [mrad]", "N");
+ TH1F* hResTheta = CreateHisto("hResTheta", "", 100, -20, 20,
+ "#theta_{rec}-#theta_{sim} [mrad]", "N");
+
+ // PID
+ Int_t partCode[] =
+ {kElectron, kMuonMinus, kPiPlus, kKPlus, kProton};
+ const char* partName[] =
+ {"electron", "muon", "pion", "kaon", "proton", "other"};
+ Double_t partFrac[] =
+ {0.01, 0.01, 0.85, 0.10, 0.05};
+ Int_t identified[6][5];
+ for (Int_t iGen = 0; iGen < 6; iGen++) {
+ for (Int_t iRec = 0; iRec < 5; iRec++) {
+ identified[iGen][iRec] = 0;
+ }
+ }
+ Int_t nIdentified = 0;
+
+ // dE/dx and TOF
+ TH2F* hDEdxRight = new TH2F("hDEdxRight", "", 300, 0, 3, 100, 0, 400);
+ hDEdxRight->SetStats(kFALSE);
+ hDEdxRight->GetXaxis()->SetTitle("p [GeV/c]");
+ hDEdxRight->GetYaxis()->SetTitle("dE/dx_{TPC}");
+ hDEdxRight->SetMarkerStyle(kFullCircle);
+ hDEdxRight->SetMarkerSize(0.4);
+ TH2F* hDEdxWrong = new TH2F("hDEdxWrong", "", 300, 0, 3, 100, 0, 400);
+ hDEdxWrong->SetStats(kFALSE);
+ hDEdxWrong->GetXaxis()->SetTitle("p [GeV/c]");
+ hDEdxWrong->GetYaxis()->SetTitle("dE/dx_{TPC}");
+ hDEdxWrong->SetMarkerStyle(kFullCircle);
+ hDEdxWrong->SetMarkerSize(0.4);
+ hDEdxWrong->SetMarkerColor(kRed);
+ TH1F* hResTOFRight = CreateHisto("hResTOFRight", "", 100, -1000, 1000,
+ "t_{TOF}-t_{track} [ps]", "N");
+ TH1F* hResTOFWrong = CreateHisto("hResTOFWrong", "", 100, -1000, 1000,
+ "t_{TOF}-t_{track} [ps]", "N");
+ hResTOFWrong->SetLineColor(kRed);
+
+ // calorimeters
+ TH1F* hEPHOS = CreateHisto("hEPHOS", "PHOS", 100, 0, 5, "E [GeV]", "N");
+ TH1F* hEEMCAL = CreateHisto("hEEMCAL", "EMCAL", 100, 0, 50, "E [GeV]", "N");
+
+ // muons
+ TH1F* hPtMUON = CreateHisto("hPtMUON", "MUON", 100, 0, 20,
+ "p_{t} [GeV/c]", "N");
+
+ // V0s and cascades
+ TH1F* hMassK0 = CreateHisto("hMassK0", "K^{0}", 100, 0.4, 0.6,
+ "M(#pi^{+}#pi^{-}) [GeV/c^{2}]", "N");
+ TH1F* hMassLambda = CreateHisto("hMassLambda", "#Lambda", 100, 1.0, 1.2,
+ "M(p#pi^{-}) [GeV/c^{2}]", "N");
+ TH1F* hMassLambdaBar = CreateHisto("hMassLambdaBar", "#bar{#Lambda}",
+ 100, 1.0, 1.2,
+ "M(#bar{p}#pi^{+}) [GeV/c^{2}]", "N");
+ Int_t nGenV0s = 0;
+ Int_t nRecV0s = 0;
+ TH1F* hMassXi = CreateHisto("hMassXi", "#Xi", 100, 1.2, 1.5,
+ "M(#Lambda#pi) [GeV/c^{2}]", "N");
+ TH1F* hMassOmega = CreateHisto("hMassOmega", "#Omega", 100, 1.5, 1.8,
+ "M(#LambdaK) [GeV/c^{2}]", "N");
+ Int_t nGenCascades = 0;
+ Int_t nRecCascades = 0;
+
+ // loop over events
+ for (Int_t iEvent = 0; iEvent < runLoader->GetNumberOfEvents(); iEvent++) {
+ runLoader->GetEvent(iEvent);
+
+ // select simulated primary particles, V0s and cascades
+ AliStack* stack = gAlice->Stack();
+ Int_t nParticles = stack->GetNtrack();
+ TArrayF vertex(3);
+ runLoader->GetHeader()->GenEventHeader()->PrimaryVertex(vertex);
+ TObjArray selParticles;
+ TObjArray selV0s;
+ TObjArray selCascades;
+ for (Int_t iParticle = 0; iParticle < nParticles; iParticle++) {
+ TParticle* particle = stack->Particle(iParticle);
+ if (!particle) continue;
+ if (particle->Pt() < 0.001) continue;
+ if (TMath::Abs(particle->Eta()) > 0.9) continue;
+ TVector3 dVertex(particle->Vx() - vertex[0],
+ particle->Vy() - vertex[1],
+ particle->Vz() - vertex[2]);
+ if (dVertex.Mag() > 0.0001) continue;
+
+ switch (TMath::Abs(particle->GetPdgCode())) {
+ case kElectron:
+ case kMuonMinus:
+ case kPiPlus:
+ case kKPlus:
+ case kProton: {
+ if (particle->Pt() > minPt) {
+ selParticles.Add(particle);
+ nGen++;
+ hGen->Fill(particle->Pt());
+ }
+ break;
+ }
+ case kK0Short:
+ case kLambda0: {
+ if (particle->Pt() > cutPtV0) {
+ nGenV0s++;
+ selV0s.Add(particle);
+ }
+ break;
+ }
+ case kXiMinus:
+ case kOmegaMinus: {
+ if (particle->Pt() > cutPtCascade) {
+ nGenCascades++;
+ selCascades.Add(particle);
+ }
+ break;
+ }
+ default: break;
+ }
+ }
+
+ // get the event summary data
+ tree->GetEvent(iEvent);
+ if (!esd) {
+ Error("CheckESD", "no ESD object found for event %d", iEvent);
+ return kFALSE;
+ }
+
+ // loop over tracks
+ for (Int_t iTrack = 0; iTrack < esd->GetNumberOfTracks(); iTrack++) {
+ AliESDtrack* track = esd->GetTrack(iTrack);
+
+ // select tracks of selected particles
+ Int_t label = TMath::Abs(track->GetLabel());
+ if (label > stack->GetNtrack()) continue; // background
+ TParticle* particle = stack->Particle(label);
+ if (!selParticles.Contains(particle)) continue;
+ if ((track->GetStatus() & AliESDtrack::kITSrefit) == 0) continue;
+ if (track->GetConstrainedChi2() > 1e9) continue;
+ selParticles.Remove(particle); // don't count multiple tracks
+
+ nRec++;
+ hRec->Fill(particle->Pt());
+ if (track->GetLabel() < 0) nFake++;
+
+ // resolutions
+ Double_t p[3];
+ track->GetConstrainedPxPyPz(p);
+ TVector3 pTrack(p);
+ hResPtInv->Fill(100. * (1./pTrack.Pt() - 1./particle->Pt()) *
+ particle->Pt());
+ hResPhi->Fill(1000. * (pTrack.Phi() - particle->Phi()));
+ hResTheta->Fill(1000. * (pTrack.Theta() - particle->Theta()));
+
+ // PID
+ if ((track->GetStatus() & AliESDtrack::kESDpid) == 0) continue;
+ Int_t iGen = 5;
+ for (Int_t i = 0; i < 5; i++) {
+ if (TMath::Abs(particle->GetPdgCode()) == partCode[i]) iGen = i;
+ }
+ Double_t probability[5];
+ track->GetESDpid(probability);
+ Double_t pMax = 0;
+ Int_t iRec = 0;
+ for (Int_t i = 0; i < 5; i++) {
+ probability[i] *= partFrac[i];
+ if (probability[i] > pMax) {
+ pMax = probability[i];
+ iRec = i;
+ }
+ }
+ identified[iGen][iRec]++;
+ if (iGen == iRec) nIdentified++;
+
+ // dE/dx and TOF
+ Double_t time[5];
+ track->GetIntegratedTimes(time);
+ if (iGen == iRec) {
+ hDEdxRight->Fill(pTrack.Mag(), track->GetTPCsignal());
+ if ((track->GetStatus() & AliESDtrack::kTOFpid) != 0) {
+ hResTOFRight->Fill(track->GetTOFsignal() - time[iRec]);
+ }
+ } else {
+ hDEdxWrong->Fill(pTrack.Mag(), track->GetTPCsignal());
+ if ((track->GetStatus() & AliESDtrack::kTOFpid) != 0) {
+ hResTOFWrong->Fill(track->GetTOFsignal() - time[iRec]);
+ }
+ }
+ }
+
+ // loop over muon tracks
+ {
+ for (Int_t iTrack = 0; iTrack < esd->GetNumberOfMuonTracks(); iTrack++) {
+ AliESDMuonTrack* muonTrack = esd->GetMuonTrack(iTrack);
+ Double_t ptInv = TMath::Abs(muonTrack->GetInverseBendingMomentum());
+ if (ptInv > 0.001) {
+ hPtMUON->Fill(1./ptInv);
+ }
+ }
+ }
+
+ // loop over V0s
+ for (Int_t iV0 = 0; iV0 < esd->GetNumberOfV0s(); iV0++) {
+ AliESDv0* v0 = esd->GetV0(iV0);
+ if (v0->GetOnFlyStatus()) continue;
+ v0->ChangeMassHypothesis(kK0Short);
+ hMassK0->Fill(v0->GetEffMass());
+ v0->ChangeMassHypothesis(kLambda0);
+ hMassLambda->Fill(v0->GetEffMass());
+ v0->ChangeMassHypothesis(kLambda0Bar);
+ hMassLambdaBar->Fill(v0->GetEffMass());
+
+ Int_t negLabel = TMath::Abs(esd->GetTrack(v0->GetNindex())->GetLabel());
+ if (negLabel > stack->GetNtrack()) continue; // background
+ Int_t negMother = stack->Particle(negLabel)->GetMother(0);
+ if (negMother < 0) continue;
+ Int_t posLabel = TMath::Abs(esd->GetTrack(v0->GetPindex())->GetLabel());
+ if (posLabel > stack->GetNtrack()) continue; // background
+ Int_t posMother = stack->Particle(posLabel)->GetMother(0);
+ if (negMother != posMother) continue;
+ TParticle* particle = stack->Particle(negMother);
+ if (!selV0s.Contains(particle)) continue;
+ selV0s.Remove(particle);
+ nRecV0s++;
+ }
+
+ // loop over Cascades
+ for (Int_t iCascade = 0; iCascade < esd->GetNumberOfCascades();
+ iCascade++) {
+ AliESDcascade* cascade = esd->GetCascade(iCascade);
+ Double_t v0q;
+ cascade->ChangeMassHypothesis(v0q,kXiMinus);
+ hMassXi->Fill(cascade->GetEffMass());
+ cascade->ChangeMassHypothesis(v0q,kOmegaMinus);
+ hMassOmega->Fill(cascade->GetEffMass());
+
+ Int_t negLabel = TMath::Abs(esd->GetTrack(cascade->GetNindex())
+ ->GetLabel());
+ if (negLabel > stack->GetNtrack()) continue; // background
+ Int_t negMother = stack->Particle(negLabel)->GetMother(0);
+ if (negMother < 0) continue;
+ Int_t posLabel = TMath::Abs(esd->GetTrack(cascade->GetPindex())
+ ->GetLabel());
+ if (posLabel > stack->GetNtrack()) continue; // background
+ Int_t posMother = stack->Particle(posLabel)->GetMother(0);
+ if (negMother != posMother) continue;
+ Int_t v0Mother = stack->Particle(negMother)->GetMother(0);
+ if (v0Mother < 0) continue;
+ Int_t bacLabel = TMath::Abs(esd->GetTrack(cascade->GetBindex())
+ ->GetLabel());
+ if (bacLabel > stack->GetNtrack()) continue; // background
+ Int_t bacMother = stack->Particle(bacLabel)->GetMother(0);
+ if (v0Mother != bacMother) continue;
+ TParticle* particle = stack->Particle(v0Mother);
+ if (!selCascades.Contains(particle)) continue;
+ selCascades.Remove(particle);
+ nRecCascades++;
+ }
+
+ // loop over the calorimeters clusters
+ {
+ Int_t nCaloCluster = esd->GetNumberOfCaloClusters();
+ cout<<"CaloClusters "<<nCaloCluster<<endl;
+ for (Int_t iCluster=0; iCluster<nCaloCluster; iCluster++){
+ if(esd->GetCaloCluster(iCluster)->IsPHOS())
+ hEPHOS->Fill(esd->GetCaloCluster(iCluster)->E());
+ if(esd->GetCaloCluster(iCluster)->GetClusterType()==AliESDCaloCluster::kEMCALClusterv1) {
+ hEEMCAL->Fill(esd->GetCaloCluster(iCluster)->E());
+ //cout<<esd->GetCaloCluster(iCluster)->E()<<endl;
+ }
+ }
+ }
+ }
+
+ // perform checks
+ if (nGen < checkNGenLow) {
+ Warning("CheckESD", "low number of generated particles: %d", Int_t(nGen));
+ }
+
+ TH1F* hEff = CreateEffHisto(hGen, hRec);
+
+ Info("CheckESD", "%d out of %d tracks reconstructed including %d "
+ "fake tracks", nRec, nGen, nFake);
+ if (nGen > 0) {
+ // efficiency
+ Double_t eff = nRec*1./nGen;
+ Double_t effError = TMath::Sqrt(eff*(1.-eff) / nGen);
+ Double_t fake = nFake*1./nGen;
+ Double_t fakeError = TMath::Sqrt(fake*(1.-fake) / nGen);
+ Info("CheckESD", "eff = (%.1f +- %.1f) %% fake = (%.1f +- %.1f) %%",
+ 100.*eff, 100.*effError, 100.*fake, 100.*fakeError);
+
+ if (eff < checkEffLow - checkEffSigma*effError) {
+ Warning("CheckESD", "low efficiency: (%.1f +- %.1f) %%",
+ 100.*eff, 100.*effError);
+ }
+ if (fake > checkFakeHigh + checkFakeSigma*fakeError) {
+ Warning("CheckESD", "high fake: (%.1f +- %.1f) %%",
+ 100.*fake, 100.*fakeError);
+ }
+
+ // resolutions
+ Double_t res, resError;
+ if (FitHisto(hResPtInv, res, resError)) {
+ Info("CheckESD", "relative inverse pt resolution = (%.1f +- %.1f) %%",
+ res, resError);
+ if (res > checkResPtInvHigh + checkResPtInvSigma*resError) {
+ Warning("CheckESD", "bad pt resolution: (%.1f +- %.1f) %%",
+ res, resError);
+ }
+ }
+
+ if (FitHisto(hResPhi, res, resError)) {
+ Info("CheckESD", "phi resolution = (%.1f +- %.1f) mrad", res, resError);
+ if (res > checkResPhiHigh + checkResPhiSigma*resError) {
+ Warning("CheckESD", "bad phi resolution: (%.1f +- %.1f) mrad",
+ res, resError);
+ }
+ }
+
+ if (FitHisto(hResTheta, res, resError)) {
+ Info("CheckESD", "theta resolution = (%.1f +- %.1f) mrad",
+ res, resError);
+ if (res > checkResThetaHigh + checkResThetaSigma*resError) {
+ Warning("CheckESD", "bad theta resolution: (%.1f +- %.1f) mrad",
+ res, resError);
+ }
+ }
+
+ // PID
+ if (nRec > 0) {
+ Double_t eff = nIdentified*1./nRec;
+ Double_t effError = TMath::Sqrt(eff*(1.-eff) / nRec);
+ Info("CheckESD", "PID eff = (%.1f +- %.1f) %%",
+ 100.*eff, 100.*effError);
+ if (eff < checkPIDEffLow - checkPIDEffSigma*effError) {
+ Warning("CheckESD", "low PID efficiency: (%.1f +- %.1f) %%",
+ 100.*eff, 100.*effError);
+ }
+ }
+
+ printf("%9s:", "gen\\rec");
+ for (Int_t iRec = 0; iRec < AliPID::kSPECIES; iRec++) {
+ printf("%9s", partName[iRec]);
+ }
+ printf("\n");
+ for (Int_t iGen = 0; iGen < AliPID::kSPECIES+1; iGen++) {
+ printf("%9s:", partName[iGen]);
+ for (Int_t iRec = 0; iRec < AliPID::kSPECIES; iRec++) {
+ printf("%9d", identified[iGen][iRec]);
+ }
+ printf("\n");
+ }
+
+ if (FitHisto(hResTOFRight, res, resError)) {
+ Info("CheckESD", "TOF resolution = (%.1f +- %.1f) ps", res, resError);
+ if (res > checkResTOFHigh + checkResTOFSigma*resError) {
+ Warning("CheckESD", "bad TOF resolution: (%.1f +- %.1f) ps",
+ res, resError);
+ }
+ }
+
+ // calorimeters
+ if (hEPHOS->Integral() < checkPHOSNLow) {
+ Warning("CheckESD", "low number of PHOS particles: %d",
+ Int_t(hEPHOS->Integral()));
+ } else {
+ Double_t mean = hEPHOS->GetMean();
+ if (mean < checkPHOSEnergyLow) {
+ Warning("CheckESD", "low mean PHOS energy: %.1f GeV", mean);
+ } else if (mean > checkPHOSEnergyHigh) {
+ Warning("CheckESD", "high mean PHOS energy: %.1f GeV", mean);
+ }
+ }
+
+ if (hEEMCAL->Integral() < checkEMCALNLow) {
+ Warning("CheckESD", "low number of EMCAL particles: %d",
+ Int_t(hEEMCAL->Integral()));
+ } else {
+ Double_t mean = hEEMCAL->GetMean();
+ if (mean < checkEMCALEnergyLow) {
+ Warning("CheckESD", "low mean EMCAL energy: %.1f GeV", mean);
+ } else if (mean > checkEMCALEnergyHigh) {
+ Warning("CheckESD", "high mean EMCAL energy: %.1f GeV", mean);
+ }
+ }
+
+ // muons
+ if (hPtMUON->Integral() < checkMUONNLow) {
+ Warning("CheckESD", "low number of MUON particles: %d",
+ Int_t(hPtMUON->Integral()));
+ } else {
+ Double_t mean = hPtMUON->GetMean();
+ if (mean < checkMUONPtLow) {
+ Warning("CheckESD", "low mean MUON pt: %.1f GeV/c", mean);
+ } else if (mean > checkMUONPtHigh) {
+ Warning("CheckESD", "high mean MUON pt: %.1f GeV/c", mean);
+ }
+ }
+
+ // V0s
+ if (nGenV0s > 0) {
+ Double_t eff = nRecV0s*1./nGenV0s;
+ Double_t effError = TMath::Sqrt(eff*(1.-eff) / nGenV0s);
+ if (effError == 0) effError = checkV0EffLow / TMath::Sqrt(1.*nGenV0s);
+ Info("CheckESD", "V0 eff = (%.1f +- %.1f) %%",
+ 100.*eff, 100.*effError);
+ if (eff < checkV0EffLow - checkV0EffSigma*effError) {
+ Warning("CheckESD", "low V0 efficiency: (%.1f +- %.1f) %%",
+ 100.*eff, 100.*effError);
+ }
+ }
+
+ // Cascades
+ if (nGenCascades > 0) {
+ Double_t eff = nRecCascades*1./nGenCascades;
+ Double_t effError = TMath::Sqrt(eff*(1.-eff) / nGenCascades);
+ if (effError == 0) effError = checkV0EffLow /
+ TMath::Sqrt(1.*nGenCascades);
+ Info("CheckESD", "Cascade eff = (%.1f +- %.1f) %%",
+ 100.*eff, 100.*effError);
+ if (eff < checkCascadeEffLow - checkCascadeEffSigma*effError) {
+ Warning("CheckESD", "low Cascade efficiency: (%.1f +- %.1f) %%",
+ 100.*eff, 100.*effError);
+ }
+ }
+ }
+
+ // draw the histograms if not in batch mode
+ if (!gROOT->IsBatch()) {
+ new TCanvas;
+ hEff->DrawCopy();
+ new TCanvas;
+ hResPtInv->DrawCopy("E");
+ new TCanvas;
+ hResPhi->DrawCopy("E");
+ new TCanvas;
+ hResTheta->DrawCopy("E");
+ new TCanvas;
+ hDEdxRight->DrawCopy();
+ hDEdxWrong->DrawCopy("SAME");
+ new TCanvas;
+ hResTOFRight->DrawCopy("E");
+ hResTOFWrong->DrawCopy("SAME");
+ new TCanvas;
+ hEPHOS->DrawCopy("E");
+ new TCanvas;
+ hEEMCAL->DrawCopy("E");
+ new TCanvas;
+ hPtMUON->DrawCopy("E");
+ new TCanvas;
+ hMassK0->DrawCopy("E");
+ new TCanvas;
+ hMassLambda->DrawCopy("E");
+ new TCanvas;
+ hMassLambdaBar->DrawCopy("E");
+ new TCanvas;
+ hMassXi->DrawCopy("E");
+ new TCanvas;
+ hMassOmega->DrawCopy("E");
+ }
+
+ // write the output histograms to a file
+ TFile* outputFile = TFile::Open("check.root", "recreate");
+ if (!outputFile || !outputFile->IsOpen()) {
+ Error("CheckESD", "opening output file check.root failed");
+ return kFALSE;
+ }
+ hEff->Write();
+ hResPtInv->Write();
+ hResPhi->Write();
+ hResTheta->Write();
+ hDEdxRight->Write();
+ hDEdxWrong->Write();
+ hResTOFRight->Write();
+ hResTOFWrong->Write();
+ hEPHOS->Write();
+ hEEMCAL->Write();
+ hPtMUON->Write();
+ hMassK0->Write();
+ hMassLambda->Write();
+ hMassLambdaBar->Write();
+ hMassXi->Write();
+ hMassOmega->Write();
+ outputFile->Close();
+ delete outputFile;
+
+ // clean up
+ delete hGen;
+ delete hRec;
+ delete hEff;
+ delete hResPtInv;
+ delete hResPhi;
+ delete hResTheta;
+ delete hDEdxRight;
+ delete hDEdxWrong;
+ delete hResTOFRight;
+ delete hResTOFWrong;
+ delete hEPHOS;
+ delete hEEMCAL;
+ delete hPtMUON;
+ delete hMassK0;
+ delete hMassLambda;
+ delete hMassLambdaBar;
+ delete hMassXi;
+ delete hMassOmega;
+
+ delete esd;
+ esdFile->Close();
+ delete esdFile;
+
+ runLoader->UnloadHeader();
+ runLoader->UnloadKinematics();
+ delete runLoader;
+
+ // result of check
+ Info("CheckESD", "check of ESD was successfull");
+ return kTRUE;
+}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff