KfTrackProducerBase Class Reference

#include <KfTrackProducerBase.h>

Inheritance diagram for KfTrackProducerBase:

Public Member Functions
	KfTrackProducerBase (bool trajectoryInEvent, bool split)
	Constructor. More...
virtual void	putInEvt (edm::Event &, const Propagator *prop, const MeasurementTracker *measTk, std::unique_ptr< TrackingRecHitCollection > &, std::unique_ptr< reco::TrackCollection > &, std::unique_ptr< reco::TrackExtraCollection > &, std::unique_ptr< std::vector< Trajectory > > &, std::unique_ptr< std::vector< int > > &, AlgoProductCollection &, TransientTrackingRecHitBuilder const *, const TrackerTopology *ttopo, int BeforeOrAfter=0)
	Put produced collections in the event. More...
Public Member Functions inherited from TrackProducerBase< reco::Track >
const edm::ParameterSet &	getConf () const
virtual void	getFromES (const edm::EventSetup &, edm::ESHandle< TrackerGeometry > &, edm::ESHandle< MagneticField > &, edm::ESHandle< TrajectoryFitter > &, edm::ESHandle< Propagator > &, edm::ESHandle< MeasurementTracker > &, edm::ESHandle< TransientTrackingRecHitBuilder > &)
	Get needed services from the Event Setup. More...
virtual void	getFromEvt (edm::Event &, edm::Handle< TrackCandidateCollection > &, reco::BeamSpot &)
	Get TrackCandidateCollection from the Event (needed by TrackProducer) More...
virtual void	getFromEvt (edm::Event &, edm::Handle< TrackView > &, reco::BeamSpot &)
	Get TrackCollection from the Event (needed by TrackRefitter) More...
void	initTrackProducerBase (const edm::ParameterSet &conf, edm::ConsumesCollector cc, const edm::EDGetToken &src)
	Call this method in inheriting class' constructor. More...
virtual void	produce (edm::Event &, const edm::EventSetup &)=0
	Method where the procduction take place. To be implemented in concrete classes. More...
void	setAlias (std::string alias)
	set the aliases of produced collections More...
void	setSecondHitPattern (Trajectory *traj, reco::Track &track, const Propagator *prop, const MeasurementTrackerEvent *measTk, const TrackerTopology *ttopo)
	TrackProducerBase (bool trajectoryInEvent=false)
	Constructor. More...
virtual	~TrackProducerBase () noexcept(false)
	Destructor. More...

Protected Member Functions
void	setClusterRemovalInfo (const edm::InputTag &clusterRemovalInfo)
	Sets the information on cluster removal, and turns it on. More...

Protected Attributes
edm::InputTag	clusterRemovalInfo_
bool	rekeyClusterRefs_
Protected Attributes inherited from TrackProducerBase< reco::Track >
std::string	alias_
edm::EDGetTokenT< reco::BeamSpot >	bsSrc_
edm::ESGetToken< TransientTrackingRecHitBuilder, TransientRecHitRecord >	builderSrc_
edm::ParameterSet	conf_
edm::ESGetToken< TrajectoryFitter, TrajectoryFitter::Record >	fitterSrc_
edm::ESGetToken< MeasurementTracker, CkfComponentsRecord >	measTkSrc_
edm::ESGetToken< MagneticField, IdealMagneticFieldRecord >	mfSrc_
edm::EDGetTokenT< MeasurementTrackerEvent >	mteSrc_
edm::ESGetToken< Propagator, TrackingComponentsRecord >	propagatorSrc_
edm::OrphanHandle< TrackCollection >	rTracks_
edm::ESGetToken< NavigationSchool, NavigationSchoolRecord >	schoolSrc_
edm::EDGetToken	src_
edm::ESHandle< NavigationSchool >	theSchool
edm::ESGetToken< TrackerGeometry, TrackerDigiGeometryRecord >	trackGeomSrc_
bool	trajectoryInEvent_
bool	useSchool_

Private Attributes
bool	useSplitting

