Kieran/ffmpeg-the-third
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

util/frame/audio: improve ergonomics

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/lib.rs
View File

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

117
src/util/frame/audio.rs
View File

@ -1,11 +1,13 @@
use libc::c_int;
use std::mem;
use std::slice;
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 ::{ChannelLayout, Samples};
use ::util::format;
use ::ChannelLayout;
use ::util::format::Sample;
use super::Frame;
#[derive(PartialEq, Eq)]
@ -15,6 +17,14 @@ impl Audio {
pub unsafe fn wrap(ptr: *mut AVFrame) -> Self {
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 {
@ -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 {
let mut frame = Audio::empty();
frame.set_format(format);
frame.set_length(length);
frame.set_channel_layout(layout);
av_frame_get_buffer(frame.as_mut_ptr(), 1);
frame.alloc(format, samples, layout);
frame
}
}
pub fn format(&self) -> format::Sample {
pub fn format(&self) -> Sample {
unsafe {
if (*self.as_ptr()).format == -1 {
format::Sample::None
Sample::None
}
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 {
(*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 {
(*self.as_ptr()).nb_samples as usize
}
}
pub fn set_length(&mut self, value: usize) {
pub fn set_samples(&mut self, value: usize) {
unsafe {
(*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 {
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 {
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 {
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
@ -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
}
}

1
src/util/mod.rs
View File

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

143
src/util/samples.rs
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
}
}