149 lines
4.7 KiB
Python
149 lines
4.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright © 2009 Pierre Raybaut
|
|
# Licensed under the terms of the MIT License
|
|
|
|
"""
|
|
MatplotlibWidget
|
|
================
|
|
|
|
Example of matplotlib widget for PyQt4
|
|
|
|
Copyright © 2009 Pierre Raybaut
|
|
This software is licensed under the terms of the MIT License
|
|
|
|
Derived from 'embedding_in_pyqt4.py':
|
|
Copyright © 2005 Florent Rougon, 2006 Darren Dale
|
|
"""
|
|
from QDSToolbar import CustomToolbar
|
|
|
|
__version__ = "1.0.0"
|
|
|
|
from PyQt4.QtGui import QSizePolicy, QWidget, QVBoxLayout
|
|
from PyQt4.QtCore import QSize, Qt
|
|
|
|
from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as Canvas
|
|
from matplotlib.figure import Figure
|
|
|
|
from matplotlib import rcParams
|
|
|
|
rcParams['font.size'] = 9
|
|
|
|
|
|
class MatplotlibWidget(Canvas):
|
|
"""
|
|
MatplotlibWidget inherits PyQt4.QtGui.QWidget
|
|
and matplotlib.backend_bases.FigureCanvasBase
|
|
|
|
Options: option_name (default_value)
|
|
-------
|
|
parent (None): parent widget
|
|
title (''): figure title
|
|
xlabel (''): X-axis label
|
|
ylabel (''): Y-axis label
|
|
xlim (None): X-axis limits ([min, max])
|
|
ylim (None): Y-axis limits ([min, max])
|
|
xscale ('linear'): X-axis scale
|
|
yscale ('linear'): Y-axis scale
|
|
width (4): width in inches
|
|
height (3): height in inches
|
|
dpi (100): resolution in dpi
|
|
hold (False): if False, figure will be cleared each time plot is called
|
|
|
|
Widget attributes:
|
|
-----------------
|
|
figure: instance of matplotlib.figure.Figure
|
|
axes: figure axes
|
|
|
|
Example:
|
|
-------
|
|
self.widget = MatplotlibWidget(self, yscale='log', hold=True)
|
|
from numpy import linspace
|
|
x = linspace(-10, 10)
|
|
self.widget.axes.plot(x, x**2)
|
|
self.wdiget.axes.plot(x, x**3)
|
|
"""
|
|
|
|
def __init__(self, parent=None, title='', xlabel='', ylabel='',
|
|
xlim=None, ylim=None, xscale='linear', yscale='linear',
|
|
width=6, height=4, dpi=100, hold=False):
|
|
self.figure = Figure(figsize=(width, height), dpi=dpi)
|
|
self.axes = self.figure.add_subplot(111)
|
|
self.axes.set_title(title)
|
|
self.axes.set_xlabel(xlabel)
|
|
self.axes.set_ylabel(ylabel)
|
|
if xscale is not None:
|
|
self.axes.set_xscale(xscale)
|
|
if yscale is not None:
|
|
self.axes.set_yscale(yscale)
|
|
if xlim is not None:
|
|
self.axes.set_xlim(*xlim)
|
|
if ylim is not None:
|
|
self.axes.set_ylim(*ylim)
|
|
self.axes.hold(hold)
|
|
|
|
Canvas.__init__(self, self.figure)
|
|
self.setParent(parent)
|
|
|
|
Canvas.setSizePolicy(self, QSizePolicy.Expanding, QSizePolicy.Expanding)
|
|
Canvas.updateGeometry(self)
|
|
|
|
def sizeHint(self):
|
|
w, h = self.get_width_height()
|
|
return QSize(w, h)
|
|
|
|
def minimumSizeHint(self):
|
|
return QSize(10, 10)
|
|
|
|
|
|
#===============================================================================
|
|
# Example
|
|
#===============================================================================
|
|
if __name__ == '__main__':
|
|
import sys
|
|
from PyQt4.QtGui import QMainWindow, QApplication
|
|
from numpy import linspace
|
|
|
|
class ApplicationWindow(QMainWindow):
|
|
def __init__(self):
|
|
QMainWindow.__init__(self)
|
|
self.mplwidget = MatplotlibWidget(self, title='Example',
|
|
xlabel='Linear scale',
|
|
ylabel='Log scale',
|
|
hold=True, yscale='log')
|
|
self.mplwidget.setFocus()
|
|
self.setCentralWidget(self.mplwidget)
|
|
self.plot(self.mplwidget.axes)
|
|
|
|
def plot(self, axes):
|
|
x = linspace(-10, 10)
|
|
axes.plot(x, x ** 2)
|
|
axes.plot(x, x ** 3)
|
|
|
|
app = QApplication(sys.argv)
|
|
win = ApplicationWindow()
|
|
win.show()
|
|
sys.exit(app.exec_())
|
|
|
|
|
|
class PlotWidget(QWidget):
|
|
def __init__(self, parent=None):
|
|
QWidget.__init__(self)
|
|
super(PlotWidget, self).__init__(parent)
|
|
self.mplwidget = MatplotlibWidget(hold=True,
|
|
xlim=(1e-2, 1e7),
|
|
xscale='log',
|
|
yscale='log')
|
|
self.canvas = self.mplwidget.figure.canvas # shortcut
|
|
self.canvas.axes.grid(True)
|
|
#self.bbox_size = self.canvas.axes.bbox.size
|
|
self.toolbar = CustomToolbar(self.canvas, self.mplwidget, parent)
|
|
self.toolbar.setToolButtonStyle(Qt.ToolButtonTextUnderIcon)
|
|
layout = QVBoxLayout(parent)
|
|
#self.mplwidget.setLayout(layout)
|
|
layout.addWidget(self.canvas)
|
|
layout.addWidget(self.mplwidget)
|
|
layout.addWidget(self.toolbar)
|
|
self._bg_cache = None
|
|
self._axvlims = []
|
|
self._axvname = [] |