def setPlotWidget(self):
self.plotWidget = QwtPlot(self.plotFrame)
sizePolicy = QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(1)
sizePolicy.setHeightForWidth(
self.plotWidget.sizePolicy().hasHeightForWidth())
self.plotWidget.setSizePolicy(sizePolicy)
self.plotWidget.setMinimumSize(QSize(0, 0))
self.plotWidget.setAutoFillBackground(False)
self.plotWidget.setCanvasBackground(Qt.white)
self.plotWidget.plotLayout().setAlignCanvasToScales(True)
zoomer = QwtPlotZoomer(QwtPlot.xBottom, QwtPlot.yLeft,
QwtPicker.DragSelection, QwtPicker.AlwaysOff,
self.plotWidget.canvas())
zoomer.setRubberBandPen(QPen(Qt.blue))
# The pen to draw zone of zooming.
picker = QwtPlotPicker(QwtPlot.xBottom, QwtPlot.yLeft,
QwtPicker.NoSelection,
QwtPlotPicker.CrossRubberBand,
QwtPicker.AlwaysOn, self.plotWidget.canvas())
picker.setTrackerPen(QPen(Qt.green))
grid = QwtPlotGrid()
grid.setPen(QPen(QColor('grey'), 0, Qt.DotLine))
grid.attach(self.plotWidget)
layout = self.plotLayout
layout.addWidget(self.plotWidget)
class QwtPlotE(QwtPlot):
"""a slightly enhanced version of QwtPlot which already has some sugar on it"""
def __init__(self, *arg):
QwtPlot.__init__(self, *arg)
self.plotLayout().setAlignCanvasToScales(True)
self.setCanvasBackground(Qt.white)
self.legend = QwtLegend()
self.legend.setItemMode(QwtLegend.ClickableItem)
self.insertLegend(self.legend, QwtPlot.TopLegend)
self.grid = QwtPlotGrid()
self.grid.attach(self)
self.grid.setPen(QPen(Qt.black, 0, Qt.DotLine))
self.zoomer = QwtPlotZoomer(QwtPlot.xBottom, QwtPlot.yLeft, QwtPicker.DragSelection,
QwtPicker.AlwaysOn, self.canvas())
self.zoomer.setRubberBandPen(QPen(Qt.green))
self.zoomer.setTrackerPen(QPen(Qt.red))
self.magnifier = QwtPlotMagnifier(self.canvas())
self.panner = QwtPlotPanner(self.canvas())
self.panner.setMouseButton(Qt.LeftButton, Qt.ControlModifier)
self.connect(self, SIGNAL("legendClicked(QwtPlotItem*)"), self.togglePlotItemVisibility)
def togglePlotItemVisibility(self, plotItem):
"""Toggle the visibility of a plot item"""
plotItem.setVisible(not plotItem.isVisible())
self.replot()
def changePlotWidget(self, library, frame_for_plot):
if library == "Qwt5" and has_qwt:
plotWdg = QwtPlot(frame_for_plot)
sizePolicy = QSizePolicy(QSizePolicy.Expanding,
QSizePolicy.Expanding)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(
plotWdg.sizePolicy().hasHeightForWidth())
plotWdg.setSizePolicy(sizePolicy)
plotWdg.setMinimumSize(QSize(0, 0))
plotWdg.setAutoFillBackground(False)
#Decoration
plotWdg.setCanvasBackground(Qt.white)
plotWdg.plotLayout().setAlignCanvasToScales(True)
zoomer = QwtPlotZoomer(QwtPlot.xBottom, QwtPlot.yLeft,
QwtPicker.DragSelection,
QwtPicker.AlwaysOff, plotWdg.canvas())
zoomer.setRubberBandPen(QPen(Qt.blue))
if platform.system() != "Windows":
# disable picker in Windows due to crashes
picker = QwtPlotPicker(QwtPlot.xBottom, QwtPlot.yLeft,
QwtPicker.NoSelection,
QwtPlotPicker.CrossRubberBand,
QwtPicker.AlwaysOn, plotWdg.canvas())
picker.setTrackerPen(QPen(Qt.green))
#self.dockwidget.qwtPlot.insertLegend(QwtLegend(), QwtPlot.BottomLegend);
grid = Qwt.QwtPlotGrid()
grid.setPen(QPen(QColor('grey'), 0, Qt.DotLine))
grid.attach(plotWdg)
return plotWdg
elif library == "Matplotlib" and has_mpl:
from matplotlib.figure import Figure
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg
fig = Figure((1.0, 1.0),
linewidth=0.0,
subplotpars=matplotlib.figure.SubplotParams(left=0,
bottom=0,
right=1,
top=1,
wspace=0,
hspace=0))
font = {'family': 'arial', 'weight': 'normal', 'size': 12}
matplotlib.rc('font', **font)
rect = fig.patch
rect.set_facecolor((0.9, 0.9, 0.9))
self.subplot = fig.add_axes((0.07, 0.15, 0.92, 0.82))
self.subplot.set_xbound(0, 1000)
self.subplot.set_ybound(0, 1000)
self.manageMatplotlibAxe(self.subplot)
canvas = FigureCanvasQTAgg(fig)
sizePolicy = QSizePolicy(QSizePolicy.Expanding,
QSizePolicy.Expanding)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
canvas.setSizePolicy(sizePolicy)
return canvas
