From 911c44bda5a435ab9292ed53dba92999e8921bfb Mon Sep 17 00:00:00 2001
From: Andrej Babic <andrej.babic@psi.ch>
Date: Tue, 19 May 2020 16:11:22 +0200
Subject: [PATCH] Implement buffered fast queue

---
 core-buffer/include/BufferedFastQueue.hpp | 32 +++++++++++
 core-buffer/src/BufferedFastQueue.cpp     | 66 +++++++++++++++++++++++
 2 files changed, 98 insertions(+)
 create mode 100644 core-buffer/include/BufferedFastQueue.hpp
 create mode 100644 core-buffer/src/BufferedFastQueue.cpp

diff --git a/core-buffer/include/BufferedFastQueue.hpp b/core-buffer/include/BufferedFastQueue.hpp
new file mode 100644
index 0000000..b690f3e
--- /dev/null
+++ b/core-buffer/include/BufferedFastQueue.hpp
@@ -0,0 +1,32 @@
+#ifndef SF_DAQ_BUFFER_BUFFEREDFASTQUEUE_HPP
+#define SF_DAQ_BUFFER_BUFFEREDFASTQUEUE_HPP
+
+#include "FastQueue.hpp"
+#include "WriterH5Writer.hpp"
+
+
+class BufferedFastQueue {
+    FastQueue<ImageMetadataBuffer>& queue_;
+    const size_t buffer_n_pulses_;
+    const size_t n_modules_;
+
+    ImageMetadataBuffer* queue_meta_buffer_ = nullptr;
+    char* queue_data_buffer_ = nullptr;
+    int current_slot_id_ = -1;
+
+    ImageMetadata image_metadata_;
+
+public:
+    BufferedFastQueue(FastQueue<ImageMetadataBuffer>& queue,
+                      const size_t buffer_n_pulses,
+                      const size_t n_modules);
+
+    ImageMetadata* get_metadata_buffer();
+    char* get_data_buffer();
+
+    void commit();
+    void finalize();
+};
+
+
+#endif //SF_DAQ_BUFFER_BUFFEREDFASTQUEUE_HPP
diff --git a/core-buffer/src/BufferedFastQueue.cpp b/core-buffer/src/BufferedFastQueue.cpp
new file mode 100644
index 0000000..dcc8a4d
--- /dev/null
+++ b/core-buffer/src/BufferedFastQueue.cpp
@@ -0,0 +1,66 @@
+#include "BufferedFastQueue.hpp"
+#include <thread>
+
+using namespace std;
+using namespace core_buffer;
+
+BufferedFastQueue::BufferedFastQueue(
+        FastQueue<ImageMetadataBuffer>& queue,
+        const size_t buffer_n_pulses,
+        const size_t n_modules) :
+            buffer_n_pulses_(buffer_n_pulses),
+            queue_(queue),
+            n_modules_(n_modules)
+{
+    while ((current_slot_id_ = queue_.reserve()) == -1){
+        this_thread::sleep_for(chrono::milliseconds(5));
+    }
+
+    queue_meta_buffer_ = queue_.get_metadata_buffer(current_slot_id_);
+    queue_meta_buffer_->n_pulses_in_buffer = 0;
+    queue_data_buffer_ = queue_.get_data_buffer(current_slot_id_);
+}
+
+ImageMetadata* BufferedFastQueue::get_metadata_buffer()
+{
+    return &image_metadata_;
+}
+
+char* BufferedFastQueue::get_data_buffer()
+{
+    auto index = queue_meta_buffer_->n_pulses_in_buffer;
+    auto image_size = MODULE_N_BYTES * n_modules_;
+
+    return queue_data_buffer_ + (index * image_size);
+}
+
+void BufferedFastQueue::commit()
+{
+    auto index = queue_meta_buffer_->n_pulses_in_buffer;
+
+    queue_meta_buffer_->pulse_id[index] = image_metadata_.pulse_id;
+    queue_meta_buffer_->frame_index[index] = image_metadata_.frame_index;
+    queue_meta_buffer_->daq_rec[index] = image_metadata_.daq_rec;
+    queue_meta_buffer_->is_good_frame[index] = image_metadata_.is_good_frame;
+    queue_meta_buffer_->data_n_bytes[index] = image_metadata_.data_n_bytes;
+
+    queue_meta_buffer_->n_pulses_in_buffer++;
+
+    if (queue_meta_buffer_->n_pulses_in_buffer == buffer_n_pulses_) {
+        queue_.commit();
+
+        while ((current_slot_id_ = queue_.reserve()) == -1){
+            this_thread::sleep_for(chrono::milliseconds(5));
+        }
+
+        queue_meta_buffer_ = queue_.get_metadata_buffer(current_slot_id_);
+        queue_meta_buffer_->n_pulses_in_buffer = 0;
+        queue_data_buffer_ = queue_.get_data_buffer(current_slot_id_);
+    }
+}
+
+void BufferedFastQueue::finalize() {
+    if (queue_meta_buffer_->n_pulses_in_buffer > 0) {
+        queue_.commit();
+    }
+}
\ No newline at end of file