#include <TrackWithVertexSelector.h>

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

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

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

bool	operator() (const reco::Track &t, const edm::Event &iEvent)

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

bool	operator() (const reco::TrackRef &t, const edm::Event &iEvent)

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

bool	testTrack (const reco::TrackRef &t) const

bool	testVertices (const reco::Track &t, const reco::VertexCollection &vtxs) const

bool	testVertices (const reco::TrackRef &t, const reco::VertexCollection &vtxs) const

	TrackWithVertexSelector (const edm::ParameterSet &iConfig, edm::ConsumesCollector &&iC)

	TrackWithVertexSelector (const edm::ParameterSet &iConfig, edm::ConsumesCollector &iC)

	~TrackWithVertexSelector ()

Private Types
typedef math::XYZPoint	Point

Private Attributes
double	d0Max_

double	dzMax_

double	etaMax_

double	etaMin_

double	normalizedChi2_

double	nSigmaDtVertex_

uint32_t	numberOfLostHits_

uint32_t	numberOfValidHits_

uint32_t	numberOfValidPixelHits_

uint32_t	nVertices_

double	ptErrorCut_

double	ptMax_

double	ptMin_

std::string	quality_

double	rhoVtx_

const edm::ValueMap< float > *	timeresoscoll_ = nullptr

edm::EDGetTokenT< edm::ValueMap< float > >	timeResosToken_

const edm::ValueMap< float > *	timescoll_ = nullptr

edm::EDGetTokenT< edm::ValueMap< float > >	timesToken_

const reco::VertexCollection *	vcoll_ = nullptr

edm::EDGetTokenT< reco::VertexCollection >	vertexToken_

bool	vtxFallback_

double	zetaVtx_

Detailed Description

Definition at line 17 of file TrackWithVertexSelector.h.

Member Typedef Documentation

◆ Point

typedef math::XYZPoint TrackWithVertexSelector::Point

private

Definition at line 63 of file TrackWithVertexSelector.h.

Constructor & Destructor Documentation

◆ TrackWithVertexSelector() [1/2]

TrackWithVertexSelector::TrackWithVertexSelector	(	const edm::ParameterSet &	iConfig,
		edm::ConsumesCollector &&	iC
	)

inlineexplicit

Definition at line 19 of file TrackWithVertexSelector.h.

20 : TrackWithVertexSelector(iConfig, iC) {}

◆ TrackWithVertexSelector() [2/2]

TrackWithVertexSelector::TrackWithVertexSelector	(	const edm::ParameterSet &	iConfig,
		edm::ConsumesCollector &	iC
	)

explicit

Definition at line 11 of file TrackWithVertexSelector.cc.

     : numberOfValidHits_(iConfig.getParameter<uint32_t>("numberOfValidHits")),
       numberOfValidPixelHits_(iConfig.getParameter<uint32_t>("numberOfValidPixelHits")),
       numberOfLostHits_(iConfig.getParameter<uint32_t>("numberOfLostHits")),
       normalizedChi2_(iConfig.getParameter<double>("normalizedChi2")),
       ptMin_(iConfig.getParameter<double>("ptMin")),
       ptMax_(iConfig.getParameter<double>("ptMax")),
       etaMin_(iConfig.getParameter<double>("etaMin")),
       etaMax_(iConfig.getParameter<double>("etaMax")),
       dzMax_(iConfig.getParameter<double>("dzMax")),
       d0Max_(iConfig.getParameter<double>("d0Max")),
       ptErrorCut_(iConfig.getParameter<double>("ptErrorCut")),
       quality_(iConfig.getParameter<std::string>("quality")),
       nVertices_(iConfig.getParameter<bool>("useVtx") ? iConfig.getParameter<uint32_t>("nVertices") : 0),
       vertexToken_(iC.consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("vertexTag"))),
       timesToken_(iC.consumes<edm::ValueMap<float> >(iConfig.getParameter<edm::InputTag>("timesTag"))),
       timeResosToken_(iC.consumes<edm::ValueMap<float> >(iConfig.getParameter<edm::InputTag>("timeResosTag"))),
       vtxFallback_(iConfig.getParameter<bool>("vtxFallback")),
       zetaVtx_(iConfig.getParameter<double>("zetaVtx")),
       rhoVtx_(iConfig.getParameter<double>("rhoVtx")),
       nSigmaDtVertex_(iConfig.getParameter<double>("nSigmaDtVertex")) {}

