Machine_learning_2023/Interface/grid_draw.py

import sys
import pygame


class Colors:
    BLACK = 0, 0, 0
    WHITE = 255, 255, 255
    RED = 255, 0, 0
    GREEN = 0, 255, 0


DEFAULT_COLOR = Colors.WHITE
RADIUS_SIZE_COEFFICIENT = 3


def default_color(func):
    def wrap(*args, **kwargs):
        if "color" not in kwargs:
            kwargs["color"] = DEFAULT_COLOR
        result = func(*args, **kwargs)
        return result

    return wrap


class GridDraw:
    def __init__(self, width=None, height=None, background=None):
        self.width = width if width != None else 100
        self.height = height if height != None else 100
        self.background = background if background != None else Colors.BLACK

        pygame.init()
        self.screen = pygame.display.set_mode((self.width, self.height))

    def start_draw(self):
        self.screen.fill(Colors.BLACK)

    def end_draw(self):
        pygame.display.flip()

    @default_color
    def line(self, x_1, y_1, x_2, y_2, color=None):
        pygame.draw.line(self.screen, color, (x_1, y_1), (x_2, y_2))

    @default_color
    def board(self, tiles_x, tiles_y, color=None):
        tiles_width = self.width / tiles_x
        tiles_height = self.height / tiles_y

        for i in range(1, tiles_x):
            self.line(tiles_width * i, 0, tiles_width * i, self.height, color=color)

        for i in range(1, tiles_y):
            self.line(0, tiles_height * i, self.width, tiles_height * i, color=color)

    @default_color
    def circle(self, x, y, tile_height, color=None):
        radius = tile_height / RADIUS_SIZE_COEFFICIENT
        pygame.draw.circle(self.screen, color, (x, y), radius)