TRD/AliTRDtrack.h

   1 #ifndef ALITRDTRACK_H
   2 #define ALITRDTRACK_H
   3
   4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 #include <AliKalmanTrack.h>
   8 #include <TMath.h>
   9
  10 #include "AliTRDgeometry.h"
  11 #include "AliESDtrack.h"
  12 #include "TVector2.h"
  13
  14 class AliTRDcluster;
  15 class AliTPCtrack;
  16 class AliESDtrack;
  17
  18 const unsigned kMAXCLUSTERSPERTRACK=210;
  19
  20 class AliTRDtrack : public AliKalmanTrack {
  21
  22   //////////////////////////////////////////////////////////////////
  23   // Represents a reconstructed TRD track                         //
  24   //////////////////////////////////////////////////////////////////
  25
  26   friend class AliTRDtracker;
  27
  28 public:
  29
  30    AliTRDtrack():AliKalmanTrack(){fBackupTrack=0;}
  31    AliTRDtrack(const AliTRDcluster *c, UInt_t index, const Double_t xx[5],
  32                const Double_t cc[15], Double_t xr, Double_t alpha);
  33    AliTRDtrack(const AliTRDtrack& t);
  34    AliTRDtrack(const AliKalmanTrack& t, Double_t alpha);
  35    AliTRDtrack(const AliESDtrack& t);
  36    ~AliTRDtrack();
  37    Int_t    Compare(const TObject *o) const;
  38    void     CookdEdx(Double_t low=0.05, Double_t up=0.55);
  39    Float_t  StatusForTOF();
  40    Double_t GetAlpha() const {return fAlpha;}
  41    Int_t    GetSector() const {
  42      //if (fabs(fAlpha) < AliTRDgeometry::GetAlpha()/2) return 0;
  43      return Int_t(TVector2::Phi_0_2pi(fAlpha)/AliTRDgeometry::GetAlpha())%AliTRDgeometry::kNsect;}
  44
  45    Double_t GetC()     const {return fC;}
  46    Int_t    GetClusterIndex(Int_t i) const {return fIndex[i];}
  47    Float_t  GetClusterdQdl(Int_t i) const {return fdQdl[i];}
  48
  49    void     GetCovariance(Double_t cov[15]) const;
  50    Double_t GetdEdx()  const {return fdEdx;}
  51    Double_t GetPIDsignal()  const {return GetdEdx();}
  52    Float_t  GetPIDsignals(Int_t i) const {return fdEdxPlane[i];}
  53    Int_t    GetPIDTimBin(Int_t i) const {return fTimBinPlane[i];}
  54    Double_t GetEta()   const {return fE;}
  55
  56    void     GetExternalCovariance(Double_t cov[15]) const ;
  57    void     GetExternalParameters(Double_t& xr, Double_t x[5]) const ;
  58
  59    Double_t GetLikelihoodElectron() const { return fLhElectron; };
  60
  61    Double_t Get1Pt()   const {return (1e-9*TMath::Abs(fC)/fC + fC)*GetConvConst(); }
  62    Double_t GetP()     const {
  63      return TMath::Abs(GetPt())*sqrt(1.+GetTgl()*GetTgl());
  64    }
  65    virtual  Double_t GetPredictedChi2(const AliCluster*) const { return 0.0; };
  66    Double_t GetPredictedChi2(const AliTRDcluster *c, Double_t h01) const ;
  67    Double_t GetPt()    const {return 1./Get1Pt();}
  68    void     GetPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const ;
  69    void     GetGlobalXYZ(Double_t &x, Double_t &y, Double_t &z) const ;
  70    Int_t    GetSeedLabel() const { return fSeedLab; }
  71    Double_t GetSigmaC2()   const {return fCcc;}
  72    Double_t GetSigmaTgl2() const {return fCtt;}
  73    Double_t GetSigmaY2()   const {return fCyy;}
  74    Double_t GetSigmaZ2()   const {return fCzz;}
  75    Double_t GetSnp()  const {return fX*fC - fE;}
  76    Double_t GetTgl()  const {return fT;}
  77    Double_t GetX()    const {return fX;}
  78    Double_t GetY()    const {return fY;}
  79    Double_t GetZ()    const {return fZ;}
  80    UInt_t * GetBackupIndexes()  {return fIndexBackup;}
  81    UInt_t * GetIndexes()  {return fIndex;}
  82    //-----------------------------------------------------------------
  83    // This function calculates the Y-coordinate of a track at the plane x=xk.
  84    // Needed for matching with the TOF (I.Belikov)
  85    Double_t GetYat(Double_t xk) const {
  86       Double_t c1=fC*fX - fE, r1=TMath::Sqrt(1.- c1*c1);
  87       Double_t c2=fC*xk - fE, r2=TMath::Sqrt(1.- c2*c2);
  88       return fY + (xk-fX)*(c1+c2)/(r1+r2);
  89    }
  90    //-----------------------------------------------------------------
  91    Int_t    GetProlongation(Double_t xk, Double_t &y, Double_t &z);
  92
  93    void     SetStop(Bool_t stop) {fStopped=stop;}
  94    Bool_t   GetStop() const {return fStopped;}
  95
  96    Int_t    PropagateTo(Double_t xr, Double_t x0=8.72, Double_t rho=5.86e-3);
  97    void     ResetCovariance();
  98    void     ResetCovariance(Float_t mult);
  99    void     ResetClusters() { SetChi2(0.); SetNumberOfClusters(0); }
 100    Int_t    Rotate(Double_t angle);
 101
 102    void     SetdEdx(Float_t dedx) {fdEdx=dedx;}
 103    void     SetPIDsignals(Float_t dedx, Int_t i) {fdEdxPlane[i]=dedx;}
 104    void     SetPIDTimBin(Int_t timbin, Int_t i) {fTimBinPlane[i]=timbin;}
 105    void     SetLikelihoodElectron(Float_t l) { fLhElectron = l; };
 106
 107    void     SetSampledEdx(Float_t q, Int_t i) {
 108                Double_t s=GetSnp(), t=GetTgl();
 109                q*= TMath::Sqrt((1-s*s)/(1+t*t));
 110                fdQdl[i]=q;
 111             }
 112    void     SetSampledEdx(Float_t q) {
 113               Double_t s=GetSnp(), t=GetTgl();
 114               q*= TMath::Sqrt((1-s*s)/(1+t*t));
 115               fdQdl[fNdedx]=q;
 116               fNdedx++;
 117             }
 118
 119    void     SetSeedLabel(Int_t lab) { fSeedLab=lab; }
 120
 121    Int_t    Update(const AliCluster*, Double_t, UInt_t) { return 0; };
 122    Int_t    Update(const AliTRDcluster* c, Double_t chi2, UInt_t i,
 123                    Double_t h01);
 124    Int_t    UpdateMI(const AliTRDcluster* c, Double_t chi2, UInt_t i,
 125                    Double_t h01, Int_t plane);
 126
 127   void      AddNWrong() {fNWrong++;}
 128
 129   Int_t     GetNWrong() const {return fNWrong;}
 130   Int_t     GetNRotate() const {return fNRotate;}
 131   Int_t     GetNCross() const {return fNCross;}
 132   void      IncCross() {fNCross++; if (fBackupTrack) fBackupTrack->IncCross();}
 133   AliTRDtrack *  GetBackupTrack(){return fBackupTrack;}
 134   void      MakeBackupTrack();
 135
 136 protected:
 137
 138    Int_t    fSeedLab;               // track label taken from seeding
 139    Float_t  fdEdx;                  // dE/dx
 140    Float_t  fdEdxPlane[kNPlane];    // dE/dx from all 6 planes
 141    Int_t    fTimBinPlane[kNPlane];  // time bin of Max cluster from all 6 planes
 142
 143    Double_t fAlpha;         // rotation angle
 144    Double_t fX;             // running local X-coordinate of the track (time bin)
 145    Bool_t   fStopped;       // track stop indication
 146
 147    Double_t fY;             // Y-coordinate of the track
 148    Double_t fZ;             // Z-coordinate of the track
 149    Double_t fE;             // C*x0
 150    Double_t fT;             // tangent of the track momentum dip angle
 151    Double_t fC;             // track curvature
 152
 153    Double_t fCyy;                         // covariance
 154    Double_t fCzy, fCzz;                   // matrix
 155    Double_t fCey, fCez, fCee;             // of the
 156    Double_t fCty, fCtz, fCte, fCtt;       // track
 157    Double_t fCcy, fCcz, fCce, fCct, fCcc; // parameters
 158
 159    UInt_t  fIndex[kMAXCLUSTERSPERTRACK];       // global indexes of clusters
 160    UInt_t  fIndexBackup[kMAXCLUSTERSPERTRACK]; // backup indexes of clusters - used in iterations
 161    Float_t fdQdl[kMAXCLUSTERSPERTRACK];        // cluster amplitudes corrected
 162                                                // for track angles
 163
 164    Float_t fLhElectron;    // Likelihood to be an electron
 165    Int_t   fNWrong;        // number of wrong clusters
 166    Int_t   fNRotate;       // number of rotation
 167    Int_t   fNCross;        // number of the cross materials
 168    Int_t   fNExpected;     // expected number of cluster
 169    Int_t   fNLast;         // number of clusters in last 2 layers
 170    Int_t   fNExpectedLast; // number of expected clusters on last 2 layers
 171    Int_t   fNdedx;         // number of clusters for dEdx measurment
 172    Float_t fChi2Last;      // chi2 in the  last 2 layers
 173
 174    AliTRDtrack * fBackupTrack; //! backup track
 175
 176    ClassDef(AliTRDtrack,2) // TRD reconstructed tracks
 177
 178 };
 179
 180
 181 #endif