video: Add VideoAffineTransformationMeta, CropMeta and RegionOfInterestMeta

Fixes https://gitlab.freedesktop.org/gstreamer/gstreamer-rs/-/issues/218
This commit is contained in:
Sebastian Dröge 2020-06-02 11:18:42 +03:00
parent ede4588e0c
commit 9236ace284
2 changed files with 261 additions and 3 deletions

View file

@ -69,8 +69,15 @@ mod video_rectangle;
pub use video_rectangle::*;
mod video_overlay_composition;
pub use video_overlay_composition::*;
mod video_meta;
pub use video_meta::*;
pub mod video_meta;
#[cfg(any(feature = "v1_16", feature = "dox"))]
pub use video_meta::VideoCaptionMeta;
#[cfg(any(feature = "v1_18", feature = "dox"))]
pub use video_meta::{VideoAFDMeta, VideoBarMeta};
pub use video_meta::{
VideoAffineTransformationMeta, VideoCropMeta, VideoMeta, VideoOverlayCompositionMeta,
VideoRegionOfInterestMeta,
};
#[cfg(any(feature = "v1_10", feature = "dox"))]
mod video_time_code;
#[cfg(any(feature = "v1_10", feature = "dox"))]

View file

@ -7,9 +7,10 @@
// except according to those terms.
use std::fmt;
use std::ptr;
use glib;
use glib::translate::{from_glib, from_glib_none, ToGlib};
use glib::translate::{from_glib, from_glib_none, FromGlib, ToGlib, ToGlibPtr};
use gst;
use gst::prelude::*;
use gst_sys;
@ -138,6 +139,256 @@ impl fmt::Debug for VideoMeta {
}
}
#[repr(C)]
pub struct VideoCropMeta(gst_video_sys::GstVideoCropMeta);
unsafe impl Send for VideoCropMeta {}
unsafe impl Sync for VideoCropMeta {}
impl VideoCropMeta {
pub fn add(
buffer: &mut gst::BufferRef,
rect: (u32, u32, u32, u32),
) -> gst::MetaRefMut<Self, gst::meta::Standalone> {
skip_assert_initialized!();
unsafe {
let meta = gst_sys::gst_buffer_add_meta(
buffer.as_mut_ptr(),
gst_video_sys::gst_video_crop_meta_get_info(),
ptr::null_mut(),
) as *mut gst_video_sys::GstVideoCropMeta;
{
let meta = &mut *meta;
meta.x = rect.0;
meta.y = rect.1;
meta.width = rect.2;
meta.height = rect.3;
}
Self::from_mut_ptr(buffer, meta)
}
}
pub fn get_rect(&self) -> (u32, u32, u32, u32) {
(self.0.x, self.0.y, self.0.width, self.0.height)
}
pub fn set_rect(&mut self, rect: (u32, u32, u32, u32)) {
self.0.x = rect.0;
self.0.y = rect.1;
self.0.width = rect.2;
self.0.height = rect.3;
}
}
unsafe impl MetaAPI for VideoCropMeta {
type GstType = gst_video_sys::GstVideoCropMeta;
fn get_meta_api() -> glib::Type {
unsafe { from_glib(gst_video_sys::gst_video_crop_meta_api_get_type()) }
}
}
impl fmt::Debug for VideoCropMeta {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f.debug_struct("VideoCropMeta")
.field("rect", &self.get_rect())
.finish()
}
}
#[repr(C)]
pub struct VideoRegionOfInterestMeta(gst_video_sys::GstVideoRegionOfInterestMeta);
unsafe impl Send for VideoRegionOfInterestMeta {}
unsafe impl Sync for VideoRegionOfInterestMeta {}
impl VideoRegionOfInterestMeta {
pub fn add<'a>(
buffer: &'a mut gst::BufferRef,
roi_type: &str,
rect: (u32, u32, u32, u32),
) -> gst::MetaRefMut<'a, Self, gst::meta::Standalone> {
skip_assert_initialized!();
unsafe {
let meta = gst_video_sys::gst_buffer_add_video_region_of_interest_meta(
buffer.as_mut_ptr(),
roi_type.to_glib_none().0,
rect.0,
rect.1,
rect.2,
rect.3,
);
Self::from_mut_ptr(buffer, meta)
}
}
pub fn get_rect(&self) -> (u32, u32, u32, u32) {
(self.0.x, self.0.y, self.0.w, self.0.h)
}
pub fn get_id(&self) -> i32 {
self.0.id
}
pub fn get_parent_id(&self) -> i32 {
self.0.parent_id
}
pub fn get_roi_type<'a>(&self) -> &'a str {
glib::Quark::from_glib(self.0.roi_type).to_string()
}
#[cfg(feature = "v1_14")]
pub fn get_params(&self) -> ParamsIter {
ParamsIter {
_meta: self,
list: self.0.params,
}
}
#[cfg(feature = "v1_14")]
pub fn get_param<'b>(&self, name: &'b str) -> Option<&gst::StructureRef> {
self.get_params().find(|s| s.get_name() == name)
}
pub fn set_rect(&mut self, rect: (u32, u32, u32, u32)) {
self.0.x = rect.0;
self.0.y = rect.1;
self.0.w = rect.2;
self.0.h = rect.3;
}
pub fn set_id(&mut self, id: i32) {
self.0.id = id
}
pub fn set_parent_id(&mut self, id: i32) {
self.0.parent_id = id
}
#[cfg(feature = "v1_14")]
pub fn add_param(&mut self, s: gst::Structure) {
unsafe {
gst_video_sys::gst_video_region_of_interest_meta_add_param(&mut self.0, s.into_ptr());
}
}
}
#[cfg(feature = "v1_14")]
pub struct ParamsIter<'a> {
_meta: &'a VideoRegionOfInterestMeta,
list: *const glib_sys::GList,
}
#[cfg(feature = "v1_14")]
impl<'a> Iterator for ParamsIter<'a> {
type Item = &'a gst::StructureRef;
fn next(&mut self) -> Option<&'a gst::StructureRef> {
if self.list.is_null() {
return None;
}
unsafe {
let data = (*self.list).data;
assert!(!data.is_null());
self.list = (*self.list).next;
let s = gst::StructureRef::from_glib_borrow(data as *const gst_sys::GstStructure);
Some(s)
}
}
}
unsafe impl MetaAPI for VideoRegionOfInterestMeta {
type GstType = gst_video_sys::GstVideoRegionOfInterestMeta;
fn get_meta_api() -> glib::Type {
unsafe { from_glib(gst_video_sys::gst_video_region_of_interest_meta_api_get_type()) }
}
}
impl fmt::Debug for VideoRegionOfInterestMeta {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
let mut f = f.debug_struct("VideoRegionOfInterestMeta");
f.field("roi_type", &self.get_roi_type())
.field("rect", &self.get_rect())
.field("id", &self.get_id())
.field("parent_id", &self.get_parent_id());
#[cfg(feature = "v1_14")]
{
f.field("params", &self.get_params().collect::<Vec<_>>());
}
f.finish()
}
}
#[repr(C)]
pub struct VideoAffineTransformationMeta(gst_video_sys::GstVideoAffineTransformationMeta);
unsafe impl Send for VideoAffineTransformationMeta {}
unsafe impl Sync for VideoAffineTransformationMeta {}
impl VideoAffineTransformationMeta {
pub fn add<'a>(
buffer: &'a mut gst::BufferRef,
matrix: Option<&[f32; 16]>,
) -> gst::MetaRefMut<'a, Self, gst::meta::Standalone> {
skip_assert_initialized!();
unsafe {
let meta = gst_sys::gst_buffer_add_meta(
buffer.as_mut_ptr(),
gst_video_sys::gst_video_affine_transformation_meta_get_info(),
ptr::null_mut(),
) as *mut gst_video_sys::GstVideoAffineTransformationMeta;
if let Some(matrix) = matrix {
let meta = &mut *meta;
meta.matrix.copy_from_slice(matrix);
}
Self::from_mut_ptr(buffer, meta)
}
}
pub fn get_matrix(&self) -> &[f32; 16] {
&self.0.matrix
}
pub fn set_matrix(&mut self, matrix: &[f32; 16]) {
self.0.matrix.copy_from_slice(matrix);
}
pub fn apply_matrix(&mut self, matrix: &[f32; 16]) {
unsafe {
gst_video_sys::gst_video_affine_transformation_meta_apply_matrix(&mut self.0, matrix);
}
}
}
unsafe impl MetaAPI for VideoAffineTransformationMeta {
type GstType = gst_video_sys::GstVideoAffineTransformationMeta;
fn get_meta_api() -> glib::Type {
unsafe { from_glib(gst_video_sys::gst_video_affine_transformation_meta_api_get_type()) }
}
}
impl fmt::Debug for VideoAffineTransformationMeta {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f.debug_struct("VideoAffineTransformationMeta")
.field("matrix", &self.get_matrix())
.finish()
}
}
#[repr(C)]
pub struct VideoOverlayCompositionMeta(gst_video_sys::GstVideoOverlayCompositionMeta);