#include <SimpleTrackRefitter.h>

Public Member Functions
std::vector< Trajectory >	refitTrack (const reco::Track &newTrack, const uint32_t ExcludedDetId=0)
	The main methods. More...

std::vector< Trajectory >	refitTrack (const TrajectorySeed &seed, const TrackingRecHitCollection &hits, const uint32_t ExcludedDetid=0)

std::vector< Trajectory >	refitTrack (const TrajectorySeed &seed, const reco::Track &theT, const uint32_t ExcludedDetid=0)

void	setServices (const edm::EventSetup &es)

	SimpleTrackRefitter (const edm::ParameterSet &)
	Constructor. More...

virtual	~SimpleTrackRefitter ()
	Destructor. More...

Private Member Functions
Trajectory	createStartingTrajectory (const TrajectorySeed &seed) const

void	destroyServices ()

void	initServices (const bool &seedAlongMomentum)

std::vector < TrajectoryMeasurement >	seedMeasurements (const TrajectorySeed &seed) const

TrajectoryStateOnSurface	startingTSOS (const TrajectorySeed &seed) const

Private Attributes
edm::ParameterSet	conf_

bool	isCosmics_

const MagneticField *	magfield

const TransientTrackingRecHitBuilder *	RHBuilder

const Chi2MeasurementEstimator *	theEstimator

const TrajectoryFitter *	theFitter

const Propagator *	thePropagator

const Propagator *	thePropagatorOp

const TrajectorySmoother *	theSmoother

const KFUpdator *	theUpdator

const TrackerGeometry *	tracker

const TrajectoryStateTransform	tsTransform

Detailed Description

This class refits a track (from cosmic or standard tracking). The final result is a Trajectory refitted and smoothed. To make the refitting (and the smoothing) the usual KF tools are used. An arbitrary DetUnit can be ignored during the fit, so that the resulting trajectory can be used for unbiased residual studies on that Det.

Definition at line 46 of file SimpleTrackRefitter.h.

Constructor & Destructor Documentation

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

Constructor.

Definition at line 25 of file SimpleTrackRefitter.cc.

References conf_, edm::ParameterSet::getParameter(), and isCosmics_.

                                                                       :
 thePropagator(0),thePropagatorOp(0),theUpdator(0),theEstimator(0),RHBuilder(0),theSmoother(0),theFitter(0),tsTransform(),conf_(iConfig)
 {
   isCosmics_ = conf_.getParameter<bool>("cosmic");
 }

SimpleTrackRefitter::~SimpleTrackRefitter ( )

virtual

Destructor.

Definition at line 31 of file SimpleTrackRefitter.cc.

 {
 
 }

Member Function Documentation

Trajectory SimpleTrackRefitter::createStartingTrajectory ( const TrajectorySeed & seed ) const

private

Definition at line 251 of file SimpleTrackRefitter.cc.

References TrajectorySeed::direction(), i, query::result, and seedMeasurements().

Referenced by refitTrack().

 {
   Trajectory result( seed, seed.direction());
   std::vector<TrajectoryMeasurement> seedMeas = seedMeasurements(seed);
   if ( !seedMeas.empty()) {
     for (std::vector<TrajectoryMeasurement>::const_iterator i=seedMeas.begin(); i!=seedMeas.end(); i++){
       result.push(*i);
     }
   }
   return result;
 }

void SimpleTrackRefitter::destroyServices ( )

private

Definition at line 88 of file SimpleTrackRefitter.cc.

References theEstimator, theFitter, thePropagator, thePropagatorOp, theSmoother, and theUpdator.

Referenced by refitTrack().

 {
   if(thePropagator) { delete thePropagator; thePropagator=0; }
   if(thePropagatorOp) { delete thePropagatorOp; thePropagatorOp =0; }
   if(theUpdator) { delete theUpdator; theUpdator=0; }
   if(theEstimator) { delete theEstimator; theEstimator=0; }
   if(theSmoother) { delete theSmoother; theSmoother=0; }
   if(theFitter) { delete theFitter; theFitter=0; }
 }

void SimpleTrackRefitter::initServices ( const bool & seedAlongMomentum )

private

Definition at line 36 of file SimpleTrackRefitter.cc.

References alongMomentum, Chi2MeasurementEstimatorESProducer_cfi::Chi2MeasurementEstimator, KFTrajectoryFitterESProducer_cfi::KFTrajectoryFitter, KFTrajectorySmootherESProducer_cfi::KFTrajectorySmoother, magfield, oppositeToMomentum, theEstimator, theFitter, thePropagator, thePropagatorOp, theSmoother, and theUpdator.

