Add AVCodec::iterate() (#162)

larksuite · Mar 17, 2024 · 6e9cd04 · 6e9cd04
1 parent 42ca429
commit 6e9cd04
Showing 1 changed file with 49 additions and 1 deletion.
diff --git a/src/avcodec/codec.rs b/src/avcodec/codec.rs
@@ -1,5 +1,5 @@
 use std::{
-    ffi::CStr,
+    ffi::{c_void, CStr},
     mem,
     ops::Drop,
     ptr::{self, NonNull},
@@ -54,6 +54,26 @@ impl AVCodec {
     pub fn long_name(&self) -> &CStr {
         unsafe { CStr::from_ptr(self.long_name) }
     }
+
+    /// Iterate over all registered codecs.
+    pub fn iterate() -> AVCodecIter {
+        AVCodecIter {
+            opaque: std::ptr::null_mut(),
+        }
+    }
+}
+
+pub struct AVCodecIter {
+    opaque: *mut c_void,
+}
+
+impl Iterator for AVCodecIter {
+    type Item = AVCodecRef<'static>;
+
+    fn next(&mut self) -> Option<Self::Item> {
+        let ptr = unsafe { ffi::av_codec_iterate(&mut self.opaque) }.upgrade()?;
+        Some(unsafe { AVCodecRef::from_raw(ptr) })
+    }
 }
 
 impl<'codec> AVCodec {
@@ -389,3 +409,31 @@ impl Drop for AVSubtitle {
         }
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use cstr::cstr;
+
+    #[test]
+    fn test_av_codec_iterator() {
+        assert!(AVCodec::iterate().count() > 10);
+
+        let iter = AVCodec::iterate();
+        for codec in iter {
+            if codec.name() == cstr!("h264") {
+                assert_eq!(
+                    codec.long_name(),
+                    cstr!("H.264 / AVC / MPEG-4 AVC / MPEG-4 part 10")
+                );
+            }
+            if codec.name() == cstr!("vnull") {
+                assert_eq!(codec.long_name(), cstr!("null video"));
+            }
+            if codec.name() == cstr!("anull") {
+                assert_eq!(codec.long_name(), cstr!("null audio"));
+            }
+            println!("codec: {:?}: {:?}", codec.name(), codec.long_name());
+        }
+    }
+}