◆ ~TrackWithVertexSelector()

TrackWithVertexSelector::~TrackWithVertexSelector ( )

Definition at line 33 of file TrackWithVertexSelector.cc.

33 {}

Member Function Documentation

◆ init() [1/2]

void TrackWithVertexSelector::init ( const edm::Event & event )

Definition at line 35 of file TrackWithVertexSelector.cc.

                                                         {
   edm::Handle<reco::VertexCollection> hVtx;
   event.getByToken(vertexToken_, hVtx);
   vcoll_ = hVtx.product();
  
   edm::Handle<edm::ValueMap<float> > hTimes;
   event.getByToken(timesToken_, hTimes);
   timescoll_ = hTimes.isValid() ? hTimes.product() : nullptr;
  
   edm::Handle<edm::ValueMap<float> > hTimeResos;
   event.getByToken(timeResosToken_, hTimeResos);
   timeresoscoll_ = hTimeResos.isValid() ? hTimeResos.product() : nullptr;
 }

References edm::HandleBase::isValid(), edm::Handle< T >::product(), timeresoscoll_, timeResosToken_, timescoll_, timesToken_, vcoll_, and vertexToken_.

◆ init() [2/2]

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

inline

Definition at line 24 of file TrackWithVertexSelector.h.

24 { init(event); }

References init().

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

◆ operator()() [1/4]

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

Definition at line 116 of file TrackWithVertexSelector.cc.

                                                                  {
   if (!testTrack(t))
     return false;
   if (nVertices_ == 0)
     return true;
   return testVertices(t, *vcoll_);
 }

References nVertices_, OrderedSet::t, testTrack(), testVertices(), and vcoll_.

◆ operator()() [2/4]

bool TrackWithVertexSelector::operator()	(	const reco::Track &	t,
		const edm::Event &	iEvent
	)

inline

Definition at line 28 of file TrackWithVertexSelector.h.

                                                               {
     init(iEvent);
     return (*this)(t);
   }

References iEvent, init(), and OrderedSet::t.

◆ operator()() [3/4]

bool TrackWithVertexSelector::operator() ( const reco::TrackRef & t ) const

Definition at line 124 of file TrackWithVertexSelector.cc.

                                                                        {
   if (!testTrack(tref))
     return false;
   if (nVertices_ == 0)
     return true;
   return testVertices(tref, *vcoll_);
 }

References nVertices_, testTrack(), testVertices(), and vcoll_.

◆ operator()() [4/4]

bool TrackWithVertexSelector::operator()	(	const reco::TrackRef &	t,
		const edm::Event &	iEvent
	)

inline

Definition at line 34 of file TrackWithVertexSelector.h.

                                                                  {
     init(iEvent);
     return (*this)(t);
   }

References iEvent, init(), and OrderedSet::t.

◆ testTrack() [1/2]

bool TrackWithVertexSelector::testTrack ( const reco::Track & t ) const

Definition at line 49 of file TrackWithVertexSelector.cc.

                                                                 {
   using std::abs;
   if ((t.numberOfValidHits() >= numberOfValidHits_) &&
       (static_cast<unsigned int>(t.hitPattern().numberOfValidPixelHits()) >= numberOfValidPixelHits_) &&
       (t.numberOfLostHits() <= numberOfLostHits_) && (t.normalizedChi2() <= normalizedChi2_) &&
       (t.ptError() / t.pt() * std::max(1., t.normalizedChi2()) <= ptErrorCut_) &&
       (t.quality(t.qualityByName(quality_))) && (t.pt() >= ptMin_) && (t.pt() <= ptMax_) && (abs(t.eta()) <= etaMax_) &&
       (abs(t.eta()) >= etaMin_) && (abs(t.dz()) <= dzMax_) && (abs(t.d0()) <= d0Max_)) {
     return true;
   }
   return false;
 }

References funct::abs(), d0Max_, dzMax_, etaMax_, etaMin_, SiStripPI::max, normalizedChi2_, numberOfLostHits_, numberOfValidHits_, numberOfValidPixelHits_, ptErrorCut_, ptMax_, ptMin_, quality_, and OrderedSet::t.

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

◆ testTrack() [2/2]

bool TrackWithVertexSelector::testTrack ( const reco::TrackRef & t ) const

Definition at line 62 of file TrackWithVertexSelector.cc.

62 { return testTrack(*tref); }

