TrackDetectorAssociator Class Reference

#include <TrackDetectorAssociator.h>

Public Types
typedef TrackAssociatorParameters	AssociatorParameters
enum	Direction { Any, InsideOut, OutsideIn }

Public Member Functions
TrackDetMatchInfo	associate (const edm::Event &, const edm::EventSetup &, const FreeTrajectoryState &, const AssociatorParameters &)
TrackDetMatchInfo	associate (const edm::Event &iEvent, const edm::EventSetup &iSetup, const AssociatorParameters &parameters, const FreeTrajectoryState *innerState, const FreeTrajectoryState *outerState=nullptr)
TrackDetMatchInfo	associate (const edm::Event &, const edm::EventSetup &, const reco::Track &, const AssociatorParameters &, Direction direction=Any)
	associate using reco::Track More...
TrackDetMatchInfo	associate (const edm::Event &, const edm::EventSetup &, const SimTrack &, const SimVertex &, const AssociatorParameters &)
	associate using a simulated track More...
TrackDetMatchInfo	associate (const edm::Event &, const edm::EventSetup &, const GlobalVector &, const GlobalPoint &, const int, const AssociatorParameters &)
	associate using 3-momentum, vertex and charge More...
const CachedTrajectory &	getCachedTrajector () const
	trajector information More...
void	setPropagator (const Propagator *)
	use a user configured propagator More...
	TrackDetectorAssociator ()
void	useDefaultPropagator ()
	use the default propagator More...
	~TrackDetectorAssociator ()

Static Public Member Functions
static bool	crossedIP (const reco::Track &track)
static FreeTrajectoryState	getFreeTrajectoryState (const MagneticField *, const reco::Track &)
	get FreeTrajectoryState from different track representations More...
static FreeTrajectoryState	getFreeTrajectoryState (const MagneticField *, const SimTrack &, const SimVertex &)
static FreeTrajectoryState	getFreeTrajectoryState (const MagneticField *, const GlobalVector &, const GlobalPoint &, const int)

Private Member Functions
bool	addTAMuonSegmentMatch (TAMuonChamberMatch &, const RecSegment *, const AssociatorParameters &)
void	fillCaloTowers (const edm::Event &, TrackDetMatchInfo &, const AssociatorParameters &)
void	fillCaloTruth (const edm::Event &, TrackDetMatchInfo &, const AssociatorParameters &)
void	fillEcal (const edm::Event &, TrackDetMatchInfo &, const AssociatorParameters &)
void	fillHcal (const edm::Event &, TrackDetMatchInfo &, const AssociatorParameters &)
void	fillHO (const edm::Event &, TrackDetMatchInfo &, const AssociatorParameters &)
void	fillMuon (const edm::Event &, TrackDetMatchInfo &, const AssociatorParameters &)
void	fillPreshower (const edm::Event &iEvent, TrackDetMatchInfo &info, const AssociatorParameters &)
DetIdAssociator::MapRange	getMapRange (const std::pair< float, float > &delta, const float dR)
math::XYZPoint	getPoint (const GlobalPoint &point)
math::XYZPoint	getPoint (const LocalPoint &point)
void	getTAMuonChamberMatches (std::vector< TAMuonChamberMatch > &matches, const AssociatorParameters &parameters, std::set< DetId > occupancy)
math::XYZVector	getVector (const GlobalVector &vec)
math::XYZVector	getVector (const LocalVector &vec)
void	init (const edm::EventSetup &, const AssociatorParameters &)

Private Attributes
CachedTrajectory	cachedTrajectory_
const DetIdAssociator *	caloDetIdAssociator_
std::unique_ptr< Propagator >	defProp_
const DetIdAssociator *	ecalDetIdAssociator_
const DetIdAssociator *	hcalDetIdAssociator_
const DetIdAssociator *	hoDetIdAssociator_
const Propagator *	ivProp_
const DetIdAssociator *	muonDetIdAssociator_
const DetIdAssociator *	preshowerDetIdAssociator_
const CaloGeometry *	theCaloGeometry_
edm::ESWatcher< IdealMagneticFieldRecord >	theMagneticFieldWatcher_
const GlobalTrackingGeometry *	theTrackingGeometry_
bool	useDefaultPropagator_

Detailed Description

Definition at line 44 of file TrackDetectorAssociator.h.

Member Typedef Documentation

AssociatorParameters

typedef TrackAssociatorParameters TrackDetectorAssociator::AssociatorParameters

Definition at line 49 of file TrackDetectorAssociator.h.

Member Enumeration Documentation

Direction

enum TrackDetectorAssociator::Direction

Enumerator
Any
InsideOut
OutsideIn

Definition at line 50 of file TrackDetectorAssociator.h.

{ Any, InsideOut, OutsideIn };

Constructor & Destructor Documentation

TrackDetectorAssociator()

TrackDetectorAssociator::TrackDetectorAssociator ( )

explicit

Definition at line 73 of file TrackDetectorAssociator.cc.

                                                 {
  ivProp_ = nullptr;
  useDefaultPropagator_ = false;
}

~TrackDetectorAssociator()

TrackDetectorAssociator::~TrackDetectorAssociator ( )

default

Member Function Documentation

addTAMuonSegmentMatch()

bool TrackDetectorAssociator::addTAMuonSegmentMatch ( TAMuonChamberMatch &  matchedChamber, const RecSegment *  segment, const AssociatorParameters &  parameters  )

private

Definition at line 868 of file TrackDetectorAssociator.cc.

References funct::abs(), relativeConstraints::chamber, spr::deltaEta, reco::deltaPhi(), PV3DBase< T, PVType, FrameType >::eta(), f, TrajectoryStateOnSurface::freeState(), TrackingRecHit::geographicalId(), TAMuonSegmentMatch::hasPhi, DTRecSegment4D::hasPhi(), TAMuonSegmentMatch::hasZed, DTRecSegment4D::hasZed(), hfClusterShapes_cfi::hits, TAMuonChamberMatch::id, DTRecSegment2D::ist0Valid(), RecSegment::localDirection(), RecSegment::localDirectionError(), TrackingRecHit::localPosition(), TrackingRecHit::localPositionError(), LogTrace, M_PI, PV3DBase< T, PVType, FrameType >::perp(), PV3DBase< T, PVType, FrameType >::phi(), DTRecSegment4D::phiSegment(), FreeTrajectoryState::position(), DetId::rawId(), TAMuonSegmentMatch::segmentGlobalPosition, TAMuonSegmentMatch::segmentLocalDirection, TAMuonSegmentMatch::segmentLocalErrorDxDz, TAMuonSegmentMatch::segmentLocalErrorDyDz, TAMuonSegmentMatch::segmentLocalErrorXDxDz, TAMuonSegmentMatch::segmentLocalErrorXX, TAMuonSegmentMatch::segmentLocalErrorXY, TAMuonSegmentMatch::segmentLocalErrorYDyDz, TAMuonSegmentMatch::segmentLocalErrorYY, TAMuonSegmentMatch::segmentLocalPosition, TAMuonChamberMatch::segments, DTRecSegment2D::specificRecHits(), TAMuonSegmentMatch::t0, DTRecSegment2D::t0(), FrontierCondition_GT_autoExpress_cfi::t0, TAMuonChamberMatch::tState, LocalError::xx(), LocalError::xy(), and LocalError::yy().

                                                                                            {
  LogTrace("TrackAssociator") << "Segment local position: " << segment->localPosition() << "\n"
                              << std::hex << segment->geographicalId().rawId() << "\n";

  const GeomDet* chamber = muonDetIdAssociator_->getGeomDet(matchedChamber.id);
  TrajectoryStateOnSurface trajectoryStateOnSurface = matchedChamber.tState;
  GlobalPoint segmentGlobalPosition = chamber->toGlobal(segment->localPosition());

  LogTrace("TrackAssociator") << "Segment global position: " << segmentGlobalPosition
                              << " \t (R_xy,eta,phi): " << segmentGlobalPosition.perp() << ","
                              << segmentGlobalPosition.eta() << "," << segmentGlobalPosition.phi() << "\n";

  LogTrace("TrackAssociator") << "\teta hit: " << segmentGlobalPosition.eta()
                              << " \tpropagator: " << trajectoryStateOnSurface.freeState()->position().eta() << "\n"
                              << "\tphi hit: " << segmentGlobalPosition.phi()
                              << " \tpropagator: " << trajectoryStateOnSurface.freeState()->position().phi()
                              << std::endl;

  bool isGood = false;
  bool isDTWithoutY = false;
  const DTRecSegment4D* dtseg = dynamic_cast<const DTRecSegment4D*>(segment);
  if (dtseg && (!dtseg->hasZed()))
    isDTWithoutY = true;

  float deltaPhi(std::abs(segmentGlobalPosition.phi() - trajectoryStateOnSurface.freeState()->position().phi()));
  if (deltaPhi > float(M_PI))
    deltaPhi = std::abs(deltaPhi - float(M_PI) * 2.f);
  float deltaEta = std::abs(segmentGlobalPosition.eta() - trajectoryStateOnSurface.freeState()->position().eta());

  if (isDTWithoutY) {
    isGood = deltaPhi < parameters.dRMuon;
    // Be in chamber
    isGood &= deltaEta < .3f;
  } else
    isGood = deltaEta * deltaEta + deltaPhi * deltaPhi < parameters.dRMuon * parameters.dRMuon;

  if (isGood) {
    TAMuonSegmentMatch muonSegment;
    muonSegment.segmentGlobalPosition = getPoint(segmentGlobalPosition);
    muonSegment.segmentLocalPosition = getPoint(segment->localPosition());
    muonSegment.segmentLocalDirection = getVector(segment->localDirection());
    muonSegment.segmentLocalErrorXX = segment->localPositionError().xx();
    muonSegment.segmentLocalErrorYY = segment->localPositionError().yy();
    muonSegment.segmentLocalErrorXY = segment->localPositionError().xy();
    muonSegment.segmentLocalErrorDxDz = segment->localDirectionError().xx();
    muonSegment.segmentLocalErrorDyDz = segment->localDirectionError().yy();

    // DANGEROUS - compiler cannot guaranty parameters ordering
    // AlgebraicSymMatrix segmentCovMatrix = segment->parametersError();
    // muonSegment.segmentLocalErrorXDxDz = segmentCovMatrix[2][0];
    // muonSegment.segmentLocalErrorYDyDz = segmentCovMatrix[3][1];
    muonSegment.segmentLocalErrorXDxDz = -999.f;
    muonSegment.segmentLocalErrorYDyDz = -999.f;
    muonSegment.hasZed = true;
    muonSegment.hasPhi = true;

    // timing information
    muonSegment.t0 = 0.f;
    if (dtseg) {
      if ((dtseg->hasPhi()) && (!isDTWithoutY)) {
        int phiHits = dtseg->phiSegment()->specificRecHits().size();
        //    int zHits = dtseg->zSegment()->specificRecHits().size();
        int hits = 0;
        double t0 = 0.;
        // TODO: cuts on hit numbers not optimized in any way yet...
        if (phiHits > 5 && dtseg->phiSegment()->ist0Valid()) {
          t0 += dtseg->phiSegment()->t0() * phiHits;
          hits += phiHits;
          LogTrace("TrackAssociator") << " Phi t0: " << dtseg->phiSegment()->t0() << " hits: " << phiHits;
        }
        // the z segments seem to contain little useful information...
        //    if (zHits>3) {
        //      t0+=s->zSegment()->t0()*zHits;
        //      hits+=zHits;
        //      LogTrace("TrackAssociator") << "   Z t0: " << s->zSegment()->t0() << " hits: " << zHits << std::endl;
        //    }
        if (hits)
          muonSegment.t0 = t0 / hits;
        //    LogTrace("TrackAssociator") << " --- t0: " << muonSegment.t0 << std::endl;
      } else {
        // check and set dimensionality
        if (isDTWithoutY)
          muonSegment.hasZed = false;
        if (!dtseg->hasPhi())
          muonSegment.hasPhi = false;
      }
    }
    matchedChamber.segments.push_back(muonSegment);
  }

  return isGood;
}

