from __future__ import annotations

from collections.abc import Callable

import PIL.Image
import PIL.ImageEnhance
from PySide6 import QtWidgets
from PySide6.QtCore import Qt
from PySide6.QtGui import QImage


class BrightnessContrastDialog(QtWidgets.QDialog):
    _base_value = 50

    img: PIL.Image.Image

    def __init__(
        self,
        img: PIL.Image.Image,
        callback: Callable[[QImage], None],
        parent: QtWidgets.QWidget | None = None,
    ) -> None:
        super().__init__(parent)
        self.setWindowTitle(self.tr("Brightness/Contrast"))
        self.setModal(True)

        self._on_image_changed = callback

        sliders = {}
        layouts = {}
        for title in ["Brightness:", "Contrast:"]:
            layout = QtWidgets.QHBoxLayout()
            title_label = QtWidgets.QLabel(self.tr(title))
            title_label.setFixedWidth(75)
            layout.addWidget(title_label)
            #
            slider = QtWidgets.QSlider(Qt.Orientation.Horizontal)
            slider.setRange(0, 3 * self._base_value)
            slider.setValue(self._base_value)
            layout.addWidget(slider)
            #
            value_label = QtWidgets.QLabel(f"{slider.value() / self._base_value:.2f}")
            value_label.setAlignment(Qt.AlignmentFlag.AlignRight)
            layout.addWidget(value_label)
            #
            slider.valueChanged.connect(lambda _: self.apply())
            slider.valueChanged.connect(
                lambda _,
                value_label_=value_label,
                slider_=slider: value_label_.setText(
                    f"{slider_.value() / self._base_value:.2f}"
                )
            )
            layouts[title] = layout
            sliders[title] = slider

        self.slider_brightness = sliders["Brightness:"]
        self.slider_contrast = sliders["Contrast:"]
        del sliders

        v_layout = QtWidgets.QVBoxLayout()
        v_layout.addLayout(layouts["Brightness:"])
        v_layout.addLayout(layouts["Contrast:"])
        del layouts
        self.setLayout(v_layout)

        self._alpha = None
        if "A" in img.getbands():
            self._alpha = img.getchannel("A")
        if img.mode != "RGB":
            img = img.convert("RGB")
        self.img = img

    def apply(self) -> None:
        img: PIL.Image.Image = self.img
        enhancers = [
            (
                self.slider_brightness.value() / self._base_value,
                PIL.ImageEnhance.Brightness,
            ),
            (
                self.slider_contrast.value() / self._base_value,
                PIL.ImageEnhance.Contrast,
            ),
        ]
        for factor, enhancer_cls in enhancers:
            if factor == 1.0:
                continue
            img = enhancer_cls(img).enhance(factor)

        fmt: QImage.Format
        if self._alpha is None:
            fmt = QImage.Format.Format_RGB888
        else:
            img = img.convert("RGBA")
            img.putalpha(self._alpha)
            fmt = QImage.Format.Format_RGBA8888

        qimage = QImage(
            img.tobytes(), img.width, img.height, img.width * len(img.getbands()), fmt
        )
        self._on_image_changed(qimage)
