[u/mrichter/AliRoot.git] / PWG4 / AliCaloPID.h

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

/* History of cvs commits:
 *
 * $Log$
 *
 *
 */

//_________________________________________________________________________
// Class for track/cluster acceptance selection
// Selection in Central barrel, EMCAL and PHOS
//
//*-- Author: Gustavo Conesa (INFN-LNF)

// --- ROOT system ---
#include <TObject.h> 
class TString ;
class Riostream ;
class TLorentzVector ;
class TFormula ;

//--- AliRoot system ---
class AliLog ;

class AliCaloPID : public TObject {

public: 

  AliCaloPID() ; // ctor
  AliCaloPID(const AliCaloPID & g) ; // cpy ctor
  AliCaloPID & operator = (const AliCaloPID & g) ;//cpy assignment
  virtual ~AliCaloPID() ;//virtual dtor

  enum PidType {
    kPhoton = 22,
    kPi0 = 111,
    kEta = 221, 
    kElectron = 11, 
    kEleCon = -11, 
    kNeutralHadron = 2112, 
    kChargedHadron = 211, 
    kNeutralUnknown = 130, 
    kChargedUnknown=321
  };

  enum PhotonStatusType {
    kPromptPhoton=2,
    kFragmentPhoton=3,
    kPi0DecayPhoton=4, 
    kEtaDecayPhoton=5, 
    kOtherDecayPhoton=6,
    kUnknown=7
  };

  void InitParameters();
  
  void Print(const Option_t * opt)const;
  
  Int_t GetPdg(const TString calo, const Double_t * pid, const Float_t energy) const ;
  Int_t GetPdg(const TString calo,  const TLorentzVector mom, const Double_t l0, const Double_t l1, const Double_t disp, const Double_t tof, const Double_t distCPV) const ;

  //Weight getters
  Float_t  GetEMCALPhotonWeight() const  { return  fEMCALPhotonWeight  ; }
  Float_t  GetEMCALPi0Weight() const     {  return fEMCALPi0Weight  ; }
  Float_t  GetEMCALElectronWeight() const  { return  fEMCALElectronWeight  ; }
  Float_t  GetEMCALChargeWeight() const     {  return fEMCALChargeWeight  ; }
  Float_t  GetEMCALNeutralWeight() const     {  return fEMCALNeutralWeight  ; }
  Float_t  GetPHOSPhotonWeight() const   {  return fPHOSPhotonWeight  ; }
  Float_t  GetPHOSPi0Weight() const   {  return fPHOSPi0Weight  ; }
  Float_t  GetPHOSElectronWeight() const   {  return fPHOSElectronWeight  ; }
  Float_t  GetPHOSChargeWeight() const   {  return fPHOSChargeWeight  ; }
  Float_t  GetPHOSNeutralWeight() const   {  return fPHOSNeutralWeight  ; }
  
  Bool_t  IsPHOSPIDWeightFormulaOn() const   {  return fPHOSWeightFormula  ; } 
  TFormula * GetPHOSPhotonWeightFormula() const     {  return fPHOSPhotonWeightFormula  ; } 
  TFormula * GetPHOSPi0WeightFormula() const    {  return fPHOSPi0WeightFormula  ; }
  
  //Weight setters
  void SetEMCALPhotonWeight(Float_t  w){  fEMCALPhotonWeight = w ; }
  void SetEMCALPi0Weight(Float_t  w){  fEMCALPi0Weight = w ; }
  void SetEMCALElectronWeight(Float_t  w){  fEMCALElectronWeight = w ; }
  void SetEMCALChargeWeight(Float_t  w){  fEMCALChargeWeight = w ; }
  void SetEMCALNeutralWeight(Float_t  w){  fEMCALNeutralWeight = w ; }
  void SetPHOSPhotonWeight(Float_t  w){  fPHOSPhotonWeight = w ; }
  void SetPHOSPi0Weight(Float_t  w){  fPHOSPi0Weight = w ; }
  void SetPHOSElectronWeight(Float_t  w){  fPHOSElectronWeight = w ; }
  void SetPHOSChargeWeight(Float_t  w){  fPHOSChargeWeight = w ; }
  void SetPHOSNeutralWeight(Float_t  w){  fPHOSNeutralWeight = w ; }
  
