leonarski_f
4a852b4d6b
Build Packages / build:rpm (rocky9_nocuda) (push) Successful in 12m57s
Build Packages / build:rpm (rocky8_nocuda) (push) Successful in 13m4s
Build Packages / build:rpm (ubuntu2404_nocuda) (push) Successful in 11m18s
Build Packages / build:rpm (ubuntu2204_nocuda) (push) Successful in 13m12s
Build Packages / build:rpm (rocky8_sls9) (push) Successful in 13m51s
Build Packages / build:rpm (rocky9_sls9) (push) Successful in 13m59s
Build Packages / build:rpm (ubuntu2204) (push) Successful in 10m45s
Build Packages / build:rpm (rocky8) (push) Successful in 12m29s
Build Packages / build:rpm (ubuntu2404) (push) Successful in 12m2s
Build Packages / Generate python client (push) Successful in 24s
Build Packages / XDS test (durin plugin) (push) Successful in 9m50s
Build Packages / Create release (push) Has been skipped
Build Packages / build:rpm (rocky9) (push) Successful in 14m15s
Build Packages / Build documentation (push) Successful in 1m6s
Build Packages / DIALS test (push) Successful in 13m10s
Build Packages / XDS test (JFJoch plugin) (push) Successful in 6m45s
Build Packages / XDS test (neggia plugin) (push) Successful in 5m58s
Build Packages / Unit tests (push) Successful in 1h20m42s
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.132. * jfjoch_broker: Add better locking for detector object - should help, when detector initialization takes too long * jfjoch_writer: Enable writing single, integrated HDF5 file with both data and metadata * XDS plugin: Add generation of Jungfraujoch plugin for XDS * CI: Add tests with XDS and DIALS (`xia2.ssx`) Reviewed-on: #43
50 lines
2.1 KiB
C++
50 lines
2.1 KiB
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#include "HDF5DataFilePluginAzInt.h"
|
|
|
|
HDF5DataFilePluginAzInt::HDF5DataFilePluginAzInt(const StartMessage &message) :
|
|
az_int_bin_to_q(message.az_int_bin_to_q),
|
|
az_int_bin_to_two_theta(message.az_int_bin_to_two_theta),
|
|
az_int_bin_to_phi(message.az_int_bin_to_phi),
|
|
azimuthal_bins(message.az_int_phi_bin_count.value_or(1)),
|
|
q_bins(message.az_int_q_bin_count.value_or(1)){
|
|
}
|
|
|
|
void HDF5DataFilePluginAzInt::OpenFile(HDF5File &data_file, const DataMessage &msg, size_t images_per_file) {
|
|
if (az_int_bin_to_q.empty() || q_bins <= 0 || azimuthal_bins <= 0)
|
|
return;
|
|
|
|
std::vector<hsize_t> dim = {azimuthal_bins, q_bins};
|
|
|
|
HDF5Group(data_file, "/entry/azint").NXClass("NXcollection");
|
|
data_file.SaveVector("/entry/azint/bin_to_q", az_int_bin_to_q, dim);
|
|
if (!az_int_bin_to_two_theta.empty())
|
|
data_file.SaveVector("/entry/azint/bin_to_two_theta", az_int_bin_to_two_theta, dim);
|
|
if (!az_int_bin_to_phi.empty())
|
|
data_file.SaveVector("/entry/azint/bin_to_phi", az_int_bin_to_phi, dim);
|
|
|
|
az_int_image.reserve(images_per_file * azimuthal_bins * q_bins);
|
|
}
|
|
|
|
void HDF5DataFilePluginAzInt::Write(const DataMessage &msg, uint64_t image_number) {
|
|
if (az_int_bin_to_q.empty() || q_bins <= 0 || azimuthal_bins <= 0)
|
|
return;
|
|
|
|
if (image_number >= max_image_number || (max_image_number == 0)) {
|
|
max_image_number = image_number;
|
|
az_int_image.resize((max_image_number + 1) * azimuthal_bins * q_bins);
|
|
}
|
|
|
|
if (!msg.az_int_profile.empty() && (msg.az_int_profile.size() == azimuthal_bins * q_bins)) {
|
|
for (int i = 0; i < azimuthal_bins * q_bins; i++)
|
|
az_int_image[image_number * azimuthal_bins * q_bins + i] = msg.az_int_profile[i];
|
|
}
|
|
}
|
|
|
|
void HDF5DataFilePluginAzInt::WriteFinal(HDF5File &data_file) {
|
|
if (!az_int_image.empty())
|
|
data_file.SaveVector("/entry/azint/image", az_int_image,
|
|
{(hsize_t) (max_image_number + 1), azimuthal_bins, q_bins});
|
|
}
|