import vapoursynth as vs

from tensura_common import (TenSuraS2Config, TenSuraS2BDSource, antialias, deband, denoise, descale,
                            edgefix, finalize, megurumono_scenefilter, regrain)

from yt_common.automation import SelfRunner
from yt_common.chapters import Chapter
from yt_common.source import FileTrim
from yt_common.video import Zone

from lvsfunc.mask import BoundingBox
from lvsfunc.misc import replace_ranges
from lvsfunc.types import Range

from typing import List, Tuple

import os

DESC: str = os.path.basename(os.path.splitext(__file__)[0])
CONFIG: TenSuraS2Config = TenSuraS2Config(DESC)
BDMV: str = "../bdmv/[BDMV][210326][Tensei Shitara Slime Datta Ken 2nd Season][Vol.1]/BDMV/STREAM"
SOURCE: TenSuraS2BDSource = TenSuraS2BDSource(
    FileTrim(os.path.join(BDMV, "00013.m2ts"), (24, -24))
)

MEGURUMONO: Tuple[int, int] = (2873, 5034)

CHAPTERS: List[Chapter] = [
    Chapter("Intro", 0),
    Chapter("OP", MEGURUMONO[0]),
    Chapter("Part A", MEGURUMONO[1]+1),
    Chapter("Part B", 22561),
    Chapter("ED", 31883),
]

NOD3: List[Range] = [
]

NODEN: List[Range] = [
    (21749, 21849),
]

ZONES: List[Zone] = [
]

ZONES += [Zone(r, 0.75) for r in NOD3 + NODEN if isinstance(r, tuple)]

AA_STRONG: List[Range] = [
]

SANGNOM: List[Tuple[Range, List[BoundingBox]]] = [
]

DEBAND_STRONG: List[Range] = [
    (0, 500),
    (11519, 11615),
    (14845, 14906),
    (15137, 15234),
    (15667, 15817),
    (18848, 19536),
    (19897, 20004),
    (20749, 20814),
    (21013, 21340),
    (21485, 21748),
    (22812, 22894),
    (26719, 26769),
    (31307, 31426),
    (31596, 31882),
]

DEBAND_NUCLEAR: List[Range] = [
]

core = vs.core


def filter() -> vs.VideoNode:
    src = SOURCE.source()
    ef = edgefix(src)
    den = denoise(ef)
    den = replace_ranges(den, src, NODEN)
    descaled = descale(den)
    deb = deband(descaled, strong=DEBAND_STRONG, nuclear=DEBAND_NUCLEAR)
    deb = replace_ranges(deb, src, NOD3)
    aa = antialias(deb, strong=AA_STRONG, sangnom=SANGNOM)
    grain = regrain(aa)
    scenefilter = grain
    if MEGURUMONO:
        megurumono = megurumono_scenefilter(src, MEGURUMONO[0])
        scenefilter = replace_ranges(scenefilter, megurumono, MEGURUMONO)
    final = finalize(scenefilter)
    src.set_output(1)
    final.set_output(0)
    return final


if __name__ == "__main__":
    SelfRunner(CONFIG, SOURCE, filter, chapters=CHAPTERS, zones=ZONES)
else:
    filter()