FastTSGFromPropagation Class Reference

#include <FastTSGFromPropagation.h>

Inheritance diagram for FastTSGFromPropagation:

Classes
struct	increasingEstimate
struct	isInvalid

Public Member Functions
	FastTSGFromPropagation (const edm::ParameterSet &pset, edm::ConsumesCollector &iC)
	constructor More...
	FastTSGFromPropagation (const edm::ParameterSet &par, const MuonServiceProxy *, edm::ConsumesCollector &iC)
void	init (const MuonServiceProxy *) override
	initialize More...
void	setEvent (const edm::Event &) override
	set an event More...
void	trackerSeeds (const TrackCand &, const TrackingRegion &, const TrackerTopology *tTopo, std::vector< TrajectorySeed > &) override
	generate seed(s) for a track More...
	~FastTSGFromPropagation () override
	destructor More...
Public Member Functions inherited from TrackerSeedGenerator
const edm::Event *	getEvent () const
	TrackerSeedGenerator ()
virtual void	trackerSeeds (const TrackCand &, const TrackingRegion &, const TrackerTopology *, BTSeedCollection &)
virtual	~TrackerSeedGenerator ()
	destructor More...

Private Member Functions
void	adjust (FreeTrajectoryState &) const
	adjust the error matrix of the FTS More...
void	adjust (TrajectoryStateOnSurface &) const
	adjust the error matrix of the TSOS More...
TrajectorySeed	createSeed (const TrajectoryStateOnSurface &, const DetId &) const
	create a hitless seed from a trajectory state More...
TrajectorySeed	createSeed (const TrajectoryStateOnSurface &tsos, const edm::OwnVector< TrackingRecHit > &container, const DetId &id) const
	create a seed from a trajectory state More...
double	dxyDis (const TrajectoryStateOnSurface &tsos) const
const Chi2MeasurementEstimator *	estimator () const
std::vector< TrajectoryMeasurement >	findMeasurements (const DetLayer *, const TrajectoryStateOnSurface &) const
	look for measurements on the first compatible layer More...
std::vector< TrajectoryMeasurement >	findMeasurements_new (const DetLayer *, const TrajectoryStateOnSurface &) const
	look for measurements on the first compatible layer (faster way) More...
void	getRescalingFactor (const TrackCand &staMuon)
TrajectoryStateOnSurface	innerState (const TrackCand &) const
TrajectoryStateOnSurface	outerTkState (const TrackCand &) const
bool	passSelection (const TrajectoryStateOnSurface &) const
	check some quantity and beam-spot compatibility and decide to continue More...
edm::ESHandle< Propagator >	propagator () const
void	stateOnDet (const TrajectoryStateOnSurface &ts, unsigned int detid, PTrajectoryStateOnDet &pts) const
	A mere copy (without memory leak) of an existing tracking method. More...
const LayerMeasurements *	tkLayerMeasurements () const
const TrajectoryStateUpdator *	updator () const
void	validMeasurements (std::vector< TrajectoryMeasurement > &) const
	select valid measurements More...
double	zDis (const TrajectoryStateOnSurface &tsos) const

Private Attributes
edm::EDGetTokenT< reco::BeamSpot >	beamSpot_
edm::Handle< FastTrackerRecHitCombinationCollection >	recHitCombinations
edm::EDGetTokenT< FastTrackerRecHitCombinationCollection >	recHitCombinationsToken_
bool	theAdjustAtIp
edm::Handle< reco::BeamSpot >	theBeamSpot
unsigned long long	theCacheId_MT
unsigned long long	theCacheId_TG
std::string	theCategory
const edm::ParameterSet	theConfig
std::unique_ptr< MuonErrorMatrix >	theErrorMatrixAdjuster
std::unique_ptr< const Chi2MeasurementEstimator >	theEstimator
double	theFixedErrorRescaling
double	theFlexErrorRescaling
const TrackerGeometry *	theGeometry
double	theMaxChi2
edm::ESHandle< MeasurementTracker >	theMeasTracker
edm::Handle< MeasurementTrackerEvent >	theMeasTrackerEvent
edm::EDGetTokenT< MeasurementTrackerEvent >	theMeasurementTrackerEventToken_
std::unique_ptr< const DirectTrackerNavigation >	theNavigation
std::string	thePropagatorName
std::string	theResetMethod
bool	theSelectStateFlag
const MuonServiceProxy *	theService
double	theSigmaZ
edm::EDGetTokenT< edm::SimTrackContainer >	theSimTrackCollectionToken_
edm::Handle< edm::SimTrackContainer >	theSimTracks
LayerMeasurements	theTkLayerMeasurements
edm::ESHandle< GeometricSearchTracker >	theTracker
edm::ESHandle< TransientTrackingRecHitBuilder >	theTTRHBuilder
bool	theUpdateStateFlag
std::unique_ptr< const TrajectoryStateUpdator >	theUpdator
bool	theUseVertexStateFlag

Additional Inherited Members
Public Types inherited from TrackerSeedGenerator
typedef std::vector< TrajectorySeed >	BTSeedCollection
typedef std::pair< const Trajectory *, reco::TrackRef >	TrackCand
Protected Attributes inherited from TrackerSeedGenerator
const edm::Event *	theEvent
const MuonServiceProxy *	theProxyService

Detailed Description

Emulate TSGFromPropagation in RecoMuon

Author

Hwidong Yoo - Purdue University

Tracker Seed Generator by propagating and updating a standAlone muon to the first 2 (or 1) rechits it meets in tracker system

Emulate TSGFromPropagation in RecoMuon

Author

Hwidong Yoo - Purdue University

Definition at line 50 of file FastTSGFromPropagation.h.

Constructor & Destructor Documentation

