PyQt5绘制闭合多边形：解决线条绘制顺序依赖问题

核心思路 #

要摆脱绘制顺序的限制，核心是从点的连接关系和整体连通性两个维度判断闭合：

1. 所有端点的连接次数（度数）必须为2——每个点恰好被两条线条衔接，这是闭合形状的基础特征
2. 所有线条必须构成单一连通环——不能是多个独立闭合图形或零散线条的组合

代码实现示例 #

假设你的代码中用self.lines存储所有绘制的线条（每条线条为(QPoint(start), QPoint(end))的元组），添加以下检测与多边形生成逻辑：

from collections import defaultdict

def is_shape_closed(lines):
    # 统计每个点的连接度数
    point_degree = defaultdict(int)
    # 用坐标元组作为哈希键（QPoint不可直接哈希）
    for start, end in lines:
        start_key = (start.x(), start.y())
        end_key = (end.x(), end.y())
        point_degree[start_key] += 1
        point_degree[end_key] += 1
    
    # 检查所有点度数是否为2
    for degree in point_degree.values():
        if degree != 2:
            return False
    
    # 用并查集验证整体连通性
    if not lines:
        return False
    parent = {}

    def find(p):
        if parent[p] != p:
            parent[p] = find(parent[p])
        return parent[p]

    def union(p1, p2):
        root1 = find(p1)
        root2 = find(p2)
        if root1 != root2:
            parent[root2] = root1
    
    # 初始化并查集并合并线条端点
    for start, end in lines:
        s = (start.x(), start.y())
        e = (end.x(), end.y())
        parent.setdefault(s, s)
        parent.setdefault(e, e)
        union(s, e)
    
    # 所有点必须属于同一个连通分量
    root = find(next(iter(parent.keys())))
    for p in parent.keys():
        if find(p) != root:
            return False
    
    return True

def get_closed_polygon(lines):
    # 构建点的邻接表
    adj = defaultdict(list)
    for start, end in lines:
        s = (start.x(), start.y())
        e = (end.x(), end.y())
        adj[s].append(e)
        adj[e].append(s)
    
    # 遍历邻接表生成有序多边形顶点
    polygon = []
    current = next(iter(adj.keys()))
    prev_point = None
    while len(polygon) < len(adj):
        polygon.append(current)
        # 排除前一个点，获取下一个衔接点
        next_points = [p for p in adj[current] if p != prev_point]
        if not next_points:
            break
        prev_point = current
        current = next_points[0]
    
    # 转换回QPoint列表
    return [QPoint(p[0], p[1]) for p in polygon]

在绘制流程中调用 #

在你的绘图函数里，加入以下逻辑即可实现自动检测与填充：

painter = QPainter(self)
# 先绘制所有线条
for start, end in self.lines:
    painter.drawLine(start, end)

# 检测并填充闭合形状
if is_shape_closed(self.lines):
    polygon = get_closed_polygon(self.lines)
    painter.setBrush(QColor(255, 0, 0))  # 红色填充
    painter.drawPolygon(QPolygon(polygon))

关键细节说明 #

• 用坐标元组替代QPoint作为哈希键，解决QPoint无法直接存入字典的问题
• 并查集确保所有线条属于同一个连通环，避免多个独立闭合图形被误判
• 邻接表遍历自动生成有序多边形顶点，完全不依赖原始线条的绘制顺序

内容的提问来源于stack exchange，提问作者Ahoora