TrackExtrapolateG4e Class Reference

geant4e-based track extrapolation. More...

#include <TrackExtrapolateG4e.h>

Public Member Functions
	~TrackExtrapolateG4e ()
	destructor
void	initialize (double minPt, double minKE, std::vector< Const::ChargedStable > &hypotheses)
	Initialize for track extrapolation by the EXT module.
void	initialize (double meanDt, double maxDt, double maxSeparation, double maxKLMTrackClusterDistance, double maxECLTrackClusterDistance, double minPt, double minKE, bool addHitsToRecoTrack, std::vector< Const::ChargedStable > &hypotheses)
	Initialize for track extrapolation by the MUID module.
void	beginRun (bool flag)
	Perform beginning-of-run actions.
void	event (bool flag)
	Performs track extrapolation for all tracks in one event.
void	endRun (bool flag)
	Perform end-of-run actions.
void	terminate (bool flag)
	Terminates this singleton.
void	extrapolate (int pdgCode, double tof, const G4ThreeVector &position, const G4ThreeVector &momentum, const G4ErrorSymMatrix &covariance)
	Performs track extrapolation for a single track (specified in genfit2 units).

Static Public Member Functions
static TrackExtrapolateG4e *	getInstance ()
	Get the singleton's address.

Private Member Functions
	TrackExtrapolateG4e ()
	constructor is hidden; user calls TrackExtrapolateG4e::getInstance() instead
	TrackExtrapolateG4e (TrackExtrapolateG4e &)
	copy constructor is hidden; user calls TrackExtrapolateG4e::getInstance() instead
void	swim (ExtState &, G4ErrorFreeTrajState &, const std::vector< std::pair< ECLCluster *, G4ThreeVector > > *, const std::vector< std::pair< KLMCluster *, G4ThreeVector > > *, std::vector< std::map< const Track *, double > > *)
	Swim a single track (MUID) until it stops or leaves the target cylinder.
void	swim (ExtState &, G4ErrorFreeTrajState &)
	Swim a single track (EXT) until it stops or leaves the target cylinder.
void	findClosestTrackToKLMClusters ()
	Find the closest Track to each KLMCluster and fill the corresponding fields of KLMCluster objects.
void	registerVolumes ()
	Register the list of geant4 physical volumes whose entry/exit points will be saved during extrapolation.
void	getVolumeID (const G4TouchableHandle &, Const::EDetector &, int &)
	Get the physical volume information for a geant4 physical volume.
void	fromG4eToPhasespace (const G4ErrorFreeTrajState &, G4ErrorSymMatrix &)
	Convert the geant4e 5x5 covariance to phasespace 6x6 covariance.
void	fromPhasespaceToG4e (const G4ThreeVector &, const G4ErrorSymMatrix &, G4ErrorTrajErr &)
	Convert the phasespace covariance to geant4e covariance.
void	fromPhasespaceToG4e (const TVector3 &, const TMatrixDSym &, G4ErrorTrajErr &)
	Convert the phasespace covariance to geant4e covariance.
ExtState	getStartPoint (const Track &, int, G4ErrorFreeTrajState &)
	Get the start point for a new reconstructed track with specific PDG hypothesis.
void	createExtHit (ExtHitStatus, const ExtState &, const G4ErrorFreeTrajState &, const G4StepPoint *, const G4TouchableHandle &)
	Create another EXT extrapolation hit for a track candidate.
void	createECLHit (const ExtState &, const G4ErrorFreeTrajState &, const G4StepPoint *, const G4StepPoint *, const G4TouchableHandle &, const std::pair< ECLCluster *, G4ThreeVector > &, double, double)
	Create another EXT ECL-crystal-crossing hit for a track candidate.
bool	createMuidHit (ExtState &, G4ErrorFreeTrajState &, KLMMuidLikelihood *, std::vector< std::map< const Track *, double > > *)
	Create another MUID extrapolation hit for a track candidate.
bool	findBarrelIntersection (ExtState &, const G4ThreeVector &, Intersection &)
	Find the intersection point of the track with the crossed BKLM plane.
bool	findEndcapIntersection (ExtState &, const G4ThreeVector &, Intersection &)
	Find the intersection point of the track with the crossed EKLM plane.
bool	findMatchingBarrelHit (Intersection &, const Track *)
	Find the matching BKLM 2D hit nearest the intersection point of the track with the crossed BKLM plane.
bool	findMatchingEndcapHit (Intersection &, const Track *)
	Find the matching EKLM 2D hit nearest the intersection point of the track with the crossed EKLM plane.
void	adjustIntersection (Intersection &, const double *, const G4ThreeVector &, const G4ThreeVector &)
	Nudge the track using the matching hit.
void	finishTrack (const ExtState &, KLMMuidLikelihood *, bool)
	Complete muon identification after end of track extrapolation.

Private Attributes
bool	m_ExtInitialized
	Flag to indicate that EXT initialize() has been called.
bool	m_MuidInitialized
	Flag to indicate that MUID initialize() has been called.
double	m_MeanDt
	Mean hit - trigger time (ns)
double	m_MaxDt
	Coincidence window half-width for in-time KLM hits (ns)
double	m_MagneticField
	Magnetic field z component (gauss) at origin.
double	m_MaxDistSqInVariances
	user-defined maximum squared-distance (in number of variances) for matching hit to extrapolation
double	m_MaxKLMTrackClusterDistance
	user-defined maximum distance (mm) between KLMCluster and associated track (for KLID)
double	m_MaxECLTrackClusterDistance
	user-defined maximum distance (mm) between ECLCluster and associated track (for EID)
double	m_MinPt
	Minimum transverse momentum in MeV/c for extrapolation to be started.
double	m_MinKE
	Minimum kinetic energy in MeV for extrapolation to continue.
Simulation::ExtManager *	m_ExtMgr
	Pointer to the ExtManager singleton.
const std::vector< Const::ChargedStable > *	m_HypothesesExt
	ChargedStable hypotheses for EXT.
const std::vector< Const::ChargedStable > *	m_HypothesesMuid
	ChargedStable hypotheses for MUID.
std::vector< Const::ChargedStable > *	m_DefaultHypotheses
	Default ChargedStable hypotheses (needed as call argument but not used)
std::map< G4VPhysicalVolume *, enum VolTypes > *	m_EnterExit
	Pointers to geant4 physical volumes whose entry/exit points will be saved.
std::vector< G4VPhysicalVolume * > *	m_BKLMVolumes
	Pointers to BKLM geant4 sensitive (physical) volumes.
Simulation::ExtCylSurfaceTarget *	m_TargetExt
	virtual "target" cylinder for EXT (boundary beyond which extrapolation ends)
Simulation::ExtCylSurfaceTarget *	m_TargetMuid
	virtual "target" cylinder for MUID (boundary beyond which extrapolation ends)
DBObjPtr< COILGeometryPar >	m_COILGeometryPar
	Conditions-database object for COIL geometry.
DBObjPtr< BeamPipeGeo >	m_BeamPipeGeo
	Conditions-database object for beam pipe geometry.
double	m_MinRadiusSq
	Minimum squared radius (cm) outside of which extrapolation will continue.
double	m_OffsetZ
	offset (cm) along z axis of KLM midpoint from IP
int	m_BarrelNSector
	Number of barrel sectors.
double	m_BarrelMaxR
	maximum radius (cm) of the barrel
double	m_BarrelMinR
	minimum radius (cm) of the barrel
double	m_BarrelHalfLength
	half-length (cm) of the barrel
int	m_OutermostActiveBarrelLayer
	outermost barrel layer that is active for muon identification (user-defined)
double	m_BarrelPhiStripVariance [BKLMElementNumbers::getMaximalLayerNumber()+1]
	BKLM RPC phi-measuring strip position variance (cm^2) by layer.
double	m_BarrelZStripVariance [BKLMElementNumbers::getMaximalLayerNumber()+1]
	BKLM RPC z-measuring strip position variance (cm^2) by layer.
double	m_BarrelScintVariance
	BKLM scintillator strip position variance (cm^2)
double	m_BarrelModuleMiddleRadius [2][BKLMElementNumbers::getMaximalSectorNumber()+1][BKLMElementNumbers::getMaximalLayerNumber()+1]
	hit-plane radius (cm) at closest distance to IP of each barrel end | sector | layer
G4ThreeVector	m_BarrelSectorPerp [BKLMElementNumbers::getMaximalSectorNumber()+1]
	normal unit vector of each barrel sector
G4ThreeVector	m_BarrelSectorPhi [BKLMElementNumbers::getMaximalSectorNumber()+1]
	azimuthal unit vector of each barrel sector
double	m_EndcapMaxR
	maximum radius (cm) of the endcaps
double	m_EndcapMinR
	minimum radius (cm) of the endcaps
double	m_EndcapMiddleZ
	midpoint along z (cm) of the forward endcap from the KLM midpoint
double	m_EndcapHalfLength
	half-length (cm) of either endcap
int	m_OutermostActiveForwardEndcapLayer
	outermost forward-endcap layer that is active for muon identification (user-defined)
int	m_OutermostActiveBackwardEndcapLayer
	outermost backward-endcap layer that is active for muon identification (user-defined)
double	m_EndcapScintVariance
	EKLM scintillator strip position variance (cm^2)
double	m_EndcapModuleMiddleZ [BKLMElementNumbers::getMaximalLayerNumber()+1]
	hit-plane z (cm) of each IP layer relative to KLM midpoint
bool	m_addHitsToRecoTrack = false
	Parameter to add the found hits also to the reco tracks or not. Is turned off by default.
std::map< int, MuidBuilder * >	m_MuidBuilderMap
	PDF for the charged final state particle hypotheses.
const EKLMElementNumbers *	m_eklmElementNumbers
	EKLM element numbers.
const KLMElementNumbers *	m_klmElementNumbers
	KLM element numbers.
const EKLM::TransformDataGlobalAligned *	m_eklmTransformData
	EKLM transformation data.
DBObjPtr< KLMChannelStatus >	m_klmChannelStatus
	Conditions-database object for KLM channel status.
DBObjPtr< KLMStripEfficiency >	m_klmStripEfficiency
	Conditions-database object for KLM strip efficiency.
DBObjPtr< KLMLikelihoodParameters >	m_klmLikelihoodParameters
	Conditions-database object for KLM likelihood parameters.
StoreArray< ECLCluster >	m_eclClusters
	ECL clusters.
StoreArray< ExtHit >	m_extHits
	Ext hits.
StoreArray< KLMHit2d >	m_klmHit2ds
	KLM 2d hits.
StoreArray< KLMCluster >	m_klmClusters
	KLM clusters.
StoreArray< KLMMuidHit >	m_klmMuidHits
	KLM muid hits.
StoreArray< KLMMuidLikelihood >	m_klmMuidLikelihoods
	KLM muid likelihoods.
StoreArray< RecoTrack >	m_recoTracks
	Reco tracks.
StoreArray< Track >	m_tracks
	Tracks.
StoreArray< TrackClusterSeparation >	m_trackClusterSeparations
	Track cluster separation.

Static Private Attributes
static TrackExtrapolateG4e *	m_Singleton = nullptr
	Stores pointer to the singleton class.

Detailed Description

geant4e-based track extrapolation.

This class extrapolates tracks outward from the outer perimeter of the CDC using geant4e.

This class requires a valid geometry in memory (gGeoManager). Therefore, a geometry building module should have been executed before this module is called.

This class has the same functions as a module - and these are called from the ExtModule's so-named functions - but also has an entry that can be called to extrapolate a single user-defined track.

Definition at line 168 of file TrackExtrapolateG4e.h.

Constructor & Destructor Documentation

~TrackExtrapolateG4e()

~TrackExtrapolateG4e

(

)

destructor

Definition at line 133 of file TrackExtrapolateG4e.cc.

{
}

TrackExtrapolateG4e()

TrackExtrapolateG4e ( )

private

constructor is hidden; user calls TrackExtrapolateG4e::getInstance() instead

Definition at line 79 of file TrackExtrapolateG4e.cc.

                                         :
  m_ExtInitialized(false), // initialized later
  m_MuidInitialized(false), // initialized later
  m_MeanDt(0.0), // initialized later
  m_MaxDt(0.0), // initialized later
  m_MagneticField(0.0), // initialized later
  m_MaxDistSqInVariances(0.0), // initialized later
  m_MaxKLMTrackClusterDistance(0.0), // initialized later
  m_MaxECLTrackClusterDistance(0.0), // initialized later
  m_MinPt(0.0), // initialized later
  m_MinKE(0.0), // initialized later
  m_ExtMgr(nullptr), // initialized later
  m_HypothesesExt(nullptr), // initialized later
  m_HypothesesMuid(nullptr), // initialized later
  m_DefaultHypotheses(nullptr), // initialized later
  m_EnterExit(nullptr), // initialized later
  m_BKLMVolumes(nullptr), // initialized later
  m_TargetExt(nullptr), // initialized later
  m_TargetMuid(nullptr), // initialized later
  m_MinRadiusSq(0.0), // initialized later
  m_OffsetZ(0.0), // initialized later
  m_BarrelNSector(0), // initialized later
  m_BarrelMaxR(0.0), // initialized later
  m_BarrelMinR(0.0), // initialized later
  m_BarrelHalfLength(0.0), // initialized later
  m_OutermostActiveBarrelLayer(0), // initialized later
  m_BarrelScintVariance(0.0), // initialized later
  m_EndcapMaxR(0.0), // initialized later
  m_EndcapMinR(0.0), // initialized later
  m_EndcapMiddleZ(0.0), // initialized later
  m_EndcapHalfLength(0.0), // initialized later
  m_OutermostActiveForwardEndcapLayer(0), // initialized later
  m_OutermostActiveBackwardEndcapLayer(0), // initialized later
  m_EndcapScintVariance(0.0), // initialized later
  m_eklmTransformData(nullptr) // initialized later
{
  for (int j = 0; j < BKLMElementNumbers::getMaximalLayerNumber() + 1; ++j) {
    m_BarrelPhiStripVariance[j] = 0.0;
    m_BarrelZStripVariance[j] = 0.0;
    m_BarrelPhiStripVariance[j] = 0.0;
    m_EndcapModuleMiddleZ[j] = 0.0;
  }
  for (int s = 0; s < BKLMElementNumbers::getMaximalSectorNumber() + 1; ++s) {
    for (int j = 0; j < BKLMElementNumbers::getMaximalLayerNumber() + 1; ++j) {
      m_BarrelModuleMiddleRadius[0][s][j] = 0.0;
      m_BarrelModuleMiddleRadius[1][s][j] = 0.0;
    }
    m_BarrelSectorPerp[s] = G4ThreeVector(0.0, 0.0, 0.0);
    m_BarrelSectorPhi[s] = G4ThreeVector(0.0, 0.0, 0.0);
  }
  m_klmElementNumbers = &(KLMElementNumbers::Instance());
  m_eklmElementNumbers = &(EKLMElementNumbers::Instance());
}

Member Function Documentation

adjustIntersection()

void adjustIntersection ( Intersection & , const double * , const G4ThreeVector & , const G4ThreeVector &  )

private

Nudge the track using the matching hit.

Definition at line 1670 of file TrackExtrapolateG4e.cc.

