StopAfterNEvents.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/EDFilter.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
class StopAfterNEvents : public edm::EDFilter {
public:
  StopAfterNEvents(const edm::ParameterSet&);
  ~StopAfterNEvents() override;
 
private:
  bool filter(edm::Event&, edm::EventSetup const&) override;
  const int nMax_;
  int n_;
  const bool verbose_;
};
 
#include <iostream>
 
using namespace std;
using namespace edm;
 
StopAfterNEvents::StopAfterNEvents(const ParameterSet& pset)
    : nMax_(pset.getParameter<int>("maxEvents")), n_(0), verbose_(pset.getUntrackedParameter<bool>("verbose", false)) {}
 
StopAfterNEvents::~StopAfterNEvents() {}
 
bool StopAfterNEvents::filter(Event&, EventSetup const&) {
  if (n_ < 0)
    return true;
  n_++;
  bool ret = n_ <= nMax_;
  if (verbose_)
    cout << ">>> filtering event" << n_ << "/" << nMax_ << "(" << (ret ? "true" : "false") << ")" << endl;
  return ret;
}
 
#include "FWCore/Framework/interface/MakerMacros.h"
 
DEFINE_FWK_MODULE(StopAfterNEvents);