ObjectPairFilter.h

Go to the documentation of this file.

#ifndef RecoAlgos_ObjectPairFilter_h
#define RecoAlgos_ObjectPairFilter_h

#include "FWCore/Framework/interface/EDFilter.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "CommonTools/UtilAlgos/interface/ParameterAdapter.h"
#include <vector>
#include <algorithm>

template<typename C, typename S>
class ObjectPairFilter : public edm::EDFilter {
public:
  explicit ObjectPairFilter( const edm::ParameterSet & cfg ) :
    select_( reco::modules::make<S>( cfg ) ),
    srcToken_( consumes<C>( cfg.template getParameter<edm::InputTag>( "src" ) ) ),
    minNumber_( 1 ) {
    std::vector<std::string> ints = cfg.template getParameterNamesForType<unsigned int>();
    const std::string minNumber( "minNumber" );
    bool foundMinNumber = std::find( ints.begin(), ints.end(), minNumber ) != ints.end();
    if ( foundMinNumber )
      minNumber_ = cfg.template getParameter<unsigned int>( minNumber );
  }

private:
  bool filter( edm::Event& evt, const edm::EventSetup&) override {
    edm::Handle<C> source;
    evt.getByToken( srcToken_, source );
    size_t n = 0;
    for( typename C::const_iterator i = source->begin(); i != source->end(); ++ i )
      for( typename C::const_iterator j = i + 1; j != source->end(); ++ j ) {
    if ( select_( * i, * j ) ) n ++;
    if ( n >= minNumber_ ) return true;
      }
    return false;
  }
  S select_;
  edm::EDGetTokenT<C> srcToken_;
  unsigned int minNumber_;
};

#endif