First implementation of ESD classes (Yu.Belikov)
[u/mrichter/AliRoot.git] / STEER / AliKalmanTrack.h
1 #ifndef ALIKALMANTRACK_H
2 #define ALIKALMANTRACK_H
3
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 /* $Id$ */
8
9 //-------------------------------------------------------------------------
10 //                          Class AliKalmanTrack
11 //
12 //         Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch 
13 //-------------------------------------------------------------------------
14
15 #include <TObject.h>
16
17 class AliCluster;
18
19 class AliKalmanTrack : public TObject {
20 public:
21   AliKalmanTrack();
22   AliKalmanTrack(const AliKalmanTrack &t);
23
24   virtual ~AliKalmanTrack(){};
25   void SetLabel(Int_t lab) {fLab=lab;}
26
27   Bool_t   IsSortable() const {return kTRUE;}
28   Int_t    GetLabel()   const {return fLab;}
29   Double_t GetChi2()    const {return fChi2;}
30   Double_t GetMass()    const {return fMass;}
31   Int_t    GetNumberOfClusters() const {return fN;}
32   virtual Int_t GetClusterIndex(Int_t) const { //reserved for AliTracker
33     Warning("GetClusterIndex(Int_t)","Method must be overloaded !\n");
34     return 0;
35   } 
36   virtual Double_t GetPIDsignal() const {
37     Warning("GetPIDsignal()","Method must be overloaded !\n");
38     return 0.;
39   }
40
41   virtual Double_t GetDCA(const AliKalmanTrack *,Double_t &,Double_t &) const; 
42   virtual 
43   Double_t PropagateToDCA(AliKalmanTrack *, Double_t d=0., Double_t x0=0.); 
44   virtual Double_t GetAlpha() const {
45     Warning("GetAlpha()","Method must be overloaded !\n");
46     return 0.;
47   }
48   virtual Double_t GetSigmaY2() const {
49     Warning("GetSigmaY2()","Method must be overloaded !\n");
50     return 0.;
51   }
52   virtual Double_t GetSigmaZ2() const {
53     Warning("GetSigmaZ2()","Method must be overloaded !\n");
54     return 0.;
55   }
56
57   virtual Int_t Compare(const TObject *) const {return 0;} 
58
59   virtual void GetExternalParameters(Double_t &/*xr*/, Double_t /*x*/[5]) const {}
60   virtual void GetExternalCovariance(Double_t /*cov*/[15]) const {}
61
62   virtual Double_t GetX() const;
63   virtual Double_t GetdEdx() const;
64
65   virtual Double_t GetY() const;
66   virtual Double_t GetZ() const;
67   virtual Double_t GetSnp() const;
68   virtual Double_t GetTgl() const;
69   virtual Double_t Get1Pt() const;
70
71   virtual Double_t Phi() const;
72   virtual Double_t SigmaPhi() const;
73   virtual Double_t Theta() const;
74   virtual Double_t SigmaTheta() const;
75   virtual Double_t Px() const;
76   virtual Double_t Py() const;
77   virtual Double_t Pz() const;
78   virtual Double_t Pt() const;
79   virtual Double_t SigmaPt() const;
80   virtual Double_t P() const;
81
82   virtual Double_t GetPredictedChi2(const AliCluster *) const {return 0.;}
83   virtual 
84     Int_t PropagateTo(Double_t /*xr*/, Double_t /*x0*/, Double_t /*rho*/) {return 0;}
85   virtual Int_t Update(const AliCluster*, Double_t /*chi2*/, UInt_t) {return 0;}
86
87   static void SetConvConst(Double_t cc) {fgConvConst=cc;}
88   Double_t GetConvConst() const {return fgConvConst;}
89
90   static void SetMagneticField(Double_t f) {// f - Magnetic field in T
91     fgConvConst=100/0.299792458/f;
92   }
93   Double_t GetMagneticField() const {return 100/0.299792458/fgConvConst;}
94
95   // Time integration (S.Radomski@gsi.de)
96   void   StartTimeIntegral();
97   void SetIntegratedLength(Double_t l) {fIntegratedLength=l;}
98   void SetIntegratedTimes(const Double_t *times);
99
100   Bool_t IsStartedTimeIntegral() const {return fStartTimeIntegral;}
101   void     AddTimeStep(Double_t length);
102   void GetIntegratedTimes(Double_t *times) const;
103   Double_t GetIntegratedTime(Int_t pdg) const;
104   Double_t GetIntegratedLength() const {return fIntegratedLength;}
105   void PrintTime() const;
106   
107
108 protected:
109   void SetChi2(Double_t chi2) {fChi2=chi2;} 
110   void SetMass(Double_t mass) {fMass=mass;}
111   void SetNumberOfClusters(Int_t n) {fN=n;} 
112
113  private:
114   Int_t fLab;             // track label
115   Double_t fChi2;         // total chi2 value for this track
116   Double_t fMass;         // mass hypothesis
117   Int_t fN;               // number of associated clusters
118
119   static Double_t fgConvConst; //conversion constant cm -> GeV/c
120
121   // variables for time integration (S.Radomski@gsi.de)
122   static const Int_t fgkTypes = 5;  // Number of track types (e,mu,pi,k,p)
123   Bool_t  fStartTimeIntegral;       // indicator wether integrate time
124   Float_t fIntegratedTime[5];       // integrated time
125   Float_t fIntegratedLength;        // integrated length
126   
127   ClassDef(AliKalmanTrack,2)    // Reconstructed track
128 };
129
130 #endif
131
132