#include <CosmicTrajectoryBuilder.h>

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

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
const edm::ESGetToken < TransientTrackingRecHitBuilder, TransientRecHitRecord >	builderToken_

double	chi2cut

std::string	geometry

TkClonerImpl	hitCloner

TransientTrackingRecHit::RecHitContainer	hits

const MagneticField *	magfield

const edm::ESGetToken < MagneticField, IdealMagneticFieldRecord >	magfieldToken_

const TransientTrackingRecHitBuilder *	RHBuilder

bool	seed_plus

Chi2MeasurementEstimator *	theEstimator

KFTrajectoryFitter *	theFitter

int	theMinHits

PropagatorWithMaterial *	thePropagator

PropagatorWithMaterial *	thePropagatorOp

KFTrajectorySmoother *	theSmoother

KFUpdator *	theUpdator

const TrackerGeometry *	tracker

const edm::ESGetToken < TrackerGeometry, TrackerDigiGeometryRecord >	trackerToken_

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 76 of file CosmicTrajectoryBuilder.h.

typedef TrajectoryStateOnSurface CosmicTrajectoryBuilder::TSOS

private

Definition at line 75 of file CosmicTrajectoryBuilder.h.

Constructor & Destructor Documentation

CosmicTrajectoryBuilder::CosmicTrajectoryBuilder	(	const edm::ParameterSet &	conf,
		edm::ConsumesCollector	iC
	)

Definition at line 20 of file CosmicTrajectoryBuilder.cc.

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

     : magfieldToken_(iC.esConsumes()),
       trackerToken_(iC.esConsumes()),
       builderToken_(iC.esConsumes(edm::ESInputTag("", conf.getParameter<std::string>("TTRHBuilder")))) {
   //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 34 of file CosmicTrajectoryBuilder.cc.

34 {}

Member Function Documentation

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

private

Definition at line 190 of file CosmicTrajectoryBuilder.cc.

References funct::abs(), TransientTrackingRecHitBuilder::build(), chi2cut, Trajectory::chiSquared(), Trajectory::firstMeasurement(), Trajectory::foundHits(), TrajectoryStateOnSurface::globalPosition(), hits, TrackerGeometry::idToDet(), sistrip::SpyUtilities::isValid(), TrajectoryStateOnSurface::isValid(), Trajectory::lastMeasurement(), TrajectoryStateOnSurface::localPosition(), LogDebug, Propagator::propagate(), Trajectory::push(), qualityFilter(), RHBuilder, Trajectory::seed(), GeomDet::surface(), 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 110 of file CosmicTrajectoryBuilder.cc.

References TrajectorySeed::direction(), mps_fire::i, eostools::move(), mps_fire::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 96 of file CosmicTrajectoryBuilder.h.

References RHBuilder.

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

96 { return RHBuilder; }

CosmicTrajectoryBuilder::RHBuilder

const TransientTrackingRecHitBuilder * RHBuilder

Definition: CosmicTrajectoryBuilder.h:125

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

Definition at line 36 of file CosmicTrajectoryBuilder.cc.

References alongMomentum, builderToken_, chi2cut, Chi2MeasurementEstimator_cfi::Chi2MeasurementEstimator, edm::EventSetup::getData(), hitCloner, magfield, magfieldToken_, oppositeToMomentum, RHBuilder, seed_plus, KFTrajectorySmoother::setHitCloner(), theEstimator, theFitter, thePropagator, thePropagatorOp, theSmoother, theUpdator, tracker, and trackerToken_.

Referenced by run().

                                                                          {
   // FIXME: this is a memory leak generator
 
   //services
   magfield = &es.getData(magfieldToken_);
   tracker = &es.getData(trackerToken_);
 
   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);
 
   RHBuilder = &es.getData(builderToken_);
   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 269 of file CosmicTrajectoryBuilder.cc.

References Trajectory::foundHits(), geometry, hits, Trajectory::recHits(), 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 66 of file CosmicTrajectoryBuilder.cc.

References AddHit(), alongMomentum, createStartingTrajectory(), hits, init(), iseed, SortHits(), theEstimator, theFitter, thePropagator, thePropagatorOp, theSmoother, theUpdator, TrajectorySmoother::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 118 of file CosmicTrajectoryBuilder.cc.

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

Referenced by createStartingTrajectory().

                                                                                                            {
   std::vector<TrajectoryMeasurement> result;
   auto const &hitRange = seed.recHits();
   for (auto ihit = hitRange.begin(); ihit != hitRange.end(); 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.end() - 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 138 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;
   float yref = 0.;
   for (auto const &recHit : seed.recHits()) {
     yref = RHBuilder->build(&recHit)->globalPosition().y();
     hits.push_back((RHBuilder->build(&recHit)));
     LogDebug("CosmicTrackFinder") << "SEED HITS" << RHBuilder->build(&recHit)->globalPosition();
   }
 
   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));
   }
 
   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));
   }
 
   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));
   }
 
   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 183 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

