|
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
|
#ifndef HLTrigger_JetMET_plugins_HLTMultiplicityValueProducerFromNestedCollection_h
#define HLTrigger_JetMET_plugins_HLTMultiplicityValueProducerFromNestedCollection_h
#include <memory>
#include "FWCore/Framework/interface/Frameworkfwd.h"
#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/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
template <class INP_TYPE, class OUT_TYPE>
class HLTMultiplicityValueProducerFromNestedCollection : public edm::global::EDProducer<> {
public:
explicit HLTMultiplicityValueProducerFromNestedCollection(edm::ParameterSet const&);
~HLTMultiplicityValueProducerFromNestedCollection() override {}
static void fillDescriptions(edm::ConfigurationDescriptions&);
protected:
void produce(edm::StreamID, edm::Event&, edm::EventSetup const&) const override;
edm::EDGetTokenT<INP_TYPE> const src_token_;
OUT_TYPE const defaultValue_;
};
template <class INP_TYPE, class OUT_TYPE>
HLTMultiplicityValueProducerFromNestedCollection<INP_TYPE, OUT_TYPE>::HLTMultiplicityValueProducerFromNestedCollection(
edm::ParameterSet const& iConfig)
: src_token_(consumes<INP_TYPE>(iConfig.getParameter<edm::InputTag>("src"))),
defaultValue_(iConfig.getParameter<OUT_TYPE>("defaultValue")) {
produces<OUT_TYPE>();
}
template <class INP_TYPE, class OUT_TYPE>
void HLTMultiplicityValueProducerFromNestedCollection<INP_TYPE, OUT_TYPE>::produce(
edm::StreamID, edm::Event& iEvent, edm::EventSetup const& iSetup) const {
auto const& objHandle(iEvent.getHandle(src_token_));
if (objHandle.isValid()) {
LogDebug("Input") << "size of input collection: " << objHandle->size();
OUT_TYPE objMult(0);
for (auto const& obj : *objHandle) {
objMult += obj.size();
}
LogDebug("Output") << "size of output objects: " << objMult;
iEvent.put(std::make_unique<OUT_TYPE>(objMult));
} else {
iEvent.put(std::make_unique<OUT_TYPE>(defaultValue_));
}
}
template <class INP_TYPE, class OUT_TYPE>
void HLTMultiplicityValueProducerFromNestedCollection<INP_TYPE, OUT_TYPE>::fillDescriptions(
edm::ConfigurationDescriptions& descriptions) {
edm::ParameterSetDescription desc;
desc.add<edm::InputTag>("src", edm::InputTag(""))->setComment("input collection");
desc.add<OUT_TYPE>("defaultValue", 0)->setComment("default output value (used when input collection is unavailable)");
descriptions.addWithDefaultLabel(desc);
}
#endif // HLTrigger_JetMET_plugins_HLTMultiplicityValueProducerFromNestedCollection_h
|