SCREEN_WIDTH = COLUMS * SIDE
SCREEN_HEIGHT = ROWS * SIDE
FOUR_NEIGH = {pygame.K_UP: (-1, 0), pygame.K_LEFT: (0, -1), pygame.K_RIGHT: (0, 1), pygame.K_DOWN: (1, 0)}
DIRE = {pygame.K_LEFT: pygame.K_RIGHT, pygame.K_RIGHT: pygame.K_LEFT,
pygame.K_UP: pygame.K_DOWN, pygame.K_DOWN: pygame.K_UP}
class Cell(object):
def init(self, screen, rect):
self.screen = screen
self.rect = rect
self.walls = {key: False for key in FOUR_NEIGH}
self.visited = 0
self.type = ‘nothing’
def draw(self):
if self.type == ‘end’:
self.screen.fill(0x11ff11, self.rect)
elif self.type == ‘body’:
pygame.draw.circle(self.screen, 0xff1122, self.rect.center, (SIDE - 4) // 2)
if not self.walls[pygame.K_LEFT]:
start = (self.rect.left, self.rect.top)
end = (self.rect.left, self.rect.top + self.r