associate() [1/5]

TrackDetMatchInfo TrackDetectorAssociator::associate	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		const FreeTrajectoryState &	fts,
		const AssociatorParameters &	parameters
	)

propagate a track across the whole detector and find associated objects. Association is done in two modes 1) an object is associated to a track only if crossed by track 2) an object is associated to a track if it is withing an eta-phi cone of some radius with respect to a track. (the cone origin is at (0,0,0)) Trajectory bending in eta-phi is taking into account when matching is performed

associate using FreeTrajectoryState

Definition at line 114 of file TrackDetectorAssociator.cc.

References iEvent.

Referenced by EopTreeWriter::analyze(), ValidIsoTrkCalib::analyze(), EcalCosmicsHists::analyze(), muonisolation::CaloExtractorByAssociator::deposits(), cms::MuonMETValueMapProducer::determine_deltax_deltay(), MuonIdProducer::fillMuonId(), and InterestingTrackEcalDetIdProducer::produce().

                                                                                             {
  return associate(iEvent, iSetup, parameters, &fts);
}

associate() [2/5]

TrackDetMatchInfo TrackDetectorAssociator::associate	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		const AssociatorParameters &	parameters,
		const FreeTrajectoryState *	innerState,
		const FreeTrajectoryState *	outerState = nullptr
	)

associate using inner and outer most states of a track in the silicon tracker.

Definition at line 121 of file TrackDetectorAssociator.cc.

References funct::abs(), Exception, iEvent, info(), reco::init(), edm::isNotFinite(), LogTrace, PV3DBase< T, PVType, FrameType >::mag(), CosmicsPD_Skims::maxZ, CosmicGenFilterHelix_cff::minZ, SteppingHelixStateInfo::momentum(), or, PV3DBase< T, PVType, FrameType >::perp(), FreeTrajectoryState::position(), PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

                                                                                            {
  TrackDetMatchInfo info;
  if (!parameters.useEcal && !parameters.useCalo && !parameters.useHcal && !parameters.useHO && !parameters.useMuon &&
      !parameters.usePreshower)
    throw cms::Exception("ConfigurationError")
        << "Configuration error! No subdetector was selected for the track association.";

  SteppingHelixStateInfo trackOrigin(*innerState);
  info.stateAtIP = *innerState;
  cachedTrajectory_.setStateAtIP(trackOrigin);

  init(iSetup, parameters);
  // get track trajectory
  // ECAL points (EB+EE)
  // If the phi angle between a track entrance and exit points is more
  // than 2 crystals, it is possible that the track will cross 3 crystals
  // and therefore one has to check at least 3 points along the track
  // trajectory inside ECAL. In order to have a chance to cross 4 crystalls
  // in the barrel, a track should have P_t as low as 3 GeV or smaller
  // If it's necessary, number of points along trajectory can be increased

  info.setCaloGeometry(theCaloGeometry_);

  cachedTrajectory_.reset_trajectory();
  // estimate propagation outer boundaries based on
  // requested sub-detector information. For now limit
  // propagation region only if muon matching is not
  // requested.
  double HOmaxR = hoDetIdAssociator_->volume().maxR();
  double HOmaxZ = hoDetIdAssociator_->volume().maxZ();
  double minR = ecalDetIdAssociator_->volume().minR();
  double minZ = preshowerDetIdAssociator_->volume().minZ();
  cachedTrajectory_.setMaxHORadius(HOmaxR);
  cachedTrajectory_.setMaxHOLength(HOmaxZ * 2.);
  cachedTrajectory_.setMinDetectorRadius(minR);
  cachedTrajectory_.setMinDetectorLength(minZ * 2.);

  if (parameters.useMuon) {
    double maxR = muonDetIdAssociator_->volume().maxR();
    double maxZ = muonDetIdAssociator_->volume().maxZ();
    cachedTrajectory_.setMaxDetectorRadius(maxR);
    cachedTrajectory_.setMaxDetectorLength(maxZ * 2.);
  } else {
    cachedTrajectory_.setMaxDetectorRadius(HOmaxR);
    cachedTrajectory_.setMaxDetectorLength(HOmaxZ * 2.);
  }

  // If track extras exist and outerState is before HO maximum, then use outerState
  if (outerState) {
    if (outerState->position().perp() < HOmaxR && std::abs(outerState->position().z()) < HOmaxZ) {
      LogTrace("TrackAssociator") << "Using outerState as trackOrigin at Rho=" << outerState->position().perp()
                                  << "  Z=" << outerState->position().z() << "\n";
      trackOrigin = SteppingHelixStateInfo(*outerState);
    } else if (innerState) {
      LogTrace("TrackAssociator") << "Using innerState as trackOrigin at Rho=" << innerState->position().perp()
                                  << "  Z=" << innerState->position().z() << "\n";
      trackOrigin = SteppingHelixStateInfo(*innerState);
    }
  }

  if (trackOrigin.momentum().mag() == 0)
    return info;
  if (edm::isNotFinite(trackOrigin.momentum().x()) or edm::isNotFinite(trackOrigin.momentum().y()) or
      edm::isNotFinite(trackOrigin.momentum().z()))
    return info;
  if (!cachedTrajectory_.propagateAll(trackOrigin))
    return info;

  // get trajectory in calorimeters
  cachedTrajectory_.findEcalTrajectory(ecalDetIdAssociator_->volume());
  cachedTrajectory_.findHcalTrajectory(hcalDetIdAssociator_->volume());
  cachedTrajectory_.findHOTrajectory(hoDetIdAssociator_->volume());
  cachedTrajectory_.findPreshowerTrajectory(preshowerDetIdAssociator_->volume());

  info.trkGlobPosAtEcal = getPoint(cachedTrajectory_.getStateAtEcal().position());
  info.trkGlobPosAtHcal = getPoint(cachedTrajectory_.getStateAtHcal().position());
  info.trkGlobPosAtHO = getPoint(cachedTrajectory_.getStateAtHO().position());

  info.trkMomAtEcal = cachedTrajectory_.getStateAtEcal().momentum();
  info.trkMomAtHcal = cachedTrajectory_.getStateAtHcal().momentum();
  info.trkMomAtHO = cachedTrajectory_.getStateAtHO().momentum();

  if (parameters.useEcal)
    fillEcal(iEvent, info, parameters);
  if (parameters.useCalo)
    fillCaloTowers(iEvent, info, parameters);
  if (parameters.useHcal)
    fillHcal(iEvent, info, parameters);
  if (parameters.useHO)
    fillHO(iEvent, info, parameters);
  if (parameters.usePreshower)
    fillPreshower(iEvent, info, parameters);
  if (parameters.useMuon)
    fillMuon(iEvent, info, parameters);
  if (parameters.truthMatch)
    fillCaloTruth(iEvent, info, parameters);

  return info;
}

