Draft: Added support for plotting result from many simulations. Solves #209

b_asic/gui_utils/plot_window.py

@@ -38,7 +38,7 @@ class PlotWindow(QWidget):
     sim_result : dict
         Simulation results of the form obtained from :attr:`~b_asic.simulation.Simulation.results`.
         Alternative, sim_result can be a list, ['name1', sim_result1, 'name2', sim_result2, ...]
-    sfg_name : str, optional
+    figure_name : str, optional
         The name of the SFG.
     parent : optional
         The parent window.
@@ -47,7 +47,7 @@ class PlotWindow(QWidget):
     def __init__(
         self,
         sim_result: Mapping[ResultKey, Sequence[Num]],
-        sfg_name: Optional[str] = None,
+        figure_name: Optional[str] = None,
     ):
         super().__init__()
         self.setWindowFlags(
@@ -57,8 +57,8 @@ class PlotWindow(QWidget):
             | Qt.WindowMaximizeButtonHint
         )
         title = (
-            f"Simulation results: {sfg_name}"
-            if sfg_name is not None
+            f"Simulation results: {figure_name}"
+            if figure_name is not None
             else "Simulation results"
         )
         self.setWindowTitle(title)
@@ -238,8 +238,8 @@ class PlotWindow(QWidget):
         # Add additional checkboxes
         self._legend = self._plot_axes.legend()
         self._legend_checkbox = QCheckBox("&Legend")
-        self._legend_checkbox.stateChanged.connect(self._legend_checkbox_change)
         self._legend_checkbox.setCheckState(Qt.CheckState.Checked)
+        self._legend_checkbox.stateChanged.connect(self._legend_checkbox_change)
         self._legend_checkbox.setIcon(get_icon('legend'))
         listlayout.addWidget(self._legend_checkbox)
         # self.ontop_checkbox = QCheckBox("&On top")
@@ -261,12 +261,60 @@ class PlotWindow(QWidget):
         # self.plotcanvas.draw()
         self._auto_redraw = True
 
+    def add_result(self, name, result) -> None:
+        """
+        Add another result to the plot.
+
+        The new signals are added in natural order without sorting them.
+        ---
+        name: str
+          The name of the result
+        result: dict
+          The result from one simulation.
+        """
+        markers = ".ov<^>s+*xd|_"
+        ix = 0
+        self._auto_redraw = False
+
+        def addline(key, vector, checked):
+            line = self._plot_axes.plot(np.real(res), fmt, label=key)
+            self._lines[key] = line[0]
+            list_item = QListWidgetItem(key)
+            list_item.setCheckState(Qt.CheckState.Unchecked)  # will add it if checked
+            self._checklist.addItem(list_item)
+            self._lines[key].remove()  # remove the line from plot. Keep it in _lines.
+            list_item.setCheckState(checked)  # will add it if checked
+
+        for key, res in result.items():
+            key2 = name + "." + key
+            ischecked = Qt.CheckState.Unchecked
+            if re.fullmatch(r"[0-9]+", key):
+                key2 = name + '.out' + key
+                ischecked = Qt.CheckState.Checked
+
+            if len(res) <= 100:
+                fmt = markers[ix] + '-'
+                ix = (ix + 1) % len(markers)
+            else:
+                fmt = '-'
+
+            if all(np.imag(np.real_if_close(res)) == 0):
+                # real: add one line with corresponding checkbox
+                addline(key2, np.real(res), ischecked)
+            else:
+                # complex: add '_re', '_im', '_mag', '_ang'
+                addline(key2 + "_re", np.real(res), ischecked)
+                addline(key2 + "_im", np.imag(res), ischecked)
+                addline(key2 + "_mag", np.absolute(res), Qt.CheckState.Unchecked)
+                addline(key2 + "_ang", np.angle(res), Qt.CheckState.Unchecked)
+        self._auto_redraw = True
+        self._update_legend()
+        self._plot_canvas.draw()
+
     def _legend_checkbox_change(self, check_state):
         self._legend.set(visible=(check_state == Qt.CheckState.Checked))
