Files
ffmpeg-the-third/src/codec/decoder/mod.rs

141 lines
2.4 KiB
Rust

pub mod video;
pub use self::video::Video;
pub mod audio;
pub use self::audio::Audio;
pub mod subtitle;
pub use self::subtitle::Subtitle;
pub mod slice;
pub mod conceal;
pub use self::conceal::Conceal;
pub mod check;
pub use self::check::Check;
use std::ffi::CString;
use std::ptr;
use std::slice::from_raw_parts;
use std::ops::Deref;
use ffi::*;
use super::{Id, Profile};
use super::context::Opened;
use ::{Codec, Error, Discard, Rational};
use ::media;
pub struct Decoder(pub Opened);
impl Decoder {
pub fn video(self) -> Result<Video, Error> {
if self.medium() == media::Type::Video {
Ok(Video(self))
}
else {
Err(Error::InvalidData)
}
}
pub fn audio(self) -> Result<Audio, Error> {
if self.medium() == media::Type::Audio {
Ok(Audio(self))
}
else {
Err(Error::InvalidData)
}
}
pub fn subtitle(self) -> Result<Subtitle, Error> {
if self.medium() == media::Type::Subtitle {
Ok(Subtitle(self))
}
else {
Err(Error::InvalidData)
}
}
pub fn conceal(&mut self, value: Conceal) {
unsafe {
(*self.ptr).error_concealment = value.bits();
}
}
pub fn check(&mut self, value: Check) {
unsafe {
(*self.ptr).err_recognition = value.bits();
}
}
pub fn profile(&self) -> Profile {
unsafe {
Profile::from((self.id(), (*self.ptr).profile))
}
}
pub fn skip_loop_filter(&mut self, value: Discard) {
unsafe {
(*self.ptr).skip_loop_filter = value.into();
}
}
pub fn skip_idct(&mut self, value: Discard) {
unsafe {
(*self.ptr).skip_idct = value.into();
}
}
pub fn skip_frame(&mut self, value: Discard) {
unsafe {
(*self.ptr).skip_frame = value.into();
}
}
pub fn subtitle_header(&self) -> &[u8] {
unsafe {
from_raw_parts((*self.ptr).subtitle_header, (*self.ptr).subtitle_header_size as usize)
}
}
pub fn time_base(&self) -> Rational {
unsafe {
Rational((*self.ptr).time_base)
}
}
}
impl Deref for Decoder {
type Target = Opened;
fn deref(&self) -> &<Self as Deref>::Target {
&self.0
}
}
pub fn find(id: Id) -> Option<Codec<'static>> {
unsafe {
let ptr = avcodec_find_decoder(id.into());
if ptr == ptr::null_mut() {
None
}
else {
Some(Codec::wrap(ptr))
}
}
}
pub fn find_by_name(name: &str) -> Option<Codec<'static>> {
unsafe {
let ptr = avcodec_find_decoder_by_name(CString::new(name).unwrap().as_ptr());
if ptr == ptr::null_mut() {
None
}
else {
Some(Codec::wrap(ptr))
}
}
}