CosmicTrajectoryBuilder Class Reference

#include <CosmicTrajectoryBuilder.h>

Public Member Functions
	CosmicTrajectoryBuilder (const edm::ParameterSet &conf)
Trajectory	createStartingTrajectory (const TrajectorySeed &seed) const
void	init (const edm::EventSetup &es, bool)
void	run (const TrajectorySeedCollection &collseed, const SiStripRecHit2DCollection &collstereo, const SiStripRecHit2DCollection &collrphi, const SiStripMatchedRecHit2DCollection &collmatched, const SiPixelRecHitCollection &collpixel, const edm::EventSetup &es, edm::Event &e, std::vector< Trajectory > &trajoutput)
	Runs the algorithm. More...
	~CosmicTrajectoryBuilder ()

Private Types
typedef TrajectoryMeasurement	TM
typedef TrajectoryStateOnSurface	TSOS

Private Member Functions
void	AddHit (Trajectory &traj, std::vector< const TrackingRecHit * >Hits)
bool	qualityFilter (Trajectory traj)
std::vector < TrajectoryMeasurement >	seedMeasurements (const TrajectorySeed &seed) const
std::vector< const TrackingRecHit * >	SortHits (const SiStripRecHit2DCollection &collstereo, const SiStripRecHit2DCollection &collrphi, const SiStripMatchedRecHit2DCollection &collmatched, const SiPixelRecHitCollection &collpixel, const TrajectorySeed &seed)
TSOS	startingTSOS (const TrajectorySeed &seed) const
void	updateTrajectory (Trajectory &traj, const TM &tm, const TransientTrackingRecHit &hit) const

Private Attributes
double	chi2cut
edm::ParameterSet	conf_
std::string	geometry
TransientTrackingRecHit::RecHitContainer	hits
edm::ESHandle< MagneticField >	magfield
const TransientTrackingRecHitBuilder *	RHBuilder
bool	seed_plus
Chi2MeasurementEstimator *	theEstimator
const KFTrajectoryFitter *	theFitter
int	theMinHits
PropagatorWithMaterial *	thePropagator
PropagatorWithMaterial *	thePropagatorOp
const KFTrajectorySmoother *	theSmoother
KFUpdator *	theUpdator
edm::ESHandle< TrackerGeometry >	tracker
std::vector< Trajectory >	trajFit

Detailed Description

Definition at line 73 of file CosmicTrajectoryBuilder.h.

Member Typedef Documentation

typedef TrajectoryMeasurement CosmicTrajectoryBuilder::TM

private

Definition at line 77 of file CosmicTrajectoryBuilder.h.

typedef TrajectoryStateOnSurface CosmicTrajectoryBuilder::TSOS

private

Definition at line 76 of file CosmicTrajectoryBuilder.h.

Constructor & Destructor Documentation

CosmicTrajectoryBuilder::CosmicTrajectoryBuilder

(

const edm::ParameterSet &

conf

)

Definition at line 21 of file CosmicTrajectoryBuilder.cc.

References chi2cut, conf_, geometry, edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), and theMinHits.

                                                                            : conf_(conf) { 
  //minimum number of hits per tracks

  theMinHits=conf_.getParameter<int>("MinHits");
  //cut on chi2
  chi2cut=conf_.getParameter<double>("Chi2Cut");
  edm::LogInfo("CosmicTrackFinder")<<"Minimum number of hits "<<theMinHits<<" Cut on Chi2= "<<chi2cut;

  geometry=conf_.getUntrackedParameter<std::string>("GeometricStructure","STANDARD");
}

CosmicTrajectoryBuilder::~CosmicTrajectoryBuilder

(

)

Definition at line 33 of file CosmicTrajectoryBuilder.cc.

                                                  {
}

Member Function Documentation

void CosmicTrajectoryBuilder::AddHit ( Trajectory &  traj, std::vector< const TrackingRecHit * >  Hits  )

private

Definition at line 259 of file CosmicTrajectoryBuilder.cc.

References abs, TransientTrackingRecHitBuilder::build(), chi2cut, Trajectory::chiSquared(), Chi2MeasurementEstimator::estimate(), Trajectory::firstMeasurement(), KFTrajectoryFitter::fit(), Trajectory::foundHits(), TrajectoryStateOnSurface::globalPosition(), hits, TrajectoryStateOnSurface::isValid(), Trajectory::lastMeasurement(), TrajectoryStateOnSurface::localPosition(), LogDebug, PropagatorWithMaterial::propagate(), Trajectory::push(), qualityFilter(), RHBuilder, Trajectory::seed(), theEstimator, theFitter, thePropagator, thePropagatorOp, theUpdator, tracker, trajFit, KFUpdator::update(), TrajectoryMeasurement::updatedState(), PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

