TrackExtrapolator Class Reference

#include <RecoTracker/TrackExtrapolator/src/TrackExtrapolator.cc>

Inheritance diagram for TrackExtrapolator:

Public Member Functions
	TrackExtrapolator (const edm::ParameterSet &)
	~TrackExtrapolator () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

Private Member Functions
void	produce (edm::Event &, const edm::EventSetup &) override
bool	propagateTrackToVolume (const reco::Track &fTrack, const MagneticField &fField, const Propagator &fPropagator, const FiducialVolume &volume, reco::TrackBase::Point &resultPos, reco::TrackBase::Vector &resultMom)
	track quality of the tracks we care about More...

Private Attributes
const edm::ESGetToken < DetIdAssociator, DetIdAssociatorRecord >	ecalDetIdAssociatorToken_
const edm::ESGetToken < MagneticField, IdealMagneticFieldRecord >	fieldToken_
	Input tracks. More...
const edm::ESGetToken < Propagator, TrackingComponentsRecord >	propagatorToken_
const reco::TrackBase::TrackQuality	trackQuality_
const edm::EDGetTokenT < reco::TrackCollection >	tracksSrc_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
using	CacheTypes = CacheContexts< T...>
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T...>
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Description: Extrapolates tracks to Calo Face. Migrating this functionality from RecoJets/JetAssociationAlgorithms/JetTracksAssociatorDRCalo.h, which will now essentially be defunct.

Implementation:

Definition at line 58 of file TrackExtrapolator.h.

Constructor & Destructor Documentation

TrackExtrapolator::TrackExtrapolator ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 8 of file TrackExtrapolator.cc.

References Exception, edm::ParameterSet::getParameter(), AlCaHLTBitMon_QueryRunRegistry::string, trackQuality_, and reco::TrackBase::undefQuality.

    : tracksSrc_(consumes(iConfig.getParameter<edm::InputTag>("trackSrc"))),
      fieldToken_(esConsumes()),
      propagatorToken_(esConsumes(edm::ESInputTag("", "SteppingHelixPropagatorAlong"))),
      ecalDetIdAssociatorToken_(esConsumes(edm::ESInputTag("", "EcalDetIdAssociator"))),
      trackQuality_(reco::TrackBase::qualityByName(iConfig.getParameter<std::string>("trackQuality"))) {
  if (trackQuality_ == reco::TrackBase::undefQuality) {  // we have a problem
    throw cms::Exception("InvalidInput") << "Unknown trackQuality value '"
                                         << iConfig.getParameter<std::string>("trackQuality")
                                         << "'. See possible values in 'reco::TrackBase::qualityByName'";
  }

  produces<std::vector<reco::TrackExtrapolation>>();
}

TrackExtrapolator::~TrackExtrapolator ( )

override

Definition at line 23 of file TrackExtrapolator.cc.

{}

Member Function Documentation

void TrackExtrapolator::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

overrideprivate

Definition at line 30 of file TrackExtrapolator.cc.

References ecalDetIdAssociatorToken_, fieldToken_, edm::Event::getByToken(), edm::EventSetup::getData(), goodTracks_cfi::goodTracks, eostools::move(), propagateTrackToVolume(), HLT_FULL_cff::propagator, propagatorToken_, edm::Event::put(), Reconstruction_hiPF_cff::trackQuality, trackQuality_, and tracksSrc_.

                                                                             {
  // get stuff from Event Setup
  auto const& field = iSetup.getData(fieldToken_);
  auto const& propagator = iSetup.getData(propagatorToken_);
  FiducialVolume const& ecalvolume = iSetup.getData(ecalDetIdAssociatorToken_).volume();

  // get stuff from Event
  edm::Handle<reco::TrackCollection> tracks_h;
  iEvent.getByToken(tracksSrc_, tracks_h);

  auto extrapolations = std::make_unique<std::vector<reco::TrackExtrapolation>>();

  // Get list of tracks we want to extrapolate
  std::vector<reco::TrackRef> goodTracks;
  for (reco::TrackCollection::const_iterator trkBegin = tracks_h->begin(), trkEnd = tracks_h->end(), itrk = trkBegin;
       itrk != trkEnd;
       ++itrk) {
    reco::TrackBase::TrackQuality trackQuality = reco::TrackBase::TrackQuality(trackQuality_);

    // Cut on track quality
    if (itrk->quality(trackQuality)) {
      goodTracks.push_back(reco::TrackRef(tracks_h, itrk - trkBegin));
    }
  }
  std::vector<reco::TrackBase::Point> vresultPos(1);
  std::vector<reco::TrackBase::Vector> vresultMom(1);

  // Now loop through the list of tracks and extrapolate them
  for (std::vector<reco::TrackRef>::const_iterator trkBegin = goodTracks.begin(),
                                                   trkEnd = goodTracks.end(),
                                                   itrk = trkBegin;
       itrk != trkEnd;
       ++itrk) {
    if (propagateTrackToVolume(**itrk, field, propagator, ecalvolume, vresultPos[0], vresultMom[0])) {
      extrapolations->push_back(reco::TrackExtrapolation(*itrk, vresultPos, vresultMom));
    }
  }
  iEvent.put(std::move(extrapolations));
}

