PHOS/AliPHOSPIDv1.h

   1 #ifndef ALIPHOSPIDV1_H
   2 #define ALIPHOSPIDV1_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8
   9 //_________________________________________________________________________
  10 // Implementation version v1 of the PHOS particle identifier
  11 // Identification is based on information from CPV and EMC
  12 // Oh yeah
  13 //*-- Author: Yves Schutz (SUBATECH), Gustavo Conesa.
  14
  15 // --- ROOT system ---
  16 //class TFormula ;
  17 class TVector3 ;
  18 class TMatrixD ;
  19 class TPrincipal ;
  20
  21 // --- Standard library ---
  22
  23 // --- AliRoot header files ---
  24 class AliPHOSEmcRecPoint ;
  25 class AliPHOSRecPoint ;
  26
  27 #include "AliPHOSPID.h"
  28
  29 class  AliPHOSPIDv1 : public AliPHOSPID {
  30
  31 public:
  32
  33   AliPHOSPIDv1() ;          // ctor
  34   AliPHOSPIDv1(const char* headerFile, const char * tsBranch = "Default", const char * from = 0) ;
  35
  36   virtual ~AliPHOSPIDv1() ; // dtor
  37
  38   virtual void Exec(Option_t * option) ;
  39   virtual char * GetRecParticlesBranch()const {return (char*) fRecParticlesTitle.Data() ;}
  40   virtual char * GetTrackSegmentsBranch()const{return (char*) fTrackSegmentsTitle.Data(); }
  41   virtual const Int_t GetRecParticlesInRun() const  {return fRecParticlesInRun ;}
  42
  43   virtual void Print(Option_t * option) const {}
  44   void Print() ;
  45   // Get CpvtoEmcDistanceCut and TimeGate parameters depending on the custer energy and
  46   // Purity-Efficiency point (possible options "HIGH EFFICIENCY" "MEDIUM EFFICIENCY" "LOW
  47   // EFFICIENCY" and 3 more options changing EFFICIENCY by PURITY)
  48   Double_t GetCpvtoEmcDistanceCut(const Float_t Cluster_En, const TString Eff_Pur)  ;
  49   Double_t GetTimeGate(const Float_t Cluster_En, const TString Eff_Pur)  ;
  50
  51   //Get files that contain the PCA
  52   const TString GetPrincipalFile5( )const {return fFileName5 ;}
  53   const TString GetPrincipalFilePar5( )const {return fFileNamePar5 ;}
  54   const TString GetPrincipalFile100( )const {return fFileName100 ;}
  55   const TString GetPrincipalFilePar100( )const {return fFileNamePar100 ;}
  56
  57   // Set all parameters necessary in the PID depending on the custer energy and
  58   // Purity-Efficiency point (possible options "HIGH EFFICIENCY" "MEDIUM EFFICIENCY" "LOW
  59   // EFFICIENCY" and 3 more options changing EFFICIENCY by PURITY)
  60   void SetCpvtoEmcDistanceCut(Float_t Cluster_En, TString Eff_Pur, Float_t cut)  ;
  61   void SetTimeGate(Float_t Cluster_En, TString Eff_Pur, Float_t gate)  ;
  62   void SetEllipseXCenter(Float_t Cluster_En, TString Eff_Pur, Float_t x)  ;
  63   void SetEllipseYCenter(Float_t Cluster_En, TString Eff_Pur, Float_t y)  ;
  64   void SetEllipseAParameter(Float_t Cluster_En, TString Eff_Pur, Float_t a)  ;
  65   void SetEllipseBParameter(Float_t Cluster_En, TString Eff_Pur, Float_t b)  ;
  66   void SetEllipseAngle(Float_t Cluster_En, TString Eff_Pur, Float_t angle)  ;
  67   void SetEllipseParameters(Float_t Cluster_En, TString Eff_Pur, Float_t x, Float_t y,Float_t a, Float_t b,Float_t angle) ;
  68
  69   //Get and Set energy calibration parameters
  70   Float_t GetACalParameter() {return fACalParameter ;}
  71   Float_t GetBCalParameter() {return fBCalParameter ;}
  72   Float_t GetCCalParameter() {return fCCalParameter ;}
  73   void SetACalParameter(Float_t a) { fACalParameter = a ;}
  74   void SetBCalParameter(Float_t b) { fBCalParameter = b ;}
  75   void SetCCalParameter(Float_t c) { fCCalParameter = c ;}
  76
  77
  78   Float_t GetEnergyAnalysisCut() {return  fEnergyAnalysisCut ;}
  79   void SetEnergyAnalysisCut(Float_t e) {  fEnergyAnalysisCut = e ;}
  80
  81   virtual void SetTrackSegmentsBranch(const char* title) { fTrackSegmentsTitle = title;}
  82   virtual void SetRecParticlesBranch (const char* title) { fRecParticlesTitle = title;}
  83   virtual const char * Version() const { return "pid-v1" ; }
  84
  85  private:
  86
  87   const TString BranchName() const ;
  88   virtual void Init() ;
  89   virtual void InitParameters() ;
  90   void     MakeRecParticles(void ) ;
  91   // Relative Distance CPV-EMC
  92   Float_t  GetDistance(AliPHOSEmcRecPoint * emc, AliPHOSRecPoint * cpv, Option_t * Axis)const ;
  93   Int_t    GetPrincipalSign(Double_t* P, Int_t ell, Int_t eff_pur)const ; //Principal cut
  94   TVector3 GetMomentumDirection(AliPHOSEmcRecPoint * emc, AliPHOSRecPoint * cpv)const ;
  95   void     PrintRecParticles(Option_t * option) ;
  96   // Gives in wich cluster energy range is the event
  97   void     GetClusterOption(const Float_t Cluster_En,const Bool_t range) ;
  98   // Gives the Efficiency-Purity point.
  99   Int_t    GetEffPurOption(const TString Eff_Pur)const ;
 100   virtual  void WriteRecParticles(Int_t event) ;
 101   void     SetParameters() ; //Fills the matrix of parameters
 102   //Selects principal and parameters file in function of energy range.
 103   void     GetAnalysisParameters(Float_t ClusterEn) ;
 104   Double_t CalibratedEnergy(Float_t e)  ; //Calibrates energy.
 105
 106
 107  private:
 108
 109   Bool_t                 fDefaultInit;        //! Says if the task was created by defaut ctor (only parameters are initialized)
 110   TString    fFileName5 ;     // File that contains the Principal file for analysis from 0.5 to 5 GeV
 111   TString    fFileName100 ;   // File that contains the Principal file for analysis from 0.5 to 100 GeV
 112   TString    fFileNamePar5 ;  // File that contains the parameters for analysis from 0.5 to 5 GeV
 113   TString    fFileNamePar100 ;// File that contains the parameters for analysis from 0.5 to 100 GeV
 114
 115
 116   TString    fFrom ;              // name of Recpoints and TrackSegments
 117   TString    fHeaderFileName ;    // file name with event header
 118   TString    fTrackSegmentsTitle; // branch name with track segments
 119   TString    fRecPointsTitle ;    // branch name with rec points
 120   TString    fRecParticlesTitle ; // branch name with rec particles
 121
 122   Int_t                      fNEvent ;            //! current event number
 123   AliPHOSClusterizer *       fClusterizer ;       //! clusterizer
 124   AliPHOSTrackSegmentMaker * fTSMaker ;           //! track segment maker
 125
 126   TPrincipal *               fPrincipal5 ;        //! TPrincipal from fFileName5
 127   TPrincipal *               fPrincipal100 ;      //! TPrincipal from fFileName100
 128   TPrincipal *               fPrincipal ;         //! TPrincipal copy
 129   Double_t *                 fX ;                 //! Principal data
 130   Double_t *                 fP ;                 //! Principal eigenvalues
 131
 132   Int_t                      fRecParticlesInRun ; //! Total number of recparticles in one run
 133
 134   TMatrixD *                 fParameters5 ;       //! Matrix of identification Parameters 0.5 to 5 GeV
 135   TMatrixD *                 fParameters100 ;     //! Matrix of identification Parameters 5-100 GeV
 136   TMatrixD *                 fParameters;         //! Matrix copy of identification Parameters
 137   Float_t                    fEnergyAnalysisCut;   // Energy to change from one PCA to the other.
 138   Int_t                      fCluster;            // Cluster energy range to choose parameters
 139   Int_t                      fClusterrcpv;        // Cluster energy range to choos rcpv parameters
 140   Int_t                      fMatrixExtraRow;     // Different size of the parameters file. Depends on range
 141
 142   Float_t   fACalParameter ;// A parameter energy calibration Encal=A+B*En+C*En^2
 143   Float_t   fBCalParameter ;// B parameter energy calibration Encal=A+B*En+C*En^2
 144   Float_t   fCCalParameter ;// B parameter energy calibration Encal=A+B*En+C*En^2
 145
 146   ClassDef( AliPHOSPIDv1,5)  // Particle identifier implementation version 1
 147
 148 };
 149
 150 #endif // AliPHOSPIDV1_H