Referenced by run().

                                                       {


  unsigned int icosm2;
  unsigned int ibestdet;
  float chi2min;
  for (unsigned int icosmhit=0;icosmhit<Hits.size();icosmhit++){
    GlobalPoint gphit=RHBuilder->build(Hits[icosmhit])->globalPosition();
    unsigned int iraw= Hits[icosmhit]->geographicalId().rawId();
    LogDebug("CosmicTrackFinder")<<" HIT POSITION "<< gphit;
    //RC TransientTrackingRecHit* tmphit=RHBuilder->build(Hits[icosmhit]);
    TransientTrackingRecHit::RecHitPointer tmphit=RHBuilder->build(Hits[icosmhit]);
     TSOS prSt= thePropagator->propagate(traj.lastMeasurement().updatedState(),
                    tracker->idToDet(Hits[icosmhit]->geographicalId())->surface());

    //After propagating the trajectory to a detector,
    //find the most compatible hit in the det
    chi2min=20000000;
    ibestdet=1000;
     if (prSt.isValid()){
       LocalPoint  prLoc = prSt.localPosition();
       LogDebug("CosmicTrackFinder") <<"STATE PROPAGATED AT DET "<<iraw<<" "<<prSt;
       for(icosm2=icosmhit;icosm2<Hits.size();icosm2++){

     if (iraw==Hits[icosm2]->geographicalId().rawId()){   
       TransientTrackingRecHit::RecHitPointer tmphit=RHBuilder->build(Hits[icosm2]);
       float contr= theEstimator->estimate(prSt, *tmphit).second;
       if (contr<chi2min) {
         chi2min=contr;
         ibestdet=icosm2;
       }
       if (icosm2!=icosmhit)    icosmhit++;

     }
     else  icosm2=Hits.size();
       }

       if(chi2min<chi2cut) 
     LogDebug("CosmicTrackFinder")<<"Chi2 contribution for hit at "
                      <<RHBuilder->build(Hits[ibestdet])->globalPosition()
                      <<" is "<<chi2min;

       if(traj.foundHits()<3 &&(chi2min<chi2cut)){
     //check on the first hit after the seed
     GlobalVector ck=RHBuilder->build(Hits[ibestdet])->globalPosition()-
       traj.firstMeasurement().updatedState().globalPosition();
     if ((abs(ck.x()/ck.y())>2)||(abs(ck.z()/ck.y())>2))  chi2min=300;
       }
       if (chi2min<chi2cut){
     if ( abs(prLoc.x()) < 25  && abs(prLoc.y()) < 25 ){
       TransientTrackingRecHit::RecHitPointer tmphitbestdet=RHBuilder->build(Hits[ibestdet]);
       TSOS UpdatedState= theUpdator->update( prSt, *tmphitbestdet);
       if (UpdatedState.isValid()){

         traj.push(TM(prSt,UpdatedState,RHBuilder->build(Hits[ibestdet])
              , chi2min));
         LogDebug("CosmicTrackFinder") <<
           "STATE UPDATED WITH HIT AT POSITION "
                       <<tmphitbestdet->globalPosition()
                       <<UpdatedState<<" "
                       <<traj.chiSquared();

         hits.push_back(&(*tmphitbestdet));
       }
     }else LogDebug("CosmicTrackFinder")<<" Hits outside module surface "<< prLoc;
       }else LogDebug("CosmicTrackFinder")<<" State can not be updated with hit at position " <<gphit;
     }else LogDebug("CosmicTrackFinder")<<" State can not be propagated at det "<< iraw;    
  }
  
  

  if ( qualityFilter( traj)){
    const TrajectorySeed& tmpseed=traj.seed();
    if (thePropagatorOp->propagate(traj.lastMeasurement().updatedState(),
                   tracker->idToDet((*hits.begin())->geographicalId())->surface()).isValid()){
      TSOS startingState= 
    thePropagatorOp->propagate(traj.lastMeasurement().updatedState(),
                   tracker->idToDet((*hits.begin())->geographicalId())->surface());
      
      trajFit = theFitter->fit(tmpseed,hits, startingState );
    }
  }
  
}

Trajectory CosmicTrajectoryBuilder::createStartingTrajectory

