import math from PySide6.QtCore import Qt, Signal, QPointF, QObject from PySide6.QtGui import QPen, QBrush, QColor, QPainterPath from PySide6.QtWidgets import QGraphicsEllipseItem, QGraphicsPathItem, \ QGraphicsTextItem, QGraphicsItem,QGraphicsLineItem class CustomLineItem(QGraphicsPathItem, QObject): pressMouse = Signal(object, int) # 定义鼠标按下信号 def __init__(self, sx, sy, ex, ey, edge_id, is_path=False, seq=2, parent=None, **kwargs): QGraphicsPathItem.__init__(self, parent) QObject.__init__(self, **kwargs) self.x = (ex + sx) // 2 self.y = (ey + sy) // 2 self.id = edge_id self.seq = seq if ex - sx != 0: m = (ey - sy) / (ex - sx) # 求得线段的斜率 self.angle_rad = math.atan(m) # 求弧度 self.angle = math.degrees(self.angle_rad) # 计算度数 else: self.angle_rad = math.pi / 2 self.angle = 90 self.off_x_center = math.cos(self.angle_rad + math.pi / 2) * 30 self.off_y_center = math.sin(self.angle_rad + math.pi / 2) * 30 self.pen_line = QPen(QColor(200, 200, 200, 100)) self.pen_line.setWidth(3) self.selected_pen_line = QPen(QColor(100, 255, 100, 255)) self.selected_pen_line.setWidth(3) if is_path: # 如果是路径 self.pen_line = QPen(QColor(0, 200, 200, 100)) self.pen_line.setWidth(5) # 设置画笔的宽度为5 # 创建画笔路径 path = QPainterPath() if self.seq == 0: path.moveTo(sx, sy) # 设置起点 path.lineTo(ex, ey) # 设置终点 elif self.seq == 1: path.moveTo(sx, sy) # 设置路径的起点 path.lineTo(self.x+self.off_x_center, self.y+self.off_y_center) # 设置路径的终点 path.lineTo(ex, ey) # 设置终点 elif self.seq == 2: path.moveTo(sx, sy) # 设置路径的起点 path.lineTo(self.x-self.off_x_center, self.y-self.off_y_center) # 设置路径的终点 path.lineTo(ex, ey) # 设置终点 # 将画笔路径设置给GraphicsPathItem self.setPath(path) # 设置路径 self.setPen(self.pen_line) # 设置画笔 self.setZValue(-1) # 设置z轴坐标 self.setFlag(QGraphicsItem.ItemIsSelectable, True) # 将连线设置为可选 # self.set_text("ID: " + str(self.id)) def set_text(self, text): # 清除之前的文本 if hasattr(self, 'text_item'): self.scene().removeItem(self.text_item) del self.text_item # 添加文本 self.text_item = QGraphicsTextItem(text, parent=self) # 修改字体属性 font = self.text_item.font() font.setPointSize(12) # 设置字体大小 self.text_item.setFont(font) # 计算理想的文本位置 tw = self.text_item.boundingRect().width()/2 th = self.text_item.boundingRect().height()/2 self.text_item.setRotation(self.angle) # 调整显示角度 # 调整路径信息显示方向和位置 ly = tw*math.sin(self.angle_rad) lx = tw*math.cos(self.angle_rad) my = th*math.sin(self.angle_rad+math.pi/2) mx = th*math.cos(self.angle_rad+math.pi/2) if self.seq == 0: self.text_item.setPos(self.x-lx-mx ,self.y-ly-my) elif self.seq == 1: self.text_item.setPos(self.x+self.off_x_center-lx-mx, self.y+self.off_y_center-ly-my) elif self.seq == 2: self.text_item.setPos(self.x-self.off_x_center-lx-mx, self.y-self.off_y_center-ly-my) def mousePressEvent(self, event): print("连接线") if event.button() == Qt.LeftButton: # 如果是鼠标左键按下 self.pressMouse.emit(event, self.id) # 鼠标左键按下时发送鼠标坐标信号和标签 for item in self.scene().items(): # 遍历场景中的所有图形项 if isinstance(item, (CustomLineItem, CustomEllipseItem)) and item.isSelected(): # 如果是线或者圆形并且被选中 item.setSelected(False) # 取消选中 self.setSelected(True) # 选中当前图形项 def itemChange(self, change, value): if change == QGraphicsItem.ItemSelectedChange: if value: self.setPen(self.selected_pen_line) else: self.setPen(self.pen_line) return super().itemChange(change, value) # 取消选中时恢复以前的状态 class CustomEllipseItem(QGraphicsEllipseItem, QObject): # 圆形对象 pressMouse = Signal(object, int) # 定义鼠标按下信号 releaseMouse = Signal(object) # 定义鼠标释放信号 collideMouse = Signal(int, int) # 定义碰撞信号发送自身节点的id和连线碰撞的节点id def __init__(self, x, y, id, text, is_must=False, radius=18, parent=None, **kwargs): ''' :param radius: 半径 :param color: 颜色带透明度 ''' super().__init__(x - radius, y - radius, radius * 2, radius * 2, parent) QObject.__init__(self, **kwargs) # 为了获得自定义信号和槽的能力 pen_circle = QPen(QColor(0, 0, 0, 255)) # 黑色 pen_circle.setWidth(0) # 设置画笔的宽度为0 self.brush_circle = QBrush(QColor(255, 255, 0, 255)) # 创建一个黄色色且透明度为255的刷子 if is_must: # 必经点设置为红色 self.brush_circle = QBrush(QColor(255, 100, 100, 255)) # 创建一个红色且透明度为100的刷子 self.setPen(pen_circle) self.setBrush(self.brush_circle) self.x = x self.y = y self.text = str(text) self.id = id # self.setFlag(QGraphicsEllipseItem.ItemIsMovable, True) self.setFlag(QGraphicsEllipseItem.ItemIsSelectable, True) # 设置为可选 # 设置鼠标悬停时的光标为抓手 self.setAcceptHoverEvents(True) # 设置接受鼠标悬停事件 self.set_text(self.text) # 设置文本 self.mouse_pressed = False # 鼠标按下标志 self.line_item = None # 初始化线 def set_text(self, text): # 清除之前的文本 if hasattr(self, 'text_item'): self.scene().removeItem(self.text_item) del self.text_item self.text_item = QGraphicsTextItem(text, parent=self) # 添加文本 font = self.text_item.font() # 修改字体属性 font.setPointSize(20) # 设置字体大小 self.text_item.setFont(font) # 设置格式 self.text_item.setPos(self.x - self.text_item.boundingRect().width() / 2, # 设置位置 self.y - self.text_item.boundingRect().height() / 2) def __str__(self): return f"圆形对象：x={self.x}, y={self.y}" def mousePressEvent(self, event): # 鼠标按下执行 if event.button() == Qt.LeftButton: self.pressMouse.emit(event, self.id) # 鼠标左键按下时发送鼠标坐标信号和标签 self.mouse_pressed = True for item in self.scene().items(): if isinstance(item, (CustomLineItem, CustomEllipseItem)) and item.isSelected(): item.setSelected(False) self.setSelected(True) self.setSelected(True) def mouseMoveEvent(self, event): if self.mouse_pressed: # 鼠标按住的情况下更新直线 mouse_x = event.scenePos().x() mouse_y = event.scenePos().y() if self.line_item is not None: # 如果存在 self.scene().removeItem(self.line_item) # 删除线 pen_circle = QPen(QColor(200, 200, 200, 255)) pen_circle.setWidth(2) self.line_item = QGraphicsLineItem(self.x, self.y, mouse_x, mouse_y) # print(mouse_x, mouse_y) # 鼠标移动时的坐标 self.line_item.setPen(pen_circle) self.line_item.setZValue(-1) self.scene().addItem(self.line_item) # 重新画线 # 获取线的端点坐标 # line_start = QPointF(self.x, self.y) line_end = QPointF(mouse_x, mouse_y) # 检查线的端点是否与其他图形项碰撞 for item in self.scene().items(): if isinstance(item, CustomEllipseItem) and item != self: if item.contains(line_end): # 如果线的端点与其他图形项碰撞 # print(f"Line endpoint collides with {item}") self.collideMouse.emit(self.id, item.id) def mouseReleaseEvent(self, event): # 鼠标释放执行 if event.button() == Qt.LeftButton: self.mouse_pressed = False # 鼠标松开时删除直线 if self.line_item is not None: self.scene().removeItem(self.line_item) self.line_item = None self.releaseMouse.emit(event) def hoverEnterEvent(self, event): # 鼠标悬停时设置光标为抓手 self.setCursor(Qt.PointingHandCursor) # 鼠标悬停时显示带背景的文字 info_text = QGraphicsTextItem(f"{self.id}-({self.x},{self.y})", self) info_text.setPos(self.x - 20, self.y + 20) self.info_text = info_text def hoverLeaveEvent(self, event): # 鼠标离开时恢复光标 self.unsetCursor() # 鼠标离开时移除信息 if hasattr(self, 'info_text'): self.scene().removeItem(self.info_text) del self.info_text def contextMenuEvent(self, event): # 在右键点击时选中图形项 self.setSelected(True) event.accept() def itemChange(self, change, value): if change == QGraphicsItem.ItemSelectedChange: if value: self.setBrush(QBrush(QColor(100, 255, 100, 255))) # 选中时设置为绿色 else: self.setBrush(self.brush_circle) # 取消选中时恢复原来的颜色 return super().itemChange(change, value)