From d48ee4147254b8019b45647d6a10d3d0c83a5fc0 Mon Sep 17 00:00:00 2001 From: Kostya Shishkov Date: Tue, 16 May 2017 14:09:44 +0200 Subject: [PATCH] codec register --- src/lib.rs | 2 +- src/register.rs | 173 ++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 174 insertions(+), 1 deletion(-) create mode 100644 src/register.rs diff --git a/src/lib.rs b/src/lib.rs index a8ba225..5147a43 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -5,4 +5,4 @@ pub mod demuxers; pub mod formats; pub mod frame; pub mod io; - +pub mod register; diff --git a/src/register.rs b/src/register.rs new file mode 100644 index 0000000..76ff16f --- /dev/null +++ b/src/register.rs @@ -0,0 +1,173 @@ +use std::fmt; + +#[derive(Debug,Clone,Copy)] +#[allow(dead_code)] +pub enum CodecType { + Video, + Audio, + Subtitles, + Data, + None, +} + +impl fmt::Display for CodecType { + fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { + match *self { + CodecType::Video => write!(f, "Video"), + CodecType::Audio => write!(f, "Audio"), + CodecType::Subtitles => write!(f, "Subtitles"), + CodecType::Data => write!(f, "Data"), + CodecType::None => write!(f, "-"), + } + } +} + +const CODEC_CAP_INTRAONLY:u32 = 0x000001; +const CODEC_CAP_LOSSLESS:u32 = 0x000002; +const CODEC_CAP_REORDER:u32 = 0x000004; +const CODEC_CAP_HYBRID:u32 = 0x000008; +const CODEC_CAP_SCALABLE:u32 = 0x000010; + +pub struct CodecDescription { + name: &'static str, + fname: &'static str, + ctype: CodecType, + caps: u32, +} + +impl CodecDescription { + pub fn get_name(&self) -> &'static str { self.name } + pub fn get_full_name(&self) -> &'static str { self.fname } + pub fn get_codec_type(&self) -> CodecType { self.ctype } + pub fn is_intraonly(&self) -> bool { (self.caps & CODEC_CAP_INTRAONLY) != 0 } + pub fn is_lossless(&self) -> bool { (self.caps & CODEC_CAP_LOSSLESS) != 0 } + pub fn has_reorder(&self) -> bool { (self.caps & CODEC_CAP_REORDER) != 0 } + pub fn is_hybrid(&self) -> bool { (self.caps & CODEC_CAP_HYBRID) != 0 } + pub fn is_scalable(&self) -> bool { (self.caps & CODEC_CAP_SCALABLE) != 0 } +} + +impl fmt::Display for CodecDescription { + fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { + let mut out = format!("{}", self.fname); + if self.caps != 0 { + let mut capfmt = "".to_string(); + if (self.caps & CODEC_CAP_INTRAONLY) != 0 { + capfmt = format!("{} Intra-only", capfmt); + } + if (self.caps & CODEC_CAP_LOSSLESS) != 0 { + capfmt = format!("{} Lossless", capfmt); + } + if (self.caps & CODEC_CAP_REORDER) != 0 { + capfmt = format!("{} Frame reorder", capfmt); + } + if (self.caps & CODEC_CAP_HYBRID) != 0 { + capfmt = format!("{} Can be lossy and lossless", capfmt); + } + if (self.caps & CODEC_CAP_SCALABLE) != 0 { + capfmt = format!("{} Scalable", capfmt); + } + out = format!("{} ({})", out, capfmt); + } + write!(f, "{}", out) + } +} + +macro_rules! desc { + (video; $n:expr, $fn:expr) => ({ + CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Video, + caps: 0 } + }); + (video; $n:expr, $fn:expr, $c:expr) => ({ + CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Video, + caps: $c } + }); + (video-ll; $n:expr, $fn:expr) => ({ + CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Video, + caps: CODEC_CAP_LOSSLESS | CODEC_CAP_INTRAONLY } + }); + (video-llp; $n:expr, $fn:expr) => ({ + CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Video, + caps: CODEC_CAP_LOSSLESS } + }); + (video-im; $n:expr, $fn:expr) => ({ + CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Video, + caps: CODEC_CAP_INTRAONLY } + }); + (video-modern; $n:expr, $fn:expr) => ({ + CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Video, + caps: CODEC_CAP_REORDER | CODEC_CAP_HYBRID } + }); + (audio; $n:expr, $fn:expr) => ({ + CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Audio, + caps: 0 } + }); + (audio-ll; $n:expr, $fn:expr) => ({ + CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Audio, + caps: CODEC_CAP_LOSSLESS | CODEC_CAP_INTRAONLY } + }); +} + +pub fn get_codec_description(name: &str) -> Option<&'static CodecDescription> { + for i in 0..CODEC_REGISTER.len() { + if CODEC_REGISTER[i].name == name { + return Some(&CODEC_REGISTER[i]); + } + } + None +} + +static CODEC_REGISTER: &'static [CodecDescription] = &[ + desc!(video-im; "video-indeo1", "Intel Raw IF09"), + desc!(video-im; "video-indeo2", "Intel Indeo 2"), + desc!(video; "video-indeo3", "Intel Indeo 3"), + desc!(video; "video-indeo4", "Intel Indeo 4", CODEC_CAP_REORDER | CODEC_CAP_SCALABLE), + desc!(video; "video-indeo5", "Intel Indeo 5", CODEC_CAP_REORDER | CODEC_CAP_SCALABLE), + desc!(audio; "audio-iac", "Intel Indeo audio"), + desc!(audio; "audio-imc", "Intel Music Coder"), +]; + +static AVI_VIDEO_CODEC_REGISTER: &'static [(&[u8;4], &str)] = &[ + (b"IF09", "video-indeo1"), + (b"RT21", "video-indeo2"), + (b"IV31", "video-indeo3"), + (b"IV32", "video-indeo3"), + (b"IV41", "video-indeo4"), + (b"IV50", "video-indeo5"), +]; + +static WAV_CODEC_REGISTER: &'static [(u16, &str)] = &[ + (0x0000, "audio-pcm"), + (0x0401, "audio-imc"), + (0x0402, "audio-iac"), +]; + +pub fn find_codec_from_avi_fourcc(fcc: &[u8;4]) -> Option<&'static str> { + for i in 0..AVI_VIDEO_CODEC_REGISTER.len() { + let (fourcc, name) = AVI_VIDEO_CODEC_REGISTER[i]; + if fourcc == fcc { return Some(name); } + } + None +} + +pub fn find_codec_from_wav_twocc(tcc: u16) -> Option<&'static str> { + for i in 0..WAV_CODEC_REGISTER.len() { + let (twocc, name) = WAV_CODEC_REGISTER[i]; + if twocc == tcc { return Some(name); } + } + None +} + +#[cfg(test)] +mod test { + use super::*; + + #[test] + fn test_register() { + let c1 = find_codec_from_avi_fourcc(b"IV41").unwrap(); + let c2 = find_codec_from_wav_twocc(0x401).unwrap(); + println!("found {} and {}", c1, c2); + let cd1 = get_codec_description(c1).unwrap(); + let cd2 = get_codec_description(c2).unwrap(); + println!("got {} and {}", cd1, cd2); + } +} \ No newline at end of file -- 2.39.5