FastTSGFromPropagation::FastTSGFromPropagation	(	const edm::ParameterSet &	pset,
		edm::ConsumesCollector &	iC
	)

constructor

Definition at line 46 of file FastTSGFromPropagation.cc.

                                                                                                      : FastTSGFromPropagation(iConfig, nullptr, iC) {
}

FastTSGFromPropagation::FastTSGFromPropagation	(	const edm::ParameterSet &	par,
		const MuonServiceProxy *	service,
		edm::ConsumesCollector &	iC
	)

Definition at line 49 of file FastTSGFromPropagation.cc.

                                                                                                                                        : 
  theCategory("FastSimulation|Muons|FastTSGFromPropagation"),
  theTkLayerMeasurements(), theTracker(), theNavigation(), theService(service), theUpdator(), theEstimator(), theSigmaZ(0.0), theConfig (iConfig),
  theSimTrackCollectionToken_(iC.consumes<edm::SimTrackContainer>(theConfig.getParameter<edm::InputTag>("SimTrackCollectionLabel"))),
  recHitCombinationsToken_(iC.consumes<FastTrackerRecHitCombinationCollection>(theConfig.getParameter<edm::InputTag>("HitProducer"))),
  beamSpot_(iC.consumes<reco::BeamSpot>(iConfig.getParameter<edm::InputTag>("beamSpot"))),
  theMeasurementTrackerEventToken_(iC.consumes<MeasurementTrackerEvent>(iConfig.getParameter<edm::InputTag>("MeasurementTrackerEvent"))) {
}

FastTSGFromPropagation::~FastTSGFromPropagation ( )

override

destructor

Definition at line 58 of file FastTSGFromPropagation.cc.

References LogTrace, and theCategory.

                                                {
  LogTrace(theCategory) << " FastTSGFromPropagation dtor called ";
}

Member Function Documentation

void FastTSGFromPropagation::adjust ( FreeTrajectoryState &  state ) const

private

adjust the error matrix of the FTS

Definition at line 505 of file FastTSGFromPropagation.cc.

References FreeTrajectoryState::curvilinearError(), FreeTrajectoryState::momentum(), MuonErrorMatrix::multiply(), FreeTrajectoryState::parameters(), FreeTrajectoryState::rescaleError(), theErrorMatrixAdjuster, theFixedErrorRescaling, theFlexErrorRescaling, and theResetMethod.

Referenced by innerState(), outerTkState(), and propagator().

                                                                     {

  //rescale the error
  if ( theResetMethod == "discreate" ) {
     state.rescaleError(theFlexErrorRescaling);
     return;
  }

  //rescale the error
  if ( theResetMethod == "fixed" || !theErrorMatrixAdjuster) {
     state.rescaleError(theFixedErrorRescaling);
     return;
  }

  CurvilinearTrajectoryError oMat = state.curvilinearError();
  CurvilinearTrajectoryError sfMat = theErrorMatrixAdjuster->get(state.momentum());//FIXME with position
  MuonErrorMatrix::multiply(oMat, sfMat);
  
  state = FreeTrajectoryState(state.parameters(),
                  oMat);
}

void FastTSGFromPropagation::adjust ( TrajectoryStateOnSurface &  state ) const

private

adjust the error matrix of the TSOS

Definition at line 527 of file FastTSGFromPropagation.cc.

References TrajectoryStateOnSurface::curvilinearError(), TrajectoryStateOnSurface::globalMomentum(), TrajectoryStateOnSurface::globalParameters(), MuonErrorMatrix::multiply(), TrajectoryStateOnSurface::rescaleError(), TrajectoryStateOnSurface::surface(), TrajectoryStateOnSurface::surfaceSide(), theErrorMatrixAdjuster, theFixedErrorRescaling, theFlexErrorRescaling, theResetMethod, and TrajectoryStateOnSurface::weight().

                                                                          {

  //rescale the error
  if ( theResetMethod == "discreate" ) {
     state.rescaleError(theFlexErrorRescaling);
     return;
  }

  if ( theResetMethod == "fixed" || !theErrorMatrixAdjuster) {
     state.rescaleError(theFixedErrorRescaling);
     return;
  }

  CurvilinearTrajectoryError oMat = state.curvilinearError();
  CurvilinearTrajectoryError sfMat = theErrorMatrixAdjuster->get(state.globalMomentum());//FIXME with position
  MuonErrorMatrix::multiply(oMat, sfMat);
  
  state = TrajectoryStateOnSurface(state.weight(),
                                   state.globalParameters(),
                   oMat,
                   state.surface(),
                   state.surfaceSide());
}

TrajectorySeed FastTSGFromPropagation::createSeed ( const TrajectoryStateOnSurface &  tsos, const DetId &  id  ) const

private

create a hitless seed from a trajectory state

Definition at line 418 of file FastTSGFromPropagation.cc.

Referenced by propagator(), and trackerSeeds().

                                                                                                             {

  edm::OwnVector<TrackingRecHit> container;
  return createSeed(tsos, container, id);

}

TrajectorySeed FastTSGFromPropagation::createSeed ( const TrajectoryStateOnSurface &  tsos, const edm::OwnVector< TrackingRecHit > &  container, const DetId &  id  ) const

private

create a seed from a trajectory state

Definition at line 425 of file FastTSGFromPropagation.cc.

References oppositeToMomentum, and trajectoryStateTransform::persistentState().

                                                                                                                                                            {

  PTrajectoryStateOnDet seedTSOS = trajectoryStateTransform::persistentState(tsos,id.rawId());
  return TrajectorySeed(seedTSOS,container,oppositeToMomentum);

}

double FastTSGFromPropagation::dxyDis ( const TrajectoryStateOnSurface &  tsos ) const

private

Definition at line 488 of file FastTSGFromPropagation.cc.

