version 1.0.0-rc.24

2024-11-17 14:55:09 +01:00
parent aeeae33ad9
commit adc13ff33e
361 changed files with 12513 additions and 3314 deletions
--- a/writer/HDF5DataFilePluginROI.cpp
+++ b/writer/HDF5DataFilePluginROI.cpp
@@ -0,0 +1,32 @@
+// Copyright (2019-2024) Paul Scherrer Institute
+
+#include "HDF5DataFilePluginROI.h"
+
+#define RESERVE_IMAGES 1000
+
+void HDF5DataFilePluginROI::OpenFile(HDF5File &data_file, const DataMessage &msg) {}
+
+void HDF5DataFilePluginROI::Write(const DataMessage &msg, uint64_t image_number) {
+    for (const auto &r: msg.roi) {
+        if (roi_data.find(r.first) != roi_data.end()) {
+            roi_data[r.first].max.reserve(RESERVE_IMAGES);
+            roi_data[r.first].sum.reserve(RESERVE_IMAGES);
+        }
+        roi_data[r.first].max[image_number] = r.second.max_count;
+        roi_data[r.first].sum[image_number] = r.second.sum;
+    }
+}
+
+void HDF5DataFilePluginROI::WriteFinal(HDF5File &data_file) {
+    if (roi_data.empty())
+        return;
+
+    HDF5Group group(data_file, "/entry/roi");
+    group.NXClass("NXcollection");
+
+    for (const auto &r: roi_data) {
+        HDF5Group group_roi(data_file, "/entry/roi/" + r.first);
+        group.SaveVector("max", r.second.max.vec());
+        group.SaveVector("sum", r.second.sum.vec());
+    }
+}