associate() [3/5]

TrackDetMatchInfo TrackDetectorAssociator::associate	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		const reco::Track &	track,
		const AssociatorParameters &	parameters,
		Direction	direction = Any
	)

associate using reco::Track

Definition at line 1032 of file TrackDetectorAssociator.cc.

References funct::abs(), Calorimetry_cff::bField, Exception, edm::EventSetup::getData(), iEvent, trajectoryStateTransform::initialFreeState(), trajectoryStateTransform::innerFreeState(), LogTrace, trajectoryStateTransform::outerFreeState(), PV3DBase< T, PVType, FrameType >::perp(), PV3DBase< T, PVType, FrameType >::phi(), FreeTrajectoryState::position(), mps_fire::result, HLT_2023v12_cff::track, and PV3DBase< T, PVType, FrameType >::z().

                                                                            {
  double currentStepSize = cachedTrajectory_.getPropagationStep();

  const MagneticField* bField = &iSetup.getData(parameters.bFieldToken);

  if (track.extra().isNull()) {
    if (direction != InsideOut)
      throw cms::Exception("FatalError") << "No TrackExtra information is available and association is done with "
                                            "something else than InsideOut track.\n"
                                         << "Either change the parameter or provide needed data!\n";
    LogTrace("TrackAssociator") << "Track Extras not found\n";
    FreeTrajectoryState initialState = trajectoryStateTransform::initialFreeState(track, bField);
    return associate(iEvent, iSetup, parameters, &initialState);  // 5th argument is null pointer
  }

  LogTrace("TrackAssociator") << "Track Extras found\n";
  FreeTrajectoryState innerState = trajectoryStateTransform::innerFreeState(track, bField);
  FreeTrajectoryState outerState = trajectoryStateTransform::outerFreeState(track, bField);
  FreeTrajectoryState referenceState = trajectoryStateTransform::initialFreeState(track, bField);

  LogTrace("TrackAssociator") << "inner track state (rho, z, phi):" << track.innerPosition().Rho() << ", "
                              << track.innerPosition().z() << ", " << track.innerPosition().phi() << "\n";
  LogTrace("TrackAssociator") << "innerFreeState (rho, z, phi):" << innerState.position().perp() << ", "
                              << innerState.position().z() << ", " << innerState.position().phi() << "\n";

  LogTrace("TrackAssociator") << "outer track state (rho, z, phi):" << track.outerPosition().Rho() << ", "
                              << track.outerPosition().z() << ", " << track.outerPosition().phi() << "\n";
  LogTrace("TrackAssociator") << "outerFreeState (rho, z, phi):" << outerState.position().perp() << ", "
                              << outerState.position().z() << ", " << outerState.position().phi() << "\n";

  // InsideOut first
  if (crossedIP(track)) {
    switch (direction) {
      case InsideOut:
      case Any:
        return associate(iEvent, iSetup, parameters, &referenceState, &outerState);
        break;
      case OutsideIn: {
        cachedTrajectory_.setPropagationStep(-std::abs(currentStepSize));
        TrackDetMatchInfo result = associate(iEvent, iSetup, parameters, &innerState, &referenceState);
        cachedTrajectory_.setPropagationStep(currentStepSize);
        return result;
        break;
      }
    }
  } else {
    switch (direction) {
      case InsideOut:
        return associate(iEvent, iSetup, parameters, &innerState, &outerState);
        break;
      case OutsideIn: {
        cachedTrajectory_.setPropagationStep(-std::abs(currentStepSize));
        TrackDetMatchInfo result = associate(iEvent, iSetup, parameters, &outerState, &innerState);
        cachedTrajectory_.setPropagationStep(currentStepSize);
        return result;
        break;
      }
      case Any: {
        // check if we deal with clear outside-in case
        if (track.innerPosition().Dot(track.innerMomentum()) < 0 &&
            track.outerPosition().Dot(track.outerMomentum()) < 0) {
          cachedTrajectory_.setPropagationStep(-std::abs(currentStepSize));
          TrackDetMatchInfo result;
          if (track.innerPosition().Mag2() < track.outerPosition().Mag2())
            result = associate(iEvent, iSetup, parameters, &innerState, &outerState);
          else
            result = associate(iEvent, iSetup, parameters, &outerState, &innerState);
          cachedTrajectory_.setPropagationStep(currentStepSize);
          return result;
        }
      }
    }
  }

  // all other cases
  return associate(iEvent, iSetup, parameters, &innerState, &outerState);
}

associate() [4/5]

TrackDetMatchInfo TrackDetectorAssociator::associate	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		const SimTrack &	track,
		const SimVertex &	vertex,
		const AssociatorParameters &	parameters
	)

associate using a simulated track

Definition at line 1114 of file TrackDetectorAssociator.cc.

References Calorimetry_cff::bField, edm::EventSetup::getData(), iEvent, HLT_2023v12_cff::track, and bphysicsOniaDQM_cfi::vertex.

                                                                                             {
  auto const* bField = &iSetup.getData(parameters.bFieldToken);
  return associate(iEvent, iSetup, getFreeTrajectoryState(bField, track, vertex), parameters);
}

associate() [5/5]

TrackDetMatchInfo TrackDetectorAssociator::associate	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		const GlobalVector &	momentum,
		const GlobalPoint &	vertex,
		const int	charge,
		const AssociatorParameters &	parameters
	)

associate using 3-momentum, vertex and charge

Definition at line 1123 of file TrackDetectorAssociator.cc.

References Calorimetry_cff::bField, reco::charge(), edm::EventSetup::getData(), iEvent, and bphysicsOniaDQM_cfi::vertex.

                                                                                             {
  auto const* bField = &iSetup.getData(parameters.bFieldToken);
  return associate(iEvent, iSetup, getFreeTrajectoryState(bField, momentum, vertex, charge), parameters);
}

crossedIP()

bool TrackDetectorAssociator::crossedIP ( const reco::Track &  track )

static

Definition at line 1133 of file TrackDetectorAssociator.cc.

References HLT_2023v12_cff::track.

Referenced by MuonIdProducer::produce().

                                                              {
  bool crossed = true;
  crossed &= (track.innerPosition().rho() > 3);  // something close to active volume
  crossed &= (track.outerPosition().rho() > 3);  // something close to active volume
  crossed &=
      ((track.innerPosition().x() * track.innerMomentum().x() + track.innerPosition().y() * track.innerMomentum().y() <
        0) !=
       (track.outerPosition().x() * track.outerMomentum().x() + track.outerPosition().y() * track.outerMomentum().y() <
        0));
  return crossed;
}

fillCaloTowers()

void TrackDetectorAssociator::fillCaloTowers ( const edm::Event &  iEvent, TrackDetMatchInfo &  info, const AssociatorParameters &  parameters  )

private

Definition at line 299 of file TrackDetectorAssociator.cc.

References eleHcalExtractorBlocks_cff::caloTowers, HLT_2023v12_cff::distance, iEvent, info(), CachedTrajectory::IpToHcal, LogTrace, and l1tHGCalTowerProducer_cfi::tower.

                                                                                     {
  // use ECAL and HCAL trajectories to match a tower. (HO isn't used for matching).
  std::vector<GlobalPoint> trajectory;
  const std::vector<SteppingHelixStateInfo>& ecalTrajectoryStates = cachedTrajectory_.getEcalTrajectory();
  const std::vector<SteppingHelixStateInfo>& hcalTrajectoryStates = cachedTrajectory_.getHcalTrajectory();
  for (std::vector<SteppingHelixStateInfo>::const_iterator itr = ecalTrajectoryStates.begin();
       itr != ecalTrajectoryStates.end();
       itr++)
    trajectory.push_back(itr->position());
  for (std::vector<SteppingHelixStateInfo>::const_iterator itr = hcalTrajectoryStates.begin();
       itr != hcalTrajectoryStates.end();
       itr++)
    trajectory.push_back(itr->position());

  if (trajectory.empty()) {
    LogTrace("TrackAssociator") << "HCAL trajectory is empty; moving on\n";
    info.isGoodCalo = false;
    return;
  }
  info.isGoodCalo = true;

  // find crossed CaloTowers
  edm::Handle<CaloTowerCollection> caloTowers;
  iEvent.getByToken(parameters.caloTowersToken, caloTowers);
  if (!caloTowers.isValid())
    throw cms::Exception("FatalError") << "Unable to find CaloTowers in event!\n";

  std::set<DetId> caloTowerIdsInRegion;
  if (parameters.accountForTrajectoryChangeCalo) {
    // get trajectory change with respect to initial state
    DetIdAssociator::MapRange mapRange =
        getMapRange(cachedTrajectory_.trajectoryDelta(CachedTrajectory::IpToHcal), parameters.dRHcalPreselection);
    caloTowerIdsInRegion = caloDetIdAssociator_->getDetIdsCloseToAPoint(trajectory[0], mapRange);
  } else
    caloTowerIdsInRegion = caloDetIdAssociator_->getDetIdsCloseToAPoint(trajectory[0], parameters.dRHcalPreselection);

  LogTrace("TrackAssociator") << "Towers in the region: " << caloTowerIdsInRegion.size();

  auto caloTowerIdsInAConeBegin = caloTowerIdsInRegion.begin();
  auto caloTowerIdsInAConeEnd = caloTowerIdsInRegion.end();
  std::set<DetId> caloTowerIdsInAConeTmp;
  if (!caloDetIdAssociator_->selectAllInACone(parameters.dRHcal)) {
    caloTowerIdsInAConeTmp =
        caloDetIdAssociator_->getDetIdsInACone(caloTowerIdsInRegion, trajectory, parameters.dRHcal);
    caloTowerIdsInAConeBegin = caloTowerIdsInAConeTmp.begin();
    caloTowerIdsInAConeEnd = caloTowerIdsInAConeTmp.end();
  }
  LogTrace("TrackAssociator") << "Towers in the cone: "
                              << std::distance(caloTowerIdsInAConeBegin, caloTowerIdsInAConeEnd);

  info.crossedTowerIds = caloDetIdAssociator_->getCrossedDetIds(caloTowerIdsInRegion, trajectory);
  const std::vector<DetId>& crossedCaloTowerIds = info.crossedTowerIds;
  LogTrace("TrackAssociator") << "Towers crossed: " << crossedCaloTowerIds.size();

  // add CaloTowers
  for (std::vector<DetId>::const_iterator itr = crossedCaloTowerIds.begin(); itr != crossedCaloTowerIds.end(); itr++) {
    CaloTowerCollection::const_iterator tower = (*caloTowers).find(*itr);
    if (tower != (*caloTowers).end())
      info.crossedTowers.push_back(&*tower);
    else
      LogTrace("TrackAssociator") << "Crossed CaloTower is not found for DetId: " << (*itr).rawId();
  }

  for (std::set<DetId>::const_iterator itr = caloTowerIdsInAConeBegin; itr != caloTowerIdsInAConeEnd; itr++) {
    CaloTowerCollection::const_iterator tower = (*caloTowers).find(*itr);
    if (tower != (*caloTowers).end())
      info.towers.push_back(&*tower);
    else
      LogTrace("TrackAssociator") << "CaloTower from the cone is not found for DetId: " << (*itr).rawId();
  }
}

