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 **************************************************************************/
16 ////////////////////////////////////////////////////////////////////////////////
18 // This class contains all code which is used to compute any of the values
19 // which can be of interest within a resonance analysis. Besides the obvious
20 // invariant mass, it allows to compute other utility values on all possible
21 // targets, in order to allow a wide spectrum of binning and checks.
22 // When needed, this object can also define a binning in the variable which
23 // it is required to compute, which is used for initializing axes of output
24 // histograms (see AliRsnFunction).
25 // The value computation requires this object to be passed the object whose
26 // informations will be used. This object can be of any allowed input type
27 // (track, Daughter, event), then this class must inherit from AliRsnTarget.
28 // Then, when value computation is attempted, a check on target type is done
29 // and computation is successful only if expected target matches that of the
31 // In some cases, the value computation can require a support external object,
32 // which must then be passed to this class. It can be of any type inheriting
35 // authors: A. Pulvirenti (alberto.pulvirenti@ct.infn.it)
36 // M. Vala (martin.vala@cern.ch)
38 ////////////////////////////////////////////////////////////////////////////////
40 #include <Riostream.h>
41 #include "AliVVertex.h"
42 #include "AliMultiplicity.h"
43 #include "AliESDtrackCuts.h"
44 #include "AliESDpid.h"
45 #include "AliAODPid.h"
46 #include "AliCentrality.h"
47 #include "AliESDUtils.h"
48 #include "AliPIDResponse.h"
50 #include "AliRsnEvent.h"
51 #include "AliRsnDaughter.h"
52 #include "AliRsnMother.h"
53 #include "AliRsnDaughterDef.h"
54 #include "AliRsnDaughterDef.h"
56 #include "AliRsnValueDaughter.h"
58 ClassImp(AliRsnValueDaughter)
60 //_____________________________________________________________________________
61 AliRsnValueDaughter::AliRsnValueDaughter(const char *name, EType type) :
62 AliRsnValue(name, AliRsnTarget::kDaughter),
70 //_____________________________________________________________________________
71 AliRsnValueDaughter::AliRsnValueDaughter(const AliRsnValueDaughter ©) :
80 //_____________________________________________________________________________
81 AliRsnValueDaughter &AliRsnValueDaughter::operator=(const AliRsnValueDaughter ©)
84 // Assignment operator.
85 // Works like copy constructor.
88 AliRsnValue::operator=(copy);
96 //_____________________________________________________________________________
97 const char *AliRsnValueDaughter::GetTypeName() const
100 // This method returns a string to give a name to each possible
101 // computation value.
105 case kP: return "SingleTrackPtot";
106 case kPt: return "SingleTrackPt";
107 case kPtpc: return "SingleTrackPtpc";
108 case kEta: return "SingleTrackEta";
109 case kY: return "SingleTrackY";
110 case kMass: return "SingleTrackMass";
111 case kITSsignal: return "SingleTrackITSsignal";
112 case kTPCsignal: return "SingleTrackTPCsignal";
113 case kTOFsignal: return "SingleTrackTOFsignal";
114 case kTPCnsigmaPi: return "SingleTrackTPCnsigmaPion";
115 case kTPCnsigmaK: return "SingleTrackTPCnsigmaKaon";
116 case kTPCnsigmaP: return "SingleTrackTPCnsigmaProton";
117 case kTOFnsigmaPi: return "SingleTrackTOFnsigmaPion";
118 case kTOFnsigmaK: return "SingleTrackTOFnsigmaKaon";
119 case kTOFnsigmaP: return "SingleTrackTOFnsigmaProton";
120 case kTOFdeltaPi: return "SingleTrackTOFdeltaPion";
121 case kTOFdeltaK: return "SingleTrackTOFdeltaKaon";
122 case kTOFdeltaP: return "SingleTrackTOFdeltaProton";
123 case kNITSclusters: return "SingleTrackNITSclusters";
124 case kNTPCclusters: return "SingleTrackNTPCclusters";
125 case kNTPCcrossedRows: return "SingleTrackNTPCcrossedRows";
126 case kNTPCcrossedRowsFclusters: return "SingleTrackNTPCcrossedRowsFclusters";
127 case kITSchi2: return "SingleTrackITSchi2";
128 case kTPCchi2: return "SingleTrackTPCchi2";
129 case kDCAXY: return "SingleTrackDCAz";
130 case kDCAZ: return "SingleTrackDCAz";
131 case kCharge: return "SingleTrackCharge";
132 case kPhi: return "SingleTrackPhi";
133 case kPhiOuterTPC: return "SingleTrackPhiOuterTPC";
134 case kV0DCA: return "V0DCAToPrimaryVertex";
135 case kV0Radius: return "V0Radius";
136 case kV0Mass: return "V0Mass";
137 case kV0P: return "V0Momentum";
138 case kV0Pt: return "V0TransverseMomentum";
139 case kDaughterDCA: return "V0DaughterDCA";
140 case kCosPointAng: return "V0CosineOfPointingAngle";
141 case kLambdaProtonPIDCut: return "V0LambdaProtonNsigma";
142 case kAntiLambdaAntiProtonPIDCut: return "V0AntiLambdaAntiProtonNsigma";
143 case kLambdaPionPIDCut: return "V0LambdaPionNsigma";
144 case kAntiLambdaAntiPionPIDCut: return "V0AntiLambdaPionNsigma";
145 default: return "Undefined";
149 //_____________________________________________________________________________
150 Bool_t AliRsnValueDaughter::Eval(TObject *object)
153 // Evaluation of the required value.
154 // Checks that the passed object is of the right type
155 // and if this check is successful, computes the required value.
156 // The output of the function tells if computing was successful,
157 // and the values must be taken with GetValue().
160 // coherence check, which also casts object
161 // to AliRsnTarget data members and returns kFALSE
162 // in case the object is NULL
163 if (!TargetOK(object)) return kFALSE;
165 // set a reference to the mother momentum
166 AliVParticle *ref = fDaughter->GetRef();
167 AliVParticle *refMC = fDaughter->GetRefMC();
168 AliVTrack *track = fDaughter->Ref2Vtrack();
169 AliESDv0 *v0esd = fDaughter->Ref2ESDv0();
170 AliAODv0 *v0aod = fDaughter->Ref2AODv0();
171 AliESDEvent *lESDEvent = fEvent->GetRefESD();
173 Double_t xPrimaryVertex = -999.9;
174 Double_t yPrimaryVertex = -999.9;
175 Double_t zPrimaryVertex = -999.9;
179 xPrimaryVertex = lESDEvent->GetPrimaryVertex()->GetX();
180 yPrimaryVertex = lESDEvent->GetPrimaryVertex()->GetY();
181 zPrimaryVertex = lESDEvent->GetPrimaryVertex()->GetZ();
185 if (fUseMCInfo && !refMC) {
189 if (!fUseMCInfo && !ref) {
194 // compute value depending on types in the enumeration
195 // if the type does not match any available choice, or if
196 // the computation is not doable due to any problem
197 // (not initialized support object, wrong values, risk of floating point errors)
198 // the method returng kFALSE and sets the computed value to a meaningless number
201 fComputedValue = (fUseMCInfo ? refMC->P() : ref->P());
205 fComputedValue = (fUseMCInfo ? refMC->Pt() : ref->Pt());
209 fComputedValue = (fUseMCInfo ? refMC->Eta() : ref->Eta());
213 fComputedValue = (fUseMCInfo ? refMC->Y() : ref->Y());
217 fComputedValue = (fUseMCInfo ? refMC->M() : ref->M());
222 fComputedValue = track->GetTPCmomentum();
225 AliWarning("Cannot get TPC momentum for non-track object");
226 fComputedValue = 0.0;
232 fComputedValue = track->GetITSsignal();
235 AliWarning("Cannot get ITS signal for non-track object");
236 fComputedValue = 0.0;
242 fComputedValue = track->GetTPCsignal();
245 AliWarning("Cannot get TPC signal for non-track object");
246 fComputedValue = 0.0;
252 fComputedValue = track->GetTOFsignal();
255 AliWarning("Cannot get TOF signal for non-track object");
256 fComputedValue = 0.0;
262 AliPIDResponse *pid = fEvent->GetPIDResponse();
263 fComputedValue = pid->NumberOfSigmasTPC(track, AliPID::kPion);
266 AliWarning("Cannot get TOF signal for non-track object");
267 fComputedValue = 0.0;
273 AliPIDResponse *pid = fEvent->GetPIDResponse();
274 fComputedValue = pid->NumberOfSigmasTPC(track, AliPID::kKaon);
277 AliWarning("Cannot get TOF signal for non-track object");
278 fComputedValue = 0.0;
284 AliPIDResponse *pid = fEvent->GetPIDResponse();
285 fComputedValue = pid->NumberOfSigmasTPC(track, AliPID::kProton);
288 AliWarning("Cannot get TOF signal for non-track object");
289 fComputedValue = 0.0;
295 AliPIDResponse *pid = fEvent->GetPIDResponse();
296 fComputedValue = pid->NumberOfSigmasTOF(track, AliPID::kPion);
299 AliWarning("Cannot get TOF signal for non-track object");
300 fComputedValue = 0.0;
306 AliPIDResponse *pid = fEvent->GetPIDResponse();
307 fComputedValue = pid->NumberOfSigmasTOF(track, AliPID::kKaon);
310 AliWarning("Cannot get TOF signal for non-track object");
311 fComputedValue = 0.0;
317 AliPIDResponse *pid = fEvent->GetPIDResponse();
318 fComputedValue = pid->NumberOfSigmasTOF(track, AliPID::kProton);
321 AliWarning("Cannot get TOF signal for non-track object");
322 fComputedValue = 0.0;
328 AliPIDResponse *pid = fEvent->GetPIDResponse();
329 pid->GetSignalDelta(AliPIDResponse::kTOF, track, AliPID::kPion, fComputedValue, kFALSE);//(==AliPIDResponse::kDetPidOk);
332 AliWarning("Cannot get TOF delta for non-track object");
333 fComputedValue = 0.0;
339 AliPIDResponse *pid = fEvent->GetPIDResponse();
340 pid->GetSignalDelta(AliPIDResponse::kTOF, track, AliPID::kKaon, fComputedValue, kFALSE);//(==AliPIDResponse::kDetPidOk);
343 AliWarning("Cannot get TOF delta for non-track object");
344 fComputedValue = 0.0;
350 AliPIDResponse *pid = fEvent->GetPIDResponse();
351 pid->GetSignalDelta(AliPIDResponse::kTOF, track, AliPID::kProton, fComputedValue, kFALSE);//(==AliPIDResponse::kDetPidOk);
354 AliWarning("Cannot get TOF delta for non-track object");
355 fComputedValue = 0.0;
361 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
363 fComputedValue = trackESD->GetITSclusters(0);
365 fComputedValue = ((AliAODTrack *)track)->GetITSNcls();
369 AliWarning("Cannot get n ITS clusters for non-track object");
370 fComputedValue = 0.0;
376 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
378 fComputedValue = trackESD->GetTPCclusters(0);
380 fComputedValue = ((AliAODTrack *)track)->GetTPCNcls();
384 AliWarning("Cannot get n TPC clusters for non-track object");
385 fComputedValue = 0.0;
389 case kNTPCcrossedRows:
391 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
393 fComputedValue = trackESD->GetTPCCrossedRows();
395 fComputedValue = ((AliAODTrack *)track)->GetTPCNCrossedRows();
399 AliWarning("Cannot get n TPC crossed rows for non-track object");
400 fComputedValue = 0.0;
404 case kNTPCcrossedRowsFclusters:
406 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
407 fComputedValue = 1.0;
409 if (trackESD->GetTPCNclsF()>0) fComputedValue = trackESD->GetTPCCrossedRows() / trackESD->GetTPCNclsF();
411 Float_t nCrossedRows = ((AliAODTrack*) track)->GetTPCNCrossedRows();
412 Float_t nFcls = ((AliAODTrack*) track)->GetTPCNclsF();
413 if (nFcls>0) fComputedValue = nCrossedRows / nFcls;
417 AliWarning("Cannot get n TPC crossed rows/findable clusters for non-track object");
418 fComputedValue = 0.0;
424 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
426 UShort_t nClustersITS = trackESD->GetITSclusters(0);
427 fComputedValue = trackESD->GetITSchi2()/Float_t(nClustersITS);
429 fComputedValue = ((AliAODTrack *)track)->Chi2perNDF();
433 AliWarning("Cannot get ITS chi^2 for non-track object");
434 fComputedValue = 0.0;
439 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
441 UShort_t nClustersTPC = trackESD->GetTPCclusters(0);
442 fComputedValue = trackESD->GetTPCchi2()/Float_t(nClustersTPC);
444 fComputedValue = ((AliAODTrack *)track)->Chi2perNDF();
448 AliWarning("Cannot get TPC chi^2 for non-track object");
449 fComputedValue = 0.0;
454 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
456 Float_t b[2], bCov[3];
457 trackESD->GetImpactParameters(b, bCov);
458 fComputedValue = b[0];
460 Double_t b[2]= {-999,-999}, cov[3];
461 AliAODVertex *vertex = fEvent->GetRefAOD()->GetPrimaryVertex();
463 track->PropagateToDCA(vertex, fEvent->GetRefAOD()->GetMagneticField(),kVeryBig, b, cov);
464 fComputedValue = b[0];
466 fComputedValue = -999;
471 AliWarning("Cannot get TPC chi^2 for non-track object");
472 fComputedValue = 0.0;
477 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
479 Float_t b[2], bCov[3];
480 trackESD->GetImpactParameters(b, bCov);
481 fComputedValue = b[1];
483 Double_t b[2]= {-999,-999}, cov[3];
484 AliAODVertex *vertex = fEvent->GetRefAOD()->GetPrimaryVertex();
486 track->PropagateToDCA(vertex, fEvent->GetRefAOD()->GetMagneticField(),kVeryBig, b, cov);
487 fComputedValue = b[1];
489 fComputedValue = -999;
495 AliWarning("Cannot get TPC chi^2 for non-track object");
496 fComputedValue = 0.0;
501 fComputedValue = (fUseMCInfo ? refMC->Charge() : ref->Charge());
505 fComputedValue = (fUseMCInfo ? (refMC->Phi()*TMath::RadToDeg()) : (ref->Phi()*TMath::RadToDeg()));
510 Double_t pos[3]={0.,0.,0.};
511 Double_t phiOut = -999.0;
512 Double_t radius = 278.;//TPC outer (vessel) = 278 cm, TOF radius (active surf.) = 378 cm; ref. PPR.1
513 AliExternalTrackParam etp; //thanks to Andrea and Cristina
514 etp.CopyFromVTrack(track);
515 if(etp.GetXYZAt(radius, 5., pos)){
516 phiOut=TMath::ATan2(pos[1],pos[0])*TMath::RadToDeg();
517 if (phiOut<0) phiOut+= (2*TMath::Pi()*TMath::RadToDeg());
519 fComputedValue = phiOut;
521 AliWarning("Cannot get phi at outer TPC radius for non-track object");
522 fComputedValue = -99.0;
529 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
530 fComputedValue = v0ESD->GetD(xPrimaryVertex,yPrimaryVertex,zPrimaryVertex);
534 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
535 fComputedValue = v0AOD->DcaV0ToPrimVertex();
539 fComputedValue = -999;
545 Double_t v0Position[3]; // from $ALICE_ROOT/ANALYSIS/AliESDV0Cuts.cxx
546 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
547 v0ESD->GetXYZ(v0Position[0],v0Position[1],v0Position[2]);
548 fComputedValue = TMath::Sqrt(TMath::Power(v0Position[0],2) + TMath::Power(v0Position[1],2));
552 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
553 fComputedValue = v0AOD->RadiusV0();
557 fComputedValue = -999;
563 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
564 fComputedValue = v0ESD->GetEffMass();
568 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
569 fComputedValue = v0AOD->MassLambda();
573 fComputedValue = -999;
579 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
580 fComputedValue = v0ESD->P();
584 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
585 fComputedValue = v0AOD->Ptot2V0();
589 fComputedValue = -999;
595 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
596 fComputedValue = v0ESD->Pt();
600 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
601 fComputedValue = v0AOD->Pt2V0();
605 fComputedValue = -999;
611 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
612 fComputedValue = v0ESD->GetDcaV0Daughters();
616 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
617 fComputedValue = v0AOD->DcaV0Daughters();
621 fComputedValue = -999;
627 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
628 fComputedValue = v0ESD->GetV0CosineOfPointingAngle();
632 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
633 AliAODVertex *vertex = fEvent->GetRefAOD()->GetPrimaryVertex();
634 fComputedValue = v0AOD->CosPointingAngle(vertex);
638 fComputedValue = -999;
642 case kLambdaProtonPIDCut:
643 if(v0esd && lESDEvent) {
644 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
645 // retrieve the V0 daughters
646 UInt_t lIdxPos = (UInt_t) TMath::Abs(v0ESD->GetPindex());
647 AliESDtrack *pTrack = lESDEvent->GetTrack(lIdxPos);
648 AliPIDResponse *pid = fEvent->GetPIDResponse();
649 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(pTrack, AliPID::kProton));
653 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
654 AliAODTrack *pTrack = (AliAODTrack *) (v0AOD->GetSecondaryVtx()->GetDaughter(0));
655 AliPIDResponse *pid = fEvent->GetPIDResponse();
656 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(pTrack, AliPID::kProton));
660 fComputedValue = -999;
664 case kAntiLambdaAntiProtonPIDCut:
665 if(v0esd && lESDEvent) {
666 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
667 // retrieve the V0 daughters
668 UInt_t lIdxNeg = (UInt_t) TMath::Abs(v0ESD->GetNindex());
669 AliESDtrack *nTrack = lESDEvent->GetTrack(lIdxNeg);
670 AliPIDResponse *pid = fEvent->GetPIDResponse();
671 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(nTrack, AliPID::kProton));
675 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
676 AliAODTrack *nTrack = (AliAODTrack *) (v0AOD->GetSecondaryVtx()->GetDaughter(1));
677 AliPIDResponse *pid = fEvent->GetPIDResponse();
678 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(nTrack, AliPID::kProton));
682 fComputedValue = -999;
686 case kLambdaPionPIDCut:
687 if(v0esd && lESDEvent) {
688 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
689 // retrieve the V0 daughters
690 UInt_t lIdxNeg = (UInt_t) TMath::Abs(v0ESD->GetNindex());
691 AliESDtrack *nTrack = lESDEvent->GetTrack(lIdxNeg);
692 AliPIDResponse *pid = fEvent->GetPIDResponse();
693 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(nTrack, AliPID::kPion));
697 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
698 AliAODTrack *nTrack = (AliAODTrack *) (v0AOD->GetSecondaryVtx()->GetDaughter(1));
699 AliPIDResponse *pid = fEvent->GetPIDResponse();
700 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(nTrack, AliPID::kPion));
704 fComputedValue = -999;
708 case kAntiLambdaAntiPionPIDCut:
709 if(v0esd && lESDEvent) {
710 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
711 // retrieve the V0 daughters
712 UInt_t lIdxPos = (UInt_t) TMath::Abs(v0ESD->GetPindex());
713 AliESDtrack *pTrack = lESDEvent->GetTrack(lIdxPos);
714 AliPIDResponse *pid = fEvent->GetPIDResponse();
715 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(pTrack, AliPID::kPion));
719 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
720 AliAODTrack *pTrack = (AliAODTrack *) (v0AOD->GetSecondaryVtx()->GetDaughter(0));
721 AliPIDResponse *pid = fEvent->GetPIDResponse();
722 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(pTrack, AliPID::kPion));
726 fComputedValue = -999;
731 AliError(Form("[%s] Invalid value type for this computation", GetName()));