mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2026-10-22 08:21:27 +02:00
Erik Fröjdh
dd409cfe41
- Default `ClusterVector` move operations simplifying the code and fixing a bug. - Fixed inconsistent frame number type (uint64/int32) - Validate Python masks as one-dimensional, C-contiguous Boolean arrays, handle empty masks safely, and reserve filtered storage based on the selected cluster count. - Align the C++ and Python API documentation with the implementation, including concise `hitmap` and reduction documentation and a correctly rendered constructor example.
94 lines
4.1 KiB
C++
94 lines
4.1 KiB
C++
// SPDX-License-Identifier: MPL-2.0
|
|
/************************************************
|
|
* @file test-Cluster.cpp
|
|
* @short test case for generic Cluster, ClusterVector, and calculate_eta2
|
|
***********************************************/
|
|
|
|
#include "aare/Cluster.hpp"
|
|
#include "aare/CalculateEta.hpp"
|
|
#include "aare/ClusterFile.hpp"
|
|
|
|
// #include "catch.hpp"
|
|
#include <array>
|
|
#include <catch2/catch_all.hpp>
|
|
#include <catch2/catch_test_macros.hpp>
|
|
|
|
using namespace aare;
|
|
|
|
TEST_CASE("Test sum of Cluster", "[cluster]") {
|
|
Cluster<int, 2, 2> cluster{0, 0, {1, 2, 3, 4}};
|
|
|
|
CHECK(cluster.sum() == 10);
|
|
}
|
|
|
|
using ClusterTypes = std::variant<Cluster<int, 2, 2>, Cluster<int, 3, 3>,
|
|
Cluster<int, 5, 5>, Cluster<int, 2, 3>>;
|
|
|
|
using ReducibleTo3x3ClusterTypes =
|
|
std::variant<Cluster<int, 3, 4>, Cluster<int, 4, 3>, Cluster<int, 5, 5>>;
|
|
|
|
TEST_CASE("Test reduce to 2x2 Cluster", "[cluster]") {
|
|
auto [cluster, expected_reduced_cluster] = GENERATE(
|
|
std::make_tuple(ClusterTypes{Cluster<int, 2, 2>{5, 5, {1, 2, 3, 4}}},
|
|
Cluster<int, 2, 2>{5, 5, {1, 2, 3, 4}}),
|
|
std::make_tuple(
|
|
ClusterTypes{Cluster<int, 3, 3>{5, 5, {1, 1, 1, 1, 3, 2, 1, 2, 2}}},
|
|
Cluster<int, 2, 2>{5, 5, {3, 2, 2, 2}}),
|
|
std::make_tuple(
|
|
ClusterTypes{Cluster<int, 3, 3>{5, 5, {1, 1, 1, 2, 3, 1, 2, 2, 1}}},
|
|
Cluster<int, 2, 2>{5, 5, {2, 3, 2, 2}}),
|
|
std::make_tuple(
|
|
ClusterTypes{Cluster<int, 3, 3>{5, 5, {2, 2, 1, 2, 3, 1, 1, 1, 1}}},
|
|
Cluster<int, 2, 2>{5, 5, {2, 2, 2, 3}}),
|
|
std::make_tuple(
|
|
ClusterTypes{Cluster<int, 3, 3>{5, 5, {1, 2, 2, 1, 3, 2, 1, 1, 1}}},
|
|
Cluster<int, 2, 2>{5, 5, {2, 2, 3, 2}}),
|
|
std::make_tuple(ClusterTypes{Cluster<int, 5, 5>{
|
|
5, 5, {1, 1, 1, 1, 1, 1, 1, 2, 2, 1, 1, 1, 3,
|
|
2, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1}}},
|
|
Cluster<int, 2, 2>{5, 5, {2, 2, 3, 2}}),
|
|
std::make_tuple(ClusterTypes{Cluster<int, 5, 5>{
|
|
5, 5, {1, 1, 1, 1, 1, 1, 2, 2, 1, 1, 1, 2, 3,
|
|
1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1}}},
|
|
Cluster<int, 2, 2>{5, 5, {2, 2, 2, 3}}),
|
|
std::make_tuple(
|
|
ClusterTypes{Cluster<int, 2, 3>{5, 5, {2, 2, 3, 2, 1, 1}}},
|
|
Cluster<int, 2, 2>{5, 5, {2, 2, 3, 2}}));
|
|
|
|
auto reduced_cluster = std::visit(
|
|
[](const auto &clustertype) { return reduce_to_2x2(clustertype); },
|
|
cluster);
|
|
|
|
CHECK(reduced_cluster.x == expected_reduced_cluster.x);
|
|
CHECK(reduced_cluster.y == expected_reduced_cluster.y);
|
|
CHECK(std::equal(reduced_cluster.data.begin(),
|
|
reduced_cluster.data.begin() + 4,
|
|
expected_reduced_cluster.data.begin()));
|
|
}
|
|
|
|
TEST_CASE("Test reduce to 3x3 Cluster", "[cluster]") {
|
|
auto [cluster, expected_reduced_cluster] = GENERATE(
|
|
std::make_tuple(
|
|
ReducibleTo3x3ClusterTypes{Cluster<int, 3, 4>{
|
|
5, 5, {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11}}},
|
|
Cluster<int, 3, 3>{5, 5, {3, 4, 5, 6, 7, 8, 9, 10, 11}}),
|
|
std::make_tuple(
|
|
ReducibleTo3x3ClusterTypes{Cluster<int, 4, 3>{
|
|
5, 5, {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11}}},
|
|
Cluster<int, 3, 3>{5, 5, {1, 2, 3, 5, 6, 7, 9, 10, 11}}),
|
|
std::make_tuple(ReducibleTo3x3ClusterTypes{Cluster<int, 5, 5>{
|
|
5, 5, {1, 1, 1, 1, 1, 1, 2, 1, 1, 1, 2, 2, 3,
|
|
1, 1, 1, 2, 1, 1, 1, 1, 1, 1, 1, 1}}},
|
|
Cluster<int, 3, 3>{5, 5, {2, 1, 1, 2, 3, 1, 2, 1, 1}}));
|
|
|
|
auto reduced_cluster = std::visit(
|
|
[](const auto &clustertype) { return reduce_to_3x3(clustertype); },
|
|
cluster);
|
|
|
|
CHECK(reduced_cluster.x == expected_reduced_cluster.x);
|
|
CHECK(reduced_cluster.y == expected_reduced_cluster.y);
|
|
CHECK(std::equal(reduced_cluster.data.begin(),
|
|
reduced_cluster.data.begin() + 9,
|
|
expected_reduced_cluster.data.begin()));
|
|
}
|