[u/mrichter/AliRoot.git] / HMPID / AliHMPIDCluster.h

#ifndef AliHMPIDCluster_h
#define AliHMPIDCluster_h
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

#include "AliHMPIDDigit.h"  //DigAdd()
#include <TObjArray.h>     //DigAdd()      
class TClonesArray;        //Solve()

class AliHMPIDCluster :public TObject
{
public:
  enum EClusterStatus {kFrm,kCoG,kLo1,kUnf,kMax,kNot,kEdg,kSi1,kNoLoc,kEmp=-1};      //status flags    
  AliHMPIDCluster():TObject( ),fCh(-1),fSi(-1),fSt(kEmp),fBox(-1),fNlocMax(-1),fMaxQpad(-1),fMaxQ(-1),fQRaw(0),fQ(0),fErrQ(-1),fX(0),fErrX(-1),fY(0),fErrY(-1),fChi2(-1),fDigs(0) {} //empty ctor
  
  
  AliHMPIDCluster           (const AliHMPIDCluster &c):TObject(c),fCh(c.fCh),fSi(c.fSi),fSt(c.fSt),fBox(c.fBox),fNlocMax(c.fNlocMax),fMaxQpad(c.fMaxQpad),fMaxQ(c.fMaxQ),fQRaw(c.fQRaw),
                                                                  fQ (c.fQ ),fErrQ(c.fErrQ),
                                                                  fX (c.fX ),fErrX(c.fErrX),
                                                                  fY (c.fY ),fErrY(c.fErrY),fChi2(c.fChi2),fDigs(0)                  {}//copy ctor
  AliHMPIDCluster &operator=(const AliHMPIDCluster &c) {if(this == &c)return *this;TObject::operator=(c);          
                                                        fSi=c.fSi;  fSt=c.fSt; fCh=c.fCh; fBox=c.fBox;fNlocMax=c.fNlocMax;fMaxQpad=c.fMaxQpad; fMaxQ=c.fMaxQ;fQRaw=c.fQRaw;
                                                        fQ=c.fQ; fErrQ=c.fErrQ; 
                                                        fX=c.fX; fErrX=c.fErrX;
                                                        fY=c.fY; fErrY=c.fErrY; fChi2=c.fChi2;fDigs=c.fDigs ? new TObjArray(*c.fDigs):0; return *this;}
    
  virtual ~AliHMPIDCluster(                                     )                                                                        {if(fDigs) delete fDigs; fDigs=0;}
//framework part                   
         void           Draw   (Option_t *opt=""                                  );                       //overloaded TObject::Print() to draw cluster in current canvas
         void           Print  (Option_t *opt=""                                  )const;                  //overloaded TObject::Print() to print cluster info
  static void           FitFunc(Int_t &, Double_t *, Double_t &, Double_t *, Int_t);                       //fit function to be used by MINUIT
//private part  
         void           CoG      (                                         );                                                      //calculates center of gravity
         void           CorrSin  (                                         );                                                      //sinoidal correction   
         Int_t          Ch       (                                         )const{return fCh;                                    } //chamber number
  inline void           DigAdd   (AliHMPIDDigit *pDig                      );                                                      //add new digit ot the cluster
         AliHMPIDDigit* Dig      (Int_t i                                  )const{return (AliHMPIDDigit*)fDigs->At(i);           } //pointer to i-th digi 
  inline Bool_t         IsInPc   ();                                                                                        //check if is in the current PC
  inline void           Reset    (                                         );                                                      //cleans the cluster
         Int_t          Size     (                                         )const{return fSi;                                    } //returns number of pads in formed cluster 
         Int_t          Solve    (TClonesArray *pCluLst,Bool_t isUnfold    );                                                      //solve cluster: MINUIT fit or CoG
         Int_t          Status  (                                          ) const{return fSt;}                                   //Status of cluster                                  
         Double_t       QRaw     (                                         )const{return fQRaw;                                  } //raw cluster charge in QDC channels 
         Double_t       Q        (                                         )const{return fQ;                                     } //given cluster charge in QDC channels 
         Double_t       Qe       (                                         )const{return fErrQ;                                  } //Error in cluster charge in QDC channels 
         Double_t       X        (                                         )const{return fX;                                     } //cluster x position in LRS
         Double_t       Xe       (                                         )const{return fErrX;                                  } //cluster charge in QDC channels 
         Double_t       Y        (                                         )const{return fY;                                     } //cluster y position in LRS 
         Double_t       Ye       (                                         )const{return fErrY;                                    } //cluster charge in QDC channels 
         Double_t       Chi2     (                                         )const{return fChi2;                                  } 
protected:
  Int_t         fCh;          //chamber number
  Int_t         fSi;          //size of the formed cluster from which this cluster deduced
  Int_t         fSt;          //flag to mark the quality of the cluster
  Int_t         fBox;         //box contaning this cluster
  Int_t         fNlocMax;     //number of local maxima in formed cluster
  Int_t         fMaxQpad;     //abs pad number of a pad with the highest charge
  Double_t      fMaxQ;        //that max charge value
  Double_t      fQRaw;        //QDC value of the raw cluster
  Double_t      fQ;           //QDC value of the actual cluster
  Double_t      fErrQ;        //error on Q
  Double_t      fX;           //local x postion, [cm]
  Double_t      fErrX;        //error on x postion, [cm]
  Double_t      fY;           //local y postion, [cm]
  Double_t      fErrY;        //error on y postion, [cm]
  Double_t      fChi2;        //some estimator of the fit quality
  TObjArray    *fDigs;        //! list of digits forming this cluster
  ClassDef(AliHMPIDCluster,6) //HMPID cluster class
};//class AliHMPIDCluster

