|
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
|
#ifndef DataFormats_L1Trigger_HGCalMulticluster_h
#define DataFormats_L1Trigger_HGCalMulticluster_h
#include "DataFormats/Common/interface/Ptr.h"
#include "DataFormats/L1Trigger/interface/BXVector.h"
#include "DataFormats/L1THGCal/interface/HGCalClusterT.h"
#include "DataFormats/L1THGCal/interface/HGCalCluster.h"
#include <boost/iterator/transform_iterator.hpp>
#include <functional>
namespace l1t {
class HGCalMulticluster : public HGCalClusterT<l1t::HGCalCluster> {
public:
HGCalMulticluster() : hOverEValid_(false) {}
HGCalMulticluster(const LorentzVector p4, int pt = 0, int eta = 0, int phi = 0);
HGCalMulticluster(const edm::Ptr<l1t::HGCalCluster>& tc, float fraction = 1);
~HGCalMulticluster() override;
float hOverE() const {
// --- this below would be faster when reading old objects, as HoE will only be computed once,
// --- but it may not be allowed by CMS rules because of the const_cast
// --- and could potentially cause a data race
// if (!hOverEValid_) (const_cast<HGCalMulticluster*>(this))->saveHOverE();
// --- this below is safe in any case
return hOverEValid_ ? hOverE_ : l1t::HGCalClusterT<l1t::HGCalCluster>::hOverE();
}
void saveHOverE() {
hOverE_ = l1t::HGCalClusterT<l1t::HGCalCluster>::hOverE();
hOverEValid_ = true;
}
enum EnergyInterpretation { EM = 0 };
void saveEnergyInterpretation(const HGCalMulticluster::EnergyInterpretation eInt, double energy);
double iEnergy(const HGCalMulticluster::EnergyInterpretation eInt) const {
return energy() * interpretationFraction(eInt);
}
double iPt(const HGCalMulticluster::EnergyInterpretation eInt) const { return pt() * interpretationFraction(eInt); }
math::XYZTLorentzVector iP4(const HGCalMulticluster::EnergyInterpretation eInt) const {
return p4() * interpretationFraction(eInt);
}
math::PtEtaPhiMLorentzVector iPolarP4(const HGCalMulticluster::EnergyInterpretation eInt) const {
return math::PtEtaPhiMLorentzVector(pt() * interpretationFraction(eInt), eta(), phi(), 0.);
}
private:
template <typename Iter>
struct KeyGetter {
const typename Iter::value_type::first_type& operator()(const typename Iter::value_type& p) const {
return p.first;
}
};
template <typename Iter>
boost::transform_iterator<KeyGetter<Iter>, Iter> key_iterator(Iter itr) const {
return boost::make_transform_iterator<KeyGetter<Iter>, Iter>(itr, KeyGetter<Iter>());
}
public:
typedef boost::transform_iterator<KeyGetter<std::map<EnergyInterpretation, double>::const_iterator>,
std::map<EnergyInterpretation, double>::const_iterator>
EnergyInterpretation_const_iterator;
std::pair<EnergyInterpretation_const_iterator, EnergyInterpretation_const_iterator> energyInterpretations() const {
return std::make_pair(key_iterator(energyInterpretationFractions_.cbegin()),
key_iterator(energyInterpretationFractions_.cend()));
}
EnergyInterpretation_const_iterator interpretations_begin() const {
return key_iterator(energyInterpretationFractions_.cbegin());
}
EnergyInterpretation_const_iterator interpretations_end() const {
return key_iterator(energyInterpretationFractions_.cend());
}
size_type interpretations_size() const { return energyInterpretationFractions_.size(); }
private:
double interpretationFraction(const HGCalMulticluster::EnergyInterpretation eInt) const;
float hOverE_;
bool hOverEValid_;
std::map<EnergyInterpretation, double> energyInterpretationFractions_;
};
typedef BXVector<HGCalMulticluster> HGCalMulticlusterBxCollection;
} // namespace l1t
#endif
|