SegmentTrackVarSet Class Reference

Class to compute floating point variables from a segment to track match which can be recorded as a flat TNtuple or serve as input to a MVA method. More...

#include <SegmentTrackVarSet.h>

Inheritance diagram for SegmentTrackVarSet:

Public Types
using	Object
	Type from which variables should be extracted.

Public Member Functions
bool	extract (const BaseSegmentTrackFilter::Object *testPair) final
	Generate and assign the contained variables.
std::vector< Named< Float_t * > >	getNamedVariables (const std::string &prefix) override
	Getter for the named references to the individual variables Base implementation returns empty vector.
std::vector< Named< Float_t * > >	getNamedVariables ()
	Getter for the named references to the individual variables.
MayBePtr< Float_t >	find (const std::string &varName) override
	Pointer to the variable with the given name.
virtual bool	extract (const Object *obj)
	Main method that extracts the variable values from the complex object.
bool	extract (const Object &obj)
	Method for extraction from an object instead of a pointer.
std::map< std::string, Float_t >	getNamedValues (const std::string &prefix) const
	Getter for a map of names to float values.
std::map< std::string, Float_t >	getNamedValues () const
	Getter for a map of names to float values.
void	initialize () override
	Receive and dispatch signal before the start of the event processing.
void	beginRun () override
	Receive and dispatch signal for the beginning of a new run.
void	beginEvent () override
	Receive and dispatch signal for the start of a new event.
void	endRun () override
	Receive and dispatch signal for the end of the run.
void	terminate () override
	Receive and dispatch Signal for termination of the event processing.
virtual void	exposeParameters (ModuleParamList *moduleParamList, const std::string &prefix)
	Expose the set of parameters of the filter to the module parameter list.

Protected Member Functions
Float_t	get () const
	Getter for the value of the ith variable. Static version.
Float_t &	var ()
	Reference getter for the value of the ith variable. Static version.
AssignFinite< Float_t >	finitevar ()
	Reference getter for the value of the ith variable. Transforms non-finite values to finite value.
void	addProcessingSignalListener (ProcessingSignalListener *psl)
	Register a processing signal listener to be notified.
int	getNProcessingSignalListener ()
	Get the number of currently registered listeners.

Static Protected Member Functions
static constexpr int	named (const char *name)
	Getter for the index from the name.

Private Types
using	Super
	Type of the super class.

Private Attributes
FixedSizeNamedFloatTuple< SegmentTrackVarNames >	m_variables
	Memory for nVars floating point values.
std::vector< ProcessingSignalListener * >	m_subordinaryProcessingSignalListeners
	References to subordinary signal processing listener contained in this findlet.
bool	m_initialized
	Flag to keep track whether initialization happened before.
bool	m_terminated
	Flag to keep track whether termination happened before.
std::string	m_initializedAs
	Name of the type during initialisation.

Static Private Attributes
static const size_t	nVars
	Number of floating point values represented by this class.

Detailed Description

Class to compute floating point variables from a segment to track match which can be recorded as a flat TNtuple or serve as input to a MVA method.

Definition at line 77 of file SegmentTrackVarSet.h.

Member Typedef Documentation

Object

using Object

inherited

Type from which variables should be extracted.

Definition at line 44 of file VarSet.h.

Super

using Super

privateinherited

Type of the super class.

Definition at line 40 of file VarSet.h.

Member Function Documentation

addProcessingSignalListener()

void addProcessingSignalListener ( ProcessingSignalListener * psl )

protectedinherited

Register a processing signal listener to be notified.

Definition at line 53 of file CompositeProcessingSignalListener.cc.

{
  m_subordinaryProcessingSignalListeners.push_back(psl);
}

beginEvent()

void beginEvent ( )

overrideinherited

Receive and dispatch signal for the start of a new event.

Definition at line 36 of file CompositeProcessingSignalListener.cc.

{
  Super::beginEvent();
  for (ProcessingSignalListener* psl : m_subordinaryProcessingSignalListeners) {
    psl->beginEvent();
  }
}

beginRun()

void beginRun ( )

overrideinherited

Receive and dispatch signal for the beginning of a new run.

Definition at line 33 of file CompositeProcessingSignalListener.cc.

{
  Super::beginRun();
  for (ProcessingSignalListener* psl : m_subordinaryProcessingSignalListeners) {
    psl->beginRun();
  }
}

endRun()

void endRun ( )

overrideinherited

Receive and dispatch signal for the end of the run.

Definition at line 39 of file CompositeProcessingSignalListener.cc.

{
  for (ProcessingSignalListener* psl : reversedRange(m_subordinaryProcessingSignalListeners)) {
    psl->endRun();
  }
  Super::endRun();
}

