PVObjectSelector.h

Go to the documentation of this file.

#ifndef Analysis_AnalysisFilters_interface_PVObjectSelector_h
#define Analysis_AnalysisFilters_interface_PVObjectSelector_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/Selector.h"
#include "DataFormats/VertexReco/interface/Vertex.h"

#include <vector>
#include <string>

// make a selector for this selection
class PVObjectSelector : public Selector<reco::Vertex> {
public:
  PVObjectSelector() {}

#ifndef __GCCXML__
  PVObjectSelector(edm::ParameterSet const& params, edm::ConsumesCollector&& iC) : PVObjectSelector(params) {}
#endif

  PVObjectSelector(edm::ParameterSet const& params) {
    push_back("PV NDOF", params.getParameter<double>("minNdof"));
    push_back("PV Z", params.getParameter<double>("maxZ"));
    push_back("PV RHO", params.getParameter<double>("maxRho"));
    set("PV NDOF");
    set("PV Z");
    set("PV RHO");

    indexNDOF_ = index_type(&bits_, "PV NDOF");
    indexZ_ = index_type(&bits_, "PV Z");
    indexRho_ = index_type(&bits_, "PV RHO");

    if (params.exists("cutsToIgnore"))
      setIgnoredCuts(params.getParameter<std::vector<std::string> >("cutsToIgnore"));

    retInternal_ = getBitTemplate();
  }

  bool operator()(reco::Vertex const& pv, pat::strbitset& ret) override {
    if (pv.isFake())
      return false;

    if (pv.ndof() >= cut(indexNDOF_, double()) || ignoreCut(indexNDOF_)) {
      passCut(ret, indexNDOF_);
      if (fabs(pv.z()) <= cut(indexZ_, double()) || ignoreCut(indexZ_)) {
        passCut(ret, indexZ_);
        if (fabs(pv.position().Rho()) <= cut(indexRho_, double()) || ignoreCut(indexRho_)) {
          passCut(ret, indexRho_);
        }
      }
    }

    setIgnored(ret);

    return (bool)ret;
  }

  using Selector<reco::Vertex>::operator();

private:
  index_type indexNDOF_;
  index_type indexZ_;
  index_type indexRho_;
};

#endif