References TrajectoryStateOnSurface::globalMomentum(), TrajectoryStateOnSurface::globalPosition(), PV3DBase< T, PVType, FrameType >::perp(), PV3DBase< T, PVType, FrameType >::x(), and PV3DBase< T, PVType, FrameType >::y().

Referenced by propagator().

                                                                                {
  return fabs(( - tsos.globalPosition().x() * tsos.globalMomentum().y() + tsos.globalPosition().y() * tsos.globalMomentum().x() )/tsos.globalMomentum().perp());
}

const Chi2MeasurementEstimator* FastTSGFromPropagation::estimator ( void  ) const

inlineprivate

Definition at line 84 of file FastTSGFromPropagation.h.

References theEstimator.

Referenced by findMeasurements(), findMeasurements_new(), and trackerSeeds().

{ return theEstimator.get(); }

std::vector< TrajectoryMeasurement > FastTSGFromPropagation::findMeasurements ( const DetLayer *  nl, const TrajectoryStateOnSurface &  staState  ) const

private

look for measurements on the first compatible layer

Definition at line 466 of file FastTSGFromPropagation.cc.

References estimator(), LayerMeasurements::measurements(), propagator(), mps_fire::result, tkLayerMeasurements(), and validMeasurements().

Referenced by propagator().

                                                                                                                                            {

  std::vector<TrajectoryMeasurement> result = tkLayerMeasurements()->measurements((*nl), staState, *propagator(), *estimator());
  validMeasurements(result);
  return result;
}

std::vector< TrajectoryMeasurement > FastTSGFromPropagation::findMeasurements_new ( const DetLayer *  nl, const TrajectoryStateOnSurface &  staState  ) const

private

look for measurements on the first compatible layer (faster way)

Definition at line 441 of file FastTSGFromPropagation.cc.

References GeometricSearchDet::compatibleDets(), estimator(), propagator(), mps_fire::result, and theMeasTrackerEvent.

Referenced by propagator(), and trackerSeeds().

                                                                                                                                                {

  std::vector<TrajectoryMeasurement> result;

  std::vector<DetLayer::DetWithState> compatDets = nl->compatibleDets(staState, *propagator(), *estimator());
  if ( compatDets.empty() )  return result;

  for (std::vector<DetLayer::DetWithState>::const_iterator idws = compatDets.begin(); idws != compatDets.end(); ++idws) {
     if ( idws->second.isValid() && (idws->first) )  {
         std::vector<TrajectoryMeasurement> tmptm = 
           theMeasTrackerEvent->idToDet(idws->first->geographicalId()).fastMeasurements(idws->second, idws->second, *propagator(), *estimator());
         //validMeasurements(tmptm);
//         if ( tmptm.size() > 2 ) {
//            std::stable_sort(tmptm.begin(),tmptm.end(),increasingEstimate());
//            result.insert(result.end(),tmptm.begin(), tmptm.begin()+2);
//         } else {
            result.insert(result.end(),tmptm.begin(), tmptm.end());
//         }
     }
  }
  
  return result;

}

void FastTSGFromPropagation::getRescalingFactor ( const TrackCand &  staMuon )

private

Definition at line 496 of file FastTSGFromPropagation.cc.

References EnergyCorrector::pt, and theFlexErrorRescaling.

Referenced by propagator(), and trackerSeeds().

                                                                        {
    float pt = (staMuon.second)->pt();
    if ( pt < 13.0 ) theFlexErrorRescaling = 3; 
    else if ( pt < 30.0 ) theFlexErrorRescaling = 5;
    else theFlexErrorRescaling = 10;
    return;
}

void FastTSGFromPropagation::init ( const MuonServiceProxy *  service )

overridevirtual

initialize

Reimplemented from TrackerSeedGenerator.

Definition at line 284 of file FastTSGFromPropagation.cc.

References Chi2MeasurementEstimator_cfi::Chi2MeasurementEstimator, TSGForRoadSearch_cfi::errorMatrixPset, geometry, edm::ParameterSet::getParameter(), AlCaHLTBitMon_QueryRunRegistry::string, theAdjustAtIp, theCacheId_MT, theCacheId_TG, theConfig, theErrorMatrixAdjuster, theEstimator, theFixedErrorRescaling, theFlexErrorRescaling, theGeometry, theMaxChi2, theNavigation, thePropagatorName, theResetMethod, theSelectStateFlag, theService, theSigmaZ, theTracker, theTTRHBuilder, theUpdateStateFlag, theUpdator, and theUseVertexStateFlag.

                                                                 {

  theMaxChi2 = theConfig.getParameter<double>("MaxChi2");

  theFixedErrorRescaling = theConfig.getParameter<double>("ErrorRescaling");

  theFlexErrorRescaling = 1.0;

  theResetMethod = theConfig.getParameter<std::string>("ResetMethod");

  if (theResetMethod != "discrete" && theResetMethod != "fixed" && theResetMethod != "matrix"  ) {
    edm::LogError("FastTSGFromPropagation") 
      <<"Wrong error rescaling method: "<<theResetMethod <<"\n"
      <<"Possible choices are: discrete, fixed, matrix.\n"
      <<"Use discrete method" <<std::endl;
    theResetMethod = "discrete"; 
  }

  theEstimator.reset( new Chi2MeasurementEstimator(theMaxChi2));

  theCacheId_MT = 0;

  theCacheId_TG = 0;

  thePropagatorName = theConfig.getParameter<std::string>("Propagator");

  theService = service;

  theUseVertexStateFlag = theConfig.getParameter<bool>("UseVertexState");

  theUpdateStateFlag = theConfig.getParameter<bool>("UpdateState");

  theSelectStateFlag = theConfig.getParameter<bool>("SelectState");

  theUpdator.reset(new KFUpdator());

  theSigmaZ = theConfig.getParameter<double>("SigmaZ");

  edm::ParameterSet errorMatrixPset = theConfig.getParameter<edm::ParameterSet>("errorMatrixPset");
  if ( theResetMethod == "matrix" && !errorMatrixPset.empty()){
    theAdjustAtIp = errorMatrixPset.getParameter<bool>("atIP");
    theErrorMatrixAdjuster.reset(new MuonErrorMatrix(errorMatrixPset));
  } else {
    theAdjustAtIp =false;
    theErrorMatrixAdjuster.reset();
  }

  theService->eventSetup().get<TrackerRecoGeometryRecord>().get(theTracker); 
  theNavigation.reset(new DirectTrackerNavigation(theTracker));

  edm::ESHandle<TrackerGeometry>        geometry;
  theService->eventSetup().get<TrackerDigiGeometryRecord>().get(geometry);
  theGeometry = &(*geometry);

  theService->eventSetup().get<TransientRecHitRecord>().get("WithTrackAngle", theTTRHBuilder);

}

