Use with x.get_frame instead of x.get_frame globally

This should help prevent some minor memory leaks. Since it only happens upon init, the memory leaks ultimately shouldn't get worse over time (unless called in i.e. a frame eval). However, since a lot of functions call `get_prop`, this might negatively impact memory for scripts that depend on a lot of these functions.

Author: LightArrowsEXE

vstools/enums/other.py

@@ -174,7 +174,8 @@ def from_clip(cls: type[SarSelf], clip: HoldsPropValueT) -> SarSelf:
         if isinstance(clip, vs.RawFrame):
             props = clip.props
         elif isinstance(clip, vs.RawNode):
-            props = clip.get_frame(0).props
+            with clip.get_frame(0) as frame:
+                props = frame.props
         else:
             props = clip
 

vstools/exceptions/color.py

@@ -60,7 +60,8 @@ def check(func: FuncExceptT, to_check: vs.VideoNode) -> None:
         """
 
         try:
-            to_check.get_frame(0)
+            with to_check.get_frame(0):
+                pass
         except vs.Error as e:
             if 'no path between colorspaces' in str(e):
                 raise InvalidColorspacePathError(func, e)

vstools/functions/heuristics.py

@@ -59,7 +59,8 @@ def video_heuristics(
     heuristics = dict[str, PropEnum]()
 
     if props is True:
-        props_dict = clip.get_frame(0).props
+        with clip.get_frame(0) as frame:
+            props_dict = frame.props
     else:
         props_dict = props or None
 

vstools/utils/info.py

@@ -30,7 +30,8 @@ def get_var_infos(frame: vs.VideoNode | vs.VideoFrame) -> tuple[vs.VideoFormat,
     if isinstance(frame, vs.VideoNode) and not (
         frame.width and frame.height and frame.format
     ):
-        frame = frame.get_frame(0)
+        with frame.get_frame(0) as frame:
+            pass
 
     assert frame.format
 

vstools/utils/misc.py

@@ -85,10 +85,10 @@ def match_clip(
         clip = clip.resize.Bicubic(format=ref.format.id, matrix=Matrix.from_video(ref))
 
     if matrices:
-        ref_frame = ref.get_frame(0)
-        clip = clip.std.SetFrameProps(
-            _Matrix=Matrix(ref_frame), _Transfer=Transfer(ref_frame), _Primaries=Primaries(ref_frame)
-        )
+        with ref.get_frame(0) as ref_frame:
+            clip = clip.std.SetFrameProps(
+                _Matrix=Matrix(ref_frame), _Transfer=Transfer(ref_frame), _Primaries=Primaries(ref_frame)
+            )
 
     return clip.std.AssumeFPS(fpsnum=ref.fps.numerator, fpsden=ref.fps.denominator)
 

vstools/utils/props.py

@@ -78,7 +78,8 @@ def get_prop(
     if isinstance(obj, vs.RawFrame):
         props = obj.props
     elif isinstance(obj, vs.RawNode):
-        props = obj.get_frame(0).props
+        with obj.get_frame(0) as frame:
+            props = frame.props
     else:
         props = obj
 

vstools/utils/vs_proxy.py

@@ -169,10 +169,12 @@ def clear_cache() -> None:
         try:
             for output in get_outputs().values():
                 if isinstance(output, VideoOutputTuple):
-                    output.clip.get_frame(0)
+                    with output.clip.get_frame(0):
+                        pass
                     break
         except Exception:
-            core.std.BlankClip().get_frame(0)
+            with core.std.BlankClip().get_frame(0):
+                pass
         core.max_cache_size = cache_size
     except Exception:
         ...