From 843d3749b71e90f3bc2572c68229e8f0954a3f9f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebastian=20Dr=C3=B6ge?= <sebastian@centricular.com>
Date: Sat, 8 Jan 2022 23:34:19 +0200
Subject: [PATCH] caps: Add bindings for `foreach()`, map_in_place()` and
 `filter_map_in_place()`

---
 gstreamer/src/caps.rs | 105 ++++++++++++++++++++++++++++++++++++++++++
 gstreamer/src/lib.rs  |   2 +-
 2 files changed, 106 insertions(+), 1 deletion(-)

diff --git a/gstreamer/src/caps.rs b/gstreamer/src/caps.rs
index 68c8ea7ef..4dd9c9e83 100644
--- a/gstreamer/src/caps.rs
+++ b/gstreamer/src/caps.rs
@@ -475,6 +475,111 @@ impl CapsRef {
     pub fn serialize(&self, flags: crate::SerializeFlags) -> glib::GString {
         unsafe { from_glib_full(ffi::gst_caps_serialize(&self.0, flags.into_glib())) }
     }
+
+    #[doc(alias = "gst_caps_foreach")]
+    pub fn foreach<F: FnMut(&CapsFeaturesRef, &StructureRef) -> std::ops::ControlFlow<()>>(
+        &self,
+        mut func: F,
+    ) -> bool {
+        unsafe {
+            unsafe extern "C" fn trampoline<
+                F: FnMut(&CapsFeaturesRef, &StructureRef) -> std::ops::ControlFlow<()>,
+            >(
+                features: *mut ffi::GstCapsFeatures,
+                s: *mut ffi::GstStructure,
+                user_data: glib::ffi::gpointer,
+            ) -> glib::ffi::gboolean {
+                let func = &mut *(user_data as *mut F);
+                let res = func(
+                    CapsFeaturesRef::from_glib_borrow(features),
+                    StructureRef::from_glib_borrow(s),
+                );
+
+                matches!(res, std::ops::ControlFlow::Continue(_)).into_glib()
+            }
+            let func = &mut func as *mut F;
+            from_glib(ffi::gst_caps_foreach(
+                self.as_ptr(),
+                Some(trampoline::<F>),
+                func as glib::ffi::gpointer,
+            ))
+        }
+    }
+
+    #[doc(alias = "gst_caps_map_in_place")]
+    pub fn map_in_place<
+        F: FnMut(&mut CapsFeaturesRef, &mut StructureRef) -> std::ops::ControlFlow<()>,
+    >(
+        &mut self,
+        mut func: F,
+    ) -> bool {
+        unsafe {
+            unsafe extern "C" fn trampoline<
+                F: FnMut(&mut CapsFeaturesRef, &mut StructureRef) -> std::ops::ControlFlow<()>,
+            >(
+                features: *mut ffi::GstCapsFeatures,
+                s: *mut ffi::GstStructure,
+                user_data: glib::ffi::gpointer,
+            ) -> glib::ffi::gboolean {
+                let func = &mut *(user_data as *mut F);
+                let res = func(
+                    CapsFeaturesRef::from_glib_borrow_mut(features),
+                    StructureRef::from_glib_borrow_mut(s),
+                );
+
+                matches!(res, std::ops::ControlFlow::Continue(_)).into_glib()
+            }
+            let func = &mut func as *mut F;
+            from_glib(ffi::gst_caps_map_in_place(
+                self.as_mut_ptr(),
+                Some(trampoline::<F>),
+                func as glib::ffi::gpointer,
+            ))
+        }
+    }
+
+    #[doc(alias = "gst_caps_filter_and_map_in_place")]
+    pub fn filter_map_in_place<
+        F: FnMut(&mut CapsFeaturesRef, &mut StructureRef) -> CapsFilterMapAction,
+    >(
+        &mut self,
+        mut func: F,
+    ) {
+        unsafe {
+            unsafe extern "C" fn trampoline<
+                F: FnMut(&mut CapsFeaturesRef, &mut StructureRef) -> CapsFilterMapAction,
+            >(
+                features: *mut ffi::GstCapsFeatures,
+                s: *mut ffi::GstStructure,
+                user_data: glib::ffi::gpointer,
+            ) -> glib::ffi::gboolean {
+                let func = &mut *(user_data as *mut F);
+
+                let res = func(
+                    CapsFeaturesRef::from_glib_borrow_mut(features),
+                    StructureRef::from_glib_borrow_mut(s),
+                );
+
+                match res {
+                    CapsFilterMapAction::Keep => glib::ffi::GTRUE,
+                    CapsFilterMapAction::Remove => glib::ffi::GFALSE,
+                }
+            }
+
+            let func = &mut func as *mut F;
+            ffi::gst_caps_filter_and_map_in_place(
+                self.as_mut_ptr(),
+                Some(trampoline::<F>),
+                func as glib::ffi::gpointer,
+            );
+        }
+    }
+}
+
+#[derive(Debug)]
+pub enum CapsFilterMapAction {
+    Keep,
+    Remove,
 }
 
 macro_rules! define_iter(
diff --git a/gstreamer/src/lib.rs b/gstreamer/src/lib.rs
index a5f93ff3f..7c0a585f3 100644
--- a/gstreamer/src/lib.rs
+++ b/gstreamer/src/lib.rs
@@ -65,7 +65,7 @@ pub use crate::structure::{Structure, StructureRef};
 mod structure_serde;
 
 pub mod caps;
-pub use crate::caps::{Caps, CapsRef};
+pub use crate::caps::{Caps, CapsFilterMapAction, CapsRef};
 mod caps_features;
 #[cfg(feature = "ser_de")]
 mod caps_serde;