Merge branch 'transformscale-use-ffi' into 'main'

meta: Make MetaTransform ffi wrappers

See merge request gstreamer/gstreamer-rs!1776
This commit is contained in:
Olivier Crête 2025-08-25 13:16:27 +00:00
commit 4258907e43
2 changed files with 75 additions and 46 deletions

View file

@ -1201,20 +1201,32 @@ pub mod tags {
gst::impl_meta_tag!(Colorspace, crate::ffi::GST_META_TAG_VIDEO_COLORSPACE_STR); gst::impl_meta_tag!(Colorspace, crate::ffi::GST_META_TAG_VIDEO_COLORSPACE_STR);
} }
#[derive(Debug, Clone, PartialEq, Eq)] #[repr(transparent)]
pub struct VideoMetaTransformScale<'a> { #[doc(alias = "GstVideoMetaTransform")]
in_info: &'a crate::VideoInfo, pub struct VideoMetaTransformScale(ffi::GstVideoMetaTransform);
out_info: &'a crate::VideoInfo,
}
impl<'a> VideoMetaTransformScale<'a> { unsafe impl Sync for VideoMetaTransformScale {}
pub fn new(in_info: &'a crate::VideoInfo, out_info: &'a crate::VideoInfo) -> Self { unsafe impl Send for VideoMetaTransformScale {}
impl VideoMetaTransformScale {
pub fn new(in_info: &crate::VideoInfo, out_info: &crate::VideoInfo) -> Self {
skip_assert_initialized!(); skip_assert_initialized!();
VideoMetaTransformScale { in_info, out_info } Self(ffi::GstVideoMetaTransform {
in_info: mut_override(in_info.to_glib_none().0),
out_info: mut_override(out_info.to_glib_none().0),
})
}
pub fn in_info(&self) -> &crate::VideoInfo {
unsafe { &*(self.0.in_info as *const crate::VideoInfo) }
}
pub fn out_info(&self) -> &crate::VideoInfo {
unsafe { &*(self.0.out_info as *const crate::VideoInfo) }
} }
} }
unsafe impl<'a> gst::meta::MetaTransform<'a> for VideoMetaTransformScale<'a> { unsafe impl gst::meta::MetaTransform for VideoMetaTransformScale {
type GLibType = ffi::GstVideoMetaTransform; type GLibType = ffi::GstVideoMetaTransform;
#[doc(alias = "gst_video_meta_transform_scale_get_quark")] #[doc(alias = "gst_video_meta_transform_scale_get_quark")]
@ -1222,14 +1234,8 @@ unsafe impl<'a> gst::meta::MetaTransform<'a> for VideoMetaTransformScale<'a> {
unsafe { from_glib(ffi::gst_video_meta_transform_scale_get_quark()) } unsafe { from_glib(ffi::gst_video_meta_transform_scale_get_quark()) }
} }
fn to_raw<T: MetaAPI>( fn as_ptr(&self) -> *const ffi::GstVideoMetaTransform {
&self, &self.0
_meta: &gst::MetaRef<T>,
) -> Result<ffi::GstVideoMetaTransform, glib::BoolError> {
Ok(ffi::GstVideoMetaTransform {
in_info: mut_override(self.in_info.to_glib_none().0),
out_info: mut_override(self.out_info.to_glib_none().0),
})
} }
} }

View file

