/ visualizer.py
visualizer.py
1 import time 2 import pygame 3 4 5 class Visualizer: 6 BLACK = (0, 0, 0) 7 GREEN = (0, 255, 0) 8 SCALE = 16 9 10 def __init__(self): 11 pygame.init() 12 self.gameDisplay = pygame.display.set_mode((56 * Visualizer.SCALE, 56 * Visualizer.SCALE)) 13 14 15 def visualize(self, positions): 16 self.gameDisplay.fill(Visualizer.BLACK) 17 18 for tup in positions: 19 piece, position = tup 20 piece = (Visualizer.SCALE * piece[0], Visualizer.SCALE * piece[1]) 21 position = (Visualizer.SCALE * position[0], Visualizer.SCALE * position[1]) 22 23 border = 5 24 pygame.draw.polygon(self.gameDisplay, Visualizer.GREEN, (position, (position[0] + piece[0], position[1]), (position[0] + piece[0], position[1] + piece[1]), (position[0], position[1] + piece[1]))) 25 pygame.draw.polygon(self.gameDisplay, Visualizer.BLACK, ((position[0] + border, position[1] + border), (position[0] + piece[0] - border, position[1] + border), (position[0] + piece[0] - border, position[1] + piece[1] - border), (position[0] + border, position[1] + piece[1] - border))) 26 27 pygame.display.update() 28 time.sleep(.01)