fillCaloTruth()

void TrackDetectorAssociator::fillCaloTruth ( const edm::Event &  iEvent, TrackDetMatchInfo &  info, const AssociatorParameters &  parameters  )

private

Definition at line 965 of file TrackDetectorAssociator.cc.

References funct::abs(), electronAnalyzer_cfi::DeltaR, f, iEvent, info(), edm::HandleBase::isValid(), cscDigiValidation_cfi::simTrack, TrackCandidateProducer_cfi::simTracks, and HGCalValidator_cfi::simVertices.

                                                                                    {
  // get list of simulated tracks and their vertices
  using namespace edm;
  Handle<SimTrackContainer> simTracks;
  iEvent.getByToken(parameters.simTracksToken, simTracks);
  if (!simTracks.isValid())
    throw cms::Exception("FatalError") << "No simulated tracks found\n";

  Handle<SimVertexContainer> simVertices;
  iEvent.getByToken(parameters.simVerticesToken, simVertices);
  if (!simVertices.isValid())
    throw cms::Exception("FatalError") << "No simulated vertices found\n";

  // get sim calo hits
  Handle<PCaloHitContainer> simEcalHitsEB;
  iEvent.getByToken(parameters.simEcalHitsEBToken, simEcalHitsEB);
  if (!simEcalHitsEB.isValid())
    throw cms::Exception("FatalError") << "No simulated ECAL EB hits found\n";

  Handle<PCaloHitContainer> simEcalHitsEE;
  iEvent.getByToken(parameters.simEcalHitsEEToken, simEcalHitsEE);
  if (!simEcalHitsEE.isValid())
    throw cms::Exception("FatalError") << "No simulated ECAL EE hits found\n";

  Handle<PCaloHitContainer> simHcalHits;
  iEvent.getByToken(parameters.simHcalHitsToken, simHcalHits);
  if (!simHcalHits.isValid())
    throw cms::Exception("FatalError") << "No simulated HCAL hits found\n";

  // find truth partner
  SimTrackContainer::const_iterator simTrack = simTracks->begin();
  for (; simTrack != simTracks->end(); ++simTrack) {
    math::XYZVector simP3(simTrack->momentum().x(), simTrack->momentum().y(), simTrack->momentum().z());
    math::XYZVector recoP3(info.stateAtIP.momentum().x(), info.stateAtIP.momentum().y(), info.stateAtIP.momentum().z());
    if (ROOT::Math::VectorUtil::DeltaR(recoP3, simP3) < 0.1)
      break;
  }
  if (simTrack != simTracks->end()) {
    info.simTrack = &(*simTrack);
    float ecalTrueEnergy(0);
    float hcalTrueEnergy(0);

    // loop over calo hits
    for (PCaloHitContainer::const_iterator hit = simEcalHitsEB->begin(); hit != simEcalHitsEB->end(); ++hit)
      if (hit->geantTrackId() == info.simTrack->genpartIndex())
        ecalTrueEnergy += hit->energy();

    for (PCaloHitContainer::const_iterator hit = simEcalHitsEE->begin(); hit != simEcalHitsEE->end(); ++hit)
      if (hit->geantTrackId() == info.simTrack->genpartIndex())
        ecalTrueEnergy += hit->energy();

    for (PCaloHitContainer::const_iterator hit = simHcalHits->begin(); hit != simHcalHits->end(); ++hit)
      if (hit->geantTrackId() == info.simTrack->genpartIndex())
        hcalTrueEnergy += hit->energy();

    info.ecalTrueEnergy = ecalTrueEnergy;
    info.hcalTrueEnergy = hcalTrueEnergy;
    info.hcalTrueEnergyCorrected = hcalTrueEnergy;
    if (std::abs(info.trkGlobPosAtHcal.eta()) < 1.3f)
      info.hcalTrueEnergyCorrected = hcalTrueEnergy * 113.2f;
    else if (std::abs(info.trkGlobPosAtHcal.eta()) < 3.0f)
      info.hcalTrueEnergyCorrected = hcalTrueEnergy * 167.2f;
  }
}

fillEcal()

void TrackDetectorAssociator::fillEcal ( const edm::Event &  iEvent, TrackDetMatchInfo &  info, const AssociatorParameters &  parameters  )

private

Definition at line 225 of file TrackDetectorAssociator.cc.

References filterRecHits_cfi::EBRecHits, filterRecHits_cfi::EERecHits, iEvent, info(), CachedTrajectory::IpToEcal, and LogTrace.

                                                                               {
  const std::vector<SteppingHelixStateInfo>& trajectoryStates = cachedTrajectory_.getEcalTrajectory();

  for (std::vector<SteppingHelixStateInfo>::const_iterator itr = trajectoryStates.begin();
       itr != trajectoryStates.end();
       itr++)
    LogTrace("TrackAssociator") << "ECAL trajectory point (rho, z, phi): " << itr->position().perp() << ", "
                                << itr->position().z() << ", " << itr->position().phi();

  std::vector<GlobalPoint> coreTrajectory;
  for (std::vector<SteppingHelixStateInfo>::const_iterator itr = trajectoryStates.begin();
       itr != trajectoryStates.end();
       itr++)
    coreTrajectory.push_back(itr->position());

  if (coreTrajectory.empty()) {
    LogTrace("TrackAssociator") << "ECAL track trajectory is empty; moving on\n";
    info.isGoodEcal = false;
    return;
  }
  info.isGoodEcal = true;

  // Find ECAL crystals
  edm::Handle<EBRecHitCollection> EBRecHits;
  iEvent.getByToken(parameters.EBRecHitsToken, EBRecHits);
  if (!EBRecHits.isValid())
    throw cms::Exception("FatalError") << "Unable to find EBRecHitCollection in the event!\n";

  edm::Handle<EERecHitCollection> EERecHits;
  iEvent.getByToken(parameters.EERecHitsToken, EERecHits);
  if (!EERecHits.isValid())
    throw cms::Exception("FatalError") << "Unable to find EERecHitCollection in event!\n";

  std::set<DetId> ecalIdsInRegion;
  if (parameters.accountForTrajectoryChangeCalo) {
    // get trajectory change with respect to initial state
    DetIdAssociator::MapRange mapRange =
        getMapRange(cachedTrajectory_.trajectoryDelta(CachedTrajectory::IpToEcal), parameters.dREcalPreselection);
    ecalIdsInRegion = ecalDetIdAssociator_->getDetIdsCloseToAPoint(coreTrajectory[0], mapRange);
  } else
    ecalIdsInRegion = ecalDetIdAssociator_->getDetIdsCloseToAPoint(coreTrajectory[0], parameters.dREcalPreselection);
  LogTrace("TrackAssociator") << "ECAL hits in the region: " << ecalIdsInRegion.size();
  if (parameters.dREcalPreselection > parameters.dREcal)
    ecalIdsInRegion = ecalDetIdAssociator_->getDetIdsInACone(ecalIdsInRegion, coreTrajectory, parameters.dREcal);
  LogTrace("TrackAssociator") << "ECAL hits in the cone: " << ecalIdsInRegion.size();
  info.crossedEcalIds = ecalDetIdAssociator_->getCrossedDetIds(ecalIdsInRegion, coreTrajectory);
  const std::vector<DetId>& crossedEcalIds = info.crossedEcalIds;
  LogTrace("TrackAssociator") << "ECAL crossed hits " << crossedEcalIds.size();

  // add EcalRecHits
  for (std::vector<DetId>::const_iterator itr = crossedEcalIds.begin(); itr != crossedEcalIds.end(); itr++) {
    std::vector<EcalRecHit>::const_iterator ebHit = (*EBRecHits).find(*itr);
    std::vector<EcalRecHit>::const_iterator eeHit = (*EERecHits).find(*itr);
    if (ebHit != (*EBRecHits).end())
      info.crossedEcalRecHits.push_back(&*ebHit);
    else if (eeHit != (*EERecHits).end())
      info.crossedEcalRecHits.push_back(&*eeHit);
    else
      LogTrace("TrackAssociator") << "Crossed EcalRecHit is not found for DetId: " << itr->rawId();
  }
  for (std::set<DetId>::const_iterator itr = ecalIdsInRegion.begin(); itr != ecalIdsInRegion.end(); itr++) {
    std::vector<EcalRecHit>::const_iterator ebHit = (*EBRecHits).find(*itr);
    std::vector<EcalRecHit>::const_iterator eeHit = (*EERecHits).find(*itr);
    if (ebHit != (*EBRecHits).end())
      info.ecalRecHits.push_back(&*ebHit);
    else if (eeHit != (*EERecHits).end())
      info.ecalRecHits.push_back(&*eeHit);
    else
      LogTrace("TrackAssociator") << "EcalRecHit from the cone is not found for DetId: " << itr->rawId();
  }
}