@ -235,7 +235,7 @@ impl<'a, T> MetaRef<'a, T> {
pub fn transform<MT>(&self, buffer: &mut BufferRef, data: &'a MT) -> Result<(), glib::BoolError> pub fn transform<MT>(&self, buffer: &mut BufferRef, data: &'a MT) -> Result<(), glib::BoolError>
where where
T: MetaAPI, T: MetaAPI,
MT: MetaTransform<'a>, MT: MetaTransform,
{ {
unsafe { unsafe {
let info = *(*self.upcast_ref().as_ptr()).info; let info = *(*self.upcast_ref().as_ptr()).info;
@ -245,15 +245,13 @@ impl<'a, T> MetaRef<'a, T> {
)); ));
}; };
let data = data.to_raw(self)?;
glib::result_from_gboolean!( glib::result_from_gboolean!(
transform_func( transform_func(
buffer.as_mut_ptr(), buffer.as_mut_ptr(),
mut_override(self.upcast_ref().as_ptr()), mut_override(self.upcast_ref().as_ptr()),
mut_override(self.buffer.as_ptr()), mut_override(self.buffer.as_ptr()),
MT::quark().into_glib(), MT::quark().into_glib(),
mut_override(&data) as *mut _, mut_override(data.as_ptr() as *mut _),
), ),
"Failed to transform meta" "Failed to transform meta"
) )
@ -493,7 +491,7 @@ impl<'a, T, U> MetaRefMut<'a, T, U> {
) -> Result<(), glib::BoolError> ) -> Result<(), glib::BoolError>
where where
T: MetaAPI, T: MetaAPI,
MT: MetaTransform<'a>, MT: MetaTransform,
{ {
self.as_meta_ref().transform(buffer, data) self.as_meta_ref().transform(buffer, data)
} }
@ -1103,46 +1101,71 @@ pub mod tags {
); );
} }
pub unsafe trait MetaTransform<'a> { pub unsafe trait MetaTransform {
type GLibType; type GLibType;
fn quark() -> glib::Quark; fn quark() -> glib::Quark;
fn to_raw<T: MetaAPI>(&self, meta: &MetaRef<T>) -> Result<Self::GLibType, glib::BoolError>; fn as_ptr(&self) -> *const Self::GLibType;
} }
#[derive(Debug, Clone, PartialEq, Eq)] #[repr(transparent)]
pub struct MetaTransformCopy { #[doc(alias = "GstMetaTransformCopy")]
range: (Bound<usize>, Bound<usize>), pub struct MetaTransformCopy(ffi::GstMetaTransformCopy);
region: bool,
}
impl MetaTransformCopy { impl MetaTransformCopy {
pub fn new(region: bool, range: impl RangeBounds<usize>) -> Self { pub fn new(range: impl RangeBounds<usize>) -> Self {
skip_assert_initialized!(); skip_assert_initialized!();
MetaTransformCopy {
range: (range.start_bound().cloned(), range.end_bound().cloned()), let region = !(matches!(range.start_bound(), Bound::Unbounded | Bound::Included(0))
region, && range.end_bound() == Bound::Unbounded);
let (offset, size) = if region {
(
match range.start_bound() {
Bound::Included(idx) => *idx,
Bound::Excluded(idx) => *idx + 1,
Bound::Unbounded => 0,
},
match range.end_bound() {
Bound::Included(idx) => *idx + 1,
Bound::Excluded(idx) => *idx,
Bound::Unbounded => usize::MAX,
},
)
} else {
(0, usize::MAX)
};
Self(ffi::GstMetaTransformCopy {
region: region.into_glib(),
offset,
size,
})
}
pub fn range(&self) -> Option<(Bound<usize>, Bound<usize>)> {
if self.0.region == glib::ffi::GFALSE {
None
} else {
let end = if self.0.size == usize::MAX {
Bound::Unbounded
} else {
Bound::Excluded(self.0.size)
};
Some((Bound::Included(self.0.offset), end))
} }
} }
} }
unsafe impl MetaTransform<'_> for MetaTransformCopy { unsafe impl MetaTransform for MetaTransformCopy {
type GLibType = ffi::GstMetaTransformCopy; type GLibType = ffi::GstMetaTransformCopy;
fn quark() -> glib::Quark { fn quark() -> glib::Quark {
static QUARK: std::sync::OnceLock<glib::Quark> = std::sync::OnceLock::new(); static QUARK: std::sync::OnceLock<glib::Quark> = std::sync::OnceLock::new();
*QUARK.get_or_init(|| glib::Quark::from_static_str(glib::gstr!("gst-copy"))) *QUARK.get_or_init(|| glib::Quark::from_static_str(glib::gstr!("gst-copy")))
} }
fn to_raw<T: MetaAPI>( fn as_ptr(&self) -> *const ffi::GstMetaTransformCopy {
&self, &self.0
meta: &MetaRef<T>,
) -> Result<ffi::GstMetaTransformCopy, glib::BoolError> {
let (offset, size) = meta.buffer.byte_range_into_offset_len(self.range)?;
Ok(ffi::GstMetaTransformCopy {
region: self.region.into_glib(),
offset,
size,
})
} }
} }
@ -1304,7 +1327,7 @@ mod tests {
{ {
let meta = buffer.meta::<ReferenceTimestampMeta>().unwrap(); let meta = buffer.meta::<ReferenceTimestampMeta>().unwrap();
let buffer_dest = buffer_dest.get_mut().unwrap(); let buffer_dest = buffer_dest.get_mut().unwrap();
meta.transform(buffer_dest, &MetaTransformCopy::new(false, ..)) meta.transform(buffer_dest, &MetaTransformCopy::new(..))
.unwrap(); .unwrap();
} }