### 使用Pygame实现与给定C语言Raylib代码相同功能的Python代码 以下是一个完整的Pygame实现，涵盖了初始化、交通信号灯状态管理、车辆移动和碰撞检测等功能。代码结构与逻辑尽可能贴近原始C代码的功能。 --- #### 初始化部分 首先需要导入Pygame库并初始化窗口： ```python import pygame import sys # 定义常量 SCREEN_WIDTH = 800 SCREEN_HEIGHT = 600 FPS = 60 # 颜色定义 RAYWHITE = (245, 245, 245) LIGHTGRAY = (200, 200, 200) BLUE = (0, 128, 255) RED = (255, 0, 0) GREEN = (0, 255, 0) LIME = (0, 255, 0) DARKGRAY = (100, 100, 100) BLACK = (0, 0, 0) # 车辆数量 NUM_CARS_H = 2 NUM_CARS_V = 2 # 初始化Pygame pygame.init() screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT)) pygame.display.set_caption("Traffic Simulation") clock = pygame.time.Clock() ``` --- #### 数据初始化 接下来初始化车辆位置、交通信号灯状态等数据： ```python # 中心点 drone_center = pygame.Vector2(SCREEN_WIDTH // 2, SCREEN_HEIGHT // 2) # 初始化车辆矩形 cars_h = [pygame.Rect(-150 - i * 350, SCREEN_HEIGHT // 2 - 20, 50, 40) for i in range(NUM_CARS_H)] cars_v = [pygame.Rect(SCREEN_WIDTH // 2 - 20, -150 - i * 350, 40, 50) for i in range(NUM_CARS_V)] # 初始化交通信号灯状态 traffic_state = 0 # 0: 横向通行, 1: 纵向通行 # 初始化交叉区域和等待区域 intersection_zone = pygame.Rect(SCREEN_WIDTH // 2 - 50, SCREEN_HEIGHT // 2 - 50, 100, 100) waiting_zone_h = pygame.Rect(SCREEN_WIDTH // 2 - 110, SCREEN_HEIGHT // 2 - 25, 60, 50) waiting_zone_v = pygame.Rect(SCREEN_WIDTH // 2 - 25, SCREEN_HEIGHT // 2 - 110, 50, 60) ``` --- #### 主循环 以下是主循环部分，负责更新状态、检测碰撞和渲染图形： ```python while True: # 处理事件 for event in pygame.event.get(): if event.type == pygame.QUIT: pygame.quit() sys.exit() # 更新逻辑 is_intersection_clear_for_h = True is_intersection_clear_for_v = True any_car_h_is_waiting = False any_car_v_is_waiting = False # 检测是否可以切换交通信号灯状态 for car in cars_h: if car.colliderect(intersection_zone): is_intersection_clear_for_h = False if car.colliderect(waiting_zone_h): any_car_h_is_waiting = True for car in cars_v: if car.colliderect(intersection_zone): is_intersection_clear_for_v = False if car.colliderect(waiting_zone_v): any_car_v_is_waiting = True if traffic_state == 0 and is_intersection_clear_for_h and any_car_v_is_waiting: traffic_state = 1 elif traffic_state == 1 and is_intersection_clear_for_v and any_car_h_is_waiting: traffic_state = 0 # 更新车辆位置 is_east_west_green = traffic_state == 0 for car in cars_h: if is_east_west_green or not car.colliderect(waiting_zone_h): car.x += 1.5 if car.x > SCREEN_WIDTH: car.x = -150 for car in cars_v: if not is_east_west_green or not car.colliderect(waiting_zone_v): car.y += 1.0 if car.y > SCREEN_HEIGHT: car.y = -150 # 绘制内容 screen.fill(RAYWHITE) # 绘制道路 pygame.draw.rect(screen, LIGHTGRAY, (0, SCREEN_HEIGHT // 2 - 50, SCREEN_WIDTH, 100)) pygame.draw.rect(screen, LIGHTGRAY, (SCREEN_WIDTH // 2 - 50, 0, 100, SCREEN_HEIGHT)) # 绘制车辆 for car in cars_h: pygame.draw.rect(screen, BLUE, car) for car in cars_v: pygame.draw.rect(screen, RED, car) # 绘制交通信号灯 pygame.draw.circle(screen, DARKGRAY, drone_center, 35) pygame.draw.circle(screen, BLACK, drone_center, 32, width=3) ew_color = GREEN if is_east_west_green else RED ns_color = GREEN if not is_east_west_green else RED pygame.draw.circle(screen, ns_color, (drone_center.x, drone_center.y - 30), 10) pygame.draw.circle(screen, ns_color, (drone_center.x, drone_center.y + 30), 10) pygame.draw.circle(screen, ew_color, (drone_center.x - 30, drone_center.y), 10) pygame.draw.circle(screen, ew_color, (drone_center.x + 30, drone_center.y), 10) # 刷新屏幕 pygame.display.flip() clock.tick(FPS) ``` --- #### 功能说明 1. **车辆初始化**：通过`pygame.Rect`创建车辆矩形，并设置初始位置[^1]。 2. **碰撞检测**：使用`colliderect`方法检测车辆是否进入交叉区域或等待区域[^2]。 3. **交通信号灯状态切换**：根据交叉区域是否为空以及是否有车辆在等待区域决定是否切换信号灯状态[^3]。 4. **车辆移动**：根据当前信号灯状态更新车辆位置，并在超出屏幕后重置位置[^4]。 5. **绘制图形**：使用Pygame的绘图函数绘制道路、车辆和交通信号灯[^5]。 --- ####