mx/Jungfraujoch
0
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases 33 Wiki Activity

v1.0.0-rc.59

Browse Source
This commit is contained in:
leonarski_f
2025-07-03 22:32:07 +02:00
parent 5f8b1d7de7
commit f5ef540ca0
138 changed files with 190 additions and 160 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
writer/HDF5DataFilePluginAzInt.cpp
View File

@@ -3,7 +3,7 @@
#include "HDF5DataFilePluginAzInt.h"
#define RESERVE_IMAGES 1000
#define RESERVE_IMAGES 10000
HDF5DataFilePluginAzInt::HDF5DataFilePluginAzInt(const StartMessage &message) :
az_int_bin_to_q(message.az_int_bin_to_q),
@@ -26,30 +26,27 @@ void HDF5DataFilePluginAzInt::OpenFile(HDF5File &data_file, const DataMessage &m
if (!az_int_bin_to_phi.empty())
data_file.SaveVector("/entry/azint/bin_to_phi", az_int_bin_to_phi, dim);
max_image_number = msg.number;
HDF5DataType data_type(0.0f);
HDF5DataSpace data_space = HDF5DataSpace({max_image_number+1, azimuthal_bins, q_bins},
{H5S_UNLIMITED, azimuthal_bins, q_bins});
HDF5Dcpl dcpl;
dcpl.SetChunking({1, azimuthal_bins, q_bins});
dataset = std::make_unique<HDF5DataSet>(data_file, "/entry/azint/image", data_type, data_space, dcpl);
az_int_image.reserve(RESERVE_IMAGES * azimuthal_bins * q_bins);
az_int_image.resize(msg.number * 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 (static_cast<int64_t>(image_number) > max_image_number) {
if (static_cast<int64_t>(image_number) >= max_image_number) {
max_image_number = image_number;
dataset->SetExtent({max_image_number+1, azimuthal_bins, q_bins});
az_int_image.resize((max_image_number + 1) * azimuthal_bins * q_bins);
}
if (!msg.az_int_profile.empty() && (msg.az_int_profile.size() == az_int_bin_to_q.size())) {
dataset->SetExtent({max_image_number+1, azimuthal_bins, q_bins});
dataset->WriteVec(msg.az_int_profile, {image_number, 0, 0}, {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});
}