OnFlyStatus added (B. Hyppolyte)
[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 AliESDEvent;
15 //class AliESDv0;
16
17 class AliAODv0 : public AliAODRecoDecay {
18
19 public:
20
21   AliAODv0();
22   //  AliAODv0(AliESDv0 *rV0Vertex, AliESDEvent *rEvent);
23   AliAODv0(AliAODVertex *rAODVertex, Double_t rDcaV0Daughters, Double_t rDcaV0ToPrimVertex,
24            Double_t *rMomPos, Double_t *rMomNeg, Double_t *rDcaDaughterToPrimVertex);
25   virtual ~AliAODv0();
26
27   AliAODv0(const AliAODv0& rAliAODv0);
28   AliAODv0& operator=(const AliAODv0& rAliAODv0);
29
30   //  void     Fill(AliESDv0 *rV0Vertex, AliESDEvent *rEvent);
31   void     Fill(AliAODVertex *rAODVertex, Double_t rDcaV0Daughters, Double_t rDcaV0ToPrimVertex,
32                 Double_t *rMomPos, Double_t *rMomNeg, Double_t *rDcaDaughterToPrimVertex);
33   void     ResetV0();
34   void     Print(Option_t* option = "") const;
35
36   void     SetOnFlyStatus(Bool_t status){fOnFlyStatus=status;}
37   Bool_t   GetOnFlyStatus() const {return fOnFlyStatus;}
38
39   Double_t DecayVertexV0X() const;
40   Double_t DecayVertexV0Y() const;
41   Double_t DecayVertexV0Z() const;
42
43   Double_t DecayLengthV0(double*) const;
44                      
45   Double_t DcaV0Daughters() const;
46   Double_t DcaV0ToPrimVertex() const;
47   Double_t DcaPosToPrimVertex() const; 
48   Double_t DcaNegToPrimVertex() const; 
49   Double_t RadiusV0()           const;
50   Double_t OpenAngleV0()        const;
51
52   Double_t MomPosX() const;
53   Double_t MomPosY() const;
54   Double_t MomPosZ() const;
55   Double_t MomNegX() const;
56   Double_t MomNegY() const;
57   Double_t MomNegZ() const;
58
59   //  Int_t    KeyPos()  const;
60   //  Int_t    KeyNeg()  const;
61
62   Double_t Chi2V0()  const;
63
64   Double_t MomV0X()  const;
65   Double_t MomV0Y()  const;
66   Double_t MomV0Z()  const;
67
68   Double_t Ptot2Pos() const;
69   Double_t Ptot2Neg() const;
70   Double_t Ptot2V0()  const;
71   Double_t Pt2V0()    const;
72   Double_t MomPosAlongV0() const;
73   Double_t MomNegAlongV0() const;
74   Double_t AlphaV0() const;
75   Double_t PtArmV0() const;
76   Double_t EPosProton() const;
77   Double_t ENegProton() const;
78   Double_t EPosPion() const;
79   Double_t ENegPion() const;
80   Double_t ELambda() const;
81   Double_t EK0Short() const;
82   Double_t MassLambda() const;
83   Double_t MassAntiLambda() const;
84   Double_t MassK0Short() const;
85   Double_t RapK0Short() const;
86   Double_t RapLambda() const;
87   Double_t PseudoRapV0()    const;
88   Double_t PseudoRapPos()   const;
89   Double_t PseudoRapNeg()   const;
90
91
92 protected:
93   Double32_t fDcaV0ToPrimVertex;    // dca of V0 to primary vertex 
94   Bool_t     fOnFlyStatus;          // if kTRUE, then this V0 is recontructed
95                                     // "on fly" during the tracking
96   ClassDef(AliAODv0,2)
97 };
98
99 inline Double_t AliAODv0::DecayVertexV0X() const {return this->GetSecVtxX();}
100 inline Double_t AliAODv0::DecayVertexV0Y() const {return this->GetSecVtxY();}
101 inline Double_t AliAODv0::DecayVertexV0Z() const {return this->GetSecVtxZ();}
102
103 inline Double_t AliAODv0::DecayLengthV0(double *tParentVertexPosition) const {
104   return ::sqrt(::pow(DecayVertexV0X() - tParentVertexPosition[0],2) +
105                 ::pow(DecayVertexV0Y() - tParentVertexPosition[1],2) +
106                 ::pow(DecayVertexV0Z() - tParentVertexPosition[2],2));
107 }
108
109 inline Double_t AliAODv0::DcaV0Daughters() const {return fDCA[0];}
110 inline Double_t AliAODv0::DcaV0ToPrimVertex() const {return fDcaV0ToPrimVertex;}
111 inline Double_t AliAODv0::DcaPosToPrimVertex() const {return fd0[0];}
112 inline Double_t AliAODv0::DcaNegToPrimVertex() const {return fd0[1];}
113
114 inline Double_t AliAODv0::RadiusV0() const {
115   return RadiusSecVtx();
116 }
117
118 inline Double_t AliAODv0::OpenAngleV0() const {
119   Double_t lPtot1xPtot2 = PxProng(0)*PxProng(1)+PyProng(0)*PyProng(1)+PzProng(0)*PzProng(1);
120   Double_t lPtot1Ptot2_2 = Ptot2Pos()*Ptot2Neg();
121   return ::acos(lPtot1xPtot2/::sqrt(lPtot1Ptot2_2) );
122 }
123
124 inline Double_t AliAODv0::MomPosX() const {return fPx[0];}
125 inline Double_t AliAODv0::MomPosY() const {return fPy[0];}
126 inline Double_t AliAODv0::MomPosZ() const {return fPz[0];}
127 inline Double_t AliAODv0::MomNegX() const {return fPx[1];}
128 inline Double_t AliAODv0::MomNegY() const {return fPy[1];}
129 inline Double_t AliAODv0::MomNegZ() const {return fPz[1];}
130
131 // inline Int_t AliAODv0::KeyPos() const {return fKeyPos;}
132 // inline Int_t AliAODv0::KeyNeg() const {return fKeyNeg;}
133
134 inline Double_t AliAODv0::Chi2V0() const {return GetSecondaryVtx()->GetChi2perNDF();}
135
136 // Compare eventually AliAODv0::MomV0X() and AliAODRecoDecay::Px()
137 inline Double_t AliAODv0::MomV0X() const {return MomPosX()+MomNegX();}
138 inline Double_t AliAODv0::MomV0Y() const {return MomPosY()+MomNegY();}
139 inline Double_t AliAODv0::MomV0Z() const {return MomPosZ()+MomNegZ();}
140
141 inline Double_t AliAODv0::Ptot2Pos() const {
142   return (::pow(MomPosX(),2) + ::pow(MomPosY(),2) + ::pow(MomPosZ(),2) );
143 }
144 inline Double_t AliAODv0::Ptot2Neg() const {
145   return (::pow(MomNegX(),2) + ::pow(MomNegY(),2) + ::pow(MomNegZ(),2) );
146 }
147 inline Double_t AliAODv0::Ptot2V0() const {return ( Pt2V0() + ::pow(MomV0Z(),2) );}
148 inline Double_t AliAODv0::Pt2V0() const {
149   return (::pow(MomV0X(),2) + ::pow(MomV0Y(),2) );
150 }
151
152 inline Double_t AliAODv0::MomPosAlongV0() const {
153   Double_t lPtot2V0 = Ptot2V0();
154   if (lPtot2V0)
155     return (MomPosX()*MomV0X() +
156             MomPosY()*MomV0Y() +
157             MomPosZ()*MomV0Z()) / ::sqrt(lPtot2V0);
158   return 0.;
159 }
160
161 inline Double_t AliAODv0::MomNegAlongV0() const {
162   Double_t lPtot2V0 = Ptot2V0();
163   if (lPtot2V0)
164     return (MomNegX()*MomV0X() +
165             MomNegY()*MomV0Y() +
166             MomNegZ()*MomV0Z()) / ::sqrt(lPtot2V0);
167   return 0.;
168 }
169
170 inline Double_t AliAODv0::AlphaV0() const {
171   return Alpha();
172 }
173 inline Double_t AliAODv0::PtArmV0() const {
174   return QtProng(0);
175 }
176
177 inline Double_t AliAODv0::EPosProton() const {
178   return EProng(0,2212);
179 }
180
181 inline Double_t AliAODv0::ENegProton() const {
182   return EProng(1,2212);
183 }
184
185 inline Double_t AliAODv0::EPosPion() const {
186   return EProng(0,211);
187 }
188
189 inline Double_t AliAODv0::ENegPion() const {
190   return EProng(1,211);
191 }
192
193 inline Double_t AliAODv0::ELambda() const {
194   return E(3122);
195 }
196
197 inline Double_t AliAODv0::EK0Short() const {
198   return E(310);
199 }
200
201 inline Double_t AliAODv0::MassLambda() const {
202   return InvMass2Prongs(0,1,2212,211);
203 }
204
205 inline Double_t AliAODv0::MassAntiLambda() const {
206   return InvMass2Prongs(0,1,211,2212);
207 }
208
209 inline Double_t AliAODv0::MassK0Short() const {
210   return InvMass2Prongs(0,1,211,211);
211 }
212
213 inline Double_t AliAODv0::RapK0Short() const {
214   return Y(310);
215 }
216
217 inline Double_t AliAODv0::RapLambda() const {
218   return Y(3122);
219 }
220
221 inline Double_t AliAODv0::PseudoRapV0() const {
222   return Eta();
223 }
224
225 inline Double_t AliAODv0::PseudoRapPos()   const {
226   return EtaProng(0);
227 }
228
229 inline Double_t AliAODv0::PseudoRapNeg()   const {
230   return EtaProng(1);
231 }
232 //----------------------------------------------------------------------------
233
234 #endif