|
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
|
/** \class edm::ThinningProducer
\author W. David Dagenhart, created 11 June 2014
*/
#include "FWCore/Framework/interface/one/EDProducer.h"
#include "DataFormats/TestObjects/interface/ThingCollection.h"
#include "DataFormats/TestObjects/interface/TrackOfThings.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/EDGetToken.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Framework/interface/MakerMacros.h"
namespace edm {
class EventSetup;
}
namespace edmtest {
class TrackOfThingsProducer : public edm::one::EDProducer<> {
public:
explicit TrackOfThingsProducer(edm::ParameterSet const&);
~TrackOfThingsProducer() override;
void produce(edm::Event&, edm::EventSetup const&) override;
private:
void incrementKey(std::vector<unsigned int>::const_iterator& key) const;
edm::EDGetTokenT<ThingCollection> inputToken_;
std::vector<unsigned int> keysToReference_;
};
TrackOfThingsProducer::TrackOfThingsProducer(edm::ParameterSet const& pset) {
inputToken_ = consumes<ThingCollection>(pset.getParameter<edm::InputTag>("inputTag"));
keysToReference_ = pset.getParameter<std::vector<unsigned int> >("keysToReference");
produces<TrackOfThingsCollection>();
}
TrackOfThingsProducer::~TrackOfThingsProducer() {}
void TrackOfThingsProducer::incrementKey(std::vector<unsigned int>::const_iterator& key) const {
++key;
if (key == keysToReference_.end())
key = keysToReference_.begin();
}
void TrackOfThingsProducer::produce(edm::Event& event, edm::EventSetup const&) {
edm::Handle<ThingCollection> inputCollection = event.getHandle(inputToken_);
auto result = std::make_unique<TrackOfThingsCollection>();
// Arbitrarily fabricate some fake data with TrackOfThings pointing to
// Thing objects in products written to the event by a different module.
// The numbers in the keys here are made up, passed in via the configuration
// and have no meaning other than that we will later check that we can
// read out what we put in here.
std::vector<unsigned int>::const_iterator key = keysToReference_.begin();
for (unsigned int i = 0; i < 5; ++i) {
edmtest::TrackOfThings trackOfThings;
trackOfThings.ref1 = edm::Ref<ThingCollection>(inputCollection, *key);
incrementKey(key);
trackOfThings.ref2 = edm::Ref<ThingCollection>(inputCollection, *key);
incrementKey(key);
trackOfThings.ptr1 = edm::Ptr<Thing>(inputCollection, *key);
incrementKey(key);
trackOfThings.ptr2 = edm::Ptr<Thing>(inputCollection, *key);
incrementKey(key);
trackOfThings.refToBase1 = edm::RefToBase<Thing>(trackOfThings.ref1);
for (auto iKey : keysToReference_) {
trackOfThings.refVector1.push_back(edm::Ref<ThingCollection>(inputCollection, iKey));
trackOfThings.ptrVector1.push_back(edm::Ptr<Thing>(inputCollection, iKey));
trackOfThings.refToBaseVector1.push_back(
edm::RefToBase<Thing>(edm::Ref<ThingCollection>(inputCollection, iKey)));
}
result->push_back(trackOfThings);
}
event.put(std::move(result));
}
} // namespace edmtest
using edmtest::TrackOfThingsProducer;
DEFINE_FWK_MODULE(TrackOfThingsProducer);
|