diff --git a/Interface/grid_draw.py b/Interface/grid_draw.py index 92471f9..0f74e86 100644 --- a/Interface/grid_draw.py +++ b/Interface/grid_draw.py @@ -32,8 +32,6 @@ class GridDraw(): self.screen = pygame.display.set_mode((self.width, self.height)) def start_draw(self): - for event in pygame.event.get(): - if event.type == pygame.QUIT: sys.exit() self.screen.fill(Colors.BLACK) def end_draw(self): diff --git a/Interface/vacuum_render.py b/Interface/vacuum_render.py index 29f1b52..3732c3e 100644 --- a/Interface/vacuum_render.py +++ b/Interface/vacuum_render.py @@ -1,14 +1,31 @@ from Interface.grid_draw import GridDraw, Colors +import sys +import pygame # dummy function def initial_draw(): grid = GridDraw(500, 500) - + tile_width = 500 / 10 + tile_height = 500 / 10 + x = tile_width / 2 + y = tile_height / 2 + radius = 15 while True: grid.start_draw() - grid.board(10, 10) - grid.circle(75, 75, 20, color=Colors.RED) - grid.circle(225, 175, 20, color=Colors.GREEN) - - grid.end_draw() + for event in pygame.event.get(): + if event.type == pygame.QUIT: + sys.exit() + if event.type == pygame.KEYDOWN: + if event.key == pygame.K_LEFT: + x -= tile_width + if event.key == pygame.K_RIGHT: + x += tile_width + if event.key == pygame.K_UP: + y -= tile_height + if event.key == pygame.K_DOWN: + y += tile_height + grid.circle(x, y, 20, color=Colors.RED) + grid.end_draw() + pygame.time.delay(10) +