-        if self._auto_redraw:
-            if check_state == Qt.CheckState.Checked:
-                self._legend = self._plot_axes.legend()
-            self._plot_canvas.draw()
+        self._update_legend()
+        self._plot_canvas.draw()
 
     # def _ontop_checkbox_change(self, checkState):
     # Bugg: It seems the window closes if you change the WindowStaysOnTopHint.
@@ -281,10 +329,12 @@ class PlotWindow(QWidget):
             self._checklist.item(x).setCheckState(Qt.CheckState.Checked)
         self._auto_redraw = True
         self._update_legend()
+        self._plot_canvas.draw()
 
     def _update_legend(self):
-        self._legend = self._plot_axes.legend()
-        self._plot_canvas.draw()
+        # if self._legend_checkbox.checkState == Qt.CheckState.Checked:
+        if self._legend_checkbox.isChecked():
+            self._legend = self._plot_axes.legend()
 
     def _button_none_click(self, event):
         self._auto_redraw = False
@@ -292,6 +342,7 @@ class PlotWindow(QWidget):
             self._checklist.item(x).setCheckState(Qt.CheckState.Unchecked)
         self._auto_redraw = True
         self._update_legend()
+        self._plot_canvas.draw()
 
     def _item_change(self, listitem):
         key = listitem.text()
@@ -301,6 +352,7 @@ class PlotWindow(QWidget):
             self._lines[key].remove()
         if self._auto_redraw:
             self._update_legend()
+            self._plot_canvas.draw()
 
     def _relim(self, event=None):
         self._plot_axes.relim(True)
@@ -313,6 +365,16 @@ class PlotWindow(QWidget):
 def start_simulation_dialog(
     sim_results: Dict[str, List[complex]], sfg_name: Optional[str] = None
 ):
+    """
+    deprecated::
+        Use :func:`show_simulation_result` instead.
+    """
+    show_simulation_result(sim_results, sfg_name)
+
+
+def show_simulation_result(
+    sim_results: Dict[str, List[complex]], figure_name: Optional[str] = None
+) -> PlotWindow:
     """
     Display the simulation results window.
 
@@ -321,19 +383,34 @@ def start_simulation_dialog(
     sim_results : dict
         Simulation results of the form obtained from :attr:`~b_asic.simulation.Simulation.results`.
         Alternative, sim_result can be a list, ['name1', sim_result1, 'name2', sim_result2, ...]
-    sfg_name : str, optional
+    figure_name : str, optional
         The name of the SFG.
+    Note: This starts the GUI event loop using `QApplication.instance().exec()`, and the function
+    will return when the window is closed.
+    If this is not desired, you can in ipython start the event loop using `%gui qt`, and define
+    a global variable `noqtexec = 1`.
+    Effect if no event loop is running: The windows turns black and do not reponse.
+    Effect if the event loop is started twice: The function will never return.
     """
     if not QApplication.instance():
         app = QApplication(sys.argv)
     else:
         app = QApplication.instance()
-    win = PlotWindow(sim_result=sim_results, sfg_name=sfg_name)
+    win = PlotWindow(sim_result=sim_results, figure_name=figure_name)
     win.show()
-    app.exec_()
+
+    # I did not get "if 'noqtexec' in globals()" to work. Try this instead:
+    try:
+        if noqtexec != 1:
+            raise NameError()
+    except NameError:
+        app.exec()
+    return win
 
 
 # Simple test of the dialog
+from time import sleep
+
 if __name__ == "__main__":
     res1 = {
         '0': [1.5, 1.6, 1.5, 1],
@@ -360,8 +437,18 @@ if __name__ == "__main__":
         't3': [0, 0, 0, 1],
     }
     res3 = {
-        '0': np.random.rand(200).tolist(),
+        '0': np.random.rand(50),
         '1': np.random.rand(200).tolist(),
     }
-    # start_simulation_dialog(res3)
-    start_simulation_dialog(['Real', res1, 'Cpx', res2, res3], "Test data")
+    # res4 = {
+    #    '0': np.random.rand(60).tolist(),
+    #    '1': np.random.rand(220).tolist(),
+    #    't4': np.random.rand(50).tolist(),
+    # }
+    win = show_simulation_result(['r1', res1, res2])
+    try:
+        if noqtexec == 1:
+            sleep(5)
+            win.add_result('hej', res3)
+    except NameError:
+        pass