#!/usr/bin/env python
import sys
from PyQt4 import Qt
import PyQt4.Qwt5 as Qwt
from PyQt4.Qwt5.anynumpy import *
class ErrorBarPlotCurve(Qwt.QwtPlotCurve):
def __init__(self,
x = [], y = [], dx = None, dy = None,
curvePen = Qt.QPen(Qt.Qt.NoPen),
curveStyle = Qwt.QwtPlotCurve.Lines,
curveSymbol = Qwt.QwtSymbol(),
errorPen = Qt.QPen(Qt.Qt.NoPen),
errorCap = 0,
errorOnTop = False,
):
"""A curve of x versus y data with error bars in dx and dy.
Horizontal error bars are plotted if dx is not None.
Vertical error bars are plotted if dy is not None.
x and y must be sequences with a shape (N,) and dx and dy must be
sequences (if not None) with a shape (), (N,), or (2, N):
- if dx or dy has a shape () or (N,), the error bars are given by
(x-dx, x+dx) or (y-dy, y+dy),
- if dx or dy has a shape (2, N), the error bars are given by
(x-dx[0], x+dx[1]) or (y-dy[0], y+dy[1]).
curvePen is the pen used to plot the curve
curveStyle is the style used to plot the curve
curveSymbol is the symbol used to plot the symbols
errorPen is the pen used to plot the error bars
errorCap is the size of the error bar caps
errorOnTop is a boolean:
- if True, plot the error bars on top of the curve,
- if False, plot the curve on top of the error bars.
"""
Qwt.QwtPlotCurve.__init__(self)
self.setData(x, y, dx, dy)
self.setPen(curvePen)
self.setStyle(curveStyle)
self.setSymbol(curveSymbol)
self.errorPen = errorPen
self.errorCap = errorCap
self.errorOnTop = errorOnTop
# __init__()
def setData(self, x, y, dx = None, dy = None):
"""Set x versus y data with error bars in dx and dy.
Horizontal error bars are plotted if dx is not None.
Vertical error bars are plotted if dy is not None.
x and y must be sequences with a shape (N,) and dx and dy must be
sequences (if not None) with a shape (), (N,), or (2, N):
- if dx or dy has a shape () or (N,), the error bars are given by
(x-dx, x+dx) or (y-dy, y+dy),
- if dx or dy has a shape (2, N), the error bars are given by
(x-dx[0], x+dx[1]) or (y-dy[0], y+dy[1]).
"""
self.__x = asarray(x, Float)
if len(self.__x.shape) != 1:
raise RuntimeError, 'len(asarray(x).shape) != 1'
self.__y = asarray(y, Float)
if len(self.__y.shape) != 1:
raise RuntimeError, 'len(asarray(y).shape) != 1'
if len(self.__x) != len(self.__y):
raise RuntimeError, 'len(asarray(x)) != len(asarray(y))'
if dx is None:
self.__dx = None
else:
self.__dx = asarray(dx, Float)
if len(self.__dx.shape) not in [0, 1, 2]:
raise RuntimeError, 'len(asarray(dx).shape) not in [0, 1, 2]'
if dy is None:
self.__dy = dy
else:
self.__dy = asarray(dy, Float)
if len(self.__dy.shape) not in [0, 1, 2]:
raise RuntimeError, 'len(asarray(dy).shape) not in [0, 1, 2]'
Qwt.QwtPlotCurve.setData(self, self.__x, self.__y)
# setData()
def boundingRect(self):
"""Return the bounding rectangle of the data, error bars included.
"""
if self.__dx is None:
xmin = min(self.__x)
xmax = max(self.__x)
elif len(self.__dx.shape) in [0, 1]:
xmin = min(self.__x - self.__dx)
xmax = max(self.__x + self.__dx)
else:
xmin = min(self.__x - self.__dx[0])
xmax = max(self.__x + self.__dx[1])
if self.__dy is None:
ymin = min(self.__y)
ymax = max(self.__y)
elif len(self.__dy.shape) in [0, 1]:
ymin = min(self.__y - self.__dy)
ymax = max(self.__y + self.__dy)
else:
ymin = min(self.__y - self.__dy[0])
ymax = max(self.__y + self.__dy[1])
return Qt.QRectF(xmin, ymin, xmax-xmin, ymax-ymin)
# boundingRect()
def drawFromTo(self, painter, xMap, yMap, first, last = -1):
"""Draw an interval of the curve, including the error bars
painter is the QPainter used to draw the curve
xMap is the Qwt.QwtDiMap used to map x-values to pixels
yMap is the Qwt.QwtDiMap used to map y-values to pixels
first is the index of the first data point to draw
last is the index of the last data point to draw. If last < 0, last
is transformed to index the last data point
"""
if last < 0:
last = self.dataSize() - 1
if self.errorOnTop:
Qwt.QwtPlotCurve.drawFromTo(self, painter, xMap, yMap, first, last)
# draw the error bars
painter.save()
painter.setPen(self.errorPen)
# draw the error bars with caps in the x direction
if self.__dx is not None:
# draw the bars
if len(self.__dx.shape) in [0, 1]:
xmin = (self.__x - self.__dx)
xmax = (self.__x + self.__dx)
else:
xmin = (self.__x - self.__dx[0])
xmax = (self.__x + self.__dx[1])
y = self.__y
n, i = len(y), 0
lines = []
while i < n:
yi = yMap.transform(y[i])
lines.append(Qt.QLine(xMap.transform(xmin[i]), yi,
xMap.transform(xmax[i]), yi))
i += 1
painter.drawLines(lines)
if self.errorCap > 0:
# draw the caps
cap = self.errorCap/2
n, i, = len(y), 0
lines = []
while i < n:
yi = yMap.transform(y[i])
lines.append(
Qt.QLine(xMap.transform(xmin[i]), yi - cap,
xMap.transform(xmin[i]), yi + cap))
lines.append(
Qt.QLine(xMap.transform(xmax[i]), yi - cap,
xMap.transform(xmax[i]), yi + cap))
i += 1
painter.drawLines(lines)
# draw the error bars with caps in the y direction
if self.__dy is not None:
# draw the bars
if len(self.__dy.shape) in [0, 1]:
ymin = (self.__y - self.__dy)
ymax = (self.__y + self.__dy)
else:
ymin = (self.__y - self.__dy[0])
ymax = (self.__y + self.__dy[1])
x = self.__x
n, i, = len(x), 0
lines = []
while i < n:
xi = xMap.transform(x[i])
lines.append(
Qt.QLine(xi, yMap.transform(ymin[i]),
xi, yMap.transform(ymax[i])))
i += 1
painter.drawLines(lines)
# draw the caps
if self.errorCap > 0:
cap = self.errorCap/2
n, i, j = len(x), 0, 0
lines = []
while i < n:
xi = xMap.transform(x[i])
lines.append(
Qt.QLine(xi - cap, yMap.transform(ymin[i]),
xi + cap, yMap.transform(ymin[i])))
lines.append(
Qt.QLine(xi - cap, yMap.transform(ymax[i]),
xi + cap, yMap.transform(ymax[i])))
i += 1
painter.drawLines(lines)
painter.restore()
if not self.errorOnTop:
Qwt.QwtPlotCurve.drawFromTo(self, painter, xMap, yMap, first, last)
# drawFromTo()
# class ErrorBarPlotCurve
def make():
# create a plot with a white canvas
demo = Qwt.QwtPlot(Qwt.QwtText("Errorbar Demonstation"))
demo.setCanvasBackground(Qt.Qt.white)
demo.plotLayout().setAlignCanvasToScales(True)
grid = Qwt.QwtPlotGrid()
grid.attach(demo)
grid.setPen(Qt.QPen(Qt.Qt.black, 0, Qt.Qt.DotLine))
# calculate data and errors for a curve with error bars
x = arange(0, 10.1, 0.5, Float)
y = sin(x)
dy = 0.2 * abs(y)
# dy = (0.15 * abs(y), 0.25 * abs(y)) # uncomment for asymmetric error bars
dx = 0.2 # all error bars the same size
errorOnTop = False # uncomment to draw the curve on top of the error bars
# errorOnTop = True # uncomment to draw the error bars on top of the curve
curve = ErrorBarPlotCurve(
x = x,
y = y,
dx = dx,
dy = dy,
curvePen = Qt.QPen(Qt.Qt.black, 2),
curveSymbol = Qwt.QwtSymbol(Qwt.QwtSymbol.Ellipse,
Qt.QBrush(Qt.Qt.red),
Qt.QPen(Qt.Qt.black, 2),
Qt.QSize(9, 9)),
errorPen = Qt.QPen(Qt.Qt.blue, 2),
errorCap = 10,
errorOnTop = errorOnTop,
)
curve.attach(demo)
demo.resize(400, 300)
demo.show()
return demo
# make()
def main(args):
app = Qt.QApplication(args)
demo = make()
zoomer = Qwt.QwtPlotZoomer(Qwt.QwtPlot.xBottom,
Qwt.QwtPlot.yLeft,
Qwt.QwtPicker.DragSelection,
Qwt.QwtPicker.AlwaysOff,
demo.canvas())
zoomer.setRubberBandPen(Qt.QPen(Qt.Qt.green))
picker = Qwt.QwtPlotPicker(Qwt.QwtPlot.xBottom,
Qwt.QwtPlot.yLeft,
Qwt.QwtPicker.NoSelection,
Qwt.QwtPlotPicker.CrossRubberBand,
Qwt.QwtPicker.AlwaysOn,
demo.canvas())
picker.setTrackerPen(Qt.QPen(Qt.Qt.red))
sys.exit(app.exec_())
# main()
# Admire!
if __name__ == '__main__':
if 'settracemask' in sys.argv:
# for debugging, requires: python configure.py --trace ...
import sip
sip.settracemask(0x3f)
main(sys.argv)
# Local Variables: ***
# mode: python ***
# End: ***
|