# -*- coding: utf-8 -*-
import logging

from PySide6 import QtWidgets, QtGui

from fastflix.models.fastflix_app import FastFlixApp
from fastflix.models.encode import AudioTrack
from fastflix.language import t

__all__ = ["AudioConversion"]

logger = logging.getLogger("fastflix")

# audio_disposition_options = [
#     "dub",
#     "original",
#     "comment",
#     "visual_impaired",
# ]
#
# subtitle_disposition_options = [
#     "dub",
#     "original",
#     "comment",
#     "lyrics",
#     "karaoke",
#     "hearing_impaired",
# ]

channel_list = {
    "mono": 1,
    "stereo": 2,
    "2.1": 3,
    "3.0": 3,
    "3.0(back)": 3,
    "3.1": 4,
    "4.0": 4,
    "quad": 4,
    "quad(side)": 4,
    "5.0": 5,
    "5.1": 6,
    "5.1(side)": 6,
    "6.0": 6,
    "6.0(front)": 6,
    "hexagonal": 6,
    "6.1": 7,
    "6.1(front)": 7,
    "7.0": 7,
    "7.0(front)": 7,
    "7.1": 8,
    "7.1(wide)": 8,
}

venc_channels = {
    "mono": 1,
    "stereo": 2,
    "2.1": 3,
    "3.0": 3,
    "3.0(back)": 3,
    "3.1": 4,
    "4.0": 4,
    "quad": 4,
    "quad(side)": 4,
    "5.0": 5,
    "5.1": 6,
    "6.0": 6,
    "6.0(front)": 6,
    "hexagonal": 6,
    "6.1": 7,
    "6.1(front)": 7,
    "7.0": 7,
    "7.0(front)": 7,
    "7.1": 8,
    "7.1(wide)": 8,
}

back_channel_list = {
    1: "mono",
    2: "stereo",
    3: "2.1",
    4: "3.1",
    5: "5.0",
    6: "5.1",
    7: "6.1",
    8: "7.1",
}


class AudioConversion(QtWidgets.QWidget):
    def __init__(self, app: FastFlixApp, track_index, encoders, audio_track_update):
        super().__init__(None)
        self.app = app
        self.audio_track_update = audio_track_update
        self.setWindowTitle(f"Audio Conversion for Track {track_index}")
        self.setMinimumWidth(400)
        self.audio_track: AudioTrack = self.app.fastflix.current_video.audio_tracks[track_index]

        # Conversion

        self.conversion_codec = QtWidgets.QComboBox()
        self.conversion_codec.addItems([t("None")] + list(sorted(encoders)))

        if self.audio_track.conversion_codec:
            self.conversion_codec.setCurrentText(self.audio_track.conversion_codec)
        self.conversion_codec.currentIndexChanged.connect(self.codec_changed)

        conversion_layout = QtWidgets.QHBoxLayout()
        conversion_layout.addWidget(QtWidgets.QLabel(t("Codec")))
        conversion_layout.addWidget(self.conversion_codec, 2)

        # AQ vs Bitrate

        self.aq = QtWidgets.QComboBox()
        self.aq.addItems(
            [
                f"0 - {t('Near Lossless')}",
                "1",
                f"2 - {t('High Quality')}",
                "3",
                f"4 - {t('Medium Quality')}",
                "5",
                f"6 {t('Low Quality')}",
                "7",
                "8",
                "9",
                t("Custom Bitrate"),
            ]
        )
        self.aq.setMinimumWidth(100)
        self.aq.currentIndexChanged.connect(self.set_aq)
        self.bitrate = QtWidgets.QLineEdit()
        self.bitrate.setFixedWidth(50)
        self.bitrate.setValidator(QtGui.QDoubleValidator())

        if self.audio_track.conversion_aq:
            self.aq.setCurrentIndex(self.audio_track.conversion_aq)
            self.bitrate.setDisabled(True)
        elif self.audio_track.conversion_bitrate:
            self.aq.setCurrentText(t("Custom Bitrate"))
            self.bitrate.setText(self.audio_track.conversion_bitrate)
            self.bitrate.setEnabled(True)

        elif self.conversion_codec.currentText() in ["libopus"]:
            self.aq.setCurrentIndex(10)
        else:
            self.aq.setCurrentIndex(3)

        quality_layout = QtWidgets.QHBoxLayout()
        quality_layout.addWidget(QtWidgets.QLabel(t("Audio Quality")))
        quality_layout.addWidget(self.aq, 1)
        quality_layout.addWidget(QtWidgets.QLabel(t("Bitrate")))
        quality_layout.addWidget(self.bitrate)
        quality_layout.addWidget(QtWidgets.QLabel("kb/s"))

        channel_layout = self.audio_track.raw_info.get("channel_layout")

        self.downmix = QtWidgets.QComboBox()

        if "encc" in app.fastflix.current_video.video_settings.video_encoder_settings.name.lower():
            self.downmix.addItems([t("None")] + list(venc_channels.keys()))
        else:
            self.downmix.addItems([t("None")] + list(channel_list.keys()))

        try:
            if channel_layout:
                self.downmix.setCurrentText(channel_layout)
            else:
                guess = back_channel_list[self.audio_track.raw_info.get("channels")]
                logger.warning(f"Channel layout not found for {self.audio_track.title}, guessing {guess}")
                self.downmix.setCurrentText(guess)
        except Exception:
            self.downmix.setCurrentIndex(2)
        if self.audio_track.downmix:
            self.downmix.setCurrentText(self.audio_track.downmix)

        downmix_layout = QtWidgets.QHBoxLayout()
        downmix_layout.addWidget(QtWidgets.QLabel(t("Channel Layout")))
        downmix_layout.addWidget(self.downmix, 2)

        # Yes No

        yes_no_layout = QtWidgets.QHBoxLayout()
        cancel = QtWidgets.QPushButton(t("Cancel"))
        cancel.clicked.connect(self.close)
        yes_no_layout.addWidget(cancel)
        save = QtWidgets.QPushButton(t("Save"))
        save.clicked.connect(self.save)
        yes_no_layout.addWidget(save)

        layout = QtWidgets.QVBoxLayout()
        layout.addLayout(conversion_layout)
        layout.addLayout(quality_layout)
        layout.addLayout(downmix_layout)
        layout.addLayout(yes_no_layout)

        self.setLayout(layout)

    def set_aq(self):
        index = self.aq.currentIndex()
        if index == 10:
            self.bitrate.setEnabled(True)
        else:
            self.bitrate.setDisabled(True)

    def codec_changed(self):
        if self.conversion_codec.currentText() in ["libopus"]:
            self.aq.setCurrentIndex(10)
            self.aq.setDisabled(True)
            # self.bitrate.setEnabled(True)
        else:
            self.aq.setEnabled(True)
            # self.bitrate.setDisabled(True)

    def save(self):
        if self.conversion_codec.currentIndex() != 0:
            self.audio_track.conversion_codec = self.conversion_codec.currentText()
        else:
            self.audio_track.conversion_codec = ""

        if self.aq.currentIndex() != 10:
            self.audio_track.conversion_aq = self.aq.currentIndex()
            self.audio_track.conversion_bitrate = None
        else:
            self.audio_track.conversion_bitrate = self.bitrate.text()
            self.audio_track.conversion_aq = None

        if self.downmix.currentIndex() != 0:
            self.audio_track.downmix = self.downmix.currentText()
        else:
            self.audio_track.downmix = None
        self.audio_track_update()
        self.close()
