From 3d98a9e9087c6b7d319e1f434c3a4b4ed9a26bdc Mon Sep 17 00:00:00 2001
From: Andrej Babic <andrej.babic@psi.ch>
Date: Wed, 27 May 2020 13:07:04 +0200
Subject: [PATCH] Add start and stop pulse id to JF writer

---
 sf-writer/include/JFH5Writer.hpp |  7 +++++--
 sf-writer/src/JFH5Writer.cpp     | 12 +++++++-----
 2 files changed, 12 insertions(+), 7 deletions(-)

diff --git a/sf-writer/include/JFH5Writer.hpp b/sf-writer/include/JFH5Writer.hpp
index 4e96e9a..b8c559c 100644
--- a/sf-writer/include/JFH5Writer.hpp
+++ b/sf-writer/include/JFH5Writer.hpp
@@ -9,8 +9,10 @@
 
 class JFH5Writer {
 
-    const size_t n_images_;
+    const size_t start_pulse_id_;
+    const uint64_t stop_pulse_id_;
     const size_t n_modules_;
+    const size_t n_images_;
     size_t current_write_index_;
 
     H5::H5File file_;
@@ -24,7 +26,8 @@ class JFH5Writer {
 
 public:
     JFH5Writer(const std::string& output_file,
-               const size_t n_images,
+               const uint64_t start_pulse_id,
+               const uint64_t stop_pulse_id,
                const size_t n_modules);
     ~JFH5Writer();
     void write(const ImageMetadataBlock* metadata, const char* data);
diff --git a/sf-writer/src/JFH5Writer.cpp b/sf-writer/src/JFH5Writer.cpp
index 1abd2fb..5b1b5a7 100644
--- a/sf-writer/src/JFH5Writer.cpp
+++ b/sf-writer/src/JFH5Writer.cpp
@@ -12,12 +12,14 @@
 using namespace std;
 using namespace core_buffer;
 
-JFH5Writer::JFH5Writer(
-        const string& output_file,
-        const size_t n_images,
-        const size_t n_modules) :
-        n_images_(n_images),
+JFH5Writer::JFH5Writer(const std::string& output_file,
+                       const uint64_t start_pulse_id,
+                       const uint64_t stop_pulse_id,
+                       const size_t n_modules) :
+        start_pulse_id_(start_pulse_id),
+        stop_pulse_id_(stop_pulse_id),
         n_modules_(n_modules),
+        n_images_(stop_pulse_id - start_pulse_id + 1),
         current_write_index_(0)
 {