Source code for edifice.components.plotting

import time
import threading
from ..base_components import CustomWidget
from .._component import Component, register_props

from ..qt import QT_VERSION
if QT_VERSION == "PyQt5":
    from PyQt5 import QtWidgets
    from PyQt5 import QtCore
else:
    from PySide2 import QtCore, QtWidgets

try:
    MATPLOTLIB_LOADED = True
    from matplotlib.backends.backend_qt5agg import FigureCanvas
    from matplotlib.figure import Figure as MatplotlibFigure
except:
    MATPLOTLIB_LOADED = False


class Figure(CustomWidget):

    @register_props
    def __init__(self, plot_fun):
        super().__init__()
        self.figure_added = False
        self.figure_canvas = None
        self.subplots = None
        self.current_plot_fun = None
        self.current_plotted_fun = None
        self.plot_thread_should_run = True
        self.plot_thread = threading.Thread(target=self.plot)

[docs]    def did_mount(self):
        self.plot_thread.start()

[docs]    def will_unmount(self):
        self.plot_thread_should_run = False
        self.plot_thread.join()

[docs]    def plot(self):
        time.sleep(0.5)
        while self.plot_thread_should_run:
            plot_fun = self.current_plot_fun
            if plot_fun is not None and self.current_plotted_fun != plot_fun:
                self.subplots.clear()
                plot_fun(self.subplots)
                self.figure_canvas.draw()
                self.figure_canvas.figure.tight_layout()
                self.current_plotted_fun = plot_fun
            time.sleep(.016)

[docs]    def create_widget(self):
        widget = QtWidgets.QWidget()
        layout = QtWidgets.QVBoxLayout(widget)
        self.figure_added = False
        return widget

[docs]    def paint(self, widget, newprops):
        if "plot_fun" in newprops:
            if self.figure_added:
                self.current_plot_fun = self.props.plot_fun
            else:
                self.figure_canvas = FigureCanvas(MatplotlibFigure(figsize=(5, 3)))
                self.subplots = self.figure_canvas.figure.subplots()
                self.current_plot_fun = self.props.plot_fun
                widget.layout().addWidget(self.figure_canvas)
                self.figure_added = True


if not MATPLOTLIB_LOADED:
[docs]    def Figure(*args, **kwargs):
        raise ValueError("To use the Figure component, you must install matplotlib, e.g. by `pip install matplotlib`")
Edifice documentation

Source code for edifice.components.plotting
