]> git.uio.no Git - u/mrichter/AliRoot.git/blame_incremental - TRD/AliTRDtrackV1.h
git://git.uio.no / u / mrichter / AliRoot.git / blame_incremental
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (non-incremental) | history | HEAD
recMC.C submitMC.sh
[u/mrichter/AliRoot.git] / TRD / AliTRDtrackV1.h
... / ...
CommitLineData
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
26class AliTRDcluster;
27class AliESDtrack;
28class AliTRDReconstructor;
29class AliTRDtrackV1 : public AliKalmanTrack
30{
31public:
32 enum ETRDtrackV1Size {
33 kNdet = AliTRDgeometry::kNdet
34 ,kNstacks = AliTRDgeometry::kNstack*AliTRDgeometry::kNsector
35 ,kNplane = AliTRDgeometry::kNlayer
36 ,kNcham = AliTRDgeometry::kNstack
37 ,kNsect = AliTRDgeometry::kNsector
38 ,kNslice = 3
39 ,kNMLPslice = 8
40 ,kMAXCLUSTERSPERTRACK = 210
41 };
42
43 // bits from 0-13 are reserved by ROOT (see TObject.h)
44 enum ETRDtrackV1Status {
45 kOwner = BIT(14)
46 ,kStopped = BIT(15)
47 };
48
49 AliTRDtrackV1();
50 AliTRDtrackV1(AliTRDseedV1 *trklts, const Double_t p[5], const Double_t cov[15], Double_t x, Double_t alpha);
51 AliTRDtrackV1(const AliESDtrack &ref);
52 AliTRDtrackV1(const AliTRDtrackV1 &ref);
53 virtual ~AliTRDtrackV1();
54 AliTRDtrackV1 &operator=(const AliTRDtrackV1 &ref) { *(new(this) AliTRDtrackV1(ref)); return *this; }
55
56 Bool_t CookPID();
57 Bool_t CookLabel(Float_t wrong);
58 AliTRDtrackV1* GetBackupTrack() const {return fBackupTrack;}
59 Double_t GetBudget(Int_t i) const { return fBudget[i];}
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 AliExternalTrackParam*
75 GetTrackLow() const { return fTrackLow;}
76 AliExternalTrackParam*
77 GetTrackHigh() const { return fTrackHigh;}
78 UShort_t* GetTrackletIndexes() { return &fTrackletIndex[0];}
79
80 Bool_t IsEqual(const TObject *inTrack) const;
81 Bool_t IsOwner() const {return TestBit(kOwner);};
82 Bool_t IsStopped() const {return TestBit(kStopped);};
83 Bool_t IsElectron() const;
84
85 void MakeBackupTrack();
86 void Print(Option_t *o="") const;
87
88 Bool_t PropagateTo(Double_t xr, Double_t x0 = 8.72, Double_t rho = 5.86e-3);
89 Int_t PropagateToR(Double_t xr, Double_t step);
90 Bool_t Rotate(Double_t angle, Bool_t absolute = kFALSE);
91 void SetBudget(Int_t i, Double_t b) {if(i>=0 && i<3) fBudget[i] = b;}
92 void SetEdep(Double32_t inDE){fDE = inDE;};
93 void SetNumberOfClusters();
94 void SetOwner();
95 void SetPID(Short_t is, Double_t inPID){if (is >=0 && is < AliPID::kSPECIES) fPID[is]=inPID;};
96 void SetPIDquality(UChar_t inPIDquality){fPIDquality = inPIDquality;};
97 void SetStopped(Bool_t stop) {SetBit(kStopped, stop);}
98 void SetTracklet(AliTRDseedV1 *trklt, Int_t index);
99 void SetTrackLow();
100 void SetTrackHigh(const AliExternalTrackParam *op=0x0);
101 inline void SetReconstructor(const AliTRDReconstructor *rec);
102 inline Float_t StatusForTOF();
103 void UnsetTracklet(Int_t plane);
104 Bool_t Update(AliTRDseedV1 *tracklet, Double_t chi2);
105 //Bool_t Update(const AliTRDcluster *c, Double_t chi2, Int_t index, Double_t h01){ return AliTRDtrack::Update(c,chi2,index,h01); };
106 Bool_t Update(const AliCluster *, Double_t, Int_t) { return kFALSE; };
107 void UpdateESDtrack(AliESDtrack *t);
108
109private:
110 UChar_t fPIDquality; // No of planes used for PID calculation
111 UShort_t fTrackletIndex[kNplane];// Tracklets index in the tracker list
112 Double32_t fPID[AliPID::kSPECIES];// PID probabilities
113 Double32_t fBudget[3]; // Integrated material budget
114 Double32_t fDE; // Integrated delta energy
115 const AliTRDReconstructor *fReconstructor;//! reconstructor link
116 AliTRDseedV1 *fTracklet[kNplane]; // Tracklets array defining the track
117 AliTRDtrackV1 *fBackupTrack; // Backup track
118 AliExternalTrackParam *fTrackLow; // parameters of the track which enter TRD from below (TPC)
119 AliExternalTrackParam *fTrackHigh; // parameters of the track which enter TRD from above (HMPID, PHOS)
120
121 ClassDef(AliTRDtrackV1, 4) // new TRD track
122};
123
124//____________________________________________________
125inline Float_t AliTRDtrackV1::GetMomentum(Int_t plane) const
126{
127 return plane >=0 && plane < kNplane && fTrackletIndex[plane] != 0xff ? fTracklet[plane]->GetMomentum() : -1.;
128}
129
130//____________________________________________________
131inline Int_t AliTRDtrackV1::GetNCross()
132{
133 Int_t ncross = 0;
134 for(Int_t ip=0; ip<kNplane; ip++){
135 if(!fTracklet[ip]) continue;
136 ncross += fTracklet[ip]->IsRowCross();
137 }
138 return ncross;
139}
140
141//____________________________________________________
142inline Int_t AliTRDtrackV1::GetNumberOfTracklets() const
143{
144 Int_t n = 0;
145 for(Int_t ip=0; ip<kNplane; ip++){
146 if(!fTracklet[ip]) continue;
147 n++;
148 }
149 return n;
150}
151
152
153//____________________________________________________
154inline void AliTRDtrackV1::SetReconstructor(const AliTRDReconstructor *rec)
155{
156 for(Int_t ip=0; ip<kNplane; ip++){
157 if(!fTracklet[ip]) continue;
158 fTracklet[ip]->SetReconstructor(rec);
159 }
160 fReconstructor = rec;
161}
162
163
164//____________________________________________________________________________
165inline Float_t AliTRDtrackV1::StatusForTOF()
166{
167 //
168 // Defines the status of the TOF extrapolation
169 //
170
171 if(!fTracklet[5]) return 0.;
172
173 // Definition of res ????
174 Float_t res = /*(0.2 + 0.8 * (fN / (fNExpected + 5.0))) **/ (0.4 + 0.6 * fTracklet[5]->GetN() / 20.0);
175 res *= (0.25 + 0.8 * 40.0 / (40.0 + fBudget[2]));
176 return res;
177}
178
179#endif
180
181
182