алгоритм заполнения наводнений в python

Question

Я пытаюсь заполнить цвет в многоугольнике, используя рекурсивный алгоритм наводнения. Я кодирую в python 2.7. Я иду на полпути к заполнению, затем падает. Это происходит все время, когда я запускаю код. Есть ли что-то неправильно с кодом, или это просто проблема в моем ноутбуке. Пожалуйста помоги? Спасибо. Вот код.

import pygame 
    import numpy as np 
    from pygame.locals import * 
    import sys 
    def flood_fill(name, x, y, color): 
     get = name.get_at((x, y)) 
     if get == (0, 0, 0, 255): 
      screen.set_at((x, y), red) 
      pygame.display.flip() 
      flood_fill(name, x + 1, y, color) 
      flood_fill(name, x - 1, y, color) 
      flood_fill(name, x, y + 1, color) 
      flood_fill(name, x, y - 1, color) 

    if __name__ == '__main__': 
     pygame.init() 
     sys.setrecursionlimit(1500000000) 
     width = 640 
     height = 480 
     resolution = (width, height) 
     screen = pygame.display.set_mode(resolution) 
     black = (0, 0, 0) 
     white = (255, 255, 255) 
     red = (255, 0, 0) 
     pygame.draw.rect(screen, white, [20, 20, 250, 100], 2) 
     flood_fill(screen, 100, 50, red) 
     while True: 
      for event in pygame.event.get(): 
       if event.type == QUIT: 
        pygame.quit() 
        sys.exit() 
      pygame.display.flip() 

Answer 1

Вы должны предоставить ошибку, что вы получаете, но на основе кода, я думаю, что ошибка является IndexError, и вы должны проверить, если пиксельные координаты являются действительными координатами т.е. они всегда находятся внутри области из поверхность, которая находится в вашем случае screen.

Согласно documentation: If the pixel position is outside the area of the Surface an IndexError exception will be raised.

Просто убедитесь, что get_at метод получает действительные параметры или окружать его с try\except блока.

Answer 2

Довольно уверен, что ваш предел рекурсии является проблемой. Все setrecursionlimit действительно устраняет предупреждения, которые помешали бы вам делать глупые вещи; он не волшебным образом создает транзисторы в вашем CPU для хранения стека вызовов 1500000000.

Реализация заливки в стеке, как правило, является плохим. Вместо этого используйте стекную структуру данных стека.