#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

std::string	geometry

TkClonerImpl	hitCloner

TransientTrackingRecHit::RecHitContainer	hits

edm::ESHandle< MagneticField >	magfield

const TransientTrackingRecHitBuilder *	RHBuilder

bool	seed_plus

std::string	theBuilderName

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

◆ TM

typedef TrajectoryMeasurement CosmicTrajectoryBuilder::TM

private

Definition at line 76 of file CosmicTrajectoryBuilder.h.

◆ TSOS

typedef TrajectoryStateOnSurface CosmicTrajectoryBuilder::TSOS

private

Definition at line 75 of file CosmicTrajectoryBuilder.h.

Constructor & Destructor Documentation

◆ CosmicTrajectoryBuilder()

CosmicTrajectoryBuilder::CosmicTrajectoryBuilder ( const edm::ParameterSet & conf )

Definition at line 21 of file CosmicTrajectoryBuilder.cc.

                                                                             {
   //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");
   theBuilderName = conf.getParameter<std::string>("TTRHBuilder");
 }

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

◆ ~CosmicTrajectoryBuilder()

CosmicTrajectoryBuilder::~CosmicTrajectoryBuilder ( )

Definition at line 33 of file CosmicTrajectoryBuilder.cc.

33 {}

Member Function Documentation

◆ AddHit()

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

private

Definition at line 194 of file CosmicTrajectoryBuilder.cc.

                                                                                                  {
   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);
     }
   }
 }

References funct::abs(), trackAssociatorByChi2_cfi::chi2cut, Trajectory::chiSquared(), Trajectory::firstMeasurement(), Trajectory::foundHits(), TrajectoryStateOnSurface::globalPosition(), hfClusterShapes_cfi::hits, TrajectoryStateOnSurface::isValid(), Trajectory::lastMeasurement(), TrajectoryStateOnSurface::localPosition(), LogDebug, Trajectory::push(), Trajectory::seed(), PbPb_ZMuSkimMuonDPG_cff::tracker, TrajectoryStateOnSurface::update(), TrajectoryMeasurement::updatedState(), PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

◆ createStartingTrajectory()

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

Definition at line 112 of file CosmicTrajectoryBuilder.cc.

                                                                                              {
   Trajectory result(seed, seed.direction());
   std::vector<TM> &&seedMeas = seedMeasurements(seed);
   for (auto &i : seedMeas)
     result.push(std::move(i));
   return result;
 }

References mps_fire::i, eostools::move(), mps_fire::result, and SurveyInfoScenario_cff::seed.

◆ hitBuilder()

const TransientTrackingRecHitBuilder* CosmicTrajectoryBuilder::hitBuilder ( ) const

inline

Definition at line 96 of file CosmicTrajectoryBuilder.h.

96 { return RHBuilder; }

References RHBuilder.

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

◆ init()

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

Definition at line 35 of file CosmicTrajectoryBuilder.cc.

                                                                          {
   // 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;
   es.get<TransientRecHitRecord>().get(theBuilderName, 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);
 }

References alongMomentum, trackAssociatorByChi2_cfi::chi2cut, Chi2MeasurementEstimator_cfi::Chi2MeasurementEstimator, edm::EventSetup::get(), get, volumeBasedMagneticField_160812_cfi::magfield, oppositeToMomentum, edm::ESHandle< T >::product(), and PbPb_ZMuSkimMuonDPG_cff::tracker.

◆ qualityFilter()

bool CosmicTrajectoryBuilder::qualityFilter ( const Trajectory & traj )

private

Definition at line 273 of file CosmicTrajectoryBuilder.cc.

                                                                   {
   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;
   }
 }

References Trajectory::foundHits(), hfClusterShapes_cfi::hits, and Trajectory::recHits().

◆ run()

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 68 of file CosmicTrajectoryBuilder.cc.

                                                                   {
   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++;
   }
 }

References alongMomentum, hfClusterShapes_cfi::hits, init, and iseed.

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

◆ seedMeasurements()

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

private

Definition at line 120 of file CosmicTrajectoryBuilder.cc.

                                                                                                            {
   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;
 }

References rpcPointValidation_cfi::recHit, mps_fire::result, SurveyInfoScenario_cff::seed, GeomDet::surface(), and TrackInfoProducer_cfi::updatedState.

◆ SortHits()

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

private

Definition at line 140 of file CosmicTrajectoryBuilder.cc.

                                                                                              {
   //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();
   }
  
   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;
 }