{
  // Use the gain matrix formalism to get the corrected track parameters.
  // R. Fruhwirth, Application of Kalman Filtering, NIM A262 (1987) 444-450
  // Equations (7)
  // x_k^{k-1} = extPar[] 6 elements before filtering
  // C_k^{k-1} = extCov[] 6x6 symmetric before filtering
  // r_k^{k-1} = residual[] 2 elements before filtering
  // h_k = 2x6 projects cartesian coordinates to measurement-plane coordinates
  // H_k = @h_k/@x = jacobian[] 2x6 Jacobian of projection to measurement plane
  // R_k^{k-1} = correction[] 2x2 before Invert()
  // G_k = R^(-1) = correction[] 2x2 after Invert()
  // K_k = gain[] 6x2 Kalman gain matrix
  // x_k = extPar[] 6 elements after filtering
  // C_k = extCov[] 6x6 symmetric after filtering
  // r_k = residual[] 2 elements after filtering
  // Use the relation K*H*C = (C*H^T*R^-1)*H*C = C*(H^T*R^-1*H)*C^T
 
  // In most cases, extPos0 is the same as intersection.position.  They differ only when
  // the nearest BKLM hit is in the sector adjacent to that of intersection.position.
  G4ThreeVector extPos(extPos0);
  G4ThreeVector extMom(intersection.momentum);
  G4ThreeVector extDir(extMom.unit());
  G4ThreeVector diffPos(hitPos - extPos);
  G4ErrorSymMatrix extCov(intersection.covariance);
  // Track parameters (x,y,z,px,py,pz) before correction
  G4ErrorMatrix extPar(6, 1); // initialized to all zeroes
  extPar[0][0] = extPos.x();
  extPar[1][0] = extPos.y();
  extPar[2][0] = extPos.z();
  extPar[3][0] = extMom.x();
  extPar[4][0] = extMom.y();
  extPar[5][0] = extMom.z();
  G4ThreeVector nA;  // unit vector normal to the readout plane
  G4ThreeVector nB;  // unit vector along phi- or x-readout direction (for barrel or endcap)
  G4ThreeVector nC;  // unit vector along z- or y-readout direction (for barrel or endcap)
  if (intersection.inBarrel) {
    nA = m_BarrelSectorPerp[intersection.sector];
    nB = m_BarrelSectorPhi[intersection.sector];
    nC = G4ThreeVector(0.0, 0.0, 1.0);
  } else {
    double out = (intersection.isForward ? 1.0 : -1.0);
    nA = G4ThreeVector(0.0, 0.0, out);
    nB = G4ThreeVector(out, 0.0, 0.0);
    nC = G4ThreeVector(0.0, out, 0.0);
  }
  // Don't adjust the extrapolation if the track is nearly tangent to the readout plane.
  double extDirA = extDir * nA;
  if (std::fabs(extDirA) < 1.0E-6)
    return;
  double extDirBA = extDir * nB / extDirA;
  double extDirCA = extDir * nC / extDirA;
  // Move the extrapolated coordinate (at most a tiny amount!) to the plane of the hit.
  // If the moved point is outside the KLM, don't do Kalman filtering.
  G4ThreeVector move = extDir * ((diffPos * nA) / extDirA);
  extPos += move;
  diffPos -= move;
  intersection.positionAtHitPlane = extPos;
  // Projection jacobian onto the nB-nC measurement plane
  G4ErrorMatrix jacobian(2, 6); // initialized to all zeroes
  jacobian[0][0] = nB.x()  - nA.x() * extDirBA;
  jacobian[0][1] = nB.y()  - nA.y() * extDirBA;
  jacobian[0][2] = nB.z()  - nA.z() * extDirBA;
  jacobian[1][0] = nC.x()  - nA.x() * extDirCA;
  jacobian[1][1] = nC.y()  - nA.y() * extDirCA;
  jacobian[1][2] = nC.z()  - nA.z() * extDirCA;
  // Residuals of EXT track and KLM hit on the nB-nC measurement plane
  G4ErrorMatrix residual(2, 1); // initialized to all zeroes
  residual[0][0] = diffPos.x() * jacobian[0][0] + diffPos.y() * jacobian[0][1] + diffPos.z() * jacobian[0][2];
  residual[1][0] = diffPos.x() * jacobian[1][0] + diffPos.y() * jacobian[1][1] + diffPos.z() * jacobian[1][2];
  // Measurement errors in the detector plane
  G4ErrorSymMatrix hitCov(2, 0); // initialized to all zeroes
  hitCov[0][0] = localVariance[0];
  hitCov[1][1] = localVariance[1];
  // No magnetic field: increase the hit uncertainty
  if (m_MagneticField == 0.0) {
    hitCov[0][0] *= 10.0;
    hitCov[1][1] *= 10.0;
  }
  // Now get the correction matrix: combined covariance of EXT and KLM hit.
  // 1st dimension = nB, 2nd dimension = nC.
  G4ErrorSymMatrix correction(extCov.similarity(jacobian) + hitCov);
  // Ignore the best hit if it is too far from the extrapolated-track intersection in the hit's plane
  if (residual[0][0] * residual[0][0] > correction[0][0] * m_MaxDistSqInVariances)
    return;
  if (residual[1][0] * residual[1][0] > correction[1][1] * m_MaxDistSqInVariances)
    return;
  int fail = 0;
  correction.invert(fail);
  if (fail != 0)
    return;
  // Matrix inversion succeeded and is reasonable.
  // Evaluate chi-squared increment assuming that the Kalman filter
  // won't be able to adjust the extrapolated track's position (fall-back).
  intersection.chi2 = (correction.similarityT(residual))[0][0];
  // Do the Kalman filtering
  G4ErrorMatrix gain((extCov * jacobian.T()) * correction);
  G4ErrorSymMatrix HRH(correction.similarityT(jacobian));
  extCov -= HRH.similarity(extCov);
  extPar += gain * residual;
  extPos.set(extPar[0][0], extPar[1][0], extPar[2][0]);
  extMom.set(extPar[3][0], extPar[4][0], extPar[5][0]);
  // Calculate the chi-squared increment using the Kalman-filtered state
  correction = hitCov - extCov.similarity(jacobian);
  correction.invert(fail);
  if (fail != 0)
    return;
  diffPos = hitPos - extPos;
  residual[0][0] = diffPos.x() * jacobian[0][0] + diffPos.y() * jacobian[0][1] + diffPos.z() * jacobian[0][2];
  residual[1][0] = diffPos.x() * jacobian[1][0] + diffPos.y() * jacobian[1][1] + diffPos.z() * jacobian[1][2];
  intersection.chi2 = (correction.similarityT(residual))[0][0];
  // Update the position, momentum and covariance of the point
  // Project the corrected extrapolation to the plane of the original
  // extrapolation's intersection.position. (Note: intersection.position is the same as
  // extPos0 in all cases except when nearest BKLM hit is in adjacent
  // sector, for which extPos0 is a projected position to the hit's plane.)
  // Also, leave the momentum magnitude unchanged.
  intersection.position = extPos + extDir * (((intersection.position - extPos) * nA) / extDirA);
  intersection.momentum = intersection.momentum.mag() * extMom.unit();
  intersection.covariance = extCov;
}

beginRun()

void beginRun

(

bool

flag

)

Perform beginning-of-run actions.

Parameters

flag	True if called by Muid module, false if called by Ext module.

Definition at line 333 of file TrackExtrapolateG4e.cc.

{
  B2DEBUG(20, (byMuid ? "muid" : "ext"));
  if (byMuid) {
    if (!m_klmChannelStatus.isValid())
      B2FATAL("KLM channel status data are not available.");
    if (!m_klmStripEfficiency.isValid())
      B2FATAL("KLM strip efficiency data are not available.");
    if (!m_klmLikelihoodParameters.isValid())
      B2FATAL("KLM likelihood parameters are not available.");
    std::vector<int> muidPdgCodes = MuidElementNumbers::getPDGVector();
    if (!m_MuidBuilderMap.empty()) {
      if (m_klmLikelihoodParameters.hasChanged()) { /* Clear m_MuidBuilderMap if KLMLikelihoodParameters payload changed. */
        for (auto const& muidBuilder : m_MuidBuilderMap)
          delete muidBuilder.second;
        m_MuidBuilderMap.clear();
      } else /* Return if m_MuidBuilderMap is already initialized. */
        return;
    }
    for (int pdg : muidPdgCodes)
      m_MuidBuilderMap.insert(std::pair<int, MuidBuilder*>(pdg, new MuidBuilder(pdg)));
  }
}

createExtHit()

void createExtHit ( ExtHitStatus status, const ExtState & extState, const G4ErrorFreeTrajState & g4eState, const G4StepPoint * stepPoint, const G4TouchableHandle & touch )

private

Create another EXT extrapolation hit for a track candidate.

Definition at line 1213 of file TrackExtrapolateG4e.cc.

{
  Const::EDetector detID(Const::EDetector::invalidDetector);
  int copyID(0);
  getVolumeID(touch, detID, copyID);
  G4ThreeVector pos(stepPoint->GetPosition() / CLHEP::cm);
  G4ThreeVector mom(stepPoint->GetMomentum() / CLHEP::GeV);
  if (extState.isCosmic)
    mom = -mom;
  G4ErrorSymMatrix covariance(6, 0);
  fromG4eToPhasespace(g4eState, covariance);
  ExtHit* extHit = m_extHits.appendNew(extState.tof, extState.pdgCode, detID, copyID, status,
                                       extState.isCosmic,
                                       pos, mom, covariance);
  // If called standalone, there will be no associated track
  if (extState.track != nullptr)
    extState.track->addRelationTo(extHit);
}

createMuidHit()

bool createMuidHit ( ExtState & extState, G4ErrorFreeTrajState & g4eState, KLMMuidLikelihood * klmMuidLikelihood, std::vector< std::map< const Track *, double > > * bklmHitUsed )

private

Create another MUID extrapolation hit for a track candidate.

Definition at line 1237 of file TrackExtrapolateG4e.cc.

{
 
  Intersection intersection;
  intersection.hit = -1;
  intersection.chi2 = -1.0;
  intersection.position = g4eState.GetPosition() / CLHEP::cm;
  intersection.momentum = g4eState.GetMomentum() / CLHEP::GeV;
  G4ThreeVector prePos = g4eState.GetG4Track()->GetStep()->GetPreStepPoint()->GetPosition() / CLHEP::cm;
  G4ThreeVector oldPosition(prePos.x(), prePos.y(), prePos.z());
  double r = intersection.position.perp();
  double z = std::fabs(intersection.position.z() - m_OffsetZ);
 
  // Is the track in the barrel?
  if ((r > m_BarrelMinR) && (r < m_BarrelMaxR) && (z < m_BarrelHalfLength)) {
    // Did the track cross the inner midplane of a detector module?
    if (findBarrelIntersection(extState, oldPosition, intersection)) {
      fromG4eToPhasespace(g4eState, intersection.covariance);
      if (findMatchingBarrelHit(intersection, extState.track)) {
        (*bklmHitUsed)[intersection.hit].insert(std::pair<const Track*, double>(extState.track, intersection.chi2));
        extState.extLayerPattern |= (0x00000001 << intersection.layer);
        float layerBarrelEfficiency = 1.;
        for (int plane = 0; plane <= BKLMElementNumbers::getMaximalPlaneNumber(); plane++) {
          layerBarrelEfficiency *= m_klmStripEfficiency->getBarrelEfficiency(
                                     intersection.isForward ? BKLMElementNumbers::c_ForwardSection : BKLMElementNumbers::c_BackwardSection,
                                     intersection.sector + 1, intersection.layer + 1, plane, 1);
        }
        klmMuidLikelihood->setExtBKLMEfficiencyValue(intersection.layer, layerBarrelEfficiency);
        if (extState.lastBarrelExtLayer < intersection.layer) {
          extState.lastBarrelExtLayer = intersection.layer;
        }
        extState.hitLayerPattern |= (0x00000001 << intersection.layer);
        if (extState.lastBarrelHitLayer < intersection.layer) {
          extState.lastBarrelHitLayer = intersection.layer;
        }
        // If the updated point is outside the barrel, discard it and the Kalman-fitter adjustment
        r = intersection.position.perp();
        z = std::fabs(intersection.position.z() - m_OffsetZ);
        if ((r <= m_BarrelMinR) || (r >= m_BarrelMaxR) || (z >= m_BarrelHalfLength)) {
          intersection.chi2 = -1.0;
        }
      } else {
        // Record a no-hit track crossing if this step is strictly within a barrel sensitive volume
        std::vector<G4VPhysicalVolume*>::iterator j = find(m_BKLMVolumes->begin(), m_BKLMVolumes->end(),
                                                           g4eState.GetG4Track()->GetVolume());
        if (j != m_BKLMVolumes->end()) {
          bool isDead = true; // by default, the nearest orthogonal strips are dead
          int section = intersection.isForward ?
                        BKLMElementNumbers::c_ForwardSection :
                        BKLMElementNumbers::c_BackwardSection;
          int sector = intersection.sector + 1; // from 0-based to 1-based enumeration
          int layer = intersection.layer + 1; // from 0-based to 1-based enumeration
          const bklm::Module* m = bklm::GeometryPar::instance()->findModule(section, sector, layer); // uses 1-based enumeration
          if (m) {
            const CLHEP::Hep3Vector localPosition = m->globalToLocal(intersection.position); // uses and returns position in cm
            int zStrip = m->getZStripNumber(localPosition);
            int phiStrip = m->getPhiStripNumber(localPosition);
            if (zStrip >= 0 && phiStrip >= 0) {
              uint16_t channel1, channel2;
              channel1 = m_klmElementNumbers->channelNumberBKLM(
                           section, sector, layer,
                           BKLMElementNumbers::c_ZPlane, zStrip);
              channel2 = m_klmElementNumbers->channelNumberBKLM(
                           section, sector, layer,
                           BKLMElementNumbers::c_PhiPlane, phiStrip);
              enum KLMChannelStatus::ChannelStatus status1, status2;
              status1 = m_klmChannelStatus->getChannelStatus(channel1);
              status2 = m_klmChannelStatus->getChannelStatus(channel2);
              if (status1 == KLMChannelStatus::c_Unknown ||
                  status2 == KLMChannelStatus::c_Unknown)
                B2ERROR("No KLM channel status data."
                        << LogVar("Section", section)
                        << LogVar("Sector", sector)
                        << LogVar("Layer", layer)
                        << LogVar("Z strip", zStrip)
                        << LogVar("Phi strip", phiStrip));
              isDead = (status1 == KLMChannelStatus::c_Dead ||
                        status2 == KLMChannelStatus::c_Dead);
            }
          }
          if (!isDead) {
            extState.extLayerPattern |= (0x00000001 << intersection.layer); // valid extrapolation-crossing of the layer but no matching hit
            float layerBarrelEfficiency = 1.;
            for (int plane = 0; plane <= BKLMElementNumbers::getMaximalPlaneNumber(); plane++) {
              layerBarrelEfficiency *= m_klmStripEfficiency->getBarrelEfficiency(
                                         intersection.isForward ? BKLMElementNumbers::c_ForwardSection : BKLMElementNumbers::c_BackwardSection,
                                         intersection.sector + 1, intersection.layer + 1, plane, 1);
            }
            klmMuidLikelihood->setExtBKLMEfficiencyValue(intersection.layer, layerBarrelEfficiency);
          } else {
            klmMuidLikelihood->setExtBKLMEfficiencyValue(intersection.layer, 0);
          }
          if (extState.lastBarrelExtLayer < intersection.layer) {
            extState.lastBarrelExtLayer = intersection.layer;
          }
        }
      }
    }
  }
 
  // Is the track in the endcap?
  if ((r > m_EndcapMinR) && (std::fabs(z - m_EndcapMiddleZ) < m_EndcapHalfLength)) {
    // Did the track cross the inner midplane of a detector module?
    if (findEndcapIntersection(extState, oldPosition, intersection)) {
      fromG4eToPhasespace(g4eState, intersection.covariance);
      if (findMatchingEndcapHit(intersection, extState.track)) {
        extState.extLayerPattern |= (0x00008000 << intersection.layer);
        float layerEndcapEfficiency = 1.;
        for (int plane = 1; plane <= EKLMElementNumbers::getMaximalPlaneNumber(); plane++) {
          layerEndcapEfficiency *= m_klmStripEfficiency->getEndcapEfficiency(
                                     intersection.isForward ? EKLMElementNumbers::c_ForwardSection : EKLMElementNumbers::c_BackwardSection,
                                     intersection.sector + 1, intersection.layer + 1, plane, 1);
        }
        klmMuidLikelihood->setExtEKLMEfficiencyValue(intersection.layer, layerEndcapEfficiency);
        if (extState.lastEndcapExtLayer < intersection.layer) {
          extState.lastEndcapExtLayer = intersection.layer;
        }
        extState.hitLayerPattern |= (0x00008000 << intersection.layer);
        if (extState.lastEndcapHitLayer < intersection.layer) {
          extState.lastEndcapHitLayer = intersection.layer;
        }
        // If the updated point is outside the endcap, discard it and the Kalman-fitter adjustment
        r = intersection.position.perp();
        z = std::fabs(intersection.position.z() - m_OffsetZ);
        if ((r <= m_EndcapMinR) || (r >= m_EndcapMaxR) || (std::fabs(z - m_EndcapMiddleZ) >= m_EndcapHalfLength)) {
          intersection.chi2 = -1.0;
        }
      } else {
        bool isDead = true;
        int result, strip1, strip2;
        result = m_eklmTransformData->getStripsByIntersection(
                   intersection.position, &strip1, &strip2);
        if (result == 0) {
          uint16_t channel1, channel2;
          channel1 = m_klmElementNumbers->channelNumberEKLM(strip1);
          channel2 = m_klmElementNumbers->channelNumberEKLM(strip2);
          enum KLMChannelStatus::ChannelStatus status1, status2;
          status1 = m_klmChannelStatus->getChannelStatus(channel1);
          status2 = m_klmChannelStatus->getChannelStatus(channel2);
          if (status1 == KLMChannelStatus::c_Unknown ||
              status2 == KLMChannelStatus::c_Unknown)
            B2ERROR("Incomplete KLM channel status data.");
          isDead = (status1 == KLMChannelStatus::c_Dead ||
                    status2 == KLMChannelStatus::c_Dead);
        }
        if (!isDead) {
          extState.extLayerPattern |= (0x00008000 << intersection.layer); // valid extrapolation-crossing of the layer but no matching hit
          float layerEndcapEfficiency = 1.;
          for (int plane = 1; plane <= EKLMElementNumbers::getMaximalPlaneNumber(); plane++) {
            layerEndcapEfficiency *= m_klmStripEfficiency->getEndcapEfficiency(
                                       intersection.isForward ? EKLMElementNumbers::c_ForwardSection : EKLMElementNumbers::c_BackwardSection,
                                       intersection.sector + 1, intersection.layer + 1, plane, 1);
          }
          klmMuidLikelihood->setExtEKLMEfficiencyValue(intersection.layer, layerEndcapEfficiency);
        } else {
          klmMuidLikelihood->setExtEKLMEfficiencyValue(intersection.layer, 0);
        }
        if (extState.lastEndcapExtLayer < intersection.layer) {
          extState.lastEndcapExtLayer = intersection.layer;
        }
      }
    }
  }
 
  // Create a new MuidHit and RelationEntry between it and the track.
  // Adjust geant4e's position, momentum and covariance based on matching hit and tell caller to update the geant4e state.
  if (intersection.chi2 >= 0.0) {
    ROOT::Math::XYZVector tpos(intersection.position.x(),
                               intersection.position.y(),
                               intersection.position.z());
    ROOT::Math::XYZVector tposAtHitPlane(intersection.positionAtHitPlane.x(),
                                         intersection.positionAtHitPlane.y(),
                                         intersection.positionAtHitPlane.z());
    KLMMuidHit* klmMuidHit = m_klmMuidHits.appendNew(extState.pdgCode,
                                                     intersection.inBarrel, intersection.isForward,
                                                     intersection.sector, intersection.layer,
                                                     tpos, tposAtHitPlane,
                                                     extState.tof, intersection.time,
                                                     intersection.chi2);
    if (extState.track != nullptr) { extState.track->addRelationTo(klmMuidHit); }
    G4Point3D newPos(intersection.position.x() * CLHEP::cm,
                     intersection.position.y() * CLHEP::cm,
                     intersection.position.z() * CLHEP::cm);
    g4eState.SetPosition(newPos);
    G4Vector3D newMom(intersection.momentum.x() * CLHEP::GeV,
                      intersection.momentum.y() * CLHEP::GeV,
                      intersection.momentum.z() * CLHEP::GeV);
    g4eState.SetMomentum(newMom);
    G4ErrorTrajErr covG4e;
    fromPhasespaceToG4e(intersection.momentum, intersection.covariance, covG4e);
    g4eState.SetError(covG4e);
    extState.chi2 += intersection.chi2;
    extState.nPoint += 2; // two (orthogonal) independent hits per detector layer
    return true;
  }
 
  // Tell caller that the geant4e state was not modified.
  return false;
 
}

