diff --git a/visualisation_components.py b/visualisation_components.py index 2bd0016..e01f8b0 100644 --- a/visualisation_components.py +++ b/visualisation_components.py @@ -222,8 +222,11 @@ def plot(self, data: list[float] | None = None) -> None: self._changing_plot = False def export(self) -> None: - return NotImplemented - + scene = self._plotWidget.sceneObj + if scene.exportDialog is None: + from pyqtgraph.GraphicsScene import exportDialog + scene.exportDialog = exportDialog.ExportDialog(scene) + scene.exportDialog.show(self._plotItem) class ScatterPlot3D(QVBoxLayout, PlotLayout): """3d scatterplot layout, inherits plotlayout""" @@ -693,7 +696,11 @@ def clear(self) -> None: self._update_plot(yticks, widths) def export(self) -> None: - return NotImplemented + scene = self._plotWidget.sceneObj + if scene.exportDialog is None: + from pyqtgraph.GraphicsScene import exportDialog + scene.exportDialog = exportDialog.ExportDialog(scene) + scene.exportDialog.show(self._plotWidget.getPlotItem()) class Table(QTableWidget):