Machine_learning_2023/Interface/helpers.py

import pygame

BLACK = 0, 0, 0
WHITE = 255, 255, 255


def rect(screen, color, position_x, position_y, size_x, size_y):
    pygame.draw.rect(screen, color, (position_x, position_y, size_x, size_y))


def line(screen, color, x_1, y_1, x_2, y_2):
    pygame.draw.line(screen, color, (x_1, y_1), (x_2, y_2))


def create_screen(size):
    return pygame.display.set_mode(size)


def draw_board(screen, color, width, height, tiles_x, tiles_y):
    tiles_width = width / tiles_x
    tiles_height = height / tiles_y

    temp_x = tiles_width
    for i in range(tiles_x-1):
        line(screen, color, temp_x, 0, temp_x, height)
        temp_x += tiles_width

    temp_y = tiles_height
    for i in range(tiles_y-1):
        line(screen, color, 0, temp_y, width, temp_y)
        temp_y += tiles_height