|
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
|
#include "CommonTools/CandAlgos/interface/GenJetParticleSelector.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "SimGeneral/HepPDTRecord/interface/PdtEntry.h"
#include "FWCore/Utilities/interface/EDMException.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include <algorithm>
using namespace std;
using namespace edm;
GenJetParticleSelector::GenJetParticleSelector(const ParameterSet& cfg, edm::ConsumesCollector& iC)
: stableOnly_(cfg.getParameter<bool>("stableOnly")),
partons_(false),
bInclude_(false),
tableToken_(iC.esConsumes()) {
const string excludeString("excludeList");
const string includeString("includeList");
vpdt includeList, excludeList;
vector<string> vPdtParams = cfg.getParameterNamesForType<vpdt>();
bool found = std::find(vPdtParams.begin(), vPdtParams.end(), includeString) != vPdtParams.end();
if (found)
includeList = cfg.getParameter<vpdt>(includeString);
found = find(vPdtParams.begin(), vPdtParams.end(), excludeString) != vPdtParams.end();
if (found)
excludeList = cfg.getParameter<vpdt>(excludeString);
const string partonsString("partons");
vector<string> vBoolParams = cfg.getParameterNamesForType<bool>();
found = find(vBoolParams.begin(), vBoolParams.end(), partonsString) != vBoolParams.end();
if (found)
partons_ = cfg.getParameter<bool>(partonsString);
bool bExclude = false;
if (!includeList.empty())
bInclude_ = true;
if (!excludeList.empty())
bExclude = true;
if (bInclude_ && bExclude) {
throw cms::Exception("ConfigError", "not allowed to use both includeList and excludeList at the same time\n");
} else if (bInclude_) {
pdtList_ = includeList;
} else {
pdtList_ = excludeList;
}
if (stableOnly_ && partons_) {
throw cms::Exception("ConfigError", "not allowed to have both stableOnly and partons true at the same time\n");
}
}
void GenJetParticleSelector::fillPSetDescription(edm::ParameterSetDescription& desc) {
desc.add<bool>("stableOnly", true);
}
bool GenJetParticleSelector::operator()(const reco::Candidate& p) {
int status = p.status();
int id = abs(p.pdgId());
if ((!stableOnly_ || status == 1) && !partons_ && ((pIds_.find(id) == pIds_.end()) ^ bInclude_))
return true;
else if (partons_ && (p.numberOfDaughters() > 0 && (p.daughter(0)->pdgId() == 91 || p.daughter(0)->pdgId() == 92)) &&
(((pIds_.find(id) == pIds_.end()) ^ bInclude_)))
return true;
else
return false;
}
void GenJetParticleSelector::init(const edm::EventSetup& es) {
auto const& pdt = es.getData(tableToken_);
for (vpdt::iterator i = pdtList_.begin(); i != pdtList_.end(); ++i)
i->setup(pdt);
}
|