|
|
|
#include <KfTrackProducerBase.h>
Public Member Functions | |
| KfTrackProducerBase (bool trajectoryInEvent, bool split) | |
| Constructor. | |
| virtual void | putInEvt (edm::Event &, const Propagator *prop, const MeasurementTracker *measTk, std::auto_ptr< TrackingRecHitCollection > &, std::auto_ptr< reco::TrackCollection > &, std::auto_ptr< reco::TrackExtraCollection > &, std::auto_ptr< std::vector< Trajectory > > &, AlgoProductCollection &) |
| Put produced collections in the event. | |
Private Attributes | |
| bool | useSplitting |
Produce Tracks from TrackCandidates
Definition at line 18 of file KfTrackProducerBase.h.
| KfTrackProducerBase::KfTrackProducerBase | ( | bool | trajectoryInEvent, |
| bool | split | ||
| ) | [inline, explicit] |
Constructor.
Definition at line 22 of file KfTrackProducerBase.h.
:
TrackProducerBase<reco::Track>(trajectoryInEvent),useSplitting(split) {}
| void KfTrackProducerBase::putInEvt | ( | edm::Event & | evt, |
| const Propagator * | prop, | ||
| const MeasurementTracker * | measTk, | ||
| std::auto_ptr< TrackingRecHitCollection > & | selHits, | ||
| std::auto_ptr< reco::TrackCollection > & | selTracks, | ||
| std::auto_ptr< reco::TrackExtraCollection > & | selTrackExtras, | ||
| std::auto_ptr< std::vector< Trajectory > > & | selTrajectories, | ||
| AlgoProductCollection & | algoResults | ||
| ) | [virtual] |
Put produced collections in the event.
Definition at line 19 of file KfTrackProducerBase.cc.
References reco::TrackExtraBase::add(), alongMomentum, TrackProducerBase< reco::Track >::clusterRemovalInfo_, TrajectoryStateOnSurface::curvilinearError(), Trajectory::direction(), Trajectory::firstMeasurement(), edm::Event::getRefBeforePut(), TrajectoryStateOnSurface::globalParameters(), i, edm::ESHandleBase::isValid(), j, Trajectory::lastMeasurement(), LogDebug, LogTrace, GlobalTrajectoryParameters::momentum(), AlCaHLTBitMon_ParallelJobs::p, GlobalTrajectoryParameters::position(), edm::Event::put(), TrajectoryMeasurement::recHit(), Trajectory::recHits(), ClusterRemovalRefSetter::reKey(), TrackProducerBase< reco::Track >::rekeyClusterRefs_, TrackProducerBase< reco::Track >::rTracks_, Trajectory::seedRef(), reco::Track::setExtra(), reco::TrackBase::setHitPattern(), reco::TrackExtra::setResiduals(), TrackProducerBase< reco::Track >::setSecondHitPattern(), matplotRender::t, TrackProducerBase< reco::Track >::theSchool, TrackProducerBase< reco::Track >::trajectoryInEvent_, trajectoryToResiduals(), TrajectoryMeasurement::updatedState(), useSplitting, v, PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().
Referenced by MTFTrackProducer::produce(), TrackProducer::produce(), DAFTrackProducer::produce(), and TrackRefitter::produce().
{
TrackingRecHitRefProd rHits = evt.getRefBeforePut<TrackingRecHitCollection>();
reco::TrackExtraRefProd rTrackExtras = evt.getRefBeforePut<reco::TrackExtraCollection>();
edm::Ref<reco::TrackExtraCollection>::key_type idx = 0;
edm::Ref<reco::TrackExtraCollection>::key_type hidx = 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;
for(AlgoProductCollection::iterator i=algoResults.begin(); i!=algoResults.end();i++){
Trajectory * theTraj = (*i).first;
if(trajectoryInEvent_) {
selTrajectories->push_back(*theTraj);
iTjRef++;
}
// const TrajectoryFitter::RecHitContainer& transHits = theTraj->recHits(useSplitting); // NO: the return type in Trajectory is by VALUE
TrajectoryFitter::RecHitContainer transHits = theTraj->recHits(useSplitting);
reco::Track * theTrack = (*i).second.first;
// 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;
reco::Track t = * theTrack;
selTracks->push_back( t );
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())
{
NavigationSetter setter( *theSchool );
setSecondHitPattern(theTraj,track,prop,measTk);
}
//==============================================================
selTrackExtras->push_back( reco::TrackExtra (outpos, outmom, true, inpos, inmom, true,
outertsos.curvilinearError(), outerId,
innertsos.curvilinearError(), innerId,
seedDir, theTraj->seedRef()));
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
size_t ih = 0;
if (theTraj->direction() == alongMomentum) {
for( TrajectoryFitter::RecHitContainer::const_iterator j = transHits.begin();
j != transHits.end(); j ++ ) {
if ((**j).hit()!=0){
TrackingRecHit * hit = (**j).hit()->clone();
track.setHitPattern( * hit, ih ++ );
selHits->push_back( hit );
tx.add( TrackingRecHitRef( rHits, hidx ++ ) );
}
}
}else{
for( TrajectoryFitter::RecHitContainer::const_iterator j = transHits.end()-1;
j != transHits.begin()-1; --j ) {
if ((**j).hit()!=0){
TrackingRecHit * hit = (**j).hit()->clone();
track.setHitPattern( * hit, ih ++ );
selHits->push_back( hit );
tx.add( TrackingRecHitRef( rHits, hidx ++ ) );
}
}
}
// ----
tx.setResiduals(trajectoryToResiduals(*theTraj));
delete theTrack;
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") << "=================================================";
rTracks_ = evt.put( selTracks );
evt.put( selTrackExtras );
evt.put( selHits );
if(trajectoryInEvent_) {
edm::OrphanHandle<std::vector<Trajectory> > rTrajs = evt.put(selTrajectories);
// Now Create traj<->tracks association map
std::auto_ptr<TrajTrackAssociationCollection> trajTrackMap( new TrajTrackAssociationCollection() );
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( trajTrackMap );
}
}
bool KfTrackProducerBase::useSplitting [private] |
Definition at line 38 of file KfTrackProducerBase.h.
Referenced by putInEvt().
1.7.3