]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PHOS/AliPHOSPIDv1.h
Important comment added.
[u/mrichter/AliRoot.git] / PHOS / AliPHOSPIDv1.h
index df2507c42029a59d71bac6de327683b39dc8aaa8..71fade128146cc065c41df635abc31f1c5e4cf05 100644 (file)
 //*-- Author: Yves Schutz (SUBATECH), Gustavo Conesa.
 
 // --- ROOT system ---
-//class TFormula ;
 class TVector3 ;
-class TMatrixD ;
+class TMatrix ;
 class TPrincipal ;
 
 // --- Standard library ---
 
 // --- AliRoot header files ---
 class AliPHOSEmcRecPoint ;
-class AliPHOSRecPoint ;
+class AliPHOSCpvRecPoint ;
 
 #include "AliPHOSPID.h"
 
 class  AliPHOSPIDv1 : public AliPHOSPID {
-
+  
 public:
-
-  AliPHOSPIDv1() ;          // ctor            
-  AliPHOSPIDv1(const char* headerFile, const char * tsBranch = "Default", const char * from = 0) ;
-   
-  virtual ~AliPHOSPIDv1() ; // dtor
-
-  virtual void Exec(Option_t * option) ;
-  virtual char * GetRecParticlesBranch()const {return (char*) fRecParticlesTitle.Data() ;}      
-  virtual char * GetTrackSegmentsBranch()const{return (char*) fTrackSegmentsTitle.Data(); }
-  virtual const Int_t GetRecParticlesInRun() const  {return fRecParticlesInRun ;}  
-  virtual void Print(Option_t * option) const {}
-  void Print() ; 
-  // Get CpvtoEmcDistanceCut and TimeGate parameters depending on the custer energy and 
-  // Purity-Efficiency point (possible options "HIGH EFFICIENCY" "MEDIUM EFFICIENCY" "LOW  
-  // EFFICIENCY" and 3 more options changing EFFICIENCY by PURITY)
-  Double_t GetCpvtoEmcDistanceCut(const Float_t Cluster_En, const TString Eff_Pur)const  ;
-  Double_t GetTimeGate(const Float_t Cluster_En, const TString Eff_Pur)const  ;
-  const TString GetPrincipalFile( )const {return fFileName ;}
-  const TString GetPrincipalFilePar( )const {return fFileNamePar ;}
-  void  SetPrincipalFileOptions(TString OptFileName) ;
-
-  // Set all parameters necessary in the PID depending on the custer energy and 
-  // Purity-Efficiency point (possible options "HIGH EFFICIENCY" "MEDIUM EFFICIENCY" "LOW  
-  // EFFICIENCY" and 3 more options changing EFFICIENCY by PURITY)
-  void SetCpvtoEmcDistanceCut(Float_t Cluster_En, TString Eff_Pur, Float_t cut)  ; 
-  void SetTimeGate(Float_t Cluster_En, TString Eff_Pur, Float_t gate)  ; 
-  void SetEllipseXCenter(Float_t Cluster_En, TString Eff_Pur, Float_t x)  ;    
-  void SetEllipseYCenter(Float_t Cluster_En, TString Eff_Pur, Float_t y)  ;   
-  void SetEllipseAParameter(Float_t Cluster_En, TString Eff_Pur, Float_t a)  ;  
-  void SetEllipseBParameter(Float_t Cluster_En, TString Eff_Pur, Float_t b)  ; 
-  void SetEllipseAngle(Float_t Cluster_En, TString Eff_Pur, Float_t angle)  ;    
-  void SetEllipseParameters(Float_t Cluster_En, TString Eff_Pur, Float_t x, Float_t y,Float_t a, Float_t b,Float_t angle) ;  
   
+  AliPHOSPIDv1() ;          // ctor   
+  AliPHOSPIDv1(const TString alirunFileNameFile, const TString eventFolderName = AliConfig::GetDefaultEventFolderName()) ;
+  AliPHOSPIDv1(const AliPHOSPIDv1 & pid) ;          // cpy ctor            
   
-  virtual void SetTrackSegmentsBranch(const char* title) { fTrackSegmentsTitle = title;}
-  virtual void SetRecParticlesBranch (const char* title) { fRecParticlesTitle = title;} 
+  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;
+
+  // 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 Print() const ; 
+
   virtual const char * Version() const { return "pid-v1" ; }  
+
+  AliPHOSPIDv1 & operator = (const AliPHOSPIDv1 & /*pid*/) { return *this ;} 
+  
+private:
   
- private:
-
-  const TString AliPHOSPIDv1::BranchName() const ; 
-  virtual void Init() ;
-  void     MakeRecParticles(void ) ;
-  Float_t  GetDistance(AliPHOSEmcRecPoint * emc, AliPHOSRecPoint * cpv, Option_t * Axis)const ; // Relative Distance CPV-EMC
-  Int_t    GetPrincipalSign(Double_t* P, Int_t ell, Int_t eff_pur)const ; //Principal cut
-  TVector3 GetMomentumDirection(AliPHOSEmcRecPoint * emc, AliPHOSRecPoint * cpv)const ;
-  void     PrintRecParticles(Option_t * option) ;
-  void     SetParameters() ;
-  virtual void WriteRecParticles(Int_t event) ; 
-
- private:
-
-  TString                fFileName ;          // Name of the file which contains the Principal file
-  TString                fFileNamePar ;       // Name of the file which contains the parameters
-  TString                fOptFileName ;       // choose de parameters and principal file
-  TString                fFrom ;              // name of Recpoints and TrackSegments 
-  TString                fHeaderFileName ;    // file name with event header
-  TString                fTrackSegmentsTitle; // branch name with track segments
-  TString                fRecPointsTitle ;    // branch name with rec points
-  TString                fRecParticlesTitle ; // branch name with rec particles
-  Int_t                      fNEvent ;            //! current event number
-  AliPHOSClusterizer *       fClusterizer ;       //! clusterizer
-  AliPHOSTrackSegmentMaker * fTSMaker ;           //! track segment maker
-  TPrincipal *               fPrincipal ;         //! TPrincipal from fFileName 
-  Int_t                      fRecParticlesInRun ; //! Total number of recparticles in one run
-  Double_t *                 fX ;                 //! Principal data 
-  Double_t *                 fP ;                 //! Principal eigenvalues
-  TMatrixD *                 fParameters ;        //! Matrix of all identification Parameters
-
-  ClassDef( AliPHOSPIDv1,2)  // Particle identifier implementation version 1
+  const TString BranchName() const ; 
+  virtual void  Init() ;
+  virtual void  InitParameters() ;
+  void          MakeRecParticles(void ) ;
+  void          MakePID(void ) ;
+ // Relative Distance CPV-EMC
+  Float_t GetDistance     (AliPHOSEmcRecPoint * emc, AliPHOSCpvRecPoint * cpv, Option_t * axis)const ; 
+  Int_t   GetCPVBit       (AliPHOSEmcRecPoint * emc, AliPHOSCpvRecPoint * cpv, 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(); 
+
+private:
+
+  Bool_t      fDefaultInit;              //! kTRUE if the task was created by defaut ctor (only parameters are initialized)
+  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
+  TMatrix    *fParameters;               //! Matrix of identification Parameters
+  // response function parameters
+  // ToF
+  Double_t fTphoton[3] ;                 // gaussian response for photon
+  TFormula * fTFphoton ;                 // the formula   
+  Double_t fTelectron[3] ;               // gaussian response for electrons
+  TFormula * fTFelectron ;               // the formula   
+  Double_t fTchargedhadron[3] ;          // landau   response for charged hadrons
+  TFormula * fTFchargedhadron ;          // the formula   
+  Double_t fTneutralhadron[3] ;          // landau   response for neutral hadrons
+  TFormula * fTFneutralhadron ;          // the formula   
+
+
+  ClassDef( AliPHOSPIDv1,10)  // Particle identifier implementation version 1
 
 };