]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - TRD/AliTRDcluster.h
forgot to include baseLinkDef file update in yesterdays commit; sorry, and thanks...
[u/mrichter/AliRoot.git] / TRD / AliTRDcluster.h
index de440616e1111206acdae97947de871b5743ba52..52a46b279092a5585664981bc8f5e4c225124fe1 100644 (file)
 
 /* $Id$ */
 
+///////////////////////////////////////////////////////////////////////////////
+//                                                                           //
+//  TRD cluster                                                              //
+//                                                                           //
+///////////////////////////////////////////////////////////////////////////////
 
-#include "AliCluster.h"  
+#include "AliCluster.h"
 
-class AliTRDrecPoint;
+class AliTRDtrackletWord;
 
 class AliTRDcluster : public AliCluster {
-
- public:
+public:
+  enum ETRDclusterStatus { 
+    kInChamber = BIT(16) // Out of fiducial volume of chamber (signal tails)
+   ,kFivePad   = BIT(17) // Deconvoluted clusters
+   ,kLUT       = BIT(18)
+   ,kGAUS      = BIT(19)
+   ,kCOG       = BIT(20)
+  };
+  enum ETRDclusterMask { 
+    kMaskedLeft   = 0
+   ,kMaskedCenter = 1
+   ,kMaskedRight  = 2
+  };
 
   AliTRDcluster();
+  AliTRDcluster(Int_t det, UChar_t col, UChar_t row, UChar_t time, const Short_t *sig, UShort_t volid);
+  AliTRDcluster(
+    Int_t det, Float_t q, Float_t *pos, Float_t *sig,
+    Int_t *tracks, Char_t npads, Short_t *signals,
+    UChar_t col, UChar_t row, UChar_t time,
+    Char_t timebin, Float_t center, UShort_t volid);
+  AliTRDcluster(const AliTRDtrackletWord *const tracklet, Int_t det, UShort_t volid);
   AliTRDcluster(const AliTRDcluster &c);
-  AliTRDcluster(const AliTRDrecPoint &p);
+  virtual ~AliTRDcluster() {};
 
-  virtual void    AddTrackIndex(Int_t *i); 
+  virtual void     AddTrackIndex(Int_t *i); 
+  void     Clear(Option_t *o="");
+  
+  Bool_t   IsEqual(const TObject *otherCluster) const;
+  Bool_t   IsInChamber() const             { return TestBit(kInChamber);       }
+  Bool_t   IsMasked() const                { return fClusterMasking ? kTRUE : kFALSE; }
+  Bool_t   IsShared() const                { return IsClusterShared();}
+  Bool_t   IsUsed() const                  { return IsClusterUsed(); }
+  Bool_t   IsFivePad() const               { return TestBit(kFivePad);}
+  inline Bool_t IsRPhiMethod(ETRDclusterStatus m) const;
 
+  UChar_t  GetPadMaskedPosition() const    { return fClusterMasking & 7; }
+  UChar_t  GetPadMaskedStatus() const      { return fClusterMasking >> 3; }
+  Int_t    GetDetector() const             { return fDetector;      }
+  Int_t    GetLocalTimeBin() const         { return fLocalTimeBin;  }
+  Float_t  GetQ() const                    { return fQ;             }
+  Int_t    GetNPads() const                { return fNPads;         }
+  Float_t  GetCenter() const               { return fCenter;        }
+  Int_t    GetPadCol() const               { return fPadCol;        }
+  Int_t    GetPadRow() const               { return fPadRow;        }
+  Int_t    GetPadTime() const              { return fPadTime;       }
+  Short_t *GetSignals()                    { return fSignals;       }
+  Float_t  GetSumS() const;
+  static Double_t  GetSX(Int_t tb, Double_t z=-1);
+  static Double_t  GetSYdrift(Int_t tb, Int_t ly=0, Double_t z=-1);
+  static Double_t  GetSYcharge(Float_t q);
+  static Double_t  GetSYprf(Int_t ly, Double_t center, Double_t s2);
+  static Double_t  GetXcorr(Int_t tb, Double_t z=-1);
+  static Double_t  GetYcorr(Int_t ly, Float_t y);
+  Float_t  GetXloc(Double_t t0, Double_t vd, Double_t *const q=0x0, Double_t *const xq=0x0, Double_t z = 0.2);
+  Float_t  GetYloc(Double_t y0, Double_t s2, Double_t W, Double_t *const y1=0x0, Double_t *const y2=0x0);
 
-  Int_t   IsUsed() const      { return (fQ < 0) ? 1 : 0; }
-  void    Use(Int_t = 0)               { fQ = -fQ; }
-  
-  Bool_t  From2pad() const    { return TestBit(k2pad); }
-  Bool_t  From3pad() const    { return TestBit(k3pad); }
-  Bool_t  From4pad() const    { return TestBit(k4pad); }
-  Bool_t  From5pad() const    { return TestBit(k5pad); }
-  Bool_t  FromLarge() const   { return TestBit(kLarge);}
-  Bool_t  Isolated() const    { return (TestBit(k2pad) || TestBit(k3pad)); }
-  void    SetDetector(Int_t d)         { fDetector  = d; }
-  void    SetLocalTimeBin(Int_t t)     { fTimeBin   = t; }
-  void    SetQ(Float_t q)              { fQ         = q; }
-  void    SetX(Float_t x)              { fX         = x; }
-  void    SetSignals(Short_t *signals);
-  
-  Int_t   GetDetector() const          { return fDetector; }
-  Int_t   GetLocalTimeBin() const      { return fTimeBin;  }
-  Float_t GetQ() const                 { return fQ; }
-  Float_t GetX() const                 { return fX; }
-
-  void    Set2pad()                    { SetBit(k2pad); fNPads=2; }
-  void    Set3pad()                    { SetBit(k3pad); fNPads=3; }
-  void    Set4pad()                    { SetBit(k4pad); fNPads=4; }
-  void    Set5pad()                    { SetBit(k5pad); fNPads=5; }
-  void    SetLarge()                   { SetBit(kLarge);fNPads=6; }
-  Int_t   GetNPads() const {return fNPads;}
-  void    SetCenter(Float_t center){fCenter =center;}
-  Float_t GetCenter() const {return fCenter;}
-  Float_t GetSumS() const;
-  Float_t GetCenterS() const;
- protected:
-
-  enum {
-    k2pad  = 0x00000001,   // 2 pad cluster
-    k3pad  = 0x00000002,   // 3 pad cluster
-    k4pad  = 0x00000004,   // 4 pad cluster
-    k5pad  = 0x00000008,   // 5 pad cluster
-    kLarge = 0x00000016    // Large cluster
-  };
-  
-  Int_t   fDetector;       // TRD detector number
-  Float_t  fX;              // local (in this detector) x pos (first order proportional to time bin (depends on local drift velocity!), (x=0 corresponds timebin=0, thus x increases TOWARDS interaction point )
-  Char_t    fTimeBin;        // Time bin number within the detector
-  Float_t   fQ;              // amplitude 
-  Char_t    fNPads;          // number of pads in cluster
-  Float_t   fCenter;         // center of the cluster relative to the pad 
-  Short_t   fSignals[7];     // signals in the cluster
-  
-  ClassDef(AliTRDcluster,3) // Cluster for the TRD
+  void     Print(Option_t* o="") const;
+
+  void     SetLocalTimeBin(Char_t t)       { fLocalTimeBin = t;     }
+  void     SetInChamber(Bool_t in = kTRUE) { SetBit(kInChamber,in); }
+  void     SetNPads(Int_t n)               { fNPads = n; }
+  void     SetPadMaskedPosition(UChar_t position);
+  void     SetPadMaskedStatus(UChar_t status);
+  void     SetPadCol(UChar_t inPadCol){ fPadCol = inPadCol;}
+  void     SetPadRow(UChar_t inPadRow){ fPadRow = inPadRow;}
+  void     SetPadTime(UChar_t inPadTime){ fPadTime = inPadTime;}
+  inline void SetRPhiMethod(ETRDclusterStatus m);
+  void     SetDetector(Short_t inDetector){ fDetector = inDetector;}
+  void     SetQ(Float_t inQ){ fQ = inQ;}
+  void     SetClusterMasking(UChar_t inClusterMasking){ fClusterMasking = inClusterMasking;}
+  void     SetShared(Bool_t sh  = kTRUE)   { SetBit(AliCluster::kShared,sh);    }
+  void     SetSigmaY2(Float_t s2, Float_t dt, Float_t exb, Float_t x0, Float_t z=-1., Float_t tgp=0.);
+  void     Use(Int_t u = 1)                  { SetBit(AliCluster::kUsed, u ? kTRUE : kFALSE);              }
+  void     SetFivePad(Bool_t b = kTRUE) { SetBit(kFivePad,b);}
+
+protected:
+  UChar_t fPadCol;         //  Central pad number in column direction
+  UChar_t fPadRow;         //  Central pad number in row direction
+  UChar_t fPadTime;        //  Uncalibrated time bin number
+  Char_t  fLocalTimeBin;   //  T0-calibrated time bin number
+  UChar_t fNPads;          //  Number of pads in cluster
+  UChar_t fClusterMasking; //  Bit field containing cluster status information;
+  Short_t fDetector;       //  TRD detector number
+  Short_t fSignals[7];     //  Signals in the cluster
+  Float_t fQ;              //  Amplitude 
+  Float_t fCenter;         //  Center of the cluster relative to the pad 
+
+private:
+  Float_t  GetDYcog(Double_t *const y1=0x0, Double_t *const y2=0x0);
+  Float_t  GetDYlut(Double_t *const y1=0x0, Double_t *const y2=0x0);
+  Float_t  GetDYgauss(Double_t sw, Double_t *const y1=0x0, Double_t *const y2=0x0);
+  static void      FillLUT();
+
+  static const Int_t   fgkNlut;              //!  Number of bins of the LUT
+  static Double_t     *fgLUT;                //! The lookup table
+
+  ClassDef(AliTRDcluster, 7)        //  Cluster for the TRD
 };
 
+//________________________________________________
+inline Bool_t AliTRDcluster::IsRPhiMethod(ETRDclusterStatus m) const
+{
+  if(m==kLUT && TestBit(kLUT)) return kTRUE;
+  else if(m==kGAUS && TestBit(kGAUS)) return kTRUE;
+  else if(m==kCOG && (!TestBit(kLUT)&&!TestBit(kGAUS))) return kTRUE;
+
+  return kFALSE;
+}
+
+//________________________________________________
+inline void AliTRDcluster::SetRPhiMethod(ETRDclusterStatus m)
+{
+  SetBit(kCOG,0);SetBit(kLUT,0);SetBit(kGAUS,0);
+  switch(m){
+  case kCOG: SetBit(kCOG); break;
+  case kLUT: SetBit(kLUT); break;
+  case kGAUS: SetBit(kGAUS); break;
+  default: SetBit(kLUT); break;
+  }
+}
 
 #endif