fixed python bindings - only read headers of modules that are in the roi
All checks were successful
Build on RHEL8 / build (push) Successful in 2m51s
Build on RHEL9 / build (push) Successful in 2m53s

This commit is contained in:
2025-06-16 11:07:00 +02:00
parent 35114cde9d
commit 9f8eee5d08
5 changed files with 35 additions and 7 deletions

View File

@ -65,7 +65,7 @@ void define_raw_file_io_bindings(py::module &m) {
header = py::array_t<DetectorHeader>(n_frames);
} else {
header = py::array_t<DetectorHeader>(
{self.n_modules(), n_frames});
{self.n_modules_in_roi(), n_frames});
}
// py::array_t<DetectorHeader> header({self.n_mod(), n_frames});
@ -103,5 +103,6 @@ void define_raw_file_io_bindings(py::module &m) {
.def_property_readonly("geometry", &RawFile::geometry)
.def_property_readonly("detector_type", &RawFile::detector_type)
.def_property_readonly("master", &RawFile::master)
.def_property_readonly("n_modules", &RawFile::n_modules);
.def_property_readonly("n_modules", &RawFile::n_modules)
.def_property_readonly("n_modules_in_roi", &RawFile::n_modules_in_roi);
}

View File

@ -0,0 +1,14 @@
import pytest
from aare import RawFile
@pytest.mark.files
def test_read_rawfile_with_roi(test_data_path):
# Starting with f1 there is now 7 frames left in the series of files
print(test_data_path)
with RawFile(test_data_path / "raw/SingleChipROI/Data_master_0.json") as f:
headers, frames = f.read()
assert headers.size == 10100
assert frames.shape == (10100, 256, 256)