ParticleFlow/interface/GenericPFCandidateSelectorDefinition.h

0001 #ifndef CommonTools_ParticleFlow_GenericPFCandidateSelectorDefinition
0002 #define CommonTools_ParticleFlow_GenericPFCandidateSelectorDefinition
0003
0004 /**
0005    \class    pf2pat::GenericPFCandidateSelectorDefinition GenericPFCandidateSelectorDefinition.h "CommonTools/ParticleFlow/interface/GenericPFCandidateSelectorDefinition.h"
0006    \brief    Selects PFCandidates basing on cuts provided with string cut parser
0007
0008    \author   Giovanni Petrucciani
0009    \version  $Id: GenericPFCandidateSelectorDefinition.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 "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
0016 #include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
0017 #include "CommonTools/ParticleFlow/interface/PFCandidateSelectorDefinition.h"
0018 #include "CommonTools/Utils/interface/StringCutObjectSelector.h"
0019
0020 namespace pf2pat {
0021
0022   struct GenericPFCandidateSelectorDefinition : public PFCandidateSelectorDefinition {
0023     GenericPFCandidateSelectorDefinition(const edm::ParameterSet& cfg, edm::ConsumesCollector&& iC)
0024         : selector_(cfg.getParameter<std::string>("cut")) {}
0025
0026     void select(const HandleToCollection& hc, const edm::Event& e, const edm::EventSetup& s) {
0027       selected_.clear();
0028
0029       unsigned key = 0;
0030       for (collection::const_iterator pfc = hc->begin(); pfc != hc->end(); ++pfc, ++key) {
0031         if (selector_(*pfc)) {
0032           selected_.push_back(reco::PFCandidate(*pfc));
0033           reco::PFCandidatePtr ptrToMother(hc, key);
0034           selected_.back().setSourceCandidatePtr(ptrToMother);
0035         }
0036       }
0037     }
0038
0039   private:
0040     StringCutObjectSelector<reco::PFCandidate> selector_;
0041   };
0042 }  // namespace pf2pat
0043
0044 #endif