version 0.1 of updated plot window. Still, buggy legend

b_asic/GUI/plot_window.py

+# TODO's:
+# * Solve the legend update. That isn't working at all.
+# * Change labels "All" and "None" into buttons.
+# * Make it work with the main_window (probably requires a rebase or merge)
+
+import re
+import sys
+
+from matplotlib.backends.backend_qt5agg import (
+    FigureCanvasQTAgg as FigureCanvas,
+)
+from matplotlib.figure import Figure
+from matplotlib.ticker import MaxNLocator
+from qtpy.QtCore import Qt
+from qtpy.QtGui import QKeySequence
+
+# Intereme imports for the Plot class:
+from qtpy.QtWidgets import (  # QFrame,; QScrollArea,; QLineEdit,
+    QApplication,
+    QCheckBox,
+    QDialog,
+    QFileDialog,
+    QHBoxLayout,
+    QLabel,
+    QListWidget,
+    QListWidgetItem,
+    QPushButton,
+    QShortcut,
+    QSizePolicy,
+    QVBoxLayout,
+)
+
+
+class PlotCanvas(FigureCanvas):
+    def __init__(self, logger, parent=None, width=5, height=4, dpi=100):
+        fig = Figure(figsize=(width, height), dpi=dpi)
+        super().__init__(fig)
+        self.axes = fig.add_subplot(111)
+        self.axes.xaxis.set_major_locator(MaxNLocator(integer=True))
+        self.legend = self.axes.legend()
+        self.logger = logger
+
+        FigureCanvas.updateGeometry(self)
+        self.save_figure = QShortcut(QKeySequence("Ctrl+S"), self)
+        self.save_figure.activated.connect(self._save_plot_figure)
+
+    def _save_plot_figure(self):
+        self.logger.info(f"Saving plot of figure: {self.sfg.name}.")
+        file_choices = "PNG (*.png)|*.png"
+        path, ext = QFileDialog.getSaveFileName(
+            self, "Save file", "", file_choices
+        )
+        path = path.encode("utf-8")
+        if not path[-4:] == file_choices[-4:].encode("utf-8"):
+            path += file_choices[-4:].encode("utf-8")
+
+        if path:
+            self.print_figure(path.decode(), dpi=self.dpi)
+            self.logger.info(f"Saved plot: {self.sfg.name} to path: {path}.")
+
+
+class PlotWindow(QDialog):
+    def __init__(
+        self,
+        window,
+        sfg_name,
+        sim_result,
+        logger,
+        parent=None,
+        width=5,
+        height=4,
+        dpi=100,
+    ):
+        super().__init__()
+        self._window = window
+        self.setWindowFlags(Qt.WindowTitleHint | Qt.WindowCloseButtonHint)
+        self.setWindowTitle("Simulation result")
+        self.sim_result = sim_result
+        self._auto_redraw = False
+
+        # Categorise sim_results into inputs, outputs, delays, others
+        sim_res_ins = {}
+        sim_res_outs = {}
+        sim_res_delays = {}
+        sim_res_others = {}
+
+        for key in sim_result:
+            if re.fullmatch("in[0-9]+", key):
+                sim_res_ins[key] = sim_result[key]
+            elif re.fullmatch("[0-9]+", key):
+                sim_res_outs[key] = sim_result[key]
+            elif re.fullmatch("t[0-9]+", key):
+                sim_res_delays[key] = sim_result[key]
+            else:
+                sim_res_others[key] = sim_result[key]
+
+        # Layout: ############################################
+        # | list |       |
+        # | ...  | plot  |
+        # | misc |       |
+
+        self.dialog_layout = QHBoxLayout()
+        self.setLayout(self.dialog_layout)
+
+        listlayout = QVBoxLayout()
+        self.plotcanvas = PlotCanvas(
+            logger=logger, parent=self, width=5, height=4, dpi=100
+        )
+
+        self.dialog_layout.addLayout(listlayout)
+        self.dialog_layout.addWidget(self.plotcanvas)
+
+        ########### Plot: ##############
+        # Do this before the list layout, as the list layout will re/set visibility
+        # Note: The order is of importens. Interesting lines last, to be on top.
+        self._lines = {}
+        for key in (
+            sim_res_others | sim_res_delays | sim_res_ins | sim_res_outs
+        ):
+            line = self.plotcanvas.axes.plot(
+                sim_result[key], visible=False, label=key
+            )
+            self._lines[key] = line
+
+        ########### List layout: ##############
+
+        # Add two labels for selecting all/none:
+        hlayout = QHBoxLayout()
+        labelAll = QLabel("All")
+        labelAll.mousePressEvent = self.labelAll_click
+        labelNone = QLabel("None")
+        labelNone.mousePressEvent = self.labelNone_click
+        hlayout.addWidget(labelAll)
+        hlayout.addWidget(labelNone)
+        listlayout.addLayout(hlayout)
+
+        # Add the entire list
+        self.checklist = QListWidget()
+        self.checklist.itemChanged.connect(self.item_change)
+        listitems = {}
+        for key in (
+            sim_res_ins | sim_res_outs | sim_res_delays | sim_res_others
+        ):
+            listitem = QListWidgetItem(key)
+            listitems[key] = listitem
+            self.checklist.addItem(listitem)
+            listitem.setCheckState(
+                Qt.Unchecked
+            )  # CheckState: Qt.{Unchecked, PartiallyChecked, Checked}
+        for key in sim_res_outs:
+            listitems[key].setCheckState(
+                Qt.Checked
+            )  # CheckState: Qt.{Unchecked, PartiallyChecked, Checked}
+        self.checklist.setFixedWidth(150)
+        listlayout.addWidget(self.checklist)
+
+        # Add a "legend" checkbox, connected to the plot.
+        self.legend_checkbox = QCheckBox("&Legend")
+        self.legend_checkbox.stateChanged.connect(self.legend_checkbox_change)
+        self.legend_checkbox.setCheckState(Qt.Checked)
+        listlayout.addWidget(self.legend_checkbox)
+
+        # Add "Close" buttons
+        buttonClose = QPushButton("&Close", self)
+        buttonClose.clicked.connect(self.close)
+        listlayout.addWidget(buttonClose)
+
+        # Done. Tell the functions below to redraw the canvas when needed.
+        # self.plotcanvas.draw()
+        self._auto_redraw = True
+
+    def legend_checkbox_change(self, checkState):
+        self.plotcanvas.legend.set(visible=(checkState == Qt.Checked))
+        if self._auto_redraw:
+            if checkState == Qt.Checked:
+                self.plotcanvas.legend = self.plotcanvas.axes.legend()
+            self.plotcanvas.draw()
+            # self.plotcanvas.legend
+        # if checkState == Qt.Checked:
+        #    print('legend on')
+        # else:
+        #    print('legend off')
+
+    def labelAll_click(self, event):
+        for x in range(self.checklist.count()):
+            self.checklist.item(x).setCheckState(Qt.Checked)
+
+    def labelNone_click(self, event):
+        for x in range(self.checklist.count()):
+            self.checklist.item(x).setCheckState(Qt.Unchecked)
+
+    def item_change(self, listitem):
+        key = listitem.text()
+        self._lines[key][0].set(visible=(listitem.checkState() == Qt.Checked))
+        if self._auto_redraw:
+            if self.legend_checkbox.checkState == Qt.Checked:
+                self.plotcanvas.legend = self.plotcanvas.axes.legend()
+            self.plotcanvas.draw()
+        # print(f"lines[{key}].set(visible={listitem.checkState() == Qt.Checked}). autodraw={self._auto_redraw}")
+        # print("Arg:", listitem)
+
+
+# Simple test of the dialog
+if __name__ == "__main__":
+    app = QApplication(sys.argv)
+    # sim_res = {"c1": [3, 6, 7], "c2": [4, 5, 5], "bfly1.0": [7, 0, 0], "bfly1.1": [-1, 0, 2], "0": [1, 2, 3]}
+    sim_res = {
+        '0': [0.5, 0.5, 0, 0],
+        'add1': [0.5, 0.5, 0, 0],
+        'cmul1': [0, 0.5, 0, 0],
+        'cmul2': [0.5, 0, 0, 0],
+        'in1': [1, 0, 0, 0],
+        't1': [0, 1, 0, 0],
+    }
+    win = PlotWindow(
+        window=None, sim_result=sim_res, sfg_name="hej", logger=print
+    )
+    win.exec_()
+    # win.show()
+
+
+# This is the original, used as a quick reference. Do not instantiate this:
+class Plot(FigureCanvas):
+    def __init__(
+        self, simulation, sfg, window, parent=None, width=5, height=4, dpi=100
+    ):
+        self.simulation = simulation
+        self.sfg = sfg
+        self.dpi = dpi
+        self._window = window
+
+        fig = Figure(figsize=(width, height), dpi=dpi)
+        fig.suptitle(sfg.name, fontsize=20)
+        self.axes = fig.add_subplot(111)
+
+        FigureCanvas.__init__(self, fig)
+        self.setParent(parent)
+
+        FigureCanvas.setSizePolicy(
+            self, QSizePolicy.Expanding, QSizePolicy.Expanding
+        )
+        FigureCanvas.updateGeometry(self)
+        self.save_figure = QShortcut(QKeySequence("Ctrl+S"), self)
+        self.save_figure.activated.connect(self._save_plot_figure)
+        self._plot_values_sfg()
+
+    def _save_plot_figure(self):
+        self._window.logger.info(f"Saving plot of figure: {self.sfg.name}.")
+        file_choices = "PNG (*.png)|*.png"
+        path, ext = QFileDialog.getSaveFileName(
+            self, "Save file", "", file_choices
+        )
+        path = path.encode("utf-8")
+        if not path[-4:] == file_choices[-4:].encode("utf-8"):
+            path += file_choices[-4:].encode("utf-8")
+
+        if path:
+            self.print_figure(path.decode(), dpi=self.dpi)
+            self._window.logger.info(
+                f"Saved plot: {self.sfg.name} to path: {path}."
+            )
+
+    def _plot_values_sfg(self):
+        x_axis = list(range(len(self.simulation.results["0"])))
+        for _output in range(self.sfg.output_count):
+            y_axis = self.simulation.results[str(_output)]
+            self.axes.plot(x_axis, y_axis)