Skip to content

Commit

Permalink
Update vardefunc stubs
Browse files Browse the repository at this point in the history
  • Loading branch information
LightArrowsEXE committed Jun 19, 2021
1 parent bf9adda commit 4770381
Show file tree
Hide file tree
Showing 8 changed files with 156 additions and 2 deletions.
8 changes: 7 additions & 1 deletion stubs/vardefunc/__init__.pyi
Original file line number Diff line number Diff line change
@@ -1 +1,7 @@
from . import deband as deband, util as util
from . import deband as deband, mask, misc, noise as noise, placebo as placebo, scale as scale, sharp as sharp, util as util
from typing import Any

drm: Any
dcm: Any
lcm = mask.luma_credit_mask
gk = misc.generate_keyframes
18 changes: 17 additions & 1 deletion stubs/vardefunc/deband.pyi
Original file line number Diff line number Diff line change
@@ -1,7 +1,23 @@
import vapoursynth as vs
from .util import FormatError as FormatError
from typing import Any, List, Union
from typing import Any, Dict, List, Union

core: Any

class F3kdb:
radius: int
thy: int
thcb: int
thcr: int
gry: int
grc: int
sample_mode: int
use_neo: bool
f3kdb_args: Dict[str, Any]
def __init__(self, radius: int=..., threshold: Union[int, List[int]]=..., grain: Union[int, List[int]]=..., sample_mode: int=..., use_neo: bool=..., **kwargs: Any) -> None: ...
def deband(self, clip: vs.VideoNode) -> vs.VideoNode: ...
def grain(self, clip: vs.VideoNode) -> vs.VideoNode: ...

def dumb3kdb(clip: vs.VideoNode, radius: int=..., threshold: Union[int, List[int]]=..., grain: Union[int, List[int]]=..., sample_mode: int=..., use_neo: bool=..., **kwargs: Any) -> vs.VideoNode: ...
def f3kbilateral(clip: vs.VideoNode, radius: int=..., threshold: Union[int, List[int]]=..., grain: Union[int, List[int]]=..., f3kdb_args: Dict[str, Any]=..., lf_args: Dict[str, Any]=...) -> vs.VideoNode: ...
def lfdeband(clip: vs.VideoNode) -> vs.VideoNode: ...
53 changes: 53 additions & 0 deletions stubs/vardefunc/mask.pyi
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
import abc
import lvsfunc
import vapoursynth as vs
from .util import FormatError as FormatError, get_sample_type as get_sample_type, mae_expr as mae_expr, max_expr as max_expr, pick_px_op as pick_px_op
from abc import ABC
from typing import Any, Dict, List, Optional, Union

core: Any

class EdgeDetect(ABC, metaclass=abc.ABCMeta):
def get_mask(self, clip: vs.VideoNode, lthr: float=..., hthr: Optional[float]=..., multi: float=...) -> vs.VideoNode: ...

class Laplacian1(EdgeDetect): ...
class Laplacian2(EdgeDetect): ...
class Laplacian3(EdgeDetect): ...
class Laplacian4(EdgeDetect): ...
class ExLaplacian1(EdgeDetect): ...
class ExLaplacian2(EdgeDetect): ...
class ExLaplacian3(EdgeDetect): ...
class ExLaplacian4(EdgeDetect): ...
class Kayyali(EdgeDetect): ...
class LoG(EdgeDetect): ...
class Roberts(EdgeDetect): ...
class Prewitt(EdgeDetect): ...
class PrewittStd(EdgeDetect): ...
class ExPrewitt(EdgeDetect): ...
class Sobel(EdgeDetect): ...
class SobelStd(EdgeDetect): ...
class ExSobel(EdgeDetect): ...
class Scharr(EdgeDetect): ...
class FDOG(EdgeDetect): ...
class Kroon(EdgeDetect): ...
class FreyChen(EdgeDetect): ...
class FreyChenG41(EdgeDetect): ...
class TEdge(EdgeDetect): ...
class TEdgeTedgemask(EdgeDetect): ...
class Robinson3(EdgeDetect): ...
class Robinson5(EdgeDetect): ...
class Kirsch(EdgeDetect): ...
class ExKirsch(EdgeDetect): ...

