PhysicsTools/SelectorUtils/interface/RunLumiSelector.h - master - githubmirror/cmssw

PhysicsTools/SelectorUtils/interface/RunLumiSelector.h
69 строк2 KB
Cms Build
Clang-Format
05 июн 2019, 13:58
05 июн 2019, 13:584bb99cd
#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