ObjectCountEventSelector.h

Go to the documentation of this file.

#ifndef CommonTools_UtilAlgos_ObjectCountEventSelector_h
#define CommonTools_UtilAlgos_ObjectCountEventSelector_h

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "CommonTools/UtilAlgos/interface/ParameterAdapter.h"
#include "CommonTools/UtilAlgos/interface/CollectionFilterTrait.h"
#include "CommonTools/UtilAlgos/interface/EventSelectorBase.h"

template<typename C,
     typename S = AnySelector,
     typename N = MinNumberSelector,
         typename CS = typename helper::CollectionFilterTrait<C, S, N>::type>
class ObjectCountEventSelector : public EventSelectorBase
{
 public:
  explicit ObjectCountEventSelector( const edm::ParameterSet & cfg, edm::ConsumesCollector && iC ) :
    srcToken_( iC.consumes<C>(cfg.template getParameter<edm::InputTag>( "src" ) ) ),
    select_( reco::modules::make<S>( cfg, iC ) ),
    sizeSelect_( reco::modules::make<N>( cfg, iC ) ) {
  }

  static void fillPSetDescription(edm::ParameterSetDescription& desc) {
    desc.add<edm::InputTag>("src", edm::InputTag());
    reco::modules::fillPSetDescription<S>(desc);
    reco::modules::fillPSetDescription<N>(desc);
  }

  bool operator()(edm::Event& evt, const edm::EventSetup&) const override {
    edm::Handle<C> source;
    evt.getByToken( srcToken_, source );
    return CS::filter( * source, select_, sizeSelect_ );
  }

 private:
  edm::EDGetTokenT<C> srcToken_;

  S select_;

  N sizeSelect_;
};

#endif