mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2025-07-11 19:31:50 +02:00
fixed python bindings - only read headers of modules that are in the roi
This commit is contained in:
@ -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);
|
||||
}
|
14
python/tests/test_RawFile.py
Normal file
14
python/tests/test_RawFile.py
Normal 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)
|
||||
|
Reference in New Issue
Block a user