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 kMass: return "SingleTrackMass";
110 case kITSsignal: return "SingleTrackITSsignal";
111 case kTPCsignal: return "SingleTrackTPCsignal";
112 case kTOFsignal: return "SingleTrackTOFsignal";
113 case kTPCnsigmaPi: return "SingleTrackTPCnsigmaPion";
114 case kTPCnsigmaK: return "SingleTrackTPCnsigmaKaon";
115 case kTPCnsigmaP: return "SingleTrackTPCnsigmaProton";
116 case kTOFnsigmaPi: return "SingleTrackTOFnsigmaPion";
117 case kTOFnsigmaK: return "SingleTrackTOFnsigmaKaon";
118 case kTOFnsigmaP: return "SingleTrackTOFnsigmaProton";
119 case kTOFdeltaPi: return "SingleTrackTOFdeltaPion";
120 case kTOFdeltaK: return "SingleTrackTOFdeltaKaon";
121 case kTOFdeltaP: return "SingleTrackTOFdeltaProton";
122 case kNITSclusters: return "SingleTrackNITSclusters";
123 case kNTPCclusters: return "SingleTrackNTPCclusters";
124 case kNTPCcrossedRows: return "SingleTrackNTPCcrossedRows";
125 case kNTPCcrossedRowsFclusters: return "SingleTrackNTPCcrossedRowsFclusters";
126 case kITSchi2: return "SingleTrackITSchi2";
127 case kTPCchi2: return "SingleTrackTPCchi2";
128 case kDCAXY: return "SingleTrackDCAz";
129 case kDCAZ: return "SingleTrackDCAz";
130 case kCharge: return "SingleTrackCharge";
131 case kPhi: return "SingleTrackPhi";
132 case kPhiOuterTPC: return "SingleTrackPhiOuterTPC";
133 case kV0DCA: return "V0DCAToPrimaryVertex";
134 case kDaughterDCA: return "V0DaughterDCA";
135 case kCosPointAng: return "V0CosineOfPointingAngle";
136 case kLambdaProtonPIDCut: return "V0LambdaProtonNsigma";
137 case kAntiLambdaAntiProtonPIDCut: return "V0AntiLambdaAntiProtonNsigma";
138 case kLambdaPionPIDCut: return "V0LambdaPionNsigma";
139 case kAntiLambdaAntiPionPIDCut: return "V0AntiLambdaPionNsigma";
140 default: return "Undefined";
144 //_____________________________________________________________________________
145 Bool_t AliRsnValueDaughter::Eval(TObject *object)
148 // Evaluation of the required value.
149 // Checks that the passed object is of the right type
150 // and if this check is successful, computes the required value.
151 // The output of the function tells if computing was successful,
152 // and the values must be taken with GetValue().
155 // coherence check, which also casts object
156 // to AliRsnTarget data members and returns kFALSE
157 // in case the object is NULL
158 if (!TargetOK(object)) return kFALSE;
160 // set a reference to the mother momentum
161 AliVParticle *ref = fDaughter->GetRef();
162 AliVParticle *refMC = fDaughter->GetRefMC();
163 AliVTrack *track = fDaughter->Ref2Vtrack();
164 AliESDv0 *v0esd = fDaughter->Ref2ESDv0();
165 AliAODv0 *v0aod = fDaughter->Ref2AODv0();
166 AliESDEvent *lESDEvent = fEvent->GetRefESD();
168 Double_t xPrimaryVertex = -999.9;
169 Double_t yPrimaryVertex = -999.9;
170 Double_t zPrimaryVertex = -999.9;
174 xPrimaryVertex = lESDEvent->GetPrimaryVertex()->GetX();
175 yPrimaryVertex = lESDEvent->GetPrimaryVertex()->GetY();
176 zPrimaryVertex = lESDEvent->GetPrimaryVertex()->GetZ();
180 if (fUseMCInfo && !refMC) {
184 if (!fUseMCInfo && !ref) {
189 // compute value depending on types in the enumeration
190 // if the type does not match any available choice, or if
191 // the computation is not doable due to any problem
192 // (not initialized support object, wrong values, risk of floating point errors)
193 // the method returng kFALSE and sets the computed value to a meaningless number
196 fComputedValue = (fUseMCInfo ? refMC->P() : ref->P());
200 fComputedValue = (fUseMCInfo ? refMC->Pt() : ref->Pt());
204 fComputedValue = (fUseMCInfo ? refMC->Eta() : ref->Eta());
208 fComputedValue = (fUseMCInfo ? refMC->M() : ref->M());
213 fComputedValue = track->GetTPCmomentum();
216 AliWarning("Cannot get TPC momentum for non-track object");
217 fComputedValue = 0.0;
223 fComputedValue = track->GetITSsignal();
226 AliWarning("Cannot get ITS signal for non-track object");
227 fComputedValue = 0.0;
233 fComputedValue = track->GetTPCsignal();
236 AliWarning("Cannot get TPC signal for non-track object");
237 fComputedValue = 0.0;
243 fComputedValue = track->GetTOFsignal();
246 AliWarning("Cannot get TOF signal for non-track object");
247 fComputedValue = 0.0;
253 AliPIDResponse *pid = fEvent->GetPIDResponse();
254 fComputedValue = pid->NumberOfSigmasTPC(track, AliPID::kPion);
257 AliWarning("Cannot get TOF signal for non-track object");
258 fComputedValue = 0.0;
264 AliPIDResponse *pid = fEvent->GetPIDResponse();
265 fComputedValue = pid->NumberOfSigmasTPC(track, AliPID::kKaon);
268 AliWarning("Cannot get TOF signal for non-track object");
269 fComputedValue = 0.0;
275 AliPIDResponse *pid = fEvent->GetPIDResponse();
276 fComputedValue = pid->NumberOfSigmasTPC(track, AliPID::kProton);
279 AliWarning("Cannot get TOF signal for non-track object");
280 fComputedValue = 0.0;
286 AliPIDResponse *pid = fEvent->GetPIDResponse();
287 fComputedValue = pid->NumberOfSigmasTOF(track, AliPID::kPion);
290 AliWarning("Cannot get TOF signal for non-track object");
291 fComputedValue = 0.0;
297 AliPIDResponse *pid = fEvent->GetPIDResponse();
298 fComputedValue = pid->NumberOfSigmasTOF(track, AliPID::kKaon);
301 AliWarning("Cannot get TOF signal for non-track object");
302 fComputedValue = 0.0;
308 AliPIDResponse *pid = fEvent->GetPIDResponse();
309 fComputedValue = pid->NumberOfSigmasTOF(track, AliPID::kProton);
312 AliWarning("Cannot get TOF signal for non-track object");
313 fComputedValue = 0.0;
319 AliPIDResponse *pid = fEvent->GetPIDResponse();
320 pid->GetSignalDelta(AliPIDResponse::kTOF, track, AliPID::kPion, fComputedValue, kFALSE);//(==AliPIDResponse::kDetPidOk);
323 AliWarning("Cannot get TOF delta for non-track object");
324 fComputedValue = 0.0;
330 AliPIDResponse *pid = fEvent->GetPIDResponse();
331 pid->GetSignalDelta(AliPIDResponse::kTOF, track, AliPID::kKaon, fComputedValue, kFALSE);//(==AliPIDResponse::kDetPidOk);
334 AliWarning("Cannot get TOF delta for non-track object");
335 fComputedValue = 0.0;
341 AliPIDResponse *pid = fEvent->GetPIDResponse();
342 pid->GetSignalDelta(AliPIDResponse::kTOF, track, AliPID::kProton, fComputedValue, kFALSE);//(==AliPIDResponse::kDetPidOk);
345 AliWarning("Cannot get TOF delta for non-track object");
346 fComputedValue = 0.0;
352 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
354 fComputedValue = trackESD->GetITSclusters(0);
356 fComputedValue = ((AliAODTrack *)track)->GetITSNcls();
360 AliWarning("Cannot get n ITS clusters for non-track object");
361 fComputedValue = 0.0;
367 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
369 fComputedValue = trackESD->GetTPCclusters(0);
371 fComputedValue = ((AliAODTrack *)track)->GetTPCNcls();
375 AliWarning("Cannot get n TPC clusters for non-track object");
376 fComputedValue = 0.0;
380 case kNTPCcrossedRows:
382 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
384 fComputedValue = trackESD->GetTPCCrossedRows();
386 fComputedValue = ((AliAODTrack *)track)->GetTPCNCrossedRows();
390 AliWarning("Cannot get n TPC crossed rows for non-track object");
391 fComputedValue = 0.0;
395 case kNTPCcrossedRowsFclusters:
397 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
398 fComputedValue = 1.0;
400 if (trackESD->GetTPCNclsF()>0) fComputedValue = trackESD->GetTPCCrossedRows() / trackESD->GetTPCNclsF();
402 Float_t nCrossedRows = ((AliAODTrack*) track)->GetTPCNCrossedRows();
403 Float_t nFcls = ((AliAODTrack*) track)->GetTPCNclsF();
404 if (nFcls>0) fComputedValue = nCrossedRows / nFcls;
408 AliWarning("Cannot get n TPC crossed rows/findable clusters for non-track object");
409 fComputedValue = 0.0;
415 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
417 UShort_t nClustersITS = trackESD->GetITSclusters(0);
418 fComputedValue = trackESD->GetITSchi2()/Float_t(nClustersITS);
420 fComputedValue = ((AliAODTrack *)track)->Chi2perNDF();
424 AliWarning("Cannot get ITS chi^2 for non-track object");
425 fComputedValue = 0.0;
430 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
432 UShort_t nClustersTPC = trackESD->GetTPCclusters(0);
433 fComputedValue = trackESD->GetTPCchi2()/Float_t(nClustersTPC);
435 fComputedValue = ((AliAODTrack *)track)->Chi2perNDF();
439 AliWarning("Cannot get TPC chi^2 for non-track object");
440 fComputedValue = 0.0;
445 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
447 Float_t b[2], bCov[3];
448 trackESD->GetImpactParameters(b, bCov);
449 fComputedValue = b[0];
451 Double_t b[2]= {-999,-999}, cov[3];
452 AliAODVertex *vertex = fEvent->GetRefAOD()->GetPrimaryVertex();
454 track->PropagateToDCA(vertex, fEvent->GetRefAOD()->GetMagneticField(),kVeryBig, b, cov);
455 fComputedValue = b[0];
457 fComputedValue = -999;
462 AliWarning("Cannot get TPC chi^2 for non-track object");
463 fComputedValue = 0.0;
468 AliESDtrack *trackESD = dynamic_cast<AliESDtrack *>(track);
470 Float_t b[2], bCov[3];
471 trackESD->GetImpactParameters(b, bCov);
472 fComputedValue = b[1];
474 Double_t b[2]= {-999,-999}, cov[3];
475 AliAODVertex *vertex = fEvent->GetRefAOD()->GetPrimaryVertex();
477 track->PropagateToDCA(vertex, fEvent->GetRefAOD()->GetMagneticField(),kVeryBig, b, cov);
478 fComputedValue = b[1];
480 fComputedValue = -999;
486 AliWarning("Cannot get TPC chi^2 for non-track object");
487 fComputedValue = 0.0;
492 fComputedValue = (fUseMCInfo ? refMC->Charge() : ref->Charge());
496 fComputedValue = (fUseMCInfo ? (refMC->Phi()*TMath::RadToDeg()) : (ref->Phi()*TMath::RadToDeg()));
501 Double_t pos[3]={0.,0.,0.};
502 Double_t phiOut = -999.0;
503 Double_t radius = 278.;//TPC outer (vessel) = 278 cm, TOF radius (active surf.) = 378 cm; ref. PPR.1
504 AliExternalTrackParam etp; //thanks to Andrea and Cristina
505 etp.CopyFromVTrack(track);
506 if(etp.GetXYZAt(radius, 5., pos)){
507 phiOut=TMath::ATan2(pos[1],pos[0])*TMath::RadToDeg();
508 if (phiOut<0) phiOut+= (2*TMath::Pi()*TMath::RadToDeg());
510 fComputedValue = phiOut;
512 AliWarning("Cannot get phi at outer TPC radius for non-track object");
513 fComputedValue = -99.0;
520 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
521 fComputedValue = v0ESD->GetD(xPrimaryVertex,yPrimaryVertex,zPrimaryVertex);
525 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
526 fComputedValue = v0AOD->DcaV0ToPrimVertex();
530 fComputedValue = -999;
536 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
537 fComputedValue = v0ESD->GetDcaV0Daughters();
541 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
542 fComputedValue = v0AOD->DcaV0Daughters();
546 fComputedValue = -999;
552 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
553 fComputedValue = v0ESD->GetV0CosineOfPointingAngle();
557 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
558 fComputedValue = TMath::Cos(v0AOD->OpenAngleV0());
562 fComputedValue = -999;
566 case kLambdaProtonPIDCut:
567 if(v0esd && lESDEvent) {
568 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
569 // retrieve the V0 daughters
570 UInt_t lIdxPos = (UInt_t) TMath::Abs(v0ESD->GetPindex());
571 AliESDtrack *pTrack = lESDEvent->GetTrack(lIdxPos);
572 AliPIDResponse *pid = fEvent->GetPIDResponse();
573 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(pTrack, AliPID::kProton));
577 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
578 AliAODTrack *pTrack = (AliAODTrack *) (v0AOD->GetSecondaryVtx()->GetDaughter(0));
579 AliPIDResponse *pid = fEvent->GetPIDResponse();
580 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(pTrack, AliPID::kProton));
584 fComputedValue = -999;
588 case kAntiLambdaAntiProtonPIDCut:
589 if(v0esd && lESDEvent) {
590 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
591 // retrieve the V0 daughters
592 UInt_t lIdxNeg = (UInt_t) TMath::Abs(v0ESD->GetNindex());
593 AliESDtrack *nTrack = lESDEvent->GetTrack(lIdxNeg);
594 AliPIDResponse *pid = fEvent->GetPIDResponse();
595 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(nTrack, AliPID::kProton));
599 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
600 AliAODTrack *nTrack = (AliAODTrack *) (v0AOD->GetSecondaryVtx()->GetDaughter(1));
601 AliPIDResponse *pid = fEvent->GetPIDResponse();
602 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(nTrack, AliPID::kProton));
606 fComputedValue = -999;
610 case kLambdaPionPIDCut:
611 if(v0esd && lESDEvent) {
612 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
613 // retrieve the V0 daughters
614 UInt_t lIdxNeg = (UInt_t) TMath::Abs(v0ESD->GetNindex());
615 AliESDtrack *nTrack = lESDEvent->GetTrack(lIdxNeg);
616 AliPIDResponse *pid = fEvent->GetPIDResponse();
617 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(nTrack, AliPID::kPion));
621 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
622 AliAODTrack *nTrack = (AliAODTrack *) (v0AOD->GetSecondaryVtx()->GetDaughter(1));
623 AliPIDResponse *pid = fEvent->GetPIDResponse();
624 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(nTrack, AliPID::kPion));
628 fComputedValue = -999;
632 case kAntiLambdaAntiPionPIDCut:
633 if(v0esd && lESDEvent) {
634 AliESDv0 *v0ESD = dynamic_cast<AliESDv0 *>(v0esd);
635 // retrieve the V0 daughters
636 UInt_t lIdxPos = (UInt_t) TMath::Abs(v0ESD->GetPindex());
637 AliESDtrack *pTrack = lESDEvent->GetTrack(lIdxPos);
638 AliPIDResponse *pid = fEvent->GetPIDResponse();
639 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(pTrack, AliPID::kPion));
643 AliAODv0 *v0AOD = dynamic_cast<AliAODv0 *>(v0aod);
644 AliAODTrack *pTrack = (AliAODTrack *) (v0AOD->GetSecondaryVtx()->GetDaughter(0));
645 AliPIDResponse *pid = fEvent->GetPIDResponse();
646 fComputedValue = TMath::Abs(pid->NumberOfSigmasTPC(pTrack, AliPID::kPion));
650 fComputedValue = -999;
655 AliError(Form("[%s] Invalid value type for this computation", GetName()));