Filip Leonarski
64002f1e29
Some checks failed
Build Packages / build:rpm (rocky8_nocuda) (push) Successful in 11m14s
Build Packages / build:rpm (ubuntu2204_nocuda) (push) Successful in 10m43s
Build Packages / build:rpm (rocky9_nocuda) (push) Successful in 11m35s
Build Packages / build:rpm (ubuntu2404_nocuda) (push) Successful in 9m20s
Build Packages / build:rpm (rocky8_sls9) (push) Successful in 10m23s
Build Packages / Generate python client (push) Successful in 39s
Build Packages / build:rpm (rocky9_sls9) (push) Successful in 11m24s
Build Packages / Create release (push) Has been skipped
Build Packages / Build documentation (push) Successful in 1m0s
Build Packages / build:rpm (rocky8) (push) Successful in 10m35s
Build Packages / build:rpm (ubuntu2204) (push) Successful in 10m35s
Build Packages / build:rpm (rocky9) (push) Successful in 11m17s
Build Packages / build:rpm (ubuntu2404) (push) Successful in 9m9s
Build Packages / Unit tests (push) Failing after 1h18m57s
This is an UNSTABLE release. The release has significant modifications and bug fixes, if things go wrong, it is better to revert to 1.0.0-rc.124. * jfjoch_broker: Significant improvements in TCP image socket, as a viable alternative for ZeroMQ sockets (only a single port on broker side, dynamically change number of writers, acknowledgments for written files) * jfjoch_broker: Delta phi is calculated also for still data in Bragg prediction * jfjoch_broker: Image pusher statistics are accessible via the REST interface * jfjoch_writer: Supports TCP image socket and for these auto-forking option Reviewed-on: #36 Co-authored-by: Filip Leonarski <filip.leonarski@psi.ch> Co-committed-by: Filip Leonarski <filip.leonarski@psi.ch>
57 lines
2.3 KiB
C++
57 lines
2.3 KiB
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#include "HDF5DataFilePluginROI.h"
|
|
|
|
void HDF5DataFilePluginROI::OpenFile(HDF5File &data_file, const DataMessage &msg, size_t in_images_per_file) {
|
|
images_per_file = in_images_per_file;
|
|
|
|
for (const auto &r: msg.roi) {
|
|
auto &entry = roi_data[r.first];
|
|
entry.max.reserve(images_per_file);
|
|
entry.sum.reserve(images_per_file);
|
|
entry.sum_sq.reserve(images_per_file);
|
|
entry.npixel.reserve(images_per_file);
|
|
entry.x.reserve(images_per_file);
|
|
entry.y.reserve(images_per_file);
|
|
}
|
|
}
|
|
|
|
void HDF5DataFilePluginROI::Write(const DataMessage &msg, uint64_t image_number) {
|
|
for (const auto &r: msg.roi) {
|
|
if (!roi_data.contains(r.first)) {
|
|
roi_data[r.first].max.reserve(images_per_file);
|
|
roi_data[r.first].sum.reserve(images_per_file);
|
|
roi_data[r.first].sum_sq.reserve(images_per_file);
|
|
roi_data[r.first].npixel.reserve(images_per_file);
|
|
roi_data[r.first].x.reserve(images_per_file);
|
|
roi_data[r.first].y.reserve(images_per_file);
|
|
}
|
|
|
|
roi_data[r.first].max[image_number] = r.second.max_count;
|
|
roi_data[r.first].sum[image_number] = r.second.sum;
|
|
roi_data[r.first].npixel[image_number] = r.second.pixels;
|
|
roi_data[r.first].sum_sq[image_number] = r.second.sum_square;
|
|
|
|
roi_data[r.first].x[image_number] = static_cast<double>(r.second.x_weighted) / static_cast<double>(r.second.sum);
|
|
roi_data[r.first].y[image_number] = static_cast<double>(r.second.y_weighted) / static_cast<double>(r.second.sum);
|
|
}
|
|
}
|
|
|
|
void HDF5DataFilePluginROI::WriteFinal(HDF5File &data_file) {
|
|
if (roi_data.empty())
|
|
return;
|
|
|
|
HDF5Group(data_file, "/entry/roi").NXClass("NXcollection");
|
|
|
|
for (const auto &r: roi_data) {
|
|
HDF5Group group_roi(data_file, "/entry/roi/" + r.first);
|
|
group_roi.SaveVector("max", r.second.max.vec());
|
|
group_roi.SaveVector("sum", r.second.sum.vec());
|
|
group_roi.SaveVector("sum_sq", r.second.sum_sq.vec());
|
|
group_roi.SaveVector("npixel", r.second.npixel.vec());
|
|
group_roi.SaveVector("x", r.second.x.vec());
|
|
group_roi.SaveVector("y", r.second.y.vec());
|
|
}
|
|
}
|