gstreamer_allocators/
drm_dumb_allocator.rs

1use std::{fmt, mem, os::unix::prelude::IntoRawFd};
2
3use glib::{prelude::*, translate::*};
4use gst::{Memory, MemoryRef};
5
6use crate::{ffi, DRMDumbAllocator, DmaBufMemory};
7
8gst::memory_object_wrapper!(
9    DRMDumbMemory,
10    DRMDumbMemoryRef,
11    gst::ffi::GstMemory,
12    |mem: &gst::MemoryRef| { unsafe { from_glib(ffi::gst_is_drm_dumb_memory(mem.as_mut_ptr())) } },
13    Memory,
14    MemoryRef
15);
16
17impl fmt::Debug for DRMDumbMemory {
18    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
19        DRMDumbMemoryRef::fmt(self, f)
20    }
21}
22
23impl fmt::Debug for DRMDumbMemoryRef {
24    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
25        MemoryRef::fmt(self, f)
26    }
27}
28
29impl DRMDumbMemoryRef {
30    #[doc(alias = "gst_drm_dumb_memory_get_handle")]
31    pub fn fd(&self) -> u32 {
32        skip_assert_initialized!();
33        unsafe { ffi::gst_drm_dumb_memory_get_handle(self.as_mut_ptr()) }
34    }
35
36    #[doc(alias = "gst_drm_dumb_memory_export_dmabuf")]
37    pub fn export_dmabuf(&self) -> Result<DmaBufMemory, glib::BoolError> {
38        skip_assert_initialized!();
39        unsafe {
40            Option::<DmaBufMemory>::from_glib_full(ffi::gst_drm_dumb_memory_export_dmabuf(
41                self.as_mut_ptr(),
42            ))
43            .ok_or_else(|| glib::bool_error!("Failed to export as dmabuf"))
44        }
45    }
46}
47
48impl DRMDumbAllocator {
49    /// Creates a new [`DRMDumbAllocator`][crate::DRMDumbAllocator] for the specific file desciptor. This
50    /// function can fail if the file descriptor is not a DRM device or if
51    /// the DRM device does not support DUMB allocation.
52    /// ## `drm_fd`
53    /// file descriptor of the DRM device
54    ///
55    /// # Returns
56    ///
57    /// a new DRM Dumb allocator. Use `gst_object_unref()`
58    ///  to release the allocator after usage.
59    #[doc(alias = "gst_drm_dumb_allocator_new_with_fd")]
60    #[doc(alias = "new_with_fd")]
61    pub fn with_fd<A: IntoRawFd>(drm_fd: A) -> Result<DRMDumbAllocator, glib::BoolError> {
62        assert_initialized_main_thread!();
63        unsafe {
64            Option::<gst::Allocator>::from_glib_full(ffi::gst_drm_dumb_allocator_new_with_fd(
65                drm_fd.into_raw_fd(),
66            ))
67            .map(|o| o.unsafe_cast())
68            .ok_or_else(|| glib::bool_error!("Failed to create allocator"))
69        }
70    }
71}
72
73pub trait DRMDumbAllocatorExtManual: IsA<DRMDumbAllocator> + 'static {
74    #[doc(alias = "gst_drm_dumb_allocator_alloc")]
75    unsafe fn alloc_drm_dumb(
76        &self,
77        drm_fourcc: u32,
78        width: u32,
79        height: u32,
80    ) -> Result<(gst::Memory, u32), glib::BoolError> {
81        skip_assert_initialized!();
82        let mut out_pitch = mem::MaybeUninit::uninit();
83        Option::<_>::from_glib_full(ffi::gst_drm_dumb_allocator_alloc(
84            self.as_ref().to_glib_none().0,
85            drm_fourcc,
86            width,
87            height,
88            out_pitch.as_mut_ptr(),
89        ))
90        .ok_or_else(|| glib::bool_error!("Failed to allocate memory"))
91        .map(|mem| (mem, unsafe { out_pitch.assume_init() }))
92    }
93}
94
95impl<O: IsA<DRMDumbAllocator>> DRMDumbAllocatorExtManual for O {}
gstreamer_allocators/drm_dumb_allocator.rs

gstreamer_allocators/
drm_dumb_allocator.rs
