]>
Commit | Line | Data |
---|---|---|
1 | #ifndef ALIESDV0_H | |
2 | #define ALIESDV0_H | |
3 | ||
4 | /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
5 | * See cxx source for full Copyright notice */ | |
6 | ||
7 | /* $Id$ */ | |
8 | ||
9 | //------------------------------------------------------------------------- | |
10 | // ESD V0 Vertex Class | |
11 | // This class is part of the Event Summary Data set of classes | |
12 | // Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch | |
13 | // Modified by: Marian Ivanov, CERN, Marian.Ivanov@cern.ch | |
14 | // and Boris Hippolyte,IPHC, hippolyt@in2p3.fr | |
15 | //------------------------------------------------------------------------- | |
16 | ||
17 | #include <TPDGCode.h> | |
18 | ||
19 | #include "AliExternalTrackParam.h" | |
20 | #include "AliVParticle.h" | |
21 | ||
22 | class AliESDV0Params; | |
23 | ||
24 | class AliESDv0 : public AliVParticle { | |
25 | public: | |
26 | AliESDv0(); | |
27 | AliESDv0(const AliExternalTrackParam &t1, Int_t i1, | |
28 | const AliExternalTrackParam &t2, Int_t i2); | |
29 | ||
30 | AliESDv0(const AliESDv0& v0); | |
31 | virtual ~AliESDv0(); | |
32 | AliESDv0& operator=(const AliESDv0& v0); | |
33 | virtual void Copy(TObject &obj) const; | |
34 | ||
35 | // Start with AliVParticle functions | |
36 | virtual Double_t Px() const { return fNmom[0]+fPmom[0]; } | |
37 | virtual Double_t Py() const { return fNmom[1]+fPmom[1]; } | |
38 | virtual Double_t Pz() const { return fNmom[2]+fPmom[2]; } | |
39 | virtual Double_t Pt() const { return TMath::Sqrt(Px()*Px()+Py()*Py()); } | |
40 | virtual Double_t P() const { | |
41 | return TMath::Sqrt(Px()*Px()+Py()*Py()+Pz()*Pz()); | |
42 | } | |
43 | virtual Bool_t PxPyPz(Double_t p[3]) const { p[0] = Px(); p[1] = Py(); p[2] = Pz(); return kTRUE; } | |
44 | virtual Double_t Xv() const { return fPos[0]; } | |
45 | virtual Double_t Yv() const { return fPos[1]; } | |
46 | virtual Double_t Zv() const { return fPos[2]; } | |
47 | virtual Bool_t XvYvZv(Double_t x[3]) const { x[0] = Xv(); x[1] = Yv(); x[2] = Zv(); return kTRUE; } | |
48 | virtual Double_t OneOverPt() const { return (Pt() != 0.) ? 1./Pt() : -999.; } | |
49 | virtual Double_t Phi() const {return TMath::Pi()+TMath::ATan2(-Py(),-Px()); } | |
50 | virtual Double_t Theta() const {return 0.5*TMath::Pi()-TMath::ATan(Pz()/(Pt()+1.e-13)); } | |
51 | virtual Double_t E() const; // default is KOs but can be changed via ChangeMassHypothesis (defined in the .cxx) | |
52 | virtual Double_t M() const { return GetEffMass(); } | |
53 | virtual Double_t Eta() const { return 0.5*TMath::Log((P()+Pz())/(P()-Pz()+1.e-13)); } | |
54 | virtual Double_t Y() const; | |
55 | virtual Short_t Charge() const { return 0; } | |
56 | virtual Int_t GetLabel() const { return -1; } // temporary | |
57 | virtual const Double_t *PID() const { return 0; } // return PID object ? (to be discussed!) | |
58 | ||
59 | // Then extend the AliVParticle functions | |
60 | Double_t E(Int_t pdg) const; | |
61 | Double_t Y(Int_t pdg) const; | |
62 | ||
63 | // Now the functions for analysis consistency | |
64 | Double_t RapK0Short() const; | |
65 | Double_t RapLambda() const; | |
66 | Double_t AlphaV0() const; | |
67 | Double_t PtArmV0() const; | |
68 | ||
69 | // Eventually the older functions | |
70 | Double_t ChangeMassHypothesis(Int_t code=kK0Short); | |
71 | ||
72 | Int_t GetPdgCode() const {return fPdgCode;} | |
73 | Double_t GetEffMass(UInt_t p1, UInt_t p2) const; | |
74 | Double_t GetEffMass() const {return fEffMass;} | |
75 | Double_t GetChi2V0() const {return fChi2V0;} | |
76 | void GetPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const; | |
77 | void GetNPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const; | |
78 | void GetPPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const; | |
79 | void GetXYZ(Double_t &x, Double_t &y, Double_t &z) const; | |
80 | Float_t GetD(Double_t x0,Double_t y0,Double_t z0) const; | |
81 | Int_t GetNindex() const {return fNidx;} | |
82 | Int_t GetPindex() const {return fPidx;} | |
83 | void SetDcaV0Daughters(Double_t rDcaV0Daughters=0.); | |
84 | Double_t GetDcaV0Daughters() const {return fDcaV0Daughters;} | |
85 | Float_t GetV0CosineOfPointingAngle(Double_t refPointX, Double_t refPointY, Double_t refPointZ) const; | |
86 | Double_t GetV0CosineOfPointingAngle() const {return fPointAngle;} | |
87 | void SetV0CosineOfPointingAngle(Double_t cpa) {fPointAngle=cpa;} | |
88 | void SetOnFlyStatus(Bool_t status){fOnFlyStatus=status;} | |
89 | Bool_t GetOnFlyStatus() const {return fOnFlyStatus;} | |
90 | const AliExternalTrackParam *GetParamP() const {return &fParamP;} | |
91 | const AliExternalTrackParam *GetParamN() const {return &fParamN;} | |
92 | ||
93 | ||
94 | ||
95 | // **** The following member functions need to be revised *** | |
96 | ||
97 | void GetPosCov(Double_t cov[6])const ; // getter for the covariance matrix of the V0 position | |
98 | Double_t GetSigmaY(); // sigma of y coordinate at vertex posistion | |
99 | Double_t GetSigmaZ(); // sigma of z coordinate at vertex posistion | |
100 | Double_t GetSigmaAP0(); // calculate sigma of Point angle resolution at vertex pos. | |
101 | Double_t GetSigmaD0(); // calculate sigma of position resolution at vertex pos. | |
102 | Double_t GetEffectiveSigmaAP0(); // calculate sigma of point angle resolution at vertex pos. effecive parameterization | |
103 | Double_t GetEffectiveSigmaD0(); // calculate sigma of position resolution at vertex pos. | |
104 | Double_t GetMinimaxSigmaAP0(); // calculate mini-max sigma of point angle resolution | |
105 | Double_t GetMinimaxSigmaD0(); // calculate mini-max sigma of dca resolution | |
106 | Double_t GetLikelihoodAP(Int_t mode0, Int_t mode1); // get likelihood for point angle | |
107 | Double_t GetLikelihoodD(Int_t mode0, Int_t mode1); // get likelihood for DCA | |
108 | Double_t GetLikelihoodC(Int_t mode0, Int_t mode1) const; // get likelihood for Causality | |
109 | // | |
110 | // | |
111 | static const AliESDV0Params & GetParameterization(){return fgkParams;} | |
112 | void SetParamP(const AliExternalTrackParam & paramP) {fParamP = paramP;} | |
113 | void SetParamN(const AliExternalTrackParam & paramN) {fParamN = paramN;} | |
114 | void SetStatus(Int_t status){fStatus=status;} | |
115 | Int_t GetStatus() const {return fStatus;} | |
116 | Int_t GetIndex(Int_t i) const {return (i==0) ? fNidx : fPidx;} | |
117 | void SetIndex(Int_t i, Int_t ind); | |
118 | const Double_t *GetAnglep() const {return fAngle;} | |
119 | Double_t GetRr() const {return fRr;} | |
120 | Double_t GetDistSigma() const {return fDistSigma;} | |
121 | void SetDistSigma(Double_t ds) {fDistSigma=ds;} | |
122 | Float_t GetChi2Before() const {return fChi2Before;} | |
123 | void SetChi2Before(Float_t cb) {fChi2Before=cb;} | |
124 | Float_t GetChi2After() const {return fChi2After;} | |
125 | void SetChi2After(Float_t ca) {fChi2After=ca;} | |
126 | Float_t GetNAfter() const {return fNAfter;} | |
127 | void SetNAfter(Short_t na) {fNAfter=na;} | |
128 | Short_t GetNBefore() const {return fNBefore;} | |
129 | void SetNBefore(Short_t nb) {fNBefore=nb;} | |
130 | void SetCausality(Float_t pb0, Float_t pb1, Float_t pa0, Float_t pa1); | |
131 | const Double_t * GetCausalityP() const {return fCausality;} | |
132 | void SetClusters(const Int_t *clp, const Int_t *clm); | |
133 | const Int_t * GetClusters(Int_t i) const {return fClusters[i];} | |
134 | void SetNormDCAPrim(Float_t nd0, Float_t nd1){fNormDCAPrim[0] = nd0; fNormDCAPrim[1]=nd1;} | |
135 | const Double_t *GetNormDCAPrimP() const {return fNormDCAPrim;} | |
136 | // Dummy | |
137 | Int_t PdgCode() const {return 0;} | |
138 | ||
139 | //virtual Bool_t GetPxPyPz(Double_t */*p*/) const { return kFALSE; } | |
140 | virtual void SetID(Short_t /*id*/) {;} | |
141 | ||
142 | ||
143 | protected: | |
144 | AliExternalTrackParam fParamN; // external parameters of negative particle | |
145 | AliExternalTrackParam fParamP; // external parameters of positive particle | |
146 | ||
147 | // CKBrev: tkink about revision | |
148 | ||
149 | Double32_t fEffMass; // reconstructed V0's effective mass | |
150 | Double32_t fDcaV0Daughters; // dca between V0's daughters | |
151 | Double32_t fChi2V0; // V0's chi2 value | |
152 | Double32_t fPos[3]; // V0's position (global) | |
153 | Double32_t fPosCov[6]; // covariance matrix of the vertex position | |
154 | Double32_t fNmom[3]; // momentum of the negative daughter (global) | |
155 | Double32_t fPmom[3]; // momentum of the positive daughter (global) | |
156 | Double32_t fNormDCAPrim[2]; // normalize distance to the primary vertex CKBrev | |
157 | Double32_t fRr; //rec position of the vertex CKBrev | |
158 | Double32_t fDistSigma; //sigma of distance CKBrev | |
159 | Double32_t fChi2Before; //chi2 of the tracks before V0 CKBrev | |
160 | Double32_t fChi2After; // chi2 of the tracks after V0 CKBrev | |
161 | ||
162 | ||
163 | Double32_t fCausality[4]; //[0,1,8] causality information - see comments in SetCausality CKBrev | |
164 | Double32_t fAngle[3]; //[-2*pi,2*pi,16]three angles CKBrev | |
165 | Double32_t fPointAngleFi; //[-1,1,16]point angle fi CKBrev | |
166 | Double32_t fPointAngleTh; //[-1,1,16]point angle theta CKBrev | |
167 | Double32_t fPointAngle; //[-1,1,16] cosine of the pointing angle | |
168 | ||
169 | ||
170 | Int_t fPdgCode; // reconstructed V0's type (PDG code) | |
171 | Int_t fClusters[2][6]; //! its clusters CKBrev | |
172 | Int_t fNidx; // index of the negative daughter | |
173 | Int_t fPidx; // index of the positive daughter | |
174 | ||
175 | ||
176 | ||
177 | Short_t fStatus; //status CKBrev | |
178 | Short_t fNBefore; // number of possible points before V0 CKBrev | |
179 | Short_t fNAfter; // number of possible points after V0 CKBrev | |
180 | ||
181 | Bool_t fOnFlyStatus; // if kTRUE, then this V0 is recontructed | |
182 | // "on fly" during the tracking | |
183 | ||
184 | // | |
185 | // parameterization coefficients | |
186 | static const AliESDV0Params fgkParams; //! resolution and likelihood parameterization | |
187 | ||
188 | private: | |
189 | ||
190 | ClassDef(AliESDv0,5) // ESD V0 vertex | |
191 | }; | |
192 | ||
193 | inline | |
194 | void AliESDv0::GetNPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const { | |
195 | px=fNmom[0]; py=fNmom[1]; pz=fNmom[2]; | |
196 | } | |
197 | ||
198 | inline | |
199 | void AliESDv0::GetPPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const { | |
200 | px=fPmom[0]; py=fPmom[1]; pz=fPmom[2]; | |
201 | } | |
202 | ||
203 | inline | |
204 | void AliESDv0::SetDcaV0Daughters(Double_t rDcaV0Daughters){ | |
205 | fDcaV0Daughters=rDcaV0Daughters; | |
206 | } | |
207 | ||
208 | inline | |
209 | void AliESDv0::SetIndex(Int_t i, Int_t ind) { | |
210 | if(i==0) | |
211 | fNidx=ind; | |
212 | else | |
213 | fPidx=ind; | |
214 | } | |
215 | ||
216 | #endif |