Back to home page

Project CMSSW displayed by LXR

 
 

    


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

0001 #ifndef CommonTools_ParticleFlow_GenericPFJetSelectorDefinition
0002 #define CommonTools_ParticleFlow_GenericPFJetSelectorDefinition
0003 
0004 #include "FWCore/Framework/interface/Event.h"
0005 #include "FWCore/Framework/interface/ConsumesCollector.h"
0006 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0007 #include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
0008 #include "DataFormats/JetReco/interface/PFJet.h"
0009 #include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
0010 #include "CommonTools/ParticleFlow/interface/PFJetSelectorDefinition.h"
0011 #include "CommonTools/Utils/interface/StringCutObjectSelector.h"
0012 
0013 namespace pf2pat {
0014 
0015   struct GenericPFJetSelectorDefinition : public PFJetSelectorDefinition {
0016     GenericPFJetSelectorDefinition(const edm::ParameterSet& cfg, edm::ConsumesCollector&& iC)
0017         : selector_(cfg.getParameter<std::string>("cut")) {}
0018 
0019     static void fillPSetDescription(edm::ParameterSetDescription& desc) { desc.add<std::string>("cut", ""); }
0020 
0021     void select(const HandleToCollection& hc, const edm::Event& e, const edm::EventSetup& s) {
0022       selected_.clear();
0023 
0024       unsigned key = 0;
0025       for (collection::const_iterator pfc = hc->begin(); pfc != hc->end(); ++pfc, ++key) {
0026         if (selector_(*pfc)) {
0027           selected_.push_back(reco::PFJet(*pfc));
0028           reco::CandidatePtr ptrToMother(hc, key);
0029           selected_.back().setSourceCandidatePtr(ptrToMother);
0030         }
0031       }
0032     }
0033 
0034   private:
0035     StringCutObjectSelector<reco::PFJet> selector_;
0036   };
0037 }  // namespace pf2pat
0038 
0039 #endif