|
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
|
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/JetReco/interface/CaloJet.h"
#include "DataFormats/JetReco/interface/CaloJetCollection.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DQM/DataScouting/interface/DiJetVarProducer.h"
#include "TLorentzVector.h"
#include "TVector3.h"
#include <memory>
#include <vector>
//
// constructors and destructor
//
DiJetVarProducer::DiJetVarProducer(const edm::ParameterSet &iConfig)
: inputJetTag_(iConfig.getParameter<edm::InputTag>("inputJetTag")),
wideJetDeltaR_(iConfig.getParameter<double>("wideJetDeltaR")) {
// register your products
// produces<std::vector<double> >("dijetvariables");
produces<std::vector<math::PtEtaPhiMLorentzVector>>("widejets");
// set Token(-s)
inputJetTagToken_ = consumes<reco::CaloJetCollection>(iConfig.getParameter<edm::InputTag>("inputJetTag"));
LogDebug("") << "Input Jet Tag: " << inputJetTag_.encode() << " ";
LogDebug("") << "Radius Parameter Wide Jet: " << wideJetDeltaR_ << ".";
}
// ------------ method called to produce the data ------------
void DiJetVarProducer::produce(edm::StreamID, edm::Event &iEvent, const edm::EventSetup &iSetup) const {
using namespace std;
using namespace edm;
using namespace reco;
// ## The output collections
// std::unique_ptr<std::vector<double> > dijetvariables(new
// std::vector<double>);
std::unique_ptr<std::vector<math::PtEtaPhiMLorentzVector>> widejets(new std::vector<math::PtEtaPhiMLorentzVector>);
// ## Get jet collection
edm::Handle<reco::CaloJetCollection> calojets_handle;
iEvent.getByToken(inputJetTagToken_, calojets_handle);
// cout << "size: " << calojets_handle->size() << endl;
// ## Wide Jet Algorithm
// At least two jets
if (calojets_handle->size() >= 2) {
TLorentzVector wj1_tmp;
TLorentzVector wj2_tmp;
TLorentzVector wj1;
TLorentzVector wj2;
TLorentzVector wdijet;
// // Loop over all the input jets
// for(reco::CaloJetCollection::const_iterator it =
// calojets_handle->begin(); it != calojets_handle->end(); ++it)
// {
// cout << "jet: " << it->pt() << " " << it->eta() << " " << it->phi()
// << endl;
// }
// Find two leading jets
TLorentzVector jet1, jet2;
reco::CaloJetCollection::const_iterator j1 = calojets_handle->begin();
reco::CaloJetCollection::const_iterator j2 = j1;
++j2;
jet1.SetPtEtaPhiM(j1->pt(), j1->eta(), j1->phi(), j1->mass());
jet2.SetPtEtaPhiM(j2->pt(), j2->eta(), j2->phi(), j2->mass());
// cout << "j1: " << jet1.Pt() << " " << jet1.Eta() << " " << jet1.Phi() <<
// endl; cout << "j2: " << jet2.Pt() << " " << jet2.Eta() << " " <<
// jet2.Phi() << endl;
// Create wide jets (radiation recovery algorithm)
for (reco::CaloJetCollection::const_iterator it = calojets_handle->begin(); it != calojets_handle->end(); ++it) {
TLorentzVector currentJet;
currentJet.SetPtEtaPhiM(it->pt(), it->eta(), it->phi(), it->mass());
double DeltaR1 = currentJet.DeltaR(jet1);
double DeltaR2 = currentJet.DeltaR(jet2);
if (DeltaR1 < DeltaR2 && DeltaR1 < wideJetDeltaR_) {
wj1_tmp += currentJet;
} else if (DeltaR2 < wideJetDeltaR_) {
wj2_tmp += currentJet;
}
}
// Re-order the wide jets in pT
if (wj1_tmp.Pt() > wj2_tmp.Pt()) {
wj1 = wj1_tmp;
wj2 = wj2_tmp;
} else {
wj1 = wj2_tmp;
wj2 = wj1_tmp;
}
// Create dijet system
wdijet = wj1 + wj2;
// cout << "j1 wide: " << wj1.Pt() << " " << wj1.Eta() << " " <<
// wj1.Phi() << " " << wj1.M() << endl; cout << "j2 wide: " <<
// wj2.Pt() << " " << wj2.Eta() << " " << wj2.Phi() << " " << wj2.M()
// << endl; cout << "MJJWide: " << wdijet.M() << endl; cout <<
// "DeltaEtaJJWide: " << fabs(wj1.Eta()-wj2.Eta()) << endl; cout <<
// "DeltaPhiJJWide: " << fabs(wj1.DeltaPhi(wj2)) << endl;
// // Put variables in the container
// dijetvariables->push_back( wdijet.M() ); //0 =
// MJJWide dijetvariables->push_back( fabs(wj1.Eta()-wj2.Eta()) );
// //1 = DeltaEtaJJWide dijetvariables->push_back(
// fabs(wj1.DeltaPhi(wj2)) ); //2 = DeltaPhiJJWide
// Put widejets in the container
math::PtEtaPhiMLorentzVector wj1math(wj1.Pt(), wj1.Eta(), wj1.Phi(), wj1.M());
math::PtEtaPhiMLorentzVector wj2math(wj2.Pt(), wj2.Eta(), wj2.Phi(), wj2.M());
widejets->push_back(wj1math);
widejets->push_back(wj2math);
}
// else
// {
// // Put variables in the container
// dijetvariables->push_back( -1 ); //0 = MJJWide
// dijetvariables->push_back( -1 ); //1 = DeltaEtaJJWide
// dijetvariables->push_back( -1 ); //2 = DeltaPhiJJWide
// }
// ## Put objects in the Event
// iEvent.put(std::move(dijetvariables), "dijetvariables");
iEvent.put(std::move(widejets), "widejets");
}
DEFINE_FWK_MODULE(DiJetVarProducer);
|