soifunc 0.14.0__py3-none-any.whl → 0.14.1__py3-none-any.whl

soifunc/denoise.py

@@ -4,9 +4,9 @@ from typing import Callable, Optional
 
 import vsdenoise
 from vsdenoise import DFTTest, bm3d, mc_degrain, nl_means
-from vstools import core, depth, get_u, get_y, join, split, vs
+from vstools import check_variable, core, join, split, vs
 
-__all__ = ["MCDenoise", "magic_denoise", "hqbm3d", "mc_dfttest"]
+__all__ = ["MCDenoise", "magic_denoise", "hqbm3d", "mc_dfttest", "Stab"]
 
 
 def hqbm3d(
@@ -218,3 +218,47 @@ def magic_denoise(clip: vs.VideoNode) -> vs.VideoNode:
         tbsize=3,
         ssystem=1,
     )
+
+
+##############################################################################
+# Original script by g-force converted into a stand alone script by McCauley #
+# then copied from havsfunc when they deleted it                             #
+# latest version from December 10, 2008                                      #
+##############################################################################
+def Stab(clp, dxmax=4, dymax=4, mirror=0):
+    if not isinstance(clp, vs.VideoNode):
+        raise vs.Error("Stab: this is not a clip")
+
+    clp = scdetect(clp, 25 / 255)
+    temp = clp.misc.AverageFrames([1] * 15, scenechange=True)
+    inter = core.std.Interleave(
+        [
+            core.rgvs.Repair(
+                temp, clp.misc.AverageFrames([1] * 3, scenechange=True), mode=[1]
+            ),
+            clp,
+        ]
+    )
+    mdata = inter.mv.DepanEstimate(trust=0, dxmax=dxmax, dymax=dymax)
+    last = inter.mv.DepanCompensate(data=mdata, offset=-1, mirror=mirror)
+    return last[::2]
+
+
+def scdetect(clip: vs.VideoNode, threshold: float = 0.1) -> vs.VideoNode:
+    def _copy_property(n: int, f: list[vs.VideoFrame]) -> vs.VideoFrame:
+        fout = f[0].copy()
+        fout.props["_SceneChangePrev"] = f[1].props["_SceneChangePrev"]
+        fout.props["_SceneChangeNext"] = f[1].props["_SceneChangeNext"]
+        return fout
+
+    assert check_variable(clip, scdetect)
+
+    sc = clip
+    if clip.format.color_family == vs.RGB:
+        sc = clip.resize.Point(format=vs.GRAY8, matrix_s="709")
+
+    sc = sc.misc.SCDetect(threshold)
+    if clip.format.color_family == vs.RGB:
+        sc = clip.std.ModifyFrame([clip, sc], _copy_property)
+
+    return sc

{soifunc-0.14.0.dist-info → soifunc-0.14.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: soifunc
-Version: 0.14.0
+Version: 0.14.1
 Summary: Soichiro's VapourSynth Functions Collection
 License: MIT
 License-File: LICENSE

{soifunc-0.14.0.dist-info → soifunc-0.14.1.dist-info}/RECORD

@@ -1,10 +1,10 @@
 soifunc/__init__.py,sha256=H6BWoCLRW2ZD47wQtL72SIZvTpD6REH7cUqIYWvCn0k,174
 soifunc/deband.py,sha256=8mT1FqKAteUAs1XJYi8WvmlcJtyQN2OxM5mfiujTKX4,1887
-soifunc/denoise.py,sha256=rqWSbSd01Ppy0BZ8IsxJaarnCosF2gtoa7URagS6ESw,7233
+soifunc/denoise.py,sha256=pWmGbFU5IRU0AAPi56a57NNBq-zaNUd7IG-nfdCgTeA,8897
 soifunc/interpolate.py,sha256=2Agg1N4GG9h6lkl7soy7G--ahVk3msbDfKGuRbj2Syo,7878
 soifunc/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 soifunc/resize.py,sha256=sOJpkovIEgaKvABEJTLfvYfKj6zvGf9pFRQMoWb7XdY,1998
-soifunc-0.14.0.dist-info/METADATA,sha256=mnto3BsBUHNojCaGOlJtGY5C8TM27afht5qQwH-sfTE,1287
-soifunc-0.14.0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-soifunc-0.14.0.dist-info/licenses/LICENSE,sha256=vgEDSMEV1J2nMiCgXE5_sjNtw2VT7_lP7rkAnrFKOWI,1068
-soifunc-0.14.0.dist-info/RECORD,,
+soifunc-0.14.1.dist-info/METADATA,sha256=xVcBWO_TJ36Fx2Mc60QwRH3XFeKMVYjhPM1NxWZZ1Gc,1287
+soifunc-0.14.1.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+soifunc-0.14.1.dist-info/licenses/LICENSE,sha256=vgEDSMEV1J2nMiCgXE5_sjNtw2VT7_lP7rkAnrFKOWI,1068
+soifunc-0.14.1.dist-info/RECORD,,