* Enhancements for XFEL

* Enhancements for EIGER
* Writer is more flexible and capable of handling DECTRIS data
This commit is contained in:
2024-03-05 20:41:47 +01:00
parent 71d862b706
commit d315506633
165 changed files with 5440 additions and 2230 deletions

View File

@@ -0,0 +1,48 @@
// 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();
}