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 */
6 //-------------------------------------------------------------------------
9 // Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
10 // dEdx analysis by: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
11 //-------------------------------------------------------------------------
13 #include <AliKalmanTrack.h>
14 #include "AliITSRecoParam.h"
15 #include "AliITSgeomTGeo.h"
23 //_____________________________________________________________________________
24 class AliITStrackV2 : public AliKalmanTrack {
27 AliITStrackV2(AliESDtrack& t,Bool_t c=kFALSE) throw (const Char_t *);
28 AliITStrackV2(const AliITStrackV2& t);
29 ~AliITStrackV2(){fESDtrack=0;}
31 void SetCheckInvariant(Bool_t check=kTRUE) {fCheckInvariant=check;}
32 Bool_t CorrectForMeanMaterial(Double_t xOverX0, Double_t xTimesRho,
33 Bool_t anglecorr=kFALSE) {
34 return AliExternalTrackParam::CorrectForMeanMaterial(xOverX0,xTimesRho,GetMass(),anglecorr);
36 Bool_t CorrectForMaterial(Double_t d, Double_t x0=AliITSRecoParam::GetX0Air()) {
37 // deprecated: use CorrectForMeanMaterial instead
38 return AliExternalTrackParam::CorrectForMaterial(d,x0,GetMass());
40 Bool_t PropagateTo(Double_t xr, Double_t d, Double_t x0=AliITSRecoParam::GetX0Air());
41 Bool_t PropagateToTGeo(Double_t xToGo, Int_t nstep, Double_t &xOverX0, Double_t &xTimesRho, Bool_t addTime=kTRUE);
42 Bool_t PropagateToTGeo(Double_t xToGo, Int_t nstep=1, Bool_t addTime=kTRUE) {
43 Double_t dummy1,dummy2; return PropagateToTGeo(xToGo,nstep,dummy1,dummy2,addTime);
45 Double_t GetPredictedChi2(const AliCluster *cluster) const;
46 Bool_t Update(const AliCluster *cl, Double_t chi2, Int_t i);
48 Bool_t PropagateToVertex(const AliESDVertex *v,Double_t d=0.,Double_t x0=0.);
49 Bool_t Propagate(Double_t alpha, Double_t xr);
50 Bool_t Propagate(Double_t xr) { return Propagate(GetAlpha(),xr); }
51 Bool_t MeanBudgetToPrimVertex(Double_t xyz[3], Double_t step, Double_t &d) const;
52 Bool_t Improve(Double_t x0,Double_t xyz[3],Double_t ers[3]);
54 void SetdEdx(Double_t dedx) {fdEdx=dedx;}
55 void SetSampledEdx(Float_t q, Int_t i);
56 Float_t GetSampledEdx(Int_t i) const {return fdEdxSample[i];}
57 void CookdEdx(Double_t low=0., Double_t up=0.51);
58 void SetDetectorIndex(Int_t i) {SetLabel(i);}
60 void UpdateESDtrack(ULong_t flags) const;
62 AliESDtrack *GetESDtrack() const {return fESDtrack;}
64 Int_t GetDetectorIndex() const {return GetLabel();}
65 Double_t GetdEdx() const {return fdEdx;}
66 Double_t GetPIDsignal() const {return GetdEdx();}
67 using AliExternalTrackParam::GetC;
68 Double_t GetC() const {return AliExternalTrackParam::GetC(GetBz());}
69 Double_t GetD(Double_t x, Double_t y) const {
70 return AliExternalTrackParam::GetD(x,y,GetBz());
72 void GetDZ(Double_t xv, Double_t yv, Double_t zv, Float_t dz[2]) const {
73 return AliExternalTrackParam::GetDZ(xv,yv,zv,GetBz(),dz);
76 Bool_t GetGlobalXYZat(Double_t xloc,Double_t &x,Double_t &y,Double_t &z) const;
77 Bool_t GetPhiZat(Double_t r,Double_t &phi,Double_t &z) const;
78 Bool_t GetLocalXat(Double_t r,Double_t &xloc) const;
80 Int_t Compare(const TObject *o) const;
81 Int_t GetClusterIndex(Int_t i) const {return fIndex[i];}
82 void SetModuleIndex(Int_t ilayer,Int_t idx) {fModule[ilayer]=idx;}
83 Int_t GetModuleIndex(Int_t ilayer) const {return fModule[ilayer];}
84 void SetModuleIndexInfo(Int_t ilayer,Int_t idet,Int_t status=1,Float_t xloc=0,Float_t zloc=0);
85 Bool_t GetModuleIndexInfo(Int_t ilayer,Int_t &idet,Int_t &status,Float_t &xloc,Float_t &zloc) const;
86 Bool_t Invariant() const;
88 void SetExtraCluster(Int_t ilayer, Int_t idx) {fIndex[AliITSgeomTGeo::kNLayers+ilayer]=idx;}
89 Int_t GetExtraCluster(Int_t ilayer) const {return fIndex[AliITSgeomTGeo::kNLayers+ilayer];}
91 void SetExtraModule(Int_t ilayer, Int_t idx) {fModule[AliITSgeomTGeo::kNLayers+ilayer]=idx;}
92 Int_t GetExtraModule(Int_t ilayer) const {return fModule[AliITSgeomTGeo::kNLayers+ilayer];}
96 Bool_t fCheckInvariant; // check the covariance matrix
98 Double_t fdEdx; // dE/dx
100 static const Int_t fgkWARN; //! used for debugging purposes
101 Float_t fdEdxSample[4]; // array of dE/dx samples b.b.
103 Int_t fIndex[2*AliITSgeomTGeo::kNLayers]; // indices of associated clusters
105 Int_t fModule[2*AliITSgeomTGeo::kNLayers]; // indices of crossed modules:
106 // see SetModuleIndexInfo()
107 AliESDtrack *fESDtrack; //! pointer to the connected ESD track
110 AliITStrackV2 &operator=(const AliITStrackV2 &tr);
111 ClassDef(AliITStrackV2,8) //ITS reconstructed track
114 inline void AliITStrackV2::SetSampledEdx(Float_t q, Int_t i) {
115 //----------------------------------------------------------------------
116 // This function stores dEdx sample corrected for the track segment length
117 // Origin: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
118 //----------------------------------------------------------------------
121 Double_t s=GetSnp(), t=GetTgl();
122 q *= TMath::Sqrt((1-s*s)/(1+t*t));
127 inline void AliITStrackV2::SetModuleIndexInfo(Int_t ilayer,Int_t idet,Int_t status,
128 Float_t xloc,Float_t zloc) {
129 //----------------------------------------------------------------------
130 // This function encodes in the module number also the status of cluster association
131 // "status" can have the following values:
132 // 1 "found" (cluster is associated),
133 // 2 "dead" (module is dead from OCDB),
134 // 3 "skipped" (module or layer forced to be skipped),
135 // 4 "outinz" (track out of z acceptance),
136 // 5 "nocls" (no clusters in the road),
137 // 6 "norefit" (cluster rejected during refit)
138 // 7 "deadzspd" (holes in z in SPD)
139 // WARNING: THIS METHOD HAS TO BE SYNCHRONIZED WITH AliESDtrack::GetITSModuleIndexInfo()!
140 //----------------------------------------------------------------------
145 // same detector numbering as in AliITSCalib classes
146 if(ilayer==1) idet+=AliITSgeomTGeo::GetNLadders(1)*AliITSgeomTGeo::GetNDetectors(1);
147 if(ilayer==3) idet+=AliITSgeomTGeo::GetNLadders(3)*AliITSgeomTGeo::GetNDetectors(3);
148 if(ilayer==5) idet+=AliITSgeomTGeo::GetNLadders(5)*AliITSgeomTGeo::GetNDetectors(5);
151 Int_t xInt = Int_t(xloc*10.);
152 Int_t zInt = Int_t(zloc*10.);
154 if(TMath::Abs(xloc*10.-(Float_t)xInt)>0.5){
162 if(TMath::Abs(zloc*10.-(Float_t)zInt)>0.5){
171 if(xInt>=0 && zInt>=0) signs=10000;
172 if(xInt>=0 && zInt<0) signs=20000;
173 if(xInt<0 && zInt>=0) signs=30000;
174 if(xInt<0 && zInt<0) signs=40000;
176 Int_t modindex = signs;
178 modindex += TMath::Abs(zInt);
179 modindex += TMath::Abs(xInt)*100;
181 modindex += status*100000;
183 modindex += idet*1000000;
185 SetModuleIndex(ilayer,modindex);
189 inline Bool_t AliITStrackV2::GetModuleIndexInfo(Int_t ilayer,Int_t &idet,Int_t &status,
190 Float_t &xloc,Float_t &zloc) const {
191 //----------------------------------------------------------------------
192 // This function encodes in the module number also the status of cluster association
193 // "status" can have the following values:
194 // 1 "found" (cluster is associated),
195 // 2 "dead" (module is dead from OCDB),
196 // 3 "skipped" (module or layer forced to be skipped),
197 // 4 "outinz" (track out of z acceptance),
198 // 5 "nocls" (no clusters in the road),
199 // 6 "norefit" (cluster rejected during refit),
200 // 7 "deadzspd" (holes in z in SPD)
201 // Also given are the coordinates of the crossing point of track and module
202 // (in the local module ref. system)
203 // WARNING: THIS METHOD HAS TO BE SYNCHRONIZED WITH AliESDtrack::GetITSModuleIndexInfo()!
204 //----------------------------------------------------------------------
207 if(fModule[ilayer]==-1) {
208 AliError("fModule was not set !");
211 xloc=-99.; zloc=-99.;
215 Int_t module = fModule[ilayer];
217 idet = Int_t(module/1000000);
219 module -= idet*1000000;
221 status = Int_t(module/100000);
223 module -= status*100000;
225 Int_t signs = Int_t(module/10000);
229 Int_t xInt = Int_t(module/100);
234 if(signs==1) { xInt*=1; zInt*=1; }
235 if(signs==2) { xInt*=1; zInt*=-1; }
236 if(signs==3) { xInt*=-1; zInt*=1; }
237 if(signs==4) { xInt*=-1; zInt*=-1; }
239 xloc = 0.1*(Float_t)xInt;
240 zloc = 0.1*(Float_t)zInt;
242 if(status==4) idet = -1;