Equivalent to. More...

#include <TrackFindingCDCTestWithSimpleSimulation.h>

Inheritance diagram for TrackFindingCDCTestWithSimpleSimulation:

Public Member Functions
void	SetUp () override
	Preparations before the test.

void	simulate (const std::initializer_list< Helix > &helices)
	Populate the event with hits generated from the given helices.

void	simulate (const std::vector< Helix > &helices)
	Populate the event with hits generated from the given helices.

void	simulate (const std::initializer_list< CDCTrajectory3D > &trajectories)
	Populate the event with hits generated from the given trajectories.

void	simulate (const std::vector< CDCTrajectory3D > &trajectories)
	Populate the event with hits generated from the given trajectories.

void	loadPreparedEvent ()
	Populate the event with hits hard codes presimulated hits.

void	fillCaches ()
	Prepare a set of hits, axial hits, segments, axial segments and track hits.

void	plotMCTracks ()
	Add the Monte Carlo tracks to the event plot.

void	plotMCTrajectories ()
	Add the Monte Carlo trajectories to the event plot.

void	saveDisplay (const std::string &svgFileName)
	Save content of the plotter to svg file only if running in unrestricted mode –gtest_also_run_disabled_tests.

template<class AFunction >
TimeItResult	timeIt (size_t nExecutions, bool activateCallgrind, const AFunction &function, const std::function< void()> &setUp=doNothing, const std::function< void()> &tearDown=doNothing)
	Repeat a time critical simulation section a couple of times.

template<class... Ts>
void	draw (Ts &&... args)
	Forwarding draw class to the plotter instance.

void	TearDown () override
	Clean up after test.

Static Public Member Functions
static void	SetUpTestCase ()
	Implementation of hook method to setup the environment for all tests in this test case Unpacks the wire topology from the CDCGeometryPar.

static void	TearDownTestCase ()
	Implementation of hook method to clean up the environment for all tests in this test case.

Protected Attributes
LogConfig::ELogLevel	m_savedLogLevel = LogConfig::c_Info
	Memory for the log level that was set before the test.

unsigned int	m_savedDebugLogInfo = 100
	Memory for the log info of debug that was set before the test.

const std::array< std::string, 6 >	m_colors {{ "red", "blue", "green", "yellow", "violet", "cyan" }}
	Some colors to cycle for plotting.

CDCSimpleSimulation	m_simpleSimulation
	Simple simulation generating the hits.

std::vector< CDCTrajectory3D >	m_mcTrajectories
	Memory for the Monte Carlo trajectories of the current event.

std::vector< CDCTrack >	m_mcTracks
	Memory for the Monte Carlo tracks of the current event.

std::vector< CDCSegment2D >	m_mcSegment2Ds
	Memory for the Monte Carlo segments of the current event.

std::vector< const CDCSegment2D * >	m_mcAxialSegment2Ds
	Memory for the axial Monte Carlo segments of the current event.

std::vector< const CDCWireHit * >	m_axialWireHits
	Memory for the axial hits of the current event.

std::vector< const CDCWireHit * >	m_wireHits
	Memory for the hits of the current event.

Private Attributes
EventDataPlotter	m_plotter
	Plotter facility to generate visual representations.

Detailed Description

Equivalent to.

See also: TrackFindingCDCTestWithTopology for disabled tests.

Definition at line 35 of file TrackFindingCDCTestWithSimpleSimulation.h.

Constructor & Destructor Documentation

◆ TrackFindingCDCTestWithSimpleSimulation()

TrackFindingCDCTestWithSimpleSimulation ( )

inline

Definition at line 39 of file TrackFindingCDCTestWithSimpleSimulation.h.

                                                : m_simpleSimulation()
      {
      }

Member Function Documentation

◆ draw()

void draw ( Ts &&... args )

inline

Forwarding draw class to the plotter instance.

Definition at line 203 of file TrackFindingCDCTestWithSimpleSimulation.h.

      {
        m_plotter.draw(std::forward<Ts>(args) ...);
      }

◆ fillCaches()

void fillCaches ( )

inline

Prepare a set of hits, axial hits, segments, axial segments and track hits.

