|
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
|
#ifdef __clang__
#pragma GCC diagnostic ignored "-Wc++20-extensions"
#endif
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
namespace edm {
/**
* This class is the physical EDProducer part of the SwitchProducer
* infrastructure. It must be configured only with the
* SwitchProducer python construct.
*
* The purposes of this EDProducer are
* - Create the consumes() links to the chosen case to make the prefetching work correclty
* - Forward the produces() information to create proper ProductDescription objects
*/
class SwitchProducer : public global::EDProducer<> {
public:
explicit SwitchProducer(ParameterSet const& iConfig);
~SwitchProducer() override = default;
static void fillDescriptions(ConfigurationDescriptions& descriptions);
void produce(StreamID, Event& e, EventSetup const& c) const final {}
};
SwitchProducer::SwitchProducer(ParameterSet const& iConfig) {
auto const& moduleLabel = iConfig.getParameter<std::string>("@module_label");
auto const& chosenLabel = iConfig.getUntrackedParameter<std::string>("@chosen_case");
auto const& processName = iConfig.getUntrackedParameter<std::string>("@process_name");
callWhenNewProductsRegistered([=, this](edm::ProductDescription const& iBranch) {
if (iBranch.moduleLabel() == chosenLabel and iBranch.processName() == processName) {
if (iBranch.branchType() != InEvent) {
throw Exception(errors::UnimplementedFeature)
<< "SwitchProducer does not support non-event branches. Got " << iBranch.branchType()
<< " for SwitchProducer with label " << moduleLabel << " whose chosen case is " << chosenLabel << ".";
}
// With consumes, create the connection to the chosen case EDProducer for prefetching
this->consumes(edm::TypeToGet{iBranch.unwrappedTypeID(), PRODUCT_TYPE},
edm::InputTag{iBranch.moduleLabel(), iBranch.productInstanceName(), iBranch.processName()});
// With produces, create a producer-like ProductDescription
// early-enough for it to be flagged as non-OnDemand in case
// the SwithcProducer is on a Path
this->produces(iBranch.unwrappedTypeID(), iBranch.productInstanceName()).setSwitchAlias(iBranch.moduleLabel());
}
});
}
void SwitchProducer::fillDescriptions(ConfigurationDescriptions& descriptions) {
ParameterSetDescription desc;
desc.add<std::vector<std::string>>("@all_cases");
desc.addUntracked<std::string>("@chosen_case");
desc.addUntracked<std::string>("@process_name");
descriptions.addDefault(desc);
}
} // namespace edm
using edm::SwitchProducer;
DEFINE_FWK_MODULE(SwitchProducer);
|