exposeParameters()

virtual void exposeParameters ( ModuleParamList * moduleParamList, const std::string & prefix )

inlinevirtualinherited

Expose the set of parameters of the filter to the module parameter list.

Parameters

moduleParamList	Module parameter list to add the new parameters to
prefix	Prefix (or sometimes postfix) to be added to all nely added parameters.

Definition at line 47 of file CompositeProcessingSignalListener.h.

      {};

extract() [1/3]

bool extract ( const BaseSegmentTrackFilter::Object * testPair )

final

Generate and assign the contained variables.

Definition at line 35 of file SegmentTrackVarSet.cc.

{
  const CDCTrack* track = testPair->getFrom();
  const CDCSegment2D* segment = testPair->getTo();
 
  assert(segment);
  assert(track);
 
  double maxmimumTrajectoryDistanceFront = 0;
  double maxmimumTrajectoryDistanceBack = 0;
  double maxmimumHitDistanceFront = 0;
  double maxmimumHitDistanceBack = 0;
  double outOfCDC = 0; // 0 means no, 1 means yes
  double hitsInSameRegion = 0;
  double hitsInCommon = 0;
 
  const CDCRecoHit2D& front = segment->front();
  const CDCRecoHit2D& back = segment->back();
 
  // Calculate distances
  const CDCTrajectory3D& trajectoryTrack3D = track->getStartTrajectory3D();
  const CDCTrajectory2D& trajectoryTrack2D = trajectoryTrack3D.getTrajectory2D();
  const CDCTrajectorySZ& szTrajectoryTrack = trajectoryTrack3D.getTrajectorySZ();
  double radius = trajectoryTrack2D.getGlobalCircle().absRadius();
 
  maxmimumTrajectoryDistanceFront = trajectoryTrack2D.getDist2D(front.getWireHit().getRefPos2D());
  maxmimumTrajectoryDistanceBack = trajectoryTrack2D.getDist2D(back.getWireHit().getRefPos2D());
 
  var<named("z_distance")>() = 0;
  var<named("theta_distance")>() = 0;
 
  if (segment->getStereoKind() == EStereoKind::c_Axial) {
    CDCTrajectory2D& trajectorySegment = segment->getTrajectory2D();
    if (not trajectoryTrack2D.isFitted()) {
      const CDCRiemannFitter& fitter = CDCRiemannFitter::getFitter();
      fitter.update(trajectorySegment, *segment);
    }
  } else {
    CDCObservations2D observations;
    for (const CDCRecoHit2D& recoHit : *segment) {
      const CDCRLWireHit& rlWireHit = recoHit.getRLWireHit();
      CDCRecoHit3D recoHit3D = CDCRecoHit3D::reconstruct(rlWireHit, trajectoryTrack2D);
      double s = recoHit3D.getArcLength2D();
      double z = recoHit3D.getRecoZ();
      observations.fill(s, z);
    }
 
    if (observations.size() > 3) {
      const CDCSZFitter& fitter = CDCSZFitter::getFitter();
      const CDCTrajectorySZ& szTrajectorySegments = fitter.fit(observations);
 
      double startZTrack = szTrajectoryTrack.getZ0();
      double startZSegments = szTrajectorySegments.getZ0();
 
      var<named("z_distance")>() = startZTrack - startZSegments;
      var<named("theta_distance")>() = szTrajectoryTrack.getTanLambda() - szTrajectorySegments.getTanLambda();
    }
  }
 
  // Calculate if it is out of the CDC
  const ROOT::Math::XYZVector& frontRecoPos3D = front.reconstruct3D(trajectoryTrack2D);
  const ROOT::Math::XYZVector& backRecoPos3D = back.reconstruct3D(trajectoryTrack2D);
 
  if (segment->getStereoKind() != EStereoKind::c_Axial) {
    double forwardZ = front.getWire().getWireLine().forwardZ();
    double backwardZ = front.getWire().getWireLine().backwardZ();
 
    if (frontRecoPos3D.z() > forwardZ or frontRecoPos3D.z() < backwardZ or backRecoPos3D.z() > forwardZ
        or backRecoPos3D.z() < backwardZ) {
      outOfCDC = 1.0;
    }
  }
 
  // Get perpS of track in the beginning and the end
  double perpSOfFront = trajectoryTrack2D.calcArcLength2D(segment->front().getRecoPos2D());
  double perpSOfBack = trajectoryTrack2D.calcArcLength2D(segment->back().getRecoPos2D());
 
  double perpSMinimum = std::min(perpSOfFront, perpSOfBack);
  double perpSMaximum = std::max(perpSOfFront, perpSOfBack);
 
  // Count number of hits in the same region
  for (const CDCRecoHit3D& recoHit : *track) {
    if (recoHit.getArcLength2D() < 0.8 * perpSMinimum or
        recoHit.getArcLength2D() > 1.2 * perpSMaximum) {
      continue;
    }
    if (recoHit.getISuperLayer() == segment->getISuperLayer()) {
      hitsInSameRegion++;
    } else if (abs(recoHit.getISuperLayer() - segment->getISuperLayer()) == 1) {
      double distanceFront = (front.getWireHit().getRefPos2D() - recoHit.getRecoPos2D()).R();
      if (distanceFront > maxmimumHitDistanceFront) {
        maxmimumHitDistanceFront = distanceFront;
      }
      double distanceBack = (back.getWireHit().getRefPos2D() - recoHit.getRecoPos2D()).R();
      if (distanceBack > maxmimumHitDistanceBack) {
        maxmimumHitDistanceBack = distanceBack;
      }
    }
  }
 
  // Count number of common hits
  for (const CDCRecoHit3D& trackHit : *track) {
    if (std::find_if(segment->begin(), segment->end(), [&trackHit](const CDCRecoHit2D & segmentHit) {
    return segmentHit.getWireHit().getHit() == trackHit.getWireHit().getHit();
    }) != segment->end()) {
      hitsInCommon += 1;
    }
  }
 
  // Make a fit with all the hits and one with only the hits in the near range
  CDCObservations2D observationsFull;
  CDCObservations2D observationsNeigh;
 
  // Collect the observations
  bool isAxialSegment = segment->getStereoKind() != EStereoKind::c_Axial;
 
  for (const CDCRecoHit3D& recoHit : *track) {
    if (isAxialSegment and recoHit.getStereoKind() == EStereoKind::c_Axial) {
      observationsFull.fill(recoHit.getWireHit().getRefPos2D());
      if (abs(recoHit.getISuperLayer() - segment->getISuperLayer()) < 3) {
        observationsNeigh.fill(recoHit.getWireHit().getRefPos2D());
      }
    } else if (not isAxialSegment and recoHit.getStereoKind() != EStereoKind::c_Axial) {
      double s = recoHit.getArcLength2D();
      double z = recoHit.getRecoZ();
      observationsFull.fill(s, z);
      if (abs(recoHit.getISuperLayer() - segment->getISuperLayer()) < 3) {
        observationsNeigh.fill(s, z);
      }
    }
  }
 
  const CDCTrajectorySZ& trajectorySZ = track->getStartTrajectory3D().getTrajectorySZ();
  double tanLambda = trajectorySZ.getTanLambda();
 
  bool hasZInformation = tanLambda != 0;
  double max_hit_z_distance = -1;
  double sum_hit_z_distance = 0;
  double stereo_quad_tree_distance = 0;
 
  if (hasZInformation) {
    double thetaFirstSegmentHit = -10;
 
    for (const CDCRecoHit2D& recoHit2D : *segment) {
      ROOT::Math::XYZVector reconstructedPosition = recoHit2D.reconstruct3D(trajectoryTrack2D);
      const ROOT::Math::XYVector& recoPos2D = recoHit2D.getRecoPos2D();
      double perpS = trajectoryTrack2D.calcArcLength2D(recoPos2D);
 
 
      double current_z_distance = std::abs(trajectorySZ.getZDist(perpS, reconstructedPosition.z()));
      if (std::isnan(current_z_distance)) {
        continue;
      }
 
      if (thetaFirstSegmentHit == -10) {
        thetaFirstSegmentHit = reconstructedPosition.Theta();
      }
      sum_hit_z_distance += current_z_distance;
      if (current_z_distance > max_hit_z_distance) {
        max_hit_z_distance = current_z_distance;
      }
    }
 
    double thetaTrack = trajectoryTrack3D.getFlightDirection3DAtSupport().Theta();
    stereo_quad_tree_distance = thetaTrack - thetaFirstSegmentHit;
  }
 
 
  for (const CDCRecoHit2D& recoHit : *segment) {
    if (isAxialSegment) {
      observationsFull.fill(recoHit.getRecoPos2D());
      observationsNeigh.fill(recoHit.getRecoPos2D());
    } else {
      const CDCRLWireHit& rlWireHit = recoHit.getRLWireHit();
      CDCRecoHit3D recoHit3D = CDCRecoHit3D::reconstruct(rlWireHit, trajectoryTrack2D);
      double s = recoHit3D.getArcLength2D();
      double z = recoHit3D.getRecoZ();
      observationsFull.fill(s, z);
      observationsNeigh.fill(s, z);
    }
  }
 
  // Do the fit
  var<named("fit_neigh")>() = 0;
  var<named("fit_full")>() = 0;
  if (segment->getStereoKind() == EStereoKind::c_Axial) {
    const CDCRiemannFitter& fitter = CDCRiemannFitter::getFitter();
    var<named("fit_full")>() = fitter.fit(observationsFull).getPValue();
  } else {
    const CDCSZFitter& fitter = CDCSZFitter::getFitter();
    var<named("fit_full")>() = toFinite(fitter.fit(observationsFull).getPValue(), 0);
 
    if (observationsNeigh.size() > 3) {
      var<named("fit_neigh")>() = toFinite(fitter.fit(observationsNeigh).getPValue(), 0);
    } else {
      var<named("fit_neigh")>() = 0;
    }
  }
 
  if (observationsFull.size() == observationsNeigh.size()) {
    var<named("fit_neigh")>() = -1;
  }
 
  var<named("is_stereo")>() = segment->getStereoKind() != EStereoKind::c_Axial;
  var<named("segment_size")>() = segment->size();
  var<named("track_size")>() = track->size();
  var<named("mean_hit_z_distance")>() = sum_hit_z_distance;
  var<named("max_hit_z_distance")>() = max_hit_z_distance;
  var<named("stereo_quad_tree_distance")>() = toFinite(stereo_quad_tree_distance, 0);
 
  var<named("pt_of_track")>() = toFinite(std::isnan(trajectoryTrack2D.getAbsMom2D()) ? 0.0 : trajectoryTrack2D.getAbsMom2D(), 0);
  var<named("track_is_curler")>() = VectorUtil::hasNAN(trajectoryTrack2D.getExit());
 
  var<named("superlayer_already_full")>() = not VectorUtil::hasNAN(trajectoryTrack2D.getOuterExit()) and hitsInSameRegion > 5;
 
  var<named("maxmimum_trajectory_distance_front")>() = toFinite(maxmimumTrajectoryDistanceFront, 999);
  var<named("maxmimum_trajectory_distance_back")>() = toFinite(maxmimumTrajectoryDistanceBack, 999);
 
  var<named("maxmimum_hit_distance_front")>() = maxmimumHitDistanceFront;
  var<named("maxmimum_hit_distance_back")>() = maxmimumHitDistanceBack;
 
  var<named("out_of_CDC")>() = outOfCDC;
  var<named("hits_in_same_region")>() = hitsInSameRegion;
 
  var<named("number_of_hits_in_common")>() = hitsInCommon;
 
  var<named("segment_super_layer")>() = segment->getISuperLayer();
 
  double phiBetweenTrackAndSegment = ROOT::Math::VectorUtil::DeltaPhi(trajectoryTrack2D.getMom2DAtSupport(),
                                     segment->front().getRecoPos2D());
 
  var<named("phi_between_track_and_segment")>() = toFinite(phiBetweenTrackAndSegment, 0);
  var<named("perp_s_of_front")>() = toFinite(perpSOfFront / radius, 0);
  var<named("perp_s_of_back")>() = toFinite(perpSOfBack / radius, 0);
 
  return true;
}

