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