From b4370be011dd55c8939014734e0392e3a2714b85 Mon Sep 17 00:00:00 2001
From: Andrej Babic <andrej.babic@psi.ch>
Date: Wed, 29 Aug 2018 16:01:50 +0200
Subject: [PATCH] Improve perf tool for writing to disk

---
 test/h5_write_perf.cpp | 86 +++++++++++++++++++++++++++++++++---------
 1 file changed, 68 insertions(+), 18 deletions(-)

diff --git a/test/h5_write_perf.cpp b/test/h5_write_perf.cpp
index cbe1d9a..48a956b 100644
--- a/test/h5_write_perf.cpp
+++ b/test/h5_write_perf.cpp
@@ -1,20 +1,47 @@
 #include <iostream>
 #include <sstream>
 #include <stdexcept>
+#include <unistd.h>
+#include <string>
 
 #include "H5Writer.hpp"
 
 using namespace std;
+using namespace std::chrono;
+
+void write_frame(H5Writer& writer, size_t index, char* buffer, size_t buffer_length, 
+    char* metadata_buffer, size_t metadata_buffer_length, size_t n_metadata, size_t n_modules) {
+
+    writer.write_data("data",
+                      index, 
+                      buffer,
+                      {n_modules * 512, 1024},
+                      buffer_length, 
+                      "uint16",
+                      "little");
+
+    for (size_t meta_index=0; meta_index < n_metadata; meta_index++) {
+        
+        writer.write_data(to_string(meta_index),
+                          index, 
+                          metadata_buffer,
+                          {n_modules},
+                          metadata_buffer_length, 
+                          "uint64",
+                          "little");
+    }
+} 
 
 int main (int argc, char *argv[])
 {
-    if (argc != 5) {
+    if (argc != 6) {
         cout << endl;
-        cout << "Usage: h5_write_perf [output_file] [n_frames] [frame_size_x] [frame_size_y]" << endl;
+        cout << "Usage: h5_write_perf [output_file] [n_frames] [n_modules] [frame_rate] [n_metadata]" << endl;
         cout << "\toutput_file: Name of the output file." << endl;
         cout << "\tn_frames: Number of images to write." << endl;
-        cout << "\frame_size_x: Frame width in pixels." << endl;
-        cout << "\frame_size_y: Frame height in pixels." << endl;
+        cout << "\tn_modules: Numbers of 512*1024 modules." << endl;
+        cout << "\tframe_rate: Frame rate in Hz." << endl;
+        cout << "\tn_metadata: Number of metadata datasets to be written." << endl;
         cout << endl;
 
         exit(-1);
@@ -22,29 +49,52 @@ int main (int argc, char *argv[])
 
     string output_file = string(argv[1]);
     int n_frames =  atoi(argv[2]);
-    int frame_size_x = atoi(argv[3]);
-    int frame_size_y = atoi(argv[4]);
+    int n_modules = atoi(argv[3]);
+    int frame_rate = atoi(argv[4]);
+    int n_metadata = atoi(argv[5]);
+
+    size_t buffer_length = n_modules * 512 * 1024 * sizeof(u_int16_t);
+    char* buffer = new char[buffer_length];
+   
+    size_t metadata_buffer_length = 4 * n_modules;
+    char* metadata_buffer = new char[metadata_buffer_length];
 
     H5Writer writer(output_file, n_frames, n_frames, n_frames);
 
-    size_t buffer_length = frame_size_x * frame_size_y * sizeof(u_int16_t);
-    char* buffer = new char[buffer_length];
+    // Initialize all datasets;
+    write_frame(writer, 0, buffer, buffer_length, metadata_buffer, metadata_buffer_length, n_metadata, n_modules);
 
-    string dataset_name = "data";
+    auto total_sleep_time = 0.0;
+    auto total_write_time = 0.0;
+    auto missed_frames = 0;
+    
+    auto start_time_frame = std::chrono::system_clock::now();
+    
+    for (int index=0; index<n_frames; index++) {
+        
+        write_frame(writer, index, buffer, buffer_length, metadata_buffer, metadata_buffer_length, n_metadata, n_modules);
 
-    for (size_t index=0; index<n_frames; index++) {
-        writer.write_data(dataset_name,
-                          index, 
-                          buffer,
-                          {frame_size_y, frame_size_x},
-                          buffer_length, 
-                          "uint16",
-                          "little");
+        auto time_diff = duration<float, milli>(std::chrono::system_clock::now() - start_time_frame).count();
+        auto sleep_time = (1.0/frame_rate*1000) - time_diff;
+
+        if (sleep_time < 0) {
+            cout << "Not in time for frame " << index << endl;
+            sleep_time = 0;
+            missed_frames++;
+        }
+
+        usleep(sleep_time*1000);
+
+        total_sleep_time += sleep_time;
+        total_write_time += time_diff;
+
+        start_time_frame = std::chrono::system_clock::now();
     }
 
     writer.close_file();
 
+    cout << "total sleep: " << total_sleep_time/1000 << " total write: " << total_write_time/1000;
+    cout << " missed frames: " << missed_frames/float(n_frames)*100 << "%" <<endl;
     
-
     return 0;
 }