App.py

u"""Файл с классом приложения и необходимыми для него классами"""
from tkinter import Button, HORIZONTAL, Menu, Scale, Tk, Toplevel, Text, END, Label

from Paint import Paint
from Dict import Dict

START_CANVAS_SIZE = 700
START_FIGURE_SIZE = 10
START_SYMMETRY_NUMBER = 8
_DEBUG = True


class FigSizer(Toplevel):
    u"""Окно, которое открывается для выбора размера фигуры"""
    def __init__(self, default_size=START_FIGURE_SIZE):
        Toplevel.__init__(self)
        self.figsize = Scale(self, from_=1, to=50, orient=HORIZONTAL)
        self.figsize.set(default_size)
        self.figsize.pack()
        self.label = Label(self, text=Dict['size_window_text'])
        self.label.pack()
        self.button = Button(self, text=Dict['size_window_button'], command=self.quit)
        self.button.pack()
        self.title(Dict['size_window_title'])
        self.protocol('WM_DELETE_WINDOW', self.quit)
        # по центру экрана
        self.update()
        x_center = (self.winfo_screenwidth() - self.winfo_width()) / 2
        y_center = (self.winfo_screenheight() - self.winfo_height()) / 2
        self.wm_geometry('+%d+%d' % (x_center, y_center))


class NumSymmetry(Toplevel):
    u"""Окошко выбора числа симметричных отражений"""
    def __init__(self, default_num_symm=START_SYMMETRY_NUMBER):
        Toplevel.__init__(self)
        self.num_symm = Scale(self, from_=-8, to=16,
                              orient=HORIZONTAL)
        self.num_symm.set(default_num_symm)
        self.num_symm.pack()
        self.label = Label(self, text=Dict['symm_window_text'])
        self.label.pack()
        self.button = Button(self, text=Dict['symm_window_button'], command=self.quit)
        self.button.pack()
        self.title(Dict['symm_window_title'])
        self.protocol('WM_DELETE_WINDOW', self.quit)
        # по центру экрана
        self.update()
        x_center = (self.winfo_screenwidth() - self.winfo_width()) / 2
        y_center = (self.winfo_screenheight() - self.winfo_height()) / 2
        self.wm_geometry('+%d+%d' % (x_center, y_center))


class HelpWindow(Toplevel):
    u"""Окно с помощью"""
    def __init__(self):
        with open(Dict["help_filename"], encoding="utf8") as in_file:
            help_text = in_file.read()
        Toplevel.__init__(self)
        self.text_field = Text(self, height=30, width=100)
        self.text_field.pack()
        self.text_field.insert(END, help_text)
        self.title(Dict['help_window_title'])
        self.protocol('WM_DELETE_WINDOW', self.quit)
        self.update()
        x_center = (self.winfo_screenwidth() - self.winfo_width()) / 2
        y_center = (self.winfo_screenheight() - self.winfo_height()) / 2
        self.wm_geometry('+%d+%d' % (x_center, y_center))