extract() [2/3]

bool extract ( const Object & obj )

inlineinherited

Method for extraction from an object instead of a pointer.

Definition at line 57 of file BaseVarSet.h.

      {
        return extract(&obj);
      }

extract() [3/3]

virtual bool extract ( const Object * obj )

inlinevirtualinherited

Main method that extracts the variable values from the complex object.

Returns indication whether the extraction could be completed successfully. Base implementation returns always true.

Parameters

obj	dummy object not used

Returns

: always true for the base implementation

Definition at line 51 of file BaseVarSet.h.

      {
        return true;
      }

find()

MayBePtr< Float_t > find ( const std::string & varName )

inlineoverridevirtualinherited

Pointer to the variable with the given name.

Returns nullptr if not found.

Reimplemented from BaseVarSet< SegmentTrackVarNames::Object >.

Definition at line 64 of file VarSet.h.

      {
        return m_variables.find(varName);
      }

finitevar()

AssignFinite< Float_t > finitevar ( )

inlineprotectedinherited

Reference getter for the value of the ith variable. Transforms non-finite values to finite value.

Definition at line 130 of file VarSet.h.

      {
        static_assert(I < nVars, "Requested variable index exceeds number of variables.");
        return AssignFinite<Float_t>(m_variables[I]);
      }

