|
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
|
#ifndef CondFormats_EcalShowerContainmentCorrections_h
#define CondFormats_EcalShowerContainmentCorrections_h
// -*- C++ -*-
//
// Package: CondFormats
// Class : EcalShowerContainmentCorrections
//
/**\class EcalShowerContainmentCorrections EcalShowerContainmentCorrections.h src/CondFormats/interface/EcalShowerContainmentCorrections.h
* Description: Holds the coefficients of a polynomial that describes
* the shower containment.
*
* Usage example(for real code see
* CalibCalorimetry/EcalCorrectionModules/test) :
* \code
ESHandle<EcalShowerContainmentCorrections> pGapCorr;
iESetup.get<EcalShowerContainmentCorrectionsRcd>().get(pGapCorr);
PositionCalc pos(...);
Hep3Vector clusterPos= pos.CalculatePos(...);
math::XYZPoint mathpoint(clusterPos.x(),clusterPos.y(),clusterPos.z());
double correction3x3 = pGapCorr->correction3x3(centerXtal,mathpoint);
double correction5x5 = pGapCorr->correction5x5(centerXtal,mathpoint);
* \endcode
* \author Stefano Argiro'
* \date Fri Mar 2 16:50:49 CET 2007
* \id $Id: EcalShowerContainmentCorrections.h,v 1.1 2007/05/15 20:37:22 argiro Exp $
*/
#include "CondFormats/Serialization/interface/Serializable.h"
#include <vector>
#include <algorithm>
#include <map>
#include <DataFormats/Math/interface/Point3D.h>
class EBDetId;
class EcalShowerContainmentCorrections {
public:
/// Structure defining the container for correction coefficients
/** data[0-3] : 3x3, x, right
* data[4-7] : 3x3, x, left
* data[8-11] : 3x3, y, right
* data[12-15] : 3x3, y, left
* data[16-19] : 5x5, x, right
* data[20-23] : 5x5, x, left
* data[24-27] : 5x5, y, right
* data[28-31] : 5x5, y, left
*/
struct Coefficients {
Coefficients() {
for (unsigned int i = 0; i < Coefficients::kSize; ++i)
data[i] = 0;
}
Coefficients(const Coefficients& coeff) { std::copy(coeff.data, coeff.data + Coefficients::kSize, data); }
Coefficients& operator=(const Coefficients& coeff) {
if (this == &coeff)
return *this;
std::copy(coeff.data, coeff.data + Coefficients::kSize, data);
return *this;
}
///The degree of the polynomial used as correction function plus one
static const int kPolynomialDegree = 4;
///Number of types of correction: Left, right, 3x3, 5x5, x, y
static const int kNTypes = 8;
static const unsigned int kSize = kPolynomialDegree * kNTypes;
double data[kSize];
COND_SERIALIZABLE;
};
/// Get the correction coefficients for the given xtal
/** Return zero coefficients in case the correction is not available
for that xtal */
const Coefficients correctionCoefficients(const EBDetId& centerxtal) const;
/// Fill the correction coefficients for a given xtal, part of group @group
/** Do not replace if xtal is already there*/
void fillCorrectionCoefficients(const EBDetId& xtal, int group, const Coefficients& coefficients);
/// Fill the correction coefficients for a given Ecal module
/** Assume that corresponding modules in different supermodules use
the same coefficients*/
void fillCorrectionCoefficients(const int supermodule, const int module, const Coefficients& coefficients);
/// The correction factor for 3x3 matrix
/** @param pos is the distance in cm from the center of the xtal
* as calculated in RecoEcal/EgammaCoreTools/interface/PositionCalc.h
* The valid return value is in the range (0,1] (divide by this
* value to apply the correction)
* Returns -1 if correction is not avaiable for that xtal*/
const double correction3x3(const EBDetId& xtal, const math::XYZPoint& pos) const;
/// The correction factor for 5x5 matrix
/** @param pos is the distance in cm from the center of the xtal
* as calculated in RecoEcal/EgammaCoreTools/interface/PositionCalc.h
* The return value is in the range (0,1] (divide by this
* value to apply the correction)
* Returns -1 if correction is not avaiable for that xtal*/
const double correction5x5(const EBDetId& xtal, const math::XYZPoint& pos) const;
private:
enum Direction { eX, eY };
enum Type { e3x3, e5x5 };
/** Calculate the correction for the given direction and type */
const double correctionXY(const EBDetId& xtal, double position, Direction dir, Type type) const;
typedef std::map<EBDetId, int> GroupMap;
/// Maps in which group a particular xtal has been placed
GroupMap groupmap_;
/// Holds the coeffiecients. The index corresponds to the group
std::vector<Coefficients> coefficients_;
COND_SERIALIZABLE;
};
#endif
|