leonarski_f
d315506633
* Enhancements for EIGER * Writer is more flexible and capable of handling DECTRIS data
49 lines
1.7 KiB
C++
49 lines
1.7 KiB
C++
// Copyright (2019-2024) Paul Scherrer Institute
|
|
|
|
#include "HDF5DataFilePluginAzInt.h"
|
|
|
|
HDF5DataFilePluginAzInt::HDF5DataFilePluginAzInt(const std::vector<float> &in_rad_int_bin_to_q) :
|
|
rad_int_bin_to_q(in_rad_int_bin_to_q) {}
|
|
|
|
void HDF5DataFilePluginAzInt::SetupSWMRFile(HDF5File &data_file, const DataMessage &msg) {
|
|
if (rad_int_bin_to_q.empty())
|
|
return;
|
|
|
|
HDF5Group(data_file, "/entry/azint").NXClass("NXcollection");
|
|
|
|
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()});
|
|
HDF5DataSet(data_file, "/entry/azint/result", HDF5DataType(0.0f), data_space_rad_int, dcpl_rad_int);
|
|
}
|
|
|
|
void HDF5DataFilePluginAzInt::OpenFile(HDF5File &data_file, const DataMessage &msg) {
|
|
if (rad_int_bin_to_q.empty())
|
|
return;
|
|
data_set_az_int = std::make_unique<HDF5DataSet>(data_file, "/entry/azint/result");
|
|
}
|
|
|
|
void HDF5DataFilePluginAzInt::Write(const DataMessage &msg) {
|
|
if (rad_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()});
|
|
}
|
|
|
|
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()});
|
|
}
|
|
|
|
void HDF5DataFilePluginAzInt::Flush() {
|
|
if (rad_int_bin_to_q.empty())
|
|
return;
|
|
|
|
data_set_az_int->Flush();
|
|
}
|