CosmicTrajectoryBuilder Class Reference

#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

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,
		edm::ConsumesCollector	iC
	)

Definition at line 20 of file CosmicTrajectoryBuilder.cc.

References chi2cut, 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::~CosmicTrajectoryBuilder

(

)

Definition at line 34 of file CosmicTrajectoryBuilder.cc.

{}

Member Function Documentation

AddHit()

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, PropagatorWithMaterial::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);
    }
  }
}

createStartingTrajectory()

Trajectory CosmicTrajectoryBuilder::createStartingTrajectory

(

const TrajectorySeed &

seed

)

const

Definition at line 110 of file CosmicTrajectoryBuilder.cc.

References mps_fire::i, eostools::move(), mps_fire::result, fileCollector::seed, 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;
}

hitBuilder()

const TransientTrackingRecHitBuilder* CosmicTrajectoryBuilder::hitBuilder ( ) const

inline

Definition at line 96 of file CosmicTrajectoryBuilder.h.

References RHBuilder.

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

{ return RHBuilder; }

init()

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

qualityFilter()

bool CosmicTrajectoryBuilder::qualityFilter ( const Trajectory &  traj )

private

Definition at line 269 of file CosmicTrajectoryBuilder.cc.

References Trajectory::foundHits(), 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;
  }
}

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

seedMeasurements()

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

private

Definition at line 118 of file CosmicTrajectoryBuilder.cc.

References TransientTrackingRecHitBuilder::build(), rpcPointValidation_cfi::recHit, mps_fire::result, RHBuilder, fileCollector::seed, 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;
}

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

References TransientTrackingRecHitBuilder::build(), edmNew::DetSetVector< T >::data(), hits, LogDebug, rpcPointValidation_cfi::recHit, RHBuilder, fileCollector::seed, 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;
}

startingTSOS()

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

private

Definition at line 183 of file CosmicTrajectoryBuilder.cc.

References magfield, fileCollector::seed, 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;
}

updateTrajectory()

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

private

Member Data Documentation

builderToken_

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

private

Definition at line 117 of file CosmicTrajectoryBuilder.h.

Referenced by init().

chi2cut

double CosmicTrajectoryBuilder::chi2cut

private

Definition at line 131 of file CosmicTrajectoryBuilder.h.

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

geometry

std::string CosmicTrajectoryBuilder::geometry

private

Definition at line 136 of file CosmicTrajectoryBuilder.h.

hitCloner

TkClonerImpl CosmicTrajectoryBuilder::hitCloner

private

Definition at line 126 of file CosmicTrajectoryBuilder.h.

Referenced by init().

hits

TransientTrackingRecHit::RecHitContainer CosmicTrajectoryBuilder::hits

private

Definition at line 134 of file CosmicTrajectoryBuilder.h.

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

magfield

const MagneticField* CosmicTrajectoryBuilder::magfield

private

Definition at line 118 of file CosmicTrajectoryBuilder.h.

Referenced by init(), and startingTSOS().

magfieldToken_

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

private

Definition at line 115 of file CosmicTrajectoryBuilder.h.

Referenced by init().

RHBuilder

const TransientTrackingRecHitBuilder* CosmicTrajectoryBuilder::RHBuilder

private

Definition at line 125 of file CosmicTrajectoryBuilder.h.

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

seed_plus

bool CosmicTrajectoryBuilder::seed_plus

private

Definition at line 135 of file CosmicTrajectoryBuilder.h.

Referenced by init(), and SortHits().

theEstimator

Chi2MeasurementEstimator* CosmicTrajectoryBuilder::theEstimator

private

Definition at line 124 of file CosmicTrajectoryBuilder.h.

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

theFitter

KFTrajectoryFitter* CosmicTrajectoryBuilder::theFitter

private

Definition at line 128 of file CosmicTrajectoryBuilder.h.

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

theMinHits

int CosmicTrajectoryBuilder::theMinHits

private

Definition at line 130 of file CosmicTrajectoryBuilder.h.

Referenced by CosmicTrajectoryBuilder(), and qualityFilter().

thePropagator

PropagatorWithMaterial* CosmicTrajectoryBuilder::thePropagator

private

Definition at line 121 of file CosmicTrajectoryBuilder.h.

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

thePropagatorOp

PropagatorWithMaterial* CosmicTrajectoryBuilder::thePropagatorOp

private

Definition at line 122 of file CosmicTrajectoryBuilder.h.

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

theSmoother

KFTrajectorySmoother* CosmicTrajectoryBuilder::theSmoother

private

Definition at line 127 of file CosmicTrajectoryBuilder.h.

Referenced by init(), and run().

theUpdator

KFUpdator* CosmicTrajectoryBuilder::theUpdator

private

Definition at line 123 of file CosmicTrajectoryBuilder.h.

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

tracker

const TrackerGeometry* CosmicTrajectoryBuilder::tracker

private

Definition at line 119 of file CosmicTrajectoryBuilder.h.

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

trackerToken_

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

private

Definition at line 116 of file CosmicTrajectoryBuilder.h.

Referenced by init().

trajFit

std::vector<Trajectory> CosmicTrajectoryBuilder::trajFit

private

Definition at line 132 of file CosmicTrajectoryBuilder.h.

Referenced by AddHit(), and run().