fillHcal()

void TrackDetectorAssociator::fillHcal ( const edm::Event &  iEvent, TrackDetMatchInfo &  info, const AssociatorParameters &  parameters  )

private

Definition at line 396 of file TrackDetectorAssociator.cc.

References universalConfigTemplate::collection, HLT_2023v12_cff::distance, iEvent, DetIdInfo::info(), info(), CachedTrajectory::IpToHcal, and LogTrace.

                                                                               {
  const std::vector<SteppingHelixStateInfo>& trajectoryStates = cachedTrajectory_.getHcalTrajectory();

  std::vector<GlobalPoint> coreTrajectory;
  for (std::vector<SteppingHelixStateInfo>::const_iterator itr = trajectoryStates.begin();
       itr != trajectoryStates.end();
       itr++)
    coreTrajectory.push_back(itr->position());

  if (coreTrajectory.empty()) {
    LogTrace("TrackAssociator") << "HCAL trajectory is empty; moving on\n";
    info.isGoodHcal = false;
    return;
  }
  info.isGoodHcal = true;

  // find crossed Hcals
  edm::Handle<HBHERecHitCollection> collection;
  iEvent.getByToken(parameters.HBHEcollToken, collection);
  if (!collection.isValid())
    throw cms::Exception("FatalError") << "Unable to find HBHERecHits in event!\n";

  std::set<DetId> idsInRegion;
  if (parameters.accountForTrajectoryChangeCalo) {
    // get trajectory change with respect to initial state
    DetIdAssociator::MapRange mapRange =
        getMapRange(cachedTrajectory_.trajectoryDelta(CachedTrajectory::IpToHcal), parameters.dRHcalPreselection);
    idsInRegion = hcalDetIdAssociator_->getDetIdsCloseToAPoint(coreTrajectory[0], mapRange);
  } else
    idsInRegion = hcalDetIdAssociator_->getDetIdsCloseToAPoint(coreTrajectory[0], parameters.dRHcalPreselection);

  LogTrace("TrackAssociator") << "HCAL hits in the region: " << idsInRegion.size() << "\n"
                              << DetIdInfo::info(idsInRegion, nullptr);

  auto idsInAConeBegin = idsInRegion.begin();
  auto idsInAConeEnd = idsInRegion.end();
  std::set<DetId> idsInAConeTmp;
  if (!hcalDetIdAssociator_->selectAllInACone(parameters.dRHcal)) {
    idsInAConeTmp = hcalDetIdAssociator_->getDetIdsInACone(idsInRegion, coreTrajectory, parameters.dRHcal);
    idsInAConeBegin = idsInAConeTmp.begin();
    idsInAConeEnd = idsInAConeTmp.end();
  }
  LogTrace("TrackAssociator") << "HCAL hits in the cone: " << std::distance(idsInAConeBegin, idsInAConeEnd) << "\n"
                              << DetIdInfo::info(std::set<DetId>(idsInAConeBegin, idsInAConeEnd), nullptr);
  info.crossedHcalIds = hcalDetIdAssociator_->getCrossedDetIds(idsInRegion, coreTrajectory);
  const std::vector<DetId>& crossedIds = info.crossedHcalIds;
  LogTrace("TrackAssociator") << "HCAL hits crossed: " << crossedIds.size() << "\n"
                              << DetIdInfo::info(crossedIds, nullptr);

  // add Hcal
  for (std::vector<DetId>::const_iterator itr = crossedIds.begin(); itr != crossedIds.end(); itr++) {
    HBHERecHitCollection::const_iterator hit = (*collection).find(*itr);
    if (hit != (*collection).end())
      info.crossedHcalRecHits.push_back(&*hit);
    else
      LogTrace("TrackAssociator") << "Crossed HBHERecHit is not found for DetId: " << itr->rawId();
  }
  for (std::set<DetId>::const_iterator itr = idsInAConeBegin; itr != idsInAConeEnd; itr++) {
    HBHERecHitCollection::const_iterator hit = (*collection).find(*itr);
    if (hit != (*collection).end())
      info.hcalRecHits.push_back(&*hit);
    else
      LogTrace("TrackAssociator") << "HBHERecHit from the cone is not found for DetId: " << itr->rawId();
  }
}

fillHO()

void TrackDetectorAssociator::fillHO ( const edm::Event &  iEvent, TrackDetMatchInfo &  info, const AssociatorParameters &  parameters  )

private

Definition at line 464 of file TrackDetectorAssociator.cc.

References universalConfigTemplate::collection, HLT_2023v12_cff::distance, iEvent, info(), CachedTrajectory::IpToHO, and LogTrace.

                                                                             {
  const std::vector<SteppingHelixStateInfo>& trajectoryStates = cachedTrajectory_.getHOTrajectory();

  std::vector<GlobalPoint> coreTrajectory;
  for (std::vector<SteppingHelixStateInfo>::const_iterator itr = trajectoryStates.begin();
       itr != trajectoryStates.end();
       itr++)
    coreTrajectory.push_back(itr->position());

  if (coreTrajectory.empty()) {
    LogTrace("TrackAssociator") << "HO trajectory is empty; moving on\n";
    info.isGoodHO = false;
    return;
  }
  info.isGoodHO = true;

  // find crossed HOs
  edm::Handle<HORecHitCollection> collection;
  iEvent.getByToken(parameters.HOcollToken, collection);
  if (!collection.isValid())
    throw cms::Exception("FatalError") << "Unable to find HORecHits in event!\n";

  std::set<DetId> idsInRegion;
  if (parameters.accountForTrajectoryChangeCalo) {
    // get trajectory change with respect to initial state
    DetIdAssociator::MapRange mapRange =
        getMapRange(cachedTrajectory_.trajectoryDelta(CachedTrajectory::IpToHO), parameters.dRHcalPreselection);
    idsInRegion = hoDetIdAssociator_->getDetIdsCloseToAPoint(coreTrajectory[0], mapRange);
  } else
    idsInRegion = hoDetIdAssociator_->getDetIdsCloseToAPoint(coreTrajectory[0], parameters.dRHcalPreselection);

  LogTrace("TrackAssociator") << "idsInRegion.size(): " << idsInRegion.size();

  auto idsInAConeBegin = idsInRegion.begin();
  auto idsInAConeEnd = idsInRegion.end();
  std::set<DetId> idsInAConeTmp;
  if (!hoDetIdAssociator_->selectAllInACone(parameters.dRHcal)) {
    idsInAConeTmp = hoDetIdAssociator_->getDetIdsInACone(idsInRegion, coreTrajectory, parameters.dRHcal);
    idsInAConeBegin = idsInAConeTmp.begin();
    idsInAConeEnd = idsInAConeTmp.end();
  }
  LogTrace("TrackAssociator") << "idsInACone.size(): " << std::distance(idsInAConeBegin, idsInAConeEnd);
  info.crossedHOIds = hoDetIdAssociator_->getCrossedDetIds(idsInRegion, coreTrajectory);
  const std::vector<DetId>& crossedIds = info.crossedHOIds;

  // add HO
  for (std::vector<DetId>::const_iterator itr = crossedIds.begin(); itr != crossedIds.end(); itr++) {
    HORecHitCollection::const_iterator hit = (*collection).find(*itr);
    if (hit != (*collection).end())
      info.crossedHORecHits.push_back(&*hit);
    else
      LogTrace("TrackAssociator") << "Crossed HORecHit is not found for DetId: " << itr->rawId();
  }

  for (std::set<DetId>::const_iterator itr = idsInAConeBegin; itr != idsInAConeEnd; itr++) {
    HORecHitCollection::const_iterator hit = (*collection).find(*itr);
    if (hit != (*collection).end())
      info.hoRecHits.push_back(&*hit);
    else
      LogTrace("TrackAssociator") << "HORecHit from the cone is not found for DetId: " << itr->rawId();
  }
}

fillMuon()

void TrackDetectorAssociator::fillMuon ( const edm::Event &  iEvent, TrackDetMatchInfo &  info, const AssociatorParameters &  parameters  )

private

Definition at line 721 of file TrackDetectorAssociator.cc.

