109 lines
3.1 KiB
C++
109 lines
3.1 KiB
C++
// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#include <catch2/catch_all.hpp>
|
|
#include "../image_analysis/spot_finding/ImageSpotFinder.h"
|
|
|
|
TEST_CASE("ImageSpotFinder_SignalToNoise") {
|
|
size_t width = 100;
|
|
size_t height = 100;
|
|
std::vector<float> resolution(width * height, 2.0);
|
|
|
|
resolution[width * 50 + 50] = 1.0;
|
|
|
|
ImageSpotFinder s(resolution, width, height);
|
|
|
|
std::vector<int32_t> input (width * height);
|
|
for (int i = 0; i < width * height; i++)
|
|
input[i] = (i % 2) * 5 + 5;
|
|
|
|
input[width * 50 + 50] = 20;
|
|
input[width * 25 + 26] = 16;
|
|
input[width * 75 + 25] = 12;
|
|
|
|
// Mean is 7.5, std. dev is 2.5
|
|
SpotFindingSettings settings{
|
|
.signal_to_noise_threshold = 3.0,
|
|
.photon_count_threshold = 0,
|
|
.min_pix_per_spot = 1,
|
|
.max_pix_per_spot = 20,
|
|
.high_resolution_limit = 0.5,
|
|
.low_resolution_limit = 3.0,
|
|
};
|
|
|
|
auto spots = s.Run(input.data(), settings);
|
|
|
|
REQUIRE(spots.size() == 2);
|
|
REQUIRE(spots[0].RawCoord().y == 25);
|
|
REQUIRE(spots[1].RawCoord().y == 50);
|
|
}
|
|
|
|
TEST_CASE("ImageSpotFinder_SignalToNoise_Resolution") {
|
|
size_t width = 100;
|
|
size_t height = 100;
|
|
std::vector<float> resolution(width * height, 2.0);
|
|
|
|
resolution[width * 50 + 50] = 1.0;
|
|
|
|
ImageSpotFinder s(resolution, width, height);
|
|
|
|
std::vector<int32_t> input (width * height);
|
|
for (int i = 0; i < width * height; i++)
|
|
input[i] = (i % 2) * 5 + 5;
|
|
|
|
input[width * 50 + 50] = 20;
|
|
input[width * 25 + 26] = 16;
|
|
input[width * 75 + 25] = 12;
|
|
|
|
// Mean is 7.5, std. dev is 2.5
|
|
SpotFindingSettings settings{
|
|
.signal_to_noise_threshold = 3.0,
|
|
.photon_count_threshold = 0,
|
|
.min_pix_per_spot = 1,
|
|
.max_pix_per_spot = 20,
|
|
.high_resolution_limit = 1.5,
|
|
.low_resolution_limit = 3.0,
|
|
};
|
|
|
|
auto spots = s.Run(input.data(), settings);
|
|
|
|
REQUIRE(spots.size() == 1);
|
|
REQUIRE(spots[0].RawCoord().x == 26);
|
|
REQUIRE(spots[0].RawCoord().y == 25);
|
|
}
|
|
|
|
|
|
TEST_CASE("ImageSpotFinder_CountThreshold_Resolution") {
|
|
size_t width = 100;
|
|
size_t height = 100;
|
|
std::vector<float> resolution(width * height, 2.0);
|
|
|
|
resolution[width * 50 + 50] = 1.0;
|
|
|
|
ImageSpotFinder s(resolution, width, height);
|
|
|
|
std::vector<int32_t> input (width * height);
|
|
for (int i = 0; i < width * height; i++)
|
|
input[i] = (i % 2) * 5 + 5;
|
|
|
|
input[width * 50 + 50] = 20;
|
|
input[width * 25 + 26] = 16;
|
|
input[width * 75 + 25] = 12;
|
|
|
|
// Mean is 7.5, std. dev is 2.5
|
|
SpotFindingSettings settings{
|
|
.signal_to_noise_threshold = -1,
|
|
.photon_count_threshold = 11,
|
|
.min_pix_per_spot = 1,
|
|
.max_pix_per_spot = 20,
|
|
.high_resolution_limit = 1.5,
|
|
.low_resolution_limit = 3.0,
|
|
};
|
|
|
|
auto spots = s.Run(input.data(), settings);
|
|
|
|
REQUIRE(spots.size() == 2);
|
|
REQUIRE(spots[0].RawCoord().y == 25);
|
|
REQUIRE(spots[1].RawCoord().y == 75);
|
|
}
|