5
0

tensura: s2: tv: 14

This commit is contained in:
louis f 2021-07-14 08:15:35 -04:00
parent 22d1a0adf6
commit ddf478ed79
Signed by: louis
GPG Key ID: 44D7E1DE4E23D6F2
5 changed files with 133 additions and 18 deletions

View File

@ -0,0 +1,96 @@
import vapoursynth as vs
from tensura_common import (TenSuraS2Config, TenSuraS2AODSource, antialias, cr_prefilter,
deband, dehalo, denoise, edgefix, finalize, regrain)
from yt_common.automation import SelfRunner
from yt_common.chapters import Chapter
from yt_common.video import Zone
from lvsfunc.mask import BoundingBox
from lvsfunc.dehardsub import HardsubASS, HardsubMask, bounded_dehardsub
from lvsfunc.types import Range
from lvsfunc.util import replace_ranges, normalize_ranges
from typing import List, Optional, Tuple
import os
DESC: str = os.path.basename(os.path.splitext(__file__)[0])
CONFIG: TenSuraS2Config = TenSuraS2Config(DESC)
SOURCE: TenSuraS2AODSource = TenSuraS2AODSource(CONFIG, [(24, -24)])
OP: Optional[int] = 1128
CHAPTERS: List[Chapter] = [
Chapter("Intro", 0),
Chapter("OP", 1128),
Chapter("Part A", 3285),
Chapter("Part B", 14530),
Chapter("ED", 31769),
Chapter("Next", 33927),
]
DEHARDSUB: List[HardsubMask] = [
HardsubASS("./ger.ass"),
]
NOD3: List[Range] = [
]
if OP is not None:
NOD3 += [
(OP+1520, OP+1576)
]
NODEN: List[Range] = [
(15459, 15592),
]
if OP is not None:
NODEN += [
(OP+1385, OP+1519)
]
ZONES: List[Zone] = [
]
ZONES += [Zone(r, 0.75) for r in normalize_ranges(SOURCE.source(), NOD3 + NODEN)]
AA_STRONG: List[Range] = [
]
SANGNOM: List[Tuple[Range, List[BoundingBox]]] = [
]
core = vs.core
def filter_basic() -> vs.VideoNode:
aod, ref = SOURCE.dhs_source()
return finalize(bounded_dehardsub(aod[0], ref, DEHARDSUB))
def filter() -> vs.VideoNode:
aod, ref = SOURCE.dhs_source()
pre = cr_prefilter(ref)
src = bounded_dehardsub(aod[0], pre, DEHARDSUB)
ef = edgefix(src)
den = denoise(ef)
den = replace_ranges(den, src, NODEN)
deb = deband(den)
deb = replace_ranges(deb, src, NOD3)
aa = antialias(deb, strong=AA_STRONG, sangnom=SANGNOM)
dh = dehalo(aa)
dh = replace_ranges(dh, aa, NOD3 + NODEN)
grain = regrain(dh)
final = finalize(grain)
src.set_output(1)
final.set_output(0)
return final
if __name__ == "__main__":
SelfRunner(CONFIG, SOURCE, filter, filter_basic, chapters=CHAPTERS, zones=ZONES)
else:
filter()

View File

@ -2,3 +2,4 @@ from .config import (TenSuraS2Config, TenSuraS2Source, TenSuraS2BDSource, TenSur
TenSuraS1BDConfig, TenSuraS1BDSource)
from .filter import (antialias, deband, denoise, descale, dehalo, edgefix, finalize, # noqa: F401
megurumono_scenefilter, regrain)
from .tv import (cr_prefilter, deband_tv, regrain_tv) # noqa: F401

View File

@ -69,12 +69,6 @@ def deband(clip: vs.VideoNode, strong: Optional[List[Range]] = None,
return core.std.MaskedMerge(deb, clip, dmask)
def deband_tv(clip: vs.VideoNode) -> vs.VideoNode:
dmask = detail_mask(clip)
deb = dumb3kdb(clip, radius=16, threshold=40)
return core.std.MaskedMerge(deb, clip, dmask)
def antialias(clip: vs.VideoNode, strong: Optional[List[Range]] = None,
sangnom: Optional[List[Tuple[Range, List[BoundingBox]]]] = None) -> vs.VideoNode:
clamp = sraa_clamp(clip, mask=mask_strong)
@ -112,18 +106,6 @@ def regrain(clip: vs.VideoNode) -> vs.VideoNode:
return grain
def regrain_tv(clip: vs.VideoNode) -> vs.VideoNode:
mask_bright = clip.std.PlaneStats().adg.Mask(10)
mask_dark = clip.std.PlaneStats().adg.Mask(25)
sgrain_l = core.std.MaskedMerge(clip, clip.grain.Add(var=0.1, constant=True, seed=393), mask_bright.std.Invert())
sgrain_h = core.std.MaskedMerge(clip, clip.grain.Add(var=0.15, uvar=0.1, constant=True, seed=393), mask_bright)
sgrain_h = core.std.MaskedMerge(clip, sgrain_h, mask_dark.std.Invert())
sgrain = sgrain_h.std.MergeDiff(clip.std.MakeDiff(sgrain_l))
dgrain = core.std.MaskedMerge(clip, clip.grain.Add(var=0.25, uvar=0.1, constant=False, seed=393), mask_dark)
grain = dgrain.std.MergeDiff(clip.std.MakeDiff(sgrain))
return grain
def finalize(clip: vs.VideoNode) -> vs.VideoNode:
return depth(clip, 10)

View File

@ -0,0 +1,34 @@
import vapoursynth as vs
from lvsfunc.kernels import Bicubic
from lvsfunc.mask import detail_mask
from lvsfunc.scale import descale
from vardefunc.deband import dumb3kdb
from vsutil import depth
core = vs.core
def cr_prefilter(clip: vs.VideoNode) -> vs.VideoNode:
rescaled = depth(descale(clip, height=810, kernel=Bicubic(b=0, c=1/2)), 16)
return rescaled
def deband_tv(clip: vs.VideoNode) -> vs.VideoNode:
dmask = detail_mask(clip)
deb = dumb3kdb(clip, radius=16, threshold=40)
return core.std.MaskedMerge(deb, clip, dmask)
def regrain_tv(clip: vs.VideoNode) -> vs.VideoNode:
mask_bright = clip.std.PlaneStats().adg.Mask(10)
mask_dark = clip.std.PlaneStats().adg.Mask(25)
sgrain_l = core.std.MaskedMerge(clip, clip.grain.Add(var=0.1, constant=True, seed=393), mask_bright.std.Invert())
sgrain_h = core.std.MaskedMerge(clip, clip.grain.Add(var=0.15, uvar=0.1, constant=True, seed=393), mask_bright)
sgrain_h = core.std.MaskedMerge(clip, sgrain_h, mask_dark.std.Invert())
sgrain = sgrain_h.std.MergeDiff(clip.std.MakeDiff(sgrain_l))
dgrain = core.std.MaskedMerge(clip, clip.grain.Add(var=0.25, uvar=0.1, constant=False, seed=393), mask_dark)
grain = dgrain.std.MergeDiff(clip.std.MakeDiff(sgrain))
return grain

View File

@ -0,0 +1,2 @@
# these don't have to make sense
x264 - --demuxer y4m --input-depth 10 --output-depth 10 --colormatrix bt709 --colorprim bt709 --transfer bt709 --preset fast --crf 16 --aq-mode 3 --aq-strength 0.85 --qcomp 0.70 --psy-rd 0.85:0.0 --deblock -1:-1 --frames {frames:d} --output {filename:s}.h264