Reconstruction and PID using transition radiation photons: first implementation ...
[u/mrichter/AliRoot.git] / TRD / AliTRDtrack.h
CommitLineData
46d29e70 1#ifndef ALITRDTRACK_H
b3a5a838 2#define ALITRDTRACK_H
46d29e70 3
4/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
b3a5a838 5 * See cxx source for full Copyright notice */
46d29e70 6
5443e65e 7#include <AliKalmanTrack.h>
b3a5a838 8#include <TMath.h>
46d29e70 9
9c9d2487 10#include "AliBarrelTrack.h"
11#include "AliTRDgeometry.h"
12#include "TVector2.h"
13
46d29e70 14class AliTRDcluster;
b3a5a838 15class AliTPCtrack;
79e94bf8 16class AliESDtrack;
5443e65e 17
18const unsigned kMAX_CLUSTERS_PER_TRACK=210;
46d29e70 19
5443e65e 20class AliTRDtrack : public AliKalmanTrack {
46d29e70 21
22// Represents reconstructed TRD track
23
46d29e70 24public:
25
16d9fbba 26 AliTRDtrack():AliKalmanTrack(){fBackupTrack=0;}
a819a5f7 27 AliTRDtrack(const AliTRDcluster *c, UInt_t index, const Double_t xx[5],
46d29e70 28 const Double_t cc[15], Double_t xr, Double_t alpha);
29 AliTRDtrack(const AliTRDtrack& t);
5443e65e 30 AliTRDtrack(const AliKalmanTrack& t, Double_t alpha);
79e94bf8 31 AliTRDtrack(const AliESDtrack& t);
16d9fbba 32 ~AliTRDtrack();
a819a5f7 33 Int_t Compare(const TObject *o) const;
34 void CookdEdx(Double_t low=0.05, Double_t up=0.70);
46d29e70 35
36 Double_t GetAlpha() const {return fAlpha;}
9c9d2487 37 Int_t GetSector() const {
38 //if (fabs(fAlpha) < AliTRDgeometry::GetAlpha()/2) return 0;
39 return Int_t(TVector2::Phi_0_2pi(fAlpha)/AliTRDgeometry::GetAlpha())%AliTRDgeometry::kNsect;}
40
46d29e70 41 Double_t GetC() const {return fC;}
42 Int_t GetClusterIndex(Int_t i) const {return fIndex[i];}
a819a5f7 43 Float_t GetClusterdQdl(Int_t i) const {return fdQdl[i];}
5443e65e 44
46d29e70 45 void GetCovariance(Double_t cov[15]) const;
c5507f6d 46 Double_t GetdEdx() const {return fdEdx;}
79e94bf8 47 Double_t GetPIDsignal() const {return GetdEdx();}
eab5961e 48 Float_t GetPIDsignals(Int_t i) const {return fdEdxPlane[i];}
49 Int_t GetPIDTimBin(Int_t i) const {return fTimBinPlane[i];}
46d29e70 50 Double_t GetEta() const {return fE;}
5443e65e 51
52 void GetExternalCovariance(Double_t cov[15]) const ;
53 void GetExternalParameters(Double_t& xr, Double_t x[5]) const ;
54
55 Double_t GetLikelihoodElectron() const { return fLhElectron; };
56
b8dc2353 57 Double_t Get1Pt() const {return (1e-9*TMath::Abs(fC)/fC + fC)*GetConvConst(); }
46d29e70 58 Double_t GetP() const {
59 return TMath::Abs(GetPt())*sqrt(1.+GetTgl()*GetTgl());
60 }
fd621f36 61 Double_t GetPredictedChi2(const AliTRDcluster*, Double_t h01) const ;
5443e65e 62 Double_t GetPt() const {return 1./Get1Pt();}
46d29e70 63 void GetPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const ;
5443e65e 64 void GetGlobalXYZ(Double_t &x, Double_t &y, Double_t &z) const ;
65 Int_t GetSeedLabel() const { return fSeedLab; }
46d29e70 66 Double_t GetSigmaC2() const {return fCcc;}
67 Double_t GetSigmaTgl2() const {return fCtt;}
68 Double_t GetSigmaY2() const {return fCyy;}
69 Double_t GetSigmaZ2() const {return fCzz;}
5443e65e 70 Double_t GetSnp() const {return fX*fC - fE;}
71 Double_t GetTgl() const {return fT;}
72 Double_t GetX() const {return fX;}
73 Double_t GetY() const {return fY;}
74 Double_t GetZ() const {return fZ;}
46e2d86c 75 UInt_t * GetBackupIndexes() {return fIndexBackup;}
76 UInt_t * GetIndexes() {return fIndex;}
77
46d29e70 78
3c625a9b 79 Double_t GetYat(Double_t xk) const {
1e9bb598 80//-----------------------------------------------------------------
81// This function calculates the Y-coordinate of a track at the plane x=xk.
82// Needed for matching with the TOF (I.Belikov)
83//-----------------------------------------------------------------
84 Double_t c1=fC*fX - fE, r1=TMath::Sqrt(1.- c1*c1);
85 Double_t c2=fC*xk - fE, r2=TMath::Sqrt(1.- c2*c2);
86 return fY + (xk-fX)*(c1+c2)/(r1+r2);
87 }
3c625a9b 88 void SetStop(Bool_t stop) {fStopped=stop;}
89 Bool_t GetStop() const {return fStopped;}
1e9bb598 90
b3a5a838 91 Int_t PropagateTo(Double_t xr, Double_t x0=8.72, Double_t rho=5.86e-3);
5443e65e 92 void ResetCovariance();
46e2d86c 93 void ResetCovariance(Float_t mult);
1e9bb598 94 void ResetClusters() { SetChi2(0.); SetNumberOfClusters(0); }
46d29e70 95 Int_t Rotate(Double_t angle);
96
46d29e70 97 void SetdEdx(Float_t dedx) {fdEdx=dedx;}
eab5961e 98 void SetPIDsignals(Float_t dedx, Int_t i) {fdEdxPlane[i]=dedx;}
99 void SetPIDTimBin(Int_t timbin, Int_t i) {fTimBinPlane[i]=timbin;}
5443e65e 100 void SetLikelihoodElectron(Float_t l) { fLhElectron = l; };
101
102 void SetSampledEdx(Float_t q, Int_t i) {
103 Double_t s=GetSnp(), t=GetTgl();
104 q*= TMath::Sqrt((1-s*s)/(1+t*t));
105 fdQdl[i]=q;
106 }
107
108 void SetSeedLabel(Int_t lab) { fSeedLab=lab; }
109
fd621f36 110 Int_t Update(const AliTRDcluster* c, Double_t chi2, UInt_t i,
b8dc2353 111 Double_t h01);
46e2d86c 112 Int_t UpdateMI(const AliTRDcluster* c, Double_t chi2, UInt_t i,
3c625a9b 113 Double_t h01, Int_t plane);
fd621f36 114
9c9d2487 115 //
116 void GetBarrelTrack(AliBarrelTrack *track);
117 void AddNWrong() {fNWrong++;}
118
119 Int_t GetNWrong() const {return fNWrong;}
120 Int_t GetNRotate() const {return fNRotate;}
16d9fbba 121 Int_t GetNCross() const {return fNCross;}
122 void IncCross() {fNCross++;}
123 AliTRDtrack * GetBackupTrack(){return fBackupTrack;}
124 void MakeBackupTrack();
9c9d2487 125 //
126
127
bbf92647 128protected:
129
5443e65e 130 Int_t fSeedLab; // track label taken from seeding
bbf92647 131 Float_t fdEdx; // dE/dx
eab5961e 132 Float_t fdEdxPlane[kNPlane]; // dE/dx from all 6 planes
133 Int_t fTimBinPlane[kNPlane]; // time bin of Max cluster from all 6 planes
bbf92647 134
135 Double_t fAlpha; // rotation angle
136 Double_t fX; // running local X-coordinate of the track (time bin)
3c625a9b 137 Bool_t fStopped; // track stop indication
b8dc2353 138
139 Double_t fY; // Y-coordinate of the track
140 Double_t fZ; // Z-coordinate of the track
141 Double_t fE; // C*x0
142 Double_t fT; // tangent of the track momentum dip angle
143 Double_t fC; // track curvature
bbf92647 144
145 Double_t fCyy; // covariance
146 Double_t fCzy, fCzz; // matrix
b3a5a838 147 Double_t fCey, fCez, fCee; // of the
148 Double_t fCty, fCtz, fCte, fCtt; // track
149 Double_t fCcy, fCcz, fCce, fCct, fCcc; // parameters
b8dc2353 150
5443e65e 151 UInt_t fIndex[kMAX_CLUSTERS_PER_TRACK]; // global indexes of clusters
46e2d86c 152 UInt_t fIndexBackup[kMAX_CLUSTERS_PER_TRACK]; //backup indexes of clusters - used in iterations
5443e65e 153 Float_t fdQdl[kMAX_CLUSTERS_PER_TRACK]; // cluster amplitudes corrected
154 // for track angles
b8dc2353 155
a819a5f7 156 Float_t fLhElectron; // Likelihood to be an electron
9c9d2487 157 Int_t fNWrong; // number of wrong clusters
16d9fbba 158 Int_t fNRotate; // number of rotation
159 Int_t fNCross; // number of the cross materials
160 AliTRDtrack * fBackupTrack; //! backup track
a819a5f7 161 ClassDef(AliTRDtrack,2) // TRD reconstructed tracks
a2b90f83 162
46d29e70 163};
164
165
46d29e70 166#endif