]> git.uio.no Git - u/mrichter/AliRoot.git/blob - TRD/AliTRDtrack.h
Rename macros
[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 /* $Id$ */
8
9 ///////////////////////////////////////////////////////////////////////////////
10 //                                                                           //
11 //  Represents a reconstructed TRD track                                     //
12 //                                                                           //
13 ///////////////////////////////////////////////////////////////////////////////
14
15 #include "AliKalmanTrack.h"
16
17 #include "AliTRDtracklet.h"
18
19 class AliESDtrack;
20 class AliTrackReference;
21
22 const unsigned kMAXCLUSTERSPERTRACK = 210; 
23
24 class AliTRDtrack : public AliKalmanTrack {
25    
26   enum { kNdet      = 540
27        , kNstacks   =  90
28        , kNplane    =   6
29        , kNcham     =   5
30        , kNsect     =  18
31        , kNslice    =   3
32        , kNtimeBins =  22 };
33
34   friend class AliTRDtracker;
35
36  public:
37
38    AliTRDtrack();
39    AliTRDtrack(const AliTRDcluster *c, Int_t index, const Double_t xx[5]
40               ,const Double_t cc[15], Double_t xr, Double_t alpha);  
41    AliTRDtrack(const AliTRDtrack &t);    
42    AliTRDtrack(const AliESDtrack &t);
43    ~AliTRDtrack();
44    AliTRDtrack(const AliKalmanTrack &t, Double_t alpha); 
45
46            void     ResetClusters()                                         { SetChi2(0.0); 
47                                                                               SetNumberOfClusters(0);              }
48            Int_t    Compare(const TObject *o) const;
49            void     AddNWrong()                                             { fNWrong++;                           }  
50            void     IncCross()                                              { fNCross++; 
51                                                                               if (fBackupTrack) fBackupTrack->IncCross();  }
52
53            Int_t    GetSector() const;
54            Float_t  GetClusterdQdl(Int_t i) const                           { return fdQdl[i];                     }    
55            Double_t GetdEdx() const                                         { return fdEdx;                        }
56            Int_t    GetNdedx() const                                        { return fNdedx;                       }
57            Double_t GetPIDsignal() const                                    { return GetdEdx();                    }
58            Int_t    GetClusterIndex(Int_t i) const                          { return fIndex[i];                    }
59            Double_t GetC() const                                            { return AliExternalTrackParam::GetC(GetBz()); }
60            Double_t GetPredictedChi2(const AliTRDcluster *c, Double_t h01) const;
61            Float_t  GetPIDsignals(Int_t iPlane, Int_t iSlice) const         { return fdEdxPlane[iPlane][iSlice];   }
62            Int_t    GetPIDTimBin(Int_t i) const                             { return fTimBinPlane[i];              }
63            Double_t GetLikelihoodElectron() const                           { return fLhElectron;                  }
64            Int_t    GetSeedLabel() const                                    { return fSeedLab;                     }
65            Int_t   *GetBackupIndexes()                                      { return fIndexBackup;                 }
66            Int_t   *GetIndexes()                                            { return fIndex;                       }
67            Int_t    GetProlongation(Double_t xk, Double_t &y, Double_t &z);
68            Bool_t   GetStop() const                                         { return fStopped;                     }
69            Int_t    GetNWrong() const                                       { return fNWrong;                      }
70            Int_t    GetNRotate() const                                      { return fNRotate;                     }
71            Int_t    GetNCross() const                                       { return fNCross;                      }
72   AliTRDtrack      *GetBackupTrack()                                        { return fBackupTrack;                 }
73
74            void     SetdEdx(Double_t dedx)                                  { fdEdx                      = dedx;   }
75            void     SetStop(Bool_t stop)                                    { fStopped                   = stop;   }
76            void     SetPIDsignals(Float_t dedx, Int_t iPlane, Int_t iSlice) { fdEdxPlane[iPlane][iSlice] = dedx;   }
77            void     SetPIDTimBin(Int_t timbin, Int_t i)                     { fTimBinPlane[i]            = timbin; }
78            void     SetLikelihoodElectron(Float_t l)                        { fLhElectron                = l;      }
79            void     SetSampledEdx(Float_t q, Int_t i);
80            void     SetSampledEdx(Float_t q);
81            void     SetSeedLabel(Int_t lab)                                 { fSeedLab                   = lab;    }
82
83            Int_t    PropagateToX(Double_t xr, Double_t step);
84            Int_t    PropagateToR(Double_t xr, Double_t step);
85            Int_t    UpdateMI(const AliTRDcluster *c, Double_t chi2, Int_t i, Double_t h01, Int_t plane);
86            void     MakeBackupTrack();
87            Bool_t   PropagateTo(Double_t xr, Double_t x0 = 8.72, Double_t rho = 5.86e-3);
88            Bool_t   Update(const AliTRDcluster *c, Double_t chi2, Int_t i, Double_t h01);
89            Bool_t   Rotate(Double_t angle, Bool_t absolute = kFALSE);
90            void     CookdEdx(Double_t low = 0.05, Double_t up = 0.7);   
91            Float_t  StatusForTOF();
92  
93  protected:
94
95   AliTRDtrack      &operator=(const AliTRDtrack &t);
96
97            Double_t GetBz() const;
98            Bool_t   Update(const AliCluster */*c*/, Double_t /*chi2*/, Int_t /*idx*/) { return 0;   }
99            Double_t GetPredictedChi2(const AliCluster* /*c*/) const                   { return 0.0; }
100
101            Int_t    fSeedLab;                           //  Track label taken from seeding  
102            Float_t  fdEdx;                              //  dE/dx (truncated mean)
103            Float_t  fDE;                                //  Integrated delta energy
104            Float_t  fdEdxPlane[kNplane][kNslice];       //  dE/dx from all 6 planes in 3 slices each
105            Int_t    fTimBinPlane[kNplane];              //  Time bin of Max cluster from all 6 planes
106
107            Bool_t   fStopped;                           //  Track stop indication
108            Int_t    fIndex[kMAXCLUSTERSPERTRACK];       //  Global indexes of clusters  
109            Int_t    fIndexBackup[kMAXCLUSTERSPERTRACK]; //  Backup indexes of clusters - used in iterations
110            Float_t  fdQdl[kMAXCLUSTERSPERTRACK];        //  Cluster amplitudes corrected for track angles    
111                            
112            Float_t  fLhElectron;                        //  Likelihood to be an electron    
113            Int_t    fNWrong;                            //  Number of wrong clusters
114            Int_t    fNRotate;                           //  Number of rotation
115            Int_t    fNCross;                            //  Number of the cross materials
116            Int_t    fNExpected;                         //  Expected number of cluster
117            Int_t    fNLast;                             //  Number of clusters in last 2 layers
118            Int_t    fNExpectedLast;                     //  Number of expected clusters on last 2 layers
119            Int_t    fNdedx;                             //  Number of clusters for dEdx measurment
120            Float_t  fChi2Last;                          //  Chi2 in the  last 2 layers
121    AliTRDtracklet   fTracklets[6];                      //  Tracklets
122            Float_t  fBudget[3];                         //  Integrated material budget
123    AliTRDtrack     *fBackupTrack;                       //! Backup track
124
125    ClassDef(AliTRDtrack,8)                              //  TRD reconstructed tracks
126
127 };                     
128
129 #endif