Flood Fill算法,工作几秒钟,然后出错。 (pygame的)

时间:2017-04-01 16:22:38

标签: python pygame flood-fill

尝试在pygame中实现泛洪填充算法,就像学习经验一样。我认为我有一个良好的开端,它主要是功能性的,但经过几秒钟的正常工作后,它给了我一个最大的递归深度误差。

import pygame, random, math
from pygame.locals import *

class GameMain():
done = False
color_bg = Color('white')

def __init__(self, width=800, height=800):
    pygame.init()

    self.width, self.height = width, height
    self.screen = pygame.display.set_mode((self.width, self.height))
    self.clock = pygame.time.Clock()

def main_loop(self):
    while not self.done:
        self.handle_events()
        self.draw()
        self.clock.tick(60)
    pygame.quit()

def draw(self):
    self.screen.fill(self.color_bg)
    pygame.draw.rect(self.screen,Color("grey30"), [100,100,400,300],2)
    pygame.draw.rect(self.screen,Color("grey30"), [ 300,300,400,300],2)
    pygame.display.flip()

def handle_events(self):
    events = pygame.event.get()

    # keystates
    keys = pygame.key.get_pressed()

    # events
    for event in events:
        if event.type == pygame.QUIT:
            self.done = True
        if event.type == KEYDOWN:
            if event.key == K_ESCAPE:
                self.done = True
        if event.type == MOUSEBUTTONDOWN:
            x,y = pygame.mouse.get_pos()
            coord = [x,y]
            self.flood_fill(self.screen,coord)
def flood_fill(self,screen,coord):

    if screen.get_at((coord[0],coord[1])) == Color("grey30"):
        return
    screen.set_at((coord[0],coord[1]),Color("grey30"))
    pygame.display.flip()
    self.flood_fill(self.screen, [coord[0] +1, coord[1]])
    self.flood_fill(self.screen, [coord[0] -1, coord[1]])
    self.flood_fill(self.screen, [coord[0], coord[1]+1])
    self.flood_fill(self.screen, [coord[0], coord[1]-1])


if __name__ == "__main__":
game = GameMain()
game.main_loop()

我的程序运行了一两秒,并改变了几个x,y坐标的颜色,但随后它变得疯狂,我得到一个递归深度错误。不知道为什么它会工作一秒钟然后失败。

1 个答案:

答案 0 :(得分:0)

Python不是特别设计或倾向于递归操作。在CPython实现中,递归深度限制为 1000

您可以通过以下方式查看:

import sys
print(sys.getrecursionlimit())

因此,您可以使用sys.setrecursionlimit()来覆盖限制。

但是,这并不是特别安全/最好的选择。

最好重写代码以实现迭代实现。

编辑: 道歉,我意识到我没有指出你应该在哪里看代码。

递归实际上发生在flood_fill()方法中,因为它实际上再次调用self.floor_fill()四(4)次。

有关详细信息,请阅读:http://seriously.dontusethiscode.com/2013/04/14/setrecursionlimit.html