Skip to content

Commit

Permalink
Rewrite again and rename to telop_resample
Browse files Browse the repository at this point in the history
  • Loading branch information
@Setsugennoao
Setsugennoao committed Aug 4, 2024
1 parent 0b95ccf commit ca82c48
Showing 1 changed file with 88 additions and 106 deletions.
194 changes: 88 additions & 106 deletions vsdeinterlace/funcs.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,149 +2,131 @@

from typing import Any

from stgpytools import KwargsT
from stgpytools import CustomIntEnum
from vsdenoise import MVTools
from vstools import InvalidFramerateError, check_variable, core, vs

__all__ = [
'pulldown_credits'
'telop_resample'
]


def pulldown_credits(
bobbed_clip: vs.VideoNode, frame_ref: int, interlaced: bool = True, decimate: bool = False,
**mv_args: Any
) -> vs.VideoNode:
"""
Deinterlacing function for interlaced credits (60i/30p) on top of telecined video (24p).
class telop_resample(CustomIntEnum):
VID_24t_TXT_60i = 0
VID_24d_TXT_60i = 1
VID_24t_TXT_30p = 2

The credits are interpolated and decimated to match the output clip.
def __call__(self, bobbed_clip: vs.VideoNode, pattern: int, **mv_args: Any) -> vs.VideoNode:
"""
:param bobbed_clip: Bobbed clip. Framerate must be 60000/1001.
:param pattern: First frame in the pattern.
:param mv_args: Arguments to pass on to MVTools, used for motion compensation.
The function assumes you're passing a telecined clip (that's native 24p).
If your credits are 30p (as opposed to 60i), you should set `interlaced` to False.
:return: Decimated clip with text resampled down to 24p.
The recommended way to use this filter is to trim out the area with interlaced credits,
apply this function, and `vstools.insert_clip` the clip back into a properly IVTC'd clip.
:raises InvalidFramerateError: Bobbed clip does not have a framerate of 60000/1001 (59.94)
"""

:param bobbed_clip: Bobbed clip. Framerate must be 60000/1001.
:param frame_ref: First frame in the pattern. Expected pattern is ABBCD,
except for when ``decimate`` is enabled, in which case it's AABCD.
:param tff: Top-field-first. `False` sets it to Bottom-Field-First.
:param interlaced: 60i credits. Set to false for 30p credits.
:param decimate: Decimate input clip as opposed to IVTC.
:param mv_args: Arguments to pass on to MVTools, used for motion compensation.
assert check_variable(bobbed_clip, telop_resample)

:return: IVTC'd/decimated clip with credits pulled down to 24p.
InvalidFramerateError.check(telop_resample, bobbed_clip, (60000, 1001))

:raises ModuleNotFoundError: Dependencies are missing.
:raises ValueError: Clip does not have a framerate of 30000/1001 (29.97).
:raises TopFieldFirstError: No automatic ``tff`` can be determined.
:raises InvalidFramerateError: Bobbed clip does not have a framerate of 60000/1001 (59.94)
"""
invpos = (5 - pattern * 2 % 5) % 5

assert check_variable(bobbed_clip, pulldown_credits)
offset = [0, 0, -1, 1, 1][pattern]
pattern = [0, 1, 0, 0, 1][pattern]
direction = [-1, -1, 1, 1, 1][pattern]

InvalidFramerateError.check(pulldown_credits, bobbed_clip, (60000, 1001))
ivtc_fps, ivtc_fps_div = (dict(fpsnum=x, fpsden=1001) for x in (24000, 12000))

field_ref = frame_ref * 2
frame_ref %= 5
invpos = (5 - field_ref) % 5
pos = []
assumefps = 0

offset = [0, 0, -1, 1, 1][frame_ref]
pattern = [0, 1, 0, 0, 1][frame_ref]
direction = [-1, -1, 1, 1, 1][frame_ref]
interlaced = self in (telop_resample.VID_24t_TXT_60i, telop_resample.VID_24d_TXT_60i)
decimate = self is telop_resample.VID_24d_TXT_60i

mv_args = KwargsT(range_conversion=1.0, pel=2, refine=0) | mv_args
cycle = 10 // (1 + interlaced)

ivtc_fps, ivtc_fps_div = (dict(fpsnum=x, fpsden=1001) for x in (24000, 12000))
def bb(idx: int, cut: bool = False) -> vs.VideoNode:
if cut:
return bobbed_clip[cycle:].std.SelectEvery(cycle, [idx])
return bobbed_clip.std.SelectEvery(cycle, [idx])

pos = []
assumefps = 0
def intl(clips: list[vs.VideoNode], toreverse: bool, halv: list[int]) -> vs.VideoNode:
clips = [c[::2] if i in halv else c for i, c in enumerate(clips)]
if not toreverse:
clips = list(reversed(clips))
return core.std.Interleave(clips)

cycle = 10 // (1 + interlaced)
if interlaced:
if decimate:
cleanpos = 4
pos = [1, 2]

