Skip to main content

gstreamer_audio/subclass/
audio_aggregator.rs

1// Take a look at the license at the top of the repository in the LICENSE file.
2
3use std::ptr;
4
5use glib::translate::*;
6use gst_base::{prelude::*, subclass::prelude::*};
7
8use crate::{AudioAggregator, AudioAggregatorPad, ffi};
9
10pub trait AudioAggregatorImpl: AggregatorImpl + ObjectSubclass<Type: IsA<AudioAggregator>> {
11    fn create_output_buffer(&self, num_frames: u32) -> Option<gst::Buffer> {
12        self.parent_create_output_buffer(num_frames)
13    }
14
15    #[allow(clippy::too_many_arguments)]
16    fn aggregate_one_buffer(
17        &self,
18        pad: &AudioAggregatorPad,
19        inbuf: &gst::BufferRef,
20        in_offset: u32,
21        outbuf: &mut gst::BufferRef,
22        out_offset: u32,
23        num_frames: u32,
24    ) -> bool {
25        self.parent_aggregate_one_buffer(pad, inbuf, in_offset, outbuf, out_offset, num_frames)
26    }
27}
28
29pub trait AudioAggregatorImplExt: AudioAggregatorImpl {
30    fn parent_create_output_buffer(&self, num_frames: u32) -> Option<gst::Buffer> {
31        unsafe {
32            let data = Self::type_data();
33            let parent_class = data.as_ref().parent_class() as *mut ffi::GstAudioAggregatorClass;
34            let f = (*parent_class)
35                .create_output_buffer
36                .expect("Missing parent function `create_output_buffer`");
37
38            from_glib_full(f(
39                self.obj()
40                    .unsafe_cast_ref::<AudioAggregator>()
41                    .to_glib_none()
42                    .0,
43                num_frames,
44            ))
45        }
46    }
47
48    fn parent_aggregate_one_buffer(
49        &self,
50        pad: &AudioAggregatorPad,
51        inbuf: &gst::BufferRef,
52        in_offset: u32,
53        outbuf: &mut gst::BufferRef,
54        out_offset: u32,
55        num_frames: u32,
56    ) -> bool {
57        unsafe {
58            let data = Self::type_data();
59            let parent_class = data.as_ref().parent_class() as *mut ffi::GstAudioAggregatorClass;
60            let f = (*parent_class)
61                .aggregate_one_buffer
62                .expect("Missing parent function `aggregate_one_buffer`");
63
64            from_glib(f(
65                self.obj()
66                    .unsafe_cast_ref::<AudioAggregator>()
67                    .to_glib_none()
68                    .0,
69                pad.to_glib_none().0,
70                inbuf.as_mut_ptr(),
71                in_offset,
72                outbuf.as_mut_ptr(),
73                out_offset,
74                num_frames,
75            ))
76        }
77    }
78}
79
80impl<T: AudioAggregatorImpl> AudioAggregatorImplExt for T {}
81
82unsafe impl<T: AudioAggregatorImpl> IsSubclassable<T> for AudioAggregator {
83    fn class_init(klass: &mut glib::Class<Self>) {
84        Self::parent_class_init::<T>(klass);
85
86        let klass = klass.as_mut();
87        klass.create_output_buffer = Some(audio_aggregator_create_output_buffer::<T>);
88        klass.aggregate_one_buffer = Some(audio_aggregator_aggregate_one_buffer::<T>);
89    }
90}
91
92unsafe extern "C" fn audio_aggregator_create_output_buffer<T: AudioAggregatorImpl>(
93    ptr: *mut ffi::GstAudioAggregator,
94    num_frames: u32,
95) -> *mut gst::ffi::GstBuffer {
96    unsafe {
97        let instance = &*(ptr as *mut T::Instance);
98        let imp = instance.imp();
99
100        gst::panic_to_error!(imp, None, { imp.create_output_buffer(num_frames) })
101            .map(|buffer| buffer.into_glib_ptr())
102            .unwrap_or(ptr::null_mut())
103    }
104}
105
106unsafe extern "C" fn audio_aggregator_aggregate_one_buffer<T: AudioAggregatorImpl>(
107    ptr: *mut ffi::GstAudioAggregator,
108    pad: *mut ffi::GstAudioAggregatorPad,
109    inbuf: *mut gst::ffi::GstBuffer,
110    in_offset: u32,
111    outbuf: *mut gst::ffi::GstBuffer,
112    out_offset: u32,
113    num_frames: u32,
114) -> glib::ffi::gboolean {
115    unsafe {
116        let instance = &*(ptr as *mut T::Instance);
117        let imp = instance.imp();
118
119        gst::panic_to_error!(imp, true, {
120            imp.aggregate_one_buffer(
121                &from_glib_borrow(pad),
122                gst::BufferRef::from_ptr(inbuf),
123                in_offset,
124                gst::BufferRef::from_mut_ptr(outbuf),
125                out_offset,
126                num_frames,
127            )
128        })
129        .into_glib()
130    }
131}