soifunc 0.9.2__tar.gz → 0.10.1__tar.gz

{soifunc-0.9.2 → soifunc-0.10.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: soifunc
-Version: 0.9.2
+Version: 0.10.1
 Summary: Soichiro's VapourSynth Functions Collection
 License: MIT
 Author: Josh Holmer
@@ -11,7 +11,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: vapoursynth (>=68)
-Requires-Dist: vsjetpack (>=0.3.4,<0.4.0)
+Requires-Dist: vsjetpack (>=0.4.0,<0.5.0)
 Description-Content-Type: text/markdown
 
 ## soifunc

{soifunc-0.9.2 → soifunc-0.10.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "soifunc"
-version = "0.9.2"
+version = "0.10.1"
 description = "Soichiro's VapourSynth Functions Collection"
 authors = ["Josh Holmer <jholmer.in@gmail.com>"]
 license = "MIT"
@@ -9,12 +9,12 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = ">=3.12,<4.0"
 vapoursynth = ">=68"
-vsjetpack = "^0.3.4"
+vsjetpack = "^0.4.0"
 
 [tool.poetry.group.dev.dependencies]
-black = "^24.4.2"
-isort = "^5.13.2"
-pre-commit = "^2.21.0"
+black = "^25.1.0"
+isort = "^6.0.1"
+pre-commit = "^4.2.0"
 
 [build-system]
 requires = ["poetry-core"]

{soifunc-0.9.2 → soifunc-0.10.1}/soifunc/resize.py

@@ -4,17 +4,15 @@ from dataclasses import dataclass
 from inspect import getfullargspec
 from typing import Any
 
+from vsaa.antialiasers.nnedi3 import Nnedi3SS
 from vskernels import (
-    Catrom,
-    EwaLanczos,
     Hermite,
-    KeepArScaler,
     Scaler,
     ScalerT,
     Spline36,
 )
-from vsscale import SSIM, GenericScaler, Waifu2x
-from vstools import check_variable_format, join, vs
+from vsscale import SSIM, ArtCNN, GenericScaler
+from vstools import check_variable_format, inject_self, is_gpu_available, join, vs
 
 __all__ = [
     "good_resize",
@@ -22,43 +20,6 @@ __all__ = [
 ]
 
 
-@dataclass
-class GoodScaler(KeepArScaler):
-    """High quality resizing filter based on opinionated defaults"""
-
-    def __init__(
-        self,
-        luma_scaler: ScalerT,
-        chroma_scaler: ScalerT,
-        **kwargs: Any,
-    ) -> None:
-        self.scaler = HybridScaler(luma_scaler, chroma_scaler)
-        super().__init__(**kwargs)
-
-    @property
-    def kernel_radius(self) -> int:
-        return self.scaler.kernel_radius
-
-    def scale_function(  # type:ignore
-        self,
-        clip: vs.VideoNode,
-        width: int,
-        height: int,
-        shift: tuple[float, float] = (0, 0),
-        **kwargs: Any,
-    ) -> vs.VideoNode:
-        if (width, height) == (clip.width, clip.height):
-            return clip
-
-        anime = kwargs.get("anime", False)
-        gpu = kwargs.get("gpu", None)
-        use_waifu2x = kwargs.get("use_waifu2x", None)
-
-        if anime and use_waifu2x:
-            return Waifu2x(cuda=gpu).scale(clip, width, height, shift)
-        return self.scaler.scale(clip, width, height, shift)
-
-
 def good_resize(
     clip: vs.VideoNode,
     width: int,
@@ -66,7 +27,6 @@ def good_resize(
     shift: tuple[float, float] = (0, 0),
     gpu: bool | None = None,
     anime: bool = False,
-    use_waifu2x: bool = False,
 ) -> vs.VideoNode:
     """High quality resizing filter
 
@@ -81,34 +41,33 @@ def good_resize(
     shift: tuple[float, float], optional
         Horizontal and vertical amount of shift to apply.
     gpu: bool, optional
-        Whether to allow usage of GPU for Waifu2x.
+        Whether to allow usage of GPU for ArtCNN.
         Defaults to None, which will auto-select based on available mlrt and hardware.
     anime: bool, optional
         Enables scalers that are better tuned toward anime.
         Defaults to False.
-    use_waifu2x: bool, optional
-        Enables Waifu2x. Will fall back to EwaLanczos if this is False.
-        Defaults to False, since Waifu2x can be a pain to set up.
     """
 
+    if gpu is None:
+        gpu = is_gpu_available()
+
     is_upscale = clip.width < width or clip.height < height
+    chroma_scaler = Spline36()
     if anime:
-        if is_upscale and not use_waifu2x:
-            luma_scaler = EwaLanczos()
-            chroma_scaler = Spline36()
+        if is_upscale:
+            if gpu:
+                luma_scaler = ArtCNN()
+            else:
+                luma_scaler = Nnedi3SS(scaler=Hermite(sigmoid=True))
         else:
-            # w2x handles upscaling differently, so we only need to specify the downscale kernel for it
-            luma_scaler = Catrom(sigmoid=True)
-            chroma_scaler = Catrom(sigmoid=True)
+            luma_scaler = Hermite(sigmoid=True)
     elif is_upscale:
-        luma_scaler = EwaLanczos()
-        chroma_scaler = Spline36()
+        luma_scaler = Nnedi3SS(scaler=SSIM())
     else:
         luma_scaler = SSIM()
-        chroma_scaler = Spline36()
 
-    return GoodScaler(luma_scaler, chroma_scaler).scale(
-        clip, width, height, shift=shift, gpu=gpu, anime=anime, use_waifu2x=use_waifu2x
+    return HybridScaler(luma_scaler, chroma_scaler).scale(
+        clip, width, height, shift=shift
     )
 
 
@@ -123,7 +82,7 @@ class HybridScaler(GenericScaler):
         self._luma = Scaler.ensure_obj(self.luma_scaler)
         self._chroma = Scaler.ensure_obj(self.chroma_scaler)
 
-    @property
+    @inject_self.cached.property
     def kernel_radius(self) -> int:
         return self._luma.kernel_radius