Elaborating split mode to write data only to the split files
[u/mrichter/AliRoot.git] / PHOS / AliPHOSPIDv1.h
CommitLineData
26d4b141 1#ifndef ALIPHOSPIDV1_H
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
b2a60966 8
9//_________________________________________________________________________
10// Implementation version v1 of the PHOS particle identifier
148b2bba 11// Identification is based on information from CPV and EMC
2f04ed65 12// Oh yeah
148b2bba 13//*-- Author: Yves Schutz (SUBATECH), Gustavo Conesa.
6ad0bfa0 14
15// --- ROOT system ---
148b2bba 16//class TFormula ;
7acf6008 17class TVector3 ;
148b2bba 18class TMatrixD ;
19class TPrincipal ;
6ad0bfa0 20
21// --- Standard library ---
22
23// --- AliRoot header files ---
7acf6008 24class AliPHOSEmcRecPoint ;
25class AliPHOSRecPoint ;
6ad0bfa0 26
26d4b141 27#include "AliPHOSPID.h"
6ad0bfa0 28
26d4b141 29class AliPHOSPIDv1 : public AliPHOSPID {
6ad0bfa0 30
31public:
32
7acf6008 33 AliPHOSPIDv1() ; // ctor
fbf811ec 34 AliPHOSPIDv1(const char* headerFile, const char * tsBranch = "Default", const Bool_t toSplit=kFALSE) ;
e0ed2e49 35
7acf6008 36 virtual ~AliPHOSPIDv1() ; // dtor
37
a496c46c 38 virtual void Exec(Option_t * option) ;
fbf811ec 39 // virtual char * GetRecParticlesBranch()const {return (char*) fRecParticlesTitle.Data() ;}
40 // virtual char * GetTrackSegmentsBranch()const{return (char*) fTrackSegmentsTitle.Data(); }
2b60655b 41 virtual const Int_t GetRecParticlesInRun() const {return fRecParticlesInRun ;}
148b2bba 42
a496c46c 43 virtual void Print(Option_t * option) const {}
44 void Print() ;
148b2bba 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)
e61ea9b9 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 ;}
148b2bba 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) ;
f0a4c9e9 68
e61ea9b9 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
6f969528 77
78 Float_t GetEnergyAnalysisCut() {return fEnergyAnalysisCut ;}
79 void SetEnergyAnalysisCut(Float_t e) { fEnergyAnalysisCut = e ;}
80
fbf811ec 81 // virtual void SetTrackSegmentsBranch(const char* title) { fTrackSegmentsTitle = title;}
82 // virtual void SetRecParticlesBranch (const char* title) { fRecParticlesTitle = title;}
7b7c1533 83 virtual const char * Version() const { return "pid-v1" ; }
79bf6a2f 84
148b2bba 85 private:
86
c6987a88 87 const TString BranchName() const ;
79bf6a2f 88 virtual void Init() ;
8d0f3f77 89 virtual void InitParameters() ;
7acf6008 90 void MakeRecParticles(void ) ;
e61ea9b9 91 // Relative Distance CPV-EMC
92 Float_t GetDistance(AliPHOSEmcRecPoint * emc, AliPHOSRecPoint * cpv, Option_t * Axis)const ;
148b2bba 93 Int_t GetPrincipalSign(Double_t* P, Int_t ell, Int_t eff_pur)const ; //Principal cut
9688c1dd 94 TVector3 GetMomentumDirection(AliPHOSEmcRecPoint * emc, AliPHOSRecPoint * cpv)const ;
7acf6008 95 void PrintRecParticles(Option_t * option) ;
e61ea9b9 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
09fc14a0 106
107 private:
108
92f521a9 109 Bool_t fDefaultInit; //! Says if the task was created by defaut ctor (only parameters are initialized)
e61ea9b9 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
fbf811ec 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
148b2bba 121
a496c46c 122 Int_t fNEvent ; //! current event number
fbf811ec 123 // AliPHOSClusterizer * fClusterizer ; //! clusterizer
124 // AliPHOSTrackSegmentMaker * fTSMaker ; //! track segment maker
e61ea9b9 125
126 TPrincipal * fPrincipal5 ; //! TPrincipal from fFileName5
127 TPrincipal * fPrincipal100 ; //! TPrincipal from fFileName100
128 TPrincipal * fPrincipal ; //! TPrincipal copy
a496c46c 129 Double_t * fX ; //! Principal data
130 Double_t * fP ; //! Principal eigenvalues
e61ea9b9 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
6f969528 137 Float_t fEnergyAnalysisCut; // Energy to change from one PCA to the other.
e61ea9b9 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
7acf6008 145
6f969528 146 ClassDef( AliPHOSPIDv1,5) // Particle identifier implementation version 1
6ad0bfa0 147
148};
149
26d4b141 150#endif // AliPHOSPIDV1_H