101 lines
2.4 KiB
Python
101 lines
2.4 KiB
Python
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()
|