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...
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	setClusterRemovalInfo (const edm::InputTag &clusterRemovalInfo)
	Sets the information on cluster removal, and turns it on. More...
void	setConf (const edm::ParameterSet &conf)
	Set parameter set. More...
void	setSecondHitPattern (Trajectory *traj, reco::Track &track, const Propagator *prop, const MeasurementTrackerEvent *measTk, const TrackerTopology *ttopo)
void	setSrc (const edm::EDGetToken &src, const edm::EDGetTokenT< reco::BeamSpot > &bsSrc, const edm::EDGetTokenT< MeasurementTrackerEvent > &mteSrc)
	set label of source collection More...
	TrackProducerBase (bool trajectoryInEvent=false)
	Constructor. More...
virtual	~TrackProducerBase ()(false)
	Destructor. More...

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 >
Protected Attributes inherited from TrackProducerBase< reco::Track >
std::string	alias_
edm::EDGetTokenT< reco::BeamSpot >	bsSrc_
edm::InputTag	clusterRemovalInfo_
edm::ParameterSet	conf_
edm::EDGetTokenT< MeasurementTrackerEvent >	mteSrc_
bool	rekeyClusterRefs_
edm::OrphanHandle< TrackCollection >	rTracks_
edm::EDGetToken	src_
edm::ESHandle< NavigationSchool >	theSchool
bool	trajectoryInEvent_

Detailed Description

Produce Tracks from TrackCandidates

Author

cerati

Definition at line 16 of file KfTrackProducerBase.h.

Constructor & Destructor Documentation

KfTrackProducerBase::KfTrackProducerBase ( bool  trajectoryInEvent, bool  split  )

inlineexplicit

Constructor.

Definition at line 20 of file KfTrackProducerBase.h.

References putInEvt().

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

Member Function Documentation

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 22 of file KfTrackProducerBase.cc.

References alongMomentum, reco::TrackBase::appendHitPattern(), TrackProducerBase< reco::Track >::clusterRemovalInfo_, TrajectoryStateOnSurface::curvilinearError(), edm::Event::getByToken(), edm::Event::getRefBeforePut(), TrajectoryStateOnSurface::globalParameters(), mps_fire::i, training_settings::idx, edm::ESHandleBase::isValid(), LogDebug, LogTrace, GlobalTrajectoryParameters::momentum(), eostools::move(), TrackProducerBase< reco::Track >::mteSrc_, AlCaHLTBitMon_ParallelJobs::p, GlobalTrajectoryParameters::position(), edm::Event::put(), reco::TrackExtraBase::recHitsSize(), ClusterRemovalRefSetter::reKey(), TrackProducerBase< reco::Track >::rekeyClusterRefs_, TrackProducerBase< reco::Track >::rTracks_, reco::Track::setExtra(), reco::TrackExtraBase::setHits(), reco::TrackExtra::setResiduals(), TrackProducerBase< reco::Track >::setSecondHitPattern(), reco::TrackExtraBase::setTrajParams(), TrackProducerBase< reco::Track >::theSchool, HiIsolationCommonParameters_cff::track, TrackProducerBase< reco::Track >::trajectoryInEvent_, trajectoryToResiduals(), reco::TrackExtraBase::trajParams(), useSplitting, findQualityFiles::v, PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

Referenced by KfTrackProducerBase(), 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) );
  }
}

Member Data Documentation

bool KfTrackProducerBase::useSplitting

private

Definition at line 44 of file KfTrackProducerBase.h.

Referenced by putInEvt().