mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2025-06-08 13:40:39 +02:00
28d7e8c07a
* Read and write cluster files (save work) * add reading test * use define for examples env variable and fix ci * read and write cluster files (working) * fix cluster CI
114 lines
3.4 KiB
C++
114 lines
3.4 KiB
C++
#include "aare/file_io/ClusterFile.hpp"
|
|
#include "aare/utils/compare_files.hpp"
|
|
#include "test_config.hpp"
|
|
#include <catch2/catch_test_macros.hpp>
|
|
#include <iostream>
|
|
#include <random>
|
|
using aare::Cluster;
|
|
using aare::ClusterFile;
|
|
using aare::ClusterFileConfig;
|
|
|
|
TEST_CASE("Read a cluster file") {
|
|
auto fpath = test_data_path() / "clusters" / "single_frame_97_clustrers.clust";
|
|
REQUIRE(std::filesystem::exists(fpath));
|
|
ClusterFile cf(fpath, "r");
|
|
|
|
SECTION("Read the header of the file") {
|
|
REQUIRE(cf.count() == 97);
|
|
REQUIRE(cf.frame() == 135);
|
|
}
|
|
SECTION("Read a single cluster") {
|
|
Cluster c = cf.read();
|
|
REQUIRE(c.x == 1);
|
|
REQUIRE(c.y == 200);
|
|
for (int i = 0; i < 9; i++) {
|
|
REQUIRE(c.data[i] == i);
|
|
}
|
|
}
|
|
SECTION("Read a single cluster using iread") {
|
|
Cluster c = cf.iread(0);
|
|
REQUIRE(c.x == 1);
|
|
REQUIRE(c.y == 200);
|
|
for (int i = 0; i < 9; i++) {
|
|
REQUIRE(c.data[i] == i);
|
|
}
|
|
}
|
|
SECTION("Read a cluster using seek") {
|
|
cf.seek(0);
|
|
Cluster c = cf.read();
|
|
REQUIRE(c.x == 1);
|
|
REQUIRE(c.y == 200);
|
|
for (int i = 0; i < 9; i++) {
|
|
REQUIRE(c.data[i] == i);
|
|
}
|
|
c = cf.read();
|
|
REQUIRE(c.x == 2);
|
|
REQUIRE(c.y == 201);
|
|
for (int i = 0; i < 9; i++) {
|
|
REQUIRE(c.data[i] == i + 9);
|
|
}
|
|
}
|
|
SECTION("check out of bound reading") {
|
|
REQUIRE_THROWS_AS(cf.iread(97), std::runtime_error);
|
|
REQUIRE_NOTHROW(cf.seek(97));
|
|
REQUIRE_THROWS_AS(cf.read(), std::runtime_error);
|
|
REQUIRE_THROWS_AS(cf.read(1), std::runtime_error);
|
|
REQUIRE_NOTHROW(cf.seek(0));
|
|
REQUIRE_NOTHROW(cf.read(97));
|
|
}
|
|
|
|
SECTION("test read multiple clusters") {
|
|
std::vector<Cluster> cluster = cf.read(97);
|
|
REQUIRE(cluster.size() == 97);
|
|
int offset = 0;
|
|
int data_offset = 0;
|
|
for (auto c : cluster) {
|
|
REQUIRE(c.x == offset + 1);
|
|
REQUIRE(c.y == offset + 200);
|
|
for (int i = 0; i < 9; i++) {
|
|
REQUIRE(c.data[i] == data_offset + i);
|
|
}
|
|
|
|
offset++;
|
|
data_offset += 9;
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST_CASE("write a cluster file") {
|
|
|
|
auto const FRAME_NUMBER = 1461041991;
|
|
auto const TOTAL_CLUSTERS = 214748;
|
|
|
|
std::filesystem::path const fpath_out("/tmp/file.clust");
|
|
ClusterFile cf_out(fpath_out, "w", ClusterFileConfig(FRAME_NUMBER, TOTAL_CLUSTERS));
|
|
REQUIRE(cf_out.count() == 0);
|
|
REQUIRE(cf_out.frame() == FRAME_NUMBER);
|
|
|
|
// write file with random close to bounds values
|
|
int32_t offset = 0;
|
|
std::vector<Cluster> clusters(TOTAL_CLUSTERS);
|
|
for (int32_t i = 0; i < TOTAL_CLUSTERS; i++) {
|
|
Cluster c;
|
|
c.x = INT16_MAX - offset;
|
|
c.y = INT16_MAX - (offset + 200);
|
|
for (int32_t j = 0; j < 9; j++) {
|
|
if (j % 2 == 0)
|
|
c.data[j] = -(offset * 2);
|
|
else
|
|
c.data[j] = (offset * 2);
|
|
}
|
|
clusters[i] = c;
|
|
offset++;
|
|
offset %= INT16_MAX - 200;
|
|
}
|
|
cf_out.write(clusters);
|
|
REQUIRE(cf_out.count() == TOTAL_CLUSTERS);
|
|
REQUIRE(cf_out.frame() == FRAME_NUMBER);
|
|
cf_out.update_header();
|
|
REQUIRE(cf_out.count() == TOTAL_CLUSTERS);
|
|
REQUIRE(cf_out.frame() == FRAME_NUMBER);
|
|
|
|
auto data_file = test_data_path() / "clusters" / "test_writing.clust";
|
|
REQUIRE(aare::compare_files(fpath_out, data_file));
|
|
} |