get()

Float_t get ( ) const

inlineprotectedinherited

Getter for the value of the ith variable. Static version.

Definition at line 85 of file VarSet.h.

      {
        static_assert(I < nVars, "Requested variable index exceeds number of variables.");
        return m_variables.get(I);
      }

getNamedValues() [1/2]

std::map< std::string, Float_t > getNamedValues ( ) const

inlineinherited

Getter for a map of names to float values.

Definition at line 96 of file BaseVarSet.h.

      {
        const std::string prefix = "";
        return this->getNamedValues(prefix);
      }

getNamedValues() [2/2]

std::map< std::string, Float_t > getNamedValues ( const std::string & prefix ) const

inlineinherited

Getter for a map of names to float values.

Parameters

prefix

Name prefix to apply to all variable names.

Definition at line 84 of file BaseVarSet.h.

      {
        std::map<std::string, Float_t> result;
        std::vector<Named<Float_t*> > namedVariables = this->getNamedVariables(prefix);
        for (const Named<Float_t*>& namedVariable : namedVariables) {
          const Float_t* variable = namedVariable;
          result[namedVariable.getName()] = *variable;
        }
        return result;
      }

getNamedVariables() [1/2]

std::vector< Named< Float_t * > > getNamedVariables ( )

inlineinherited

