]> git.uio.no Git - u/mrichter/AliRoot.git/blob - TRD/AliTRDtrack.h
Protection against division by 0 in Binaries().
[u/mrichter/AliRoot.git] / TRD / AliTRDtrack.h
1 #ifndef ALITRDTRACK_H
2 #define ALITRDTRACK_H
3
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 #include <AliKalmanTrack.h>
8 #include <TMath.h>
9
10 #include "AliBarrelTrack.h"
11 #include "AliTRDgeometry.h"
12 #include "TVector2.h"
13
14 class AliTRDcluster;
15 class AliTPCtrack;
16
17 const unsigned kMAX_CLUSTERS_PER_TRACK=210; 
18
19 class AliTRDtrack : public AliKalmanTrack {
20
21 // Represents reconstructed TRD track
22
23 public:
24
25    AliTRDtrack():AliKalmanTrack(){}
26    AliTRDtrack(const AliTRDcluster *c, UInt_t index, const Double_t xx[5],
27                const Double_t cc[15], Double_t xr, Double_t alpha);  
28    AliTRDtrack(const AliTRDtrack& t);    
29    AliTRDtrack(const AliKalmanTrack& t, Double_t alpha); 
30
31    Int_t    Compare(const TObject *o) const;
32    void     CookdEdx(Double_t low=0.05, Double_t up=0.70);   
33
34    Double_t GetAlpha() const {return fAlpha;}
35    Int_t    GetSector() const {
36      //if (fabs(fAlpha) < AliTRDgeometry::GetAlpha()/2) return 0;
37      return Int_t(TVector2::Phi_0_2pi(fAlpha)/AliTRDgeometry::GetAlpha())%AliTRDgeometry::kNsect;}
38
39    Double_t GetC()     const {return fC;}
40    Int_t    GetClusterIndex(Int_t i) const {return fIndex[i];}    
41    Float_t  GetClusterdQdl(Int_t i) const {return fdQdl[i];}    
42
43    void     GetCovariance(Double_t cov[15]) const;  
44    Double_t GetdEdx()  const {return fdEdx;}
45    Double_t GetEta()   const {return fE;}
46
47    void     GetExternalCovariance(Double_t cov[15]) const ;   
48    void     GetExternalParameters(Double_t& xr, Double_t x[5]) const ;
49
50    Double_t GetLikelihoodElectron() const { return fLhElectron; };
51
52    Double_t Get1Pt()   const {return (1e-9*TMath::Abs(fC)/fC + fC)*GetConvConst(); } 
53    Double_t GetP()     const {  
54      return TMath::Abs(GetPt())*sqrt(1.+GetTgl()*GetTgl());
55    }
56    Double_t GetPredictedChi2(const AliTRDcluster*, Double_t h01) const ;
57    Double_t GetPt()    const {return 1./Get1Pt();}   
58    void     GetPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const ;
59    void     GetGlobalXYZ(Double_t &x, Double_t &y, Double_t &z) const ;
60    Int_t    GetSeedLabel() const { return fSeedLab; }
61    Double_t GetSigmaC2()   const {return fCcc;}
62    Double_t GetSigmaTgl2() const {return fCtt;}
63    Double_t GetSigmaY2()   const {return fCyy;}
64    Double_t GetSigmaZ2()   const {return fCzz;}
65    Double_t GetSnp()  const {return fX*fC - fE;}
66    Double_t GetTgl()  const {return fT;}
67    Double_t GetX()    const {return fX;}
68    Double_t GetY()    const {return fY;}
69    Double_t GetZ()    const {return fZ;}
70
71    Int_t    PropagateTo(Double_t xr, Double_t x0=8.72, Double_t rho=5.86e-3);
72    void     ResetCovariance();   
73    Int_t    Rotate(Double_t angle);
74
75    void     SetdEdx(Float_t dedx) {fdEdx=dedx;}  
76    void     SetLikelihoodElectron(Float_t l) { fLhElectron = l; };  
77
78    void     SetSampledEdx(Float_t q, Int_t i) {
79               Double_t s=GetSnp(), t=GetTgl();
80               q*= TMath::Sqrt((1-s*s)/(1+t*t));
81               fdQdl[i]=q;
82             }     
83
84    void     SetSeedLabel(Int_t lab) { fSeedLab=lab; }
85
86    Int_t    Update(const AliTRDcluster* c, Double_t chi2, UInt_t i, 
87                    Double_t h01);
88
89   //
90   void GetBarrelTrack(AliBarrelTrack *track);
91   void AddNWrong() {fNWrong++;}
92   
93   Int_t GetNWrong() const {return fNWrong;}
94   Int_t GetNRotate() const {return fNRotate;}
95   //
96
97
98 protected:
99
100    Int_t    fSeedLab;     // track label taken from seeding  
101    Float_t  fdEdx;        // dE/dx 
102
103    Double_t fAlpha;       // rotation angle
104    Double_t fX;           // running local X-coordinate of the track (time bin)
105
106
107    Double_t fY;             // Y-coordinate of the track
108    Double_t fZ;             // Z-coordinate of the track
109    Double_t fE;             // C*x0
110    Double_t fT;             // tangent of the track momentum dip angle
111    Double_t fC;             // track curvature
112
113    Double_t fCyy;                         // covariance
114    Double_t fCzy, fCzz;                   // matrix
115    Double_t fCey, fCez, fCee;             // of the
116    Double_t fCty, fCtz, fCte, fCtt;       // track
117    Double_t fCcy, fCcz, fCce, fCct, fCcc; // parameters   
118    
119    UInt_t  fIndex[kMAX_CLUSTERS_PER_TRACK];  // global indexes of clusters  
120    Float_t fdQdl[kMAX_CLUSTERS_PER_TRACK];   // cluster amplitudes corrected 
121                                              // for track angles    
122                            
123    Float_t fLhElectron;    // Likelihood to be an electron    
124    Int_t fNWrong;    // number of wrong clusters
125    Int_t fNRotate;
126
127    ClassDef(AliTRDtrack,2) // TRD reconstructed tracks
128
129 };                     
130
131
132 #endif