Additional Inherited Members
Public Types inherited from TrackProducerBase< reco::Track >
using	AlgoProductCollection = typename Base::AlgoProductCollection
using	Base = AlgoProductTraits< reco::Track >
using	TrackCollection = typename Base::TrackCollection
using	TrackView = typename Base::TrackView
Public Types inherited from AlgoProductTraits< reco::Track >
using	AlgoProductCollection = std::vector< AlgoProduct >
using	TrackCollection = std::vector< reco::Track >
using	TrackView = edm::View< reco::Track >

Detailed Description

Produce Tracks from TrackCandidates

Author

cerati

Definition at line 16 of file KfTrackProducerBase.h.

Constructor & Destructor Documentation

KfTrackProducerBase()

KfTrackProducerBase::KfTrackProducerBase ( bool  trajectoryInEvent, bool  split  )

inlineexplicit

Constructor.

Definition at line 19 of file KfTrackProducerBase.h.

      : TrackProducerBase<reco::Track>(trajectoryInEvent), rekeyClusterRefs_(false), useSplitting(split) {}

Member Function Documentation

putInEvt()

void KfTrackProducerBase::putInEvt ( edm::Event &  evt, const Propagator *  prop, const MeasurementTracker *  measTk, std::unique_ptr< TrackingRecHitCollection > &  selHits, std::unique_ptr< reco::TrackCollection > &  selTracks, std::unique_ptr< reco::TrackExtraCollection > &  selTrackExtras, std::unique_ptr< std::vector< Trajectory > > &  selTrajectories, std::unique_ptr< std::vector< int > > &  indecesInput, AlgoProductCollection &  algoResults, TransientTrackingRecHitBuilder const *  hitBuilder, const TrackerTopology *  ttopo, int  BeforeOrAfter = 0  )

virtual

Put produced collections in the event.

Definition at line 21 of file KfTrackProducerBase.cc.

References alongMomentum, cms::cuda::assert(), clusterRemovalInfo_, TrajectoryStateOnSurface::curvilinearError(), edm::Event::getByToken(), edm::Event::getRefBeforePut(), TrajectoryStateOnSurface::globalParameters(), mps_fire::i, heavyIonCSV_trainingSettings::idx, edm::ESHandleBase::isValid(), ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it, LogDebug, LogTrace, GlobalTrajectoryParameters::momentum(), eostools::move(), TrackProducerBase< reco::Track >::mteSrc_, AlCaHLTBitMon_ParallelJobs::p, GlobalTrajectoryParameters::position(), edm::Event::put(), reco::TrackExtraBase::recHitsSize(), ClusterRemovalRefSetter::reKey(), rekeyClusterRefs_, TrackProducerBase< reco::Track >::rTracks_, reco::TrackExtraBase::setHits(), reco::TrackExtra::setResiduals(), TrackProducerBase< reco::Track >::setSecondHitPattern(), reco::TrackExtraBase::setTrajParams(), TrackProducerBase< reco::Track >::theSchool, HLT_2024v14_cff::track, TrackProducerBase< reco::Track >::trajectoryInEvent_, trajectoryToResiduals(), reco::TrackExtraBase::trajParams(), useSplitting, and findQualityFiles::v.

