|
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
|
#ifndef CommonTools_UtilAlgos_ObjectSelectorProducer_h
#define CommonTools_UtilAlgos_ObjectSelectorProducer_h
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "CommonTools/UtilAlgos/interface/StoreManagerTrait.h"
#include "CommonTools/UtilAlgos/interface/SelectedOutputCollectionTrait.h"
#include "CommonTools/UtilAlgos/interface/NullPostProcessor.h"
#include "CommonTools/UtilAlgos/interface/EventSetupInitTrait.h"
#include <utility>
#include <vector>
#include <memory>
#include <algorithm>
/**
* This class template is like ObjectSelector, but it is an EDProducer
* instead of EDFilter. Use case is that when the filter decisions are
* ignored (cms.ignore in configuration or EDFilter returns always
* true), EDProducers are better for the unscheduled mode than
* EDFilters.
*/
template <typename Selector,
typename OutputCollection,
typename PostProcessor,
typename StoreManager,
typename Base,
typename Init>
class ObjectSelectorProducer : public Base {
public:
/// constructor
explicit ObjectSelectorProducer(const edm::ParameterSet& cfg)
: Base(cfg),
srcToken_(
this->template consumes<typename Selector::collection>(cfg.template getParameter<edm::InputTag>("src"))),
selectorInit_(this->consumesCollector()),
selector_(cfg, this->consumesCollector()),
postProcessor_(cfg, this->consumesCollector()) {
postProcessor_.init(*this);
}
/// destructor
~ObjectSelectorProducer() override {}
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
edm::ParameterSetDescription desc;
desc.add<edm::InputTag>("src", edm::InputTag(""));
Selector::fillPSetDescription(desc);
descriptions.addWithDefaultLabel(desc);
}
private:
/// process one event
void produce(edm::Event& evt, const edm::EventSetup& es) override {
selectorInit_.init(selector_, evt, es);
edm::Handle<typename Selector::collection> source;
evt.getByToken(srcToken_, source);
StoreManager manager(source);
selector_.select(source, evt, es);
manager.cloneAndStore(selector_.begin(), selector_.end(), evt);
edm::OrphanHandle<OutputCollection> filtered = manager.put(evt);
postProcessor_.process(filtered, evt);
}
/// source collection label
edm::EDGetTokenT<typename Selector::collection> srcToken_;
/// Object collection selector
Init selectorInit_;
Selector selector_;
/// post processor
PostProcessor postProcessor_;
};
#endif
|