Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2025-02-05 23:51:09

0001 #ifndef CommonTools_ParticleFlow_MuonIDPFCandidateSelectorDefinition
0002 #define CommonTools_ParticleFlow_MuonIDPFCandidateSelectorDefinition
0003 
0004 /**
0005    \class    pf2pat::MuonIDPFCandidateSelectorDefinition MuonIDPFCandidateSelectorDefinition.h "CommonTools/ParticleFlow/interface/MuonIDPFCandidateSelectorDefinition.h"
0006    \brief    Selects PFCandidates basing on cuts provided with string cut parser
0007 
0008    \author   Giovanni Petrucciani
0009    \version  $Id: MuonIDPFCandidateSelectorDefinition.h,v 1.1 2011/01/28 20:56:44 srappocc Exp $
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         // skip ones without a ref to a reco::Muon: they won't be matched anyway
0039         if (muR.isNull())
0040           continue;
0041 
0042         // convert into a pat::Muon, so that the 'muonID' method is available
0043         pat::Muon patMu(*muR);
0044 
0045         // apply muon id
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 }  // namespace pf2pat
0058 
0059 #endif