endRun()

void endRun

(

bool

flag

)

Perform end-of-run actions.

Parameters

flag	True if called by Muid module, false if called by Ext module.

Definition at line 416 of file TrackExtrapolateG4e.cc.

{
}

event()

void event

(

bool

flag

)

Performs track extrapolation for all tracks in one event.

Parameters

flag	True if called by Muid module, false if called by Ext module.

Definition at line 357 of file TrackExtrapolateG4e.cc.

{
 
  // Put geant4 in proper state (in case this module is in a separate process)
  if (G4StateManager::GetStateManager()->GetCurrentState() == G4State_Idle) {
    G4StateManager::GetStateManager()->SetNewState(G4State_GeomClosed);
  }
 
  G4ThreeVector directionAtIP, positionG4e, momentumG4e;
  G4ErrorTrajErr covG4e(5); // initialized to zeroes
 
  // Loop over the reconstructed tracks
  // Do extrapolation for each hypothesis of each reconstructed track.
  if (byMuid) { // event() called by Muid module
    G4ErrorPropagatorData::GetErrorPropagatorData()->SetTarget(m_TargetMuid);
    std::vector<std::pair<ECLCluster*, G4ThreeVector> > eclClusterInfo(m_eclClusters.getEntries());
    for (int c = 0; c < m_eclClusters.getEntries(); ++c) {
      eclClusterInfo[c].first = m_eclClusters[c];
      eclClusterInfo[c].second = G4ThreeVector(m_eclClusters[c]->getClusterPosition().X(),
                                               m_eclClusters[c]->getClusterPosition().Y(),
                                               m_eclClusters[c]->getClusterPosition().Z()) * CLHEP::cm;
    }
    std::vector<std::pair<KLMCluster*, G4ThreeVector> > klmClusterInfo(m_klmClusters.getEntries());
    for (int c = 0; c < m_klmClusters.getEntries(); ++c) {
      klmClusterInfo[c].first = m_klmClusters[c];
      klmClusterInfo[c].second = G4ThreeVector(m_klmClusters[c]->getClusterPosition().X(),
                                               m_klmClusters[c]->getClusterPosition().Y(),
                                               m_klmClusters[c]->getClusterPosition().Z()) * CLHEP::cm;
    }
    // Keep track of (re-)use of BKLMHit2ds
    std::vector<std::map<const Track*, double> > bklmHitUsed(m_klmHit2ds.getEntries());
    for (auto& b2track : m_tracks) {
      for (const auto& hypothesis : *m_HypothesesMuid) {
        int pdgCode = hypothesis.getPDGCode();
        if (hypothesis == Const::electron || hypothesis == Const::muon)
          pdgCode = -pdgCode;
        G4ErrorFreeTrajState g4eState("g4e_mu+", G4ThreeVector(), G4ThreeVector()); // will be updated
        ExtState extState = getStartPoint(b2track, pdgCode, g4eState);
        swim(extState, g4eState, &eclClusterInfo, &klmClusterInfo, &bklmHitUsed);
      } // Muid hypothesis loop
    } // Muid track loop
  } else { // event() called by Ext module
    G4ErrorPropagatorData::GetErrorPropagatorData()->SetTarget(m_TargetExt);
    for (auto& b2track : m_tracks) {
      for (const auto& hypothesis : *m_HypothesesExt) {
        int pdgCode = hypothesis.getPDGCode();
        if (hypothesis == Const::electron || hypothesis == Const::muon) pdgCode = -pdgCode;
        G4ErrorFreeTrajState g4eState("g4e_mu+", G4ThreeVector(), G4ThreeVector()); // will be updated
        ExtState extState = getStartPoint(b2track, pdgCode, g4eState);
        swim(extState, g4eState);
      } // Ext hypothesis loop
    } // Ext track loop
  } // byMuid
 
  if (byMuid) {
    findClosestTrackToKLMClusters();
  }
}

extrapolate()

void extrapolate	(	int	pdgCode,
		double	tof,
		const G4ThreeVector &	position,
		const G4ThreeVector &	momentum,
		const G4ErrorSymMatrix &	covariance )

Performs track extrapolation for a single track (specified in genfit2 units).

Parameters

pdgCode	Signed PDG identifier of the particle hypothesis to be used for the extrapolation.
tof	Starting time, i.e., time of flight from the IP, at the starting point (ns).
position	Starting point of the extrapolation (cm).
momentum	Momentum of the track at the starting point (GeV/c).
covariance	Phase-space covariance matrix (6x6) at the starting point (cm, GeV/c).

Definition at line 442 of file TrackExtrapolateG4e.cc.

{
  bool isCosmic = false; // DIVOT
  if ((!m_ExtInitialized) && (!m_MuidInitialized)) {
    // No EXT nor MUID module in analysis path ==> mimic ext::initialize() with reasonable defaults.
    // The default values are taken from the EXT module's parameter definitions.
    Simulation::ExtManager* extMgr = Simulation::ExtManager::GetManager();
    extMgr->Initialize("Ext", "default", 0.0, 0.25, false, 0, std::vector<std::string>());
    // Redefine geant4e step length, magnetic field step limitation (fraction of local curvature radius),
    // and kinetic energy loss limitation (maximum fractional energy loss) by communicating with
    // the geant4 UI.  (Commands were defined in ExtMessenger when physics list was set up.)
    // *NOTE* If module muid runs after this, its G4UImanager commands will override these.
    G4UImanager::GetUIpointer()->ApplyCommand("/geant4e/limits/stepLength 250 mm");
    G4UImanager::GetUIpointer()->ApplyCommand("/geant4e/limits/magField 0.001");
    G4UImanager::GetUIpointer()->ApplyCommand("/geant4e/limits/energyLoss 0.05");
    m_DefaultHypotheses = new std::vector<Const::ChargedStable>; // not used
    initialize(0.1, 0.002, *m_DefaultHypotheses);
  }
 
  // Put geant4 in proper state (in case this module is in a separate process)
  if (G4StateManager::GetStateManager()->GetCurrentState() == G4State_Idle) {
    G4StateManager::GetStateManager()->SetNewState(G4State_GeomClosed);
  }
 
  G4ErrorPropagatorData::GetErrorPropagatorData()->SetTarget(m_TargetExt);
 
  // Do extrapolation for selected hypothesis (pion, electron, muon, kaon, proton,
  // deuteron) for the selected track until calorimeter exit.
 
  G4ThreeVector positionG4e = position * CLHEP::cm; // convert from genfit2 units (cm) to geant4 units (mm)
  G4ThreeVector momentumG4e = momentum * CLHEP::GeV; // convert from genfit2 units (GeV/c) to geant4 units (MeV/c)
  if (isCosmic)
    momentumG4e = -momentumG4e;
  G4ErrorSymMatrix covarianceG4e(5, 0); // in Geant4e units (GeV/c, cm)
  fromPhasespaceToG4e(momentum, covariance, covarianceG4e);
  G4String nameG4e("g4e_" + G4ParticleTable::GetParticleTable()->FindParticle(pdgCode)->GetParticleName());
  G4ErrorFreeTrajState g4eState(nameG4e, positionG4e, momentumG4e, covarianceG4e);
  ExtState extState = { nullptr, pdgCode, isCosmic, tof, 0.0,                         // for EXT and MUID
                        momentumG4e.unit(), 0.0, 0, 0, 0, 0, 0, 0, 0, 0, 0, false  // for MUID only
                      };
  swim(extState, g4eState);
}

findBarrelIntersection()

bool findBarrelIntersection ( ExtState & extState, const G4ThreeVector & oldPosition, Intersection & intersection )

private

Find the intersection point of the track with the crossed BKLM plane.

Definition at line 1439 of file TrackExtrapolateG4e.cc.

{
  // Be generous: allow outward-moving intersection to be in the dead space between
  // largest sensitive-volume Z and m_BarrelHalfLength, not necessarily in a geant4 sensitive volume
  if (std::fabs(intersection.position.z() - m_OffsetZ) > m_BarrelHalfLength)
    return false;
  double phi = intersection.position.phi();
  if (phi < 0.0)
    phi += TWOPI;
  if (phi > TWOPI - PI_8)
    phi -= TWOPI;
  int sector = (int)((phi + PI_8) / M_PI_4);
  int section = intersection.position.z() > m_OffsetZ ?
                BKLMElementNumbers::c_ForwardSection :
                BKLMElementNumbers::c_BackwardSection;
  double oldR = oldPosition * m_BarrelSectorPerp[sector];
  double newR = intersection.position * m_BarrelSectorPerp[sector];
  for (int layer = extState.firstBarrelLayer; layer <= m_OutermostActiveBarrelLayer; ++layer) {
    if (newR <  m_BarrelModuleMiddleRadius[section][sector][layer]) break;
    if (oldR <= m_BarrelModuleMiddleRadius[section][sector][layer]) {
      extState.firstBarrelLayer = layer + 1; // ratchet outward for next call's loop starting value
      if (extState.firstBarrelLayer > m_OutermostActiveBarrelLayer) extState.escaped = true;
      intersection.inBarrel = true;
      intersection.isForward = intersection.position.z() > m_OffsetZ;
      intersection.layer = layer;
      intersection.sector = sector;
      return true;
    }
  }
  return false;
}

findClosestTrackToKLMClusters()

void findClosestTrackToKLMClusters ( )

private

Find the closest Track to each KLMCluster and fill the corresponding fields of KLMCluster objects.

Definition at line 710 of file TrackExtrapolateG4e.cc.

{
  for (auto& klmCluster : m_klmClusters) {
    const auto& trackClusterSeparations = klmCluster.getRelationsWith<TrackClusterSeparation>();
 
    // If there are no TrackClusterSeparation objects related to this cluster, set NaN
    if (trackClusterSeparations.size() == 0) {
      klmCluster.setClusterTrackSeparation(Const::doubleNaN);
      klmCluster.setClusterTrackSeparationAngle(Const::doubleNaN);
      klmCluster.setClusterTrackRotationAngle(Const::doubleNaN);
      continue;
    }
 
    // Look for the closest track by comparing the TrackClusterSeparation objects
    const auto closestSeparationIterator = std::min_element(trackClusterSeparations.begin(), trackClusterSeparations.end(),
    [](const auto & a, const auto & b) {
      return a.getDistance() < b.getDistance();
    });
 
    // We found the closest track, let's set the cluster properties accordingly
    const auto& closestSeparation = *closestSeparationIterator;
    klmCluster.setClusterTrackSeparation(closestSeparation.getDistance() / CLHEP::cm);
    klmCluster.setClusterTrackSeparationAngle(closestSeparation.getTrackClusterSeparationAngle());
    klmCluster.setClusterTrackRotationAngle(closestSeparation.getTrackRotationAngle());
  }
}