TrajectoryStateOnSurface FastTSGFromPropagation::innerState ( const TrackCand &  staMuon ) const

private

Definition at line 379 of file FastTSGFromPropagation.cc.

References adjust(), alongMomentum, trajectoryStateTransform::innerStateOnSurface(), oppositeToMomentum, and theService.

Referenced by outerTkState().

                                                                                          {

  TrajectoryStateOnSurface innerTS;

  if ( staMuon.first && staMuon.first->isValid() ) {
    if (staMuon.first->direction() == alongMomentum) {
      innerTS = staMuon.first->firstMeasurement().updatedState();
    } 
    else if (staMuon.first->direction() == oppositeToMomentum) { 
      innerTS = staMuon.first->lastMeasurement().updatedState();
    }
  } else {
    innerTS = trajectoryStateTransform::innerStateOnSurface(*(staMuon.second),*theService->trackingGeometry(), &*theService->magneticField());
  }
  //rescale the error
  adjust(innerTS);

  return  innerTS;

}

TrajectoryStateOnSurface FastTSGFromPropagation::outerTkState ( const TrackCand &  staMuon ) const

private

Definition at line 400 of file FastTSGFromPropagation.cc.

References adjust(), trajectoryStateTransform::initialFreeState(), innerState(), propagator(), mps_fire::result, theService, and theUseVertexStateFlag.

Referenced by trackerSeeds().

                                                                                            {

  TrajectoryStateOnSurface result;

  if ( theUseVertexStateFlag && staMuon.second->pt() > 1.0 ) {
    FreeTrajectoryState iniState = trajectoryStateTransform::initialFreeState(*(staMuon.second), &*theService->magneticField());
    //rescale the error at IP
    adjust(iniState); 

    StateOnTrackerBound fromInside(&*(theService->propagator("PropagatorWithMaterial")));
    result = fromInside(iniState);
  } else {
    StateOnTrackerBound fromOutside(&*propagator());
    result = fromOutside(innerState(staMuon));
  }
  return result;
}

bool FastTSGFromPropagation::passSelection ( const TrajectoryStateOnSurface &  tsos ) const

private

check some quantity and beam-spot compatibility and decide to continue

Definition at line 473 of file FastTSGFromPropagation.cc.

References edm::HandleBase::isValid(), theBeamSpot, theSelectStateFlag, theSigmaZ, reco::BeamSpot::z0(), and zDis().

Referenced by propagator(), and trackerSeeds().

                                                                                     {
  if ( !theSelectStateFlag ) return true;
  else {
     if ( theBeamSpot.isValid() ) {
       return ( ( fabs(zDis(tsos) - theBeamSpot->z0() ) < theSigmaZ) );

     } else {
       return ( ( fabs(zDis(tsos)) < theSigmaZ) ); 
//      double theDxyCut = 100;
//      return ( (zDis(tsos) < theSigmaZ) && (dxyDis(tsos) < theDxyCut) );
     }
  }

}

edm::ESHandle<Propagator> FastTSGFromPropagation::propagator ( ) const

inlineprivate

Definition at line 86 of file FastTSGFromPropagation.h.

References adjust(), createSeed(), dxyDis(), findMeasurements(), findMeasurements_new(), getRescalingFactor(), passSelection(), MuonServiceProxy::propagator(), thePropagatorName, theService, validMeasurements(), and zDis().

Referenced by findMeasurements(), findMeasurements_new(), outerTkState(), and trackerSeeds().

{return theService->propagator(thePropagatorName); }

void FastTSGFromPropagation::setEvent ( const edm::Event &  iEvent )

overridevirtual

set an event

Reimplemented from TrackerSeedGenerator.

Definition at line 342 of file FastTSGFromPropagation.cc.