(

const TrajectorySeed &

seed

)

const

Definition at line 130 of file CosmicTrajectoryBuilder.cc.

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

Referenced by run().

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

void CosmicTrajectoryBuilder::init	(	const edm::EventSetup &	es,
		bool	seedplus
	)

Definition at line 37 of file CosmicTrajectoryBuilder.cc.

References alongMomentum, chi2cut, Chi2MeasurementEstimatorESProducer_cfi::Chi2MeasurementEstimator, conf_, edm::EventSetup::get(), edm::ParameterSet::getParameter(), KFTrajectoryFitterESProducer_cfi::KFTrajectoryFitter, KFTrajectorySmootherESProducer_cfi::KFTrajectorySmoother, magfield, oppositeToMomentum, edm::ESHandle< class >::product(), RHBuilder, seed_plus, theEstimator, theFitter, thePropagator, thePropagatorOp, theSmoother, theUpdator, and tracker.

Referenced by run().

                                                                        {


  //services
  es.get<IdealMagneticFieldRecord>().get(magfield);
  es.get<TrackerDigiGeometryRecord>().get(tracker);
  
 
  
  if (seedplus) {    
    seed_plus=true;      
    thePropagator=      new PropagatorWithMaterial(alongMomentum,0.1057,&(*magfield) );      
    thePropagatorOp=    new PropagatorWithMaterial(oppositeToMomentum,0.1057,&(*magfield) );}    
  else {
    seed_plus=false;
    thePropagator=      new PropagatorWithMaterial(oppositeToMomentum,0.1057,&(*magfield) );    
    thePropagatorOp=    new PropagatorWithMaterial(alongMomentum,0.1057,&(*magfield) );
  }
  
  theUpdator=       new KFUpdator();
  theEstimator=     new Chi2MeasurementEstimator(chi2cut);
  

  edm::ESHandle<TransientTrackingRecHitBuilder> theBuilder;
  std::string builderName = conf_.getParameter<std::string>("TTRHBuilder");   
  es.get<TransientRecHitRecord>().get(builderName,theBuilder);
  

  RHBuilder=   theBuilder.product();




  theFitter=        new KFTrajectoryFitter(*thePropagator,
                       *theUpdator, 
                       *theEstimator) ;
  

  theSmoother=      new KFTrajectorySmoother(*thePropagatorOp,
                         *theUpdator,   
                         *theEstimator);
  
}

bool CosmicTrajectoryBuilder::qualityFilter ( Trajectory  traj )

private

Definition at line 347 of file CosmicTrajectoryBuilder.cc.

References Trajectory::foundHits(), geometry, hits, Trajectory::recHits(), and theMinHits.

Referenced by AddHit().

                                                     {
  int ngoodhits=0;
  if(geometry=="MTCC"){
    std::vector< ConstReferenceCountingPointer< TransientTrackingRecHit> > hits= traj.recHits();
    std::vector< ConstReferenceCountingPointer< TransientTrackingRecHit> >::const_iterator hit;
    for(hit=hits.begin();hit!=hits.end();hit++){
      unsigned int iid=(*hit)->hit()->geographicalId().rawId();
      //CHECK FOR 3 hits r-phi
      if(((iid>>0)&0x3)!=1) ngoodhits++;
    }
  }
  else ngoodhits=traj.foundHits();
  
  if ( ngoodhits >= theMinHits) {
    return true;
  }
  else {
    return false;
  }
}

void CosmicTrajectoryBuilder::run	(	const TrajectorySeedCollection &	collseed,
		const SiStripRecHit2DCollection &	collstereo,
		const SiStripRecHit2DCollection &	collrphi,
		const SiStripMatchedRecHit2DCollection &	collmatched,
		const SiPixelRecHitCollection &	collpixel,
		const edm::EventSetup &	es,
		edm::Event &	e,
		std::vector< Trajectory > &	trajoutput
	)

Runs the algorithm.

Definition at line 81 of file CosmicTrajectoryBuilder.cc.

References AddHit(), alongMomentum, createStartingTrajectory(), hits, init(), iseed, SortHits(), theEstimator, theFitter, thePropagator, thePropagatorOp, theSmoother, theUpdator, KFTrajectorySmoother::trajectories(), and trajFit.

Referenced by cms::CosmicTrackFinder::produce().

