mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2025-06-06 21:00:41 +02:00
70acfbf4ac
* read subfiles with unordered and missing frames * save work debugging * Revert "save work debugging" This reverts commit e791992a05efd754f93a80c980d17397eb4b6045. * Revert "read subfiles with unordered and missing frames" This reverts commit 1177fd129d3690db92e9597ccda62598e5a44d41. * throw when two frames have different frame numbers * write single part RawFile (working beta) * correct total number of frames in master file * add new mythen file with syncd frames * read frames with same frame number * clang-tidy fixes, formatting, add tests * improve readability in loop * fix failing tests --------- Co-authored-by: Bechir <bechir.brahem420@gmail.com>
115 lines
4.0 KiB
C++
115 lines
4.0 KiB
C++
#include "aare/file_io/File.hpp"
|
|
#include "aare/utils/logger.hpp"
|
|
#include <catch2/catch_test_macros.hpp>
|
|
#include <filesystem>
|
|
|
|
#include "test_config.hpp"
|
|
|
|
using aare::File;
|
|
|
|
TEST_CASE("Read number of frames from a jungfrau raw file") {
|
|
|
|
auto fpath = test_data_path() / "jungfrau" / "jungfrau_single_master_0.json";
|
|
REQUIRE(std::filesystem::exists(fpath));
|
|
|
|
File f(fpath, "r");
|
|
REQUIRE(f.total_frames() == 10);
|
|
}
|
|
|
|
TEST_CASE("Read frame numbers from a jungfrau raw file") {
|
|
auto fpath = test_data_path() / "jungfrau" / "jungfrau_single_master_0.json";
|
|
REQUIRE(std::filesystem::exists(fpath));
|
|
|
|
File f(fpath, "r");
|
|
|
|
// we know this file has 10 frames with frame numbers 1 to 10
|
|
// f0 1,2,3
|
|
// f1 4,5,6
|
|
// f2 7,8,9
|
|
// f3 10
|
|
for (size_t i = 0; i < 10; i++) {
|
|
CHECK(f.frame_number(i) == i + 1);
|
|
}
|
|
}
|
|
|
|
TEST_CASE("Read data from a jungfrau 500k single port raw file") {
|
|
auto fpath = test_data_path() / "jungfrau" / "jungfrau_single_master_0.json";
|
|
REQUIRE(std::filesystem::exists(fpath));
|
|
|
|
File f(fpath, "r");
|
|
|
|
// we know this file has 10 frames with pixel 0,0 being: 2123, 2051, 2109, 2117, 2089, 2095, 2072, 2126, 2097, 2102
|
|
std::vector<uint16_t> pixel_0_0 = {2123, 2051, 2109, 2117, 2089, 2095, 2072, 2126, 2097, 2102};
|
|
for (size_t i = 0; i < 10; i++) {
|
|
auto frame = f.read();
|
|
CHECK(frame.rows() == 512);
|
|
CHECK(frame.cols() == 1024);
|
|
CHECK(frame.view<uint16_t>()(0, 0) == pixel_0_0[i]);
|
|
}
|
|
}
|
|
|
|
TEST_CASE("Read frame numbers from a raw file") {
|
|
auto fpath = test_data_path() / "eiger" / "eiger_500k_16bit_master_0.json";
|
|
REQUIRE(std::filesystem::exists(fpath));
|
|
|
|
// we know this file has 3 frames with frame numbers 14, 15, 16
|
|
std::vector<size_t> frame_numbers = {14, 15, 16};
|
|
|
|
File f(fpath, "r");
|
|
for (size_t i = 0; i < 3; i++) {
|
|
CHECK(f.frame_number(i) == frame_numbers[i]);
|
|
}
|
|
}
|
|
|
|
TEST_CASE("Compare reading from a numpy file with a raw file") {
|
|
auto fpath_raw = test_data_path() / "jungfrau" / "jungfrau_single_master_0.json";
|
|
REQUIRE(std::filesystem::exists(fpath_raw));
|
|
|
|
auto fpath_npy = test_data_path() / "jungfrau" / "jungfrau_single_0.npy";
|
|
REQUIRE(std::filesystem::exists(fpath_npy));
|
|
|
|
File raw(fpath_raw, "r");
|
|
File npy(fpath_npy, "r");
|
|
|
|
CHECK(raw.total_frames() == 10);
|
|
CHECK(npy.total_frames() == 10);
|
|
|
|
for (size_t i = 0; i < 10; ++i) {
|
|
auto raw_frame = raw.read();
|
|
auto npy_frame = npy.read();
|
|
CHECK(raw_frame.view<uint16_t>() == npy_frame.view<uint16_t>());
|
|
}
|
|
}
|
|
|
|
TEST_CASE("Read multipart files") {
|
|
auto fpath = test_data_path() / "jungfrau" / "jungfrau_double_master_0.json";
|
|
REQUIRE(std::filesystem::exists(fpath));
|
|
|
|
File f(fpath, "r");
|
|
|
|
// we know this file has 10 frames check read_multiport.py for the values
|
|
std::vector<uint16_t> pixel_0_0 = {2099, 2121, 2108, 2084, 2084, 2118, 2066, 2108, 2112, 2116};
|
|
std::vector<uint16_t> pixel_0_1 = {2842, 2796, 2865, 2798, 2805, 2817, 2852, 2789, 2792, 2833};
|
|
std::vector<uint16_t> pixel_255_1023 = {2149, 2037, 2115, 2102, 2118, 2090, 2036, 2071, 2073, 2142};
|
|
std::vector<uint16_t> pixel_511_1023 = {3231, 3169, 3167, 3162, 3168, 3160, 3171, 3171, 3169, 3171};
|
|
std::vector<uint16_t> pixel_1_0 = {2748, 2614, 2665, 2629, 2618, 2630, 2631, 2634, 2577, 2598};
|
|
|
|
for (size_t i = 0; i < 10; i++) {
|
|
auto frame = f.read();
|
|
CHECK(frame.rows() == 512);
|
|
CHECK(frame.cols() == 1024);
|
|
CHECK(frame.view<uint16_t>()(0, 0) == pixel_0_0[i]);
|
|
CHECK(frame.view<uint16_t>()(0, 1) == pixel_0_1[i]);
|
|
CHECK(frame.view<uint16_t>()(1, 0) == pixel_1_0[i]);
|
|
CHECK(frame.view<uint16_t>()(255, 1023) == pixel_255_1023[i]);
|
|
CHECK(frame.view<uint16_t>()(511, 1023) == pixel_511_1023[i]);
|
|
}
|
|
}
|
|
|
|
TEST_CASE("Read file with unordered frames") {
|
|
auto fpath = test_data_path() / "mythen" / "scan242_master_3.raw";
|
|
REQUIRE(std::filesystem::exists(fpath));
|
|
File f(fpath, "r");
|
|
REQUIRE_THROWS(f.read());
|
|
}
|