]> git.uio.no Git - u/mrichter/AliRoot.git/blob - STEER/AliAODv0.h
Updates for pile-up vertex (F. Prino)
[u/mrichter/AliRoot.git] / STEER / AliAODv0.h
1 #ifndef ALIAODV0_H
2 #define ALIAODV0_H
3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 //-------------------------------------------------------------------------
7 //     Analysis Oriented Data (AOD) V0 vertex class
8 //     Authors: B.Hippolyte, IPHC, hippolyt@in2p3.fr 
9 //              G.Van Buren, BNL,  gene@bnl.gov      (original STAR MuDsts)
10 //-------------------------------------------------------------------------
11
12 #include "AliAODRecoDecay.h"
13
14 class AliAODv0 : public AliAODRecoDecay {
15
16 public:
17
18   AliAODv0();
19   AliAODv0(AliAODVertex *rAODVertex, Double_t rDcaV0Daughters, Double_t rDcaV0ToPrimVertex,
20            const Double_t *rMomPos, const Double_t *rMomNeg, Double_t *rDcaDaughterToPrimVertex);
21   virtual ~AliAODv0();
22
23   AliAODv0(const AliAODv0& rAliAODv0);
24   AliAODv0& operator=(const AliAODv0& rAliAODv0);
25
26   void     Fill(AliAODVertex *rAODVertex, Double_t rDcaV0Daughters, Double_t rDcaV0ToPrimVertex,
27                 const Double_t *rMomPos, const Double_t *rMomNeg, const Double_t *rDcaDaughterToPrimVertex);
28   void     ResetV0();
29   void     Print(Option_t* option = "") const;
30
31   void     SetOnFlyStatus(Bool_t status){fOnFlyStatus=status;}
32   Bool_t   GetOnFlyStatus() const {return fOnFlyStatus;}
33
34   Double_t DecayVertexV0X() const;
35   Double_t DecayVertexV0Y() const;
36   Double_t DecayVertexV0Z() const;
37
38   Double_t DecayLengthV0(const Double_t *) const;
39                      
40   Double_t DcaV0Daughters()     const;
41   Double_t DcaV0ToPrimVertex()  const;
42   Double_t DcaPosToPrimVertex() const; 
43   Double_t DcaNegToPrimVertex() const; 
44   Double_t RadiusV0()           const;
45   Double_t OpenAngleV0()        const;
46
47   Double_t MomPosX() const;
48   Double_t MomPosY() const;
49   Double_t MomPosZ() const;
50   Double_t MomNegX() const;
51   Double_t MomNegY() const;
52   Double_t MomNegZ() const;
53
54   Double_t Chi2V0()  const;
55
56   Double_t MomV0X()  const;
57   Double_t MomV0Y()  const;
58   Double_t MomV0Z()  const;
59
60   Double_t Ptot2Pos() const;
61   Double_t Ptot2Neg() const;
62   Double_t Ptot2V0()  const;
63   Double_t Pt2V0()    const;
64   Double_t MomPosAlongV0() const;
65   Double_t MomNegAlongV0() const;
66   Double_t AlphaV0() const;
67   Double_t PtArmV0() const;
68   Double_t EPosProton() const;
69   Double_t ENegProton() const;
70   Double_t EPosPion() const;
71   Double_t ENegPion() const;
72   Double_t ELambda() const;
73   Double_t EK0Short() const;
74   Double_t MassLambda() const;
75   Double_t MassAntiLambda() const;
76   Double_t MassK0Short() const;
77   Double_t RapK0Short() const;
78   Double_t RapLambda() const;
79   Double_t PseudoRapV0()    const;
80   Double_t PseudoRapPos()   const;
81   Double_t PseudoRapNeg()   const;
82   
83   Short_t  GetPosID()       const;
84   Short_t  GetNegID()       const;
85   Int_t    GetLabel()       const {return -1;} // Dummy
86   Int_t    PdgCode()        const {return  0;} // Dummy
87   
88 protected:
89   Double32_t fDcaV0ToPrimVertex;    // dca of V0 to primary vertex 
90   Bool_t     fOnFlyStatus;          // if kTRUE, then this V0 is recontructed
91                                     // "on fly" during the tracking
92   ClassDef(AliAODv0,2)
93 };
94
95 inline Double_t AliAODv0::DecayVertexV0X() const {return this->GetSecVtxX();}
96 inline Double_t AliAODv0::DecayVertexV0Y() const {return this->GetSecVtxY();}
97 inline Double_t AliAODv0::DecayVertexV0Z() const {return this->GetSecVtxZ();}
98
99 inline Double_t AliAODv0::DecayLengthV0(const Double_t *tParentVertexPosition) const {
100   return ::sqrt(::pow(DecayVertexV0X() - tParentVertexPosition[0],2) +
101                 ::pow(DecayVertexV0Y() - tParentVertexPosition[1],2) +
102                 ::pow(DecayVertexV0Z() - tParentVertexPosition[2],2));
103 }
104
105 inline Double_t AliAODv0::DcaV0Daughters() const {return fDCA[0];}
106 inline Double_t AliAODv0::DcaV0ToPrimVertex() const {return fDcaV0ToPrimVertex;}
107 inline Double_t AliAODv0::DcaPosToPrimVertex() const {return fd0[0];}
108 inline Double_t AliAODv0::DcaNegToPrimVertex() const {return fd0[1];}
109
110 inline Double_t AliAODv0::RadiusV0() const {
111   return RadiusSecVtx();
112 }
113
114 inline Double_t AliAODv0::OpenAngleV0() const {
115   Double_t lScalPtot1Ptot2 = PxProng(0)*PxProng(1)+PyProng(0)*PyProng(1)+PzProng(0)*PzProng(1);
116   Double_t lPtot1xPtot2 = Ptot2Pos()*Ptot2Neg();
117   return ::acos(lScalPtot1Ptot2/::sqrt(lPtot1xPtot2) );
118 }
119
120 inline Double_t AliAODv0::MomPosX() const {return fPx[0];}
121 inline Double_t AliAODv0::MomPosY() const {return fPy[0];}
122 inline Double_t AliAODv0::MomPosZ() const {return fPz[0];}
123 inline Double_t AliAODv0::MomNegX() const {return fPx[1];}
124 inline Double_t AliAODv0::MomNegY() const {return fPy[1];}
125 inline Double_t AliAODv0::MomNegZ() const {return fPz[1];}
126
127 inline Double_t AliAODv0::Chi2V0() const {return GetSecondaryVtx()->GetChi2perNDF();}
128
129 // Compare eventually AliAODv0::MomV0X() and AliAODRecoDecay::Px()
130 inline Double_t AliAODv0::MomV0X() const {return MomPosX()+MomNegX();}
131 inline Double_t AliAODv0::MomV0Y() const {return MomPosY()+MomNegY();}
132 inline Double_t AliAODv0::MomV0Z() const {return MomPosZ()+MomNegZ();}
133
134 inline Double_t AliAODv0::Ptot2Pos() const {
135   return (::pow(MomPosX(),2) + ::pow(MomPosY(),2) + ::pow(MomPosZ(),2) );
136 }
137 inline Double_t AliAODv0::Ptot2Neg() const {
138   return (::pow(MomNegX(),2) + ::pow(MomNegY(),2) + ::pow(MomNegZ(),2) );
139 }
140 inline Double_t AliAODv0::Ptot2V0() const {return ( Pt2V0() + ::pow(MomV0Z(),2) );}
141 inline Double_t AliAODv0::Pt2V0() const {
142   return (::pow(MomV0X(),2) + ::pow(MomV0Y(),2) );
143 }
144
145 inline Double_t AliAODv0::MomPosAlongV0() const {
146   Double_t lPtot2V0 = Ptot2V0();
147   if (lPtot2V0)
148     return (MomPosX()*MomV0X() +
149             MomPosY()*MomV0Y() +
150             MomPosZ()*MomV0Z()) / ::sqrt(lPtot2V0);
151   return 0.;
152 }
153
154 inline Double_t AliAODv0::MomNegAlongV0() const {
155   Double_t lPtot2V0 = Ptot2V0();
156   if (lPtot2V0)
157     return (MomNegX()*MomV0X() +
158             MomNegY()*MomV0Y() +
159             MomNegZ()*MomV0Z()) / ::sqrt(lPtot2V0);
160   return 0.;
161 }
162
163 inline Double_t AliAODv0::AlphaV0() const {
164   return Alpha();
165 }
166 inline Double_t AliAODv0::PtArmV0() const {
167   return QtProng(0);
168 }
169
170 inline Double_t AliAODv0::EPosProton() const {
171   return EProng(0,2212);
172 }
173
174 inline Double_t AliAODv0::ENegProton() const {
175   return EProng(1,2212);
176 }
177
178 inline Double_t AliAODv0::EPosPion() const {
179   return EProng(0,211);
180 }
181
182 inline Double_t AliAODv0::ENegPion() const {
183   return EProng(1,211);
184 }
185
186 inline Double_t AliAODv0::ELambda() const {
187   return E(3122);
188 }
189
190 inline Double_t AliAODv0::EK0Short() const {
191   return E(310);
192 }
193
194 inline Double_t AliAODv0::MassLambda() const {
195   return InvMass2Prongs(0,1,2212,211);
196 }
197
198 inline Double_t AliAODv0::MassAntiLambda() const {
199   return InvMass2Prongs(0,1,211,2212);
200 }
201
202 inline Double_t AliAODv0::MassK0Short() const {
203   return InvMass2Prongs(0,1,211,211);
204 }
205
206 inline Double_t AliAODv0::RapK0Short() const {
207   return Y(310);
208 }
209
210 inline Double_t AliAODv0::RapLambda() const {
211   return Y(3122);
212 }
213
214 inline Double_t AliAODv0::PseudoRapV0() const {
215   return Eta();
216 }
217
218 inline Double_t AliAODv0::PseudoRapPos()   const {
219   return EtaProng(0);
220 }
221
222 inline Double_t AliAODv0::PseudoRapNeg()   const {
223   return EtaProng(1);
224 }
225 //----------------------------------------------------------------------------
226
227 #endif