gstreamer-rs/gstreamer/src/sample.rs

251 lines
6.9 KiB
Rust
Raw Normal View History

2017-07-28 17:04:15 +00:00
// Copyright (C) 2017 Sebastian Dröge <sebastian@centricular.com>
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.
2017-12-30 10:02:33 +00:00
use std::fmt;
2018-04-01 08:30:03 +00:00
use std::ptr;
2017-07-28 17:04:15 +00:00
2018-12-08 11:06:44 +00:00
use glib::translate::{from_glib_full, from_glib_none, mut_override, ToGlibPtr};
2017-07-28 17:04:15 +00:00
use crate::Buffer;
use crate::BufferList;
use crate::BufferListRef;
use crate::BufferRef;
use crate::Caps;
use crate::CapsRef;
use crate::FormattedSegment;
use crate::FormattedValue;
use crate::Segment;
use crate::Structure;
use crate::StructureRef;
gst_define_mini_object_wrapper!(Sample, SampleRef, ffi::GstSample, || {
ffi::gst_sample_get_type()
});
2017-07-28 17:04:15 +00:00
#[derive(Debug, Clone)]
pub struct SampleBuilder<'a> {
buffer: Option<&'a Buffer>,
buffer_list: Option<&'a BufferList>,
caps: Option<&'a Caps>,
segment: Option<&'a Segment>,
info: Option<Structure>,
}
impl<'a> SampleBuilder<'a> {
pub fn buffer(self, buffer: &'a Buffer) -> Self {
Self {
buffer: Some(buffer),
buffer_list: None,
..self
}
}
pub fn buffer_list(self, buffer_list: &'a BufferList) -> Self {
Self {
buffer: None,
buffer_list: Some(buffer_list),
..self
}
}
pub fn caps(self, caps: &'a Caps) -> Self {
Self {
caps: Some(caps),
..self
}
}
pub fn segment<F: FormattedValue>(self, segment: &'a FormattedSegment<F>) -> Self {
Self {
segment: Some(segment.upcast_ref()),
..self
}
}
pub fn info(self, info: Structure) -> Self {
Self {
info: Some(info),
..self
}
}
pub fn build(self) -> Sample {
2017-07-28 17:04:15 +00:00
assert_initialized_main_thread!();
2017-07-28 17:04:15 +00:00
unsafe {
let info = self.info.map(|i| i.into_ptr()).unwrap_or(ptr::null_mut());
2017-07-28 17:04:15 +00:00
let sample: Sample = from_glib_full(ffi::gst_sample_new(
self.buffer.to_glib_none().0,
self.caps.to_glib_none().0,
self.segment.to_glib_none().0,
mut_override(info),
));
if let Some(buffer_list) = self.buffer_list {
ffi::gst_sample_set_buffer_list(
sample.to_glib_none().0,
buffer_list.to_glib_none().0,
);
}
sample
2017-07-28 17:04:15 +00:00
}
}
}
impl Sample {
pub fn builder<'a>() -> SampleBuilder<'a> {
SampleBuilder {
buffer: None,
buffer_list: None,
caps: None,
segment: None,
info: None,
}
}
2017-07-28 17:04:15 +00:00
}
impl SampleRef {
pub fn get_buffer(&self) -> Option<&BufferRef> {
unsafe {
let ptr = ffi::gst_sample_get_buffer(self.as_mut_ptr());
if ptr.is_null() {
None
} else {
Some(BufferRef::from_ptr(ptr))
}
}
2017-07-28 17:04:15 +00:00
}
pub fn get_buffer_owned(&self) -> Option<Buffer> {
unsafe {
self.get_buffer()
.map(|buffer| from_glib_none(buffer.as_ptr()))
}
}
pub fn get_buffer_list(&self) -> Option<&BufferListRef> {
unsafe {
let ptr = ffi::gst_sample_get_buffer_list(self.as_mut_ptr());
if ptr.is_null() {
None
} else {
Some(BufferListRef::from_ptr(ptr))
}
}
2017-08-03 07:15:20 +00:00
}
2017-07-28 17:04:15 +00:00
pub fn get_buffer_list_owned(&self) -> Option<BufferList> {
unsafe {
self.get_buffer_list()
.map(|list| from_glib_none(list.as_ptr()))
}
}
pub fn get_caps(&self) -> Option<&CapsRef> {
unsafe {
let ptr = ffi::gst_sample_get_caps(self.as_mut_ptr());
if ptr.is_null() {
None
} else {
Some(CapsRef::from_ptr(ptr))
}
}
2017-07-28 17:04:15 +00:00
}
pub fn get_caps_owned(&self) -> Option<Caps> {
unsafe { self.get_caps().map(|caps| from_glib_none(caps.as_ptr())) }
}
2017-07-28 17:04:15 +00:00
pub fn get_segment(&self) -> Option<Segment> {
unsafe { from_glib_none(ffi::gst_sample_get_segment(self.as_mut_ptr())) }
2017-07-28 17:04:15 +00:00
}
pub fn get_info(&self) -> Option<&StructureRef> {
2017-07-28 17:04:15 +00:00
unsafe {
let ptr = ffi::gst_sample_get_info(self.as_mut_ptr());
2017-07-28 17:04:15 +00:00
if ptr.is_null() {
None
} else {
Some(StructureRef::from_glib_borrow(ptr))
}
}
}
#[cfg(any(feature = "v1_16", feature = "dox"))]
#[cfg_attr(feature = "dox", doc(cfg(feature = "v1_16")))]
pub fn set_buffer(&mut self, buffer: Option<&Buffer>) {
unsafe { ffi::gst_sample_set_buffer(self.as_mut_ptr(), buffer.to_glib_none().0) }
}
#[cfg(any(feature = "v1_16", feature = "dox"))]
#[cfg_attr(feature = "dox", doc(cfg(feature = "v1_16")))]
pub fn set_buffer_list(&mut self, buffer_list: Option<&BufferList>) {
unsafe { ffi::gst_sample_set_buffer_list(self.as_mut_ptr(), buffer_list.to_glib_none().0) }
}
#[cfg(any(feature = "v1_16", feature = "dox"))]
#[cfg_attr(feature = "dox", doc(cfg(feature = "v1_16")))]
pub fn set_caps(&mut self, caps: Option<&Caps>) {
unsafe { ffi::gst_sample_set_caps(self.as_mut_ptr(), caps.to_glib_none().0) }
}
#[cfg(any(feature = "v1_16", feature = "dox"))]
#[cfg_attr(feature = "dox", doc(cfg(feature = "v1_16")))]
pub fn set_segment(&mut self, segment: Option<&Segment>) {
unsafe { ffi::gst_sample_set_segment(self.as_mut_ptr(), segment.to_glib_none().0) }
}
#[cfg(any(feature = "v1_16", feature = "dox"))]
#[cfg_attr(feature = "dox", doc(cfg(feature = "v1_16")))]
pub fn set_info(&mut self, info: Option<Structure>) {
unsafe {
ffi::gst_sample_set_info(
self.as_mut_ptr(),
info.map(|i| i.into_ptr()).unwrap_or(ptr::null_mut()),
);
}
}
2017-07-28 17:04:15 +00:00
}
impl fmt::Debug for Sample {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
SampleRef::fmt(self, f)
}
}
2017-12-30 10:02:33 +00:00
impl fmt::Debug for SampleRef {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f.debug_struct("Sample")
.field("buffer", &self.get_buffer())
.field("caps", &self.get_caps())
.field("segment", &self.get_segment())
.field("info", &self.get_info())
.finish()
}
}
#[cfg(test)]
mod tests {
#[test]
fn test_sample_new_with_info() {
use crate::Sample;
use crate::Structure;
crate::init().unwrap();
let info = Structure::builder("sample.info")
.field("f3", &123i32)
.build();
let sample = Sample::builder().info(info).build();
assert!(sample.get_info().is_some());
}
}