#include <RecoTrackSelectorBase.h>

Inheritance diagram for RecoTrackSelectorBase:

Public Member Functions
void	init (const edm::Event &event, const edm::EventSetup &es)

bool	operator() (const reco::Track &t) const

bool	operator() (const reco::Track &t, const reco::Track::Point &vertex) const

bool	operator() (const reco::TrackRef &tref) const

	RecoTrackSelectorBase ()

	RecoTrackSelectorBase (const edm::ParameterSet &cfg)

	RecoTrackSelectorBase (const edm::ParameterSet &cfg, edm::ConsumesCollector &iC)

Private Attributes
std::vector< reco::TrackBase::TrackAlgorithm >	algorithm_

std::vector< reco::TrackBase::TrackAlgorithm >	algorithmMask_

edm::EDGetTokenT< reco::BeamSpot >	bsSrcToken_

bool	invertRapidityCut_

double	lip_

double	maxChi2_

double	maxRapidity_

double	meanPhi_

int	min3DLayer_

int	minHit_

int	minLayer_

int	minPixelHit_

double	minRapidity_

std::vector< reco::TrackBase::TrackAlgorithm >	originalAlgorithm_

double	ptMin_

std::vector< reco::TrackBase::TrackQuality >	quality_

double	rangePhi_

double	tip_

bool	usePV_

reco::Track::Point	vertex_

edm::EDGetTokenT< reco::VertexCollection >	vertexToken_

Detailed Description

Definition at line 16 of file RecoTrackSelectorBase.h.

Constructor & Destructor Documentation

◆ RecoTrackSelectorBase() [1/3]

RecoTrackSelectorBase::RecoTrackSelectorBase ( )

inline

Definition at line 18 of file RecoTrackSelectorBase.h.

18 {}

◆ RecoTrackSelectorBase() [2/3]

RecoTrackSelectorBase::RecoTrackSelectorBase ( const edm::ParameterSet & cfg )

inline

Definition at line 19 of file RecoTrackSelectorBase.h.

       : ptMin_(cfg.getParameter<double>("ptMin")),
         minRapidity_(cfg.getParameter<double>("minRapidity")),
         maxRapidity_(cfg.getParameter<double>("maxRapidity")),
         meanPhi_((cfg.getParameter<double>("minPhi") + cfg.getParameter<double>("maxPhi")) / 2.),
         rangePhi_((cfg.getParameter<double>("maxPhi") - cfg.getParameter<double>("minPhi")) / 2.),
         tip_(cfg.getParameter<double>("tip")),
         lip_(cfg.getParameter<double>("lip")),
         maxChi2_(cfg.getParameter<double>("maxChi2")),
         minHit_(cfg.getParameter<int>("minHit")),
         minPixelHit_(cfg.getParameter<int>("minPixelHit")),
         minLayer_(cfg.getParameter<int>("minLayer")),
         min3DLayer_(cfg.getParameter<int>("min3DLayer")),
         usePV_(false),
         invertRapidityCut_(cfg.getParameter<bool>("invertRapidityCut")) {
     const auto minPhi = cfg.getParameter<double>("minPhi");
     const auto maxPhi = cfg.getParameter<double>("maxPhi");
     if (minPhi >= maxPhi) {
       throw cms::Exception("Configuration")
           << "RecoTrackSelectorPhase: minPhi (" << minPhi << ") must be smaller than maxPhi (" << maxPhi
           << "). The range is constructed from minPhi to maxPhi around their average.";
     }
     if (minPhi >= M_PI) {
       throw cms::Exception("Configuration")
           << "RecoTrackSelectorPhase: minPhi (" << minPhi
           << ") must be smaller than PI. The range is constructed from minPhi to maxPhi around their average.";
     }
     if (maxPhi <= -M_PI) {
       throw cms::Exception("Configuration")
           << "RecoTrackSelectorPhase: maxPhi (" << maxPhi
           << ") must be larger than -PI. The range is constructed from minPhi to maxPhi around their average.";
     }
  
     for (const std::string& quality : cfg.getParameter<std::vector<std::string> >("quality"))
       quality_.push_back(reco::TrackBase::qualityByName(quality));
     for (const std::string& algorithm : cfg.getParameter<std::vector<std::string> >("algorithm"))
       algorithm_.push_back(reco::TrackBase::algoByName(algorithm));
     for (const std::string& algorithm : cfg.getParameter<std::vector<std::string> >("originalAlgorithm"))
       originalAlgorithm_.push_back(reco::TrackBase::algoByName(algorithm));
     for (const std::string& algorithm : cfg.getParameter<std::vector<std::string> >("algorithmMaskContains"))
       algorithmMask_.push_back(reco::TrackBase::algoByName(algorithm));
   }

