#include <SimpleTrackRefitter.h>

Public Member Functions
std::vector< Trajectory >	refitTrack (const reco::Track &newTrack, const uint32_t ExcludedDetId=0)
	The main methods.
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.
virtual	~SimpleTrackRefitter ()
	Destructor.
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 align::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;

}