HLTCountNumberOfObject.h

Go to the documentation of this file.

#ifndef HLTrigger_HLTCountNumberOfObject_H

// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDFilter.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "FWCore/MessageService/interface/MessageLogger.h"

#include "HLTrigger/HLTcore/interface/HLTFilter.h"
#include "DataFormats/HLTReco/interface/TriggerFilterObjectWithRefs.h"
#include "HLTrigger/HLTcore/interface/defaultModuleLabel.h"

template <class OColl>
class HLTCountNumberOfObject : public HLTFilter {
public:
  explicit HLTCountNumberOfObject(const edm::ParameterSet& iConfig)
      : HLTFilter(iConfig),
        src_(iConfig.getParameter<edm::InputTag>("src")),
        minN_(iConfig.getParameter<int>("MinN")),
        maxN_(iConfig.getParameter<int>("MaxN")) {
    srcToken_ = consumes<OColl>(src_);
  }

  ~HLTCountNumberOfObject() override {}

  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
    edm::ParameterSetDescription desc;
    makeHLTFilterDescription(desc);
    desc.add<edm::InputTag>("src", edm::InputTag(""));
    desc.add<int>("MinN", 0);
    desc.add<int>("MaxN", 99999);
    descriptions.add(defaultModuleLabel<HLTCountNumberOfObject<OColl>>(), desc);
  }

private:
  bool hltFilter(edm::Event& iEvent,
                 const edm::EventSetup&,
                 trigger::TriggerFilterObjectWithRefs& filterproduct) const override {
    edm::Handle<OColl> oHandle;
    iEvent.getByToken(srcToken_, oHandle);
    int s = oHandle->size();
    bool answer = true;
    if (minN_ != -1)
      answer = answer && (s >= minN_);
    if (maxN_ != -1)
      answer = answer && (s <= maxN_);
    LogDebug("HLTCountNumberOfObject") << module(iEvent) << " sees: " << s
                                       << " objects. Filtere answer is: " << (answer ? "true" : "false");

    return answer;
  }

  edm::InputTag src_;
  edm::EDGetTokenT<OColl> srcToken_;
  int minN_, maxN_;
};

#endif