[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                               */
//
// Implementation class: AliHMPIDCluster
//   
//   class to reconstruct clester in HMPID
//   it is forseen to Solve (split) the raw cluster in 
//   several clusters (# local maxima in the raw cluster -> deconvolution
//   according to a Mathieson profile of the charge 
//
#include "AliGeomManager.h"
#include "AliCluster3D.h"
#include "AliHMPIDDigit.h"  //DigAdd()
#include <TObjArray.h>     //DigAdd()      
class TClonesArray;        //Solve()

class AliHMPIDCluster :public AliCluster3D
{
public:
  enum EClusterStatus {kFrm,kCoG,kLo1,kUnf,kMax,kNot,kEdg,kSi1,kNoLoc,kAbn,kBig,kEmp=-1};      //status flags    
      AliHMPIDCluster():AliCluster3D(),
                          fCh(-1),fSi(-1),fSt(kEmp),fBox(-1),fNlocMax(-1),fMaxQpad(-1),fMaxQ(-1),fQRaw(0),
                          fQ(0),fErrQ(-1),fXX(0),fErrX(-1),fYY(0),fErrY(-1),fChi2(-1),fDigs(0),fParam(AliHMPIDParam::Instance())
      {
      }//ctor
  

      AliHMPIDCluster(const AliHMPIDCluster &c):AliCluster3D(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),
                        fXX (c.fXX ),fErrX(c.fErrX),
                        fYY (c.fYY ),fErrY(c.fErrY),fChi2(c.fChi2),fDigs(0),fParam(c.fParam)                  {}//copy ctor
   virtual ~AliHMPIDCluster();//dtor   {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 &iNpars, Double_t* deriv, Double_t &chi2, Double_t *par, Int_t iflag);//fit function to be used by MINUIT
//private part  
         Int_t          Box      (                                         )const{return fBox;                                   } //Dimension of the cluster
         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
  void           SetClusterParams(Double_t xL,Double_t yL,Int_t iCh  );                                                            //Set AliCluster3D part
         Int_t          Size     (                                         )const{return fSi;                                    } //returns number of pads in formed cluster 
         Int_t          Solve    (TClonesArray *pCluLst,Int_t *pSigmaCut, 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 fXX;                                    } //cluster x position in LRS
         Double_t       Xe       (                                         )const{return fErrX;                                  } //cluster charge in QDC channels 
         Double_t       Y        (                                         )const{return fYY;                                    } //cluster y position in LRS 
         Double_t       Ye       (                                         )const{return fErrY;                                  } //cluster charge in QDC channels 
         Double_t       Chi2     (                                         )const{return fChi2;                                  } //chi2 of the fit
         void           DoCorrSin(Bool_t doCorrSin                         ){fgDoCorrSin=doCorrSin;}                               // Set sinoidal correction
         void           SetX     (Double_t x                               ){fXX=x;}                                               // Setter
         void           SetY     (Double_t y                               ){fYY=y;}                                               // Setter
         void           SetSize  (Int_t size                               ){fSi=size;}                                            // Setter
         void           FindClusterSize(Int_t i,Int_t *pSigmaCut);                                                                 //Find the clusterSize of deconvoluted clusters 
 virtual void	        Clear(const Option_t*) { delete [] fDigs; fDigs=0; delete [] fParam; fParam=0; }
         
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      fXX;           //local x postion, [cm]
  Double_t      fErrX;        //error on x postion, [cm]
  Double_t      fYY;           //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
  static  Bool_t fgDoCorrSin; //flag to switch on/off correction for Sinusoidal to cluster reco
  AliHMPIDParam *fParam;      //!Pointer to AliHMPIDParam
  
private:
/*
  AliHMPIDCluster &operator=(const AliHMPIDCluster &c) {if(this == &c)return *this;AliCluster3D::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; 
                                                        fXX=c.fXX; fErrX=c.fErrX;
                                                        fYY=c.fYY; fErrY=c.fErrY; fChi2=c.fChi2;fDigs=c.fDigs ? new TObjArray(*c.fDigs):0; return *this;}
*/
//  AliHMPIDCluster(const AliHMPIDCluster& c);              //dummy copy constructor
  AliHMPIDCluster &operator=(const AliHMPIDCluster& c);   //dummy assignment operator
      
  
  ClassDef(AliHMPIDCluster,9) //HMPID cluster class
};

//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
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;
  fXX=fYY=0;
  fCh=fSi=fBox=fNlocMax=fMaxQpad=-1;
  fMaxQ=fErrQ=fErrX=fErrY=fChi2=-1; //empty ctor
}
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Bool_t AliHMPIDCluster::IsInPc()
{
  //Check if (X,Y) position is inside the PC limits
  //Arguments:
  //  Returns: True or False
  Int_t pc = ((AliHMPIDDigit*)fDigs->At(0))->Pc();
 
  
  if ( fXX < AliHMPIDParam::MinPcX(pc) || fXX > AliHMPIDParam::MaxPcX(pc) || 
       fYY < AliHMPIDParam::MinPcY(pc) || fYY > AliHMPIDParam::MaxPcY(pc) ) return kFALSE;
  
  return kTRUE;
  
}

