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() { }

  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:
  virtual 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