  void UsePHOSPIDWeightFormula(Bool_t par)  { fPHOSWeightFormula  = par; } 
  void SetPHOSPhotonWeightFormula(TFormula * photon)    {  fPHOSPhotonWeightFormula  = photon; } 
  void SetPHOSPi0WeightFormula(TFormula * pi0)   {  fPHOSPi0WeightFormula  = pi0; }
  
 private:
  
  Float_t      fEMCALPhotonWeight; //Bayesian PID weight for photons in EMCAL 
  Float_t      fEMCALPi0Weight;  //Bayesian PID weight for pi0 in EMCAL 
  Float_t      fEMCALElectronWeight; //Bayesian PID weight for electrons in EMCAL 
  Float_t      fEMCALChargeWeight;  //Bayesian PID weight for charged hadrons in EMCAL 
  Float_t      fEMCALNeutralWeight;  //Bayesian PID weight for neutral hadrons in EMCAL 
  Float_t      fPHOSPhotonWeight; //Bayesian PID weight for photons in PHOS 
  Float_t      fPHOSPi0Weight; //Bayesian PID weight for pi0 in PHOS 
  Float_t      fPHOSElectronWeight; //Bayesian PID weight for electrons in PHOS 
  Float_t      fPHOSChargeWeight; //Bayesian PID weight for charged hadrons in PHOS 
  Float_t      fPHOSNeutralWeight; //Bayesian PID weight for neutral hadrons in PHOS 

  Bool_t  fPHOSWeightFormula ; //Use parametrized weight threshold, function of energy
  TFormula * fPHOSPhotonWeightFormula ; //Formula for photon weight
  TFormula * fPHOSPi0WeightFormula ; //Formula for pi0 weight

  
  ClassDef(AliCaloPID,1)
} ;


