Source code for

#  Pyctools - a picture processing algorithm development kit.
#  Copyright (C) 2016-19  Pyctools contributors
#  This program is free software: you can redistribute it and/or
#  modify it under the terms of the GNU General Public License as
#  published by the Free Software Foundation, either version 3 of the
#  License, or (at your option) any later version.
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  General Public License for more details.
#  You should have received a copy of the GNU General Public License
#  along with this program.  If not, see
#  <>.

__all__ = ['PlotData']
__docformat__ = 'restructuredtext en'

# matplotlib and pyctools.core.frame both import GObject, but pyctools
# will load pgi if available. pgi has to be set up before importing
# GObject, so import pyctools.core.frame before matplotlib
import pyctools.core.frame

# matplotlib sometimes chooses the wrong backend, so make it use Qt5
import matplotlib as mpl
import matplotlib.pyplot as plt

from pyctools.core.base import Component
from pyctools.core.qt import QtEventLoop

[docs] class PlotData(Component): """Plot data on a graph. The input frame's data should be a 2-D :py:class:`numpy:numpy.ndarray`. The first row contains horizontal (X) coordinates and the remaining rows contain corresponding Y values. """ with_outframe_pool = False outputs = [] event_loop = QtEventLoop def initialise(self): self.first_plot = True def process_frame(self): in_frame = self.input_buffer['input'].get() labels = eval(in_frame.metadata.get('labels', '[]')) data = in_frame.as_numpy() x = data[0] if self.first_plot: plt.ion() self.lines = [] for i, y in enumerate(data[1:]): self.lines.append(plt.plot(x, y, '-')[0]) else: for i, y in enumerate(data[1:]): self.lines[i].set_xdata(x) self.lines[i].set_ydata(y) if labels: plt.xlabel(labels[0]) if len(labels) > 1: for i in range(min(len(labels) - 1, len(self.lines))): self.lines[i].set(label=labels[i + 1]) plt.legend() if self.first_plot: plt.grid(True) else: plt.draw()