new bugs fixed by Tomek
[u/mrichter/AliRoot.git] / ITS / AliITStrackV2.h
CommitLineData
006b5f7f 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
a9a2d814 10// dEdx analysis by: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
006b5f7f 11//-------------------------------------------------------------------------
12
006b5f7f 13#include <AliKalmanTrack.h>
44347160 14#include "AliITSRecoParam.h"
e50912db 15#include "AliITSgeomTGeo.h"
006b5f7f 16
83d73500 17class AliESDtrack;
006b5f7f 18
19//_____________________________________________________________________________
8676d691 20class AliITStrackV2 : public AliKalmanTrack {
006b5f7f 21public:
22b13da0 22 AliITStrackV2();
67c3dcbe 23 AliITStrackV2(AliESDtrack& t,Bool_t c=kFALSE) throw (const Char_t *);
006b5f7f 24 AliITStrackV2(const AliITStrackV2& t);
43d2197f 25 ~AliITStrackV2(){fESDtrack=0;}
6c94f330 26
e50912db 27 Bool_t CorrectForMeanMaterial(Double_t xOverX0, Double_t xTimesRho,
28 Bool_t anglecorr=kFALSE) {
29 return AliExternalTrackParam::CorrectForMeanMaterial(xOverX0,xTimesRho,GetMass(),anglecorr);
30 }
31 Bool_t CorrectForMaterial(Double_t d, Double_t x0=AliITSRecoParam::GetX0Air()) {
32 // deprecated: use CorrectForMeanMaterial instead
6c94f330 33 return AliExternalTrackParam::CorrectForMaterial(d,x0,GetMass());
34 }
e50912db 35 Bool_t PropagateTo(Double_t xr, Double_t d, Double_t x0=AliITSRecoParam::GetX0Air());
36 Bool_t PropagateToTGeo(Double_t xToGo, Int_t nstep, Double_t &xOverX0, Double_t &xTimesRho);
37 Bool_t PropagateToTGeo(Double_t xToGo, Int_t nstep=1) {
38 Double_t dummy1,dummy2; return PropagateToTGeo(xToGo,nstep,dummy1,dummy2);
afd25725 39 }
6c94f330 40 Double_t GetPredictedChi2(const AliCluster *cluster) const;
41 Bool_t Update(const AliCluster *cl, Double_t chi2, Int_t i);
42
e50912db 43 Bool_t PropagateToVertex(const AliESDVertex *v,Double_t d=0.,Double_t x0=0.);
6c94f330 44 Bool_t Propagate(Double_t alpha, Double_t xr);
afd25725 45 Bool_t MeanBudgetToPrimVertex(Double_t xyz[3], Double_t step, Double_t &d) const;
6c94f330 46 Bool_t Improve(Double_t x0,Double_t xyz[3],Double_t ers[3]);
47
7f6ddf58 48 void SetdEdx(Double_t dedx) {fdEdx=dedx;}
a9a2d814 49 void SetSampledEdx(Float_t q, Int_t i);
b9671574 50 Float_t GetSampledEdx(Int_t i) const {return fdEdxSample[i];}
a9a2d814 51 void CookdEdx(Double_t low=0., Double_t up=0.51);
006b5f7f 52 void SetDetectorIndex(Int_t i) {SetLabel(i);}
7f6ddf58 53 void ResetClusters() { SetChi2(0.); SetNumberOfClusters(0); }
15dd636f 54 void UpdateESDtrack(ULong_t flags) const;
e43c066c 55
1aedd96e 56 AliESDtrack *GetESDtrack() const {return fESDtrack;}
e43c066c 57
006b5f7f 58 Int_t GetDetectorIndex() const {return GetLabel();}
7f6ddf58 59 Double_t GetdEdx() const {return fdEdx;}
83d73500 60 Double_t GetPIDsignal() const {return GetdEdx();}
6c94f330 61 Double_t GetC() const {return AliExternalTrackParam::GetC(GetBz());}
62 Double_t GetD(Double_t x, Double_t y) const {
63 return AliExternalTrackParam::GetD(x,y,GetBz());
64 }
65 void GetDZ(Double_t xv, Double_t yv, Double_t zv, Float_t dz[2]) const {
66 return AliExternalTrackParam::GetDZ(xv,yv,zv,GetBz(),dz);
67 }
68
e50912db 69 Bool_t GetGlobalXYZat(Double_t xloc,Double_t &x,Double_t &y,Double_t &z) const;
8602c008 70 Bool_t GetPhiZat(Double_t r,Double_t &phi,Double_t &z) const;
e50912db 71 Bool_t GetLocalXat(Double_t r,Double_t &xloc) const;
6c94f330 72
006b5f7f 73 Int_t Compare(const TObject *o) const;
006b5f7f 74 Int_t GetClusterIndex(Int_t i) const {return fIndex[i];}
6c94f330 75 Bool_t Invariant() const;
ef7253ac 76
e50912db 77 void SetExtraCluster(Int_t i, Int_t idx) {fIndex[AliITSgeomTGeo::kNLayers+i]=idx;}
78 Int_t GetExtraCluster(Int_t i) const {return fIndex[AliITSgeomTGeo::kNLayers+i];}
c84a5e9e 79
6c94f330 80protected:
81 Double_t GetBz() const ;
82 Double_t fdEdx; // dE/dx
8602c008 83
84 static const Int_t fgkWARN; //! used for debugging purposes
6c94f330 85 Float_t fdEdxSample[4]; // array of dE/dx samples b.b.
15dd636f 86
e50912db 87 Int_t fIndex[2*AliITSgeomTGeo::kNLayers]; // indices of associated clusters
15dd636f 88
ef7253ac 89 AliESDtrack *fESDtrack; //! pointer to the connected ESD track
006b5f7f 90
6c94f330 91private:
92 AliITStrackV2 &operator=(const AliITStrackV2 &tr);
8602c008 93 ClassDef(AliITStrackV2,7) //ITS reconstructed track
6c94f330 94};
a9a2d814 95
23efe5f1 96inline
97void AliITStrackV2::SetSampledEdx(Float_t q, Int_t i) {
98 //----------------------------------------------------------------------
a9a2d814 99 // This function stores dEdx sample corrected for the track segment length
100 // Origin: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
23efe5f1 101 //----------------------------------------------------------------------
a9a2d814 102 if (i<0) return;
103 if (i>3) return;
23efe5f1 104 Double_t s=GetSnp(), t=GetTgl();
23efe5f1 105 q *= TMath::Sqrt((1-s*s)/(1+t*t));
106 fdEdxSample[i]=q;
23efe5f1 107}
c84a5e9e 108
006b5f7f 109#endif
110
111