References testTrack().

◆ testVertices() [1/2]

bool TrackWithVertexSelector::testVertices	(	const reco::Track &	t,
		const reco::VertexCollection &	vtxs
	)		const

Definition at line 64 of file TrackWithVertexSelector.cc.

                                                                                                      {
   bool ok = false;
   if (!vtxs.empty()) {
     unsigned int tested = 1;
     for (reco::VertexCollection::const_iterator it = vtxs.begin(), ed = vtxs.end(); it != ed; ++it) {
       if ((std::abs(t.dxy(it->position())) < rhoVtx_) && (std::abs(t.dz(it->position())) < zetaVtx_)) {
         ok = true;
         break;
       }
       if (tested++ >= nVertices_)
         break;
     }
   } else if (vtxFallback_) {
     return ((std::abs(t.vertex().z()) < 15.9) && (t.vertex().Rho() < 0.2));
   }
   return ok;
 }

References funct::abs(), nVertices_, convertSQLiteXML::ok, rhoVtx_, OrderedSet::t, vtxFallback_, and zetaVtx_.

Referenced by operator()().

◆ testVertices() [2/2]

bool TrackWithVertexSelector::testVertices	(	const reco::TrackRef &	t,
		const reco::VertexCollection &	vtxs
	)		const

Definition at line 82 of file TrackWithVertexSelector.cc.

                                                                                                            {
   const auto &t = *tref;
   const bool timeAvailable = timescoll_ != nullptr && timeresoscoll_ != nullptr;
   bool ok = false;
   if (!vtxs.empty()) {
     unsigned int tested = 1;
     for (reco::VertexCollection::const_iterator it = vtxs.begin(), ed = vtxs.end(); it != ed; ++it) {
       const bool useTime = timeAvailable && it->t() != 0.;
       float time = useTime ? (*timescoll_)[tref] : -1.f;
       float timeReso = useTime ? (*timeresoscoll_)[tref] : -1.f;
       timeReso = (timeReso > 1e-6 ? timeReso : fakeBeamSpotTimeWidth);
  
       if (edm::isNotFinite(time)) {
         time = 0.0;
         timeReso = 2.0 * fakeBeamSpotTimeWidth;
       }
  
       const double vtxSigmaT2 = it->tError() * it->tError();
       const double vtxTrackErr = std::sqrt(vtxSigmaT2 + timeReso * timeReso);
  
       if ((std::abs(t.dxy(it->position())) < rhoVtx_) && (std::abs(t.dz(it->position())) < zetaVtx_) &&
           (!useTime || (std::abs(time - it->t()) / vtxTrackErr < nSigmaDtVertex_))) {
         ok = true;
         break;
       }
       if (tested++ >= nVertices_)
         break;
     }
   } else if (vtxFallback_) {
     return ((std::abs(t.vertex().z()) < 15.9) && (t.vertex().Rho() < 0.2));
   }
   return ok;
 }

References funct::abs(), MillePedeFileConverter_cfg::e, edm::isNotFinite(), nSigmaDtVertex_, nVertices_, convertSQLiteXML::ok, rhoVtx_, mathSSE::sqrt(), OrderedSet::t, ntuplemaker::time, timeresoscoll_, timescoll_, vtxFallback_, and zetaVtx_.

Member Data Documentation

◆ d0Max_

double TrackWithVertexSelector::d0Max_

private

Definition at line 50 of file TrackWithVertexSelector.h.

Referenced by testTrack().

◆ dzMax_

double TrackWithVertexSelector::dzMax_

private

Definition at line 50 of file TrackWithVertexSelector.h.

Referenced by testTrack().

◆ etaMax_

double TrackWithVertexSelector::etaMax_

private

Definition at line 49 of file TrackWithVertexSelector.h.

Referenced by testTrack().

◆ etaMin_

double TrackWithVertexSelector::etaMin_

private

Definition at line 49 of file TrackWithVertexSelector.h.

Referenced by testTrack().

◆ normalizedChi2_

double TrackWithVertexSelector::normalizedChi2_

private

Definition at line 48 of file TrackWithVertexSelector.h.

Referenced by testTrack().

◆ nSigmaDtVertex_

double TrackWithVertexSelector::nSigmaDtVertex_

private

Definition at line 58 of file TrackWithVertexSelector.h.

Referenced by testVertices().

◆ numberOfLostHits_

