def test_controlpointline(self):
control = ControlPointLine()
line = QGraphicsLineItem(10, 10, 200, 200)
self.scene.addItem(line)
self.scene.addItem(control)
control.setLine(line.line())
control.setFocus()
control.lineChanged.connect(line.setLine)
control.setLine(QLineF(30, 30, 180, 180))
self.assertEqual(control.line(), line.line())
self.assertEqual(line.line(), QLineF(30, 30, 180, 180))
control.lineEdited.connect(line.setLine)
self.view.show()
self.app.exec_()
self.assertEqual(control.line(), line.line())
class AxisItem(pg.GraphicsObject):
def __init__(self, parent=None, line=None, label=None, **kwargs):
super().__init__(parent, **kwargs)
self.setFlag(pg.GraphicsObject.ItemHasNoContents)
if line is None:
line = QLineF(0, 0, 1, 0)
self._spine = QGraphicsLineItem(line, self)
angle = QLineF(0, 0, 1, 0).angleTo(line)
angle = (180 - angle) % 360
dx = line.x2() - line.x1()
dy = line.y2() - line.y1()
rad = numpy.arctan2(dy, dx)
angle = (rad * 180 / numpy.pi) % 360
self._arrow = pg.ArrowItem(parent=self, angle=180 - angle)
self._arrow.setPos(self._spine.line().p2())
self._label = pg.TextItem(text=label, color=(10, 10, 10))
self._label.setParentItem(self)
self._label.setPos(self._spine.line().p2())
def setLabel(self, label):
if label != self._label:
self._label = label
self._label.setText(label)
def setPen(self, pen):
self._spine.setPen(pen)
def paint(self, painter, option, widget):
pass
def boundingRect(self):
return QRectF()
def viewTransformChanged(self):
self.__updateLabelPos()
def __updateLabelPos(self):
T = self.viewTransform()
if T is not None:
Tinv, ok = T.inverted()
else:
Tinv, ok = None, False
if not ok:
T = Tinv = QtGui.QTransform()
# map the axis spine to viewbox coord. system
viewbox_line = Tinv.map(self._spine.line())
angle = viewbox_line.angle()
# note in Qt the y axis is inverted (90 degree angle 'points' down)
left_quad = 270 <= angle <= 360 or -0.0 <= angle < 90
# position the text label along the viewbox_line
label_pos = viewbox_line.pointAt(0.90)
if left_quad:
anchor = (0.5, -0.1)
else:
anchor = (0.5, 1.1)
pos = T.map(label_pos)
self._label.setPos(pos)
self._label.anchor = pg.Point(*anchor)
self._label.updateText()
self._label.setRotation(angle if left_quad else angle - 180)
class AxisItem(pg.GraphicsObject):
def __init__(self, parent=None, line=None, label=None, **kwargs):
super().__init__(parent, **kwargs)
self.setFlag(pg.GraphicsObject.ItemHasNoContents)
if line is None:
line = QLineF(0, 0, 1, 0)
self._spine = QGraphicsLineItem(line, self)
angle = QLineF(0, 0, 1, 0).angleTo(line)
angle = (180 - angle) % 360
dx = line.x2() - line.x1()
dy = line.y2() - line.y1()
rad = numpy.arctan2(dy, dx)
angle = (rad * 180 / numpy.pi) % 360
self._arrow = pg.ArrowItem(parent=self, angle=180 - angle)
self._arrow.setPos(self._spine.line().p2())
self._label = pg.TextItem(text=label, color=(10, 10, 10))
self._label.setParentItem(self)
self._label.setPos(self._spine.line().p2())
def setLabel(self, label):
if label != self._label:
self._label = label
self._label.setText(label)
def setPen(self, pen):
self._spine.setPen(pen)
def paint(self, painter, option, widget):
pass
def boundingRect(self):
return QRectF()
def viewTransformChanged(self):
self.__updateLabelPos()
def __updateLabelPos(self):
T = self.viewTransform()
if T is not None:
Tinv, ok = T.inverted()
else:
Tinv, ok = None, False
if not ok:
T = Tinv = QtGui.QTransform()
# map the axis spine to viewbox coord. system
viewbox_line = Tinv.map(self._spine.line())
angle = viewbox_line.angle()
# note in Qt the y axis is inverted (90 degree angle 'points' down)
left_quad = 270 <= angle <= 360 or -0.0 <= angle < 90
# position the text label along the viewbox_line
label_pos = viewbox_line.pointAt(0.90)
if left_quad:
anchor = (0.5, -0.1)
else:
anchor = (0.5, 1.1)
pos = T.map(label_pos)
self._label.setPos(pos)
self._label.anchor = pg.Point(*anchor)
self._label.updateText()
self._label.setRotation(angle if left_quad else angle - 180)
