forgot to include baseLinkDef file update in yesterdays commit; sorry, and thanks...

[u/mrichter/AliRoot.git] / TRD / AliTRDcluster.h

diff --git a/TRD/AliTRDcluster.h b/TRD/AliTRDcluster.h
index 906a25913fda49714f63952259c4d91f1e2e8713..52a46b279092a5585664981bc8f5e4c225124fe1 100644 (file)
--- a/TRD/AliTRDcluster.h
+++ b/TRD/AliTRDcluster.h
@@ -13,12 +13,16 @@
 
 #include "AliCluster.h"
 
+class AliTRDtrackletWord;
+
 class AliTRDcluster : public AliCluster {
 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
@@ -27,11 +31,13 @@ public:
   };
 
   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);
   virtual ~AliTRDcluster() {};
 
@@ -44,6 +50,7 @@ public:
   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; }
@@ -57,23 +64,32 @@ public:
   Int_t    GetPadTime() const              { return fPadTime;       }
   Short_t *GetSignals()                    { return fSignals;       }
   Float_t  GetSumS() const;
-  Float_t  GetXpos(Float_t t0, Float_t vd, Float_t *const q=0x0);
-  Float_t  GetYpos(Float_t Sigma2, Float_t PadWidth, Float_t *const yPos1=0x0, Float_t *const yPos2=0x0);
+  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);
 
   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     Use(Int_t = 0)                  { SetBit(AliCluster::kUsed, kTRUE);              }
+  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:
@@ -88,6 +104,38 @@ protected:
   Float_t fQ;              //  Amplitude 
   Float_t fCenter;         //  Center of the cluster relative to the pad 
 
-  ClassDef(AliTRDcluster, 6)        //  Cluster for the TRD
+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