RunLumiSelector.h

Go to the documentation of this file.

#ifndef Analysis_AnalysisFilters_interface_RunLumiSelector_h
#define Analysis_AnalysisFilters_interface_RunLumiSelector_h
 
#ifndef __GCCXML__
#include "FWCore/Framework/interface/ConsumesCollector.h"
#endif
#include "FWCore/Common/interface/EventBase.h"
#include "DataFormats/Common/interface/Handle.h"
 
#include "PhysicsTools/SelectorUtils/interface/EventSelector.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
 
#include <vector>
#include <string>
 
// make a selector for this selection
class RunLumiSelector : public EventSelector {
public:
  RunLumiSelector() {}
 
#ifndef __GCCXML__
  RunLumiSelector(edm::ParameterSet const& params, edm::ConsumesCollector&& iC) : RunLumiSelector(params) {}
#endif
 
  RunLumiSelector(edm::ParameterSet const& params) {
    push_back("RunLumi");
 
    if (params.exists("lumisToProcess")) {
      lumis_ = params.getUntrackedParameter<std::vector<edm::LuminosityBlockRange> >("lumisToProcess");
      set("RunLumi");
    } else {
      lumis_.clear();
      set("RunLumi", false);
    }
 
    retInternal_ = getBitTemplate();
  }
 
  bool operator()(edm::EventBase const& ev, pat::strbitset& ret) override {
    if (!ignoreCut("RunLumi")) {
      bool goodLumi = false;
      for (std::vector<edm::LuminosityBlockRange>::const_iterator lumisBegin = lumis_.begin(),
                                                                  lumisEnd = lumis_.end(),
                                                                  ilumi = lumisBegin;
           ilumi != lumisEnd;
           ++ilumi) {
        if (ev.id().run() >= ilumi->startRun() && ev.id().run() <= ilumi->endRun() &&
            ev.id().luminosityBlock() >= ilumi->startLumi() && ev.id().luminosityBlock() <= ilumi->endLumi()) {
          goodLumi = true;
          break;
        }
      }
      if (goodLumi)
        passCut(ret, "RunLumi");
    } else {
      passCut(ret, "RunLumi");
    }
 
    setIgnored(ret);
    return (bool)ret;
  }
 
  using EventSelector::operator();
 
private:
  std::vector<edm::LuminosityBlockRange> lumis_;
};
 
#endif