gstreamer_base/subclass/
base_sink.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::{prelude::*, translate::*};
6use gst::subclass::prelude::*;
7
8use crate::{BaseSink, ffi};
9
10pub trait BaseSinkImpl: ElementImpl + ObjectSubclass<Type: IsA<BaseSink>> {
11    fn start(&self) -> Result<(), gst::ErrorMessage> {
12        self.parent_start()
13    }
14
15    fn stop(&self) -> Result<(), gst::ErrorMessage> {
16        self.parent_stop()
17    }
18
19    fn render(&self, buffer: &gst::Buffer) -> Result<gst::FlowSuccess, gst::FlowError> {
20        self.parent_render(buffer)
21    }
22
23    fn prepare(&self, buffer: &gst::Buffer) -> Result<gst::FlowSuccess, gst::FlowError> {
24        self.parent_prepare(buffer)
25    }
26
27    fn render_list(&self, list: &gst::BufferList) -> Result<gst::FlowSuccess, gst::FlowError> {
28        self.parent_render_list(list)
29    }
30
31    fn prepare_list(&self, list: &gst::BufferList) -> Result<gst::FlowSuccess, gst::FlowError> {
32        self.parent_prepare_list(list)
33    }
34
35    fn query(&self, query: &mut gst::QueryRef) -> bool {
36        BaseSinkImplExt::parent_query(self, query)
37    }
38
39    fn event(&self, event: gst::Event) -> bool {
40        self.parent_event(event)
41    }
42
43    fn caps(&self, filter: Option<&gst::Caps>) -> Option<gst::Caps> {
44        self.parent_caps(filter)
45    }
46
47    fn set_caps(&self, caps: &gst::Caps) -> Result<(), gst::LoggableError> {
48        self.parent_set_caps(caps)
49    }
50
51    fn fixate(&self, caps: gst::Caps) -> gst::Caps {
52        self.parent_fixate(caps)
53    }
54
55    fn unlock(&self) -> Result<(), gst::ErrorMessage> {
56        self.parent_unlock()
57    }
58
59    fn unlock_stop(&self) -> Result<(), gst::ErrorMessage> {
60        self.parent_unlock_stop()
61    }
62
63    fn propose_allocation(
64        &self,
65        query: &mut gst::query::Allocation,
66    ) -> Result<(), gst::LoggableError> {
67        self.parent_propose_allocation(query)
68    }
69}
70
71pub trait BaseSinkImplExt: BaseSinkImpl {
72    fn parent_start(&self) -> Result<(), gst::ErrorMessage> {
73        unsafe {
74            let data = Self::type_data();
75            let parent_class = data.as_ref().parent_class() as *mut ffi::GstBaseSinkClass;
76            (*parent_class)
77                .start
78                .map(|f| {
79                    if from_glib(f(self.obj().unsafe_cast_ref::<BaseSink>().to_glib_none().0)) {
80                        Ok(())
81                    } else {
82                        Err(gst::error_msg!(
83                            gst::CoreError::StateChange,
84                            ["Parent function `start` failed"]
85                        ))
86                    }
87                })
88                .unwrap_or(Ok(()))
89        }
90    }
91
92    fn parent_stop(&self) -> Result<(), gst::ErrorMessage> {
93        unsafe {
94            let data = Self::type_data();
95            let parent_class = data.as_ref().parent_class() as *mut ffi::GstBaseSinkClass;
96            (*parent_class)
97                .stop
98                .map(|f| {
99                    if from_glib(f(self.obj().unsafe_cast_ref::<BaseSink>().to_glib_none().0)) {
100                        Ok(())
101                    } else {
102                        Err(gst::error_msg!(
103                            gst::CoreError::StateChange,
104                            ["Parent function `stop` failed"]
105                        ))
106                    }
107                })
108                .unwrap_or(Ok(()))
109        }
110    }
111
112    fn parent_render(&self, buffer: &gst::Buffer) -> Result<gst::FlowSuccess, gst::FlowError> {
113        unsafe {
114            let data = Self::type_data();
115            let parent_class = data.as_ref().parent_class() as *mut ffi::GstBaseSinkClass;
116            (*parent_class)
117                .render
118                .map(|f| {
119                    try_from_glib(f(
120                        self.obj().unsafe_cast_ref::<BaseSink>().to_glib_none().0,
121                        buffer.to_glib_none().0,
122                    ))
123                })
124                .unwrap_or(Ok(gst::FlowSuccess::Ok))
125        }
126    }
127
128    fn parent_prepare(&self, buffer: &gst::Buffer) -> Result<gst::FlowSuccess, gst::FlowError> {
129        unsafe {
130            let data = Self::type_data();
131            let parent_class = data.as_ref().parent_class() as *mut ffi::GstBaseSinkClass;
132            (*parent_class)
133                .prepare
134                .map(|f| {
135                    try_from_glib(f(
136                        self.obj().unsafe_cast_ref::<BaseSink>().to_glib_none().0,
137                        buffer.to_glib_none().0,
138                    ))
139                })
140                .unwrap_or(Ok(gst::FlowSuccess::Ok))
141        }
142    }
143
144    fn parent_render_list(
145        &self,
146        list: &gst::BufferList,
147    ) -> Result<gst::FlowSuccess, gst::FlowError>;
148
149    fn parent_prepare_list(
150        &self,
151        list: &gst::BufferList,
152    ) -> Result<gst::FlowSuccess, gst::FlowError>;
153
154    fn parent_query(&self, query: &mut gst::QueryRef) -> bool {
155        unsafe {
156            let data = Self::type_data();
157            let parent_class = data.as_ref().parent_class() as *mut ffi::GstBaseSinkClass;
158            (*parent_class)
159                .query
160                .map(|f| {
161                    from_glib(f(
162                        self.obj().unsafe_cast_ref::<BaseSink>().to_glib_none().0,
163                        query.as_mut_ptr(),
164                    ))
165                })
166                .unwrap_or(false)
167        }
168    }
169
170    fn parent_event(&self, event: gst::Event) -> bool {
171        unsafe {
172            let data = Self::type_data();
173            let parent_class = data.as_ref().parent_class() as *mut ffi::GstBaseSinkClass;
174            (*parent_class)
175                .event
176                .map(|f| {
177                    from_glib(f(
178                        self.obj().unsafe_cast_ref::<BaseSink>().to_glib_none().0,
179                        event.into_glib_ptr(),
180                    ))
181                })
182                .unwrap_or(true)
183        }
184    }
185
186    fn parent_caps(&self, filter: Option<&gst::Caps>) -> Option<gst::Caps> {
187        unsafe {
188            let data = Self::type_data();
189            let parent_class = data.as_ref().parent_class() as *mut ffi::GstBaseSinkClass;
190
191            (*parent_class)
192                .get_caps
193                .map(|f| {
194                    from_glib_full(f(
195                        self.obj().unsafe_cast_ref::<BaseSink>().to_glib_none().0,
196                        filter.to_glib_none().0,
197                    ))
198                })
199                .unwrap_or(None)
200        }
201    }
202
203    fn parent_set_caps(&self, caps: &gst::Caps) -> Result<(), gst::LoggableError> {
204        unsafe {
205            let data = Self::type_data();
206            let parent_class = data.as_ref().parent_class() as *mut ffi::GstBaseSinkClass;
207            (*parent_class)
208                .set_caps
209                .map(|f| {
210                    gst::result_from_gboolean!(
211                        f(
212                            self.obj().unsafe_cast_ref::<BaseSink>().to_glib_none().0,
213                            caps.to_glib_none().0
214                        ),
215                        gst::CAT_RUST,
216                        "Parent function `set_caps` failed"
217                    )
218                })
219                .unwrap_or(Ok(()))
220        }
221    }
222
223    fn parent_fixate(&self, caps: gst::Caps) -> gst::Caps {
224        unsafe {
225            let data = Self::type_data();
226            let parent_class = data.as_ref().parent_class() as *mut ffi::GstBaseSinkClass;
227
228            match (*parent_class).fixate {
229                Some(fixate) => from_glib_full(fixate(
230                    self.obj().unsafe_cast_ref::<BaseSink>().to_glib_none().0,
231                    caps.into_glib_ptr(),
232                )),
233                None => caps,
234            }
235        }
236    }
237
238    fn parent_unlock(&self) -> Result<(), gst::ErrorMessage> {
239        unsafe {
240            let data = Self::type_data();
241            let parent_class = data.as_ref().parent_class() as *mut ffi::GstBaseSinkClass;
242            (*parent_class)
243                .unlock
244                .map(|f| {
245                    if from_glib(f(self.obj().unsafe_cast_ref::<BaseSink>().to_glib_none().0)) {
246                        Ok(())
247                    } else {
248                        Err(gst::error_msg!(
249                            gst::CoreError::Failed,
250                            ["Parent function `unlock` failed"]
251                        ))
252                    }
253                })
254                .unwrap_or(Ok(()))
255        }
256    }
257
258    fn parent_unlock_stop(&self) -> Result<(), gst::ErrorMessage> {
259        unsafe {
260            let data = Self::type_data();
261            let parent_class = data.as_ref().parent_class() as *mut ffi::GstBaseSinkClass;
262            (*parent_class)
263                .unlock_stop
264                .map(|f| {
265                    if from_glib(f(self.obj().unsafe_cast_ref::<BaseSink>().to_glib_none().0)) {
266                        Ok(())
267                    } else {
268                        Err(gst::error_msg!(
269                            gst::CoreError::Failed,
270                            ["Parent function `unlock_stop` failed"]
271                        ))
272                    }
273                })
274                .unwrap_or(Ok(()))
275        }
276    }
277
278    fn parent_propose_allocation(
279        &self,
280        query: &mut gst::query::Allocation,
281    ) -> Result<(), gst::LoggableError> {
282        unsafe {
283            let data = Self::type_data();
284            let parent_class = data.as_ref().parent_class() as *mut ffi::GstBaseSinkClass;
285            (*parent_class)
286                .propose_allocation
287                .map(|f| {
288                    gst::result_from_gboolean!(
289                        f(
290                            self.obj().unsafe_cast_ref::<BaseSink>().to_glib_none().0,
291                            query.as_mut_ptr(),
292                        ),
293                        gst::CAT_RUST,
294                        "Parent function `propose_allocation` failed",
295                    )
296                })
297                .unwrap_or(Ok(()))
298        }
299    }
300}
301
302impl<T: BaseSinkImpl> BaseSinkImplExt for T {
303    fn parent_render_list(
304        &self,
305        list: &gst::BufferList,
306    ) -> Result<gst::FlowSuccess, gst::FlowError> {
307        unsafe {
308            let data = Self::type_data();
309            let parent_class = data.as_ref().parent_class() as *mut ffi::GstBaseSinkClass;
310            (*parent_class)
311                .render_list
312                .map(|f| {
313                    try_from_glib(f(
314                        self.obj().unsafe_cast_ref::<BaseSink>().to_glib_none().0,
315                        list.to_glib_none().0,
316                    ))
317                })
318                .unwrap_or_else(|| {
319                    for buffer in list.iter() {
320                        self.render(&from_glib_borrow(buffer.as_ptr()))?;
321                    }
322                    Ok(gst::FlowSuccess::Ok)
323                })
324        }
325    }
326
327    fn parent_prepare_list(
328        &self,
329        list: &gst::BufferList,
330    ) -> Result<gst::FlowSuccess, gst::FlowError> {
331        unsafe {
332            let data = Self::type_data();
333            let parent_class = data.as_ref().parent_class() as *mut ffi::GstBaseSinkClass;
334            (*parent_class)
335                .prepare_list
336                .map(|f| {
337                    try_from_glib(f(
338                        self.obj().unsafe_cast_ref::<BaseSink>().to_glib_none().0,
339                        list.to_glib_none().0,
340                    ))
341                })
342                .unwrap_or_else(|| {
343                    for buffer in list.iter() {
344                        self.prepare(&from_glib_borrow(buffer.as_ptr()))?;
345                    }
346                    Ok(gst::FlowSuccess::Ok)
347                })
348        }
349    }
350}
351
352unsafe impl<T: BaseSinkImpl> IsSubclassable<T> for BaseSink {
353    fn class_init(klass: &mut glib::Class<Self>) {
354        Self::parent_class_init::<T>(klass);
355        let klass = klass.as_mut();
356        klass.start = Some(base_sink_start::<T>);
357        klass.stop = Some(base_sink_stop::<T>);
358        klass.render = Some(base_sink_render::<T>);
359        klass.render_list = Some(base_sink_render_list::<T>);
360        klass.prepare = Some(base_sink_prepare::<T>);
361        klass.prepare_list = Some(base_sink_prepare_list::<T>);
362        klass.query = Some(base_sink_query::<T>);
363        klass.event = Some(base_sink_event::<T>);
364        klass.get_caps = Some(base_sink_get_caps::<T>);
365        klass.set_caps = Some(base_sink_set_caps::<T>);
366        klass.fixate = Some(base_sink_fixate::<T>);
367        klass.unlock = Some(base_sink_unlock::<T>);
368        klass.unlock_stop = Some(base_sink_unlock_stop::<T>);
369        klass.propose_allocation = Some(base_sink_propose_allocation::<T>);
370    }
371}
372
373unsafe extern "C" fn base_sink_start<T: BaseSinkImpl>(
374    ptr: *mut ffi::GstBaseSink,
375) -> glib::ffi::gboolean {
376    unsafe {
377        let instance = &*(ptr as *mut T::Instance);
378        let imp = instance.imp();
379
380        gst::panic_to_error!(imp, false, {
381            match imp.start() {
382                Ok(()) => true,
383                Err(err) => {
384                    imp.post_error_message(err);
385                    false
386                }
387            }
388        })
389        .into_glib()
390    }
391}
392
393unsafe extern "C" fn base_sink_stop<T: BaseSinkImpl>(
394    ptr: *mut ffi::GstBaseSink,
395) -> glib::ffi::gboolean {
396    unsafe {
397        let instance = &*(ptr as *mut T::Instance);
398        let imp = instance.imp();
399
400        gst::panic_to_error!(imp, false, {
401            match imp.stop() {
402                Ok(()) => true,
403                Err(err) => {
404                    imp.post_error_message(err);
405                    false
406                }
407            }
408        })
409        .into_glib()
410    }
411}
412
413unsafe extern "C" fn base_sink_render<T: BaseSinkImpl>(
414    ptr: *mut ffi::GstBaseSink,
415    buffer: *mut gst::ffi::GstBuffer,
416) -> gst::ffi::GstFlowReturn {
417    unsafe {
418        let instance = &*(ptr as *mut T::Instance);
419        let imp = instance.imp();
420        let buffer = from_glib_borrow(buffer);
421
422        gst::panic_to_error!(imp, gst::FlowReturn::Error, { imp.render(&buffer).into() })
423            .into_glib()
424    }
425}
426
427unsafe extern "C" fn base_sink_prepare<T: BaseSinkImpl>(
428    ptr: *mut ffi::GstBaseSink,
429    buffer: *mut gst::ffi::GstBuffer,
430) -> gst::ffi::GstFlowReturn {
431    unsafe {
432        let instance = &*(ptr as *mut T::Instance);
433        let imp = instance.imp();
434        let buffer = from_glib_borrow(buffer);
435
436        gst::panic_to_error!(imp, gst::FlowReturn::Error, { imp.prepare(&buffer).into() })
437            .into_glib()
438    }
439}
440
441unsafe extern "C" fn base_sink_render_list<T: BaseSinkImpl>(
442    ptr: *mut ffi::GstBaseSink,
443    list: *mut gst::ffi::GstBufferList,
444) -> gst::ffi::GstFlowReturn {
445    unsafe {
446        let instance = &*(ptr as *mut T::Instance);
447        let imp = instance.imp();
448        let list = from_glib_borrow(list);
449
450        gst::panic_to_error!(imp, gst::FlowReturn::Error, {
451            imp.render_list(&list).into()
452        })
453        .into_glib()
454    }
455}
456
457unsafe extern "C" fn base_sink_prepare_list<T: BaseSinkImpl>(
458    ptr: *mut ffi::GstBaseSink,
459    list: *mut gst::ffi::GstBufferList,
460) -> gst::ffi::GstFlowReturn {
461    unsafe {
462        let instance = &*(ptr as *mut T::Instance);
463        let imp = instance.imp();
464        let list = from_glib_borrow(list);
465
466        gst::panic_to_error!(imp, gst::FlowReturn::Error, {
467            imp.prepare_list(&list).into()
468        })
469        .into_glib()
470    }
471}
472
473unsafe extern "C" fn base_sink_query<T: BaseSinkImpl>(
474    ptr: *mut ffi::GstBaseSink,
475    query_ptr: *mut gst::ffi::GstQuery,
476) -> glib::ffi::gboolean {
477    unsafe {
478        let instance = &*(ptr as *mut T::Instance);
479        let imp = instance.imp();
480        let query = gst::QueryRef::from_mut_ptr(query_ptr);
481
482        gst::panic_to_error!(imp, false, { BaseSinkImpl::query(imp, query) }).into_glib()
483    }
484}
485
486unsafe extern "C" fn base_sink_event<T: BaseSinkImpl>(
487    ptr: *mut ffi::GstBaseSink,
488    event_ptr: *mut gst::ffi::GstEvent,
489) -> glib::ffi::gboolean {
490    unsafe {
491        let instance = &*(ptr as *mut T::Instance);
492        let imp = instance.imp();
493
494        gst::panic_to_error!(imp, false, { imp.event(from_glib_full(event_ptr)) }).into_glib()
495    }
496}
497
498unsafe extern "C" fn base_sink_get_caps<T: BaseSinkImpl>(
499    ptr: *mut ffi::GstBaseSink,
500    filter: *mut gst::ffi::GstCaps,
501) -> *mut gst::ffi::GstCaps {
502    unsafe {
503        let instance = &*(ptr as *mut T::Instance);
504        let imp = instance.imp();
505        let filter = Option::<gst::Caps>::from_glib_borrow(filter);
506
507        gst::panic_to_error!(imp, None, { imp.caps(filter.as_ref().as_ref()) })
508            .map(|caps| caps.into_glib_ptr())
509            .unwrap_or(ptr::null_mut())
510    }
511}
512
513unsafe extern "C" fn base_sink_set_caps<T: BaseSinkImpl>(
514    ptr: *mut ffi::GstBaseSink,
515    caps: *mut gst::ffi::GstCaps,
516) -> glib::ffi::gboolean {
517    unsafe {
518        let instance = &*(ptr as *mut T::Instance);
519        let imp = instance.imp();
520        let caps = from_glib_borrow(caps);
521
522        gst::panic_to_error!(imp, false, {
523            match imp.set_caps(&caps) {
524                Ok(()) => true,
525                Err(err) => {
526                    err.log_with_imp(imp);
527                    false
528                }
529            }
530        })
531        .into_glib()
532    }
533}
534
535unsafe extern "C" fn base_sink_fixate<T: BaseSinkImpl>(
536    ptr: *mut ffi::GstBaseSink,
537    caps: *mut gst::ffi::GstCaps,
538) -> *mut gst::ffi::GstCaps {
539    unsafe {
540        let instance = &*(ptr as *mut T::Instance);
541        let imp = instance.imp();
542        let caps = from_glib_full(caps);
543
544        gst::panic_to_error!(imp, gst::Caps::new_empty(), { imp.fixate(caps) }).into_glib_ptr()
545    }
546}
547
548unsafe extern "C" fn base_sink_unlock<T: BaseSinkImpl>(
549    ptr: *mut ffi::GstBaseSink,
550) -> glib::ffi::gboolean {
551    unsafe {
552        let instance = &*(ptr as *mut T::Instance);
553        let imp = instance.imp();
554
555        gst::panic_to_error!(imp, false, {
556            match imp.unlock() {
557                Ok(()) => true,
558                Err(err) => {
559                    imp.post_error_message(err);
560                    false
561                }
562            }
563        })
564        .into_glib()
565    }
566}
567
568unsafe extern "C" fn base_sink_unlock_stop<T: BaseSinkImpl>(
569    ptr: *mut ffi::GstBaseSink,
570) -> glib::ffi::gboolean {
571    unsafe {
572        let instance = &*(ptr as *mut T::Instance);
573        let imp = instance.imp();
574
575        gst::panic_to_error!(imp, false, {
576            match imp.unlock_stop() {
577                Ok(()) => true,
578                Err(err) => {
579                    imp.post_error_message(err);
580                    false
581                }
582            }
583        })
584        .into_glib()
585    }
586}
587
588unsafe extern "C" fn base_sink_propose_allocation<T: BaseSinkImpl>(
589    ptr: *mut ffi::GstBaseSink,
590    query: *mut gst::ffi::GstQuery,
591) -> glib::ffi::gboolean {
592    unsafe {
593        let instance = &*(ptr as *mut T::Instance);
594        let imp = instance.imp();
595        let query = match gst::QueryRef::from_mut_ptr(query).view_mut() {
596            gst::QueryViewMut::Allocation(allocation) => allocation,
597            _ => unreachable!(),
598        };
599
600        gst::panic_to_error!(imp, false, {
601            match imp.propose_allocation(query) {
602                Ok(()) => true,
603                Err(err) => {
604                    err.log_with_imp_and_level(imp, gst::DebugLevel::Info);
605                    false
606                }
607            }
608        })
609        .into_glib()
610    }
611}
gstreamer_base/subclass/base_sink.rs

gstreamer_base/subclass/
base_sink.rs
