chartflow 0.1__py3-none-any.whl

chartflow/node/demo.py

@@ -0,0 +1,258 @@
+"""
+qnode 演示示例 - 展示节点图编辑器的基本功能
+"""
+
+from PyQt6.QtCore import QTimer
+import random
+from PyQt6.QtWidgets import QFileDialog
+import json
+
+import sys
+sys.path.insert(0, 'F:\\Python\\Python314\\Lib\\site-packages')
+
+from PyQt6.QtWidgets import QApplication, QVBoxLayout, QHBoxLayout, QWidget, QPushButton, QLabel, QComboBox
+from PyQt6.QtCore import QPointF, Qt
+from chartflow.node import NodeCanvasWindow, QNodeCanvas, NodeShape, ConnectionType
+from chartflow.node.layout_utils import DenseLayoutEngine, ViewportFitter
+from chartflow.theme import applyTheme, PALETTE
+
+
+class DemoWindow(QWidget):
+    """演示窗口 - 包含控制面板和节点画布"""
+
+    def __init__(self):
+        super().__init__()
+        self.setWindowTitle("QNode 演示 - 节点图编辑器")
+        self.resize(1200, 800)
+
+        # 主布局
+        layout = QHBoxLayout(self)
+
+        # 左侧控制面板
+        controlPanel = self.CreateControlPanel()
+        layout.addWidget(controlPanel, 0)
+
+        # 右侧节点画布窗口
+        self.nodeWindow = NodeCanvasWindow()
+        self.canvas = self.nodeWindow.canvas
+        layout.addWidget(self.nodeWindow, 1)
+
+        # 初始化一些示例节点
+        self.InitDemoNodes()
+
+    def CreateControlPanel(self):
+        """创建控制面板"""
+        panel = QWidget()
+        panel.setFixedWidth(250)
+        panel.setObjectName("controlPanel")
+
+        layout = QVBoxLayout(panel)
+        layout.setContentsMargins(12, 12, 12, 12)
+        layout.setSpacing(8)
+        layout.setAlignment(Qt.AlignmentFlag.AlignTop)
+
+        # 标题
+        title = QLabel("QNode 演示控制面板")
+        title.setStyleSheet("font-size: 15px; font-weight: 600; padding-bottom: 6px;")
+        layout.addWidget(title)
+
+        # 添加节点区域
+        layout.addWidget(self._sectionLabel("添加节点"))
+
+        self.shapeCombo = QComboBox()
+        self.shapeCombo.addItems(["椭圆 (开始/结束)", "矩形 (处理)", "菱形 (决策)"])
+        layout.addWidget(self.shapeCombo)
+
+        addBtn = QPushButton("添加节点")
+        addBtn.clicked.connect(self.AddNode)
+        layout.addWidget(addBtn)
+
+        layout.addSpacing(16)
+
+        # 布局算法区域
+        layout.addWidget(self._sectionLabel("自动布局"))
+
+        self.layoutCombo = QComboBox()
+        self.layoutCombo.addItems(["紧凑", "环形", "网格", "层次", "力导向"])
+        layout.addWidget(self.layoutCombo)
+
+        layoutBtn = QPushButton("应用布局")
+        layoutBtn.clicked.connect(self.ApplyLayout)
+        layout.addWidget(layoutBtn)
+
+        layout.addSpacing(16)
+
+        # 视图控制区域
+        layout.addWidget(self._sectionLabel("视图控制"))
+
+        fitBtn = QPushButton("适配视口")
+        fitBtn.clicked.connect(self.FitViewport)
+        layout.addWidget(fitBtn)
+
+        centerBtn = QPushButton("居中节点")
+        centerBtn.clicked.connect(self.CenterNodes)
+        layout.addWidget(centerBtn)
+
+        layout.addSpacing(16)
+
+        # 导入/导出区域
+        layout.addWidget(self._sectionLabel("导入/导出"))
+
+        exportBtn = QPushButton("导出为 JSON")
+        exportBtn.clicked.connect(self.ExportJson)
+        layout.addWidget(exportBtn)
+
+        importBtn = QPushButton("从 JSON 导入")
+        importBtn.clicked.connect(self.ImportJson)
+        layout.addWidget(importBtn)
+
+        clearBtn = QPushButton("清空画布")
+        clearBtn.clicked.connect(self.ClearCanvas)
+        layout.addWidget(clearBtn)
+
+        layout.addSpacing(16)
+
+        # 帮助信息
+        helpLabel = QLabel(
+            "操作说明\n"
+            "• 左键拖拽: 框选\n"
+            "• Ctrl+点击: 多选\n"
+            "• 从节点拖拽: 创建连接\n"
+            "• 双击节点: 编辑\n"
+            "• 右键连接: 切换线条类型\n"
+            "• 中键拖拽: 平移\n"
+            "• 空格/Alt+左键: 平移\n"
+            "• Alt 悬停节点: 调整大小光标\n"
+            "• 滚轮: 缩放\n"
+            "• Delete: 删除"
+        )
+        helpLabel.setStyleSheet(f"color: {PALETTE.text_dim}; font-size: 11px;")
+        layout.addWidget(helpLabel)
+
+        layout.addStretch()
+
+        return panel
+
+    @staticmethod
+    def _sectionLabel(text: str) -> QLabel:
+        """A small accent-colored section header used inside the control panel."""
+        label = QLabel(text)
+        label.setStyleSheet(
+            f"color: {PALETTE.accent_hover}; font-size: 11px; font-weight: 600;"
+            f" border-bottom: 1px solid {PALETTE.border}; padding-bottom: 3px;"
+        )
+        return label
+
+    def InitDemoNodes(self):
+        """初始化演示节点"""
+        # 创建流程图示例
+        start = self.canvas.addNode("开始", QPointF(0, 0), NodeShape.ELLIPSE)
+
+        inputNode = self.canvas.addNode("输入数据", QPointF(200, -100), NodeShape.RECTANGLE)
+        process = self.canvas.addNode("处理数据", QPointF(400, 0), NodeShape.RECTANGLE)
+        decision = self.canvas.addNode("验证?", QPointF(600, 0), NodeShape.DIAMOND)
+        success = self.canvas.addNode("成功", QPointF(800, -100), NodeShape.RECTANGLE)
+        error = self.canvas.addNode("错误处理", QPointF(800, 100), NodeShape.RECTANGLE)
+        end = self.canvas.addNode("结束", QPointF(1000, 0), NodeShape.ELLIPSE)
+
+        # 创建连接
+        self.canvas.addConnection(start, inputNode)
+        self.canvas.addConnection(inputNode, process)
+        self.canvas.addConnection(process, decision)
+        self.canvas.addConnection(decision, success)
+        self.canvas.addConnection(decision, error)
+        self.canvas.addConnection(success, end)
+        self.canvas.addConnection(error, end)
+
+        # 设置起始节点
+        self.canvas.startNode = start
+
+        # 适配视口 - 使用较小的边距确保节点清晰可见
+        QTimer.singleShot(100, self._fitViewportOnStartup)
+
+    def _fitViewportOnStartup(self):
+        """启动时适配视口"""
+        fitter = ViewportFitter(self.canvas)
+        # 使用较小的边距(20)确保图形清晰可见
+        fitter.fitViewport(margin=20.0)
+        # 限制最小缩放比例，避免图形过小
+        current_scale = self.canvas.transform().m11()
+        if current_scale < 0.8:
+            self.canvas.resetTransform()
+            self.canvas.scale(0.8, 0.8)
+
+    def AddNode(self):
+        """添加新节点"""
+        shapeIdx = self.shapeCombo.currentIndex()
+        shape = [NodeShape.ELLIPSE, NodeShape.RECTANGLE, NodeShape.DIAMOND][shapeIdx]
+        name = ["开始/结束", "处理节点", "决策节点"][shapeIdx]
+
+        # 在随机位置添加节点
+        x = random.randint(-200, 200)
+        y = random.randint(-200, 200)
+        node = self.canvas.addNode(name, QPointF(x, y), shape)
+
+    def ApplyLayout(self):
+        """应用布局算法"""
+        layoutNames = ["compact", "circular", "grid", "hierarchical", "force"]
+        layoutName = layoutNames[self.layoutCombo.currentIndex()]
+
+        engine = DenseLayoutEngine(self.canvas)
+        engine.autoLayout(layoutName)
+
+    def FitViewport(self):
+        """适配视口"""
+        fitter = ViewportFitter(self.canvas)
+        fitter.fitViewport()
+
+    def CenterNodes(self):
+        """居中节点"""
+        fitter = ViewportFitter(self.canvas)
+        fitter.centerOnNodes()
+
+    def ExportJson(self):
+        """导出为 JSON"""
+
+        json_str = self.canvas.exportToJson()
+        filePath, _ = QFileDialog.getSaveFileName(
+            self, "导出图", "graph.json", "JSON 文件 (*.json)"
+        )
+        if filePath:
+            with open(filePath, 'w', encoding='utf-8') as f:
+                f.write(json_str)
+            print(f"已导出到: {filePath}")
+
+    def ImportJson(self):
+        """从 JSON 导入"""
+
+        filePath, _ = QFileDialog.getOpenFileName(
+            self, "导入图", "", "JSON 文件 (*.json)"
+        )
+        if filePath:
+            with open(filePath, 'r', encoding='utf-8') as f:
+                data = json.load(f)
+            self.canvas.importData(data)
+            print(f"已从 {filePath} 导入")
+
+    def ClearCanvas(self):
+        """清空画布"""
+        self.canvas.clear()
+
+
+def main():
+    """主函数"""
+    app = QApplication(sys.argv)
+
+    # 设置应用样式与统一主题
+    app.setStyle('Fusion')
+    applyTheme(app)
+
+    # 创建并显示演示窗口
+    window = DemoWindow()
+    window.show()
+
+    sys.exit(app.exec())
+
+
+if __name__ == "__main__":
+    main()