class App(Tk):
    u"""Главный класс приложения."""

    def __init__(self):
        u"""Создание холста и запуск цикла отрисовки"""
        super(App, self).__init__()
        self.geometry('{y}x{y}'.format(y=START_CANVAS_SIZE))
        self.title(Dict['app_title'])
        # создаем сам холст и помещаем его в окно
        self.canv = Paint(self, bg='white')
        self.canv.grid(row=0, column=0, sticky='wens')
        # чтобы занимал все окно
        self.columnconfigure(index=0, weight=1)
        self.rowconfigure(index=0, weight=1)

        # добавляем главное меню
        self.main_menu = Menu(self)

        # здесь несколько вложенных меню для разных параметров
        self.param_menu = Menu(self.main_menu)

        # меню выбора фигуры
        self.brush_style = Menu(self.param_menu)
        self.brush_style.add_command(label=Dict['brush_style_circle'],
                                command=lambda: self.canv.set_style('circle'))
        self.brush_style.add_command(label=Dict['brush_style_square'],
                                command=lambda: self.canv.set_style('square'))
        self.brush_style.add_command(label=Dict['brush_style_triangle'],
                                command=lambda: self.canv.set_style('triangle'))

        # меню выбора цвета
        self.palette_choice = Menu(self.param_menu)
        self.palette_choice.add_command(label=Dict['palette_menu_random'],
                                        command=lambda: self.canv.color.define_palette(-1))
        self.palette_choice.add_command(label=Dict['palette_menu_gradual'],
                                        command=lambda: self.canv.color.define_palette(-2))
        self.palette_choice.add_command(label=Dict['palette_menu_heart'],
                                        command=lambda: self.canv.color.define_palette(-4))
        self.palette_choice.add_command(label=Dict['palette_menu_1'],
                                        command=lambda: self.canv.color.define_palette(1))
        self.palette_choice.add_command(label=Dict['palette_menu_2'],
                                        command=lambda: self.canv.color.define_palette(2))
        self.palette_choice.add_command(label=Dict['palette_menu_3'],
                                        command=lambda: self.canv.color.define_palette(3))

        # меню выбора масштабирования
        self.scale_choice = Menu(self.param_menu)
        self.scale_choice.add_command(
            label=Dict['scale_menu_const'],
            command=lambda: self.canv.set_scale_function('constant'))
        self.scale_choice.add_command(
            label=Dict['scale_menu_inverse'],
            command=lambda: self.canv.set_scale_function('inverse_dist'))
        self.scale_choice.add_command(
            label=Dict['scale_menu_manhatten'],
            command=lambda: self.canv.set_scale_function('manhatten'))
        self.scale_choice.add_command(
            label=Dict['scale_menu_square'],
            command=lambda: self.canv.set_scale_function('square_dist'))
        self.scale_choice.add_command(
            label=Dict['scale_menu_inv'],
            command=lambda: self.canv.set_scale_function('inv_Chebushev'))

        # меню выбора стандартных функций
        self.func_choice = Menu(self.main_menu)
        self.func_choice.add_command(
            label=Dict['func_menu_cardioid'],
            command=lambda: self.canv.heart(1))
        self.func_choice.add_command(
            label=Dict['func_menu_heart'],
            command=lambda: self.canv.heart(2))

        # объединение параметрических меню
        self.param_menu.add_cascade(label=Dict['menu_brush'], menu=self.brush_style)
        self.param_menu.add_cascade(label=Dict['menu_scale'], menu=self.scale_choice)
        self.param_menu.add_cascade(label=Dict['menu_palette'], menu=self.palette_choice)
        self.param_menu.add_command(label=Dict['menu_size'], command=self.select_fig_size)
        self.param_menu.add_command(label=Dict['menu_symm'], command=self.select_num_symm)

        # меню работы с файлами
        self.file_menu = Menu(self.main_menu)
        self.file_menu.add_command(
            label=Dict['file_menu_load'],
            command=self.canv.load)
        self.file_menu.add_command(
            label=Dict['file_menu_save'],
            command=self.canv.save)
        self.file_menu.add_command(
            label=Dict['file_menu_save_pic'],
            command=self.canv.save_to_png)

        # добавляем кнопки и менюшки
        self.main_menu.add_command(label=Dict['menu_help'], command=self.open_help)
        self.main_menu.add_cascade(label=Dict['menu_file'], menu=self.file_menu)
        self.main_menu.add_cascade(label=Dict['menu_param'], menu=self.param_menu)
        self.main_menu.add_command(label=Dict['menu_undo'], command=self.canv.undo)
        self.main_menu.add_command(label=Dict['menu_redo'], command=self.canv.redo)
        self.main_menu.add_cascade(label=Dict['menu_func'], menu=self.func_choice)
        self.main_menu.add_command(label=Dict['menu_clear'], command=self.canv.cleanup)
        self.config(menu=self.main_menu)

        # центрируем окно по центру экрана
        self.update_idletasks()
        x_center = (self.winfo_screenwidth() - self.winfo_width()) / 2
        y_center = (self.winfo_screenheight() - self.winfo_height()) / 2
        self.wm_geometry('+%d+%d' % (x_center, y_center))

        # self.mainloop()

    def open_help(self):
        """Открывает окно с помощью"""
        help_wind = HelpWindow()
        help_wind.mainloop()
        help_wind.destroy()

    def select_fig_size(self):
        """Установка размера фигуры"""
        fig_sizer = FigSizer(self.canv.fig_size)
        fig_sizer.mainloop()
        self.canv.fig_size = fig_sizer.figsize.get()
        fig_sizer.destroy()

    def select_num_symm(self):
        u"""Установка числа симметричных отражений"""
        num_symmetry = NumSymmetry(self.canv.num_symm)
        num_symmetry.mainloop()
        num_symm = num_symmetry.num_symm.get()
        if num_symm > 0:
            self.canv.num_symm = num_symm * 2
        else:
            self.canv.num_symm = num_symm
        self.canv.recalculate_coefficients()
        num_symmetry.destroy()