findEndcapIntersection()

bool findEndcapIntersection ( ExtState & extState, const G4ThreeVector & oldPosition, Intersection & intersection )

private

Find the intersection point of the track with the crossed EKLM plane.

Definition at line 1471 of file TrackExtrapolateG4e.cc.

{
  // Be generous: allow intersection to be in the dead space between m_EndcapMinR and innermost
  // sensitive-volume radius or between outermost sensitive-volume radius and m_EndcapMaxR,
  // not necessarily in a geant4 sensitive volume
  if (oldPosition.perp() > m_EndcapMaxR)
    return false;
  if (intersection.position.perp() < m_EndcapMinR)
    return false;
  double oldZ = std::fabs(oldPosition.z() - m_OffsetZ);
  double newZ = std::fabs(intersection.position.z() - m_OffsetZ);
  bool isForward = intersection.position.z() > m_OffsetZ;
  int outermostLayer = isForward ? m_OutermostActiveForwardEndcapLayer
                       : m_OutermostActiveBackwardEndcapLayer;
  for (int layer = extState.firstEndcapLayer; layer <= outermostLayer; ++layer) {
    if (newZ <  m_EndcapModuleMiddleZ[layer])
      break;
    if (oldZ <= m_EndcapModuleMiddleZ[layer]) {
      extState.firstEndcapLayer = layer + 1; // ratchet outward for next call's loop starting value
      if (extState.firstEndcapLayer > outermostLayer)
        extState.escaped = true;
      intersection.inBarrel = false;
      intersection.isForward = isForward;
      intersection.layer = layer;
      intersection.sector = m_eklmTransformData->getSectorByPosition(
                              isForward ? 2 : 1, intersection.position) - 1;
      return true;
    }
  }
  return false;
}

findMatchingBarrelHit()

bool findMatchingBarrelHit ( Intersection & intersection, const Track * track )

private

Find the matching BKLM 2D hit nearest the intersection point of the track with the crossed BKLM plane.

Definition at line 1503 of file TrackExtrapolateG4e.cc.

{
  G4ThreeVector extPos0(intersection.position);
  double diffBestMagSq = 1.0E60;
  int bestHit = -1;
  int matchingLayer = intersection.layer + 1;
  G4ThreeVector n(m_BarrelSectorPerp[intersection.sector]);
  for (int h = 0; h < m_klmHit2ds.getEntries(); ++h) {
    KLMHit2d* hit = m_klmHit2ds[h];
    if (hit->getSubdetector() != KLMElementNumbers::c_BKLM)
      continue;
    if (hit->getLayer() != matchingLayer)
      continue;
    if (hit->isOutOfTime())
      continue;
    if (std::fabs(hit->getTime() - m_MeanDt) > m_MaxDt)
      continue;
    G4ThreeVector diff(hit->getPositionX() - intersection.position.x(),
                       hit->getPositionY() - intersection.position.y(),
                       hit->getPositionZ() - intersection.position.z());
    double dn = diff * n; // in cm
    if (std::fabs(dn) < 2.0) {
      // Hit and extrapolated point are in the same sector
      diff -= n * dn;
      if (diff.mag2() < diffBestMagSq) {
        diffBestMagSq = diff.mag2();
        bestHit = h;
        extPos0 = intersection.position;
      }
    } else {
      // Accept a nearby hit in adjacent sector
      if (std::fabs(dn) > 50.0)
        continue;
      int sector = hit->getSector() - 1;
      int dSector = abs(intersection.sector - sector);
      if ((dSector != +1) && (dSector != m_BarrelNSector - 1))
        continue;
      // Use the normal vector of the adjacent (hit's) sector
      G4ThreeVector nHit(m_BarrelSectorPerp[sector]);
      int section = intersection.isForward ?
                    BKLMElementNumbers::c_ForwardSection :
                    BKLMElementNumbers::c_BackwardSection;
      double dn2 = intersection.position * nHit - m_BarrelModuleMiddleRadius[section][sector][intersection.layer];
      dn = diff * nHit + dn2;
      if (std::fabs(dn) > 1.0)
        continue;
      // Project extrapolated track to the hit's plane in the adjacent sector
      G4ThreeVector extDir(intersection.momentum.unit());
      double extDirA = extDir * nHit;
      if (std::fabs(extDirA) < 1.0E-6)
        continue;
      G4ThreeVector projection = extDir * (dn2 / extDirA);
      if (projection.mag() > 15.0)
        continue;
      diff += projection - nHit * dn;
      if (diff.mag2() < diffBestMagSq) {
        diffBestMagSq = diff.mag2();
        bestHit = h;
        extPos0 = intersection.position - projection;
      }
    }
  }
 
  if (bestHit >= 0) {
    KLMHit2d* hit = m_klmHit2ds[bestHit];
    intersection.isForward = (hit->getSection() == 1);
    intersection.sector = hit->getSector() - 1;
    intersection.time = hit->getTime();
    double localVariance[2] = {m_BarrelScintVariance, m_BarrelScintVariance};
    if (hit->inRPC()) {
      int nStrips = hit->getPhiStripMax() - hit->getPhiStripMin() + 1;
      double dn = nStrips - 1.5;
      double factor = std::pow((0.9 + 0.4 * dn * dn), 1.5) * 0.60; // measured-in-Belle resolution
      localVariance[0] = m_BarrelPhiStripVariance[intersection.layer] * factor;
      nStrips = hit->getZStripMax() - hit->getZStripMin() + 1;
      dn = nStrips - 1.5;
      factor = std::pow((0.9 + 0.4 * dn * dn), 1.5) * 0.55; // measured-in-Belle resolution
      localVariance[1] = m_BarrelZStripVariance[intersection.layer] * factor;
    } else {
      int nStrips = hit->getPhiStripMax() - hit->getPhiStripMin() + 1;
      localVariance[0] *= (nStrips * nStrips); // variance inflated for multi-strip hit
      nStrips = hit->getZStripMax() - hit->getZStripMin() + 1;
      localVariance[1] *= (nStrips * nStrips); // variance inflated for multi-strip hit
    }
    G4ThreeVector hitPos(hit->getPositionX(), hit->getPositionY(),
                         hit->getPositionZ());
    adjustIntersection(intersection, localVariance, hitPos, extPos0);
    if (intersection.chi2 >= 0.0) {
      intersection.hit = bestHit;
      hit->isOnTrack(true);
      if (track != nullptr) {
        track->addRelationTo(hit, intersection.chi2);
        RecoTrack* recoTrack = track->getRelatedTo<RecoTrack>();
        if (m_addHitsToRecoTrack) {
          recoTrack->addBKLMHit(hit, recoTrack->getNumberOfTotalHits() + 1);
        }
      }
    }
  }
  return intersection.chi2 >= 0.0;
 
}

findMatchingEndcapHit()

bool findMatchingEndcapHit ( Intersection & intersection, const Track * track )

private

Find the matching EKLM 2D hit nearest the intersection point of the track with the crossed EKLM plane.

Definition at line 1607 of file TrackExtrapolateG4e.cc.

{
  double diffBestMagSq = 1.0E60;
  int bestHit = -1;
  int matchingLayer = intersection.layer + 1;
  int matchingEndcap = (intersection.isForward ? 2 : 1);
  G4ThreeVector n(0.0, 0.0, (intersection.isForward ? 1.0 : -1.0));
  for (int h = 0; h < m_klmHit2ds.getEntries(); ++h) {
    KLMHit2d* hit = m_klmHit2ds[h];
    if (hit->getSubdetector() != KLMElementNumbers::c_EKLM)
      continue;
    if (hit->getLayer() != matchingLayer)
      continue;
    if (hit->getSection() != matchingEndcap)
      continue;
    // DIVOT no such function for EKLM!
    // if (hit->isOutOfTime()) continue;
    if (std::fabs(hit->getTime() - m_MeanDt) > m_MaxDt)
      continue;
    G4ThreeVector diff(hit->getPositionX() - intersection.position.x(),
                       hit->getPositionY() - intersection.position.y(),
                       hit->getPositionZ() - intersection.position.z());
    double dn = diff * n; // in cm
    if (std::fabs(dn) > 2.0)
      continue;
    diff -= n * dn;
    if (diff.mag2() < diffBestMagSq) {
      diffBestMagSq = diff.mag2();
      bestHit = h;
    }
  }
 
  if (bestHit >= 0) {
    KLMHit2d* hit = m_klmHit2ds[bestHit];
    intersection.hit = bestHit;
    intersection.isForward = (hit->getSection() == 2);
    intersection.sector = hit->getSector() - 1;
    intersection.time = hit->getTime();
    double localVariance[2] = {m_EndcapScintVariance, m_EndcapScintVariance};
    int nStrips = hit->getXStripMax() - hit->getXStripMin() + 1;
    localVariance[0] *= (nStrips * nStrips); // variance inflated for multi-strip hit
    nStrips = hit->getYStripMax() - hit->getYStripMin() + 1;
    localVariance[1] *= (nStrips * nStrips); // variance inflated for multi-strip hit
    G4ThreeVector hitPos(hit->getPositionX(), hit->getPositionY(), hit->getPositionZ());
    adjustIntersection(intersection, localVariance, hitPos, intersection.position);
    if (intersection.chi2 >= 0.0) {
      // DIVOT no such function for EKLM!
      // hit->isOnTrack(true);
      if (track != nullptr) {
        track->addRelationTo(hit, intersection.chi2);
        RecoTrack* recoTrack = track->getRelatedTo<RecoTrack>();
        if (m_addHitsToRecoTrack) {
          for (const EKLMAlignmentHit& alignmentHit : hit->getRelationsFrom<EKLMAlignmentHit>()) {
            recoTrack->addEKLMHit(&alignmentHit, recoTrack->getNumberOfTotalHits() + 1);
          }
        }
      }
    }
  }
  return intersection.chi2 >= 0.0;
 
}

finishTrack()

void finishTrack ( const ExtState & extState, KLMMuidLikelihood * klmMuidLikelihood, bool isForward )

private

Complete muon identification after end of track extrapolation.

Definition at line 1793 of file TrackExtrapolateG4e.cc.

{
  /* Done with this track: compute KLM likelihoods and fill the relative dataobject. */
  int lastExtLayer = extState.lastBarrelExtLayer + extState.lastEndcapExtLayer + 1;
  unsigned int outcome = MuidElementNumbers::calculateExtrapolationOutcome(
                           isForward, extState.escaped, extState.lastBarrelExtLayer, extState.lastEndcapExtLayer);
  klmMuidLikelihood->setOutcome(outcome);
  klmMuidLikelihood->setIsForward(isForward);
  klmMuidLikelihood->setBarrelExtLayer(extState.lastBarrelExtLayer);
  klmMuidLikelihood->setEndcapExtLayer(extState.lastEndcapExtLayer);
  klmMuidLikelihood->setBarrelHitLayer(extState.lastBarrelHitLayer);
  klmMuidLikelihood->setEndcapHitLayer(extState.lastEndcapHitLayer);
  klmMuidLikelihood->setExtLayer(lastExtLayer);
  klmMuidLikelihood->setHitLayer(((extState.lastEndcapHitLayer == -1) ?
                                  extState.lastBarrelHitLayer :
                                  extState.lastBarrelExtLayer + extState.lastEndcapHitLayer + 1));
  klmMuidLikelihood->setChiSquared(extState.chi2);
  klmMuidLikelihood->setDegreesOfFreedom(extState.nPoint);
  klmMuidLikelihood->setExtLayerPattern(extState.extLayerPattern);
  klmMuidLikelihood->setHitLayerPattern(extState.hitLayerPattern);
  /* Do KLM likelihood calculation. */
  if (outcome != MuidElementNumbers::c_NotReached) { /* Extrapolation reached KLM sensitive volume. */
    double denom = 0.0;
    int charge = klmMuidLikelihood->getCharge();
    std::vector<int> signedPdgVector = MuidElementNumbers::getPDGVector(charge);
    std::map<int, double> mapPdgPDF;
    for (int pdg : signedPdgVector) {
      auto search = m_MuidBuilderMap.find(pdg);
      if (search == m_MuidBuilderMap.end())
        B2FATAL("Something went wrong: PDF for PDG code " << pdg << " not found!");
      double pdf = (search->second)->getPDF(klmMuidLikelihood);
      denom += pdf;
      mapPdgPDF.insert(std::pair<int, double>(std::abs(pdg), pdf));
    }
    if (denom < 1.0E-20)
      klmMuidLikelihood->setJunkPDFValue(true); /* Anomaly: should be very rare. */
    else {
      for (auto const& [pdg, pdf] : mapPdgPDF) {
        klmMuidLikelihood->setPDFValue(pdf, std::abs(pdg));
        if (pdf > 0.0)
          klmMuidLikelihood->setLogL(std::log(pdf), std::abs(pdg));
      }
    }
  }
}

fromG4eToPhasespace()

void fromG4eToPhasespace ( const G4ErrorFreeTrajState & g4eState, G4ErrorSymMatrix & covariance )

private

Convert the geant4e 5x5 covariance to phasespace 6x6 covariance.

Definition at line 1062 of file TrackExtrapolateG4e.cc.

{
 
  // Convert Geant4e covariance matrix with parameters 1/p, lambda, phi, yT, zT (in GeV/c, radians, cm)
  // to phase-space covariance matrix with parameters x, y, z, px, py, pz (in GeV/c, cm)
  // (1/p) = 1/sqrt( px^2 + py^2 + pz^2 )
  // phi = atan( py / px )
  // lambda = asin( pz / sqrt( px^2 + py^2 + pz^2 )
  // xT = x * cos(lambda) * cos(phi) + y * cos(lambda) * sin(phi) + z * sin(lambda)
  // yT = -x * sin(phi) + y * cos(phi)
  // zT = -x * sin(lambda) * cos(phi) - y * sin(lambda) * sin(phi) + z * cos(lambda)
 
  G4ErrorFreeTrajParam param = g4eState.GetParameters();
  double p = 1.0 / (param.GetInvP() * CLHEP::GeV);     // in GeV/c
  double pSq = p * p;
  double lambda = param.GetLambda();    // in radians
  double sinLambda = std::sin(lambda);
  double cosLambda = std::cos(lambda);
  double phi = param.GetPhi();          // in radians
  double sinPhi = std::sin(phi);
  double cosPhi = std::cos(phi);
 
  // Transformation Jacobian 6x5 from Geant4e 5x5 to phase-space 6x6
 
  G4ErrorMatrix jacobian(6, 5, 0); // All entries are initialized to 0
 
  jacobian(4, 1) = -pSq * cosLambda * cosPhi;          // @(px)/@(1/p)
  jacobian(5, 1) = -pSq * cosLambda * sinPhi;          // @(py)/@(1/p)
  jacobian(6, 1) = -pSq * sinLambda;                   // @(pz)/@(1/p)
 
  jacobian(4, 2) = -p * sinLambda * cosPhi;            // @(px)/@(lambda)
  jacobian(5, 2) = -p * sinLambda * sinPhi;            // @(py)/@(lambda)
  jacobian(6, 2) =  p * cosLambda;                     // @(pz)/@(lambda)
 
  jacobian(4, 3) = -p * cosLambda * sinPhi;            // @(px)/@(phi)
  jacobian(5, 3) =  p * cosLambda * cosPhi;            // @(py)/@(phi)
 
  jacobian(1, 4) = -sinPhi;                            // @(x)/@(yT)
  jacobian(2, 4) =  cosPhi;                            // @(y)/@(yT)
 
  jacobian(1, 5) = -sinLambda * cosPhi;                // @(x)/@(zT)
  jacobian(2, 5) = -sinLambda * sinPhi;                // @(y)/@(zT)
  jacobian(3, 5) =  cosLambda;                         // @(z)/@(zT)
 
  G4ErrorTrajErr g4eCov = g4eState.GetError();
  covariance.assign(g4eCov.similarity(jacobian));
 
}

