]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - TRD/AliTRDcluster.h
Fix for memory leaks in digits TClonesArrays (Matevz)
[u/mrichter/AliRoot.git] / TRD / AliTRDcluster.h
index 032f073b5d40022f7ac6c5aab480a97504ef1f38..f2b7009033446df5777f4ba37405eadb54339560 100644 (file)
 
 /* $Id$ */
 
-#include <TObject.h>
-
-class AliTRDrecPoint;
-
-class AliTRDcluster : public TObject {
-
- public:
+///////////////////////////////////////////////////////////////////////////////
+//                                                                           //
+//  TRD cluster                                                              //
+//                                                                           //
+///////////////////////////////////////////////////////////////////////////////
+
+#include "AliCluster.h"
+
+class AliTRDtrackletWord;
+
+class AliTRDcluster : public AliCluster {
+  friend class AliHLTTRDCluster;
+
+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 * const 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();
-  AliTRDcluster &operator=(const AliTRDcluster &c);
-
-  virtual void    Copy(TObject &c);
-  virtual void    AddTrackIndex(Int_t *i);
-   
-          Int_t   GetDetector() const             { return fDetector; }
-          Int_t   GetLocalTimeBin() const         { return fTimeBin;  }
-
-          Float_t GetSigmaY2() const              { return fSigmaY2; }
-          Float_t GetSigmaZ2() const              { return fSigmaZ2; }
-          Float_t GetY() const                    { return fY; }
-          Float_t GetZ() const                    { return fZ; }
-          Float_t GetQ() const                    { return fQ; }
-
-          Int_t   IsUsed() const                  { return (fQ < 0) ? 1 : 0; }
-          void    Use()                           { fQ = -fQ; }
-          Int_t   GetTrackIndex(Int_t i) const    { return fTracks[i]; }
-
-          Bool_t  FromUnfolding() const           { return TestBit(kUnfold); }
-
-          void    SetDetector(Int_t d)            { fDetector  = d; }
-          void    SetLocalTimeBin(Int_t t)        { fTimeBin   = t; }
-          void    SetQ(Float_t q)                 { fQ         = q; }
-          void    SetY(Float_t y)                 { fY         = y; }
-          void    SetZ(Float_t z)                 { fZ         = z; }
-          void    SetTrackIndex(Int_t i, Int_t t) { fTracks[i] = t; } 
-          void    SetSigmaY2(Float_t s)           { fSigmaY2   = s; }
-          void    SetSigmaZ2(Float_t s)           { fSigmaZ2   = s; }
-
-          void    SetUnfolding()                  { SetBit(kUnfold); }
-
- protected:
-
-  enum {
-    kUnfold = 0x00000001    // Cluster results from unfolding procedure
-  };
-
-  Int_t    fDetector;       // TRD detector number
-  Int_t    fTimeBin;        // Time bin number within the detector
+  virtual ~AliTRDcluster() {};
+  AliTRDcluster    &operator=(const AliTRDcluster &c);
+
+  virtual void      AddTrackIndex(const Int_t * const 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
+                          , const Double_t *const q = 0x0
+                          , const 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        SetNPads(Int_t n)                           { fNPads          = n;                }
+  void        SetPadCol(UChar_t inPadCol)                 { fPadCol         = inPadCol;         }
+  void        SetPadRow(UChar_t inPadRow)                 { fPadRow         = inPadRow;         }
+  void        SetPadTime(UChar_t inPadTime)               { fPadTime        = inPadTime;        }
+  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        SetFivePad(Bool_t b = kTRUE)                { SetBit(kFivePad,b);                 }
+  void        SetInChamber(Bool_t in = kTRUE)             { SetBit(kInChamber,in);              }
+  void        SetPadMaskedPosition(UChar_t position);
+  void        SetPadMaskedStatus(UChar_t status);
+  void        SetSigmaY2(Float_t s2, Float_t dt, Float_t exb, Float_t x0, Float_t z=-1., Float_t tgp=0.);
+  inline void SetRPhiMethod(ETRDclusterStatus m);
+
+  void        Use(Int_t u = 1)                            { SetBit(AliCluster::kUsed, u ? kTRUE : kFALSE); }
+
+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(const Double_t *const y1=0x0, const Double_t *const y2=0x0);
+         Float_t   GetDYlut(const Double_t *const y1=0x0, const 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
 
-  Int_t    fTracks[3];      // labels of overlapped tracks
-  Float_t  fQ;              // amplitude 
-  Float_t  fY;              // local Rphi coordinate (cm) within tracking sector
-  Float_t  fZ;              // local Z coordinate (cm) within tracking sector
-  Float_t  fSigmaY2;        // Y variance (cm)
-  Float_t  fSigmaZ2;        // Z variance (cm)  
-
-  ClassDef(AliTRDcluster,1) // 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