typedef AliHMPIDCluster AliRICHCluster; // for backward compatibility

//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
void AliHMPIDCluster::DigAdd(AliHMPIDDigit *pDig)
{
// Adds a given digit to the list of digits belonging to this cluster, cluster is not owner of digits
// Arguments: pDig - pointer to digit to be added  
//   Returns: none  
  if(!fDigs) {fSi=0;fDigs = new TObjArray;} //create list of digits in the first invocation
  fDigs->Add(pDig);
  fSt=kFrm;
  fSi++;
}
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
void AliHMPIDCluster::Reset()
{
  if(fDigs) delete fDigs; 
  fDigs=0; 
  fSt=kEmp;
  fQRaw=fQ=0;
  fX=fY=0;
  fCh=fSi=fBox=fNlocMax=fMaxQpad=-1;
  fMaxQ=fErrQ=fErrX=fErrY=fChi2=-1; //empty ctor
}
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Bool_t AliHMPIDCluster::IsInPc()
{
  Int_t pc = ((AliHMPIDDigit*)fDigs->At(0))->Pc();
 
  
  if ( fX < AliHMPIDDigit::fMinPcX[pc] || fX > AliHMPIDDigit::fMaxPcX[pc] || 
       fY < AliHMPIDDigit::fMinPcY[pc] || fY > AliHMPIDDigit::fMaxPcY[pc] ) return kFALSE;
  
  return kTRUE;
  
}

