matplotlibwidget.py

# -*- coding: utf-8 -*-
"""
Created on Tue Jun 24 00:27:01 2014
 
@author: fvidal
"""
"""
from PyQt4 import QtGui
from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
 
from matplotlib.figure import Figure
 
class MplCanvas(FigureCanvas):
 
    def __init__(self):
        self.fig = Figure(facecolor='white')
        self.axes = self.fig.add_subplot(111)
        # We want the axes cleared every time plot() is called
        self.axes.hold(False)
        FigureCanvas.__init__(self, self.fig)
        FigureCanvas.setSizePolicy(self, QtGui.QSizePolicy.Expanding,QtGui.QSizePolicy.Expanding)
        FigureCanvas.updateGeometry(self)
 
 
class MatplotlibWidget(QtGui.QWidget):
 
    def __init__(self, parent = None):
        QtGui.QWidget.__init__(self, parent)
        self.canvas = MplCanvas()
        self.vbl = QtGui.QVBoxLayout()
        self.vbl.addWidget(self.canvas)
        self.setLayout(self.vbl)
"""
 
try:
    from PyQt5 import QtWidgets
except ModuleNotFoundError as e:
    try:
        from PySide2 import QtWidgets
    except ModuleNotFoundError as e:
        raise ModuleNotFoundError("No module named 'PyQt5' or PySide2', please install one of them\nException raised: "+e.msg)
 
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.backends.backend_qt5agg import NavigationToolbar2QT as Navigationtoolbar
from matplotlib.figure import Figure
from matplotlib import gridspec
 
# import matplotlib
# matplotlib.use('Qt5Agg')
 
# matplotlib.rcParams['backend.qt4']='PySide'
# matplotlib.style.use('ggplot')
# matplotlib.style.use('seaborn-muted')
 
 
#Embeddable matplotlib figure/canvas
class MplCanvas(FigureCanvas):
 
    def __init__(self):
        self.figfig = Figure(frameon=True)
        self.gs1gs1 = gridspec.GridSpec(1, 1)
        self.axesaxes = self.figfig.add_subplot(self.gs1gs1[0], aspect="auto")
 
        FigureCanvas.__init__(self, self.figfig)
        FigureCanvas.setSizePolicy(self, QtWidgets.QSizePolicy.Expanding,
                                   QtWidgets.QSizePolicy.Expanding)
        FigureCanvas.updateGeometry(self)
 
 
#creates embeddable matplotlib figure/canvas with toolbar
class MatplotlibWidget(QtWidgets.QWidget):
 
    def __init__(self, parent=None):
        QtWidgets.QWidget.__init__(self, parent)
        self.create_framentoolbarcreate_framentoolbar()
 
    def create_framentoolbar(self):
        self.frameframe = QtWidgets.QWidget()
        self.canvascanvas = MplCanvas()
        self.canvascanvas.setParent(self.frameframe)
        self.mpltoolbarmpltoolbar = Navigationtoolbar(self.canvascanvas, self.frameframe)
        self.vblvbl = QtWidgets.QVBoxLayout()
        self.vblvbl.addWidget(self.canvascanvas)
        self.vblvbl.addWidget(self.mpltoolbarmpltoolbar)
        self.setLayout(self.vblvbl)