References relativeConstraints::chamber, cscRecHitPSet::cscSegment, dtChamberEfficiency_cfi::cscSegments, muonGEMRecHitPSet::gemRecHit, gemRecHits_cfi::gemRecHits, iEvent, info(), edm::HandleBase::isValid(), LogTrace, simTrackMatching_cfi::me0RecHit, me0RecHits_cfi::me0RecHits, me0Segments_cfi::me0Segments, FastTimerService_cff::range, and dtTriggerPhase2PrimitiveDigis_cfi::rpcRecHits.

                                                                               {
  // Get the segments from the event
  edm::Handle<DTRecSegment4DCollection> dtSegments;
  iEvent.getByToken(parameters.dtSegmentsToken, dtSegments);
  if (!dtSegments.isValid())
    throw cms::Exception("FatalError") << "Unable to find DTRecSegment4DCollection in event!\n";

  edm::Handle<CSCSegmentCollection> cscSegments;
  iEvent.getByToken(parameters.cscSegmentsToken, cscSegments);
  if (!cscSegments.isValid())
    throw cms::Exception("FatalError") << "Unable to find CSCSegmentCollection in event!\n";

  edm::Handle<GEMSegmentCollection> gemSegments;
  if (parameters.useGEM)
    iEvent.getByToken(parameters.gemSegmentsToken, gemSegments);
  edm::Handle<ME0SegmentCollection> me0Segments;
  if (parameters.useME0)
    iEvent.getByToken(parameters.me0SegmentsToken, me0Segments);

  // Get the hits from the event only if track preselection is activated
  // Get the chambers segments/hits in the events
  std::set<DetId> occupancy_set;
  if (parameters.preselectMuonTracks) {
    edm::Handle<RPCRecHitCollection> rpcRecHits;
    iEvent.getByToken(parameters.rpcHitsToken, rpcRecHits);
    if (!rpcRecHits.isValid())
      throw cms::Exception("FatalError") << "Unable to find RPCRecHitCollection in event!\n";

    edm::Handle<GEMRecHitCollection> gemRecHits;
    if (parameters.useGEM)
      iEvent.getByToken(parameters.gemHitsToken, gemRecHits);

    edm::Handle<ME0RecHitCollection> me0RecHits;
    if (parameters.useME0)
      iEvent.getByToken(parameters.me0HitsToken, me0RecHits);

    for (const auto& dtSegment : *dtSegments) {
      occupancy_set.insert(dtSegment.geographicalId());
    }
    for (const auto& cscSegment : *cscSegments) {
      occupancy_set.insert(cscSegment.geographicalId());
    }
    for (const auto& rpcRecHit : *rpcRecHits) {
      occupancy_set.insert(rpcRecHit.geographicalId());
    }
    if (parameters.useGEM) {
      for (const auto& gemSegment : *gemSegments) {
        occupancy_set.insert(gemSegment.geographicalId());
      }
      for (const auto& gemRecHit : *gemRecHits) {
        occupancy_set.insert(gemRecHit.geographicalId());
      }
    }
    if (parameters.useME0) {
      for (const auto& me0Segment : *me0Segments) {
        occupancy_set.insert(me0Segment.geographicalId());
      }
      for (const auto& me0RecHit : *me0RecHits) {
        occupancy_set.insert(me0RecHit.geographicalId());
      }
    }
    if (occupancy_set.empty()) {
      LogTrace("TrackAssociator") << "No segments or hits were found in the event: aborting track extrapolation"
                                  << std::endl;
      return;
    }
  }


  // check the map of available segments
  // if there is no segments in a given direction at all,
  // then there is no point to fly there.
  //
  // MISSING
  // Possible solution: quick search for presence of segments
  // for the set of DetIds

  // get a set of matches corresponding to muon chambers
  std::vector<TAMuonChamberMatch> matchedChambers;
  LogTrace("TrackAssociator") << "Trying to Get ChamberMatches" << std::endl;
  getTAMuonChamberMatches(matchedChambers, parameters, occupancy_set);
  LogTrace("TrackAssociator") << "Chambers matched: " << matchedChambers.size() << "\n";

  // Iterate over all chamber matches and fill segment matching
  // info if it's available
  for (std::vector<TAMuonChamberMatch>::iterator matchedChamber = matchedChambers.begin();
       matchedChamber != matchedChambers.end();
       matchedChamber++) {
    const GeomDet* geomDet = muonDetIdAssociator_->getGeomDet((*matchedChamber).id);
    // DT chamber
    if (const DTChamber* chamber = dynamic_cast<const DTChamber*>(geomDet)) {
      // Get the range for the corresponding segments
      DTRecSegment4DCollection::range range = dtSegments->get(chamber->id());
      // Loop over the segments of this chamber
      for (DTRecSegment4DCollection::const_iterator segment = range.first; segment != range.second; segment++) {
        if (addTAMuonSegmentMatch(*matchedChamber, &(*segment), parameters)) {
          matchedChamber->segments.back().dtSegmentRef = DTRecSegment4DRef(dtSegments, segment - dtSegments->begin());
        }
      }
    }
    // CSC Chamber
    else if (const CSCChamber* chamber = dynamic_cast<const CSCChamber*>(geomDet)) {
      // Get the range for the corresponding segments
      CSCSegmentCollection::range range = cscSegments->get(chamber->id());
      // Loop over the segments
      for (CSCSegmentCollection::const_iterator segment = range.first; segment != range.second; segment++) {
        if (addTAMuonSegmentMatch(*matchedChamber, &(*segment), parameters)) {
          matchedChamber->segments.back().cscSegmentRef = CSCSegmentRef(cscSegments, segment - cscSegments->begin());
        }
      }
    } else {
      // GEM Chamber
      if (parameters.useGEM) {
        if (const GEMSuperChamber* chamber = dynamic_cast<const GEMSuperChamber*>(geomDet)) {
          // Get the range for the corresponding segments
          GEMSegmentCollection::range range = gemSegments->get(chamber->id());
          // Loop over the segments
          for (GEMSegmentCollection::const_iterator segment = range.first; segment != range.second; segment++) {
            if (addTAMuonSegmentMatch(*matchedChamber, &(*segment), parameters)) {
              matchedChamber->segments.back().gemSegmentRef =
                  GEMSegmentRef(gemSegments, segment - gemSegments->begin());
            }
          }
        }
      }
      // ME0 Chamber
      if (parameters.useME0) {
        if (const ME0Chamber* chamber = dynamic_cast<const ME0Chamber*>(geomDet)) {
          // Get the range for the corresponding segments
          ME0SegmentCollection::range range = me0Segments->get(chamber->id());
          // Loop over the segments
          for (ME0SegmentCollection::const_iterator segment = range.first; segment != range.second; segment++) {
            if (addTAMuonSegmentMatch(*matchedChamber, &(*segment), parameters)) {
              matchedChamber->segments.back().me0SegmentRef =
                  ME0SegmentRef(me0Segments, segment - me0Segments->begin());
            }
          }
        }
      }
    }
    info.chambers.push_back(*matchedChamber);
  }
}

fillPreshower()

void TrackDetectorAssociator::fillPreshower ( const edm::Event &  iEvent, TrackDetMatchInfo &  info, const AssociatorParameters &  parameters  )

private

Definition at line 373 of file TrackDetectorAssociator.cc.

References info(), and LogTrace.

                                                                                    {
  std::vector<GlobalPoint> trajectory;
  const std::vector<SteppingHelixStateInfo>& trajectoryStates = cachedTrajectory_.getPreshowerTrajectory();
  for (std::vector<SteppingHelixStateInfo>::const_iterator itr = trajectoryStates.begin();
       itr != trajectoryStates.end();
       itr++)
    trajectory.push_back(itr->position());

  if (trajectory.empty()) {
    LogTrace("TrackAssociator") << "Preshower trajectory is empty; moving on\n";
    return;
  }

  std::set<DetId> idsInRegion =
      preshowerDetIdAssociator_->getDetIdsCloseToAPoint(trajectory[0], parameters.dRPreshowerPreselection);

  LogTrace("TrackAssociator") << "Number of Preshower Ids in the region: " << idsInRegion.size();
  info.crossedPreshowerIds = preshowerDetIdAssociator_->getCrossedDetIds(idsInRegion, trajectory);
  LogTrace("TrackAssociator") << "Number of Preshower Ids in crossed: " << info.crossedPreshowerIds.size();
}

getCachedTrajector()

const CachedTrajectory& TrackDetectorAssociator::getCachedTrajector ( ) const

inline

trajector information

Definition at line 93 of file TrackDetectorAssociator.h.

References cachedTrajectory_.

{ return cachedTrajectory_; }

getFreeTrajectoryState() [1/3]

FreeTrajectoryState TrackDetectorAssociator::getFreeTrajectoryState ( const MagneticField *  bField, const reco::Track &  track  )

static

get FreeTrajectoryState from different track representations

Definition at line 557 of file TrackDetectorAssociator.cc.

References Calorimetry_cff::bField, MillePedeFileConverter_cfg::e, l1ctLayer2EG_cff::id, point, HLT_2023v12_cff::track, and trackerHitRTTI::vector.

