From 82e5b9ec0bc9af29a3e8a5904a1913d7d7c3b399 Mon Sep 17 00:00:00 2001
From: Andrej Babic <andrej.babic@psi.ch>
Date: Thu, 12 Jul 2018 15:29:07 +0200
Subject: [PATCH] Update Bernina writer

---
 sf_bernina/bernina_h5_writer.cpp | 22 ++++++++++++----------
 1 file changed, 12 insertions(+), 10 deletions(-)

diff --git a/sf_bernina/bernina_h5_writer.cpp b/sf_bernina/bernina_h5_writer.cpp
index 50c3b0f..1133821 100644
--- a/sf_bernina/bernina_h5_writer.cpp
+++ b/sf_bernina/bernina_h5_writer.cpp
@@ -11,16 +11,17 @@
 
 int main (int argc, char *argv[])
 {
-    if (argc != 7) {
+    if (argc != 8) {
         cout << endl;
         cout << "Usage: bernina_h5_writer [connection_address] [output_file] [n_frames]";
-        cout << " [rest_port] [user_id] [bsread_address]" << endl;
+        cout << " [rest_port] [user_id] [bsread_address] [n_modules]" << endl;
         cout << "\tconnection_address: Address to connect to the stream (PULL). Example: tcp://127.0.0.1:40000" << endl;
         cout << "\toutput_file: Name of the output file." << endl;
         cout << "\tn_frames: Number of images to acquire. 0 for infinity (until /stop is called)." << endl;
         cout << "\trest_port: Port to start the REST Api on." << endl;
         cout << "\tuser_id: uid under which to run the writer. -1 to leave it as it is." << endl;
         cout << "\tbsread_address: HTTP address of the bsread REST api." << endl;
+        cout << "\tn_modules: Number of detector modules to be written." << endl;
         cout << endl;
 
         exit(-1);
@@ -32,6 +33,7 @@ int main (int argc, char *argv[])
     int rest_port = atoi(argv[4]);
     int user_id = atoi(argv[5]);
     string bsread_rest_address = string(argv[6]);
+    int n_modules = atoi(argv[7]);
 
     if (user_id != -1) {
         writer_utils::set_process_id(user_id);
@@ -48,17 +50,17 @@ int main (int argc, char *argv[])
         {"is_good_frame", HeaderDataType("uint64")},
         {"daq_rec", HeaderDataType("int64")},
 
-        {"pulse_id_diff", HeaderDataType("int64", 6)},
-        {"framenum_diff", HeaderDataType("int64", 6)},
+        {"pulse_id_diff", HeaderDataType("int64", n_modules)},
+        {"framenum_diff", HeaderDataType("int64", n_modules)},
 
-        {"missing_packets_1", HeaderDataType("uint64", 6)},
-        {"missing_packets_2", HeaderDataType("uint64", 6)},
-        {"daq_recs", HeaderDataType("uint64", 6)},
+        {"missing_packets_1", HeaderDataType("uint64", n_modules)},
+        {"missing_packets_2", HeaderDataType("uint64", n_modules)},
+        {"daq_recs", HeaderDataType("uint64", n_modules)},
         
-        {"pulse_ids", HeaderDataType("uint64", 6)},
-        {"framenums", HeaderDataType("uint64", 6)},
+        {"pulse_ids", HeaderDataType("uint64", n_modules)},
+        {"framenums", HeaderDataType("uint64", n_modules)},
         
-        {"module_number", HeaderDataType("uint64", 6)}
+        {"module_number", HeaderDataType("uint64", n_modules)}
     });
     
     ZmqReceiver receiver(connect_address, config::zmq_n_io_threads, config::zmq_receive_timeout, header_values);