Referenced by refitTrack().

 {
   // dynamic services for cosmic refit
   if(seedAlongMomentum) {
     thePropagator   = new PropagatorWithMaterial(alongMomentum,0.1057,magfield );
     thePropagatorOp = new PropagatorWithMaterial(oppositeToMomentum,0.1057,magfield );
   }
   else {
     thePropagator   = new PropagatorWithMaterial(oppositeToMomentum,0.1057,magfield );
     thePropagatorOp = new PropagatorWithMaterial(alongMomentum,0.1057,magfield );
   }
   theUpdator=         new KFUpdator();
   theEstimator=       new Chi2MeasurementEstimator(30);
   theFitter=          new KFTrajectoryFitter(*thePropagator,
                          *theUpdator,   
                          *theEstimator) ;
   theSmoother=        new KFTrajectorySmoother(*thePropagatorOp,
                            *theUpdator, 
                            *theEstimator);
 }

std::vector< Trajectory > SimpleTrackRefitter::refitTrack	(	const reco::Track &	newTrack,
		const uint32_t	ExcludedDetId = `0`
	)

The main methods.

Definition at line 98 of file SimpleTrackRefitter.cc.

References TransientTrackingRecHitBuilder::build(), funct::C, TrajectoryFitter::fit(), i, TrackingRecHit::inactive, LogDebug, Propagator::magneticField(), NULL, Propagator::propagate(), reco::Track::recHitsBegin(), reco::Track::recHitsEnd(), RHBuilder, theFitter, thePropagator, tmp, and DetId::Tracker.

Referenced by SiStripFineDelayTLA::findtrackangle().

 {
   // convert the TrackingRecHit vector to a TransientTrackingRecHit vector
   LogDebug("SimpleTrackRefitter") << "Start\n";
   TransientTrackingRecHit::RecHitContainer tmp;
   for (trackingRecHit_iterator i=theT.recHitsBegin();i!=theT.recHitsEnd(); i++) {
     if ( (*i)->geographicalId().det() == DetId::Tracker ) {
       if((*i)->isValid()) {
         if((*i)->geographicalId().rawId()!=ExcludedDetid) {
       tmp.push_back(RHBuilder->build(&**i ));
     } else {
       InvalidTrackingRecHit* irh = new InvalidTrackingRecHit((*i)->geographicalId(), TrackingRecHit::inactive);
       tmp.push_back(RHBuilder->build(irh));
       delete irh;
         }
       } else {
         tmp.push_back(RHBuilder->build(&**i ));
       }
     }
   }
   LogDebug("SimpleTrackRefitter") << "Transient rechit filled" << "\n";
   // sort rechits alongmomentum
   TransientTrackingRecHit::RecHitContainer hits;
   const TransientTrackingRecHit::ConstRecHitPointer *firstHit = NULL;
   for (TransientTrackingRecHit::RecHitContainer::const_iterator it=tmp.begin(); it!=tmp.end();it++){
     if ((**it).isValid()) {
       firstHit = &(*it);
       break;
     }
   }
   const TransientTrackingRecHit::ConstRecHitPointer *lastHit = NULL;
   for (TransientTrackingRecHit::RecHitContainer::const_iterator it=tmp.end()-1; it!=tmp.begin()-1;it--){
     if ((**it).isValid()) {
       lastHit= &(*it);
       break;
     }
   }
   if ((*firstHit)->globalPosition().mag2() > ((*lastHit)->globalPosition().mag2()) ){
     for (TransientTrackingRecHit::RecHitContainer::const_iterator it=tmp.end()-1;it!=tmp.begin()-1;it--){
       hits.push_back(*it);
     }
   } else hits=tmp;
   
   // build the transient track and fit it
   std::vector<Trajectory> trajVec;
   reco::TransientTrack theTT(theT, thePropagator->magneticField() );
   TrajectoryStateOnSurface firstState=thePropagator->propagate(theTT.impactPointState(), hits.front()->det()->surface());
   AlgebraicSymMatrix C(5,1);
   C *= 100.;
   if(!firstState.isValid()) return trajVec;
   TrajectoryStateOnSurface theTSOS( firstState.localParameters(), LocalTrajectoryError(C),
                     firstState.surface(), thePropagator->magneticField()); 
   LogDebug("SimpleTrackRefitter") << "initial TSOS\n" << theTSOS << "\n";
   const TrajectorySeed seed;//empty seed: not needed
   trajVec = theFitter->fit(seed, hits, theTSOS);
   LogDebug("SimpleTrackRefitter") <<" FITTER FOUND "<< trajVec.size() << " TRAJECTORIES" <<"\n";
   return trajVec;
 }

std::vector< Trajectory > SimpleTrackRefitter::refitTrack	(	const TrajectorySeed &	seed,
		const TrackingRecHitCollection &	hits,
		const uint32_t	ExcludedDetid = `0`
	)

Definition at line 157 of file SimpleTrackRefitter.cc.

