init.py

import tkinter as tk
from tkinter import font

from constant import (
    Side_window_size, Window_name, Bg_color, Line_color, Side_color, Black,
    Main_size, Plot_size, Offset, Data_radius, Data_color, Change_figure_path,
    Hide_figure_path, Eraser_image, Export_image, Corr_image
)
from image_handler import (
    create_figureimage, slider_image, darken_image, Image, ImageTk,
    create_dynamic_button_images
)
from logic import InteractivePlot, Shape, ButtonVisual, Slider, BasicButton, SwitchVisual


def init() -> tk.Tk:
    """
    Инициализирует основное окно приложения.

    Создает основное окно, холсты для графика и бокового интерфейса,
    настраивает шрифты, фон и элементы управления.

    Returns:
        tk.Tk: Основное окно приложения.
    """
    # Инициализация основного окна
    root: tk.Tk = tk.Tk()
    root.title(Window_name)

    # Создание главного холста для графика
    main_canvas: tk.Canvas = tk.Canvas(
        root,
        width=Main_size[0],
        height=Main_size[1],
        bg=Bg_color,
        bd=0,
        highlightthickness=0
    )
    main_canvas.pack()

    # Холст для графика внутри главного холста
    plot_canvas: tk.Canvas = tk.Canvas(
        main_canvas,
        width=Plot_size[0] - 1,
        height=Plot_size[1] - 1,
        bg=Bg_color,
        bd=0,
        relief="solid",
        highlightthickness=1,
        highlightbackground=Black
    )
    plot_canvas.place(x=Offset[0], y=Offset[1])

    # Холст для бокового (побочного) окна
    side_window: tk.Canvas = tk.Canvas(
        main_canvas,
        width=Side_window_size[0],
        height=Side_window_size[1],
        bg=Side_color,
        bd=0,
        highlightthickness=0
    )
    side_window.place(x=0, y=Main_size[1] - Side_window_size[1])

    # Шрифт для текстовых элементов
    font_style: tk.font.Font = font.Font(family="Arial", size=12)

    # Инициализация фона и разметки графика
    init_bg(main_canvas, plot_canvas, font_style)

    # Метка для отображения координат (расположена вне бокового окна)
    coord_label: tk.Label = tk.Label(side_window, text="", font=font.Font(family="Arial", size=11), bg=Side_color)

    # Метка для отображения статистики
    stats_label: tk.Label = tk.Label(main_canvas, text="", font=font.Font(family="Arial", size=14), bg=Bg_color)

    # Создание объекта формы для отображения данных на графике
    shape: Shape = Shape(plot_canvas, Data_radius, Data_color)

    # Инициализация интерактивного графика
    plot: InteractivePlot = InteractivePlot(
        plot_canvas, coord_label, stats_label, shape,
        Plot_size[0] / 10, Plot_size
    )

    # Инициализация бокового интерфейса (слайдеры, кнопки и т.д.)
    init_side_interface(side_window, plot, font_style)
    return root


def init_bg(main_canvas: tk.Canvas, plot_canvas: tk.Canvas, font_style: tk.font.Font) -> None:
    """
    Инициализирует фон и базовую разметку графика.

    Рисует центральные вертикальную и горизонтальную линии на холсте графика,
    затем запускает инициализацию шкал и текстовых подписей.

    Args:
        main_canvas (tk.Canvas): Основной холст приложения.
        plot_canvas (tk.Canvas): Холст, на котором рисуется график.
        font_style (tk.font.Font): Шрифт для текстовых элементов.
    """
    # Вычисление половины размеров графика для центрирования осей.
    half_size_plot = Plot_size // 2
    scale_lines_width: int = Plot_size[0] // 1000  # Ширина линий шкалы пропорциональна размеру графика

    # Вертикальная центральная линия
    plot_canvas.create_line(
        half_size_plot[0], 0,
        half_size_plot[0], Plot_size[1],
        fill=Line_color,
        width=scale_lines_width
    )
    # Горизонтальная центральная линия
    plot_canvas.create_line(
        0, half_size_plot[1],
        Plot_size[0], half_size_plot[1],
        fill=Line_color,
        width=scale_lines_width
    )

    # Инициализация подписей и шкал
    init_scale_text(main_canvas, font_style)


def init_side_interface(side_window: tk.Canvas, plot: InteractivePlot, font_style: tk.font.Font) -> None:
    """
    Инициализирует элементы бокового интерфейса.

    Создает слайдеры и кнопки, располагает их на боковом холсте и связывает с функционалом графика.

    Args:
        side_window (tk.Canvas): Холст для бокового интерфейса.
        plot (InteractivePlot): Интерактивный график, содержащий методы управления.
        font_style (tk.font.Font): Шрифт для текстовых элементов.
    """
    # Инициализация слайдеров
    init_sliders(side_window, plot, font_style)

    # Определение размера кнопок относительно высоты бокового окна
    buttons_size = (int(Side_window_size[1] / 2.5),) * 2
    button_offset = (-1, 2)  # Смещение для изображения кнопок

    # Базовые координаты для расположения кнопок
    base_coords = (int(buttons_size[0] * 0.65),) * 2
    base_coords_2 = (base_coords[0] * 2.75, Side_window_size[1] - base_coords[1])

    # Пути к изображениям для кнопок
    images_path = [
        Change_figure_path, Hide_figure_path,
        Eraser_image, Corr_image, Export_image
    ]
    # Координаты для размещения кнопок
    buttons_coords = [
        base_coords,
        (base_coords_2[0], base_coords[1]),
        (base_coords[0], base_coords_2[1]),
        (base_coords[0] * 4.6, base_coords[1] * 2),
        base_coords_2
    ]
    # Функции, вызываемые при нажатии на кнопки
    funcs = [
        plot.generate_region.switch_figure,
        plot.generate_region.switch_dispersion,
        plot.generate_region.switch_eraser,
        plot.generate_region.switch_correlation,
        plot.generate_region.pixel_array.export,
    ]
    # Тип визуализации для кнопок: первые четыре используют SwitchVisual, пятая — ButtonVisual
    buttons_type = [SwitchVisual] * 4 + [ButtonVisual]

    # Создание кнопок с изображениями и привязка соответствующих функций
    for image, coords, func, button_type in zip(images_path, buttons_coords, funcs, buttons_type):
        change_figure_image = Image.open(image).resize(buttons_size)
        # Создание набора изображений для кнопки (обычное, затемненное, при нажатии)
        change_figure_images_3d = create_dynamic_button_images(change_figure_image, *button_offset)
        button_figure_visual = button_type(*change_figure_images_3d, button_offset)
        BasicButton(
            side_window,
            coords[0],
            coords[1],
            change_figure_images_3d[0],
            button_figure_visual,
            func,
            'center'
        )