chartflow/node/demo_arrow_drag.py

@@ -0,0 +1,46 @@
+"""
+qnode 箭头拖拽功能演示
+
+演示如何通过拖拽连接线上的箭头来创建新连接。
+"""
+
+import sys
+from PyQt6.QtWidgets import QApplication
+from PyQt6.QtCore import QPointF
+
+from chartflow.node import QNodeCanvas, QNodeItem, NodeShape
+
+
+def DemoArrowDrag():
+    """演示箭头拖拽创建新连接功能"""
+    app = QApplication(sys.argv)
+
+    # 创建画布
+    canvas = QNodeCanvas()
+    canvas.setWindowTitle("箭头拖拽演示 - 拖拽连接线上的箭头到另一个节点")
+    canvas.resize(800, 600)
+
+    # 创建三个节点
+    node_a = canvas.addNode("节点 A", QPointF(200, 300), NodeShape.ELLIPSE)
+    node_b = canvas.addNode("节点 B", QPointF(500, 200), NodeShape.RECTANGLE)
+    node_c = canvas.addNode("节点 C", QPointF(500, 400), NodeShape.RECTANGLE)
+
+    # 创建初始连接 A -> B
+    conn = canvas.addConnection(node_a, node_b)
+    print(f"创建连接: {node_a.name} -> {node_b.name}")
+
+    # 演示说明
+    print("\n=== 箭头拖拽功能说明 ===")
+    print("1. 将鼠标悬停在连接线末端的箭头上")
+    print("2. 按住鼠标左键拖拽箭头")
+    print("3. 拖拽到另一个节点上释放，会创建新连接")
+    print("4. 如果是单向连接，原连接会被移除，新连接会被创建")
+    print("5. 如果是双向连接，会保留未拖拽方向的连接")
+    print("\n提示：可以拖拽箭头到节点C来创建新连接！")
+
+    canvas.show()
+    sys.exit(app.exec())
+
+
+if __name__ == "__main__":
+    DemoArrowDrag()

