From 67d27c4618e0dea8c6c6fba20931964f5b0b0636 Mon Sep 17 00:00:00 2001
From: Andrej Babic <andrej.babic@psi.ch>
Date: Wed, 27 May 2020 15:47:45 +0200
Subject: [PATCH] Implemented read_slot_id

---
 sf-writer/src/ImageAssembler.cpp | 16 +++++++++++++++-
 sf-writer/src/ImageAssembler.hpp |  2 ++
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/sf-writer/src/ImageAssembler.cpp b/sf-writer/src/ImageAssembler.cpp
index f87270f..1bc4f24 100644
--- a/sf-writer/src/ImageAssembler.cpp
+++ b/sf-writer/src/ImageAssembler.cpp
@@ -15,6 +15,8 @@ ImageAssembler::ImageAssembler(const size_t n_modules) :
     for (size_t i=0; i<IA_N_SLOTS; i++) {
         free_slot(i);
     }
+
+    read_slot_id_ = 0;
 }
 
 ImageAssembler::~ImageAssembler()
@@ -65,17 +67,29 @@ void ImageAssembler::process(
         }
     }
 
-    buffer_status_[slot_id]--;
+    auto previous_status = buffer_status_[slot_id].fetch_sub(1);
+
+    // 1 because fetch is done before subtraction.
+    if (previous_status ==  1) {
+        write_slot_id_++;
+        read_slot_id_ %= IA_N_SLOTS;
+    }
 }
 
 int ImageAssembler::get_full_slot()
 {
+    if (buffer_status_[read_slot_id_] == 0) {
+        return read_slot_id_;
+    }
 
+    return -1;
 }
 
 void ImageAssembler::free_slot(const int slot_id)
 {
     buffer_status_[slot_id] = n_modules_;
+
+    read_slot_id_ = (int)((read_slot_id_+1) % IA_N_SLOTS);
 }
 
 ImageMetadataBlock* ImageAssembler::get_metadata_buffer(const int slot_id)
diff --git a/sf-writer/src/ImageAssembler.hpp b/sf-writer/src/ImageAssembler.hpp
index 10f7fdf..1b4d2be 100644
--- a/sf-writer/src/ImageAssembler.hpp
+++ b/sf-writer/src/ImageAssembler.hpp
@@ -15,6 +15,8 @@ class ImageAssembler {
     ImageMetadataBlock* metadata_buffer_;
     std::atomic_int* buffer_status_;
 
+    int read_slot_id_;
+
 public:
     ImageAssembler(const size_t n_modules);