Filter for two hits plus a virtual IP using QualityEstimators. More...

#include <TwoHitVirtualIPQIFilter.h>

Inheritance diagram for TwoHitVirtualIPQIFilter:

Public Types
using	Object
	Type of the object to be analysed.

using	Interface
	Mark this class as the basic interface.

Public Member Functions
TrackFindingCDC::Weight	operator() (const BasePathFilter::Object &pair) override
	Return the weight based on the quality estimator.

void	exposeParameters (ModuleParamList *moduleParamList, const std::string &prefix) override
	Expose the parameters.

void	initialize () override
	initialize quality estimator

void	beginRun () override
	set BField value for estimator

virtual bool	needsTruthInformation ()
	Indicates if the filter requires Monte Carlo information.

virtual bool	needsTruthInformation ()
	Indicates if the filter requires Monte Carlo information.

virtual Weight	operator() (const Object &obj)
	Function to evaluate the object.

Weight	operator() (const Object *obj)
	Function to evaluate the object.

virtual std::vector< float >	operator() (const std::vector< Object * > &objs)
	Function to evaluate a vector of objects Base implementation applies the function to each object.

virtual Weight	operator() (const Object &obj)
	Function to evaluate the object.

Weight	operator() (const Object *obj)
	Function to evaluate the object.

virtual std::vector< float >	operator() (const std::vector< Object * > &objs)
	Function to evaluate a vector of objects Base implementation applies the function to each object.

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.

Protected Member Functions
void	addProcessingSignalListener (ProcessingSignalListener *psl)
	Register a processing signal listener to be notified.

int	getNProcessingSignalListener ()
	Get the number of currently registered listeners.

Private Types
using	Super
	Type of the base class.

Private Attributes
SpacePoint	m_virtualIPSpacePoint
	virtual IP SpacePoint

std::string	m_EstimationMethod = "helixFit"
	Identifier which estimation methsod to use.

std::string	m_MCRecoTracksStoreArrayName = "MCRecoTracks"
	sets the name of the expected StoreArray containing MCRecoTracks. Only required for MCInfo method

std::unique_ptr< QualityEstimatorBase >	m_estimator
	pointer to the selected QualityEstimator

bool	m_MCStrictQualityEstimator = true
	Only required for MCInfo method.

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.

Detailed Description

Filter for two hits plus a virtual IP using QualityEstimators.

Definition at line 22 of file TwoHitVirtualIPQIFilter.h.

Member Typedef Documentation

◆ Interface

using Interface

inherited

Mark this class as the basic interface.

Definition at line 38 of file Filter.dcl.h.

◆ Object

using Object

inherited

Type of the object to be analysed.

Definition at line 35 of file Filter.dcl.h.

◆ Super

using Super

privateinherited

Type of the base class.

Definition at line 26 of file CompositeProcessingSignalListener.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.

56{

57 m_subordinaryProcessingSignalListeners.push_back(psl);

58}

Belle2::TrackFindingCDC::Filter

Base class for filters on a generic object type.

Definition Filter.dcl.h:31

◆ 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 ( )

overridevirtual

set BField value for estimator

BeamSpot from DB

Reimplemented from CompositeProcessingSignalListener.

Definition at line 36 of file TwoHitVirtualIPQIFilter.cc.

{
  const double bFieldZ = BFieldManager::getField(0, 0, 0).Z() / Unit::T;
  m_estimator->setMagneticFieldStrength(bFieldZ);
 
  if (m_EstimationMethod == "mcInfo") {
    QualityEstimatorMC* MCestimator = static_cast<QualityEstimatorMC*>(m_estimator.get());
    MCestimator->forceUpdateClusterNames();
  }
 
  DBObjPtr<BeamSpot> beamSpotDB;
  if (beamSpotDB.isValid()) {
    const B2Vector3D& BeamSpotPosition = (*beamSpotDB).getIPPosition();
    const TMatrixDSym posErr = (*beamSpotDB).getIPPositionCovMatrix();
    const B2Vector3D BeamSpotPositionError(sqrt(posErr[0][0]), sqrt(posErr[1][1]), sqrt(posErr[2][2]));
    m_virtualIPSpacePoint = SpacePoint(BeamSpotPosition, BeamSpotPositionError, {0.5, 0.5}, {false, false}, VxdID(0),
                                       Belle2::VXD::SensorInfoBase::VXD);
  } else {
    m_virtualIPSpacePoint = SpacePoint(B2Vector3D(0., 0., 0.), B2Vector3D(0.1, 0.1, 0.5), {0.5, 0.5}, {false, false}, VxdID(0),
                                       Belle2::VXD::SensorInfoBase::VXD);
  }
 
}

