Particle identification improvment by Gustavo
[u/mrichter/AliRoot.git] / PHOS / AliPHOSPIDv1.h
index 2e26e29..abfa667 100644 (file)
@@ -27,92 +27,75 @@ class AliPHOSRecPoint ;
 #include "AliPHOSPID.h"
 
 class  AliPHOSPIDv1 : public AliPHOSPID {
-
-public:
-
+  
+ public:
+  
   AliPHOSPIDv1() ;          // ctor            
   AliPHOSPIDv1(const char* headerFile, const char * tsBranch = "Default", const Bool_t toSplit=kFALSE) ;
-   
+  
   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)  ;
-  Double_t GetTimeGate(const Float_t Cluster_En, const TString Eff_Pur)  ;
-
+  
   //Get files that contain the PCA
-  const TString GetPrincipalFile5( )const {return fFileName5 ;}
-  const TString GetPrincipalFilePar5( )const {return fFileNamePar5 ;}
-  const TString GetPrincipalFile100( )const {return fFileName100 ;}
-  const TString GetPrincipalFilePar100( )const {return fFileNamePar100 ;}
-
-  // 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)
+  const TString GetPrincipalFile( )const {return fFileName ;}
+  const TString GetPrincipalFilePar( )const {return fFileNamePar ;}
+  
+  // Set and Get 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) ;  
  
+  const Double_t GetCpvtoEmcDistanceCut(const Float_t Cluster_En, const TString Eff_Pur) const ;
+  const Double_t GetTimeGate(const Float_t Cluster_En, const TString Eff_Pur)  const;
+  void SetEllipseParameter(TString Param, Int_t i, Double_t par)  ;    
+  const Double_t GetParameterToCalculateEllipse(const TString Param, const Int_t i) const  ;     
+  const Double_t GetEllipseParameter(const TString Param, Float_t E) const;
   //Get and Set energy calibration parameters
-  Float_t GetACalParameter() {return fACalParameter ;}
-  Float_t GetBCalParameter() {return fBCalParameter ;}
-  Float_t GetCCalParameter() {return fCCalParameter ;}
-  void SetACalParameter(Float_t a) { fACalParameter = a ;}
-  void SetBCalParameter(Float_t b) { fBCalParameter = b ;}
-  void SetCCalParameter(Float_t c) { fCCalParameter = c ;}
-
-
-  Float_t GetEnergyAnalysisCut() {return  fEnergyAnalysisCut ;}
-  void SetEnergyAnalysisCut(Float_t e) {  fEnergyAnalysisCut = e ;}
-
+  
+  void  SetCalibrationParameter(Int_t Param,Double_t param);
+  const Double_t GetCalibrationParameter(const Int_t i) const;
+  const Double_t GetCalibratedEnergy(const Float_t e)   const; //Calibrates energy.
+  
   //  virtual void SetTrackSegmentsBranch(const char* title) { fTrackSegmentsTitle = title;}
   //  virtual void SetRecParticlesBranch (const char* title) { fRecParticlesTitle = title;} 
   virtual const char * Version() const { return "pid-v1" ; }  
   
  private:
-
+  
   const TString BranchName() const ; 
   virtual void Init() ;
   virtual void InitParameters() ;
   void     MakeRecParticles(void ) ;
   // Relative Distance CPV-EMC
-  Float_t  GetDistance(AliPHOSEmcRecPoint * emc, AliPHOSRecPoint * cpv, Option_t * Axis)const ; 
-  Int_t    GetPrincipalSign(Double_t* P, Int_t ell, Int_t eff_pur)const ; //Principal cut
+  const Float_t  GetDistance(AliPHOSEmcRecPoint * emc, AliPHOSRecPoint * cpv, Option_t * Axis)const ; 
+  const Int_t    GetPrincipalSign(const Double_t* P, const Int_t eff_pur, const Float_t E)const ; //Principal cut
   TVector3 GetMomentumDirection(AliPHOSEmcRecPoint * emc, AliPHOSRecPoint * cpv)const ;
   void     PrintRecParticles(Option_t * option) ;
   // Gives in wich cluster energy range is the event
