from GraphicsObject import * import pyqtgraph.functions as fn from pyqtgraph.Qt import QtGui, QtCore class IsocurveItem(GraphicsObject): """ **Bases:** :class:`GraphicsObject ` Item displaying an isocurve of a 2D array.To align this item correctly with an ImageItem,call isocurve.setParentItem(image) """ def __init__(self, data=None, level=0, pen='w'): """ Create a new isocurve item. ============= =============================================================== **Arguments** data A 2-dimensional ndarray. Can be initialized as None, and set later using :func:`setData ` level The cutoff value at which to draw the isocurve. pen The color of the curve item. Can be anything valid for :func:`mkPen ` ============= =============================================================== """ GraphicsObject.__init__(self) self.level = level self.data = None self.path = None self.setPen(pen) self.setData(data, level) #if data is not None and level is not None: #self.updateLines(data, level) def setData(self, data, level=None): """ Set the data/image to draw isocurves for. ============= ================================================================ **Arguments** data A 2-dimensional ndarray. level The cutoff value at which to draw the curve. If level is not specified, the previous level is used. ============= ================================================================ """ if level is None: level = self.level self.level = level self.data = data self.path = None self.prepareGeometryChange() self.update() def setLevel(self, level): """Set the level at which the isocurve is drawn.""" self.level = level self.path = None self.update() def setPen(self, *args, **kwargs): """Set the pen used to draw the isocurve. Arguments can be any that are valid for :func:`mkPen `""" self.pen = fn.mkPen(*args, **kwargs) self.update() def updateLines(self, data, level): ##print "data:", data ##print "level", level #lines = fn.isocurve(data, level) ##print len(lines) #self.path = QtGui.QPainterPath() #for line in lines: #self.path.moveTo(*line[0]) #self.path.lineTo(*line[1]) #self.update() self.setData(data, level) def boundingRect(self): if self.path is None: return QtCore.QRectF() return self.path.boundingRect() def generatePath(self): self.path = QtGui.QPainterPath() if self.data is None: return lines = fn.isocurve(self.data, self.level) for line in lines: self.path.moveTo(*line[0]) self.path.lineTo(*line[1]) def paint(self, p, *args): if self.path is None: self.generatePath() p.setPen(self.pen) p.drawPath(self.path)