◆ 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()

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

overridevirtual

Expose the parameters.

Reimplemented from CompositeProcessingSignalListener.

Definition at line 23 of file TwoHitVirtualIPQIFilter.cc.

{
  moduleParamList->addParameter(TrackFindingCDC::prefixed(prefix, "trackQualityEstimationMethod"), m_EstimationMethod,
                                "Identifier which estimation method to use. Valid identifiers are: [mcInfo, tripletFit, helixFit]",
                                m_EstimationMethod);
  moduleParamList->addParameter(TrackFindingCDC::prefixed(prefix, "MCRecoTracksStoreArrayName"), m_MCRecoTracksStoreArrayName,
                                "Only required for MCInfo method. Name of StoreArray containing MCRecoTracks.",
                                m_MCRecoTracksStoreArrayName);
  moduleParamList->addParameter(TrackFindingCDC::prefixed(prefix, "MCStrictQualityEstimator"), m_MCStrictQualityEstimator,
                                "Only required for MCInfo method. If false combining several MCTracks is allowed.",
                                m_MCStrictQualityEstimator);
}

◆ getNProcessingSignalListener()

int getNProcessingSignalListener ( )

protectedinherited

Get the number of currently registered listeners.

Definition at line 56 of file CompositeProcessingSignalListener.cc.

61{

62 return m_subordinaryProcessingSignalListeners.size();

63}

◆ initialize()

void initialize ( )

overridevirtual

initialize quality estimator

Reimplemented from CompositeProcessingSignalListener.

Definition at line 62 of file TwoHitVirtualIPQIFilter.cc.

{
  // create pointer to chosen estimator
  if (m_EstimationMethod == "mcInfo") {
    StoreArray<RecoTrack> mcRecoTracks;
    mcRecoTracks.isRequired(m_MCRecoTracksStoreArrayName);
    m_estimator = std::make_unique<QualityEstimatorMC>(m_MCRecoTracksStoreArrayName, m_MCStrictQualityEstimator);
  } else if (m_EstimationMethod == "tripletFit") {
    m_estimator = std::make_unique<QualityEstimatorTripletFit>();
  }  else if (m_EstimationMethod == "helixFit") {
    m_estimator = std::make_unique<QualityEstimatorRiemannHelixFit>();
  }
  B2ASSERT("QualityEstimator could not be initialized with method: " << m_EstimationMethod, m_estimator);
}

◆ needsTruthInformation()

bool needsTruthInformation ( )

virtualinherited

Indicates if the filter requires Monte Carlo information.

Definition at line 48 of file Filter.icc.h.

41 {

42 return false;

43 }

◆ operator()() [1/7]

virtual Weight operator() ( const Object & obj )

virtualinherited

Function to evaluate the object.

Base implementation accepts all objects.

Parameters

obj	The object to be accepted or rejected.

Returns: A finite float value if the object is accepted. NAN if the object is rejected.

◆ operator()() [2/7]

Weight operator() ( const Object & obj )

virtualinherited

Function to evaluate the object.

Base implementation accepts all objects.

Parameters

obj	The object to be accepted or rejected.

Returns: A finite float value if the object is accepted. NAN if the object is rejected.

Definition at line 59 of file Filter.icc.h.

47 {

48 return 1;

49 }

◆ operator()() [3/7]

Weight operator() ( const Object * obj )

inherited

Function to evaluate the object.

Base implementation accepts all objects, except nullptr.

Parameters

obj	The object to be accepted or rejected.

Returns: A finit float value if the object is accepted. NAN if the object is rejected. Nullptr is always rejected.

◆ operator()() [4/7]

Weight operator() ( const Object * obj )

inherited

Function to evaluate the object.

Base implementation accepts all objects, except nullptr.

Parameters

obj	The object to be accepted or rejected.

Returns: A finit float value if the object is accepted. NAN if the object is rejected. Nullptr is always rejected.

Definition at line 69 of file Filter.icc.h.

53 {

54 return obj ? operator()(*obj) : NAN;

55 }

◆ operator()() [5/7]

virtual std::vector< float > operator() ( const std::vector< Object * > & objs )

virtualinherited

Function to evaluate a vector of objects Base implementation applies the function to each object.

