1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
19 // R.Bailhache <R.Bailhache@gsi.de>
25 #include "AliAnalysisManager.h"
26 #include "AliCentrality.h"
27 #include "AliAODTrack.h"
28 #include "AliAODEvent.h"
29 #include "AliAODPid.h"
30 #include "AliHFEcuts.h"
31 #include "AliHFEextraCuts.h"
32 #include "AliInputEventHandler.h"
34 #include "AliPIDResponse.h"
35 #include "AliVEvent.h"
36 #include "AliHFEpidTPC.h"
37 #include "TTreeStream.h"
38 #include "AliESDtrack.h"
39 #include "TClonesArray.h"
40 #include "AliAODMCHeader.h"
41 #include "AliHFEsignalCuts.h"
42 #include "AliAODMCParticle.h"
43 #include "AliVTrack.h"
44 #include "AliAODVertex.h"
46 #include "AliHFEdebugTreeTaskAOD.h"
48 ClassImp(AliHFEdebugTreeTaskAOD)
50 AliHFEdebugTreeTaskAOD::AliHFEdebugTreeTaskAOD():
53 fAODArrayMCInfo(NULL),
62 fFilename("HFEtree.root"),
68 AliHFEdebugTreeTaskAOD::AliHFEdebugTreeTaskAOD(const char *name):
69 AliAnalysisTaskSE(name),
71 fAODArrayMCInfo(NULL),
80 fFilename("HFEtree.root"),
83 fTPCpid = new AliHFEpidTPC("QAtpcPID");
86 AliHFEdebugTreeTaskAOD::~AliHFEdebugTreeTaskAOD(){
88 if(fDebugTree) delete fDebugTree;
89 if(fTPCpid) delete fTPCpid;
92 void AliHFEdebugTreeTaskAOD::UserCreateOutputObjects(){
94 // Create debug tree, signal cuts and track cuts
98 fDebugTree = new TTreeSRedirector(fFilename.Data());
100 // printf("testa\n");
101 fSignalCuts = new AliHFEsignalCuts("HFEsignalCuts", "HFE MC Signal definition");
104 fTrackCuts = new AliHFEcuts("fTrackCuts", "Basic HFE track cuts");
105 fTrackCuts->CreateStandardCuts();
106 fTrackCuts->SetAOD();
108 fTrackCuts->SetMinNClustersTPC(fNclustersTPC);
109 fTrackCuts->SetMinRatioTPCclusters(0);
110 fTrackCuts->SetTPCmodes(AliHFEextraCuts::kFound, AliHFEextraCuts::kFoundOverFindable);
111 fTrackCuts->SetMinNClustersTPCPID(fNclustersTPCPID);
112 fTrackCuts->SetMinNClustersITS(fNclustersITS);
114 fTrackCuts->SetUseMixedVertex(kTRUE);
115 fTrackCuts->SetVertexRange(10.);
117 fTrackCuts->Initialize();
120 fExtraCuts = new AliHFEextraCuts("hfeExtraCuts","HFE Extra Cuts");
124 void AliHFEdebugTreeTaskAOD::UserExec(Option_t *){
126 // User Exec: Fill debug Tree
130 AliPIDResponse *pid = NULL;
131 AliInputEventHandler *handler = dynamic_cast<AliInputEventHandler *>(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
133 // printf("testb\n");
134 pid = handler->GetPIDResponse();
136 AliError("No Handler");
139 // printf("testc\n");
140 AliError("No PID response");
144 AliError("No Input event");
148 AliAODTrack copyTrack;
151 Bool_t mcthere = kTRUE;
152 AliAODEvent *aodE = dynamic_cast<AliAODEvent *>(fInputEvent);
154 // printf("testd\n");
155 AliError("No AOD Event");
158 fAODMCHeader = dynamic_cast<AliAODMCHeader *>(fInputEvent->FindListObject(AliAODMCHeader::StdBranchName()));
163 fAODArrayMCInfo = dynamic_cast<TClonesArray *>(fInputEvent->FindListObject(AliAODMCParticle::StdBranchName()));
164 if(!fAODArrayMCInfo){
169 fSignalCuts->SetMCAODInfo(fAODArrayMCInfo);
170 fTrackCuts->SetMCEvent(aodE);
173 // Set for track cuts
174 fTrackCuts->SetRecEvent(fInputEvent);
176 if(!fTrackCuts->CheckEventCuts("fCutsEvRec", fInputEvent)){
177 AliDebug(1, "Event rejected by the event cuts\n");
181 fExtraCuts = new AliHFEextraCuts("hfeExtraCuts","HFE Extra Cuts");
183 fExtraCuts->SetRecEventInfo(fInputEvent);
189 Int_t run = fInputEvent->GetRunNumber();
192 UInt_t trigger = fInputHandler->IsEventSelected();
193 Bool_t isMBTrigger = trigger & AliVEvent::kMB;
194 Bool_t isCentralTrigger = trigger & AliVEvent::kCentral;
195 Bool_t isSemicentralTrigger = trigger & AliVEvent::kSemiCentral;
196 Bool_t isEMCALTrigger = trigger & AliVEvent::kEMCEJE;
198 // Get Primary Vertex
199 const AliVVertex *vertex = fInputEvent->GetPrimaryVertex();
202 Double_t ncontrib = fInputEvent->GetPrimaryVertex()->GetNContributors();
205 Float_t centrality = -1.;
206 AliCentrality *hicent = fInputEvent->GetCentrality();
207 centrality = hicent->GetCentralityPercentile("V0M");
209 // Store event selection variables
210 (*fDebugTree) << "EventDebug"
211 << "Centrality=" << centrality
212 << "VertexZ=" << vtx[2]
213 << "NumberOfContributors=" << ncontrib
219 // Look for kink mother
220 Int_t numberofvertices = aodE->GetNumberOfVertices();
221 Double_t listofmotherkink[numberofvertices];
222 Int_t numberofmotherkink = 0;
223 //printf("Number of vertices %d\n",numberofvertices);
224 for(Int_t ivertex=0; ivertex < numberofvertices; ivertex++) {
225 AliAODVertex *aodvertex = aodE->GetVertex(ivertex);
226 if(!aodvertex) continue;
227 //printf("Type %d\n",aodvertex->GetType());
228 if(aodvertex->GetType()==AliAODVertex::kKink) {
229 //printf("Find one kink\n");
230 AliAODTrack *mother = (AliAODTrack *) aodvertex->GetParent();
231 if(!mother) continue;
232 Int_t idmother = mother->GetID();
233 listofmotherkink[numberofmotherkink] = idmother;
234 //printf("ID %d\n",idmother);
235 numberofmotherkink++;
238 //printf("Number of kink mother in the events %d\n",numberofmotherkink);
241 // Loop on MC tracks only
243 AliAODMCParticle *mctrack = NULL;
245 Double_t eR,vx,vy,vz;
246 Double_t chargemc, etamc, phimc, momentummc, transversemomentummc;
247 Int_t source,pdg,signal;
249 for(Int_t itrack = 0; itrack < fAODArrayMCInfo->GetEntriesFast(); itrack++) {
250 mctrack = (AliAODMCParticle *)(fAODArrayMCInfo->At(itrack));
251 if(!mctrack) continue;
253 if(fTrackCuts->CheckParticleCuts(static_cast<UInt_t>(AliHFEcuts::kStepMCGenerated), mctrack)) signal = 1;
255 chargemc = mctrack->Charge() > 0. ? 1. : -1.;
256 momentummc = mctrack->P() * chargemc;
257 transversemomentummc = mctrack->Pt() * chargemc;
258 etamc = mctrack->Eta();
259 phimc = mctrack->Phi();
260 pdg = mctrack->GetPdgCode();
262 // Get Production Vertex in radial direction
266 eR = TMath::Sqrt(vx*vx+vy*vy);
268 // Get Mother PDG code of the particle
270 Int_t motherlabel = TMath::Abs(mctrack->GetMother());
271 if(motherlabel >= 0 && motherlabel < fAODArrayMCInfo->GetEntriesFast()){
272 AliAODMCParticle *mother = dynamic_cast<AliAODMCParticle *>(fAODArrayMCInfo->At(motherlabel));
273 if(mother) motherPdg = mother->GetPdgCode();
278 if(fSignalCuts->IsCharmElectron(mctrack)) source = 0;
279 else if(fSignalCuts->IsBeautyElectron(mctrack)) source = 1;
280 else if(fSignalCuts->IsGammaElectron(mctrack)) source = 2;
281 else if(fSignalCuts->IsNonHFElectron(mctrack)) source = 3;
282 else if(TMath::Abs(pdg) == 11) source = 4;
285 (*fDebugTree) << "MCDebug"
286 << "centrality=" << centrality
287 << "MBtrigger=" << isMBTrigger
288 << "CentralTrigger=" << isCentralTrigger
289 << "SemicentralTrigger=" << isSemicentralTrigger
290 << "EMCALtrigger=" << isEMCALTrigger
292 << "p=" << momentummc
293 << "pt=" << transversemomentummc
300 << "ProductionVertex=" << eR
301 << "motherPdg=" << motherPdg
302 << "source=" << source
308 Double_t charge, eta, phi, momentum, momentumTPC, transversemomentum;
311 // Loop on reconstructed tracks
314 TArrayI *arraytrack = new TArrayI(fInputEvent->GetNumberOfTracks());
317 AliAODTrack *track = 0x0;
318 for(Int_t itrack = 0; itrack < fInputEvent->GetNumberOfTracks(); itrack++){
320 track = dynamic_cast<AliAODTrack *>(fInputEvent->GetTrack(itrack));
322 // Cut track (Only basic track cuts)
323 // printf("testv\n");
324 if(!fTrackCuts->CheckParticleCuts(AliHFEcuts::kNcutStepsMCTrack + AliHFEcuts::kStepRecKineITSTPC, track)) continue;
327 Double_t nSigmaTOF = pid->NumberOfSigmasTOF(track, AliPID::kElectron);
328 Double_t nSigmaTPC = pid->NumberOfSigmasTPC(track, AliPID::kElectron);
329 Double_t tPCdEdx = track->GetDetPid() ? track->GetDetPid()->GetTPCsignal() : 0.;
331 copyTrack.~AliAODTrack();
332 new(©Track) AliAODTrack(*track);
333 if(fTPCpid->HasCentralityCorrection()) fTPCpid->ApplyCentralityCorrection(©Track, static_cast<Double_t>(ncontrib),AliHFEpidObject::kAODanalysis);
334 if(fTPCpid->HasEtaCorrection()) fTPCpid->ApplyEtaCorrection(©Track, AliHFEpidObject::kAODanalysis);
335 Double_t nSigmaTPCcorr = pid->NumberOfSigmasTPC(©Track, AliPID::kElectron);
336 Double_t tPCdEdxcorr = copyTrack.GetDetPid() ? copyTrack.GetDetPid()->GetTPCsignal() : 0.;
339 charge = track->Charge() > 0 ? 1. : -1.;
342 momentum = track->P() * charge;
343 transversemomentum = track->Pt() * charge;
344 momentumTPC = track->GetDetPid() ? track->GetDetPid()->GetTPCmomentum() : track->P();
347 ULong_t status = track->GetStatus();
349 if((status & AliESDtrack::kITSrefit) == AliESDtrack::kITSrefit) itsrefit = 1;
351 if((status & AliESDtrack::kTPCrefit) == AliESDtrack::kTPCrefit) tpcrefit = 1;
353 if((status & AliESDtrack::kTOFpid) == AliESDtrack::kTOFpid) tofpid = 1;
355 // ITS number of clusters
356 UChar_t nclustersITS = track->GetITSNcls();
357 // TPC number of clusters (different definitions)
358 UChar_t nclustersTPCfit = track->GetTPCNcls();
359 UChar_t nclustersTPCall = 0;
360 const TBits &clusterTPC = track->GetTPCClusterMap();
361 nclustersTPCall = clusterTPC.CountBits();
362 UChar_t nclustersTPCPID = track->GetTPCsignalN();
363 UChar_t nfindableTPC = track->GetTPCNclsF();
364 Double_t clusterRatioTPCfit = 0.0;
365 if((static_cast<Double_t>(nfindableTPC))>0.0) clusterRatioTPCfit = static_cast<Double_t>(nclustersTPCfit)/static_cast<Double_t>(nfindableTPC);
366 Double_t clusterRatioTPCall = 0.0;
367 if((static_cast<Double_t>(nfindableTPC))>0.0) clusterRatioTPCall = static_cast<Double_t>(nclustersTPCall)/static_cast<Double_t>(nfindableTPC);
368 UChar_t nclustersTPCshared = 0;
369 Float_t ncrossedRowsTPC = track->GetTPCNCrossedRows();
370 const TBits &sharedTPC = track->GetTPCSharedMap();
371 for(Int_t ibit = 0; ibit < 160; ibit++) if(sharedTPC.TestBitNumber(ibit)) nclustersTPCshared++;
372 // TRD clusters and tracklets
373 UChar_t nclustersTRD = track->GetTRDncls();
374 UChar_t ntrackletsTRDPID = track->GetTRDntrackletsPID();
375 Int_t nslicesTRD = track->GetDetPid()->GetTRDnSlices();
376 Int_t chi2TRD = track->GetTRDchi2();
377 AliAODPid* aodpid= track->GetDetPid();
378 Double_t* arraytrdsignals;
379 arraytrdsignals=aodpid->GetTRDslices();
382 for(Int_t iplane = 0; iplane < 6; iplane++){
386 for(Int_t iplane = 0; iplane < 6; iplane++){
388 for(Int_t b=(iplane*8);b<((iplane*8)+8);b++)
390 if(ntrackletsTRDPID>0)
392 if(arraytrdsignals[b]>0.001) nslicetemp++;
395 if(nslicetemp > 0) trdlayer[iplane]=1;
398 // ITS and TRD acceptance maps
399 UChar_t itsPixel = track->GetITSClusterMap();
400 Bool_t statusL0 = kFALSE;
401 if(TESTBIT(itsPixel, 0)) statusL0 = kTRUE;
402 Bool_t statusL1 = kFALSE;
403 if(TESTBIT(itsPixel, 1)) statusL1 = kTRUE;
406 Float_t dcaxy = -999.;
407 Float_t dcaz = -999.;
408 fExtraCuts->GetImpactParameters((AliVTrack *)track,dcaxy,dcaz);
412 if(fExtraCuts->IsKinkDaughter(track)) kink = 1;
415 Int_t kinkmotherpass = 0;
416 for(Int_t kinkmother = 0; kinkmother < numberofmotherkink; kinkmother++) {
417 if(track->GetID() == listofmotherkink[kinkmother]) {
423 // ID track to see if negative
424 Int_t id = track->GetID();
428 for(Int_t l=0; l < counterdc; l++){
429 Int_t iTrack2 = arraytrack->At(l);
430 if(iTrack2==id) doublec=1;
432 //printf("Doublec %d\n",doublec);
434 // Add the id at this place
435 arraytrack->AddAt(id,counterdc);
440 for(Int_t k=0; k<20; k++) {
443 if((track->TestFilterBit(u))) {
447 Int_t filter0 = filter[0];
448 Int_t filter1 = filter[1];
449 Int_t filter2 = filter[2];
450 Int_t filter3 = filter[3];
451 Int_t filter4 = filter[4];
452 Int_t filter5 = filter[5];
453 Int_t filter6 = filter[6];
454 Int_t filter7 = filter[7];
455 Int_t filter8 = filter[8];
456 Int_t filter9 = filter[9];
457 Int_t filter10 = filter[10];
458 Int_t filter11 = filter[11];
459 Int_t filter12 = filter[12];
460 Int_t filter13 = filter[13];
461 Int_t filter14 = filter[14];
462 Int_t filter15 = filter[15];
463 Int_t filter16 = filter[16];
464 Int_t filter17 = filter[17];
465 Int_t filter18 = filter[18];
466 Int_t filter19 = filter[19];
468 Int_t eventnb = fEventNumber;
474 Int_t label = TMath::Abs(track->GetLabel());
475 if(label && label < fAODArrayMCInfo->GetEntriesFast())
476 mctrack = dynamic_cast<AliAODMCParticle *>(fAODArrayMCInfo->At(label));
477 if(!mctrack) continue;
478 if(fTrackCuts->CheckParticleCuts(static_cast<UInt_t>(AliHFEcuts::kStepMCGenerated), mctrack)) signal = 1;
480 chargemc = mctrack->Charge() > 0. ? 1. : -1.;
481 momentummc = mctrack->P() * chargemc;
482 transversemomentummc = mctrack->Pt() * chargemc;
483 etamc = mctrack->Eta();
484 phimc = mctrack->Phi();
485 pdg = mctrack->GetPdgCode();
487 // Get Production Vertex in radial direction
491 eR = TMath::Sqrt(vx*vx+vy*vy);
493 // Get Mother PDG code of the particle
495 Int_t motherlabel = TMath::Abs(mctrack->GetMother());
496 if(motherlabel >= 0 && motherlabel < fAODArrayMCInfo->GetEntriesFast()){
497 AliAODMCParticle *mother = dynamic_cast<AliAODMCParticle *>(fAODArrayMCInfo->At(motherlabel));
498 if(mother) motherPdg = mother->GetPdgCode();
503 if(fSignalCuts->IsCharmElectron(mctrack)) source = 0;
504 else if(fSignalCuts->IsBeautyElectron(mctrack)) source = 1;
505 else if(fSignalCuts->IsGammaElectron(mctrack)) source = 2;
506 else if(fSignalCuts->IsNonHFElectron(mctrack)) source = 3;
507 else if(TMath::Abs(pdg) == 11) source = 4;
515 (*fDebugTree) << "PIDdebug"
516 << "centrality=" << centrality
517 << "MBtrigger=" << isMBTrigger
518 << "CentralTrigger=" << isCentralTrigger
519 << "SemicentralTrigger=" << isSemicentralTrigger
520 << "EMCALtrigger=" << isEMCALTrigger
522 << "eventnb=" << eventnb
526 << "ncontrib=" << ncontrib
530 << "ptpc=" << momentumTPC
531 << "pt=" << transversemomentum
534 << "itsrefit=" << itsrefit
535 << "tpcrefit=" << tpcrefit
536 << "tofpid=" << tofpid
537 << "nclustersTPC=" << nclustersTPCfit
538 << "nclustersTPCall=" << nclustersTPCall
539 << "nclustersTPCPID=" << nclustersTPCPID
540 << "nclustersTPCshared=" << nclustersTPCshared
541 << "ncrossedRowsTPC=" << ncrossedRowsTPC
542 << "clusterRatioTPC=" << clusterRatioTPCfit
543 << "clusterRatioTPCall=" << clusterRatioTPCall
544 << "nclustersITS=" << nclustersITS
545 << "nclustersTRD=" << nclustersTRD
546 << "ntrackletsTRD=" << ntrackletsTRDPID
547 << "nslicesTRD=" << nslicesTRD
548 << "trd0=" << trdlayer[0]
549 << "trd1=" << trdlayer[1]
550 << "trd2=" << trdlayer[2]
551 << "trd3=" << trdlayer[3]
552 << "trd4=" << trdlayer[4]
553 << "trd5=" << trdlayer[5]
554 << "chi2TRD=" << chi2TRD
555 << "statusITS0=" << statusL0
556 << "statusITS1=" << statusL1
557 << "TOFsigmaEl=" << nSigmaTOF
558 << "TPCsigmaEl=" << nSigmaTPC
559 << "TPCsigmaElcorr=" << nSigmaTPCcorr
560 << "TPCdEdx=" << tPCdEdx
561 << "TPCdEdxcorr=" << tPCdEdxcorr
564 << "kinkdaughter=" << kink
565 << "kinkmother=" << kinkmotherpass
566 << "nbofmotherkink=" << numberofmotherkink
567 << "filter0=" << filter0
568 << "filter1=" << filter1
569 << "filter2=" << filter2
570 << "filter3=" << filter3
571 << "filter4=" << filter4
572 << "filter5=" << filter5
573 << "filter6=" << filter6
574 << "filter7=" << filter7
575 << "filter8=" << filter8
576 << "filter9=" << filter9
577 << "filter10=" << filter10
578 << "filter11=" << filter11
579 << "filter12=" << filter12
580 << "filter13=" << filter13
581 << "filter14=" << filter14
582 << "filter15=" << filter15
583 << "filter16=" << filter16
584 << "filter17=" << filter17
585 << "filter18=" << filter18
586 << "filter19=" << filter19
587 << "mcp=" << momentummc
588 << "mcpt=" << transversemomentummc
592 << "source=" << source
597 << "mccharge=" << chargemc
598 << "signal=" << signal
606 arraytrack->~TArrayI();
613 void AliHFEdebugTreeTaskAOD::SetFileName(const char *filename){ fFilename = filename; }
614 void AliHFEdebugTreeTaskAOD::Terminate(Option_t *){
616 if(fDebugTree) delete fDebugTree;