import pygame from pygame import Surface from pygame.constants import QUIT from maze import Maze from draw import Button, Toast from save_ui import SaveLoadUI from config import * import sys import os screen: Surface = None # 窗口实例 clock = None # 时钟实例 textFont = None # 字体 def pygameInit(title: str = "pygame"): """初始化 pygame""" pygame.init() pygame.mixer.init() # 声音初始化 pygame.display.set_caption(title) global screen, clock, textFont # 修改全局变量 screen = pygame.display.set_mode((UI_WIDTH, UI_HEIGHT)) clock = pygame.time.Clock() # Initialize font with UTF-8 support pygame.font.init() textFont = pygame.font.Font(FONT_FILE, FONT_SIZE) if __name__ == "__main__": pygameInit("maze") maze = Maze(wall_size=WALL_SIZE, maze_size=MAZE_SIZE, file_name=DEFAULT_MAZE_FILE) # 加载图片资源 image_wall = pygame.image.load(WALL_IMAGE).convert_alpha() image_wall = pygame.transform.scale(image_wall, (WALL_SIZE, WALL_SIZE)) image_coin = pygame.image.load(COIN_IMAGE).convert_alpha() image_coin = pygame.transform.scale(image_coin, (WALL_SIZE, WALL_SIZE)) image_trap = pygame.image.load(TRAP_IMAGE).convert_alpha() image_trap = pygame.transform.scale(image_trap, (WALL_SIZE, WALL_SIZE)) # 获取按钮位置配置 button_positions = get_button_positions() # 创建按钮 button_start_texture = pygame.image.load(START_BUTTON_IMAGE).convert_alpha() button_start_texture = pygame.transform.scale(button_start_texture, BUTTON_START_SIZE) button_start = Button(pygame.rect.Rect(*button_positions['start_button'], *BUTTON_START_SIZE), button_start_texture) button_save_texture = pygame.image.load(SAVE_BUTTON_IMAGE).convert_alpha() button_save_texture = pygame.transform.scale(button_save_texture, BUTTON_SAVE_SIZE) button_save = Button(pygame.rect.Rect(*button_positions['save_button'], *BUTTON_SAVE_SIZE), button_save_texture) button_load_texture = pygame.image.load(LOAD_BUTTON_IMAGE).convert_alpha() button_load_texture = pygame.transform.scale(button_load_texture, BUTTON_SAVE_SIZE) button_load = Button(pygame.rect.Rect(*button_positions['load_button'], *BUTTON_SAVE_SIZE), button_load_texture) # 添加路径控制按钮 button_next_step = Button(pygame.rect.Rect(*button_positions['next_step_button'], *BUTTON_CONTROL_SIZE), None) button_reset_path = Button(pygame.rect.Rect(*button_positions['reset_path_button'], *BUTTON_CONTROL_SIZE), None) button_auto_play = Button(pygame.rect.Rect(*button_positions['auto_play_button'], *BUTTON_CONTROL_SIZE), None) # 提示信息 mes1 = Toast("没有生成迷宫，无法保存", UI_WIDTH, UI_HEIGHT, font=textFont) mes2 = Toast("迷宫已保存", UI_WIDTH, UI_HEIGHT, font=textFont) mes3 = Toast("存档已保存", UI_WIDTH, UI_HEIGHT, font=textFont) mes4 = Toast("存档加载成功", UI_WIDTH, UI_HEIGHT, font=textFont) mes5 = Toast("加载失败", UI_WIDTH, UI_HEIGHT, font=textFont) # 创建存档界面 save_ui = SaveLoadUI(textFont) # 路径控制变量 auto_play = False auto_play_timer = 0 auto_play_interval = AUTO_PLAY_INTERVAL running = True while running: clock.tick(FPS) # 限制帧数 screen.fill(COLOR_WHITE) # 铺底 # 自动播放逻辑 if auto_play and len(maze.full_path) > 0: auto_play_timer += 1 if auto_play_timer >= auto_play_interval: if not maze.next_path_step(): auto_play = False # 路径播放完成后停止自动播放 auto_play_timer = 0 for event in pygame.event.get(): # 首先让存档界面处理事件 save_result = save_ui.handle_event(event, maze) if save_result == "save_success": mes3.show() elif save_result == "load_success": mes4.show() auto_play = False # 加载游戏后停止自动播放 elif save_result == "load_failed": mes5.show() # 如果存档界面正在显示，不处理其他按钮事件 if not save_ui.show_save_list: button_start.handle_event(event=event) button_save.handle_event(event=event) button_load.handle_event(event=event) button_next_step.handle_event(event=event) button_reset_path.handle_event(event=event) button_auto_play.handle_event(event=event) if button_start.pressed == True: maze.generate() auto_play = False # 生成新迷宫时停止自动播放 if button_save.pressed == True: if len(maze.grid) == 0: mes1.show() else: maze.export_to_csv(DEFAULT_MAZE_FILE) mes2.text = f"迷宫已保存至{DEFAULT_MAZE_FILE}" mes2.show() if button_load.pressed == True: save_ui.update_save_list(maze) save_ui.toggle_save_list() # 路径控制 if button_next_step.pressed == True and len(maze.full_path) > 0: maze.next_path_step() if button_reset_path.pressed == True and len(maze.full_path) > 0: maze.reset_path() auto_play = False if button_auto_play.pressed == True and len(maze.full_path) > 0: auto_play = not auto_play auto_play_timer = 0 # 键盘控制 if event.type == pygame.KEYDOWN: if event.key == pygame.K_SPACE and len(maze.full_path) > 0: maze.next_path_step() elif event.key == pygame.K_r and len(maze.full_path) > 0: maze.reset_path() auto_play = False elif event.key == pygame.K_a and len(maze.full_path) > 0: auto_play = not auto_play auto_play_timer = 0 elif event.key == pygame.K_s and pygame.key.get_pressed()[pygame.K_LCTRL]: # Ctrl+S 保存包含路径的CSV if len(maze.grid) > 0: result = maze.save_game() if result: mes3.show() elif event.key == pygame.K_l and pygame.key.get_pressed()[pygame.K_LCTRL]: # Ctrl+L 打开读档界面 save_ui.update_save_list(maze) save_ui.toggle_save_list() if event.type == QUIT: running = False maze.draw(screen=screen, wall_texture=image_wall, coin_texture=image_coin, trap_texture=image_trap) button_start.draw(screen=screen) button_save.draw(screen=screen) button_load.draw(screen=screen) # 绘制路径控制按钮 if len(maze.full_path) > 0: # 绘制按钮背景 pygame.draw.rect(screen, COLOR_GRAY, button_next_step.rect) pygame.draw.rect(screen, COLOR_GRAY, button_reset_path.rect) pygame.draw.rect(screen, COLOR_GREEN if auto_play else COLOR_GRAY, button_auto_play.rect) # 绘制按钮文字 next_text = textFont.render("下一步", True, COLOR_BLACK) reset_text = textFont.render("重置", True, COLOR_BLACK) auto_text = textFont.render("自动播放" if not auto_play else "停止", True, COLOR_BLACK) screen.blit(next_text, (button_next_step.rect.x + 10, button_next_step.rect.y + 15)) screen.blit(reset_text, (button_reset_path.rect.x + 25, button_reset_path.rect.y + 15)) screen.blit(auto_text, (button_auto_play.rect.x + 5, button_auto_play.rect.y + 15)) # 显示当前步数信息 progress_text = textFont.render(f"路径进度: {maze.path_step}/{len(maze.full_path)}", True, COLOR_BLACK) screen.blit(progress_text, button_positions['progress_text']) # 显示操作提示 if len(maze.full_path) > 0: hint_text = textFont.render("空格键: 下一步 | R键: 重置 | A键: 自动播放", True, COLOR_LIGHT_GRAY) screen.blit(hint_text, button_positions['hint_text']) # 显示快捷键提示 shortcut_text = textFont.render("Ctrl+S: 保存包含路径的CSV | Ctrl+L: 读档", True, COLOR_LIGHT_GRAY) screen.blit(shortcut_text, (MAZE_SIZE + 50, 330)) mes1.draw(screen=screen) mes2.draw(screen=screen) mes3.draw(screen=screen) mes4.draw(screen=screen) mes5.draw(screen=screen) # 绘制存档界面（必须在最后绘制以显示在最上层） save_ui.draw(screen) pygame.display.flip() pygame.quit()