Changes related to the extraction of the V0 finder into a separate class (A. Dainese...
[u/mrichter/AliRoot.git] / ITS / AliITStrackMI.h
1 #ifndef ALIITSTRACKMI_H
2 #define ALIITSTRACKMI_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: Marian Ivanov, CERN, Marian.Ivanov@cern.ch 
10 //     dEdx analysis by: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
11 //-------------------------------------------------------------------------
12
13 /* $Id$ */
14
15 /*****************************************************************************
16  *                          December 18, 2000                                *
17  *  Internal view of the ITS track parametrisation as well as the order of   *
18  *           track parameters are subject for possible changes !             *
19  *  Use GetExternalParameters() and GetExternalCovariance() to access ITS    *
20  *      track information regardless of its internal representation.         *
21  * This formation is now fixed in the following way:                         *
22  *      external param0:   local Y-coordinate of a track (cm)                *
23  *      external param1:   local Z-coordinate of a track (cm)                *
24  *      external param2:   local sine of the track momentum azimuthal angle  *
25  *      external param3:   tangent of the track momentum dip angle           *
26  *      external param4:   1/pt (1/(GeV/c))                                  *
27  *****************************************************************************/
28
29 #include <AliKalmanTrack.h>
30
31 #include "AliITSRecoParam.h"
32 #include "AliITStrackV2.h"
33
34 class AliESDtrack;
35
36 //_____________________________________________________________________________
37 class AliITStrackMI : public AliITStrackV2 {
38 public:
39   AliITStrackMI();
40   AliITStrackMI(AliESDtrack& t,Bool_t c=kFALSE) throw (const Char_t *);
41   AliITStrackMI(const AliITStrackMI& t);
42   Int_t GetProlongationFast(Double_t alpha, Double_t xr,Double_t &y, Double_t &z);
43   Bool_t UpdateMI(const AliCluster *c, Double_t chi2, Int_t i);  
44
45   void SetReconstructed(Bool_t sr=kTRUE){fReconstructed = sr;}  
46   Bool_t GetReconstructed() const {return fReconstructed;}
47   void SetChi2MIP(Int_t i,Float_t val){fChi2MIP[i]=val;}
48   Float_t GetChi2MIP(Int_t i) const {return fChi2MIP[i];}  
49   void IncrementNSkipped(){fNSkipped++;} // increment by 1 the # of skipped cls
50   Float_t GetNSkipped() const {return fNSkipped;}
51   void SetNSkipped(Float_t n) {fNSkipped=n;}
52   void IncrementNUsed(){fNUsed++;} // increment by 1 the # of shared clusters
53   Float_t GetNUsed() const {return fNUsed;}
54   void SetNUsed(Float_t n) {fNUsed=n;}
55
56   Int_t Compare(const TObject *o) const;
57   Double_t GetCov33() const {return GetCovariance()[9];} // cov. matrix el. 3,3
58   //Double_t GetCov44() const {return GetCovariance()[15];}// cov. matrix el. 4,4
59   Float_t GetDy(Int_t i) const {return fDy[i];}
60   Float_t GetDz(Int_t i) const {return fDz[i];}
61   Float_t GetD(Int_t i) const {return fD[i];}
62   Double_t GetD(Double_t x, Double_t y) const
63     {return AliITStrackV2::GetD(x,y);}
64   Float_t *GetDP() {return fD;}
65   void SetD(Int_t i, Float_t d) {fD[i]=d;}
66   Float_t GetDnorm(Int_t i) const {return fDnorm[i];}
67   Float_t *GetDnormP() {return fDnorm;}
68   void SetDnorm(Int_t i, Float_t d) {fDnorm[i]=d;}
69   Float_t GetSigmaY(Int_t i) const {return fSigmaY[i];}
70   Float_t GetSigmaZ(Int_t i) const {return fSigmaZ[i];}
71   void SetSigmaY(Int_t i, Float_t s) {fSigmaY[i]=s;}
72   void SetSigmaZ(Int_t i, Float_t s) {fSigmaZ[i]=s;}
73   Float_t GetNDeadZone() const {return fNDeadZone;}
74   void SetNDeadZone(Float_t d) {fNDeadZone=d;}
75   Int_t* ClIndex() {return fClIndex;}
76   Int_t GetClIndex(Int_t i) const {return fClIndex[i];}
77   void SetClIndex(Int_t i, Int_t c) {fClIndex[i]=c;}
78   Float_t GetNormChi2(Int_t i) const {return fNormChi2[i];}
79   void SetNormChi2(Int_t i, Float_t n) {fNormChi2[i]=n;}
80   Bool_t GetConstrain() const {return fConstrain;}
81   void SetConstrain(Bool_t c) {fConstrain=c;}
82   Float_t GetExpQ() const {return fExpQ;}
83   void SetExpQ(Float_t f) {fExpQ=f;}
84   Float_t GetNormQ(Int_t i) const {return fNormQ[i];}
85   void SetNormQ(Int_t i, Float_t q) {fNormQ[i]=q;}
86   Float_t GetdEdxMismatch() const {return fdEdxMismatch;}
87   void SetdEdxMismatch(Float_t m) {fdEdxMismatch=m;}
88   Float_t GetNy(Int_t i) const {return fNy[i];}
89   void SetNy(Int_t i, Float_t f) {fNy[i]=f;}
90   Float_t GetNz(Int_t i) const {return fNz[i];}
91   void SetNz(Int_t i, Float_t f) {fNz[i]=f;}
92   Bool_t GetGoldV0() const {return fGoldV0;}
93   void SetGoldV0(Bool_t g) {fGoldV0=g;}
94   Float_t GetChi22() const {return fChi22;}
95   void SetChi22(Float_t c) {fChi22=c;}
96   Float_t GetDeadZoneProbability() const {return fDeadZoneProbability;}
97   void SetDeadZoneProbability(Float_t d) {fDeadZoneProbability=d;}
98
99   Double_t GetPredictedChi2MI(Double_t cy, Double_t cz, Double_t cerry, Double_t cerrz) const;
100   Bool_t IsGoldPrimary();
101 protected:
102
103   Float_t fNUsed;                          // number of shared clusters
104   Float_t fNSkipped;                       // number of skipped clusters
105   Float_t fNDeadZone;                     // number of clusters in dead zone
106   Float_t fDeadZoneProbability;          // probability to cross dead zone
107   Bool_t  fReconstructed;                 // reconstructed - accepted flag
108   Float_t fChi2MIP[12];                   // MIP chi squres 
109
110   Float_t fDy[12];           //dy in layer
111   Float_t fDz[12];           //dz in layer
112   Float_t fSigmaY[12];       //sigma y 
113   Float_t fSigmaZ[12];       //sigma z
114   Float_t fNy[6];              //expected size of cluster
115   Float_t fNz[6];              //expected size of cluster
116   Float_t fD[2];            //distance to the vertex
117   Float_t fDnorm[2];        // normalized distance to the vertex
118   Float_t fNormQ[6];        // normalized Q
119   Float_t fExpQ;            // expected Q
120   Float_t fNormChi2[6];     // normalized chi2 
121   Float_t fChi22;           // chi22
122   Float_t fdEdxMismatch;    
123   Bool_t fConstrain;        //indication of the vertex constrain
124   Int_t  fClIndex[6];       //cluster Index
125   Bool_t fGoldV0;           //corresponding gold V0 found
126   ClassDef(AliITStrackMI,1)   //ITS reconstructed track
127 };
128
129 #endif
130
131