]>
Commit | Line | Data |
---|---|---|
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 ALITRDSEEDV1_H | |
19 | #include "AliTRDseedV1.h" | |
20 | #endif | |
21 | ||
22 | template <typename Value> class TVectorT; | |
23 | typedef class TVectorT<Double_t> TVectorD; | |
24 | class AliESDtrack; | |
25 | class AliTRDcluster; | |
26 | class AliTRDReconstructor; | |
27 | class AliTRDtrackV1 : public AliKalmanTrack | |
28 | { | |
29 | friend class AliHLTTRDTrack; // allow HLT special access | |
30 | public: | |
31 | enum ETRDtrackSize { | |
32 | kNdet = AliTRDgeometry::kNdet | |
33 | ,kNstacks = AliTRDgeometry::kNstack*AliTRDgeometry::kNsector | |
34 | ,kNplane = AliTRDgeometry::kNlayer | |
35 | ,kNcham = AliTRDgeometry::kNstack | |
36 | ,kNsect = AliTRDgeometry::kNsector | |
37 | ,kMAXCLUSTERSPERTRACK = 210 | |
38 | }; | |
39 | ||
40 | // bits from 0-13 are reserved by ROOT (see TObject.h) | |
41 | enum ETRDtrackStatus { | |
42 | kOwner = BIT(14) | |
43 | ,kStopped = BIT(15) | |
44 | ,kKink = BIT(16) | |
45 | ,kPrimary = BIT(17) | |
46 | ,kSeeder = BIT(18) // if set other than TPC (ITS, TRD) | |
47 | }; | |
48 | ||
49 | // propagation/update error codes (up to 4 bits) | |
50 | enum ETRDtrackError { | |
51 | kProlongation = 1 | |
52 | ,kPropagation | |
53 | ,kAdjustSector | |
54 | ,kSnp | |
55 | ,kTrackletInit | |
56 | ,kUpdate | |
57 | ,kUnknown = 0xff | |
58 | }; | |
59 | ||
60 | // data/clusters/tracklet error codes (up to 4 bits/layer) | |
61 | enum ETRDlayerError { | |
62 | kGeometry = 1 | |
63 | ,kBoundary | |
64 | ,kNoClusters | |
65 | ,kNoAttach | |
66 | ,kNoClustersTracklet | |
67 | ,kNoFit | |
68 | ,kChi2 | |
69 | }; | |
70 | ||
71 | AliTRDtrackV1(); | |
72 | AliTRDtrackV1(AliTRDseedV1 * const trklts, const Double_t p[5], const Double_t cov[15], Double_t x, Double_t alpha); | |
73 | AliTRDtrackV1(const AliESDtrack &ref); | |
74 | AliTRDtrackV1(const AliTRDtrackV1 &ref); | |
75 | virtual ~AliTRDtrackV1(); | |
76 | AliTRDtrackV1 &operator=(const AliTRDtrackV1 &ref); | |
77 | virtual void Copy(TObject &ref) const; | |
78 | virtual TObject* Clone(const char* newname = "") const; | |
79 | Bool_t CookPID(); | |
80 | Double_t CookTruncatedMean(const Bool_t kinvq, const Double_t mag, const Int_t charge, const Int_t kcalib, Int_t &nch, Int_t &ncls, TVectorD *Qs=NULL, TVectorD *Xs=NULL, Int_t timeBin0=-1, Int_t timeBin1=1000, Int_t tstep=1) const; | |
81 | ||
82 | Int_t CookLabel(Float_t wrong, Int_t *labs=NULL, Float_t *freq=NULL); | |
83 | AliTRDtrackV1* GetBackupTrack() const {return fBackupTrack;} | |
84 | Double_t GetBudget(Int_t i) const { return fBudget[i];} | |
85 | AliTRDcluster* GetCluster(Int_t id); | |
86 | Int_t GetClusterIndex(Int_t id) const; | |
87 | Float_t GetEdep() const {return fDE;} | |
88 | Int_t GetESDid() const {return fESDid;} | |
89 | inline Float_t GetMomentum(Int_t plane=-1) const; | |
90 | inline Int_t GetNCross(); | |
91 | inline Int_t GetNumberOfTracklets() const; | |
92 | Double_t GetPIDsignal() const { return 0.;} | |
93 | Double_t GetPID(Int_t is) const { return (is >=0 && is < AliPID::kSPECIES) ? fPID[is] : -1.;} | |
94 | UChar_t GetNumberOfTrackletsPID() const; | |
95 | Int_t GetNumberOfPhysicsSlices() const { return fNdEdxSlices; }; | |
96 | Double_t GetPredictedChi2(const AliTRDseedV1 *tracklet, Double_t *cov) const; | |
97 | Double_t GetPredictedChi2(const AliCluster* /*c*/) const { return 0.0; } | |
98 | Int_t GetProlongation(Double_t xk, Double_t &y, Double_t &z) const; | |
99 | inline UChar_t GetStatusTRD(Int_t ly=-1) const; | |
100 | Int_t GetSector() const; | |
101 | AliTRDseedV1* GetTracklet(Int_t plane) const {return plane >=0 && plane <kNplane ? fTracklet[plane] : NULL;} | |
102 | Int_t GetTrackletIndex(Int_t plane) const { return (plane>=0 && plane<kNplane) ? fTrackletIndex[plane] : -1;} | |
103 | AliExternalTrackParam* | |
104 | GetTrackIn() const { return fTrackLow;} | |
105 | AliExternalTrackParam* | |
106 | GetTrackOut() const { return fTrackHigh;} | |
107 | const Int_t* GetTrackletIndexes() const { return &fTrackletIndex[0];} | |
108 | Bool_t IsEqual(const TObject *inTrack) const; | |
109 | Bool_t IsKink() const { return TestBit(kKink);} | |
110 | Bool_t IsOwner() const { return TestBit(kOwner);}; | |
111 | Bool_t IsPrimary() const { return TestBit(kPrimary);}; | |
112 | Bool_t IsStopped() const { return TestBit(kStopped);}; | |
113 | Bool_t IsElectron() const; | |
114 | Bool_t IsTPCseeded() const { return !TestBit(kSeeder);}; | |
115 | inline static Bool_t IsTrackError(ETRDtrackError error, UInt_t status); | |
116 | inline static Bool_t IsLayerError(ETRDlayerError error, Int_t layer, UInt_t status); | |
117 | ||
118 | Int_t MakeBackupTrack(); | |
119 | void Print(Option_t *o="") const; | |
120 | ||
121 | Bool_t PropagateTo(Double_t xr, Double_t x0 = 8.72, Double_t rho = 5.86e-3); | |
122 | Int_t PropagateToR(Double_t xr, Double_t step); | |
123 | Bool_t Rotate(Double_t angle, Bool_t absolute = kFALSE); | |
124 | void SetBudget(Int_t i, Double_t b) {if(i>=0 && i<3) fBudget[i] = b;} | |
125 | void SetEdep(Double32_t inDE){fDE = inDE;}; | |
126 | void SetESDid(Int_t id) {fESDid = id;} | |
127 | void SetKink(Bool_t k) { SetBit(kKink, k);} | |
128 | void SetPrimary(Bool_t k) { SetBit(kPrimary, k);} | |
129 | void SetNonTPCseeded(Bool_t k){ SetBit(kSeeder, k);} | |
130 | void SetNumberOfClusters(); | |
131 | void SetOwner(); | |
132 | void SetPID(Short_t is, Double_t inPID){if (is >=0 && is < AliPID::kSPECIES) fPID[is]=inPID;}; | |
133 | void SetPIDquality(UChar_t /*inPIDquality*/) const {/*fPIDquality = inPIDquality*/;}; | |
134 | inline void SetErrStat(UChar_t stat, Int_t ly=-1); | |
135 | void SetStopped(Bool_t stop) {SetBit(kStopped, stop);} | |
136 | void SetTracklet(AliTRDseedV1 *const trklt, Int_t index); | |
137 | void SetTrackIn(); | |
138 | void SetTrackOut(const AliExternalTrackParam *op=NULL); | |
139 | inline void SetReconstructor(const AliTRDReconstructor *rec); | |
140 | inline Float_t StatusForTOF(); | |
141 | void UnsetTracklet(Int_t plane); | |
142 | Bool_t Update(const AliCluster *, Double_t, Int_t) { return kFALSE; }; | |
143 | void UpdateChi2(Float_t chi2); | |
144 | void UpdateESDtrack(AliESDtrack *t); | |
145 | ||
146 | private: | |
147 | UInt_t fStatus; // Bit map for the status of propagation | |
148 | Int_t fTrackletIndex[kNplane];// Tracklets index in the tracker list | |
149 | Int_t fESDid; // ESD track id | |
150 | Double32_t fPID[AliPID::kSPECIES]; // PID probabilities | |
151 | Double32_t fBudget[3]; // Integrated material budget | |
152 | Double32_t fDE; // Integrated delta energy | |
153 | Double32_t fTruncatedMean; // Truncated mean | |
154 | Int_t fNchamberdEdx; // number of chambers used in calculating truncated mean | |
155 | Int_t fNclusterdEdx; // number of clusters used in calculating truncated mean | |
156 | Int_t fNdEdxSlices; // number of "physics slices" fill via AliTRDPIDResponse | |
157 | ||
158 | const AliTRDReconstructor *fkReconstructor;//! reconstructor link | |
159 | AliTRDtrackV1 *fBackupTrack; //! Backup track | |
160 | AliTRDseedV1 *fTracklet[kNplane]; // Tracklets array defining the track | |
161 | AliExternalTrackParam *fTrackLow; // parameters of the track which enter TRD from below (TPC) | |
162 | AliExternalTrackParam *fTrackHigh; // parameters of the track which enter TRD from above (HMPID, PHOS) | |
163 | ||
164 | ClassDef(AliTRDtrackV1, 8) // TRD track - tracklet based | |
165 | }; | |
166 | ||
167 | //____________________________________________________ | |
168 | inline Float_t AliTRDtrackV1::GetMomentum(Int_t plane) const | |
169 | { | |
170 | // Return ESD momentum stored in the tracklet reconstructed in layer = "plane". | |
171 | // By default returns the ESD momentum in first tracklet attached to track | |
172 | if(plane==-1){ | |
173 | for(Int_t i(0); i<kNplane; i++){ | |
174 | if(fTracklet[i]) return fTracklet[i]->GetMomentum(); | |
175 | } | |
176 | } else if( plane >=0 && plane < kNplane){ | |
177 | if(fTracklet[plane]) return fTracklet[plane]->GetMomentum(); | |
178 | } | |
179 | return -1.; | |
180 | } | |
181 | ||
182 | //____________________________________________________ | |
183 | inline Int_t AliTRDtrackV1::GetNCross() | |
184 | { | |
185 | Int_t ncross = 0; | |
186 | for(Int_t ip=0; ip<kNplane; ip++){ | |
187 | if(!fTracklet[ip]) continue; | |
188 | ncross += fTracklet[ip]->IsRowCross(); | |
189 | } | |
190 | return ncross; | |
191 | } | |
192 | ||
193 | //____________________________________________________ | |
194 | inline Int_t AliTRDtrackV1::GetNumberOfTracklets() const | |
195 | { | |
196 | Int_t n = 0; | |
197 | for(Int_t ip=0; ip<kNplane; ip++){ | |
198 | if(!fTracklet[ip]) continue; | |
199 | n++; | |
200 | } | |
201 | return n; | |
202 | } | |
203 | ||
204 | //____________________________________________________ | |
205 | inline UChar_t AliTRDtrackV1::GetStatusTRD(Int_t ly) const | |
206 | { | |
207 | if(ly>=-1 && ly<kNplane) return (fStatus>>((ly+1)*4))&0xf; | |
208 | return kUnknown; | |
209 | } | |
210 | ||
211 | //____________________________________________________ | |
212 | inline Bool_t AliTRDtrackV1::IsTrackError(ETRDtrackError error, UInt_t status) | |
213 | { | |
214 | return (status&0xf)==UChar_t(error); | |
215 | } | |
216 | ||
217 | //____________________________________________________ | |
218 | inline Bool_t AliTRDtrackV1::IsLayerError(ETRDlayerError error, Int_t ly, UInt_t status) | |
219 | { | |
220 | if(ly>=kNplane || ly<0) return kFALSE; | |
221 | return ((status>>((ly+1)*4))&0xf) == UChar_t(error); | |
222 | } | |
223 | ||
224 | //____________________________________________________ | |
225 | inline void AliTRDtrackV1::SetReconstructor(const AliTRDReconstructor *rec) | |
226 | { | |
227 | for(Int_t ip=0; ip<kNplane; ip++){ | |
228 | if(!fTracklet[ip]) continue; | |
229 | fTracklet[ip]->SetReconstructor(rec); | |
230 | } | |
231 | fkReconstructor = rec; | |
232 | } | |
233 | ||
234 | //____________________________________________________ | |
235 | inline void AliTRDtrackV1::SetErrStat(UChar_t status, Int_t ly) | |
236 | { | |
237 | if(ly<kNplane) fStatus|=((status&0xf)<<((ly+1)*4)); | |
238 | return; | |
239 | } | |
240 | ||
241 | ||
242 | //____________________________________________________________________________ | |
243 | inline Float_t AliTRDtrackV1::StatusForTOF() | |
244 | { | |
245 | // OBSOLETE | |
246 | // Defines the status of the TOF extrapolation | |
247 | // | |
248 | ||
249 | if(!fTracklet[5]) return 0.; | |
250 | ||
251 | // Definition of res ???? | |
252 | Float_t res = /*(0.2 + 0.8 * (fN / (fNExpected + 5.0))) **/ (0.4 + 0.6 * fTracklet[5]->GetN() / 20.0); | |
253 | res *= (0.25 + 0.8 * 40.0 / (40.0 + fBudget[2])); | |
254 | return res; | |
255 | } | |
256 | ||
257 | #endif | |
258 | ||
259 | ||
260 |