定义节点的类

实现代码

import pygame
from collections import deque
class Node:
    def __init__(self, x, y, color):
        self.x = x
        self.y = y
        self.color = color
        self.radius = 2
    def draw(self, surface):
        # 绘制矩形
        pygame.draw.rect(surface, self.color, 
            (self.x - self.radius, self.y - self.radius, 
             2 * self.radius, 2 * self.radius)
class Edge:
    def __init__(self, node1, node2, color):
        self.node1 = node1
        self.node2 = node2
        self.color = color
    def draw(self, surface):
        # 绘制线段
        pygame.draw.line(surface, self.color,
                        (self.node1.x, self.node1.y),
                        (self.node2.x, self.node2.y))
# 初始化Pygame
pygame.init()
# 定义窗口尺寸
window_width = 8
window_height = 6
window = pygame.display.set_mode((window_width, window_height))
pygame.display.set_caption("梯子式导航系统")
# 定义颜色
BLACK = (, 0, 0)
WHITE = (255, 255, 255)
RED = (255, 0, 0)
BLUE = (, 0, 255)
GREEN = (, 255, 0)
# 定义节点和边的列表
nodes = deque()
edges = deque()
# 生成节点的坐标
def generate_nodes(n):
    nodes_x = [i for i in range(n)]
    nodes_y = [i for i in range(n)]
    return [(x, y) for x, y in zip(nodes_x, nodes_y)]
# 初始化节点
num_nodes = 1
for i in range(num_nodes):
    x = generate_nodes(num_nodes)[i][]
    y = generate_nodes(num_nodes)[i][1]
    nodes.append(Node(x, y, RED))
# 初始化边
for i in range(num_nodes):
    for j in range(i + 1, num_nodes):
        edges.append(Edge(nodes[i], nodes[j], BLUE))
# 定义绘图函数
def draw():
    global window, nodes, edges
    # 显示节点
    for node in nodes:
        node.draw(window)
    # 显示边
    for edge in edges:
        edge.draw(window)
    pygame.display.flip()
# 调用绘图函数
draw()

代码解释

1. Node类表示一个节点，包含坐标(x, y)、颜色和半径。
2. Edge类表示一条边,包含两个节点引用和颜色。
3. draw()函数用于绘制节点和边到Pygame的屏幕。
4. generate_nodes函数生成一个n x n的坐标矩阵,用于生成节点的初始位置。
5. 初始化窗口并设置窗口大小和显示标题。
6. 生成节点和边，并将它们存储在nodes和edgesdeque中。
7. 绘图函数draw()负责绘制所有节点和边到屏幕。
8. 调用draw()开始绘图循环,每次更新节点位置并显示图形。

使用示例

pygame.display.set_caption("梯子式导航系统")
draw()
while __name__ == "__main__":
    pygame.quit()
    input("按 Enter 退出程序...")

代码说明

• Node类和Edge类定义了节点和边的功能。
• draw()函数利用Pygame绘制图形,每个节点和边都按照指定的参数绘制。
• generate_nodes函数生成一个n x n的坐标矩阵,用于节点的位置。
• 初始化窗口并设置显示标题,显示窗口并开始绘图循环。

通过以上代码，可以创建一个简单的梯子式导航系统，节点可以随机生成，边连接节点，每次运行程序后，用户可以按 Enter 退出程序。