References reco::TrackBase::algoByName(), qcdUeDQM_cfi::algorithm, algorithm_, algorithmMask_, looper::cfg, Exception, M_PI, HLT_2018_cff::maxPhi, HLT_2018_cff::minPhi, originalAlgorithm_, qcdUeDQM_cfi::quality, quality_, reco::TrackBase::qualityByName(), and AlCaHLTBitMon_QueryRunRegistry::string.

◆ RecoTrackSelectorBase() [3/3]

RecoTrackSelectorBase::RecoTrackSelectorBase	(	const edm::ParameterSet &	cfg,
		edm::ConsumesCollector &	iC
	)

inline

Definition at line 62 of file RecoTrackSelectorBase.h.

                                                                               : RecoTrackSelectorBase(cfg) {
     usePV_ = cfg.getParameter<bool>("usePV");
     bsSrcToken_ = iC.consumes<reco::BeamSpot>(cfg.getParameter<edm::InputTag>("beamSpot"));
     if (usePV_)
       vertexToken_ = iC.consumes<reco::VertexCollection>(cfg.getParameter<edm::InputTag>("vertexTag"));
   }

References bsSrcToken_, looper::cfg, edm::ConsumesCollector::consumes(), usePV_, and vertexToken_.

Member Function Documentation

◆ init()

void RecoTrackSelectorBase::init	(	const edm::Event &	event,
		const edm::EventSetup &	es
	)

inline

Definition at line 69 of file RecoTrackSelectorBase.h.

                                                             {
     edm::Handle<reco::BeamSpot> beamSpot;
     event.getByToken(bsSrcToken_, beamSpot);
     vertex_ = beamSpot->position();
     if (!usePV_)
       return;
     edm::Handle<reco::VertexCollection> hVtx;
     event.getByToken(vertexToken_, hVtx);
     if (hVtx->empty())
       return;
     vertex_ = (*hVtx)[0].position();
   }

References pwdgSkimBPark_cfi::beamSpot, bsSrcToken_, usePV_, vertex_, and vertexToken_.

Referenced by RecoTrackViewRefSelector::select(), RecoTrackRefSelector::select(), and RecoTrackSelector::select().

◆ operator()() [1/3]

bool RecoTrackSelectorBase::operator() ( const reco::Track & t ) const

inline

Definition at line 84 of file RecoTrackSelectorBase.h.

84 { return (*this)(t, vertex_); }

References OrderedSet::t, and vertex_.

◆ operator()() [2/3]

bool RecoTrackSelectorBase::operator()	(	const reco::Track &	t,
		const reco::Track::Point &	vertex
	)		const

inline

