From e88c2e66307b951f686f363d369d4d9539def90b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Berkay=20=C3=9Cr=C3=BCn?=
 <berkay.ueruen@aisec.fraunhofer.de>
Date: Mon, 4 Sep 2023 15:29:07 +0000
Subject: [PATCH] Fix progress bar flickering

---
 controller.py | 6 ------
 hdf5logger.py | 3 +++
 2 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/controller.py b/controller.py
index 9be2167..bdfcc5e 100755
--- a/controller.py
+++ b/controller.py
@@ -702,9 +702,6 @@ def controller(
     # Handlers are used for a graceful exit, in case of a signal
     register_signal_handlers()
 
-    pbar = tqdm(
-        total=len(faultlist), desc="Simulating faults", disable=not len(faultlist)
-    )
     itter = 0
     while 1:
         if stop_signal_received.value == 1:
@@ -789,8 +786,6 @@ def controller(
             # Find finished processes
             p["process"].join(timeout=0)
             if p["process"].is_alive() is False:
-                # Update the progress bar
-                pbar.update(1)
                 # Recalculate moving average
                 p_time_list.append(current_time - p["start_time"])
                 len_p_time_list = len(p_time_list)
@@ -803,7 +798,6 @@ def controller(
                 break
 
     clogger.debug("{} experiments remaining in queue".format(queue_output.qsize()))
-    pbar.close()
     p_logger.join()
 
     clogger.debug("Done with qemu and logger")
diff --git a/hdf5logger.py b/hdf5logger.py
index d2568ee..6a99ff9 100644
--- a/hdf5logger.py
+++ b/hdf5logger.py
@@ -506,6 +506,7 @@ def hdf5collector(
         ):
             n._f_remove(recursive=True)
 
+    pbar = tqdm(total=num_exp, desc="Simulating faults", disable=not num_exp)
     while num_exp > 0 or log_goldenrun or log_pregoldenrun or log_config:
         if stop_signal.value == 1:
             break
@@ -531,6 +532,7 @@ def hdf5collector(
                     )
                 )
             num_exp = num_exp - 1
+            pbar.update(1)
         elif exp["index"] == -2 and log_pregoldenrun:
             if "Pregoldenrun" in f.root:
                 raise ValueError("Pregoldenrun already exists!")
@@ -584,5 +586,6 @@ def hdf5collector(
 
         del exp
 
+    pbar.close()
     f.close()
     logger.debug("Data Logging done")