#endif //ALICALOPID_H
Commit	Line	Data
d92b41ad	1	#ifndef ALICALOPID_H
	2	#define ALICALOPID_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5	/* $Id: $ */
	6
	7	/* History of cvs commits:
	8	*
	9	* $Log$
	10	*
	11	*
	12	*/
	13
	14	//_________________________________________________________________________
	15	// Class for track/cluster acceptance selection
	16	// Selection in Central barrel, EMCAL and PHOS
	17	//
	18	//*-- Author: Gustavo Conesa (INFN-LNF)
	19
	20	// --- ROOT system ---
	21	#include <TObject.h>
	22	class TString ;
	23	class Riostream ;
	24	class TLorentzVector ;
	25	class TFormula ;
	26
	27	//--- AliRoot system ---
	28	class AliLog ;
	29
	30	class AliCaloPID : public TObject {
	31
	32	public:
	33
	34	AliCaloPID() ; // ctor
	35	AliCaloPID(const AliCaloPID & g) ; // cpy ctor
	36	AliCaloPID & operator = (const AliCaloPID & g) ;//cpy assignment
	37	virtual ~AliCaloPID() ;//virtual dtor
	38
	39	enum PidType {
	40	kPhoton = 22,
	41	kPi0 = 111,
	42	kEta = 221,
	43	kElectron = 11,
	44	kEleCon = -11,
	45	kNeutralHadron = 2112,
	46	kChargedHadron = 211,
	47	kNeutralUnknown = 130,
	48	kChargedUnknown=321
	49	};
	50
	51	enum PhotonStatusType {
	52	kPromptPhoton=2,
	53	kFragmentPhoton=3,
	54	kPi0DecayPhoton=4,
	55	kEtaDecayPhoton=5,
	56	kOtherDecayPhoton=6,
	57	kUnknown=7
	58	};
	59
	60	void InitParameters();
	61
	62	void Print(const Option_t * opt)const;
	63
	64	Int_t GetPdg(const TString calo, const Double_t * pid, const Float_t energy) const ;
65	Int_t GetPdg(const TString calo, const TLorentzVector mom, const Double_t l0, const Double_t l1, const Double_t disp, const Double_t tof, const Double_t distCPV) const ;
66
67	//Weight getters
68	Float_t GetEMCALPhotonWeight() const { return fEMCALPhotonWeight ; }
69	Float_t GetEMCALPi0Weight() const { return fEMCALPi0Weight ; }
70	Float_t GetEMCALElectronWeight() const { return fEMCALElectronWeight ; }
71	Float_t GetEMCALChargeWeight() const { return fEMCALChargeWeight ; }
72	Float_t GetEMCALNeutralWeight() const { return fEMCALNeutralWeight ; }
73	Float_t GetPHOSPhotonWeight() const { return fPHOSPhotonWeight ; }
74	Float_t GetPHOSPi0Weight() const { return fPHOSPi0Weight ; }
75	Float_t GetPHOSElectronWeight() const { return fPHOSElectronWeight ; }
76	Float_t GetPHOSChargeWeight() const { return fPHOSChargeWeight ; }
77	Float_t GetPHOSNeutralWeight() const { return fPHOSNeutralWeight ; }
78
79	Bool_t IsPHOSPIDWeightFormulaOn() const { return fPHOSWeightFormula ; }
80	TFormula * GetPHOSPhotonWeightFormula() const { return fPHOSPhotonWeightFormula ; }
81	TFormula * GetPHOSPi0WeightFormula() const { return fPHOSPi0WeightFormula ; }
82
83	//Weight setters
84	void SetEMCALPhotonWeight(Float_t w){ fEMCALPhotonWeight = w ; }
85	void SetEMCALPi0Weight(Float_t w){ fEMCALPi0Weight = w ; }
86	void SetEMCALElectronWeight(Float_t w){ fEMCALElectronWeight = w ; }
87	void SetEMCALChargeWeight(Float_t w){ fEMCALChargeWeight = w ; }
88	void SetEMCALNeutralWeight(Float_t w){ fEMCALNeutralWeight = w ; }
89	void SetPHOSPhotonWeight(Float_t w){ fPHOSPhotonWeight = w ; }
90	void SetPHOSPi0Weight(Float_t w){ fPHOSPi0Weight = w ; }
91	void SetPHOSElectronWeight(Float_t w){ fPHOSElectronWeight = w ; }
92	void SetPHOSChargeWeight(Float_t w){ fPHOSChargeWeight = w ; }
93	void SetPHOSNeutralWeight(Float_t w){ fPHOSNeutralWeight = w ; }
94
95	void UsePHOSPIDWeightFormula(Bool_t par) { fPHOSWeightFormula = par; }
96	void SetPHOSPhotonWeightFormula(TFormula * photon) { fPHOSPhotonWeightFormula = photon; }
97	void SetPHOSPi0WeightFormula(TFormula * pi0) { fPHOSPi0WeightFormula = pi0; }
98
99	private:
100
101	Float_t fEMCALPhotonWeight; //Bayesian PID weight for photons in EMCAL
102	Float_t fEMCALPi0Weight; //Bayesian PID weight for pi0 in EMCAL
103	Float_t fEMCALElectronWeight; //Bayesian PID weight for electrons in EMCAL
104	Float_t fEMCALChargeWeight; //Bayesian PID weight for charged hadrons in EMCAL
105	Float_t fEMCALNeutralWeight; //Bayesian PID weight for neutral hadrons in EMCAL
106	Float_t fPHOSPhotonWeight; //Bayesian PID weight for photons in PHOS
107	Float_t fPHOSPi0Weight; //Bayesian PID weight for pi0 in PHOS
108	Float_t fPHOSElectronWeight; //Bayesian PID weight for electrons in PHOS
109	Float_t fPHOSChargeWeight; //Bayesian PID weight for charged hadrons in PHOS
110	Float_t fPHOSNeutralWeight; //Bayesian PID weight for neutral hadrons in PHOS
111
112	Bool_t fPHOSWeightFormula ; //Use parametrized weight threshold, function of energy
113	TFormula * fPHOSPhotonWeightFormula ; //Formula for photon weight
114	TFormula * fPHOSPi0WeightFormula ; //Formula for pi0 weight
115
116
117	ClassDef(AliCaloPID,1)
118	} ;
119
120
121	#endif //ALICALOPID_H
122
123
124