Added separate class to handle figure properties

Author: jo-mueller

src/napari_matplotlib/base.py

@@ -5,6 +5,27 @@
 
 __all__ = ["NapariMPLWidget"]
 
+class MplCanvas(FigureCanvas):
+    """
+    Defines the canvas of the matplotlib window
+    From https://github.com/haesleinhuepf/napari-workflow-inspector/blob/main/src/napari_workflow_inspector/_dock_widget.py
+    """
+    def __init__(self):
+        self.fig = Figure()                         # create figure
+        self.axes = self.fig.add_subplot(111)       # create subplot
+
+        self.axes.spines['bottom'].set_color('white')
+        self.axes.spines['top'].set_color('white')
+        self.axes.spines['left'].set_color('white')
+        self.axes.spines['right'].set_color('white')
+        self.fig.patch.set_facecolor('#262930')
+        self.axes.set_facecolor('#262930')
+        self.axes.grid(which='major', linestyle='--', color='white', alpha=0.6)
+        self.axes.tick_params(axis='both', colors='white')
+
+        FigureCanvas.__init__(self, self.fig)       # initialize canvas
+        FigureCanvas.updateGeometry(self)
+
 
 class NapariMPLWidget(QWidget):
     """
@@ -31,7 +52,7 @@ def __init__(self, napari_viewer: napari.viewer.Viewer):
 
         self.viewer = napari_viewer
         self.figure = Figure(figsize=(5, 3), tight_layout=True)
-        self.canvas = FigureCanvas(self.figure)
+        self.canvas = MplCanvas()
         self.toolbar = NavigationToolbar2QT(self.canvas, self)
         self.axes = self.canvas.figure.subplots()