References beamSpot_, edm::Event::getByToken(), LogTrace, edm::ESHandle< T >::product(), recHitCombinations, recHitCombinationsToken_, theBeamSpot, theCacheId_MT, theCacheId_TG, theCategory, theMeasTracker, theMeasTrackerEvent, theMeasurementTrackerEventToken_, theNavigation, theService, theSimTrackCollectionToken_, theSimTracks, theTkLayerMeasurements, theTracker, and theUpdateStateFlag.

                                                            {

  iEvent.getByToken(beamSpot_, theBeamSpot);
  
  // retrieve the MC truth (SimTracks)
  iEvent.getByToken(theSimTrackCollectionToken_, theSimTracks);
  iEvent.getByToken(recHitCombinationsToken_, recHitCombinations);

  unsigned long long newCacheId_MT = theService->eventSetup().get<CkfComponentsRecord>().cacheIdentifier();

  if ( theUpdateStateFlag && newCacheId_MT != theCacheId_MT ) {
    LogTrace(theCategory) << "Measurment Tracker Geometry changed!";
    theCacheId_MT = newCacheId_MT;
    theService->eventSetup().get<CkfComponentsRecord>().get(theMeasTracker);
  }

  if ( theUpdateStateFlag ) {
     iEvent.getByToken(theMeasurementTrackerEventToken_, theMeasTrackerEvent);
     theTkLayerMeasurements = LayerMeasurements(*theMeasTracker,*theMeasTrackerEvent);
  }

  bool trackerGeomChanged = false;

  unsigned long long newCacheId_TG = theService->eventSetup().get<TrackerRecoGeometryRecord>().cacheIdentifier();

  if ( newCacheId_TG != theCacheId_TG ) {
    LogTrace(theCategory) << "Tracker Reco Geometry changed!";
    theCacheId_TG = newCacheId_TG;
    theService->eventSetup().get<TrackerRecoGeometryRecord>().get(theTracker);
    trackerGeomChanged = true;
  }

  if ( trackerGeomChanged && theTracker.product() ) {
    theNavigation.reset(new DirectTrackerNavigation(theTracker));
  }
}

void FastTSGFromPropagation::stateOnDet ( const TrajectoryStateOnSurface &  ts, unsigned int  detid, PTrajectoryStateOnDet &  pts  ) const

private

A mere copy (without memory leak) of an existing tracking method.

should check if corresponds to m

Definition at line 551 of file FastTSGFromPropagation.cc.

References TrajectoryStateOnSurface::globalMomentum(), mps_fire::i, gen::k, TrajectoryStateOnSurface::localError(), TrajectoryStateOnSurface::localParameters(), funct::m, LocalTrajectoryError::matrix(), PV3DBase< T, PVType, FrameType >::perp(), and TrajectoryStateOnSurface::surfaceSide().

{
    const AlgebraicSymMatrix55& m = ts.localError().matrix();
    int dim = 5; 
    float localErrors[15];
    int k = 0;
    for (int i=0; i<dim; ++i) {
    for (int j=0; j<=i; ++j) {
        localErrors[k++] = m(i,j);
    }
    }
    int surfaceSide = static_cast<int>(ts.surfaceSide());
    pts = PTrajectoryStateOnDet( ts.localParameters(),ts.globalMomentum().perp(),
                    localErrors, detid,
                surfaceSide);
}

const LayerMeasurements* FastTSGFromPropagation::tkLayerMeasurements ( ) const

inlineprivate

Definition at line 80 of file FastTSGFromPropagation.h.

References theTkLayerMeasurements.

Referenced by findMeasurements().

{ return &theTkLayerMeasurements; } 

void FastTSGFromPropagation::trackerSeeds ( const TrackCand &  staMuon, const TrackingRegion &  region, const TrackerTopology *  tTopo, std::vector< TrajectorySeed > &  result  )

override

generate seed(s) for a track

Definition at line 62 of file FastTSGFromPropagation.cc.

