# =====================================================
# Apckeyl_RealESRGAN
# Version 1.0
# mode_manager.py
# =====================================================

from PIL import Image


class ModeManager:

    """
    Управляет режимами обработки изображения.

    Пока выполняет только изменение
    итогового размера изображения.

    В следующих версиях именно здесь
    будут подключаться:

    • детализация
    • шумоподавление
    • модели
    • Face Restore
    • Anime
    • и т.д.
    """

    def __init__(self):

        pass

    # -------------------------------------------------

    def apply(

        self,

        image,

        mode,

        quality,

        denoise,

    ):

        width, height = image.size

        # ---------------------------------------------
        # Улучшить качество
        # ---------------------------------------------

        if mode == "Улучшить качество (рекомендуется)":

            return image.resize(

                (

                    width // 4,
                    height // 4,

                ),

                Image.Resampling.LANCZOS,

            )

        # ---------------------------------------------
        # ×2
        # ---------------------------------------------

        if mode == "Увеличить ×2":

            return image.resize(

                (

                    width // 2,
                    height // 2,

                ),

                Image.Resampling.LANCZOS,

            )

        # ---------------------------------------------
        # ×4
        # ---------------------------------------------

        if mode == "Увеличить ×4":

            return image

        # ---------------------------------------------

        return image