Can be optimized for MVA filters

Parameters

objs	A vector of pointers to objects

Returns: A vector of float or NAN values. See above

◆ operator()() [6/7]

std::vector< float > operator() ( const std::vector< Object * > & objs )

virtualinherited

Function to evaluate a vector of objects Base implementation applies the function to each object.

Can be optimized for MVA filters

Parameters

objs	A vector of pointers to objects

Returns: A vector of float or NAN values. See above

Definition at line 77 of file Filter.icc.h.

    {
      std::vector<float> out;
      for (const auto& obj : objs) {
        out.push_back(operator()(obj));
      }
      return out;
    }

◆ operator()() [7/7]

TrackFindingCDC::Weight operator() ( const BasePathFilter::Object & pair )

override

Return the weight based on the quality estimator.

Returns the QI value obtained from the fit with the chosen QualityEstimator

Definition at line 78 of file TwoHitVirtualIPQIFilter.cc.

{
  const std::vector<TrackFindingCDC::WithWeight<const VXDHoughState*>>& previousHits = pair.first;
 
  // Do nothing if path is too short or too long
  if (previousHits.size() != 1) {
    return NAN;
  }
 
  std::vector<const SpacePoint*> spacePointsVirtIP;
  spacePointsVirtIP.reserve(previousHits.size() + 2);
  // Note that the path is always created outside-in.
  // The tripletFit only works with hits being ordered inside-out, so add the hits in that direction.
  // First the virtual IP, then the hit that is currently checked, and last the single hit in the path.
  spacePointsVirtIP.emplace_back(&m_virtualIPSpacePoint);
  spacePointsVirtIP.emplace_back(pair.second->getHit());
  spacePointsVirtIP.emplace_back(previousHits.at(0)->getHit());
  const auto& estimatorResultVirtIP = m_estimator->estimateQualityAndProperties(spacePointsVirtIP);
 
  return estimatorResultVirtIP.qualityIndicator;
}

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

Member Data Documentation

◆ m_EstimationMethod

std::string m_EstimationMethod = "helixFit"

private

Identifier which estimation methsod to use.

Valid identifiers are: mcInfo, tripletFit, helixFit

Definition at line 41 of file TwoHitVirtualIPQIFilter.h.

◆ m_estimator

std::unique_ptr<QualityEstimatorBase> m_estimator

private

pointer to the selected QualityEstimator

Definition at line 45 of file TwoHitVirtualIPQIFilter.h.

◆ 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_MCRecoTracksStoreArrayName

std::string m_MCRecoTracksStoreArrayName = "MCRecoTracks"

private

sets the name of the expected StoreArray containing MCRecoTracks. Only required for MCInfo method

Definition at line 43 of file TwoHitVirtualIPQIFilter.h.

◆ m_MCStrictQualityEstimator

bool m_MCStrictQualityEstimator = true

private

Only required for MCInfo method.

Definition at line 47 of file TwoHitVirtualIPQIFilter.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_virtualIPSpacePoint

SpacePoint m_virtualIPSpacePoint

private

virtual IP SpacePoint

Definition at line 37 of file TwoHitVirtualIPQIFilter.h.

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

tracking/vxdHoughTracking/filters/pathFilters/include/TwoHitVirtualIPQIFilter.h
tracking/vxdHoughTracking/filters/pathFilters/src/TwoHitVirtualIPQIFilter.cc

Public Types

Public Member Functions

Protected Member Functions

Private Types

Private Attributes

Detailed Description

Member Typedef Documentation

◆ Interface

◆ Object

◆ Super

Member Function Documentation

◆ addProcessingSignalListener()

◆ beginEvent()

◆ beginRun()

◆ endRun()

◆ exposeParameters()

◆ getNProcessingSignalListener()

◆ initialize()

◆ needsTruthInformation()

◆ operator()() [1/7]

◆ operator()() [2/7]

◆ operator()() [3/7]

◆ operator()() [4/7]

◆ operator()() [5/7]

◆ operator()() [6/7]

◆ operator()() [7/7]

◆ terminate()

Member Data Documentation

◆ m_EstimationMethod

◆ m_estimator

◆ m_initialized

◆ m_initializedAs

◆ m_MCRecoTracksStoreArrayName

◆ m_MCStrictQualityEstimator

◆ m_subordinaryProcessingSignalListeners

◆ m_terminated

◆ m_virtualIPSpacePoint