References PV3DBase< T, PVType, FrameType >::basicVector(), TransientTrackingRecHitBuilder::build(), CoreSimTrack::charge(), ALCARECOTkAlJpsiMuMu_cff::charge, FastTrackerRecHit::clone(), createSeed(), PTrajectoryStateOnDet::detId(), estimator(), PV3DBase< T, PVType, FrameType >::eta(), findMeasurements_new(), TrajectoryStateOnSurface::freeState(), g, TrackingRecHit::geographicalId(), getRescalingFactor(), TrajectoryStateOnSurface::globalMomentum(), TrajectoryStateOnSurface::globalPosition(), TrajectorySeedHitCandidate::hit(), triggerObjects_cff::id, TrackerGeometry::idToDet(), createfilelist::int, TrajectoryStateOnSurface::isValid(), LogTrace, nullptr, oppositeToMomentum, outerTkState(), passSelection(), PV3DBase< T, PVType, FrameType >::perp(), position, propagator(), edm::OwnVector< T, P >::push_back(), DetId::rawId(), rpcPointValidation_cfi::recHit, recHitCombinations, fastTrackingUtilities::setRecHitCombinationIndex(), tkConvValidator_cfi::simTracks, GeomDet::surface(), theCategory, theGeometry, theNavigation, theResetMethod, theService, theTTRHBuilder, theUpdateStateFlag, SimTrack::trackerSurfaceMomentum(), SimTrack::trackerSurfacePosition(), trajectoryStateTransform::transientState(), TrajectoryStateUpdator::update(), updator(), and PV3DBase< T, PVType, FrameType >::y().

                                                                                        {

  if ( theResetMethod == "discrete" ) getRescalingFactor(staMuon);

  TrajectoryStateOnSurface staState = outerTkState(staMuon);

  if ( !staState.isValid() ) { 
    LogTrace(theCategory) << "Error: initial state from L2 muon is invalid.";
    return;
  }

  LogTrace(theCategory) << "begin of trackerSeed:\n staState pos: "<<staState.globalPosition()
                        << " mom: "<<staState.globalMomentum() 
                        <<"pos eta: "<<staState.globalPosition().eta()
                        <<"mom eta: "<<staState.globalMomentum().eta();


  std::vector<const DetLayer*> nls = theNavigation->compatibleLayers(*(staState.freeState()), oppositeToMomentum);

  LogTrace(theCategory) << " compatible layers: "<<nls.size();

  if ( nls.empty() ) return;

  int ndesLayer = 0;

  bool usePredictedState = false;

  if ( theUpdateStateFlag ) { //use updated states
     std::vector<TrajectoryMeasurement> alltm; 

     for (std::vector<const DetLayer*>::const_iterator inl = nls.begin();
         inl != nls.end(); inl++, ndesLayer++ ) {
         if ( (*inl == nullptr) ) break;
//         if ( (inl != nls.end()-1 ) && ( (*inl)->subDetector() == GeomDetEnumerators::TEC ) && ( (*(inl+1))->subDetector() == GeomDetEnumerators::TOB ) ) continue; 
         alltm = findMeasurements_new(*inl, staState);
         if ( (!alltm.empty()) ) {
            LogTrace(theCategory) << "final compatible layer: "<<ndesLayer;
            break;
         }
     }

     if ( alltm.empty() ) {
        LogTrace(theCategory) << " NO Measurements Found: eta: "<<staState.globalPosition().eta() <<"pt "<<staState.globalMomentum().perp();
        usePredictedState = true;
     } else {
       LogTrace(theCategory) << " Measurements for seeds: "<<alltm.size();
       std::stable_sort(alltm.begin(),alltm.end(),increasingEstimate()); 
       if ( alltm.size() > 5 ) alltm.erase(alltm.begin() + 5, alltm.end());

       const edm::SimTrackContainer* simTracks = &(*theSimTracks);
       TrajectorySeedHitCandidate theSeedHits;
       std::vector<TrajectorySeedHitCandidate> outerHits;

       //std::vector<TrajectorySeed>  tmpTS;
       bool isMatch = false;
       for (std::vector<TrajectoryMeasurement>::const_iterator itm = alltm.begin(); itm != alltm.end(); itm++) {
       const TrajectoryStateOnSurface seedState = itm->predictedState();
       double preY = seedState.globalPosition().y();

       // Check SimTrack
       FreeTrajectoryState simtrack_trackerstate;
       for( unsigned icomb = 0;icomb < recHitCombinations->size();++icomb){
           const auto & recHitCombination = (*recHitCombinations)[icomb];
           if(recHitCombination.empty())
             continue;
           int32_t simTrackId = recHitCombination.back()->simTrackId(0);
           const SimTrack & simtrack = (*simTracks)[simTrackId];
           
           GlobalPoint position(simtrack.trackerSurfacePosition().x(),
                            simtrack.trackerSurfacePosition().y(),
                            simtrack.trackerSurfacePosition().z());
           GlobalVector momentum(simtrack.trackerSurfaceMomentum().x(),
                         simtrack.trackerSurfaceMomentum().y(),
                         simtrack.trackerSurfaceMomentum().z());
           int charge = (int)simtrack.charge();
           GlobalTrajectoryParameters glb_parameters(position, momentum, charge, &*theService->magneticField().product());
           simtrack_trackerstate = FreeTrajectoryState(glb_parameters);

           unsigned int outerId = 0;
           for( const auto & recHitRef : recHitCombination) {
           theSeedHits = TrajectorySeedHitCandidate(recHitRef.get(), tTopo);
           unsigned int id = theSeedHits.hit()->geographicalId().rawId();
           if( preY < 0 ) {
               if( id > outerId ) outerId = id;
           }
           else {
               if( id > outerId ) outerId = id;
           }
           }
           for( const auto & recHitRef : recHitCombination ) {
           theSeedHits = TrajectorySeedHitCandidate(recHitRef.get(),tTopo);
           if( itm->recHit()->hit()->geographicalId().rawId() == theSeedHits.hit()->geographicalId().rawId() ) {
                   auto aTrackingRecHit = std::unique_ptr<TrackingRecHit>(theSeedHits.hit()->clone());
                   TransientTrackingRecHit::ConstRecHitPointer recHit = theTTRHBuilder->build(aTrackingRecHit.get());
                   if( !recHit ) continue;
                   TrajectoryStateOnSurface updatedTSOS = updator()->update(seedState, *(recHit));
                   if( updatedTSOS.isValid() && passSelection(updatedTSOS) ) {
               edm::OwnVector<TrackingRecHit> container;
               container.push_back(recHit->hit()->clone());
               fastTrackingUtilities::setRecHitCombinationIndex(container,icomb);
               TrajectorySeed ts = createSeed(updatedTSOS, container, recHit->geographicalId());
               // check direction
               const TrajectorySeed* aSeed = &ts;
               PTrajectoryStateOnDet PTSOD = aSeed->startingState();
               
               const GeomDet *g = theGeometry->idToDet(PTSOD.detId());
               TrajectoryStateOnSurface tsos = trajectoryStateTransform::transientState(PTSOD, &(g->surface()),  &*theService->magneticField().product());
               if( tsos.globalMomentum().basicVector()*seedState.globalMomentum().basicVector() < 0. ) continue;
               result.push_back(ts);
               isMatch = true;
               }
           }
           }
       }
       }
       if( !isMatch ) {
     // if there is no hits w.r.t. TM, find outermost hit
         for (std::vector<TrajectoryMeasurement>::const_iterator itm = alltm.begin(); itm != alltm.end(); itm++) {
       const TrajectoryStateOnSurface seedState = itm->predictedState();
       double preY = seedState.globalPosition().y();

       // Check SimTrack
       TrackingRecHit* aTrackingRecHit;
       FreeTrajectoryState simtrack_trackerstate;

       for( unsigned icomb = 0;icomb < recHitCombinations->size();++icomb){
           const auto & recHitCombination = (*recHitCombinations)[icomb];
           if(recHitCombination.empty())
             continue;
           int32_t simTrackId = recHitCombination.back()->simTrackId(0);
           const SimTrack & simtrack = (*simTracks)[simTrackId];

           GlobalPoint position(simtrack.trackerSurfacePosition().x(),
                            simtrack.trackerSurfacePosition().y(),
                            simtrack.trackerSurfacePosition().z());
           GlobalVector momentum(simtrack.trackerSurfaceMomentum().x(),
                         simtrack.trackerSurfaceMomentum().y(),
                         simtrack.trackerSurfaceMomentum().z());
           int charge = (int)simtrack.charge();
           GlobalTrajectoryParameters glb_parameters(position, momentum, charge, &*theService->magneticField().product());
           simtrack_trackerstate = FreeTrajectoryState(glb_parameters);

           unsigned int outerId = 0;
           for( const auto & recHitRef : recHitCombination ) {
           theSeedHits = TrajectorySeedHitCandidate(recHitRef.get(),tTopo);
           unsigned int id = theSeedHits.hit()->geographicalId().rawId();
           if( preY < 0 ) {
               if( id > outerId ) outerId = id;
           }
           else {
               if( id > outerId ) outerId = id;
           }
           }
           for( const auto & recHitRef : recHitCombination ) {
           theSeedHits = TrajectorySeedHitCandidate(recHitRef.get(),tTopo);
           if( outerId == theSeedHits.hit()->geographicalId().rawId() ) {
               aTrackingRecHit = theSeedHits.hit()->clone();
                   TransientTrackingRecHit::ConstRecHitPointer recHit = theTTRHBuilder->build(aTrackingRecHit);
                   if( !recHit ) continue;
                   TrajectoryStateOnSurface updatedTSOS = updator()->update(seedState, *(recHit));
                   if( updatedTSOS.isValid() && passSelection(updatedTSOS) ) {
               edm::OwnVector<TrackingRecHit> container;
               container.push_back(recHit->hit()->clone());
               fastTrackingUtilities::setRecHitCombinationIndex(container,icomb);
               TrajectorySeed ts = createSeed(updatedTSOS, container, recHit->geographicalId());
               // check direction
               const TrajectorySeed* aSeed = &ts;
               PTrajectoryStateOnDet PTSOD = aSeed->startingState();
               
               const GeomDet *g = theGeometry->idToDet(PTSOD.detId());
               TrajectoryStateOnSurface tsos = trajectoryStateTransform::transientState(PTSOD, &(g->surface()),  &*theService->magneticField().product());
               if( tsos.globalMomentum().basicVector()*seedState.globalMomentum().basicVector() < 0. ) continue;
               result.push_back(ts);
               }
           }
           }
       }
         }
       }

       /*
       for( unsigned ir = 0; ir < tmpTS.size(); ir++ ) {
     const BasicTrajectorySeed* aSeed = &((tmpTS)[ir]);
     PTrajectoryStateOnDet PTSOD = aSeed->startingState();
     
     DetId seedDetId(PTSOD.detId());
     const GeomDet * g = theGeometry->idToDet(seedDetId);
     TrajectoryStateOnSurface tsos = trajectoryStateTransform::transientState(PTSOD, &(g->surface()),  &*theService->magneticField().product());
         cout << "tsos3 = " << tsos.globalMomentum() << endl;
     if( _index == ir ) {
         cout << "tsos4 = " << tsos.globalMomentum() << endl;
         result.push_back(tmpTS[ir]);
     }
       }
       */
       LogTrace(theCategory) << "result: "<<result.size();
       return;
     }
  }
  
  if ( !theUpdateStateFlag || usePredictedState ) { //use predicted states
     LogTrace(theCategory) << "use predicted state: ";
     for (std::vector<const DetLayer*>::const_iterator inl = nls.begin();
         inl != nls.end(); inl++ ) {

         if ( !result.empty() || *inl == nullptr ) {
            break;
         }
         std::vector<DetLayer::DetWithState> compatDets = (*inl)->compatibleDets(staState, *propagator(), *estimator());
         LogTrace(theCategory) << " compatDets "<<compatDets.size();
         if ( compatDets.empty() ) continue;
         TrajectorySeed ts = createSeed(compatDets.front().second, compatDets.front().first->geographicalId());
         result.push_back(ts);

     }
     LogTrace(theCategory) << "result: "<<result.size();
     return;
  } 
  return;
}