References alongMomentum, TransientTrackingRecHitBuilder::build(), createStartingTrajectory(), destroyServices(), TrajectorySeed::direction(), Chi2MeasurementEstimator::estimate(), TrajectoryFitter::fit(), TrackingRecHit::geographicalId(), TrackerGeometry::idToDet(), TrackingRecHit::inactive, initServices(), TrajectoryStateOnSurface::isValid(), Propagator::propagate(), DetId::rawId(), RHBuilder, edm::OwnVector< T, P >::size(), GeomDet::surface(), theEstimator, theFitter, thePropagator, thePropagatorOp, theSmoother, theUpdator, tracker, TrajectorySmoother::trajectories(), and KFUpdator::update().

 {
   initServices(seed.direction()==alongMomentum);
   Trajectory traj=createStartingTrajectory(seed);
   TransientTrackingRecHit::RecHitContainer trans_hits;
   for (unsigned int icosmhit=hits.size()-1;icosmhit+1>0;icosmhit--){
     TransientTrackingRecHit::RecHitPointer tmphit;
     const TrackingRecHit* rh = &(hits[icosmhit]);
     if(rh->geographicalId().rawId() == ExcludedDetid) {
       InvalidTrackingRecHit* irh = new InvalidTrackingRecHit(rh->geographicalId(), TrackingRecHit::inactive);
       tmphit=RHBuilder->build(irh);
       delete irh;
     }
     else {
       tmphit=RHBuilder->build(rh);
     }
     trans_hits.push_back(&(*tmphit));
     if (icosmhit<hits.size()-1){
       TSOS prSt = thePropagator->propagate(traj.lastMeasurement().updatedState(),
                                            tracker->idToDet(hits[icosmhit].geographicalId())->surface());
       if(prSt.isValid() && tmphit->isValid()) {
        if(theUpdator->update(prSt,*tmphit).isValid()) {
       traj.push(TrajectoryMeasurement(prSt, theUpdator->update(prSt,*tmphit),
                       tmphit, theEstimator->estimate(prSt,*tmphit).second));
        }
       }
     }
   }
   std::vector<Trajectory> smoothtraj;
   if (thePropagatorOp->propagate(traj.lastMeasurement().updatedState(),
                  tracker->idToDet((*trans_hits.begin())->geographicalId())->surface()).isValid()){
     TSOS startingState=  TrajectoryStateWithArbitraryError()
       (thePropagatorOp->propagate(traj.lastMeasurement().updatedState(),
                   tracker->idToDet((*trans_hits.begin())->geographicalId())->surface()));
     std::vector<Trajectory> fittraj=theFitter->fit(seed,trans_hits,startingState);
     if (fittraj.size()) smoothtraj=theSmoother->trajectories(*(fittraj.begin()));
   }
   destroyServices();
   return smoothtraj;
 }

std::vector< Trajectory > SimpleTrackRefitter::refitTrack	(	const TrajectorySeed &	seed,
		const reco::Track &	theT,
		const uint32_t	ExcludedDetid = `0`
	)

Definition at line 200 of file SimpleTrackRefitter.cc.

References alongMomentum, TransientTrackingRecHitBuilder::build(), createStartingTrajectory(), destroyServices(), TrajectorySeed::direction(), Chi2MeasurementEstimator::estimate(), TrajectoryFitter::fit(), i, TrackerGeometry::idToDet(), TrackingRecHit::inactive, initServices(), TrajectoryStateOnSurface::isValid(), Propagator::propagate(), reco::Track::recHitsBegin(), reco::Track::recHitsEnd(), RHBuilder, theEstimator, theFitter, thePropagator, thePropagatorOp, theSmoother, theUpdator, tracker, TrajectorySmoother::trajectories(), and KFUpdator::update().

 {
   initServices(seed.direction()==alongMomentum);
   Trajectory traj=createStartingTrajectory(seed);
   TransientTrackingRecHit::RecHitContainer trans_hits;
   for (trackingRecHit_iterator i=theT.recHitsBegin();i!=theT.recHitsEnd(); i++) {
     TransientTrackingRecHit::RecHitPointer tmphit;
     if((*i)->geographicalId().rawId() == ExcludedDetid) {
       InvalidTrackingRecHit* irh = new InvalidTrackingRecHit((*i)->geographicalId(), TrackingRecHit::inactive);
       tmphit=RHBuilder->build(irh);
       delete irh;
     }
     else {
       tmphit=RHBuilder->build(&**i);
     }
     trans_hits.push_back(&(*tmphit));
     if (i!=theT.recHitsBegin()){
       TSOS prSt = thePropagator->propagate(traj.lastMeasurement().updatedState(),
                                            tracker->idToDet((*i)->geographicalId())->surface());
       if(prSt.isValid() && tmphit->isValid()) {
        if(theUpdator->update(prSt,*tmphit).isValid()) {
       traj.push(TrajectoryMeasurement(prSt, theUpdator->update(prSt,*tmphit),
                       tmphit, theEstimator->estimate(prSt,*tmphit).second));
        }
       }
     }
   }
   std::vector<Trajectory> smoothtraj;
   if (thePropagatorOp->propagate(traj.lastMeasurement().updatedState(),
                  tracker->idToDet((*trans_hits.begin())->geographicalId())->surface()).isValid()){
     TSOS startingState=  TrajectoryStateWithArbitraryError()
       (thePropagatorOp->propagate(traj.lastMeasurement().updatedState(),
                   tracker->idToDet((*trans_hits.begin())->geographicalId())->surface()));
     std::vector<Trajectory> fittraj=theFitter->fit(seed,trans_hits,startingState);
     if (fittraj.size()) smoothtraj=theSmoother->trajectories(*(fittraj.begin()));
   }
   destroyServices();
   return smoothtraj;
 }

