gstreamer-rs/gstreamer/src/sample.rs

// 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.

use std::fmt;
use std::ptr;

use gst_sys;

use glib;
use glib::translate::{from_glib_full, from_glib_none, mut_override, ToGlibPtr};

use miniobject::*;
use BufferListRef;
use BufferRef;
use CapsRef;
use FormattedSegment;
use FormattedValue;
use Segment;
use Structure;
use StructureRef;

gst_define_mini_object_wrapper!(Sample, SampleRef, gst_sys::GstSample, [Debug,], || {
    gst_sys::gst_sample_get_type()
});

impl Sample {
    pub fn new<F: FormattedValue>(
        buffer: Option<&BufferRef>,
        caps: Option<&CapsRef>,
        segment: Option<&FormattedSegment<F>>,
        info: Option<Structure>,
    ) -> Self {
        assert_initialized_main_thread!();
        unsafe {
            let info = info.map(|i| i.into_ptr()).unwrap_or(ptr::null_mut());

            from_glib_full(gst_sys::gst_sample_new(
                buffer
                    .map(|buffer| buffer.as_mut_ptr())
                    .unwrap_or(ptr::null_mut()),
                caps.map(|caps| caps.as_mut_ptr())
                    .unwrap_or(ptr::null_mut()),
                segment.to_glib_none().0,
                mut_override(info),
            ))
        }
    }

    pub fn with_buffer_list<F: FormattedValue>(
        buffer_list: Option<&BufferListRef>,
        caps: Option<&CapsRef>,
        segment: Option<&FormattedSegment<F>>,
        info: Option<Structure>,
    ) -> Self {
        assert_initialized_main_thread!();
        let sample = Self::new(None, caps, segment, info);
        unsafe {
            gst_sys::gst_sample_set_buffer_list(
                sample.to_glib_none().0,
                buffer_list
                    .map(|buffer_list| buffer_list.as_mut_ptr())
                    .unwrap_or(ptr::null_mut()),
            );
        }
        sample
    }
}

impl SampleRef {
    pub fn get_buffer(&self) -> Option<&BufferRef> {
        unsafe {
            let ptr = gst_sys::gst_sample_get_buffer(self.as_mut_ptr());
            if ptr.is_null() {
                None
            } else {
                Some(BufferRef::from_ptr(ptr))
            }
        }
    }

    pub fn get_buffer_list(&self) -> Option<&BufferListRef> {
        unsafe {
            let ptr = gst_sys::gst_sample_get_buffer_list(self.as_mut_ptr());
            if ptr.is_null() {
                None
            } else {
                Some(BufferListRef::from_ptr(ptr))
            }
        }
    }

    pub fn get_caps(&self) -> Option<&CapsRef> {
        unsafe {
            let ptr = gst_sys::gst_sample_get_caps(self.as_mut_ptr());
            if ptr.is_null() {
                None
            } else {
                Some(CapsRef::from_ptr(ptr))
            }
        }
    }

    pub fn get_segment(&self) -> Option<Segment> {
        unsafe { from_glib_none(gst_sys::gst_sample_get_segment(self.as_mut_ptr())) }
    }

    pub fn get_info(&self) -> Option<&StructureRef> {
        unsafe {
            let ptr = gst_sys::gst_sample_get_info(self.as_mut_ptr());
            if ptr.is_null() {
                None
            } else {
                Some(StructureRef::from_glib_borrow(ptr))
            }
        }
    }

    #[cfg(any(feature = "v1_16", feature = "dox"))]
    pub fn set_buffer(&mut self, buffer: Option<&BufferRef>) {
        unsafe {
            gst_sys::gst_sample_set_buffer(
                self.as_mut_ptr(),
                buffer
                    .map(|buffer| buffer.as_mut_ptr())
                    .unwrap_or(ptr::null_mut()),
            )
        }
    }

    #[cfg(any(feature = "v1_16", feature = "dox"))]
    pub fn set_buffer_list(&mut self, buffer_list: Option<&BufferListRef>) {
        unsafe {
            gst_sys::gst_sample_set_buffer_list(
                self.as_mut_ptr(),
                buffer_list
                    .map(|buffer_list| buffer_list.as_mut_ptr())
                    .unwrap_or(ptr::null_mut()),
            )
        }
    }

    #[cfg(any(feature = "v1_16", feature = "dox"))]
    pub fn set_caps(&mut self, caps: Option<&CapsRef>) {
        unsafe {
            gst_sys::gst_sample_set_caps(
                self.as_mut_ptr(),
                caps.map(|caps| caps.as_mut_ptr())
                    .unwrap_or(ptr::null_mut()),
            )
        }
    }

    #[cfg(any(feature = "v1_16", feature = "dox"))]
    pub fn set_segment(&mut self, segment: Option<&Segment>) {
        unsafe { gst_sys::gst_sample_set_segment(self.as_mut_ptr(), segment.to_glib_none().0) }
    }

    #[cfg(any(feature = "v1_16", feature = "dox"))]
    pub fn set_info(&mut self, info: Option<Structure>) {
        unsafe {
            gst_sys::gst_sample_set_info(
                self.as_mut_ptr(),
                info.map(|i| i.into_ptr()).unwrap_or(ptr::null_mut()),
            );
        }
    }
}

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 GenericFormattedValue;
        use Sample;
        use Structure;

        ::init().unwrap();

        let info = Structure::builder("sample.info")
            .field("f3", &123i32)
            .build();
        let sample = Sample::new::<GenericFormattedValue>(None, None, None, Some(info));

        assert!(sample.get_info().is_some());
    }
}