EdbCluster.h

Go to the documentation of this file.

#ifndef ROOT_EdbCluster
#define ROOT_EdbCluster
 
 
#include "TClonesArray.h"
 
#ifndef ROOT_EdbVirtual
#include "EdbVirtual.h"
#endif
 
//______________________________________________________________________________
class EdbCluster : public TObject, public EdbPoint3D {
 
public:
 
  Float_t    eX;       
  Float_t    eY;       
  Float_t    eZ;       
  Float_t    eArea;    
  Float_t    eVolume;  
  Int_t      eFrame;   
  Int_t      eSide;    
  Int_t      eSegment; 
 
public:
   EdbCluster() { }
   EdbCluster( float x,  float y,  float z,  
           float a,  float v, int f, int s, int seg=-1 ) :
    eX(x), eY(y), eZ(z), 
     eArea(a),eVolume(v), eFrame(f), eSide(s), eSegment(seg) {}
   virtual ~EdbCluster() { }
 
   void Print( Option_t *opt=0 ) const;
 
   void Set0(); 
   void Set(float x, float y, float z, float a, float v, int f, int s, int seg=-1)
     {eX=x; eY=y; eZ=z; eArea=a; eVolume=v; eFrame=f; eSide=s; eSegment=seg;}
   void SetArea(float a)    { eArea=a;}
   void SetVolume(float v)  { eVolume=v;}
   void SetFrame(int f)     { eFrame=f;}
   void SetSide(int s)      { eSide=s;}
   void SetSegment(int seg) { eSegment=seg;}
 
   Float_t  GetX()         const  { return eX      ;}
   Float_t  GetY()         const  { return eY      ;}
   Float_t  GetZ()         const  { return eZ      ;}
   Float_t  GetArea()      const  { return eArea   ;}
   Float_t  GetVolume()    const  { return eVolume ;}
   Int_t    GetFrame()     const  { return eFrame  ;}
   Int_t    GetSide()      const  { return eSide   ;}
   Int_t    GetSegment()   const  { return eSegment;}
 
   void AddPixelSum( int x,  int y, unsigned char pix )
     {
       eArea++;
       eVolume += pix ;
       eX += x;
       eY += y;
     }
 
   void  Normalize()
     {
       eX = eX/eArea - 0.5;
       eY = eY/eArea - 0.5;
     }
 
   //mandatory virtual functions
   
   virtual Float_t    X()          const { return eX; }
   virtual Float_t    Y()          const { return eY; }
   virtual Float_t    Z()          const { return eZ; }
   virtual void    SetX( float x )       { eX=x; }
   virtual void    SetY( float y )       { eY=y; }
   virtual void    SetZ( float z )       { eZ=z; }
 
   //other functions
   void Draw(Option_t *opt="") const;
 
   void TT() const { Test(); }
 
  Bool_t  IsSortable() const { return kTRUE; }
  Bool_t  IsEqual(const TObject *obj) const
    {
      const EdbCluster *c=(EdbCluster*)obj;
      if(c->eFrame  != eFrame)  return false;
      if(c->eX      != eX)      return false;
      if(c->eY      != eY)      return false;
      return true;
    }
 
  Int_t   Compare(const TObject *obj) const
  {
    const EdbCluster *c=(EdbCluster*)obj;
    Long_t f1=0, f2=0;
    f1 = eFrame*1000000+int(eX)*10000+int(eY);
    f2 = c->eFrame*1000000+int(c->eX)*10000+int(c->eY);
    if (f1>f2)
      return 1;
    else if (f1<f2)
      return -1;
    else
      return 0;
  }
 
   ClassDef(EdbCluster,2)  // single cluster
};
 
//______________________________________________________________________________
class EdbClustersBox : public TObject, public EdbPointsBox3D {
 
 private:
 
  TClonesArray *eClusters;  
 
 public: 
  EdbClustersBox();
  EdbClustersBox( int n );
  virtual ~EdbClustersBox();
 
  //members access functions
  void          AddCluster( EdbCluster *c );
  void          AddCluster( float x,  float y,  float z,
                float a,  float v, int f, int s, int seg );
 
  Int_t         GetN()             const;
  EdbCluster    *GetCluster(int i) const;
  TClonesArray  *GetClusters()     const { return eClusters; }
 
  // mandatory virtual functions:
  Int_t         N()       const { return GetN(); }
  EdbPoint     *At(int i)       { return (EdbPoint*)GetCluster(i); }
 
  //  void  Draw(int style=23, int   col=4, float size=1. ){}
 
  // other finctions
  // void       Print( Option_t *opt="") const;
 
  ClassDef(EdbClustersBox,2)  // collection of clusters
};
 
 
#endif /* ROOT_EdbCluster */