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 template <typename Value> class TVectorT;
  23 typedef class TVectorT<Double_t> TVectorD;
  24 class AliESDtrack;
  25 class AliTRDcluster;
  26 class AliTRDReconstructor;
  27 class AliTRDtrackV1 : public AliKalmanTrack
  28 {
  29   friend class AliHLTTRDTrack; // allow HLT special access
  30 public:
  31   enum ETRDtrackSize {
  32     kNdet      = AliTRDgeometry::kNdet
  33    ,kNstacks   = AliTRDgeometry::kNstack*AliTRDgeometry::kNsector
  34    ,kNplane    = AliTRDgeometry::kNlayer
  35    ,kNcham     = AliTRDgeometry::kNstack
  36    ,kNsect     = AliTRDgeometry::kNsector
  37    ,kNslice    =   3
  38    ,kNMLPslice =   8
  39    ,kMAXCLUSTERSPERTRACK = 210
  40   };
  41
  42   // bits from 0-13 are reserved by ROOT (see TObject.h)
  43   enum ETRDtrackStatus {
  44     kOwner     = BIT(14)
  45    ,kStopped   = BIT(15)
  46    ,kKink      = BIT(16)
  47    ,kPrimary   = BIT(17)
  48    ,kSeeder    = BIT(18)   // if set other than TPC (ITS, TRD)
  49   };
  50
  51   // propagation/update error codes (up to 4 bits)
  52   enum ETRDtrackError {
  53     kProlongation = 1
  54    ,kPropagation
  55    ,kAdjustSector
  56    ,kSnp
  57    ,kTrackletInit
  58    ,kUpdate
  59    ,kUnknown      = 0xff
  60   };
  61
  62   // data/clusters/tracklet error codes (up to 4 bits/layer)
  63   enum ETRDlayerError {
  64     kGeometry = 1
  65    ,kBoundary
  66    ,kNoClusters
  67    ,kNoAttach
  68    ,kNoClustersTracklet
  69    ,kNoFit
  70    ,kChi2
  71   };
  72
  73   AliTRDtrackV1();
  74   AliTRDtrackV1(AliTRDseedV1 * const trklts, const Double_t p[5], const Double_t cov[15], Double_t x, Double_t alpha);
  75   AliTRDtrackV1(const AliESDtrack &ref);
  76   AliTRDtrackV1(const AliTRDtrackV1 &ref);
  77   virtual ~AliTRDtrackV1();
  78   AliTRDtrackV1 &operator=(const AliTRDtrackV1 &ref);
  79   virtual void   Copy(TObject &ref) const;
  80
  81   Bool_t         CookPID();
  82   Double_t       CookTruncatedMean(const Bool_t kinvq, const Double_t mag, const Int_t charge, const Int_t kcalib, Int_t &nch, Int_t &ncls, TVectorD *Qs=NULL, TVectorD *Xs=NULL, Int_t timeBin0=-1, Int_t timeBin1=1000, Int_t tstep=1) const;
  83
  84   Int_t          CookLabel(Float_t wrong, Int_t *labs=NULL, Float_t *freq=NULL);
  85   AliTRDtrackV1* GetBackupTrack() const {return fBackupTrack;}
  86   Double_t       GetBudget(Int_t i) const { return fBudget[i];}
  87   AliTRDcluster* GetCluster(Int_t id);
  88   Int_t          GetClusterIndex(Int_t id) const;
  89   Float_t        GetEdep() const {return fDE;}
  90   Int_t          GetESDid() const {return fESDid;}
  91   inline Float_t GetMomentum(Int_t plane=-1) const;
  92   inline Int_t   GetNCross();
  93   inline Int_t   GetNumberOfTracklets() const;
  94   Double_t       GetPIDsignal() const   { return 0.;}
  95   Double_t       GetPID(Int_t is) const { return (is >=0 && is < AliPID::kSPECIES) ? fPID[is] : -1.;}
  96   UChar_t        GetNumberOfTrackletsPID() const;
  97   Double_t       GetPredictedChi2(const AliTRDseedV1 *tracklet, Double_t *cov) const;
  98   Double_t       GetPredictedChi2(const AliCluster* /*c*/) const                   { return 0.0; }
  99   Int_t          GetProlongation(Double_t xk, Double_t &y, Double_t &z) const;
 100   inline UChar_t GetStatusTRD(Int_t ly=-1) const;
 101   Int_t          GetSector() const;
 102   AliTRDseedV1*  GetTracklet(Int_t plane) const {return plane >=0 && plane <kNplane ? fTracklet[plane] : NULL;}
 103   Int_t          GetTrackletIndex(Int_t plane) const          { return (plane>=0 && plane<kNplane) ? fTrackletIndex[plane] : -1;}
 104   AliExternalTrackParam*
 105                  GetTrackIn() const  { return fTrackLow;}
 106   AliExternalTrackParam*
 107                  GetTrackOut() const  { return fTrackHigh;}
 108   const Int_t* GetTrackletIndexes() const { return &fTrackletIndex[0];}
 109
 110   Bool_t         IsEqual(const TObject *inTrack) const;
 111   Bool_t         IsKink() const    { return TestBit(kKink);}
 112   Bool_t         IsOwner() const   { return TestBit(kOwner);};
 113   Bool_t         IsPrimary() const   { return TestBit(kPrimary);};
 114   Bool_t         IsStopped() const { return TestBit(kStopped);};
 115   Bool_t         IsElectron() const;
 116   Bool_t         IsTPCseeded() const { return !TestBit(kSeeder);};
 117   inline static Bool_t IsTrackError(ETRDtrackError error, UInt_t status);
 118   inline static Bool_t IsLayerError(ETRDlayerError error, Int_t layer, UInt_t status);
 119
 120   Int_t          MakeBackupTrack();
 121   void           Print(Option_t *o="") const;
 122
 123   Bool_t         PropagateTo(Double_t xr, Double_t x0 = 8.72, Double_t rho = 5.86e-3);
 124   Int_t          PropagateToR(Double_t xr, Double_t step);
 125   Bool_t         Rotate(Double_t angle, Bool_t absolute = kFALSE);
 126   void           SetBudget(Int_t i, Double_t b) {if(i>=0 && i<3) fBudget[i] = b;}
 127   void           SetEdep(Double32_t inDE){fDE = inDE;};
 128   void           SetESDid(Int_t id) {fESDid = id;}
 129   void           SetKink(Bool_t k)        { SetBit(kKink, k);}
 130   void           SetPrimary(Bool_t k)     { SetBit(kPrimary, k);}
 131   void           SetNonTPCseeded(Bool_t k){ SetBit(kSeeder, k);}
 132   void           SetNumberOfClusters();
 133   void           SetOwner();
 134   void           SetPID(Short_t is, Double_t inPID){if (is >=0 && is < AliPID::kSPECIES) fPID[is]=inPID;};
 135   void           SetPIDquality(UChar_t /*inPIDquality*/) const {/*fPIDquality = inPIDquality*/;};
 136   inline void    SetErrStat(UChar_t stat, Int_t ly=-1);
 137   void           SetStopped(Bool_t stop) {SetBit(kStopped, stop);}
 138   void           SetTracklet(AliTRDseedV1 *const trklt,  Int_t index);
 139   void           SetTrackIn();
 140   void           SetTrackOut(const AliExternalTrackParam *op=NULL);
 141   inline void    SetReconstructor(const AliTRDReconstructor *rec);
 142   inline Float_t StatusForTOF();
 143   void           UnsetTracklet(Int_t plane);
 144   Bool_t         Update(const AliCluster *, Double_t, Int_t) { return kFALSE; };
 145   void           UpdateChi2(Float_t chi2);
 146   void           UpdateESDtrack(AliESDtrack *t);
 147
 148 private:
 149   UInt_t       fStatus;                //  Bit map for the status of propagation
 150   Int_t        fTrackletIndex[kNplane];//  Tracklets index in the tracker list
 151   Int_t        fESDid;                 //  ESD track id
 152   Double32_t   fPID[AliPID::kSPECIES]; //  PID probabilities
 153   Double32_t   fBudget[3];             //  Integrated material budget
 154   Double32_t   fDE;                    //  Integrated delta energy
 155   Double32_t   fTruncatedMean;         //  Truncated mean
 156   Int_t        fNchamberdEdx;          //  number of chambers used in calculating truncated mean
 157   Int_t        fNclusterdEdx;          //  number of clusters used in calculating truncated mean
 158
 159   const AliTRDReconstructor *fkReconstructor;//! reconstructor link
 160   AliTRDtrackV1 *fBackupTrack;         //! Backup track
 161   AliTRDseedV1  *fTracklet[kNplane];   //  Tracklets array defining the track
 162   AliExternalTrackParam *fTrackLow;    // parameters of the track which enter TRD from below (TPC)
 163   AliExternalTrackParam *fTrackHigh;   // parameters of the track which enter TRD from above (HMPID, PHOS)
 164
 165   ClassDef(AliTRDtrackV1, 7)          // TRD track - tracklet based
 166 };
 167
 168 //____________________________________________________
 169 inline Float_t AliTRDtrackV1::GetMomentum(Int_t plane) const
 170 {
 171 // Return ESD momentum stored in the tracklet reconstructed in layer = "plane".
 172 // By default returns the ESD momentum in first tracklet attached to track
 173   if(plane==-1){
 174     for(Int_t i(0); i<kNplane; i++){
 175       if(fTracklet[i]) return fTracklet[i]->GetMomentum();
 176     }
 177   } else if( plane >=0 && plane < kNplane){
 178     if(fTracklet[plane]) return fTracklet[plane]->GetMomentum();
 179   }
 180   return -1.;
 181 }
 182
 183 //____________________________________________________
 184 inline Int_t AliTRDtrackV1::GetNCross()
 185 {
 186   Int_t ncross = 0;
 187   for(Int_t ip=0; ip<kNplane; ip++){
 188     if(!fTracklet[ip]) continue;
 189     ncross += fTracklet[ip]->IsRowCross();
 190   }
 191   return ncross;
 192 }
 193
 194 //____________________________________________________
 195 inline Int_t AliTRDtrackV1::GetNumberOfTracklets() const
 196 {
 197   Int_t n = 0;
 198   for(Int_t ip=0; ip<kNplane; ip++){
 199     if(!fTracklet[ip]) continue;
 200     n++;
 201   }
 202   return n;
 203 }
 204
 205 //____________________________________________________
 206 inline UChar_t AliTRDtrackV1::GetStatusTRD(Int_t ly) const
 207 {
 208   if(ly>=-1 && ly<kNplane) return (fStatus>>((ly+1)*4))&0xf;
 209   return kUnknown;
 210 }
 211
 212 //____________________________________________________
 213 inline Bool_t AliTRDtrackV1::IsTrackError(ETRDtrackError error, UInt_t status)
 214 {
 215   return (status&0xf)==UChar_t(error);
 216 }
 217
 218 //____________________________________________________
 219 inline Bool_t AliTRDtrackV1::IsLayerError(ETRDlayerError error, Int_t ly, UInt_t status)
 220 {
 221   if(ly>=kNplane || ly<0) return kFALSE;
 222   return ((status>>((ly+1)*4))&0xf) == UChar_t(error);
 223 }
 224
 225 //____________________________________________________
 226 inline void AliTRDtrackV1::SetReconstructor(const AliTRDReconstructor *rec)
 227 {
 228   for(Int_t ip=0; ip<kNplane; ip++){
 229     if(!fTracklet[ip]) continue;
 230     fTracklet[ip]->SetReconstructor(rec);
 231   }
 232   fkReconstructor = rec;
 233 }
 234
 235 //____________________________________________________
 236 inline void AliTRDtrackV1::SetErrStat(UChar_t status, Int_t ly)
 237 {
 238   if(ly<kNplane) fStatus|=((status&0xf)<<((ly+1)*4));
 239   return;
 240 }
 241
 242
 243 //____________________________________________________________________________
 244 inline Float_t AliTRDtrackV1::StatusForTOF()
 245 {
 246   // OBSOLETE
 247   // Defines the status of the TOF extrapolation
 248   //
 249
 250   if(!fTracklet[5]) return 0.;
 251
 252   // Definition of res ????
 253   Float_t res = /*(0.2 + 0.8 * (fN / (fNExpected + 5.0))) **/ (0.4 + 0.6 * fTracklet[5]->GetN() / 20.0);
 254   res *= (0.25 + 0.8 * 40.0 / (40.0 + fBudget[2]));
 255   return res;
 256 }
 257
 258 #endif
 259
 260
 261