ffmpeg_next/util/frame/
mod.rs

1pub mod side_data;
2pub use self::side_data::SideData;
3
4pub mod video;
5pub use self::video::Video;
6
7pub mod audio;
8pub use self::audio::Audio;
9
10pub mod flag;
11pub use self::flag::Flags;
12
13use ffi::*;
14use {Dictionary, DictionaryRef};
15
16#[derive(PartialEq, Eq, Copy, Clone, Debug)]
17pub struct Packet {
18    pub duration: i64,
19    pub position: i64,
20    pub size: usize,
21
22    #[cfg(not(feature = "ffmpeg_5_0"))]
23    pub pts: i64,
24    pub dts: i64,
25}
26
27#[derive(PartialEq, Eq)]
28pub struct Frame {
29    ptr: *mut AVFrame,
30
31    _own: bool,
32}
33
34unsafe impl Send for Frame {}
35unsafe impl Sync for Frame {}
36
37impl Frame {
38    #[inline(always)]
39    pub unsafe fn wrap(ptr: *mut AVFrame) -> Self {
40        Frame { ptr, _own: false }
41    }
42
43    #[inline(always)]
44    pub unsafe fn empty() -> Self {
45        Frame {
46            ptr: av_frame_alloc(),
47            _own: true,
48        }
49    }
50
51    #[inline(always)]
52    pub unsafe fn as_ptr(&self) -> *const AVFrame {
53        self.ptr as *const _
54    }
55
56    #[inline(always)]
57    pub unsafe fn as_mut_ptr(&mut self) -> *mut AVFrame {
58        self.ptr
59    }
60
61    #[inline(always)]
62    pub unsafe fn is_empty(&self) -> bool {
63        (*self.as_ptr()).data[0].is_null()
64    }
65}
66
67impl Frame {
68    #[inline]
69    pub fn is_key(&self) -> bool {
70        unsafe { (*self.as_ptr()).key_frame == 1 }
71    }
72
73    #[inline]
74    pub fn is_corrupt(&self) -> bool {
75        self.flags().contains(Flags::CORRUPT)
76    }
77
78    #[inline]
79    pub fn packet(&self) -> Packet {
80        unsafe {
81            Packet {
82                #[cfg(not(feature = "ffmpeg_7_0"))]
83                duration: (*self.as_ptr()).pkt_duration,
84                #[cfg(feature = "ffmpeg_7_0")]
85                duration: (*self.as_ptr()).duration,
86
87                position: (*self.as_ptr()).pkt_pos,
88                size: (*self.as_ptr()).pkt_size as usize,
89
90                #[cfg(not(feature = "ffmpeg_5_0"))]
91                pts: (*self.as_ptr()).pkt_pts,
92                dts: (*self.as_ptr()).pkt_dts,
93            }
94        }
95    }
96
97    #[inline]
98    pub fn pts(&self) -> Option<i64> {
99        unsafe {
100            match (*self.as_ptr()).pts {
101                AV_NOPTS_VALUE => None,
102                pts => Some(pts),
103            }
104        }
105    }
106
107    #[inline]
108    pub fn set_pts(&mut self, value: Option<i64>) {
109        unsafe {
110            (*self.as_mut_ptr()).pts = value.unwrap_or(AV_NOPTS_VALUE);
111        }
112    }
113
114    #[inline]
115    pub fn timestamp(&self) -> Option<i64> {
116        unsafe {
117            match (*self.as_ptr()).best_effort_timestamp {
118                AV_NOPTS_VALUE => None,
119                t => Some(t),
120            }
121        }
122    }
123
124    #[inline]
125    pub fn quality(&self) -> usize {
126        unsafe { (*self.as_ptr()).quality as usize }
127    }
128
129    #[inline]
130    pub fn flags(&self) -> Flags {
131        unsafe { Flags::from_bits_truncate((*self.as_ptr()).flags) }
132    }
133
134    #[inline]
135    pub fn metadata(&self) -> DictionaryRef {
136        unsafe { DictionaryRef::wrap((*self.as_ptr()).metadata) }
137    }
138
139    #[inline]
140    pub fn set_metadata(&mut self, value: Dictionary) {
141        unsafe { (*self.as_mut_ptr()).metadata = value.disown() }
142    }
143
144    #[inline]
145    pub fn side_data(&self, kind: side_data::Type) -> Option<SideData> {
146        unsafe {
147            let ptr = av_frame_get_side_data(self.as_ptr(), kind.into());
148
149            if ptr.is_null() {
150                None
151            } else {
152                Some(SideData::wrap(ptr))
153            }
154        }
155    }
156
157    #[inline]
158    pub fn new_side_data(&mut self, kind: side_data::Type, size: usize) -> Option<SideData> {
159        unsafe {
160            let ptr = av_frame_new_side_data(self.as_mut_ptr(), kind.into(), size as _);
161
162            if ptr.is_null() {
163                None
164            } else {
165                Some(SideData::wrap(ptr))
166            }
167        }
168    }
169
170    #[inline]
171    pub fn remove_side_data(&mut self, kind: side_data::Type) {
172        unsafe {
173            av_frame_remove_side_data(self.as_mut_ptr(), kind.into());
174        }
175    }
176}
177
178impl Drop for Frame {
179    #[inline]
180    fn drop(&mut self) {
181        unsafe {
182            av_frame_free(&mut self.as_mut_ptr());
183        }
184    }
185}
ffmpeg_next/util/frame/mod.rs

ffmpeg_next/util/frame/
mod.rs
