|
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
|
#include "GeneratorInterface/GenFilters/plugins/MCPdgIndexFilter.h"
#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
MCPdgIndexFilter::MCPdgIndexFilter(const edm::ParameterSet& cfg)
: token_(consumes<edm::HepMCProduct>(
edm::InputTag(cfg.getUntrackedParameter("moduleLabel", std::string("generator")), "unsmeared"))),
pdgID(cfg.getParameter<std::vector<int> >("PdgId")),
index(cfg.getParameter<std::vector<unsigned> >("Index")),
maxIndex(*std::max_element(index.begin(), index.end())),
taggingMode(cfg.getUntrackedParameter<bool>("TagMode", false)) {
if (pdgID.size() != index.size())
edm::LogWarning("MCPdgIndexFilter") << "Configuration Error :"
<< "Sizes of array parameters 'PdgId' and 'Index' differ.";
if (taggingMode) {
auto tag = cfg.getUntrackedParameter<std::string>("Tag", "");
putToken_ = produces<bool>(tag);
edm::LogInfo("TagMode") << "Filter result in '" << tag << "', filtering disabled.";
}
}
bool MCPdgIndexFilter::filter(edm::StreamID, edm::Event& evt, const edm::EventSetup&) const {
bool result = pass(evt);
LogDebug("FilterResult") << (result ? "Pass" : "Fail");
if (!taggingMode)
return result;
evt.emplace(putToken_, result);
return true;
}
bool MCPdgIndexFilter::pass(const edm::Event& evt) const {
edm::Handle<edm::HepMCProduct> hepmc;
evt.getByToken(token_, hepmc);
const HepMC::GenEvent* genEvent = hepmc->GetEvent();
HepMC::GenEvent::particle_const_iterator p(genEvent->particles_begin()), p_end(genEvent->particles_end());
for (unsigned i = 0; p != p_end && i <= maxIndex; ++p, i++) {
LogDebug("Particle") << "index: " << i << " pdgID: " << (*p)->pdg_id();
for (unsigned j = 0; j < pdgID.size(); j++) {
if (i == index[j] && pdgID[j] != (*p)->pdg_id())
return false;
}
}
return true;
}
|