import numpy from PIL import Image import pygame from game_board import GameBoard import argparse def view_img(image=""): pygame.init() game_board = GameBoard() running = True clock = pygame.time.Clock() pause = True image = Image.open('img/' + image + ".png") new_size = 400 resized_image = image.resize((new_size, new_size)) gray_image = resized_image.convert('L') data = numpy.asarray(gray_image) threshold = 100 binary_array = (data > threshold).astype(int) game_board.tiles = binary_array while running: clock.tick(30) for event in pygame.event.get(): if event.type == pygame.QUIT: running = False if event.type == pygame.KEYDOWN: if event.key == pygame.K_x: game_board.reset() elif event.key == pygame.K_SPACE: pause = not pause print(pause) elif event.key == pygame.K_r: pause = True print(pause) game_board.tiles = binary_array if pause: if event.type == pygame.MOUSEBUTTONDOWN: if event.button == 1: game_board.paint_tile(*event.pos) elif event.button == 3: game_board.remove_tile(*event.pos) game_board.draw_window(pause) pygame.quit() if __name__ == "__main__": parser = argparse.ArgumentParser("images") parser.add_argument("image", help="which image to pick from the directory") args = parser.parse_args() view_img(args.image)