#endif
Commit	Line	Data
d3da6dc4	1	#ifndef AliHMPIDCluster_h
	2	#define AliHMPIDCluster_h
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	#include "AliHMPIDDigit.h" //DigAdd()
	7	#include <TObjArray.h> //DigAdd()
	8	class TClonesArray; //Solve()
	9
	10	class AliHMPIDCluster :public TObject
	11	{
	12	public:
e4a3eae8	13	enum EClusterStatus {kFrm,kCoG,kLo1,kUnf,kMax,kNot,kEdg,kSi1,kNoLoc,kEmp=-1}; //status flags
e4a3eae8	14	AliHMPIDCluster():TObject( ),fCh(-1),fSi(-1),fSt(kEmp),fBox(-1),fNlocMax(-1),fMaxQpad(-1),fMaxQ(-1),fQRaw(0),fQ(0),fErrQ(-1),fX(0),fErrX(-1),fY(0),fErrY(-1),fChi2(-1),fDigs(0) {} //empty ctor
d1bf51e1	15
d1bf51e1	16
e4a3eae8	17	AliHMPIDCluster (const AliHMPIDCluster &c):TObject(c),fCh(c.fCh),fSi(c.fSi),fSt(c.fSt),fBox(c.fBox),fNlocMax(c.fNlocMax),fMaxQpad(c.fMaxQpad),fMaxQ(c.fMaxQ),fQRaw(c.fQRaw),
	18	fQ (c.fQ ),fErrQ(c.fErrQ),
	19	fX (c.fX ),fErrX(c.fErrX),
	20	fY (c.fY ),fErrY(c.fErrY),fChi2(c.fChi2),fDigs(0) {}//copy ctor
d1bf51e1	21	AliHMPIDCluster &operator=(const AliHMPIDCluster &c) {if(this == &c)return *this;TObject::operator=(c);
e4a3eae8	22	fSi=c.fSi; fSt=c.fSt; fCh=c.fCh; fBox=c.fBox;fNlocMax=c.fNlocMax;fMaxQpad=c.fMaxQpad; fMaxQ=c.fMaxQ;fQRaw=c.fQRaw;
	23	fQ=c.fQ; fErrQ=c.fErrQ;
	24	fX=c.fX; fErrX=c.fErrX;
	25	fY=c.fY; fErrY=c.fErrY; fChi2=c.fChi2;fDigs=c.fDigs ? new TObjArray(c.fDigs):0; return this;}
d3da6dc4	26
d1bf51e1	27	virtual ~AliHMPIDCluster( ) {if(fDigs) delete fDigs; fDigs=0;}
d3da6dc4	28	//framework part
d1bf51e1	29	void Draw (Option_t *opt="" ); //overloaded TObject::Print() to draw cluster in current canvas
	30	void Print (Option_t *opt="" )const; //overloaded TObject::Print() to print cluster info
	31	static void FitFunc(Int_t &, Double_t , Double_t &, Double_t , Int_t); //fit function to be used by MINUIT
d3da6dc4	32	//private part
d1bf51e1	33	void CoG ( ); //calculates center of gravity
	34	void CorrSin ( ); //sinoidal correction
	35	Int_t Ch ( )const{return fCh; } //chamber number
	36	inline void DigAdd (AliHMPIDDigit *pDig ); //add new digit ot the cluster
	37	AliHMPIDDigit* Dig (Int_t i )const{return (AliHMPIDDigit*)fDigs->At(i); } //pointer to i-th digi
	38	inline Bool_t IsInPc (); //check if is in the current PC
	39	inline void Reset ( ); //cleans the cluster
	40	Int_t Size ( )const{return fSi; } //returns number of pads in formed cluster
	41	Int_t Solve (TClonesArray *pCluLst,Bool_t isUnfold ); //solve cluster: MINUIT fit or CoG
	42	Int_t Status ( ) const{return fSt;} //Status of cluster
e4a3eae8	43	Double_t QRaw ( )const{return fQRaw; } //raw cluster charge in QDC channels
	44	Double_t Q ( )const{return fQ; } //given cluster charge in QDC channels
	45	Double_t Qe ( )const{return fErrQ; } //Error in cluster charge in QDC channels
d1bf51e1	46	Double_t X ( )const{return fX; } //cluster x position in LRS
e4a3eae8	47	Double_t Xe ( )const{return fErrX; } //cluster charge in QDC channels
d1bf51e1	48	Double_t Y ( )const{return fY; } //cluster y position in LRS
e4a3eae8	49	Double_t Ye ( )const{return fErrY; } //cluster charge in QDC channels
d1bf51e1	50	Double_t Chi2 ( )const{return fChi2; }
d3da6dc4	51	protected:
d3da6dc4	52	Int_t fCh; //chamber number
e4a3eae8	53	Int_t fSi; //size of the formed cluster from which this cluster deduced
	54	Int_t fSt; //flag to mark the quality of the cluster
	55	Int_t fBox; //box contaning this cluster
	56	Int_t fNlocMax; //number of local maxima in formed cluster
d1bf51e1	57	Int_t fMaxQpad; //abs pad number of a pad with the highest charge
e4a3eae8	58	Double_t fMaxQ; //that max charge value
	59	Double_t fQRaw; //QDC value of the raw cluster
	60	Double_t fQ; //QDC value of the actual cluster
	61	Double_t fErrQ; //error on Q
	62	Double_t fX; //local x postion, [cm]
	63	Double_t fErrX; //error on x postion, [cm]
	64	Double_t fY; //local y postion, [cm]
	65	Double_t fErrY; //error on y postion, [cm]
	66	Double_t fChi2; //some estimator of the fit quality
d3da6dc4	67	TObjArray *fDigs; //! list of digits forming this cluster
e4a3eae8	68	ClassDef(AliHMPIDCluster,6) //HMPID cluster class
d3da6dc4	69	};//class AliHMPIDCluster
cf7e313e	70
	71	typedef AliHMPIDCluster AliRICHCluster; // for backward compatibility
	72
d3da6dc4	73	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	74	void AliHMPIDCluster::DigAdd(AliHMPIDDigit *pDig)
	75	{
	76	// Adds a given digit to the list of digits belonging to this cluster, cluster is not owner of digits
	77	// Arguments: pDig - pointer to digit to be added
	78	// Returns: none
d1bf51e1	79	if(!fDigs) {fSi=0;fDigs = new TObjArray;} //create list of digits in the first invocation
d3da6dc4	80	fDigs->Add(pDig);
d1bf51e1	81	fSt=kFrm;
	82	fSi++;
	83	}
	84	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	85	void AliHMPIDCluster::Reset()
	86	{
	87	if(fDigs) delete fDigs;
	88	fDigs=0;
e4a3eae8	89	fSt=kEmp;
	90	fQRaw=fQ=0;
	91	fX=fY=0;
	92	fCh=fSi=fBox=fNlocMax=fMaxQpad=-1;
	93	fMaxQ=fErrQ=fErrX=fErrY=fChi2=-1; //empty ctor
d3da6dc4	94	}
d3da6dc4	95	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
d1bf51e1	96	Bool_t AliHMPIDCluster::IsInPc()
	97	{
	98	Int_t pc = ((AliHMPIDDigit*)fDigs->At(0))->Pc();
	99
	100
	101	if ( fX < AliHMPIDDigit::fMinPcX[pc] \|\| fX > AliHMPIDDigit::fMaxPcX[pc] \|\|
	102	fY < AliHMPIDDigit::fMinPcY[pc] \|\| fY > AliHMPIDDigit::fMaxPcY[pc] ) return kFALSE;
	103
	104	return kTRUE;
	105
	106	}
	107
d3da6dc4	108	#endif