EDFilterObjectWrapper.h

Go to the documentation of this file.

#ifndef PhysicsTools_UtilAlgos_interface_EDFilterObjectWrapper_h
#define PhysicsTools_UtilAlgos_interface_EDFilterObjectWrapper_h

#include "FWCore/Framework/interface/stream/EDFilter.h"
#include "FWCore/Common/interface/EventBase.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include <boost/shared_ptr.hpp>

namespace edm {

  template<class T, class C>
  class FilterObjectWrapper : public edm::stream::EDFilter<> {

  public:
    typename C::iterator        iterator;
    typename C::const_iterator  const_iterator;

    FilterObjectWrapper(const edm::ParameterSet& cfg) : src_( consumes<C>(cfg.getParameter<edm::InputTag>("src")))
    {
      filter_ = boost::shared_ptr<T>( new T(cfg.getParameter<edm::ParameterSet>("filterParams")) );
      if ( cfg.exists("filter") ) {
    doFilter_ = cfg.getParameter<bool>("filter");
      } else {
    doFilter_ = false;
      }
      produces<C>();
    }
    virtual ~FilterObjectWrapper(){}
    virtual bool filter(edm::Event& event, const edm::EventSetup& eventSetup) override {
      // create a collection of the objects to put into the event
      std::auto_ptr<C> objsToPut( new C() );
      // get the handle to the objects in the event.
      edm::Handle<C> h_c;
      event.getByToken( src_, h_c );
      // loop through and add passing value_types to the output vector
      for ( typename C::const_iterator ibegin = h_c->begin(), iend = h_c->end(), i = ibegin; i != iend; ++i ){
    if ( (*filter_)(*i) ){
      objsToPut->push_back( *i );
    }
      }
      // put objs in the event
      bool pass = objsToPut->size() > 0;
      event.put(objsToPut);
      if ( doFilter_ )
    return pass;
      else
    return true;
    }

  protected:
    edm::EDGetTokenT<C> src_;
    boost::shared_ptr<T> filter_;
    bool doFilter_;
  };

}

#endif