/
ChainedJetCorrector.cc
83 lines (75 loc) · 2.35 KB
/
ChainedJetCorrector.cc
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
//
// Original Author: Fedor Ratnikov Feb. 16, 2007
//
// Correction which chains other corrections
//
#include "JetMETCorrections/Objects/interface/ChainedJetCorrector.h"
/// get correction using Jet information only
double ChainedJetCorrector::correction (const LorentzVector& fJet) const
{
LorentzVector jet = fJet;
double result = 1;
for (size_t i = 0; i < mCorrectors.size (); ++i) {
double scale = mCorrectors[i]->correction (jet);
jet *= scale;
result *= scale;
}
return result;
}
/// apply correction using Jet information only
double ChainedJetCorrector::correction (const reco::Jet& fJet) const
{
std::unique_ptr<reco::Jet> jet (dynamic_cast<reco::Jet*> (fJet.clone ()));
double result = 1;
for (size_t i = 0; i < mCorrectors.size (); ++i) {
double scale = mCorrectors[i]->correction (*jet);
jet->scaleEnergy (scale);
result *= scale;
}
return result;
}
/// apply correction using all event information
double ChainedJetCorrector::correction (const reco::Jet& fJet,
const edm::Event& fEvent,
const edm::EventSetup& fSetup) const
{
std::unique_ptr<reco::Jet> jet (dynamic_cast<reco::Jet*> (fJet.clone ()));
double result = 1;
for (size_t i = 0; i < mCorrectors.size (); ++i) {
double scale = mCorrectors[i]->correction (*jet, fEvent, fSetup);
jet->scaleEnergy (scale);
result *= scale;
}
return result;
}
/// apply correction using all event information and reference to the raw jet
double ChainedJetCorrector::correction (const reco::Jet& fJet,
const edm::RefToBase<reco::Jet>& fJetRef,
const edm::Event& fEvent,
const edm::EventSetup& fSetup) const
{
std::unique_ptr<reco::Jet> jet (dynamic_cast<reco::Jet*> (fJet.clone ()));
double result = 1;
for (size_t i = 0; i < mCorrectors.size (); ++i) {
double scale = mCorrectors[i]->correction (*jet, fJetRef, fEvent, fSetup);
jet->scaleEnergy (scale);
result *= scale;
}
return result;
}
/// if correction needs event information
bool ChainedJetCorrector::eventRequired () const
{
for (size_t i = 0; i < mCorrectors.size (); ++i) {
if (mCorrectors[i]->eventRequired ()) return true;
}
return false;
}
/// if correction needs jet reference
bool ChainedJetCorrector::refRequired () const
{
for (size_t i = 0; i < mCorrectors.size (); ++i) {
if (mCorrectors[i]->refRequired ()) return true;
}
return false;
}