TRD/AliTRDReconstructor.h

   1 #ifndef ALITRDRECONSTRUCTOR_H
   2 #define ALITRDRECONSTRUCTOR_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 // Class for TRD reconstruction                                              //
  11 //                                                                           //
  12 ///////////////////////////////////////////////////////////////////////////////
  13
  14 #include "AliReconstructor.h"
  15 #include "AliDetectorRecoParam.h"
  16 #include "AliTRDrecoParam.h"
  17 #include "AliTRDpidUtil.h"
  18
  19 class TClonesArray;
  20 class TTreeSRedirector;
  21 class AliRawReader;
  22 class AliTRDReconstructor: public AliReconstructor
  23 {
  24 public:
  25   enum ETRDReconstructorSteer {
  26     kDigitsConversion= BIT(0)
  27     ,kTC             = BIT(1) // tail cancelation
  28     ,kLUT            = BIT(2) // look up table for cluster position determination
  29     ,kGAUS           = BIT(3) // look up table for cluster position determination
  30     ,kClusterSharing = BIT(4) // Toggle cluster sharing
  31     ,kSteerPID       = BIT(5)
  32     ,kEightSlices    = BIT(6)
  33     ,kWriteClusters  = BIT(7)
  34     ,kWriteTracklets = BIT(8)
  35     ,kDriftGas       = BIT(9)
  36     ,kSeeding        = BIT(10)
  37     ,kVertexConstrained = BIT(11) // Perform vertex constrained fit
  38     ,kImproveTracklet   = BIT(12) // Improve tracklet in the SA TRD track finder
  39     ,kHLT            = BIT(13)
  40     ,kCosmic         = BIT(14)
  41     ,kOwner          = BIT(15)
  42     ,kNsteer         = 15       // number of tasks
  43   };
  44   enum ETRDReconstructorTask {
  45     kRawReader    = 0
  46     ,kClusterizer = 1
  47     ,kTracker     = 2
  48     ,kPID         = 3
  49     ,kNtasks      = 4  // number of reconsruction tasks
  50   };
  51   enum ETRDReconstructorGas {
  52     kXe = 0
  53     ,kAr = 1
  54   };
  55
  56   AliTRDReconstructor();
  57   AliTRDReconstructor(const AliTRDReconstructor &r);
  58   virtual ~AliTRDReconstructor();
  59   AliTRDReconstructor& operator = (const AliTRDReconstructor&)          { return *this;}
  60
  61         virtual void        Init();
  62
  63   virtual void        ConvertDigits(AliRawReader *rawReader, TTree *digitsTree) const;
  64   virtual AliTracker* CreateTracker() const;
  65   TTreeSRedirector*   GetDebugStream(ETRDReconstructorTask task) const { return task < kNtasks ? fDebugStream[task] : 0x0; }
  66
  67   virtual void        FillESD(AliRawReader *, TTree *clusterTree, AliESDEvent *esd) const { FillESD((TTree * )NULL, clusterTree, esd);                    }
  68   virtual void        FillESD(TTree *digitsTree, TTree *clusterTree, AliESDEvent *esd) const;
  69   static TClonesArray* GetClusters() {return fgClusters;}
  70   Int_t               GetNdEdxSlices() const     { return (Int_t)AliTRDpidUtil::GetNdEdxSlices(GetPIDMethod());}
  71   ETRDReconstructorGas GetDriftGas() const        { return fSteerParam&kDriftGas ? kAr : kXe;}
  72   AliTRDpidUtil::ETRDPIDMethod       GetPIDMethod() const       { return fSteerParam&kSteerPID ? AliTRDpidUtil::kNN : AliTRDpidUtil::kLQ;}
  73   static const AliTRDrecoParam* GetRecoParam() { return dynamic_cast<const AliTRDrecoParam*>(AliReconstructor::GetRecoParam(2)); }
  74   Int_t               GetStreamLevel(ETRDReconstructorTask task) const    { return fStreamLevel[task];}
  75   inline void         GetTCParams(Double_t *par) const;
  76   virtual Bool_t      HasDigitConversion() const { return fSteerParam&kDigitsConversion;  };
  77   Bool_t              HasVertexConstrained() const { return fSteerParam&kVertexConstrained; }
  78   Bool_t              HasImproveTracklets() const  { return fSteerParam&kImproveTracklet; }
  79   Bool_t              IsWritingClusters() const  { return fSteerParam&kWriteClusters;}
  80   Bool_t              IsWritingTracklets() const { return fSteerParam&kWriteTracklets;}
  81   Bool_t              IsHLT() const              { return fSteerParam&kHLT;}
  82   Bool_t              IsSeeding() const          { return fSteerParam&kSeeding;}
  83   Bool_t              IsCosmic() const           { return fSteerParam&kCosmic;}
  84   Bool_t              IsEightSlices() const      { return fSteerParam&kEightSlices;}
  85   Bool_t              UseClusterSharing() const  { return fSteerParam&kClusterSharing;}
  86   Bool_t              UseLUT() const             { return fSteerParam&kLUT;}
  87   Bool_t              UseGAUS() const             { return fSteerParam&kGAUS;}
  88   Bool_t              UseTailCancelation() const { return fSteerParam&kTC;}
  89
  90   static void         Options(UInt_t steer=0, UChar_t *stream=0x0);
  91   virtual void        Reconstruct(AliRawReader *rawReader, TTree *clusterTree) const;
  92   virtual void        Reconstruct(TTree *digitsTree, TTree *clusterTree) const;
  93
  94   static void         SetClusters(TClonesArray *clusters) {fgClusters = clusters;}
  95   void                SetOption(Option_t *opt);
  96   inline void         SetTCParams(Double_t *par);
  97   void                SetStreamLevel(Int_t level, ETRDReconstructorTask task= kTracker);
  98
  99 private:
 100   static Char_t    *fgSteerNames[kNsteer];//! steering names
 101   static Char_t    *fgSteerFlags[kNsteer];//! steering flags
 102   static Char_t    *fgTaskNames[kNtasks]; //! tasks names
 103   static Char_t    *fgTaskFlags[kNtasks]; //! tasks flags
 104   UChar_t           fStreamLevel[kNtasks];// stream level for each reconstruction task
 105   UInt_t            fSteerParam;          // steering bits
 106   Double_t          fTCParams[8];         // Tail Cancellation parameters for drift gases
 107   TTreeSRedirector *fDebugStream[kNtasks];// Debug Streamer container;
 108
 109   static TClonesArray *fgClusters;    // list of clusters for local reconstructor
 110
 111   ClassDef(AliTRDReconstructor, 2)    //  Class for the TRD reconstruction
 112
 113 };
 114
 115 //___________________________________________________
 116 inline void AliTRDReconstructor::GetTCParams(Double_t *par) const
 117 {
 118   if(!par) return;
 119   if(GetDriftGas()==kAr) memcpy(par, &fTCParams[4], 4*sizeof(Double_t));
 120   else memcpy(par, &fTCParams[0], 4*sizeof(Double_t));
 121 }
 122
 123 //___________________________________________________
 124 inline void AliTRDReconstructor::SetTCParams(Double_t *par)
 125 {
 126   if(!par) return;
 127   memcpy(fTCParams, par, 8*sizeof(Double_t));
 128 }
 129
 130 #endif