add a few methods in the v interface
[u/mrichter/AliRoot.git] / STEER / STEERBase / AliExternalTrackParam.h
CommitLineData
51ad6848 1#ifndef ALIEXTERNALTRACKPARAM_H
2#define ALIEXTERNALTRACKPARAM_H
3/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
5
6/* $Id$ */
7
49d13e89 8/*****************************************************************************
9 * "External" track parametrisation class *
10 * *
11 * external param0: local Y-coordinate of a track (cm) *
12 * external param1: local Z-coordinate of a track (cm) *
13 * external param2: local sine of the track momentum azimuthal angle *
14 * external param3: tangent of the track momentum dip angle *
15 * external param4: 1/pt (1/(GeV/c)) *
16 * *
17 * The parameters are estimated at an exact position x in a local coord. *
18 * system rotated by angle alpha with respect to the global coord.system. *
19 * Origin: I.Belikov, CERN, Jouri.Belikov@cern.ch *
20 *****************************************************************************/
6c94f330 21#include "TMath.h"
a5e407e9 22
4f6e22bd 23#include "AliVTrack.h"
1f516476 24#include "AliVMisc.h"
49d13e89 25const Double_t kVeryBig=1./kAlmost0;
7b5ef2e6 26const Double_t kMostProbablePt=0.35;
5773defd 27
58e536c5 28class AliVVertex;
0c19adf7 29class TPolyMarker3D;
c683ddc2 30
86be8934 31const Double_t kC0max=100*100, // SigmaY<=100cm
32 kC2max=100*100, // SigmaZ<=100cm
33 kC5max=1*1, // SigmaSin<=1
34 kC9max=1*1, // SigmaTan<=1
35 kC14max=100*100; // Sigma1/Pt<=100 1/GeV
36
4f6e22bd 37class AliExternalTrackParam: public AliVTrack {
51ad6848 38 public:
39 AliExternalTrackParam();
6c94f330 40 AliExternalTrackParam(const AliExternalTrackParam &);
def9660e 41 AliExternalTrackParam& operator=(const AliExternalTrackParam & trkPar);
51ad6848 42 AliExternalTrackParam(Double_t x, Double_t alpha,
43 const Double_t param[5], const Double_t covar[15]);
da4e3deb 44 AliExternalTrackParam(Double_t xyz[3],Double_t pxpypz[3],
45 Double_t cv[21],Short_t sign);
a4b2769d 46 // constructor for reinitialisation of vtable
1f516476 47 AliExternalTrackParam( AliVConstructorReinitialisationFlag f) :AliVTrack(f), fX(), fAlpha(){}
48 void Reinitialize() { new (this) AliExternalTrackParam( AliVReinitialize ); }
a4b2769d 49
1530f89c 50 virtual ~AliExternalTrackParam(){}
4ec1ca0f 51 void CopyFromVTrack(const AliVTrack *vTrack);
13545423 52
1443179a 53 template <typename T>
54 void Set(T x, T alpha, const T param[5], const T covar[15]) {
55 // Sets the parameters
c4f6245f 56 if (alpha < -TMath::Pi()) alpha += 2*TMath::Pi();
57 else if (alpha >= TMath::Pi()) alpha -= 2*TMath::Pi();
1443179a 58 fX=x; fAlpha=alpha;
59 for (Int_t i = 0; i < 5; i++) fP[i] = param[i];
60 for (Int_t i = 0; i < 15; i++) fC[i] = covar[i];
86be8934 61
62 CheckCovariance();
63
1443179a 64 }
65
a251382e 66 void SetParamOnly(double x, double alpha, const double param[5]) {
67 // Sets the parameters, neglect cov matrix
68 if (alpha < -TMath::Pi()) alpha += 2*TMath::Pi();
69 else if (alpha >= TMath::Pi()) alpha -= 2*TMath::Pi();
70 fX=x; fAlpha=alpha;
71 for (Int_t i = 0; i < 5; i++) fP[i] = param[i];
72 }
73
4f6e22bd 74 void Set(Double_t xyz[3],Double_t pxpypz[3],Double_t cv[21],Short_t sign);
ed5f2849 75
76 static void SetMostProbablePt(Double_t pt) { fgMostProbablePt=pt; }
77 static Double_t GetMostProbablePt() { return fgMostProbablePt; }
78
c9ec41e8 79 void Reset();
ed5f2849 80 void ResetCovariance(Double_t s2);
3775b0ca 81 void AddCovariance(const Double_t cov[15]);
c9ec41e8 82
49d13e89 83 const Double_t *GetParameter() const {return fP;}
84 const Double_t *GetCovariance() const {return fC;}
13545423 85 virtual Bool_t IsStartedTimeIntegral() const {return kFALSE;}
86 virtual void AddTimeStep(Double_t ) {} // dummy method, real stuff is done in AliKalmanTrack
6c94f330 87 Double_t GetAlpha() const {return fAlpha;}
c7bafca9 88 Double_t GetX() const {return fX;}
2d57349e 89 Double_t GetY() const {return fP[0];}
90 Double_t GetZ() const {return fP[1];}
6c94f330 91 Double_t GetSnp() const {return fP[2];}
3740029a 92 virtual Double_t GetTgl() const {return fP[3];}
93 virtual void GetImpactParameters(Float_t& ,Float_t&) const {}
def9660e 94 Double_t GetSigned1Pt() const {return fP[4];}
6c94f330 95
96 Double_t GetSigmaY2() const {return fC[0];}
97 Double_t GetSigmaZY() const {return fC[1];}
98 Double_t GetSigmaZ2() const {return fC[2];}
99 Double_t GetSigmaSnpY() const {return fC[3];}
100 Double_t GetSigmaSnpZ() const {return fC[4];}
101 Double_t GetSigmaSnp2() const {return fC[5];}
102 Double_t GetSigmaTglY() const {return fC[6];}
103 Double_t GetSigmaTglZ() const {return fC[7];}
104 Double_t GetSigmaTglSnp() const {return fC[8];}
105 Double_t GetSigmaTgl2() const {return fC[9];}
106 Double_t GetSigma1PtY() const {return fC[10];}
107 Double_t GetSigma1PtZ() const {return fC[11];}
108 Double_t GetSigma1PtSnp() const {return fC[12];}
109 Double_t GetSigma1PtTgl() const {return fC[13];}
110 Double_t GetSigma1Pt2() const {return fC[14];}
111
def9660e 112 // additional functions for AliVParticle
113 Double_t Px() const;
114 Double_t Py() const;
b52b9586 115 Double_t Pz() const { return Pt()*GetTgl(); }
def9660e 116 Double_t Pt() const { return TMath::Abs(GetSignedPt()); }
117 Double_t P() const { return GetP(); }
c683ddc2 118 Bool_t PxPyPz(Double_t p[3]) const { return GetPxPyPz(p); }
119
120 Double_t Xv() const;
121 Double_t Yv() const;
be15fa9c 122 Double_t Zv() const {return GetZ();}
c683ddc2 123 Bool_t XvYvZv(Double_t x[3]) const { return GetXYZ(x); }
124
def9660e 125 Double_t OneOverPt() const { return 1./Pt(); }
126 Double_t Phi() const;
127 Double_t Theta() const;
128 virtual Double_t E() const;
129 virtual Double_t M() const;
130 Double_t Eta() const;
131 virtual Double_t Y() const;
2258e165 132 virtual Short_t Charge() const { return (Short_t)GetSign(); }
67be2d29 133 virtual const Double_t *PID() const { return 0x0; }
def9660e 134
4f6e22bd 135 // additional functions from AliVTrack
136 virtual Int_t GetID() const { return -999; }
137 virtual UChar_t GetITSClusterMap() const {return 0; }
138 virtual ULong_t GetStatus() const { return 0; }
139
c9ec41e8 140 Double_t GetSign() const {return (fP[4]>0) ? 1 : -1;}
141 Double_t GetP() const;
def9660e 142 Double_t GetSignedPt() const {
6c94f330 143 return (TMath::Abs(fP[4])>kAlmost0) ? 1./fP[4]:TMath::Sign(kVeryBig,fP[4]);
144 }
1d99986f 145 Double_t Get1P() const;
2258e165 146 virtual Double_t GetC(Double_t b) const {return fP[4]*b*kB2C;}
1530f89c 147 void GetDZ(Double_t x,Double_t y,Double_t z,Double_t b,Float_t dz[2]) const;
c7bafca9 148 Double_t GetD(Double_t xv, Double_t yv, Double_t b) const;
49d13e89 149 Double_t GetLinearD(Double_t xv, Double_t yv) const;
b8e07ed6 150
116b445b 151 Bool_t CorrectForMeanMaterial(Double_t xOverX0, Double_t xTimesRho,
7dded1d5 152 Double_t mass, Bool_t anglecorr=kFALSE,
d46683db 153 Double_t (*f)(Double_t)=AliExternalTrackParam::BetheBlochSolid);
154
b8e07ed6 155 Bool_t CorrectForMeanMaterialdEdx(Double_t xOverX0, Double_t xTimesRho,
156 Double_t mass, Double_t dEdx, Bool_t anglecorr=kFALSE);
157
158 Bool_t CorrectForMeanMaterialZA(Double_t xOverX0, Double_t xTimesRho,
159 Double_t mass,
160 Double_t zOverA=0.49848,
161 Double_t density=2.33,
162 Double_t exEnergy=173e-9,
163 Double_t jp1=0.20,
164 Double_t jp2=3.00,
165 Bool_t anglecorr=kFALSE
166 );
167
9c56b409 168 //
169 // Bethe-Bloch formula parameterizations
170 //
171 static Double_t BetheBlochAleph(Double_t bg,
172 Double_t kp1=0.76176e-1,
173 Double_t kp2=10.632,
174 Double_t kp3=0.13279e-4,
175 Double_t kp4=1.8631,
176 Double_t kp5=1.9479
177 );
178 static Double_t BetheBlochGeant(Double_t bg,
179 Double_t kp0=2.33,
180 Double_t kp1=0.20,
181 Double_t kp2=3.00,
182 Double_t kp3=173e-9,
183 Double_t kp4=0.49848
184 );
185
d46683db 186 static Double_t BetheBlochSolid(Double_t bg);
187 static Double_t BetheBlochGas(Double_t bg);
188
49d13e89 189 Double_t GetPredictedChi2(Double_t p[2],Double_t cov[3]) const;
1e023a36 190
4b189f98 191 Double_t
192 GetPredictedChi2(Double_t p[3],Double_t covyz[3],Double_t covxyz[3]) const;
acdfbc78 193
194 Double_t GetPredictedChi2(const AliExternalTrackParam *t) const;
195
1e023a36 196 Bool_t
197 PropagateTo(Double_t p[3],Double_t covyz[3],Double_t covxyz[3],Double_t b);
198
e23a38cb 199 Double_t *GetResiduals(Double_t *p,Double_t *cov,Bool_t updated=kTRUE) const;
49d13e89 200 Bool_t Update(Double_t p[2],Double_t cov[3]);
201 Bool_t Rotate(Double_t alpha);
a251382e 202 Bool_t RotateParamOnly(Double_t alpha);
2d2fd909 203 Bool_t Invert();
49d13e89 204 Bool_t PropagateTo(Double_t x, Double_t b);
599b440e 205 Bool_t PropagateParamOnlyTo(Double_t xk, Double_t b);
9f2bec63 206 Bool_t Propagate(Double_t alpha, Double_t x, Double_t b);
266a0f9b 207 Bool_t PropagateBxByBz(Double_t alpha, Double_t x, Double_t b[3]);
e0302afb 208 Bool_t PropagateParamOnlyBxByBzTo(Double_t xk, const Double_t b[3]);
052daaff 209 void Propagate(Double_t len,Double_t x[3],Double_t p[3],Double_t bz) const;
210 Bool_t Intersect(Double_t pnt[3], Double_t norm[3], Double_t bz) const;
49d13e89 211
8b6e3369 212 static void g3helx3(Double_t qfield, Double_t step, Double_t vect[7]);
213 Bool_t PropagateToBxByBz(Double_t x, const Double_t b[3]);
214
c7bafca9 215 void GetHelixParameters(Double_t h[6], Double_t b) const;
216 Double_t GetDCA(const AliExternalTrackParam *p, Double_t b,
217 Double_t &xthis,Double_t &xp) const;
218 Double_t PropagateToDCA(AliExternalTrackParam *p, Double_t b);
58e536c5 219 Bool_t PropagateToDCA(const AliVVertex *vtx, Double_t b, Double_t maxd,
e99a34df 220 Double_t dz[2]=0, Double_t cov[3]=0);
266a0f9b 221 Bool_t PropagateToDCABxByBz(const AliVVertex *vtx, Double_t b[3],
222 Double_t maxd, Double_t dz[2]=0, Double_t cov[3]=0);
4c3dc2a0 223 Bool_t ConstrainToVertex(const AliVVertex* vtx, Double_t b[3]);
e99a34df 224
b1149664 225 void GetDirection(Double_t d[3]) const;
6a33e0e9 226 Bool_t GetPxPyPz(Double_t *p) const;
c9ec41e8 227 Bool_t GetXYZ(Double_t *p) const;
228 Bool_t GetCovarianceXYZPxPyPz(Double_t cv[21]) const;
229 Bool_t GetPxPyPzAt(Double_t x, Double_t b, Double_t p[3]) const;
230 Bool_t GetXYZAt(Double_t x, Double_t b, Double_t r[3]) const;
b8b98dc2 231 Double_t GetParameterAtRadius(Double_t r, Double_t bz, Int_t parType) const;
232
7cf7bb6c 233 Bool_t GetYAt(Double_t x, Double_t b, Double_t &y) const;
6c94f330 234 Bool_t GetZAt(Double_t x, Double_t b, Double_t &z) const;
c9ec41e8 235 void Print(Option_t* option = "") const;
c194ba83 236 Double_t GetSnpAt(Double_t x,Double_t b) const;
70580f26 237 Bool_t GetXatLabR(Double_t r,Double_t &x, Double_t bz, Int_t dir=0) const;
7d2e151a 238 Bool_t GetXYZatR(Double_t xr,Double_t bz, Double_t *xyz=0, Double_t* alpSect=0) const;
6c94f330 239
116b445b 240 //Deprecated
241 Bool_t CorrectForMaterial(Double_t d, Double_t x0, Double_t mass,
d46683db 242 Double_t (*f)(Double_t)=AliExternalTrackParam::BetheBlochSolid);
116b445b 243
bf00ebb8 244 Bool_t GetDistance(AliExternalTrackParam *param2, Double_t x, Double_t dist[3], Double_t b);
0e8460af 245 Int_t GetIndex(Int_t i, Int_t j) const;
3c43fb2b 246 Int_t GetLabel() const {return -1;}
6a8e543a 247 Int_t PdgCode() const {return 0;}
248
0c19adf7 249 //
250 // visualization (M. Ivanov)
251 //
252 virtual void FillPolymarker(TPolyMarker3D *pol, Float_t magf, Float_t minR, Float_t maxR, Float_t stepR);
253 virtual void DrawTrack(Float_t magF, Float_t minR, Float_t maxR, Float_t stepR);
cfdb62d4 254
255 virtual Bool_t Translate(Double_t *vTrasl,Double_t *covV);
256
86be8934 257 void CheckCovariance();
258
f2cef1b5 259 static Bool_t GetUseLogTermMS() {return fgUseLogTermMS;}
260 static void SetUseLogTermMS(Bool_t v=kTRUE) {fgUseLogTermMS = v;}
261
f8356e9e 262 //---------------------------------------------------------------------------
263 //--the calibration interface--
264 //--to be used in online calibration/QA
265 //--should also be implemented in ESD so it works offline as well
266 //-----------
267 virtual Int_t GetTrackParam ( AliExternalTrackParam & ) const {return 0;}
268 virtual Int_t GetTrackParamRefitted ( AliExternalTrackParam & ) const {return 0;}
269 virtual Int_t GetTrackParamIp ( AliExternalTrackParam & ) const {return 0;}
270 virtual Int_t GetTrackParamTPCInner ( AliExternalTrackParam & ) const {return 0;}
271 virtual Int_t GetTrackParamOp ( AliExternalTrackParam & ) const {return 0;}
272 virtual Int_t GetTrackParamCp ( AliExternalTrackParam & ) const {return 0;}
273 virtual Int_t GetTrackParamITSOut ( AliExternalTrackParam & ) const {return 0;}
274
4ec1ca0f 275 protected:
276 AliExternalTrackParam(const AliVTrack *vTrack);
277
86be8934 278/* protected: */
279 private:
6c94f330 280 Double_t &Par(Int_t i) {return fP[i];}
281 Double_t &Cov(Int_t i) {return fC[i];}
e804766b 282 protected:
bad5ac2c 283 Double32_t fX; // X coordinate for the point of parametrisation
284 Double32_t fAlpha; // Local <-->global coor.system rotation angle
285 Double32_t fP[5]; // The track parameters
286 Double32_t fC[15]; // The track parameter covariance matrix
ddfbc51a 287
ed5f2849 288 static Double32_t fgMostProbablePt; // "Most probable" pt
289 // (to be used if Bz=0)
f2cef1b5 290 static Bool_t fgUseLogTermMS; // use log term in Mult.Stattering evaluation
4f6e22bd 291 ClassDef(AliExternalTrackParam, 8)
51ad6848 292};
293
ed5f2849 294inline void AliExternalTrackParam::ResetCovariance(Double_t s2) {
295 //
296 // Reset the covarince matrix to "something big"
297 //
86be8934 298
299 s2 = TMath::Abs(s2);
300 Double_t fC0=fC[0]*s2,
301 fC2=fC[2]*s2,
302 fC5=fC[5]*s2,
303 fC9=fC[9]*s2,
304 fC14=fC[14]*s2;
305
306 if (fC0>kC0max) fC0 = kC0max;
307 if (fC2>kC2max) fC2 = kC2max;
308 if (fC5>kC5max) fC5 = kC5max;
309 if (fC9>kC9max) fC9 = kC9max;
310 if (fC14>kC14max) fC14 = kC14max;
311
312
313 fC[0] = fC0;
314 fC[1] = 0.; fC[2] = fC2;
315 fC[3] = 0.; fC[4] = 0.; fC[5] = fC5;
316 fC[6] = 0.; fC[7] = 0.; fC[8] = 0.; fC[9] = fC9;
317 fC[10]= 0.; fC[11]= 0.; fC[12]= 0.; fC[13]= 0.; fC[14] = fC14;
ed5f2849 318}
319
bf00ebb8 320
321
322
51ad6848 323#endif