window.py

from tkinter import Tk, BOTH, Canvas

from line import Line

from utils import BACKGROUND_COLOR, LINE_COLOR

class Window:
    """
    width and height are the size of the new window in pixels.
    title is the self.__root title.
    """
    def __init__(self, width: int, height: int, title: str) -> None:
        self.__root = Tk()
        self.__root.title(title)
        self.__root.protocol("WM_DELETE_WINDOW", self.close)

        self.__canvas = Canvas(self.__root, bg=BACKGROUND_COLOR, width=width, height=height)
        self.__canvas.pack()

        self.__is_window_running = False


    def redraw(self) -> None:
        """
        call self.__root.update_idletasks() and self.__root.update()
        to redraw the screen.
        """
        self.__root.update_idletasks()
        self.__root.update()
    

    def wait_for_close(self) -> None:
        """
        set self.__is_window_running to True
        call self.redraw() until self.__is_window_running is False
        """
        self.__is_window_running = True
        while self.__is_window_running is True:
            self.redraw()


    def close(self) -> None:
        """
        set self.__is_window_running to False
        """
        self.__is_window_running = False


    def draw_line(self, line: Line, fill_color: str = LINE_COLOR) -> None:
        """
        Draw a line between two Point's in a Canvas.

        line: Line class
        fill_color: TK Color (name of the color or HEX #rrggbb value)
        """
        line.draw(self.__canvas, fill_color)