]> git.uio.no Git - u/mrichter/AliRoot.git/blame - STEER/AliExternalTrackParam.h
Templeted version of AliExternalTrackParam::Set (Matthias)
[u/mrichter/AliRoot.git] / STEER / 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"
def9660e 24
49d13e89 25const Double_t kVeryBig=1./kAlmost0;
7b5ef2e6 26const Double_t kMostProbablePt=0.35;
5773defd 27
ee5dba5e 28Double_t ApproximateBetheBloch(Double_t);
29
58e536c5 30class AliVVertex;
0c19adf7 31class TPolyMarker3D;
c683ddc2 32
4f6e22bd 33class AliExternalTrackParam: public AliVTrack {
51ad6848 34 public:
35 AliExternalTrackParam();
6c94f330 36 AliExternalTrackParam(const AliExternalTrackParam &);
def9660e 37 AliExternalTrackParam& operator=(const AliExternalTrackParam & trkPar);
51ad6848 38 AliExternalTrackParam(Double_t x, Double_t alpha,
39 const Double_t param[5], const Double_t covar[15]);
4f6e22bd 40 AliExternalTrackParam(const AliVTrack *vTrack);
da4e3deb 41 AliExternalTrackParam(Double_t xyz[3],Double_t pxpypz[3],
42 Double_t cv[21],Short_t sign);
1530f89c 43 virtual ~AliExternalTrackParam(){}
ac3eaff4 44
1443179a 45 template <typename T>
46 void Set(T x, T alpha, const T param[5], const T covar[15]) {
47 // Sets the parameters
48 fX=x; fAlpha=alpha;
49 for (Int_t i = 0; i < 5; i++) fP[i] = param[i];
50 for (Int_t i = 0; i < 15; i++) fC[i] = covar[i];
51 }
52
4f6e22bd 53 void Set(Double_t xyz[3],Double_t pxpypz[3],Double_t cv[21],Short_t sign);
ed5f2849 54
55 static void SetMostProbablePt(Double_t pt) { fgMostProbablePt=pt; }
56 static Double_t GetMostProbablePt() { return fgMostProbablePt; }
57
c9ec41e8 58 void Reset();
ed5f2849 59 void ResetCovariance(Double_t s2);
3775b0ca 60 void AddCovariance(const Double_t cov[15]);
c9ec41e8 61
49d13e89 62 const Double_t *GetParameter() const {return fP;}
63 const Double_t *GetCovariance() const {return fC;}
6c94f330 64
65 Double_t GetAlpha() const {return fAlpha;}
c7bafca9 66 Double_t GetX() const {return fX;}
2d57349e 67 Double_t GetY() const {return fP[0];}
68 Double_t GetZ() const {return fP[1];}
6c94f330 69 Double_t GetSnp() const {return fP[2];}
70 Double_t GetTgl() const {return fP[3];}
def9660e 71 Double_t GetSigned1Pt() const {return fP[4];}
6c94f330 72
73 Double_t GetSigmaY2() const {return fC[0];}
74 Double_t GetSigmaZY() const {return fC[1];}
75 Double_t GetSigmaZ2() const {return fC[2];}
76 Double_t GetSigmaSnpY() const {return fC[3];}
77 Double_t GetSigmaSnpZ() const {return fC[4];}
78 Double_t GetSigmaSnp2() const {return fC[5];}
79 Double_t GetSigmaTglY() const {return fC[6];}
80 Double_t GetSigmaTglZ() const {return fC[7];}
81 Double_t GetSigmaTglSnp() const {return fC[8];}
82 Double_t GetSigmaTgl2() const {return fC[9];}
83 Double_t GetSigma1PtY() const {return fC[10];}
84 Double_t GetSigma1PtZ() const {return fC[11];}
85 Double_t GetSigma1PtSnp() const {return fC[12];}
86 Double_t GetSigma1PtTgl() const {return fC[13];}
87 Double_t GetSigma1Pt2() const {return fC[14];}
88
def9660e 89 // additional functions for AliVParticle
90 Double_t Px() const;
91 Double_t Py() const;
92 Double_t Pz() const;
93 Double_t Pt() const { return TMath::Abs(GetSignedPt()); }
94 Double_t P() const { return GetP(); }
c683ddc2 95 Bool_t PxPyPz(Double_t p[3]) const { return GetPxPyPz(p); }
96
97 Double_t Xv() const;
98 Double_t Yv() const;
99 Double_t Zv() const;
100 Bool_t XvYvZv(Double_t x[3]) const { return GetXYZ(x); }
101
def9660e 102 Double_t OneOverPt() const { return 1./Pt(); }
103 Double_t Phi() const;
104 Double_t Theta() const;
105 virtual Double_t E() const;
106 virtual Double_t M() const;
107 Double_t Eta() const;
108 virtual Double_t Y() const;
109 Short_t Charge() const { return (Short_t)GetSign(); }
67be2d29 110 virtual const Double_t *PID() const { return 0x0; }
def9660e 111
4f6e22bd 112 // additional functions from AliVTrack
113 virtual Int_t GetID() const { return -999; }
114 virtual UChar_t GetITSClusterMap() const {return 0; }
115 virtual ULong_t GetStatus() const { return 0; }
116
c9ec41e8 117 Double_t GetSign() const {return (fP[4]>0) ? 1 : -1;}
118 Double_t GetP() const;
def9660e 119 Double_t GetSignedPt() const {
6c94f330 120 return (TMath::Abs(fP[4])>kAlmost0) ? 1./fP[4]:TMath::Sign(kVeryBig,fP[4]);
121 }
1d99986f 122 Double_t Get1P() const;
1530f89c 123 Double_t GetC(Double_t b) const {return fP[4]*b*kB2C;}
124 void GetDZ(Double_t x,Double_t y,Double_t z,Double_t b,Float_t dz[2]) const;
c7bafca9 125 Double_t GetD(Double_t xv, Double_t yv, Double_t b) const;
49d13e89 126 Double_t GetLinearD(Double_t xv, Double_t yv) const;
116b445b 127 Bool_t CorrectForMeanMaterial(Double_t xOverX0, Double_t xTimesRho,
7dded1d5 128 Double_t mass, Bool_t anglecorr=kFALSE,
129 Double_t (*f)(Double_t)=ApproximateBetheBloch);
49d13e89 130 Double_t GetPredictedChi2(Double_t p[2],Double_t cov[3]) const;
1e023a36 131
4b189f98 132 Double_t
133 GetPredictedChi2(Double_t p[3],Double_t covyz[3],Double_t covxyz[3]) const;
1e023a36 134 Bool_t
135 PropagateTo(Double_t p[3],Double_t covyz[3],Double_t covxyz[3],Double_t b);
136
e23a38cb 137 Double_t *GetResiduals(Double_t *p,Double_t *cov,Bool_t updated=kTRUE) const;
49d13e89 138 Bool_t Update(Double_t p[2],Double_t cov[3]);
139 Bool_t Rotate(Double_t alpha);
140 Bool_t PropagateTo(Double_t x, Double_t b);
141 Bool_t Propagate(Double_t alpha, Double_t x, Double_t b) {
142 if (Rotate(alpha))
143 if (PropagateTo(x,b)) return kTRUE;
144 return kFALSE;
145 }
052daaff 146 void Propagate(Double_t len,Double_t x[3],Double_t p[3],Double_t bz) const;
147 Bool_t Intersect(Double_t pnt[3], Double_t norm[3], Double_t bz) const;
49d13e89 148
c7bafca9 149 void GetHelixParameters(Double_t h[6], Double_t b) const;
150 Double_t GetDCA(const AliExternalTrackParam *p, Double_t b,
151 Double_t &xthis,Double_t &xp) const;
152 Double_t PropagateToDCA(AliExternalTrackParam *p, Double_t b);
58e536c5 153 Bool_t PropagateToDCA(const AliVVertex *vtx, Double_t b, Double_t maxd,
e99a34df 154 Double_t dz[2]=0, Double_t cov[3]=0);
155
b1149664 156 void GetDirection(Double_t d[3]) const;
c9ec41e8 157 Bool_t GetPxPyPz(Double_t *p) const;
158 Bool_t GetXYZ(Double_t *p) const;
159 Bool_t GetCovarianceXYZPxPyPz(Double_t cv[21]) const;
160 Bool_t GetPxPyPzAt(Double_t x, Double_t b, Double_t p[3]) const;
161 Bool_t GetXYZAt(Double_t x, Double_t b, Double_t r[3]) const;
7cf7bb6c 162 Bool_t GetYAt(Double_t x, Double_t b, Double_t &y) const;
6c94f330 163 Bool_t GetZAt(Double_t x, Double_t b, Double_t &z) const;
c9ec41e8 164 void Print(Option_t* option = "") const;
c194ba83 165 Double_t GetSnpAt(Double_t x,Double_t b) const;
6c94f330 166
116b445b 167 //Deprecated
168 Bool_t CorrectForMaterial(Double_t d, Double_t x0, Double_t mass,
169 Double_t (*f)(Double_t)=ApproximateBetheBloch);
170
bf00ebb8 171 Bool_t GetDistance(AliExternalTrackParam *param2, Double_t x, Double_t dist[3], Double_t b);
0e8460af 172 Int_t GetIndex(Int_t i, Int_t j) const;
3c43fb2b 173 Int_t GetLabel() const {return -1;}
0c19adf7 174 //
175 // visualization (M. Ivanov)
176 //
177 virtual void FillPolymarker(TPolyMarker3D *pol, Float_t magf, Float_t minR, Float_t maxR, Float_t stepR);
178 virtual void DrawTrack(Float_t magF, Float_t minR, Float_t maxR, Float_t stepR);
179 protected:
6c94f330 180 Double_t &Par(Int_t i) {return fP[i];}
181 Double_t &Cov(Int_t i) {return fC[i];}
0c19adf7 182
183 private:
bad5ac2c 184 Double32_t fX; // X coordinate for the point of parametrisation
185 Double32_t fAlpha; // Local <-->global coor.system rotation angle
186 Double32_t fP[5]; // The track parameters
187 Double32_t fC[15]; // The track parameter covariance matrix
c9ec41e8 188
ed5f2849 189 static Double32_t fgMostProbablePt; // "Most probable" pt
190 // (to be used if Bz=0)
4f6e22bd 191 ClassDef(AliExternalTrackParam, 8)
51ad6848 192};
193
ed5f2849 194inline void AliExternalTrackParam::ResetCovariance(Double_t s2) {
195 //
196 // Reset the covarince matrix to "something big"
197 //
198 fC[0]*= s2;
199 fC[1] = 0.; fC[2]*= s2;
200 fC[3] = 0.; fC[4] = 0.; fC[5]*= s2;
201 fC[6] = 0.; fC[7] = 0.; fC[8] = 0.; fC[9]*= s2;
202 fC[10]= 0.; fC[11]= 0.; fC[12]= 0.; fC[13]= 0.; fC[14]*=s2;
203}
204
bf00ebb8 205
206
207
51ad6848 208#endif