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