-  void     GetClusterOption(const Float_t Cluster_En,const Bool_t range) ;
+  const Int_t     GetClusterOption(const Float_t Cluster_En) const;
   // Gives the Efficiency-Purity point.
-  Int_t    GetEffPurOption(const TString Eff_Pur)const ;
+  const Int_t    GetEffPurOption(const TString Eff_Pur)const ;
   virtual  void WriteRecParticles(Int_t event) ; 
   void     SetParameters() ; //Fills the matrix of parameters
-  //Selects principal and parameters file in function of energy range.
-  void     GetAnalysisParameters(Float_t ClusterEn) ;
-  Double_t CalibratedEnergy(Float_t e)  ; //Calibrates energy.
+  
  
 
  private:
 
   Bool_t                 fDefaultInit;        //! Says if the task was created by defaut ctor (only parameters are initialized)
-  TString    fFileName5 ;     // File that contains the Principal file for analysis from 0.5 to 5 GeV
-  TString    fFileName100 ;   // File that contains the Principal file for analysis from 0.5 to 100 GeV
-  TString    fFileNamePar5 ;  // File that contains the parameters for analysis from 0.5 to 5 GeV
-  TString    fFileNamePar100 ;// File that contains the parameters for analysis from 0.5 to 100 GeV
+  TString    fFileName ;  // File that contains the Principal file for analysis
+  TString    fFileNamePar ;// File that contains the parameters for analysis
   
   //  TString    fFrom ;              // name of Recpoints and TrackSegments 
   //  TString    fHeaderFileName ;    // file name with event header
   //  TString    fTrackSegmentsTitle; // branch name with track segments
@@ -123,27 +106,13 @@ public:
   //  AliPHOSClusterizer *       fClusterizer ;       //! clusterizer
   //  AliPHOSTrackSegmentMaker * fTSMaker ;           //! track segment maker
 
-  TPrincipal *               fPrincipal5 ;        //! TPrincipal from fFileName5  
-  TPrincipal *               fPrincipal100 ;      //! TPrincipal from fFileName100 
-  TPrincipal *               fPrincipal ;         //! TPrincipal copy 
+  TPrincipal *               fPrincipal ;         //! TPrincipal from pca file 
   Double_t *                 fX ;                 //! Principal data 
   Double_t *                 fP ;                 //! Principal eigenvalues
   Int_t                      fRecParticlesInRun ; //! Total number of recparticles in one run
-  TMatrixD *                 fParameters5 ;       //! Matrix of identification Parameters 0.5 to 5 GeV
-  TMatrixD *                 fParameters100 ;     //! Matrix of identification Parameters 5-100 GeV
-  TMatrixD *                 fParameters;         //! Matrix copy of identification Parameters
-  Float_t                    fEnergyAnalysisCut;   // Energy to change from one PCA to the other.
-  Int_t                      fCluster;            // Cluster energy range to choose parameters
-  Int_t                      fClusterrcpv;        // Cluster energy range to choos rcpv parameters
-  Int_t                      fMatrixExtraRow;     // Different size of the parameters file. Depends on range
-
-  Float_t   fACalParameter ;// A parameter energy calibration Encal=A+B*En+C*En^2
-  Float_t   fBCalParameter ;// B parameter energy calibration Encal=A+B*En+C*En^2
-  Float_t   fCCalParameter ;// B parameter energy calibration Encal=A+B*En+C*En^2
+  TMatrixD *                 fParameters;         //! Matrix of identification Parameters
 
-  ClassDef( AliPHOSPIDv1,5)  // Particle identifier implementation version 1
+  ClassDef( AliPHOSPIDv1,6)  // Particle identifier implementation version 1
 
 };