TRD/AliTRDcluster.h

   1 #ifndef ALITRDCLUSTER_H
   2 #define ALITRDCLUSTER_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 //  TRD cluster                                                              //
  11 //                                                                           //
  12 ///////////////////////////////////////////////////////////////////////////////
  13
  14 #include "AliCluster.h"
  15
  16 class AliTRDcluster : public AliCluster {
  17 public:
  18   enum ETRDclusterStatus {
  19     kInChamber = BIT(16) // Out of fiducial volume of chamber (signal tails)
  20    ,kFivePad   = BIT(17) // Deconvoluted clusters
  21    ,kLUT       = BIT(18)
  22    ,kGAUS      = BIT(19)
  23    ,kCOG       = BIT(20)
  24   };
  25   enum ETRDclusterMask {
  26     kMaskedLeft   = 0
  27    ,kMaskedCenter = 1
  28    ,kMaskedRight  = 2
  29   };
  30
  31   AliTRDcluster();
  32   AliTRDcluster(Int_t det, UChar_t col, UChar_t row, UChar_t time, const Short_t *sig, UShort_t volid);
  33   AliTRDcluster(
  34     Int_t det, Float_t q, Float_t *pos, Float_t *sig,
  35     Int_t *tracks, Char_t npads, Short_t *signals,
  36     UChar_t col, UChar_t row, UChar_t time,
  37     Char_t timebin, Float_t center, UShort_t volid);
  38   AliTRDcluster(const AliTRDcluster &c);
  39   virtual ~AliTRDcluster() {};
  40
  41   virtual void     AddTrackIndex(Int_t *i);
  42   void     Clear(Option_t *o="");
  43
  44   Bool_t   IsEqual(const TObject *otherCluster) const;
  45   Bool_t   IsInChamber() const             { return TestBit(kInChamber);       }
  46   Bool_t   IsMasked() const                { return fClusterMasking ? kTRUE : kFALSE; }
  47   Bool_t   IsShared() const                { return IsClusterShared();}
  48   Bool_t   IsUsed() const                  { return IsClusterUsed(); }
  49   Bool_t   IsFivePad() const               { return TestBit(kFivePad);}
  50   inline Bool_t IsRPhiMethod(ETRDclusterStatus m) const;
  51
  52   UChar_t  GetPadMaskedPosition() const    { return fClusterMasking & 7; }
  53   UChar_t  GetPadMaskedStatus() const      { return fClusterMasking >> 3; }
  54   Int_t    GetDetector() const             { return fDetector;      }
  55   Int_t    GetLocalTimeBin() const         { return fLocalTimeBin;  }
  56   Float_t  GetQ() const                    { return fQ;             }
  57   Int_t    GetNPads() const                { return fNPads;         }
  58   Float_t  GetCenter() const               { return fCenter;        }
  59   Int_t    GetPadCol() const               { return fPadCol;        }
  60   Int_t    GetPadRow() const               { return fPadRow;        }
  61   Int_t    GetPadTime() const              { return fPadTime;       }
  62   Short_t *GetSignals()                    { return fSignals;       }
  63   Float_t  GetSumS() const;
  64   static Double_t  GetSX(Int_t tb, Double_t z=-1);
  65   static Double_t  GetSYdrift(Int_t tb, Int_t ly=0, Double_t z=-1);
  66   static Double_t  GetSYcharge(Float_t q);
  67   static Double_t  GetSYprf(Int_t ly, Double_t center, Double_t s2);
  68   static Double_t  GetXcorr(Int_t tb, Double_t z=-1);
  69   static Double_t  GetYcorr(Int_t ly, Float_t y);
  70   Float_t  GetXloc(Double_t t0, Double_t vd, Double_t *const q=0x0, Double_t *const xq=0x0, Double_t z = 0.2);
  71   Float_t  GetYloc(Double_t y0, Double_t s2, Double_t W, Double_t *const y1=0x0, Double_t *const y2=0x0);
  72
  73   void     Print(Option_t* o="") const;
  74
  75   void     SetLocalTimeBin(Char_t t)       { fLocalTimeBin = t;     }
  76   void     SetInChamber(Bool_t in = kTRUE) { SetBit(kInChamber,in); }
  77   void     SetNPads(Int_t n)               { fNPads = n; }
  78   void     SetPadMaskedPosition(UChar_t position);
  79   void     SetPadMaskedStatus(UChar_t status);
  80   void     SetPadCol(UChar_t inPadCol){ fPadCol = inPadCol;}
  81   void     SetPadRow(UChar_t inPadRow){ fPadRow = inPadRow;}
  82   void     SetPadTime(UChar_t inPadTime){ fPadTime = inPadTime;}
  83   inline void SetRPhiMethod(ETRDclusterStatus m);
  84   void     SetDetector(Short_t inDetector){ fDetector = inDetector;}
  85   void     SetQ(Float_t inQ){ fQ = inQ;}
  86   void     SetClusterMasking(UChar_t inClusterMasking){ fClusterMasking = inClusterMasking;}
  87   void     SetShared(Bool_t sh  = kTRUE)   { SetBit(AliCluster::kShared,sh);    }
  88   void     SetSigmaY2(Float_t s2, Float_t dt, Float_t exb, Float_t x0, Float_t z=-1., Float_t tgp=0.);
  89   void     Use(Int_t u = 1)                  { SetBit(AliCluster::kUsed, u ? kTRUE : kFALSE);              }
  90   void     SetFivePad(Bool_t b = kTRUE) { SetBit(kFivePad,b);}
  91
  92 protected:
  93   UChar_t fPadCol;         //  Central pad number in column direction
  94   UChar_t fPadRow;         //  Central pad number in row direction
  95   UChar_t fPadTime;        //  Uncalibrated time bin number
  96   Char_t  fLocalTimeBin;   //  T0-calibrated time bin number
  97   UChar_t fNPads;          //  Number of pads in cluster
  98   UChar_t fClusterMasking; //  Bit field containing cluster status information;
  99   Short_t fDetector;       //  TRD detector number
 100   Short_t fSignals[7];     //  Signals in the cluster
 101   Float_t fQ;              //  Amplitude
 102   Float_t fCenter;         //  Center of the cluster relative to the pad
 103
 104 private:
 105   Float_t  GetDYcog(Double_t *const y1=0x0, Double_t *const y2=0x0);
 106   Float_t  GetDYlut(Double_t *const y1=0x0, Double_t *const y2=0x0);
 107   Float_t  GetDYgauss(Double_t sw, Double_t *const y1=0x0, Double_t *const y2=0x0);
 108   static void      FillLUT();
 109
 110   static const Int_t   fgkNlut;              //!  Number of bins of the LUT
 111   static Double_t     *fgLUT;                //! The lookup table
 112
 113   ClassDef(AliTRDcluster, 7)        //  Cluster for the TRD
 114 };
 115
 116 //________________________________________________
 117 inline Bool_t AliTRDcluster::IsRPhiMethod(ETRDclusterStatus m) const
 118 {
 119   if(m==kLUT && TestBit(kLUT)) return kTRUE;
 120   else if(m==kGAUS && TestBit(kGAUS)) return kTRUE;
 121   else if(m==kCOG && (!TestBit(kLUT)&&!TestBit(kGAUS))) return kTRUE;
 122
 123   return kFALSE;
 124 }
 125
 126 //________________________________________________
 127 inline void AliTRDcluster::SetRPhiMethod(ETRDclusterStatus m)
 128 {
 129   SetBit(kCOG,0);SetBit(kLUT,0);SetBit(kGAUS,0);
 130   switch(m){
 131   case kCOG: SetBit(kCOG); break;
 132   case kLUT: SetBit(kLUT); break;
 133   case kGAUS: SetBit(kGAUS); break;
 134   default: SetBit(kLUT); break;
 135   }
 136 }
 137
 138 #endif