mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2026-09-03 12:10:43 +02:00
145 lines
5.6 KiB
C++
145 lines
5.6 KiB
C++
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
#include "aare/CtbRawFile.hpp"
|
|
#include "aare/File.hpp"
|
|
#include "aare/Frame.hpp"
|
|
#include "aare/RawFile.hpp"
|
|
#include "aare/RawMasterFile.hpp"
|
|
#include "aare/RawSubFile.hpp"
|
|
|
|
#include "aare/defs.hpp"
|
|
|
|
#include <cstdint>
|
|
#include <filesystem>
|
|
#include <pybind11/iostream.h>
|
|
#include <pybind11/numpy.h>
|
|
#include <pybind11/pybind11.h>
|
|
#include <pybind11/stl.h>
|
|
#include <pybind11/stl/filesystem.h>
|
|
#include <string>
|
|
|
|
namespace py = pybind11;
|
|
using namespace ::aare;
|
|
|
|
void define_raw_master_file_bindings(py::module &m) {
|
|
|
|
py::enum_<ReadoutMode>(m, "ReadoutMode")
|
|
.value("ANALOG_ONLY", ReadoutMode::ANALOG_ONLY)
|
|
.value("DIGITAL_ONLY", ReadoutMode::DIGITAL_ONLY)
|
|
.value("ANALOG_AND_DIGITAL", ReadoutMode::ANALOG_AND_DIGITAL)
|
|
.value("TRANSCEIVER_ONLY", ReadoutMode::TRANSCEIVER_ONLY)
|
|
.value("DIGITAL_AND_TRANSCEIVER", ReadoutMode::DIGITAL_AND_TRANSCEIVER)
|
|
.value("UNKNOWN", ReadoutMode::UNKNOWN)
|
|
.export_values();
|
|
|
|
py::class_<RawMasterFile>(m, "RawMasterFile")
|
|
.def(py::init<const std::filesystem::path &>())
|
|
.def("data_fname", &RawMasterFile::data_fname, R"(
|
|
|
|
Parameters
|
|
------------
|
|
module_index : int
|
|
module index (d0, d1 .. dN)
|
|
file_index : int
|
|
file index (f0, f1 .. fN)
|
|
|
|
Returns
|
|
----------
|
|
os.PathLike
|
|
The name of the data file.
|
|
|
|
)")
|
|
.def_property_readonly("version", &RawMasterFile::version)
|
|
.def_property_readonly("detector_type", &RawMasterFile::detector_type)
|
|
.def_property_readonly("timing_mode", &RawMasterFile::timing_mode)
|
|
.def_property_readonly("image_size_in_bytes",
|
|
&RawMasterFile::image_size_in_bytes)
|
|
.def_property_readonly("frames_in_file", &RawMasterFile::frames_in_file)
|
|
.def_property_readonly("pixels_y", &RawMasterFile::pixels_y)
|
|
.def_property_readonly("pixels_x", &RawMasterFile::pixels_x)
|
|
.def_property_readonly("max_frames_per_file",
|
|
&RawMasterFile::max_frames_per_file)
|
|
.def_property_readonly("bitdepth", &RawMasterFile::bitdepth)
|
|
.def_property_readonly("frame_padding", &RawMasterFile::frame_padding)
|
|
.def_property_readonly("frame_discard_policy",
|
|
&RawMasterFile::frame_discard_policy)
|
|
|
|
.def_property_readonly("total_frames_expected",
|
|
&RawMasterFile::total_frames_expected)
|
|
.def_property_readonly("detector_layout",
|
|
&RawMasterFile::detector_layout)
|
|
.def_property_readonly("udp_interfaces_per_module",
|
|
&RawMasterFile::udp_interfaces_per_module)
|
|
.def_property_readonly("analog_samples", &RawMasterFile::analog_samples,
|
|
R"(
|
|
Number of analog samples
|
|
|
|
Returns
|
|
----------
|
|
int | None
|
|
The number of analog samples in the file (or None if not enabled)
|
|
)")
|
|
.def_property_readonly("digital_samples",
|
|
&RawMasterFile::digital_samples, R"(
|
|
Number of digital samples
|
|
|
|
Returns
|
|
----------
|
|
int | None
|
|
The number of digital samples in the file (or None if not enabled)
|
|
)")
|
|
|
|
.def_property_readonly("transceiver_samples",
|
|
&RawMasterFile::transceiver_samples)
|
|
.def_property_readonly("reading_mode", &RawMasterFile::get_reading_mode)
|
|
.def_property_readonly("number_of_rows", &RawMasterFile::number_of_rows)
|
|
.def_property_readonly("quad", &RawMasterFile::quad)
|
|
.def_property_readonly("scan_parameters",
|
|
&RawMasterFile::scan_parameters)
|
|
.def_property_readonly("roi", &RawMasterFile::roi)
|
|
.def_property_readonly(
|
|
"exptime",
|
|
[](RawMasterFile &self) -> std::optional<double> {
|
|
if (self.exptime()) {
|
|
double seconds =
|
|
std::chrono::duration<double>(*self.exptime()).count();
|
|
return seconds;
|
|
} else {
|
|
return std::nullopt;
|
|
}
|
|
})
|
|
.def_property_readonly("rois", &RawMasterFile::rois, R"(
|
|
Get the ROIs defined in the master file
|
|
|
|
Returns
|
|
----------
|
|
List[ROI]
|
|
List of ROIs (default complete ROI)
|
|
)")
|
|
.def_property_readonly("udp_port_types", &RawMasterFile::udp_port_types,
|
|
R"(
|
|
Get the types of UDP ports
|
|
|
|
Returns
|
|
----------
|
|
Optional[List[UDPPortPosition]]
|
|
Optional vector of UDP port types as strings (only present for
|
|
masterfile version >= 8.1)
|
|
)")
|
|
.def_property_readonly("disabled_udp_ports",
|
|
&RawMasterFile::disabled_udp_ports, R"(
|
|
Get the indices of disabled UDP ports
|
|
|
|
Returns
|
|
----------
|
|
Optional[List[int]]
|
|
Optional vector of disabled UDP port indices relative to UDP port types (only present for
|
|
masterfile version >= 8.1)
|
|
)")
|
|
.def_property_readonly("period", [](RawMasterFile &self) {
|
|
double seconds =
|
|
std::chrono::duration<double>(self.period()).count();
|
|
return seconds;
|
|
});
|
|
}
|