chartflow/node/demo_decorator.py

@@ -0,0 +1,63 @@
+"""
+qnode 装饰器节点边框颜色演示
+
+演示如何为不同类型的装饰器节点设置不同的边框颜色：
+- recursive 装饰的节点 → 紫罗兰色 (violet) 边框
+- behavior/sequence/selector/parallel 装饰的节点 → 皇家蓝色 (royal blue) 边框
+"""
+
+import sys
+from PyQt6.QtWidgets import QApplication
+from PyQt6.QtCore import QPointF
+
+from chartflow.node import QNodeCanvas, QNodeItem, NodeShape
+
+
+def DemoDecoratorColors():
+    """演示装饰器节点边框颜色"""
+    app = QApplication(sys.argv)
+
+    # 创建画布
+    canvas = QNodeCanvas()
+    canvas.setWindowTitle("装饰器节点边框颜色演示")
+    canvas.resize(900, 600)
+
+    # 创建普通节点（无装饰器）
+    node_normal = canvas.addNode("普通节点", QPointF(150, 150), NodeShape.ELLIPSE)
+
+    # 创建 recursive 装饰的节点 → 紫罗兰色边框
+    node_recursive = canvas.addNode("Recursive", QPointF(450, 150), NodeShape.RECTANGLE)
+    node_recursive.decorator = "recursive"
+
+    # 创建 behavior 装饰的节点 → 皇家蓝色边框
+    node_behavior = canvas.addNode("Behavior", QPointF(150, 300), NodeShape.RECTANGLE)
+    node_behavior.decorator = "behavior"
+
+    # 创建 sequence 装饰的节点 → 皇家蓝色边框
+    node_sequence = canvas.addNode("Sequence", QPointF(450, 300), NodeShape.RECTANGLE)
+    node_sequence.decorator = "sequence"
+
+    # 创建 selector 装饰的节点 → 皇家蓝色边框
+    node_selector = canvas.addNode("Selector", QPointF(150, 450), NodeShape.RECTANGLE)
+    node_selector.decorator = "selector"
+
+    # 创建 parallel 装饰的节点 → 皇家蓝色边框
+    node_parallel = canvas.addNode("Parallel", QPointF(450, 450), NodeShape.RECTANGLE)
+    node_parallel.decorator = "parallel"
+
+    # 创建菱形节点带 recursive 装饰器
+    node_diamond = canvas.addNode("Recursive\nDiamond", QPointF(750, 300), NodeShape.DIAMOND)
+    node_diamond.decorator = "recursive"
+
+    print("=== 装饰器节点边框颜色演示 ===")
+    print("紫罗兰色 (Violet) 边框: recursive 装饰的节点")
+    print("皇家蓝色 (Royal Blue) 边框: behavior, sequence, selector, parallel 装饰的节点")
+    print("")
+    print("提示: 点击节点查看选中状态下的颜色效果")
+
+    canvas.show()
+    sys.exit(app.exec())
+
+
+if __name__ == "__main__":
+    DemoDecoratorColors()