Referenced by TrackRefitter::produce(), TrackProducer::produce(), and DAFTrackProducer::produce().

                                                      {
  TrackingRecHitRefProd rHits = evt.getRefBeforePut<TrackingRecHitCollection>();
  reco::TrackExtraRefProd rTrackExtras = evt.getRefBeforePut<reco::TrackExtraCollection>();

  edm::Ref<reco::TrackExtraCollection>::key_type idx = 0;
  edm::Ref<reco::TrackCollection>::key_type iTkRef = 0;
  edm::Ref<std::vector<Trajectory> >::key_type iTjRef = 0;
  std::map<unsigned int, unsigned int> tjTkMap;

  selTracks->reserve(algoResults.size());
  selTrackExtras->reserve(algoResults.size());
  if (trajectoryInEvent_)
    selTrajectories->reserve(algoResults.size());

  for (AlgoProductCollection::iterator i = algoResults.begin(); i != algoResults.end(); i++) {
    auto theTraj = (*i).trajectory;
    (*indecesInput).push_back((*i).indexInput);
    if (trajectoryInEvent_) {
      selTrajectories->push_back(*theTraj);
      iTjRef++;
    }

    auto theTrack = (*i).track;

    // Hits are going to be re-sorted along momentum few lines later.
    // Therefore the direction stored in the TrackExtra
    // has to be "alongMomentum" as well. Anyway, this direction can be differnt from the one of the orignal
    // seed! The name seedDirection() for the Track's method (and the corresponding data member) is
    // misleading and should be changed into something like "hitsDirection()". TO BE FIXED!

    PropagationDirection seedDir = alongMomentum;

    LogDebug("TrackProducer") << "In KfTrackProducerBase::putInEvt - seedDir=" << seedDir;

    selTracks->push_back(std::move(*theTrack));
    delete theTrack;
    iTkRef++;

    // Store indices in local map (starts at 0)
    if (trajectoryInEvent_)
      tjTkMap[iTjRef - 1] = iTkRef - 1;

    //sets the outermost and innermost TSOSs
    TrajectoryStateOnSurface outertsos;
    TrajectoryStateOnSurface innertsos;
    unsigned int innerId, outerId;

    // ---  NOTA BENE: the convention is to sort hits and measurements "along the momentum".
    // This is consistent with innermost and outermost labels only for tracks from LHC collision
    if (theTraj->direction() == alongMomentum) {
      outertsos = theTraj->lastMeasurement().updatedState();
      innertsos = theTraj->firstMeasurement().updatedState();
      outerId = theTraj->lastMeasurement().recHit()->geographicalId().rawId();
      innerId = theTraj->firstMeasurement().recHit()->geographicalId().rawId();
    } else {
      outertsos = theTraj->firstMeasurement().updatedState();
      innertsos = theTraj->lastMeasurement().updatedState();
      outerId = theTraj->firstMeasurement().recHit()->geographicalId().rawId();
      innerId = theTraj->lastMeasurement().recHit()->geographicalId().rawId();
    }
    // ---
    //build the TrackExtra
    GlobalPoint v = outertsos.globalParameters().position();
    GlobalVector p = outertsos.globalParameters().momentum();
    math::XYZVector outmom(p.x(), p.y(), p.z());
    math::XYZPoint outpos(v.x(), v.y(), v.z());
    v = innertsos.globalParameters().position();
    p = innertsos.globalParameters().momentum();
    math::XYZVector inmom(p.x(), p.y(), p.z());
    math::XYZPoint inpos(v.x(), v.y(), v.z());

    reco::TrackExtraRef teref = reco::TrackExtraRef(rTrackExtras, idx++);
    reco::Track& track = selTracks->back();
    track.setExtra(teref);
    //======= I want to set the second hitPattern here =============
    if (theSchool.isValid()) {
      edm::Handle<MeasurementTrackerEvent> mte;
      evt.getByToken(mteSrc_, mte);
      // NavigationSetter setter( *theSchool );
      setSecondHitPattern(theTraj, track, prop, &*mte, ttopo);
    }
    //==============================================================

    selTrackExtras->push_back(reco::TrackExtra(outpos,
                                               outmom,
                                               true,
                                               inpos,
                                               inmom,
                                               true,
                                               outertsos.curvilinearError(),
                                               outerId,
                                               innertsos.curvilinearError(),
                                               innerId,
                                               seedDir,
                                               theTraj->seedRef()));

    // FIXME will remove this obsolete config-param in a future PR
    assert(!useSplitting);

    reco::TrackExtra& tx = selTrackExtras->back();
    // ---  NOTA BENE: the convention is to sort hits and measurements "along the momentum".
    // This is consistent with innermost and outermost labels only for tracks from LHC collisions
    reco::TrackExtra::TrajParams trajParams;
    reco::TrackExtra::Chi2sFive chi2s;
    Traj2TrackHits t2t;
    auto ih = selHits->size();
    t2t(*theTraj, *selHits, trajParams, chi2s);
    auto ie = selHits->size();
    tx.setHits(rHits, ih, ie - ih);
    tx.setTrajParams(std::move(trajParams), std::move(chi2s));
    assert(tx.trajParams().size() == tx.recHitsSize());
    for (; ih < ie; ++ih) {
      auto const& hit = (*selHits)[ih];
      track.appendHitPattern(hit, *ttopo);
    }

    // ----
    tx.setResiduals(trajectoryToResiduals(*theTraj));

    delete theTraj;
  }

  // Now we can re-set refs to hits, as they have already been cloned
  if (rekeyClusterRefs_) {
    ClusterRemovalRefSetter refSetter(evt, clusterRemovalInfo_);
    for (TrackingRecHitCollection::iterator it = selHits->begin(), ed = selHits->end(); it != ed; ++it) {
      refSetter.reKey(&*it);
    }
  }

  LogTrace("TrackingRegressionTest") << "========== TrackProducer Info ===================";
  LogTrace("TrackingRegressionTest") << "number of finalTracks: " << selTracks->size();
  for (reco::TrackCollection::const_iterator it = selTracks->begin(); it != selTracks->end(); it++) {
    LogTrace("TrackingRegressionTest") << "track's n valid and invalid hit, chi2, pt, eta : " << it->found() << " , "
                                       << it->lost() << " , " << it->normalizedChi2() << " , " << it->pt() << " , "
                                       << it->eta();
  }
  LogTrace("TrackingRegressionTest") << "=================================================";

  selTracks->shrink_to_fit();
  selTrackExtras->shrink_to_fit();
  selHits->shrink_to_fit();
  indecesInput->shrink_to_fit();
  if (BeforeOrAfter == 1) {
    rTracks_ = evt.put(std::move(selTracks), "beforeDAF");
    evt.put(std::move(selTrackExtras), "beforeDAF");
    evt.put(std::move(indecesInput), "beforeDAF");
  } else if (BeforeOrAfter == 2) {
    rTracks_ = evt.put(std::move(selTracks), "afterDAF");
    evt.put(std::move(selTrackExtras), "afterDAF");
    evt.put(std::move(indecesInput), "afterDAF");
  } else {
    rTracks_ = evt.put(std::move(selTracks));
    evt.put(std::move(selTrackExtras));
    evt.put(std::move(selHits));
    evt.put(std::move(indecesInput));
  }

  if (trajectoryInEvent_ && BeforeOrAfter == 0) {
    selTrajectories->shrink_to_fit();
    edm::OrphanHandle<std::vector<Trajectory> > rTrajs = evt.put(std::move(selTrajectories));

    // Now Create traj<->tracks association map
    std::unique_ptr<TrajTrackAssociationCollection> trajTrackMap(new TrajTrackAssociationCollection(rTrajs, rTracks_));
    for (std::map<unsigned int, unsigned int>::iterator i = tjTkMap.begin(); i != tjTkMap.end(); i++) {
      edm::Ref<std::vector<Trajectory> > trajRef(rTrajs, (*i).first);
      edm::Ref<reco::TrackCollection> tkRef(rTracks_, (*i).second);
      trajTrackMap->insert(edm::Ref<std::vector<Trajectory> >(rTrajs, (*i).first),
                           edm::Ref<reco::TrackCollection>(rTracks_, (*i).second));
    }
    evt.put(std::move(trajTrackMap));
  }
}

setClusterRemovalInfo()

void KfTrackProducerBase::setClusterRemovalInfo ( const edm::InputTag &  clusterRemovalInfo )

inlineprotected

Sets the information on cluster removal, and turns it on.

Definition at line 43 of file KfTrackProducerBase.h.

References HLT_2024v14_cff::clusterRemovalInfo, clusterRemovalInfo_, and rekeyClusterRefs_.

Referenced by TrackProducer::TrackProducer().

                                                                    {
    rekeyClusterRefs_ = true;
    clusterRemovalInfo_ = clusterRemovalInfo;
  }

Member Data Documentation

clusterRemovalInfo_

edm::InputTag KfTrackProducerBase::clusterRemovalInfo_

protected

Definition at line 49 of file KfTrackProducerBase.h.

Referenced by putInEvt(), and setClusterRemovalInfo().

rekeyClusterRefs_

bool KfTrackProducerBase::rekeyClusterRefs_

protected

Definition at line 48 of file KfTrackProducerBase.h.

Referenced by putInEvt(), and setClusterRemovalInfo().

useSplitting

bool KfTrackProducerBase::useSplitting

private

Definition at line 52 of file KfTrackProducerBase.h.

Referenced by putInEvt().