[u/mrichter/AliRoot.git] / PHOS / AliPHOSPIDv1.h

#ifndef ALIPHOSPIDV1_H
#define ALIPHOSPIDV1_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$
 * Revision 1.59  2007/03/06 06:57:46  kharlov
 * DP:calculation of distance to CPV done in TSM
 *
 * Revision 1.58  2006/04/12 11:32:03  alibrary
 * Simplification of Makefile and some small corrections
 *
 * Revision 1.57  2006/01/23 17:51:48  hristov
 * Using the recommended way of forward declarations for TVector and TMatrix (see v5-08-00 release notes). Additional clean-up
 *
 * Revision 1.56  2005/05/28 14:19:04  schutz
 * Compilation warnings fixed by T.P.
 *
 */

//_________________________________________________________________________
// Implementation version v1 of the PHOS particle identifier 
// Identification is based on information from CPV and EMC
// Oh yeah                 
//*-- Author: Yves Schutz (SUBATECH), Gustavo Conesa.

// --- Standard library ---

// --- ROOT system ---
class TCanvas ;
class TFolder ;
class TFormula;
class TPrincipal ;
class TROOT ;
class TTree ;
class TVector3 ;
#include <TMatrixDfwd.h>

// --- AliRoot header files ---
class AliPHOSClusterizerv1 ;
class AliPHOSCpvRecPoint ;
class AliPHOSEmcRecPoint ;
class AliPHOSTrackSegment ;
class AliPHOSTrackSegmentMakerv1 ;
#include "AliPHOSPID.h"
#include "AliPID.h"

class  AliPHOSPIDv1 : public AliPHOSPID {
  
public:
  
  AliPHOSPIDv1() ;          // ctor   
  AliPHOSPIDv1(const TString alirunFileNameFile, const TString eventFolderName = AliConfig::GetDefaultEventFolderName()) ;
  AliPHOSPIDv1(const AliPHOSPIDv1 & pid) ;          // cpy ctor            
  
  virtual ~AliPHOSPIDv1() ; // dtor
  
  virtual void Exec(Option_t *option);  // Does the job

  //Get file name that contain the PCA
  const TString GetFileNamePrincipal(TString particle) const;

  //Get file name that contain PID parameters
  const TString GetFileNameParameters()      const {return fFileNameParameters ;}

  // Get number of rec.particles in this run
  virtual Int_t GetRecParticlesInRun() const {return fRecParticlesInRun ;}  


  // Get PID parameters as they are defined in fParameters
  Float_t GetParameterCalibration    (Int_t i)               const;
  Float_t GetParameterCpv2Emc        (Int_t i, TString axis) const;
  Float_t GetParameterTimeGate       (Int_t i)               const;
  Float_t GetParameterToCalculateEllipse(TString particle, TString param, Int_t i) const  ;     
  Float_t GetParameterPhotonBoundary (Int_t i)               const;
  Float_t GetParameterPi0Boundary    (Int_t i)               const;

  // Get energy-dependent PID parameters
  Float_t GetCalibratedEnergy    (Float_t e)                 const;
  Float_t GetCpv2EmcDistanceCut  (TString axis, Float_t e)   const ;
  Float_t GetEllipseParameter    (TString particle, TString param, Float_t e) const;

  Double_t GetThresholdChargedNeutral () const {return  fChargedNeutralThreshold;}
  Float_t GetTOFEnergyThreshold () const {return  fTOFEnThreshold;}
  Float_t GetDispersionEnergyThreshold () const {return  fDispEnThreshold;}
  Int_t   GetDispersionMultiplicityThreshold () const {return  fDispMultThreshold;}

  //Do bayesian PID
  void SetBayesianPID(Bool_t set){ fBayesian = set ;}