Definition at line 86 of file RecoTrackSelectorBase.h.

                                                                           {
     bool quality_ok = true;
     if (!quality_.empty()) {
       quality_ok = false;
       for (unsigned int i = 0; i < quality_.size(); ++i) {
         if (t.quality(quality_[i])) {
           quality_ok = true;
           break;
         }
       }
     }
  
     bool algo_ok = true;
     if (!algorithm_.empty()) {
       if (std::find(algorithm_.begin(), algorithm_.end(), t.algo()) == algorithm_.end())
         algo_ok = false;
     }
     if (!originalAlgorithm_.empty() && algo_ok) {
       if (std::find(originalAlgorithm_.begin(), originalAlgorithm_.end(), t.originalAlgo()) == originalAlgorithm_.end())
         algo_ok = false;
     }
     if (!algorithmMask_.empty() && algo_ok) {
       if (std::find_if(algorithmMask_.begin(),
                        algorithmMask_.end(),
                        // for some reason I have to either explicitly give the return type, or use static_cast<bool>()
                        [&](reco::TrackBase::TrackAlgorithm algo) -> bool { return t.algoMask()[algo]; }) ==
           algorithmMask_.end())
         algo_ok = false;
     }
  
     const auto dphi = deltaPhi(t.phi(), meanPhi_);
  
     auto etaOk = [&](const reco::Track& p) -> bool {
       float eta = p.eta();
       if (!invertRapidityCut_)
         return (eta >= minRapidity_) && (eta <= maxRapidity_);
       else
         return (eta < minRapidity_ || eta > maxRapidity_);
     };
  
     return ((algo_ok & quality_ok) && t.hitPattern().numberOfValidHits() >= minHit_ &&
             t.hitPattern().numberOfValidPixelHits() >= minPixelHit_ &&
             t.hitPattern().trackerLayersWithMeasurement() >= minLayer_ &&
             t.hitPattern().pixelLayersWithMeasurement() + t.hitPattern().numberOfValidStripLayersWithMonoAndStereo() >=
                 min3DLayer_ &&
             fabs(t.pt()) >= ptMin_ && etaOk(t) && dphi >= -rangePhi_ && dphi <= rangePhi_ &&
             fabs(t.dxy(vertex)) <= tip_ && fabs(t.dsz(vertex)) <= lip_ && t.normalizedChi2() <= maxChi2_);
   }

References algorithm_, algorithmMask_, SiPixelRawToDigiRegional_cfi::deltaPhi, PVValHelper::eta, spr::find(), mps_fire::i, invertRapidityCut_, lip_, maxChi2_, maxRapidity_, meanPhi_, min3DLayer_, minHit_, minLayer_, minPixelHit_, minRapidity_, originalAlgorithm_, AlCaHLTBitMon_ParallelJobs::p, ptMin_, quality_, rangePhi_, OrderedSet::t, tip_, and bphysicsOniaDQM_cfi::vertex.

◆ operator()() [3/3]

bool RecoTrackSelectorBase::operator() ( const reco::TrackRef & tref ) const

inline

Definition at line 82 of file RecoTrackSelectorBase.h.

82 { return (*this)(*tref); }

Member Data Documentation

◆ algorithm_

std::vector<reco::TrackBase::TrackAlgorithm> RecoTrackSelectorBase::algorithm_

private

Definition at line 155 of file RecoTrackSelectorBase.h.

Referenced by operator()(), and RecoTrackSelectorBase().

◆ algorithmMask_

std::vector<reco::TrackBase::TrackAlgorithm> RecoTrackSelectorBase::algorithmMask_

private

Definition at line 157 of file RecoTrackSelectorBase.h.

Referenced by operator()(), and RecoTrackSelectorBase().

◆ bsSrcToken_

edm::EDGetTokenT<reco::BeamSpot> RecoTrackSelectorBase::bsSrcToken_

private

Definition at line 151 of file RecoTrackSelectorBase.h.

Referenced by init(), and RecoTrackSelectorBase().

◆ invertRapidityCut_

bool RecoTrackSelectorBase::invertRapidityCut_

private

Definition at line 149 of file RecoTrackSelectorBase.h.

Referenced by operator()().

◆ lip_

double RecoTrackSelectorBase::lip_

private

Definition at line 142 of file RecoTrackSelectorBase.h.

Referenced by operator()().