const TrajectoryStateUpdator* FastTSGFromPropagation::updator ( ) const

inlineprivate

Definition at line 82 of file FastTSGFromPropagation.h.

References theUpdator.

Referenced by trackerSeeds().

{return theUpdator.get();}

void FastTSGFromPropagation::validMeasurements ( std::vector< TrajectoryMeasurement > &  tms ) const

private

select valid measurements

Definition at line 433 of file FastTSGFromPropagation.cc.

Referenced by findMeasurements(), and propagator().

                                                                                          {

  std::vector<TrajectoryMeasurement>::iterator tmsend = std::remove_if(tms.begin(), tms.end(), isInvalid());
  tms.erase(tmsend, tms.end());
  return;

}

double FastTSGFromPropagation::zDis ( const TrajectoryStateOnSurface &  tsos ) const

private

Definition at line 492 of file FastTSGFromPropagation.cc.

References TrajectoryStateOnSurface::globalMomentum(), TrajectoryStateOnSurface::globalPosition(), PV3DBase< T, PVType, FrameType >::perp(), and PV3DBase< T, PVType, FrameType >::z().

Referenced by passSelection(), and propagator().

                                                                              {
  return tsos.globalPosition().z() - tsos.globalPosition().perp() * tsos.globalMomentum().z()/tsos.globalMomentum().perp();
}

Member Data Documentation

edm::EDGetTokenT<reco::BeamSpot> FastTSGFromPropagation::beamSpot_

private

Definition at line 177 of file FastTSGFromPropagation.h.

Referenced by setEvent().

edm::Handle<FastTrackerRecHitCombinationCollection> FastTSGFromPropagation::recHitCombinations

private

Definition at line 182 of file FastTSGFromPropagation.h.

Referenced by setEvent(), and trackerSeeds().

