1 // AliAnalysisTaskMultPbTracks
3 // Author: Michele Floris, CERN
5 // - Add chi2/cluster plot for primary, secondaries and fakes
7 #include "AliAnalysisTaskMultPbTracks.h"
8 #include "AliESDInputHandler.h"
9 #include "AliAnalysisMultPbTrackHistoManager.h"
10 #include "AliAnalysisManager.h"
11 #include "AliESDtrackCuts.h"
12 #include "AliMCEvent.h"
17 #include "AliMCParticle.h"
18 #include "AliGenEventHeader.h"
19 #include "AliESDCentrality.h"
20 #include "AliMultiplicity.h"
22 #include "AliAnalysisMultPbCentralitySelector.h"
23 #include "AliTriggerAnalysis.h"
27 ClassImp(AliAnalysisTaskMultPbTracks)
29 AliAnalysisTaskMultPbTracks::AliAnalysisTaskMultPbTracks()
30 : AliAnalysisTaskSE("TaskMultPbTracks"),
31 fESD(0),fHistoManager(0),fCentrSelector(0),fTrackCuts(0),fTrackCutsNoDCA(0),fOfflineTrigger(0), fIsMC(0),fIsTPCOnly(0), fTriggerAnalysis(0)
35 DefineOutput(1, AliAnalysisMultPbTrackHistoManager::Class());
36 DefineOutput(2, AliESDtrackCuts::Class());
37 DefineOutput(3, AliAnalysisMultPbCentralitySelector::Class());
39 fHistoManager = new AliAnalysisMultPbTrackHistoManager("histoManager","Hitograms, Multiplicity, Track analysis");
40 if(fIsMC) fHistoManager->SetSuffix("MC");
43 AliAnalysisTaskMultPbTracks::AliAnalysisTaskMultPbTracks(const char * name)
44 : AliAnalysisTaskSE(name),
45 fESD(0),fHistoManager(0),fCentrSelector(0),fTrackCuts(0),fTrackCutsNoDCA(0),fOfflineTrigger(0),fIsMC(0),fIsTPCOnly(0), fTriggerAnalysis(0)
48 // Standard constructur which should be used
51 DefineOutput(1, AliAnalysisMultPbTrackHistoManager::Class());
52 DefineOutput(2, AliESDtrackCuts::Class());
53 DefineOutput(3, AliAnalysisMultPbCentralitySelector::Class());
55 fHistoManager = new AliAnalysisMultPbTrackHistoManager("histoManager","Hitograms, Multiplicity, Track analysis");
56 if(fIsMC) fHistoManager->SetSuffix("MC");
60 AliAnalysisTaskMultPbTracks::AliAnalysisTaskMultPbTracks(const AliAnalysisTaskMultPbTracks& obj) :
61 AliAnalysisTaskSE(obj) ,fESD (0), fHistoManager(0), fCentrSelector(0), fTrackCuts(0),fTrackCutsNoDCA(0),fOfflineTrigger(0),fIsMC(0),fIsTPCOnly(0), fTriggerAnalysis(0)
65 fHistoManager= obj.fHistoManager;
66 fTrackCuts = obj.fTrackCuts;
67 fTrackCutsNoDCA = obj.fTrackCutsNoDCA;
68 fCentrSelector = obj.fCentrSelector;
69 fOfflineTrigger = obj.fOfflineTrigger;
71 fIsTPCOnly = obj.fIsTPCOnly;
72 fTriggerAnalysis = obj.fTriggerAnalysis;
76 AliAnalysisTaskMultPbTracks::~AliAnalysisTaskMultPbTracks(){
79 if(!AliAnalysisManager::GetAnalysisManager()->IsProofMode()) {
83 delete fTriggerAnalysis;
87 // Histo list should not be destroyed: fListWrapper is owner!
90 void AliAnalysisTaskMultPbTracks::UserCreateOutputObjects()
94 // For the DCA distributions, we want to use exactly the same cuts
95 // as for the other distributions, with the exception of the DCA cut
96 fTrackCutsNoDCA = new AliESDtrackCuts(*fTrackCuts); // clone cuts
97 // Reset all DCA cuts; FIXME: is this all?
98 fTrackCutsNoDCA->SetMaxDCAToVertexXY();
99 fTrackCutsNoDCA->SetMaxDCAToVertexZ ();
100 fTrackCutsNoDCA->SetMaxDCAToVertexXYPtDep();
101 fTrackCutsNoDCA->SetMaxDCAToVertexZPtDep();
103 fTriggerAnalysis = new AliTriggerAnalysis();
104 fTriggerAnalysis->SetAnalyzeMC(fIsMC);
108 void AliAnalysisTaskMultPbTracks::UserExec(Option_t *)
113 /* PostData(0) is taken care of by AliAnalysisTaskSE */
114 PostData(1,fHistoManager);
115 PostData(2,fTrackCuts);
116 PostData(3,fCentrSelector);
118 // Cache histogram pointers
119 static TH3D * hTracks [AliAnalysisMultPbTrackHistoManager::kNHistos];
120 static TH2D * hDCA [AliAnalysisMultPbTrackHistoManager::kNHistos];
121 static TH1D * hNTracks [AliAnalysisMultPbTrackHistoManager::kNHistos];
122 hTracks[AliAnalysisMultPbTrackHistoManager::kHistoGen] = fHistoManager->GetHistoPtEtaVz(AliAnalysisMultPbTrackHistoManager::kHistoGen );
123 hTracks[AliAnalysisMultPbTrackHistoManager::kHistoRec] = fHistoManager->GetHistoPtEtaVz(AliAnalysisMultPbTrackHistoManager::kHistoRec );
124 hTracks[AliAnalysisMultPbTrackHistoManager::kHistoRecPrim] = fHistoManager->GetHistoPtEtaVz(AliAnalysisMultPbTrackHistoManager::kHistoRecPrim );
125 hTracks[AliAnalysisMultPbTrackHistoManager::kHistoRecFake] = fHistoManager->GetHistoPtEtaVz(AliAnalysisMultPbTrackHistoManager::kHistoRecFake );
126 hTracks[AliAnalysisMultPbTrackHistoManager::kHistoRecSecMat] = fHistoManager->GetHistoPtEtaVz(AliAnalysisMultPbTrackHistoManager::kHistoRecSecMat );
127 hTracks[AliAnalysisMultPbTrackHistoManager::kHistoRecSecWeak] = fHistoManager->GetHistoPtEtaVz(AliAnalysisMultPbTrackHistoManager::kHistoRecSecWeak);
129 hDCA[AliAnalysisMultPbTrackHistoManager::kHistoGen] = fHistoManager->GetHistoDCA(AliAnalysisMultPbTrackHistoManager::kHistoGen );
130 hDCA[AliAnalysisMultPbTrackHistoManager::kHistoRec] = fHistoManager->GetHistoDCA(AliAnalysisMultPbTrackHistoManager::kHistoRec );
131 hDCA[AliAnalysisMultPbTrackHistoManager::kHistoRecPrim] = fHistoManager->GetHistoDCA(AliAnalysisMultPbTrackHistoManager::kHistoRecPrim );
132 hDCA[AliAnalysisMultPbTrackHistoManager::kHistoRecFake] = fHistoManager->GetHistoDCA(AliAnalysisMultPbTrackHistoManager::kHistoRecFake );
133 hDCA[AliAnalysisMultPbTrackHistoManager::kHistoRecSecMat] = fHistoManager->GetHistoDCA(AliAnalysisMultPbTrackHistoManager::kHistoRecSecMat );
134 hDCA[AliAnalysisMultPbTrackHistoManager::kHistoRecSecWeak] = fHistoManager->GetHistoDCA(AliAnalysisMultPbTrackHistoManager::kHistoRecSecWeak);
136 hNTracks[AliAnalysisMultPbTrackHistoManager::kHistoGen] = fHistoManager->GetHistoMult(AliAnalysisMultPbTrackHistoManager::kHistoGen );
137 hNTracks[AliAnalysisMultPbTrackHistoManager::kHistoRec] = fHistoManager->GetHistoMult(AliAnalysisMultPbTrackHistoManager::kHistoRec );
139 fESD = dynamic_cast<AliESDEvent*>(fInputEvent);
140 if (strcmp(fESD->ClassName(),"AliESDEvent")) {
141 AliFatal("Not processing ESDs");
144 fHistoManager->GetHistoStats()->Fill(AliAnalysisMultPbTrackHistoManager::kStatAll);
147 // Centrality selection
148 Bool_t isCentralitySelected = fCentrSelector->IsCentralityBinSelected(fESD,fTrackCuts);
149 if(!isCentralitySelected) return;
151 // AliESDCentrality *centrality = fESD->GetCentrality();
153 // AliFatal("Centrality object not available");
156 // Int_t centrBin = centrality->GetCentralityClass5(fCentralityEstimator.Data()) ;
158 // if (centrBin != fCentrBin && fCentrBin != -1) return;
161 fHistoManager->GetHistoStats()->Fill(AliAnalysisMultPbTrackHistoManager::kStatCentr);
163 Bool_t isSelected = (((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected() & fOfflineTrigger);
165 if(!isSelected) return;
166 fHistoManager->GetHistoStats()->Fill(AliAnalysisMultPbTrackHistoManager::kStatPhysSel);
171 //FIXME: which vertex do I take for MC?
173 fMCEvent->GenEventHeader()->PrimaryVertex(vertex);
174 Float_t zvGen = vertex[2];
177 AliError("No MC info found");
180 //loop on the MC event, only over primaries, which are always
181 // the first in stack.
182 // WRONG: D&B decays are produced in the transport... Need To Loop over all particles
183 // Int_t nMCTracks = fMCEvent->GetNumberOfPrimaries();
184 Int_t nMCTracks = fMCEvent->GetNumberOfTracks();
185 Int_t nPhysicalPrimaries = 0;
186 for (Int_t ipart=0; ipart<nMCTracks; ipart++) {
188 AliMCParticle *mcPart = (AliMCParticle*)fMCEvent->GetTrack(ipart);
190 // We don't care about neutrals and non-physical primaries
191 if(mcPart->Charge() == 0) continue;
193 //check if current particle is a physical primary
194 if(!IsPhysicalPrimaryAndTransportBit(ipart)) continue;
195 if(TMath::Abs(mcPart->Zv()-zvGen)>1e-6) {
196 // This cures a bug in Hijing
197 // A little hack here: I put those in the underflow bin of the process type to keep track of them
198 fHistoManager->GetHistoProcess(AliAnalysisMultPbTrackHistoManager::kHistoGen)->Fill(-1);
202 nPhysicalPrimaries++;
203 // Fill species histo and particle species
204 fHistoManager->GetHistoProcess(AliAnalysisMultPbTrackHistoManager::kHistoGen)->Fill(mcPart->Particle()->GetUniqueID());
205 fHistoManager->FillParticleID(AliAnalysisMultPbTrackHistoManager::kHistoGen, mcPart);
207 // Float_t zv = vtxESD->GetZ();
208 // Fill generated histo
209 hTracks[AliAnalysisMultPbTrackHistoManager::kHistoGen]->Fill(mcPart->Pt(),mcPart->Eta(),zvGen);
210 Int_t partCode = fHistoManager->GetLocalParticleID(mcPart);
211 cout << "Part " << partCode << endl;
212 if (partCode == AliAnalysisMultPbTrackHistoManager::kPartPiPlus ||
213 partCode == AliAnalysisMultPbTrackHistoManager::kPartPiMinus ||
214 partCode == AliAnalysisMultPbTrackHistoManager::kPartKPlus ||
215 partCode == AliAnalysisMultPbTrackHistoManager::kPartKMinus ||
216 partCode == AliAnalysisMultPbTrackHistoManager::kPartP ||
217 partCode == AliAnalysisMultPbTrackHistoManager::kPartPBar
219 cout << "Found " << partCode << endl;
221 fHistoManager->GetHistoPtEtaVz(AliAnalysisMultPbTrackHistoManager::kHistoGen, partCode);
224 hNTracks[AliAnalysisMultPbTrackHistoManager::kHistoGen]->Fill(nPhysicalPrimaries);
225 fHistoManager->GetHistoVzEvent(AliAnalysisMultPbTrackHistoManager::kHistoGen)->Fill(zvGen);
232 const AliESDVertex* vtxESD = fESD->GetPrimaryVertex();
234 // TODO: leave the cuts here or find a better place?)
235 // Quality cut on vertexer Z, as suggested by Francesco Prino
236 if(vtxESD->IsFromVertexerZ()) {
237 if (vtxESD->GetNContributors() <= 0) return;
238 if (vtxESD->GetDispersion() > 0.04) return;
239 if (vtxESD->GetZRes() > 0.25) return;
241 // "Beam gas" vertex cut
242 const AliESDVertex * vtxESDTPC= fESD->GetPrimaryVertexTPC();
243 if(vtxESDTPC->GetNContributors()<1) return;
244 if (vtxESDTPC->GetNContributors()<(-10.+0.25*fESD->GetMultiplicity()->GetNumberOfITSClusters(0))) return;
247 fHistoManager->GetHistoStats()->Fill(AliAnalysisMultPbTrackHistoManager::kStatVtx);
250 Bool_t zdcA = fTriggerAnalysis->ZDCTDCTrigger(fESD, AliTriggerAnalysis::kASide, kTRUE, kFALSE) ;
251 Bool_t zdcC = fTriggerAnalysis->ZDCTDCTrigger(fESD, AliTriggerAnalysis::kCSide, kTRUE, kFALSE) ;
253 if (!(zdcA && zdcC) && (!fIsMC)) return;
254 fHistoManager->GetHistoStats()->Fill(AliAnalysisMultPbTrackHistoManager::kStatZDCCut);
259 fHistoManager->GetHistoVzEvent(AliAnalysisMultPbTrackHistoManager::kHistoRec)->Fill(vtxESD->GetZ());
262 Int_t ntracks = fESD->GetNumberOfTracks();
263 Int_t acceptedTracks = 0;
265 for (Int_t itrack = 0; itrack<ntracks; itrack++) {
266 AliESDtrack * esdTrack = fIsTPCOnly ? fTrackCuts->GetTPCOnlyTrack(fESD,itrack) : fESD->GetTrack(itrack);// FIXMEL TrackCuts or TrackCutsNoDCA for the TPC?
267 if (!esdTrack) continue;
269 // Fill DCA distibutions, without the DCA cut, Fill the other stuff, with all cuts!
270 Bool_t accepted = fTrackCuts->AcceptTrack(esdTrack);
271 Bool_t acceptedNoDCA = fTrackCutsNoDCA->AcceptTrack(esdTrack);
273 if(accepted) acceptedTracks++;
275 // Compute weighted offset
276 // TODO: other choiches for the DCA?
277 Double_t weightedDCA = 10;
280 #if defined WEIGHTED_DCA
281 Double_t b = fESD->GetMagneticField();
284 if (esdTrack->PropagateToDCA(vtxESD, b,3., dca, cov)) {
285 Double_t det = cov[0]*cov[2]-cov[1]*cov[1];
287 AliError("DCA Covariance matrix is not positive definite!");
290 weightedDCA = (dca[0]*dca[0]*cov[2]+dca[1]*dca[1]*cov[0]-2*dca[0]*dca[1]*cov[1])/det;
291 weightedDCA = weightedDCA>0 ? TMath::Sqrt(weightedDCA) : 10;
293 // printf("dR:%e dZ%e sigR2:%e sigRZ:%e sigZ2:%e\n",dca[0],dca[1],cov[0],cov[1],cov[2]);
296 #elif defined TRANSVERSE_DCA
298 esdTrack->GetDZ(vtxESD->GetX(),vtxESD->GetY(),vtxESD->GetZ(), fESD->GetMagneticField(), xz);
303 // Alternative: p*DCA
305 // esdTrack->GetDZ(vtxESD->GetX(),vtxESD->GetY(),vtxESD->GetZ(), fESD->GetMagneticField(), xz);
306 // Float_t dca = xz[0]*esdTrack->P();
308 // This cures a bug in Hijing (displaced primaries)
310 Int_t label = TMath::Abs(esdTrack->GetLabel()); // no fakes!!!
311 if (IsPhysicalPrimaryAndTransportBit(label)) {
312 AliMCParticle *mcPart = (AliMCParticle*)fMCEvent->GetTrack(label);
313 if(!mcPart) AliFatal("No particle");
315 fMCEvent->GenEventHeader()->PrimaryVertex(vertex);
316 Float_t zvGen = vertex[2];
317 if(TMath::Abs(mcPart->Zv()-zvGen)>1e-6) continue;
322 hTracks[AliAnalysisMultPbTrackHistoManager::kHistoRec]->Fill(esdTrack->Pt(),esdTrack->Eta(),vtxESD->GetZ());
324 hDCA[AliAnalysisMultPbTrackHistoManager::kHistoRec]->Fill(weightedDCA,esdTrack->Pt());
326 // Fill separately primaries and secondaries
327 // FIXME: fakes? Is this the correct way to account for them?
328 // Get label and corresponding mcPart;
330 Int_t label = TMath::Abs(esdTrack->GetLabel()); // no fakes!!!
331 //Int_t label = esdTrack->GetLabel(); //
332 AliMCParticle *mcPart = label < 0 ? 0 : (AliMCParticle*)fMCEvent->GetTrack(label);
335 hTracks[AliAnalysisMultPbTrackHistoManager::kHistoRecFake]->Fill(esdTrack->Pt(),esdTrack->Eta(),vtxESD->GetZ());
337 hDCA[AliAnalysisMultPbTrackHistoManager::kHistoRecFake]->Fill(weightedDCA,esdTrack->Pt());
340 if(IsPhysicalPrimaryAndTransportBit(label)) {
341 // Fill species histo
342 fHistoManager->GetHistoProcess(AliAnalysisMultPbTrackHistoManager::kHistoRecPrim)->Fill(mcPart->Particle()->GetUniqueID());
344 hTracks[AliAnalysisMultPbTrackHistoManager::kHistoRecPrim]->Fill(esdTrack->Pt(),esdTrack->Eta(),vtxESD->GetZ());
345 Int_t partCode = fHistoManager->GetLocalParticleID(mcPart);
346 if (partCode == AliAnalysisMultPbTrackHistoManager::kPartPiPlus ||
347 partCode == AliAnalysisMultPbTrackHistoManager::kPartPiMinus ||
348 partCode == AliAnalysisMultPbTrackHistoManager::kPartKPlus ||
349 partCode == AliAnalysisMultPbTrackHistoManager::kPartKMinus ||
350 partCode == AliAnalysisMultPbTrackHistoManager::kPartP ||
351 partCode == AliAnalysisMultPbTrackHistoManager::kPartPBar
353 fHistoManager->GetHistoPtEtaVz(AliAnalysisMultPbTrackHistoManager::kHistoRecPrim, partCode);
356 hDCA[AliAnalysisMultPbTrackHistoManager::kHistoRecPrim]->Fill(weightedDCA,esdTrack->Pt());
360 Int_t indexMoth=mcPart->Particle()->GetFirstMother();
362 TParticle* moth = fMCEvent->Stack()->Particle(indexMoth);
363 Float_t codemoth = TMath::Abs(moth->GetPdgCode());
364 mfl = Int_t (codemoth/ TMath::Power(10, Int_t(TMath::Log10(codemoth))));
366 if(mfl==3){ // strangeness
367 fHistoManager->GetHistoProcess(AliAnalysisMultPbTrackHistoManager::kHistoRecSecWeak)->Fill(mcPart->Particle()->GetUniqueID());
369 hTracks[AliAnalysisMultPbTrackHistoManager::kHistoRecSecWeak]->Fill(esdTrack->Pt(),esdTrack->Eta(),vtxESD->GetZ());
371 hDCA[AliAnalysisMultPbTrackHistoManager::kHistoRecSecWeak]->Fill(weightedDCA,esdTrack->Pt());
373 fHistoManager->GetHistoProcess(AliAnalysisMultPbTrackHistoManager::kHistoRecSecMat)->Fill(mcPart->Particle()->GetUniqueID());
375 hTracks[AliAnalysisMultPbTrackHistoManager::kHistoRecSecMat]->Fill(esdTrack->Pt(),esdTrack->Eta(),vtxESD->GetZ());
377 hDCA[AliAnalysisMultPbTrackHistoManager::kHistoRecSecMat]->Fill(weightedDCA,esdTrack->Pt());
388 // cout << acceptedTracks << endl;
390 hNTracks[AliAnalysisMultPbTrackHistoManager::kHistoRec] ->Fill(acceptedTracks);
392 fHistoManager->GetHistoV0vsNtracks(AliAnalysisMultPbTrackHistoManager::kHistoRec)->Fill(acceptedTracks, fCentrSelector->GetCorrV0(fESD,v0));
394 // hNTracks[AliAnalysisMultPbTrackHistoManager::kHistoRec] ->Fill(fESD->GetMultiplicity()->GetNumberOfTracklets());
399 void AliAnalysisTaskMultPbTracks::Terminate(Option_t *){
405 Bool_t AliAnalysisTaskMultPbTracks::IsPhysicalPrimaryAndTransportBit(Int_t ipart) {
406 // Check if it is a primary and if it was transported
407 Bool_t physprim=fMCEvent->IsPhysicalPrimary(ipart);
408 if (!physprim) return kFALSE;
409 Bool_t transported = ((AliMCParticle*)fMCEvent->GetTrack(ipart))->Particle()->TestBit(kTransportBit);
410 if(!transported) return kFALSE;
416 // void AliAnalysisTaskEvil::PrintProcInfo()
419 // gSystem->GetProcInfo(&info);
420 // AliInfo(Form("fMemResident=%10ld kB fMemVirtual=%10ld kB",info.fMemResident,info.fMemVirtual));