gstreamer_video/subclass/
video_sink.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_base::subclass::prelude::*;
5
6use crate::{ffi, VideoSink};
7
8pub trait VideoSinkImpl: BaseSinkImpl + ObjectSubclass<Type: IsA<VideoSink>> {
9    fn show_frame(&self, buffer: &gst::Buffer) -> Result<gst::FlowSuccess, gst::FlowError> {
10        self.parent_show_frame(buffer)
11    }
12}
13
14pub trait VideoSinkImplExt: VideoSinkImpl {
15    fn parent_show_frame(&self, buffer: &gst::Buffer) -> Result<gst::FlowSuccess, gst::FlowError> {
16        unsafe {
17            let data = Self::type_data();
18            let parent_class = data.as_ref().parent_class() as *mut ffi::GstVideoSinkClass;
19            (*parent_class)
20                .show_frame
21                .map(|f| {
22                    try_from_glib(f(
23                        self.obj().unsafe_cast_ref::<VideoSink>().to_glib_none().0,
24                        buffer.to_glib_none().0,
25                    ))
26                })
27                .unwrap_or(Err(gst::FlowError::Error))
28        }
29    }
30}
31
32impl<T: VideoSinkImpl> VideoSinkImplExt for T {}
33
34unsafe impl<T: VideoSinkImpl> IsSubclassable<T> for VideoSink {
35    fn class_init(klass: &mut glib::Class<Self>) {
36        Self::parent_class_init::<T>(klass);
37        let klass = klass.as_mut();
38        klass.show_frame = Some(video_sink_show_frame::<T>);
39    }
40}
41
42unsafe extern "C" fn video_sink_show_frame<T: VideoSinkImpl>(
43    ptr: *mut ffi::GstVideoSink,
44    buffer: *mut gst::ffi::GstBuffer,
45) -> gst::ffi::GstFlowReturn {
46    let instance = &*(ptr as *mut T::Instance);
47    let imp = instance.imp();
48    let buffer = from_glib_borrow(buffer);
49
50    gst::panic_to_error!(imp, gst::FlowReturn::Error, {
51        imp.show_frame(&buffer).into()
52    })
53    .into_glib()
54}