Untitled

 avatar
unknown
plain_text
2 years ago
59 kB
5
Indexable
import vapoursynth as vs
import re
from functools import partial

# Small collection of VapourSynth functions I used at least once.
# Most are simple wrappers or ports of AviSynth functions.

# Included functions:
#
#       GradFun3mod
#       DescaleM (DebilinearM, DebicubicM etc.)
#       Downscale444
#       JIVTC
#       OverlayInter
#       AutoDeblock
#       ReplaceFrames (ReplaceFramesSimple)
#       maa
#       TemporalDegrain
#       DescaleAA
#       InsertSign

# To use all included functions, you need to have
# the following Python scripts installed:
#
# havsfunc: https://github.com/HomeOfVapourSynthEvolution/havsfunc
# mvsfunc: https://github.com/HomeOfVapourSynthEvolution/mvsfunc
# muvsfunc: https://github.com/WolframRhodium/muvsfunc
# nnedi3_rpow2: https://gist.github.com/4re/342624c9e1a144a696c6


core = vs.core


"""
VapourSynth port of Gebbi's GradFun3mod

Based on Muonium's GradFun3 port:
https://github.com/WolframRhodium/muvsfunc

If you don't use any of the newly added arguments
it will behave just like unmodified GradFun3.

Differences:

 - added smode=5 that uses a bilateral filter on the GPU (CUDA)
   output should be very similar to smode=2
 - fixed the strength of the bilateral filter when using 
   smode=2 to match the AviSynth version
 - changed argument lsb to bits (default is input bitdepth)
 - case of the resizer doesn't matter anymore
 - every resizer supported by fmtconv.resample can be specified
 - yuv444 can now be used with any output resolution
 - removed fh and fv arguments for all resizers

Requirements:

 - muvsfunc  https://github.com/WolframRhodium/muvsfunc
 - havsfunc  https://github.com/HomeOfVapourSynthEvolution/havsfunc
 - mvsfunc  https://github.com/HomeOfVapourSynthEvolution/mvsfunc
 - Bilateral  https://github.com/HomeOfVapourSynthEvolution/VapourSynth-Bilateral
 - BilateralGPU (optional, needs OpenCV 3.2 with CUDA module)  https://github.com/WolframRhodium/VapourSynth-BilateralGPU
 - fmtconv  https://github.com/EleonoreMizo/fmtconv
 - Descale (optional)  https://github.com/Frechdachs/vapoursynth-descale
 - dfttest  https://github.com/HomeOfVapourSynthEvolution/VapourSynth-DFTTest
 - nnedi3  https://github.com/dubhater/vapoursynth-nnedi3
 - nnedi3_rpow2  https://gist.github.com/4re/342624c9e1a144a696c6

Original header:

##################################################################################################################
#
#   High bitdepth tools for Avisynth - GradFun3mod r6
#       based on Dither v1.27.2
#   Author: Firesledge, slightly modified by Gebbi
#
#  What?
#       - This is a slightly modified version of the original GradFun3.
#       - It combines the usual color banding removal stuff with resizers during the process
#         for sexier results (less detail loss, especially for downscales of cartoons).
#       - This is a starter script, not everything is covered through parameters. Modify it to your needs.
#
#   Requirements (in addition to the Dither requirements):
#       - AviSynth 2.6.x
#       - Debilinear, Debicubic, DebilinearM
#       - NNEDI3 + nnedi3_resize16
#
#  Changes from the original GradFun3:
#       - yuv444 = true
#         (4:2:0 -> 4:4:4 colorspace conversion, needs 1920x1080 input)
#       - resizer = [ "none", "Debilinear", "DebilinearM", "Debicubic", "DebicubicM", "Spline16",
#         "Spline36", "Spline64", "lineart_rpow2", "lineart_rpow2_bicubic" ] 
#         (use it only for downscales)
#           NOTE: As of r2 Debicubic doesn't have 16-bit precision, so a Y (luma) plane fix by torch is used here,
#                 more info: https://mechaweaponsvidya.wordpress.com/2015/07/07/a-precise-debicubic/
#                 Without yuv444=true Dither_resize16 is used with an inverse bicubic kernel.
#       - w = 1280, h = 720
#         (output width & height for the resizers; or production resolution for resizer="lineart_rpow2")
#       - smode = 4
#         (the old GradFun3mod behaviour for legacy reasons; based on smode = 1 (dfttest);
#         not useful anymore in most cases, use smode = 2 instead (less detail loss))
#       - deb = true
#         (legacy parameter; same as resizer = "DebilinearM")
#
#  Usage examples:
#       - Source is bilinear 720p->1080p upscale (BD) with 1080p credits overlayed,
#         revert the upscale without fucking up the credits:
#               lwlibavvideosource("lol.m2ts")
#               GradFun3mod(smode=1, yuv444=true, resizer="DebilinearM")
#
#       - same as above, but bicubic Catmull-Rom upscale (outlines are kind of "blocky" and oversharped):
#               GradFun3mod(smode=1, yuv444=true, resizer="DebicubicM", b=0, c=1)
#               (you may try any value between 0 and 0.2 for b, and between 0.7 and 1 for c)
#
#       - You just want to get rid off the banding without changing the resolution:
#               GradFun3(smode=2)
#
#       - Source is 1080p production (BD), downscale to 720p:
#               GradFun3mod(smode=2, yuv444=true, resizer="Spline36")
#
#       - Source is a HDTV transportstream (or CR or whatever), downscale to 720p:
#               GradFun3mod(smode=2, resizer="Spline36")
#
#       - Source is anime, 720p->1080p upscale, keep the resolution
#         but with smoother lineart instead of bilinear upscaled shit:
#               GradFun3mod(smode=2, resizer="lineart_rpow2")
#         This won't actually resize the video but instead mask the lineart and re-upscale it using
#         nnedi3_rpow2 which often results in much better looking lineart (script mostly by Daiz).
#
#       Note: Those examples don't include parameters like thr, radius, elast, mode, ampo, ampn, staticnoise.
#             You probably don't want to use the default values.
#             For 16-bit output use:
#              GradFun3mod(lsb=true).Dither_out()
#
#  What's the production resolution of my korean cartoon?
#       - Use your eyes combined with Debilinear(1280,720) - if it looks like oversharped shit,
#         it was probably produced in a higher resolution.
#       - Use Debilinear(1280,720).BilinearResize(1920,1080) for detail loss search.
#       - Alternatively you can lookup the (estimated) production resolution at
#         http://anibin.blogspot.com  (but don't blindly trust those results)
#
#   This program is free software. It comes without any warranty, to
#   the extent permitted by applicable law. You can redistribute it
#   and/or modify it under the terms of the Do What The Fuck You Want
#   To Public License, Version 2, as published by Sam Hocevar. See
#   http://sam.zoy.org/wtfpl/COPYING for more details.
#
##################################################################################################################

"""
def GradFun3(src, thr=None, radius=None, elast=None, mask=None, mode=None, ampo=None,
                ampn=None, pat=None, dyn=None, staticnoise=None, smode=None, thr_det=None,
                debug=None, thrc=None, radiusc=None, elastc=None, planes=None, ref=None,
                yuv444=None, w=None, h=None, resizer=None, b=None, c=None, bits=None):

    try:
        import mvsfunc as mvf
    except ImportError:
        raise ImportError('GradFun3: mvsfunc not found. Download it here: https://github.com/HomeOfVapourSynthEvolution/mvsfunc')
    try:
        import muvsfunc as muf
    except ImportError:
        raise ImportError('GradFun3: muvsfunc not found. Download it here: https://github.com/WolframRhodium/muvsfunc')

    def smooth_mod(src_16, ref_16, smode, radius, thr, elast, planes):
        if smode == 0:
            return muf._GF3_smoothgrad_multistage(src_16, ref_16, radius, thr, elast, planes)
        elif smode == 1:
            return muf._GF3_dfttest(src_16, ref_16, radius, thr, elast, planes)
        elif smode == 2:
            return bilateral(src_16, ref_16, radius, thr, elast, planes)
        elif smode == 3:
            return muf._GF3_smoothgrad_multistage_3(src_16, radius, thr, elast, planes)
        elif smode == 4:
            return dfttest_mod(src_16, ref_16, radius, thr, elast, planes)
        elif smode == 5:
            return bilateral_gpu(src_16, ref_16, radius, thr, elast, planes)
        else:
            raise ValueError(funcname + ': wrong smode value!')

    def dfttest_mod(src, ref, radius, thr, elast, planes):
        hrad = max(radius * 3 // 4, 1)
        last = core.dfttest.DFTTest(src, sigma=thr * 12, sbsize=hrad * 4,
                                    sosize=hrad * 3, tbsize=1, planes=planes)
        last = mvf.LimitFilter(last, ref, thr=thr, elast=elast, planes=planes)
        return last

    def bilateral(src, ref, radius, thr, elast, planes):
        thr_1 = max(thr * 4.5, 1.25)
        thr_2 = max(thr * 9, 5.0)
        r4 = max(radius * 4 / 3, 4.0)
        r2 = max(radius * 2 / 3, 3.0)
        r1 = max(radius * 1 / 3, 2.0)
        last = src
        last = core.bilateral.Bilateral(last, ref=ref, sigmaS=r4 / 2, sigmaR=thr_1 / 255,
                                        planes=planes, algorithm=0)
        # NOTE: I get much better results if I just call Bilateral once
        #last = core.bilateral.Bilateral(last, ref=ref, sigmaS=r2 / 2, sigmaR=thr_2 / 255,
        #                                planes=planes, algorithm=0)
        #last = core.bilateral.Bilateral(last, ref=ref, sigmaS=r1 / 2, sigmaR=thr_2 / 255,
        #                                planes=planes, algorithm=0)
        last = mvf.LimitFilter(last, src, thr=thr, elast=elast, planes=planes)
        return last

    def bilateral_gpu(src, ref, radius, thr, elast, planes):
        t = max(thr * 4.5, 1.25)
        r = max(radius * 4 / 3, 4.0)
        last = core.bilateralgpu.Bilateral(src, sigma_spatial=r / 2, sigma_color=t)
        last = mvf.LimitFilter(last, ref, thr=thr, elast=elast, planes=planes)
        return last

    funcname = 'GradFun3'

    # Type checking
    kwargsdict = {'src': [src, (vs.VideoNode,)], 'thr': [thr, (int, float)], 'radius': [radius, (int,)],
                  'elast': [elast, (int, float)], 'mask': [mask, (int,)], 'mode': [mode, (int,)],
                  'ampo': [ampo, (int, float)], 'ampn': [ampn, (int, float)], 'pat': [pat, (int,)],
                  'dyn': [dyn, (bool,)], 'staticnoise': [staticnoise, (bool,)], 'smode': [smode, (int,)],
                  'thr_det': [thr_det, (int, float)], 'debug': [debug, (bool, int)], 'thrc': [thrc, (int, float)],
                  'radiusc': [radiusc, (int,)], 'elastc': [elastc, (int, float)], 'planes': [planes, (int, list)],
                  'ref': [ref, (vs.VideoNode,)], 'yuv444': [yuv444, (bool,)], 'w': [w, (int,)], 'h': [h, (int,)],
                  'resizer': [resizer, (str,)], 'b': [b, (int, float)], 'c': [c, (int, float)], 'bits': [bits, (int,)]}

    for k, v in kwargsdict.items():
        if v[0] is not None and not isinstance(v[0], v[1]):
            raise TypeError('{funcname}: "{variable}" must be {types}!'
                            .format(funcname=funcname, variable=k, types=' or '.join([TYPEDICT[t] for t in v[1]])))

    # Set defaults
    if smode is None:
        smode = 2
    if thr is None:
        thr = 0.35
    if radius is None:
        radius = 12 if smode not in [0, 3] else 9
    if elast is None:
        elast = 3.0
    if mask is None:
        mask = 2
    if thr_det is None:
        thr_det = 2 + round(max(thr - 0.35, 0) / 0.3)
    if debug is None:
        debug = False
    if thrc is None:
        thrc = thr
    if radiusc is None:
        radiusc = radius
    if elastc is None:
        elastc = elast
    if planes is None:
        planes = list(range(src.format.num_planes))
    if ref is None:
        ref = src
    if yuv444 is None:
        yuv444 = False
    if w is None:
        w = 1280
    if h is None:
        h = 720
    if resizer is None:
        resizer = ''
    if yuv444 and not resizer:
        resizer = 'spline36'
    if b is None:
        b = 1/3
    if c is None:
        c = 1/3
    if bits is None:
        bits = src.format.bits_per_sample

    # Value checking
    if src.format.color_family not in [vs.YUV, vs.GRAY]:
        raise TypeError(funcname + ': "src" must be YUV or GRAY color family!')
    if ref.format.color_family not in [vs.YUV, vs.GRAY]:
        raise TypeError(funcname + ': "ref" must be YUV or GRAY color family!')
    if thr < 0.1 or thr > 10.0:
        raise ValueError(funcname + ': "thr" must be in [0.1, 10.0]!')
    if thrc < 0.1 or thrc > 10.0:
        raise ValueError(funcname + ': "thrc" must be in [0.1, 10.0]!')
    if radius <= 0:
        raise ValueError(funcname + ': "radius" must be positive.')
    if radiusc <= 0:
        raise ValueError(funcname + ': "radiusc" must be positive.')
    if elast < 1:
        raise ValueError(funcname + ': Valid range of "elast" is [1, +inf)!')
    if elastc < 1:
        raise ValueError(funcname + ': Valid range of "elastc" is [1, +inf)!')
    if smode not in [0, 1, 2, 3, 4, 5]:
        raise ValueError(funcname + ': "smode" must be in [0, 1, 2, 3, 4, 5]!')
    if smode in [0, 3]:
        if radius not in list(range(2, 10)):
            raise ValueError(funcname + ': "radius" must be in 2-9 for smode=0 or 3 !')
        if radiusc not in list(range(2, 10)):
            raise ValueError(funcname + ': "radiusc" must be in 2-9 for smode=0 or 3 !')
    elif smode in [1, 4]:
        if radius not in list(range(1, 129)):
            raise ValueError(funcname + ': "radius" must be in 1-128 for smode=1 or smode=4 !')
        if radiusc not in list(range(1, 129)):
            raise ValueError(funcname + ': "radiusc" must be in 1-128 for smode=1 or smode=4 !')
    if thr_det <= 0.0:
        raise ValueError(funcname + ': "thr_det" must be positive!')

    ow = src.width
    oh = src.height

    src_16 = core.fmtc.bitdepth(src, bits=16, planes=planes) if src.format.bits_per_sample < 16 else src
    src_8 = core.fmtc.bitdepth(src, bits=8, dmode=1, planes=[0]) if src.format.bits_per_sample != 8 else src
    ref_16 = core.fmtc.bitdepth(ref, bits=16, planes=planes) if ref.format.bits_per_sample < 16 else ref

    # Do lineart smoothing first for sharper results
    if resizer.lower() == 'lineart_rpow2':
        src_16 = ProtectedDebiXAA(src_16, w, h, bicubic=False)
    elif resizer.lower() == 'lineart_rpow2_bicubic':
        src_16 = ProtectedDebiXAA(src_16, w, h, bicubic=True, b=b, c=c)

    # Main debanding
    chroma_flag = (thrc != thr or radiusc != radius or
                   elastc != elast) and 0 in planes and (1 in planes or 2 in planes)

    if chroma_flag:
        planes2 = [0] if 0 in planes else []
    else:
        planes2 = planes

    if not planes2:
        raise ValueError(funcname + ': no plane is processed')

    flt_y = smooth_mod(src_16, ref_16, smode, radius, thr, elast, planes2)
    if chroma_flag:
        flt_c = smooth_mod(src_16, ref_16, smode, radiusc, thrc, elastc, [x for x in planes if x != 0])
        flt = core.std.ShufflePlanes([flt_y,flt_c], [0,1,2], src.format.color_family)
    else:
        flt = flt_y

    # Edge/detail mask
    td_lo = max(thr_det * 0.75, 1.0)
    td_hi = max(thr_det, 1.0)
    mexpr = 'x {tl} - {th} {tl} - / 255 *'.format(tl=td_lo - 0.0001, th=td_hi + 0.0001)

    if mask > 0:
        dmask = mvf.GetPlane(src_8, 0)
        dmask = muf._Build_gf3_range_mask(dmask, mask)
        dmask = core.std.Expr([dmask], [mexpr])
        dmask = core.rgvs.RemoveGrain(dmask, [22])
        if mask > 1:
            dmask = core.std.Convolution(dmask, matrix=[1,2,1,2,4,2,1,2,1])
            if mask > 2:
                dmask = core.std.Convolution(dmask, matrix=[1,1,1,1,1,1,1,1,1])
        dmask = core.fmtc.bitdepth(dmask, bits=16)
        res_16 = core.std.MaskedMerge(flt, src_16, dmask, planes=planes, first_plane=True)
    else:
        res_16 = flt

    # Resizing / colorspace conversion (GradFun3mod)
    res_16_y = core.std.ShufflePlanes(res_16, planes=0, colorfamily=vs.GRAY)
    if resizer.lower() == 'debilinear':
        rkernel = Resize(res_16_y if yuv444 else res_16, w, h, kernel='bilinear', invks=True)
    elif resizer.lower() == 'debicubic':
        rkernel = Resize(res_16_y if yuv444 else res_16, w, h, kernel='bicubic', a1=b, a2=c, invks=True)
    elif resizer.lower() == 'debilinearm':
        rkernel = DebilinearM(res_16_y if yuv444 else res_16, w, h, chroma=not yuv444)
    elif resizer.lower() == 'debicubicm':
        rkernel = DebicubicM(res_16_y if yuv444 else res_16, w, h, b=b, c=c, chroma=not yuv444)
    elif resizer.lower() in ('lineart_rpow2', 'lineart_rpow2_bicubic'):
        if yuv444:
            rkernel = Resize(res_16_y, w, h, kernel='spline36')
        else:
            rkernel = res_16
    elif not resizer:
        rkernel = res_16
    else:
       rkernel = Resize(res_16_y if yuv444 else res_16, w, h, kernel=resizer.lower())

    if yuv444:
        ly = rkernel
        lu = core.std.ShufflePlanes(res_16, planes=1, colorfamily=vs.GRAY)
        lv = core.std.ShufflePlanes(res_16, planes=2, colorfamily=vs.GRAY)
        lu = Resize(lu, w, h, kernel='spline16', sx=0.25)
        lv = Resize(lv, w, h, kernel='spline16', sx=0.25)
        rkernel = core.std.ShufflePlanes([ly,lu,lv], planes=[0,0,0], colorfamily=vs.YUV)
    res_16 = rkernel

    # Dithering
    result = res_16 if bits == 16 else core.fmtc.bitdepth(res_16, bits=bits, planes=planes, dmode=mode, ampo=ampo,
                                                          ampn=ampn, dyn=dyn, staticnoise=staticnoise, patsize=pat)

    if debug:
        last = dmask
        if bits != 16:
            last = core.fmtc.bitdepth(last, bits=bits)
    else:
        last = result
    return last


# GradFun3 alias
GradFun3mod = GradFun3

# GradFun3 alias
gf3 = GradFun3


"""
VapourSynth port of DebilinearM

Currently only YUV420 and YUV422 input makes sense

Differences:

 - changed the cubic argument to descale_kernel,
   so that this function is not limited to bilinear or bicubic
 - chroma is never scaled with an inverted kernel
 - added yuv444 argument to convert to yuv444
 - added arguments to fine tune the resizers

Usage:

  It is recommended to use the function alias for the desired kernel:

    DebilinearM(clip, 1280, 720)
    DebicubicM(clip, 1280, 720, b=0, c=0.5)
    DelanczosM(clip, 1280, 720, taps=3)
    Despline16M(clip, 1280, 720)
    Despline36M(clip, 1280, 720)

Original header:

DebilinearM is a wrapper function for the Debilinear and Debicubic plugins that masks parts of the frame that aren't upscaled,
such as text overlays, and uses a regular resize kernel to downscale those areas. It works by downscaling the input
clip to the target resolution with Debilinear or Debicubic, upscaling it again, comparing it to the original clip,
and masking pixels that have a difference greater than the specified threshold.

"""
def DescaleM(src, w, h, thr=None, expand=None, inflate=None, descale_kernel=None, kernel=None, kernely=None, kerneluv=None,
             taps=None, tapsy=None, tapsuv=None, a1=None, a2=None, a1y=None, a2y=None, a1uv=None, a2uv=None, b=None, c=None,
             chroma=None, yuv444=None, showmask=None, ow=None, oh=None):

    # Type checking
    kwargsdict = {'src': [src, (vs.VideoNode,)], 'w': [w, (int,)], 'h': [h, (int,)], 'thr': [thr, (int,)],
                  'expand': [expand, (int,)], 'inflate': [inflate, (int,)],'descale_kernel': [descale_kernel, (str,)],
                  'kernel': [kernel, (str,)], 'kernely': [kernely, (str,)], 'kerneluv': [kerneluv, (str,)],
                  'taps': [taps, (int,)], 'tapsy': [tapsy, (int,)], 'tapsuv': [tapsuv, (int,)], 'a1': [a1, (int, float)],
                  'a2': [a2, (int, float)], 'a1y': [a1y, (int, float)], 'a2y': [a2y, (int, float)],
                  'a1uv': [a1uv, (int, float)], 'a2uv': [a2uv, (int, float)], 'b': [b, (int, float)],
                  'c': [c, (int, float)], 'chroma': [chroma, (bool,)], 'yuv444': [yuv444, (bool,)],
                  'showmask': [showmask, (int,)], 'ow': [ow, (int,)], 'oh': [oh, (int,)]}

    for k, v in kwargsdict.items():
        if v[0] is not None and not isinstance(v[0], v[1]):
            raise TypeError('DescaleM: "{variable}" must be {types}!'
                            .format(variable=k, types=' or '.join([TYPEDICT[t] for t in v[1]])))

    # Set defaults
    if thr is None:
        thr = 10
    if expand is None:
        expand = 1
    if inflate is None:
        inflate = 2
    if chroma is None:
        chroma = True
    if src.format.num_planes == 1:
        chroma = False
    if yuv444 is None:
        yuv444 = False
    if showmask is None:
        showmask = 0
    if descale_kernel is None:
        descale_kernel = 'bilinear'
    elif descale_kernel.lower().startswith('de'):
        descale_kernel = descale_kernel[2:]
    if kernely is None:
        kernely = kernel
    if kerneluv is None:
        kerneluv = kernel
    if tapsy is None:
        tapsy = taps
    if tapsuv is None:
        tapsuv = taps
    if a1y is None:
        a1y = a1
    if a2y is None:
        a2y = a2
    if a1uv is None:
        a1uv = a1
    if a2uv is None:
        a2uv = a2
    if ow is None:
        ow = w
    if oh is None:
        oh = h

    # Value checking
    if thr < 0 or thr > 0xFF:
        raise ValueError('DebilinearM: "thr" must be in the range of 0 and 255!')
    if showmask < 0 or showmask > 2:
        raise ValueError('DebilinearM: "showmask" must be 0, 1 or 2!')
    if yuv444 and not chroma:
        raise ValueError('DebilinearM: "yuv444=True" and "chroma=False" cannot be used at the same time!')

    src_w = src.width
    src_h = src.height

    bits = src.format.bits_per_sample
    sample_type = src.format.sample_type
    
    if sample_type == vs.INTEGER:
        maxvalue = (1 << bits) - 1
        thr = thr * maxvalue // 0xFF
    else:
        thr /= (235 - 16)

    # Resizing
    src_y = core.std.ShufflePlanes(src, planes=0, colorfamily=vs.GRAY)
    if chroma:
        src_u = core.std.ShufflePlanes(src, planes=1, colorfamily=vs.GRAY)
        src_v = core.std.ShufflePlanes(src, planes=2, colorfamily=vs.GRAY)

    dbi = Resize(src_y, w, h, kernel=descale_kernel, a1=b, a2=c, taps=taps, invks=True)
    dbi2 = Resize(dbi, src_w, src_h, kernel=descale_kernel, a1=b, a2=c, taps=taps)
    if (w, h) != (ow, oh):
        dbi = Resize(dbi, ow, oh, kernel=kernely, taps=tapsy, a1=a1y, a2=a2y)

    if chroma and yuv444:
        rs = Resize(src_y, ow, oh, kernel=kernely, taps=tapsy, a1=a1y, a2=a2y)
        rs_u = Resize(src_u, ow, oh, kernel=kerneluv, taps=tapsuv, a1=a1uv, a2=a2uv, sx=0.25)
        rs_v = Resize(src_v, ow, oh, kernel=kerneluv, taps=tapsuv, a1=a1uv, a2=a2uv, sx=0.25)
    else:
        rs = Resize(src if chroma else src_y, ow, oh, kernel=kernely, taps=tapsy, a1=a1y, a2=a2y)

    # Masking
    diffmask = core.std.Expr([src_y, dbi2], 'x y - abs')
    if showmask != 2:
        diffmask = Resize(diffmask, ow, oh, kernel='bilinear')
        diffmask = core.std.Binarize(diffmask, threshold=thr)
    for _ in range(expand):
        diffmask = core.std.Maximum(diffmask, planes=0)
    for _ in range(inflate):
        diffmask = core.std.Inflate(diffmask, planes=0)

    if chroma:
        merged = core.std.ShufflePlanes([dbi,rs_u,rs_v] if yuv444 else [dbi,rs], planes=[0,0,0] if yuv444 else [0,1,2], colorfamily=vs.YUV)
    else:
        merged = dbi

    if showmask > 0:
        out = diffmask
    else:
        if yuv444:
            rs = core.std.ShufflePlanes([rs,merged], planes=[0,1,2], colorfamily=vs.YUV)
        out = core.std.MaskedMerge(merged, rs, diffmask, planes=0)

    return out


# DescaleM alias
DebilinearM = partial(DescaleM, descale_kernel='bilinear')

# DescaleM alias
DebicubicM = partial(DescaleM, descale_kernel='bicubic')

# DescaleM alias
DelanczosM = partial(DescaleM, descale_kernel='lanczos')

# DescaleM alias
Despline16M = partial(DescaleM, descale_kernel='spline16')

# DescaleM alias
Despline36M = partial(DescaleM, descale_kernel='spline36')


"""
Wrapper for fmtconv to scale each plane individually to the same size and fix chroma shift

Will only produce correct results if input is YUV420 or YUV422 with left aligned chroma

"""
def Downscale444(clip, w=1280, h=720, kernely="spline36", kerneluv="spline16", tapsy=3, tapsuv=3, a1y=None,
                 a1uv=None, a2y=None, a2uv=None, a3y=None, a3uv=None, invks=False, invkstaps=None):
    y = core.std.ShufflePlanes(clip, planes=0, colorfamily=vs.GRAY)
    u = core.std.ShufflePlanes(clip, planes=1, colorfamily=vs.GRAY)
    v = core.std.ShufflePlanes(clip, planes=2, colorfamily=vs.GRAY)
    y = Resize(y, w=w, h=h, kernel=kernely, taps=tapsy, a1=a1y, a2=a2y, a3=a3y, invks=invks, invkstaps=invkstaps)
    u = Resize(u, w=w, h=h, kernel=kerneluv, taps=tapsuv, a1=a1uv, a2=a2uv, a3=a3uv, sx=0.25)
    v = Resize(v, w=w, h=h, kernel=kerneluv, taps=tapsuv, a1=a1uv, a2=a2uv, a3=a3uv, sx=0.25)
    out = core.std.ShufflePlanes(clips=[y,u,v], planes=[0,0,0], colorfamily=vs.YUV)
    return out


"""
VapourSynth port of JIVTC.
Original script by lovesyk (https://github.com/lovesyk/avisynth-scripts/blob/master/JIVTC.avsi)

JIVTC applies inverse telecine in a way to minimize artifacts often seen on Japanese
TV broadcasts followed by recalculating the fields that might still contain some.

Dependencies: yadifmod, nnedi3

clip   src:            Source clip. Has to be 60i (30000/1001).
int    pattern:        First frame of any clean-combed-combed-clean-clean sequence.
int    threshold (10): This setting controls with how much probability one field has to
                       look better than the other to recalculate the other one using it.
                       Since there is no point dropping a field on a still (detail loss)
                       or an action (both results will look bad) scene, keep this above 0.
bool   draft (false):  If set to true, skip recalculate step (which means keep 50% of bad fields).
clip   ivtced:         Can be used to supply a custom IVTCed clip.
                       Keep in mind that the default IVTC process gets rid of 50% of
                       bad fields which might be "restored" depending on your supplied clip.
string bobber:         Can be used to supply a custom bobber.
                       The less information the bobber uses from the other field,
                       the better the result will be.
bool show (false):     If set to true, mark those frames that were recalculated.

"""
def JIVTC(src, pattern, thr=10, draft=False, ivtced=None, bobber=None, show=False, tff=None):

    def calculate(n, f, ivtced, bobbed):
        diffprev = f[0].props.EvenDiff
        diffnext = f[1].props.OddDiff
        if diffnext > diffprev:
            prerecalc = core.std.SelectEvery(bobbed, 2, 0)
        else:
            prerecalc = core.std.SelectEvery(bobbed, 2, 1)
        if abs(diffprev - diffnext) * 0xFF < thr:
            return ivtced
        if show:
            prerecalc = core.text.Text(prerecalc, 'Recalculated')
        return prerecalc

    pattern = pattern % 5

    defivtc = core.std.SeparateFields(src, tff=tff).std.DoubleWeave()
    selectlist = [[0,3,6,8], [0,2,5,8], [0,2,4,7], [2,4,6,9], [1,4,6,8]]
    defivtc = core.std.SelectEvery(defivtc, 10, selectlist[pattern])

    ivtced = defivtc if ivtced is None else ivtced
    if bobber is None:
        bobbed = core.yadifmod.Yadifmod(ivtced, edeint=core.nnedi3.nnedi3(ivtced, 2), order=0, mode=1)
    else:
        bobbed = bobber(ivtced)

    if src.fps_num != 30000 or src.fps_den != 1001:
        raise ValueError('JIVTC: This filter can only be used with 60i clips.')
    if bobbed.fps_num != 48000 or bobbed.fps_den != 1001:
        raise ValueError('JIVTC: The bobber you specified does not double the frame rate.')

    sep = core.std.SeparateFields(ivtced)
    even = core.std.SelectEvery(sep, 2, 0)
    odd = core.std.SelectEvery(sep, 2, 1)
    diffeven = core.std.PlaneStats(even, even.std.DuplicateFrames([0]), prop='Even')
    diffodd = core.std.PlaneStats(odd, odd.std.DeleteFrames([0]), prop='Odd')
    recalc = core.std.FrameEval(ivtced, partial(calculate, ivtced=ivtced, bobbed=bobbed),
                                prop_src=[diffeven,diffodd])

    inter = core.std.Interleave([ivtced, recalc])
    selectlist = [[0,3,4,6], [0,2,5,6], [0,2,4,7], [0,2,4,7], [1,2,4,6]]
    final = core.std.SelectEvery(inter, 8, selectlist[pattern])

    out = ivtced if draft else final
    out = core.std.SetFrameProp(out, prop='_FieldBased', intval=0)
    return out


"""
VapourSynth port of OverlayInter

Based on the AviSynth script by Majin3 and the already ported
ivtc_txt60mc by Firesledge that can be found inside havsfunc

It's much faster than ivtc_txt60mc because you can limit processing
to a small part of the clip.

Original Header:
# OverlayInter 0.1 by Majin3 (06.09.2012)
# Converts 60i overlays (like scrolling credits) on top of telecined 24p video to 24p using motion interpolation.
# Required: MVTools2, QTGMC (if not using a custom bobber)
# int		pattern:			First frame of a clean-combed-combed-clean-clean sequence
# int		pos (0):			Overlay position: 0: whole screen - 1: left - 2: top - 3: right - 4: bottom
# int		size (0):			Overlay size in px from the corresponding position
# bool		show (false):		Enable this to show the area selected by "pos" and "size"
# bool		draft (false):		Enable this to speed up processing by using low-quality bobbing and motion interpolation
# string	bobber:				A custom bobber if you do not wish to use "QTGMC(Preset="Very Slow", SourceMatch=2, Lossless=2)"
# string	ivtc:				A custom IVTC if you do not wish to use simple IVTC based on "pattern"
# Based on ivtc_txt60mc 1.1 by Firesledge

"""
def OverlayInter(src, pattern, pos=0, size=0, show=False, draft=False, bobber=None, ivtc=None, tff=None):

    try:
        import havsfunc as haf
    except ImportError:
        raise ImportError('OverlayInter: havsfunc not found. Download it here: https://github.com/HomeOfVapourSynthEvolution/havsfunc')

    if bobber is None and not isinstance(tff, bool):
        raise TypeError('OverlayInter: "tff" must be set. Setting tff to True means top field first. False means bottom field first')

    field_ref = (pattern * 2) % 5
    invpos = (5 - field_ref) % 5
    pattern %= 5

    croplist = [[0,0,0,0], [size,0,0,0], [0,0,size,0], [0,size,0,0], [0,0,0,size]]
    keep = core.std.CropRel(src, *croplist[pos])

    croplist = [[0,0,0,0], [0,src.width-size-4,0,0], [0,0,0,src.height-size-4],
                [src.width-size-4,0,0,0], [0,0,src.height-size-4,0]]
    bobbed = core.std.CropRel(src, *croplist[pos])
    if draft:
        bobbed = haf.Bob(bobbed, tff=tff)
    elif bobber is None:
        bobbed = haf.QTGMC(bobbed, Preset='very slow', SourceMatch=3, Lossless=2, TFF=tff)
    else:
        bobbed = bobber(bobbed)

    if ivtc is None:
        ivtclist = [[0,3,6,8], [0,2,5,8], [0,2,4,7], [2,4,6,9], [1,4,6,8]]
        ivtc = keep.std.SeparateFields(tff=tff).std.DoubleWeave().std.SelectEvery(10, ivtclist[pattern])
    else:
        ivtc = ivtc(keep)

    if invpos > 1:
        clean = core.std.AssumeFPS(bobbed[0] + core.std.SelectEvery(bobbed, 5, [6 - invpos]),
                                   fpsnum=12000, fpsden=1001)
    else:
        clean = core.std.SelectEvery(bobbed, 5, [1 - invpos])
    if invpos > 3:
        jitter = core.std.AssumeFPS(bobbed[0] + core.std.SelectEvery(bobbed, 5, [4 - invpos, 8 - invpos]),
                                    fpsnum=24000, fpsden=1001)
    else:
        jitter = core.std.SelectEvery(bobbed, 5, [3 - invpos, 4 - invpos])

    jsup = core.mv.Super(jitter)
    vecsup = haf.DitherLumaRebuild(jitter, s0=1).mv.Super(rfilter=4)
    vectb = core.mv.Analyse(jsup if draft else vecsup, overlap=0 if draft else 4, blksize=16, isb=True)
    if not draft:
        vectb = core.mv.Recalculate(vecsup, vectb, blksize=8, overlap=2)
    vectf = core.mv.Analyse(jsup if draft else vecsup, overlap=0 if draft else 4, blksize=16, isb=False)
    if not draft:
        vectf = core.mv.Recalculate(vecsup, vectf, blksize=8, overlap=2)
    comp = core.mv.FlowInter(jitter, jsup, vectb, vectf)
    fixed = core.std.SelectEvery(comp, 2, 0)
    fixed = core.std.Interleave([clean,fixed])[invpos // 2:]

    croplist = [[0,0,0,0], [0,4,0,0], [0,0,0,4], [4,0,0,0], [0,0,4,0]]
    fixed = core.std.CropRel(fixed, *croplist[pos])

    if show:
        maxvalue = (1 << src.format.bits_per_sample) - 1
        offset = 32 * maxvalue // 0xFF if fixed.format.sample_type == vs.INTEGER else 32 / 0xFF
        fixed = core.std.Expr(fixed, ['','x {} +'.format(offset),''])

    if pos == 1:
        out = core.std.StackHorizontal([fixed,ivtc])
    elif pos == 2:
        out = core.std.StackVertical([fixed,ivtc])
    elif pos == 3:
        out = core.std.StackHorizontal([ivtc,fixed])
    elif pos == 4:
        out = core.std.StackVertical([ivtc,fixed])
    else:
        out = fixed
    out = core.std.SetFrameProp(out, prop='_FieldBased', intval=0)

    return out


"""
VapourSynth port of AutoDeblock2. Original script by joletb, vinylfreak89, eXmendiC and Gebbi.

The purpose of this script is to automatically remove MPEG2 artifacts.

Supports 8..16 bit integer YUV formats

"""
def AutoDeblock(src, edgevalue=24, db1=1, db2=6, db3=15, deblocky=True, deblockuv=True, debug=False, redfix=False,
                fastdeblock=False, adb1=3, adb2=4, adb3=8, adb1d=2, adb2d=7, adb3d=11, planes=None):

    try:
        import havsfunc as haf
    except ImportError:
        raise ImportError('AutoDeblock: havsfunc not found. Download it here: https://github.com/HomeOfVapourSynthEvolution/havsfunc')

    if src.format.color_family not in [vs.YUV]:
        raise TypeError("AutoDeblock: src must be YUV color family!")

    if src.format.bits_per_sample < 8 or src.format.bits_per_sample > 16 or src.format.sample_type != vs.INTEGER:
        raise TypeError("AutoDeblock: src must be between 8 and 16 bit integer format")

    # Scale values to handle high bit depths
    shift = src.format.bits_per_sample - 8
    edgevalue = edgevalue << shift
    maxvalue = (1 << src.format.bits_per_sample) - 1

    # Scales the output of PlaneStats (which is a float, 0-1) to 8 bit values.
    # We scale to 8 bit because all thresholds/parameters for this function are
    # specified in an 8-bit scale.
    # All processing still happens in the native bit depth of the input format.
    def to8bit(f):
        return f * 255

    def sub_props(src, f, name):

        OrigDiff_str = str(to8bit(f[0].props.OrigDiff))
        YNextDiff_str = str(to8bit(f[1].props.YNextDiff))
        return core.sub.Subtitle(src, name + f"\nOrigDiff: {OrigDiff_str}\nYNextDiff: {YNextDiff_str}")

    def eval_deblock_strength(n, f, fastdeblock, debug, unfiltered, fast, weakdeblock,
                              mediumdeblock, strongdeblock):
        unfiltered = sub_props(unfiltered, f, "unfiltered") if debug else unfiltered
        out = unfiltered
        if fastdeblock:
            if to8bit(f[0].props.OrigDiff) > adb1 and to8bit(f[1].props.YNextDiff) > adb1d:
                return sub_props(fast, f, "deblock") if debug else fast
            else:
                return unfiltered
        if to8bit(f[0].props.OrigDiff) > adb1 and to8bit(f[1].props.YNextDiff) > adb1d:
            out = sub_props(weakdeblock, f, "weakdeblock") if debug else weakdeblock
        if to8bit(f[0].props.OrigDiff) > adb2 and to8bit(f[1].props.YNextDiff) > adb2d:
            out = sub_props(mediumdeblock, f, "mediumdeblock") if debug else mediumdeblock
        if to8bit(f[0].props.OrigDiff) > adb3 and to8bit(f[1].props.YNextDiff) > adb3d:
            out = sub_props(strongdeblock, f, "strongdeblock") if debug else strongdeblock
        return out

    def fix_red(n, f, unfiltered, autodeblock):
        if (to8bit(f[0].props.YAverage) > 50 and to8bit(f[0].props.YAverage) < 130
                and to8bit(f[1].props.UAverage) > 95 and to8bit(f[1].props.UAverage) < 130
                and to8bit(f[2].props.VAverage) > 130 and to8bit(f[2].props.VAverage) < 155):
            return unfiltered
        return autodeblock

    if redfix and fastdeblock:
        raise ValueError('AutoDeblock: You cannot set both "redfix" and "fastdeblock" to True!')

    if planes is None:
        planes = []
        if deblocky: planes.append(0)
        if deblockuv: planes.extend([1,2])

    orig = core.std.Prewitt(src)
    orig = core.std.Expr(orig, f"x {edgevalue} >= {maxvalue} x ?")
    orig_d = orig.rgvs.RemoveGrain(4).rgvs.RemoveGrain(4)

    predeblock = haf.Deblock_QED(src.rgvs.RemoveGrain(2).rgvs.RemoveGrain(2))
    fast = core.dfttest.DFTTest(predeblock, tbsize=1)

    unfiltered = src
    weakdeblock = core.dfttest.DFTTest(predeblock, sigma=db1, tbsize=1, planes=planes)
    mediumdeblock = core.dfttest.DFTTest(predeblock, sigma=db2, tbsize=1, planes=planes)
    strongdeblock = core.dfttest.DFTTest(predeblock, sigma=db3, tbsize=1, planes=planes)

    difforig = core.std.PlaneStats(orig, orig_d, prop='Orig')
    diffnext = core.std.PlaneStats(src, src.std.DeleteFrames([0]), prop='YNext')
    autodeblock = core.std.FrameEval(unfiltered, partial(eval_deblock_strength, fastdeblock=fastdeblock,
                                     debug=debug, unfiltered=unfiltered, fast=fast, weakdeblock=weakdeblock,
                                     mediumdeblock=mediumdeblock, strongdeblock=strongdeblock),
                                     prop_src=[difforig,diffnext])

    if redfix:
        src = core.std.PlaneStats(src, prop='Y')
        src_u = core.std.PlaneStats(src, plane=1, prop='U')
        src_v = core.std.PlaneStats(src, plane=2, prop='V')
        autodeblock = core.std.FrameEval(unfiltered, partial(fix_red, unfiltered=unfiltered,
                                         autodeblock=autodeblock), prop_src=[src,src_u,src_v])

    return autodeblock


"""
Basically a wrapper for std.Trim and std.Splice that recreates the functionality of
AviSynth's ReplaceFramesSimple (http://avisynth.nl/index.php/RemapFrames)
that was part of the plugin RemapFrames by James D. Lin

Usage: ReplaceFrames(clipa, clipb, mappings="[200 300] [1100 1150] 400 1234")

This will replace frames 200..300, 1100..1150, 400 and 1234 from clipa with
the corresponding frames from clipb.

"""
def ReplaceFrames(clipa, clipb, mappings=None, filename=None):

    if not isinstance(clipa, vs.VideoNode):
        raise TypeError('ReplaceFrames: "clipa" must be a clip!')
    if not isinstance(clipb, vs.VideoNode):
        raise TypeError('ReplaceFrames: "clipb" must be a clip!')
    if clipa.format.id != clipb.format.id:
        raise TypeError('ReplaceFrames: "clipa" and "clipb" must have the same format!')
    if filename is not None and not isinstance(filename, str):
        raise TypeError('ReplaceFrames: "filename" must be a string!')
    if mappings is not None and not isinstance(mappings, str):
        raise TypeError('ReplaceFrames: "mappings" must be a string!')
    if mappings is None:
        mappings = ''

    if filename:
        with open(filename, 'r') as mf:
            mappings += '\n{}'.format(mf.read())
    # Some people used this as separators and wondered why it wasn't working
    mappings = mappings.replace(',', ' ').replace(':', ' ')

    frames = re.findall('\d+(?!\d*\s*\d*\s*\d*\])', mappings)
    ranges = re.findall('\[\s*\d+\s+\d+\s*\]', mappings)
    maps = []
    for range_ in ranges:
        maps.append([int(x) for x in range_.strip('[ ]').split()])
    for frame in frames:
        maps.append([int(frame), int(frame)])

    for start, end in maps:
        if start > end:
            raise ValueError('ReplaceFrames: Start frame is bigger than end frame: [{} {}]'.format(start, end))
        if end >= clipa.num_frames or end >= clipb.num_frames:
            raise ValueError('ReplaceFrames: End frame too big, one of the clips has less frames: {}'.format(end)) 

    out = clipa
    for start, end in maps:
        temp = clipb[start:end+1] 
        if start != 0:
            temp = out[:start] + temp
        if end < out.num_frames - 1:
            temp = temp + out[end+1:]
        out = temp
    return out


# ReplaceFrames alias
ReplaceFramesSimple = ReplaceFrames

# ReplaceFrames alias
rfs = ReplaceFrames


"""
This overlays a clip onto another.
Default matrix for RGB -> YUV conversion is "709"
Use "601" if you want to mimic AviSynth's Overlay()
overlay should be a list of [video, mask] or a path string to an RGBA file
If you specifiy a clip instead then a mask with max value will be generated
(RGBA videos opened by ffms2 with alpha=True are already such a list)
"""
def InsertSign(clip, overlay, start, end=None, matrix='709'):
    if start < 0:
        raise ValueError('InsertSign: "start" must not be lower than 0!')
    if isinstance(overlay, str):
        overlay = core.ffms2.Source(overlay, alpha=True)
        overlay = [overlay, overlay.std.PropToClip('_Alpha')]
    if not isinstance(overlay, list):
        overlay = [overlay, None]
    if end is None:
        end = start + overlay[0].num_frames
    else:
        end += 1
    if end > clip.num_frames:
        end = clip.num_frames
    if start >= end:
        raise ValueError('InsertSign: "start" must be smaller than or equal to "end"!')
    if matrix == '601':
        matrix = '470bg'
    clip_cf = clip.format.color_family
    overlay_cf = overlay[0].format.color_family
    before = clip[:start] if start != 0 else None
    middle = clip[start:end]
    after = clip[end:] if end != clip.num_frames else None
    matrix_s = None
    matrix_in_s = None
    if clip_cf == vs.YUV and overlay_cf == vs.RGB:
        matrix_s = matrix
    if overlay_cf == vs.YUV and clip_cf == vs.RGB:
        matrix_in_s = matrix
    sign = core.resize.Spline36(overlay[0], clip.width, clip.height, format=clip.format.id,
                                matrix_s=matrix_s, matrix_in_s=matrix_in_s,
                                dither_type='error_diffusion')
    if overlay[1] is None:
        overlay[1] = core.std.BlankClip(sign, format=vs.GRAY8, color=255)
    mask = core.resize.Bicubic(overlay[1], clip.width, clip.height)
    mask = Depth(mask, bits=clip.format.bits_per_sample, range='full', range_in='full')
    middle = core.std.MaskedMerge(middle, sign, mask)
    out = middle
    if before is not None:
        out = before + out
    if after is not None:
        out = out + after
    return out


"""
Downscale only lineart with an inverted kernel and interpolate
it back to its original resolution with NNEDI3.

Parts of higher resolution like credits are protected by a mask.

Basic idea stolen from a script made by Daiz.

"""
def DescaleAA(src, w=1280, h=720, thr=10, kernel='bilinear', b=1/3, c=1/3, taps=3,
              expand=3, inflate=3, showmask=False):

    try:
        import nnedi3_rpow2 as nnp2
    except ImportError:
        raise ImportError('DescaleAA: nnedi3_rpow2 not found. Download it here: https://gist.github.com/4re/342624c9e1a144a696c6')

    if kernel.lower().startswith('de'):
        kernel = kernel[2:]

    ow = src.width
    oh = src.height

    bits = src.format.bits_per_sample
    sample_type = src.format.sample_type
    
    if sample_type == vs.INTEGER:
        maxvalue = (1 << bits) - 1
        thr = thr * maxvalue // 0xFF
    else:
        maxvalue = 1
        thr /= (235 - 16)

    # Fix lineart
    src_y = core.std.ShufflePlanes(src, planes=0, colorfamily=vs.GRAY)
    deb = Resize(src_y, w, h, kernel=kernel, a1=b, a2=c, taps=taps, invks=True)
    sharp = nnp2.nnedi3_rpow2(deb, 2, ow, oh)
    thrlow = 4 * maxvalue // 0xFF if sample_type == vs.INTEGER else 4 / 0xFF
    thrhigh = 24 * maxvalue // 0xFF if sample_type == vs.INTEGER else 24 / 0xFF
    edgemask = core.std.Prewitt(sharp, planes=0)
    edgemask = core.std.Expr(edgemask, "x {thrhigh} >= {maxvalue} x {thrlow} <= 0 x ? ?"
                                       .format(thrhigh=thrhigh, maxvalue=maxvalue, thrlow=thrlow))
    if kernel == "bicubic" and c >= 0.7:
        edgemask = core.std.Maximum(edgemask, planes=0)
    sharp = core.resize.Point(sharp, format=src.format.id)

    # Restore true 1080p
    deb_upscale = Resize(deb, ow, oh, kernel=kernel, a1=b, a2=c, taps=taps)
    diffmask = core.std.Expr([src_y, deb_upscale], 'x y - abs')
    for _ in range(expand):
        diffmask = core.std.Maximum(diffmask, planes=0)
    for _ in range(inflate):
        diffmask = core.std.Inflate(diffmask, planes=0)

    mask = core.std.Expr([diffmask,edgemask], 'x {thr} >= 0 y ?'.format(thr=thr))
    mask = mask.std.Inflate().std.Deflate()
    out = core.std.MaskedMerge(src, sharp, mask, planes=0)

    if showmask:
        out = mask

    return out


# Legacy DescaleAA alias
def ProtectedDebiXAA(src, w=1280, h=720, thr=10, expand=3, inflate=3,
                     bicubic=False, b=1/3, c=1/3, showmask=False, bits=None):

    if bicubic:
        return DescaleAA(src, w=w, h=h, thr=thr, kernel='bicubic', b=b, c=c, taps=None,
                         expand=expand, inflate=inflate, showmask=showmask)
    else:
        return DescaleAA(src, w=w, h=h, thr=thr, kernel='bilinear', b=None, c=None, taps=None,
                         expand=expand, inflate=inflate, showmask=showmask)


"""
VapourSynth port of AviSynth's maa2 (https://github.com/AviSynth/avs-scripts)

Works on any bitdepth

"""
def maa(src, mask=None, chroma=None, ss=None, aa=None, aac=None, show=None):

    try:
        import mvsfunc as mvf
    except ImportError:
        raise ImportError('maa: mvsfunc not found. Download it here: https://github.com/HomeOfVapourSynthEvolution/mvsfunc')

    def SangNomAA(src, ss=2.0, aa=48, aac=None):
        ss_w = round(src.width * ss / 4) * 4
        ss_h = round(src.height * ss / 4) * 4
        out = core.resize.Spline36(src, ss_w, ss_h).std.Transpose()
        out = core.sangnom.SangNom(out, aa=aa if aac is None else [aa, aac, aac]).std.Transpose()
        out = core.sangnom.SangNom(out, aa=aa if aac is None else [aa, aac, aac])
        out = core.resize.Spline36(out, src.width, src.height)
        return out

    # Type checking
    kwargsdict = {'src': [src, (vs.VideoNode,)], 'mask': [mask, (int,)], 'ss': [ss, (int, float)],
                  'aa': [aa, (int,)], 'aac': [aac, (int,)], 'show': [show, (bool,)]}

    for k, v in kwargsdict.items():
        if v[0] is not None and not isinstance(v[0], v[1]):
            raise TypeError('maa: "{variable}" must be {types}!'
                            .format(variable=k, types=' or '.join([TYPEDICT[t] for t in v[1]])))

    # Set defaults
    if mask is None:
        mask = 1
    if chroma is None:
        chroma = False
    if ss is None:
        ss = 2.0
    if aa is None:
        aa = 48
    if aac is None:
        aac = aa - 8
    if show is None:
        show = False

    # Value checking
    if mask < -0xFF or mask > 1:
        raise ValueError('maa: "mask" must be between -255 and 1!')
    if ss <= 0:
        raise ValueError('maa: "ss" must be > 0!')

    bits = src.format.bits_per_sample
    sample_type = src.format.sample_type
    maxvalue = (1 << bits) - 1
    
    if sample_type == vs.INTEGER:
        mthresh = -mask * maxvalue // 0xFF if mask < 0 else 7 * maxvalue // 0xFF
        mthreshc = mthresh - 6 * maxvalue // 0xFF
    else:
        mthresh = -mask / 0xFF if mask < 0 else 7 / 0xFF
        mthreshc = mthresh - 6 / 0xFF

    if src.format.num_planes == 1:
        chroma = False

    if mask != 0:
        m = core.std.Sobel(mvf.GetPlane(src, 0))
        m = core.std.Binarize(m, mthresh)
        if chroma:
            mu = core.std.Sobel(mvf.GetPlane(src, 1))
            mu = core.std.Binarize(mu, mthreshc)
            mv = core.std.Sobel(mvf.GetPlane(src, 2))
            mv = core.std.Binarize(mv, mthreshc)
            m = core.std.ShufflePlanes([m,mu,mv], planes=[0,0,0], colorfamily=vs.YUV)
    if not chroma:
        c_aa = SangNomAA(mvf.GetPlane(src, 0), ss, aa)
    else:
        c_aa = SangNomAA(src, ss, aa, aac)

    if not chroma and src.format.num_planes != 1:
        c_aa = core.std.ShufflePlanes([c_aa, mvf.GetPlane(src, 1), mvf.GetPlane(src, 2)], planes=[0,0,0], colorfamily=vs.YUV)
    if mask == 0:
        out = c_aa
    elif show:
        out = m
    else:
        out = core.std.MaskedMerge(src, c_aa, m)
    return out


"""
VapourSynth port of TemporalDegrain (http://avisynth.nl/index.php/Temporal_Degrain)

(only 8 bit YUV input)

Differences:

- all keyword arguments are now lowercase
- hq > 0 is not implemented
- gpu=True is not implemented

"""
def TemporalDegrain(input_, denoise=None, gpu=False, sigma=16, bw=16, bh=16, pel=2,
                    blksize=8, ov=None, degrain=2, limit=255, sad1=400, sad2=300, hq=0):

    try:
        import havsfunc as haf
    except ImportError:
        raise ImportError('TemporalDegrain: havsfunc not found. Download it here: https://github.com/HomeOfVapourSynthEvolution/havsfunc')

    if not isinstance(input_, vs.VideoNode):
        raise TypeError('TemporalDegrain: "input_" must be a clip!')
    if denoise is not None and not isinstance(denoise, vs.VideoNode):
        raise TypeError('TemporalDegrain: "denoise" must be a clip!')
    if not isinstance(gpu, bool):
        raise TypeError('TemporalDegrain: "gpu" must be a bool!')
    if gpu:
        raise NotImplementedError('TemporalDegrain: "gpu=True" is not implemented!')
    if not isinstance(sigma, int):
        raise TypeError('TemporalDegrain: "sigma" must be an int!')
    if not isinstance(bw, int):
        raise TypeError('TemporalDegrain: "bw" must be an int!')
    if not isinstance(bh, int):
        raise TypeError('TemporalDegrain: "bh" must be an int!')
    if not isinstance(pel, int):
        raise TypeError('TemporalDegrain: "pel" must be an int!')
    if not isinstance(blksize, int):
        raise TypeError('TemporalDegrain: "blksize" must be an int!')
    if ov is not None and not isinstance(ov, int):
        raise TypeError('TemporalDegrain: "ov" must be an int!')
    if not isinstance(degrain, int):
        raise TypeError('TemporalDegrain: "degrain" must be an int!')
    if not isinstance(limit, int):
        raise TypeError('TemporalDegrain: "limit" must be an int!')
    if not isinstance(sad1, int):
        raise TypeError('TemporalDegrain: "sad1" must be an int!')
    if not isinstance(sad2, int):
        raise TypeError('TemporalDegrain: "sad2" must be an int!')
    if not isinstance(hq, int):
        raise TypeError('TemporalDegrain: "hq" must be an int!')
    if hq > 0:
        raise NotImplementedError('TemporalDegrain: "hq" > 0 is not implemented!')

    o = input_
    s2 = int(sigma * 0.625)
    s3 = int(sigma * 0.375)
    s4 = int(sigma * 0.250)
    ow = int(bw / 2)
    oh = int(bh / 2)
    ov = int(blksize / 2) if not ov or ov*2 > blksize else ov

    if denoise:
        filter_ = denoise
    elif gpu:
        filter_ = o.FFT3DGPU(sigma=sigma, sigma2=s2 , sigma3=s3, sigma4=s4, bt=4, bw=bw, bh=bh, ow=ow, oh=oh)  # not implemented
    else:
        filter_ = core.fft3dfilter.FFT3DFilter(o, sigma=sigma, sigma2=s2, sigma3=s3, sigma4=s4, bt=4, bw=bw, bh=bh, ow=ow, oh=oh)
    if hq >= 1:
        filter_ = filter_.HQdn3D(4,3,6,3)  # not implemented

    spat = filter_
    spatd = core.std.MakeDiff(o, spat)

    srch = filter_
    srch_super = core.mv.Super(filter_, pel=pel)

    if degrain == 3:
        bvec3 = core.mv.Analyse(srch_super, isb=True, delta=3, blksize=blksize, overlap=ov)
    else:
        bvec3 = core.std.BlankClip()
    if degrain >= 2:
        bvec2 = core.mv.Analyse(srch_super, isb=True, delta=2, blksize=blksize, overlap=ov)
    else:
        bvec2 = core.std.BlankClip()
    bvec1 = core.mv.Analyse(srch_super, isb=True,  delta=1, blksize=blksize, overlap=ov)
    fvec1  = core.mv.Analyse(srch_super, isb=False, delta=1, blksize=blksize, overlap=ov)
    if degrain >= 2:
        fvec2 = core.mv.Analyse(srch_super, isb=False, delta=2, blksize=blksize, overlap=ov)
    else:
        fvec2 = core.std.BlankClip()
    if degrain == 3:
        fvec3 = core.mv.Analyse(srch_super, isb=False, delta=3, blksize=blksize, overlap=ov)
    else:
        fvec3 = core.std.BlankClip()

    o_super = core.mv.Super(o, pel=2, levels=1)

    if degrain == 3:
        nr1 = core.mv.Degrain3(o, o_super, bvec1, fvec1, bvec2, fvec2, bvec3, fvec3, thsad=sad1, limit=limit)
    elif degrain == 2:
        nr1 = core.mv.Degrain2(o, o_super, bvec1, fvec1, bvec2, fvec2, thsad=sad1, limit=limit)
    else:
        nr1 = core.mv.Degrain1(o, o_super, bvec1, fvec1, thsad=sad1, limit=limit)
    nr1d = core.std.MakeDiff(o, nr1)

    dd = core.std.Expr([spatd, nr1d], 'x 128 - abs y 128 - abs < x y ?')
    nr1x = core.std.MakeDiff(o, dd, planes=0)

    nr1x_super = core.mv.Super(nr1x, pel=2, levels=1)

    if degrain == 3:
        nr2 = core.mv.Degrain3(nr1x, nr1x_super, bvec1, fvec1, bvec2, fvec2, bvec3, fvec3, thsad=sad2, limit=limit)
    elif degrain == 2:
        nr2 = core.mv.Degrain2(nr1x, nr1x_super, bvec1, fvec1, bvec2, fvec2, thsad=sad2, limit=limit)
    else:
        nr2 = core.mv.Degrain1(nr1x, nr1x_super, bvec1, fvec1, thsad=sad2, limit=limit)

    if hq >= 2:
        nr2.HQDn3D(0,0,4,1)  # not implemented

    s = haf.MinBlur(nr2, 1, 0)
    alld = core.std.MakeDiff(o, nr2)
    temp = core.rgvs.RemoveGrain(s, [11, 0])
    ssd = core.std.MakeDiff(s, temp)
    ssdd = core.rgvs.Repair(ssd, alld, 1)
    ssdd = core.std.Expr([ssdd, ssd], 'x 128 - abs y 128 - abs < x y ?')

    output = core.std.MergeDiff(nr2, ssdd, planes=0)
    return output


# Helpers

# Wrapper with fmtconv syntax that tries to use the internal resizers whenever it is possible
def Resize(src, w, h, sx=None, sy=None, sw=None, sh=None, kernel='spline36', taps=None, a1=None,
             a2=None, a3=None, invks=None, invkstaps=None, fulls=None, fulld=None):

    bits = src.format.bits_per_sample

    if (src.width, src.height, fulls) == (w, h, fulld):
        return src

    if kernel is None:
        kernel = 'spline36'
    kernel = kernel.lower()

    if invks and kernel == 'bilinear' and hasattr(core, 'unresize') and invkstaps is None:
        return core.unresize.Unresize(src, w, h, src_left=sx, src_top=sy)
    if invks and kernel in ['bilinear', 'bicubic', 'lanczos', 'spline16', 'spline36', 'spline64'] and hasattr(core, 'descale') and invkstaps is None:
        return Descale(src, w, h, kernel=kernel, b=a1, c=a2, taps=taps)
    if not invks:
        if kernel == 'bilinear':
            return core.resize.Bilinear(src, w, h, range=fulld, range_in=fulls, src_left=sx, src_top=sy,
                                        src_width=sw, src_height=sh)
        if kernel == 'bicubic':
            return core.resize.Bicubic(src, w, h, range=fulld, range_in=fulls, filter_param_a=a1, filter_param_b=a2,
                                       src_left=sx, src_top=sy, src_width=sw, src_height=sh)
        if kernel == 'spline16':
            return core.resize.Spline16(src, w, h, range=fulld, range_in=fulls, src_left=sx, src_top=sy,
                                        src_width=sw, src_height=sh)
        if kernel == 'spline36':
            return core.resize.Spline36(src, w, h, range=fulld, range_in=fulls, src_left=sx, src_top=sy,
                                        src_width=sw, src_height=sh)
        if kernel == 'spline64':
            return core.resize.Spline64(src, w, h, range=fulld, range_in=fulls, src_left=sx, src_top=sy,
                                        src_width=sw, src_height=sh)
        if kernel == 'lanczos':
            return core.resize.Lanczos(src, w, h, range=fulld, range_in=fulls, filter_param_a=taps,
                                       src_left=sx, src_top=sy, src_width=sw, src_height=sh)
    return Depth(core.fmtc.resample(src, w, h, sx=sx, sy=sy, sw=sw, sh=sh, kernel=kernel, taps=taps,
                              a1=a1, a2=a2, a3=a3, invks=invks, invkstaps=invkstaps, fulls=fulls, fulld=fulld), bits)


def Debilinear(src, width, height, yuv444=False, gray=False, chromaloc=None):
    return Descale(src, width, height, kernel='bilinear', taps=None, b=None, c=None, yuv444=yuv444, gray=gray, chromaloc=chromaloc)

def Debicubic(src, width, height, b=0.0, c=0.5, yuv444=False, gray=False, chromaloc=None):
    return Descale(src, width, height, kernel='bicubic', taps=None, b=b, c=c, yuv444=yuv444, gray=gray, chromaloc=chromaloc)

def Delanczos(src, width, height, taps=3, yuv444=False, gray=False, chromaloc=None):
    return Descale(src, width, height, kernel='lanczos', taps=taps, b=None, c=None, yuv444=yuv444, gray=gray, chromaloc=chromaloc)

def Despline16(src, width, height, yuv444=False, gray=False, chromaloc=None):
    return Descale(src, width, height, kernel='spline16', taps=None, b=None, c=None, yuv444=yuv444, gray=gray, chromaloc=chromaloc)

def Despline36(src, width, height, yuv444=False, gray=False, chromaloc=None):
    return Descale(src, width, height, kernel='spline36', taps=None, b=None, c=None, yuv444=yuv444, gray=gray, chromaloc=chromaloc)

def Despline64(src, width, height, yuv444=False, gray=False, chromaloc=None):
    return Descale(src, width, height, kernel='spline64', taps=None, b=None, c=None, yuv444=yuv444, gray=gray, chromaloc=chromaloc)


def Descale(src, width, height, kernel=None, custom_kernel=None, taps=None, b=None, c=None, yuv444=False, gray=False, chromaloc=None):
    src_f = src.format
    src_cf = src_f.color_family
    src_st = src_f.sample_type
    src_bits = src_f.bits_per_sample
    src_sw = src_f.subsampling_w
    src_sh = src_f.subsampling_h

    if src_cf == vs.RGB and not gray:
        rgb = to_rgbs(src).descale.Descale(width, height, kernel, custom_kernel, taps, b, c)
        return rgb.resize.Point(format=src_f.id)

    y = to_grays(src).descale.Descale(width, height, kernel, custom_kernel, taps, b, c)
    y_f = core.query_video_format(vs.GRAY, src_st, src_bits, 0, 0)
    y = y.resize.Point(format=y_f.id)

    if src_cf == vs.GRAY or gray:
        return y

    if not yuv444 and ((width % 2 and src_sw) or (height % 2 and src_sh)):
        raise ValueError('Descale: The output dimension and the subsampling are incompatible.')

    uv_f = core.query_video_format(src_cf, src_st, src_bits, 0 if yuv444 else src_sw, 0 if yuv444 else src_sh)
    uv = src.resize.Spline36(width, height, format=uv_f.id, chromaloc_s=chromaloc)

    return core.std.ShufflePlanes([y,uv], [0,1,2], vs.YUV)


def to_grays(src):
    return src.resize.Point(format=vs.GRAYS)


def to_rgbs(src):
    return src.resize.Point(format=vs.RGBS)


def get_plane(src, plane):
    return core.std.ShufflePlanes(src, plane, vs.GRAY)


def Depth(src, bits, dither_type='error_diffusion', range=None, range_in=None):
    src_f = src.format
    src_cf = src_f.color_family
    src_st = src_f.sample_type
    src_bits = src_f.bits_per_sample
    src_sw = src_f.subsampling_w
    src_sh = src_f.subsampling_h
    dst_st = vs.INTEGER if bits < 32 else vs.FLOAT

    if isinstance(range, str):
        range = RANGEDICT[range]

    if isinstance(range_in, str):
        range_in = RANGEDICT[range_in]

    if (src_bits, range_in) == (bits, range):
        return src
    out_f = core.query_video_format(src_cf, dst_st, bits, src_sw, src_sh)
    return core.resize.Point(src, format=out_f.id, dither_type=dither_type, range=range, range_in=range_in)


TYPEDICT = {vs.VideoNode: 'a clip', int: 'an int', float: 'a float', bool: 'a bool', str: 'a str', list: 'a list'}

RANGEDICT = {'limited': 0, 'full': 1}
Editor is loading...