70 lines
2.5 KiB
Python
70 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
PlotWidget.py - Convenience class--GraphicsView widget displaying a single PlotItem
|
|
Copyright 2010 Luke Campagnola
|
|
Distributed under MIT/X11 license. See license.txt for more infomation.
|
|
"""
|
|
|
|
from pyqtgraph.Qt import QtCore, QtGui
|
|
from GraphicsView import *
|
|
from pyqtgraph.graphicsItems.PlotItem import *
|
|
import exceptions
|
|
|
|
__all__ = ['PlotWidget']
|
|
class PlotWidget(GraphicsView):
|
|
|
|
#sigRangeChanged = QtCore.Signal(object, object) ## already defined in GraphicsView
|
|
|
|
"""
|
|
Widget implementing a graphicsView with a single PlotItem inside.
|
|
|
|
The following methods are wrapped directly from PlotItem: addItem, removeItem,
|
|
clear, setXRange, setYRange, setRange, setAspectLocked, setMouseEnabled. For all
|
|
other methods, use getPlotItem.
|
|
"""
|
|
def __init__(self, parent=None, **kargs):
|
|
GraphicsView.__init__(self, parent)
|
|
self.setSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding)
|
|
self.enableMouse(False)
|
|
self.plotItem = PlotItem(**kargs)
|
|
self.setCentralItem(self.plotItem)
|
|
## Explicitly wrap methods from plotItem
|
|
for m in ['addItem', 'removeItem', 'autoRange', 'clear', 'setXRange', 'setYRange', 'setRange', 'setAspectLocked', 'setMouseEnabled']:
|
|
setattr(self, m, getattr(self.plotItem, m))
|
|
#QtCore.QObject.connect(self.plotItem, QtCore.SIGNAL('viewChanged'), self.viewChanged)
|
|
self.plotItem.sigRangeChanged.connect(self.viewRangeChanged)
|
|
|
|
def close(self):
|
|
self.plotItem.close()
|
|
self.plotItem = None
|
|
#self.scene().clear()
|
|
#self.mPlotItem.close()
|
|
self.setParent(None)
|
|
GraphicsView.close(self)
|
|
|
|
def __getattr__(self, attr): ## implicitly wrap methods from plotItem
|
|
if hasattr(self.plotItem, attr):
|
|
m = getattr(self.plotItem, attr)
|
|
if hasattr(m, '__call__'):
|
|
return m
|
|
raise exceptions.NameError(attr)
|
|
|
|
def viewRangeChanged(self, view, range):
|
|
#self.emit(QtCore.SIGNAL('viewChanged'), *args)
|
|
self.sigRangeChanged.emit(self, range)
|
|
|
|
def widgetGroupInterface(self):
|
|
return (None, PlotWidget.saveState, PlotWidget.restoreState)
|
|
|
|
def saveState(self):
|
|
return self.plotItem.saveState()
|
|
|
|
def restoreState(self, state):
|
|
return self.plotItem.restoreState(state)
|
|
|
|
def getPlotItem(self):
|
|
"""Return the PlotItem contained within."""
|
|
return self.plotItem
|
|
|
|
|
|
|