Referenced by EopTreeWriter::analyze(), ValidIsoTrkCalib::analyze(), cms::MuonMETValueMapProducer::determine_deltax_deltay(), and InterestingTrackEcalDetIdProducer::produce().

                                                                                            {
  GlobalVector vector(track.momentum().x(), track.momentum().y(), track.momentum().z());

  GlobalPoint point(track.vertex().x(), track.vertex().y(), track.vertex().z());

  GlobalTrajectoryParameters tPars(point, vector, track.charge(), bField);

  // FIX THIS !!!
  // need to convert from perigee to global or helix (curvilinear) frame
  // for now just an arbitrary matrix.
  ROOT::Math::SMatrixIdentity id;
  AlgebraicSymMatrix66 covT(id);
  covT *= 1e-6;  // initialize to sigma=1e-3
  CartesianTrajectoryError tCov(covT);

  return FreeTrajectoryState(tPars, tCov);
}

getFreeTrajectoryState() [2/3]

FreeTrajectoryState TrackDetectorAssociator::getFreeTrajectoryState ( const MagneticField *  bField, const SimTrack &  track, const SimVertex &  vertex  )

static

Definition at line 529 of file TrackDetectorAssociator.cc.

References funct::abs(), Calorimetry_cff::bField, reco::charge(), point, HLT_2023v12_cff::track, trackerHitRTTI::vector, and bphysicsOniaDQM_cfi::vertex.

                                                                                             {
  GlobalVector vector(track.momentum().x(), track.momentum().y(), track.momentum().z());
  GlobalPoint point(vertex.position().x(), vertex.position().y(), vertex.position().z());

  int charge = track.type() > 0 ? -1 : 1;  // lepton convention
  if (abs(track.type()) == 211 ||          // pion
      abs(track.type()) == 321 ||          // kaon
      abs(track.type()) == 2212)
    charge = track.type() < 0 ? -1 : 1;
  return getFreeTrajectoryState(bField, vector, point, charge);
}

getFreeTrajectoryState() [3/3]

FreeTrajectoryState TrackDetectorAssociator::getFreeTrajectoryState ( const MagneticField *  bField, const GlobalVector &  momentum, const GlobalPoint &  vertex, const int  charge  )

static

Definition at line 543 of file TrackDetectorAssociator.cc.

References Calorimetry_cff::bField, reco::charge(), MillePedeFileConverter_cfg::e, l1ctLayer2EG_cff::id, and bphysicsOniaDQM_cfi::vertex.

                                                                                      {
  GlobalTrajectoryParameters tPars(vertex, momentum, charge, bField);

  ROOT::Math::SMatrixIdentity id;
  AlgebraicSymMatrix66 covT(id);
  covT *= 1e-6;  // initialize to sigma=1e-3
  CartesianTrajectoryError tCov(covT);

  return FreeTrajectoryState(tPars, tCov);
}

getMapRange()

DetIdAssociator::MapRange TrackDetectorAssociator::getMapRange ( const std::pair< float, float > &  delta, const float  dR  )

private

Definition at line 576 of file TrackDetectorAssociator.cc.

References funct::abs(), dumpMFGeometry_cfg::delta, DetIdAssociator::MapRange::dPhiMinus, DetIdAssociator::MapRange::dPhiPlus, HGC3DClusterGenMatchSelector_cfi::dR, DetIdAssociator::MapRange::dThetaMinus, DetIdAssociator::MapRange::dThetaPlus, and LogTrace.

                                                                                                               {
  DetIdAssociator::MapRange mapRange;
  mapRange.dThetaPlus = dR;
  mapRange.dThetaMinus = dR;
  mapRange.dPhiPlus = dR;
  mapRange.dPhiMinus = dR;
  if (delta.first > 0)
    mapRange.dThetaPlus += delta.first;
  else
    mapRange.dThetaMinus += std::abs(delta.first);
  if (delta.second > 0)
    mapRange.dPhiPlus += delta.second;
  else
    mapRange.dPhiMinus += std::abs(delta.second);
  LogTrace("TrackAssociator") << "Selection range: (dThetaPlus, dThetaMinus, dPhiPlus, dPhiMinus, dRPreselection): "
                              << mapRange.dThetaPlus << ", " << mapRange.dThetaMinus << ", " << mapRange.dPhiPlus
                              << ", " << mapRange.dPhiMinus << ", " << dR;
  return mapRange;
}

getPoint() [1/2]

math::XYZPoint TrackDetectorAssociator::getPoint ( const GlobalPoint &  point )

inlineprivate

Definition at line 136 of file TrackDetectorAssociator.h.

References point.

                                                               {
    return math::XYZPoint(point.x(), point.y(), point.z());
  }

getPoint() [2/2]

math::XYZPoint TrackDetectorAssociator::getPoint ( const LocalPoint &  point )

inlineprivate

Definition at line 140 of file TrackDetectorAssociator.h.

References point.

                                                              {
    return math::XYZPoint(point.x(), point.y(), point.z());
  }

getTAMuonChamberMatches()

void TrackDetectorAssociator::getTAMuonChamberMatches ( std::vector< TAMuonChamberMatch > &  matches, const AssociatorParameters &  parameters, std::set< DetId >  occupancy  )

private

Definition at line 596 of file TrackDetectorAssociator.cc.

References funct::abs(), Surface::bounds(), hcalRecHitTable_cff::detId, f, TrajectoryStateOnSurface::freeState(), CachedTrajectory::FullTrajectory, DetIdInfo::info(), TrajectoryStateOnSurface::isValid(), SteppingHelixStateInfo::isValid(), Bounds::length(), CSCWireTopology::lengthOfPlane(), TrajectoryStateOnSurface::localError(), LogTrace, oniaPATMuonsWithTrigger_cff::matches, SteppingHelixStateInfo::momentum(), CSCWireTopology::narrowWidthOfPlane(), CSCChamberSpecs::oddLayerGeometry(), SteppingHelixStateInfo::position(), FreeTrajectoryState::position(), LocalTrajectoryError::positionError(), DBoxMetadataHelper::set_intersection(), mathSSE::sqrt(), GeomDet::surface(), GloballyPositioned< T >::toLocal(), Vector3DBase< T, FrameTag >::unit(), CSCWireTopology::wideWidthOfPlane(), Bounds::width(), TrapezoidalPlaneBounds::width(), CSCWireTopology::wireAngle(), CSCLayerGeometry::wireTopology(), PV3DBase< T, PVType, FrameType >::x(), LocalError::xx(), PV3DBase< T, PVType, FrameType >::y(), CSCWireTopology::yOfWire(), and LocalError::yy().

                                                                               {
  // Strategy:
  //    Propagate through the whole detector, estimate change in eta and phi
  //    along the trajectory, add this to dRMuon and find DetIds around this
  //    direction using the map. Then propagate fast to each surface and apply
  //    final matching criteria.

  // get the direction first
  SteppingHelixStateInfo trajectoryPoint = cachedTrajectory_.getStateAtHcal();
  // If trajectory point at HCAL is not valid, try to use the outer most state of the
  // trajectory instead.
  if (!trajectoryPoint.isValid())
    trajectoryPoint = cachedTrajectory_.getOuterState();
  if (!trajectoryPoint.isValid()) {
    LogTrace("TrackAssociator")
        << "trajectory position at HCAL is not valid. Assume the track cannot reach muon detectors and skip it";
    return;
  }

  GlobalVector direction = trajectoryPoint.momentum().unit();
  LogTrace("TrackAssociator") << "muon direction: " << direction
                              << "\n\t and corresponding point: " << trajectoryPoint.position() << "\n";

  DetIdAssociator::MapRange mapRange =
      getMapRange(cachedTrajectory_.trajectoryDelta(CachedTrajectory::FullTrajectory), parameters.dRMuonPreselection);

  // and find chamber DetIds

  std::set<DetId> muonIdsInRegion = muonDetIdAssociator_->getDetIdsCloseToAPoint(trajectoryPoint.position(), mapRange);
  LogTrace("TrackAssociator") << "Number of chambers to check: " << muonIdsInRegion.size();

  if (parameters.preselectMuonTracks) {
    std::set<DetId> muonIdsInRegionOccupied;
    std::set_intersection(muonIdsInRegion.begin(),
                          muonIdsInRegion.end(),
                          occupancy.begin(),
                          occupancy.end(),
                          std::inserter(muonIdsInRegionOccupied, muonIdsInRegionOccupied.begin()));
    if (muonIdsInRegionOccupied.empty())
      return;
  }

  for (std::set<DetId>::const_iterator detId = muonIdsInRegion.begin(); detId != muonIdsInRegion.end(); detId++) {
    const GeomDet* geomDet = muonDetIdAssociator_->getGeomDet(*detId);
    TrajectoryStateOnSurface stateOnSurface = cachedTrajectory_.propagate(&geomDet->surface());
    if (!stateOnSurface.isValid()) {
      LogTrace("TrackAssociator") << "Failed to propagate the track; moving on\n\t"
                                  << "Element is not crosssed: " << DetIdInfo::info(*detId, nullptr) << "\n";
      continue;
    }
    LocalPoint localPoint = geomDet->surface().toLocal(stateOnSurface.freeState()->position());
    LocalError localError = stateOnSurface.localError().positionError();
    float distanceX = 0.f;
    float distanceY = 0.f;
    if (const CSCChamber* cscChamber = dynamic_cast<const CSCChamber*>(geomDet)) {
      const CSCChamberSpecs* chamberSpecs = cscChamber->specs();
      if (!chamberSpecs) {
        LogTrace("TrackAssociator") << "Failed to get CSCChamberSpecs from CSCChamber; moving on\n";
        continue;
      }
      const CSCLayerGeometry* layerGeometry = chamberSpecs->oddLayerGeometry(1);
      if (!layerGeometry) {
        LogTrace("TrackAssociator") << "Failed to get CSCLayerGeometry from CSCChamberSpecs; moving on\n";
        continue;
      }
      const CSCWireTopology* wireTopology = layerGeometry->wireTopology();
      if (!wireTopology) {
        LogTrace("TrackAssociator") << "Failed to get CSCWireTopology from CSCLayerGeometry; moving on\n";
        continue;
      }

      float wideWidth = wireTopology->wideWidthOfPlane();
      float narrowWidth = wireTopology->narrowWidthOfPlane();
      float length = wireTopology->lengthOfPlane();
      // If slanted, there is no y offset between local origin and symmetry center of wire plane
      float yOfFirstWire = std::abs(wireTopology->wireAngle()) > 1.E-06f ? -0.5 * length : wireTopology->yOfWire(1);
      // y offset between local origin and symmetry center of wire plane
      float yCOWPOffset = yOfFirstWire + 0.5f * length;

      // tangent of the incline angle from inside the trapezoid
      float tangent = (wideWidth - narrowWidth) / (2.f * length);
      // y position wrt bottom of trapezoid
      float yPrime = localPoint.y() + std::abs(yOfFirstWire);
      // half trapezoid width at y' is 0.5 * narrowWidth + x side of triangle with the above tangent and side y'
      float halfWidthAtYPrime = 0.5f * narrowWidth + yPrime * tangent;
      distanceX = std::abs(localPoint.x()) - halfWidthAtYPrime;
      distanceY = std::abs(localPoint.y() - yCOWPOffset) - 0.5f * length;
    } else if (dynamic_cast<const GEMChamber*>(geomDet) || dynamic_cast<const GEMSuperChamber*>(geomDet)) {
      const TrapezoidalPlaneBounds* bounds = dynamic_cast<const TrapezoidalPlaneBounds*>(&geomDet->surface().bounds());

      float wideWidth = bounds->width();
      float narrowWidth = 2.f * bounds->widthAtHalfLength() - wideWidth;
      float length = bounds->length();
      float tangent = (wideWidth - narrowWidth) / (2.f * length);
      float halfWidthAtY = tangent * localPoint.y() + 0.25f * (narrowWidth + wideWidth);

      distanceX = std::abs(localPoint.x()) - halfWidthAtY;
      distanceY = std::abs(localPoint.y()) - 0.5f * length;
    } else {
      distanceX = std::abs(localPoint.x()) - 0.5f * geomDet->surface().bounds().width();
      distanceY = std::abs(localPoint.y()) - 0.5f * geomDet->surface().bounds().length();
    }
    if ((distanceX < parameters.muonMaxDistanceX && distanceY < parameters.muonMaxDistanceY) ||
        (distanceX * distanceX <
             localError.xx() * parameters.muonMaxDistanceSigmaX * parameters.muonMaxDistanceSigmaX &&
         distanceY * distanceY <
             localError.yy() * parameters.muonMaxDistanceSigmaY * parameters.muonMaxDistanceSigmaY)) {
      LogTrace("TrackAssociator") << "found a match: " << DetIdInfo::info(*detId, nullptr) << "\n";
      TAMuonChamberMatch match;
      match.tState = stateOnSurface;
      match.localDistanceX = distanceX;
      match.localDistanceY = distanceY;
      match.id = *detId;
      matches.push_back(match);
    } else {
      LogTrace("TrackAssociator") << "chamber is too far: " << DetIdInfo::info(*detId, nullptr)
                                  << "\n\tdistanceX: " << distanceX << "\t distanceY: " << distanceY
                                  << "\t sigmaX: " << distanceX / sqrt(localError.xx())
                                  << "\t sigmaY: " << distanceY / sqrt(localError.yy()) << "\n";
    }
  }
}