{

 


  std::vector<Trajectory> trajSmooth;
  std::vector<Trajectory>::iterator trajIter;
  
  TrajectorySeedCollection::const_iterator iseed;
  unsigned int IS=0;
  for(iseed=collseed.begin();iseed!=collseed.end();iseed++){
    bool seedplus=((*iseed).direction()==alongMomentum);
    init(es,seedplus);
    hits.clear();
    trajFit.clear();
    trajSmooth.clear();
    //order all the hits
    vector<const TrackingRecHit*> allHits= SortHits(collstereo,collrphi,collmatched,collpixel,*iseed);
    Trajectory startingTraj = createStartingTrajectory(*iseed);
    AddHit(startingTraj,allHits);
    for (trajIter=trajFit.begin(); trajIter!=trajFit.end();trajIter++){
      trajSmooth=theSmoother->trajectories((*trajIter));
    }
    for (trajIter= trajSmooth.begin(); trajIter!=trajSmooth.end();trajIter++){
      if((*trajIter).isValid()){
    trajoutput.push_back((*trajIter));
      }
    }
    delete theUpdator;
    delete theEstimator;
    delete thePropagator;
    delete thePropagatorOp;
    delete theFitter;
    delete theSmoother;
    //Only the first 30 seeds are considered
    if (IS>30) return;
    IS++;

  }
}

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

private

Definition at line 145 of file CosmicTrajectoryBuilder.cc.

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

Referenced by createStartingTrajectory().

{
  std::vector<TrajectoryMeasurement> result;
  TrajectorySeed::range hitRange = seed.recHits();
  for (TrajectorySeed::const_iterator ihit = hitRange.first; 
       ihit != hitRange.second; ihit++) {
    //RC TransientTrackingRecHit* recHit = RHBuilder->build(&(*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(TM( invalidState, updatedState, recHit));

    } 
    else {
      result.push_back(TM( invalidState, recHit));
    }
    
  }

  return result;
}

vector< const TrackingRecHit * > CosmicTrajectoryBuilder::SortHits ( const SiStripRecHit2DCollection &  collstereo, const SiStripRecHit2DCollection &  collrphi, const SiStripMatchedRecHit2DCollection &  collmatched, const SiPixelRecHitCollection &  collpixel, const TrajectorySeed &  seed  )

private

Definition at line 175 of file CosmicTrajectoryBuilder.cc.

References TransientTrackingRecHitBuilder::build(), edmNew::DetSetVector< T >::data(), hits, LogDebug, TrajectorySeed::recHits(), RHBuilder, seed_plus, and tracker.

Referenced by run().

                                             {


  //The Hits with global y more than the seed are discarded
  //The Hits correspondign to the seed are discarded
  //At the end all the hits are sorted in y
  vector<const TrackingRecHit*> allHits;

  SiStripRecHit2DCollection::DataContainer::const_iterator istrip;
  TrajectorySeed::range hRange= seed.recHits();
  TrajectorySeed::const_iterator ihit;
  float yref=0.;
  for (ihit = hRange.first; 
       ihit != hRange.second; ihit++) {
    yref=RHBuilder->build(&(*ihit))->globalPosition().y();
    hits.push_back((RHBuilder->build(&(*ihit)))); 
    LogDebug("CosmicTrackFinder")<<"SEED HITS"<<RHBuilder->build(&(*ihit))->globalPosition();
  }

  
  if ((&collpixel)!=0){
    SiPixelRecHitCollection::DataContainer::const_iterator ipix;
    for(ipix=collpixel.data().begin();ipix!=collpixel.data().end();ipix++){
      float ych= RHBuilder->build(&(*ipix))->globalPosition().y();
      if ((seed_plus && (ych<yref)) || (!(seed_plus) && (ych>yref)))
    allHits.push_back(&(*ipix));
    }
  } 
  
  

  if ((&collrphi)!=0){
    for(istrip=collrphi.data().begin();istrip!=collrphi.data().end();istrip++){
      float ych= RHBuilder->build(&(*istrip))->globalPosition().y();
      if ((seed_plus && (ych<yref)) || (!(seed_plus) && (ych>yref)))
    allHits.push_back(&(*istrip));   
    }
  }




  if ((&collstereo)!=0){
    for(istrip=collstereo.data().begin();istrip!=collstereo.data().end();istrip++){
      float ych= RHBuilder->build(&(*istrip))->globalPosition().y();
      if ((seed_plus && (ych<yref)) || (!(seed_plus) && (ych>yref)))
    allHits.push_back(&(*istrip));
    }
  }

//   SiStripMatchedRecHit2DCollection::DataContainer::const_iterator istripm;
//   if ((&collmatched)!=0){
//     for(istripm=collmatched.data().begin();istripm!=collmatched.data().end();istripm++){
//       float ych= RHBuilder->build(&(*istripm))->globalPosition().y();
//       if ((seed_plus && (ych<yref)) || (!(seed_plus) && (ych>yref)))
//  allHits.push_back(&(*istripm));
//     }
//   }

  if (seed_plus){
    stable_sort(allHits.begin(),allHits.end(),CompareHitY_plus(*tracker));
  }
  else {
    stable_sort(allHits.begin(),allHits.end(),CompareHitY(*tracker));
  }

  return allHits;
}

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

