PWG1/TRD/AliTRDclusterResolution.h

   1 #ifndef ALITRDCLUSTERRESOLUTION_H
   2 #define ALITRDCLUSTERRESOLUTION_H
   3
   4 ///////////////////////////////////////////////////////////////////////////////
   5 //                                                                           //
   6 //  TRD cluster error parameterization
   7 //  Authors:                                                              //
   8 //    Alexandru Bercuci <A.Bercuci@gsi.de>                                //
   9 ////////////////////////////////////////////////////////////////////////////
  10
  11 #ifndef ALITRDRECOTASK_H
  12 #include "AliTRDrecoTask.h"
  13 #endif
  14 #include "AliLog.h"
  15
  16 class TCanvas;
  17 class TObjArray;
  18 class AliTRDclusterResolution : public AliTRDrecoTask
  19 {
  20 public:
  21   enum EAxisBinning { // bins in z and x direction
  22     kND  = 25
  23   };
  24   enum EResultContainer { // results container type
  25     kCenter = 0   // cluster2center pad calibration
  26    ,kQRes   = 1   // resolution on charge dependence
  27    ,kSigm   = 2   // sigma cluster as func of x and z
  28    ,kMean   = 3   // shift cluster as func of x and z
  29    ,kNtasks = 4   // total number os sub tasks
  30   };
  31   enum ECheckBits {
  32     kSaveAs     = BIT(22) // save intermediary results
  33    ,kCalibrated = BIT(23) // load calibration
  34   };
  35   AliTRDclusterResolution();
  36   AliTRDclusterResolution(const char *name);
  37   virtual ~AliTRDclusterResolution();
  38
  39   void          UserCreateOutputObjects();
  40   void          UserExec(Option_t *);
  41   Int_t         GetDetector() const { return fDet; }
  42   void          GetPad(Int_t &c, Int_t &r) const { c=fCol, r=fRow; return;}
  43   inline Float_t GetExB() const;
  44   inline Float_t GetVdrift() const;
  45   inline Float_t GetT0() const;
  46   inline Float_t GetGain() const;
  47   Bool_t        GetRefFigure(Int_t ifig);
  48   Bool_t        HasProcess(EResultContainer bit) const {return TESTBIT(fStatus, bit);}
  49   Bool_t        IsCalibrated() const { return TestBit(kCalibrated);}
  50
  51   TObjArray*    Histos();
  52   TObjArray*    Results() const {return fResults;};
  53
  54   Bool_t        IsVisual() const {return Bool_t(fCanvas);}
  55   Bool_t        IsSaveAs() const {return TestBit(kSaveAs);}
  56
  57   Bool_t        PostProcess();
  58   void          SetCalibrationRegion(Int_t det, Int_t col=-1, Int_t row=-1);
  59   void          SetVisual();
  60   void          SetProcess(EResultContainer bit, Bool_t v = kTRUE) {v ? SETBIT(fStatus, bit) : CLRBIT(fStatus, bit);}
  61   void          SetSaveAs(Bool_t v = kTRUE) {SetBit(kSaveAs, v);}
  62   inline void   ResetProcesses();
  63
  64 protected:
  65   void    ProcessCharge();
  66   void    ProcessCenterPad();
  67   void    ProcessSigma();
  68   void    ProcessMean();
  69
  70 private:
  71   AliTRDclusterResolution(const AliTRDclusterResolution&);
  72   AliTRDclusterResolution& operator=(const AliTRDclusterResolution&);
  73   Bool_t        LoadCalibration();
  74
  75   TCanvas    *fCanvas; //! visualization canvas
  76   TObjArray  *fInfo;   //! list of cluster info
  77   TObjArray  *fResults;// list of result graphs/histos/trees
  78   UChar_t    fStatus;  // steer parameter of the task
  79   Short_t    fDet;     // detector (-1 for all)
  80   Char_t     fCol;     // pad column (-1 for all)
  81   Char_t     fRow;     // pad row (-1 for all)
  82   Float_t    fExB;     // tg of the Lorentz angle
  83   Float_t    fVdrift;  // mean drift velocity
  84   Float_t    fT0;      // time 0
  85   Float_t    fGain;    // gain
  86   static const Float_t fgkTimeBinLength;// time bin length (invers of sampling frequency)
  87
  88   // working named variables
  89   UChar_t    fLy;      // TRD plane
  90   Float_t    fT;       // calibrated time
  91   Float_t    fX;       // local drift length
  92   Float_t    fY;       // local rphi offset
  93   Float_t    fZ;       // local anode wire offset
  94   Float_t    fR[4];    // mean/sgm resolution
  95   Float_t    fP[4];    // mean/sgm pulls
  96
  97   ClassDef(AliTRDclusterResolution, 3)  // cluster resolution
  98 };
  99
 100 //___________________________________________________
 101 inline Float_t AliTRDclusterResolution::GetExB() const
 102 {
 103   if(!IsCalibrated()) AliWarning("Instance not calibrated.");
 104   return fExB;
 105 }
 106
 107 //___________________________________________________
 108 inline Float_t AliTRDclusterResolution::GetVdrift() const
 109 {
 110   if(!IsCalibrated()) AliWarning("Instance not calibrated.");
 111   return fVdrift;
 112 }
 113
 114 //___________________________________________________
 115 inline Float_t AliTRDclusterResolution::GetT0() const
 116 {
 117   if(!IsCalibrated()) AliWarning("Instance not calibrated.");
 118   return fT0;
 119 }
 120
 121 //___________________________________________________
 122 inline Float_t AliTRDclusterResolution::GetGain() const
 123 {
 124   if(!IsCalibrated()) AliWarning("Instance not calibrated.");
 125   return fGain;
 126 }
 127
 128 //___________________________________________________
 129 inline void AliTRDclusterResolution::ResetProcesses()
 130 {
 131   CLRBIT(fStatus, kQRes);
 132   CLRBIT(fStatus, kCenter);
 133   CLRBIT(fStatus, kSigm);
 134   CLRBIT(fStatus, kMean);
 135 }
 136
 137 #endif
 138