|
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
|
#include "HLTriggerOffline/Tau/interface/HLTTauRefCombiner.h"
#include "Math/GenVector/VectorUtil.h"
using namespace edm;
using namespace std;
HLTTauRefCombiner::HLTTauRefCombiner(const edm::ParameterSet &iConfig)
: matchDeltaR_{iConfig.getParameter<double>("MatchDeltaR")},
outName_{iConfig.getParameter<string>("OutputCollection")} {
std::vector<edm::InputTag> inputCollVector_ = iConfig.getParameter<std::vector<InputTag>>("InputCollections");
for (unsigned int ii = 0; ii < inputCollVector_.size(); ++ii) {
inputColl_.push_back(consumes<LorentzVectorCollection>(inputCollVector_[ii]));
}
produces<LorentzVectorCollection>(outName_);
}
void HLTTauRefCombiner::produce(edm::StreamID, edm::Event &iEvent, const edm::EventSetup &iES) const {
unique_ptr<LorentzVectorCollection> out_product(new LorentzVectorCollection);
// Create The Handles..
std::vector<Handle<LorentzVectorCollection>> handles;
bool allCollectionsExist = true;
// Map the Handles to the collections if all collections exist
for (size_t i = 0; i < inputColl_.size(); ++i) {
edm::Handle<LorentzVectorCollection> tmp;
if (iEvent.getByToken(inputColl_[i], tmp)) {
handles.push_back(tmp);
} else {
allCollectionsExist = false;
}
}
// The reference output object collection will be the first one..
if (allCollectionsExist) {
// loop on the first collection
for (size_t i = 0; i < (handles[0])->size(); ++i) {
bool MatchedObj = true;
// get reference Vector
const LorentzVector lvRef = (*(handles[0]))[i];
// Loop on all other collections and match
for (size_t j = 1; j < handles.size(); ++j) {
if (!match(lvRef, *(handles[j])))
MatchedObj = false;
}
// If the Object is Matched Everywhere store
if (MatchedObj) {
out_product->push_back(lvRef);
}
}
// Put product to file
iEvent.put(std::move(out_product), outName_);
}
}
bool HLTTauRefCombiner::match(const LorentzVector &lv, const LorentzVectorCollection &lvcol) const {
bool matched = false;
if (!lvcol.empty())
for (LorentzVectorCollection::const_iterator it = lvcol.begin(); it != lvcol.end(); ++it) {
double delta = ROOT::Math::VectorUtil::DeltaR(lv, *it);
if (delta < matchDeltaR_) {
matched = true;
}
}
return matched;
}
|