Move sf_writer to sf-buffer

2026-04-23 05:02:41 +02:00 · 2020-04-24 14:43:32 +02:00
parent d7dcfec8f6
commit 2033241fee
4 changed files with 15 additions and 1 deletions
@@ -0,0 +1,92 @@
+#include "SFWriter.hpp"
+#include "buffer_config.hpp"
+
+using namespace std;
+using namespace core_buffer;
+
+SFWriter::SFWriter(
+        const string& output_file,
+        const size_t n_frames,
+        const size_t n_modules) :
+        n_frames_(n_frames),
+        n_modules_(n_modules),
+        current_write_index_(0)
+{
+    file_ = H5::H5File(output_file, H5F_ACC_TRUNC);
+
+    hsize_t image_dataset_dims[3] =
+            {n_frames_, n_modules * MODULE_Y_SIZE, MODULE_X_SIZE};
+
+    H5::DataSpace image_dataspace(3, image_dataset_dims);
+
+    hsize_t image_dataset_chunking[3] =
+            {1, n_modules * MODULE_Y_SIZE, MODULE_X_SIZE};
+    H5::DSetCreatPropList image_dataset_properties;
+    image_dataset_properties.setChunk(3, image_dataset_chunking);
+
+    image_dataset_ = file_.createDataSet(
+            "image",
+            H5::PredType::NATIVE_UINT16,
+            image_dataspace,
+            image_dataset_properties);
+
+    hsize_t metadata_dataset_dims[2] = {n_frames_, 1};
+    H5::DataSpace metadata_dataspace(2, metadata_dataset_dims);
+
+    pulse_id_dataset_ = file_.createDataSet(
+            "pulse_id",
+            H5::PredType::NATIVE_UINT64,
+            metadata_dataspace);
+
+    pulse_id_dataset_ = file_.createDataSet(
+            "frame_index",
+            H5::PredType::NATIVE_UINT64,
+            metadata_dataspace);
+
+    pulse_id_dataset_ = file_.createDataSet(
+            "daq_rec",
+            H5::PredType::NATIVE_UINT32,
+            metadata_dataspace);
+
+    pulse_id_dataset_ = file_.createDataSet(
+            "n_received_packets",
+            H5::PredType::NATIVE_UINT16,
+            metadata_dataspace);
+}
+
+SFWriter::~SFWriter()
+{
+    close_file();
+}
+
+void SFWriter::close_file()
+{
+    image_dataset_.close();
+    pulse_id_dataset_.close();
+    frame_index_dataset_.close();
+    daq_rec_dataset_.close();
+    n_received_packets_dataset_.close();
+
+    file_.close();
+}
+
+void SFWriter::write(char* data, std::shared_ptr<DetectorFrame> metadata) {
+    auto pulse_id = metadata->pulse_id;
+    auto frame_index = metadata->frame_index;
+    auto daq_rec = metadata->daq_rec;
+    auto n_received_packets = metadata->n_received_packets;
+
+    hsize_t buff_dim[2] = {n_modules_*MODULE_Y_SIZE, MODULE_X_SIZE};
+    H5::DataSpace buffer_space (2, buff_dim);
+
+    hsize_t disk_dim[3] = {n_frames_, n_modules_*MODULE_Y_SIZE, MODULE_X_SIZE};
+    H5::DataSpace disk_space(3, disk_dim);
+
+    hsize_t count[] = {1, n_modules_*MODULE_Y_SIZE, MODULE_X_SIZE};
+    hsize_t start[] = {current_write_index_, 0, 0};
+    disk_space.selectHyperslab(H5S_SELECT_SET, count, start);
+
+    image_dataset_.write(data, H5::PredType::NATIVE_UINT16,
+            buffer_space,
+            disk_space);
+}