import pyqtgraph as pg from pyqtgraph.Qt import QtGui, QtCore from Exporter import Exporter __all__ = ['MatplotlibExporter'] class MatplotlibExporter(Exporter): Name = "Matplotlib Window" windows = [] def __init__(self, item): Exporter.__init__(self, item) def parameters(self): return None def export(self, fileName=None): if isinstance(self.item, pg.PlotItem): mpw = MatplotlibWindow() MatplotlibExporter.windows.append(mpw) fig = mpw.getFigure() ax = fig.add_subplot(111) ax.clear() #ax.grid(True) for item in self.item.curves: x, y = item.getData() opts = item.opts pen = pg.mkPen(opts['pen']) if pen.style() == QtCore.Qt.NoPen: linestyle = '' else: linestyle = '-' color = tuple([c/255. for c in pg.colorTuple(pen.color())]) symbol = opts['symbol'] if symbol == 't': symbol = '^' symbolPen = pg.mkPen(opts['symbolPen']) symbolBrush = pg.mkBrush(opts['symbolBrush']) markeredgecolor = tuple([c/255. for c in pg.colorTuple(symbolPen.color())]) markerfacecolor = tuple([c/255. for c in pg.colorTuple(symbolBrush.color())]) if opts['fillLevel'] is not None and opts['fillBrush'] is not None: fillBrush = pg.mkBrush(opts['fillBrush']) fillcolor = tuple([c/255. for c in pg.colorTuple(fillBrush.color())]) ax.fill_between(x=x, y1=y, y2=opts['fillLevel'], facecolor=fillcolor) ax.plot(x, y, marker=symbol, color=color, linewidth=pen.width(), linestyle=linestyle, markeredgecolor=markeredgecolor, markerfacecolor=markerfacecolor) xr, yr = self.item.viewRange() ax.set_xbound(*xr) ax.set_ybound(*yr) mpw.draw() else: raise Exception("Matplotlib export currently only works with plot items") class MatplotlibWindow(QtGui.QMainWindow): def __init__(self): import pyqtgraph.widgets.MatplotlibWidget QtGui.QMainWindow.__init__(self) self.mpl = pyqtgraph.widgets.MatplotlibWidget.MatplotlibWidget() self.setCentralWidget(self.mpl) self.show() def __getattr__(self, attr): return getattr(self.mpl, attr) def closeEvent(self, ev): MatplotlibExporter.windows.remove(self)