• Python Prim 算法 生成迷宫 通过Pygame可视化

    前两篇文章，我们使用Prim算法生成了文章，这一次，我们要用Pygame将迷宫的完整生成过程可视化

    首先导入模块，这里我们要用到pygame，随机库，sys，time用于减缓速度，threading用于生成迷宫的另一个线程

    1. import pygame
    2. from pygame.locals import *
    3. import random as rd
    4. import sys
    5. import time
    6. import threading

    我们用类和对象的形式实现，定义Game类，初始化函数中设置窗口，a和b表示迷宫大小，lineWidth是迷宫墙壁画在窗口中的粗细，创建一个线程，指向createMaze（createMaze的代码下方会补充），然后启动线程

    1. class Game:
    2.     def __init__(self):
    3.         pygame.init()
    4.         self.W,self.H=900,900
    5.         self.screen=pygame.display.set_mode((self.W,self.H))
    6.  
    7.         self.a,self.b=28,39
    8.         self.lineWidth=5
    9.         thread=threading.Thread(target=self.createMaze,args=(self.a,self.b))
    10.         thread.start()

    创建listen函数，用于监听事件

    1.     def listen(self):
    2.         for event in pygame.event.get():
    3.             if event.type==QUIT:
    4.                 sys.exit()

    接下来是创建迷宫的代码，在上一篇遍历墙生成迷宫的时候，我们已经将代码完整写出来了，只要修改一小部分内容即可，要修改的地方如下：

    在函数参数中添加self

    def createMaze(self,a,b):

    然后，把函数中的maze全部改为self.maze，直接存入类变量中

    这里不再展示代码，需要算法代码的可以查看我的上一篇文章：

    Python Prim 算法 生成迷宫_Leleprogrammer的博客-CSDN博客Python Prim算法 通过遍历墙来生成迷宫，快来看看吧！https://blog.csdn.net/leleprogrammer/article/details/125472436?spm=1001.2014.3001.5501或者是在文末的最终代码中复制，要学习算法原理的，查看上一篇文章即可

    接下来，draw函数，用于绘制迷宫，这里代码不会太难，就不做太多讲解

    1.     def draw(self):
    2.         self.screen.fill((255,255,255))
    3.         for (n,face),is_wall in self.maze.items():
    4.             if is_wall:
    5.                 y=n//self.a
    6.                 x=n%self.a
    7.                 absX=x*self.width
    8.                 absY=y*self.height
    9.                 if face=="u":
    10.                     pygame.draw.line(self.screen,(0,0,0),(absX,absY),(absX+self.width,absY),self.lineWidth)
    11.                 if face=="d":
    12.                     pygame.draw.line(self.screen,(0,0,0),(absX,absY+self.height),(absX+self.width,absY+self.height),self.lineWidth)
    13.                 if face=="l":
    14.                     pygame.draw.line(self.screen,(0,0,0),(absX,absY),(absX,absY+self.height),self.lineWidth)
    15.                 if face=="r":
    16.                     pygame.draw.line(self.screen,(0,0,0),(absX+self.width,absY),(absX+self.width,absY+self.height),self.lineWidth)

    然后是run函数，主循环

    1.     def run(self):
    2.         while True:
    3.             self.listen()
    4.             self.draw()
    5.             self.refreshWindow()
    6.             pygame.display.update()

    用于显示文字的函数，这里没有用到，可以不加上，这里加上去是为了后续利用此框架编写游戏，这里还是把这个显示文字的函数给大家

    1.     @staticmethod
    2.     def print_text(name,size,text,color):
    3.         font=pygame.font.SysFont(name,size)
    4.         image=font.render(text,True,color)
    5.         return image

    最后，添加这段启动的代码

    1. if __name__ == '__main__':
    2.     game=Game()
    3.     game.run()

    这样就好啦！

    最终代码（可供参考）

    1. import pygame
    2. from pygame.locals import *
    3. import random as rd
    4. import sys
    5. import time
    6. import threading
    7.  
    8. class Game:
    9.     def __init__(self):
    10.         pygame.init()
    11.         self.W,self.H=900,900
    12.         self.screen=pygame.display.set_mode((self.W,self.H))
    13.         pygame.display.set_caption("Find the Way")
    14.  
    15.         self.a,self.b=28,39
    16.         self.lineWidth=5
    17.         thread=threading.Thread(target=self.createMaze,args=(self.a,self.b))
    18.         thread.start()
    19.  
    20.     def listen(self):
    21.         for event in pygame.event.get():
    22.             if event.type==QUIT:
    23.                 sys.exit()
    24.  
    25.     def createMaze(self,a,b):
    26.         self.maze={}
    27.         for n in range(a*b):
    28.             for face in ["u","d","l","r"]:
    29.                 self.maze[(n,face)]=1
    30.         history=[]
    31.         walls=[]
    32.         block=rd.choice(list(self.maze.keys()))[0]
    33.         history.append(block)
    34.         for face in ["u","d","l","r"]:
    35.             walls.append((block,face))
    36.         while len(walls)!=0:
    37.             time.sleep(0.001)
    38.             wall=rd.choice(walls)
    39.             twoBlocks=[wall[0]]
    40.             faces=[wall[1]]
    41.             if wall[1]=="u":
    42.                 if wall[0]-a<0:
    43.                     twoBlocks.append(None)
    44.                 else:
    45.                     twoBlocks.append(wall[0]-a)
    46.                     faces.append("d")
    47.             elif wall[1]=="r":
    48.                 if (wall[0]+1)%a!=0:
    49.                     twoBlocks.append(wall[0]+1)
    50.                     faces.append("l")
    51.                 else:
    52.                     twoBlocks.append(None)
    53.             elif wall[1]=="l":
    54.                 if wall[0]%a!=0:
    55.                     twoBlocks.append(wall[0]-1)
    56.                     faces.append("r")
    57.                 else:
    58.                     twoBlocks.append(None)
    59.             elif wall[1]=="d":
    60.                 if wall[0]+a>len(self.maze)/4-1:
    61.                     twoBlocks.append(None)
    62.                 else:
    63.                     twoBlocks.append(wall[0]+a)
    64.                     faces.append("u")
    65.             ins=[]
    66.             infaces=[]
    67.             for i,oneBlock in enumerate(twoBlocks):
    68.                 if oneBlock in history:
    69.                     ins.append(oneBlock)
    70.                     infaces.append(faces[i])
    71.             if len(ins)==1:
    72.                 mirrorFace=None
    73.                 if infaces[0]=="u":
    74.                     mirrorFace="d"
    75.                 elif infaces[0]=="d":
    76.                     mirrorFace="u"
    77.                 elif infaces[0]=="r":
    78.                     mirrorFace="l"
    79.                 elif infaces[0]=="l":
    80.                     mirrorFace="r"
    81.                 if not (None in twoBlocks):
    82.                     self.maze[(ins[0],infaces[0])]=0
    83.                     other=None
    84.                     if ins[0]==twoBlocks[0]:
    85.                         other=twoBlocks[1]
    86.                     else:
    87.                         other=twoBlocks[0]
    88.                     self.maze[(other,mirrorFace)]=0
    89.                     walls.remove(wall)
    90.                     history.append(other)
    91.                     for face in ["u","l","r","d"]:
    92.                         if self.maze.get((other,face))==1 and not ((other,face) in walls):
    93.                             walls.append((other,face))
    94.                 else:
    95.                     walls.remove(wall)
    96.                     continue
    97.             elif len(ins)==2:
    98.                 walls.remove(wall)
    99.  
    100.     def draw(self):
    101.         self.screen.fill((255,255,255))
    102.         for (n,face),is_wall in self.maze.items():
    103.             if is_wall:
    104.                 y=n//self.a
    105.                 x=n%self.a
    106.                 absX=x*self.width
    107.                 absY=y*self.height
    108.                 if face=="u":
    109.                     pygame.draw.line(self.screen,(0,0,0),(absX,absY),(absX+self.width,absY),self.lineWidth)
    110.                 if face=="d":
    111.                     pygame.draw.line(self.screen,(0,0,0),(absX,absY+self.height),(absX+self.width,absY+self.height),self.lineWidth)
    112.                 if face=="l":
    113.                     pygame.draw.line(self.screen,(0,0,0),(absX,absY),(absX,absY+self.height),self.lineWidth)
    114.                 if face=="r":
    115.                     pygame.draw.line(self.screen,(0,0,0),(absX+self.width,absY),(absX+self.width,absY+self.height),self.lineWidth)
    116.  
    117.     def run(self):
    118.         while True:
    119.             self.listen()
    120.             self.draw()
    121.             self.refreshWindow()
    122.             pygame.display.update()
    123.  
    124. if __name__ == '__main__':
    125.     game=Game()
    126.     game.run()

    本次的教学到这里也就结束啦！喜欢我的文章的，别忘了多多点赞关注支持哦！谢谢~

  • 相关阅读:
    模拟卷Leetcode【普通】341. 扁平化嵌套列表迭代器
    JUC_Volatile简述
    adb对安卓app进行抓包（ip连接设备）
    Maven笔记---超详细
    开源的容器运行时项目 Podman
    配置git-ssh
    在CentOS7下安装Oracle11教程
    算法3：链表实现队列
    gwas数据根据eaf Z 和N 求beta和se
    活动预约报名小程序已优化点与待优化点
  • 原文地址：https://blog.csdn.net/leleprogrammer/article/details/125472897