]> git.uio.no Git - u/mrichter/AliRoot.git/blame - TRD/AliTRDseedV1.h
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
modify "Add" macro names accordingly
[u/mrichter/AliRoot.git] / TRD / AliTRDseedV1.h
CommitLineData
e4f2f73d 1#ifndef ALITRDSEEDV1_H
2#define ALITRDSEEDV1_H
3/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
e44586fb 4* See cxx source for full Copyright notice */
e4f2f73d 5
6/* $Id$ */
7
8////////////////////////////////////////////////////////////////////////////
9// //
3e778975 10// The TRD offline tracklet //
e4f2f73d 11// //
12////////////////////////////////////////////////////////////////////////////
13
3e778975 14#ifndef ALITRDTRACKLETBASE_H
15#include "AliTRDtrackletBase.h"
e3cf3d02 16#endif
17
18#ifndef ROOT_TMath
19#include "TMath.h"
e4f2f73d 20#endif
21
ae4e8b84 22#ifndef ALITRDGEOMETRY_H
23#include "AliTRDgeometry.h"
24#endif
25
0906e73e 26#ifndef ALIPID_H
27#include "AliPID.h"
28#endif
29
e4f2f73d 30#ifndef ALIRIEMAN_H
31#include "AliRieman.h"
32#endif
33
f29f13a6 34#ifndef ALITRDCLUSTER_H
35#include "AliTRDcluster.h"
36#endif
37
e4f2f73d 38class TTreeSRedirector;
39
40class AliRieman;
41
eb38ed55 42class AliTRDtrackingChamber;
f3d3af1b 43class AliTRDtrackV1;
3a039a31 44class AliTRDReconstructor;
eb2b4f91 45class AliTRDpadPlane;
3e778975 46class AliTRDseedV1 : public AliTRDtrackletBase
47{
e3cf3d02 48public:
49 enum ETRDtrackletBuffers {
8d2bec9e 50 kNtb = 31 // max clusters/pad row
51 ,kNclusters = 2*kNtb // max number of clusters/tracklet
52 ,kNslices = 10 // max dEdx slices
e44586fb 53 };
e3cf3d02 54
2e2915e7 55 // bits from 0-13 are reserved by ROOT (see TObject.h)
e3cf3d02 56 enum ETRDtrackletStatus {
f29f13a6 57 kOwner = BIT(14) // owner of its clusters
58 ,kRowCross = BIT(15) // pad row cross tracklet
59 ,kCalib = BIT(16) // calibrated tracklet
60 ,kKink = BIT(17) // kink prolongation tracklet
61 ,kStandAlone = BIT(18)
e44586fb 62 };
63
ae4e8b84 64 AliTRDseedV1(Int_t det = -1);
e44586fb 65 ~AliTRDseedV1();
66 AliTRDseedV1(const AliTRDseedV1 &ref);
67 AliTRDseedV1& operator=(const AliTRDseedV1 &ref);
68
f29f13a6 69/* Bool_t AttachClustersIter(
203967fc 70 AliTRDtrackingChamber *chamber, Float_t quality,
f29f13a6 71 Bool_t kZcorr = kFALSE, AliTRDcluster *c=0x0);*/
203967fc 72 Bool_t AttachClusters(
b1957d3c 73 AliTRDtrackingChamber *chamber, Bool_t tilt = kFALSE);
203967fc 74 void Bootstrap(const AliTRDReconstructor *rec);
e3cf3d02 75 void Calibrate();
203967fc 76 void CookdEdx(Int_t nslices);
e3cf3d02 77 void CookLabels();
3e778975 78 Bool_t CookPID();
b72f4eaf 79 Bool_t Fit(Bool_t tilt=kFALSE, Bool_t zcorr=kFALSE);
203967fc 80 Bool_t Init(AliTRDtrackV1 *track);
e44586fb 81 inline void Init(const AliRieman *fit);
203967fc 82 Bool_t IsEqual(const TObject *inTracklet) const;
e3cf3d02 83 Bool_t IsCalibrated() const { return TestBit(kCalib);}
e44586fb 84 Bool_t IsOwner() const { return TestBit(kOwner);}
f29f13a6 85 Bool_t IsKink() const { return TestBit(kKink);}
3e778975 86 Bool_t IsOK() const { return GetN() > 4 && GetNUsed() < 4;}
e44586fb 87 Bool_t IsRowCross() const { return TestBit(kRowCross);}
f29f13a6 88 Bool_t IsUsable(Int_t i) const { return fClusters[i] && !fClusters[i]->IsUsed();}
89 Bool_t IsStandAlone() const { return TestBit(kStandAlone);}
e44586fb 90
e3cf3d02 91 Float_t GetC() const { return fC; }
92 Float_t GetChi2() const { return fChi2; }
93 inline Float_t GetChi2Z() const;
94 inline Float_t GetChi2Y() const;
f29f13a6 95 inline Float_t GetChi2Phi() const;
e44586fb 96 void GetCovAt(Double_t x, Double_t *cov) const;
d937ad7a 97 void GetCovXY(Double_t *cov) const { memcpy(cov, &fCov[0], 3*sizeof(Double_t));}
bb2db46c 98 void GetCovRef(Double_t *cov) const { memcpy(cov, &fRefCov, 3*sizeof(Double_t));}
99 static Double_t GetCovSqrt(Double_t *c, Double_t *d);
100 static Double_t GetCovInv(Double_t *c, Double_t *d);
203967fc 101 Float_t GetdX() const { return fdX;}
102 Float_t* GetdEdx() { return &fdEdx[0];}
e44586fb 103 Float_t GetdQdl(Int_t ic) const;
3e778975 104 Float_t GetdYdX() const { return fYfit[1]; }
105 Float_t GetdZdX() const { return fZref[1]; }
106 Int_t GetdY() const { return Int_t(GetY()/0.014);}
203967fc 107 Int_t GetDetector() const { return fDet;}
e3cf3d02 108 void GetCalibParam(Float_t &exb, Float_t &vd, Float_t &t0, Float_t &s2, Float_t &dl, Float_t &dt) const {
109 exb = fExB; vd = fVD; t0 = fT0; s2 = fS2PRF; dl = fDiffL; dt = fDiffT;}
8d2bec9e 110 AliTRDcluster* GetClusters(Int_t i) const { return i<0 || i>=kNclusters ? 0x0 : fClusters[i];}
111 Int_t GetIndexes(Int_t i) const{ return i<0 || i>=kNclusters ? -1 : fIndexes[i];}
e3cf3d02 112 Int_t GetLabels(Int_t i) const { return fLabels[i];}
b25a5e9e 113 Float_t GetMomentum() const { return fPt*TMath::Sqrt(1.+fZref[1]*fZref[1]);}
3e778975 114 Int_t GetN() const { return (Int_t)fN&0x1f;}
115 Int_t GetN2() const { return GetN();}
116 Int_t GetNUsed() const { return Int_t((fN>>5)&0x1f);}
117 Int_t GetNShared() const { return Int_t((fN>>10)&0x1f);}
e44586fb 118 Float_t GetQuality(Bool_t kZcorr) const;
dd8059a8 119 Float_t GetPadLength() const { return fPad[0];}
120 Float_t GetPadWidth() const { return fPad[1];}
ae4e8b84 121 Int_t GetPlane() const { return AliTRDgeometry::GetLayer(fDet); }
122
3e778975 123 Float_t* GetProbability(Bool_t force=kFALSE);
b25a5e9e 124 Float_t GetPt() const { return fPt; }
3e778975 125 inline Double_t GetPID(Int_t is=-1) const;
e3cf3d02 126 Float_t GetS2Y() const { return fS2Y;}
127 Float_t GetS2Z() const { return fS2Z;}
128 Float_t GetSigmaY() const { return fS2Y > 0. ? TMath::Sqrt(fS2Y) : 0.2;}
129 Float_t GetSnp() const { return fYref[1]/TMath::Sqrt(1+fYref[1]*fYref[1]);}
130 Float_t GetTgl() const { return fZref[1];}
dd8059a8 131 Float_t GetTilt() const { return fPad[2];}
3e778975 132 UInt_t GetTrackletWord() const { return 0;}
b72f4eaf 133 UShort_t GetVolumeId() const;
e3cf3d02 134 Float_t GetX0() const { return fX0;}
135 Float_t GetX() const { return fX0 - fX;}
5a7a515d 136 Float_t GetY() const { return fYfit[0] - fYfit[1] * fX;}
b1957d3c 137 Double_t GetYat(Double_t x) const { return fYfit[0] - fYfit[1] * (fX0-x);}
e3cf3d02 138 Float_t GetYfit(Int_t id) const { return fYfit[id];}
139 Float_t GetYref(Int_t id) const { return fYref[id];}
3658afdc 140 Float_t GetZ() const { return fZfit[0] - fZfit[1] * fX;}
b1957d3c 141 Double_t GetZat(Double_t x) const { return fZfit[0] - fZfit[1] * (fX0-x);}
e3cf3d02 142 Float_t GetZfit(Int_t id) const { return fZfit[id];}
143 Float_t GetZref(Int_t id) const { return fZref[id];}
3e778975 144 Int_t GetYbin() const { return Int_t(GetY()/0.016);}
dd8059a8 145 Int_t GetZbin() const { return Int_t(GetZ()/fPad[0]);}
e3cf3d02 146
ae4e8b84 147 inline AliTRDcluster* NextCluster();
71ea19a3 148 inline AliTRDcluster* PrevCluster();
e44586fb 149 void Print(Option_t *o = "") const;
71ea19a3 150 inline void ResetClusterIter(Bool_t forward = kTRUE);
e3cf3d02 151 void Reset();
ae4e8b84 152
e3cf3d02 153 void SetC(Float_t c) { fC = c;}
154 void SetChi2(Float_t chi2) { fChi2 = chi2;}
0849f128 155 void SetCovRef(const Double_t *cov) { memcpy(&fRefCov[0], cov, 3*sizeof(Double_t));}
e3cf3d02 156 void SetIndexes(Int_t i, Int_t idx) { fIndexes[i] = idx; }
157 void SetLabels(Int_t *lbls) { memcpy(fLabels, lbls, 3*sizeof(Int_t)); }
f29f13a6 158 void SetKink(Bool_t k) { SetBit(kKink, k);}
159 void SetStandAlone(Bool_t st) { SetBit(kStandAlone, st); }
b25a5e9e 160 void SetPt(Double_t pt) { fPt = pt;}
29b87567 161 void SetOwner();
eb2b4f91 162 void SetPadPlane(AliTRDpadPlane *p);
dd8059a8 163 void SetPadLength(Float_t l) { fPad[0] = l;}
164 void SetPadWidth(Float_t w) { fPad[1] = w;}
cbe97468 165 void SetTilt(Float_t tilt) { fPad[2] = tilt; }
203967fc 166 void SetDetector(Int_t d) { fDet = d; }
bee2b41e 167 void SetDX(Float_t inDX) { fdX = inDX;}
3a039a31 168 void SetReconstructor(const AliTRDReconstructor *rec) {fReconstructor = rec;}
e3cf3d02 169 void SetX0(Float_t x0) { fX0 = x0; }
170 void SetYref(Int_t i, Float_t y) { fYref[i] = y;}
171 void SetZref(Int_t i, Float_t z) { fZref[i] = z;}
f29f13a6 172// void SetUsabilityMap(Long_t um) { fUsable = um; }
b1957d3c 173 void UpDate(const AliTRDtrackV1* trk);
e3cf3d02 174 void UpdateUsed();
175 void UseClusters();
e4f2f73d 176
d937ad7a 177protected:
3e778975 178 void Copy(TObject &ref) const;
e4f2f73d 179
e44586fb 180private:
3e778975 181 inline void SetN(Int_t n);
182 inline void SetNUsed(Int_t n);
183 inline void SetNShared(Int_t n);
184
0ae89c5d 185 const AliTRDReconstructor *fReconstructor;//! local reconstructor
e3cf3d02 186 AliTRDcluster **fClusterIter; //! clusters iterator
8d2bec9e 187 Int_t fIndexes[kNclusters]; //! Indexes
e3cf3d02 188 Float_t fExB; //! tg(a_L) @ tracklet location
189 Float_t fVD; //! drift velocity @ tracklet location
190 Float_t fT0; //! time 0 @ tracklet location
191 Float_t fS2PRF; //! sigma^2 PRF for xd->0 and phi=a_L
192 Float_t fDiffL; //! longitudinal diffusion coefficient
193 Float_t fDiffT; //! transversal diffusion coefficient
71ea19a3 194 Char_t fClusterIdx; //! clusters iterator
dd8059a8 195 UShort_t fN; // number of clusters attached/used/shared
e3cf3d02 196 Short_t fDet; // TRD detector
8d2bec9e 197 AliTRDcluster *fClusters[kNclusters]; // Clusters
dd8059a8 198 Float_t fPad[3]; // local pad definition : length/width/tilt
e3cf3d02 199 Float_t fYref[2]; // Reference y
200 Float_t fZref[2]; // Reference z
201 Float_t fYfit[2]; // Y fit position +derivation
202 Float_t fZfit[2]; // Z fit position
b25a5e9e 203 Float_t fPt; // Momentum estimate @ tracklet [GeV/c]
e44586fb 204 Float_t fdX; // length of time bin
e3cf3d02 205 Float_t fX0; // anode wire position
206 Float_t fX; // radial position of the tracklet
207 Float_t fY; // r-phi position of the tracklet
208 Float_t fZ; // z position of the tracklet
209 Float_t fS2Y; // estimated resolution in the r-phi direction
210 Float_t fS2Z; // estimated resolution in the z direction
211 Float_t fC; // Curvature
212 Float_t fChi2; // Global chi2
8d2bec9e 213 Float_t fdEdx[kNslices]; // dE/dx measurements for tracklet
e3cf3d02 214 Float_t fProb[AliPID::kSPECIES]; // PID probabilities
215 Int_t fLabels[3]; // most frequent MC labels and total number of different labels
6e4d4425 216 Double_t fRefCov[3]; // covariance matrix of the track in the yz plane
d937ad7a 217 Double_t fCov[3]; // covariance matrix of the tracklet in the xy plane
e4f2f73d 218
3e778975 219 ClassDef(AliTRDseedV1, 6) // The offline TRD tracklet
e4f2f73d 220};
221
222//____________________________________________________________
e3cf3d02 223inline Float_t AliTRDseedV1::GetChi2Z() const
e4f2f73d 224{
e3cf3d02 225 Double_t dz = fZref[0]-fZfit[0]; dz*=dz;
226 Double_t cov[3]; GetCovAt(fX, cov);
227 Double_t s2 = fRefCov[2]+cov[2];
228 return s2 > 0. ? dz/s2 : 0.;
e4f2f73d 229}
230
231//____________________________________________________________
e3cf3d02 232inline Float_t AliTRDseedV1::GetChi2Y() const
e4f2f73d 233{
e3cf3d02 234 Double_t dy = fYref[0]-fYfit[0]; dy*=dy;
235 Double_t cov[3]; GetCovAt(fX, cov);
236 Double_t s2 = fRefCov[0]+cov[0];
237 return s2 > 0. ? dy/s2 : 0.;
e4f2f73d 238}
239
f29f13a6 240//____________________________________________________________
241inline Float_t AliTRDseedV1::GetChi2Phi() const
242{
243 Double_t dphi = fYref[1]-fYfit[1]; dphi*=dphi;
244 Double_t cov[3]; GetCovAt(fX, cov);
245 Double_t s2 = fRefCov[2]+cov[2];
246 return s2 > 0. ? dphi/s2 : 0.;
247}
248
3e778975 249//____________________________________________________________
250inline Double_t AliTRDseedV1::GetPID(Int_t is) const
251{
252 if(is<0) return fProb[AliPID::kElectron];
253 if(is<AliPID::kSPECIES) return fProb[is];
254 return 0.;
255}
256
e4f2f73d 257//____________________________________________________________
0906e73e 258inline void AliTRDseedV1::Init(const AliRieman *rieman)
e4f2f73d 259{
e44586fb 260 fZref[0] = rieman->GetZat(fX0);
261 fZref[1] = rieman->GetDZat(fX0);
262 fYref[0] = rieman->GetYat(fX0);
263 fYref[1] = rieman->GetDYat(fX0);
55ef6967 264 fC = rieman->GetC();
265 fChi2 = rieman->GetChi2();
e4f2f73d 266}
267
ae4e8b84 268//____________________________________________________________
269inline AliTRDcluster* AliTRDseedV1::NextCluster()
270{
71ea19a3 271// Mimic the usage of STL iterators.
272// Forward iterator
273
ae4e8b84 274 fClusterIdx++; fClusterIter++;
8d2bec9e 275 while(fClusterIdx < kNclusters){
71ea19a3 276 if(!(*fClusterIter)){
277 fClusterIdx++;
278 fClusterIter++;
279 continue;
280 }
281 return *fClusterIter;
282 }
283 return 0x0;
284}
285
286//____________________________________________________________
287inline AliTRDcluster* AliTRDseedV1::PrevCluster()
288{
289// Mimic the usage of STL iterators.
290// Backward iterator
291
292 fClusterIdx--; fClusterIter--;
293 while(fClusterIdx >= 0){
294 if(!(*fClusterIter)){
295 fClusterIdx--;
296 fClusterIter--;
297 continue;
298 }
299 return *fClusterIter;
300 }
301 return 0x0;
302}
303
304//____________________________________________________________
305inline void AliTRDseedV1::ResetClusterIter(Bool_t forward)
306{
307// Mimic the usage of STL iterators.
308// Facilitate the usage of NextCluster for forward like
309// iterator (kTRUE) and PrevCluster for backward like iterator (kFALSE)
310
311 if(forward){
312 fClusterIter = &fClusters[0]; fClusterIter--;
313 fClusterIdx=-1;
314 } else {
8d2bec9e 315 fClusterIter = &fClusters[kNclusters-1]; fClusterIter++;
316 fClusterIdx=kNclusters;
71ea19a3 317 }
ae4e8b84 318}
319
3e778975 320//____________________________________________________________
321inline void AliTRDseedV1::SetN(Int_t n)
322{
323 if(n<0 || n>= (1<<5)) return;
324 fN &= ~0x1f;
325 fN |= n;
326}
327
328//____________________________________________________________
329inline void AliTRDseedV1::SetNUsed(Int_t n)
330{
331 if(n<0 || n>= (1<<5)) return;
332 fN &= ~(0x1f<<5);
333 n <<= 5; fN |= n;
334}
335
336//____________________________________________________________
337inline void AliTRDseedV1::SetNShared(Int_t n)
338{
339 if(n<0 || n>= (1<<5)) return;
340 fN &= ~(0x1f<<10);
341 n <<= 10; fN |= n;
342}
343
344
e4f2f73d 345#endif
346
47d5d320 347
6e49cfdb 348