codec: improve decoder state transitions

This commit improves the state transitions for the `codec::*` structs as
discussed in #7.
This commit is contained in:
lummax
2015-08-25 15:20:58 +02:00
committed by meh
parent 8ed092c0c3
commit ee6004d05e
8 changed files with 185 additions and 186 deletions

View File

@ -15,110 +15,78 @@ pub use self::conceal::Conceal;
pub mod check;
pub use self::check::Check;
pub mod opened;
pub use self::opened::Opened;
use std::ptr;
use std::ffi::CString;
use std::slice::from_raw_parts;
use std::ops::{Deref, DerefMut};
use ffi::*;
use super::{Id, Profile};
use super::context::Opened;
use ::{Codec, Error, Discard, Rational};
use ::media;
use super::{Id, Context};
use ::{Codec, Error, Dictionary};
pub struct Decoder(pub Opened);
pub struct Decoder(pub Context);
impl Decoder {
pub fn open(mut self, codec: &Codec) -> Result<Opened, Error> {
unsafe {
if codec.is_decoder() {
match avcodec_open2(self.as_mut_ptr(), codec.as_ptr(), ptr::null_mut()) {
0 => Ok(Opened(self)),
e => Err(Error::from(e))
}
}
else {
Err(Error::InvalidData)
}
}
}
pub fn open_with(mut self, codec: &Codec, options: Dictionary) -> Result<Opened, Error> {
unsafe {
if codec.is_decoder() {
match avcodec_open2(self.as_mut_ptr(), codec.as_ptr(), &mut options.take()) {
0 => Ok(Opened(self)),
e => Err(Error::from(e))
}
}
else {
Err(Error::InvalidData)
}
}
}
pub fn video(self) -> Result<Video, Error> {
if self.medium() == media::Type::Video {
Ok(Video(self))
if let Some(ref codec) = find(self.id()) {
self.open(codec).and_then(|o| o.video())
}
else {
Err(Error::InvalidData)
Err(Error::DecoderNotFound)
}
}
pub fn audio(self) -> Result<Audio, Error> {
if self.medium() == media::Type::Audio {
Ok(Audio(self))
if let Some(ref codec) = find(self.id()) {
self.open(codec).and_then(|o| o.audio())
}
else {
Err(Error::InvalidData)
Err(Error::DecoderNotFound)
}
}
pub fn subtitle(self) -> Result<Subtitle, Error> {
if self.medium() == media::Type::Subtitle {
Ok(Subtitle(self))
if let Some(ref codec) = find(self.id()) {
self.open(codec).and_then(|o| o.subtitle())
}
else {
Err(Error::InvalidData)
}
}
pub fn conceal(&mut self, value: Conceal) {
unsafe {
(*self.as_mut_ptr()).error_concealment = value.bits();
}
}
pub fn check(&mut self, value: Check) {
unsafe {
(*self.as_mut_ptr()).err_recognition = value.bits();
}
}
pub fn profile(&self) -> Profile {
unsafe {
Profile::from((self.id(), (*self.as_ptr()).profile))
}
}
pub fn skip_loop_filter(&mut self, value: Discard) {
unsafe {
(*self.as_mut_ptr()).skip_loop_filter = value.into();
}
}
pub fn skip_idct(&mut self, value: Discard) {
unsafe {
(*self.as_mut_ptr()).skip_idct = value.into();
}
}
pub fn skip_frame(&mut self, value: Discard) {
unsafe {
(*self.as_mut_ptr()).skip_frame = value.into();
}
}
pub fn subtitle_header(&self) -> &[u8] {
unsafe {
from_raw_parts((*self.as_ptr()).subtitle_header, (*self.as_ptr()).subtitle_header_size as usize)
}
}
pub fn frame_rate(&self) -> Option<Rational> {
unsafe {
let value = (*self.as_ptr()).framerate;
if value == (AVRational { num: 0, den: 1 }) {
None
}
else {
Some(Rational::from(value))
}
}
}
pub fn time_base(&self) -> Rational {
unsafe {
Rational::from((*self.as_ptr()).time_base)
Err(Error::DecoderNotFound)
}
}
}
impl Deref for Decoder {
type Target = Opened;
type Target = Context;
fn deref(&self) -> &<Self as Deref>::Target {
&self.0