Definition at line 114 of file TrackFindingCDCTestWithSimpleSimulation.h.

      {
        for (size_t iTrack = 0; iTrack < m_mcTracks.size(); ++iTrack) {
          B2INFO("Size mc track " << iTrack << " : " << m_mcTracks[iTrack].size());
        }
 
        // Prepare the monte carlo segments
        for (const CDCTrack& mcTrack : m_mcTracks) {
          std::vector<CDCSegment3D> segment3DsInTrack = mcTrack.splitIntoSegments();
          for (const CDCSegment3D& segment3D :  segment3DsInTrack) {
            m_mcSegment2Ds.push_back(segment3D.stereoProjectToRef());
          }
        }
 
        // Filter the axial segments
        for (const CDCSegment2D& segment2D : m_mcSegment2Ds) {
          if (segment2D.getStereoKind() == EStereoKind::c_Axial) {
            m_mcAxialSegment2Ds.push_back(&segment2D);
          }
        }
 
        // Filter for axial hits
        for (const CDCWireHit& wireHit : m_simpleSimulation.getWireHits()) {
          if (wireHit.isAxial()) {
            m_axialWireHits.push_back(&wireHit);
          }
        }
 
        // Pick up points for all hits
        for (const CDCWireHit& wireHit : m_simpleSimulation.getWireHits()) {
          m_wireHits.push_back(&wireHit);
        }
 
        m_plotter.draw(CDCWireTopology::getInstance());
        for (const CDCWireHit& wireHit : m_simpleSimulation.getWireHits()) {
          m_plotter.draw(wireHit);
        }
      }

◆ loadPreparedEvent()

void loadPreparedEvent ( )

inline

Populate the event with hits hard codes presimulated hits.

Definition at line 102 of file TrackFindingCDCTestWithSimpleSimulation.h.

      {
        // No trajectory information present
        m_mcTrajectories.clear();
 
        // Load prepared tracks wire hits are stored in the simple simulation
        m_mcTracks = m_simpleSimulation.loadPreparedEvent();
 
        fillCaches();
      }

◆ plotMCTracks()

void plotMCTracks ( )

inline

Add the Monte Carlo tracks to the event plot.

Definition at line 154 of file TrackFindingCDCTestWithSimpleSimulation.h.

155 { for (const CDCTrack& mcTrack : m_mcTracks) m_plotter.draw(mcTrack); }

◆ plotMCTrajectories()

void plotMCTrajectories ( )

inline

Add the Monte Carlo trajectories to the event plot.

Definition at line 158 of file TrackFindingCDCTestWithSimpleSimulation.h.

      {
        for (const CDCTrajectory3D& mcTrajectory : m_mcTrajectories) {
          m_plotter.draw(mcTrajectory.getTrajectory2D());
        }
      }

◆ saveDisplay()

void saveDisplay ( const std::string & svgFileName )

inline

Save content of the plotter to svg file only if running in unrestricted mode –gtest_also_run_disabled_tests.

Definition at line 166 of file TrackFindingCDCTestWithSimpleSimulation.h.

      {
        bool run_disabled = ::testing::GTEST_FLAG(also_run_disabled_tests);
        if (run_disabled) {
          m_plotter.save(svgFileName);
        } else {
          B2INFO("Not writing display file. To activate svg display output run with --gtest_also_run_disabled_tests");
        }
      }

◆ SetUp()

void SetUp ( )

inlineoverride

Preparations before the test.

Definition at line 45 of file TrackFindingCDCTestWithSimpleSimulation.h.

      {
 
        m_savedLogLevel = Belle2::LogSystem::Instance().getLogConfig()->getLogLevel();
        m_savedDebugLogInfo = Belle2::LogSystem::Instance().getLogConfig()->getLogInfo(LogConfig::c_Debug);
        bool run_disabled = ::testing::GTEST_FLAG(also_run_disabled_tests);
        if (run_disabled) {
          Belle2::LogSystem::Instance().getLogConfig()->setLogLevel(LogConfig::ELogLevel::c_Debug);
          Belle2::LogSystem::Instance().getLogConfig()->setLogInfo(LogConfig::c_Debug, LogConfig::c_Level + LogConfig::c_Message);
        }
 
        m_plotter.clear();
        m_mcAxialSegment2Ds.clear();
        m_mcSegment2Ds.clear();
        m_mcTracks.clear();
        m_mcTrajectories.clear();
        m_axialWireHits.clear();
        m_wireHits.clear();
      }

