test and live plotting

src/qililab/result/live_plotting.py

@@ -0,0 +1,45 @@
+import h5py
+import numpy as np
+import matplotlib.pyplot as plt
+import time
+
+class LivePlot:
+    """
+    Live plotter for results saved in an HDF5 file with SWMR mode enabled.
+
+    Args:
+        path (str): Path to the HDF5 file.
+        refresh_interval (float): Interval in seconds to refresh the plot.
+    """
+
+    def __init__(self, path: str, refresh_interval: float = 1.0):
+        self.path = path
+        self.refresh_interval = refresh_interval
+        self.fig, self.ax = plt.subplots()
+        self.line, = self.ax.plot([], [], 'b-')  # Initialize an empty line
+
+    def update_plot(self, data):
+        """Update the plot with new data."""
+        x = np.arange(data.shape[0])
+        self.line.set_data(x, data[:, 0])  # Assuming we're plotting the first column only
+        self.ax.relim()
+        self.ax.autoscale_view()
+
+    def run(self):
+        """Continuously reads from the HDF5 file and updates the plot."""
+        plt.ion()  # Interactive mode on
+        with h5py.File(self.path, "r", swmr=True) as f:
+            dataset = f["results"]
+            last_size = dataset.shape[0]
+
+            while plt.fignum_exists(self.fig.number):  # Run until the plot window is closed
+                current_size = dataset.shape[0]
+                if current_size > last_size:  # Check if there's new data
+                    data = dataset[:]
+                    self.update_plot(data)
+                    plt.draw()
+                    plt.pause(0.01)  # Small pause to refresh the plot
+                    last_size = current_size
+                time.sleep(self.refresh_interval)
+
+        plt.ioff()  # Turn interactive mode off

tests/result/test_stream_results.py

@@ -50,7 +50,9 @@ def create_dataset(self, _: str, data: np.ndarray):
 
     def __exit__(self, *_):
         """mocks exit"""
-
+
+    def flush():
+        """mocks flush"""
 
 class TestStreamArray:
     """Test `StreamArray` functionalities."""