CMS 3D CMS Logo
TopDQMHelpers.cc
Go to the documentation of this file.
1 #include "DQM/Physics/interface/TopDQMHelpers.h"
2 
3 Calculate::Calculate(int maxNJets, double wMass)
4  : failed_(false),
5  maxNJets_(maxNJets),
6  wMass_(wMass),
7  massWBoson_(-1.),
8  massTopQuark_(-1.),
9  massBTopQuark_(-1.),
10  tmassWBoson_(-1),
11  tmassTopQuark_(-1) {}
12 
13 double Calculate::massWBoson(const std::vector<reco::Jet>& jets) {
14  if (!failed_ && massWBoson_ < 0) operator()(jets);
15  return massWBoson_;
16 }
17 
18 double Calculate::massTopQuark(const std::vector<reco::Jet>& jets) {
19  if (!failed_ && massTopQuark_ < 0) operator()(jets);
20  return massTopQuark_;
21 }
22 
23 double Calculate::massBTopQuark(const std::vector<reco::Jet>& jets,
24  std::vector<double> VbtagWP, double btagWP_) {
25  if (!failed_ && massBTopQuark_ < 0) operator2(jets, VbtagWP, btagWP_);
26  return massBTopQuark_;
27 }
28 
29 double Calculate::tmassWBoson(reco::RecoCandidate* mu, const reco::MET& met,
30  const reco::Jet& b) {
31  if (tmassWBoson_ < 0) operator()(b, mu, met);
32  return tmassWBoson_;
33 }
34 
35 double Calculate::tmassTopQuark(reco::RecoCandidate* lepton,
36  const reco::MET& met, const reco::Jet& b) {
37  if (tmassTopQuark_ < 0) operator()(b, lepton, met);
38  return tmassTopQuark_;
39 }
40 
41 void Calculate::operator()(const reco::Jet& bJet, reco::RecoCandidate* lepton,
42  const reco::MET& met) {
43  double metT = sqrt(pow(met.px(), 2) + pow(met.py(), 2));
44  double lepT = sqrt(pow(lepton->px(), 2) + pow(lepton->py(), 2));
45  double bT = sqrt(pow(bJet.px(), 2) + pow(bJet.py(), 2));
46  reco::Particle::LorentzVector WT = lepton->p4() + met.p4();
47  tmassWBoson_ =
48  sqrt(pow(metT + lepT, 2) - (WT.px() * WT.px()) - (WT.py() * WT.py()));
49  reco::Particle::LorentzVector topT = WT + bJet.p4();
50  tmassTopQuark_ = sqrt(pow((metT + lepT + bT), 2) - (topT.px() * topT.px()) -
51  (topT.py() * topT.py()));
52 }
53 
54 void Calculate::operator()(const std::vector<reco::Jet>& jets) {
55  if (maxNJets_ < 0) maxNJets_ = jets.size();
56  failed_ = jets.size() < (unsigned int)maxNJets_;
57  if (failed_) {
58  return;
59  }
60 
61  // associate those jets with maximum pt of the vectorial
62  // sum to the hadronic decay chain
63  double maxPt = -1.;
64  std::vector<int> maxPtIndices;
65  maxPtIndices.push_back(-1);
66  maxPtIndices.push_back(-1);
67  maxPtIndices.push_back(-1);
68 
69  for (int idx = 0; idx < maxNJets_; ++idx) {
70  for (int jdx = idx+1; jdx < maxNJets_; ++jdx) {
71  //if (jdx <= idx) continue;
72  for (int kdx = 0; kdx < maxNJets_; ++kdx) {
73  if (kdx == idx || kdx == jdx) continue;
74  reco::Particle::LorentzVector sum =
75  jets[idx].p4() + jets[jdx].p4() + jets[kdx].p4();
76  if (maxPt < 0. || maxPt < sum.pt()) {
77  maxPt = sum.pt();
78  maxPtIndices.clear();
79  maxPtIndices.push_back(idx);
80  maxPtIndices.push_back(jdx);
81  maxPtIndices.push_back(kdx);
82  }
83  }
84  }
85  }
86  massTopQuark_ = (jets[maxPtIndices[0]].p4() + jets[maxPtIndices[1]].p4() +
87  jets[maxPtIndices[2]].p4()).mass();
88 
89  // associate those jets that get closest to the W mass
90  // with their invariant mass to the W boson
91  double wDist = -1.;
92  std::vector<int> wMassIndices;
93  wMassIndices.push_back(-1);
94  wMassIndices.push_back(-1);
95  for (unsigned idx = 0; idx < maxPtIndices.size(); ++idx) {
96  for (unsigned jdx = 0; jdx < maxPtIndices.size(); ++jdx) {
97  if (jdx == idx || maxPtIndices[idx] > maxPtIndices[jdx]) continue;
98  reco::Particle::LorentzVector sum =
99  jets[maxPtIndices[idx]].p4() + jets[maxPtIndices[jdx]].p4();
100  if (wDist < 0. || wDist > fabs(sum.mass() - wMass_)) {
101  wDist = fabs(sum.mass() - wMass_);
102  wMassIndices.clear();
103  wMassIndices.push_back(maxPtIndices[idx]);
104  wMassIndices.push_back(maxPtIndices[jdx]);
105  }
106  }
107  }
108  massWBoson_ =
109  (jets[wMassIndices[0]].p4() + jets[wMassIndices[1]].p4()).mass();
110 }
111 
112 void Calculate::operator2(const std::vector<reco::Jet>& jets,
113  std::vector<double> bjet, double btagWP) {
114  if (maxNJets_ < 0) maxNJets_ = jets.size();
115  failed_ = jets.size() < (unsigned int)maxNJets_;
116  if (failed_) {
117  return;
118  }
119  if (jets.size() != bjet.size()) {
120  return;
121  }
122 
123  // associate those jets with maximum pt of the vectorial
124  // sum to the hadronic decay chain. Require ONLY 1 btagged jet
125  double maxBPt = -1.;
126  std::vector<int> maxBPtIndices;
127  maxBPtIndices.push_back(-1);
128  maxBPtIndices.push_back(-1);
129  maxBPtIndices.push_back(-1);
130  for (int idx = 0; idx < maxNJets_; ++idx) {
131  for (int jdx = idx+1; jdx < maxNJets_; ++jdx) {
132  //if (jdx <= idx) continue;
133  for (int kdx = 0; kdx < maxNJets_; ++kdx) {
134  if (kdx == idx || kdx == jdx) continue;
135  // require only 1b-jet
136  if ((bjet[idx] > btagWP && bjet[jdx] <= btagWP &&
137  bjet[kdx] <= btagWP) ||
138  (bjet[idx] <= btagWP && bjet[jdx] > btagWP &&
139  bjet[kdx] <= btagWP) ||
140  (bjet[idx] <= btagWP && bjet[jdx] <= btagWP &&
141  bjet[kdx] > btagWP)) {
142  reco::Particle::LorentzVector sum =
143  jets[idx].p4() + jets[jdx].p4() + jets[kdx].p4();
144  if (maxBPt < 0. || maxBPt < sum.pt()) {
145  maxBPt = sum.pt();
146  maxBPtIndices.clear();
147  maxBPtIndices.push_back(idx);
148  maxBPtIndices.push_back(jdx);
149  maxBPtIndices.push_back(kdx);
150  }
151  }
152  }
153  }
154  }
155  if (maxBPtIndices[0] < 0 || maxBPtIndices[1] < 0 || maxBPtIndices[2] < 0)
156  return;
157  massBTopQuark_ = (jets[maxBPtIndices[0]].p4() + jets[maxBPtIndices[1]].p4() +
158  jets[maxBPtIndices[2]].p4()).mass();
159 }
160 
161 
162 
163 
164 
165 
166 
167 
168 Calculate_miniAOD::Calculate_miniAOD(int maxNJets, double wMass)
169  : failed_(false),
170  maxNJets_(maxNJets),
171  wMass_(wMass),
172  massWBoson_(-1.),
173  massTopQuark_(-1.),
174  massBTopQuark_(-1.),
175  tmassWBoson_(-1),
176  tmassTopQuark_(-1) {}
177 
178 double Calculate_miniAOD::massWBoson(const std::vector<pat::Jet>& jets) {
179  if (!failed_ && massWBoson_ < 0) operator()(jets);
180  return massWBoson_;
181 }
182 
183 double Calculate_miniAOD::massTopQuark(const std::vector<pat::Jet>& jets) {
184  if (!failed_ && massTopQuark_ < 0) operator()(jets);
185  return massTopQuark_;
186 }
187 
188 double Calculate_miniAOD::massBTopQuark(const std::vector<pat::Jet>& jets,
189  std::vector<double> VbtagWP, double btagWP_) {
190 
191  if (!failed_ && massBTopQuark_ < 0) operator2(jets, VbtagWP, btagWP_);
192  return massBTopQuark_;
193 }
194 
195 double Calculate_miniAOD::tmassWBoson(pat::Muon* mu, const pat::MET& met,
196  const pat::Jet& b) {
197  if (tmassWBoson_ < 0) operator()(b, mu, met);
198  return tmassWBoson_;
199 }
200 
201 double Calculate_miniAOD::tmassWBoson(pat::Electron* mu, const pat::MET& met,
202  const pat::Jet& b) {
203  if (tmassWBoson_ < 0) operator()(b, mu, met);
204  return tmassWBoson_;
205 }
206 
207 
208 double Calculate_miniAOD::tmassTopQuark(pat::Electron* lepton,
209  const pat::MET& met, const pat::Jet& b) {
210  if (tmassTopQuark_ < 0) operator()(b, lepton, met);
211  return tmassTopQuark_;
212 }
213 
214 
215 double Calculate_miniAOD::tmassTopQuark(pat::Muon* lepton,
216  const pat::MET& met, const pat::Jet& b) {
217  if (tmassTopQuark_ < 0) operator()(b, lepton, met);
218  return tmassTopQuark_;
219 }
220 
221 
222 void Calculate_miniAOD::operator()(const pat::Jet& bJet, pat::Muon* lepton,
223  const pat::MET& met) {
224  double metT = sqrt(pow(met.px(), 2) + pow(met.py(), 2));
225  double lepT = sqrt(pow(lepton->px(), 2) + pow(lepton->py(), 2));
226  double bT = sqrt(pow(bJet.px(), 2) + pow(bJet.py(), 2));
227  reco::Particle::LorentzVector WT = lepton->p4() + met.p4();
228  tmassWBoson_ =
229  sqrt(pow(metT + lepT, 2) - (WT.px() * WT.px()) - (WT.py() * WT.py()));
230  reco::Particle::LorentzVector topT = WT + bJet.p4();
231  tmassTopQuark_ = sqrt(pow((metT + lepT + bT), 2) - (topT.px() * topT.px()) -
232  (topT.py() * topT.py()));
233 }
234 
235 
236 void Calculate_miniAOD::operator()(const pat::Jet& bJet, pat::Electron* lepton,
237  const pat::MET& met) {
238  double metT = sqrt(pow(met.px(), 2) + pow(met.py(), 2));
239  double lepT = sqrt(pow(lepton->px(), 2) + pow(lepton->py(), 2));
240  double bT = sqrt(pow(bJet.px(), 2) + pow(bJet.py(), 2));
241  reco::Particle::LorentzVector WT = lepton->p4() + met.p4();
242  tmassWBoson_ =
243  sqrt(pow(metT + lepT, 2) - (WT.px() * WT.px()) - (WT.py() * WT.py()));
244  reco::Particle::LorentzVector topT = WT + bJet.p4();
245  tmassTopQuark_ = sqrt(pow((metT + lepT + bT), 2) - (topT.px() * topT.px()) -
246  (topT.py() * topT.py()));
247 }
248 
249 
250 void Calculate_miniAOD::operator()(const std::vector<pat::Jet>& jets) {
251  if (maxNJets_ < 0) maxNJets_ = jets.size();
252  failed_ = jets.size() < (unsigned int)maxNJets_;
253  if (failed_) {
254  return;
255  }
256 
257  // associate those jets with maximum pt of the vectorial
258  // sum to the hadronic decay chain
259  double maxPt = -1.;
260  std::vector<int> maxPtIndices;
261  maxPtIndices.push_back(-1);
262  maxPtIndices.push_back(-1);
263  maxPtIndices.push_back(-1);
264 
265  for (int idx = 0; idx < maxNJets_; ++idx) {
266  for (int jdx = idx+1; jdx < maxNJets_; ++jdx) {
267  //if (jdx <= idx) continue;
268  for (int kdx = 0; kdx < maxNJets_; ++kdx) {
269  if (kdx == idx || kdx == jdx) continue;
270  reco::Particle::LorentzVector sum =
271  jets[idx].p4() + jets[jdx].p4() + jets[kdx].p4();
272  if (maxPt < 0. || maxPt < sum.pt()) {
273  maxPt = sum.pt();
274  maxPtIndices.clear();
275  maxPtIndices.push_back(idx);
276  maxPtIndices.push_back(jdx);
277  maxPtIndices.push_back(kdx);
278  }
279  }
280  }
281  }
282  massTopQuark_ = (jets[maxPtIndices[0]].p4() + jets[maxPtIndices[1]].p4() +
283  jets[maxPtIndices[2]].p4()).mass();
284 
285  // associate those jets that get closest to the W mass
286  // with their invariant mass to the W boson
287  double wDist = -1.;
288  std::vector<int> wMassIndices;
289  wMassIndices.push_back(-1);
290  wMassIndices.push_back(-1);
291  for (unsigned idx = 0; idx < maxPtIndices.size(); ++idx) {
292  for (unsigned jdx = 0; jdx < maxPtIndices.size(); ++jdx) {
293  if (jdx == idx || maxPtIndices[idx] > maxPtIndices[jdx]) continue;
294  reco::Particle::LorentzVector sum =
295  jets[maxPtIndices[idx]].p4() + jets[maxPtIndices[jdx]].p4();
296  if (wDist < 0. || wDist > fabs(sum.mass() - wMass_)) {
297  wDist = fabs(sum.mass() - wMass_);
298  wMassIndices.clear();
299  wMassIndices.push_back(maxPtIndices[idx]);
300  wMassIndices.push_back(maxPtIndices[jdx]);
301  }
302  }
303  }
304  massWBoson_ =
305  (jets[wMassIndices[0]].p4() + jets[wMassIndices[1]].p4()).mass();
306 }
307 
308 void Calculate_miniAOD::operator2(const std::vector<pat::Jet>& jets,
309  std::vector<double> bjet, double btagWP) {
310  if (maxNJets_ < 0) maxNJets_ = jets.size();
311  failed_ = jets.size() < (unsigned int)maxNJets_;
312 
313  if (failed_) {
314  return;
315  }
316 
317  if (jets.size() != bjet.size()) {
318  return;
319  }
320 
321  // associate those jets with maximum pt of the vectorial
322  // sum to the hadronic decay chain. Require ONLY 1 btagged jet
323  double maxBPt = -1.;
324  std::vector<int> maxBPtIndices;
325  maxBPtIndices.push_back(-1);
326  maxBPtIndices.push_back(-1);
327  maxBPtIndices.push_back(-1);
328  for (int idx = 0; idx < maxNJets_; ++idx) {
329  for (int jdx = idx+1; jdx < maxNJets_; ++jdx) {
330  //if (jdx <= idx) continue;
331  for (int kdx = 0; kdx < maxNJets_; ++kdx) {
332  if (kdx == idx || kdx == jdx) continue;
333  // require only 1b-jet
334  if ((bjet[idx] > btagWP && bjet[jdx] <= btagWP &&
335  bjet[kdx] <= btagWP) ||
336  (bjet[idx] <= btagWP && bjet[jdx] > btagWP &&
337  bjet[kdx] <= btagWP) ||
338  (bjet[idx] <= btagWP && bjet[jdx] <= btagWP &&
339  bjet[kdx] > btagWP)) {
340  reco::Particle::LorentzVector sum =
341  jets[idx].p4() + jets[jdx].p4() + jets[kdx].p4();
342  if (maxBPt < 0. || maxBPt < sum.pt()) {
343  maxBPt = sum.pt();
344  maxBPtIndices.clear();
345  maxBPtIndices.push_back(idx);
346  maxBPtIndices.push_back(jdx);
347  maxBPtIndices.push_back(kdx);
348  }
349  }
350  }
351  }
352  }
353  if (maxBPtIndices[0] < 0 || maxBPtIndices[1] < 0 || maxBPtIndices[2] < 0)
354  return;
355  massBTopQuark_ = (jets[maxBPtIndices[0]].p4() + jets[maxBPtIndices[1]].p4() +
356  jets[maxBPtIndices[2]].p4()).mass();
357 }
358 
Calculate::massWBoson_
double massWBoson_
cache of w boson mass estimate
Definition: TopDQMHelpers.h:182
pat::MET
Analysis-level MET class.
Definition: MET.h:43
Calculate_miniAOD::massBTopQuark_
double massBTopQuark_
cache of b-tagged top quark mass estimate
Definition: TopDQMHelpers.h:130
Calculate::massBTopQuark
double massBTopQuark(const std::vector< reco::Jet > &jets, std::vector< double > VbtagWP, double btagWP_)
calculate b-tagged t-quark mass estimate
Definition: TopDQMHelpers.cc:23
reco::GsfElectron::p4
const LorentzVector & p4(P4Kind kind) const
Definition: GsfElectron.cc:225
Calculate_miniAOD::tmassWBoson
double tmassWBoson(pat::Muon *lep, const pat::MET &met, const pat::Jet &b)
calculate W boson transverse mass estimate
Definition: TopDQMHelpers.cc:195
Calculate_miniAOD::massTopQuark_
double massTopQuark_
cache of top quark mass estimate
Definition: TopDQMHelpers.h:128
Calculate::massTopQuark_
double massTopQuark_
cache of top quark mass estimate
Definition: TopDQMHelpers.h:184
Calculate::tmassTopQuark_
double tmassTopQuark_
cache of top quark transverse mass estimate
Definition: TopDQMHelpers.h:190
reco::Jet
Base class for all types of Jets.
Definition: Jet.h:20
Calculate::massTopQuark
double massTopQuark(const std::vector< reco::Jet > &jets)
calculate t-quark mass estimate
Definition: TopDQMHelpers.cc:18
Calculate_miniAOD::failed_
bool failed_
indicate failed associations
Definition: TopDQMHelpers.h:120
Calculate_miniAOD::massWBoson
double massWBoson(const std::vector< pat::Jet > &jets)
calculate W boson mass estimate
Definition: TopDQMHelpers.cc:178
Calculate_miniAOD::massWBoson_
double massWBoson_
cache of w boson mass estimate
Definition: TopDQMHelpers.h:126
Calculate::massWBoson
double massWBoson(const std::vector< reco::Jet > &jets)
calculate W boson mass estimate
Definition: TopDQMHelpers.cc:13
Calculate::operator2
void operator2(const std::vector< reco::Jet > &, std::vector< double >, double)
do the calculation of the t-quark mass with one b-jet
Definition: TopDQMHelpers.cc:112
Calculate_miniAOD::tmassWBoson_
double tmassWBoson_
cache of W boson transverse mass estimate
Definition: TopDQMHelpers.h:132
Calculate::maxNJets_
int maxNJets_
max. number of jets to be considered
Definition: TopDQMHelpers.h:178
Calculate_miniAOD::operator()
void operator()(const std::vector< pat::Jet > &jets)
Definition: TopDQMHelpers.cc:250
reco::MET
Definition: MET.h:42
reco::LeafCandidate::px
virtual double px() const final
x coordinate of momentum vector
Definition: LeafCandidate.h:125
mathSSE::sqrt
T sqrt(T t)
Definition: SSEVec.h:18
fwrapper::jets
vector< PseudoJet > jets
Definition: fastjetfortran_madfks.cc:42
Calculate_miniAOD::maxNJets_
int maxNJets_
max. number of jets to be considered
Definition: TopDQMHelpers.h:122
Calculate_miniAOD::wMass_
double wMass_
paramater of the w boson mass
Definition: TopDQMHelpers.h:124
RPCpg::mu
const int mu
Definition: Constants.h:22
Calculate::wMass_
double wMass_
paramater of the w boson mass
Definition: TopDQMHelpers.h:180
Calculate::tmassTopQuark
double tmassTopQuark(reco::RecoCandidate *lep, const reco::MET &met, const reco::Jet &b)
calculate top quark transverse mass estimate
Definition: TopDQMHelpers.cc:35
Calculate_miniAOD::massTopQuark
double massTopQuark(const std::vector< pat::Jet > &jets)
calculate t-quark mass estimate
Definition: TopDQMHelpers.cc:183
Calculate_miniAOD::tmassTopQuark
double tmassTopQuark(pat::Muon *lep, const pat::MET &met, const pat::Jet &b)
calculate top quark transverse mass estimate
Definition: TopDQMHelpers.cc:215
TtFullLepEvtBuilder_cfi.maxNJets
maxNJets
maximum number of jets taken into account per event for each hypothesis (this parameter is used in th...
Definition: TtFullLepEvtBuilder_cfi.py:28
Calculate_miniAOD::operator2
void operator2(const std::vector< pat::Jet > &, std::vector< double >, double)
do the calculation of the t-quark mass with one b-jet
Definition: TopDQMHelpers.cc:308
Calculate::tmassWBoson
double tmassWBoson(reco::RecoCandidate *lep, const reco::MET &met, const reco::Jet &b)
calculate W boson transverse mass estimate
Definition: TopDQMHelpers.cc:29
RazorAnalyzer.met
met
===> hadronic RAZOR
Definition: RazorAnalyzer.py:70
Calculate_miniAOD::massBTopQuark
double massBTopQuark(const std::vector< pat::Jet > &jets, std::vector< double > VbtagWP, double btagWP_)
calculate b-tagged t-quark mass estimate
Definition: TopDQMHelpers.cc:188
training_settings.idx
idx
Definition: training_settings.py:16
b
double b
Definition: hdecay.h:120
pat::Electron
Analysis-level electron class.
Definition: Electron.h:52
pat::Jet
Analysis-level calorimeter jet class.
Definition: Jet.h:78
Calculate::operator()
void operator()(const std::vector< reco::Jet > &jets)
Definition: TopDQMHelpers.cc:54
Calculate::failed_
bool failed_
indicate failed associations
Definition: TopDQMHelpers.h:176
Calculate_miniAOD::Calculate_miniAOD
Calculate_miniAOD(int maxNJets, double wMass)
default constructor
Definition: TopDQMHelpers.cc:168
MuonErrorMatrixAnalyzer_cfi.maxPt
maxPt
Definition: MuonErrorMatrixAnalyzer_cfi.py:19
Calculate_miniAOD::tmassTopQuark_
double tmassTopQuark_
cache of top quark transverse mass estimate
Definition: TopDQMHelpers.h:134
Calculate::Calculate
Calculate(int maxNJets, double wMass)
default constructor
Definition: TopDQMHelpers.cc:3
reco::LeafCandidate::p4
virtual const LorentzVector & p4() const final
four-momentum Lorentz vector
Definition: LeafCandidate.h:99
reco::Particle::LorentzVector
math::XYZTLorentzVector LorentzVector
Lorentz vector.
Definition: Particle.h:21
pat::Muon
Analysis-level muon class.
Definition: Muon.h:49
reco::LeafCandidate::py
virtual double py() const final
y coordinate of momentum vector
Definition: LeafCandidate.h:127
funct::pow
Power< A, B >::type pow(const A &a, const B &b)
Definition: Power.h:40
Calculate::tmassWBoson_
double tmassWBoson_
cache of W boson transverse mass estimate
Definition: TopDQMHelpers.h:188
Calculate::massBTopQuark_
double massBTopQuark_
cache of b-tagged top quark mass estimate
Definition: TopDQMHelpers.h:186
pseudoTop_cfi.wMass
wMass
Definition: pseudoTop_cfi.py:12
Generated for CMSSW Reference Manual by doxygen 1.8.11