soifunc 0.8.1__py3-none-any.whl → 0.9.0__py3-none-any.whl

soifunc/deband.py

@@ -34,7 +34,7 @@ def retinex_deband(
             "The format {format.name} is not supported! It must be an 8-16bit integer YUV bit format!",
         )
 
-    mask = dre_edgemask(clip)
+    mask = dre_edgemask.CLAHE(clip)
 
     if showmask:
         return mask

soifunc/denoise.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from typing import Callable, Optional
 
 import vsdenoise
-from vsdenoise import DFTTest, FilterType, Profile
+from vsdenoise import DFTTest, FilterType, Profile, mc_degrain
 from vstools import core, vs
 
 __all__ = ["MCDenoise", "magic_denoise", "hqbm3d", "mc_dfttest"]
@@ -21,20 +21,18 @@ def hqbm3d(
 
     Sane strength values will typically be below 1.0.
     """
-    mv = vsdenoise.MVTools.denoise(
+    mv = mc_degrain(
         clip,
+        preset=vsdenoise.MVToolsPresets.HQ_SAD,
         tr=2,
-        thSAD=100,
-        block_size=32,
-        overlap=16,
-        range_conversion=3.5,
-        sad_mode=vsdenoise.SADMode.SPATIAL.same_recalc,
-        search=vsdenoise.SearchMode.DIAMOND,
-        motion=vsdenoise.MotionMode.HIGH_SAD,
+        thsad=100,
+        refine=3,
+        blksize=select_block_size(clip),
         prefilter=vsdenoise.Prefilter.DFTTEST(
             clip,
             slocation=[(0.0, 1.0), (0.2, 4.0), (0.35, 12.0), (1.0, 48.0)],
             ssystem=1,
+            full_range=3.5,
             planes=0,
         ),
         planes=None,
@@ -53,13 +51,30 @@ def mc_dfttest(
     Even at the default `thSAD` of 75, it works well at eliminating noise.
     Turn it up to 150 if you really need to nuke something.
     It does an *okay* job at preserving details, but not nearly as good
-    as bm3d, so this is not recommended on high quality sources.
-    Use it to fix sources that are already dog water.
+    as bm3d, so this is not recommended on clean, high quality sources.
+
+    The `noisy` parameter helps preserve more detail on high-quality but grainy sources,
+    but is slower.
     """
-    profile = vsdenoise.MVToolsPresets.NOISY if noisy else vsdenoise.MVToolsPresets.CMDE
-    pre = vsdenoise.Prefilter.DFTTEST(clip)
-    mc = vsdenoise.MVTools(pre, **profile)
-    return mc.degrain(ref=clip, thSAD=thSAD)
+    # TODO: Do we need to tweak anything for the `noisy` param?
+    return mc_degrain(
+        clip,
+        prefilter=vsdenoise.Prefilter.DFTTEST,
+        preset=vsdenoise.MVToolsPresets.HQ_SAD,
+        thsad=thSAD,
+        tr=2,
+        refine=3,
+        blksize=select_block_size(clip),
+    )
+
+def select_block_size(clip: vs.VideoNode):
+    if clip.width >= 1920 or clip.height >= 1080:
+        return 64
+    if clip.width >= 1600 or clip.height >= 900:
+        return 48
+    if clip.width >= 1280 or clip.height >= 720:
+        return 32
+    return 16
 
 
 def MCDenoise(

{soifunc-0.8.1.dist-info → soifunc-0.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: soifunc
-Version: 0.8.1
+Version: 0.9.0
 Summary: Soichiro's VapourSynth Functions Collection
 License: MIT
 Author: Josh Holmer
@@ -9,13 +9,9 @@ Requires-Python: >=3.12,<4.0
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: vapoursynth (>=68)
-Requires-Dist: vsdeband (>=1.2.2,<2.0.0)
-Requires-Dist: vsdenoise (>=2.7.1,<3.0.0)
-Requires-Dist: vskernels (>=3.3.1,<4.0.0)
-Requires-Dist: vsmasktools (>=1.3.2,<2.0.0)
-Requires-Dist: vsscale (>=2.2.1,<3.0.0)
-Requires-Dist: vstools (>=3.3.4,<4.0.0)
+Requires-Dist: vsjetpack (>=0.3.4,<0.4.0)
 Description-Content-Type: text/markdown
 
 ## soifunc

soifunc-0.9.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+soifunc/__init__.py,sha256=o8id_imdLYtSpVNfB-G4Pru9GtEPqdkcpyyW3yQt-6U,127
+soifunc/deband.py,sha256=GB9ksv-G1EOkro7mEN44p1vxLCnY2x4d4SqKhHxQ6Z8,1176
+soifunc/denoise.py,sha256=NiMdXZX2c0OMRU8m1KXCAZcLgP54a2y9l-F1IrcQMYQ,6822
+soifunc/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+soifunc/resize.py,sha256=irh4Mo8OidoYYXkwjvekMCtgb4Fp0YVszTQLfONJcNU,4344
+soifunc-0.9.0.dist-info/LICENSE,sha256=vgEDSMEV1J2nMiCgXE5_sjNtw2VT7_lP7rkAnrFKOWI,1068
+soifunc-0.9.0.dist-info/METADATA,sha256=mhk0o3NeyBJ6fZU5itBPHQjyvI3VXQ08epCr3sZ7Z9c,3363
+soifunc-0.9.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+soifunc-0.9.0.dist-info/RECORD,,

{soifunc-0.8.1.dist-info → soifunc-0.9.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.0
+Generator: poetry-core 2.1.2
 Root-Is-Purelib: true
 Tag: py3-none-any

soifunc-0.8.1.dist-info/RECORD

@@ -1,9 +0,0 @@
-soifunc/__init__.py,sha256=o8id_imdLYtSpVNfB-G4Pru9GtEPqdkcpyyW3yQt-6U,127
-soifunc/deband.py,sha256=XQuc2TLvvA-BDfM8AfMQOhuzQZ5AMLqJkNNNEy4HmCM,1170
-soifunc/denoise.py,sha256=5H11XSgYj1-fmzdnNVyGckB08QP0W6mYIcoXKyaLeEI,6513
-soifunc/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-soifunc/resize.py,sha256=irh4Mo8OidoYYXkwjvekMCtgb4Fp0YVszTQLfONJcNU,4344
-soifunc-0.8.1.dist-info/LICENSE,sha256=vgEDSMEV1J2nMiCgXE5_sjNtw2VT7_lP7rkAnrFKOWI,1068
-soifunc-0.8.1.dist-info/METADATA,sha256=TazN2zM4CLXDrsKNWAn7FLqBEbespbgfnJTDyBc-kVQ,3519
-soifunc-0.8.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-soifunc-0.8.1.dist-info/RECORD,,