◆ maxChi2_

double RecoTrackSelectorBase::maxChi2_

private

Definition at line 143 of file RecoTrackSelectorBase.h.

Referenced by operator()().

◆ maxRapidity_

double RecoTrackSelectorBase::maxRapidity_

private

Definition at line 138 of file RecoTrackSelectorBase.h.

Referenced by operator()().

◆ meanPhi_

double RecoTrackSelectorBase::meanPhi_

private

Definition at line 139 of file RecoTrackSelectorBase.h.

Referenced by operator()().

◆ min3DLayer_

int RecoTrackSelectorBase::min3DLayer_

private

Definition at line 147 of file RecoTrackSelectorBase.h.

Referenced by operator()().

◆ minHit_

int RecoTrackSelectorBase::minHit_

private

Definition at line 144 of file RecoTrackSelectorBase.h.

Referenced by operator()().

◆ minLayer_

int RecoTrackSelectorBase::minLayer_

private

Definition at line 146 of file RecoTrackSelectorBase.h.

Referenced by operator()().

◆ minPixelHit_

int RecoTrackSelectorBase::minPixelHit_

private

Definition at line 145 of file RecoTrackSelectorBase.h.

Referenced by operator()().

◆ minRapidity_

double RecoTrackSelectorBase::minRapidity_

private

Definition at line 137 of file RecoTrackSelectorBase.h.

Referenced by operator()().

◆ originalAlgorithm_

std::vector<reco::TrackBase::TrackAlgorithm> RecoTrackSelectorBase::originalAlgorithm_

private

Definition at line 156 of file RecoTrackSelectorBase.h.

Referenced by operator()(), and RecoTrackSelectorBase().

◆ ptMin_

double RecoTrackSelectorBase::ptMin_

private

Definition at line 136 of file RecoTrackSelectorBase.h.

Referenced by operator()().

◆ quality_

std::vector<reco::TrackBase::TrackQuality> RecoTrackSelectorBase::quality_

private

Definition at line 154 of file RecoTrackSelectorBase.h.

Referenced by operator()(), and RecoTrackSelectorBase().

◆ rangePhi_

double RecoTrackSelectorBase::rangePhi_

private

Definition at line 140 of file RecoTrackSelectorBase.h.

Referenced by operator()().

◆ tip_

double RecoTrackSelectorBase::tip_

private

Definition at line 141 of file RecoTrackSelectorBase.h.

Referenced by operator()().

◆ usePV_

bool RecoTrackSelectorBase::usePV_

private

Definition at line 148 of file RecoTrackSelectorBase.h.

Referenced by init(), and RecoTrackSelectorBase().

◆ vertex_

reco::Track::Point RecoTrackSelectorBase::vertex_

private

Definition at line 159 of file RecoTrackSelectorBase.h.

Referenced by init(), and operator()().

◆ vertexToken_

edm::EDGetTokenT<reco::VertexCollection> RecoTrackSelectorBase::vertexToken_

private

Definition at line 152 of file RecoTrackSelectorBase.h.

Referenced by init(), and RecoTrackSelectorBase().

Public Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ RecoTrackSelectorBase() [1/3]

◆ RecoTrackSelectorBase() [2/3]

◆ RecoTrackSelectorBase() [3/3]

Member Function Documentation

◆ init()

◆ operator()() [1/3]

◆ operator()() [2/3]

◆ operator()() [3/3]

Member Data Documentation

◆ algorithm_

◆ algorithmMask_

◆ bsSrcToken_

◆ invertRapidityCut_

◆ lip_

◆ maxChi2_

◆ maxRapidity_

◆ meanPhi_

◆ min3DLayer_

◆ minHit_

◆ minLayer_

◆ minPixelHit_

◆ minRapidity_

◆ originalAlgorithm_

◆ ptMin_

◆ quality_

◆ rangePhi_

◆ tip_

◆ usePV_

◆ vertex_

◆ vertexToken_