const edm::ESGetToken<TransientTrackingRecHitBuilder, TransientRecHitRecord> CosmicTrajectoryBuilder::builderToken_

private

Definition at line 117 of file CosmicTrajectoryBuilder.h.

Referenced by init().

double CosmicTrajectoryBuilder::chi2cut

private

Definition at line 131 of file CosmicTrajectoryBuilder.h.

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

std::string CosmicTrajectoryBuilder::geometry

private

Definition at line 136 of file CosmicTrajectoryBuilder.h.

Referenced by CosmicTrajectoryBuilder(), and qualityFilter().

TkClonerImpl CosmicTrajectoryBuilder::hitCloner

private

Definition at line 126 of file CosmicTrajectoryBuilder.h.

Referenced by init().

TransientTrackingRecHit::RecHitContainer CosmicTrajectoryBuilder::hits

private

Definition at line 134 of file CosmicTrajectoryBuilder.h.

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

const MagneticField* CosmicTrajectoryBuilder::magfield

private

Definition at line 118 of file CosmicTrajectoryBuilder.h.

Referenced by init(), and startingTSOS().

const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> CosmicTrajectoryBuilder::magfieldToken_

private

Definition at line 115 of file CosmicTrajectoryBuilder.h.

Referenced by init().

const TransientTrackingRecHitBuilder* CosmicTrajectoryBuilder::RHBuilder

private

Definition at line 125 of file CosmicTrajectoryBuilder.h.

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

bool CosmicTrajectoryBuilder::seed_plus

private

Definition at line 135 of file CosmicTrajectoryBuilder.h.

Referenced by init(), and SortHits().

Chi2MeasurementEstimator* CosmicTrajectoryBuilder::theEstimator

private

Definition at line 124 of file CosmicTrajectoryBuilder.h.

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

KFTrajectoryFitter* CosmicTrajectoryBuilder::theFitter

private

Definition at line 128 of file CosmicTrajectoryBuilder.h.

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

int CosmicTrajectoryBuilder::theMinHits

private

Definition at line 130 of file CosmicTrajectoryBuilder.h.

Referenced by CosmicTrajectoryBuilder(), and qualityFilter().

PropagatorWithMaterial* CosmicTrajectoryBuilder::thePropagator

private

Definition at line 121 of file CosmicTrajectoryBuilder.h.

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

PropagatorWithMaterial* CosmicTrajectoryBuilder::thePropagatorOp

private

Definition at line 122 of file CosmicTrajectoryBuilder.h.

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

KFTrajectorySmoother* CosmicTrajectoryBuilder::theSmoother

private

Definition at line 127 of file CosmicTrajectoryBuilder.h.

Referenced by init(), and run().

KFUpdator* CosmicTrajectoryBuilder::theUpdator

private

Definition at line 123 of file CosmicTrajectoryBuilder.h.

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

const TrackerGeometry* CosmicTrajectoryBuilder::tracker

private

Definition at line 119 of file CosmicTrajectoryBuilder.h.

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

const edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> CosmicTrajectoryBuilder::trackerToken_

private

Definition at line 116 of file CosmicTrajectoryBuilder.h.

Referenced by init().

std::vector<Trajectory> CosmicTrajectoryBuilder::trajFit

private

Definition at line 132 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