|
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
93
94
95
|
#ifndef UtilAlgos_NtpProducer_h
#define UtilAlgos_NtpProducer_h
/** \class NtpProducer
*
* Creates histograms defined in config file
*
* \author: Luca Lista, INFN
*
* Template parameters:
* - C : Concrete candidate collection type
*
*/
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Provenance/interface/RunLumiEventNumber.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "CommonTools/Utils/interface/StringObjectFunction.h"
template <typename C>
class NtpProducer : public edm::stream::EDProducer<> {
public:
/// constructor from parameter set
NtpProducer(const edm::ParameterSet&);
/// destructor
~NtpProducer() override;
protected:
/// process an event
void produce(edm::Event&, const edm::EventSetup&) override;
private:
/// label of the collection to be read in
edm::EDGetTokenT<C> srcToken_;
/// variable tags
std::vector<std::pair<std::string, StringObjectFunction<typename C::value_type> > > tags_;
bool lazyParser_;
std::string prefix_;
bool eventInfo_;
};
template <typename C>
NtpProducer<C>::NtpProducer(const edm::ParameterSet& par)
: srcToken_(consumes<C>(par.template getParameter<edm::InputTag>("src"))),
lazyParser_(par.template getUntrackedParameter<bool>("lazyParser", false)),
prefix_(par.template getUntrackedParameter<std::string>("prefix", "")),
eventInfo_(par.template getUntrackedParameter<bool>("eventInfo", true)) {
std::vector<edm::ParameterSet> variables = par.template getParameter<std::vector<edm::ParameterSet> >("variables");
std::vector<edm::ParameterSet>::const_iterator q = variables.begin(), end = variables.end();
if (eventInfo_) {
produces<edm::EventNumber_t>(prefix_ + "EventNumber").setBranchAlias(prefix_ + "EventNumber");
produces<unsigned int>(prefix_ + "RunNumber").setBranchAlias(prefix_ + "RunNumber");
produces<unsigned int>(prefix_ + "LumiBlock").setBranchAlias(prefix_ + "Lumiblock");
}
for (; q != end; ++q) {
std::string tag = prefix_ + q->getUntrackedParameter<std::string>("tag");
StringObjectFunction<typename C::value_type> quantity(q->getUntrackedParameter<std::string>("quantity"),
lazyParser_);
tags_.push_back(std::make_pair(tag, quantity));
produces<std::vector<float> >(tag).setBranchAlias(tag);
}
}
template <typename C>
NtpProducer<C>::~NtpProducer() {}
template <typename C>
void NtpProducer<C>::produce(edm::Event& iEvent, const edm::EventSetup&) {
edm::Handle<C> coll;
iEvent.getByToken(srcToken_, coll);
if (eventInfo_) {
std::unique_ptr<edm::EventNumber_t> event(new edm::EventNumber_t);
std::unique_ptr<unsigned int> run(new unsigned int);
std::unique_ptr<unsigned int> lumi(new unsigned int);
*event = iEvent.id().event();
*run = iEvent.id().run();
*lumi = iEvent.luminosityBlock();
iEvent.put(std::move(event), prefix_ + "EventNumber");
iEvent.put(std::move(run), prefix_ + "RunNumber");
iEvent.put(std::move(lumi), prefix_ + "LumiBlock");
}
typename std::vector<std::pair<std::string, StringObjectFunction<typename C::value_type> > >::const_iterator
q = tags_.begin(),
end = tags_.end();
for (; q != end; ++q) {
std::unique_ptr<std::vector<float> > x(new std::vector<float>);
x->reserve(coll->size());
for (typename C::const_iterator elem = coll->begin(); elem != coll->end(); ++elem) {
x->push_back(q->second(*elem));
}
iEvent.put(std::move(x), q->first);
}
}
#endif
|