rav1d-safe 0.5.5

Safe SIMD fork of rav1d - Rust AV1 decoder with archmage intrinsics
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

#![cfg_attr(not(any(feature = "asm", feature = "c-ffi")), forbid(unsafe_code))]
#[cfg(any(feature = "asm", feature = "c-ffi"))]
use crate::include::common::validate::validate_input;
use crate::include::dav1d::common::Rav1dDataProps;
use crate::include::dav1d::data::Rav1dData;
use crate::src::c_arc::CArc;
#[cfg(any(feature = "asm", feature = "c-ffi"))]
use crate::src::c_box::CBox;
#[cfg(any(feature = "asm", feature = "c-ffi"))]
use crate::src::c_box::FnFree;
#[cfg(any(feature = "asm", feature = "c-ffi"))]
use crate::src::c_box::Free;
#[cfg(any(feature = "asm", feature = "c-ffi"))]
use crate::src::error::Rav1dError::EINVAL;
#[cfg(feature = "c-ffi")]
use crate::src::error::Rav1dResult;
#[cfg(any(feature = "asm", feature = "c-ffi"))]
use crate::src::send_sync_non_null::SendSyncNonNull;
#[cfg(any(feature = "asm", feature = "c-ffi"))]
use std::ffi::c_void;
#[cfg(any(feature = "asm", feature = "c-ffi"))]
use std::ptr::NonNull;

impl From<CArc<[u8]>> for Rav1dData {
    fn from(data: CArc<[u8]>) -> Self {
        let size = data.len();
        Self {
            data: Some(data),
            m: Rav1dDataProps {
                size,
                ..Default::default()
            },
        }
    }
}

impl Rav1dData {
    #[cfg(feature = "c-ffi")]
    pub fn create(size: usize) -> Rav1dResult<Self> {
        let data = CArc::zeroed_slice(size)?;
        Ok(data.into())
    }

    /// # Safety
    ///
    /// See [`CBox::from_c`]'s safety for `data`, `free_callback`, `cookie`.
    #[cfg(any(feature = "asm", feature = "c-ffi"))]
    pub unsafe fn wrap(
        data: NonNull<[u8]>,
        free_callback: Option<FnFree>,
        cookie: Option<SendSyncNonNull<c_void>>,
    ) -> Rav1dResult<Self> {
        let free = validate_input!(free_callback.ok_or(EINVAL))?;
        let free = Free { free, cookie };
        // SAFETY: Preconditions delegate to `CBox::from_c`'s safety.
        let data = unsafe { CBox::from_c(data, free) };
        let data = CArc::wrap(data)?;
        Ok(data.into())
    }

    /// # Safety
    ///
    /// See [`CBox::from_c`]'s safety for `user_data`, `free_callback`, `cookie`.
    #[cfg(any(feature = "asm", feature = "c-ffi"))]
    pub unsafe fn wrap_user_data(
        &mut self,
        user_data: NonNull<u8>,
        free_callback: Option<FnFree>,
        cookie: Option<SendSyncNonNull<c_void>>,
    ) -> Rav1dResult {
        let free = validate_input!(free_callback.ok_or(EINVAL))?;
        let free = Free { free, cookie };
        // SAFETY: Preconditions delegate to `CBox::from_c`'s safety.
        let user_data = unsafe { CBox::from_c(user_data, free) };
        let user_data = CArc::wrap(user_data)?;
        self.m.user_data = Some(user_data);
        Ok(())
    }
}

impl AsRef<[u8]> for Rav1dData {
    fn as_ref(&self) -> &[u8] {
        match &self.data {
            Some(data) => data.as_ref(),
            None => &[],
        }
    }
}