def get_all_edge_detects(clip: vs.VideoNode, **kwargs: Any) -> List[vs.VideoNode]: ...

class Difference:
def rescale(self, clip: vs.VideoNode, height: int=..., kernel: lvsfunc.kernels.Kernel=..., thr: Union[int, float]=..., expand: int=...) -> vs.VideoNode: ...
def creditless(self, src_clip: vs.VideoNode, credit_clip: vs.VideoNode, nc_clip: vs.VideoNode, start_frame: int, thr: int, expand: int=..., *, prefilter: bool=..., bilateral_args: Dict[str, Any]=...) -> vs.VideoNode: ...
def creditless_oped(self, ep: vs.VideoNode, ncop: vs.VideoNode, nced: vs.VideoNode, opstart: Optional[int]=..., opend: Optional[int]=..., edstart: Optional[int]=..., edend: Optional[int]=..., **creditless_args: Any) -> vs.VideoNode: ...

def diff_creditless_mask(src_clip: vs.VideoNode, credit_clip: vs.VideoNode, nc_clip: vs.VideoNode, start_frame: int, thr: int, expand: int=..., *, prefilter: bool=..., bilateral_args: Dict[str, Any]=...) -> vs.VideoNode: ...
def diff_rescale_mask(clip: vs.VideoNode, height: int=..., kernel: lvsfunc.kernels.Kernel=..., thr: Union[int, float]=..., expand: int=...) -> vs.VideoNode: ...
def luma_mask(clip: vs.VideoNode, thr_lo: float, thr_hi: float, invert: bool=...) -> vs.VideoNode: ...
def luma_credit_mask(clip: vs.VideoNode, thr: int=..., edgemask: EdgeDetect=..., draft: bool=...) -> vs.VideoNode: ...
def region_mask(clip: vs.VideoNode, left: int=..., right: int=..., top: int=..., bottom: int=...) -> vs.VideoNode: ...
11 changes: 11 additions & 0 deletions stubs/vardefunc/misc.pyi
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
import vapoursynth as vs
from typing import Any, Tuple, Union

core: Any

def fade_filter(clip: vs.VideoNode, clip_a: vs.VideoNode, clip_b: vs.VideoNode, start_f: int, end_f: int) -> vs.VideoNode: ...
def merge_chroma(luma: vs.VideoNode, ref: vs.VideoNode) -> vs.VideoNode: ...
def get_chroma_shift(src_h: int, dst_h: int, aspect_ratio: float=...) -> float: ...
def get_bicubic_params(cubic_filter: str) -> Tuple[float, float]: ...
def generate_keyframes(clip: vs.VideoNode, out_path: str) -> None: ...
def set_ffms2_log_level(level: Union[str, int]=...) -> None: ...
40 changes: 40 additions & 0 deletions stubs/vardefunc/noise.pyi
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
import abc
import vapoursynth as vs
from .deband import dumb3kdb as dumb3kdb
from .mask import FDOG as FDOG
from .placebo import deband as deband
from .util import FormatError as FormatError, get_sample_type as get_sample_type, pick_px_op as pick_px_op
from abc import ABC, abstractmethod
from typing import Any, List, Sequence, Tuple, Union

core: Any

class Grainer(ABC, metaclass=abc.ABCMeta):
kwargs: Any = ...
def __init__(self, **kwargs: Any) -> None: ...
@abstractmethod
def grain(self, clip: vs.VideoNode, strength: Tuple[float, float]) -> vs.VideoNode: ...

class AddGrain(Grainer):
def grain(self, clip: vs.VideoNode, strength: Tuple[float, float]) -> vs.VideoNode: ...

class PlaceboGrain(Grainer):
def grain(self, clip: vs.VideoNode, strength: Tuple[float, float]) -> vs.VideoNode: ...

class F3kdbGrain(Grainer):
def grain(self, clip: vs.VideoNode, strength: Tuple[float, float]) -> vs.VideoNode: ...

