70 lines
3.1 KiB
C++
70 lines
3.1 KiB
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#include "HDF5DataFilePluginDetector.h"
|
|
|
|
HDF5DataFilePluginDetector::HDF5DataFilePluginDetector(const StartMessage &msg) {
|
|
integrated_format = (msg.file_format == FileWriterFormat::NXmxIntegrated);
|
|
}
|
|
|
|
void HDF5DataFilePluginDetector::OpenFile(HDF5File &in_data_file, const DataMessage &msg, size_t images_per_file) {
|
|
jf_info.reserve(images_per_file);
|
|
storage_cell.reserve(images_per_file);
|
|
receiver_aq_dev_delay.reserve(images_per_file);
|
|
receiver_free_buffers.reserve(images_per_file);
|
|
efficiency.reserve(images_per_file);
|
|
packets_received.reserve(images_per_file);
|
|
packets_expected.reserve(images_per_file);
|
|
pixel_sum.reserve(images_per_file);
|
|
processing_time.reserve(images_per_file);
|
|
}
|
|
|
|
void HDF5DataFilePluginDetector::Write(const DataMessage &msg, uint64_t image_number) {
|
|
if (msg.jf_info.has_value())
|
|
jf_info[image_number] = msg.jf_info.value();
|
|
if (msg.storage_cell.has_value())
|
|
storage_cell[image_number] = msg.storage_cell.value();
|
|
if (msg.receiver_aq_dev_delay.has_value())
|
|
receiver_aq_dev_delay[image_number] = msg.receiver_aq_dev_delay.value();
|
|
if (msg.receiver_buf_available.has_value())
|
|
receiver_free_buffers[image_number] = msg.receiver_buf_available.value();
|
|
|
|
if (msg.packets_received.has_value())
|
|
packets_received[image_number] = msg.packets_received.value();
|
|
if (msg.packets_expected.has_value())
|
|
packets_expected[image_number] = msg.packets_expected.value();
|
|
if (msg.pixel_sum.has_value())
|
|
pixel_sum[image_number] = msg.pixel_sum.value();
|
|
|
|
if (msg.image_collection_efficiency.has_value())
|
|
efficiency[image_number] = msg.image_collection_efficiency.value();
|
|
else
|
|
efficiency[image_number] = 1.0;
|
|
|
|
if (msg.processing_time_s.has_value())
|
|
processing_time[image_number] = msg.processing_time_s.value();
|
|
}
|
|
|
|
void HDF5DataFilePluginDetector::WriteFinal(HDF5File &data_file) {
|
|
const std::string prefix = integrated_format ? "/entry/instrument/detector/detectorSpecific" : "/entry/detector";
|
|
|
|
if (!jf_info.empty())
|
|
data_file.SaveVector(prefix + "/det_info", jf_info.vec());
|
|
if (!storage_cell.empty())
|
|
data_file.SaveVector(prefix + "/storage_cell_image", storage_cell.vec());
|
|
if (!receiver_aq_dev_delay.empty())
|
|
data_file.SaveVector(prefix + "/rcv_delay", receiver_aq_dev_delay.vec());
|
|
if (!receiver_free_buffers.empty())
|
|
data_file.SaveVector(prefix + "/rcv_free_send_buffers", receiver_free_buffers.vec());
|
|
if (!packets_received.empty())
|
|
data_file.SaveVector(prefix + "/packets_received", packets_received.vec());
|
|
if (!packets_expected.empty())
|
|
data_file.SaveVector(prefix + "/packets_expected", packets_expected.vec());
|
|
if (!pixel_sum.empty())
|
|
data_file.SaveVector(prefix + "/pixel_sum", pixel_sum.vec());
|
|
if (!processing_time.empty())
|
|
data_file.SaveVector(prefix + "/processing_time", processing_time.vec())->Units("s");
|
|
|
|
data_file.SaveVector(prefix + "/data_collection_efficiency_image", efficiency.vec());
|
|
}
|