Add inject_kwargs_params to get_scale/descale/resample_args

Jaded-Encoding-Thaumaturgy · Dec 14, 2023 · 9cf8ea1 · 9cf8ea1
1 parent 3ddc52b
commit 9cf8ea1
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/vskernels/kernels/abstract.py b/vskernels/kernels/abstract.py
@@ -228,6 +228,7 @@ def multi(
 
         return self.scale(clip, dst_width, dst_height, shift, **kwargs)
 
+    @inject_kwargs_params
     def get_scale_args(
         self, clip: vs.VideoNode, shift: tuple[TopShift, LeftShift] = (0, 0),
         width: int | None = None, height: int | None = None,
@@ -289,6 +290,7 @@ def descale(  # type: ignore[override]
 
         return depth(descaled, bits)
 
+    @inject_kwargs_params
     def get_descale_args(
         self, clip: vs.VideoNode, shift: tuple[TopShift, LeftShift] = (0, 0),
         width: int | None = None, height: int | None = None,
@@ -503,6 +505,7 @@ def get_params_args(
     ) -> KwargsT:
         return dict(width=width, height=height) | kwargs
 
+    @inject_kwargs_params
     def get_scale_args(
         self, clip: vs.VideoNode, shift: tuple[TopShift, LeftShift] = (0, 0),
         width: int | None = None, height: int | None = None,
@@ -514,6 +517,7 @@ def get_scale_args(
             | self.get_params_args(False, clip, width, height, **kwargs)
         )
 
+    @inject_kwargs_params
     def get_descale_args(
         self, clip: vs.VideoNode, shift: tuple[TopShift, LeftShift] = (0, 0),
         width: int | None = None, height: int | None = None,
@@ -525,6 +529,7 @@ def get_descale_args(
             | self.get_params_args(True, clip, width, height, **kwargs)
         )
 
+    @inject_kwargs_params
     def get_resample_args(
         self, clip: vs.VideoNode, format: int | VideoFormatT | HoldsVideoFormatT,
         matrix: MatrixT | None, matrix_in: MatrixT | None,

diff --git a/vskernels/kernels/fmtconv.py b/vskernels/kernels/fmtconv.py
@@ -102,6 +102,7 @@ def _clean_args(self, **kwargs: Any) -> dict[str, Any]:
 
         return kwargs
 
+    @inject_kwargs_params
     def get_scale_args(
         self, clip: vs.VideoNode, shift: tuple[TopShift, LeftShift] = (0, 0),
         width: int | None = None, height: int | None = None,

diff --git a/vskernels/kernels/placebo.py b/vskernels/kernels/placebo.py
@@ -66,6 +66,7 @@ def scale(  # type: ignore[override]
             trc=Transfer.from_param_or_video(curve, clip).value_libplacebo
         )
 
+    @inject_kwargs_params
     def get_scale_args(
         self, clip: vs.VideoNode, shift: tuple[TopShift, LeftShift] = (0, 0),
         width: int | None = None, height: int | None = None,