chartflow 0.1__py3-none-any.whl

chartflow/node/_node_markers.py

@@ -0,0 +1,648 @@
+"""
+节点标记辅助类模块
+
+定义节点上附着的浮动标记/按钮辅助类:
+  - HoverButton: 左上角编辑按钮(📖)
+  - StartMarker: 右上角起点标记(🔷)
+  - SingleMarker: 左下角执行状态标记(⚡)
+  - CustomMarker: 右下角用户自定义标记(★)
+
+这些类仅依赖 Qt 与模块常量, 不依赖 NodeItem, 故独立存放以打破循环导入。
+node_item.py 通过 re-export 保持 ``from chartflow.node.node_item import HoverButton``
+等历史导入路径不变。
+"""
+
+from typing import Optional
+
+from PyQt6.QtCore import Qt, QPointF, QRectF, QObject, pyqtProperty, pyqtSignal
+from PyQt6.QtGui import QBrush, QPen, QColor
+from PyQt6.QtWidgets import (
+    QGraphicsItem, QGraphicsEllipseItem, QGraphicsTextItem,
+)
+
+from chartflow.node._node_base import (
+    Z_VALUE_BUTTON,
+    Z_VALUE_MARKER,
+    ANIMATION_DURATION_SHOW,
+    ANIMATION_DURATION_HIDE,
+)
+
+
+class HoverButton(QObject):
+    """
+    悬浮按钮 - 用于节点编辑的浮动按钮
+    包含圆形背景和编辑图标
+    """
+    clicked = pyqtSignal()  # 点击信号
+
+    def __init__(self, parent: QGraphicsItem = None) -> None:
+        super().__init__()
+
+        self._parent = parent
+        self._radius: float = 15.0
+
+        # 创建圆形背景（浅黄灰色）
+        self._ellipse_item = QGraphicsEllipseItem(parent)
+        self._ellipse_item.setBrush(QBrush(QColor(240, 230, 200)))  # 浅黄灰色
+        self._ellipse_item.setPen(QPen(QColor(180, 130, 80), 2))  # 橙灰色描边，2像素宽
+        self._ellipse_item.setZValue(Z_VALUE_BUTTON)  # 确保在最上层
+        self._ellipse_item.setAcceptHoverEvents(True)
+        self._ellipse_item.setToolTip("编辑窗口已打开")  # 添加 tooltip
+
+        # 创建编辑图标文本（书本图标）
+        self._text_item = QGraphicsTextItem("📖", parent)
+        self._text_item.setZValue(Z_VALUE_BUTTON + 1)  # 确保在圆形背景之上
+        self._text_item.setAcceptHoverEvents(False)
+        self._text_item.setTextInteractionFlags(Qt.TextInteractionFlag.NoTextInteraction)
+
+        # 初始隐藏（只有当编辑窗口打开时才显示）
+        self._opacity: float = 0.0
+        self._updateOpacity()
+
+        # 设置按钮位置（左上角）
+        self._updatePosition()
+
+    def _updateOpacity(self) -> None:
+        """更新透明度"""
+        if not hasattr(self, '_ellipse_item') or self._ellipse_item is None:
+            return
+        try:
+            self._ellipse_item.setOpacity(self._opacity)
+            self._text_item.setOpacity(self._opacity)
+        except RuntimeError:
+            pass
+
+    def getOpacity(self) -> float:
+        """获取当前透明度"""
+        return self._opacity
+
+    def setOpacity(self, value: float) -> None:
+        """设置透明度"""
+        self._opacity = value
+        self._updateOpacity()
+
+    # 创建属性用于动画
+    opacity = pyqtProperty(float, getOpacity, setOpacity)
+
+    def _isValid(self) -> bool:
+        """检查 C++ 对象是否仍然有效"""
+        if not hasattr(self, '_ellipse_item') or self._ellipse_item is None:
+            return False
+        try:
+            self._ellipse_item.isVisible()
+            return True
+        except RuntimeError:
+            return False
+
+    def _updatePosition(self) -> None:
+        """更新按钮位置（节点左上角）"""
+        if not self._isValid():
+            return
+        if self._parent is None:
+            return
+
+        try:
+            parent_rect = self._parent.rect()
+            center = parent_rect.center()
+
+            # 计算节点左上角的偏移位置
+            offset_x = -parent_rect.width() / 2 + self._radius
+            offset_y = -parent_rect.height() / 2 + self._radius
+
+            # 设置圆形背景位置
+            ellipse_rect = QRectF(
+                center.x() + offset_x - self._radius,
+                center.y() + offset_y - self._radius,
+                self._radius * 2,
+                self._radius * 2
+            )
+            self._ellipse_item.setRect(ellipse_rect)
+
+            # 设置文本居中
+            text_rect = self._text_item.boundingRect()
+            text_pos = QPointF(
+                center.x() + offset_x - text_rect.width() / 2,
+                center.y() + offset_y - text_rect.height() / 2
+            )
+            self._text_item.setPos(text_pos)
+        except RuntimeError:
+            pass
+
+    def setRadius(self, radius: float) -> None:
+        """设置按钮半径"""
+        self._radius = radius
+        self._updatePosition()
+
+    def show(self) -> None:
+        """显示按钮（带动画）"""
+        if not self._isValid():
+            return
+        try:
+            self._updatePosition()
+            self._ellipse_item.show()
+            self._text_item.show()
+        except RuntimeError:
+            pass
+
+    def hide(self) -> None:
+        """隐藏按钮"""
+        if not self._isValid():
+            return
+        try:
+            self._ellipse_item.hide()
+            self._text_item.hide()
+        except RuntimeError:
+            pass
+
+    def contains(self, pos: QPointF) -> bool:
+        """检查点是否在按钮内"""
+        if not self._isValid():
+            return False
+        try:
+            return self._ellipse_item.contains(pos) or self._text_item.contains(pos)
+        except RuntimeError:
+            return False
+
+    def mousePressEvent(self, event) -> bool:
+        """处理鼠标按下事件"""
+        if not self._isValid():
+            return False
+        try:
+            local_pos = self._ellipse_item.mapFromScene(event.scenePos())
+            if self._ellipse_item.contains(local_pos):
+                self.clicked.emit()
+                return True
+
+            text_local_pos = self._text_item.mapFromScene(event.scenePos())
+            if self._text_item.contains(text_local_pos):
+                self.clicked.emit()
+                return True
+        except RuntimeError:
+            pass
+
+        return False
+
+
+class StartMarker(QObject):
+    """
+    起点标记 - 用于在节点右上角显示起点状态
+    包含圆形背景和🔷图标
+    """
+
+    def __init__(self, parent: QGraphicsItem = None) -> None:
+        super().__init__()
+
+        self._parent = parent
+        self._radius: float = 15.0
+
+        # 创建圆形背景
+        self._ellipse_item = QGraphicsEllipseItem(parent)
+        self._ellipse_item.setBrush(QBrush(QColor(200, 240, 220)))  # 浅绿色背景
+        self._ellipse_item.setPen(QPen(QColor(80, 150, 100), 2))  # 绿灰色描边，2像素宽
+        self._ellipse_item.setZValue(Z_VALUE_MARKER)  # 确保在最上层
+        self._ellipse_item.setAcceptHoverEvents(False)
+        self._ellipse_item.setToolTip("起点节点")  # 添加 tooltip
+
+        # 创建图标文本（🔷）
+        self._text_item = QGraphicsTextItem("🔷", parent)
+        self._text_item.setZValue(Z_VALUE_MARKER + 1)  # 确保在圆形背景之上
+        self._text_item.setAcceptHoverEvents(False)
+        self._text_item.setTextInteractionFlags(Qt.TextInteractionFlag.NoTextInteraction)
+
+        # 初始隐藏
+        self._opacity: float = 0.0
+        self._updateOpacity()
+
+        # 设置按钮位置（右上角）
+        self._updatePosition()
+
+    def _isValid(self) -> bool:
+        """检查 C++ 对象是否仍然有效"""
+        if not hasattr(self, '_ellipse_item') or self._ellipse_item is None:
+            return False
+        try:
+            self._ellipse_item.isVisible()
+            return True
+        except RuntimeError:
+            return False
+
+    def _updateOpacity(self) -> None:
+        """更新透明度"""
+        if not self._isValid():
+            return
+        try:
+            self._ellipse_item.setOpacity(self._opacity)
+            self._text_item.setOpacity(self._opacity)
+        except RuntimeError:
+            pass
+
+    def getOpacity(self) -> float:
+        """获取当前透明度"""
+        return self._opacity
+
+    def setOpacity(self, value: float) -> None:
+        """设置透明度"""
+        self._opacity = value
+        self._updateOpacity()
+
+    # 创建属性用于动画
+    opacity = pyqtProperty(float, getOpacity, setOpacity)
+
+    def _updatePosition(self) -> None:
+        """更新标记位置（节点右上角）"""
+        if not self._isValid():
+            return
+        if self._parent is None:
+            return
+
+        try:
+            parent_rect = self._parent.rect()
+            center = parent_rect.center()
+
+            # 计算节点右上角的偏移位置
+            offset_x = parent_rect.width() / 2 - self._radius
+            offset_y = -parent_rect.height() / 2 + self._radius
+
+            # 设置圆形背景位置
+            ellipse_rect = QRectF(
+                center.x() + offset_x - self._radius,
+                center.y() + offset_y - self._radius,
+                self._radius * 2,
+                self._radius * 2
+            )
+            self._ellipse_item.setRect(ellipse_rect)
+
+            # 设置文本居中
+            text_rect = self._text_item.boundingRect()
+            text_pos = QPointF(
+                center.x() + offset_x - text_rect.width() / 2,
+                center.y() + offset_y - text_rect.height() / 2
+            )
+            self._text_item.setPos(text_pos)
+        except RuntimeError:
+            pass
+
+    def setRadius(self, radius: float) -> None:
+        """设置标记半径"""
+        self._radius = radius
+        self._updatePosition()
+
+    def show(self) -> None:
+        """显示标记（带动画）"""
+        if not self._isValid():
+            return
+        try:
+            self._updatePosition()
+            self._ellipse_item.show()
+            self._text_item.show()
+        except RuntimeError:
+            pass
+
+    def hide(self) -> None:
+        """隐藏标记"""
+        if not self._isValid():
+            return
+        try:
+            self._ellipse_item.hide()
+            self._text_item.hide()
+        except RuntimeError:
+            pass
+
+
+class SingleMarker(QObject):
+    """
+    Single标记 - 用于在节点左下角显示当前执行状态
+    包含圆形背景和⚡图标（表示正在执行）
+    """
+
+    def __init__(self, parent: QGraphicsItem = None) -> None:
+        super().__init__()
+
+        self._parent = parent
+        self._radius: float = 15.0
+
+        # 创建圆形背景（橙黄色 - 表示活跃/执行中）
+        self._ellipse_item = QGraphicsEllipseItem(parent)
+        self._ellipse_item.setBrush(QBrush(QColor(255, 200, 100)))  # 橙黄色背景
+        self._ellipse_item.setPen(QPen(QColor(200, 150, 50), 2))  # 深橙色描边，2像素宽
+        self._ellipse_item.setZValue(Z_VALUE_MARKER)  # 确保在最上层
+        self._ellipse_item.setAcceptHoverEvents(False)
+        self._ellipse_item.setToolTip("正在执行")  # 添加 tooltip
+
+        # 创建图标文本（⚡）
+        self._text_item = QGraphicsTextItem("⚡", parent)
+        self._text_item.setZValue(Z_VALUE_MARKER + 1)  # 确保在圆形背景之上
+        self._text_item.setAcceptHoverEvents(False)
+        self._text_item.setTextInteractionFlags(Qt.TextInteractionFlag.NoTextInteraction)
+
+        # 初始隐藏
+        self._opacity: float = 0.0
+        self._updateOpacity()
+
+        # 设置标记位置（左下角）
+        self._updatePosition()
+
+    def _isValid(self) -> bool:
+        """检查 C++ 对象是否仍然有效"""
+        if not hasattr(self, '_ellipse_item') or self._ellipse_item is None:
+            return False
+        try:
+            self._ellipse_item.isVisible()
+            return True
+        except RuntimeError:
+            return False
+
+    def _updateOpacity(self) -> None:
+        """更新透明度"""
+        if not self._isValid():
+            return
+        try:
+            self._ellipse_item.setOpacity(self._opacity)
+            self._text_item.setOpacity(self._opacity)
+        except RuntimeError:
+            pass
+
+    def getOpacity(self) -> float:
+        """获取当前透明度"""
+        return self._opacity
+
+    def setOpacity(self, value: float) -> None:
+        """设置透明度"""
+        self._opacity = value
+        self._updateOpacity()
+
+    # 创建属性用于动画
+    opacity = pyqtProperty(float, getOpacity, setOpacity)
+
+    def _updatePosition(self) -> None:
+        """更新标记位置（节点左下角）"""
+        if not self._isValid():
+            return
+        if self._parent is None:
+            return
+
+        try:
+            parent_rect = self._parent.rect()
+            center = parent_rect.center()
+
+            # 计算节点左下角的偏移位置
+            offset_x = -parent_rect.width() / 2 + self._radius
+            offset_y = parent_rect.height() / 2 - self._radius
+
+            # 设置圆形背景位置
+            ellipse_rect = QRectF(
+                center.x() + offset_x - self._radius,
+                center.y() + offset_y - self._radius,
+                self._radius * 2,
+                self._radius * 2
+            )
+            self._ellipse_item.setRect(ellipse_rect)
+
+            # 设置文本居中
+            text_rect = self._text_item.boundingRect()
+            text_pos = QPointF(
+                center.x() + offset_x - text_rect.width() / 2,
+                center.y() + offset_y - text_rect.height() / 2
+            )
+            self._text_item.setPos(text_pos)
+        except RuntimeError:
+            pass
+
+    def setRadius(self, radius: float) -> None:
+        """设置标记半径"""
+        self._radius = radius
+        self._updatePosition()
+
+    def show(self) -> None:
+        """显示标记（带动画）"""
+        if not self._isValid():
+            return
+        try:
+            self._updatePosition()
+            self._ellipse_item.show()
+            self._text_item.show()
+        except RuntimeError:
+            pass
+
+    def hide(self) -> None:
+        """隐藏标记"""
+        if not self._isValid():
+            return
+        try:
+            self._ellipse_item.hide()
+            self._text_item.hide()
+        except RuntimeError:
+            pass
+
+
+class CustomMarker(QObject):
+    """
+    Custom标记 - 用户自定义标识
+    支持自定义显示字符、背景色、前景色、边框颜色和厚度、半径
+    默认位置在节点右下角
+    """
+
+    # 默认样式值
+    DEFAULT_TEXT = "★"
+    DEFAULT_BG_COLOR = QColor(227, 242, 253)  # 浅蓝色 #E3F2FD
+    DEFAULT_FG_COLOR = QColor(21, 101, 192)   # 深蓝色 #1565C0
+    DEFAULT_BORDER_COLOR = QColor(100, 181, 246)  # 中蓝色 #64B5F6
+    DEFAULT_BORDER_WIDTH = 2
+    DEFAULT_RADIUS = 15.0
+
+    def __init__(self, parent: QGraphicsItem = None) -> None:
+        super().__init__()
+
+        self._parent = parent
+        self._radius: float = self.DEFAULT_RADIUS
+        self._text: str = self.DEFAULT_TEXT
+        self._bg_color: QColor = self.DEFAULT_BG_COLOR
+        self._fg_color: QColor = self.DEFAULT_FG_COLOR
+        self._border_color: QColor = self.DEFAULT_BORDER_COLOR
+        self._border_width: float = self.DEFAULT_BORDER_WIDTH
+
+        # 创建圆形背景
+        self._ellipse_item = QGraphicsEllipseItem(parent)
+        self._ellipse_item.setBrush(QBrush(self._bg_color))
+        self._ellipse_item.setPen(QPen(self._border_color, self._border_width))
+        self._ellipse_item.setZValue(Z_VALUE_MARKER)
+        self._ellipse_item.setAcceptHoverEvents(False)
+        self._ellipse_item.setToolTip("自定义标记")
+
+        # 创建图标文本
+        self._text_item = QGraphicsTextItem(self._text, parent)
+        self._text_item.setDefaultTextColor(self._fg_color)
+        self._text_item.setZValue(Z_VALUE_MARKER + 1)
+        self._text_item.setAcceptHoverEvents(False)
+        self._text_item.setTextInteractionFlags(Qt.TextInteractionFlag.NoTextInteraction)
+
+        # 初始隐藏
+        self._opacity: float = 0.0
+        self._updateOpacity()
+
+        # 设置标记位置（右下角）
+        self._updatePosition()
+
+    def _isValid(self) -> bool:
+        """检查 C++ 对象是否仍然有效"""
+        if not hasattr(self, '_ellipse_item') or self._ellipse_item is None:
+            return False
+        try:
+            self._ellipse_item.isVisible()
+            return True
+        except RuntimeError:
+            return False
+
+    def _updateOpacity(self) -> None:
+        """更新透明度"""
+        if not self._isValid():
+            return
+        try:
+            self._ellipse_item.setOpacity(self._opacity)
+            self._text_item.setOpacity(self._opacity)
+        except RuntimeError:
+            pass
+
+    def getOpacity(self) -> float:
+        """获取当前透明度"""
+        return self._opacity
+
+    def setOpacity(self, value: float) -> None:
+        """设置透明度"""
+        self._opacity = value
+        self._updateOpacity()
+
+    # 创建属性用于动画
+    opacity = pyqtProperty(float, getOpacity, setOpacity)
+
+    def setText(self, text: str) -> None:
+        """设置显示字符"""
+        self._text = text if text else self.DEFAULT_TEXT
+        if self._isValid():
+            self._text_item.setPlainText(self._text)
+            self._updatePosition()
+
+    def setBgColor(self, color: QColor) -> None:
+        """设置背景色"""
+        self._bg_color = color if color else self.DEFAULT_BG_COLOR
+        if self._isValid():
+            self._ellipse_item.setBrush(QBrush(self._bg_color))
+
+    def setFgColor(self, color: QColor) -> None:
+        """设置前景色（文本颜色）"""
+        self._fg_color = color if color else self.DEFAULT_FG_COLOR
+        if self._isValid():
+            self._text_item.setDefaultTextColor(self._fg_color)
+
+    def setBorderColor(self, color: QColor) -> None:
+        """设置边框颜色"""
+        self._border_color = color if color else self.DEFAULT_BORDER_COLOR
+        if self._isValid():
+            self._ellipse_item.setPen(QPen(self._border_color, self._border_width))
+
+    def setBorderWidth(self, width: float) -> None:
+        """设置边框厚度"""
+        self._border_width = width if width > 0 else self.DEFAULT_BORDER_WIDTH
+        if self._isValid():
+            self._ellipse_item.setPen(QPen(self._border_color, self._border_width))
+
+    def setRadius(self, radius: float) -> None:
+        """设置标记半径"""
+        self._radius = radius if radius > 0 else self.DEFAULT_RADIUS
+        self._updatePosition()
+
+    def setStyle(
+        self,
+        text: str = None,
+        bg_color: QColor = None,
+        fg_color: QColor = None,
+        border_color: QColor = None,
+        border_width: float = None,
+        radius: float = None
+    ) -> None:
+        """批量设置样式"""
+        if text is not None:
+            self._text = text
+        if bg_color is not None:
+            self._bg_color = bg_color
+        if fg_color is not None:
+            self._fg_color = fg_color
+        if border_color is not None:
+            self._border_color = border_color
+        if border_width is not None:
+            self._border_width = border_width
+        if radius is not None:
+            self._radius = radius
+
+        if self._isValid():
+            self._text_item.setPlainText(self._text)
+            self._text_item.setDefaultTextColor(self._fg_color)
+            self._ellipse_item.setBrush(QBrush(self._bg_color))
+            self._ellipse_item.setPen(QPen(self._border_color, self._border_width))
+            self._updatePosition()
+
+    def getStyle(self) -> dict:
+        """获取当前样式配置"""
+        return {
+            "text": self._text,
+            "bg_color": self._bg_color.name(),
+            "fg_color": self._fg_color.name(),
+            "border_color": self._border_color.name(),
+            "border_width": self._border_width,
+            "radius": self._radius
+        }
+
+    def _updatePosition(self) -> None:
+        """更新标记位置（节点右下角）"""
+        if not self._isValid():
+            return
+        if self._parent is None:
+            return
+
+        try:
+            parent_rect = self._parent.rect()
+            center = parent_rect.center()
+
+            # 计算节点右下角的偏移位置
+            offset_x = parent_rect.width() / 2 - self._radius
+            offset_y = parent_rect.height() / 2 - self._radius
+
+            # 设置圆形背景位置
+            ellipse_rect = QRectF(
+                center.x() + offset_x - self._radius,
+                center.y() + offset_y - self._radius,
+                self._radius * 2,
+                self._radius * 2
+            )
+            self._ellipse_item.setRect(ellipse_rect)
+
+            # 设置文本居中
+            text_rect = self._text_item.boundingRect()
+            text_pos = QPointF(
+                center.x() + offset_x - text_rect.width() / 2,
+                center.y() + offset_y - text_rect.height() / 2
+            )
+            self._text_item.setPos(text_pos)
+        except RuntimeError:
+            pass
+
+    def show(self) -> None:
+        """显示标记（带动画）"""
+        if not self._isValid():
+            return
+        try:
+            self._updatePosition()
+            self._ellipse_item.show()
+            self._text_item.show()
+        except RuntimeError:
+            pass
+
+    def hide(self) -> None:
+        """隐藏标记"""
+        if not self._isValid():
+            return
+        try:
+            self._ellipse_item.hide()
+            self._text_item.hide()
+        except RuntimeError:
+            pass