◆ SetUpTestCase()

void SetUpTestCase ( )

staticinherited

Implementation of hook method to setup the environment for all tests in this test case Unpacks the wire topology from the CDCGeometryPar.

Manually open the database and prepare the cdc geometry parameters.

Definition at line 19 of file TrackFindingCDCTestWithTopology.cc.

{
  TrackFindingCDC::CDCGeometryLoader::loadDatabase();
 
  // Prepare the wires for the cdc track finders.
  TrackFindingCDC::CDCWireTopology::getInstance();
}

◆ simulate() [1/4]

void simulate ( const std::initializer_list< CDCTrajectory3D > & trajectories )

inline

Populate the event with hits generated from the given trajectories.

Definition at line 85 of file TrackFindingCDCTestWithSimpleSimulation.h.

      {
        simulate(std::vector<CDCTrajectory3D>(trajectories));
      }

◆ simulate() [2/4]

void simulate ( const std::initializer_list< Helix > & helices )

inline

Populate the event with hits generated from the given helices.

Definition at line 66 of file TrackFindingCDCTestWithSimpleSimulation.h.

      {
        simulate(std::vector<Helix>(helices));
      }

◆ simulate() [3/4]

void simulate ( const std::vector< CDCTrajectory3D > & trajectories )

inline

Populate the event with hits generated from the given trajectories.

Definition at line 91 of file TrackFindingCDCTestWithSimpleSimulation.h.

      {
        m_mcTrajectories = trajectories;
 
        // Construct tracks. Wire hits are stored in the simple simulation
        m_mcTracks = m_simpleSimulation.simulate(trajectories);
 
        fillCaches();
      }

◆ simulate() [4/4]

void simulate ( const std::vector< Helix > & helices )

inline

Populate the event with hits generated from the given helices.

Definition at line 73 of file TrackFindingCDCTestWithSimpleSimulation.h.

      {
        std::vector<CDCTrajectory3D> trajectories;
        trajectories.reserve(helices.size());
        for (const Helix& helix : helices) {
          // had to make the implicit convesion to an explicit conversion. Maybe there is a more elegant way to do it
          trajectories.emplace_back(UncertainHelix(helix));
        }
        simulate(trajectories);
      }

◆ TearDown()

void TearDown ( )

inlineoverride

Clean up after test.

Definition at line 209 of file TrackFindingCDCTestWithSimpleSimulation.h.

      {
        Belle2::LogSystem::Instance().getLogConfig()->setLogLevel(m_savedLogLevel);
        Belle2::LogSystem::Instance().getLogConfig()->setLogInfo(LogConfig::c_Debug, m_savedDebugLogInfo);
 
        m_plotter.clear();
        m_mcAxialSegment2Ds.clear();
        m_mcSegment2Ds.clear();
        m_mcTracks.clear();
        m_mcTrajectories.clear();
        m_axialWireHits.clear();
        m_wireHits.clear();
      }

◆ TearDownTestCase()

void TearDownTestCase ( )

staticinherited

Implementation of hook method to clean up the environment for all tests in this test case.

Definition at line 28 of file TrackFindingCDCTestWithTopology.cc.

{
  TrackFindingCDC::CDCGeometryLoader::closeDatabase();
}

◆ timeIt()

TimeItResult timeIt	(	size_t	nExecutions,
		bool	activateCallgrind,
		const AFunction &	function,
		const std::function< void()> &	setUp = `doNothing`,
		const std::function< void()> &	tearDown = `doNothing`
	)

inline

Repeat a time critical simulation section a couple of times.

The repeated execution is only done if the test is run in unrestricted mode. The default is to run it once.

Definition at line 183 of file TrackFindingCDCTestWithSimpleSimulation.h.

      {
        bool run_disabled = ::testing::GTEST_FLAG(also_run_disabled_tests);
        if (not run_disabled) {
          nExecutions = 1;
        }
        return Belle2::TrackFindingCDC::timeIt(nExecutions,
                                               activateCallgrind,
                                               function,
                                               setUp,
                                               tearDown);
 
      }