Automatic treatment of the magnetic field value
[u/mrichter/AliRoot.git] / ITS / AliITSTrackV1.h
1 #ifndef ALIITSTRACKV1_H
2 #define ALIITSTRACKV1_H
3 //   ITS Track Class
4 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
5 // It contain all the usefull information for the track and the method to calculate, modify or extract them
6 #include <TObject.h>
7 #include <TMatrix.h>
8 #include <TVector.h>
9
10 #include "../TPC/AliTPCtrack.h"
11
12 class TObjArray;
13 class AliITSRad;
14
15 class AliITSTrackV1 : public TObject { 
16
17 public:
18
19   AliITSTrackV1();  // default constructor
20   AliITSTrackV1(AliTPCtrack &obj);  // copy constructor
21   AliITSTrackV1(const AliITSTrackV1 &cobj);  // copy constructor
22   AliITSTrackV1 &operator=(AliITSTrackV1 obj);  // operator =
23   ~AliITSTrackV1(); // default destructor
24   Int_t GetNumClust() const { return fNumClustInTrack;}  // gets the num of cluster in a track
25   void AddClustInTrack() { fNumClustInTrack++;}  // adds a cluster in track
26   TObjArray *GetListOfCluster() const { return flistCluster;}  // gets the list of cluster in the track
27   void SetChi2(Double_t chi2) { fChi2 = chi2;}   // sets the chisquare value for the track
28   Double_t GetChi2() const { return fChi2;}            // gets the chisquare value for the track
29   Double_t GetZ() const {return fX1;}            // gets the Z value for the track
30   Double_t GetTgl() const {return fX3;}          // gets the tgl value for the track
31   Double_t Getrtrack() const{return frtrack;}    // gets the raius value for the current track
32   Double_t Getphi()  const{return fX0;}          // gets the phi value for the track
33   Double_t GetC() const {return fX4;}            // gets the curvature value for the track
34   Double_t GetD() const{return fX2;}             // gets the radial impact parameter for the track
35   Double_t GetPt() const {return 0.299792458*0.2/(fX4*100.);} // gets the transvers momentum value for the
36                                                               //track           
37   void SetVertex(TVector &vert) { for(Int_t i=0;i<3;i++) fVertex(i) = vert(i);} // sets the vertex
38                                                                                 // cohordinates
39   void SetErrorVertex(TVector &evert) {for(Int_t i=0;i<3;i++) fErrorVertex(i) = evert(i);} // sets the errors
40                                                                                     //for vertex cohordinates
41
42   void LmTPC(); // trasform state vector and covariance matrix from local TPC to master
43   TVector GetVertex() const { return fVertex;}
44   TVector GetErrorVertex() const { return fErrorVertex;}
45   Long_t  GetLabel() const { return flabel;}
46   void SetLabel(Long_t label) { flabel = label;}
47   Int_t  GetLayer() const { return fLayer;}
48   
49
50   void PutCElements(Double_t C00, Double_t C10, Double_t C11, Double_t C20, Double_t C21, 
51   Double_t C22, Double_t C30, Double_t C31, Double_t C32, Double_t C33, Double_t C40, 
52   Double_t C41, Double_t C42, Double_t C43, Double_t C44);  // put elements of covariance matrix
53   
54   void GetCElements(Double_t &C00, Double_t &C10, Double_t &C11, Double_t &C20, Double_t &C21, 
55   Double_t &C22, Double_t &C30, Double_t &C31, Double_t &C32, Double_t &C33, Double_t &C40, 
56   Double_t &C41, Double_t &C42, Double_t &C43, Double_t &C44) const;  // get elements of covariance matrix
57    
58   void GetXElements(Double_t &X0, Double_t &X1, Double_t &X2, Double_t &X3, Double_t &X4) const; // get elements
59                                                                                            // of state vector
60   void PutXElements(Double_t X0, Double_t X1, Double_t X2, Double_t X3, Double_t X4);  // put elements
61                                                                                        // of state vector
62     
63   void SetLayer(Int_t layer) { fLayer = layer;}      // set current layer
64   AliTPCtrack *GetTPCtrack() const { return fTPCtrack;}    // get hte TPC track
65
66   void PutCluster(Int_t layerc, TVector vecclust);   // put information for clusters
67   TVector GetLabTrack(Int_t lay);                    // get the label of the track
68   void Search(TVector VecTotLabref, Long_t &labref, Int_t &freq); // determine the label and the frequency of
69                                                                   // it for the current track
70   Float_t GetZclusterTrack(Int_t lay) {return ((Float_t) (*fClusterInTrack)(lay,2));} // get the Z
71                              //cohordinate of the cluster gelonging to the track for a given layer
72   void GetClusters();     // prints the clusters belonging to the current track
73   Int_t GetLabTPC() const {return (*fTPCtrack).GetLabel();}  // get the TPC label for the current track
74   Int_t GetIdPoint(Int_t lay) {return ((Int_t) (*fClusterInTrack)(lay,4));}  // get the label identifiing the
75                                                                              //point of the track
76   Int_t GetIdModule(Int_t lay) {return ((Int_t) (*fClusterInTrack)(lay,5));} // get the label identifiing the
77                                                                              // module of the track
78   Float_t GetIdParticle(Int_t lay) {return (*fClusterInTrack)(lay,3);}       // get the label to identify
79                                                                              // the particle      
80   Int_t DoNotCross(Double_t rk) const;  // determine if the track cross a layer 
81   Double_t ArgA(Double_t rk) const;     // quantity usefull in propagation
82   Double_t Arga(Double_t rk) const;     // quantity usefull in propagation
83   Double_t ArgB(Double_t rk) const;     // quantity usefull in propagation
84   Double_t ArgC(Double_t rk) const;     // quantity usefull in propagation             
85   void  Propagation(Double_t rk) ;      // propagation of the track to a layer of radius rk
86
87   Double_t GetSigmaphi() const{return fC00;}    // gets the phi variance
88   Double_t GetSigmaZ() const{return  fC11;}     // gets the Z variance
89   void AddEL(AliITSRad *rl,Double_t signdE,  Bool_t flagtot, Double_t mass=0.1396);  // adds the energy loss
90   void AddMS(AliITSRad *rl);  // modify the covariance matrix to take into account the multiple scattering
91   void Correct(Double_t rk);  // correct the track to take into account the real detector geometry
92   void SetDv(Double_t x) {fDv=x;}  // sets the radial impact parameter for vertex constraint 
93   void SetZv(Double_t x) {fZv=x;}  // sets longitudinal impact parameter for vertex constraint
94   Double_t GetDv() const {return fDv;}   // gets the radial impact parameter for vertex constraint 
95   Double_t GetZv() const {return fZv;}   // gets longitudinal impact parameter for vertex constraint
96   void SetsigmaDv( Double_t x) {fsigmaDv=x;} // sets sigma for Dv extraction
97   void SetsigmaZv( Double_t x) {fsigmaZv=x;} // sets sigma for Zv extraction
98   Double_t GetsigmaDv() const {return fsigmaDv;}   // gets sigma for Dv extraction
99   Double_t GetsigmaZv() const {return fsigmaZv;}   // gets sigma for Zv extraction
100   void PrimaryTrack(AliITSRad *rl);   // calculation of part of covariance matrix for vertex constraint
101   void Setd2(TVector &x) {for(Int_t i=0; i<6; i++){fd2(i)=x(i);}} // sets the vector fd2
102   void Settgl2(TVector &x) {for(Int_t i=0; i<6; i++){ftgl2(i)=x(i);}}  // sets the vector ftgl2
103   void Setdtgl(TVector &x) {for(Int_t i=0; i<6; i++){fdtgl(i)=x(i);}}  // sets the vector fdtgl
104   TVector Getd2() const { return fd2;}  // gets the vector fd2
105   TVector Gettgl2() const { return ftgl2;}  // gets the vector ftgl2
106   TVector Getdtgl() const { return fdtgl;}  // gets the vectoe dtgl
107   Double_t Getd2(Int_t i){return (Double_t)fd2(i);}     // gets the i element of the vector fd2
108   Double_t Gettgl2(Int_t i){return (Double_t)ftgl2(i);} // gets the i element of the vector tgl2
109   Double_t Getdtgl(Int_t i){return (Double_t)fdtgl(i);} // gets the i element of the vector fdtgl
110   Double_t GetxoTPC() const {return fxoTPC;}  // gets fxoTPC
111  
112                         
113 //////////////////////////////////////////////////////////////////////////////////////// 
114
115  private:  
116    
117   AliTPCtrack     *fTPCtrack;           // reference to TPC track
118
119   Double_t        fX0,fX1,fX2,fX3,fX4;  // state vector: |phi/z/D/tgl/C 
120   Double_t        frtrack;               // radius of courrent layer     
121   
122   Double_t        fC00, fC10, fC11,     // Covariance Matrix
123                   fC20, fC21, fC22,     //      "       "
124                                                 fC30, fC31, fC32,     //      "       " 
125                                                 fC33, fC40, fC41,     //      "       " 
126                                                 fC42, fC43, fC44;     //      "       "
127       
128   Double_t        fChi2;                // fChi^2 of track         
129   TObjArray       *flistCluster;        // list of clusters of the track
130   Int_t           fNumClustInTrack;     // total number of clusters
131   Long_t          flabel;               // label of the track
132   TVector         fVertex;              // vertex coordinates of the track
133   TVector         fErrorVertex;         // error on the vertex coordinates
134   Int_t           fLayer;               // current Layer of the track
135   TMatrix        *fClusterInTrack;      // matrix of clusters belonging to the  track
136                                         // row index = layer-1; 
137                                         // cols index = master coordinates of the clusters
138
139   Double_t        fFieldFactor;         // magnetic field factor                                                                                                         
140                                                                                                            
141   Double_t          fDv;                // radial impact parameter for vertex  constraint
142   Double_t          fZv;                // longitudinal impact parameter for vertex constraint
143   Double_t          fsigmaDv;           // sigma for Dv extraction
144   Double_t          fsigmaZv;           // sigma for Zv extraction
145   TVector           fd2;                // C(2,2)  for primary track
146   TVector           ftgl2;              // C(3,3)  for primary track
147   TVector           fdtgl;              // C(2,3)  for primary track
148
149   Double_t          fxoTPC;             // cohordinate xo of the helix center, got from the TPC track 
150                    
151  // Int_t freq; //provvisorio      
152
153  
154   ClassDef(AliITSTrackV1, 1)
155    
156 };
157
158 #endif
159