[u/mrichter/AliRoot.git] / EMCAL / AliEMCALRecoUtils.h

#ifndef ALIEMCALRECOUTILS_H
#define ALIEMCALRECOUTILS_H

/* $Id: AliEMCALRecoUtils.h 33808 2009-07-15 09:48:08Z gconesab $ */

///////////////////////////////////////////////////////////////////////////////
//
// Class AliEMCALRecoUtils
// Some utilities to recalculate the cluster position or energy linearity
//
//
// Author:  Gustavo Conesa (LPSC- Grenoble) 
///////////////////////////////////////////////////////////////////////////////

//Root includes
#include "TNamed.h"

//AliRoot includes
class AliVCluster;
class AliVCaloCells;
#include "AliLog.h"
class AliEMCALGeoUtils;

class AliEMCALRecoUtils : public TNamed {
  
public:
  
  AliEMCALRecoUtils();
  AliEMCALRecoUtils(const AliEMCALRecoUtils&); 
  AliEMCALRecoUtils& operator=(const AliEMCALRecoUtils&); 
  virtual ~AliEMCALRecoUtils() {;}
  
  enum NonlinearityFunctions{kPi0MC=0,kPi0GammaGamma=1,kPi0GammaConversion=2,kNoCorrection=3};
  
  //Position recalculation
  void     RecalculateClusterPosition(AliEMCALGeoUtils *geom, AliVCaloCells* cells, AliVCluster* clu, const Int_t iParticle); 
  void     GetMaxEnergyCell(AliEMCALGeoUtils *geom, AliVCaloCells* cells, AliVCluster* clu, 
                            Int_t & absId,  Int_t& iSupMod, Int_t& ieta, Int_t& iphi);
  
  Float_t  GetMisalShift(const Int_t i) const {
    if(i < 15 ){return fMisalShift[i]; }
    else { AliInfo(Form("Index %d larger than 15, do nothing\n",i)); return 0.;}
  }
  Float_t  *GetMisalShiftArray() {return fMisalShift; }

  void     SetMisalShift(const Int_t i, const Float_t shift) {
    if(i < 15 ){fMisalShift[i] = shift; }
    else { AliInfo(Form("Index %d larger than 15, do nothing\n",i));}
  }
  void     SetMisalShiftArray(Float_t * misal) 
  { for(Int_t i = 0; i < 15; i++)fMisalShift[i] = misal[i]; }

  //Non Linearity
  
  Float_t CorrectClusterEnergyLinearity(AliVCluster* clu);
  
  Float_t  GetNonLinearityParam(const Int_t i) const {
    if(i < 6 ){return fNonLinearityParams[i]; }
    else { AliInfo(Form("Index %d larger than 6, do nothing\n",i)); return 0.;}
  }
  void     SetNonLinearityParam(const Int_t i, const Float_t param) {
    if(i < 6 ){fNonLinearityParams[i] = param; }
    else { AliInfo(Form("Index %d larger than 6, do nothing\n",i));}
  }
  
  Int_t GetNonLinearityFunction() const {return fNonLinearityFunction;}
  void  SetNonLinearityFunction(Int_t fun) {fNonLinearityFunction = fun ;}
  
  void Print(const Option_t*) const;
  
private:
  
  Float_t fMisalShift[15];        // Shift parameters
  Int_t   fNonLinearityFunction;  // Non linearity function choice
  Float_t fNonLinearityParams[6]; // Parameters for the non linearity function

  ClassDef(AliEMCALRecoUtils, 1)
  
};

#endif // ALIEMCALRECOUTILS_H
Commit	Line	Data
d9b3567c	1	#ifndef ALIEMCALRECOUTILS_H
	2	#define ALIEMCALRECOUTILS_H
	3
	4	/* $Id: AliEMCALRecoUtils.h 33808 2009-07-15 09:48:08Z gconesab $ */
	5
	6	///////////////////////////////////////////////////////////////////////////////
	7	//
	8	// Class AliEMCALRecoUtils
	9	// Some utilities to recalculate the cluster position or energy linearity
	10	//
	11	//
	12	// Author: Gustavo Conesa (LPSC- Grenoble)
	13	///////////////////////////////////////////////////////////////////////////////
	14
	15	//Root includes
	16	#include "TNamed.h"
	17
	18	//AliRoot includes
	19	class AliVCluster;
	20	class AliVCaloCells;
	21	#include "AliLog.h"
	22	class AliEMCALGeoUtils;
	23
	24	class AliEMCALRecoUtils : public TNamed {
	25
	26	public:
	27
	28	AliEMCALRecoUtils();
	29	AliEMCALRecoUtils(const AliEMCALRecoUtils&);
	30	AliEMCALRecoUtils& operator=(const AliEMCALRecoUtils&);
	31	virtual ~AliEMCALRecoUtils() {;}
	32
	33	enum NonlinearityFunctions{kPi0MC=0,kPi0GammaGamma=1,kPi0GammaConversion=2,kNoCorrection=3};
	34
	35	//Position recalculation
	36	void RecalculateClusterPosition(AliEMCALGeoUtils geom, AliVCaloCells cells, AliVCluster* clu, const Int_t iParticle);
	37	void GetMaxEnergyCell(AliEMCALGeoUtils geom, AliVCaloCells cells, AliVCluster* clu,
	38	Int_t & absId, Int_t& iSupMod, Int_t& ieta, Int_t& iphi);
	39
	40	Float_t GetMisalShift(const Int_t i) const {
	41	if(i < 15 ){return fMisalShift[i]; }
	42	else { AliInfo(Form("Index %d larger than 15, do nothing\n",i)); return 0.;}
	43	}
	44	Float_t *GetMisalShiftArray() {return fMisalShift; }
	45
	46	void SetMisalShift(const Int_t i, const Float_t shift) {
	47	if(i < 15 ){fMisalShift[i] = shift; }
	48	else { AliInfo(Form("Index %d larger than 15, do nothing\n",i));}
	49	}
	50	void SetMisalShiftArray(Float_t * misal)
	51	{ for(Int_t i = 0; i < 15; i++)fMisalShift[i] = misal[i]; }
	52
	53	//Non Linearity
	54
	55	Float_t CorrectClusterEnergyLinearity(AliVCluster* clu);
	56
	57	Float_t GetNonLinearityParam(const Int_t i) const {
	58	if(i < 6 ){return fNonLinearityParams[i]; }
	59	else { AliInfo(Form("Index %d larger than 6, do nothing\n",i)); return 0.;}
	60	}
	61	void SetNonLinearityParam(const Int_t i, const Float_t param) {
	62	if(i < 6 ){fNonLinearityParams[i] = param; }
	63	else { AliInfo(Form("Index %d larger than 6, do nothing\n",i));}
	64	}
65
66	Int_t GetNonLinearityFunction() const {return fNonLinearityFunction;}
67	void SetNonLinearityFunction(Int_t fun) {fNonLinearityFunction = fun ;}
68
69	void Print(const Option_t*) const;
70
71	private:
72
73	Float_t fMisalShift[15]; // Shift parameters
74	Int_t fNonLinearityFunction; // Non linearity function choice
75	Float_t fNonLinearityParams[6]; // Parameters for the non linearity function
76
77	ClassDef(AliEMCALRecoUtils, 1)
78
79	};
80
81	#endif // ALIEMCALRECOUTILS_H
82
83