fromPhasespaceToG4e() [1/2]

void fromPhasespaceToG4e ( const G4ThreeVector & momentum, const G4ErrorSymMatrix & covariance, G4ErrorTrajErr & covG4e )

private

Convert the phasespace covariance to geant4e covariance.

Definition at line 1164 of file TrackExtrapolateG4e.cc.

{
 
  // Convert phase-space covariance matrix with parameters x, y, z, px, py, pz (in genfit2 units: cm, GeV/c)
  // to Geant4e covariance matrix with parameters 1/p, lambda, phi, yT, zT (in geant4e units: GeV/c, radians, cm)
  // xT = x * cos(lambda) * cos(phi) + y * cos(lambda) * sin(phi) + z * sin(lambda)
  // yT = -x * sin(phi) + y * cos(phi)
  // zT = -x * sin(lambda) * cos(phi) - y * sin(lambda) * sin(phi) + z * cos(lambda)
  // (1/p) = 1/sqrt( px^2 + py^2 + pz^2 )
  // phi = atan( py / px )
  // lambda = asin( pz / sqrt( px^2 + py^2 + pz^2 )
 
  G4ErrorSymMatrix temp(covariance);
 
  double pInvSq = 1.0 / momentum.mag2();
  double pInv   = std::sqrt(pInvSq);
  double pPerpInv = 1.0 / momentum.perp();
  double sinLambda = momentum.cosTheta();
  double cosLambda = std::sqrt(1.0 - sinLambda * sinLambda);
  double phi = momentum.phi();
  double cosPhi = std::cos(phi);
  double sinPhi = std::sin(phi);
 
  // Transformation Jacobian 5x6 from phase-space 6x6 to Geant4e 5x5
  G4ErrorMatrix jacobian(5, 6, 0);
 
  jacobian(1, 4) = -pInvSq * cosLambda * cosPhi;       // @(1/p)/@(px)
  jacobian(1, 5) = -pInvSq * cosLambda * sinPhi;       // @(1/p)/@(py)
  jacobian(1, 6) = -pInvSq * sinLambda;                // @(1/p)/@(pz)
 
  jacobian(2, 4) = -pInv * sinLambda * cosPhi;         // @(lambda)/@(px)
  jacobian(2, 5) = -pInv * sinLambda * sinPhi;         // @(lambda)/@(py)
  jacobian(2, 6) =  pInv * cosLambda;                  // @(lambda)/@(pz)
 
  jacobian(3, 4) = -pPerpInv * sinPhi;                 // @(phi)/@(px)
  jacobian(3, 5) =  pPerpInv * cosPhi;                 // @(phi)/@(py)
 
  jacobian(4, 1) = -sinPhi;                            // @(yT)/@(x)
  jacobian(4, 2) =  cosPhi;                            // @(yT)/@(y)
 
  jacobian(5, 1) = -sinLambda * cosPhi;                // @(zT)/@(x)
  jacobian(5, 2) = -sinLambda * sinPhi;                // @(zT)/@(y)
  jacobian(5, 3) =  cosLambda;                         // @(zT)/@(z)
  covG4e = temp.similarity(jacobian);
 
}

fromPhasespaceToG4e() [2/2]

void fromPhasespaceToG4e ( const TVector3 & momentum, const TMatrixDSym & covariance, G4ErrorTrajErr & covG4e )

private

Convert the phasespace covariance to geant4e covariance.

Definition at line 1111 of file TrackExtrapolateG4e.cc.

{
 
  // Convert phase-space covariance matrix with parameters x, y, z, px, py, pz (in genfit2 units: cm, GeV/c)
  // to Geant4e covariance matrix with parameters 1/p, lambda, phi, yT, zT (in geant4e units: GeV/c, radians, cm)
  // xT = x * cos(lambda) * cos(phi) + y * cos(lambda) * sin(phi) + z * sin(lambda)
  // yT = -x * sin(phi) + y * cos(phi)
  // zT = -x * sin(lambda) * cos(phi) - y * sin(lambda) * sin(phi) + z * cos(lambda)
  // (1/p) = 1/sqrt( px^2 + py^2 + pz^2 )
  // phi = atan( py / px )
  // lambda = asin( pz / sqrt( px^2 + py^2 + pz^2 )
 
  G4ErrorSymMatrix temp(6, 0);
  for (int k = 0; k < 6; ++k) {
    for (int j = k; j < 6; ++j) {
      temp[j][k] = covariance[j][k];
    }
  }
 
  double pInvSq = 1.0 / momentum.Mag2();
  double pInv   = std::sqrt(pInvSq);
  double pPerpInv = 1.0 / momentum.Perp();
  double sinLambda = momentum.CosTheta();
  double cosLambda = std::sqrt(1.0 - sinLambda * sinLambda);
  double phi = momentum.Phi();
  double cosPhi = std::cos(phi);
  double sinPhi = std::sin(phi);
 
  // Transformation Jacobian 5x6 from phase-space 6x6 to Geant4e 5x5
  G4ErrorMatrix jacobian(5, 6, 0); // All entries are initialized to 0
 
  jacobian(1, 4) = -pInvSq * cosLambda * cosPhi;       // @(1/p)/@(px)
  jacobian(1, 5) = -pInvSq * cosLambda * sinPhi;       // @(1/p)/@(py)
  jacobian(1, 6) = -pInvSq * sinLambda;                // @(1/p)/@(pz)
 
  jacobian(2, 4) = -pInv * sinLambda * cosPhi;         // @(lambda)/@(px)
  jacobian(2, 5) = -pInv * sinLambda * sinPhi;         // @(lambda)/@(py)
  jacobian(2, 6) =  pInv * cosLambda;                  // @(lambda)/@(pz)
 
  jacobian(3, 4) = -pPerpInv * sinPhi;                 // @(phi)/@(px)
  jacobian(3, 5) =  pPerpInv * cosPhi;                 // @(phi)/@(py)
 
  jacobian(4, 1) = -sinPhi;                            // @(yT)/@(x)
  jacobian(4, 2) =  cosPhi;                            // @(yT)/@(y)
 
  jacobian(5, 1) = -sinLambda * cosPhi;                // @(zT)/@(x)
  jacobian(5, 2) = -sinLambda * sinPhi;                // @(zT)/@(y)
  jacobian(5, 3) =  cosLambda;                         // @(zT)/@(z)
 
  covG4e = temp.similarity(jacobian);
 
}

getInstance()

TrackExtrapolateG4e * getInstance ( )

static

Get the singleton's address.

Definition at line 72 of file TrackExtrapolateG4e.cc.

{
  if (m_Singleton == nullptr)
    m_Singleton = new TrackExtrapolateG4e;
  return m_Singleton;
}

getStartPoint()

ExtState getStartPoint ( const Track & b2track, int pdgCode, G4ErrorFreeTrajState & g4eState )

private

Get the start point for a new reconstructed track with specific PDG hypothesis.

Definition at line 969 of file TrackExtrapolateG4e.cc.

{
  ExtState extState = {&b2track, pdgCode, false, 0.0, 0.0,                               // for EXT and MUID
                       G4ThreeVector(0, 0, 1), 0.0, 0, 0, 0, -1, -1, -1, -1, 0, 0, false // for MUID only
                      };
  RecoTrack* recoTrack = b2track.getRelatedTo<RecoTrack>();
  if (recoTrack == nullptr) {
    B2WARNING("Track without associated RecoTrack: skipping extrapolation for this track.");
    extState.pdgCode = 0; // prevent start of extrapolation in swim()
    return extState;
  }
  const genfit::AbsTrackRep* trackRep = recoTrack->getCardinalRepresentation();
  // check for a valid track fit
  if (!recoTrack->wasFitSuccessful(trackRep)) {
    B2WARNING("RecoTrack fit failed for cardinal representation: skipping extrapolation for this track.");
    extState.pdgCode = 0; // prevent start of extrapolation in swim()
    return extState;
  }
  int charge = int(trackRep->getPDGCharge());
  if (charge != 0) {
    extState.pdgCode *= charge;
  } else {
    charge = 1; // should never happen but persist if it does
  }
  TVector3 firstPosition, firstMomentum, lastPosition, lastMomentum; // initialized to zeroes
  TMatrixDSym firstCov(6), lastCov(6);                               // initialized to zeroes
  try {
    const genfit::MeasuredStateOnPlane& firstState = recoTrack->getMeasuredStateOnPlaneFromFirstHit(trackRep);
    trackRep->getPosMomCov(firstState, firstPosition, firstMomentum, firstCov);
    const genfit::MeasuredStateOnPlane& lastState = recoTrack->getMeasuredStateOnPlaneFromLastHit(trackRep);
    trackRep->getPosMomCov(lastState, lastPosition, lastMomentum, lastCov);
    // in genfit units (cm, GeV/c)
    extState.tof = lastState.getTime(); // DIVOT: must be revised when IP profile (reconstructed beam spot) become available!
    if (lastPosition.Mag2() < firstPosition.Mag2()) {
      firstPosition = lastPosition;
      firstMomentum = -lastMomentum;
      firstCov = lastCov;
      trackRep->getPosMomCov(firstState, lastPosition, lastMomentum, lastCov);
      lastMomentum *= -1.0; // extrapolate backwards instead of forwards
      extState.isCosmic = true;
      extState.tof = firstState.getTime(); // DIVOT: must be revised when IP profile (reconstructed beam spot) become available!
    }
 
    G4ParticleDefinition* particle = G4ParticleTable::GetParticleTable()->FindParticle(extState.pdgCode);
    if (extState.pdgCode != trackRep->getPDG()) {
      double pSq = lastMomentum.Mag2();
      double mass = particle->GetPDGMass() / CLHEP::GeV;
      extState.tof *= std::sqrt((pSq + mass * mass) / (pSq + lastState.getMass() * lastState.getMass()));
    }
 
    extState.directionAtIP.set(firstMomentum.Unit().X(), firstMomentum.Unit().Y(), firstMomentum.Unit().Z());
    if (m_MagneticField != 0.0) { // in gauss
      double radius = (firstMomentum.Perp() * CLHEP::GeV / CLHEP::eV) / (CLHEP::c_light * charge * m_MagneticField); // in cm
      double centerPhi = extState.directionAtIP.phi() - M_PI_2;
      double centerX = firstPosition.X() + radius * std::cos(centerPhi);
      double centerY = firstPosition.Y() + radius * std::sin(centerPhi);
      double pocaPhi = atan2(charge * centerY, charge * centerX) + M_PI;
      double ipPerp = extState.directionAtIP.perp();
      if (ipPerp > 0.0) {
        extState.directionAtIP.setX(+std::sin(pocaPhi) * ipPerp);
        extState.directionAtIP.setY(-std::cos(pocaPhi) * ipPerp);
      }
    } else {
      // No field: replace flaky covariance matrix with a diagonal one measured in 1.5T field
      // for a 10 GeV/c track ... and replace momentum magnitude with fixed 10 GeV/c
      lastCov *= 0.0;
      lastCov[0][0] = 5.0E-5;
      lastCov[1][1] = 1.0E-7;
      lastCov[2][2] = 5.0E-4;
      lastCov[3][3] = 3.5E-3;
      lastCov[4][4] = 3.5E-3;
      lastMomentum = lastMomentum.Unit() * 10.0;
    }
 
    G4ThreeVector posG4e(lastPosition.X() * CLHEP::cm, lastPosition.Y() * CLHEP::cm,
                         lastPosition.Z() * CLHEP::cm); // in Geant4 units (mm)
    G4ThreeVector momG4e(lastMomentum.X() * CLHEP::GeV, lastMomentum.Y() * CLHEP::GeV,
                         lastMomentum.Z() * CLHEP::GeV);  // in Geant4 units (MeV/c)
    G4ErrorSymMatrix covG4e(5, 0); // in Geant4e units (GeV/c, cm)
    fromPhasespaceToG4e(lastMomentum, lastCov, covG4e); // in Geant4e units (GeV/c, cm)
    g4eState.SetData("g4e_" + particle->GetParticleName(), posG4e, momG4e);
    g4eState.SetParameters(posG4e, momG4e); // compute private-state parameters from momG4e
    g4eState.SetError(covG4e);
  } catch (const genfit::Exception&) {
    B2WARNING("genfit::MeasuredStateOnPlane() exception: skipping extrapolation for this track. initial momentum = ("
              << firstMomentum.X() << "," << firstMomentum.Y() << "," << firstMomentum.Z() << ")");
    extState.pdgCode = 0; // prevent start of extrapolation in swim()
  }
 
  return extState;
}

getVolumeID()

void getVolumeID ( const G4TouchableHandle & touch, Const::EDetector & detID, int & copyID )

private

Get the physical volume information for a geant4 physical volume.

Definition at line 883 of file TrackExtrapolateG4e.cc.

{
 
  // default values
  detID = Const::EDetector::invalidDetector;
  copyID = 0;
 
  G4VPhysicalVolume* pv = touch->GetVolume(0);
  std::map<G4VPhysicalVolume*, enum VolTypes>::iterator it = m_EnterExit->find(pv);
  if (it == m_EnterExit->end())
    return;
 
  switch (it->second) {
    case VOLTYPE_CDC:
      detID = Const::EDetector::CDC;
      copyID = pv->GetCopyNo();
      return;
    case VOLTYPE_TOP1:
      detID = Const::EDetector::TOP;
      copyID = -(pv->GetCopyNo()); // negative to distinguish module and quartz hits
      return;
    case VOLTYPE_TOP2:
      detID = Const::EDetector::TOP;
      if (touch->GetHistoryDepth() >= 1)
        copyID = touch->GetVolume(1)->GetCopyNo();
      return;
    case VOLTYPE_TOP3:
      detID = Const::EDetector::TOP;
      if (touch->GetHistoryDepth() >= 2)
        copyID = touch->GetVolume(2)->GetCopyNo();
      return;
    case VOLTYPE_ARICH1:
      detID = Const::EDetector::ARICH;
      copyID = 12345;
      return;
    case VOLTYPE_ARICH2:
      detID = Const::EDetector::ARICH;
      copyID = 6789;
      return;
    case VOLTYPE_ARICH3:
      detID = Const::EDetector::ARICH;
      if (touch->GetHistoryDepth() >= 2)
        copyID = touch->GetVolume(2)->GetCopyNo();
      return;
    case VOLTYPE_ECL:
      detID = Const::EDetector::ECL;
      copyID = ECL::ECLGeometryPar::Instance()->ECLVolumeToCellID(touch());
      return;
    case VOLTYPE_BKLM1: // BKLM RPCs
      detID = Const::EDetector::BKLM;
      if (touch->GetHistoryDepth() == DEPTH_RPC) {
        // int plane = touch->GetCopyNumber(0);
        int layer = touch->GetCopyNumber(4);
        int sector = touch->GetCopyNumber(6);
        int section = touch->GetCopyNumber(7);
        copyID = BKLMElementNumbers::moduleNumber(section, sector, layer);
      }
      return;
    case VOLTYPE_BKLM2: // BKLM scints
      detID = Const::EDetector::BKLM;
      if (touch->GetHistoryDepth() == DEPTH_SCINT) {
        int strip = touch->GetCopyNumber(1);
        int plane = (touch->GetCopyNumber(2) == BKLM_INNER) ?
                    BKLMElementNumbers::c_PhiPlane :
                    BKLMElementNumbers::c_ZPlane;
        int layer = touch->GetCopyNumber(6);
        int sector = touch->GetCopyNumber(8);
        int section = touch->GetCopyNumber(9);
        copyID = BKLMElementNumbers::channelNumber(
                   section, sector, layer, plane, strip);
        BKLMStatus::setMaximalStrip(copyID, strip);
      }
      return;
    case VOLTYPE_EKLM:
      detID = Const::EDetector::EKLM;
      copyID = EKLMElementNumbers::Instance().stripNumber(
                 touch->GetVolume(7)->GetCopyNo(),
                 touch->GetVolume(6)->GetCopyNo(),
                 touch->GetVolume(5)->GetCopyNo(),
                 touch->GetVolume(4)->GetCopyNo(),
                 touch->GetVolume(1)->GetCopyNo());
      return;
  }
 
}

