Add a matplotlib figure editor

Author: siddhantwahal

ets_tutorial/util/__init__.py

@@ -0,0 +1,36 @@
+import matplotlib
+from matplotlib.backends.backend_qt5agg import (
+    FigureCanvasQTAgg as FigureCanvas,
+    NavigationToolbar2QT as NavigationToolbar
+)
+from pyface.qt import QtGui
+from traitsui.api import BasicEditorFactory
+from traitsui.qt4.editor import Editor
+
+matplotlib.use('Qt5Agg')
+
+
+class _MplFigureEditor(Editor):
+
+    scrollable = True
+
+    def init(self, parent):
+        self.set_tooltip()
+        self.control = self._create_mpl_canvas(parent)
+
+    def update_editor(self):
+        pass
+
+    def _create_mpl_canvas(self, parent):
+        control = QtGui.QWidget()
+        canvas = FigureCanvas(figure=self.value)
+        toolbar = NavigationToolbar(canvas, control)
+        layout = QtGui.QVBoxLayout()
+        layout.addWidget(toolbar)
+        layout.addWidget(canvas)
+        control.setLayout(layout)
+        return control
+
+
+class MplFigureEditor(BasicEditorFactory):
+    klass = _MplFigureEditor

ets_tutorial/util/mpl_figure_editor.py

@@ -0,0 +1,37 @@
+import unittest
+
+import numpy as np
+from matplotlib.figure import Figure
+from traits.api import HasTraits, Instance
+from traitsui.api import Item, View
+from traitsui.testing.api import UITester, IsVisible
+
+from ets_tutorial.util.mpl_figure_editor import MplFigureEditor
+
+
+class Plot(HasTraits):
+    figure = Instance(Figure, ())
+
+    view = View(
+        Item("figure", editor=MplFigureEditor(), show_label=False),
+    )
+
+    def _figure_default(self):
+        figure = Figure()
+        axes = figure.add_subplot(111)
+        time = np.linspace(0, 2 * np.pi, 200)
+        axes.plot(
+            np.sin(time) * (1 + 0.5 * np.cos(11 * time)),
+            np.cos(time) * (1 + 0.5 * np.cos(11 * time)),
+        )
+        return figure
+
+
+class TestMplFigureEditor(unittest.TestCase):
+    def test_mpl_figure_editor(self):
+        tester = UITester()
+        plot = Plot()
+        with tester.create_ui(plot) as ui:
+            # smoke test: just check if Plot can create its UI
+            figure = tester.find_by_name(ui, "figure")
+            figure.inspect(IsVisible())