STEER/AOD/AliAODTrdTrack.h

   1 #ifndef ALIAODTRDTRACK_H
   2 #define ALIAODTRDTRACK_H
   3
   4 //
   5 // format for the TRD tracks calculated in the
   6 // Global Tracking Unit, used for the TRD L1 trigger
   7 // Author: Jochen Klein <jochen.klein@cern.ch>
   8
   9 #include "TRef.h"
  10 #include "TClonesArray.h"
  11 #include "AliVTrack.h"
  12 #include "AliVTrdTrack.h"
  13 #include "AliAODTrdTracklet.h"
  14
  15 class AliAODTrdTrack : public AliVTrdTrack {
  16  public:
  17
  18   AliAODTrdTrack();
  19   AliAODTrdTrack(const AliVTrdTrack &rhs);
  20   virtual ~AliAODTrdTrack() {};
  21   AliAODTrdTrack(const AliAODTrdTrack& track);
  22   AliAODTrdTrack& operator=(const AliAODTrdTrack& track);
  23   virtual void Copy(TObject &obj) const;
  24
  25   virtual Int_t GetA()         const { return fA; }
  26   virtual Int_t GetLayerMask() const { return fLayerMask; }
  27   virtual Int_t GetPID()       const { return fPID; }
  28   virtual Int_t GetPt()        const;
  29   virtual Int_t GetStack()     const { return fGlobalStack%5; }
  30   virtual Int_t GetSector()    const { return fGlobalStack/5; }
  31
  32   virtual Bool_t GetTrackInTime() const { return (fFlagsTiming & 0x1); }
  33   virtual UChar_t GetFlagsTiming() const { return fFlagsTiming; }
  34
  35   virtual Int_t GetLabel()     const { return fLabel; }
  36
  37   virtual Double_t Pt()        const { return GetPt() / 128.; }
  38
  39   Int_t GetNTracklets() const {
  40     Int_t count = 0;
  41     for (Int_t iLayer = 0; iLayer < 6; ++iLayer)
  42       count += (fLayerMask >> iLayer) & 1;
  43     return count;
  44   }
  45   virtual AliAODTrdTracklet* GetTracklet(Int_t idx) const { return (AliAODTrdTracklet*) fTracklets[idx]; }
  46
  47   virtual AliVTrack* GetTrackMatch() const { return (AliVTrack*) fTrackMatch.GetObject(); }
  48
  49   virtual void SetA(Int_t a) { fA = a; }
  50   virtual void SetLayerMask(Int_t mask) { fLayerMask = mask; }
  51   virtual void SetPID(Int_t pid) { fPID = pid; }
  52   virtual void SetLabel(Int_t label) { fLabel = label; }
  53   virtual void SetSector(Int_t sector) { fGlobalStack = 5*sector + (fGlobalStack%5); }
  54   virtual void SetStack(Int_t stack) { fGlobalStack = 5*(fGlobalStack%5) + stack; }
  55
  56   void AddTracklet(const AliVTrdTracklet& trkl, Int_t layer) { new (fTracklets[layer]) AliAODTrdTracklet(trkl); }
  57   void SetTrackMatchReference(AliVTrack *trk) { fTrackMatch = trk; }
  58
  59   virtual Bool_t IsSortable() const { return kFALSE; }
  60   virtual Int_t Compare(const TObject* /* obj */) const { return 0; }
  61
  62  protected:
  63
  64   Char_t   fGlobalStack;                  // stack (0-89) in which the track was found
  65                                           // (unique because of stack-wise tracking)
  66   UChar_t  fPID;                          // electron PID for this track
  67   UChar_t  fLayerMask;                    // mask of contributing tracklets
  68   Int_t    fA;                            // transverse offset from nominal primary vertex
  69   UChar_t  fFlagsTiming;                  // timing flags
  70
  71   TClonesArray fTracklets;                // array of contributing tracklets
  72   TRef fTrackMatch;                       // reference to matched global track
  73
  74   Int_t fLabel;                           // Track label
  75
  76   ClassDef(AliAODTrdTrack,1)
  77 };
  78
  79 #endif