Coverity
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAODRecoDecayHF.h
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                               */
5
6 /* $Id$ */ 
7
8 //***********************************************************
9 // Class AliAODRecoDecayHF
10 // base class for AOD reconstructed heavy-flavour decays
11 // Author: A.Dainese, andrea.dainese@lnl.infn.it
12 //***********************************************************
13
14 #include <TRef.h>
15 #include <TList.h>
16 #include "AliAODTrack.h"
17 #include "AliAODRecoDecay.h"
18
19 class AliAODEvent;
20 class AliRDHFCuts;
21 class AliKFParticle;
22
23 class AliAODRecoDecayHF : public AliAODRecoDecay {
24
25  public:
26
27   AliAODRecoDecayHF();
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();
37
38   AliAODRecoDecayHF(const AliAODRecoDecayHF& source);
39   AliAODRecoDecayHF& operator=(const AliAODRecoDecayHF& source); 
40    
41
42   // primary vertex
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);
55
56
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());}
86
87   // prongs
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);}
92   
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;}}
96
97   // ITS clustermap for daughters
98   Bool_t DaughterHasPointOnITSLayer(Int_t dg,Int_t l) const;
99
100   // check if it is like-sign
101   Bool_t IsLikeSign() const;
102
103   // list of cuts
104   void SetListOfCutsRef(TObject *obj) {fListOfCuts=obj;}
105   TList *GetListOfCuts() const {return (TList*)(fListOfCuts.GetObject());}
106   AliRDHFCuts *GetCuts(const char* name) const;
107
108   // vertexing KF:
109   AliKFParticle *ApplyVertexingKF(Int_t *iprongs,Int_t nprongs,Int_t *pdgs,
110                                  Bool_t topoCostraint,Double_t bzkG,
111                                  Double_t *mass) const;
112   
113   // misalign
114   void Misalign(TString misal="null");
115
116   // selection map
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;}
120
121   Int_t   NumberOfFakeDaughters() const;
122
123   Bool_t  HasBadDaughters() const; // TPC+ITS tracks not passing the StandardCuts2010 with loose DCA
124
125  protected:
126
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
133
134   ClassDef(AliAODRecoDecayHF,5)  // base class for AOD reconstructed heavy-flavour decays
135 };
136
137 inline void AliAODRecoDecayHF::SetProngIDs(Int_t nIDs,UShort_t *id) 
138 {
139   if(nIDs!=GetNProngs()) { 
140     printf("Wrong number of IDs, must be nProngs\n");
141     return;
142   }
143   if(fProngID) delete [] fProngID;
144   fProngID = new UShort_t[nIDs];
145   for(Int_t i=0;i<nIDs;i++) 
146     fProngID[i] = id[i]; 
147   return;
148 }
149
150 inline Bool_t AliAODRecoDecayHF::IsLikeSign() const
151 {
152   // check if it is like-sign
153
154   Int_t ndg=GetNDaughters();
155   if(!ndg) {
156     printf("Daughters not available\n");
157     return kFALSE;
158   }
159   Int_t chargeDg0 = ((AliAODTrack*)GetDaughter(0))->Charge();
160
161   for(Int_t i=1; i<ndg; i++) {
162     if(chargeDg0!=((AliAODTrack*)GetDaughter(i))->Charge()) return kFALSE;
163   }
164
165   return kTRUE;
166 }
167
168 inline AliRDHFCuts *AliAODRecoDecayHF::GetCuts(const char* name) const
169
170   // returns the analysis cuts
171
172   TList *list = GetListOfCuts();
173   if(!list) return 0;
174
175
176   return (AliRDHFCuts*)list->FindObject(name);
177 }
178
179 inline Bool_t AliAODRecoDecayHF::DaughterHasPointOnITSLayer(Int_t dg,Int_t l) const
180 {
181   // ITS clustermap for daughters
182
183   if(l<0 || l>5) {
184     printf("ERROR: layer has to be in the range 0-5\n");
185     return kFALSE;
186   }
187   AliAODTrack *t = (AliAODTrack*)GetDaughter(dg);
188   if(!t) return kFALSE;
189
190   return TESTBIT(t->GetITSClusterMap(),l);
191 }
192
193 inline Int_t AliAODRecoDecayHF::NumberOfFakeDaughters() const 
194 {
195   // Count number of daughters with negative label
196
197   Int_t nfakes=0;
198   for(Int_t i=0; i<GetNDaughters(); i++) {
199     AliAODTrack *track=(AliAODTrack*)GetDaughter(i);
200
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++;
206       }
207       continue;
208     }
209
210     if(track->GetLabel()<0) nfakes++;
211   }
212   return nfakes;
213 }
214
215 inline void AliAODRecoDecayHF::Setd0errProngs(Int_t nprongs,Double_t *d0err) 
216 {
217   if(nprongs!=GetNProngs()) { 
218     printf("Wrong number of momenta, must be nProngs");
219     return;
220   }
221   if(!fd0) {
222     fd0err = new Double32_t[nprongs];
223   }
224   for(Int_t i=0;i<nprongs;i++) {
225     fd0err[i] = d0err[i]; 
226   }
227
228   return;
229 }
230
231 inline Bool_t AliAODRecoDecayHF::HasBadDaughters() const {
232
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)
234
235   for(Int_t iDau=0; iDau<GetNDaughters(); iDau++){
236     AliAODTrack* at=(AliAODTrack*)GetDaughter(iDau);
237     if(at->Charge()==0) continue;
238     if(at->GetTPCNcls()==0) continue;
239     if(!(at->TestFilterMask(BIT(4)))) return kTRUE;
240   }
241   return kFALSE;
242 }
243
244
245 #endif
246