private

Definition at line 249 of file CosmicTrajectoryBuilder.cc.

References magfield, TrajectorySeed::startingState(), and trajectoryStateTransform::transientState().

Referenced by seedMeasurements().

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

}

void CosmicTrajectoryBuilder::updateTrajectory ( Trajectory &  traj, const TM &  tm, const TransientTrackingRecHit &  hit  ) const

private

Member Data Documentation

double CosmicTrajectoryBuilder::chi2cut

private

Definition at line 135 of file CosmicTrajectoryBuilder.h.

Referenced by AddHit(), CosmicTrajectoryBuilder(), and init().

edm::ParameterSet CosmicTrajectoryBuilder::conf_

private

Definition at line 122 of file CosmicTrajectoryBuilder.h.

Referenced by CosmicTrajectoryBuilder(), and init().

std::string CosmicTrajectoryBuilder::geometry

private

Definition at line 140 of file CosmicTrajectoryBuilder.h.

Referenced by CosmicTrajectoryBuilder(), Vispa.Gui.ConnectableWidget.ConnectableWidget::leaveEvent(), and qualityFilter().

TransientTrackingRecHit::RecHitContainer CosmicTrajectoryBuilder::hits

private

Definition at line 138 of file CosmicTrajectoryBuilder.h.

Referenced by AddHit(), qualityFilter(), run(), and SortHits().

edm::ESHandle<MagneticField> CosmicTrajectoryBuilder::magfield

private

Definition at line 120 of file CosmicTrajectoryBuilder.h.

Referenced by init(), and startingTSOS().

const TransientTrackingRecHitBuilder* CosmicTrajectoryBuilder::RHBuilder

private

Definition at line 128 of file CosmicTrajectoryBuilder.h.

Referenced by AddHit(), init(), seedMeasurements(), and SortHits().

bool CosmicTrajectoryBuilder::seed_plus

private

Definition at line 139 of file CosmicTrajectoryBuilder.h.

Referenced by init(), and SortHits().

Chi2MeasurementEstimator* CosmicTrajectoryBuilder::theEstimator

private

Definition at line 127 of file CosmicTrajectoryBuilder.h.

Referenced by AddHit(), init(), and run().

const KFTrajectoryFitter* CosmicTrajectoryBuilder::theFitter

private

Definition at line 130 of file CosmicTrajectoryBuilder.h.

Referenced by AddHit(), init(), and run().

int CosmicTrajectoryBuilder::theMinHits

private

Definition at line 134 of file CosmicTrajectoryBuilder.h.

Referenced by CosmicTrajectoryBuilder(), and qualityFilter().

PropagatorWithMaterial* CosmicTrajectoryBuilder::thePropagator

private

Definition at line 124 of file CosmicTrajectoryBuilder.h.

Referenced by AddHit(), init(), and run().

PropagatorWithMaterial* CosmicTrajectoryBuilder::thePropagatorOp

private

Definition at line 125 of file CosmicTrajectoryBuilder.h.

Referenced by AddHit(), init(), and run().

const KFTrajectorySmoother* CosmicTrajectoryBuilder::theSmoother

private

Definition at line 129 of file CosmicTrajectoryBuilder.h.

Referenced by init(), and run().

KFUpdator* CosmicTrajectoryBuilder::theUpdator

private

Definition at line 126 of file CosmicTrajectoryBuilder.h.

Referenced by AddHit(), init(), and run().

edm::ESHandle<TrackerGeometry> CosmicTrajectoryBuilder::tracker

private

Definition at line 121 of file CosmicTrajectoryBuilder.h.

Referenced by AddHit(), init(), and SortHits().

std::vector<Trajectory> CosmicTrajectoryBuilder::trajFit

private

Definition at line 136 of file CosmicTrajectoryBuilder.h.

Referenced by AddHit(), and run().