|
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
|
#include <string>
#include "DataFormats/TestObjects/interface/OtherThingCollection.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "OtherThingAlgorithm.h"
#include "FWCore/Utilities/interface/EDGetToken.h"
#include "FWCore/Utilities/interface/EDPutToken.h"
#include "FWCore/Utilities/interface/InputTag.h"
namespace edmtest {
class OtherThingProducer : public edm::global::EDProducer<> {
public:
explicit OtherThingProducer(edm::ParameterSet const& ps);
~OtherThingProducer() override;
void produce(edm::StreamID, edm::Event& e, edm::EventSetup const& c) const override;
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
private:
OtherThingAlgorithm alg_;
edm::EDGetTokenT<ThingCollection> thingToken_;
edm::EDPutTokenT<OtherThingCollection> putToken_;
bool useRefs_;
bool refsAreTransient_;
bool thingMissing_;
};
OtherThingProducer::OtherThingProducer(edm::ParameterSet const& pset) : alg_(), refsAreTransient_(false) {
putToken_ = produces<OtherThingCollection>("testUserTag");
useRefs_ = pset.getUntrackedParameter<bool>("useRefs");
thingMissing_ = pset.getUntrackedParameter<bool>("thingMissing");
if (useRefs_ or thingMissing_) {
thingToken_ = consumes<ThingCollection>(pset.getParameter<edm::InputTag>("thingTag"));
}
refsAreTransient_ = pset.getUntrackedParameter<bool>("transient");
}
// Virtual destructor needed.
OtherThingProducer::~OtherThingProducer() {}
// Functions that gets called by framework every event
void OtherThingProducer::produce(edm::StreamID, edm::Event& e, edm::EventSetup const&) const {
// Step A: Get Inputs
// Step B: Create empty output
auto result = std::make_unique<OtherThingCollection>(); //Empty
// Step C: Get data for algorithm
edm::Handle<ThingCollection> parentHandle;
if (useRefs_ or thingMissing_) {
parentHandle = e.getHandle(thingToken_);
//If not here, throw exception
if (thingMissing_) {
if (parentHandle.isValid()) {
throw cms::Exception("TestFailure")
<< "The ThingCollection is available when it was expected to not be available";
}
} else {
*parentHandle;
}
}
// Step D: Invoke the algorithm, passing in inputs (NONE) and getting back outputs.
alg_.run(parentHandle, *result, useRefs_, refsAreTransient_);
// Step E: Put outputs into event
e.put(putToken_, std::move(result));
}
void OtherThingProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
edm::ParameterSetDescription desc;
desc.add<edm::InputTag>("thingTag", edm::InputTag("Thing"))->setComment("Where to get the ThingCollection");
desc.addUntracked<bool>("useRefs", true)
->setComment("Actually get the ThingCollection and build edm::Refs to the contained items.");
desc.addUntracked<bool>("transient", false)
->setComment("If true, then the Refs constructed by the ThingCollection can not be persisted");
desc.addUntracked<bool>("thingMissing", false)->setComment("If true, expect that thing collection is missing");
descriptions.add("otherThingProd", desc);
}
} // namespace edmtest
using edmtest::OtherThingProducer;
DEFINE_FWK_MODULE(OtherThingProducer);
|