bool TrackExtrapolator::propagateTrackToVolume ( const reco::Track &  fTrack, const MagneticField &  fField, const Propagator &  fPropagator, const FiducialVolume &  volume, reco::TrackBase::Point &  resultPos, reco::TrackBase::Vector &  resultMom  )

private

track quality of the tracks we care about

Propagate a track to a given radius, given the magnetic field and the propagator. Store the resulting position, momentum, and direction.

Definition at line 72 of file TrackExtrapolator.cc.

References Plane::build(), Cylinder::build(), reco::TrackBase::charge(), reco::Track::extra(), TrajectoryStateOnSurface::globalMomentum(), TrajectoryStateOnSurface::globalPosition(), edm::Ref< C, T, F >::isAvailable(), TrajectoryStateOnSurface::isValid(), FiducialVolume::minR(), FiducialVolume::minZ(), gpuPixelDoublets::minz, reco::Track::outerPx(), reco::Track::outerPy(), reco::Track::outerPz(), reco::Track::outerX(), reco::Track::outerY(), reco::Track::outerZ(), Propagator::propagate(), reco::TrackBase::px(), reco::TrackBase::py(), reco::TrackBase::pz(), reco::btau::trackMomentum, reco::TrackBase::vx(), reco::TrackBase::vy(), reco::TrackBase::vz(), and PV3DBase< T, PVType, FrameType >::z().

Referenced by produce().

                                                                               {
  GlobalPoint trackPosition(fTrack.vx(), fTrack.vy(), fTrack.vz());   // reference point
  GlobalVector trackMomentum(fTrack.px(), fTrack.py(), fTrack.pz());  // reference momentum
  if (fTrack.extra().isAvailable()) {                                 // use outer point information, if available
    trackPosition = GlobalPoint(fTrack.outerX(), fTrack.outerY(), fTrack.outerZ());
    trackMomentum = GlobalVector(fTrack.outerPx(), fTrack.outerPy(), fTrack.outerPz());
  }

  GlobalTrajectoryParameters trackParams(trackPosition, trackMomentum, fTrack.charge(), &fField);
  FreeTrajectoryState trackState(trackParams);

  TrajectoryStateOnSurface propagatedInfo = fPropagator.propagate(
      trackState, *Cylinder::build(volume.minR(), Surface::PositionType(0, 0, 0), Surface::RotationType()));

  // if the track went through either side of the endcaps, repropagate the track
  double minz = volume.minZ();
  if (propagatedInfo.isValid() && propagatedInfo.globalPosition().z() > minz) {
    propagatedInfo =
        fPropagator.propagate(trackState, *Plane::build(Surface::PositionType(0, 0, minz), Surface::RotationType()));

  } else if (propagatedInfo.isValid() && propagatedInfo.globalPosition().z() < -minz) {
    propagatedInfo =
        fPropagator.propagate(trackState, *Plane::build(Surface::PositionType(0, 0, -minz), Surface::RotationType()));
  }

  if (propagatedInfo.isValid()) {
    resultPos = propagatedInfo.globalPosition();
    resultMom = propagatedInfo.globalMomentum();
    return true;
  } else {
    return false;
  }
}

Member Data Documentation

const edm::ESGetToken<DetIdAssociator, DetIdAssociatorRecord> TrackExtrapolator::ecalDetIdAssociatorToken_

private

Definition at line 71 of file TrackExtrapolator.h.

Referenced by produce().

const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> TrackExtrapolator::fieldToken_

private

Input tracks.

Definition at line 69 of file TrackExtrapolator.h.

Referenced by produce().

const edm::ESGetToken<Propagator, TrackingComponentsRecord> TrackExtrapolator::propagatorToken_

private

Definition at line 70 of file TrackExtrapolator.h.

Referenced by produce().

const reco::TrackBase::TrackQuality TrackExtrapolator::trackQuality_

private

Definition at line 72 of file TrackExtrapolator.h.

Referenced by produce(), and TrackExtrapolator().

const edm::EDGetTokenT<reco::TrackCollection> TrackExtrapolator::tracksSrc_

private

Definition at line 68 of file TrackExtrapolator.h.

Referenced by produce().