1 #ifndef ALIAODRECODECAYHF_H
2 #define ALIAODRECODECAYHF_H
3 /* Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
8 //***********************************************************
9 // Class AliAODRecoDecayHF
10 // base class for AOD reconstructed heavy-flavour decays
11 // Author: A.Dainese, andrea.dainese@lnl.infn.it
12 //***********************************************************
16 #include "AliAODTrack.h"
17 #include "AliAODRecoDecay.h"
23 class AliAODRecoDecayHF : public AliAODRecoDecay {
28 AliAODRecoDecayHF(AliAODVertex *vtx2,Int_t nprongs,Short_t charge,
29 Double_t *px,Double_t *py,Double_t *pz,
30 Double_t *d0,Double_t *d0err);
31 AliAODRecoDecayHF(AliAODVertex *vtx2,Int_t nprongs,Short_t charge,
32 Double_t *d0,Double_t *d0err);
33 AliAODRecoDecayHF(Double_t vtx1[3],Double_t vtx2[3],
34 Int_t nprongs,Short_t charge,
35 Double_t *px,Double_t *py,Double_t *pz,Double_t *d0);
36 virtual ~AliAODRecoDecayHF();
38 AliAODRecoDecayHF(const AliAODRecoDecayHF& source);
39 AliAODRecoDecayHF& operator=(const AliAODRecoDecayHF& source);
43 void SetPrimaryVtxRef(TObject *vtx) { fEventPrimaryVtx = vtx; }
44 AliAODVertex* GetPrimaryVtxRef() const { return (AliAODVertex*)(fEventPrimaryVtx.GetObject()); }
45 void SetOwnPrimaryVtx(const AliAODVertex *vtx) { UnsetOwnPrimaryVtx(); fOwnPrimaryVtx = new AliAODVertex(*vtx);}
46 void CheckOwnPrimaryVtx() const
47 {if(!fOwnPrimaryVtx) printf("fOwnPrimaryVtx not set"); return;}
48 AliAODVertex* GetOwnPrimaryVtx() const {return fOwnPrimaryVtx;}
49 void GetOwnPrimaryVtx(Double_t vtx[3]) const
50 {CheckOwnPrimaryVtx();fOwnPrimaryVtx->GetPosition(vtx);}
51 void UnsetOwnPrimaryVtx() {if(fOwnPrimaryVtx) {delete fOwnPrimaryVtx; fOwnPrimaryVtx=0;} return;}
52 AliAODVertex* GetPrimaryVtx() const { return (GetOwnPrimaryVtx() ? GetOwnPrimaryVtx() : GetPrimaryVtxRef()); }
53 AliAODVertex* RemoveDaughtersFromPrimaryVtx(AliAODEvent *aod);
54 void RecalculateImpPars(AliAODVertex *vtxAODNew,AliAODEvent *aod);
57 // kinematics & topology
58 Double_t DecayLength2() const
59 { return AliAODRecoDecay::DecayLength2(GetPrimaryVtx());}
60 Double_t DecayLength() const
61 { return AliAODRecoDecay::DecayLength(GetPrimaryVtx());}
62 Double_t DecayLengthError() const
63 { return AliAODRecoDecay::DecayLengthError(GetPrimaryVtx());}
64 Double_t NormalizedDecayLength() const
65 { return AliAODRecoDecay::NormalizedDecayLength(GetPrimaryVtx());}
66 Double_t NormalizedDecayLength2() const
67 { return AliAODRecoDecay::NormalizedDecayLength2(GetPrimaryVtx());}
68 Double_t DecayLengthXY() const
69 { return AliAODRecoDecay::DecayLengthXY(GetPrimaryVtx());}
70 Double_t DecayLengthXYError() const
71 { return AliAODRecoDecay::DecayLengthXYError(GetPrimaryVtx());}
72 Double_t NormalizedDecayLengthXY() const
73 { return AliAODRecoDecay::NormalizedDecayLengthXY(GetPrimaryVtx());}
74 Double_t Ct(UInt_t pdg) const
75 { return AliAODRecoDecay::Ct(pdg,GetPrimaryVtx());}
76 Double_t CosPointingAngle() const
77 { return AliAODRecoDecay::CosPointingAngle(GetPrimaryVtx());}
78 Double_t CosPointingAngleXY() const
79 { return AliAODRecoDecay::CosPointingAngleXY(GetPrimaryVtx());}
80 Double_t ImpParXY() const
81 { return AliAODRecoDecay::ImpParXY(GetPrimaryVtx());}
82 Double_t QtProngFlightLine(Int_t ip) const
83 { return AliAODRecoDecay::QtProngFlightLine(ip,GetPrimaryVtx());}
84 Double_t QlProngFlightLine(Int_t ip) const
85 { return AliAODRecoDecay::QlProngFlightLine(ip,GetPrimaryVtx());}
88 Double_t Getd0errProng(Int_t ip) const {return fd0err[ip];}
89 void Setd0errProngs(Int_t nprongs,Double_t *d0);
90 Double_t Normalizedd0Prong(Int_t ip) const
91 {return Getd0Prong(ip)/Getd0errProng(ip);}
93 void SetProngIDs(Int_t nIDs,UShort_t *id);
94 UShort_t GetProngID(Int_t ip) const
95 {if(fProngID) {return fProngID[ip];} else {return 9999;}}
97 // ITS clustermap for daughters
98 Bool_t DaughterHasPointOnITSLayer(Int_t dg,Int_t l) const;
100 // check if it is like-sign
101 Bool_t IsLikeSign() const;
104 void SetListOfCutsRef(TObject *obj) {fListOfCuts=obj;}
105 TList *GetListOfCuts() const {return (TList*)(fListOfCuts.GetObject());}
106 AliRDHFCuts *GetCuts(const char* name) const;
109 AliKFParticle *ApplyVertexingKF(Int_t *iprongs,Int_t nprongs,Int_t *pdgs,
110 Bool_t topoCostraint,Double_t bzkG,
111 Double_t *mass) const;
114 void Misalign(TString misal="null");
117 void SetSelectionBit(Int_t i) {SETBIT(fSelectionMap,i); return;}
118 Bool_t HasSelectionBit(Int_t i) const {return TESTBIT(fSelectionMap,i);}
119 ULong_t GetSelectionMap() const {return fSelectionMap;}
121 Int_t NumberOfFakeDaughters() const;
123 Bool_t HasBadDaughters() const; // TPC+ITS tracks not passing the StandardCuts2010 with loose DCA
127 AliAODVertex *fOwnPrimaryVtx; // primary vertex for this candidate
128 TRef fEventPrimaryVtx; // ref to primary vertex of the event
129 TRef fListOfCuts; // ref to the list of analysis cuts
130 Double_t *fd0err; //[fNProngs] error on prongs rphi impact param [cm]
131 UShort_t *fProngID; //[fNProngs] track ID of daughters
132 ULong_t fSelectionMap; // used to store outcome of selection in AliAnalysisVertexingHF
134 ClassDef(AliAODRecoDecayHF,5) // base class for AOD reconstructed heavy-flavour decays
137 inline void AliAODRecoDecayHF::SetProngIDs(Int_t nIDs,UShort_t *id)
139 if(nIDs!=GetNProngs()) {
140 printf("Wrong number of IDs, must be nProngs\n");
143 if(fProngID) delete [] fProngID;
144 fProngID = new UShort_t[nIDs];
145 for(Int_t i=0;i<nIDs;i++)
150 inline Bool_t AliAODRecoDecayHF::IsLikeSign() const
152 // check if it is like-sign
154 Int_t ndg=GetNDaughters();
156 printf("Daughters not available\n");
159 Int_t chargeDg0 = ((AliAODTrack*)GetDaughter(0))->Charge();
161 for(Int_t i=1; i<ndg; i++) {
162 if(chargeDg0!=((AliAODTrack*)GetDaughter(i))->Charge()) return kFALSE;
168 inline AliRDHFCuts *AliAODRecoDecayHF::GetCuts(const char* name) const
170 // returns the analysis cuts
172 TList *list = GetListOfCuts();
176 return (AliRDHFCuts*)list->FindObject(name);
179 inline Bool_t AliAODRecoDecayHF::DaughterHasPointOnITSLayer(Int_t dg,Int_t l) const
181 // ITS clustermap for daughters
184 printf("ERROR: layer has to be in the range 0-5\n");
187 AliAODTrack *t = (AliAODTrack*)GetDaughter(dg);
188 if(!t) return kFALSE;
190 return TESTBIT(t->GetITSClusterMap(),l);
193 inline Int_t AliAODRecoDecayHF::NumberOfFakeDaughters() const
195 // Count number of daughters with negative label
198 for(Int_t i=0; i<GetNDaughters(); i++) {
199 AliAODTrack *track=(AliAODTrack*)GetDaughter(i);
201 if(track->Charge()==0) { // this is a two prong decay
202 AliAODRecoDecay *rd=(AliAODRecoDecay*)GetDaughter(i);
203 for(Int_t j=0; j<rd->GetNDaughters(); j++) {
204 AliAODTrack *track2=(AliAODTrack*)GetDaughter(j);
205 if(track2->GetLabel()<0) nfakes++;
210 if(track->GetLabel()<0) nfakes++;
215 inline void AliAODRecoDecayHF::Setd0errProngs(Int_t nprongs,Double_t *d0err)
217 if(nprongs!=GetNProngs()) {
218 printf("Wrong number of momenta, must be nProngs");
222 fd0err = new Double32_t[nprongs];
224 for(Int_t i=0;i<nprongs;i++) {
225 fd0err[i] = d0err[i];
231 inline Bool_t AliAODRecoDecayHF::HasBadDaughters() const {
233 // checks if at least one of ITS+TPC daughters didn't pass the StandardCuts2010 with loose DCA (ie doesn't have the BIT(4) set)
235 for(Int_t iDau=0; iDau<GetNDaughters(); iDau++){
236 if (GetDaughter(iDau)==NULL) continue;
237 AliAODTrack* at=dynamic_cast<AliAODTrack*>(GetDaughter(iDau));
239 if(at->Charge()==0) continue;
240 if(at->GetTPCNcls()==0) continue;
241 if(!(at->TestFilterMask(BIT(4)))) return kTRUE;