File indexing completed on 2025-02-05 23:51:09
0001 #ifndef CommonTools_ParticleFlow_MuonIDPFCandidateSelectorDefinition
0002 #define CommonTools_ParticleFlow_MuonIDPFCandidateSelectorDefinition
0003
0004
0005
0006
0007
0008
0009
0010
0011
0012 #include "FWCore/Framework/interface/Event.h"
0013 #include "FWCore/Framework/interface/ConsumesCollector.h"
0014 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0015 #include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
0016 #include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
0017 #include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
0018 #include "DataFormats/Common/interface/ValueMap.h"
0019 #include "DataFormats/PatCandidates/interface/Muon.h"
0020 #include "CommonTools/Utils/interface/StringCutObjectSelector.h"
0021 #include "CommonTools/ParticleFlow/interface/PFCandidateSelectorDefinition.h"
0022
0023 namespace pf2pat {
0024
0025 struct MuonIDPFCandidateSelectorDefinition : public PFCandidateSelectorDefinition {
0026 MuonIDPFCandidateSelectorDefinition(const edm::ParameterSet& cfg, edm::ConsumesCollector&& iC)
0027 : muonCut_(cfg.getParameter<std::string>("cut")) {}
0028
0029 static void fillPSetDescription(edm::ParameterSetDescription& desc) { desc.add<std::string>("cut", ""); }
0030
0031 void select(const HandleToCollection& hc, const edm::Event& e, const edm::EventSetup& s) {
0032 selected_.clear();
0033
0034 unsigned key = 0;
0035 for (collection::const_iterator pfc = hc->begin(); pfc != hc->end(); ++pfc, ++key) {
0036 reco::MuonRef muR = pfc->muonRef();
0037
0038
0039 if (muR.isNull())
0040 continue;
0041
0042
0043 pat::Muon patMu(*muR);
0044
0045
0046 if (muonCut_(patMu)) {
0047 selected_.push_back(reco::PFCandidate(*pfc));
0048 reco::PFCandidatePtr ptrToMother(hc, key);
0049 selected_.back().setSourceCandidatePtr(ptrToMother);
0050 }
0051 }
0052 }
0053
0054 private:
0055 StringCutObjectSelector<pat::Muon> muonCut_;
0056 };
0057 }
0058
0059 #endif