]> git.nihav.org Git - nihav.git/blob - nihav-registry/src/register.rs
add another FOURCC for TrueMotion 1
[nihav.git] / nihav-registry / src / register.rs
1 //! Global registry of codec information.
2 //!
3 //! This module contains codec information from technical level that allows user to retrieve information about codec type and features without creating and invoking a decoder for such codec.
4 use std::fmt;
5
6 /// Codec types.
7 #[derive(Debug,Clone,Copy,PartialEq)]
8 #[allow(dead_code)]
9 pub enum CodecType {
10 /// Video codec.
11 Video,
12 /// Audio codec.
13 Audio,
14 /// Subtitle codec.
15 Subtitles,
16 /// Some special codec (e.g. some event stream or separate timecodes stream).
17 Data,
18 /// Dummy codec.
19 None,
20 }
21
22 impl fmt::Display for CodecType {
23 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
24 match *self {
25 CodecType::Video => write!(f, "Video"),
26 CodecType::Audio => write!(f, "Audio"),
27 CodecType::Subtitles => write!(f, "Subtitles"),
28 CodecType::Data => write!(f, "Data"),
29 CodecType::None => write!(f, "-"),
30 }
31 }
32 }
33
34 /// Codec capability flag for intra-only codecs.
35 pub const CODEC_CAP_INTRAONLY:u32 = 0x0001;
36 /// Codec capability flag for lossless codecs.
37 pub const CODEC_CAP_LOSSLESS:u32 = 0x0002;
38 /// Codec capability flag for codecs with frame reordering.
39 pub const CODEC_CAP_REORDER:u32 = 0x0004;
40 /// Codec capability flag for codecs that can be both lossy and lossless.
41 pub const CODEC_CAP_HYBRID:u32 = 0x0008;
42 /// Codec capability flag for codecs with scalability features.
43 pub const CODEC_CAP_SCALABLE:u32 = 0x0010;
44 /// Codec capability flag for codecs with complex frame reordering.
45 pub const CODEC_CAP_COMPLEX_REORDER:u32 = 0x0020;
46
47 /// Codec description structure.
48 #[derive(Clone)]
49 pub struct CodecDescription {
50 /// Short codec name.
51 ///
52 /// Short codec name is used inside NihAV as the unique identifier.
53 pub name: &'static str,
54 /// Full codec name.
55 pub fname: &'static str,
56 /// Codec type.
57 pub ctype: CodecType,
58 /// Codec capabilities.
59 pub caps: u32,
60 }
61
62 impl CodecDescription {
63 /// Returns short codec name.
64 pub fn get_name(&self) -> &'static str { self.name }
65 /// Returns full codec name.
66 pub fn get_full_name(&self) -> &'static str { self.fname }
67 /// Returns codec type.
68 pub fn get_codec_type(&self) -> CodecType { self.ctype }
69 /// Reports whether the codec has only intra frames or not.
70 pub fn is_intraonly(&self) -> bool { (self.caps & CODEC_CAP_INTRAONLY) != 0 }
71 /// Reports whether the codec is lossless.
72 pub fn is_lossless(&self) -> bool { (self.caps & CODEC_CAP_LOSSLESS) != 0 }
73 /// Reports whether the codec requires frame reordering.
74 pub fn has_reorder(&self) -> bool { (self.caps & CODEC_CAP_REORDER) != 0 }
75 /// Reports whether the codec can be either lossless or lossy.
76 pub fn is_hybrid(&self) -> bool { (self.caps & CODEC_CAP_HYBRID) != 0 }
77 /// Reports whether codec supports scalability.
78 ///
79 /// Scalability means that codec can be decoded in reduced resolution by design.
80 pub fn is_scalable(&self) -> bool { (self.caps & CODEC_CAP_SCALABLE) != 0 }
81 }
82
83 impl fmt::Display for CodecDescription {
84 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
85 let mut out = self.fname.to_string();
86 if self.caps != 0 {
87 let mut capfmt = "".to_string();
88 if (self.caps & CODEC_CAP_INTRAONLY) != 0 {
89 capfmt = format!("{} Intra-only", capfmt);
90 }
91 if (self.caps & CODEC_CAP_LOSSLESS) != 0 {
92 capfmt = format!("{} Lossless", capfmt);
93 }
94 if (self.caps & CODEC_CAP_REORDER) != 0 {
95 capfmt = format!("{} Frame reorder", capfmt);
96 }
97 if (self.caps & CODEC_CAP_HYBRID) != 0 {
98 capfmt = format!("{} Can be lossy and lossless", capfmt);
99 }
100 if (self.caps & CODEC_CAP_SCALABLE) != 0 {
101 capfmt = format!("{} Scalable", capfmt);
102 }
103 out = format!("{} ({})", out, capfmt);
104 }
105 write!(f, "{}", out)
106 }
107 }
108
109 macro_rules! desc {
110 (video; $n:expr, $fn:expr) => ({
111 CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Video,
112 caps: 0 }
113 });
114 (video; $n:expr, $fn:expr, $c:expr) => ({
115 CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Video,
116 caps: $c }
117 });
118 (video-ll; $n:expr, $fn:expr) => ({
119 CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Video,
120 caps: CODEC_CAP_LOSSLESS | CODEC_CAP_INTRAONLY }
121 });
122 (video-llp; $n:expr, $fn:expr) => ({
123 CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Video,
124 caps: CODEC_CAP_LOSSLESS }
125 });
126 (video-im; $n:expr, $fn:expr) => ({
127 CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Video,
128 caps: CODEC_CAP_INTRAONLY }
129 });
130 (video-modern; $n:expr, $fn:expr) => ({
131 CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Video,
132 caps: CODEC_CAP_REORDER | CODEC_CAP_HYBRID }
133 });
134 (audio; $n:expr, $fn:expr) => ({
135 CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Audio,
136 caps: 0 }
137 });
138 (audio-ll; $n:expr, $fn:expr) => ({
139 CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Audio,
140 caps: CODEC_CAP_LOSSLESS | CODEC_CAP_INTRAONLY }
141 });
142 (audio-hyb; $n:expr, $fn:expr) => ({
143 CodecDescription{ name: $n, fname: $fn, ctype: CodecType::Audio,
144 caps: CODEC_CAP_HYBRID }
145 });
146 }
147
148 /// Returns codec description for the provided codec short name if it is found.
149 pub fn get_codec_description(name: &str) -> Option<&'static CodecDescription> {
150 CODEC_REGISTER.iter().find(|&reg| reg.name == name)
151 }
152
153 static CODEC_REGISTER: &[CodecDescription] = &[
154 desc!(audio-ll; "pcm", "PCM"),
155 desc!(audio; "alaw", "A-law PCM"),
156 desc!(audio; "ulaw", "mu-law PCM"),
157
158 desc!(video-im; "indeo1", "Intel Raw IF09"),
159 desc!(video-im; "indeo2", "Intel Indeo 2"),
160 desc!(video; "indeo3", "Intel Indeo 3"),
161 desc!(video; "indeo4", "Intel Indeo 4", CODEC_CAP_REORDER | CODEC_CAP_SCALABLE),
162 desc!(video; "indeo5", "Intel Indeo 5", CODEC_CAP_REORDER | CODEC_CAP_SCALABLE),
163 desc!(video; "indeo5s", "Intel Indeo 5 Scalable", CODEC_CAP_SCALABLE),
164 desc!(video; "intel263", "Intel I263", CODEC_CAP_REORDER),
165 desc!(audio; "iac", "Intel Indeo audio"),
166 desc!(audio; "imc", "Intel Music Coder"),
167
168 desc!(video; "realvideo1", "Real Video 1"),
169 desc!(video; "realvideo2", "Real Video 2", CODEC_CAP_REORDER),
170 desc!(video; "realvideo3", "Real Video 3", CODEC_CAP_REORDER),
171 desc!(video; "realvideo4", "Real Video 4", CODEC_CAP_REORDER),
172 desc!(video; "realvideo6", "Real Video 6", CODEC_CAP_REORDER),
173 desc!(video; "clearvideo", "ClearVideo"),
174 desc!(video; "clearvideo_rm", "ClearVideo"),
175 desc!(audio; "ra14.4", "RealAudio 14.4"),
176 desc!(audio; "ra28.8", "RealAudio 28.8"),
177 desc!(audio; "cook", "RealAudio Cooker"),
178 desc!(audio; "ralf", "RealAudio Lossless"),
179 desc!(audio; "aac", "AAC"),
180 desc!(audio; "ac3", "ETSI TS 102 366"),
181 desc!(audio; "atrac3", "Sony Atrac3"),
182 desc!(audio; "sipro", "Sipro Labs ADPCM"),
183
184
185 desc!(video-ll; "rawvideo", "Raw video data"),
186 desc!(video-ll; "rawvideo-ms", "Raw video data"),
187
188 desc!(video; "cinepak", "Cinepak"),
189
190 desc!(video-llp; "zmbv", "Zip Motion Blocks Video"),
191
192 desc!(video; "msvideo1", "MS Video 1"),
193 desc!(video; "msrle", "MS RLE"),
194 desc!(audio; "ms-adpcm", "MS ADPCM"),
195 desc!(audio; "ima-adpcm-ms", "IMA ADPCM (MS variant)"),
196
197 desc!(video; "qt-smc", "Apple Graphics"),
198 desc!(video; "qt-rle", "Apple Animation"),
199 desc!(video; "apple-video", "Apple video"),
200 desc!(video; "sorenson-video", "Sorenson Video"),
201 desc!(video; "sorenson-video3", "Sorenson Video 3", CODEC_CAP_REORDER),
202 desc!(audio-ll; "alac", "Apple Lossless Audio Codec"),
203 desc!(audio; "mace-3", "MACE 3:1"),
204 desc!(audio; "mace-6", "MACE 6:1"),
205 desc!(audio; "ima-adpcm-qt", "IMA ADPCM (Apple variant)"),
206 desc!(audio; "qdesign-music", "QDesign Music"),
207 desc!(audio; "qdesign-music2", "QDesign Music v2"),
208 desc!(audio; "qualcomm-purevoice", "Qualcomm PureVoice"),
209
210 desc!(video-ll; "arm_rawvideo", "Acorn Replay Movie raw video formats"),
211 desc!(audio; "arm_rawaudio", "Acorn Replay Movie raw audio formats"),
212 desc!(video; "movinglines", "Acorn Moving Lines"),
213 desc!(video; "movingblocks", "Acorn Moving Blocks"),
214 desc!(video; "movingblockshq", "Acorn Moving Blocks HQ"),
215 desc!(video; "supermovingblocks", "Acorn Super Moving Blocks"),
216 desc!(video; "linepack", "Henrik Pedersen's LinePack"),
217 desc!(video; "movie16_3", "Henrik Pedersen's Movie 16:3"),
218 desc!(video; "escape100", "Eidos Escape 100"),
219 desc!(video; "escape102", "Eidos Escape 102"),
220 desc!(video; "escape122", "Eidos Escape 122"),
221 desc!(video; "escape124", "Eidos Escape 124"),
222 desc!(video; "escape130", "Eidos Escape 130"),
223
224 desc!(video; "truemotion1", "TrueMotion 1"),
225 desc!(video-im; "truemotionrt", "TrueMotion RT"),
226 desc!(video; "truemotion2", "TrueMotion 2"),
227 desc!(video; "truemotion2x", "TrueMotion 2X"),
228 desc!(video; "vp3", "VP3"),
229 desc!(video; "vp4", "VP4"),
230 desc!(video; "vp5", "VP5"),
231 desc!(video; "vp6", "VP6"),
232 desc!(video; "vp6f", "VP6 (in Flash)"),
233 desc!(video; "vp6a", "VP6 with alpha"),
234 desc!(video; "vp7", "VP7"),
235 desc!(video; "vp8", "VP8"),
236 desc!(video; "vp9", "VP9"),
237 desc!(audio; "adpcm-dk3", "Duck DK3 ADPCM"),
238 desc!(audio; "adpcm-dk4", "Duck DK4 ADPCM"),
239 desc!(audio; "on2avc-500", "On2 AVC"),
240 desc!(audio; "on2avc-501", "On2 AVC"),
241
242 desc!(video; "flv263", "Sorenson H.263"),
243 desc!(video-llp; "flashsv", "Flash Screen Video"),
244 desc!(video-llp; "flashsv2", "Flash Screen Video 2"),
245 desc!(audio; "asao", "N*llym*s*r ASAO"),
246 desc!(audio; "flv-adpcm", "Flash ADPCM"),
247
248 desc!(audio; "mp1", "MPEG Audio Layer I"),
249 desc!(audio; "mp2", "MPEG Audio Layer II"),
250 desc!(audio; "mp3", "MPEG Audio Layer III"),
251 desc!(audio; "speex", "Speex"),
252
253 desc!(video; "gdv-video", "Gremlin Digital Video - video"),
254 desc!(audio; "gdv-audio", "Gremlin Digital Video - audio"),
255 desc!(video-im; "arxel-video", "Arxel Tribe Video"),
256 desc!(video; "beam-fcp", "Beam Software Animation"),
257 desc!(video; "beam-video", "Beam Software Video"),
258 desc!(video; "bmv-video", "BMV video"),
259 desc!(audio; "bmv-audio", "BMV audio"),
260 desc!(video; "bmv3-video", "DW Noir BMV video"),
261 desc!(audio; "bmv3-audio", "DW Noir BMV audio"),
262 desc!(video; "dp-sga", "Digital Pictures SGA video"),
263 desc!(video; "fable-imax", "Fable IMAX video"),
264 desc!(video; "fst-video", "FutureVision video"),
265 desc!(audio; "fst-audio", "FutureVision audio"),
266 desc!(video; "hl-fmv-video", "Highlander FMV video"),
267 desc!(video-llp; "ipma", "Imagination Pilots Matte Animation"),
268 desc!(video-llp; "ipma2", "Imagination Pilots Matte Animation v2"),
269 desc!(video; "legend-q-video", "Legend Entertainment Q video"),
270 desc!(video; "midivid", "MidiVid"),
271 desc!(video; "midivid3", "MidiVid 3"),
272 desc!(video-ll; "midivid-ll", "MidiVid Lossless"),
273 desc!(video-ll; "rbt-video", "Sierra Robot video"),
274 desc!(audio; "rbt-audio", "Sierra Robot audio"),
275 desc!(video; "seq-video", "Sierra Sequence video"),
276 desc!(video; "smushv1", "SMUSH Video paletted"),
277 desc!(video; "smushv2", "SMUSH Video 16-bit"),
278 desc!(video; "smush-iact", "SMUSH IACT Audio"),
279 desc!(video; "smush-vima", "SMUSH VIMA Audio"),
280 desc!(video; "vmd-video", "VMD video"),
281 desc!(audio; "vmd-audio", "VMD audio"),
282 desc!(video; "vxvideo", "Actimagine Vx"),
283 desc!(audio; "vxaudio", "Actimagine Sx"),
284
285 desc!(video; "smacker-video", "Smacker video"),
286 desc!(audio; "smacker-audio", "Smacker audio"),
287 desc!(video; "bink-video", "Bink video"),
288 desc!(video; "bink2-video", "Bink2 video"),
289 desc!(audio; "bink-audio-dct", "Bink audio (DCT)"),
290 desc!(audio; "bink-audio-rdft", "Bink audio (RDFT)"),
291
292 desc!(audio; "lhst15f8", "L&H StreamTalk 15kbps at 8 kHz"),
293 desc!(audio; "lhst250f11", "L&H StreamTalk 25kbps at 11 kHz"),
294 desc!(audio; "lhst500f22", "L&H StreamTalk 50kpbs at 22 kHz"),
295 desc!(audio; "lhst48", "L&H StreamTalk CELP Codec 4.8kbps at 8 kHz"),
296
297 desc!(video; "vivo1", "VivoActive Video 1.0"),
298 desc!(video; "vivo2", "VivoActive Video 2.0", CODEC_CAP_REORDER),
299 desc!(audio; "g723.1", "ITU G.723.1"),
300 desc!(audio; "siren", "Polycom Siren"),
301
302 desc!(audio-ll; "ape", "Monkey's Audio"),
303 desc!(audio-ll; "flac", "Free Lossless Audio Codec"),
304 desc!(audio-ll; "tta", "True Audio codec"),
305 desc!(audio-hyb; "wavpack", "WavPack"),
306
307 desc!(video-ll; "gif", "GIF"),
308 desc!(video-im; "jpeg", "JPEG"),
309 desc!(video; "h264", "ITU H.264", CODEC_CAP_COMPLEX_REORDER | CODEC_CAP_HYBRID),
310
311 desc!(video-im; "mwv1", "Aware MotionWavelets"),
312 ];
313
314 static AVI_VIDEO_CODEC_REGISTER: &[(&[u8;4], &str)] = &[
315 (&[1, 0, 0, 0], "msrle"),
316 (&[2, 0, 0, 0], "msrle"),
317
318 (b"CRAM", "msvideo1"),
319 (b"MSVC", "msvideo1"),
320 (b"WHAM", "msvideo1"),
321
322 (b"MJPG", "jpeg"),
323
324 (b"IF09", "indeo1"),
325 (b"RT21", "indeo2"),
326 (b"IV31", "indeo3"),
327 (b"IV32", "indeo3"),
328 (b"IV41", "indeo4"),
329 (b"IV50", "indeo5"),
330 (b"I263", "intel263"),
331
332 (b"UCOD", "clearvideo"),
333 (b"cvid", "cinepak"),
334 (b"ZMBV", "zmbv"),
335
336 (b"Ipma", "ipma"),
337 (b"Ip20", "ipma2"),
338
339 (b"MVDV", "midivid"),
340 (b"MV30", "midivid3"),
341 (b"MVLZ", "midivid-ll"),
342
343 (b"tmot", "truemotion1"),
344 (b"DUCK", "truemotion1"),
345 (b"TR20", "truemotionrt"),
346 (b"TM20", "truemotion2"),
347 (b"TM2A", "truemotion2x"),
348 (b"TM2X", "truemotion2x"),
349 (b"VP30", "vp3"),
350 (b"VP31", "vp3"),
351 (b"VP40", "vp4"),
352 (b"VP50", "vp5"),
353 (b"VP60", "vp6"),
354 (b"VP61", "vp6"),
355 (b"VP62", "vp6"),
356 (b"VP6A", "vp6a"),
357 (b"VP70", "vp7"),
358
359 (b"MWV1", "mwv1"),
360 ];
361
362 static WAV_CODEC_REGISTER: &[(u16, &str)] = &[
363 (0x0000, "unknown"),
364 (0x0001, "pcm"),
365 (0x0002, "ms-adpcm"),
366 (0x0003, "pcm"),
367 (0x0011, "ima-adpcm-ms"),
368 (0x0061, "adpcm-dk4"),
369 (0x0062, "adpcm-dk3"),
370 (0x0401, "imc"),
371 (0x0402, "iac"),
372 (0x0500, "on2avc-500"),
373 (0x0501, "on2avc-501"),
374 ];
375
376 static MOV_VIDEO_CODEC_REGISTER: &[(&[u8;4], &str)] = &[
377 (b"cvid", "cinepak"),
378 (b"jpeg", "jpeg"),
379 //(b"raw ", "raw"),
380 //(b"Yuv2", "raw"),
381 (b"smc ", "qt-smc"),
382 (b"rle ", "qt-rle"),
383 (b"rpza", "apple-video"),
384 (b"kpcd", "kodak-photocd"),
385 //(b"mpeg", "mpeg-video"),
386 (b"mjpa", "mjpeg-a"),
387 (b"mjpb", "mjpeg-b"),
388 (b"svqi", "sorenson-video"),
389 (b"SVQ1", "sorenson-video"),
390 (b"svq3", "sorenson-video3"),
391 (b"SVQ3", "sorenson-video3"),
392
393 (b"IV31", "indeo3"),
394 (b"IV32", "indeo3"),
395
396 (b"UCOD", "clearvideo"),
397
398 (b"VP30", "vp3"),
399 (b"VP31", "vp3"),
400
401 (b"avc1", "h264"),
402 ];
403
404 static MOV_AUDIO_CODEC_REGISTER: &[(&[u8;4], &str)] = &[
405 (b"NONE", "pcm"),
406 (b"raw ", "pcm"),
407 (b"twos", "pcm"),
408 (b"sowt", "pcm"),
409 (b"fl32", "pcm"),
410 (b"fl64", "pcm"),
411 (b"in24", "pcm"),
412 (b"in32", "pcm"),
413 (b"MAC3", "mace-3"),
414 (b"MAC6", "mace-6"),
415 (b"ima4", "ima-adpcm-qt"),
416 (b"ulaw", "ulaw"),
417 (b"alaw", "alaw"),
418 (b"dvca", "dv-audio"),
419 (b"QDMC", "qdesign-music"),
420 (b"QDM2", "qdesign-music2"),
421 (b"Qclp", "qualcomm-purevoice"),
422 //(b".mp3", "mpeg-layer3"),
423
424 (b"mp4a", "aac"),
425
426 (b"alac", "alac"),
427 ];
428
429 /// Returns video codec short name for provided FOURCC (used in AVI format).
430 pub fn find_codec_from_avi_fourcc(fcc: &[u8;4]) -> Option<&'static str> {
431 for (fourcc, name) in AVI_VIDEO_CODEC_REGISTER.iter() {
432 if *fourcc == fcc { return Some(name); }
433 }
434 None
435 }
436
437 /// Returns FOURCC (used in AVI format) for provided codec name.
438 pub fn find_avi_fourcc(codecname: &str) -> Option<[u8; 4]> {
439 for (fourcc, name) in AVI_VIDEO_CODEC_REGISTER.iter() {
440 if *name == codecname { return Some(**fourcc); }
441 }
442 None
443 }
444
445 /// Returns known audio codec short name for provided TWOCC (used in WAV and AVI format).
446 pub fn find_codec_from_wav_twocc(tcc: u16) -> Option<&'static str> {
447 for (twocc, name) in WAV_CODEC_REGISTER.iter() {
448 if *twocc == tcc { return Some(name); }
449 }
450 None
451 }
452
453 /// Returns TWOCC (used in WAV and AVI format for provided codec name.
454 pub fn find_wav_twocc(codecname: &str) -> Option<u16> {
455 for (twocc, name) in WAV_CODEC_REGISTER.iter() {
456 if *name == codecname { return Some(*twocc); }
457 }
458 None
459 }
460
461 /// Returns video codec short name for provided FOURCC (used in MOV format).
462 pub fn find_codec_from_mov_video_fourcc(fcc: &[u8;4]) -> Option<&'static str> {
463 for (fourcc, name) in MOV_VIDEO_CODEC_REGISTER.iter() {
464 if *fourcc == fcc { return Some(name); }
465 }
466 None
467 }
468
469 /// Returns known audio codec short name for provided FOURCC (used in MOV format).
470 pub fn find_codec_from_mov_audio_fourcc(fcc: &[u8;4]) -> Option<&'static str> {
471 for (fourcc, name) in MOV_AUDIO_CODEC_REGISTER.iter() {
472 if *fourcc == fcc { return Some(name); }
473 }
474 None
475 }
476
477 #[cfg(test)]
478 mod test {
479 use super::*;
480
481 #[test]
482 fn test_register() {
483 let c1 = find_codec_from_avi_fourcc(b"IV41").unwrap();
484 let c2 = find_codec_from_wav_twocc(0x401).unwrap();
485 println!("found {} and {}", c1, c2);
486 let cd1 = get_codec_description(c1).unwrap();
487 let cd2 = get_codec_description(c2).unwrap();
488 println!("got {} and {}", cd1, cd2);
489 }
490 }