added using()

tikuma-lsuhsc · tikuma-lsuhsc · commit aa4e455ea9b3 · 2025-10-23T20:01:56.000-04:00
diff --git a/src/ffmpegio/__init__.py b/src/ffmpegio/__init__.py
@@ -46,6 +46,7 @@
     logger.warning(str(e))
 
 use = plugins.use
+using = plugins.using
 
 def __getattr__(name):
     if name == "ffmpeg_ver":
@@ -70,7 +71,7 @@ def __getattr__(name):
 # fmt:off
 __all__ = ["ffmpeg_info", "get_path", "set_path", "is_ready", "ffmpeg", "ffprobe",
     "transcode", "caps", "probe", "audio", "image", "video", "media", "devices",
-    "open", "ffmpegprocess", "FFmpegError", "FilterGraph", "FFConcat", "use"]
+    "open", "ffmpegprocess", "FFmpegError", "FilterGraph", "FFConcat", "use", "using"]
 # fmt:on
 
 __version__ = "0.11.1"
diff --git a/src/ffmpegio/plugins/__init__.py b/src/ffmpegio/plugins/__init__.py
@@ -100,6 +100,24 @@ def use(name: Literal["read_numpy", "read_bytes"] | str):
         pm.register(plugin)
 
 
+def using(
+    media_type: Literal["video", "audio"],
+) -> Literal["read_numpy", "read_bytes"] | str:
+    """Returns current rawdata primary read mode
+
+    :para media_type: Specifies the target media stream type
+
+    """
+
+    name = "bytes_to_audio" if media_type == "audio" else "bytes_to_video"
+    read_plugin = pm.subset_hook_caller(name, ()).get_hookimpls()[-1].plugin_name
+
+    return {
+        "ffmpegio.plugins.rawdata_numpy": "read_numpy",
+        "ffmpegio.plugins.rawdata_bytes": "read_bytes",
+    }.get(read_plugin, read_plugin)
+
+
 def initialize():
     """initilaize manager and load builtin plugins"""
 
diff --git a/tests/test_plugins.py b/tests/test_plugins.py
@@ -1,6 +1,9 @@
+import logging
+
 from ffmpegio.utils import prod
 from ffmpegio import plugins
 
+logging.basicConfig(level=logging.INFO)
 
 def test_rawdata_bytes():
     hook = plugins.get_hook()
@@ -32,6 +35,8 @@ def test_use():
     import numpy as np
 
     plugins.use("read_numpy")
+    assert plugins.using('video')=='read_numpy'
+    assert plugins.using('audio')=='read_numpy'
 
     dtype = "|u1"
     shape = (2, 2, 3)
@@ -42,6 +47,9 @@ def test_use():
     assert isinstance(data, np.ndarray)
 
     plugins.use("read_bytes")
+    assert plugins.using('video')=='read_bytes'
+    assert plugins.using('audio')=='read_bytes'
+
     hook = plugins.get_hook()
     data = hook.bytes_to_video(b=b, dtype=dtype, shape=shape, squeeze=False)
     assert isinstance(data, dict)
