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

Improvements before MAX IV test

Browse Source
This commit is contained in:
leonarski_f
2024-04-25 20:11:58 +02:00
parent 2c8e1fd83d
commit ea70b27e85
80 changed files with 1835 additions and 1781 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
writer/HDF5DataFilePluginAzInt.cpp
View File

@@ -2,38 +2,37 @@
#include "HDF5DataFilePluginAzInt.h"
#define RESERVE_IMAGES 1000
HDF5DataFilePluginAzInt::HDF5DataFilePluginAzInt(const std::vector<float> &in_rad_int_bin_to_q) :
rad_int_bin_to_q(in_rad_int_bin_to_q) {}
az_int_bin_to_q(in_rad_int_bin_to_q) {}
void HDF5DataFilePluginAzInt::OpenFile(HDF5File &data_file, const DataMessage &msg) {
if (rad_int_bin_to_q.empty())
if (az_int_bin_to_q.empty())
return;
HDF5Group(data_file, "/entry/azint").NXClass("NXcollection");
data_file.SaveVector("/entry/azint/binToQ", az_int_bin_to_q);
data_file.SaveVector("/entry/azint/binToQ", rad_int_bin_to_q);
HDF5DataSpace data_space_rad_int({1, rad_int_bin_to_q.size()}, {H5S_UNLIMITED, rad_int_bin_to_q.size()});
HDF5Dcpl dcpl_rad_int;
dcpl_rad_int.SetChunking({1, rad_int_bin_to_q.size()});
data_set_az_int = std::make_unique<HDF5DataSet>(data_file, "/entry/azint/result", HDF5DataType(0.0f),
data_space_rad_int, dcpl_rad_int);
az_int.reserve(RESERVE_IMAGES * az_int_bin_to_q.size());
}
void HDF5DataFilePluginAzInt::Write(const DataMessage &msg) {
if (rad_int_bin_to_q.empty())
void HDF5DataFilePluginAzInt::Write(const DataMessage &msg, uint64_t image_number) {
if (az_int_bin_to_q.empty())
return;
size_t image_number = msg.number;
if (image_number >= max_image_number) {
max_image_number = image_number;
data_set_az_int->SetExtent({max_image_number + 1, rad_int_bin_to_q.size()});
az_int.resize((max_image_number + 1) * az_int_bin_to_q.size());
}
if (!msg.az_int_profile.empty() && (msg.az_int_profile.size() == rad_int_bin_to_q.size()))
data_set_az_int->WriteVec(msg.az_int_profile, {image_number, 0}, {1, rad_int_bin_to_q.size()});
if (!msg.az_int_profile.empty() && (msg.az_int_profile.size() == az_int_bin_to_q.size())) {
for (int i = 0; i < az_int_bin_to_q.size(); i++)
az_int[image_number * az_int_bin_to_q.size() + i] = msg.az_int_profile[i];
}
}
void HDF5DataFilePluginAzInt::WriteFinal(HDF5File &data_file) {}
void HDF5DataFilePluginAzInt::WriteFinal(HDF5File &data_file) {
if (!az_int.empty())
data_file.SaveVector("/entry/azint/result", az_int, {(hsize_t) (max_image_number + 1), az_int_bin_to_q.size()});
}