edm::EDGetTokenT<FastTrackerRecHitCombinationCollection> FastTSGFromPropagation::recHitCombinationsToken_

private

Definition at line 176 of file FastTSGFromPropagation.h.

Referenced by setEvent().

bool FastTSGFromPropagation::theAdjustAtIp

private

Definition at line 170 of file FastTSGFromPropagation.h.

Referenced by init().

edm::Handle<reco::BeamSpot> FastTSGFromPropagation::theBeamSpot

private

Definition at line 180 of file FastTSGFromPropagation.h.

Referenced by passSelection(), and setEvent().

unsigned long long FastTSGFromPropagation::theCacheId_MT

private

Definition at line 131 of file FastTSGFromPropagation.h.

Referenced by init(), and setEvent().

unsigned long long FastTSGFromPropagation::theCacheId_TG

private

Definition at line 132 of file FastTSGFromPropagation.h.

Referenced by init(), and setEvent().

std::string FastTSGFromPropagation::theCategory

private

Definition at line 134 of file FastTSGFromPropagation.h.

Referenced by setEvent(), trackerSeeds(), and ~FastTSGFromPropagation().

const edm::ParameterSet FastTSGFromPropagation::theConfig

private

Definition at line 174 of file FastTSGFromPropagation.h.

Referenced by init().

std::unique_ptr<MuonErrorMatrix> FastTSGFromPropagation::theErrorMatrixAdjuster

private

Definition at line 168 of file FastTSGFromPropagation.h.

Referenced by adjust(), and init().

std::unique_ptr<const Chi2MeasurementEstimator> FastTSGFromPropagation::theEstimator

private

Definition at line 150 of file FastTSGFromPropagation.h.

Referenced by estimator(), and init().

double FastTSGFromPropagation::theFixedErrorRescaling

private

Definition at line 156 of file FastTSGFromPropagation.h.

Referenced by adjust(), and init().

double FastTSGFromPropagation::theFlexErrorRescaling

private

Definition at line 154 of file FastTSGFromPropagation.h.

Referenced by adjust(), getRescalingFactor(), and init().

const TrackerGeometry* FastTSGFromPropagation::theGeometry

private

Definition at line 144 of file FastTSGFromPropagation.h.

Referenced by init(), and trackerSeeds().

double FastTSGFromPropagation::theMaxChi2

private

Definition at line 152 of file FastTSGFromPropagation.h.

Referenced by init().

edm::ESHandle<MeasurementTracker> FastTSGFromPropagation::theMeasTracker

private

Definition at line 140 of file FastTSGFromPropagation.h.

Referenced by setEvent().

edm::Handle<MeasurementTrackerEvent> FastTSGFromPropagation::theMeasTrackerEvent

private

Definition at line 183 of file FastTSGFromPropagation.h.

Referenced by findMeasurements_new(), and setEvent().

edm::EDGetTokenT<MeasurementTrackerEvent> FastTSGFromPropagation::theMeasurementTrackerEventToken_

private

Definition at line 178 of file FastTSGFromPropagation.h.

Referenced by setEvent().

std::unique_ptr<const DirectTrackerNavigation> FastTSGFromPropagation::theNavigation

private

Definition at line 142 of file FastTSGFromPropagation.h.

Referenced by init(), setEvent(), and trackerSeeds().

std::string FastTSGFromPropagation::thePropagatorName

private

Definition at line 166 of file FastTSGFromPropagation.h.

Referenced by init(), and propagator().

std::string FastTSGFromPropagation::theResetMethod

private

Definition at line 162 of file FastTSGFromPropagation.h.

Referenced by adjust(), init(), and trackerSeeds().

bool FastTSGFromPropagation::theSelectStateFlag

private

Definition at line 164 of file FastTSGFromPropagation.h.

Referenced by init(), and passSelection().

const MuonServiceProxy* FastTSGFromPropagation::theService

private

Definition at line 146 of file FastTSGFromPropagation.h.

Referenced by init(), innerState(), outerTkState(), propagator(), setEvent(), and trackerSeeds().

double FastTSGFromPropagation::theSigmaZ

private

Definition at line 172 of file FastTSGFromPropagation.h.

Referenced by init(), and passSelection().

edm::EDGetTokenT<edm::SimTrackContainer> FastTSGFromPropagation::theSimTrackCollectionToken_

private

Definition at line 175 of file FastTSGFromPropagation.h.

Referenced by setEvent().

edm::Handle<edm::SimTrackContainer> FastTSGFromPropagation::theSimTracks

private

Definition at line 181 of file FastTSGFromPropagation.h.

Referenced by setEvent().

LayerMeasurements FastTSGFromPropagation::theTkLayerMeasurements

private

Definition at line 136 of file FastTSGFromPropagation.h.

Referenced by setEvent(), and tkLayerMeasurements().

edm::ESHandle<GeometricSearchTracker> FastTSGFromPropagation::theTracker

private

Definition at line 138 of file FastTSGFromPropagation.h.

Referenced by init(), and setEvent().

edm::ESHandle<TransientTrackingRecHitBuilder> FastTSGFromPropagation::theTTRHBuilder

private

Definition at line 184 of file FastTSGFromPropagation.h.

Referenced by init(), and trackerSeeds().

bool FastTSGFromPropagation::theUpdateStateFlag

private

Definition at line 160 of file FastTSGFromPropagation.h.

Referenced by init(), setEvent(), and trackerSeeds().

std::unique_ptr<const TrajectoryStateUpdator> FastTSGFromPropagation::theUpdator

private

Definition at line 148 of file FastTSGFromPropagation.h.

Referenced by init(), and updator().

bool FastTSGFromPropagation::theUseVertexStateFlag

private

Definition at line 158 of file FastTSGFromPropagation.h.

Referenced by init(), and outerTkState().