  // Set PID parameters to change appropriate element of fParameters
  void SetParameterCalibration   (Int_t i, Float_t param);
  void SetParameterCpv2Emc       (Int_t i, TString axis, Float_t cut)  ; 
  void SetParameterTimeGate      (Int_t i, Float_t gate)  ; 
  void SetParameterToCalculateEllipse(TString particle, TString param, Int_t i, Float_t value) ;
  void SetParameterPhotonBoundary(Int_t i, Float_t param);
  void SetParameterPi0Boundary   (Int_t i, Float_t param);

  void SetThresholdChargedNeutral (Double_t th) {fChargedNeutralThreshold = th;}
  void SetTOFEnergyThreshold (Float_t th)  {fTOFEnThreshold = th;}
  void SetDispersionEnergyThreshold (Float_t th) {fDispEnThreshold = th;}
  void SetDispersionMultiplicityThreshold (Int_t th)  {fDispMultThreshold = th;}

  //Switch to "on flyght" mode, without writing to TreeR and file  
  void SetWriting(Bool_t toWrite = kFALSE){fWrite = toWrite;} 
  void Print(const Option_t * = "") const ; 

  void GetVertex(void) ; //Extracts vertex in current event

  virtual const char * Version() const { return "pid-v1" ; }  

  AliPHOSPIDv1 & operator = (const AliPHOSPIDv1 & /*pid*/) { return *this ;} 
  
private:
  
  const TString BranchName() const ; 
  virtual void  Init() ;
  virtual void  InitParameters() ;
  void          MakeRecParticles(void ) ;
  void          MakePID(void) ;

  //Functions to calculate the PID probability 
  //  Double_t ChargedHadronDistProb(Double_t  x, Double_t y, Double_t * parg, Double_t * parl) ;
  Double_t GausF   (Double_t x, Double_t y, Double_t *par) ; //gaussian probability, parameter dependence a+b/(x*x)+c/x
  Double_t GausPol2(Double_t x, Double_t y, Double_t *par) ; //gaussian probability, parameter dependence a+b*x+c*x*x
  Double_t LandauF(Double_t x, Double_t y, Double_t *par) ; //gaussian probability, parameter dependence  a+b/(x*x)+c/x
  Double_t LandauPol2(Double_t x, Double_t y, Double_t *par) ; //gaussian probability, parameter dependence a+b*x+c*x*x
 // Relative Distance CPV-EMC
  Int_t   GetCPVBit       (AliPHOSTrackSegment * ts, Int_t EffPur, Float_t e) const;
  Int_t   GetPrincipalBit (TString particle, const Double_t* P, Int_t EffPur, Float_t e)const ; //Principal cut
  Int_t   GetHardPhotonBit(AliPHOSEmcRecPoint * emc) const;
  Int_t   GetHardPi0Bit   (AliPHOSEmcRecPoint * emc) const;
  TVector3      GetMomentumDirection(AliPHOSEmcRecPoint * emc, AliPHOSCpvRecPoint * cpv)const ;
  void          PrintRecParticles(Option_t * option) ;
  virtual void  WriteRecParticles() ; 
  void          SetParameters() ; //Fills the matrix of parameters
  void          Unload(); 

  //PID population
  void SetInitPID(const Double_t * pid) ;
  void GetInitPID(Double_t * pid) const ;

private:
  Bool_t      fBayesian ;                 //  Do PID bayesian
  Bool_t      fDefaultInit;              //! kTRUE if the task was created by defaut ctor (only parameters are initialized)
  Bool_t      fWrite ;                   //! To write result to file 
  Int_t       fNEvent ;                  //! current event number
  TString     fFileNamePrincipalPhoton ; //  File name of the photon principals
  TString     fFileNamePrincipalPi0 ;    //  File name of the pi0 principals
  TString     fFileNameParameters ;      //  File name with PID parameters
  TPrincipal *fPrincipalPhoton ;         //! TPrincipal from photon pca file 
  TPrincipal *fPrincipalPi0 ;            //! TPrincipal from pi0 pca file 
  Double_t   *fX ;                       //! Shower shape for the principal data 
  Double_t   *fPPhoton ;                 //! Principal photon eigenvalues
  Double_t   *fPPi0 ;                    //! Principal pi0 eigenvalues
  Int_t       fRecParticlesInRun ;       //! Total number of recparticles in one run
  TMatrixF    *fParameters;               //! Matrix of identification Parameters