chartflow/node/demo_ports.py

@@ -0,0 +1,69 @@
+"""
+端口(ports)演示 - 展示节点的输入/输出端口
+
+运行:
+    python -m chartflow.node.demo_ports
+或:
+    python F:/Python/Python312/Lib/site-packages/chartflow/node/demo_ports.py
+
+交互:
+    - hover 节点超过 0.5s: 显示所有端口 label(外侧文本)
+    - hover 单个端口 dot: 该端口立即显示 label + DodgerBlue 描边
+"""
+
+import sys
+
+from PyQt6.QtCore import QPointF
+from PyQt6.QtWidgets import QApplication
+
+from chartflow.node import NodeCanvasWindow, NodeShape
+from chartflow.node.port_item import QNodePortSpec
+from chartflow.theme import applyTheme
+
+
+def build_counter_node(canvas) -> None:
+    """构造一个带完整端口的计数器节点(模仿 uipcweb 的 Counter16)"""
+    node = canvas.addNode("Counter16", QPointF(0, 0), NodeShape.RECTANGLE)
+    node.ports = {
+        "left": [
+            QNodePortSpec(name="clk", shape="triangle", size=10, hit="1"),
+            QNodePortSpec(name="rst", shape="triangle", size=10, hit="1"),
+            QNodePortSpec(name="cmp", shape="trapezoid", size=12, hit="16", label="cmp[15:0]"),
+            QNodePortSpec(name="en", shape="circle", size=10, hit="1"),
+        ],
+        "right": [
+            QNodePortSpec(name="cnt", shape="trapezoid", size=12, hit="16", label="cnt[15:0]"),
+            QNodePortSpec(name="rco", shape="circle", size=10, hit="1"),
+            QNodePortSpec(name="flag", shape="square", size=10, hit="8"),
+        ],
+    }
+
+
+def build_simple_node(canvas) -> None:
+    """一个最小节点: 只有两个端口，label 未指定(默认等同 name)"""
+    node = canvas.addNode("Buf", QPointF(260, 0), NodeShape.RECTANGLE)
+    node.ports = {
+        "left": [QNodePortSpec(name="in", shape="triangle", size=9, hit="1")],
+        "right": [QNodePortSpec(name="out", shape="triangle", size=9, hit="1")],
+    }
+
+
+def main() -> None:
+    app = QApplication(sys.argv)
+    applyTheme(app)
+
+    win = NodeCanvasWindow()
+    canvas = win.canvas
+
+    build_counter_node(canvas)
+    build_simple_node(canvas)
+
+    win.resize(960, 600)
+    win.setWindowTitle("chartflow · ports demo")
+    win.show()
+
+    sys.exit(app.exec())
+
+
+if __name__ == "__main__":
+    main()