References edmNew::DetSetVector< T >::data(), hfClusterShapes_cfi::hits, LogDebug, SurveyInfoScenario_cff::seed, and PbPb_ZMuSkimMuonDPG_cff::tracker.

◆ startingTSOS()

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

private

Definition at line 187 of file CosmicTrajectoryBuilder.cc.

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

References volumeBasedMagneticField_160812_cfi::magfield, SurveyInfoScenario_cff::seed, and trajectoryStateTransform::transientState().

◆ updateTrajectory()

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

private

Member Data Documentation

◆ chi2cut

double CosmicTrajectoryBuilder::chi2cut

private

Definition at line 128 of file CosmicTrajectoryBuilder.h.

◆ geometry

std::string CosmicTrajectoryBuilder::geometry

private

Definition at line 133 of file CosmicTrajectoryBuilder.h.

◆ hitCloner

TkClonerImpl CosmicTrajectoryBuilder::hitCloner

private

Definition at line 123 of file CosmicTrajectoryBuilder.h.

◆ hits

TransientTrackingRecHit::RecHitContainer CosmicTrajectoryBuilder::hits

private

Definition at line 131 of file CosmicTrajectoryBuilder.h.

◆ magfield

edm::ESHandle<MagneticField> CosmicTrajectoryBuilder::magfield

private

Definition at line 115 of file CosmicTrajectoryBuilder.h.

◆ RHBuilder

const TransientTrackingRecHitBuilder* CosmicTrajectoryBuilder::RHBuilder

private

Definition at line 122 of file CosmicTrajectoryBuilder.h.

Referenced by hitBuilder().

◆ seed_plus

bool CosmicTrajectoryBuilder::seed_plus

private

Definition at line 132 of file CosmicTrajectoryBuilder.h.

◆ theBuilderName

std::string CosmicTrajectoryBuilder::theBuilderName

private

Definition at line 134 of file CosmicTrajectoryBuilder.h.

◆ theEstimator

Chi2MeasurementEstimator* CosmicTrajectoryBuilder::theEstimator

private

Definition at line 121 of file CosmicTrajectoryBuilder.h.

◆ theFitter

KFTrajectoryFitter* CosmicTrajectoryBuilder::theFitter

private

Definition at line 125 of file CosmicTrajectoryBuilder.h.

◆ theMinHits

int CosmicTrajectoryBuilder::theMinHits

private

Definition at line 127 of file CosmicTrajectoryBuilder.h.

◆ thePropagator

PropagatorWithMaterial* CosmicTrajectoryBuilder::thePropagator

private

Definition at line 118 of file CosmicTrajectoryBuilder.h.

◆ thePropagatorOp

PropagatorWithMaterial* CosmicTrajectoryBuilder::thePropagatorOp

private

Definition at line 119 of file CosmicTrajectoryBuilder.h.

◆ theSmoother

KFTrajectorySmoother* CosmicTrajectoryBuilder::theSmoother

private

Definition at line 124 of file CosmicTrajectoryBuilder.h.

◆ theUpdator

KFUpdator* CosmicTrajectoryBuilder::theUpdator

private

Definition at line 120 of file CosmicTrajectoryBuilder.h.

◆ tracker

edm::ESHandle<TrackerGeometry> CosmicTrajectoryBuilder::tracker

private

Definition at line 116 of file CosmicTrajectoryBuilder.h.

◆ trajFit

std::vector<Trajectory> CosmicTrajectoryBuilder::trajFit

private

Definition at line 129 of file CosmicTrajectoryBuilder.h.

Public Member Functions

Private Types

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ TM

◆ TSOS

Constructor & Destructor Documentation

◆ CosmicTrajectoryBuilder()

◆ ~CosmicTrajectoryBuilder()

Member Function Documentation

◆ AddHit()

◆ createStartingTrajectory()

◆ hitBuilder()

◆ init()

◆ qualityFilter()

◆ run()

◆ seedMeasurements()

◆ SortHits()

◆ startingTSOS()

◆ updateTrajectory()

Member Data Documentation

◆ chi2cut

◆ geometry

◆ hitCloner

◆ hits

◆ magfield

◆ RHBuilder

◆ seed_plus

◆ theBuilderName

◆ theEstimator

◆ theFitter

◆ theMinHits

◆ thePropagator

◆ thePropagatorOp

◆ theSmoother

◆ theUpdator

◆ tracker

◆ trajFit