|
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 HLTCollectionProducer_h
#define HLTCollectionProducer_h
//
// Package: HLTCollectionProducer
// Class: HLTCollectionProducer
//
/**\class HLTCollectionProducer
Extract objects from trigger::TriggerFilterObjectWithRefs and fill
them into a new collection (Based on GetJetsFromHLTobject.h)
*/
//
// Original Author: Ian Tomalin
//
#include <string>
#include <vector>
// user include files
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/Common/interface/RefVector.h"
#include "DataFormats/HLTReco/interface/TriggerFilterObjectWithRefs.h"
#include "HLTrigger/HLTcore/interface/defaultModuleLabel.h"
//
// class declaration
//
template <typename T>
class HLTCollectionProducer : public edm::global::EDProducer<> {
public:
explicit HLTCollectionProducer(const edm::ParameterSet&);
~HLTCollectionProducer() override;
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
void produce(edm::StreamID, edm::Event&, const edm::EventSetup&) const override;
private:
const edm::InputTag hltObjectTag_;
const edm::EDGetTokenT<trigger::TriggerFilterObjectWithRefs> hltObjectToken_;
const std::vector<int> triggerTypes_;
};
//
// constructors and destructor
//
template <typename T>
HLTCollectionProducer<T>::HLTCollectionProducer(const edm::ParameterSet& iConfig)
: hltObjectTag_(iConfig.getParameter<edm::InputTag>("HLTObject")),
hltObjectToken_(consumes<trigger::TriggerFilterObjectWithRefs>(hltObjectTag_)),
triggerTypes_(iConfig.getParameter<std::vector<int>>("TriggerTypes")) {
produces<std::vector<T>>();
}
template <typename T>
HLTCollectionProducer<T>::~HLTCollectionProducer() {}
template <typename T>
void HLTCollectionProducer<T>::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
edm::ParameterSetDescription desc;
desc.add<edm::InputTag>("HLTObject", edm::InputTag("TriggerFilterObjectWithRefs"));
std::vector<int> triggerTypes;
desc.add<std::vector<int>>("TriggerTypes", triggerTypes);
descriptions.add(defaultModuleLabel<HLTCollectionProducer<T>>(), desc);
}
//
// member functions
//
// ------------ method called to produce the data ------------
template <typename T>
void HLTCollectionProducer<T>::produce(edm::StreamID iStreamID,
edm::Event& iEvent,
const edm::EventSetup& iSetup) const {
std::unique_ptr<std::vector<T>> collection(new std::vector<T>());
// get hold of collection of TriggerFilterObjectsWithRefs
edm::Handle<trigger::TriggerFilterObjectWithRefs> hltObject;
iEvent.getByToken(hltObjectToken_, hltObject);
std::vector<edm::Ref<std::vector<T>>> objectRefs;
for (size_t t = 0; t < triggerTypes_.size(); ++t) {
objectRefs.clear();
hltObject->getObjects(triggerTypes_[t], objectRefs);
for (size_t i = 0; i < objectRefs.size(); ++i) {
collection->push_back(*objectRefs[i]);
}
}
iEvent.put(std::move(collection));
}
#endif // HLTCollectionProducer_h
|