102 lines
2.7 KiB
Python
102 lines
2.7 KiB
Python
import vapoursynth as vs
|
|
|
|
from tensura_common import TenSuraS2BDConfig, antialias, deband, denoise, descale, edgefix, finalize, regrain
|
|
|
|
from yt_common.automation import SelfRunner, Zone
|
|
from yt_common.chapters import Chapter, Edition
|
|
from yt_common.source import FileTrim, SimpleSource
|
|
|
|
from lvsfunc.mask import BoundingBox
|
|
from lvsfunc.misc import replace_ranges
|
|
from lvsfunc.types import Range
|
|
|
|
from typing import List, Tuple
|
|
|
|
import os
|
|
|
|
EPNUM: int = int(os.path.basename(os.path.splitext(__file__)[0]))
|
|
CONFIG: TenSuraS2BDConfig = TenSuraS2BDConfig(EPNUM)
|
|
BDMV: str = "../bdmv/[BDMV][210526][Tensei Shitara Slime Datta Ken 2nd Season][Vol.2]/BDMV/STREAM"
|
|
SOURCE: SimpleSource = SimpleSource([
|
|
FileTrim(os.path.join(BDMV, "00011.m2ts"), (None, -49)),
|
|
FileTrim(os.path.join(BDMV, "00017.m2ts"), (None, 3158)),
|
|
FileTrim(os.path.join(BDMV, "00017.m2ts"), (33342, 33927)),
|
|
], asrc=[
|
|
# NC version has different (quieter) audio, so make sure we use the regular version
|
|
FileTrim(os.path.join(BDMV, "00011.m2ts"), (None, -49)),
|
|
FileTrim(os.path.join(BDMV, "00011.m2ts"), (None, 3158)),
|
|
FileTrim(os.path.join(BDMV, "00011.m2ts"), (33342, 33927)),
|
|
])
|
|
|
|
EDITIONS: List[Edition] = [
|
|
Edition(chapters=[
|
|
Chapter("Intro", 0),
|
|
Chapter("Part A", 840),
|
|
Chapter("Part B", 13980),
|
|
Chapter("Next", 33927, 34047),
|
|
], default=True, ordered=True),
|
|
Edition(chapters=[
|
|
Chapter("Intro - NC", 34047),
|
|
Chapter("Part A - NC", 34887, 37205),
|
|
Chapter("Part A", 3158),
|
|
Chapter("Part B", 13980, 33342),
|
|
Chapter("Part B - NC", 37205, -1),
|
|
Chapter("Next", 33927, 34047),
|
|
], default=False, ordered=True),
|
|
]
|
|
|
|
NOD3: List[Range] = [
|
|
(4862, 5023),
|
|
(8264, 8692),
|
|
(10200, 10576),
|
|
(17993, 18828),
|
|
(33927, 34046),
|
|
]
|
|
|
|
NODEN: List[Range] = [
|
|
(16484, 17992),
|
|
(18829, 19528),
|
|
]
|
|
|
|
ZONES: List[Zone] = [
|
|
]
|
|
|
|
ZONES += [Zone(r, 0.75) for r in NOD3 + NODEN if isinstance(r, tuple)]
|
|
|
|
AA_STRONG: List[Range] = [
|
|
(14544, 14588),
|
|
(14716, 14763),
|
|
(15493, 15633),
|
|
(19529, 19600),
|
|
]
|
|
|
|
SANGNOM: List[Tuple[Range, List[BoundingBox]]] = [
|
|
]
|
|
|
|
DEBAND_NUCLEAR: List[Range] = [
|
|
]
|
|
|
|
core = vs.core
|
|
|
|
|
|
def filter() -> vs.VideoNode:
|
|
src = SOURCE.source()
|
|
ef = edgefix(src)
|
|
den = denoise(ef)
|
|
replace_ranges(den, src, NODEN)
|
|
descaled = descale(den)
|
|
deb = deband(descaled, nuclear=DEBAND_NUCLEAR)
|
|
deb = replace_ranges(deb, src, NOD3)
|
|
aa = antialias(deb, strong=AA_STRONG, sangnom=SANGNOM)
|
|
grain = regrain(aa)
|
|
final = finalize(grain)
|
|
src.set_output(1)
|
|
final.set_output(0)
|
|
return final
|
|
|
|
|
|
if __name__ == "__main__":
|
|
SelfRunner(CONFIG, SOURCE, filter, editions=EDITIONS, zones=ZONES)
|
|
else:
|
|
filter()
|