util/frame/audio: improve ergonomics

This commit is contained in:
meh 2015-06-08 15:51:41 +02:00
parent 065337925f
commit 02ea528c26
4 changed files with 95 additions and 167 deletions

View File

@ -15,7 +15,6 @@ pub use util::color;
pub use util::chroma; pub use util::chroma;
pub use util::time; pub use util::time;
pub use util::frame::{self, Frame}; pub use util::frame::{self, Frame};
pub use util::samples::Samples;
pub use util::channel_layout::{self, ChannelLayout}; pub use util::channel_layout::{self, ChannelLayout};
#[cfg(feature = "format")] #[cfg(feature = "format")]

View File

@ -1,11 +1,13 @@
use libc::c_int;
use std::mem; use std::mem;
use std::slice;
use std::ops::{Deref, DerefMut}; use std::ops::{Deref, DerefMut};
use std::any::TypeId;
use std::marker::Reflect;
use libc::{int64_t, c_ulonglong}; use libc::{c_int, int64_t, c_ulonglong};
use ffi::*; use ffi::*;
use ::{ChannelLayout, Samples}; use ::ChannelLayout;
use ::util::format; use ::util::format::Sample;
use super::Frame; use super::Frame;
#[derive(PartialEq, Eq)] #[derive(PartialEq, Eq)]
@ -15,6 +17,14 @@ impl Audio {
pub unsafe fn wrap(ptr: *mut AVFrame) -> Self { pub unsafe fn wrap(ptr: *mut AVFrame) -> Self {
Audio(Frame::wrap(ptr)) Audio(Frame::wrap(ptr))
} }
pub unsafe fn alloc(&mut self, format: Sample, samples: usize, layout: ChannelLayout) {
self.set_format(format);
self.set_samples(samples);
self.set_channel_layout(layout);
av_frame_get_buffer(self.as_mut_ptr(), 1);
}
} }
impl Audio { impl Audio {
@ -24,32 +34,27 @@ impl Audio {
} }
} }
pub fn new(format: format::Sample, length: usize, layout: ChannelLayout) -> Self { pub fn new(format: Sample, samples: usize, layout: ChannelLayout) -> Self {
unsafe { unsafe {
let mut frame = Audio::empty(); let mut frame = Audio::empty();
frame.alloc(format, samples, layout);
frame.set_format(format);
frame.set_length(length);
frame.set_channel_layout(layout);
av_frame_get_buffer(frame.as_mut_ptr(), 1);
frame frame
} }
} }
pub fn format(&self) -> format::Sample { pub fn format(&self) -> Sample {
unsafe { unsafe {
if (*self.as_ptr()).format == -1 { if (*self.as_ptr()).format == -1 {
format::Sample::None Sample::None
} }
else { else {
format::Sample::from(mem::transmute::<_, AVSampleFormat>(((*self.as_ptr()).format))) Sample::from(mem::transmute::<_, AVSampleFormat>(((*self.as_ptr()).format)))
} }
} }
} }
pub fn set_format(&mut self, value: format::Sample) { pub fn set_format(&mut self, value: Sample) {
unsafe { unsafe {
(*self.as_mut_ptr()).format = mem::transmute::<AVSampleFormat, c_int>(value.into()); (*self.as_mut_ptr()).format = mem::transmute::<AVSampleFormat, c_int>(value.into());
} }
@ -91,27 +96,70 @@ impl Audio {
} }
} }
pub fn length(&self) -> usize { pub fn samples(&self) -> usize {
unsafe { unsafe {
(*self.as_ptr()).nb_samples as usize (*self.as_ptr()).nb_samples as usize
} }
} }
pub fn set_length(&mut self, value: usize) { pub fn set_samples(&mut self, value: usize) {
unsafe { unsafe {
(*self.as_mut_ptr()).nb_samples = value as c_int; (*self.as_mut_ptr()).nb_samples = value as c_int;
} }
} }
pub fn samples(&self) -> Samples { pub fn is_planar(&self) -> bool {
self.format().is_planar()
}
pub fn is_packed(&self) -> bool {
self.format().is_packed()
}
pub fn planes(&self) -> usize {
unsafe { unsafe {
Samples::wrap(self.as_ptr() as *mut AVPicture, self.format(), self.rate(), self.length(), self.channels(), self.channel_layout()) if (*self.as_ptr()).linesize[0] == 0 {
return 0;
}
}
if self.is_packed() {
1
}
else {
self.samples()
} }
} }
pub fn samples_mut(&mut self) -> Samples { pub fn plane<T: Reflect + 'static>(&self, index: usize) -> &[T] {
if index >= self.planes() {
panic!("out of bounds");
}
if !valid::<T>(self.format()) {
panic!("unsupported type");
}
unsafe { unsafe {
Samples::wrap(self.as_ptr() as *mut AVPicture, self.format(), self.rate(), self.length(), self.channels(), self.channel_layout()) slice::from_raw_parts(
mem::transmute((*self.as_ptr()).data[index]),
mem::size_of::<T>() * self.samples())
}
}
pub fn plane_mut<T: Reflect + 'static>(&mut self, index: usize) -> &[T] {
if index >= self.planes() {
panic!("out of bounds");
}
if !valid::<T>(self.format()) {
panic!("unsupported type");
}
unsafe {
slice::from_raw_parts_mut(
mem::transmute((*self.as_mut_ptr()).data[index]),
mem::size_of::<T>() * self.samples())
} }
} }
} }
@ -134,7 +182,7 @@ impl DerefMut for Audio {
impl Clone for Audio { impl Clone for Audio {
fn clone(&self) -> Self { fn clone(&self) -> Self {
let mut cloned = Audio::new(self.format(), self.length(), self.channel_layout()); let mut cloned = Audio::new(self.format(), self.samples(), self.channel_layout());
cloned.clone_from(self); cloned.clone_from(self);
cloned cloned
@ -147,3 +195,28 @@ impl Clone for Audio {
} }
} }
} }
fn valid<T: Reflect + 'static>(format: Sample) -> bool {
match format {
Sample::None =>
false,
Sample::U8(..) if TypeId::of::<T>() != TypeId::of::<u8>() =>
false,
Sample::I16(..) if TypeId::of::<T>() != TypeId::of::<i16>() =>
false,
Sample::I32(..) if TypeId::of::<T>() != TypeId::of::<i32>() =>
false,
Sample::F32(..) if TypeId::of::<T>() != TypeId::of::<f32>() =>
false,
Sample::F64(..) if TypeId::of::<T>() != TypeId::of::<f64>() =>
false,
_ =>
true
}
}

