codec/subtitle: refactor Rect handling

src/codec/subtitle/rect.rs

@@ -0,0 +1,150 @@
+use std::marker::PhantomData;
+use std::ffi::CStr;
+use std::str::from_utf8_unchecked;
+
+use ffi::*;
+use super::{Type, Flags};
+use ::{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((*ptr).kind) {
+			Type::None   => Rect::None(ptr),
+			Type::Bitmap => Rect::Bitmap(Bitmap::wrap(ptr)),
+			Type::Text   => Rect::Text(Text::wrap(ptr)),
+			Type::Ass    => Rect::Ass(Ass::wrap(ptr))
+		}
+	}
+
+	pub unsafe fn as_ptr(&self) -> *const AVSubtitleRect {
+		match self {
+			&Rect::None(ptr)     => ptr,
+			&Rect::Bitmap(ref b) => b.as_ptr(),
+			&Rect::Text(ref t)   => t.as_ptr(),
+			&Rect::Ass(ref a)    => 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: 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
+		}
+	}
+
+	// XXX: must split Picture and PictureMut
+	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: 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: 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())
+		}
+	}
+}