]>
Commit | Line | Data |
---|---|---|
10b59615 | 1 | #ifndef ALITRDTRACKONLINE_H |
2 | #define ALITRDTRACKONLINE_H | |
3 | ||
4 | #include "TObject.h" | |
5 | #include "TList.h" | |
6 | ||
7 | #include "Math/IFunction.h" | |
8 | #include "Math/Minimizer.h" | |
9 | ||
10 | #include "AliTRDgeometry.h" | |
11 | #include "AliTRDpadPlane.h" | |
12 | ||
13 | class AliVTrdTracklet; | |
14 | class AliTRDtrackPosition; | |
15 | class AliTRDtrackOnline; | |
16 | class AliTRDgeometry; | |
17 | ||
18 | class AliTRDtrackParametrization : public TNamed | |
19 | { | |
20 | public: | |
21 | AliTRDtrackParametrization(const char* name = "", const char* title = ""); | |
22 | ~AliTRDtrackParametrization() {} | |
23 | ||
24 | virtual void SetParams(ROOT::Math::Minimizer * minim) = 0; | |
25 | virtual void GetParams(ROOT::Math::Minimizer * minim) = 0; | |
26 | virtual void SetValues(const Double_t *par) = 0; | |
27 | virtual Int_t GetNDim() const = 0; | |
28 | ||
29 | virtual void UpdateTitle() {} | |
30 | ||
31 | virtual AliTRDtrackPosition ExtrapolateToLayer(Int_t layer) = 0; | |
32 | virtual AliTRDtrackPosition ExtrapolateToX(Float_t x) = 0; | |
33 | ||
34 | Bool_t IsFitGood() const { return fFitGood; } | |
35 | ||
36 | protected: | |
37 | Bool_t fFitGood; | |
38 | ||
39 | ClassDef(AliTRDtrackParametrization, 1); | |
40 | }; | |
41 | ||
42 | ||
43 | class AliTRDtrackParametrizationStraightLine : public AliTRDtrackParametrization | |
44 | { | |
45 | public: | |
46 | AliTRDtrackParametrizationStraightLine(); | |
47 | AliTRDtrackParametrizationStraightLine(Double_t offsetY, Double_t slopeY, | |
48 | Double_t offsetZ, Double_t slopeZ); | |
49 | ||
50 | virtual void SetParams(ROOT::Math::Minimizer * minim); | |
51 | virtual void GetParams(ROOT::Math::Minimizer * minim); | |
52 | virtual void SetValues(const Double_t *par); | |
53 | virtual Int_t GetNDim() const { return 4; } | |
54 | ||
55 | AliTRDtrackPosition ExtrapolateToLayer(Int_t layer); | |
56 | AliTRDtrackPosition ExtrapolateToX(Float_t x); | |
57 | ||
58 | Double_t GetOffsetY() const { return fOffsetY; } | |
59 | Double_t GetOffsetZ() const { return fOffsetZ; } | |
60 | Double_t GetSlopeY() const { return fSlopeY; } | |
61 | Double_t GetSlopeZ() const { return fSlopeZ; } | |
62 | ||
63 | void Print(Option_t *option = "") const; | |
64 | ||
65 | protected: | |
66 | Double_t fOffsetY; | |
67 | Double_t fSlopeY; | |
68 | Double_t fOffsetZ; | |
69 | Double_t fSlopeZ; | |
70 | ||
71 | ClassDef(AliTRDtrackParametrizationStraightLine, 1); | |
72 | }; | |
73 | ||
74 | ||
75 | class AliTRDtrackParametrizationCurved : public AliTRDtrackParametrization | |
76 | { | |
77 | public: | |
78 | AliTRDtrackParametrizationCurved(); | |
79 | ||
80 | virtual void SetParams(ROOT::Math::Minimizer * minim); | |
81 | virtual void GetParams(ROOT::Math::Minimizer * minim); | |
82 | virtual void SetValues(const Double_t *par); | |
83 | virtual Int_t GetNDim() const { return 4; } | |
84 | ||
85 | AliTRDtrackPosition ExtrapolateToLayer(Int_t layer); | |
86 | AliTRDtrackPosition ExtrapolateToX(Float_t x); | |
87 | ||
88 | Float_t GetY(Float_t x); | |
89 | ||
90 | void Print(Option_t *option = "") const; | |
91 | ||
92 | protected: | |
93 | // parameters | |
94 | Double_t fRadiusInv; | |
95 | Double_t fOffsetY; | |
96 | Double_t fOffsetZ; | |
97 | Double_t fSlopeZ; | |
98 | ||
99 | // fixed values | |
100 | Double_t fOffsetX; | |
101 | ||
102 | ClassDef(AliTRDtrackParametrizationCurved, 1); | |
103 | }; | |
104 | ||
105 | ||
106 | class AliTRDtrackOnline : public TObject | |
107 | { | |
108 | public: | |
109 | AliTRDtrackOnline(); | |
110 | ~AliTRDtrackOnline(); | |
111 | ||
112 | void AddTracklet(AliVTrdTracklet *trkl); | |
113 | ||
114 | Bool_t Fit(ROOT::Math::Minimizer *minim); | |
115 | ||
116 | Int_t GetNTracklets() const { return fNTracklets; } | |
117 | AliVTrdTracklet* GetTracklet(Int_t i) const { return i < fNTracklets ? (AliVTrdTracklet*) fTracklets[i] : 0x0; } | |
118 | ||
119 | AliTRDtrackPosition ExtrapolateToLayer(Int_t layer); | |
120 | ||
121 | void AddParametrization(AliTRDtrackParametrization *param) { fTrackParametrizations.Add(param); } | |
122 | const TList& GetParametrizations() const { return fTrackParametrizations; } | |
123 | ||
124 | void Print(Option_t *option = "") const; | |
125 | ||
126 | static Float_t GetX(AliVTrdTracklet *trkl) { return fgGeometry->GetTime0(trkl->GetDetector() % 6); } | |
20af9b67 | 127 | static Float_t GetZ(AliVTrdTracklet *trkl) { return |
128 | fgGeometry->GetPadPlane(trkl->GetDetector())->GetRowPos(trkl->GetBinZ()) - | |
129 | .5 * fgGeometry->GetPadPlane(trkl->GetDetector())->GetRowSize(trkl->GetBinZ()); } | |
10b59615 | 130 | static AliTRDgeometry *fgGeometry; |
131 | ||
132 | protected: | |
133 | static const Int_t fgkMaxTracklets = 10; | |
134 | ||
135 | Int_t fNTracklets; | |
136 | TObjArray fTracklets; | |
137 | ||
138 | TList fTrackParametrizations; | |
139 | ||
140 | ClassDef(AliTRDtrackOnline, 1); | |
141 | }; | |
142 | ||
143 | ||
144 | class AliTRDtrackPosition : public TObject | |
145 | { | |
146 | public: | |
147 | AliTRDtrackPosition(Float_t y, Float_t z, Float_t dy = 0.); | |
148 | ~AliTRDtrackPosition(); | |
149 | ||
150 | Float_t GetY() const { return fY; } | |
151 | Float_t GetZ() const { return fZ; } | |
152 | Float_t GetdY() const { return fDy; } | |
153 | ||
154 | Float_t Distance(AliVTrdTracklet *trkl) const; | |
155 | ||
156 | protected: | |
157 | Float_t fY; | |
158 | Float_t fZ; | |
159 | Float_t fDy; | |
160 | ||
161 | ClassDef(AliTRDtrackPosition, 1); | |
162 | }; | |
163 | ||
164 | ||
165 | class AliTRDtrackResiduals : public ROOT::Math::IBaseFunctionMultiDim | |
166 | { | |
167 | public: | |
168 | AliTRDtrackResiduals(const AliTRDtrackOnline *track, AliTRDtrackParametrization *param); | |
169 | AliTRDtrackResiduals(const AliTRDtrackResiduals &rhs); | |
170 | AliTRDtrackResiduals& operator=(const AliTRDtrackResiduals &rhs); | |
171 | ~AliTRDtrackResiduals() {} | |
172 | ||
173 | AliTRDtrackResiduals* Clone() const; | |
174 | UInt_t NDim() const { return fParam->GetNDim(); } | |
175 | Double_t DoEval(const Double_t *par) const; | |
176 | ||
177 | protected: | |
178 | const AliTRDtrackOnline *fTrack; // reference to track being fitted | |
179 | AliTRDtrackParametrization *fParam; // reference to the used parametrization | |
180 | ||
181 | static AliTRDgeometry *fgGeometry; | |
182 | }; | |
183 | ||
184 | #endif |