propagate cluster error parametrization
[u/mrichter/AliRoot.git] / TRD / AliTRDtrackV1.h
1 #ifndef ALITRDTRACKV1_H
2 #define ALITRDTRACKV1_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
8 ///////////////////////////////////////////////////////////////////////////////
9 //                                                                           //
10 //  Represents a reconstructed TRD track                                     //
11 //                                                                           //
12 ///////////////////////////////////////////////////////////////////////////////
13
14 #ifndef ALIKALMANTRACK_H
15 #include "AliKalmanTrack.h"
16 #endif
17
18 #ifndef ALIESDTRACK_H
19 #include "AliESDtrack.h"
20 #endif
21
22 #ifndef ALITRDSEEDV1_H
23 #include "AliTRDseedV1.h"
24 #endif
25
26 class AliTRDcluster;
27 class AliESDtrack;
28 class AliTRDReconstructor;
29 class AliTRDtrackV1 : public AliKalmanTrack
30 {
31 public:
32   enum { kMAXCLUSTERSPERTRACK = 210 };
33     
34   enum { kNdet      = 540
35         , kNstacks   =  90
36         , kNplane    =   AliESDtrack::kTRDnPlanes
37         , kNcham     =   5
38         , kNsect     =  18
39         , kNslice    =   3
40         , kNMLPslice =   8 };
41   
42   // bits from 0-13 are reserved by ROOT (see TObject.h)
43   enum AliTRDtrackStatus {
44           kOwner   = BIT(14)
45         , kStopped = BIT(15) 
46   };
47
48   AliTRDtrackV1();
49   AliTRDtrackV1(AliTRDseedV1 *trklts, const Double_t p[5], const Double_t cov[15], Double_t x, Double_t alpha);
50   AliTRDtrackV1(const AliESDtrack &ref);
51   AliTRDtrackV1(const AliTRDtrackV1 &ref);
52   virtual ~AliTRDtrackV1();
53   AliTRDtrackV1 &operator=(const AliTRDtrackV1 &ref) { *(new(this) AliTRDtrackV1(ref)); return *this; }
54   
55   Bool_t         CookPID();
56   Bool_t         CookLabel(Float_t wrong);
57   AliTRDtrackV1* GetBackupTrack() const {return fBackupTrack;}
58   Double_t       GetBudget(Int_t i) const { return fBudget[i];}
59   Double_t       GetC() const { return AliExternalTrackParam::GetC(GetBz());}
60   AliTRDcluster* GetCluster(Int_t id);
61   Int_t          GetClusterIndex(Int_t id) const;
62   Float_t        GetEdep() const {return fDE;}
63   inline Float_t GetMomentum(Int_t plane) const;
64   inline Int_t   GetNCross();
65   inline Int_t   GetNumberOfTracklets() const;
66   Double_t       GetPIDsignal() const   { return 0.;}
67   Double_t       GetPID(Int_t is) const { return (is >=0 && is < AliPID::kSPECIES) ? fPID[is] : -1.;}
68   UChar_t        GetPIDquality() const  { return fPIDquality;}
69   Double_t       GetPredictedChi2(const AliTRDseedV1 *tracklet) const;
70   Double_t       GetPredictedChi2(const AliCluster* /*c*/) const                   { return 0.0; }
71   Int_t          GetProlongation(Double_t xk, Double_t &y, Double_t &z);
72   AliTRDseedV1*  GetTracklet(Int_t plane) const {return plane >=0 && plane <kNplane ? fTracklet[plane] : 0x0;}
73   Int_t          GetTrackletIndex(Int_t plane) const          { return (plane>=0 && plane<kNplane) ? fTrackletIndex[plane] : -1;} 
74   UShort_t*      GetTrackletIndexes() {return &fTrackletIndex[0];}
75   
76   Bool_t         IsEqual(const TObject *inTrack) const;
77   Bool_t         IsOwner() const {return TestBit(kOwner);};
78   Bool_t         IsStopped() const {return TestBit(kStopped);};
79   Bool_t         IsElectron() const;
80   
81   void           MakeBackupTrack();
82   void           Print(Option_t *o="") const;
83
84   Bool_t         PropagateTo(Double_t xr, Double_t x0 = 8.72, Double_t rho = 5.86e-3);
85   Int_t          PropagateToR(Double_t xr, Double_t step);
86   Bool_t         Rotate(Double_t angle, Bool_t absolute = kFALSE);
87   void           SetBudget(Int_t i, Double_t b) {if(i>=0 && i<3) fBudget[i] = b;}
88   void           SetEdep(Double32_t inDE){fDE = inDE;};
89   void           SetNumberOfClusters();
90   void           SetOwner();
91   void           SetPID(Short_t is, Double_t inPID){if (is >=0 && is < AliPID::kSPECIES) fPID[is]=inPID;};
92   void           SetPIDquality(UChar_t inPIDquality){fPIDquality = inPIDquality;};
93   void           SetStopped(Bool_t stop) {SetBit(kStopped, stop);}
94   void           SetTracklet(AliTRDseedV1 *trklt,  Int_t index);
95   inline void    SetReconstructor(const AliTRDReconstructor *rec);
96   inline Float_t StatusForTOF();
97   void           UnsetTracklet(Int_t plane);
98   Bool_t         Update(AliTRDseedV1 *tracklet, Double_t chi2);
99   //Bool_t         Update(const AliTRDcluster *c, Double_t chi2, Int_t index, Double_t h01){ return AliTRDtrack::Update(c,chi2,index,h01); };
100   Bool_t         Update(const AliCluster *, Double_t, Int_t)                        { return kFALSE; };
101   void           UpdateESDtrack(AliESDtrack *t);
102
103 protected:
104   Double_t       GetBz() const;
105
106 private:
107   UChar_t      fPIDquality;           //  No of planes used for PID calculation 
108   UShort_t     fTrackletIndex[kNplane];//  Tracklets index in the tracker list
109   Double32_t   fPID[AliPID::kSPECIES];//  PID probabilities
110   Double32_t   fBudget[3];            //  Integrated material budget
111   Double32_t   fDE;                   //  Integrated delta energy
112   const AliTRDReconstructor *fReconstructor;//! reconstructor link 
113   AliTRDseedV1 *fTracklet[kNplane];   //  Tracklets array defining the track
114   AliTRDtrackV1 *fBackupTrack;        // Backup track
115
116
117   ClassDef(AliTRDtrackV1, 3)          // new TRD track
118 };
119
120 //____________________________________________________
121 inline Float_t AliTRDtrackV1::GetMomentum(Int_t plane) const
122 {
123   return plane >=0 && plane < kNplane && fTrackletIndex[plane] != 0xff ? fTracklet[plane]->GetMomentum() : -1.;
124 }
125
126 //____________________________________________________
127 inline Int_t AliTRDtrackV1::GetNCross()
128 {
129   Int_t ncross = 0;
130   for(Int_t ip=0; ip<kNplane; ip++){
131     if(!fTracklet[ip]) continue;
132     ncross += fTracklet[ip]->IsRowCross();
133   }
134   return ncross;
135 }
136
137 //____________________________________________________
138 inline Int_t AliTRDtrackV1::GetNumberOfTracklets() const
139 {
140   Int_t n = 0;
141   for(Int_t ip=0; ip<kNplane; ip++){
142     if(!fTracklet[ip]) continue;
143     n++;
144   }
145   return n;
146 }
147
148
149 //____________________________________________________
150 inline void AliTRDtrackV1::SetReconstructor(const AliTRDReconstructor *rec)
151 {
152   for(Int_t ip=0; ip<kNplane; ip++){
153     if(!fTracklet[ip]) continue;
154     fTracklet[ip]->SetReconstructor(rec);
155   }
156   fReconstructor = rec;
157 }
158
159
160 //____________________________________________________________________________
161 inline Float_t AliTRDtrackV1::StatusForTOF()
162 {
163   //
164   // Defines the status of the TOF extrapolation
165   //
166
167   if(!fTracklet[5]) return 0.;
168
169   // Definition of res ????
170   Float_t res = /*(0.2 + 0.8 * (fN / (fNExpected + 5.0))) **/ (0.4 + 0.6 * fTracklet[5]->GetN() / 20.0);
171   res *= (0.25 + 0.8 * 40.0 / (40.0 + fBudget[2]));
172   return res;
173 }
174
175 #endif
176
177
178