Back to home page

Project CMSSW displayed by LXR

 
 

    


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

0001 #ifndef CommonTools_ParticleFlow_PdgIdPFCandidateSelectorDefinition
0002 #define CommonTools_ParticleFlow_PdgIdPFCandidateSelectorDefinition
0003 
0004 #include "FWCore/Common/interface/EventBase.h"
0005 #include "FWCore/Framework/interface/EventSetup.h"
0006 #include "FWCore/Framework/interface/ConsumesCollector.h"
0007 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0008 #include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
0009 #include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
0010 #include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
0011 #include "CommonTools/ParticleFlow/interface/PFCandidateSelectorDefinition.h"
0012 
0013 namespace pf2pat {
0014 
0015   class PdgIdPFCandidateSelectorDefinition : public PFCandidateSelectorDefinition {
0016   public:
0017     PdgIdPFCandidateSelectorDefinition(const edm::ParameterSet& cfg, edm::ConsumesCollector&& iC)
0018         : pdgIds_(cfg.getParameter<std::vector<int> >("pdgId")) {}
0019 
0020     static void fillPSetDescription(edm::ParameterSetDescription& desc) { desc.add<std::vector<int> >("pdgId", {}); }
0021 
0022     void select(const HandleToCollection& hc, const edm::EventBase& e, const edm::EventSetup& s) {
0023       selected_.clear();
0024 
0025       unsigned key = 0;
0026       for (collection::const_iterator pfc = hc->begin(); pfc != hc->end(); ++pfc, ++key) {
0027         for (unsigned iId = 0; iId < pdgIds_.size(); iId++) {
0028           if (pfc->pdgId() == pdgIds_[iId]) {
0029             selected_.push_back(reco::PFCandidate(*pfc));
0030             reco::PFCandidatePtr ptrToMother(hc, key);
0031             selected_.back().setSourceCandidatePtr(ptrToMother);
0032             break;
0033           }
0034         }
0035       }
0036     }
0037 
0038   private:
0039     std::vector<int> pdgIds_;
0040   };
0041 
0042 }  // namespace pf2pat
0043 
0044 #endif