base/baseparse: Add support for overriding stop vfunc

This commit is contained in:
Sebastian Dröge 2020-06-30 23:45:08 +03:00
parent 03893f895c
commit 9911de6591

View file

@ -28,6 +28,10 @@ pub trait BaseParseImpl: BaseParseImplExt + ElementImpl + Send + Sync + 'static
self.parent_start(element)
}
fn stop(&self, element: &BaseParse) -> Result<(), gst::ErrorMessage> {
self.parent_stop(element)
}
fn set_sink_caps(
&self,
element: &BaseParse,
@ -57,6 +61,8 @@ pub trait BaseParseImpl: BaseParseImplExt + ElementImpl + Send + Sync + 'static
pub trait BaseParseImplExt {
fn parent_start(&self, element: &BaseParse) -> Result<(), gst::ErrorMessage>;
fn parent_stop(&self, element: &BaseParse) -> Result<(), gst::ErrorMessage>;
fn parent_set_sink_caps(
&self,
element: &BaseParse,
@ -99,6 +105,27 @@ impl<T: BaseParseImpl + ObjectImpl> BaseParseImplExt for T {
}
}
fn parent_stop(&self, element: &BaseParse) -> Result<(), gst::ErrorMessage> {
unsafe {
let data = self.get_type_data();
let parent_class =
data.as_ref().get_parent_class() as *mut gst_base_sys::GstBaseParseClass;
(*parent_class)
.stop
.map(|f| {
if from_glib(f(element.to_glib_none().0)) {
Ok(())
} else {
Err(gst_error_msg!(
gst::CoreError::StateChange,
["Parent function `stop` failed"]
))
}
})
.unwrap_or(Ok(()))
}
}
fn parent_set_sink_caps(
&self,
element: &BaseParse,
@ -197,6 +224,7 @@ where
unsafe {
let klass = &mut *(self as *mut Self as *mut gst_base_sys::GstBaseParseClass);
klass.start = Some(base_parse_start::<T>);
klass.stop = Some(base_parse_stop::<T>);
klass.set_sink_caps = Some(base_parse_set_sink_caps::<T>);
klass.handle_frame = Some(base_parse_handle_frame::<T>);
klass.convert = Some(base_parse_convert::<T>);
@ -227,6 +255,29 @@ where
.to_glib()
}
unsafe extern "C" fn base_parse_stop<T: ObjectSubclass>(
ptr: *mut gst_base_sys::GstBaseParse,
) -> glib_sys::gboolean
where
T: BaseParseImpl,
T::Instance: PanicPoison,
{
let instance = &*(ptr as *mut T::Instance);
let imp = instance.get_impl();
let wrap: Borrowed<BaseParse> = from_glib_borrow(ptr);
gst_panic_to_error!(&wrap, &instance.panicked(), false, {
match imp.stop(&wrap) {
Ok(()) => true,
Err(err) => {
wrap.post_error_message(&err);
false
}
}
})
.to_glib()
}
unsafe extern "C" fn base_parse_set_sink_caps<T: ObjectSubclass>(
ptr: *mut gst_base_sys::GstBaseParse,
caps: *mut gst_sys::GstCaps,