getVector() [1/2]

math::XYZVector TrackDetectorAssociator::getVector ( const GlobalVector &  vec )

inlineprivate

Definition at line 144 of file TrackDetectorAssociator.h.

{ return math::XYZVector(vec.x(), vec.y(), vec.z()); }

getVector() [2/2]

math::XYZVector TrackDetectorAssociator::getVector ( const LocalVector &  vec )

inlineprivate

Definition at line 146 of file TrackDetectorAssociator.h.

{ return math::XYZVector(vec.x(), vec.y(), vec.z()); }

init()

void TrackDetectorAssociator::init ( const edm::EventSetup &  iSetup, const AssociatorParameters &  parameters  )

private

Definition at line 87 of file TrackDetectorAssociator.cc.

References anyDirection, Calorimetry_cff::bField, edm::EventSetup::getData(), and eostools::move().

                                                                                                      {
  // access the calorimeter geometry
  theCaloGeometry_ = &iSetup.getData(parameters.theCaloGeometryToken);

  // get the tracking Geometry
  theTrackingGeometry_ = &iSetup.getData(parameters.theTrackingGeometryToken);

  if (useDefaultPropagator_ && (!defProp_ || theMagneticFieldWatcher_.check(iSetup))) {
    // setup propagator
    const MagneticField* bField = &iSetup.getData(parameters.bFieldToken);

    auto prop = std::make_unique<SteppingHelixPropagator>(bField, anyDirection);
    prop->setMaterialMode(false);
    prop->applyRadX0Correction(true);
    // prop->setDebug(true); // tmp
    defProp_ = std::move(prop);
    setPropagator(defProp_.get());
  }

  ecalDetIdAssociator_ = &iSetup.getData(parameters.ecalDetIdAssociatorToken);
  hcalDetIdAssociator_ = &iSetup.getData(parameters.hcalDetIdAssociatorToken);
  hoDetIdAssociator_ = &iSetup.getData(parameters.hoDetIdAssociatorToken);
  caloDetIdAssociator_ = &iSetup.getData(parameters.caloDetIdAssociatorToken);
  muonDetIdAssociator_ = &iSetup.getData(parameters.muonDetIdAssociatorToken);
  preshowerDetIdAssociator_ = &iSetup.getData(parameters.preshowerDetIdAssociatorToken);
}

setPropagator()

void TrackDetectorAssociator::setPropagator

(

const Propagator *

ptr

)

use a user configured propagator

Definition at line 80 of file TrackDetectorAssociator.cc.

Referenced by muonisolation::CaloExtractorByAssociator::deposits(), and MuonIdProducer::init().

                                                                 {
  ivProp_ = ptr;
  cachedTrajectory_.setPropagator(ivProp_);
}

useDefaultPropagator()

void TrackDetectorAssociator::useDefaultPropagator

(

)

use the default propagator

Definition at line 85 of file TrackDetectorAssociator.cc.

Referenced by EopTreeWriter::analyze(), EcalCosmicsHists::EcalCosmicsHists(), InterestingTrackEcalDetIdProducer::InterestingTrackEcalDetIdProducer(), cms::MuonMETValueMapProducer::MuonMETValueMapProducer(), pat::PATIsolatedTrackProducer::PATIsolatedTrackProducer(), and ValidIsoTrkCalib::ValidIsoTrkCalib().

{ useDefaultPropagator_ = true; }

Member Data Documentation

cachedTrajectory_

CachedTrajectory TrackDetectorAssociator::cachedTrajectory_

private

Definition at line 150 of file TrackDetectorAssociator.h.

Referenced by getCachedTrajector().

caloDetIdAssociator_

const DetIdAssociator* TrackDetectorAssociator::caloDetIdAssociator_

private

Definition at line 156 of file TrackDetectorAssociator.h.

defProp_

std::unique_ptr<Propagator> TrackDetectorAssociator::defProp_

private

Definition at line 149 of file TrackDetectorAssociator.h.

ecalDetIdAssociator_

const DetIdAssociator* TrackDetectorAssociator::ecalDetIdAssociator_

private

Definition at line 153 of file TrackDetectorAssociator.h.

hcalDetIdAssociator_

const DetIdAssociator* TrackDetectorAssociator::hcalDetIdAssociator_

private

Definition at line 154 of file TrackDetectorAssociator.h.

hoDetIdAssociator_

const DetIdAssociator* TrackDetectorAssociator::hoDetIdAssociator_

private

Definition at line 155 of file TrackDetectorAssociator.h.

ivProp_

const Propagator* TrackDetectorAssociator::ivProp_

private

Definition at line 148 of file TrackDetectorAssociator.h.

muonDetIdAssociator_

const DetIdAssociator* TrackDetectorAssociator::muonDetIdAssociator_

private

Definition at line 157 of file TrackDetectorAssociator.h.

preshowerDetIdAssociator_

const DetIdAssociator* TrackDetectorAssociator::preshowerDetIdAssociator_

private

Definition at line 158 of file TrackDetectorAssociator.h.

theCaloGeometry_

const CaloGeometry* TrackDetectorAssociator::theCaloGeometry_

private

Definition at line 160 of file TrackDetectorAssociator.h.

theMagneticFieldWatcher_

edm::ESWatcher<IdealMagneticFieldRecord> TrackDetectorAssociator::theMagneticFieldWatcher_

private

Definition at line 163 of file TrackDetectorAssociator.h.

theTrackingGeometry_

const GlobalTrackingGeometry* TrackDetectorAssociator::theTrackingGeometry_

private

Definition at line 161 of file TrackDetectorAssociator.h.

useDefaultPropagator_

bool TrackDetectorAssociator::useDefaultPropagator_

private

Definition at line 151 of file TrackDetectorAssociator.h.