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

#ifndef ALIEMCALPIDUTILS_H
#define ALIEMCALPIDUTILS_H

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

///////////////////////////////////////////////////////////////////////////////
// Class AliEMCALPIDUtils
// Compute PID weights for all the clusters
///////////////////////////////////////////////////////////////////////////////

//Root includes
#include "TNamed.h"
class TArrayD ;

//AliRoot includes
#include "AliPID.h" 

class AliEMCALPIDUtils : public TNamed {

public:
  
  AliEMCALPIDUtils();
/*   AliEMCALPIDUtils(Bool_t reconstructor); */
  //virtual ~AliEMCALPIDUtils() { }
  
  void     ComputePID(Double_t energy, Double_t lambda0); // give the PID of a cluster

  void     InitParameters();
  void     SetLowFluxParam();
  void     SetHighFluxParam();

  TArrayD  DistLambda0(const Double_t energy, const Int_t nature) ; // compute lambda0 distributions
  
  Double_t DistEnergy(const Double_t energy, const Int_t nature) ;

  Double_t GetPID(Int_t idx) const {if (idx>=0&&idx<3) return fPID[idx]; else return 0.;}
  Double_t GetPIDFinal(Int_t idx) const {if (idx>=0&&idx<AliPID::kSPECIESCN) return fPIDFinal[idx]; else return 0.;}
  Double_t GetPIDWeight(Int_t idx) const {if (idx>=0&&idx<3) return fPIDWeight[idx]; else return 0.;}
  
  void    SetPID(Double_t val, Int_t idx) {if (idx>=0&&idx<3) fPID[idx] = val;}
  void    SetPIDFinal(Double_t val, Int_t idx) {if (idx>=0&&idx<AliPID::kSPECIESCN) fPIDFinal[idx] = val;}
  void    SetPIDWeight(Double_t val, Int_t idx) {if (idx>=0&&idx<3) fPIDWeight[idx] = val;}
  void    SetPrintInfo(Bool_t yesno) {fPrintInfo = yesno;}
	
   	
	
 private:
  
  Double_t Polynomial(const Double_t x, const Double_t *params) const ;
  Double_t Polynomialinv(const Double_t x, const Double_t *params) const ;
  Double_t PolynomialMixed1(const Double_t x, const Double_t *params) const ;
  Double_t PolynomialMixed2(const Double_t x, const Double_t *params) const ;
  Double_t Polynomial0(const Double_t *params) const ;
  Double_t PowerExp(const Double_t x, const Double_t *params) const ;
	
protected: 	
  Bool_t   fPrintInfo;          // flag to decide if details about PID must be printed
  
  Double_t fGamma[6][6];            // Parameter to Compute PID for photons
  Double_t fGamma1to10[6][6];       // Parameter to Compute PID not used
  Double_t fHadron[6][6];	        // Parameter to Compute PID for hadrons, 1 to 10 GeV
  Double_t fHadron1to10[6][6];	    // Parameter to Compute PID for hadrons, 1 to 10 GeV
  Double_t fPiZero[6][6];           // Parameter to Compute PID for pi0
  Double_t fHadronEnergyProb[6]; 	// Parameter to Compute PID for energy ponderation for hadrons  	 
  Double_t fPiZeroEnergyProb[6]; 	// Parameter to Compute PID for energy ponderation for Pi0  	 
  Double_t fGammaEnergyProb[6]; 	// Parameter to Compute PID for energy ponderation for gamma  	 
   
  Float_t fPID[3];
  
  Float_t  fPIDFinal[AliPID::kSPECIESCN+1]; // final PID format
  Float_t  fPIDWeight[3];                  // order: gamma, pi0, hadrons,
  Double_t fProbGamma;	                  // probility to be a Gamma
  Double_t fProbPiZero;	                  // probility to be a PiO
  Double_t fProbHadron;	                  // probility to be a Hadron
  Double_t fWeightHadronEnergy;	          // Weight for a  a Hadron to have a given energy  (parametr from a flat distrib from 0 to 100)
  Double_t fWeightGammaEnergy;	          // Weight for a  Gamma to have a given energy  (for the moment =1.)
  Double_t fWeightPiZeroEnergy;	          // Weight for a Pi0 Hadron to have a given energy (for the moment =1.)
  
  ClassDef(AliEMCALPIDUtils, 2)

};

#endif // ALIEMCALPIDUTILS_H
Commit	Line	Data
0c5b726e	1	#ifndef ALIEMCALPIDUTILS_H
	2	#define ALIEMCALPIDUTILS_H
	3
	4	/* $Id: AliEMCALPIDUtils.h 33808 2009-07-15 09:48:08Z gconesab $ */
	5
	6	///////////////////////////////////////////////////////////////////////////////
	7	// Class AliEMCALPIDUtils
	8	// Compute PID weights for all the clusters
	9	///////////////////////////////////////////////////////////////////////////////
	10
	11	//Root includes
