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