|
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
|
#ifndef CommonTools_ParticleFlow_MuonIDPFCandidateSelectorDefinition
#define CommonTools_ParticleFlow_MuonIDPFCandidateSelectorDefinition
/**
\class pf2pat::MuonIDPFCandidateSelectorDefinition MuonIDPFCandidateSelectorDefinition.h "CommonTools/ParticleFlow/interface/MuonIDPFCandidateSelectorDefinition.h"
\brief Selects PFCandidates basing on cuts provided with string cut parser
\author Giovanni Petrucciani
\version $Id: MuonIDPFCandidateSelectorDefinition.h,v 1.1 2011/01/28 20:56:44 srappocc Exp $
*/
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/Common/interface/ValueMap.h"
#include "DataFormats/PatCandidates/interface/Muon.h"
#include "CommonTools/Utils/interface/StringCutObjectSelector.h"
#include "CommonTools/ParticleFlow/interface/PFCandidateSelectorDefinition.h"
namespace pf2pat {
struct MuonIDPFCandidateSelectorDefinition : public PFCandidateSelectorDefinition {
MuonIDPFCandidateSelectorDefinition(const edm::ParameterSet& cfg, edm::ConsumesCollector&& iC)
: muonCut_(cfg.getParameter<std::string>("cut")) {}
static void fillPSetDescription(edm::ParameterSetDescription& desc) { desc.add<std::string>("cut", ""); }
void select(const HandleToCollection& hc, const edm::Event& e, const edm::EventSetup& s) {
selected_.clear();
unsigned key = 0;
for (collection::const_iterator pfc = hc->begin(); pfc != hc->end(); ++pfc, ++key) {
reco::MuonRef muR = pfc->muonRef();
// skip ones without a ref to a reco::Muon: they won't be matched anyway
if (muR.isNull())
continue;
// convert into a pat::Muon, so that the 'muonID' method is available
pat::Muon patMu(*muR);
// apply muon id
if (muonCut_(patMu)) {
selected_.push_back(reco::PFCandidate(*pfc));
reco::PFCandidatePtr ptrToMother(hc, key);
selected_.back().setSourceCandidatePtr(ptrToMother);
}
}
}
private:
StringCutObjectSelector<pat::Muon> muonCut_;
};
} // namespace pf2pat
#endif
|