def bb(idx: int, cut: bool = False) -> vs.VideoNode:
if cut:
return bobbed_clip[cycle:].std.SelectEvery(cycle, [idx])
return bobbed_clip.std.SelectEvery(cycle, [idx])
if invpos > 2:
pos = [6, 7]
assumefps = 2
elif invpos > 1:
pos = [2, 6]
assumefps = 1
else:
cleanpos = 1
pos = [3, 4]

def intl(clips: list[vs.VideoNode], toreverse: bool, halv: list[int]) -> vs.VideoNode:
clips = [c[::2] if i in halv else c for i, c in enumerate(clips)]
if not toreverse:
clips = list(reversed(clips))
return core.std.Interleave(clips)
if invpos > 1:
cleanpos = 6
assumefps = 1

# 60i credits. Start of ABBCD
if interlaced:
if decimate:
cleanpos = 4
pos = [1, 2]
if invpos > 3:
pos = [4, 8]
assumefps = 1

if invpos > 2:
pos = [6, 7]
assumefps = 2
elif invpos > 1:
pos = [2, 6]
assumefps = 1
else:
cleanpos = 1
pos = [3, 4]
clean = bobbed_clip.std.SelectEvery(cycle, [cleanpos - invpos])
jitter = bobbed_clip.std.SelectEvery(cycle, [p - invpos for p in pos])

if invpos > 1:
cleanpos = 6
assumefps = 1
if assumefps:
jitter = core.std.AssumeFPS(
bobbed_clip[0] * assumefps + jitter, **(ivtc_fps_div if cleanpos == 6 else ivtc_fps)
)

if invpos > 3:
pos = [4, 8]
assumefps = 1
comp = MVTools(jitter, **mv_args).flow_interpolate()

clean = bobbed_clip.std.SelectEvery(cycle, [cleanpos - invpos])
jitter = bobbed_clip.std.SelectEvery(cycle, [p - invpos for p in pos])
out = intl([comp, clean], decimate, [0])
offs = 3 if decimate else 2

if assumefps:
jitter = core.std.AssumeFPS(
bobbed_clip[0] * assumefps + jitter, **(ivtc_fps_div if cleanpos == 6 else ivtc_fps)
)
return out[invpos // offs:]

comp = MVTools(jitter, **mv_args).flow_interpolate()
if pattern == 0:
c1pos = [0, 2, 7, 5]
c2pos = [3, 4, 9, 8]

out = intl([comp, clean], decimate, [0])
offs = 3 if decimate else 2
if offset == -1:
c1pos = [2, 7, 5, 10]

return out[invpos // offs:]
if offset == 1:
c2pos = []
c2 = core.std.Interleave([bb(4), bb(5), bb(0, True), bb(9)])
else:
c1pos = [2, 4, 9, 7]
c2pos = [0, 1, 6, 5]

# 30i credits
if pattern == 0:
c1pos = [0, 2, 7, 5]
c2pos = [3, 4, 9, 8]
if offset == 1:
c1pos = []
c1 = core.std.Interleave([bb(3), bb(5), bb(0, True), bb(8)])

if offset == -1:
c1pos = [2, 7, 5, 10]
if offset == -1:
c2pos = [1, 6, 5, 10]

if offset == 1:
c2pos = []
c2 = core.std.Interleave([bb(4), bb(5), bb(0, True), bb(9)])
else:
c1pos = [2, 4, 9, 7]
c2pos = [0, 1, 6, 5]
if c1pos:
c1 = bobbed_clip.std.SelectEvery(cycle, [c + offset for c in c1pos])

if offset == 1:
c1pos = []
c1 = core.std.Interleave([bb(3), bb(5), bb(0, True), bb(8)])
if c2pos:
c2 = bobbed_clip.std.SelectEvery(cycle, [c + offset for c in c2pos])

if offset == -1:
c2pos = [1, 6, 5, 10]

if c1pos:
c1 = bobbed_clip.std.SelectEvery(cycle, [c + offset for c in c1pos])

if c2pos:
c2 = bobbed_clip.std.SelectEvery(cycle, [c + offset for c in c2pos])

if offset == -1:
c1, c2 = (core.std.AssumeFPS(bobbed_clip[0] + c, **ivtc_fps) for c in (c1, c2))
c1, c2 = (core.std.AssumeFPS(bobbed_clip[0] + c, **ivtc_fps) for c in (c1, c2))

fix1 = MVTools(c1, **mv_args).flow_interpolate(time=50 + direction * 25)
fix2 = MVTools(c2, **mv_args).flow_interpolate()
fix1 = MVTools(c1, **mv_args).flow_interpolate(time=50 + direction * 25)
fix2 = MVTools(c2, **mv_args).flow_interpolate()

return intl([fix1, fix2], pattern == 0, [0, 1])
return intl([fix1, fix2], pattern == 0, [0, 1])

0 comments on commit ca82c48

Please sign in to comment.