  TVector3   fVtx ;                      //! Vertex position in current event

  //Initial pid population
  Double_t fInitPID[AliPID::kSPECIESN] ; // Initial population to do bayesian PID
  // pid probability function parameters
  // ToF
  Double_t fTphoton[3] ;       // gaussian tof response for photon
  TFormula * fTFphoton ;       // the formula   
  Double_t fTpiong[3] ;        // gaussian tof response for pions
  TFormula * fTFpiong ;        // the formula
  Double_t fTkaong[3] ;        // landau tof response for kaons
  TFormula * fTFkaong ;        // the formula
  Double_t fTkaonl[3] ;        // landau tof response for kaons
  TFormula * fTFkaonl ;        // the formula
  Double_t fThhadrong[3] ;     // gaus   tof response for heavy hadrons
  TFormula * fTFhhadrong ;     // the formula
  Double_t fThhadronl[3] ;     // landau   tof response for heavy hadrons
  TFormula * fTFhhadronl ;     // the formula

  //Shower dispersion
  Double_t fDmuon[3]    ;     // gaussian ss response for muon 
  TFormula * fDFmuon    ;     // the formula 
  Double_t fDphoton[10] ;     // gaussian ss response for EM
  Double_t fDpi0[10]    ;     // gaussian ss response for pi0
  Double_t fDhadron[10] ;     // gaussian ss response for hadrons

  Double_t fXelectron[10] ;   // gaussian emc-cpv distance response for electron
  Double_t fXcharged[10]  ;   // landau emc-cpv distance response for charged part (no elect) */
  Double_t fZelectron[10] ;   // gaussian emc-cpv distance response for electron
  Double_t fZcharged[10]  ;   // landau emc-cpv distance response for charged part (no elect) */


  Double_t fERecWeightPar[4] ;  // gaussian tof response for photon
  TFormula * fERecWeight ;      // the formula   
  Double_t fChargedNeutralThreshold ; //Threshold to differentiate between charged and neutral
  Float_t  fTOFEnThreshold;           //Maximum energy to use TOF
  Float_t  fDispEnThreshold;          //Minimum energy to use shower shape
  Int_t    fDispMultThreshold ;       //Minimum multiplicity to use shower shape

  ClassDef( AliPHOSPIDv1,12)  // Particle identifier implementation version 1

};

#endif // AliPHOSPIDV1_H
Commit	Line	Data
26d4b141	1	#ifndef ALIPHOSPIDV1_H
26d4b141	2	#define ALIPHOSPIDV1_H
6ad0bfa0	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id$ */
	7
702ab87e	8	/* History of cvs commits:
	9	*
	10	* $Log$
407d15b3	11	* Revision 1.59 2007/03/06 06:57:46 kharlov
	12	* DP:calculation of distance to CPV done in TSM
	13	*
26aa7e4a	14	* Revision 1.58 2006/04/12 11:32:03 alibrary
	15	* Simplification of Makefile and some small corrections
	16	*
8faf5ae4	17	* Revision 1.57 2006/01/23 17:51:48 hristov
	18	* Using the recommended way of forward declarations for TVector and TMatrix (see v5-08-00 release notes). Additional clean-up
	19	*
e8d02863	20	* Revision 1.56 2005/05/28 14:19:04 schutz
	21	* Compilation warnings fixed by T.P.
	22	*
702ab87e	23	*/
b2a60966	24
	25	//_________________________________________________________________________
	26	// Implementation version v1 of the PHOS particle identifier
