mediadecode_ffmpeg/
boundary.rs

1//! Boundary conversions between FFmpeg's bindgen integers and the
2//! unified [`mediadecode`] vocabulary.
3//!
4//! Centralised so the rest of the crate never compares raw
5//! `AVPixelFormat` integers against literals or transmutes back into
6//! the bindgen enum (UB hazard when the value isn't in the enum's
7//! discriminant set).
8
9use core::ffi::c_int;
10
11use ffmpeg_next::{Packet, ffi::AVPixelFormat};
12use mediadecode::{
13  PixelFormat, Timestamp,
14  channel::AudioChannelLayout,
15  frame::{AudioFrame, Dimensions, Plane, SubtitleFrame, VideoFrame},
16  packet::{AudioPacket, PacketFlags as MdPacketFlags, SubtitlePacket, VideoPacket},
17  subtitle::SubtitlePayload,
18};
19
20use crate::{
21  FfmpegBuffer,
22  extras::{
23    AudioFrameExtra, AudioPacketExtra, SubtitleFrameExtra, SubtitlePacketExtra, VideoFrameExtra,
24    VideoPacketExtra,
25  },
26  sample_format::SampleFormat,
27};
28
29/// Maps a raw `AVFrame.format` integer (i.e. the value of an
30/// `AVPixelFormat` enum variant) onto [`mediadecode::PixelFormat`].
31///
32/// Returns [`PixelFormat::None`] for raw integers we don't have a
33/// mapping for — including `AV_PIX_FMT_NONE` itself and the
34/// hardware-frame markers (`AV_PIX_FMT_VIDEOTOOLBOX` / `_VAAPI` /
35/// `_CUDA` / `_D3D11` / …), since those never describe CPU-side pixel
36/// data and the unified enum intentionally doesn't carry them. Use
37/// [`is_hardware_pix_fmt`] to identify HW frames before transferring
38/// to a CPU format.
39///
40/// mediaframe 0.3 struck `PixelFormat::Unknown(u32)`, so the raw
41/// integer no longer rides along in the returned value; the caller's
42/// own `raw` is the place it survives. Every consumer in this crate
43/// already treats the fall-through as "not a deliverable CPU format"
44/// (`pixdesc::to_av_pixel_format`, `is_supported_cpu_pix_fmt` and the
45/// geometry tables all reject it), so the rejection is unchanged.
46///
47/// The match never constructs an `AVPixelFormat` from a runtime
48/// value; it compares the input against `AVPixelFormat::AV_PIX_FMT_X
49/// as i32` constants. Sound regardless of which discriminant set the
50/// linked FFmpeg version exposes.
51pub const fn from_av_pixel_format(raw: i32) -> PixelFormat {
52  // Mirrors `crate::pixdesc::to_av_pixel_format` arm-for-arm (its
53  // inverse). Every deliverable CPU format plus the non-deliverable
54  // formats `to_av` still resolves a constant for (monochrome / PAL /
55  // sub-byte-packed RGB / Bayer) is mapped here, so a frame's raw
56  // `format` integer always lands on the same `PixelFormat` the round
57  // trip would produce. Deliverability (HWACCEL / BAYER / PAL /
58  // BITSTREAM rejection) is enforced separately by
59  // `pixdesc::is_deliverable` / the convert layer — this boundary is
60  // identity-only.
61  //
62  // BE-tagged formats map to mediadecode's distinct `*Be` variants
63  // (never folded onto the LE canonical). Folding BE onto LE silently
64  // corrupted pixel data: each >8-bit sample is byte-swapped between
65  // BE and LE, and the convert path exports the AVBufferRef bytes
66  // verbatim with no endian conversion, so a consumer reading a
67  // BE-tagged frame's planes as LE samples would see every sample
68  // byte-reversed. Mapping to the `*Be` variant keeps the format
69  // distinct so the convert layer can handle (or reject) it correctly.
70  //
71  // The match never constructs an `AVPixelFormat` from a runtime
72  // value; it compares the input against `AVPixelFormat::AV_PIX_FMT_X
73  // as i32` constants. Sound regardless of which discriminant set the
74  // linked FFmpeg version exposes.
75  match raw {
76    // Planar YUV 8-bit.
77    x if x == AVPixelFormat::AV_PIX_FMT_YUV420P as i32 => PixelFormat::Yuv420p,
78    x if x == AVPixelFormat::AV_PIX_FMT_YUV422P as i32 => PixelFormat::Yuv422p,
79    x if x == AVPixelFormat::AV_PIX_FMT_YUV440P as i32 => PixelFormat::Yuv440p,
80    x if x == AVPixelFormat::AV_PIX_FMT_YUV444P as i32 => PixelFormat::Yuv444p,
81    x if x == AVPixelFormat::AV_PIX_FMT_YUV411P as i32 => PixelFormat::Yuv411p,
82    x if x == AVPixelFormat::AV_PIX_FMT_YUV410P as i32 => PixelFormat::Yuv410p,
83    // Deprecated JPEG-range planar YUV (yuvj-family).
84    x if x == AVPixelFormat::AV_PIX_FMT_YUVJ411P as i32 => PixelFormat::Yuvj411p,
85    x if x == AVPixelFormat::AV_PIX_FMT_YUVJ420P as i32 => PixelFormat::Yuvj420p,
86    x if x == AVPixelFormat::AV_PIX_FMT_YUVJ422P as i32 => PixelFormat::Yuvj422p,
87    x if x == AVPixelFormat::AV_PIX_FMT_YUVJ440P as i32 => PixelFormat::Yuvj440p,
88    x if x == AVPixelFormat::AV_PIX_FMT_YUVJ444P as i32 => PixelFormat::Yuvj444p,
89    // Planar YUV 4:2:0 high-bit.
90    x if x == AVPixelFormat::AV_PIX_FMT_YUV420P9LE as i32 => PixelFormat::Yuv420p9Le,
91    x if x == AVPixelFormat::AV_PIX_FMT_YUV420P9BE as i32 => PixelFormat::Yuv420p9Be,
92    x if x == AVPixelFormat::AV_PIX_FMT_YUV420P10LE as i32 => PixelFormat::Yuv420p10Le,
93    x if x == AVPixelFormat::AV_PIX_FMT_YUV420P10BE as i32 => PixelFormat::Yuv420p10Be,
94    x if x == AVPixelFormat::AV_PIX_FMT_YUV420P12LE as i32 => PixelFormat::Yuv420p12Le,
95    x if x == AVPixelFormat::AV_PIX_FMT_YUV420P12BE as i32 => PixelFormat::Yuv420p12Be,
96    x if x == AVPixelFormat::AV_PIX_FMT_YUV420P14LE as i32 => PixelFormat::Yuv420p14Le,
97    x if x == AVPixelFormat::AV_PIX_FMT_YUV420P14BE as i32 => PixelFormat::Yuv420p14Be,
98    x if x == AVPixelFormat::AV_PIX_FMT_YUV420P16LE as i32 => PixelFormat::Yuv420p16Le,
99    x if x == AVPixelFormat::AV_PIX_FMT_YUV420P16BE as i32 => PixelFormat::Yuv420p16Be,
100    // Planar YUV 4:2:2 high-bit.
101    x if x == AVPixelFormat::AV_PIX_FMT_YUV422P9LE as i32 => PixelFormat::Yuv422p9Le,
102    x if x == AVPixelFormat::AV_PIX_FMT_YUV422P9BE as i32 => PixelFormat::Yuv422p9Be,
103    x if x == AVPixelFormat::AV_PIX_FMT_YUV422P10LE as i32 => PixelFormat::Yuv422p10Le,
104    x if x == AVPixelFormat::AV_PIX_FMT_YUV422P10BE as i32 => PixelFormat::Yuv422p10Be,
105    x if x == AVPixelFormat::AV_PIX_FMT_YUV422P12LE as i32 => PixelFormat::Yuv422p12Le,
106    x if x == AVPixelFormat::AV_PIX_FMT_YUV422P12BE as i32 => PixelFormat::Yuv422p12Be,
107    x if x == AVPixelFormat::AV_PIX_FMT_YUV422P14LE as i32 => PixelFormat::Yuv422p14Le,
108    x if x == AVPixelFormat::AV_PIX_FMT_YUV422P14BE as i32 => PixelFormat::Yuv422p14Be,
109    x if x == AVPixelFormat::AV_PIX_FMT_YUV422P16LE as i32 => PixelFormat::Yuv422p16Le,
110    x if x == AVPixelFormat::AV_PIX_FMT_YUV422P16BE as i32 => PixelFormat::Yuv422p16Be,
111    // Planar YUV 4:4:0 high-bit.
112    x if x == AVPixelFormat::AV_PIX_FMT_YUV440P10LE as i32 => PixelFormat::Yuv440p10Le,
113    x if x == AVPixelFormat::AV_PIX_FMT_YUV440P10BE as i32 => PixelFormat::Yuv440p10Be,
114    x if x == AVPixelFormat::AV_PIX_FMT_YUV440P12LE as i32 => PixelFormat::Yuv440p12Le,
115    x if x == AVPixelFormat::AV_PIX_FMT_YUV440P12BE as i32 => PixelFormat::Yuv440p12Be,
116    // Planar YUV 4:4:4 high-bit.
117    x if x == AVPixelFormat::AV_PIX_FMT_YUV444P9LE as i32 => PixelFormat::Yuv444p9Le,
118    x if x == AVPixelFormat::AV_PIX_FMT_YUV444P9BE as i32 => PixelFormat::Yuv444p9Be,
119    x if x == AVPixelFormat::AV_PIX_FMT_YUV444P10LE as i32 => PixelFormat::Yuv444p10Le,
120    x if x == AVPixelFormat::AV_PIX_FMT_YUV444P10BE as i32 => PixelFormat::Yuv444p10Be,
121    x if x == AVPixelFormat::AV_PIX_FMT_YUV444P12LE as i32 => PixelFormat::Yuv444p12Le,
122    x if x == AVPixelFormat::AV_PIX_FMT_YUV444P12BE as i32 => PixelFormat::Yuv444p12Be,
123    x if x == AVPixelFormat::AV_PIX_FMT_YUV444P14LE as i32 => PixelFormat::Yuv444p14Le,
124    x if x == AVPixelFormat::AV_PIX_FMT_YUV444P14BE as i32 => PixelFormat::Yuv444p14Be,
125    x if x == AVPixelFormat::AV_PIX_FMT_YUV444P16LE as i32 => PixelFormat::Yuv444p16Le,
126    x if x == AVPixelFormat::AV_PIX_FMT_YUV444P16BE as i32 => PixelFormat::Yuv444p16Be,
127    // MSB-packed YUV 4:4:4.
128    x if x == AVPixelFormat::AV_PIX_FMT_YUV444P10MSBLE as i32 => PixelFormat::Yuv444p10MsbLe,
129    x if x == AVPixelFormat::AV_PIX_FMT_YUV444P10MSBBE as i32 => PixelFormat::Yuv444p10MsbBe,
130    x if x == AVPixelFormat::AV_PIX_FMT_YUV444P12MSBLE as i32 => PixelFormat::Yuv444p12MsbLe,
131    x if x == AVPixelFormat::AV_PIX_FMT_YUV444P12MSBBE as i32 => PixelFormat::Yuv444p12MsbBe,
132    // Planar YUVA.
133    x if x == AVPixelFormat::AV_PIX_FMT_YUVA420P as i32 => PixelFormat::Yuva420p,
134    x if x == AVPixelFormat::AV_PIX_FMT_YUVA422P as i32 => PixelFormat::Yuva422p,
135    x if x == AVPixelFormat::AV_PIX_FMT_YUVA444P as i32 => PixelFormat::Yuva444p,
136    x if x == AVPixelFormat::AV_PIX_FMT_YUVA420P9LE as i32 => PixelFormat::Yuva420p9Le,
137    x if x == AVPixelFormat::AV_PIX_FMT_YUVA420P9BE as i32 => PixelFormat::Yuva420p9Be,
138    x if x == AVPixelFormat::AV_PIX_FMT_YUVA422P9LE as i32 => PixelFormat::Yuva422p9Le,
139    x if x == AVPixelFormat::AV_PIX_FMT_YUVA422P9BE as i32 => PixelFormat::Yuva422p9Be,
140    x if x == AVPixelFormat::AV_PIX_FMT_YUVA444P9LE as i32 => PixelFormat::Yuva444p9Le,
141    x if x == AVPixelFormat::AV_PIX_FMT_YUVA444P9BE as i32 => PixelFormat::Yuva444p9Be,
142    x if x == AVPixelFormat::AV_PIX_FMT_YUVA420P10LE as i32 => PixelFormat::Yuva420p10Le,
143    x if x == AVPixelFormat::AV_PIX_FMT_YUVA420P10BE as i32 => PixelFormat::Yuva420p10Be,
144    x if x == AVPixelFormat::AV_PIX_FMT_YUVA422P10LE as i32 => PixelFormat::Yuva422p10Le,
145    x if x == AVPixelFormat::AV_PIX_FMT_YUVA422P10BE as i32 => PixelFormat::Yuva422p10Be,
146    x if x == AVPixelFormat::AV_PIX_FMT_YUVA444P10LE as i32 => PixelFormat::Yuva444p10Le,
147    x if x == AVPixelFormat::AV_PIX_FMT_YUVA444P10BE as i32 => PixelFormat::Yuva444p10Be,
148    x if x == AVPixelFormat::AV_PIX_FMT_YUVA422P12LE as i32 => PixelFormat::Yuva422p12Le,
149    x if x == AVPixelFormat::AV_PIX_FMT_YUVA422P12BE as i32 => PixelFormat::Yuva422p12Be,
150    x if x == AVPixelFormat::AV_PIX_FMT_YUVA444P12LE as i32 => PixelFormat::Yuva444p12Le,
151    x if x == AVPixelFormat::AV_PIX_FMT_YUVA444P12BE as i32 => PixelFormat::Yuva444p12Be,
152    x if x == AVPixelFormat::AV_PIX_FMT_YUVA420P16LE as i32 => PixelFormat::Yuva420p16Le,
153    x if x == AVPixelFormat::AV_PIX_FMT_YUVA420P16BE as i32 => PixelFormat::Yuva420p16Be,
154    x if x == AVPixelFormat::AV_PIX_FMT_YUVA422P16LE as i32 => PixelFormat::Yuva422p16Le,
155    x if x == AVPixelFormat::AV_PIX_FMT_YUVA422P16BE as i32 => PixelFormat::Yuva422p16Be,
156    x if x == AVPixelFormat::AV_PIX_FMT_YUVA444P16LE as i32 => PixelFormat::Yuva444p16Le,
157    x if x == AVPixelFormat::AV_PIX_FMT_YUVA444P16BE as i32 => PixelFormat::Yuva444p16Be,
158    // Semi-planar YUV 8-bit.
159    x if x == AVPixelFormat::AV_PIX_FMT_NV12 as i32 => PixelFormat::Nv12,
160    x if x == AVPixelFormat::AV_PIX_FMT_NV21 as i32 => PixelFormat::Nv21,
161    x if x == AVPixelFormat::AV_PIX_FMT_NV16 as i32 => PixelFormat::Nv16,
162    x if x == AVPixelFormat::AV_PIX_FMT_NV24 as i32 => PixelFormat::Nv24,
163    x if x == AVPixelFormat::AV_PIX_FMT_NV42 as i32 => PixelFormat::Nv42,
164    x if x == AVPixelFormat::AV_PIX_FMT_NV20LE as i32 => PixelFormat::Nv20Le,
165    x if x == AVPixelFormat::AV_PIX_FMT_NV20BE as i32 => PixelFormat::Nv20Be,
166    // Semi-planar YUV high-bit.
167    x if x == AVPixelFormat::AV_PIX_FMT_P010LE as i32 => PixelFormat::P010Le,
168    x if x == AVPixelFormat::AV_PIX_FMT_P010BE as i32 => PixelFormat::P010Be,
169    x if x == AVPixelFormat::AV_PIX_FMT_P012LE as i32 => PixelFormat::P012Le,
170    x if x == AVPixelFormat::AV_PIX_FMT_P012BE as i32 => PixelFormat::P012Be,
171    x if x == AVPixelFormat::AV_PIX_FMT_P016LE as i32 => PixelFormat::P016Le,
172    x if x == AVPixelFormat::AV_PIX_FMT_P016BE as i32 => PixelFormat::P016Be,
173    x if x == AVPixelFormat::AV_PIX_FMT_P210LE as i32 => PixelFormat::P210Le,
174    x if x == AVPixelFormat::AV_PIX_FMT_P210BE as i32 => PixelFormat::P210Be,
175    x if x == AVPixelFormat::AV_PIX_FMT_P212LE as i32 => PixelFormat::P212Le,
176    x if x == AVPixelFormat::AV_PIX_FMT_P212BE as i32 => PixelFormat::P212Be,
177    x if x == AVPixelFormat::AV_PIX_FMT_P216LE as i32 => PixelFormat::P216Le,
178    x if x == AVPixelFormat::AV_PIX_FMT_P216BE as i32 => PixelFormat::P216Be,
179    x if x == AVPixelFormat::AV_PIX_FMT_P410LE as i32 => PixelFormat::P410Le,
180    x if x == AVPixelFormat::AV_PIX_FMT_P410BE as i32 => PixelFormat::P410Be,
181    x if x == AVPixelFormat::AV_PIX_FMT_P412LE as i32 => PixelFormat::P412Le,
182    x if x == AVPixelFormat::AV_PIX_FMT_P412BE as i32 => PixelFormat::P412Be,
183    x if x == AVPixelFormat::AV_PIX_FMT_P416LE as i32 => PixelFormat::P416Le,
184    x if x == AVPixelFormat::AV_PIX_FMT_P416BE as i32 => PixelFormat::P416Be,
185    // Packed YUV 8-bit.
186    x if x == AVPixelFormat::AV_PIX_FMT_YUYV422 as i32 => PixelFormat::Yuyv422,
187    x if x == AVPixelFormat::AV_PIX_FMT_UYVY422 as i32 => PixelFormat::Uyvy422,
188    x if x == AVPixelFormat::AV_PIX_FMT_YVYU422 as i32 => PixelFormat::Yvyu422,
189    x if x == AVPixelFormat::AV_PIX_FMT_UYYVYY411 as i32 => PixelFormat::Uyyvyy411,
190    // Packed YUV high-bit.
191    x if x == AVPixelFormat::AV_PIX_FMT_Y210LE as i32 => PixelFormat::Y210Le,
192    x if x == AVPixelFormat::AV_PIX_FMT_Y210BE as i32 => PixelFormat::Y210Be,
193    x if x == AVPixelFormat::AV_PIX_FMT_Y212LE as i32 => PixelFormat::Y212Le,
194    x if x == AVPixelFormat::AV_PIX_FMT_Y212BE as i32 => PixelFormat::Y212Be,
195    x if x == AVPixelFormat::AV_PIX_FMT_Y216LE as i32 => PixelFormat::Y216Le,
196    x if x == AVPixelFormat::AV_PIX_FMT_Y216BE as i32 => PixelFormat::Y216Be,
197    x if x == AVPixelFormat::AV_PIX_FMT_XV30LE as i32 => PixelFormat::Xv30Le,
198    x if x == AVPixelFormat::AV_PIX_FMT_XV30BE as i32 => PixelFormat::Xv30Be,
199    x if x == AVPixelFormat::AV_PIX_FMT_V30XLE as i32 => PixelFormat::V30xLe,
200    x if x == AVPixelFormat::AV_PIX_FMT_V30XBE as i32 => PixelFormat::V30xBe,
201    x if x == AVPixelFormat::AV_PIX_FMT_XV36LE as i32 => PixelFormat::Xv36Le,
202    x if x == AVPixelFormat::AV_PIX_FMT_XV36BE as i32 => PixelFormat::Xv36Be,
203    x if x == AVPixelFormat::AV_PIX_FMT_XV48LE as i32 => PixelFormat::Xv48Le,
204    x if x == AVPixelFormat::AV_PIX_FMT_XV48BE as i32 => PixelFormat::Xv48Be,
205    x if x == AVPixelFormat::AV_PIX_FMT_VUYA as i32 => PixelFormat::Vuya,
206    x if x == AVPixelFormat::AV_PIX_FMT_VUYX as i32 => PixelFormat::Vuyx,
207    x if x == AVPixelFormat::AV_PIX_FMT_AYUV as i32 => PixelFormat::Ayuv,
208    x if x == AVPixelFormat::AV_PIX_FMT_AYUV64LE as i32 => PixelFormat::Ayuv64Le,
209    x if x == AVPixelFormat::AV_PIX_FMT_AYUV64BE as i32 => PixelFormat::Ayuv64Be,
210    x if x == AVPixelFormat::AV_PIX_FMT_UYVA as i32 => PixelFormat::Uyva,
211    x if x == AVPixelFormat::AV_PIX_FMT_VYU444 as i32 => PixelFormat::Vyu444,
212    // XYZ.
213    x if x == AVPixelFormat::AV_PIX_FMT_XYZ12LE as i32 => PixelFormat::Xyz12Le,
214    x if x == AVPixelFormat::AV_PIX_FMT_XYZ12BE as i32 => PixelFormat::Xyz12Be,
215    // Packed RGB 8-bit.
216    x if x == AVPixelFormat::AV_PIX_FMT_RGB24 as i32 => PixelFormat::Rgb24,
217    x if x == AVPixelFormat::AV_PIX_FMT_BGR24 as i32 => PixelFormat::Bgr24,
218    x if x == AVPixelFormat::AV_PIX_FMT_RGBA as i32 => PixelFormat::Rgba,
219    x if x == AVPixelFormat::AV_PIX_FMT_BGRA as i32 => PixelFormat::Bgra,
220    x if x == AVPixelFormat::AV_PIX_FMT_ARGB as i32 => PixelFormat::Argb,
221    x if x == AVPixelFormat::AV_PIX_FMT_ABGR as i32 => PixelFormat::Abgr,
222    x if x == AVPixelFormat::AV_PIX_FMT_RGB0 as i32 => PixelFormat::Rgbx,
223    x if x == AVPixelFormat::AV_PIX_FMT_BGR0 as i32 => PixelFormat::Bgrx,
224    x if x == AVPixelFormat::AV_PIX_FMT_0RGB as i32 => PixelFormat::Xrgb,
225    x if x == AVPixelFormat::AV_PIX_FMT_0BGR as i32 => PixelFormat::Xbgr,
226    x if x == AVPixelFormat::AV_PIX_FMT_X2RGB10LE as i32 => PixelFormat::X2Rgb10Le,
227    x if x == AVPixelFormat::AV_PIX_FMT_X2RGB10BE as i32 => PixelFormat::X2Rgb10Be,
228    x if x == AVPixelFormat::AV_PIX_FMT_X2BGR10LE as i32 => PixelFormat::X2Bgr10Le,
229    x if x == AVPixelFormat::AV_PIX_FMT_X2BGR10BE as i32 => PixelFormat::X2Bgr10Be,
230    // Gbr24p shares AV_PIX_FMT_GBRP's discriminant; mapped to Gbrp above.
231    // Packed RGB high-bit.
232    x if x == AVPixelFormat::AV_PIX_FMT_RGB48LE as i32 => PixelFormat::Rgb48Le,
233    x if x == AVPixelFormat::AV_PIX_FMT_RGB48BE as i32 => PixelFormat::Rgb48Be,
234    x if x == AVPixelFormat::AV_PIX_FMT_BGR48LE as i32 => PixelFormat::Bgr48Le,
235    x if x == AVPixelFormat::AV_PIX_FMT_BGR48BE as i32 => PixelFormat::Bgr48Be,
236    x if x == AVPixelFormat::AV_PIX_FMT_RGBA64LE as i32 => PixelFormat::Rgba64Le,
237    x if x == AVPixelFormat::AV_PIX_FMT_RGBA64BE as i32 => PixelFormat::Rgba64Be,
238    x if x == AVPixelFormat::AV_PIX_FMT_BGRA64LE as i32 => PixelFormat::Bgra64Le,
239    x if x == AVPixelFormat::AV_PIX_FMT_BGRA64BE as i32 => PixelFormat::Bgra64Be,
240    x if x == AVPixelFormat::AV_PIX_FMT_RGB96LE as i32 => PixelFormat::Rgb96Le,
241    x if x == AVPixelFormat::AV_PIX_FMT_RGB96BE as i32 => PixelFormat::Rgb96Be,
242    x if x == AVPixelFormat::AV_PIX_FMT_RGBA128LE as i32 => PixelFormat::Rgba128Le,
243    x if x == AVPixelFormat::AV_PIX_FMT_RGBA128BE as i32 => PixelFormat::Rgba128Be,
244    // Packed RGB float / half-float.
245    x if x == AVPixelFormat::AV_PIX_FMT_RGBF16LE as i32 => PixelFormat::Rgbf16Le,
246    x if x == AVPixelFormat::AV_PIX_FMT_RGBF16BE as i32 => PixelFormat::Rgbf16Be,
247    x if x == AVPixelFormat::AV_PIX_FMT_RGBF32LE as i32 => PixelFormat::Rgbf32Le,
248    x if x == AVPixelFormat::AV_PIX_FMT_RGBF32BE as i32 => PixelFormat::Rgbf32Be,
249    x if x == AVPixelFormat::AV_PIX_FMT_RGBAF16LE as i32 => PixelFormat::Rgbaf16Le,
250    x if x == AVPixelFormat::AV_PIX_FMT_RGBAF16BE as i32 => PixelFormat::Rgbaf16Be,
251    x if x == AVPixelFormat::AV_PIX_FMT_RGBAF32LE as i32 => PixelFormat::Rgbaf32Le,
252    x if x == AVPixelFormat::AV_PIX_FMT_RGBAF32BE as i32 => PixelFormat::Rgbaf32Be,
253    // Planar GBR.
254    x if x == AVPixelFormat::AV_PIX_FMT_GBRP as i32 => PixelFormat::Gbrp,
255    x if x == AVPixelFormat::AV_PIX_FMT_GBRP9LE as i32 => PixelFormat::Gbrp9Le,
256    x if x == AVPixelFormat::AV_PIX_FMT_GBRP9BE as i32 => PixelFormat::Gbrp9Be,
257    x if x == AVPixelFormat::AV_PIX_FMT_GBRP10LE as i32 => PixelFormat::Gbrp10Le,
258    x if x == AVPixelFormat::AV_PIX_FMT_GBRP10BE as i32 => PixelFormat::Gbrp10Be,
259    x if x == AVPixelFormat::AV_PIX_FMT_GBRP10MSBLE as i32 => PixelFormat::Gbrp10MsbLe,
260    x if x == AVPixelFormat::AV_PIX_FMT_GBRP10MSBBE as i32 => PixelFormat::Gbrp10MsbBe,
261    x if x == AVPixelFormat::AV_PIX_FMT_GBRP12LE as i32 => PixelFormat::Gbrp12Le,
262    x if x == AVPixelFormat::AV_PIX_FMT_GBRP12BE as i32 => PixelFormat::Gbrp12Be,
263    x if x == AVPixelFormat::AV_PIX_FMT_GBRP12MSBLE as i32 => PixelFormat::Gbrp12MsbLe,
264    x if x == AVPixelFormat::AV_PIX_FMT_GBRP12MSBBE as i32 => PixelFormat::Gbrp12MsbBe,
265    x if x == AVPixelFormat::AV_PIX_FMT_GBRP14LE as i32 => PixelFormat::Gbrp14Le,
266    x if x == AVPixelFormat::AV_PIX_FMT_GBRP14BE as i32 => PixelFormat::Gbrp14Be,
267    x if x == AVPixelFormat::AV_PIX_FMT_GBRP16LE as i32 => PixelFormat::Gbrp16Le,
268    x if x == AVPixelFormat::AV_PIX_FMT_GBRP16BE as i32 => PixelFormat::Gbrp16Be,
269    x if x == AVPixelFormat::AV_PIX_FMT_GBRPF16LE as i32 => PixelFormat::Gbrpf16Le,
270    x if x == AVPixelFormat::AV_PIX_FMT_GBRPF16BE as i32 => PixelFormat::Gbrpf16Be,
271    x if x == AVPixelFormat::AV_PIX_FMT_GBRPF32LE as i32 => PixelFormat::Gbrpf32Le,
272    x if x == AVPixelFormat::AV_PIX_FMT_GBRPF32BE as i32 => PixelFormat::Gbrpf32Be,
273    // Planar GBRA.
274    x if x == AVPixelFormat::AV_PIX_FMT_GBRAP as i32 => PixelFormat::Gbrap,
275    x if x == AVPixelFormat::AV_PIX_FMT_GBRAP10LE as i32 => PixelFormat::Gbrap10Le,
276    x if x == AVPixelFormat::AV_PIX_FMT_GBRAP10BE as i32 => PixelFormat::Gbrap10Be,
277    x if x == AVPixelFormat::AV_PIX_FMT_GBRAP12LE as i32 => PixelFormat::Gbrap12Le,
278    x if x == AVPixelFormat::AV_PIX_FMT_GBRAP12BE as i32 => PixelFormat::Gbrap12Be,
279    x if x == AVPixelFormat::AV_PIX_FMT_GBRAP14LE as i32 => PixelFormat::Gbrap14Le,
280    x if x == AVPixelFormat::AV_PIX_FMT_GBRAP14BE as i32 => PixelFormat::Gbrap14Be,
281    x if x == AVPixelFormat::AV_PIX_FMT_GBRAP16LE as i32 => PixelFormat::Gbrap16Le,
282    x if x == AVPixelFormat::AV_PIX_FMT_GBRAP16BE as i32 => PixelFormat::Gbrap16Be,
283    x if x == AVPixelFormat::AV_PIX_FMT_GBRAP32LE as i32 => PixelFormat::Gbrap32Le,
284    x if x == AVPixelFormat::AV_PIX_FMT_GBRAP32BE as i32 => PixelFormat::Gbrap32Be,
285    x if x == AVPixelFormat::AV_PIX_FMT_GBRAPF16LE as i32 => PixelFormat::Gbrapf16Le,
286    x if x == AVPixelFormat::AV_PIX_FMT_GBRAPF16BE as i32 => PixelFormat::Gbrapf16Be,
287    x if x == AVPixelFormat::AV_PIX_FMT_GBRAPF32LE as i32 => PixelFormat::Gbrapf32Le,
288    x if x == AVPixelFormat::AV_PIX_FMT_GBRAPF32BE as i32 => PixelFormat::Gbrapf32Be,
289    // Greyscale.
290    x if x == AVPixelFormat::AV_PIX_FMT_GRAY8 as i32 => PixelFormat::Gray8,
291    x if x == AVPixelFormat::AV_PIX_FMT_GRAY9LE as i32 => PixelFormat::Gray9Le,
292    x if x == AVPixelFormat::AV_PIX_FMT_GRAY9BE as i32 => PixelFormat::Gray9Be,
293    x if x == AVPixelFormat::AV_PIX_FMT_GRAY10LE as i32 => PixelFormat::Gray10Le,
294    x if x == AVPixelFormat::AV_PIX_FMT_GRAY10BE as i32 => PixelFormat::Gray10Be,
295    x if x == AVPixelFormat::AV_PIX_FMT_GRAY12LE as i32 => PixelFormat::Gray12Le,
296    x if x == AVPixelFormat::AV_PIX_FMT_GRAY12BE as i32 => PixelFormat::Gray12Be,
297    x if x == AVPixelFormat::AV_PIX_FMT_GRAY14LE as i32 => PixelFormat::Gray14Le,
298    x if x == AVPixelFormat::AV_PIX_FMT_GRAY14BE as i32 => PixelFormat::Gray14Be,
299    x if x == AVPixelFormat::AV_PIX_FMT_GRAY16LE as i32 => PixelFormat::Gray16Le,
300    x if x == AVPixelFormat::AV_PIX_FMT_GRAY16BE as i32 => PixelFormat::Gray16Be,
301    x if x == AVPixelFormat::AV_PIX_FMT_GRAY32LE as i32 => PixelFormat::Gray32Le,
302    x if x == AVPixelFormat::AV_PIX_FMT_GRAY32BE as i32 => PixelFormat::Gray32Be,
303    x if x == AVPixelFormat::AV_PIX_FMT_GRAYF16LE as i32 => PixelFormat::Grayf16Le,
304    x if x == AVPixelFormat::AV_PIX_FMT_GRAYF16BE as i32 => PixelFormat::Grayf16Be,
305    x if x == AVPixelFormat::AV_PIX_FMT_GRAYF32LE as i32 => PixelFormat::Grayf32Le,
306    x if x == AVPixelFormat::AV_PIX_FMT_GRAYF32BE as i32 => PixelFormat::Grayf32Be,
307    x if x == AVPixelFormat::AV_PIX_FMT_YA8 as i32 => PixelFormat::Ya8,
308    x if x == AVPixelFormat::AV_PIX_FMT_YA16LE as i32 => PixelFormat::Ya16Le,
309    x if x == AVPixelFormat::AV_PIX_FMT_YA16BE as i32 => PixelFormat::Ya16Be,
310    x if x == AVPixelFormat::AV_PIX_FMT_YAF16LE as i32 => PixelFormat::Yaf16Le,
311    x if x == AVPixelFormat::AV_PIX_FMT_YAF16BE as i32 => PixelFormat::Yaf16Be,
312    x if x == AVPixelFormat::AV_PIX_FMT_YAF32LE as i32 => PixelFormat::Yaf32Le,
313    x if x == AVPixelFormat::AV_PIX_FMT_YAF32BE as i32 => PixelFormat::Yaf32Be,
314    x if x == AVPixelFormat::AV_PIX_FMT_MONOWHITE as i32 => PixelFormat::Monowhite,
315    x if x == AVPixelFormat::AV_PIX_FMT_MONOBLACK as i32 => PixelFormat::Monoblack,
316    x if x == AVPixelFormat::AV_PIX_FMT_PAL8 as i32 => PixelFormat::Pal8,
317    x if x == AVPixelFormat::AV_PIX_FMT_RGB4 as i32 => PixelFormat::Rgb4,
318    x if x == AVPixelFormat::AV_PIX_FMT_RGB4_BYTE as i32 => PixelFormat::Rgb4Byte,
319    x if x == AVPixelFormat::AV_PIX_FMT_RGB8 as i32 => PixelFormat::Rgb8,
320    x if x == AVPixelFormat::AV_PIX_FMT_BGR4 as i32 => PixelFormat::Bgr4,
321    x if x == AVPixelFormat::AV_PIX_FMT_BGR4_BYTE as i32 => PixelFormat::Bgr4Byte,
322    x if x == AVPixelFormat::AV_PIX_FMT_BGR8 as i32 => PixelFormat::Bgr8,
323    x if x == AVPixelFormat::AV_PIX_FMT_RGB444LE as i32 => PixelFormat::Rgb444Le,
324    x if x == AVPixelFormat::AV_PIX_FMT_RGB444BE as i32 => PixelFormat::Rgb444Be,
325    x if x == AVPixelFormat::AV_PIX_FMT_BGR444LE as i32 => PixelFormat::Bgr444Le,
326    x if x == AVPixelFormat::AV_PIX_FMT_BGR444BE as i32 => PixelFormat::Bgr444Be,
327    x if x == AVPixelFormat::AV_PIX_FMT_RGB555LE as i32 => PixelFormat::Rgb555Le,
328    x if x == AVPixelFormat::AV_PIX_FMT_RGB555BE as i32 => PixelFormat::Rgb555Be,
329    x if x == AVPixelFormat::AV_PIX_FMT_BGR555LE as i32 => PixelFormat::Bgr555Le,
330    x if x == AVPixelFormat::AV_PIX_FMT_BGR555BE as i32 => PixelFormat::Bgr555Be,
331    x if x == AVPixelFormat::AV_PIX_FMT_RGB565LE as i32 => PixelFormat::Rgb565Le,
332    x if x == AVPixelFormat::AV_PIX_FMT_RGB565BE as i32 => PixelFormat::Rgb565Be,
333    x if x == AVPixelFormat::AV_PIX_FMT_BGR565LE as i32 => PixelFormat::Bgr565Le,
334    x if x == AVPixelFormat::AV_PIX_FMT_BGR565BE as i32 => PixelFormat::Bgr565Be,
335    x if x == AVPixelFormat::AV_PIX_FMT_BAYER_BGGR8 as i32 => PixelFormat::BayerBggr8,
336    x if x == AVPixelFormat::AV_PIX_FMT_BAYER_RGGB8 as i32 => PixelFormat::BayerRggb8,
337    x if x == AVPixelFormat::AV_PIX_FMT_BAYER_GBRG8 as i32 => PixelFormat::BayerGbrg8,
338    x if x == AVPixelFormat::AV_PIX_FMT_BAYER_GRBG8 as i32 => PixelFormat::BayerGrbg8,
339    x if x == AVPixelFormat::AV_PIX_FMT_BAYER_BGGR16LE as i32 => PixelFormat::BayerBggr16Le,
340    x if x == AVPixelFormat::AV_PIX_FMT_BAYER_BGGR16BE as i32 => PixelFormat::BayerBggr16Be,
341    x if x == AVPixelFormat::AV_PIX_FMT_BAYER_RGGB16LE as i32 => PixelFormat::BayerRggb16Le,
342    x if x == AVPixelFormat::AV_PIX_FMT_BAYER_RGGB16BE as i32 => PixelFormat::BayerRggb16Be,
343    x if x == AVPixelFormat::AV_PIX_FMT_BAYER_GBRG16LE as i32 => PixelFormat::BayerGbrg16Le,
344    x if x == AVPixelFormat::AV_PIX_FMT_BAYER_GBRG16BE as i32 => PixelFormat::BayerGbrg16Be,
345    x if x == AVPixelFormat::AV_PIX_FMT_BAYER_GRBG16LE as i32 => PixelFormat::BayerGrbg16Le,
346    x if x == AVPixelFormat::AV_PIX_FMT_BAYER_GRBG16BE as i32 => PixelFormat::BayerGrbg16Be,
347    _ => PixelFormat::None,
348  }
349}
350
351/// Returns `true` when `raw` is one of FFmpeg's hardware-frame markers
352/// (`AV_PIX_FMT_VIDEOTOOLBOX` / `_VAAPI` / `_CUDA` / `_D3D11` /
353/// `_DRM_PRIME` / `_MEDIACODEC` / `_VULKAN`). Used by the HW probe to
354/// identify GPU-resident frames before triggering
355/// `av_hwframe_transfer_data`.
356pub const fn is_hardware_pix_fmt(raw: i32) -> bool {
357  matches!(
358    raw,
359    x if x == AVPixelFormat::AV_PIX_FMT_VIDEOTOOLBOX as i32
360      || x == AVPixelFormat::AV_PIX_FMT_VAAPI as i32
361      || x == AVPixelFormat::AV_PIX_FMT_CUDA as i32
362      || x == AVPixelFormat::AV_PIX_FMT_D3D11 as i32
363      || x == AVPixelFormat::AV_PIX_FMT_DRM_PRIME as i32
364      || x == AVPixelFormat::AV_PIX_FMT_MEDIACODEC as i32
365      || x == AVPixelFormat::AV_PIX_FMT_VULKAN as i32
366  )
367}
368
369/// Fallible counterpart to ffmpeg-next's `Packet::copy`.
370///
371/// The upstream helper calls `Packet::new(size)` (which silently
372/// truncates `size` to `c_int` and ignores `av_new_packet`'s return
373/// code) and then panics via `data_mut().unwrap().write_all(...).unwrap()`
374/// if the allocation failed. From a safe public decoder API we want
375/// the OOM / oversized-payload paths to surface as
376/// `ffmpeg_next::Error` rather than aborting the process — every
377/// `send_packet` path goes through this helper.
378///
379/// Failure modes:
380/// * payload larger than `c_int::MAX` (would overflow `AVPacket.size`)
381///   → `ffmpeg_next::Error::Other { errno: libc::EINVAL }`.
382/// * `av_new_packet` allocation failure (signalled by `data_mut()`
383///   returning `None`) → `ffmpeg_next::Error::Other { errno:
384///   libc::ENOMEM }`.
385fn try_packet_copy(data: &[u8]) -> std::result::Result<Packet, ffmpeg_next::Error> {
386  // FFmpeg's `AVPacket.size` is `c_int`. A payload larger than that
387  // can't fit in a single packet — refuse rather than truncate via
388  // `as c_int` inside `Packet::new`.
389  if data.len() > c_int::MAX as usize {
390    return Err(ffmpeg_next::Error::Other {
391      errno: libc::EINVAL,
392    });
393  }
394  // `Packet::new(size)` calls `av_new_packet(&mut pkt, size as
395  // c_int)` and ignores the return code; on OOM it returns a
396  // `Packet` whose `.data` is null. We detect that via
397  // `data_mut()` (returns `None` on null) and copy via
398  // `copy_nonoverlapping` so we never go through `data_mut()
399  // .unwrap().write_all().unwrap()` — the upstream `Packet::copy`'s
400  // double panic.
401  let mut pkt = Packet::new(data.len());
402  match pkt.data_mut() {
403    Some(slot) if slot.len() == data.len() => {
404      // SAFETY: `slot` is a `&mut [u8]` of `data.len()` bytes;
405      // `data` is a `&[u8]` of the same length. Non-overlapping
406      // because `slot` is a fresh allocation.
407      if !data.is_empty() {
408        unsafe {
409          core::ptr::copy_nonoverlapping(data.as_ptr(), slot.as_mut_ptr(), data.len());
410        }
411      }
412      Ok(pkt)
413    }
414    _ => Err(ffmpeg_next::Error::Other {
415      errno: libc::ENOMEM,
416    }),
417  }
418}
419
420/// Centralised mediadecode→AV packet flag mapping so the three
421/// packet-conversion helpers stay aligned.
422fn map_md_flags_to_av(flags: MdPacketFlags) -> ffmpeg_next::packet::Flags {
423  let mut av_flags = ffmpeg_next::packet::Flags::empty();
424  if flags.contains(MdPacketFlags::KEY) {
425    av_flags |= ffmpeg_next::packet::Flags::KEY;
426  }
427  if flags.contains(MdPacketFlags::CORRUPT) {
428    av_flags |= ffmpeg_next::packet::Flags::CORRUPT;
429  }
430  // ffmpeg-next 8.x doesn't expose a DISCARD flag constant on
431  // `packet::Flags`; the upstream `AV_PKT_FLAG_DISCARD` bit is
432  // documented as a demuxer hint and rarely set on packets passed
433  // to a decoder. We forward KEY and CORRUPT (the meaningful subset)
434  // and silently drop DISCARD until ffmpeg-next adds it.
435  av_flags
436}
437
438/// Builds an `ffmpeg::Packet` from a [`mediadecode::VideoPacket`]
439/// parameterized by [`crate::extras::VideoPacketExtra`] and
440/// [`crate::FfmpegBuffer`].
441///
442/// The compressed bytes are **copied** into a new packet allocation —
443/// zero-copy passthrough of the FfmpegBuffer's underlying AVBufferRef
444/// is a future optimization (would need to wire an `AVBufferRef` into
445/// `AVPacket.buf` directly via `av_packet_alloc` + manual buffer set).
446/// PTS / DTS / duration / flags / stream_index are propagated.
447///
448/// Returns `Err(ffmpeg_next::Error)` on:
449/// * payload larger than `c_int::MAX` (would overflow `AVPacket.size`);
450/// * `av_new_packet` allocation failure (OOM).
451pub fn ffmpeg_packet_from_video_packet(
452  packet: &mediadecode::packet::VideoPacket<VideoPacketExtra, FfmpegBuffer>,
453) -> std::result::Result<Packet, ffmpeg_next::Error> {
454  let mut out = try_packet_copy(packet.data().as_ref())?;
455  if let Some(ts) = packet.pts() {
456    out.set_pts(Some(ts.pts()));
457  }
458  if let Some(ts) = packet.dts() {
459    out.set_dts(Some(ts.pts()));
460  }
461  if let Some(d) = packet.duration() {
462    out.set_duration(d.pts());
463  }
464  out.set_flags(map_md_flags_to_av(packet.flags()));
465  out.set_stream(packet.extra().stream_index() as usize);
466  Ok(out)
467}
468
469/// Builds an `ffmpeg::Packet` from a [`mediadecode::AudioPacket`].
470/// Same shape as [`ffmpeg_packet_from_video_packet`] — bytes are
471/// copied; pts/dts/duration/flags/stream_index are forwarded. Same
472/// failure modes.
473pub fn ffmpeg_packet_from_audio_packet(
474  packet: &mediadecode::packet::AudioPacket<AudioPacketExtra, FfmpegBuffer>,
475) -> std::result::Result<Packet, ffmpeg_next::Error> {
476  let mut out = try_packet_copy(packet.data().as_ref())?;
477  if let Some(ts) = packet.pts() {
478    out.set_pts(Some(ts.pts()));
479  }
480  if let Some(ts) = packet.dts() {
481    out.set_dts(Some(ts.pts()));
482  }
483  if let Some(d) = packet.duration() {
484    out.set_duration(d.pts());
485  }
486  out.set_flags(map_md_flags_to_av(packet.flags()));
487  out.set_stream(packet.extra().stream_index() as usize);
488  Ok(out)
489}
490
491/// Builds an `ffmpeg::Packet` from a [`mediadecode::SubtitlePacket`].
492/// Bytes copied; pts/duration/flags/stream_index forwarded. Subtitle
493/// packets have no `dts` in the mediadecode model. Same failure
494/// modes as [`ffmpeg_packet_from_video_packet`].
495pub fn ffmpeg_packet_from_subtitle_packet(
496  packet: &mediadecode::packet::SubtitlePacket<SubtitlePacketExtra, FfmpegBuffer>,
497) -> std::result::Result<Packet, ffmpeg_next::Error> {
498  let mut out = try_packet_copy(packet.data().as_ref())?;
499  if let Some(ts) = packet.pts() {
500    out.set_pts(Some(ts.pts()));
501  }
502  if let Some(d) = packet.duration() {
503    out.set_duration(d.pts());
504  }
505  out.set_flags(map_md_flags_to_av(packet.flags()));
506  out.set_stream(packet.extra().stream_index() as usize);
507  Ok(out)
508}
509
510// ---------------------------------------------------------------------------
511//  Safe wrappers — `&ffmpeg::Packet` → `mediadecode::*Packet`.
512// ---------------------------------------------------------------------------
513
514/// Wraps a borrowed [`ffmpeg::Packet`] as a
515/// [`mediadecode::packet::VideoPacket`]. The compressed payload is
516/// shared with the source `AVPacket` via refcount bump (no copy).
517/// Timestamps, duration, key/corrupt flags, and the source stream
518/// index are forwarded to the produced packet.
519///
520/// Returns `None` when the source packet has no buffer attached
521/// (empty packet — typical after EOF). Caller can also fill in
522/// [`VideoPacketExtra::byte_pos`] / `side_data` post-construction
523/// if they need those.
524pub fn video_packet_from_ffmpeg(
525  packet: &Packet,
526) -> Option<VideoPacket<VideoPacketExtra, FfmpegBuffer>> {
527  let buf = FfmpegBuffer::from_packet(packet)?;
528  let mut out = VideoPacket::new(buf, VideoPacketExtra::new(packet.stream() as i32))
529    .with_flags(md_flags_from_av(packet.flags()));
530  if let Some(p) = packet.pts() {
531    out = out.with_pts(Some(Timestamp::new(p, mediadecode::Timebase::default())));
532  }
533  if let Some(d) = packet.dts() {
534    out = out.with_dts(Some(Timestamp::new(d, mediadecode::Timebase::default())));
535  }
536  let dur = packet.duration();
537  if dur > 0 {
538    out = out.with_duration(Some(Timestamp::new(dur, mediadecode::Timebase::default())));
539  }
540  Some(out)
541}
542
543/// Wraps a borrowed [`ffmpeg::Packet`] as a
544/// [`mediadecode::packet::AudioPacket`]. Same shape as
545/// [`video_packet_from_ffmpeg`] — refcounted payload, forwarded
546/// metadata.
547pub fn audio_packet_from_ffmpeg(
548  packet: &Packet,
549) -> Option<AudioPacket<AudioPacketExtra, FfmpegBuffer>> {
550  let buf = FfmpegBuffer::from_packet(packet)?;
551  let mut out = AudioPacket::new(buf, AudioPacketExtra::new(packet.stream() as i32))
552    .with_flags(md_flags_from_av(packet.flags()));
553  if let Some(p) = packet.pts() {
554    out = out.with_pts(Some(Timestamp::new(p, mediadecode::Timebase::default())));
555  }
556  if let Some(d) = packet.dts() {
557    out = out.with_dts(Some(Timestamp::new(d, mediadecode::Timebase::default())));
558  }
559  let dur = packet.duration();
560  if dur > 0 {
561    out = out.with_duration(Some(Timestamp::new(dur, mediadecode::Timebase::default())));
562  }
563  Some(out)
564}
565
566/// Wraps a borrowed [`ffmpeg::Packet`] as a
567/// [`mediadecode::packet::SubtitlePacket`]. Subtitle packets have no
568/// `dts` in the mediadecode model; everything else mirrors
569/// [`video_packet_from_ffmpeg`].
570pub fn subtitle_packet_from_ffmpeg(
571  packet: &Packet,
572) -> Option<SubtitlePacket<SubtitlePacketExtra, FfmpegBuffer>> {
573  let buf = FfmpegBuffer::from_packet(packet)?;
574  let mut out = SubtitlePacket::new(buf, SubtitlePacketExtra::new(packet.stream() as i32))
575    .with_flags(md_flags_from_av(packet.flags()));
576  if let Some(p) = packet.pts() {
577    out = out.with_pts(Some(Timestamp::new(p, mediadecode::Timebase::default())));
578  }
579  let dur = packet.duration();
580  if dur > 0 {
581    out = out.with_duration(Some(Timestamp::new(dur, mediadecode::Timebase::default())));
582  }
583  Some(out)
584}
585
586fn md_flags_from_av(flags: ffmpeg_next::packet::Flags) -> MdPacketFlags {
587  let mut out = MdPacketFlags::empty();
588  if flags.contains(ffmpeg_next::packet::Flags::KEY) {
589    out |= MdPacketFlags::KEY;
590  }
591  if flags.contains(ffmpeg_next::packet::Flags::CORRUPT) {
592    out |= MdPacketFlags::CORRUPT;
593  }
594  out
595}
596
597// ---------------------------------------------------------------------------
598//  Empty-frame placeholders for `receive_frame` destinations.
599// ---------------------------------------------------------------------------
600
601/// Constructs an empty [`mediadecode::frame::VideoFrame`] suitable as
602/// the destination argument to
603/// [`mediadecode::decoder::VideoStreamDecoder::receive_frame`]. The
604/// decoder overwrites the frame on success; this just provides a
605/// well-formed slot.
606///
607/// All four plane slots get a 1-byte `FfmpegBuffer` placeholder
608/// (the array shape requires a buffer in every slot, but
609/// `plane_count = 0` reports them as inactive).
610///
611/// # Panics
612///
613/// Panics on FFmpeg-side OOM (the per-plane 1-byte allocation
614/// failed). Callers who need to recover from OOM should use
615/// [`try_empty_video_frame`].
616pub fn empty_video_frame() -> VideoFrame<PixelFormat, VideoFrameExtra, FfmpegBuffer> {
617  try_empty_video_frame().expect("empty_video_frame: av_buffer_alloc returned null (OOM)")
618}
619
620/// Fallible counterpart to [`empty_video_frame`]. Returns `None` if
621/// any of the four placeholder allocations fails.
622pub fn try_empty_video_frame() -> Option<VideoFrame<PixelFormat, VideoFrameExtra, FfmpegBuffer>> {
623  let planes = [
624    Plane::new(FfmpegBuffer::try_empty()?, 0),
625    Plane::new(FfmpegBuffer::try_empty()?, 0),
626    Plane::new(FfmpegBuffer::try_empty()?, 0),
627    Plane::new(FfmpegBuffer::try_empty()?, 0),
628  ];
629  Some(VideoFrame::new(
630    Dimensions::new(0, 0),
631    // mediaframe 0.3's named "no format yet" member, and its
632    // `Default` — the state a descriptor is in before a decoder has
633    // said what it produces, which is exactly this placeholder.
634    PixelFormat::None,
635    planes,
636    0,
637    VideoFrameExtra::default(),
638  ))
639}
640
641/// Constructs an empty [`mediadecode::frame::AudioFrame`] suitable as
642/// the destination argument to
643/// [`mediadecode::decoder::AudioStreamDecoder::receive_frame`]. Same
644/// behaviour as [`empty_video_frame`] — eight 1-byte plane
645/// placeholders, `plane_count = 0`.
646///
647/// # Panics
648///
649/// Panics on FFmpeg-side OOM. See [`try_empty_audio_frame`] for the
650/// fallible variant.
651pub fn empty_audio_frame()
652-> AudioFrame<SampleFormat, AudioChannelLayout, AudioFrameExtra, FfmpegBuffer> {
653  try_empty_audio_frame().expect("empty_audio_frame: av_buffer_alloc returned null (OOM)")
654}
655
656/// Fallible counterpart to [`empty_audio_frame`]. Returns `None` if
657/// any of the eight placeholder allocations fails.
658pub fn try_empty_audio_frame()
659-> Option<AudioFrame<SampleFormat, AudioChannelLayout, AudioFrameExtra, FfmpegBuffer>> {
660  let planes = [
661    Plane::new(FfmpegBuffer::try_empty()?, 0),
662    Plane::new(FfmpegBuffer::try_empty()?, 0),
663    Plane::new(FfmpegBuffer::try_empty()?, 0),
664    Plane::new(FfmpegBuffer::try_empty()?, 0),
665    Plane::new(FfmpegBuffer::try_empty()?, 0),
666    Plane::new(FfmpegBuffer::try_empty()?, 0),
667    Plane::new(FfmpegBuffer::try_empty()?, 0),
668    Plane::new(FfmpegBuffer::try_empty()?, 0),
669  ];
670  Some(AudioFrame::new(
671    0,
672    0,
673    0,
674    SampleFormat::NONE,
675    AudioChannelLayout::default(),
676    planes,
677    0,
678    AudioFrameExtra::default(),
679  ))
680}
681
682/// Constructs an empty [`mediadecode::frame::SubtitleFrame`] suitable
683/// as the destination argument to
684/// [`mediadecode::decoder::SubtitleDecoder::receive_frame`]. The
685/// payload is an empty `Text` placeholder; the decoder overwrites
686/// it on success.
687///
688/// # Panics
689///
690/// Panics on FFmpeg-side OOM. See [`try_empty_subtitle_frame`] for
691/// the fallible variant.
692pub fn empty_subtitle_frame() -> SubtitleFrame<SubtitleFrameExtra, FfmpegBuffer> {
693  try_empty_subtitle_frame().expect("empty_subtitle_frame: av_buffer_alloc returned null (OOM)")
694}
695
696/// Fallible counterpart to [`empty_subtitle_frame`]. Returns `None`
697/// if the placeholder allocation fails.
698pub fn try_empty_subtitle_frame() -> Option<SubtitleFrame<SubtitleFrameExtra, FfmpegBuffer>> {
699  let buf = FfmpegBuffer::copy_from_slice(&[]).or_else(FfmpegBuffer::try_empty)?;
700  Some(SubtitleFrame::new(
701    SubtitlePayload::Text {
702      text: buf,
703      language: None,
704    },
705    SubtitleFrameExtra::default(),
706  ))
707}
708
709#[cfg(test)]
710mod tests {
711  use super::*;
712
713  #[test]
714  fn nv12_round_trips() {
715    assert_eq!(
716      from_av_pixel_format(AVPixelFormat::AV_PIX_FMT_NV12 as i32),
717      PixelFormat::Nv12,
718    );
719  }
720
721  #[test]
722  fn p010be_maps_to_p010be() {
723    // BE must map to the BE variant — the previous "fold to LE"
724    // mapping silently corrupted P010BE pixel data via the safe
725    // export path. The unsupported-format gate in `convert::av_frame_to_video_frame`
726    // is the right place to reject BE today.
727    assert_eq!(
728      from_av_pixel_format(AVPixelFormat::AV_PIX_FMT_P010BE as i32),
729      PixelFormat::P010Be,
730    );
731  }
732
733  #[test]
734  fn unnamed_raw_maps_to_none() {
735    assert_eq!(from_av_pixel_format(-99_999), PixelFormat::None);
736  }
737
738  #[test]
739  fn av_pix_fmt_none_maps_to_none() {
740    assert_eq!(
741      from_av_pixel_format(AVPixelFormat::AV_PIX_FMT_NONE as i32),
742      PixelFormat::None,
743    );
744  }
745
746  #[test]
747  fn hw_formats_detected() {
748    assert!(is_hardware_pix_fmt(
749      AVPixelFormat::AV_PIX_FMT_VIDEOTOOLBOX as i32
750    ));
751    assert!(is_hardware_pix_fmt(AVPixelFormat::AV_PIX_FMT_VAAPI as i32));
752    assert!(is_hardware_pix_fmt(AVPixelFormat::AV_PIX_FMT_CUDA as i32));
753    assert!(is_hardware_pix_fmt(AVPixelFormat::AV_PIX_FMT_D3D11 as i32));
754  }
755
756  #[test]
757  fn cpu_formats_not_detected_as_hw() {
758    assert!(!is_hardware_pix_fmt(AVPixelFormat::AV_PIX_FMT_NV12 as i32));
759    assert!(!is_hardware_pix_fmt(
760      AVPixelFormat::AV_PIX_FMT_YUV420P as i32
761    ));
762    assert!(!is_hardware_pix_fmt(AVPixelFormat::AV_PIX_FMT_NONE as i32));
763  }
764
765  #[test]
766  fn hw_formats_map_to_none_in_pixel_format() {
767    // HW sentinels intentionally don't have a mediadecode::PixelFormat
768    // representation — they're not CPU pixel data.
769    assert_eq!(
770      from_av_pixel_format(AVPixelFormat::AV_PIX_FMT_VIDEOTOOLBOX as i32),
771      PixelFormat::None,
772    );
773    assert_eq!(
774      from_av_pixel_format(AVPixelFormat::AV_PIX_FMT_VAAPI as i32),
775      PixelFormat::None,
776    );
777  }
778}
mediadecode_ffmpeg/boundary.rs

mediadecode_ffmpeg/
boundary.rs
