STEER/AOD/AliAODTrack.h

   1 #ifndef AliAODTrack_H
   2 #define AliAODTrack_H
   3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 //-------------------------------------------------------------------------
   9 //     AOD track implementation of AliVTrack
  10 //     Author: Markus Oldenburg, CERN
  11 //-------------------------------------------------------------------------
  12
  13 #include <TRef.h>
  14 #include <TBits.h>
  15
  16 #include "AliVTrack.h"
  17 #include "AliAODVertex.h"
  18 #include "AliAODRedCov.h"
  19 #include "AliAODPid.h"
  20
  21
  22 class AliVVertex;
  23
  24 class AliAODTrack : public AliVTrack {
  25
  26  public:
  27
  28   enum AODTrk_t {kUndef = -1,
  29                  kPrimary,
  30                  kSecondary,
  31                  kOrphan};
  32
  33   enum AODTrkBits_t {
  34     kIsDCA=BIT(14),   // set if fPosition is the DCA and not the position of the first point
  35     kUsedForVtxFit=BIT(15), // set if this track was used to fit the vertex it is attached to
  36     kUsedForPrimVtxFit=BIT(16), // set if this track was used to fit the primary vertex
  37     kIsTPCConstrained=BIT(17), // set if this track is a SA TPC track constrained to the SPD vertex, needs to be skipped in any track loop to avoid double counting
  38     kIsHybridTPCCG=BIT(18), // set if this track can be used as a hybrid track i.e. Gbobal tracks with certain slecetion plus the TPC constrained tracks that did not pass the selection
  39     kIsGlobalConstrained=BIT(19), // set if this track is a global track constrained to the vertex, needs to be skipped in any track loop to avoid double counting
  40     kIsHybridGCG=BIT(20)// set if this track can be used as a hybrid track i.e. tracks with certain slecetion plus the global constraint tracks that did not pass the selection
  41   };
  42
  43
  44   enum AODTrkFilterBits_t {
  45     kTrkTPCOnly = BIT(1), // Standard TPC only tracks
  46     kTrkITSsa   = BIT(2), // ITS standalone
  47     kTrkITSConstrained = BIT(3), // Pixel OR necessary for the electrons
  48     kTrkElectronsPID = BIT(4),    // PID for the electrons
  49     kTrkGlobalNoDCA = BIT(5), // standard cuts with very loose DCA
  50     kTrkGlobal = BIT(6),  // standard cuts with tight DCA cut
  51     kTrkGlobalSDD = BIT(7), // standard cuts with tight DCA but with requiring the first SDD cluster instead of an SPD cluster tracks selected by this cut are exclusive to those selected by the previous cut
  52     kTrkTPCOnlyConstrained = BIT(8) // TPC only tracks: TPConly information constrained to SPD vertex in the filter below
  53   };
  54
  55
  56   enum AODTrkPID_t {
  57     kElectron     =  0,
  58     kMuon         =  1,
  59     kPion         =  2,
  60     kKaon         =  3,
  61     kProton       =  4,
  62     kDeuteron     =  5,
  63     kTriton       =  6,
  64     kHelium3      =  7,
  65     kAlpha        =  8,
  66     kUnknown      =  9,
  67     kMostProbable = -1
  68   };
  69
  70   AliAODTrack();
  71   AliAODTrack(Short_t id,
  72               Int_t label,
  73               Double_t p[3],
  74               Bool_t cartesian,
  75               Double_t x[3],
  76               Bool_t dca,
  77               Double_t covMatrix[21],
  78               Short_t q,
  79               UChar_t itsClusMap,
  80               Double_t pid[10],
  81               AliAODVertex *prodVertex,
  82               Bool_t usedForVtxFit,
  83               Bool_t usedForPrimVtxFit,
  84               AODTrk_t ttype=kUndef,
  85               UInt_t selectInfo=0,
  86               Float_t chi2perNDF = -999.);
  87
  88   AliAODTrack(Short_t id,
  89               Int_t label,
  90               Float_t p[3],
  91               Bool_t cartesian,
  92               Float_t x[3],
  93               Bool_t dca,
  94               Float_t covMatrix[21],
  95               Short_t q,
  96               UChar_t itsClusMap,
  97               Float_t pid[10],
  98               AliAODVertex *prodVertex,
  99               Bool_t usedForVtxFit,
 100               Bool_t usedForPrimVtxFit,
 101               AODTrk_t ttype=kUndef,
 102               UInt_t selectInfo=0,
 103               Float_t chi2perNDF = -999.);
 104
 105   virtual ~AliAODTrack();
 106   AliAODTrack(const AliAODTrack& trk);
 107   AliAODTrack& operator=(const AliAODTrack& trk);
 108
 109   // kinematics
 110   virtual Double_t OneOverPt() const { return (fMomentum[0] != 0.) ? 1./fMomentum[0] : -999.; }
 111   virtual Double_t Phi()       const { return fMomentum[1]; }
 112   virtual Double_t Theta()     const { return fMomentum[2]; }
 113
 114   virtual Double_t Px() const { return fMomentum[0] * TMath::Cos(fMomentum[1]); }
 115   virtual Double_t Py() const { return fMomentum[0] * TMath::Sin(fMomentum[1]); }
 116   virtual Double_t Pz() const { return fMomentum[0] / TMath::Tan(fMomentum[2]); }
 117   virtual Double_t Pt() const { return fMomentum[0]; }
 118   virtual Double_t P()  const { return TMath::Sqrt(Pt()*Pt()+Pz()*Pz()); }
 119   virtual Bool_t   PxPyPz(Double_t p[3]) const { p[0] = Px(); p[1] = Py(); p[2] = Pz(); return kTRUE; }
 120
 121   virtual Double_t Xv() const { return GetProdVertex() ? GetProdVertex()->GetX() : -999.; }
 122   virtual Double_t Yv() const { return GetProdVertex() ? GetProdVertex()->GetY() : -999.; }
 123   virtual Double_t Zv() const { return GetProdVertex() ? GetProdVertex()->GetZ() : -999.; }
 124   virtual Bool_t   XvYvZv(Double_t x[3]) const { x[0] = Xv(); x[1] = Yv(); x[2] = Zv(); return kTRUE; }
 125
 126   Double_t Chi2perNDF()  const { return fChi2perNDF; }
 127   UShort_t GetTPCNcls()  const { return fTPCClusterMap.CountBits();}
 128
 129   virtual Double_t M() const { return M(GetMostProbablePID()); }
 130   Double_t M(AODTrkPID_t pid) const;
 131   virtual Double_t E() const { return E(GetMostProbablePID()); }
 132   Double_t E(AODTrkPID_t pid) const;
 133   Double_t E(Double_t m) const { return TMath::Sqrt(P()*P() + m*m); }
 134   virtual Double_t Y() const { return Y(GetMostProbablePID()); }
 135   Double_t Y(AODTrkPID_t pid) const;
 136   Double_t Y(Double_t m) const;
 137
 138   virtual Double_t Eta() const { return -TMath::Log(TMath::Tan(0.5 * fMomentum[2])); }
 139
 140   virtual Short_t  Charge() const {return fCharge; }
 141
 142   virtual Bool_t   PropagateToDCA(const AliVVertex *vtx,
 143           Double_t b, Double_t maxd, Double_t dz[2], Double_t covar[3]);
 144
 145   // PID
 146   virtual const Double_t *PID() const { return fPID; }
 147   AODTrkPID_t GetMostProbablePID() const;
 148   void ConvertAliPIDtoAODPID();
 149   void SetDetPID(AliAODPid *aodpid) {fDetPid = aodpid;}
 150
 151   template <class T> void GetPID(T *pid) const {
 152     for(Int_t i=0; i<10; ++i) pid[i]=fPID[i];}
 153
 154   template <class T> void SetPID(const T *pid) {
 155     if(pid) for(Int_t i=0; i<10; ++i) fPID[i]=pid[i];
 156     else {for(Int_t i=0; i<10; fPID[i++]=0.) ; fPID[AliAODTrack::kUnknown]=1.;}}
 157
 158   Bool_t IsOn(Int_t mask) const {return (fFlags&mask)>0;}
 159   ULong_t GetStatus() const { return GetFlags(); }
 160   ULong_t GetFlags() const { return fFlags; }
 161
 162   Int_t   GetID() const { return (Int_t)fID; }
 163   Int_t   GetLabel() const { return fLabel; }
 164   Char_t  GetType() const { return fType;}
 165   Bool_t  IsPrimaryCandidate() const;
 166   Bool_t  GetUsedForVtxFit() const { return TestBit(kUsedForVtxFit); }
 167   Bool_t  GetUsedForPrimVtxFit() const { return TestBit(kUsedForPrimVtxFit); }
 168
 169   Bool_t  IsHybridGlobalConstrainedGlobal() const { return TestBit(kIsHybridGCG); }
 170   Bool_t  IsHybridTPCConstrainedGlobal() const { return TestBit(kIsHybridTPCCG); }
 171   Bool_t  IsTPCOnly() const { return IsTPCConstrained(); } // obsolete bad naming
 172   Bool_t  IsTPCConstrained() const { return TestBit(kIsTPCConstrained); }
 173   Bool_t  IsGlobalConstrained() const { return TestBit(kIsGlobalConstrained); }
 174   //
 175   Int_t   GetTOFBunchCrossing(Double_t b=0) const;
 176   //
 177   template <class T> void GetP(T *p) const {
 178     p[0]=fMomentum[0]; p[1]=fMomentum[1]; p[2]=fMomentum[2];}
 179
 180 //  template <class T> void GetPxPyPz(T *p) const {
 181 //    p[0] = Px(); p[1] = Py(); p[2] = Pz();}
 182   Bool_t GetPxPyPz(Double_t *p) const;
 183
 184   template <class T> Bool_t GetPosition(T *x) const {
 185     x[0]=fPosition[0]; x[1]=fPosition[1]; x[2]=fPosition[2];
 186     return TestBit(kIsDCA);}
 187
 188   template <class T> void SetCovMatrix(const T *covMatrix) {
 189     if(!fCovMatrix) fCovMatrix=new AliAODRedCov<6>();
 190     fCovMatrix->SetCovMatrix(covMatrix);}
 191
 192   template <class T> Bool_t GetCovMatrix(T *covMatrix) const {
 193     if(!fCovMatrix) return kFALSE;
 194     fCovMatrix->GetCovMatrix(covMatrix); return kTRUE;}
 195
 196   Bool_t GetXYZ(Double_t *p) const {
 197     return GetPosition(p); }
 198
 199   Bool_t GetCovarianceXYZPxPyPz(Double_t cv[21]) const {
 200     return GetCovMatrix(cv);}
 201
 202   void RemoveCovMatrix() {delete fCovMatrix; fCovMatrix=NULL;}
 203
 204   Double_t XAtDCA() const { return fPositionAtDCA[0]; }
 205   Double_t YAtDCA() const { return fPositionAtDCA[1]; }
 206   Double_t ZAtDCA() const {
 207     if (IsMuonTrack()) return fPosition[2];
 208     else if (TestBit(kIsDCA)) return fPosition[1];
 209     else return -999.; }
 210   Bool_t   XYZAtDCA(Double_t x[3]) const { x[0] = XAtDCA(); x[1] = YAtDCA(); x[2] = ZAtDCA(); return kTRUE; }
 211
 212   Double_t DCA() const {
 213     if (IsMuonTrack()) return TMath::Sqrt(XAtDCA()*XAtDCA() + YAtDCA()*YAtDCA());
 214     else if (TestBit(kIsDCA)) return fPosition[0];
 215     else return -999.; }
 216
 217   Double_t PxAtDCA() const { return fMomentumAtDCA[0]; }
 218   Double_t PyAtDCA() const { return fMomentumAtDCA[1]; }
 219   Double_t PzAtDCA() const { return fMomentumAtDCA[2]; }
 220   Double_t PAtDCA() const { return TMath::Sqrt(PxAtDCA()*PxAtDCA() + PyAtDCA()*PyAtDCA() + PzAtDCA()*PzAtDCA()); }
 221   Bool_t   PxPyPzAtDCA(Double_t p[3]) const { p[0] = PxAtDCA(); p[1] = PyAtDCA(); p[2] = PzAtDCA(); return kTRUE; }
 222
 223   Double_t GetRAtAbsorberEnd() const { return fRAtAbsorberEnd; }
 224
 225   UChar_t  GetITSClusterMap() const       { return (UChar_t)(fITSMuonClusterMap&0xff); }
 226   Int_t    GetITSNcls() const;
 227   Bool_t   HasPointOnITSLayer(Int_t i) const { return TESTBIT(GetITSClusterMap(),i); }
 228   UShort_t GetHitsPatternInTrigCh() const { return (UShort_t)((fITSMuonClusterMap&0xff00)>>8); }
 229   UInt_t   GetMUONClusterMap() const      { return (fITSMuonClusterMap&0x3ff0000)>>16; }
 230   UInt_t   GetITSMUONClusterMap() const   { return fITSMuonClusterMap; }
 231
 232   Bool_t  TestFilterBit(UInt_t filterBit) const {return (Bool_t) ((filterBit & fFilterMap) != 0);}
 233   Bool_t  TestFilterMask(UInt_t filterMask) const {return (Bool_t) ((filterMask & fFilterMap) == filterMask);}
 234   void    SetFilterMap(UInt_t i){fFilterMap = i;}
 235   UInt_t  GetFilterMap(){return fFilterMap;}
 236
 237   const TBits& GetTPCClusterMap() const {return fTPCClusterMap;}
 238   Float_t GetTPCClusterInfo(Int_t nNeighbours=3, Int_t type=0, Int_t row0=0, Int_t row1=159) const;
 239
 240   const TBits& GetTPCSharedMap() const {return fTPCSharedMap;}
 241   void    SetTPCClusterMap(const TBits amap) {fTPCClusterMap = amap;}
 242   void    SetTPCSharedMap(const TBits amap) {fTPCSharedMap = amap;}
 243   void    SetTPCPointsF(UShort_t  findable){fTPCnclsF = findable;}
 244
 245   UShort_t GetTPCNclsF() const { return fTPCnclsF;}
 246
 247   // Calorimeter Cluster
 248   Int_t GetEMCALcluster() const {return fCaloIndex;}
 249   void SetEMCALcluster(Int_t index) {fCaloIndex=index;}
 250   Bool_t IsEMCAL() const {return fFlags&kEMCALmatch;}
 251
 252   Int_t GetPHOScluster() const {return fCaloIndex;}
 253   void SetPHOScluster(Int_t index) {fCaloIndex=index;}
 254   Bool_t IsPHOS() const {return fFlags&kPHOSmatch;}
 255
 256   //pid signal interface
 257   Double_t  GetITSsignal()       const { return fDetPid?fDetPid->GetITSsignal():0.;    }
 258   Double_t  GetTPCsignal()       const { return fDetPid?fDetPid->GetTPCsignal():0.;    }
 259   UShort_t  GetTPCsignalN()      const { return fDetPid?fDetPid->GetTPCsignalN():0;    }
 260   Double_t  GetTPCmomentum()     const { return fDetPid?fDetPid->GetTPCmomentum():0.;  }
 261   Double_t  GetTOFsignal()       const { return fDetPid?fDetPid->GetTOFsignal():0.; }
 262   void      GetIntegratedTimes(Double_t *times) const {if (fDetPid) fDetPid->GetIntegratedTimes(times); }
 263   Double_t  GetTRDslice(Int_t plane, Int_t slice) const;
 264   Double_t  GetTRDmomentum(Int_t plane, Double_t */*sp*/=0x0) const;
 265   UChar_t   GetTRDncls(Int_t layer = -1) const;
 266   UChar_t   GetTRDntrackletsPID() const;
 267   void      GetHMPIDpid(Double_t *p) const { if (fDetPid) fDetPid->GetHMPIDprobs(p); }
 268
 269
 270   AliAODPid    *GetDetPid() const { return fDetPid; }
 271   AliAODVertex *GetProdVertex() const { return (AliAODVertex*)fProdVertex.GetObject(); }
 272
 273   // print
 274   void  Print(const Option_t *opt = "") const;
 275
 276   // setters
 277   void SetFlags(ULong_t flags) { fFlags = flags; }
 278   void SetStatus(ULong_t flags) { fFlags|=flags; }
 279   void ResetStatus(ULong_t flags) { fFlags&=~flags; }
 280
 281   void SetID(Short_t id) { fID = id; }
 282   void SetLabel(Int_t label) { fLabel = label; }
 283
 284   template <class T> void SetPosition(const T *x, Bool_t isDCA = kFALSE);
 285   void SetDCA(Double_t d, Double_t z);
 286   void SetUsedForVtxFit(Bool_t used = kTRUE) { used ? SetBit(kUsedForVtxFit) : ResetBit(kUsedForVtxFit); }
 287   void SetUsedForPrimVtxFit(Bool_t used = kTRUE) { used ? SetBit(kUsedForPrimVtxFit) : ResetBit(kUsedForPrimVtxFit); }
 288
 289   void SetIsTPCOnly(Bool_t b = kTRUE) { SetIsTPCConstrained(b); }// obsolete bad naming
 290
 291   void SetIsTPCConstrained(Bool_t b = kTRUE) { b ? SetBit(kIsTPCConstrained) : ResetBit(kIsTPCConstrained); }
 292   void SetIsHybridTPCConstrainedGlobal(Bool_t hybrid = kTRUE) { hybrid ? SetBit(kIsHybridTPCCG) : ResetBit(kIsHybridTPCCG); }
 293
 294   void SetIsGlobalConstrained(Bool_t b = kTRUE) { b ? SetBit(kIsGlobalConstrained) : ResetBit(kIsGlobalConstrained); }
 295   void SetIsHybridGlobalConstrainedGlobal(Bool_t hybrid = kTRUE) { hybrid ? SetBit(kIsHybridGCG) : ResetBit(kIsHybridGCG); }
 296
 297
 298
 299   void SetOneOverPt(Double_t oneOverPt) { fMomentum[0] = 1. / oneOverPt; }
 300   void SetPt(Double_t pt) { fMomentum[0] = pt; };
 301   void SetPhi(Double_t phi) { fMomentum[1] = phi; }
 302   void SetTheta(Double_t theta) { fMomentum[2] = theta; }
 303   template <class T> void SetP(const T *p, Bool_t cartesian = kTRUE);
 304   void SetP() {fMomentum[0]=fMomentum[1]=fMomentum[2]=-999.;}
 305
 306   void SetXYAtDCA(Double_t x, Double_t y) {fPositionAtDCA[0] = x; fPositionAtDCA[1] = y;}
 307   void SetPxPyPzAtDCA(Double_t pX, Double_t pY, Double_t pZ) {fMomentumAtDCA[0] = pX; fMomentumAtDCA[1] = pY; fMomentumAtDCA[2] = pZ;}
 308
 309   void SetRAtAbsorberEnd(Double_t r) { fRAtAbsorberEnd = r; }
 310
 311   void SetCharge(Short_t q) { fCharge = q; }
 312   void SetChi2perNDF(Double_t chi2perNDF) { fChi2perNDF = chi2perNDF; }
 313
 314   void SetITSClusterMap(UChar_t itsClusMap)                 { fITSMuonClusterMap = (fITSMuonClusterMap&0xffffff00)|(((UInt_t)itsClusMap)&0xff); }
 315   void SetHitsPatternInTrigCh(UShort_t hitsPatternInTrigCh) { fITSMuonClusterMap = (fITSMuonClusterMap&0xffff00ff)|((((UInt_t)hitsPatternInTrigCh)&0xff)<<8); }
 316   void SetMuonClusterMap(UInt_t muonClusMap)                { fITSMuonClusterMap = (fITSMuonClusterMap&0xfc00ffff)|((muonClusMap&0x3ff)<<16); }
 317   void SetITSMuonClusterMap(UInt_t itsMuonClusMap)          { fITSMuonClusterMap = itsMuonClusMap; }
 318
 319   Int_t GetMatchTrigger() const {return fITSMuonClusterMap>>30;}
 320                                         //  0 Muon track does not match trigger
 321                                         //  1 Muon track match but does not pass pt cut
 322                                         //  2 Muon track match Low pt cut
 323                                         //  3 Muon track match High pt cut
 324   void     SetMatchTrigger(Int_t MatchTrigger);
 325   Bool_t   MatchTrigger() const { return (GetMatchTrigger()>0); }         //  Muon track matches trigger track
 326   Bool_t   MatchTriggerLowPt()   const  { return (GetMatchTrigger()>1); } //  Muon track matches trigger track and passes Low pt cut
 327   Bool_t   MatchTriggerHighPt()  const  { return (GetMatchTrigger()>2); } //  Muon track matches trigger track and passes High pt cut
 328   Bool_t   MatchTriggerDigits()  const;                                   //  Muon track matches trigger digits
 329   Double_t GetChi2MatchTrigger() const  { return fChi2MatchTrigger;}
 330   void     SetChi2MatchTrigger(Double_t Chi2MatchTrigger) {fChi2MatchTrigger = Chi2MatchTrigger; }
 331   Bool_t   HitsMuonChamber(Int_t MuonChamber, Int_t cathode = -1) const;  // Check if track hits Muon chambers
 332   Bool_t   IsMuonTrack() const { return (GetMUONClusterMap()>0) ? kTRUE : kFALSE; }
 333
 334   void     Connected(Bool_t flag) {flag ? SETBIT(fITSMuonClusterMap,26) : CLRBIT(fITSMuonClusterMap,26);}
 335   Bool_t   IsConnected() const {return TESTBIT(fITSMuonClusterMap,26);}
 336
 337   void     SetProdVertex(TObject *vertex) { fProdVertex = vertex; }
 338   void     SetType(AODTrk_t ttype) { fType=ttype; }
 339
 340
 341
 342   // Dummy
 343   Int_t    PdgCode() const {return 0;}
 344
 345  private :
 346
 347   // Momentum & position
 348   Double32_t    fMomentum[3];       // momemtum stored in pt, phi, theta
 349   Double32_t    fPosition[3];       // position of first point on track or dca
 350
 351   Double32_t    fMomentumAtDCA[3];  // momentum (px,py,pz) at DCA
 352   Double32_t    fPositionAtDCA[2];  // trasverse position (x,y) at DCA
 353
 354   Double32_t    fRAtAbsorberEnd;    // transverse position r at the end of the muon absorber
 355
 356   Double32_t    fChi2perNDF;        // chi2/NDF of momentum fit
 357   Double32_t    fChi2MatchTrigger;  // chi2 of trigger/track matching
 358   Double32_t    fPID[10];           // [0.,1.,8] pointer to PID object
 359
 360   ULong_t       fFlags;             // reconstruction status flags
 361   Int_t         fLabel;             // track label, points back to MC track
 362
 363   UInt_t        fITSMuonClusterMap; // map of ITS and muon clusters, one bit per layer
 364                                     // (ITS: bit 1-8, muon trigger: bit 9-16, muon tracker: bit 17-26, muon match trigger: bit 31-32)
 365   UInt_t        fFilterMap;         // filter information, one bit per set of cuts
 366
 367   TBits         fTPCClusterMap;     // Map of clusters, one bit per padrow; 1 if has a cluster on given padrow
 368   TBits         fTPCSharedMap;      // Map of clusters, one bit per padrow; 1 if has a shared cluster on given padrow
 369   UShort_t      fTPCnclsF;          // findable clusters
 370
 371   Short_t       fID;                // unique track ID, points back to the ESD track
 372
 373   Char_t        fCharge;            // particle charge
 374   Char_t        fType;              // Track Type
 375
 376   Int_t         fCaloIndex;         // index of associated EMCAL/PHOS cluster (AliAODCaloCluster)
 377
 378
 379   AliAODRedCov<6> *fCovMatrix;      // covariance matrix (x, y, z, px, py, pz)
 380   AliAODPid    *fDetPid;            // more detailed or detector specific pid information
 381   TRef          fProdVertex;        // vertex of origin
 382
 383   ClassDef(AliAODTrack, 14);
 384 };
 385
 386 inline Bool_t  AliAODTrack::IsPrimaryCandidate() const
 387 {
 388     // True of track passes primary particle selection (independent of type)
 389     //
 390     if (fFilterMap) {
 391         return kTRUE;
 392     } else {
 393         return kFALSE;
 394     }
 395 }
 396
 397 inline Int_t AliAODTrack::GetITSNcls() const
 398 {
 399   // Number of points in ITS
 400   Int_t n=0;
 401   for(Int_t i=0;i<6;i++) if(HasPointOnITSLayer(i)) n++;
 402   return n;
 403 }
 404
 405 #endif