Skip to main content

gstreamer_gl/subclass/
gl_base_src.rs

1use ffi::{GstGLBaseSrc, GstGLMemory};
2use glib::translate::*;
3use gst::{CAT_RUST, LoggableError, result_from_gboolean};
4use gst_base::subclass::prelude::*;
5
6use crate::{GLAPI, GLBaseSrc, GLMemory, ffi, prelude::*};
7
8pub trait GLBaseSrcImpl: PushSrcImpl + ObjectSubclass<Type: IsA<GLBaseSrc>> {
9    const SUPPORTED_GL_API: GLAPI;
10
11    /// called in the GL thread to setup the element GL state.
12    fn gl_start(&self) -> Result<(), LoggableError> {
13        self.parent_gl_start()
14    }
15
16    /// called in the GL thread to setup the element GL state.
17    fn gl_stop(&self) {
18        self.parent_gl_stop()
19    }
20
21    /// called in the GL thread to fill the current video texture.
22    fn fill_gl_memory(&self, memory: &GLMemory) -> Result<(), LoggableError> {
23        self.parent_fill_gl_memory(memory)
24    }
25}
26
27pub trait GLBaseSrcImplExt: GLBaseSrcImpl {
28    fn parent_gl_start(&self) -> Result<(), LoggableError> {
29        unsafe {
30            let data = Self::type_data();
31            let parent_class = data.as_ref().parent_class() as *mut ffi::GstGLBaseSrcClass;
32
33            (*parent_class)
34                .gl_start
35                .map(|f| {
36                    result_from_gboolean!(
37                        f(self.obj().unsafe_cast_ref::<GLBaseSrc>().to_glib_none().0),
38                        CAT_RUST,
39                        "Parent function `gl_start` failed",
40                    )
41                })
42                .unwrap_or(Ok(()))
43        }
44    }
45
46    fn parent_gl_stop(&self) {
47        unsafe {
48            let data = Self::type_data();
49            let parent_class = data.as_ref().parent_class() as *mut ffi::GstGLBaseSrcClass;
50
51            if let Some(f) = (*parent_class).gl_stop {
52                f(self.obj().unsafe_cast_ref::<GLBaseSrc>().to_glib_none().0)
53            }
54        }
55    }
56
57    fn parent_fill_gl_memory(&self, memory: &GLMemory) -> Result<(), LoggableError> {
58        unsafe {
59            let data = Self::type_data();
60            let parent_class = data.as_ref().parent_class() as *mut ffi::GstGLBaseSrcClass;
61
62            (*parent_class)
63                .fill_gl_memory
64                .map(|f| {
65                    result_from_gboolean!(
66                        f(
67                            self.obj().unsafe_cast_ref::<GLBaseSrc>().to_glib_none().0,
68                            mut_override(memory.to_glib_none().0),
69                        ),
70                        CAT_RUST,
71                        "Parent function `fill_gl_memory` failed",
72                    )
73                })
74                .unwrap_or(Ok(()))
75        }
76    }
77}
78
79impl<T: GLBaseSrcImpl> GLBaseSrcImplExt for T {}
80
81unsafe impl<T: GLBaseSrcImpl> IsSubclassable<T> for GLBaseSrc {
82    fn class_init(klass: &mut glib::Class<Self>) {
83        Self::parent_class_init::<T>(klass);
84        let klass = klass.as_mut();
85        klass.supported_gl_api = T::SUPPORTED_GL_API.into_glib();
86        klass.gl_start = Some(gl_start::<T>);
87        klass.gl_stop = Some(gl_stop::<T>);
88        klass.fill_gl_memory = Some(fill_gl_memory::<T>);
89    }
90}
91
92unsafe extern "C" fn gl_start<T: GLBaseSrcImpl>(ptr: *mut GstGLBaseSrc) -> glib::ffi::gboolean {
93    unsafe {
94        let instance = &*(ptr as *mut T::Instance);
95        let imp = instance.imp();
96
97        gst::panic_to_error!(imp, false, {
98            match imp.gl_start() {
99                Ok(()) => true,
100                Err(err) => {
101                    err.log_with_imp(imp);
102                    false
103                }
104            }
105        })
106        .into_glib()
107    }
108}
109
110unsafe extern "C" fn gl_stop<T: GLBaseSrcImpl>(ptr: *mut GstGLBaseSrc) {
111    unsafe {
112        let instance = &*(ptr as *mut T::Instance);
113        let imp = instance.imp();
114
115        gst::panic_to_error!(imp, (), { imp.gl_stop() })
116    }
117}
118
119unsafe extern "C" fn fill_gl_memory<T: GLBaseSrcImpl>(
120    ptr: *mut GstGLBaseSrc,
121    memory: *mut GstGLMemory,
122) -> glib::ffi::gboolean {
123    unsafe {
124        let instance = &*(ptr as *mut T::Instance);
125        let imp = instance.imp();
126
127        gst::panic_to_error!(imp, false, {
128            match imp.fill_gl_memory(&from_glib_borrow(memory)) {
129                Ok(()) => true,
130                Err(err) => {
131                    err.log_with_imp(imp);
132                    false
133                }
134            }
135        })
136        .into_glib()
137    }
138}