f21fc003	12	#include "TNamed.h"
0c5b726e	13	class TArrayD ;
	14
	15	//AliRoot includes
	16	#include "AliPID.h"
	17
f21fc003	18	class AliEMCALPIDUtils : public TNamed {
0c5b726e	19
	20	public:
	21
	22	AliEMCALPIDUtils();
	23	/* AliEMCALPIDUtils(Bool_t reconstructor); */
	24	//virtual ~AliEMCALPIDUtils() { }
	25
	26	void ComputePID(Double_t energy, Double_t lambda0); // give the PID of a cluster
	27
	28	void InitParameters();
	29	void SetLowFluxParam();
	30	void SetHighFluxParam();
	31
	32	TArrayD DistLambda0(const Double_t energy, const Int_t nature) ; // compute lambda0 distributions
	33
	34	Double_t DistEnergy(const Double_t energy, const Int_t nature) ;
	35
	36	Double_t GetPID(Int_t idx) const {if (idx>=0&&idx<3) return fPID[idx]; else return 0.;}
00a38d07	37	Double_t GetPIDFinal(Int_t idx) const {if (idx>=0&&idx<AliPID::kSPECIESCN) return fPIDFinal[idx]; else return 0.;}
0c5b726e	38	Double_t GetPIDWeight(Int_t idx) const {if (idx>=0&&idx<3) return fPIDWeight[idx]; else return 0.;}
	39
	40	void SetPID(Double_t val, Int_t idx) {if (idx>=0&&idx<3) fPID[idx] = val;}
00a38d07	41	void SetPIDFinal(Double_t val, Int_t idx) {if (idx>=0&&idx<AliPID::kSPECIESCN) fPIDFinal[idx] = val;}
0c5b726e	42	void SetPIDWeight(Double_t val, Int_t idx) {if (idx>=0&&idx<3) fPIDWeight[idx] = val;}
	43	void SetPrintInfo(Bool_t yesno) {fPrintInfo = yesno;}
	44
	45
	46
	47	private:
	48
	49	Double_t Polynomial(const Double_t x, const Double_t *params) const ;
	50	Double_t Polynomialinv(const Double_t x, const Double_t *params) const ;
	51	Double_t PolynomialMixed1(const Double_t x, const Double_t *params) const ;
	52	Double_t PolynomialMixed2(const Double_t x, const Double_t *params) const ;
	53	Double_t Polynomial0(const Double_t *params) const ;
	54	Double_t PowerExp(const Double_t x, const Double_t *params) const ;
	55
	56	protected:
	57	Bool_t fPrintInfo; // flag to decide if details about PID must be printed
	58
	59	Double_t fGamma[6][6]; // Parameter to Compute PID for photons
	60	Double_t fGamma1to10[6][6]; // Parameter to Compute PID not used
	61	Double_t fHadron[6][6]; // Parameter to Compute PID for hadrons, 1 to 10 GeV
	62	Double_t fHadron1to10[6][6]; // Parameter to Compute PID for hadrons, 1 to 10 GeV
	63	Double_t fPiZero[6][6]; // Parameter to Compute PID for pi0
	64	Double_t fHadronEnergyProb[6]; // Parameter to Compute PID for energy ponderation for hadrons
	65	Double_t fPiZeroEnergyProb[6]; // Parameter to Compute PID for energy ponderation for Pi0
	66	Double_t fGammaEnergyProb[6]; // Parameter to Compute PID for energy ponderation for gamma
	67
	68	Float_t fPID[3];
	69
00a38d07	70	Float_t fPIDFinal[AliPID::kSPECIESCN+1]; // final PID format
0c5b726e	71	Float_t fPIDWeight[3]; // order: gamma, pi0, hadrons,
	72	Double_t fProbGamma; // probility to be a Gamma
	73	Double_t fProbPiZero; // probility to be a PiO
	74	Double_t fProbHadron; // probility to be a Hadron
	75	Double_t fWeightHadronEnergy; // Weight for a a Hadron to have a given energy (parametr from a flat distrib from 0 to 100)
	76	Double_t fWeightGammaEnergy; // Weight for a Gamma to have a given energy (for the moment =1.)
	77	Double_t fWeightPiZeroEnergy; // Weight for a Pi0 Hadron to have a given energy (for the moment =1.)
	78
f21fc003	79	ClassDef(AliEMCALPIDUtils, 2)
0c5b726e	80
	81	};
	82
	83	#endif // ALIEMCALPIDUTILS_H
	84
	85