class Graigasm:
thrs: List[float]
strengths: List[Tuple[float, float]]
sizes: List[float]
sharps: List[float]
overflows: List[float]
grainers: List[Grainer]
def __init__(self, thrs: Sequence[float], strengths: Sequence[Tuple[float, float]], sizes: Sequence[float], sharps: Sequence[float], *, overflows: Union[float, Sequence[float]]=..., grainers: Union[Grainer, Sequence[Grainer]]=...) -> None: ...
def graining(self, clip: vs.VideoNode, *, prefilter: vs.VideoNode=..., show_masks: bool=...) -> vs.VideoNode: ...
@staticmethod
def _m__(x: int, mod: int) -> int: ...

def decsiz(clip: vs.VideoNode, sigmaS: float=..., sigmaR: float=..., min_in: Union[int, float]=..., max_in: Union[int, float]=..., gamma: float=..., protect_mask: vs.VideoNode=..., prefilter: bool=..., planes: List[int]=..., show_mask: bool=...) -> vs.VideoNode: ...
def adaptative_regrain(denoised: vs.VideoNode, new_grained: vs.VideoNode, original_grained: vs.VideoNode, range_avg: Tuple[float, float]=..., luma_scaling: int=...) -> vs.VideoNode: ...
8 changes: 8 additions & 0 deletions stubs/vardefunc/placebo.pyi
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
import vapoursynth as vs
from .util import FormatError as FormatError
from typing import Any, List, Union

core: Any

def deband(clip: vs.VideoNode, radius: float=..., threshold: Union[float, List[float]]=..., iterations: int=..., grain: Union[float, List[float]]=..., chroma: bool=..., **kwargs: Any) -> vs.VideoNode: ...
def shader(clip: vs.VideoNode, width: int, height: int, shader_file: str, luma_only: bool=..., **kwargs: Any) -> vs.VideoNode: ...
13 changes: 13 additions & 0 deletions stubs/vardefunc/scale.pyi
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
import lvsfunc
import vapoursynth as vs
from .placebo import shader as shader
from .sharp import z4usm as z4usm
from typing import Any, Callable, Dict, List, Optional, Union

core: Any

def nnedi3cl_double(clip: vs.VideoNode, scaler: lvsfunc.kernels.Kernel=..., correct_shift: bool=..., use_znedi: bool=..., **nnedi3_args: Any) -> vs.VideoNode: ...
def nnedi3_upscale(clip: vs.VideoNode, scaler: lvsfunc.kernels.Kernel=..., correct_shift: bool=..., use_znedi: bool=..., **nnedi3_args: Any) -> vs.VideoNode: ...
def eedi3_upscale(clip: vs.VideoNode, scaler: lvsfunc.kernels.Kernel=..., correct_shift: bool=..., nnedi3_args: Optional[Dict[str, Any]]=..., eedi3_args: Optional[Dict[str, Any]]=...) -> vs.VideoNode: ...
def fsrcnnx_upscale(clip: vs.VideoNode, width: int=..., height: int=..., shader_file: str=..., downscaler: Callable[[vs.VideoNode, int, int], vs.VideoNode]=..., upscaled_smooth: Optional[vs.VideoNode]=..., strength: float=..., profile: str=..., lmode: int=..., overshoot: float=..., undershoot: float=..., sharpener: Callable[[vs.VideoNode], vs.VideoNode]=...) -> vs.VideoNode: ...
def to_444(clip: vs.VideoNode, width: int=..., height: int=..., join_planes: bool=..., znedi: bool=..., scaler: lvsfunc.kernels.Kernel=...) -> Union[vs.VideoNode, List[vs.VideoNode]]: ...
7 changes: 7 additions & 0 deletions stubs/vardefunc/sharp.pyi
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
import vapoursynth as vs
from .util import FormatError as FormatError
from typing import Any

core: Any

def z4usm(clip: vs.VideoNode, radius: int=..., strength: float=...) -> vs.VideoNode: ...

0 comments on commit 4770381

Please sign in to comment.