Machine_learning_2023/Interface/vacuum_render.py

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)
        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)