148b2bba	27	// Identification is based on information from CPV and EMC
2f04ed65	28	// Oh yeah
148b2bba	29	//*-- Author: Yves Schutz (SUBATECH), Gustavo Conesa.
6ad0bfa0	30
8faf5ae4	31	// --- Standard library ---
e8d02863	32
8faf5ae4	33	// --- ROOT system ---
c947e71a	34	class TCanvas ;
c947e71a	35	class TFolder ;
acb5beb7	36	class TFormula;
8faf5ae4	37	class TPrincipal ;
	38	class TROOT ;
	39	class TTree ;
	40	class TVector3 ;
	41	#include <TMatrixDfwd.h>
	42
6ad0bfa0	43	// --- AliRoot header files ---
c947e71a	44	class AliPHOSClusterizerv1 ;
8faf5ae4	45	class AliPHOSCpvRecPoint ;
8faf5ae4	46	class AliPHOSEmcRecPoint ;
26aa7e4a	47	class AliPHOSTrackSegment ;
c947e71a	48	class AliPHOSTrackSegmentMakerv1 ;
26d4b141	49	#include "AliPHOSPID.h"
fb7b51ad	50	#include "AliPID.h"
8faf5ae4	51
26d4b141	52	class AliPHOSPIDv1 : public AliPHOSPID {
50739f15	53
e3817e5f	54	public:
50739f15	55
88cb7938	56	AliPHOSPIDv1() ; // ctor
e191bb57	57	AliPHOSPIDv1(const TString alirunFileNameFile, const TString eventFolderName = AliConfig::GetDefaultEventFolderName()) ;
88cb7938	58	AliPHOSPIDv1(const AliPHOSPIDv1 & pid) ; // cpy ctor
e3817e5f	59
7acf6008	60	virtual ~AliPHOSPIDv1() ; // dtor
50739f15	61
eabde521	62	virtual void Exec(Option_t *option); // Does the job
e3817e5f	63
	64	//Get file name that contain the PCA
	65	const TString GetFileNamePrincipal(TString particle) const;
	66
	67	//Get file name that contain PID parameters
	68	const TString GetFileNameParameters() const {return fFileNameParameters ;}
	69
	70	// Get number of rec.particles in this run
fc7e2f43	71	virtual Int_t GetRecParticlesInRun() const {return fRecParticlesInRun ;}
e3817e5f	72
35adb638	73
e3817e5f	74	// Get PID parameters as they are defined in fParameters
fc7e2f43	75	Float_t GetParameterCalibration (Int_t i) const;
	76	Float_t GetParameterCpv2Emc (Int_t i, TString axis) const;
	77	Float_t GetParameterTimeGate (Int_t i) const;
	78	Float_t GetParameterToCalculateEllipse(TString particle, TString param, Int_t i) const ;
	79	Float_t GetParameterPhotonBoundary (Int_t i) const;
	80	Float_t GetParameterPi0Boundary (Int_t i) const;
e3817e5f	81
e3817e5f	82	// Get energy-dependent PID parameters
fc7e2f43	83	Float_t GetCalibratedEnergy (Float_t e) const;
	84	Float_t GetCpv2EmcDistanceCut (TString axis, Float_t e) const ;
	85	Float_t GetEllipseParameter (TString particle, TString param, Float_t e) const;
e3817e5f	86
fb7b51ad	87	Double_t GetThresholdChargedNeutral () const {return fChargedNeutralThreshold;}
2924941c	88	Float_t GetTOFEnergyThreshold () const {return fTOFEnThreshold;}
	89	Float_t GetDispersionEnergyThreshold () const {return fDispEnThreshold;}
	90	Int_t GetDispersionMultiplicityThreshold () const {return fDispMultThreshold;}
fb7b51ad	91
c947e71a	92	//Do bayesian PID
	93	void SetBayesianPID(Bool_t set){ fBayesian = set ;}
	94
e3817e5f	95	// Set PID parameters to change appropriate element of fParameters
	96	void SetParameterCalibration (Int_t i, Float_t param);
	97	void SetParameterCpv2Emc (Int_t i, TString axis, Float_t cut) ;
	98	void SetParameterTimeGate (Int_t i, Float_t gate) ;
	99	void SetParameterToCalculateEllipse(TString particle, TString param, Int_t i, Float_t value) ;
	100	void SetParameterPhotonBoundary(Int_t i, Float_t param);
	101	void SetParameterPi0Boundary (Int_t i, Float_t param);
	102
fb7b51ad	103	void SetThresholdChargedNeutral (Double_t th) {fChargedNeutralThreshold = th;}
2924941c	104	void SetTOFEnergyThreshold (Float_t th) {fTOFEnThreshold = th;}
	105	void SetDispersionEnergyThreshold (Float_t th) {fDispEnThreshold = th;}
	106	void SetDispersionMultiplicityThreshold (Int_t th) {fDispMultThreshold = th;}
fb7b51ad	107
adcca1e6	108	//Switch to "on flyght" mode, without writing to TreeR and file
5d0435dd	109	void SetWriting(Bool_t toWrite = kFALSE){fWrite = toWrite;}
702ab87e	110	void Print(const Option_t * = "") const ;
e3817e5f	111
407d15b3	112	void GetVertex(void) ; //Extracts vertex in current event
407d15b3	113
8e8eae84	114	virtual const char * Version() const { return "pid-v1" ; }
e3817e5f	115
8c140292	116	AliPHOSPIDv1 & operator = (const AliPHOSPIDv1 & /pid/) { return *this ;}
e3817e5f	117
e3817e5f	118	private:
50739f15	119
c6987a88	120	const TString BranchName() const ;
e3817e5f	121	virtual void Init() ;
	122	virtual void InitParameters() ;
	123	void MakeRecParticles(void ) ;
35adb638	124	void MakePID(void) ;
	125
	126	//Functions to calculate the PID probability
	127	// Double_t ChargedHadronDistProb(Double_t x, Double_t y, Double_t * parg, Double_t * parl) ;
17323043	128	Double_t GausF (Double_t x, Double_t y, Double_t par) ; //gaussian probability, parameter dependence a+b/(xx)+c/x
	129	Double_t GausPol2(Double_t x, Double_t y, Double_t par) ; //gaussian probability, parameter dependence a+bx+cxx
	130	Double_t LandauF(Double_t x, Double_t y, Double_t par) ; //gaussian probability, parameter dependence a+b/(xx)+c/x
	131	Double_t LandauPol2(Double_t x, Double_t y, Double_t par) ; //gaussian probability, parameter dependence a+bx+cxx
2cc71c1e	132	// Relative Distance CPV-EMC
26aa7e4a	133	Int_t GetCPVBit (AliPHOSTrackSegment * ts, Int_t EffPur, Float_t e) const;
fc7e2f43	134	Int_t GetPrincipalBit (TString particle, const Double_t* P, Int_t EffPur, Float_t e)const ; //Principal cut
	135	Int_t GetHardPhotonBit(AliPHOSEmcRecPoint * emc) const;
	136	Int_t GetHardPi0Bit (AliPHOSEmcRecPoint * emc) const;
8d4608b5	137	TVector3 GetMomentumDirection(AliPHOSEmcRecPoint * emc, AliPHOSCpvRecPoint * cpv)const ;
e3817e5f	138	void PrintRecParticles(Option_t * option) ;
90cceaf6	139	virtual void WriteRecParticles() ;
e3817e5f	140	void SetParameters() ; //Fills the matrix of parameters
adcca1e6	141	void Unload();
e3817e5f	142
35adb638	143	//PID population
	144	void SetInitPID(const Double_t * pid) ;
	145	void GetInitPID(Double_t * pid) const ;
e3817e5f	146
35adb638	147	private:
35adb638	148	Bool_t fBayesian ; // Do PID bayesian
e3817e5f	149	Bool_t fDefaultInit; //! kTRUE if the task was created by defaut ctor (only parameters are initialized)
adcca1e6	150	Bool_t fWrite ; //! To write result to file
e3817e5f	151	Int_t fNEvent ; //! current event number
	152	TString fFileNamePrincipalPhoton ; // File name of the photon principals
	153	TString fFileNamePrincipalPi0 ; // File name of the pi0 principals
	154	TString fFileNameParameters ; // File name with PID parameters
	155	TPrincipal *fPrincipalPhoton ; //! TPrincipal from photon pca file
	156	TPrincipal *fPrincipalPi0 ; //! TPrincipal from pi0 pca file
	157	Double_t *fX ; //! Shower shape for the principal data
	158	Double_t *fPPhoton ; //! Principal photon eigenvalues
	159	Double_t *fPPi0 ; //! Principal pi0 eigenvalues
	160	Int_t fRecParticlesInRun ; //! Total number of recparticles in one run
e8d02863	161	TMatrixF *fParameters; //! Matrix of identification Parameters
cc1fe362	162
407d15b3	163	TVector3 fVtx ; //! Vertex position in current event
407d15b3	164
35adb638	165	//Initial pid population
304864ab	166	Double_t fInitPID[AliPID::kSPECIESN] ; // Initial population to do bayesian PID
35adb638	167	// pid probability function parameters
35adb638	168	// ToF
2924941c	169	Double_t fTphoton[3] ; // gaussian tof response for photon
	170	TFormula * fTFphoton ; // the formula
	171	Double_t fTpiong[3] ; // gaussian tof response for pions
	172	TFormula * fTFpiong ; // the formula
	173	Double_t fTkaong[3] ; // landau tof response for kaons
	174	TFormula * fTFkaong ; // the formula
	175	Double_t fTkaonl[3] ; // landau tof response for kaons
	176	TFormula * fTFkaonl ; // the formula
	177	Double_t fThhadrong[3] ; // gaus tof response for heavy hadrons
	178	TFormula * fTFhhadrong ; // the formula
	179	Double_t fThhadronl[3] ; // landau tof response for heavy hadrons
	180	TFormula * fTFhhadronl ; // the formula
35adb638	181
35adb638	182	//Shower dispersion
2924941c	183	Double_t fDmuon[3] ; // gaussian ss response for muon
	184	TFormula * fDFmuon ; // the formula
	185	Double_t fDphoton[10] ; // gaussian ss response for EM
	186	Double_t fDpi0[10] ; // gaussian ss response for pi0
	187	Double_t fDhadron[10] ; // gaussian ss response for hadrons
	188
	189	Double_t fXelectron[10] ; // gaussian emc-cpv distance response for electron
	190	Double_t fXcharged[10] ; // landau emc-cpv distance response for charged part (no elect) */
	191	Double_t fZelectron[10] ; // gaussian emc-cpv distance response for electron
	192	Double_t fZcharged[10] ; // landau emc-cpv distance response for charged part (no elect) */
	193
cc1fe362	194
fb7b51ad	195	Double_t fERecWeightPar[4] ; // gaussian tof response for photon
fb7b51ad	196	TFormula * fERecWeight ; // the formula
2924941c	197	Double_t fChargedNeutralThreshold ; //Threshold to differentiate between charged and neutral
	198	Float_t fTOFEnThreshold; //Maximum energy to use TOF
	199	Float_t fDispEnThreshold; //Minimum energy to use shower shape
	200	Int_t fDispMultThreshold ; //Minimum multiplicity to use shower shape
	201
	202	ClassDef( AliPHOSPIDv1,12) // Particle identifier implementation version 1
6ad0bfa0	203
	204	};
	205
26d4b141	206	#endif // AliPHOSPIDV1_H