use std::{ffi::CStr, marker::PhantomData, str::from_utf8_unchecked};
use super::{Flags, Type};
use crate::ffi::*;
#[cfg(not(feature = "ffmpeg_5_0"))]
use {crate::format, Picture};
pub enum Rect<'a> {
None(*const AVSubtitleRect),
Bitmap(Bitmap<'a>),
Text(Text<'a>),
Ass(Ass<'a>),
}
impl<'a> Rect<'a> {
pub unsafe fn wrap(ptr: *const AVSubtitleRect) -> Self {
match Type::from(unsafe { (*ptr).type_ }) {
Type::None => Rect::None(ptr),
Type::Bitmap => Rect::Bitmap(unsafe { Bitmap::wrap(ptr) }),
Type::Text => Rect::Text(unsafe { Text::wrap(ptr) }),
Type::Ass => Rect::Ass(unsafe { Ass::wrap(ptr) }),
}
}
pub unsafe fn as_ptr(&self) -> *const AVSubtitleRect {
match *self {
Rect::None(ptr) => ptr,
Rect::Bitmap(ref b) => unsafe { b.as_ptr() },
Rect::Text(ref t) => unsafe { t.as_ptr() },
Rect::Ass(ref a) => unsafe { a.as_ptr() },
}
}
}
impl<'a> Rect<'a> {
pub fn flags(&self) -> Flags {
unsafe {
Flags::from_bits_truncate(match *self {
Rect::None(ptr) => (*ptr).flags,
Rect::Bitmap(ref b) => (*b.as_ptr()).flags,
Rect::Text(ref t) => (*t.as_ptr()).flags,
Rect::Ass(ref a) => (*a.as_ptr()).flags,
})
}
}
}
pub struct Bitmap<'a> {
ptr: *const AVSubtitleRect,
_marker: PhantomData<&'a ()>,
}
impl<'a> Bitmap<'a> {
pub unsafe fn wrap(ptr: *const AVSubtitleRect) -> Self {
Bitmap { ptr, _marker: PhantomData }
}
pub unsafe fn as_ptr(&self) -> *const AVSubtitleRect {
self.ptr
}
}
impl<'a> Bitmap<'a> {
pub fn x(&self) -> usize {
unsafe { (*self.as_ptr()).x as usize }
}
pub fn y(&self) -> usize {
unsafe { (*self.as_ptr()).y as usize }
}
pub fn width(&self) -> u32 {
unsafe { (*self.as_ptr()).w as u32 }
}
pub fn height(&self) -> u32 {
unsafe { (*self.as_ptr()).h as u32 }
}
pub fn colors(&self) -> usize {
unsafe { (*self.as_ptr()).nb_colors as usize }
}
#[cfg(not(feature = "ffmpeg_5_0"))]
pub fn picture(&self, format: format::Pixel) -> Picture<'a> {
unsafe { Picture::wrap(&(*self.as_ptr()).pict as *const _ as *mut _, format, (*self.as_ptr()).w as u32, (*self.as_ptr()).h as u32) }
}
}
pub struct Text<'a> {
ptr: *const AVSubtitleRect,
_marker: PhantomData<&'a ()>,
}
impl<'a> Text<'a> {
pub unsafe fn wrap(ptr: *const AVSubtitleRect) -> Self {
Text { ptr, _marker: PhantomData }
}
pub unsafe fn as_ptr(&self) -> *const AVSubtitleRect {
self.ptr
}
}
impl<'a> Text<'a> {
pub fn get(&self) -> &str {
unsafe { from_utf8_unchecked(CStr::from_ptr((*self.as_ptr()).text).to_bytes()) }
}
}
pub struct Ass<'a> {
ptr: *const AVSubtitleRect,
_marker: PhantomData<&'a ()>,
}
impl<'a> Ass<'a> {
pub unsafe fn wrap(ptr: *const AVSubtitleRect) -> Self {
Ass { ptr, _marker: PhantomData }
}
pub unsafe fn as_ptr(&self) -> *const AVSubtitleRect {
self.ptr
}
}
impl<'a> Ass<'a> {
pub fn get(&self) -> &str {
unsafe { from_utf8_unchecked(CStr::from_ptr((*self.as_ptr()).ass).to_bytes()) }
}
}