def init_sliders(side_window: tk.Canvas, plot: InteractivePlot, font_style: tk.font.Font) -> None:
    """
    Создает и размещает слайдеры на боковом холсте.

    Настраивает фон слайдера, кнопку слайдера и текстовые метки,
    а также связывает слайдеры с соответствующими методами графика.

    Args:
        side_window (tk.Canvas): Холст для бокового интерфейса.
        plot (InteractivePlot): Интерактивный график, содержащий методы управления.
        font_style (tk.font.Font): Шрифт для текстовых меток.
    """
    slider_x: int = Side_window_size[0] // 2
    # Вертикальные позиции слайдеров: 30% и 70% от высоты бокового окна
    sliders_y = (int(Side_window_size[1] * 0.3), int(Side_window_size[1] * 0.7))
    slider_width: int = int(Side_window_size[0] * 0.5)
    slider_height: int = int(Side_window_size[1] * 0.2)
    text_x: float = slider_width * 1.62

    # Создание фонового изображения для слайдера
    slider_background = slider_image(Data_color, Bg_color, slider_width * 2, slider_height)
    # Создание изображения кнопки слайдера; параметр "circle" задаёт форму
    slider_button_img = create_figureimage(
        "circle",
        Image.new("RGBA", (slider_height, slider_height), Line_color)
    )
    # Генерация трех состояний кнопки: стандартное, затемненное и при нажатии
    slider_button, darker_button, pushed_button = map(
        ImageTk.PhotoImage,
        (
            slider_button_img,
            darken_image(slider_button_img, 0.8),
            darken_image(slider_button_img, 0.5)
        )
    )
    slider_visual = ButtonVisual(slider_button, darker_button, pushed_button)

    # Тексты для подписей слайдеров
    texts = ("Dispersion", "Intensive")
    # Функции, привязанные к слайдерам
    funcs = [
        plot.generate_region.set_dispersion,
        plot.generate_region.set_intensive
    ]

    # Создание каждого слайдера с привязкой к функции и размещение текстовой подписи
    for slider_y_val, text, func in zip(sliders_y, texts, funcs):
        Slider(
            side_window,
            slider_background,
            Side_color,
            slider_x,
            slider_y_val,
            slider_button,
            slider_visual,
            func,
            anchor="center"
        )

        side_window.create_text(
            text_x,
            slider_y_val,
            text=text,
            anchor="center",
            font=font_style
        )


def init_scale_text(main_canvas: tk.Canvas, font_style: tk.font.Font) -> None:
    """
    Инициализирует шкалы и текстовые подписи на основном холсте.

    Создает линии и подписи для горизонтальной и вертикальной шкал,
    отображающих значения от 0 до 1.

    Args:
        main_canvas (tk.Canvas): Основной холст приложения.
        font_style (tk.font.Font): Шрифт для подписей.
    """
    half_scale: int = Offset[0] // 2

    # Координаты для вертикальной шкалы
    vert_x: int = half_scale
    vert_line_start_x: float = half_scale * 1.75
    vert_line_end_x: int = Offset[1]

    # Координаты для горизонтальной шкалы
    horiz_y: float = Plot_size[1] + Offset[1] * 1.5
    horiz_line_y_start: float = Plot_size[1] + Offset[1] * 1.25
    horiz_line_y_end: float = Plot_size[1] + Offset[1]

    # Создание линий и текстовых меток для значений от 0 до 1
    for number in (0, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1):
        # Вычисление позиций для горизонтальной шкалы
        horiz_x: int = Offset[1] + int(number * Plot_size[1])
        vert_y: int = Offset[1] + int(number * Plot_size[1])

        # Рисование линии и подписи для горизонтальной шкалы
        main_canvas.create_line(
            horiz_x, horiz_line_y_start,
            horiz_x, horiz_line_y_end,
            fill=Black,
            width=Plot_size[1] // 1000
        )
        main_canvas.create_text(
            horiz_x, horiz_y,
            text=str(number),
            font=font_style,
            fill=Black
        )

        # Рисование линии и подписи для вертикальной шкалы
        main_canvas.create_line(
            vert_line_start_x, vert_y,
            vert_line_end_x, vert_y,
            fill=Black,
            width=Plot_size[1] // 1000
        )
        main_canvas.create_text(
            vert_x, vert_y,
            text=str(round(1 - number, 1)),
            font=font_style,
            fill=Black
        )