std::vector< TrajectoryMeasurement > SimpleTrackRefitter::seedMeasurements ( const TrajectorySeed & seed ) const

private

Definition at line 263 of file SimpleTrackRefitter.cc.

References TransientTrackingRecHitBuilder::build(), TrackerGeometry::idToDet(), TrajectorySeed::recHits(), query::result, RHBuilder, startingTSOS(), GeomDet::surface(), and tracker.

Referenced by createStartingTrajectory().

 {
   std::vector<TrajectoryMeasurement> result;
   TrajectorySeed::range hitRange = seed.recHits();
   for (TrajectorySeed::const_iterator ihit = hitRange.first; ihit != hitRange.second; ihit++) {
     TransientTrackingRecHit::RecHitPointer recHit = RHBuilder->build(&(*ihit));
     const GeomDet* hitGeomDet = tracker->idToDet( ihit->geographicalId());
     TSOS invalidState( new BasicSingleTrajectoryState( hitGeomDet->surface()));
     if (ihit == hitRange.second - 1) {
       TSOS  updatedState=startingTSOS(seed);
       result.push_back(TrajectoryMeasurement( invalidState, updatedState, recHit));
     } 
     else {
       result.push_back(TrajectoryMeasurement( invalidState, recHit));
     }
   }
   return result;
 }

void SimpleTrackRefitter::setServices ( const edm::EventSetup & es )

Definition at line 57 of file SimpleTrackRefitter.cc.

References conf_, edm::EventSetup::get(), edm::ParameterSet::getParameter(), isCosmics_, LogDebug, magfield, LargeD0_PixelPairStep_cff::propagator, RHBuilder, theFitter, thePropagator, and tracker.

Referenced by SiStripFineDelayTLA::init().

 {
   // get geometry
   edm::ESHandle<TrackerGeometry> estracker;
   es.get<TrackerDigiGeometryRecord>().get(estracker);
   tracker=&(*estracker);
   // get magnetic field
   edm::ESHandle<MagneticField> esmagfield;
   es.get<IdealMagneticFieldRecord>().get(esmagfield);
   magfield=&(*esmagfield);
   // get the builder
   edm::ESHandle<TransientTrackingRecHitBuilder> builder;
   LogDebug("SimpleTrackRefitter") << "get also the TransientTrackingRecHitBuilder" << "\n";
   std::string builderName = conf_.getParameter<std::string>("TTRHBuilder");   
   es.get<TransientRecHitRecord>().get(builderName,builder);
   RHBuilder=&(*builder);
   if(isCosmics_) return;
   // get the fitter
   edm::ESHandle<TrajectoryFitter> fitter;
   LogDebug("SimpleTrackRefitter") << "get the fitter from the ES" << "\n";
   std::string fitterName = conf_.getParameter<std::string>("Fitter");   
   es.get<TrajectoryFitter::Record>().get(fitterName,fitter);
   theFitter=&(*fitter);
   // get the propagator
   edm::ESHandle<Propagator> propagator;
   LogDebug("SimpleTrackRefitter") << "get also the propagator" << "\n";
   std::string propagatorName = conf_.getParameter<std::string>("Propagator");   
   es.get<TrackingComponentsRecord>().get(propagatorName,propagator);
   thePropagator=&(*propagator);
 }

TrajectoryStateOnSurface SimpleTrackRefitter::startingTSOS ( const TrajectorySeed & seed ) const

private

Definition at line 242 of file SimpleTrackRefitter.cc.

References TrackerGeometry::idToDet(), magfield, TrajectorySeed::startingState(), tracker, TrajectoryStateTransform::transientState(), and tsTransform.

Referenced by seedMeasurements().

 {
   PTrajectoryStateOnDet pState( seed.startingState());
   const GeomDet* gdet  = tracker->idToDet(DetId(pState.detId()));
   TSOS  State= tsTransform.transientState( pState, &(gdet->surface()), &(*magfield));
   return State;
 
 }