TRD/AliTRDtrackV1.h

   1 #ifndef ALITRDTRACKV1_H
   2 #define ALITRDTRACKV1_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 ///////////////////////////////////////////////////////////////////////////////
   9 //                                                                           //
  10 //  Represents a reconstructed TRD track                                     //
  11 //                                                                           //
  12 ///////////////////////////////////////////////////////////////////////////////
  13
  14 //#ifndef ALIKALMANTRACK_H
  15 #include "AliKalmanTrack.h"
  16 //#endif
  17
  18 //#ifndef ALITRDSEEDV1_H
  19 #include "AliTRDseedV1.h"
  20 //#endif
  21
  22 class AliTRDcluster;
  23 class AliESDtrack;
  24 class AliTRDReconstructor;
  25 class AliTRDtrackV1 : public AliKalmanTrack
  26 {
  27   friend class AliHLTTRDTrack;
  28 public:
  29   enum ETRDtrackSize {
  30     kNdet      = AliTRDgeometry::kNdet
  31    ,kNstacks   = AliTRDgeometry::kNstack*AliTRDgeometry::kNsector
  32    ,kNplane    = AliTRDgeometry::kNlayer
  33    ,kNcham     = AliTRDgeometry::kNstack
  34    ,kNsect     = AliTRDgeometry::kNsector
  35    ,kNslice    =   3
  36    ,kNMLPslice =   8
  37    ,kMAXCLUSTERSPERTRACK = 210
  38   };
  39
  40   // bits from 0-13 are reserved by ROOT (see TObject.h)
  41   enum ETRDtrackStatus {
  42     kOwner     = BIT(14)
  43    ,kStopped   = BIT(15)
  44    ,kKink      = BIT(16)
  45   };
  46
  47   // propagation/update error codes (up to 4 bits)
  48   enum ETRDtrackError {
  49     kProlongation = 1
  50    ,kPropagation
  51    ,kAdjustSector
  52    ,kSnp
  53    ,kTrackletInit
  54    ,kUpdate
  55    ,kUnknown      = 0xff
  56   };
  57
  58   // data/clusters/tracklet error codes (up to 4 bits/layer)
  59   enum ETRDlayerError {
  60     kGeometry = 1
  61    ,kBoundary
  62    ,kNoClusters
  63    ,kNoAttach
  64    ,kNoClustersTracklet
  65    ,kNoFit
  66    ,kChi2
  67   };
  68
  69   AliTRDtrackV1();
  70   AliTRDtrackV1(AliTRDseedV1 * const trklts, const Double_t p[5], const Double_t cov[15], Double_t x, Double_t alpha);
  71   AliTRDtrackV1(const AliESDtrack &ref);
  72   AliTRDtrackV1(const AliTRDtrackV1 &ref);
  73   virtual ~AliTRDtrackV1();
  74   AliTRDtrackV1 &operator=(const AliTRDtrackV1 &ref) { *(new(this) AliTRDtrackV1(ref)); return *this; }
  75
  76   Bool_t         CookPID();
  77   Bool_t         CookLabel(Float_t wrong);
  78   AliTRDtrackV1* GetBackupTrack() const {return fBackupTrack;}
  79   Double_t       GetBudget(Int_t i) const { return fBudget[i];}
  80   AliTRDcluster* GetCluster(Int_t id);
  81   Int_t          GetClusterIndex(Int_t id) const;
  82   Float_t        GetEdep() const {return fDE;}
  83   Int_t          GetESDid() const {return fESDid;}
  84   inline Float_t GetMomentum(Int_t plane) const;
  85   inline Int_t   GetNCross();
  86   inline Int_t   GetNumberOfTracklets() const;
  87   Double_t       GetPIDsignal() const   { return 0.;}
  88   Double_t       GetPID(Int_t is) const { return (is >=0 && is < AliPID::kSPECIES) ? fPID[is] : -1.;}
  89   UChar_t        GetNumberOfTrackletsPID() const;
  90   Double_t       GetPredictedChi2(const AliTRDseedV1 *tracklet, Double_t *cov) const;
  91   Double_t       GetPredictedChi2(const AliCluster* /*c*/) const                   { return 0.0; }
  92   Int_t          GetProlongation(Double_t xk, Double_t &y, Double_t &z);
  93   inline UChar_t GetStatusTRD(Int_t ly=-1) const;
  94   Int_t          GetSector() const;
  95   AliTRDseedV1*  GetTracklet(Int_t plane) const {return plane >=0 && plane <kNplane ? fTracklet[plane] : NULL;}
  96   Int_t          GetTrackletIndex(Int_t plane) const          { return (plane>=0 && plane<kNplane) ? fTrackletIndex[plane] : -1;}
  97   AliExternalTrackParam*
  98                  GetTrackLow() const  { return fTrackLow;}
  99   AliExternalTrackParam*
 100                  GetTrackHigh() const  { return fTrackHigh;}
 101   const Int_t* GetTrackletIndexes() const { return &fTrackletIndex[0];}
 102
 103   Bool_t         IsEqual(const TObject *inTrack) const;
 104   Bool_t         IsKink() const    { return TestBit(kKink);}
 105   Bool_t         IsOwner() const   { return TestBit(kOwner);};
 106   Bool_t         IsStopped() const { return TestBit(kStopped);};
 107   Bool_t         IsElectron() const;
 108   inline static Bool_t IsTrackError(ETRDtrackError error, UInt_t status);
 109   inline static Bool_t IsLayerError(ETRDlayerError error, Int_t layer, UInt_t status);
 110
 111   void           MakeBackupTrack();
 112   void           Print(Option_t *o="") const;
 113
 114   Bool_t         PropagateTo(Double_t xr, Double_t x0 = 8.72, Double_t rho = 5.86e-3);
 115   Int_t          PropagateToR(Double_t xr, Double_t step);
 116   Bool_t         Rotate(Double_t angle, Bool_t absolute = kFALSE);
 117   void           SetBudget(Int_t i, Double_t b) {if(i>=0 && i<3) fBudget[i] = b;}
 118   void           SetEdep(Double32_t inDE){fDE = inDE;};
 119   void           SetESDid(Int_t id) {fESDid = id;}
 120   void           SetKink(Bool_t k)        { SetBit(kKink, k);}
 121   void           SetNumberOfClusters();
 122   UChar_t        SetNumberOfTrackletsPID(Bool_t recalc);
 123   void           SetOwner();
 124   void           SetPID(Short_t is, Double_t inPID){if (is >=0 && is < AliPID::kSPECIES) fPID[is]=inPID;};
 125   void           SetPIDquality(UChar_t /*inPIDquality*/) const {/*fPIDquality = inPIDquality*/;};
 126   inline void    SetStatus(UChar_t stat, Int_t ly=-1);
 127   void           SetStopped(Bool_t stop) {SetBit(kStopped, stop);}
 128   void           SetTracklet(AliTRDseedV1 *const trklt,  Int_t index);
 129   void           SetTrackLow();
 130   void           SetTrackHigh(const AliExternalTrackParam *op=NULL);
 131   inline void    SetReconstructor(const AliTRDReconstructor *rec);
 132   inline Float_t StatusForTOF();
 133   void           UnsetTracklet(Int_t plane);
 134   Bool_t         Update(Double_t *p, Double_t *cov, Double_t chi2);
 135   Bool_t         Update(const AliCluster *, Double_t, Int_t) { return kFALSE; };
 136   void           UpdateESDtrack(AliESDtrack *t);
 137
 138 private:
 139   UInt_t       fStatus;                //  Bit map for the status of propagation
 140   Int_t        fTrackletIndex[kNplane];//  Tracklets index in the tracker list
 141   Int_t        fESDid;                 //  ESD track id
 142   Double32_t   fPID[AliPID::kSPECIES]; //  PID probabilities
 143   Double32_t   fBudget[3];             //  Integrated material budget
 144   Double32_t   fDE;                    //  Integrated delta energy
 145   const AliTRDReconstructor *fkReconstructor;//! reconstructor link
 146   AliTRDtrackV1 *fBackupTrack;         //! Backup track
 147   AliTRDseedV1  *fTracklet[kNplane];   //  Tracklets array defining the track
 148   AliExternalTrackParam *fTrackLow;    // parameters of the track which enter TRD from below (TPC)
 149   AliExternalTrackParam *fTrackHigh;   // parameters of the track which enter TRD from above (HMPID, PHOS)
 150
 151   ClassDef(AliTRDtrackV1, 7)          // TRD track - tracklet based
 152 };
 153
 154 //____________________________________________________
 155 inline Float_t AliTRDtrackV1::GetMomentum(Int_t plane) const
 156 {
 157   return plane >=0 && plane < kNplane && fTrackletIndex[plane] != -1 ? fTracklet[plane]->GetMomentum() : -1.;
 158 }
 159
 160 //____________________________________________________
 161 inline Int_t AliTRDtrackV1::GetNCross()
 162 {
 163   Int_t ncross = 0;
 164   for(Int_t ip=0; ip<kNplane; ip++){
 165     if(!fTracklet[ip]) continue;
 166     ncross += fTracklet[ip]->IsRowCross();
 167   }
 168   return ncross;
 169 }
 170
 171 //____________________________________________________
 172 inline Int_t AliTRDtrackV1::GetNumberOfTracklets() const
 173 {
 174   Int_t n = 0;
 175   for(Int_t ip=0; ip<kNplane; ip++){
 176     if(!fTracklet[ip]) continue;
 177     n++;
 178   }
 179   return n;
 180 }
 181
 182 //____________________________________________________
 183 inline UChar_t AliTRDtrackV1::GetStatusTRD(Int_t ly) const
 184 {
 185   if(ly>=-1 && ly<kNplane) return (fStatus>>((ly+1)*4))&0xf;
 186   return kUnknown;
 187 }
 188
 189 //____________________________________________________
 190 inline Bool_t AliTRDtrackV1::IsTrackError(ETRDtrackError error, UInt_t status)
 191 {
 192   return (status&0xf)==UChar_t(error);
 193 }
 194
 195 //____________________________________________________
 196 inline Bool_t AliTRDtrackV1::IsLayerError(ETRDlayerError error, Int_t ly, UInt_t status)
 197 {
 198   if(ly>=kNplane || ly<0) return kFALSE;
 199   return ((status>>((ly+1)*4))&0xf) == UChar_t(error);
 200 }
 201
 202 //____________________________________________________
 203 inline void AliTRDtrackV1::SetReconstructor(const AliTRDReconstructor *rec)
 204 {
 205   for(Int_t ip=0; ip<kNplane; ip++){
 206     if(!fTracklet[ip]) continue;
 207     fTracklet[ip]->SetReconstructor(rec);
 208   }
 209   fkReconstructor = rec;
 210 }
 211
 212 //____________________________________________________
 213 inline void AliTRDtrackV1::SetStatus(UChar_t status, Int_t ly)
 214 {
 215   if(ly<kNplane) fStatus|=((status&0xf)<<((ly+1)*4));
 216   return;
 217 }
 218
 219
 220 //____________________________________________________________________________
 221 inline Float_t AliTRDtrackV1::StatusForTOF()
 222 {
 223   // OBSOLETE
 224   // Defines the status of the TOF extrapolation
 225   //
 226
 227   if(!fTracklet[5]) return 0.;
 228
 229   // Definition of res ????
 230   Float_t res = /*(0.2 + 0.8 * (fN / (fNExpected + 5.0))) **/ (0.4 + 0.6 * fTracklet[5]->GetN() / 20.0);
 231   res *= (0.25 + 0.8 * 40.0 / (40.0 + fBudget[2]));
 232   return res;
 233 }
 234
 235 #endif
 236
 237
 238