|
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
|
#ifndef MCMatchSelector_h
#define MCMatchSelector_h
/* \class MCMatchSelector
*
* Extended version of MCTruthPairSelector. Preselects matches
* based on charge, pdgId and status.
*/
#include <set>
#include "FWCore/ParameterSet/interface/ParameterSet.h"
namespace reco {
template <typename T1, typename T2>
class MCMatchSelector {
public:
MCMatchSelector(const edm::ParameterSet& cfg) : checkCharge_(cfg.getParameter<bool>("checkCharge")) {
std::vector<int> ids = cfg.getParameter<std::vector<int> >("mcPdgId");
for (std::vector<int>::const_iterator i = ids.begin(); i != ids.end(); ++i)
ids_.insert(*i);
std::vector<int> status = cfg.getParameter<std::vector<int> >("mcStatus");
for (std::vector<int>::const_iterator i = status.begin(); i != status.end(); ++i)
status_.insert(*i);
}
/// true if match is possible
bool operator()(const T1& c, const T2& mc) const {
if (checkCharge_ && c.charge() != mc.charge())
return false;
if (!ids_.empty()) {
if (ids_.find(abs(mc.pdgId())) == ids_.end())
return false;
}
if (status_.empty())
return true;
return status_.find(mc.status()) != status_.end();
}
private:
bool checkCharge_;
std::set<int> ids_;
std::set<int> status_;
};
} // namespace reco
#endif
|