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 "AliTRDpidUtil.h"
  17 #include "AliTRDrecoParam.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     ,kProcTracklets  = BIT(15) // process online tracklets
  42     ,kOwner          = BIT(16)
  43     ,kNsteer         = 16       // number of tasks
  44   };
  45   enum ETRDReconstructorTask {
  46     kRawReader    = 0
  47     ,kClusterizer = 1
  48     ,kTracker     = 2
  49     ,kPID         = 3
  50     ,kNtasks      = 4  // number of reconsruction tasks
  51   };
  52   enum ETRDReconstructorGas {
  53     kXe = 0
  54     ,kAr = 1
  55   };
  56
  57   AliTRDReconstructor();
  58   AliTRDReconstructor(const AliTRDReconstructor &r);
  59   virtual ~AliTRDReconstructor();
  60   AliTRDReconstructor& operator = (const AliTRDReconstructor&)          { return *this;}
  61
  62         virtual void        Init();
  63
  64   virtual void        ConvertDigits(AliRawReader *rawReader, TTree *digitsTree) const;
  65   virtual AliTracker* CreateTracker() const;
  66   TTreeSRedirector*   GetDebugStream(ETRDReconstructorTask task) const { return task < kNtasks ? fDebugStream[task] : 0x0; }
  67
  68   virtual void        FillESD(AliRawReader *, TTree *clusterTree, AliESDEvent *esd) const { FillESD((TTree * )NULL, clusterTree, esd);                    }
  69   virtual void        FillESD(TTree *digitsTree, TTree *clusterTree, AliESDEvent *esd) const;
  70   static TClonesArray* GetClusters() {return fgClusters;}
  71   static TClonesArray* GetTracklets() {return fgTracklets;}
  72   Int_t               GetNdEdxSlices() const     { return (Int_t)AliTRDpidUtil::GetNdEdxSlices(GetPIDMethod());}
  73   ETRDReconstructorGas GetDriftGas() const        { return fSteerParam&kDriftGas ? kAr : kXe;}
  74   AliTRDpidUtil::ETRDPIDMethod       GetPIDMethod() const       { return fSteerParam&kSteerPID ? AliTRDpidUtil::kNN : AliTRDpidUtil::kLQ;}
  75   static const AliTRDrecoParam* GetRecoParam() { return dynamic_cast<const AliTRDrecoParam*>(AliReconstructor::GetRecoParam(2)); }
  76   Int_t               GetStreamLevel(ETRDReconstructorTask task) const    { return fStreamLevel[task];}
  77   inline void         GetTCParams(Double_t *par) const;
  78   virtual Bool_t      HasDigitConversion() const { return fSteerParam&kDigitsConversion;  };
  79   Bool_t              HasVertexConstrained() const { return fSteerParam&kVertexConstrained; }
  80   Bool_t              HasImproveTracklets() const  { return fSteerParam&kImproveTracklet; }
  81   Bool_t              IsWritingClusters() const  { return fSteerParam&kWriteClusters;}
  82   Bool_t              IsWritingTracklets() const { return fSteerParam&kWriteTracklets;}
  83   Bool_t              IsHLT() const              { return fSteerParam&kHLT;}
  84   Bool_t              IsSeeding() const          { return fSteerParam&kSeeding;}
  85   Bool_t              IsCosmic() const           { return fSteerParam&kCosmic;}
  86   Bool_t              IsEightSlices() const      { return fSteerParam&kEightSlices;}
  87   Bool_t              UseClusterSharing() const  { return fSteerParam&kClusterSharing;}
  88   Bool_t              UseLUT() const             { return fSteerParam&kLUT;}
  89   Bool_t              UseGAUS() const             { return fSteerParam&kGAUS;}
  90   Bool_t              UseTailCancelation() const { return fSteerParam&kTC;}
  91   Bool_t              IsProcessingTracklets() const { return fSteerParam&kProcTracklets;}
  92
  93   static void         Options(UInt_t steer=0, UChar_t *stream=0x0);
  94   virtual void        Reconstruct(AliRawReader *rawReader, TTree *clusterTree) const;
  95   virtual void        Reconstruct(TTree *digitsTree, TTree *clusterTree) const;
  96
  97   static void         SetClusters(TClonesArray *clusters) {fgClusters = clusters;}
  98   static void         SetTracklets(TClonesArray *tracklets) {fgTracklets = tracklets;}
  99   void                SetOption(Option_t *opt);
 100   inline void         SetTCParams(Double_t *par);
 101   void                SetStreamLevel(Int_t level, ETRDReconstructorTask task= kTracker);
 102
 103 private:
 104   static Char_t    *fgSteerNames[kNsteer];//! steering names
 105   static Char_t    *fgSteerFlags[kNsteer];//! steering flags
 106   static Char_t    *fgTaskNames[kNtasks]; //! tasks names
 107   static Char_t    *fgTaskFlags[kNtasks]; //! tasks flags
 108   UChar_t           fStreamLevel[kNtasks];// stream level for each reconstruction task
 109   UInt_t            fSteerParam;          // steering bits
 110   Double_t          fTCParams[8];         // Tail Cancellation parameters for drift gases
 111   TTreeSRedirector *fDebugStream[kNtasks];// Debug Streamer container;
 112
 113   static TClonesArray *fgClusters;    // list of clusters for local reconstructor
 114   static TClonesArray *fgTracklets;   // list of online tracklets for local reconstructor
 115
 116   ClassDef(AliTRDReconstructor, 2)    //  Class for the TRD reconstruction
 117
 118 };
 119
 120 //___________________________________________________
 121 inline void AliTRDReconstructor::GetTCParams(Double_t *par) const
 122 {
 123   if(!par) return;
 124   if(GetDriftGas()==kAr) memcpy(par, &fTCParams[4], 4*sizeof(Double_t));
 125   else memcpy(par, &fTCParams[0], 4*sizeof(Double_t));
 126 }
 127
 128 //___________________________________________________
 129 inline void AliTRDReconstructor::SetTCParams(Double_t *par)
 130 {
 131   if(!par) return;
 132   memcpy(fTCParams, par, 8*sizeof(Double_t));
 133 }
 134
 135 #endif