initialize() [1/2]

void initialize	(	double	meanDt,
		double	maxDt,
		double	maxSeparation,
		double	maxKLMTrackClusterDistance,
		double	maxECLTrackClusterDistance,
		double	minPt,
		double	minKE,
		bool	addHitsToRecoTrack,
		std::vector< Const::ChargedStable > &	hypotheses )

Initialize for track extrapolation by the MUID module.

Parameters

meanDt	Mean value of the in-time window (ns).
maxDt	Half-width of the in-time window (ns).
maxSeparation	Maximum separation between track crossing and matching hit in detector plane (number of sigmas).
maxKLMTrackClusterDistance	Maximum distance between associated track and KLMCluster (cm), criterion for matching relation Track->KLMCluster on MDST.
maxECLTrackClusterDistance	Maximum distance between associated track and ECLCluster (cm).
minPt	Minimum transverse momentum to begin extrapolation (GeV/c).
minKE	Minimum kinetic energy to continue extrapolation (GeV/c).
addHitsToRecoTrack	Parameter to add the found hits also to the reco tracks or not. Is turned off by default.
hypotheses	Vector of charged-particle hypotheses used in extrapolation of each track.

Definition at line 191 of file TrackExtrapolateG4e.cc.

{
  m_MuidInitialized = true;
  m_addHitsToRecoTrack = addHitsToRecoTrack;
 
  // Define required objects, register the new ones and relations
  m_eclClusters.isRequired();
  m_klmHit2ds.isRequired();
  m_klmClusters.isRequired();
  m_recoTracks.isRequired();
  m_tracks.isRequired();
  m_extHits.registerInDataStore();
  m_klmMuidLikelihoods.registerInDataStore();
  m_klmMuidHits.registerInDataStore();
  m_trackClusterSeparations.registerInDataStore();
  m_tracks.registerRelationTo(m_extHits);
  m_tracks.registerRelationTo(m_klmMuidLikelihoods);
  m_tracks.registerRelationTo(m_klmMuidHits);
  m_tracks.registerRelationTo(m_klmHit2ds);
  m_tracks.registerRelationTo(m_trackClusterSeparations);
  m_tracks.registerRelationTo(m_klmClusters);
  m_klmClusters.registerRelationTo(m_trackClusterSeparations);
  m_eclClusters.registerRelationTo(m_extHits);
  RecoTrack::registerRequiredRelations(m_recoTracks);
 
  // Save the in-time cut's central value and width for valid hits
  m_MeanDt = meanDt;
  m_MaxDt = maxDt;
 
  // Save the magnetic field z component (gauss) at the origin
  m_MagneticField = BFieldManager::getField(0, 0, 0).Z() / Unit::T * CLHEP::tesla / CLHEP::gauss;
 
  // Convert from sigma to variance for hit-position uncertainty
  m_MaxDistSqInVariances = maxKLMTrackHitDistance * maxKLMTrackHitDistance;
 
  // Convert user cutoff values to geant4 units
  m_MaxKLMTrackClusterDistance = std::max(0.0, maxKLMTrackClusterDistance) * CLHEP::cm; // mm
  m_MaxECLTrackClusterDistance = std::max(0.0, maxECLTrackClusterDistance) * CLHEP::cm; // mm
  m_MinPt = std::max(0.0, minPt) * CLHEP::GeV;
  m_MinKE = std::max(0.0, minKE) * CLHEP::GeV;
 
  // Save pointer to the list of particle hypotheses for EXT extrapolation
  m_HypothesesMuid = &hypotheses;
 
  // Define the list of volumes that will have their entry and/or
  // exit points stored during the extrapolation.
  registerVolumes();
 
  // Store the address of the ExtManager (used later)
  m_ExtMgr = Simulation::ExtManager::GetManager();
 
  // Set up the EXT-specific geometry (might have already been done by EXT)
  if (m_TargetExt == nullptr) {
    if (!m_COILGeometryPar.isValid())
      B2FATAL("Coil geometry data are not available.");
    double offsetZ = m_COILGeometryPar->getGlobalOffsetZ();
    double rMinCoil = m_COILGeometryPar->getCryoRmin();
    double halfLength = m_COILGeometryPar->getCryoLength();
    m_COILGeometryPar.addCallback([this, &offsetZ, &rMinCoil, &halfLength]() {
      offsetZ = m_COILGeometryPar->getGlobalOffsetZ();
      rMinCoil = m_COILGeometryPar->getCryoRmin();
      halfLength = m_COILGeometryPar->getCryoLength();
    });
    m_TargetExt = new Simulation::ExtCylSurfaceTarget(rMinCoil, offsetZ - halfLength, offsetZ + halfLength);
    G4ErrorPropagatorData::GetErrorPropagatorData()->SetTarget(m_TargetExt);
  }
  if (!m_BeamPipeGeo.isValid())
    B2FATAL("Beam pipe geometry data are not available.");
  double beampipeRadius = m_BeamPipeGeo->getParameter("Lv2OutBe.R2") * CLHEP::cm; // mm
  m_BeamPipeGeo.addCallback([this, &beampipeRadius]() {
    beampipeRadius = m_BeamPipeGeo->getParameter("Lv2OutBe.R2") * CLHEP::cm;
  });
  m_MinRadiusSq = beampipeRadius * beampipeRadius; // mm^2
 
  // Set up the MUID-specific geometry
  bklm::GeometryPar* bklmGeometry = bklm::GeometryPar::instance();
  const EKLM::GeometryData& eklmGeometry = EKLM::GeometryData::Instance();
  m_BarrelHalfLength = bklmGeometry->getHalfLength() * CLHEP::cm; // in G4 units (mm)
  m_EndcapHalfLength = 0.5 * eklmGeometry.getSectionPosition()->getLength(); // in G4 units (mm)
  m_OffsetZ = bklmGeometry->getOffsetZ() * CLHEP::cm; // in G4 units (mm)
  double minZ = m_OffsetZ - (m_BarrelHalfLength + 2.0 * m_EndcapHalfLength);
  double maxZ = m_OffsetZ + (m_BarrelHalfLength + 2.0 * m_EndcapHalfLength);
  m_BarrelNSector = bklmGeometry->getNSector();
  m_BarrelMaxR = bklmGeometry->getOuterRadius() * CLHEP::cm / std::cos(M_PI / m_BarrelNSector); // in G4 units (mm)
  m_TargetMuid = new Simulation::ExtCylSurfaceTarget(m_BarrelMaxR, minZ, maxZ);
  G4ErrorPropagatorData::GetErrorPropagatorData()->SetTarget(m_TargetMuid);
 
  m_BarrelHalfLength /= CLHEP::cm;                   // now in G4e units (cm)
  m_EndcapHalfLength /= CLHEP::cm;                   // now in G4e units (cm)
  m_OffsetZ /= CLHEP::cm;                            // now in G4e units (cm)
  m_BarrelMinR = bklmGeometry->getGap1InnerRadius(); // in G4e units (cm)
  m_BarrelMaxR /= CLHEP::cm;                         // now in G4e units (cm)
  m_EndcapMinR = eklmGeometry.getSectionPosition()->getInnerR() / CLHEP::cm; // in G4e units (cm)
  m_EndcapMaxR = eklmGeometry.getSectionPosition()->getOuterR() / CLHEP::cm; // in G4e units (cm)
  m_EndcapMiddleZ = m_BarrelHalfLength + m_EndcapHalfLength;                // in G4e units (cm)
 
  // Measurement uncertainties and acceptance windows
  double width = eklmGeometry.getStripGeometry()->getWidth() / CLHEP::cm; // in G4e units (cm)
  m_EndcapScintVariance = width * width / 12.0;
  width = bklmGeometry->getScintHalfWidth() * 2.0;                        // in G4e units (cm)
  m_BarrelScintVariance = width * width / 12.0;
  for (int layer = 1; layer <= BKLMElementNumbers::getMaximalLayerNumber(); ++layer) {
    const bklm::Module* module =
      bklmGeometry->findModule(BKLMElementNumbers::c_ForwardSection, 1, layer);
    width = module->getPhiStripWidth(); // in G4e units (cm)
    m_BarrelPhiStripVariance[layer - 1] = width * width / 12.0;
    width = module->getZStripWidth(); // in G4e units (cm)
    m_BarrelZStripVariance[layer - 1] = width * width / 12.0;
  }
 
  // KLM geometry (for associating KLM hit with extrapolated crossing point)
  m_OutermostActiveBarrelLayer = BKLMElementNumbers::getMaximalLayerNumber() - 1; // zero-based counting
  for (int sector = 1; sector <= BKLMElementNumbers::getMaximalSectorNumber(); ++sector) {
    double phi = M_PI_4 * (sector - 1);
    m_BarrelSectorPerp[sector - 1].set(std::cos(phi), std::sin(phi), 0.0);
    m_BarrelSectorPhi[sector - 1].set(-std::sin(phi), std::cos(phi), 0.0);
  }
  KLMChannelIndex klmLayers(KLMChannelIndex::c_IndexLevelLayer);
  for (KLMChannelIndex& klmLayer : klmLayers) {
    if (klmLayer.getSubdetector() == KLMElementNumbers::c_BKLM)
      m_BarrelModuleMiddleRadius[klmLayer.getSection()][klmLayer.getSector() - 1][klmLayer.getLayer() - 1] =
        bklmGeometry->getActiveMiddleRadius(klmLayer.getSection(), klmLayer.getSector(), klmLayer.getLayer()); // in G4e units (cm)
  }
  double dz(eklmGeometry.getLayerShiftZ() / CLHEP::cm); // in G4e units (cm)
  double z0((eklmGeometry.getSectionPosition()->getZ()
             + eklmGeometry.getLayerShiftZ()
             - 0.5 * eklmGeometry.getSectionPosition()->getLength()
             - 0.5 * eklmGeometry.getLayerPosition()->getLength()) / CLHEP::cm); // in G4e units (cm)
  m_OutermostActiveForwardEndcapLayer = m_eklmElementNumbers->getMaximalDetectorLayerNumber(EKLMElementNumbers::c_ForwardSection) -
                                        1; // zero-based counting
  m_OutermostActiveBackwardEndcapLayer = m_eklmElementNumbers->getMaximalDetectorLayerNumber(EKLMElementNumbers::c_BackwardSection) -
                                         1; // zero-based counting
  for (int layer = 1; layer <= EKLMElementNumbers::getMaximalLayerNumber(); ++layer) {
    m_EndcapModuleMiddleZ[layer - 1] = z0 + dz * (layer - 1); // in G4e units (cm)
  }
 
  m_eklmTransformData = &(EKLM::TransformDataGlobalAligned::Instance());
}

initialize() [2/2]

void initialize	(	double	minPt,
		double	minKE,
		std::vector< Const::ChargedStable > &	hypotheses )

Initialize for track extrapolation by the EXT module.

Parameters

minPt	Minimum transverse momentum to begin extrapolation (GeV/c).
minKE	Minimum kinetic energy to continue extrapolation (GeV/c).
hypotheses	Vector of charged-particle hypotheses used in extrapolation of each track.

Definition at line 138 of file TrackExtrapolateG4e.cc.

{
  m_ExtInitialized = true;
 
  // Define required objects, register the new ones and relations
  m_recoTracks.isRequired();
  m_tracks.isRequired();
  m_extHits.registerInDataStore();
  m_tracks.registerRelationTo(m_extHits);
 
  // Save the magnetic field z component (gauss) at the origin
  m_MagneticField = BFieldManager::getField(0, 0, 0).Z() / Unit::T * CLHEP::tesla / CLHEP::gauss;
 
  // Convert user cutoff values to geant4 units
  m_MinPt = std::max(0.0, minPt) * CLHEP::GeV;
  m_MinKE = std::max(0.0, minKE) * CLHEP::GeV;
 
  // Save pointer to the list of particle hypotheses for EXT extrapolation
  m_HypothesesExt = &hypotheses;
 
  // Define the list of volumes that will have their entry and/or
  // exit points stored during the extrapolation.
  registerVolumes();
 
  // Store the address of the ExtManager (used later)
  m_ExtMgr = Simulation::ExtManager::GetManager();
 
  // Set up the EXT-specific geometry (might have already been done by MUID)
  if (m_TargetExt == nullptr) {
    if (!m_COILGeometryPar.isValid())
      B2FATAL("Coil geometry data are not available.");
    double offsetZ = m_COILGeometryPar->getGlobalOffsetZ();
    double rMinCoil = m_COILGeometryPar->getCryoRmin();
    double halfLength = m_COILGeometryPar->getCryoLength();
    m_COILGeometryPar.addCallback([this, &offsetZ, &rMinCoil, &halfLength]() {
      offsetZ = m_COILGeometryPar->getGlobalOffsetZ();
      rMinCoil = m_COILGeometryPar->getCryoRmin();
      halfLength = m_COILGeometryPar->getCryoLength();
    });
    m_TargetExt = new Simulation::ExtCylSurfaceTarget(rMinCoil, offsetZ - halfLength, offsetZ + halfLength);
    G4ErrorPropagatorData::GetErrorPropagatorData()->SetTarget(m_TargetExt);
  }
  if (!m_BeamPipeGeo.isValid())
    B2FATAL("Beam pipe geometry data are not available.");
  double beampipeRadius = m_BeamPipeGeo->getParameter("Lv2OutBe.R2") * CLHEP::cm; // mm
  m_BeamPipeGeo.addCallback([this, &beampipeRadius]() {
    beampipeRadius = m_BeamPipeGeo->getParameter("Lv2OutBe.R2") * CLHEP::cm;
  });
  m_MinRadiusSq = beampipeRadius * beampipeRadius; // mm^2
}

registerVolumes()

void registerVolumes ( )

private

Register the list of geant4 physical volumes whose entry/exit points will be saved during extrapolation.

Definition at line 819 of file TrackExtrapolateG4e.cc.

