Attaching extra clusters from the overlapped ITS modules
[u/mrichter/AliRoot.git] / ITS / AliITStrackV2.h
1 #ifndef ALIITSTRACKV2_H
2 #define ALIITSTRACKV2_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 //-------------------------------------------------------------------------
7 //                       ITS Track Class
8 //
9 //        Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch 
10 //     dEdx analysis by: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
11 //-------------------------------------------------------------------------
12
13
14 /*****************************************************************************
15  *                          December 18, 2000                                *
16  *  Internal view of the ITS track parametrisation as well as the order of   *
17  *           track parameters are subject for possible changes !             *
18  *  Use GetExternalParameters() and GetExternalCovariance() to access ITS    *
19  *      track information regardless of its internal representation.         *
20  * This formation is now fixed in the following way:                         *
21  *      external param0:   local Y-coordinate of a track (cm)                *
22  *      external param1:   local Z-coordinate of a track (cm)                *
23  *      external param2:   local sine of the track momentum azimuthal angle  *
24  *      external param3:   tangent of the track momentum dip angle           *
25  *      external param4:   1/pt (1/(GeV/c))                                  *
26  *****************************************************************************/
27
28 #include <AliKalmanTrack.h>
29
30 #include "AliITSrecoV2.h"
31
32 class AliESDtrack;
33 class AliStrLine;
34
35 //_____________________________________________________________________________
36 class AliITStrackV2 : public AliKalmanTrack {
37 public:
38   AliITStrackV2();
39   AliITStrackV2(AliESDtrack& t,Bool_t c=kFALSE) throw (const Char_t *);
40   AliITStrackV2(const AliITStrackV2& t);
41   Int_t PropagateToVertex(Double_t d=0., Double_t x0=0.);
42   Int_t Propagate(Double_t alpha, Double_t xr);
43   virtual Int_t CorrectForMaterial(Double_t d, Double_t x0=21.82);
44   Int_t PropagateTo(Double_t xr, Double_t d, Double_t x0=21.82);
45   Double_t PropagateToDCA(AliKalmanTrack *p, Double_t d=0., Double_t x0=0.); 
46   Int_t Update(const AliCluster* cl,Double_t chi2,UInt_t i);
47   Int_t Improve(Double_t x0,Double_t xyz[3],Double_t ers[3]);
48   void SetdEdx(Double_t dedx) {fdEdx=dedx;}
49   void SetSampledEdx(Float_t q, Int_t i);
50   void CookdEdx(Double_t low=0., Double_t up=0.51);
51   void SetDetectorIndex(Int_t i) {SetLabel(i);}
52   void ResetCovariance();
53   void ResetClusters() { SetChi2(0.); SetNumberOfClusters(0); }
54   void UpdateESDtrack(ULong_t flags) const;
55   
56   AliESDtrack *GetESDtrack() const {return fESDtrack;}
57
58   Int_t GetDetectorIndex() const {return GetLabel();}
59   Double_t GetX()    const {return fX;}
60   Double_t GetAlpha()const {return fAlpha;}
61   Double_t GetdEdx() const {return fdEdx;}
62   Double_t GetPIDsignal() const {return GetdEdx();}
63   Double_t GetY()    const {return fP0;}
64   Double_t GetZ()    const {return fP1;}
65   Double_t GetSnp()  const {return fP2;}
66   Double_t GetTgl()  const {return fP3;}
67   Double_t GetC()    const {return fP4;}
68   Double_t Get1Pt() const {
69       return (TMath::Sign(1e-9,fP4) + fP4)*GetLocalConvConst();
70   }
71   Double_t GetD(Double_t x=0, Double_t y=0) const;
72   Double_t GetZat(Double_t x=0) const;
73
74   Double_t GetSigmaY2() const {return fC00;}
75   Double_t GetSigmaZ2() const {return fC11;}
76   Int_t Compare(const TObject *o) const;
77   void GetExternalParameters(Double_t& xr, Double_t x[5]) const ;
78   void GetExternalCovariance(Double_t cov[15]) const ;
79   Int_t GetClusterIndex(Int_t i) const {return fIndex[i];}
80   Int_t GetGlobalXYZat(Double_t r,Double_t &x,Double_t &y,Double_t &z) const;
81   void ApproximateHelixWithLine(Double_t xk, AliStrLine *line);
82   Double_t GetPredictedChi2(const AliCluster *cluster) const;
83   Int_t Invariant() const;
84
85   void  SetExtraCluster(Int_t i, Int_t idx) {fIndex[kMaxLayer+i]=idx;}
86   Int_t GetExtraCluster(Int_t i) const {return fIndex[kMaxLayer+i];}
87  
88 protected:
89   void GetXYZ(Float_t r[3]) const;
90
91   Double_t fX;              // X-coordinate of this track (reference plane)
92   Double_t fAlpha;          // rotation angle
93
94   Double_t fdEdx;           // dE/dx
95
96   Double_t fP0;             // Y-coordinate of a track 
97   Double_t fP1;             // Z-coordinate of a track
98   Double_t fP2;             // sine of the track momentum azimuthal angle
99   Double_t fP3;             // tangent of the track momentum dip angle
100   Double_t fP4;             // track curvature
101
102   Double_t fC00;                         // covariance
103   Double_t fC10, fC11;                   // matrix
104   Double_t fC20, fC21, fC22;             // of the
105   Double_t fC30, fC31, fC32, fC33;       // track
106   Double_t fC40, fC41, fC42, fC43, fC44; // parameters 
107
108   Int_t fIndex[2*kMaxLayer]; // indices of associated clusters 
109   Float_t fdEdxSample[4];    // array of dE/dx samples b.b.
110
111   AliESDtrack *fESDtrack;    //! pointer to the connected ESD track
112   ClassDef(AliITStrackV2,5)  //ITS reconstructed track
113 };
114
115 inline 
116 void AliITStrackV2::GetExternalParameters(Double_t& xr, Double_t x[5]) const {
117   //---------------------------------------------------------------------
118   // This function return external ITS track representation
119   //---------------------------------------------------------------------
120      xr=fX;          
121      x[0]=GetY(); x[1]=GetZ(); x[2]=GetSnp(); x[3]=GetTgl(); x[4]=Get1Pt();
122 }
123
124 inline
125 void AliITStrackV2::SetSampledEdx(Float_t q, Int_t i) {
126   //----------------------------------------------------------------------
127   // This function stores dEdx sample corrected for the track segment length 
128   // Origin: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
129   //----------------------------------------------------------------------
130   if (i<0) return;
131   if (i>3) return;
132   Double_t s=GetSnp(), t=GetTgl();
133   q *= TMath::Sqrt((1-s*s)/(1+t*t));
134   fdEdxSample[i]=q;
135 }
136
137 inline void AliITStrackV2::GetXYZ(Float_t r[3]) const {
138   //---------------------------------------------------------------------
139   // Returns the position of the track in the global coord. system 
140   //---------------------------------------------------------------------
141   Double_t cs=TMath::Cos(fAlpha), sn=TMath::Sin(fAlpha);
142   r[0]=fX*cs - fP0*sn; r[1]=fX*sn + fP0*cs; r[2]=fP1;
143 }
144
145 #endif
146
147