chartflow/node/enums.py

@@ -0,0 +1,35 @@
+"""
+枚举定义模块
+
+遵循代码规范:
+- enum类不能定义函数，数据和方法必须分离
+"""
+
+from enum import Enum, auto
+
+
+class NodeState(Enum):
+    """节点状态枚举"""
+    IDLE = auto()
+    HOVER = auto()
+    SELECTED = auto()
+
+
+class NodeShape(Enum):
+    """节点形状枚举"""
+    ELLIPSE = auto()    # 椭圆形/圆形
+    RECTANGLE = auto()  # 矩形
+    DIAMOND = auto()    # 菱形
+
+
+class ArrowState(Enum):
+    """箭头状态枚举"""
+    IDLE = auto()
+    HOVER = auto()
+    SELECTED = auto()
+
+
+class ConnectionType(Enum):
+    """连线类型枚举"""
+    BEZIER = auto()     # 贝塞尔曲线
+    STRAIGHT = auto()   # 直线

chartflow/node/example.py

@@ -0,0 +1,84 @@
+"""
+qnode库使用示例
+
+展示如何使用qnode库创建节点图
+"""
+
+from PyQt6.QtWidgets import QVBoxLayout, QLabel, QTextEdit
+
+
+import sys
+from PyQt6.QtWidgets import QApplication
+from PyQt6.QtCore import QPointF
+
+from chartflow.node import (
+    NodeCanvasWindow, QNodeCanvas, QNodeItem,
+    NodeShape, ConnectionType, QNodeEditor
+)
+
+
+class MyNodeEditor(QNodeEditor):
+    """自定义节点编辑器示例"""
+    
+    def __init__(self, node, parent=None):
+        super().__init__(node, parent)
+        self.resize(400, 300)
+        
+        
+        layout = QVBoxLayout(self)
+        
+        # 显示节点信息
+        info_label = QLabel(f"Editing node: {node.name}")
+        layout.addWidget(info_label)
+        
+        # 添加文本编辑器
+        self.text_edit = QTextEdit()
+        self.text_edit.setPlainText(node.data.get("content", ""))
+        layout.addWidget(self.text_edit)
+    
+    def save(self) -> bool:
+        """保存编辑器内容到节点"""
+        self.node.data["content"] = self.text_edit.toPlainText()
+        return True
+
+
+def main():
+    """主函数"""
+    app = QApplication(sys.argv)
+    
+    # 创建主窗口
+    window = NodeCanvasWindow()
+    canvas = window.canvas
+    
+    # 添加一些节点
+    node1 = canvas.addNode("Start", QPointF(0, 0), NodeShape.ELLIPSE)
+    node2 = canvas.addNode("Process", QPointF(200, 100), NodeShape.RECTANGLE)
+    node3 = canvas.addNode("Decision", QPointF(400, 0), NodeShape.DIAMOND)
+    node4 = canvas.addNode("End", QPointF(600, 100), NodeShape.ELLIPSE)
+    
+    # 设置节点编辑器
+    node2.editorClass = MyNodeEditor
+    
+    # 添加一些数据
+    node2.data["description"] = "This is a process node"
+    node2.data["content"] = "Process logic here..."
+    
+    # 创建连接
+    canvas.addConnection(node1, node2)
+    canvas.addConnection(node2, node3)
+    canvas.addConnection(node3, node4)
+    
+    # 设置双向连接示例
+    canvas.addConnection(node2, node4, is_bidirectional=True)
+    
+    # 设置起点节点
+    canvas.startNode = node1
+    
+    # 显示窗口
+    window.show()
+    
+    sys.exit(app.exec())
+
+
+if __name__ == "__main__":
+    main()