InteligentnySaper/classes/system.py

import pygame


class Window:
    window: None
    width: int
    height: int
    title: str
    icon_path: str
    paused: bool
    pause_menu: None

    search: pygame.Surface

    def __init__(self, width: int = 640, height: int = 480, title="", icon_path=""):
        self.set_resolution(width, height)
        self.set_title(title)
        self.set_icon(icon_path)
        self.mount()
        self.paused = False

        self.pause_menu = pygame.Surface((width, height))
        self.pause_menu.set_alpha(128)
        self.pause_menu.fill((0, 0, 0))
        self.search = pygame.Surface((width, height), flags=pygame.SRCALPHA)

    def set_resolution(self, width: int, height: int):
        self.width = width
        self.height = height

    def set_title(self, title: str):
        self.title = title

    def set_icon(self, icon_path: str):
        self.icon_path = icon_path

    def mount(self):
        self.window = pygame.display.set_mode((self.width, self.height))
        pygame.display.set_caption(self.title)
        if self.icon_path != "":
            icon = pygame.image.load(self.icon_path)
            pygame.display.set_icon(icon)

    def pause(self, paused: bool):
        self.paused = paused
        if self.paused:
            self.window.blit(self.pause_menu, (0, 0))
            pygame.display.update()

    def draw_search(self, pos1: list = [0, 0], pos2: list = [0, 0], tile_size: int = 64, map=None, saper=None):
        map.draw_tiles()
        map.draw_objects()
        saper.draw(self.window, 0.1)
        self.window.blit(self.pause_menu, (0, 0))
        self.search = pygame.Surface((self.width, self.height), flags=pygame.SRCALPHA)
        pos1 = [pos1[0] * tile_size + (tile_size / 2), pos1[1] * tile_size + (tile_size / 2)]
        pos2 = [pos2[0] * tile_size + (tile_size / 2), pos2[1] * tile_size + (tile_size / 2)]
        pygame.draw.line(self.search, pygame.Color(255, 0, 0), pos1, pos2, 5)

        n = 10
        p1 = 0
        p2 = 0
        p3 = 0
        if (pos2[0] - pos1[0]) != 0:
            a1 = (pos2[1] - pos1[1]) / (pos2[0] - pos1[0])
            b1 = (pos1[1] - a1 * pos1[0])
            if a1 != 0:
                a2 = -(1 / a1)
                b2 = pos2[1] - a2 * pos2[0]
                y = a2 * (pos2[0] + n) + b2
                p1 = (pos2[0] + n, y)
                y = a2 * (pos2[0] - n) + b2
                p2 = (pos2[0] - n, y)
                if pos1[0] > pos2[0]:
                    y = a1 * (pos2[0] - n) + b1
                    p3 = (pos2[0] - n, y)
                else:
                    y = a1 * (pos2[0] + n) + b1
                    p3 = (pos2[0] + n, y)
            else:
                p1 = (pos2[0], pos2[1] + n)
                p2 = (pos2[0], pos2[1] - n)
                if pos1[0] > pos2[0]:
                    p3 = (pos2[0] - n, pos2[1])
                else:
                    p3 = (pos2[0] + n, pos2[1])
        else:
            p1 = (pos2[0] - n, pos2[1])
            p2 = (pos2[0] + n, pos2[1])
            if pos1[1] > pos2[1]:
                p3 = (pos2[0], pos2[1] - n)
            else:
                p3 = (pos2[0], pos2[1] + n)

        pygame.draw.polygon(self.search, pygame.Color(255, 0, 0), (p1, p2, p3))
        self.window.blit(self.search, (0, 0))
        pygame.display.update()