{
  G4PhysicalVolumeStore* pvStore = G4PhysicalVolumeStore::GetInstance();
  if (pvStore->size() == 0) {
    B2FATAL("No geometry defined. Please create the geometry first.");
  }
  if (m_EnterExit != nullptr) // Only do this once
    return;
  m_EnterExit = new std::map<G4VPhysicalVolume*, enum VolTypes>;
  m_BKLMVolumes = new std::vector<G4VPhysicalVolume*>;
  for (std::vector<G4VPhysicalVolume*>::iterator iVol = pvStore->begin();
       iVol != pvStore->end(); ++iVol) {
    const G4String name = (*iVol)->GetName();
    // Do not store ExtHits in CDC, so let's start from TOP and ARICH.
    // TOP doesn't have one envelope; it has 16 "TOPModule"s
    if (name.find("TOPModule") != std::string::npos) {
      (*m_EnterExit)[*iVol] = VOLTYPE_TOP1;
    }
    // TOP quartz bar (=sensitive)
    else if (name.find("_TOPPrism_") != std::string::npos ||
             name.find("_TOPBarSegment") != std::string::npos ||
             name.find("_TOPMirrorSegment") != std::string::npos) {
      (*m_EnterExit)[*iVol] = VOLTYPE_TOP2;
    }
    // TOP quartz glue (not sensitive?)
    else if (name.find("TOPBarSegment1Glue") != std::string::npos ||
             name.find("TOPBarSegment2Glue") != std::string::npos ||
             name.find("TOPMirrorSegmentGlue") != std::string::npos) {
      (*m_EnterExit)[*iVol] = VOLTYPE_TOP3;
      // ARICH volumes
    } else if (name == "ARICH.AerogelSupportPlate") {
      (*m_EnterExit)[*iVol] = VOLTYPE_ARICH1;
    } else if (name == "ARICH.AerogelImgPlate") {
      (*m_EnterExit)[*iVol] = VOLTYPE_ARICH2;
    } else if (name.find("ARICH.HAPDWindow") != std::string::npos) {
      (*m_EnterExit)[*iVol] = VOLTYPE_ARICH3;
    }
    // ECL crystal
    else if (name.find("lv_barrel_crystal_") != std::string::npos ||
             name.find("lv_forward_crystal_") != std::string::npos ||
             name.find("lv_backward_crystal_") != std::string::npos) {
      (*m_EnterExit)[*iVol] = VOLTYPE_ECL;
    }
    // Barrel KLM: BKLM.Layer**GasPhysical for RPCs or BKLM.Layer**ChimneyGasPhysical for RPCs
    //             BKLM.ScintActiveType*Physical for scintillator strips
    else if (name.compare(0, 5, "BKLM.") == 0) {
      if (name.find("GasPhysical") != std::string::npos) {
        (*m_EnterExit)[*iVol] = VOLTYPE_BKLM1;
      } else if (name.find("ScintActiveType") != std::string::npos) {
        (*m_EnterExit)[*iVol] = VOLTYPE_BKLM2;
      } else if ((name.find("ScintType") != std::string::npos) ||
                 (name.find("ElectrodePhysical") != std::string::npos)) {
        m_BKLMVolumes->push_back(*iVol);
      }
    }
    // Endcap KLM: StripSensitive_*
    else if (name.compare(0, 14, "StripSensitive") == 0) {
      (*m_EnterExit)[*iVol] = VOLTYPE_EKLM;
    }
  }
 
}

swim() [1/2]

void swim ( ExtState & extState, G4ErrorFreeTrajState & g4eState )

private

Swim a single track (EXT) until it stops or leaves the target cylinder.

Definition at line 738 of file TrackExtrapolateG4e.cc.

{
  if (extState.pdgCode == 0)
    return;
  if (g4eState.GetMomentum().perp() <= m_MinPt)
    return;
  if (m_TargetExt->GetDistanceFromPoint(g4eState.GetPosition()) < 0.0)
    return;
  G4ParticleDefinition* particle = G4ParticleTable::GetParticleTable()->FindParticle(extState.pdgCode);
  double mass = particle->GetPDGMass();
  double minPSq = (mass + m_MinKE) * (mass + m_MinKE) - mass * mass;
  G4ErrorMode propagationMode = (extState.isCosmic ? G4ErrorMode_PropBackwards : G4ErrorMode_PropForwards);
  m_ExtMgr->InitTrackPropagation(propagationMode);
  while (true) {
    const G4int        errCode       = m_ExtMgr->PropagateOneStep(&g4eState, propagationMode);
    G4Track*           track         = g4eState.GetG4Track();
    const G4Step*      step          = track->GetStep();
    const G4StepPoint* preStepPoint  = step->GetPreStepPoint();
    const G4StepPoint* postStepPoint = step->GetPostStepPoint();
    G4TouchableHandle  preTouch      = preStepPoint->GetTouchableHandle();
    G4VPhysicalVolume* pVol          = preTouch->GetVolume();
    const G4int        preStatus     = preStepPoint->GetStepStatus();
    const G4int        postStatus    = postStepPoint->GetStepStatus();
    G4ThreeVector      pos           = track->GetPosition(); // this is at postStepPoint
    G4ThreeVector      mom           = track->GetMomentum(); // ditto
    // First step on this track?
    if (extState.isCosmic)
      mom = -mom;
    if (preStatus == fUndefined) {
      if (m_EnterExit->find(pVol) != m_EnterExit->end()) {
        createExtHit(EXT_FIRST, extState, g4eState, preStepPoint, preTouch);
      }
    }
    // Ignore the zero-length step by PropagateOneStep() at each boundary
    if (step->GetStepLength() > 0.0) {
      double dt = step->GetDeltaTime();
      double dl = step->GetStepLength() / track->GetMaterial()->GetRadlen();
      if (preStatus == fGeomBoundary) {      // first step in this volume?
        if (m_EnterExit->find(pVol) != m_EnterExit->end()) {
          createExtHit(EXT_ENTER, extState, g4eState, preStepPoint, preTouch);
        }
      }
      if (extState.isCosmic) {
        extState.tof -= dt;
        extState.length -= dl;
      } else {
        extState.tof += dt;
        extState.length += dl;
      }
      // Last step in this volume?
      if (postStatus == fGeomBoundary) {
        if (m_EnterExit->find(pVol) != m_EnterExit->end()) {
          createExtHit(EXT_EXIT, extState, g4eState, postStepPoint, preTouch);
        }
      }
    }
    // Post-step momentum too low?
    if (errCode || (mom.mag2() < minPSq)) {
      if (m_EnterExit->find(pVol) != m_EnterExit->end()) {
        createExtHit(EXT_STOP, extState, g4eState, postStepPoint, preTouch);
      }
      break;
    }
    // Detect escapes from the imaginary target cylinder.
    if (m_TargetExt->GetDistanceFromPoint(pos) < 0.0) {
      if (m_EnterExit->find(pVol) != m_EnterExit->end()) {
        createExtHit(EXT_ESCAPE, extState, g4eState, postStepPoint, preTouch);
      }
      break;
    }
    // Stop extrapolating as soon as the track curls inward too much
    if (pos.perp2() < m_MinRadiusSq) {
      break;
    }
  } // track-extrapolation "infinite" loop
 
  m_ExtMgr->EventTermination(propagationMode);
 
}

swim() [2/2]

void swim ( ExtState & extState, G4ErrorFreeTrajState & g4eState, const std::vector< std::pair< ECLCluster *, G4ThreeVector > > * eclClusterInfo, const std::vector< std::pair< KLMCluster *, G4ThreeVector > > * klmClusterInfo, std::vector< std::map< const Track *, double > > * bklmHitUsed )

private

Swim a single track (MUID) until it stops or leaves the target cylinder.

Definition at line 491 of file TrackExtrapolateG4e.cc.

{
  if (extState.pdgCode == 0)
    return;
  if (g4eState.GetMomentum().perp() <= m_MinPt)
    return;
  if (m_TargetMuid->GetDistanceFromPoint(g4eState.GetPosition()) < 0.0)
    return;
  G4ParticleDefinition* particle = G4ParticleTable::GetParticleTable()->FindParticle(extState.pdgCode);
  double mass = particle->GetPDGMass();
  double minPSq = (mass + m_MinKE) * (mass + m_MinKE) - mass * mass;
  // Create structures for ECLCluster-Track matching
  std::vector<double> eclClusterDistance;
  std::vector<ExtHit> eclHit1, eclHit2, eclHit3;
  if (eclClusterInfo != nullptr) {
    eclClusterDistance.resize(eclClusterInfo->size(), 1.0E10); // "positive infinity"
    ExtHit tempExtHit(.0, extState.pdgCode, Const::EDetector::ECL, 0, EXT_FIRST,
                      extState.isCosmic,
                      G4ThreeVector(), G4ThreeVector(), G4ErrorSymMatrix(6));
    eclHit1.resize(eclClusterInfo->size(), tempExtHit);
    eclHit2.resize(eclClusterInfo->size(), tempExtHit);
    eclHit3.resize(eclClusterInfo->size(), tempExtHit);
  }
  // Create structures for KLMCluster-Track matching
  std::vector<TrackClusterSeparation> klmHit;
  if (klmClusterInfo != nullptr) {
    klmHit.resize(klmClusterInfo->size()); // initialize each to huge distance
  }
  KLMMuidLikelihood* klmMuidLikelihood = m_klmMuidLikelihoods.appendNew(); // rest of this object will be filled later
  klmMuidLikelihood->setPDGCode(extState.pdgCode);
  if (extState.track != nullptr)
    extState.track->addRelationTo(klmMuidLikelihood);
  G4ErrorMode propagationMode = (extState.isCosmic ? G4ErrorMode_PropBackwards : G4ErrorMode_PropForwards);
  m_ExtMgr->InitTrackPropagation(propagationMode);
  while (true) {
    const G4int        errCode       = m_ExtMgr->PropagateOneStep(&g4eState, propagationMode);
    G4Track*           track         = g4eState.GetG4Track();
    const G4Step*      step          = track->GetStep();
    const G4StepPoint* preStepPoint  = step->GetPreStepPoint();
    const G4StepPoint* postStepPoint = step->GetPostStepPoint();
    G4TouchableHandle  preTouch      = preStepPoint->GetTouchableHandle();
    G4VPhysicalVolume* pVol          = preTouch->GetVolume();
    const G4int        preStatus     = preStepPoint->GetStepStatus();
    const G4int        postStatus    = postStepPoint->GetStepStatus();
    G4ThreeVector      pos           = track->GetPosition(); // this is at postStepPoint
    G4ThreeVector      mom           = track->GetMomentum(); // ditto
    // Ignore the zero-length step by PropagateOneStep() at each boundary
    if (extState.isCosmic) mom = -mom;
    if (step->GetStepLength() > 0.0) {
      double dt = step->GetDeltaTime();
      double dl = step->GetStepLength() / track->GetMaterial()->GetRadlen();
      if (preStatus == fGeomBoundary) {      // first step in this volume?
        if (m_TargetExt->GetDistanceFromPoint(pos) < 0.0) { // only hits outside ECL during muid
          if (m_EnterExit->find(pVol) != m_EnterExit->end()) {
            createExtHit(EXT_ENTER, extState, g4eState, preStepPoint, preTouch);
          }
        }
      }
      if (extState.isCosmic) {
        extState.tof -= dt;
        extState.length -= dl;
      } else {
        extState.tof += dt;
        extState.length += dl;
      }
      if (postStatus == fGeomBoundary) { // last step in this volume?
        if (m_TargetExt->GetDistanceFromPoint(pos) < 0.0) { // only hits outside ECL during muid
          if (m_EnterExit->find(pVol) != m_EnterExit->end()) {
            createExtHit(EXT_EXIT, extState, g4eState, postStepPoint, preTouch);
          }
        }
      }
      if (createMuidHit(extState, g4eState, klmMuidLikelihood, bklmHitUsed)) {
        // Force geant4e to update its G4Track from the Kalman-updated state
        m_ExtMgr->GetPropagator()->SetStepN(0);
      }
      if (eclClusterInfo != nullptr) {
        for (unsigned int c = 0; c < eclClusterInfo->size(); ++c) {
          G4ThreeVector eclPos((*eclClusterInfo)[c].second);
          G4ThreeVector prePos(preStepPoint->GetPosition());
          G4ThreeVector diff(prePos - eclPos);
          double distance = diff.mag();
          if (distance < m_MaxECLTrackClusterDistance) {
            // fallback ECLNEAR in case no ECLCROSS is found
            if (distance < eclClusterDistance[c]) {
              eclClusterDistance[c] = distance;
              G4ErrorSymMatrix covariance(6, 0);
              fromG4eToPhasespace(g4eState, covariance);
              eclHit3[c].update(EXT_ECLNEAR, extState.tof, pos / CLHEP::cm, mom / CLHEP::GeV, covariance);
            }
            // find position of crossing of the track with the ECLCluster's sphere
            if (eclHit1[c].getStatus() == EXT_FIRST) {
              if (pos.mag2() >= eclPos.mag2()) {
                double r = eclPos.mag();
                double preD = prePos.mag() - r;
                double postD = pos.mag() - r;
                double f = postD / (postD - preD);
                G4ThreeVector midPos = pos + (prePos - pos) * f;
                double tof = extState.tof + dt * f * (extState.isCosmic ? +1 : -1); // in ns, at end of step
                G4ErrorSymMatrix covariance(6, 0);
                fromG4eToPhasespace(g4eState, covariance);
                eclHit1[c].update(EXT_ECLCROSS, tof, midPos / CLHEP::cm, mom / CLHEP::GeV, covariance);
              }
            }
          }
          // find closest distance to the radial line to the ECLCluster
          if (eclHit2[c].getStatus() == EXT_FIRST) {
            G4ThreeVector delta(pos - prePos);
            G4ThreeVector perp(eclPos.cross(delta));
            double perpMag2 = perp.mag2();
            if (perpMag2 > 1.0E-10) {
              double dist = std::fabs(diff * perp) / std::sqrt(perpMag2);
              if (dist < m_MaxECLTrackClusterDistance) {
                double f = eclPos * (prePos.cross(perp)) / perpMag2;
                if ((f > -0.5) && (f <= 1.0)) {
                  G4ThreeVector midPos(prePos + f * delta);
                  double length = extState.length + dl * (1.0 - f) * (extState.isCosmic ? +1 : -1);
                  G4ErrorSymMatrix covariance(6, 0);
                  fromG4eToPhasespace(g4eState, covariance);
                  eclHit2[c].update(EXT_ECLDL, length, midPos / CLHEP::cm, mom / CLHEP::GeV, covariance);
                }
              }
            }
          }
        }
      }
      if (klmClusterInfo != nullptr) {
        for (unsigned int c = 0; c < klmClusterInfo->size(); ++c) {
          G4ThreeVector klmPos = (*klmClusterInfo)[c].second;
          G4ThreeVector separation = klmPos - pos;
          double distance = separation.mag();
          if (distance < klmHit[c].getDistance()) {
            klmHit[c].setDistance(distance);
            klmHit[c].setTrackClusterAngle(mom.angle(separation));
            klmHit[c].setTrackClusterSeparationAngle(mom.angle(klmPos));
            klmHit[c].setTrackRotationAngle(extState.directionAtIP.angle(mom));
            klmHit[c].setTrackClusterInitialSeparationAngle(extState.directionAtIP.angle(klmPos));
          }
        }
      }
    }
    // Post-step momentum too low?
    if (errCode || (mom.mag2() < minPSq)) {
      break;
    }
    // Detect escapes from the imaginary target cylinder.
    if (m_TargetMuid->GetDistanceFromPoint(pos) < 0.0) {
      break;
    }
    // Stop extrapolating as soon as the track curls inward too much
    if (pos.perp2() < m_MinRadiusSq) {
      break;
    }
  } // track-extrapolation "infinite" loop
 
  m_ExtMgr->EventTermination(propagationMode);
 
  finishTrack(extState, klmMuidLikelihood, (g4eState.GetPosition().z() > m_OffsetZ));
 
  if (eclClusterInfo != nullptr) {
    for (unsigned int c = 0; c < eclClusterInfo->size(); ++c) {
      if (eclHit1[c].getStatus() != EXT_FIRST) {
        ExtHit* h = m_extHits.appendNew(eclHit1[c]);
        (*eclClusterInfo)[c].first->addRelationTo(h);
        if (extState.track != nullptr) {
          extState.track->addRelationTo(h);
        }
      }
      if (eclHit2[c].getStatus() != EXT_FIRST) {
        ExtHit* h = m_extHits.appendNew(eclHit2[c]);
        (*eclClusterInfo)[c].first->addRelationTo(h);
        if (extState.track != nullptr) {
          extState.track->addRelationTo(h);
        }
      }
      if (eclHit3[c].getStatus() != EXT_FIRST) {
        ExtHit* h = m_extHits.appendNew(eclHit3[c]);
        (*eclClusterInfo)[c].first->addRelationTo(h);
        if (extState.track != nullptr) {
          extState.track->addRelationTo(h);
        }
      }
    }
  }
 
  if (klmClusterInfo != nullptr) {
    // here we set a relation only to the closest KLMCluster
    // and we don't set any relation if the distance is too large
    double minDistance = m_MaxKLMTrackClusterDistance;
    unsigned int closestCluster = 0;
    for (unsigned int c = 0; c < klmClusterInfo->size(); ++c) {
      if (klmHit[c].getDistance() > 1.0E9) {
        continue;
      }
      TrackClusterSeparation* h = m_trackClusterSeparations.appendNew(klmHit[c]);
      if (h != nullptr) {
        (*klmClusterInfo)[c].first->addRelationTo(h); // relation KLMCluster to TrackClusterSeparation
        if (extState.track != nullptr) {
          extState.track->addRelationTo(h); // relation Track to TrackClusterSeparation
        }
      }
      if (klmHit[c].getDistance() < minDistance) {
        closestCluster = c;
        minDistance = klmHit[c].getDistance();
      }
    }
    if (minDistance < m_MaxKLMTrackClusterDistance) {
      // set the relation Track to KLMCluster, using the distance as weight
      // but for being consistent with the basf2 conventions, store the distance in cm
      if (extState.track != nullptr) {
        extState.track->addRelationTo((*klmClusterInfo)[closestCluster].first, 1. / (minDistance / CLHEP::cm));
      }
    }
  }
}

