TRD/AliTRDmcTrack.h

   1 #ifndef ALITRDMCTRACK_H
   2 #define ALITRDMCTRACK_H
   3
   4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 ///////////////////////////////////////////////////////////////////////////////
   8 //                                                                           //
   9 //  TRD MC track                                                             //
  10 //  Used for efficiency estimates and matching of reconstructed tracks       //
  11 //  to MC particles                                                          //
  12 //                                                                           //
  13 ///////////////////////////////////////////////////////////////////////////////
  14
  15 #include <TObject.h>
  16
  17 class AliTRDgeometry;
  18
  19 const Int_t kMAX_TB=30;
  20
  21 class AliTRDmcTrack : public TObject {
  22
  23 // Represents TRD related info about generated track
  24
  25 public:
  26
  27   AliTRDmcTrack();
  28   AliTRDmcTrack(Int_t label, Int_t seedLabel, Bool_t primary, Float_t mass,
  29                 Int_t charge, Int_t pdg);
  30
  31   void SetPin(Int_t plane, Double_t px, Double_t py, Double_t pz)
  32               { Pin[plane][0] = px; Pin[plane][1] = py; Pin[plane][2] = pz; }
  33
  34   void SetPout(Int_t plane, Double_t px, Double_t py, Double_t pz)
  35               {Pout[plane][0] = px; Pout[plane][1] = py; Pout[plane][2] = pz;}
  36
  37   void SetXYZin(Int_t plane, Double_t x, Double_t y, Double_t z)
  38     { XYZin[plane][0] = x; XYZin[plane][1] = y; XYZin[plane][2] = z; }
  39
  40   void SetXYZout(Int_t plane, Double_t x, Double_t y, Double_t z)
  41     { XYZout[plane][0] = x; XYZout[plane][1] = y; XYZout[plane][2] = z; }
  42
  43   void GetPxPyPzXYZ(Double_t &px, Double_t &py, Double_t &pz,
  44                     Double_t &x,  Double_t &y,  Double_t &z,
  45                     Int_t opt = 0) const;
  46
  47   void GetPlanePxPyPz(Double_t &px, Double_t &py, Double_t &pz
  48                      ,Int_t plane, Int_t opt = 0) const;
  49
  50   void GetXYZin(Int_t plane, Double_t &x, Double_t &y, Double_t &z) const
  51     { x = XYZin[plane][0]; y = XYZin[plane][1]; z = XYZin[plane][2]; return; }
  52
  53   void GetXYZout(Int_t plane, Double_t &x, Double_t &y, Double_t &z) const
  54     {x = XYZout[plane][0]; y = XYZout[plane][1]; z = XYZout[plane][2]; return;}
  55
  56   void Update(Int_t ltb, Int_t p, Int_t n, Int_t index)
  57                                               { fIndex[ltb][p][n] = index; }
  58
  59   Int_t   GetTrackIndex()          const { return fLab;      }
  60   Int_t   GetSeedLabel()           const { return fSeedLab;  }
  61   void    SetSeedLabel(Int_t l)          { fSeedLab = l;     }
  62   void    SetNumberOfClusters(Int_t n)   { fN = n;           }
  63   Bool_t  IsPrimary()              const { return fPrimary;  }
  64   Float_t GetMass()                const { return fMass;     }
  65   Int_t   GetCharge()              const { return fCharge;   }
  66   Int_t   GetPdgCode()             const { return fPDG;      }
  67
  68   Int_t   GetNumberOfClusters()    const { return fN;        }
  69   Int_t   GetClusterIndex(Int_t ltb, Int_t p, Int_t n)
  70                                    const { return fIndex[ltb][p][n]; }
  71
  72 protected:
  73
  74    Int_t    fLab;             // Track index
  75    Int_t    fSeedLab;         // Seed track index
  76    Bool_t   fPrimary;         // TRUE if it's a primary particle
  77    Float_t  fMass;            // Mass of the MC track
  78    Int_t    fCharge;          // Charge of the MC track
  79    Int_t    fPDG;             // PDG code of the MC track
  80
  81    Int_t  fN;               // Number of TRD clusters associated with the track
  82    Int_t  fIndex[kMAX_TB][6][2]; // Indices of these clusters
  83
  84    Double_t Pin[6][3];        // Px,Py,Pz at the entrance of each TRD plane
  85    Double_t Pout[6][3];       // Px,Py,Pz at the exit of each TRD plane
  86
  87    Double_t XYZin[6][3];        // x,y,z at the entrance of the TRD
  88    Double_t XYZout[6][3];        // x,y,z at the exit of the TRD
  89
  90    ClassDef(AliTRDmcTrack,1)  // TRD MC track
  91
  92 };
  93
  94 #endif