Getter for the named references to the individual variables.

Definition at line 74 of file BaseVarSet.h.

      {
        const std::string prefix = "";
        return this->getNamedVariables(prefix);
      }

getNamedVariables() [2/2]

std::vector< Named< Float_t * > > getNamedVariables ( const std::string & prefix )

inlineoverridevirtualinherited

Getter for the named references to the individual variables Base implementation returns empty vector.

Reimplemented from BaseVarSet< SegmentTrackVarNames::Object >.

Definition at line 55 of file VarSet.h.

      {
        return m_variables.getNamedVariables(prefix);
      }

getNProcessingSignalListener()

int getNProcessingSignalListener ( )

protectedinherited

Get the number of currently registered listeners.

Definition at line 56 of file CompositeProcessingSignalListener.cc.

{
  return m_subordinaryProcessingSignalListeners.size();
}

initialize()

void initialize ( )

overrideinherited

Receive and dispatch signal before the start of the event processing.

Definition at line 30 of file CompositeProcessingSignalListener.cc.

{
  Super::initialize();
  for (ProcessingSignalListener* psl : m_subordinaryProcessingSignalListeners) {
    psl->initialize();
  }
}

named()

static constexpr int named ( const char * name )

inlinestaticconstexprprotectedinherited

Getter for the index from the name.

Looks through the associated names and returns the right index if found Returns nVars (one after the last element) if not found.

Parameters

name	Name of the sought variable

Returns

Index of the name, nVars if not found.

Definition at line 78 of file VarSet.h.

      {
        return index<nVars>(AVarNames::getName, name);
      }

terminate()

void terminate ( )

overrideinherited

Receive and dispatch Signal for termination of the event processing.

Definition at line 42 of file CompositeProcessingSignalListener.cc.

{
  for (ProcessingSignalListener* psl : reversedRange(m_subordinaryProcessingSignalListeners)) {
    psl->terminate();
  }
  Super::terminate();
}

var()

Float_t & var ( )

inlineprotectedinherited

Reference getter for the value of the ith variable. Static version.

Definition at line 93 of file VarSet.h.

      {
        static_assert(I < nVars, "Requested variable index exceeds number of variables.");
        return m_variables[I];
      }

Member Data Documentation

m_initialized

bool m_initialized

privateinherited

Flag to keep track whether initialization happened before.

Definition at line 52 of file ProcessingSignalListener.h.

m_initializedAs

std::string m_initializedAs

privateinherited

Name of the type during initialisation.

Definition at line 58 of file ProcessingSignalListener.h.

m_subordinaryProcessingSignalListeners

std::vector<ProcessingSignalListener*> m_subordinaryProcessingSignalListeners

privateinherited

References to subordinary signal processing listener contained in this findlet.

Definition at line 60 of file CompositeProcessingSignalListener.h.

m_terminated

bool m_terminated

privateinherited

Flag to keep track whether termination happened before.

Definition at line 55 of file ProcessingSignalListener.h.

m_variables

FixedSizeNamedFloatTuple<SegmentTrackVarNames> m_variables

privateinherited

Memory for nVars floating point values.

Definition at line 138 of file VarSet.h.

nVars

const size_t nVars

staticprivateinherited

Number of floating point values represented by this class.

Definition at line 48 of file VarSet.h.

---

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

• tracking/trackFindingCDC/filters/segmentTrack/include/SegmentTrackVarSet.h
• tracking/trackFindingCDC/filters/segmentTrack/src/SegmentTrackVarSet.cc