terminate()

void terminate

(

bool

flag

)

Terminates this singleton.

Parameters

flag	True if called by Muid module, false if called by Ext module.

Definition at line 420 of file TrackExtrapolateG4e.cc.

{
  if (m_DefaultHypotheses != nullptr)
    delete m_DefaultHypotheses;
  if (byMuid) {
    delete m_TargetMuid;
    for (auto const& muidBuilder : m_MuidBuilderMap)
      delete muidBuilder.second;
  }
  if (m_TargetExt != nullptr) {
    delete m_TargetExt;
    m_TargetExt = nullptr;
  }
  if (m_EnterExit != nullptr) {
    delete m_EnterExit;
    delete m_BKLMVolumes;
    m_ExtMgr->RunTermination();
    m_EnterExit = nullptr;
    m_BKLMVolumes = nullptr;
  }
}

Member Data Documentation

m_addHitsToRecoTrack

bool m_addHitsToRecoTrack = false

private

Parameter to add the found hits also to the reco tracks or not. Is turned off by default.

Definition at line 424 of file TrackExtrapolateG4e.h.

m_BarrelHalfLength

double m_BarrelHalfLength

private

half-length (cm) of the barrel

Definition at line 375 of file TrackExtrapolateG4e.h.

m_BarrelMaxR

double m_BarrelMaxR

private

maximum radius (cm) of the barrel

Definition at line 369 of file TrackExtrapolateG4e.h.

m_BarrelMinR

double m_BarrelMinR

private

minimum radius (cm) of the barrel

Definition at line 372 of file TrackExtrapolateG4e.h.

m_BarrelModuleMiddleRadius

double m_BarrelModuleMiddleRadius[2][BKLMElementNumbers::getMaximalSectorNumber()+1][BKLMElementNumbers::getMaximalLayerNumber()+1]

private

hit-plane radius (cm) at closest distance to IP of each barrel end | sector | layer

Definition at line 390 of file TrackExtrapolateG4e.h.

m_BarrelNSector

int m_BarrelNSector

private

Number of barrel sectors.

Definition at line 366 of file TrackExtrapolateG4e.h.

m_BarrelPhiStripVariance

double m_BarrelPhiStripVariance[BKLMElementNumbers::getMaximalLayerNumber()+1]

private

BKLM RPC phi-measuring strip position variance (cm^2) by layer.

Definition at line 381 of file TrackExtrapolateG4e.h.

m_BarrelScintVariance

double m_BarrelScintVariance

private

BKLM scintillator strip position variance (cm^2)

Definition at line 387 of file TrackExtrapolateG4e.h.

m_BarrelSectorPerp

G4ThreeVector m_BarrelSectorPerp[BKLMElementNumbers::getMaximalSectorNumber()+1]

private

normal unit vector of each barrel sector

Definition at line 394 of file TrackExtrapolateG4e.h.

m_BarrelSectorPhi

G4ThreeVector m_BarrelSectorPhi[BKLMElementNumbers::getMaximalSectorNumber()+1]

private

azimuthal unit vector of each barrel sector

Definition at line 397 of file TrackExtrapolateG4e.h.

m_BarrelZStripVariance

double m_BarrelZStripVariance[BKLMElementNumbers::getMaximalLayerNumber()+1]

private

BKLM RPC z-measuring strip position variance (cm^2) by layer.

Definition at line 384 of file TrackExtrapolateG4e.h.

m_BeamPipeGeo

DBObjPtr<BeamPipeGeo> m_BeamPipeGeo

private

Conditions-database object for beam pipe geometry.

Definition at line 357 of file TrackExtrapolateG4e.h.

m_BKLMVolumes

std::vector<G4VPhysicalVolume*>* m_BKLMVolumes

private

Pointers to BKLM geant4 sensitive (physical) volumes.

Definition at line 345 of file TrackExtrapolateG4e.h.

m_COILGeometryPar

DBObjPtr<COILGeometryPar> m_COILGeometryPar

private

Conditions-database object for COIL geometry.

Definition at line 354 of file TrackExtrapolateG4e.h.

m_DefaultHypotheses

std::vector<Const::ChargedStable>* m_DefaultHypotheses

private

Default ChargedStable hypotheses (needed as call argument but not used)

Definition at line 339 of file TrackExtrapolateG4e.h.

m_eclClusters

StoreArray<ECLCluster> m_eclClusters

private

ECL clusters.

Definition at line 448 of file TrackExtrapolateG4e.h.

m_eklmElementNumbers

const EKLMElementNumbers* m_eklmElementNumbers

private

EKLM element numbers.

Definition at line 430 of file TrackExtrapolateG4e.h.

m_eklmTransformData

const EKLM::TransformDataGlobalAligned* m_eklmTransformData

private

EKLM transformation data.

Definition at line 436 of file TrackExtrapolateG4e.h.

m_EndcapHalfLength

double m_EndcapHalfLength

private

half-length (cm) of either endcap

Definition at line 409 of file TrackExtrapolateG4e.h.

m_EndcapMaxR

double m_EndcapMaxR

private

maximum radius (cm) of the endcaps

Definition at line 400 of file TrackExtrapolateG4e.h.

m_EndcapMiddleZ

double m_EndcapMiddleZ

private

midpoint along z (cm) of the forward endcap from the KLM midpoint

Definition at line 406 of file TrackExtrapolateG4e.h.

m_EndcapMinR

double m_EndcapMinR

private

minimum radius (cm) of the endcaps

Definition at line 403 of file TrackExtrapolateG4e.h.

m_EndcapModuleMiddleZ

double m_EndcapModuleMiddleZ[BKLMElementNumbers::getMaximalLayerNumber()+1]

private

hit-plane z (cm) of each IP layer relative to KLM midpoint

Definition at line 421 of file TrackExtrapolateG4e.h.

m_EndcapScintVariance

double m_EndcapScintVariance

private

EKLM scintillator strip position variance (cm^2)

Definition at line 418 of file TrackExtrapolateG4e.h.

m_EnterExit

std::map<G4VPhysicalVolume*, enum VolTypes>* m_EnterExit

private

Pointers to geant4 physical volumes whose entry/exit points will be saved.

Definition at line 342 of file TrackExtrapolateG4e.h.

m_extHits

StoreArray<ExtHit> m_extHits

private

Ext hits.

Definition at line 451 of file TrackExtrapolateG4e.h.

m_ExtInitialized

bool m_ExtInitialized

private

Flag to indicate that EXT initialize() has been called.

Definition at line 300 of file TrackExtrapolateG4e.h.

m_ExtMgr

Simulation::ExtManager* m_ExtMgr

private

Pointer to the ExtManager singleton.

Definition at line 330 of file TrackExtrapolateG4e.h.

m_HypothesesExt

const std::vector<Const::ChargedStable>* m_HypothesesExt

private

ChargedStable hypotheses for EXT.

Definition at line 333 of file TrackExtrapolateG4e.h.

m_HypothesesMuid

const std::vector<Const::ChargedStable>* m_HypothesesMuid

private

ChargedStable hypotheses for MUID.

Definition at line 336 of file TrackExtrapolateG4e.h.

m_klmChannelStatus

DBObjPtr<KLMChannelStatus> m_klmChannelStatus

private

Conditions-database object for KLM channel status.

Definition at line 439 of file TrackExtrapolateG4e.h.

m_klmClusters

StoreArray<KLMCluster> m_klmClusters

private

KLM clusters.

Definition at line 457 of file TrackExtrapolateG4e.h.

m_klmElementNumbers

const KLMElementNumbers* m_klmElementNumbers

private

KLM element numbers.

Definition at line 433 of file TrackExtrapolateG4e.h.

m_klmHit2ds

StoreArray<KLMHit2d> m_klmHit2ds

private

KLM 2d hits.

Definition at line 454 of file TrackExtrapolateG4e.h.

m_klmLikelihoodParameters

DBObjPtr<KLMLikelihoodParameters> m_klmLikelihoodParameters

private

Conditions-database object for KLM likelihood parameters.

Definition at line 445 of file TrackExtrapolateG4e.h.

m_klmMuidHits

StoreArray<KLMMuidHit> m_klmMuidHits

private

KLM muid hits.

Definition at line 460 of file TrackExtrapolateG4e.h.

m_klmMuidLikelihoods

StoreArray<KLMMuidLikelihood> m_klmMuidLikelihoods

private

KLM muid likelihoods.

Definition at line 463 of file TrackExtrapolateG4e.h.

m_klmStripEfficiency

DBObjPtr<KLMStripEfficiency> m_klmStripEfficiency

private

Conditions-database object for KLM strip efficiency.

Definition at line 442 of file TrackExtrapolateG4e.h.

m_MagneticField

double m_MagneticField

private

Magnetic field z component (gauss) at origin.

Definition at line 312 of file TrackExtrapolateG4e.h.

m_MaxDistSqInVariances

double m_MaxDistSqInVariances

private

user-defined maximum squared-distance (in number of variances) for matching hit to extrapolation

Definition at line 315 of file TrackExtrapolateG4e.h.

m_MaxDt

double m_MaxDt

private

Coincidence window half-width for in-time KLM hits (ns)

Definition at line 309 of file TrackExtrapolateG4e.h.

m_MaxECLTrackClusterDistance

double m_MaxECLTrackClusterDistance

private

user-defined maximum distance (mm) between ECLCluster and associated track (for EID)

Definition at line 321 of file TrackExtrapolateG4e.h.

m_MaxKLMTrackClusterDistance

double m_MaxKLMTrackClusterDistance

private

user-defined maximum distance (mm) between KLMCluster and associated track (for KLID)

Definition at line 318 of file TrackExtrapolateG4e.h.

m_MeanDt

double m_MeanDt

private

Mean hit - trigger time (ns)

Definition at line 306 of file TrackExtrapolateG4e.h.

m_MinKE

double m_MinKE

private

Minimum kinetic energy in MeV for extrapolation to continue.

Definition at line 327 of file TrackExtrapolateG4e.h.

m_MinPt

double m_MinPt

private

Minimum transverse momentum in MeV/c for extrapolation to be started.

Definition at line 324 of file TrackExtrapolateG4e.h.

m_MinRadiusSq

double m_MinRadiusSq

private

Minimum squared radius (cm) outside of which extrapolation will continue.

Definition at line 360 of file TrackExtrapolateG4e.h.

m_MuidBuilderMap

std::map<int, MuidBuilder*> m_MuidBuilderMap

private

PDF for the charged final state particle hypotheses.

Definition at line 427 of file TrackExtrapolateG4e.h.

m_MuidInitialized

bool m_MuidInitialized

private

Flag to indicate that MUID initialize() has been called.

Definition at line 303 of file TrackExtrapolateG4e.h.

m_OffsetZ

double m_OffsetZ

private

offset (cm) along z axis of KLM midpoint from IP

Definition at line 363 of file TrackExtrapolateG4e.h.

m_OutermostActiveBackwardEndcapLayer

int m_OutermostActiveBackwardEndcapLayer

private

outermost backward-endcap layer that is active for muon identification (user-defined)

Definition at line 415 of file TrackExtrapolateG4e.h.

m_OutermostActiveBarrelLayer

int m_OutermostActiveBarrelLayer

private

outermost barrel layer that is active for muon identification (user-defined)

Definition at line 378 of file TrackExtrapolateG4e.h.

m_OutermostActiveForwardEndcapLayer

int m_OutermostActiveForwardEndcapLayer

private

outermost forward-endcap layer that is active for muon identification (user-defined)

Definition at line 412 of file TrackExtrapolateG4e.h.

m_recoTracks

StoreArray<RecoTrack> m_recoTracks

private

Reco tracks.

Definition at line 466 of file TrackExtrapolateG4e.h.

m_Singleton

TrackExtrapolateG4e * m_Singleton = nullptr

staticprivate

Stores pointer to the singleton class.

Definition at line 297 of file TrackExtrapolateG4e.h.

m_TargetExt

Simulation::ExtCylSurfaceTarget* m_TargetExt

private

virtual "target" cylinder for EXT (boundary beyond which extrapolation ends)

Definition at line 348 of file TrackExtrapolateG4e.h.

m_TargetMuid

Simulation::ExtCylSurfaceTarget* m_TargetMuid

private

virtual "target" cylinder for MUID (boundary beyond which extrapolation ends)

Definition at line 351 of file TrackExtrapolateG4e.h.

m_trackClusterSeparations

StoreArray<TrackClusterSeparation> m_trackClusterSeparations

private

Track cluster separation.

Definition at line 472 of file TrackExtrapolateG4e.h.

m_tracks

StoreArray<Track> m_tracks

private

Tracks.

Definition at line 469 of file TrackExtrapolateG4e.h.

---

The documentation for this class was generated from the following files:

• tracking/trackExtrapolateG4e/include/TrackExtrapolateG4e.h
• tracking/trackExtrapolateG4e/src/TrackExtrapolateG4e.cc