Skip to content

Commit

Permalink
feat: Add test for space point edm (acts-project#3690)
Browse files Browse the repository at this point in the history
Adds some unit test for the space point edm, with also checks on the types

Requires acts-project#3689 to compile (already merged)
  • Loading branch information
CarloVarni authored Oct 4, 2024
1 parent bcf1a19 commit 68cf91d
Show file tree
Hide file tree
Showing 2 changed files with 204 additions and 0 deletions.
1 change: 1 addition & 0 deletions Tests/UnitTests/Core/EventData/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -15,5 +15,6 @@ add_unittest(ParticleHypothesis ParticleHypothesisTests.cpp)
add_unittest(MultiTrajectoryHelpers MultiTrajectoryHelpersTests.cpp)
add_unittest(SubspaceHelpers SubspaceHelpersTests.cpp)
add_unittest(SeedEdm SeedEdmTests.cpp)
add_unittest(SpacePointContainerEdm SpacePointContainerEdmTests.cpp)

add_non_compile_test(MultiTrajectory TrackContainerComplianceTests.cpp)
203 changes: 203 additions & 0 deletions Tests/UnitTests/Core/EventData/SpacePointContainerEdmTests.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,203 @@
// This file is part of the ACTS project.
//
// Copyright (C) 2016 CERN for the benefit of the ACTS project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at https://mozilla.org/MPL/2.0/.

#include <boost/test/unit_test.hpp>

#include "Acts/Definitions/Algebra.hpp"
#include "Acts/EventData/SpacePointContainer.hpp"
#include "Acts/Utilities/HashedString.hpp"
#include "Acts/Utilities/Holders.hpp"

#include <any>
#include <cmath>
#include <iterator>
#include <ranges>
#include <vector>

namespace Acts::Test {

struct SpacePoint {
SpacePoint() = default;
SpacePoint(float ix, float iy, float iz, float ivarR, float ivarZ)
: x(ix), y(iy), z(iz), varR(ivarR), varZ(ivarZ) {}
float x{};
float y{};
float z{};
float varR{};
float varZ{};
};
using SpacePointCollection = std::vector<SpacePoint>;

class Adapter {
public:
friend Acts::SpacePointContainer<Adapter, Acts::detail::RefHolder>;
friend Acts::SpacePointContainer<Adapter, Acts::detail::ValueHolder>;
using value_type = SpacePoint;
using ValueType = value_type;

Adapter(SpacePointCollection&&) = delete;
Adapter(const SpacePointCollection& externalCollection)
: m_storage(&externalCollection) {}

private:
std::size_t size_impl() const { return storage().size(); }

float x_impl(std::size_t idx) const { return storage()[idx].x; };
float y_impl(std::size_t idx) const { return storage()[idx].y; };
float z_impl(std::size_t idx) const { return storage()[idx].z; };
float varianceR_impl(std::size_t idx) const { return storage()[idx].varR; }
float varianceZ_impl(std::size_t idx) const { return storage()[idx].varZ; }

const SpacePoint& get_impl(std::size_t idx) const { return storage()[idx]; }

std::any component_impl(Acts::HashedString key, std::size_t /*n*/) const {
using namespace Acts::HashedStringLiteral;
switch (key) {
case "TopStripVector"_hash:
case "BottomStripVector"_hash:
case "StripCenterDistance"_hash:
case "TopStripCenterPosition"_hash:
return Acts::Vector3(0., 0., 0.);
default:
throw std::runtime_error("no such component " + std::to_string(key));
}
}

const SpacePointCollection& storage() const { return *m_storage; }

private:
const SpacePointCollection* m_storage{};
};

BOOST_AUTO_TEST_CASE(spacepoint_container_edm_traits) {
using adapter_t = Acts::Test::Adapter;
using container_t =
Acts::SpacePointContainer<adapter_t, Acts::detail::RefHolder>;
using proxy_t = Acts::SpacePointProxy<container_t>;
using iterator_t = Acts::ContainerIndexIterator<container_t, proxy_t&, false>;

static_assert(std::ranges::range<container_t>);
static_assert(std::same_as<typename iterator_t::iterator_category,
std::random_access_iterator_tag>);
static_assert(
std::same_as<typename std::iterator_traits<iterator_t>::iterator_category,
std::random_access_iterator_tag>);
}

BOOST_AUTO_TEST_CASE(spacepoint_container_edm_constructors) {
std::size_t nExternalPoints = 10;
SpacePointCollection externalCollection(nExternalPoints);

Acts::SpacePointContainerConfig spConfig;
Acts::SpacePointContainerOptions spOptions;

Acts::Test::Adapter adapterForRef(externalCollection);
Acts::SpacePointContainer<Acts::Test::Adapter, Acts::detail::RefHolder>
spContainerRef(spConfig, spOptions, adapterForRef);

Acts::SpacePointContainer<Acts::Test::Adapter, Acts::detail::ValueHolder>
spContainerVal(spConfig, spOptions,
Acts::Test::Adapter(externalCollection));

BOOST_CHECK_EQUAL(spContainerRef.size(), nExternalPoints);
BOOST_CHECK_EQUAL(spContainerVal.size(), nExternalPoints);
}

BOOST_AUTO_TEST_CASE(spacepoint_container_edm_functionalities) {
std::size_t nExternalPoints = 100;
SpacePointCollection externalCollection;
externalCollection.reserve(nExternalPoints);
for (std::size_t i = 0; i < nExternalPoints; ++i) {
externalCollection.emplace_back(1.f * i, 1.5f * i, 2.f * i, 2.5f * i,
3.f * i);
}

Acts::SpacePointContainerConfig spConfig;
spConfig.useDetailedDoubleMeasurementInfo = true;
Acts::SpacePointContainerOptions spOptions;

Acts::Test::Adapter adapter(externalCollection);
Acts::SpacePointContainer<Acts::Test::Adapter, Acts::detail::RefHolder>
spContainer(spConfig, spOptions, adapter);

BOOST_CHECK_EQUAL(spContainer.size(), nExternalPoints);
BOOST_CHECK_EQUAL(spContainer.size(), externalCollection.size());
BOOST_CHECK_EQUAL(spContainer.end() - spContainer.begin(), nExternalPoints);
BOOST_CHECK_EQUAL(std::distance(spContainer.begin(), spContainer.end()),
nExternalPoints);
BOOST_CHECK_EQUAL(std::distance(std::ranges::begin(spContainer),
std::ranges::end(spContainer)),
nExternalPoints);

using proxy_t = Acts::SpacePointProxy<
Acts::SpacePointContainer<Acts::Test::Adapter, Acts::detail::RefHolder>>;
static_assert(std::same_as<typename decltype(spContainer)::ValueType,
Acts::Test::SpacePoint>);
static_assert(
std::same_as<typename decltype(spContainer)::ProxyType, proxy_t>);
static_assert(
std::same_as<typename decltype(spContainer)::value_type, proxy_t>);
static_assert(
std::same_as<typename proxy_t::ContainerType, decltype(spContainer)>);
static_assert(
std::same_as<typename proxy_t::ValueType, Acts::Test::SpacePoint>);

using iterator_t =
Acts::ContainerIndexIterator<decltype(spContainer), proxy_t&, false>;
using const_iterator_t =
Acts::ContainerIndexIterator<decltype(spContainer), const proxy_t&, true>;
static_assert(
std::same_as<iterator_t, typename decltype(spContainer)::iterator>);
static_assert(std::same_as<const_iterator_t,
typename decltype(spContainer)::const_iterator>);
static_assert(
std::same_as<iterator_t, decltype(std::ranges::begin(spContainer))>);
static_assert(std::same_as<const_iterator_t,
decltype(std::ranges::cbegin(spContainer))>);

std::size_t n = 0ul;
for (const proxy_t& proxy : spContainer) {
float refX = 1.f * n;
float refY = 1.5f * n;
float refZ = 2.f * n;
float refCovR = 2.5f * n;
float refCovZ = 3.f * n;
float refRadius = std::hypot(refX, refY);
float refPhi = std::atan2(refY, refX);

BOOST_CHECK_EQUAL(proxy.index(), n);
BOOST_CHECK_EQUAL(proxy.x(), refX);
BOOST_CHECK_EQUAL(proxy.y(), refY);
BOOST_CHECK_EQUAL(proxy.z(), refZ);
BOOST_CHECK_EQUAL(proxy.radius(), refRadius);
BOOST_CHECK_EQUAL(proxy.phi(), refPhi);
BOOST_CHECK_EQUAL(proxy.varianceR(), refCovR);
BOOST_CHECK_EQUAL(proxy.varianceZ(), refCovZ);

const Acts::Vector3& topStripVector = proxy.topStripVector();
const Acts::Vector3& bottomStripVector = proxy.bottomStripVector();
const Acts::Vector3& stripCenterDistance = proxy.stripCenterDistance();
const Acts::Vector3& topStripCenterPosition =
proxy.topStripCenterPosition();

for (std::size_t i = 0; i < 3; ++i) {
BOOST_CHECK_EQUAL(topStripVector[i], 0.);
BOOST_CHECK_EQUAL(bottomStripVector[i], 0.);
BOOST_CHECK_EQUAL(stripCenterDistance[i], 0.);
BOOST_CHECK_EQUAL(topStripCenterPosition[i], 0.);
}

const Acts::Test::SpacePoint& sp = proxy.externalSpacePoint();
BOOST_CHECK_EQUAL(&sp, &externalCollection[n]);

++n;
}
BOOST_CHECK_EQUAL(n, nExternalPoints);
}

} // namespace Acts::Test

0 comments on commit 68cf91d

Please sign in to comment.