View File

@ -8,7 +8,6 @@ pub mod format;
pub mod frame; pub mod frame;
pub mod chroma; pub mod chroma;
pub mod time; pub mod time;
pub mod samples;
pub mod channel_layout; pub mod channel_layout;
use std::ffi::CStr; use std::ffi::CStr;

View File

@ -1,143 +0,0 @@
use std::slice;
use std::mem;
use std::any::TypeId;
use std::marker::{Reflect, PhantomData};
use ffi::*;
use ::util::format::Sample;
use ::{Error, ChannelLayout};
pub struct Samples<'a> {
pub ptr: *mut AVPicture,
format: Sample,
rate: u32,
number: usize,
channels: u16,
layout: ChannelLayout,
_marker: PhantomData<&'a ()>,
}
impl<'a> Samples<'a> {
pub fn wrap(ptr: *mut AVPicture, format: Sample, rate: u32, number: usize, channels: u16, layout: ChannelLayout) -> Self {
Samples {
ptr: ptr,
format: format,
rate: rate,
number: number,
channels: channels,
layout: layout,
_marker: PhantomData,
}
}
pub fn format(&self) -> Sample {
self.format
}
pub fn rate(&self) -> u32 {
self.rate
}
pub fn number(&self) -> usize {
self.number
}
pub fn channels(&self) -> u16 {
self.channels
}
pub fn channel_layout(&self) -> ChannelLayout {
self.layout
}
pub fn is_planar(&self) -> bool {
self.format.is_planar()
}
pub fn is_packed(&self) -> bool {
self.format.is_planar()
}
pub fn data<T: Reflect + 'static>(&self) -> Result<Vec<&[T]>, Error> {
if !valid::<T>(self.format) {
return Err(Error::InvalidData)
}
let mut result = Vec::new();
unsafe {
let size = (*self.ptr).linesize[0];
if self.is_planar() {
for i in 0 .. self.channels {
result.push(slice::from_raw_parts(
mem::transmute((*self.ptr).data[i as usize]),
size as usize / mem::size_of::<T>()));
}
}
else {
result.push(slice::from_raw_parts(
mem::transmute((*self.ptr).data[0]),
size as usize / mem::size_of::<T>()));
}
}
Ok(result)
}
pub fn data_mut<T: Reflect + 'static>(&mut self) -> Result<Vec<&mut [T]>, Error> {
if !valid::<T>(self.format) {
return Err(Error::InvalidData)
}
let mut result = Vec::new();
unsafe {
let size = (*self.ptr).linesize[0];
if self.is_planar() {
for i in 0 .. self.channels {
result.push(slice::from_raw_parts_mut(
mem::transmute((*self.ptr).data[i as usize]),
size as usize / mem::size_of::<T>()));
}
}
else {
result.push(slice::from_raw_parts_mut(
mem::transmute((*self.ptr).data[0]),
size as usize / mem::size_of::<T>()));
}
}
Ok(result)
}
}
pub fn valid<T: Reflect + 'static>(format: Sample) -> bool {
match format {
Sample::None =>
false,
Sample::U8(..) if TypeId::of::<T>() != TypeId::of::<u8>() =>
false,
Sample::I16(..) if TypeId::of::<T>() != TypeId::of::<i16>() =>
false,
Sample::I32(..) if TypeId::of::<T>() != TypeId::of::<i32>() =>
false,
Sample::F32(..) if TypeId::of::<T>() != TypeId::of::<f32>() =>
false,
Sample::F64(..) if TypeId::of::<T>() != TypeId::of::<f64>() =>
false,
_ =>
true
}
}