Including St2 in the new geometry segmentation (Christian)

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

diff --git a/TRD/AliTRDcluster.h b/TRD/AliTRDcluster.h
index 475614848e6f244eb34b5a0abda6027d9b4d81d2..189bc20dbbe04ea53f8c7fb6add17e1c488936f8 100644 (file)
--- a/TRD/AliTRDcluster.h
+++ b/TRD/AliTRDcluster.h
@@ -5,45 +5,61 @@
 
 /* $Id$ */
 
-#include <TObject.h>
+
+#include "AliCluster.h"  
 
 class AliTRDrecPoint;
 
-class AliTRDcluster : public TObject {
+class AliTRDcluster : public AliCluster {
 
  public:
 
-  AliTRDcluster();
-  AliTRDcluster(AliTRDrecPoint *rp);
-  
-  virtual Int_t   GetDetector() const           { return fDetector; };
-  virtual Int_t   GetLocalTimeBin() const       { return fTimeBin; }
+  AliTRDcluster() : AliCluster() { fQ=0; fTimeBin=0; fDetector=0; fNPads=0; }
+  AliTRDcluster(const AliTRDcluster &c);
+  AliTRDcluster(const AliTRDrecPoint &p);
 
-  virtual Float_t GetSigmaY2() const            { return fSigmaY2; }
-  virtual Float_t GetSigmaZ2() const            { return fSigmaZ2; }
-  virtual Float_t GetY() const                  { return fY; }
-  virtual Float_t GetZ() const                  { return fZ; }
-  virtual Float_t GetQ() const                  { return fQ; }
+  virtual void    AddTrackIndex(Int_t *i); 
 
-  Int_t   IsUsed() const                        { return (fQ<0) ? 1 : 0; }
-  void    Use()                                 { fQ=-fQ; }
-  Int_t   GetTrackIndex(Int_t i) const          { return fTracks[i]; }
 
+  Int_t   IsUsed() const      { return (fQ < 0) ? 1 : 0; }
+  void    Use()               { 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; }
+  
+  Int_t   GetDetector() const          { return fDetector; }
+  Int_t   GetLocalTimeBin() const      { return fTimeBin;  }
+  Float_t GetQ() const                 { return fQ; }
 
+  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;}
  protected:
 
-  Int_t    fDetector;    // TRD detector number
-  Int_t    fTimeBin;     // Time bin number within the detector
-
-  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) // Reconstructed point for the TRD
+  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
+  Int_t    fTimeBin;        // Time bin number within the detector
+  Float_t  fQ;              // amplitude 
+  Int_t    fNPads;          // number of pads in cluster
+  ClassDef(AliTRDcluster,1) // Cluster for the TRD
  
 };