- fix for z position in on-line track fit
[u/mrichter/AliRoot.git] / TRD / AliTRDtrackOnline.h
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); }
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()); }
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