/data/refman/pasoursint/CMSSW_5_3_3/src/CommonTools/ParticleFlow/interface/GenericPFJetSelectorDefinition.h

Go to the documentation of this file.

#ifndef CommonTools_ParticleFlow_GenericPFJetSelectorDefinition
#define CommonTools_ParticleFlow_GenericPFJetSelectorDefinition


#include "DataFormats/JetReco/interface/PFJet.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "CommonTools/ParticleFlow/interface/PFJetSelectorDefinition.h"
#include "CommonTools/Utils/interface/StringCutObjectSelector.h"

namespace pf2pat {

  struct GenericPFJetSelectorDefinition : public PFJetSelectorDefinition {
    
    GenericPFJetSelectorDefinition ( const edm::ParameterSet & cfg ) :
      selector_( cfg.getParameter< std::string >( "cut" ) ) { }
    
    void select( const HandleToCollection & hc, 
                 const edm::Event & e,
                 const edm::EventSetup& s) {
      selected_.clear();
      
      unsigned key=0;
      for( collection::const_iterator pfc = hc->begin(); 
           pfc != hc->end(); ++pfc, ++key) {
        
        if( selector_(*pfc) ) {
          selected_.push_back( reco::PFJet(*pfc) );
          reco::CandidatePtr ptrToMother( hc, key );
          selected_.back().setSourceCandidatePtr( ptrToMother );
          
        }
      }
    }
    
    private:
    StringCutObjectSelector<reco::PFJet> selector_;
  };
}

#endif