forked from h2gglobe/h2gglobe
/
PhotonFix.h
174 lines (138 loc) · 5.32 KB
/
PhotonFix.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
#ifndef PhotonFix_Defined_hh
#define PhotonFix_Defined_hh
//-------------------------------------------------------//
// Project: PhotonFix
// Author: Paul Dauncey (p.dauncey@imperial.ac.uk)
// Modified: 11/07/2011
// Admins: Paul Dauncey (p.dauncey@imperial.ac.uk)
// Matt Kenzie (matthew.william.kenzie@cern.ch)
//-------------------------------------------------------//
/*
Does post-reco fixes to ECAL photon energy and estimates resolution.
This can run outside of the usual CMS software framework but requires
access to a file 'EcalGaps.dat' which must be in the same directory as
that used to run.
To run within CMSSW use PhotonFixCMS.h (which can access the geometry
directly - go to "RecoEcal/EgammaCoreTools/plugins/PhotonFixCMS.h"
for details.
Before instantiating any objects of PhotonFix, the constants must be
initialised in the first event using
PhotonFix::initialise("3_8");
The string gives the reco version used. Valid strings are
"3_8", "3_11", "4_2" and "Nominal", where the latter gives no correction
to the energy and a nominal resolution value. There is also "4_2e" which
provides corrections for electrons which are reconstructed as photons (to
aid with testing the performance of these corrections in data).
Make objects using
PhotonFix a(energy,eta,phi,r9);
where energy is the photon energy, eta and phi are the ECAL
cluster positions (NB from the Supercluster object, _not_ the
Photon object, as the latter gives eta and phi directions,
not positions), and r9 is the R9 value of the SC.
Get the corrected energy using
a.fixedEnergy();
and the resolution using
a.sigmaEnergy();
*/
#include <iostream>
#include <string>
class PhotonFix {
public:
PhotonFix(double e, double eta, double phi, double r9);
// Must be called before instantiating any PhotonFix objects
static bool initialise(const std::string &,const std::string &);
static bool initialised() ;
// Used by above; do not call directly
static bool initialiseParameters(const std::string &s);
static bool initialiseGeometry(const std::string &s);
void setup();
// Corrected energy and sigma
double fixedEnergy() const;
double sigmaEnergy() const;
// Input values
double rawEnergy() const;
double eta() const;
double phi() const;
double r9() const;
// Derived EB crystal, submodule and module relative coordinates
double etaC() const;
double etaS() const;
double etaM() const;
double phiC() const;
double phiS() const;
double phiM() const;
// Derived EE zeta, crystal, subcrystal and D-module relative coordinates
double xZ() const;
double xC() const;
double xS() const;
double xM() const;
double yZ() const;
double yC() const;
double yS() const;
double yM() const;
// Return arrays containing positions of ecal gaps
static void barrelCGap(unsigned i, unsigned j, unsigned k, double c);
static void barrelSGap(unsigned i, unsigned j, unsigned k, double c);
static void barrelMGap(unsigned i, unsigned j, unsigned k, double c);
static void endcapCrystal(unsigned i, unsigned j, bool c);
static void endcapCGap(unsigned i, unsigned j, unsigned k, double c);
static void endcapSGap(unsigned i, unsigned j, unsigned k, double c);
static void endcapMGap(unsigned i, unsigned j, unsigned k, double c);
void print() const;
// Input and output the fit parameters
static void setParameters(unsigned be, unsigned hl, const double *p);
static void getParameters(unsigned be, unsigned hl, double *p);
static void dumpParameters(std::ostream &o);
static void printParameters(std::ostream &o);
// Utility functions
static double GetaPhi(double f0, double f1);
static double asinh(double s);
static void dumpGaps(std::ostream &o);
private:
// Utility functions
static double dPhi(double f0, double f1);
static double aPhi(double f0, double f1);
static double expCorrection(double a, const double *p);
static double gausCorrection(double a, const double *p);
// Actual data for each instantiated object
unsigned _be,_hl;
double _e,_eta,_phi,_r9;
double _aC,_aS,_aM,_bC,_bS,_bM;
// Constants
static const double _onePi;
static const double _twoPi;
// Initialisation flag
static bool _initialised;
// Parameters for fixes
static double _meanScale[2][2][4];
static double _meanAT[2][2][4];
static double _meanAC[2][2][4];
static double _meanAS[2][2][4];
static double _meanAM[2][2][4];
static double _meanBT[2][2][4];
static double _meanBC[2][2][4];
static double _meanBS[2][2][4];
static double _meanBM[2][2][4];
static double _meanR9[2][2][4];
// Parameters for resolution
static double _sigmaScale[2][2][4];
static double _sigmaAT[2][2][4];
static double _sigmaAC[2][2][4];
static double _sigmaAS[2][2][4];
static double _sigmaAM[2][2][4];
static double _sigmaBT[2][2][4];
static double _sigmaBC[2][2][4];
static double _sigmaBS[2][2][4];
static double _sigmaBM[2][2][4];
static double _sigmaR9[2][2][4];
// EB gap positions
static double _barrelCGap[169][360][2];
static double _barrelSGap[33][180][2];
static double _barrelMGap[7][18][2];
// EE crystal existence and gap positions
static bool _endcapCrystal[100][100];
static double _endcapCGap[2][7080][2];
static double _endcapSGap[2][264][2];
static double _endcapMGap[2][1][2];
};
#endif