QGraphicsItem 圆角线段

要在 QGraphicsItem 上绘制圆角线段，您需要实现 QGraphicsItem 的 paint() 函数并在其中使用 QPainter 绘制线段。

以下是一个简单的示例：

from PyQt5.QtCore import QPointF, QRectF
from PyQt5.QtGui import QBrush, QColor, QPainter, QPen
from PyQt5.QtWidgets import QGraphicsItem


class RoundedLineItem(QGraphicsItem):

    def __init__(self, start: QPointF, end: QPointF, radius: float):
        super().__init__()

        self._start = start
        self._end = end
        self._radius = radius

        self.setFlag(QGraphicsItem.ItemIsSelectable)

    def boundingRect(self) -> QRectF:
        pen_width = self.pen().widthF()
        extra = (pen_width + self._radius) / 2.0
        return QRectF(self._start, self._end).normalized().adjusted(-extra, -extra, extra, extra)

    def paint(self, painter: QPainter, option, widget=None):
        pen = self.pen()
        brush = self.brush()

        # 计算圆角线段的中心点和长度
        line_vec = self._end - self._start
        length = line_vec.length()
        center = self._start + line_vec / 2.0

        # 计算圆角线段两端的点
        start_cap = self._start + QPointF(self._radius / 2.0, 0)
        end_cap = self._end - QPointF(self._radius / 2.0, 0)

        # 绘制线段
        painter.setPen(pen)
        painter.setBrush(brush)
        painter.drawChord(QRectF(center.x() - length / 2.0, center.y() - self._radius / 2.0, length, self._radius), 0, 16 * 360)
        painter.drawLine(start_cap, end_cap)

    def shape(self) -> QPainterPath:
        path = super().shape()

        # 添加圆角线段的路径
        line_path = QPainterPath()
        line_path.moveTo(self._start)
        line_path.lineTo(self._end)
        path.addPath(line_path)

        return path

    def setRadius(self, radius: float):
        self._radius = radius
        self.update()

    def setPen(self, pen: QPen):
        super().setPen(pen)
        self.update()

    def setBrush(self, brush: QBrush):
        super().setBrush(brush)
        self.update()

在这个例子中，我们使用 QPainter 的 drawChord() 函数绘制了一个圆弧来实现圆角线段的效果。注意圆弧的起始角度和结束角度都是 0 和 16 * 360，这意味着它是一个完整的圆弧。

为了使圆角线段能够被正确地选择和拖动，我们还重写了 shape() 函数，并将圆角线段的路径添加到 QGraphicsItem 的原始路径中。

您可以像这样使用 RoundedLineItem：

from PyQt5.QtCore import QPointF
from PyQt5.QtGui import QPen
from PyQt5.QtWidgets import QGraphicsScene, QGraphicsView, QApplication

if __name__ == '__main__':
    app = QApplication([])

    scene = QGraphicsScene()
    view = QGraphicsView(scene)

    line_item = RoundedLineItem(QPointF(0, 0), QPointF(100, 100), 10)
    line_item.setPen(QPen(Qt.red, 3))

    scene.addItem(line_item)
    view.show()

    app.exec_()

这将在 QGraphicsView 中显示一条从 (0, 0) 到 (100, 100) 的红色圆角线段。