updated read_columns in python

This commit is contained in:
mazzol_a committed 2026-10-06 16:05:10 +02:00
1 parent 03208a5154
commit 43cecc3779
3 files changed
+12 -4

No files matched your search

+9 -2
View File
@@ -81,8 +81,15 @@ void define_jungfrau_data_file_io_bindings(py::module &m) {
.def(
"read_columns",
[](JungfrauDataFile &self,
const std::vector<size_t> column_indices) {
return self.read_columns(column_indices);
const std::vector<size_t> &column_indices) {
auto frame = new Frame(self.read_columns(column_indices));
return py::array_t<uint16_t>(
{frame->rows(), frame->cols()},
{frame->cols() * sizeof(uint16_t), sizeof(uint16_t)},
reinterpret_cast<uint16_t *>(frame->data()),
py::capsule(frame, [](void *ptr) {
delete reinterpret_cast<Frame *>(ptr);
}));
},
R"(
Read the columns at the current frame.