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 #include <AliKalmanTrack.h>
  14 #include "AliITSRecoParam.h"
  15 #include "AliITSgeomTGeo.h"
  16
  17 class AliESDtrack;
  18
  19 //_____________________________________________________________________________
  20 class AliITStrackV2 : public AliKalmanTrack {
  21 public:
  22   AliITStrackV2();
  23   AliITStrackV2(AliESDtrack& t,Bool_t c=kFALSE) throw (const Char_t *);
  24   AliITStrackV2(const AliITStrackV2& t);
  25   ~AliITStrackV2(){fESDtrack=0;}
  26
  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
  33     return AliExternalTrackParam::CorrectForMaterial(d,x0,GetMass());
  34   }
  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);
  39   }
  40   Double_t GetPredictedChi2(const AliCluster *cluster) const;
  41   Bool_t Update(const AliCluster *cl, Double_t chi2, Int_t i);
  42
  43   Bool_t PropagateToVertex(const AliESDVertex *v,Double_t d=0.,Double_t x0=0.);
  44   Bool_t Propagate(Double_t alpha, Double_t xr);
  45   Bool_t MeanBudgetToPrimVertex(Double_t xyz[3], Double_t step, Double_t &d) const;
  46   Bool_t Improve(Double_t x0,Double_t xyz[3],Double_t ers[3]);
  47
  48   void SetdEdx(Double_t dedx) {fdEdx=dedx;}
  49   void SetSampledEdx(Float_t q, Int_t i);
  50   Float_t GetSampledEdx(Int_t i) const {return fdEdxSample[i];}
  51   void CookdEdx(Double_t low=0., Double_t up=0.51);
  52   void SetDetectorIndex(Int_t i) {SetLabel(i);}
  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 GetdEdx() const {return fdEdx;}
  60   Double_t GetPIDsignal() const {return GetdEdx();}
  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
  69   Bool_t GetGlobalXYZat(Double_t xloc,Double_t &x,Double_t &y,Double_t &z) const;
  70   Bool_t GetPhiZat(Double_t r,Double_t &phi,Double_t &z) const;
  71   Bool_t GetLocalXat(Double_t r,Double_t &xloc) const;
  72
  73   Int_t Compare(const TObject *o) const;
  74   Int_t GetClusterIndex(Int_t i) const {return fIndex[i];}
  75   Bool_t Invariant() const;
  76
  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];}
  79
  80 protected:
  81   Double_t GetBz() const ;
  82   Double_t fdEdx;            // dE/dx
  83
  84   static const Int_t fgkWARN; //! used for debugging purposes
  85   Float_t fdEdxSample[4];    // array of dE/dx samples b.b.
  86
  87   Int_t fIndex[2*AliITSgeomTGeo::kNLayers]; // indices of associated clusters
  88
  89   AliESDtrack *fESDtrack;    //! pointer to the connected ESD track
  90
  91 private:
  92   AliITStrackV2 &operator=(const AliITStrackV2 &tr);
  93   ClassDef(AliITStrackV2,7)  //ITS reconstructed track
  94 };
  95
  96 inline
  97 void AliITStrackV2::SetSampledEdx(Float_t q, Int_t i) {
  98   //----------------------------------------------------------------------
  99   // This function stores dEdx sample corrected for the track segment length
 100   // Origin: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
 101   //----------------------------------------------------------------------
 102   if (i<0) return;
 103   if (i>3) return;
 104   Double_t s=GetSnp(), t=GetTgl();
 105   q *= TMath::Sqrt((1-s*s)/(1+t*t));
 106   fdEdxSample[i]=q;
 107 }
 108
 109 #endif
 110
 111