Script supersceeded by AliForwarddNdetaTask.C and
[u/mrichter/AliRoot.git] / TPC / AliTPCtrack.h
1 #ifndef ALITPCTRACK_H
2 #define ALITPCTRACK_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5 /* $Id$ */
6
7 //-------------------------------------------------------
8 //                    TPC Track Class
9 //
10 //   Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch 
11 //
12 // The track parameterization is fixed in the following way:                   
13 //      param0:   local Y-coordinate of a track (cm)                
14 //      param1:   local Z-coordinate of a track (cm)                
15 //      param2:   local sine of the track momentum azimuth angle    
16 //      param3:   tangent of the track momentum dip angle           
17 //      param4:   1/pt (1/(GeV/c))                                  
18 //
19 //-------------------------------------------------------
20
21 #include <AliKalmanTrack.h>
22 #include <TMath.h>
23
24 #include "AliTPCreco.h"
25 #include "AliExternalTrackParam.h"
26 class AliESDtrack;
27 class AliESDVertex;
28 class TTreeSRedirector;
29 //_____________________________________________________________________________
30 class AliTPCtrack : public AliKalmanTrack {
31 public:
32   AliTPCtrack();
33   AliTPCtrack(Double_t x, Double_t alpha, const Double_t p[5], 
34               const Double_t cov[15], Int_t index); 
35   AliTPCtrack(const AliESDtrack& t, TTreeSRedirector *pcstream=0);
36   AliTPCtrack(const AliTPCtrack& t);
37   AliTPCtrack& operator=(const AliTPCtrack& o);
38   virtual ~AliTPCtrack() {}
39
40   Int_t Compare(const TObject *o) const;
41
42   void SetdEdx(Double_t dedx) {fdEdx=dedx;}
43   Double_t GetdEdx()  const {return fdEdx;}
44   Double_t GetPIDsignal()  const {return GetdEdx();}
45
46   Int_t GetClusterIndex(Int_t i) const {return fIndex[i];}
47   void  SetClusterIndex(Int_t i, Int_t idx) {fIndex[i]=idx;}
48
49   Double_t GetC()           const {return AliExternalTrackParam::GetC(GetBz());}
50   Double_t GetC(Double_t b) const {return AliExternalTrackParam::GetC(b);}
51
52   Double_t GetPredictedChi2(const AliCluster *cluster) const;
53   Bool_t PropagateTo(Double_t xr, Double_t rho=0.9e-3, Double_t x0=28.94);
54   Bool_t Update(const AliCluster *c, Double_t chi2, Int_t i);
55   Bool_t Rotate(Double_t alpha) {
56     return AliExternalTrackParam::Rotate(GetAlpha()+alpha);
57   }
58   
59   Bool_t PropagateToVertex(const AliESDVertex *v, 
60                            Double_t rho=1.2e-3, Double_t x0=36.66);
61   void ResetClusters() {SetNumberOfClusters(0); SetChi2(0.);}
62   void UpdatePoints();//update points 
63   Float_t* GetPoints() {return fPoints;}
64
65   Float_t Density(Int_t row0, Int_t row1); //calculate cluster density
66   Float_t Density2(Int_t row0, Int_t row1); //calculate cluster density
67   Double_t GetD(Double_t x=0, Double_t y=0) const {
68      return AliExternalTrackParam::GetD(x,y,GetBz());
69   }
70   AliExternalTrackParam & GetReference(){ return fReference;}
71   void UpdateReference(){ new (&fReference) AliExternalTrackParam(*this);}
72   Int_t   GetKinkIndex(Int_t i) const{ return fKinkIndexes[i];}
73   Int_t*  GetKinkIndexes() { return fKinkIndexes;}
74   Int_t   GetV0Index(Int_t i) const{ return fV0Indexes[i];}
75   Int_t*  GetV0Indexes() { return fV0Indexes;}
76
77   void SetFirstPoint(Int_t f) {fFirstPoint=f;}
78   void SetLastPoint(Int_t f) {fLastPoint=f;}
79   void SetRemoval(Int_t f) {fRemoval=f;}
80   void SetLab2(Int_t f) {fLab2=f;}
81   void SetKinkIndex(Int_t i, Int_t idx) {fKinkIndexes[i]=idx;}
82   void SetBConstrain(Bool_t b) {fBConstrain=b;}
83   void SetNFoundable(Int_t n) {fNFoundable=n;}
84   void SetNShared(Int_t s) {fNShared=s;}
85
86   Int_t GetFirstPoint() const {return fFirstPoint;}
87   Int_t GetLastPoint() const {return fLastPoint;}
88   Int_t GetRemoval() const {return fRemoval;}
89   Int_t GetLab2() const {return fLab2;}
90   Bool_t GetBConstrain() const {return fBConstrain;}
91   Int_t GetNShared() const {return fNShared;}
92   Int_t GetNFoundable() const {return fNFoundable;}
93
94 protected: 
95   Double_t fdEdx;           // dE/dx
96
97   Int_t fIndex[kMaxRow];       // indices of associated clusters 
98   Float_t fPoints[4];            //first, max dens row  end points of the track and max density
99   // MI addition
100   Float_t fSdEdx;           // sigma of dedx 
101   //
102   Int_t   fNFoundable;      //number of foundable clusters - dead zone taken to the account
103   Bool_t  fBConstrain;   // indicate seeding with vertex constrain
104   Int_t   fLastPoint;     // last  cluster position     
105   Int_t   fFirstPoint;    // first cluster position
106   Int_t fRemoval;         // removal factor
107   Int_t fTrackType;       // track type - 0 - normal - 1 - kink -  2 -V0  3- double found
108   Int_t fLab2;            // index of corresponding track (kink, V0, double)
109   Int_t   fNShared;       // number of shared points 
110   AliExternalTrackParam   fReference; // track parameters at the middle of the chamber
111   Float_t  fKinkPoint[12];      //radius, of kink,  dfi and dtheta
112   Int_t    fKinkIndexes[3];     // kink indexes - minus = mother + daughter
113   Int_t    fV0Indexes[3];     // kink indexes - minus = mother + daughter
114
115   ClassDef(AliTPCtrack,4)   // Time Projection Chamber reconstructed tracks
116 };
117
118 #endif
119