#include <CosmicTrajectoryBuilder.h>

Public Member Functions
	CosmicTrajectoryBuilder (const edm::ParameterSet &conf)

Trajectory	createStartingTrajectory (const TrajectorySeed &seed) const

const TransientTrackingRecHitBuilder *	hitBuilder () 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, const std::vector< const TrackingRecHit * > &Hits)

bool	qualityFilter (const 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

TkClonerImpl	hitCloner

TransientTrackingRecHit::RecHitContainer	hits

edm::ESHandle< MagneticField >	magfield

const TransientTrackingRecHitBuilder *	RHBuilder

bool	seed_plus

Chi2MeasurementEstimator *	theEstimator

KFTrajectoryFitter *	theFitter

int	theMinHits

PropagatorWithMaterial *	thePropagator

PropagatorWithMaterial *	thePropagatorOp

KFTrajectorySmoother *	theSmoother

KFUpdator *	theUpdator

edm::ESHandle< TrackerGeometry >	tracker

std::vector< Trajectory >	trajFit

Detailed Description

Definition at line 74 of file CosmicTrajectoryBuilder.h.

Member Typedef Documentation

typedef TrajectoryMeasurement CosmicTrajectoryBuilder::TM

private

Definition at line 78 of file CosmicTrajectoryBuilder.h.

typedef TrajectoryStateOnSurface CosmicTrajectoryBuilder::TSOS

private

Definition at line 77 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(), AlCaHLTBitMon_QueryRunRegistry::string, 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.

33 {

34 }

Member Function Documentation

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

private

Definition at line 254 of file CosmicTrajectoryBuilder.cc.

References funct::abs(), TransientTrackingRecHitBuilder::build(), chi2cut, Trajectory::chiSquared(), Chi2MeasurementEstimator::estimate(), Trajectory::firstMeasurement(), Trajectory::foundHits(), TrajectoryStateOnSurface::globalPosition(), hits, TrajectoryStateOnSurface::isValid(), Trajectory::lastMeasurement(), TrajectoryStateOnSurface::localPosition(), LogDebug, Trajectory::push(), qualityFilter(), RHBuilder, Trajectory::seed(), theEstimator, theFitter, thePropagator, thePropagatorOp, theUpdator, tracker, trajFit, 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(std::move(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 131 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);
   for (auto & i : seedMeas) result.push(std::move(i));
   return result;
 }

const TransientTrackingRecHitBuilder* CosmicTrajectoryBuilder::hitBuilder ( ) const

inline

Definition at line 99 of file CosmicTrajectoryBuilder.h.

References RHBuilder.

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

99 {return RHBuilder;}

CosmicTrajectoryBuilder::RHBuilder

const TransientTrackingRecHitBuilder * RHBuilder

Definition: CosmicTrajectoryBuilder.h:132

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(), hitCloner, KFTrajectoryFitterESProducer_cfi::KFTrajectoryFitter, KFTrajectorySmootherESProducer_cfi::KFTrajectorySmoother, magfield, oppositeToMomentum, edm::ESHandle< class >::product(), RHBuilder, seed_plus, AlCaHLTBitMon_QueryRunRegistry::string, theEstimator, theFitter, thePropagator, thePropagatorOp, theSmoother, theUpdator, and tracker.

Referenced by run().

                                                                         {
 
   // FIXME: this is a memory leak generator
 
   //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();
   hitCloner = static_cast<TkTransientTrackingRecHitBuilder const *>(RHBuilder)->cloner();
 
 
 
   theFitter=        new KFTrajectoryFitter(*thePropagator,
                        *theUpdator, 
                        *theEstimator) ;
   theFitter->setHitCloner(&hitCloner);
 
   theSmoother=      new KFTrajectorySmoother(*thePropagatorOp,
                          *theUpdator,   
                          *theEstimator);
   theSmoother->setHitCloner(&hitCloner);
 }

bool CosmicTrajectoryBuilder::qualityFilter ( const Trajectory & traj )

private

Definition at line 342 of file CosmicTrajectoryBuilder.cc.

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

Referenced by AddHit().

                                                             {
   int ngoodhits=0;
   if(geometry=="MTCC"){
     auto hits = traj.recHits();
     for(auto 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 82 of file CosmicTrajectoryBuilder.cc.

References AddHit(), alongMomentum, createStartingTrajectory(), hits, init(), iseed, SortHits(), theEstimator, theFitter, thePropagator, thePropagatorOp, theSmoother, theUpdator, 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 141 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.emplace_back(invalidState, updatedState, recHit);
     } 
     else {
       result.emplace_back(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 170 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 244 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 140 of file CosmicTrajectoryBuilder.h.

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

edm::ParameterSet CosmicTrajectoryBuilder::conf_

private

Definition at line 126 of file CosmicTrajectoryBuilder.h.

Referenced by CosmicTrajectoryBuilder(), and init().

std::string CosmicTrajectoryBuilder::geometry

private

Definition at line 145 of file CosmicTrajectoryBuilder.h.

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

TkClonerImpl CosmicTrajectoryBuilder::hitCloner

private

Definition at line 133 of file CosmicTrajectoryBuilder.h.

Referenced by init().

TransientTrackingRecHit::RecHitContainer CosmicTrajectoryBuilder::hits

private

Definition at line 143 of file CosmicTrajectoryBuilder.h.

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

edm::ESHandle<MagneticField> CosmicTrajectoryBuilder::magfield

private

Definition at line 124 of file CosmicTrajectoryBuilder.h.

Referenced by init(), and startingTSOS().

const TransientTrackingRecHitBuilder* CosmicTrajectoryBuilder::RHBuilder

private

Definition at line 132 of file CosmicTrajectoryBuilder.h.

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

bool CosmicTrajectoryBuilder::seed_plus

private

Definition at line 144 of file CosmicTrajectoryBuilder.h.

Referenced by init(), and SortHits().

Chi2MeasurementEstimator* CosmicTrajectoryBuilder::theEstimator

private

Definition at line 131 of file CosmicTrajectoryBuilder.h.

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

KFTrajectoryFitter* CosmicTrajectoryBuilder::theFitter

private

Definition at line 135 of file CosmicTrajectoryBuilder.h.

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

int CosmicTrajectoryBuilder::theMinHits

private

Definition at line 139 of file CosmicTrajectoryBuilder.h.

Referenced by CosmicTrajectoryBuilder(), and qualityFilter().

PropagatorWithMaterial* CosmicTrajectoryBuilder::thePropagator

private

Definition at line 128 of file CosmicTrajectoryBuilder.h.

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

PropagatorWithMaterial* CosmicTrajectoryBuilder::thePropagatorOp

private

Definition at line 129 of file CosmicTrajectoryBuilder.h.

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

KFTrajectorySmoother* CosmicTrajectoryBuilder::theSmoother

private

Definition at line 134 of file CosmicTrajectoryBuilder.h.

Referenced by init(), and run().

KFUpdator* CosmicTrajectoryBuilder::theUpdator

private

Definition at line 130 of file CosmicTrajectoryBuilder.h.

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

edm::ESHandle<TrackerGeometry> CosmicTrajectoryBuilder::tracker

private

Definition at line 125 of file CosmicTrajectoryBuilder.h.

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

std::vector<Trajectory> CosmicTrajectoryBuilder::trajFit

private

Definition at line 141 of file CosmicTrajectoryBuilder.h.

Referenced by AddHit(), and run().

Public Member Functions

Private Types

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation