256 lines
8.5 KiB
Python
256 lines
8.5 KiB
Python
|
from pyqtgraph.Qt import QtGui, QtCore
|
||
|
|
from pyqtgraph.Point import Point
|
||
|
|
from UIGraphicsItem import UIGraphicsItem
|
||
|
|
import pyqtgraph.functions as fn
|
||
|
|
import numpy as np
|
||
|
|
import weakref
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = ['InfiniteLine']
|
||
|
|
class InfiniteLine(UIGraphicsItem):
|
||
|
|
"""
|
||
|
|
Displays a line of infinite length.
|
||
|
|
This line may be dragged to indicate a position in data coordinates.
|
||
|
|
"""
|
||
|
|
|
||
|
|
sigDragged = QtCore.Signal(object)
|
||
|
|
sigPositionChangeFinished = QtCore.Signal(object)
|
||
|
|
sigPositionChanged = QtCore.Signal(object)
|
||
|
|
|
||
|
|
def __init__(self, pos=None, angle=90, pen=None, movable=False, bounds=None):
|
||
|
|
"""
|
||
|
|
Initialization options:
|
||
|
|
pos - Position of the line. This can be a QPointF or a single value for vertical/horizontal lines.
|
||
|
|
angle - Angle of line in degrees. 0 is horizontal, 90 is vertical.
|
||
|
|
pen - Pen to use when drawing line
|
||
|
|
movable - If True, the line can be dragged to a new position by the user
|
||
|
|
bounds - Optional [min, max] bounding values. Bounds are only valid if the line is vertical or horizontal.
|
||
|
|
"""
|
||
|
|
|
||
|
|
UIGraphicsItem.__init__(self)
|
||
|
|
|
||
|
|
if bounds is None: ## allowed value boundaries for orthogonal lines
|
||
|
|
self.maxRange = [None, None]
|
||
|
|
else:
|
||
|
|
self.maxRange = bounds
|
||
|
|
self.moving = False
|
||
|
|
self.setMovable(movable)
|
||
|
|
self.p = [0, 0]
|
||
|
|
self.setAngle(angle)
|
||
|
|
if pos is None:
|
||
|
|
pos = Point(0,0)
|
||
|
|
self.setPos(pos)
|
||
|
|
|
||
|
|
if pen is None:
|
||
|
|
pen = (200, 200, 100)
|
||
|
|
self.setPen(pen)
|
||
|
|
self.currentPen = self.pen
|
||
|
|
#self.setFlag(self.ItemSendsScenePositionChanges)
|
||
|
|
|
||
|
|
def setMovable(self, m):
|
||
|
|
self.movable = m
|
||
|
|
self.setAcceptHoverEvents(m)
|
||
|
|
|
||
|
|
def setBounds(self, bounds):
|
||
|
|
"""Set the (minimum, maximum) allowable values when dragging."""
|
||
|
|
self.maxRange = bounds
|
||
|
|
self.setValue(self.value())
|
||
|
|
|
||
|
|
def setPen(self, pen):
|
||
|
|
self.pen = fn.mkPen(pen)
|
||
|
|
self.currentPen = self.pen
|
||
|
|
self.update()
|
||
|
|
|
||
|
|
def setAngle(self, angle):
|
||
|
|
"""
|
||
|
|
Takes angle argument in degrees.
|
||
|
|
0 is horizontal; 90 is vertical.
|
||
|
|
|
||
|
|
Note that the use of value() and setValue() changes if the line is
|
||
|
|
not vertical or horizontal.
|
||
|
|
"""
|
||
|
|
self.angle = ((angle+45) % 180) - 45 ## -45 <= angle < 135
|
||
|
|
self.resetTransform()
|
||
|
|
self.rotate(self.angle)
|
||
|
|
self.update()
|
||
|
|
|
||
|
|
def setPos(self, pos):
|
||
|
|
if type(pos) in [list, tuple]:
|
||
|
|
newPos = pos
|
||
|
|
elif isinstance(pos, QtCore.QPointF):
|
||
|
|
newPos = [pos.x(), pos.y()]
|
||
|
|
else:
|
||
|
|
if self.angle == 90:
|
||
|
|
newPos = [pos, 0]
|
||
|
|
elif self.angle == 0:
|
||
|
|
newPos = [0, pos]
|
||
|
|
else:
|
||
|
|
raise Exception("Must specify 2D coordinate for non-orthogonal lines.")
|
||
|
|
|
||
|
|
## check bounds (only works for orthogonal lines)
|
||
|
|
if self.angle == 90:
|
||
|
|
if self.maxRange[0] is not None:
|
||
|
|
newPos[0] = max(newPos[0], self.maxRange[0])
|
||
|
|
if self.maxRange[1] is not None:
|
||
|
|
newPos[0] = min(newPos[0], self.maxRange[1])
|
||
|
|
elif self.angle == 0:
|
||
|
|
if self.maxRange[0] is not None:
|
||
|
|
newPos[1] = max(newPos[1], self.maxRange[0])
|
||
|
|
if self.maxRange[1] is not None:
|
||
|
|
newPos[1] = min(newPos[1], self.maxRange[1])
|
||
|
|
|
||
|
|
if self.p != newPos:
|
||
|
|
self.p = newPos
|
||
|
|
UIGraphicsItem.setPos(self, Point(self.p))
|
||
|
|
self.update()
|
||
|
|
self.sigPositionChanged.emit(self)
|
||
|
|
|
||
|
|
def getXPos(self):
|
||
|
|
return self.p[0]
|
||
|
|
|
||
|
|
def getYPos(self):
|
||
|
|
return self.p[1]
|
||
|
|
|
||
|
|
def getPos(self):
|
||
|
|
return self.p
|
||
|
|
|
||
|
|
def value(self):
|
||
|
|
if self.angle%180 == 0:
|
||
|
|
return self.getYPos()
|
||
|
|
elif self.angle%180 == 90:
|
||
|
|
return self.getXPos()
|
||
|
|
else:
|
||
|
|
return self.getPos()
|
||
|
|
|
||
|
|
def setValue(self, v):
|
||
|
|
self.setPos(v)
|
||
|
|
|
||
|
|
## broken in 4.7
|
||
|
|
#def itemChange(self, change, val):
|
||
|
|
#if change in [self.ItemScenePositionHasChanged, self.ItemSceneHasChanged]:
|
||
|
|
#self.updateLine()
|
||
|
|
#print "update", change
|
||
|
|
#print self.getBoundingParents()
|
||
|
|
#else:
|
||
|
|
#print "ignore", change
|
||
|
|
#return GraphicsObject.itemChange(self, change, val)
|
||
|
|
|
||
|
|
def boundingRect(self):
|
||
|
|
br = UIGraphicsItem.boundingRect(self)
|
||
|
|
|
||
|
|
## add a 4-pixel radius around the line for mouse interaction.
|
||
|
|
|
||
|
|
#print "line bounds:", self, br
|
||
|
|
dt = self.deviceTransform()
|
||
|
|
if dt is None:
|
||
|
|
return QtCore.QRectF()
|
||
|
|
lineDir = Point(dt.map(Point(1, 0)) - dt.map(Point(0,0))) ## direction of line in pixel-space
|
||
|
|
orthoDir = Point(lineDir[1], -lineDir[0]) ## orthogonal to line in pixel-space
|
||
|
|
try:
|
||
|
|
norm = orthoDir.norm() ## direction of one pixel orthogonal to line
|
||
|
|
except ZeroDivisionError:
|
||
|
|
return br
|
||
|
|
|
||
|
|
dti = dt.inverted()[0]
|
||
|
|
px = Point(dti.map(norm)-dti.map(Point(0,0))) ## orthogonal pixel mapped back to item coords
|
||
|
|
px = px[1] ## project to y-direction
|
||
|
|
|
||
|
|
br.setBottom(-px*4)
|
||
|
|
br.setTop(px*4)
|
||
|
|
return br.normalized()
|
||
|
|
|
||
|
|
def paint(self, p, *args):
|
||
|
|
UIGraphicsItem.paint(self, p, *args)
|
||
|
|
br = self.boundingRect()
|
||
|
|
p.setPen(self.currentPen)
|
||
|
|
p.drawLine(Point(br.right(), 0), Point(br.left(), 0))
|
||
|
|
#p.drawRect(self.boundingRect())
|
||
|
|
|
||
|
|
def dataBounds(self, axis, frac=1.0):
|
||
|
|
if axis == 0:
|
||
|
|
return None ## x axis should never be auto-scaled
|
||
|
|
else:
|
||
|
|
return (0,0)
|
||
|
|
|
||
|
|
#def mousePressEvent(self, ev):
|
||
|
|
#if self.movable and ev.button() == QtCore.Qt.LeftButton:
|
||
|
|
#ev.accept()
|
||
|
|
#self.pressDelta = self.mapToParent(ev.pos()) - QtCore.QPointF(*self.p)
|
||
|
|
#else:
|
||
|
|
#ev.ignore()
|
||
|
|
|
||
|
|
#def mouseMoveEvent(self, ev):
|
||
|
|
#self.setPos(self.mapToParent(ev.pos()) - self.pressDelta)
|
||
|
|
##self.emit(QtCore.SIGNAL('dragged'), self)
|
||
|
|
#self.sigDragged.emit(self)
|
||
|
|
#self.hasMoved = True
|
||
|
|
|
||
|
|
#def mouseReleaseEvent(self, ev):
|
||
|
|
#if self.hasMoved and ev.button() == QtCore.Qt.LeftButton:
|
||
|
|
#self.hasMoved = False
|
||
|
|
##self.emit(QtCore.SIGNAL('positionChangeFinished'), self)
|
||
|
|
#self.sigPositionChangeFinished.emit(self)
|
||
|
|
|
||
|
|
def mouseDragEvent(self, ev):
|
||
|
|
if self.movable and ev.button() == QtCore.Qt.LeftButton:
|
||
|
|
if ev.isStart():
|
||
|
|
self.moving = True
|
||
|
|
self.cursorOffset = self.pos() - self.mapToParent(ev.buttonDownPos())
|
||
|
|
self.startPosition = self.pos()
|
||
|
|
ev.accept()
|
||
|
|
|
||
|
|
if not self.moving:
|
||
|
|
return
|
||
|
|
|
||
|
|
#pressDelta = self.mapToParent(ev.buttonDownPos()) - Point(self.p)
|
||
|
|
self.setPos(self.cursorOffset + self.mapToParent(ev.pos()))
|
||
|
|
self.sigDragged.emit(self)
|
||
|
|
if ev.isFinish():
|
||
|
|
self.moving = False
|
||
|
|
self.sigPositionChangeFinished.emit(self)
|
||
|
|
#else:
|
||
|
|
#print ev
|
||
|
|
|
||
|
|
|
||
|
|
def mouseClickEvent(self, ev):
|
||
|
|
if self.moving and ev.button() == QtCore.Qt.RightButton:
|
||
|
|
ev.accept()
|
||
|
|
self.setPos(self.startPosition)
|
||
|
|
self.moving = False
|
||
|
|
self.sigDragged.emit(self)
|
||
|
|
self.sigPositionChangeFinished.emit(self)
|
||
|
|
|
||
|
|
def hoverEvent(self, ev):
|
||
|
|
if (not ev.isExit()) and ev.acceptDrags(QtCore.Qt.LeftButton):
|
||
|
|
self.currentPen = fn.mkPen(255, 0,0)
|
||
|
|
else:
|
||
|
|
self.currentPen = self.pen
|
||
|
|
self.update()
|
||
|
|
|
||
|
|
#def hoverEnterEvent(self, ev):
|
||
|
|
#print "line hover enter"
|
||
|
|
#ev.ignore()
|
||
|
|
#self.updateHoverPen()
|
||
|
|
|
||
|
|
#def hoverMoveEvent(self, ev):
|
||
|
|
#print "line hover move"
|
||
|
|
#ev.ignore()
|
||
|
|
#self.updateHoverPen()
|
||
|
|
|
||
|
|
#def hoverLeaveEvent(self, ev):
|
||
|
|
#print "line hover leave"
|
||
|
|
#ev.ignore()
|
||
|
|
#self.updateHoverPen(False)
|
||
|
|
|
||
|
|
#def updateHoverPen(self, hover=None):
|
||
|
|
#if hover is None:
|
||
|
|
#scene = self.scene()
|
||
|
|
#hover = scene.claimEvent(self, QtCore.Qt.LeftButton, scene.Drag)
|
||
|
|
|
||
|
|
#if hover:
|
||
|
|
#self.currentPen = fn.mkPen(255, 0,0)
|
||
|
|
#else:
|
||
|
|
#self.currentPen = self.pen
|
||
|
|
#self.update()
|
||
|
|
|