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)