#endif
Commit	Line	Data
	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	// Implementation class: AliHMPIDCluster
	7	//
	8	// class to reconstruct clester in HMPID
	9	// it is forseen to Solve (split) the raw cluster in
	10	// several clusters (# local maxima in the raw cluster -> deconvolution
	11	// according to a Mathieson profile of the charge
	12	//
	13	#include "AliGeomManager.h"
	14	#include "AliCluster3D.h"
	15	#include "AliHMPIDDigit.h" //DigAdd()
	16	#include <TObjArray.h> //DigAdd()
	17	class TClonesArray; //Solve()
	18
	19	class AliHMPIDCluster :public AliCluster3D
	20	{
	21	public:
	22	enum EClusterStatus {kFrm,kCoG,kLo1,kUnf,kMax,kNot,kEdg,kSi1,kNoLoc,kAbn,kBig,kEmp=-1}; //status flags
	23	AliHMPIDCluster():AliCluster3D(),
	24	fCh(-1),fSi(-1),fSt(kEmp),fBox(-1),fNlocMax(-1),fMaxQpad(-1),fMaxQ(-1),fQRaw(0),
	25	fQ(0),fErrQ(-1),fXX(0),fErrX(-1),fYY(0),fErrY(-1),fChi2(-1),fDigs(0),fParam(AliHMPIDParam::Instance())
	26	{
	27	}//ctor
	28
	29
	30	AliHMPIDCluster(const AliHMPIDCluster &c):AliCluster3D(c),
	31	fCh(c.fCh),fSi(c.fSi),fSt(c.fSt),fBox(c.fBox),fNlocMax(c.fNlocMax),fMaxQpad(c.fMaxQpad),fMaxQ(c.fMaxQ),fQRaw(c.fQRaw),
	32	fQ (c.fQ ),fErrQ(c.fErrQ),
	33	fXX (c.fXX ),fErrX(c.fErrX),
	34	fYY (c.fYY ),fErrY(c.fErrY),fChi2(c.fChi2),fDigs(0),fParam(c.fParam) {}//copy ctor
	35	virtual ~AliHMPIDCluster();//dtor {if(fDigs) delete fDigs; fDigs=0;}
	36	//framework part
	37	void Draw (Option_t *opt="" ); //overloaded TObject::Print() to draw cluster in current canvas
	38	void Print (Option_t *opt="" )const; //overloaded TObject::Print() to print cluster info
	39	static void FitFunc(Int_t &iNpars, Double_t* deriv, Double_t &chi2, Double_t *par, Int_t iflag);//fit function to be used by MINUIT
	40	//private part
	41	Int_t Box ( )const{return fBox; } //Dimension of the cluster
	42	void CoG ( ); //calculates center of gravity
	43	void CorrSin ( ); //sinoidal correction
	44	Int_t Ch ( )const{return fCh; } //chamber number
	45	inline void DigAdd (AliHMPIDDigit *pDig ); //add new digit ot the cluster
	46	AliHMPIDDigit* Dig (Int_t i )const{return (AliHMPIDDigit*)fDigs->At(i); } //pointer to i-th digi
	47	inline Bool_t IsInPc (); //check if is in the current PC
	48	inline void Reset ( ); //cleans the cluster
	49	void SetClusterParams(Double_t xL,Double_t yL,Int_t iCh ); //Set AliCluster3D part
	50	Int_t Size ( )const{return fSi; } //returns number of pads in formed cluster
	51	Int_t Solve (TClonesArray pCluLst,Int_t pSigmaCut, Bool_t isUnfold); //solve cluster: MINUIT fit or CoG
	52	Int_t Status ( ) const{return fSt;} //Status of cluster
	53	Double_t QRaw ( )const{return fQRaw; } //raw cluster charge in QDC channels
	54	Double_t Q ( )const{return fQ; } //given cluster charge in QDC channels
	55	Double_t Qe ( )const{return fErrQ; } //Error in cluster charge in QDC channels
	56	Double_t X ( )const{return fXX; } //cluster x position in LRS
	57	Double_t Xe ( )const{return fErrX; } //cluster charge in QDC channels
	58	Double_t Y ( )const{return fYY; } //cluster y position in LRS
	59	Double_t Ye ( )const{return fErrY; } //cluster charge in QDC channels
	60	Double_t Chi2 ( )const{return fChi2; } //chi2 of the fit
	61	void DoCorrSin(Bool_t doCorrSin ){fgDoCorrSin=doCorrSin;} // Set sinoidal correction
	62	void SetX (Double_t x ){fXX=x;} // Setter
	63	void SetY (Double_t y ){fYY=y;} // Setter
	64	void SetSize (Int_t size ){fSi=size;} // Setter
	65	void FindClusterSize(Int_t i,Int_t *pSigmaCut); //Find the clusterSize of deconvoluted clusters
	66	virtual void Clear(const Option_t*) { delete [] fDigs; fDigs=0; delete [] fParam; fParam=0; }
	67
	68	protected:
	69	Int_t fCh; //chamber number
	70	Int_t fSi; //size of the formed cluster from which this cluster deduced
	71	Int_t fSt; //flag to mark the quality of the cluster
	72	Int_t fBox; //box contaning this cluster
	73	Int_t fNlocMax; //number of local maxima in formed cluster
	74	Int_t fMaxQpad; //abs pad number of a pad with the highest charge
	75	Double_t fMaxQ; //that max charge value
	76	Double_t fQRaw; //QDC value of the raw cluster
	77	Double_t fQ; //QDC value of the actual cluster
	78	Double_t fErrQ; //error on Q
	79	Double_t fXX; //local x postion, [cm]
	80	Double_t fErrX; //error on x postion, [cm]
	81	Double_t fYY; //local y postion, [cm]
	82	Double_t fErrY; //error on y postion, [cm]
	83	Double_t fChi2; //some estimator of the fit quality
	84	TObjArray *fDigs; //! list of digits forming this cluster
	85	static Bool_t fgDoCorrSin; //flag to switch on/off correction for Sinusoidal to cluster reco
	86	AliHMPIDParam *fParam; //!Pointer to AliHMPIDParam
	87
	88	private:
	89	/*
	90	AliHMPIDCluster &operator=(const AliHMPIDCluster &c) {if(this == &c)return *this;AliCluster3D::operator=(c);
	91	fSi=c.fSi; fSt=c.fSt; fCh=c.fCh; fBox=c.fBox;fNlocMax=c.fNlocMax;fMaxQpad=c.fMaxQpad; fMaxQ=c.fMaxQ;fQRaw=c.fQRaw;
	92	fQ=c.fQ; fErrQ=c.fErrQ;
	93	fXX=c.fXX; fErrX=c.fErrX;
	94	fYY=c.fYY; fErrY=c.fErrY; fChi2=c.fChi2;fDigs=c.fDigs ? new TObjArray(c.fDigs):0; return this;}
	95	*/
	96	// AliHMPIDCluster(const AliHMPIDCluster& c); //dummy copy constructor
	97	AliHMPIDCluster &operator=(const AliHMPIDCluster& c); //dummy assignment operator
	98
	99
	100	ClassDef(AliHMPIDCluster,9) //HMPID cluster class
	101	};
	102
	103	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	104	void AliHMPIDCluster::DigAdd(AliHMPIDDigit *pDig)
	105	{
	106	// Adds a given digit to the list of digits belonging to this cluster, cluster is not owner of digits
	107	// Arguments: pDig - pointer to digit to be added
	108	// Returns: none
	109	if(!fDigs) {fSi=0;fDigs = new TObjArray;} //create list of digits in the first invocation
	110	fDigs->Add(pDig);
	111	fSt=kFrm;
	112	fSi++;
	113	}
	114	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	115	void AliHMPIDCluster::Reset()
	116	{
	117	//
	118	//
	119	//
	120	if(fDigs) delete fDigs;
	121	fDigs=0;
	122	fSt=kEmp;
	123	fQRaw=fQ=0;
	124	fXX=fYY=0;
	125	fCh=fSi=fBox=fNlocMax=fMaxQpad=-1;
	126	fMaxQ=fErrQ=fErrX=fErrY=fChi2=-1; //empty ctor
	127	}
	128	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	129	Bool_t AliHMPIDCluster::IsInPc()
	130	{
	131	//Check if (X,Y) position is inside the PC limits
	132	//Arguments:
	133	// Returns: True or False
	134	Int_t pc = ((AliHMPIDDigit*)fDigs->At(0))->Pc();
	135
	136
	137	if ( fXX < AliHMPIDParam::MinPcX(pc) \|\| fXX > AliHMPIDParam::MaxPcX(pc) \|\|
	138	fYY < AliHMPIDParam::MinPcY(pc) \|\| fYY > AliHMPIDParam::MaxPcY(pc) ) return kFALSE;
	139
	140	return kTRUE;
	141
	142	}
	143
	144	#endif