gstreamer_base/subclass/
aggregator_pad.rs

1// Take a look at the license at the top of the repository in the LICENSE file.
2
3use glib::{prelude::*, translate::*};
4use gst::subclass::prelude::*;
5
6use crate::{ffi, Aggregator, AggregatorPad};
7
8pub trait AggregatorPadImpl: AggregatorPadImplExt + PadImpl {
9    /// Optional
10    ///  Called when the pad has received a flush stop, this is the place
11    ///  to flush any information specific to the pad, it allows for individual
12    ///  pads to be flushed while others might not be.
13    fn flush(&self, aggregator: &Aggregator) -> Result<gst::FlowSuccess, gst::FlowError> {
14        self.parent_flush(aggregator)
15    }
16
17    /// Optional
18    ///  Called before input buffers are queued in the pad, return [`true`]
19    ///  if the buffer should be skipped.
20    fn skip_buffer(&self, aggregator: &Aggregator, buffer: &gst::Buffer) -> bool {
21        self.parent_skip_buffer(aggregator, buffer)
22    }
23}
24
25mod sealed {
26    pub trait Sealed {}
27    impl<T: super::AggregatorPadImplExt> Sealed for T {}
28}
29
30pub trait AggregatorPadImplExt: sealed::Sealed + ObjectSubclass {
31    fn parent_flush(&self, aggregator: &Aggregator) -> Result<gst::FlowSuccess, gst::FlowError> {
32        unsafe {
33            let data = Self::type_data();
34            let parent_class = data.as_ref().parent_class() as *mut ffi::GstAggregatorPadClass;
35            (*parent_class)
36                .flush
37                .map(|f| {
38                    try_from_glib(f(
39                        self.obj()
40                            .unsafe_cast_ref::<AggregatorPad>()
41                            .to_glib_none()
42                            .0,
43                        aggregator.to_glib_none().0,
44                    ))
45                })
46                .unwrap_or(Ok(gst::FlowSuccess::Ok))
47        }
48    }
49
50    fn parent_skip_buffer(&self, aggregator: &Aggregator, buffer: &gst::Buffer) -> bool {
51        unsafe {
52            let data = Self::type_data();
53            let parent_class = data.as_ref().parent_class() as *mut ffi::GstAggregatorPadClass;
54            (*parent_class)
55                .skip_buffer
56                .map(|f| {
57                    from_glib(f(
58                        self.obj()
59                            .unsafe_cast_ref::<AggregatorPad>()
60                            .to_glib_none()
61                            .0,
62                        aggregator.to_glib_none().0,
63                        buffer.to_glib_none().0,
64                    ))
65                })
66                .unwrap_or(false)
67        }
68    }
69}
70
71impl<T: AggregatorPadImpl> AggregatorPadImplExt for T {}
72unsafe impl<T: AggregatorPadImpl> IsSubclassable<T> for AggregatorPad {
73    fn class_init(klass: &mut glib::Class<Self>) {
74        Self::parent_class_init::<T>(klass);
75        let klass = klass.as_mut();
76        klass.flush = Some(aggregator_pad_flush::<T>);
77        klass.skip_buffer = Some(aggregator_pad_skip_buffer::<T>);
78    }
79}
80
81unsafe extern "C" fn aggregator_pad_flush<T: AggregatorPadImpl>(
82    ptr: *mut ffi::GstAggregatorPad,
83    aggregator: *mut ffi::GstAggregator,
84) -> gst::ffi::GstFlowReturn {
85    let instance = &*(ptr as *mut T::Instance);
86    let imp = instance.imp();
87
88    let res: gst::FlowReturn = imp.flush(&from_glib_borrow(aggregator)).into();
89    res.into_glib()
90}
91
92unsafe extern "C" fn aggregator_pad_skip_buffer<T: AggregatorPadImpl>(
93    ptr: *mut ffi::GstAggregatorPad,
94    aggregator: *mut ffi::GstAggregator,
95    buffer: *mut gst::ffi::GstBuffer,
96) -> glib::ffi::gboolean {
97    let instance = &*(ptr as *mut T::Instance);
98    let imp = instance.imp();
99
100    imp.skip_buffer(&from_glib_borrow(aggregator), &from_glib_borrow(buffer))
101        .into_glib()
102}