uint32_t TrackWithVertexSelector::numberOfLostHits_

private

Definition at line 47 of file TrackWithVertexSelector.h.

Referenced by testTrack().

◆ numberOfValidHits_

uint32_t TrackWithVertexSelector::numberOfValidHits_

private

Definition at line 45 of file TrackWithVertexSelector.h.

Referenced by testTrack().

◆ numberOfValidPixelHits_

uint32_t TrackWithVertexSelector::numberOfValidPixelHits_

private

Definition at line 46 of file TrackWithVertexSelector.h.

Referenced by testTrack().

◆ nVertices_

uint32_t TrackWithVertexSelector::nVertices_

private

Definition at line 54 of file TrackWithVertexSelector.h.

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

◆ ptErrorCut_

double TrackWithVertexSelector::ptErrorCut_

private

Definition at line 51 of file TrackWithVertexSelector.h.

Referenced by testTrack().

◆ ptMax_

double TrackWithVertexSelector::ptMax_

private

Definition at line 49 of file TrackWithVertexSelector.h.

Referenced by testTrack().

◆ ptMin_

double TrackWithVertexSelector::ptMin_

private

Definition at line 49 of file TrackWithVertexSelector.h.

Referenced by testTrack().

◆ quality_

std::string TrackWithVertexSelector::quality_

private

Definition at line 52 of file TrackWithVertexSelector.h.

Referenced by testTrack().

◆ rhoVtx_

double TrackWithVertexSelector::rhoVtx_

private

Definition at line 58 of file TrackWithVertexSelector.h.

Referenced by testVertices().

◆ timeresoscoll_

const edm::ValueMap<float>* TrackWithVertexSelector::timeresoscoll_ = nullptr

private

Definition at line 62 of file TrackWithVertexSelector.h.

Referenced by init(), and testVertices().

◆ timeResosToken_

edm::EDGetTokenT<edm::ValueMap<float> > TrackWithVertexSelector::timeResosToken_

private

Definition at line 56 of file TrackWithVertexSelector.h.

Referenced by init().

◆ timescoll_

const edm::ValueMap<float>* TrackWithVertexSelector::timescoll_ = nullptr

private

Definition at line 61 of file TrackWithVertexSelector.h.

Referenced by init(), and testVertices().

◆ timesToken_

edm::EDGetTokenT<edm::ValueMap<float> > TrackWithVertexSelector::timesToken_

private

Definition at line 56 of file TrackWithVertexSelector.h.

Referenced by init().

◆ vcoll_

const reco::VertexCollection* TrackWithVertexSelector::vcoll_ = nullptr

private

Definition at line 60 of file TrackWithVertexSelector.h.

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

◆ vertexToken_

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

private

Definition at line 55 of file TrackWithVertexSelector.h.

Referenced by init().

◆ vtxFallback_

bool TrackWithVertexSelector::vtxFallback_

private

Definition at line 57 of file TrackWithVertexSelector.h.

Referenced by testVertices().

◆ zetaVtx_

double TrackWithVertexSelector::zetaVtx_

private

Definition at line 58 of file TrackWithVertexSelector.h.

Referenced by testVertices().

Public Member Functions

Private Types

Private Attributes

Detailed Description

Member Typedef Documentation

◆ Point

Constructor & Destructor Documentation

◆ TrackWithVertexSelector() [1/2]

◆ TrackWithVertexSelector() [2/2]

◆ ~TrackWithVertexSelector()

Member Function Documentation

◆ init() [1/2]

◆ init() [2/2]

◆ operator()() [1/4]

◆ operator()() [2/4]

◆ operator()() [3/4]

◆ operator()() [4/4]

◆ testTrack() [1/2]

◆ testTrack() [2/2]

◆ testVertices() [1/2]

◆ testVertices() [2/2]

Member Data Documentation

◆ d0Max_

◆ dzMax_

◆ etaMax_

◆ etaMin_

◆ normalizedChi2_

◆ nSigmaDtVertex_

◆ numberOfLostHits_

◆ numberOfValidHits_

◆ numberOfValidPixelHits_

◆ nVertices_

◆ ptErrorCut_

◆ ptMax_

◆ ptMin_

◆ quality_

◆ rhoVtx_

◆ timeresoscoll_

◆ timeResosToken_

◆ timescoll_

◆ timesToken_

◆ vcoll_

◆ vertexToken_

◆ vtxFallback_

◆ zetaVtx_