core: fix most clippy warnings
[nihav.git] / nihav-core / src / formats.rs
1 use std::str::FromStr;
2 use std::string::*;
3 use std::fmt;
4
5 #[derive(Debug,Copy,Clone,PartialEq)]
6 pub struct NASoniton {
7 bits: u8,
8 be: bool,
9 packed: bool,
10 planar: bool,
11 float: bool,
12 signed: bool,
13 }
14
15 pub const SONITON_FLAG_BE :u32 = 0x01;
16 pub const SONITON_FLAG_PACKED :u32 = 0x02;
17 pub const SONITON_FLAG_PLANAR :u32 = 0x04;
18 pub const SONITON_FLAG_FLOAT :u32 = 0x08;
19 pub const SONITON_FLAG_SIGNED :u32 = 0x10;
20
21 pub const SND_U8_FORMAT: NASoniton = NASoniton { bits: 8, be: false, packed: false, planar: false, float: false, signed: false };
22 pub const SND_S16_FORMAT: NASoniton = NASoniton { bits: 16, be: false, packed: false, planar: false, float: false, signed: true };
23 pub const SND_S16P_FORMAT: NASoniton = NASoniton { bits: 16, be: false, packed: false, planar: true, float: false, signed: true };
24 pub const SND_F32P_FORMAT: NASoniton = NASoniton { bits: 32, be: false, packed: false, planar: true, float: true, signed: true };
25
26 impl NASoniton {
27 pub fn new(bits: u8, flags: u32) -> Self {
28 let is_be = (flags & SONITON_FLAG_BE) != 0;
29 let is_pk = (flags & SONITON_FLAG_PACKED) != 0;
30 let is_pl = (flags & SONITON_FLAG_PLANAR) != 0;
31 let is_fl = (flags & SONITON_FLAG_FLOAT) != 0;
32 let is_sg = (flags & SONITON_FLAG_SIGNED) != 0;
33 NASoniton { bits, be: is_be, packed: is_pk, planar: is_pl, float: is_fl, signed: is_sg }
34 }
35
36 pub fn get_bits(self) -> u8 { self.bits }
37 pub fn is_be(self) -> bool { self.be }
38 pub fn is_packed(self) -> bool { self.packed }
39 pub fn is_planar(self) -> bool { self.planar }
40 pub fn is_float(self) -> bool { self.float }
41 pub fn is_signed(self) -> bool { self.signed }
42
43 pub fn get_audio_size(self, length: u64) -> usize {
44 if self.packed {
45 ((length * u64::from(self.bits) + 7) >> 3) as usize
46 } else {
47 (length * u64::from((self.bits + 7) >> 3)) as usize
48 }
49 }
50 }
51
52 impl fmt::Display for NASoniton {
53 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
54 let fmt = if self.float { "float" } else if self.signed { "int" } else { "uint" };
55 let end = if self.be { "BE" } else { "LE" };
56 write!(f, "({} bps, {} planar: {} packed: {} {})", self.bits, end, self.packed, self.planar, fmt)
57 }
58 }
59
60 #[derive(Debug,Clone,Copy,PartialEq)]
61 pub enum NAChannelType {
62 C, L, R, Cs, Ls, Rs, Lss, Rss, LFE, Lc, Rc, Lh, Rh, Ch, LFE2, Lw, Rw, Ov, Lhs, Rhs, Chs, Ll, Rl, Cl, Lt, Rt, Lo, Ro
63 }
64
65 impl NAChannelType {
66 pub fn is_center(self) -> bool {
67 match self {
68 NAChannelType::C => true, NAChannelType::Ch => true,
69 NAChannelType::Cl => true, NAChannelType::Ov => true,
70 NAChannelType::LFE => true, NAChannelType::LFE2 => true,
71 NAChannelType::Cs => true, NAChannelType::Chs => true,
72 _ => false,
73 }
74 }
75 pub fn is_left(self) -> bool {
76 match self {
77 NAChannelType::L => true, NAChannelType::Ls => true,
78 NAChannelType::Lss => true, NAChannelType::Lc => true,
79 NAChannelType::Lh => true, NAChannelType::Lw => true,
80 NAChannelType::Lhs => true, NAChannelType::Ll => true,
81 NAChannelType::Lt => true, NAChannelType::Lo => true,
82 _ => false,
83 }
84 }
85 pub fn is_right(self) -> bool {
86 match self {
87 NAChannelType::R => true, NAChannelType::Rs => true,
88 NAChannelType::Rss => true, NAChannelType::Rc => true,
89 NAChannelType::Rh => true, NAChannelType::Rw => true,
90 NAChannelType::Rhs => true, NAChannelType::Rl => true,
91 NAChannelType::Rt => true, NAChannelType::Ro => true,
92 _ => false,
93 }
94 }
95 }
96
97 #[derive(Clone,Copy,Debug,PartialEq)]
98 pub struct ChannelParseError {}
99
100 impl FromStr for NAChannelType {
101 type Err = ChannelParseError;
102
103 fn from_str(s: &str) -> Result<Self, Self::Err> {
104 match s {
105 "C" => Ok(NAChannelType::C),
106 "L" => Ok(NAChannelType::L),
107 "R" => Ok(NAChannelType::R),
108 "Cs" => Ok(NAChannelType::Cs),
109 "Ls" => Ok(NAChannelType::Ls),
110 "Rs" => Ok(NAChannelType::Rs),
111 "Lss" => Ok(NAChannelType::Lss),
112 "Rss" => Ok(NAChannelType::Rss),
113 "LFE" => Ok(NAChannelType::LFE),
114 "Lc" => Ok(NAChannelType::Lc),
115 "Rc" => Ok(NAChannelType::Rc),
116 "Lh" => Ok(NAChannelType::Lh),
117 "Rh" => Ok(NAChannelType::Rh),
118 "Ch" => Ok(NAChannelType::Ch),
119 "LFE2" => Ok(NAChannelType::LFE2),
120 "Lw" => Ok(NAChannelType::Lw),
121 "Rw" => Ok(NAChannelType::Rw),
122 "Ov" => Ok(NAChannelType::Ov),
123 "Lhs" => Ok(NAChannelType::Lhs),
124 "Rhs" => Ok(NAChannelType::Rhs),
125 "Chs" => Ok(NAChannelType::Chs),
126 "Ll" => Ok(NAChannelType::Ll),
127 "Rl" => Ok(NAChannelType::Rl),
128 "Cl" => Ok(NAChannelType::Cl),
129 "Lt" => Ok(NAChannelType::Lt),
130 "Rt" => Ok(NAChannelType::Rt),
131 "Lo" => Ok(NAChannelType::Lo),
132 "Ro" => Ok(NAChannelType::Ro),
133 _ => Err(ChannelParseError{}),
134 }
135 }
136 }
137
138 impl fmt::Display for NAChannelType {
139 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
140 let name = match *self {
141 NAChannelType::C => "C".to_string(),
142 NAChannelType::L => "L".to_string(),
143 NAChannelType::R => "R".to_string(),
144 NAChannelType::Cs => "Cs".to_string(),
145 NAChannelType::Ls => "Ls".to_string(),
146 NAChannelType::Rs => "Rs".to_string(),
147 NAChannelType::Lss => "Lss".to_string(),
148 NAChannelType::Rss => "Rss".to_string(),
149 NAChannelType::LFE => "LFE".to_string(),
150 NAChannelType::Lc => "Lc".to_string(),
151 NAChannelType::Rc => "Rc".to_string(),
152 NAChannelType::Lh => "Lh".to_string(),
153 NAChannelType::Rh => "Rh".to_string(),
154 NAChannelType::Ch => "Ch".to_string(),
155 NAChannelType::LFE2 => "LFE2".to_string(),
156 NAChannelType::Lw => "Lw".to_string(),
157 NAChannelType::Rw => "Rw".to_string(),
158 NAChannelType::Ov => "Ov".to_string(),
159 NAChannelType::Lhs => "Lhs".to_string(),
160 NAChannelType::Rhs => "Rhs".to_string(),
161 NAChannelType::Chs => "Chs".to_string(),
162 NAChannelType::Ll => "Ll".to_string(),
163 NAChannelType::Rl => "Rl".to_string(),
164 NAChannelType::Cl => "Cl".to_string(),
165 NAChannelType::Lt => "Lt".to_string(),
166 NAChannelType::Rt => "Rt".to_string(),
167 NAChannelType::Lo => "Lo".to_string(),
168 NAChannelType::Ro => "Ro".to_string(),
169 };
170 write!(f, "{}", name)
171 }
172 }
173
174 #[derive(Clone,Default)]
175 pub struct NAChannelMap {
176 ids: Vec<NAChannelType>,
177 }
178
179 const MS_CHANNEL_MAP: [NAChannelType; 11] = [
180 NAChannelType::L,
181 NAChannelType::R,
182 NAChannelType::C,
183 NAChannelType::LFE,
184 NAChannelType::Ls,
185 NAChannelType::Rs,
186 NAChannelType::Lss,
187 NAChannelType::Rss,
188 NAChannelType::Cs,
189 NAChannelType::Lc,
190 NAChannelType::Rc,
191 ];
192
193 impl NAChannelMap {
194 pub fn new() -> Self { NAChannelMap { ids: Vec::new() } }
195 pub fn add_channel(&mut self, ch: NAChannelType) {
196 self.ids.push(ch);
197 }
198 pub fn add_channels(&mut self, chs: &[NAChannelType]) {
199 for e in chs.iter() {
200 self.ids.push(*e);
201 }
202 }
203 pub fn num_channels(&self) -> usize {
204 self.ids.len()
205 }
206 pub fn get_channel(&self, idx: usize) -> NAChannelType {
207 self.ids[idx]
208 }
209 pub fn find_channel_id(&self, t: NAChannelType) -> Option<u8> {
210 for i in 0..self.ids.len() {
211 if self.ids[i] as i32 == t as i32 { return Some(i as u8); }
212 }
213 None
214 }
215 pub fn from_ms_mapping(chmap: u32) -> Self {
216 let mut cm = NAChannelMap::new();
217 for (i, ch) in MS_CHANNEL_MAP.iter().enumerate() {
218 if ((chmap >> i) & 1) != 0 {
219 cm.add_channel(*ch);
220 }
221 }
222 cm
223 }
224 }
225
226 impl fmt::Display for NAChannelMap {
227 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
228 let mut map = String::new();
229 for el in self.ids.iter() {
230 if !map.is_empty() { map.push(','); }
231 map.push_str(&*el.to_string());
232 }
233 write!(f, "{}", map)
234 }
235 }
236
237 impl FromStr for NAChannelMap {
238 type Err = ChannelParseError;
239
240 fn from_str(s: &str) -> Result<Self, Self::Err> {
241 let mut chm = NAChannelMap::new();
242 for tok in s.split(',') {
243 chm.add_channel(NAChannelType::from_str(tok)?);
244 }
245 Ok(chm)
246 }
247 }
248
249 #[derive(Debug,Clone,Copy,PartialEq)]
250 pub enum RGBSubmodel {
251 RGB,
252 SRGB,
253 }
254
255 impl fmt::Display for RGBSubmodel {
256 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
257 let name = match *self {
258 RGBSubmodel::RGB => "RGB".to_string(),
259 RGBSubmodel::SRGB => "sRGB".to_string(),
260 };
261 write!(f, "{}", name)
262 }
263 }
264
265 #[derive(Debug,Clone,Copy,PartialEq)]
266 pub enum YUVSubmodel {
267 YCbCr,
268 YIQ,
269 YUVJ,
270 }
271
272 impl fmt::Display for YUVSubmodel {
273 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
274 let name = match *self {
275 YUVSubmodel::YCbCr => "YCbCr".to_string(),
276 YUVSubmodel::YIQ => "YIQ".to_string(),
277 YUVSubmodel::YUVJ => "YUVJ".to_string(),
278 };
279 write!(f, "{}", name)
280 }
281 }
282
283 #[derive(Debug, Clone,Copy,PartialEq)]
284 pub enum ColorModel {
285 RGB(RGBSubmodel),
286 YUV(YUVSubmodel),
287 CMYK,
288 HSV,
289 LAB,
290 XYZ,
291 }
292
293 impl ColorModel {
294 pub fn get_default_components(self) -> usize {
295 match self {
296 ColorModel::CMYK => 4,
297 _ => 3,
298 }
299 }
300 pub fn is_rgb(self) -> bool {
301 match self {
302 ColorModel::RGB(_) => true,
303 _ => false,
304 }
305 }
306 pub fn is_yuv(self) -> bool {
307 match self {
308 ColorModel::YUV(_) => true,
309 _ => false,
310 }
311 }
312 pub fn get_short_name(self) -> &'static str {
313 match self {
314 ColorModel::RGB(_) => "rgb",
315 ColorModel::YUV(_) => "yuv",
316 ColorModel::CMYK => "cmyk",
317 ColorModel::HSV => "hsv",
318 ColorModel::LAB => "lab",
319 ColorModel::XYZ => "xyz",
320 }
321 }
322 }
323
324 impl fmt::Display for ColorModel {
325 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
326 let name = match *self {
327 ColorModel::RGB(fmt) => format!("RGB({})", fmt).to_string(),
328 ColorModel::YUV(fmt) => format!("YUV({})", fmt).to_string(),
329 ColorModel::CMYK => "CMYK".to_string(),
330 ColorModel::HSV => "HSV".to_string(),
331 ColorModel::LAB => "LAB".to_string(),
332 ColorModel::XYZ => "XYZ".to_string(),
333 };
334 write!(f, "{}", name)
335 }
336 }
337
338 #[derive(Clone,Copy,PartialEq)]
339 pub struct NAPixelChromaton {
340 pub h_ss: u8,
341 pub v_ss: u8,
342 pub packed: bool,
343 pub depth: u8,
344 pub shift: u8,
345 pub comp_offs: u8,
346 pub next_elem: u8,
347 }
348
349 pub const FORMATON_FLAG_BE :u32 = 0x01;
350 pub const FORMATON_FLAG_ALPHA :u32 = 0x02;
351 pub const FORMATON_FLAG_PALETTE :u32 = 0x04;
352
353 pub const MAX_CHROMATONS: usize = 5;
354
355 #[derive(Clone,Copy,PartialEq)]
356 pub struct NAPixelFormaton {
357 pub model: ColorModel,
358 pub components: u8,
359 pub comp_info: [Option<NAPixelChromaton>; MAX_CHROMATONS],
360 pub elem_size: u8,
361 pub be: bool,
362 pub alpha: bool,
363 pub palette: bool,
364 }
365
366 macro_rules! chromaton {
367 ($hs: expr, $vs: expr, $pck: expr, $d: expr, $sh: expr, $co: expr, $ne: expr) => ({
368 Some(NAPixelChromaton{ h_ss: $hs, v_ss: $vs, packed: $pck, depth: $d, shift: $sh, comp_offs: $co, next_elem: $ne })
369 });
370 (yuv8; $hs: expr, $vs: expr, $co: expr) => ({
371 Some(NAPixelChromaton{ h_ss: $hs, v_ss: $vs, packed: false, depth: 8, shift: 0, comp_offs: $co, next_elem: 1 })
372 });
373 (packrgb; $d: expr, $s: expr, $co: expr, $ne: expr) => ({
374 Some(NAPixelChromaton{ h_ss: 0, v_ss: 0, packed: true, depth: $d, shift: $s, comp_offs: $co, next_elem: $ne })
375 });
376 (pal8; $co: expr) => ({
377 Some(NAPixelChromaton{ h_ss: 0, v_ss: 0, packed: true, depth: 8, shift: 0, comp_offs: $co, next_elem: 3 })
378 });
379 }
380
381 pub const YUV420_FORMAT: NAPixelFormaton = NAPixelFormaton { model: ColorModel::YUV(YUVSubmodel::YUVJ), components: 3,
382 comp_info: [
383 chromaton!(0, 0, false, 8, 0, 0, 1),
384 chromaton!(yuv8; 1, 1, 1),
385 chromaton!(yuv8; 1, 1, 2),
386 None, None],
387 elem_size: 0, be: false, alpha: false, palette: false };
388
389 pub const YUV410_FORMAT: NAPixelFormaton = NAPixelFormaton { model: ColorModel::YUV(YUVSubmodel::YUVJ), components: 3,
390 comp_info: [
391 chromaton!(0, 0, false, 8, 0, 0, 1),
392 chromaton!(yuv8; 2, 2, 1),
393 chromaton!(yuv8; 2, 2, 2),
394 None, None],
395 elem_size: 0, be: false, alpha: false, palette: false };
396 pub const YUVA410_FORMAT: NAPixelFormaton = NAPixelFormaton { model: ColorModel::YUV(YUVSubmodel::YUVJ), components: 4,
397 comp_info: [
398 chromaton!(0, 0, false, 8, 0, 0, 1),
399 chromaton!(yuv8; 2, 2, 1),
400 chromaton!(yuv8; 2, 2, 2),
401 chromaton!(0, 0, false, 8, 0, 3, 1),
402 None],
403 elem_size: 0, be: false, alpha: true, palette: false };
404
405 pub const PAL8_FORMAT: NAPixelFormaton = NAPixelFormaton { model: ColorModel::RGB(RGBSubmodel::RGB), components: 3,
406 comp_info: [
407 chromaton!(pal8; 0),
408 chromaton!(pal8; 1),
409 chromaton!(pal8; 2),
410 None, None],
411 elem_size: 3, be: false, alpha: false, palette: true };
412
413 pub const RGB565_FORMAT: NAPixelFormaton = NAPixelFormaton { model: ColorModel::RGB(RGBSubmodel::RGB), components: 3,
414 comp_info: [
415 chromaton!(packrgb; 5, 11, 0, 2),
416 chromaton!(packrgb; 6, 5, 0, 2),
417 chromaton!(packrgb; 5, 0, 0, 2),
418 None, None],
419 elem_size: 2, be: false, alpha: false, palette: false };
420
421 pub const RGB24_FORMAT: NAPixelFormaton = NAPixelFormaton { model: ColorModel::RGB(RGBSubmodel::RGB), components: 3,
422 comp_info: [
423 chromaton!(packrgb; 8, 0, 0, 3),
424 chromaton!(packrgb; 8, 0, 1, 3),
425 chromaton!(packrgb; 8, 0, 2, 3),
426 None, None],
427 elem_size: 3, be: false, alpha: false, palette: false };
428
429 impl NAPixelChromaton {
430 pub fn new(h_ss: u8, v_ss: u8, packed: bool, depth: u8, shift: u8, comp_offs: u8, next_elem: u8) -> Self {
431 Self { h_ss, v_ss, packed, depth, shift, comp_offs, next_elem }
432 }
433 pub fn get_subsampling(self) -> (u8, u8) { (self.h_ss, self.v_ss) }
434 pub fn is_packed(self) -> bool { self.packed }
435 pub fn get_depth(self) -> u8 { self.depth }
436 pub fn get_shift(self) -> u8 { self.shift }
437 pub fn get_offset(self) -> u8 { self.comp_offs }
438 pub fn get_step(self) -> u8 { self.next_elem }
439
440 pub fn get_width(self, width: usize) -> usize {
441 (width + ((1 << self.h_ss) - 1)) >> self.h_ss
442 }
443 pub fn get_height(self, height: usize) -> usize {
444 (height + ((1 << self.v_ss) - 1)) >> self.v_ss
445 }
446 pub fn get_linesize(self, width: usize) -> usize {
447 let d = self.depth as usize;
448 if self.packed {
449 (self.get_width(width) * d + d - 1) >> 3
450 } else {
451 self.get_width(width)
452 }
453 }
454 pub fn get_data_size(self, width: usize, height: usize) -> usize {
455 let nh = (height + ((1 << self.v_ss) - 1)) >> self.v_ss;
456 self.get_linesize(width) * nh
457 }
458 }
459
460 impl fmt::Display for NAPixelChromaton {
461 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
462 let pfmt = if self.packed {
463 let mask = ((1 << self.depth) - 1) << self.shift;
464 format!("packed(+{},{:X}, step {})", self.comp_offs, mask, self.next_elem)
465 } else {
466 format!("planar({},{})", self.comp_offs, self.next_elem)
467 };
468 write!(f, "({}x{}, {})", self.h_ss, self.v_ss, pfmt)
469 }
470 }
471
472 impl NAPixelFormaton {
473 pub fn new(model: ColorModel,
474 comp1: Option<NAPixelChromaton>,
475 comp2: Option<NAPixelChromaton>,
476 comp3: Option<NAPixelChromaton>,
477 comp4: Option<NAPixelChromaton>,
478 comp5: Option<NAPixelChromaton>,
479 flags: u32, elem_size: u8) -> Self {
480 let mut chromatons: [Option<NAPixelChromaton>; MAX_CHROMATONS] = [None; MAX_CHROMATONS];
481 let mut ncomp = 0;
482 let be = (flags & FORMATON_FLAG_BE) != 0;
483 let alpha = (flags & FORMATON_FLAG_ALPHA) != 0;
484 let palette = (flags & FORMATON_FLAG_PALETTE) != 0;
485 if let Some(c) = comp1 { chromatons[0] = Some(c); ncomp += 1; }
486 if let Some(c) = comp2 { chromatons[1] = Some(c); ncomp += 1; }
487 if let Some(c) = comp3 { chromatons[2] = Some(c); ncomp += 1; }
488 if let Some(c) = comp4 { chromatons[3] = Some(c); ncomp += 1; }
489 if let Some(c) = comp5 { chromatons[4] = Some(c); ncomp += 1; }
490 NAPixelFormaton { model,
491 components: ncomp,
492 comp_info: chromatons,
493 elem_size,
494 be, alpha, palette }
495 }
496
497 pub fn get_model(&self) -> ColorModel { self.model }
498 pub fn get_num_comp(&self) -> usize { self.components as usize }
499 pub fn get_chromaton(&self, idx: usize) -> Option<NAPixelChromaton> {
500 if idx < self.comp_info.len() { return self.comp_info[idx]; }
501 None
502 }
503 pub fn is_be(self) -> bool { self.be }
504 pub fn has_alpha(self) -> bool { self.alpha }
505 pub fn is_paletted(self) -> bool { self.palette }
506 pub fn get_elem_size(self) -> u8 { self.elem_size }
507 pub fn is_unpacked(&self) -> bool {
508 if self.palette { return false; }
509 for chr in self.comp_info.iter() {
510 if let Some(ref chromaton) = chr {
511 if chromaton.is_packed() { return false; }
512 }
513 }
514 true
515 }
516 pub fn get_max_depth(&self) -> u8 {
517 let mut mdepth = 0;
518 for chr in self.comp_info.iter() {
519 if let Some(ref chromaton) = chr {
520 mdepth = mdepth.max(chromaton.depth);
521 }
522 }
523 mdepth
524 }
525 pub fn get_max_subsampling(&self) -> u8 {
526 let mut ssamp = 0;
527 for chr in self.comp_info.iter() {
528 if let Some(ref chromaton) = chr {
529 let (ss_v, ss_h) = chromaton.get_subsampling();
530 ssamp = ssamp.max(ss_v).max(ss_h);
531 }
532 }
533 ssamp
534 }
535 }
536
537 impl fmt::Display for NAPixelFormaton {
538 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
539 let end = if self.be { "BE" } else { "LE" };
540 let palstr = if self.palette { "palette " } else { "" };
541 let astr = if self.alpha { "alpha " } else { "" };
542 let mut str = format!("Formaton for {} ({}{}elem {} size {}): ", self.model, palstr, astr,end, self.elem_size);
543 for i in 0..self.comp_info.len() {
544 if let Some(chr) = self.comp_info[i] {
545 str = format!("{} {}", str, chr);
546 }
547 }
548 write!(f, "[{}]", str)
549 }
550 }
551
552 #[cfg(test)]
553 mod test {
554 use super::*;
555
556 #[test]
557 fn test_fmt() {
558 println!("{}", SND_S16_FORMAT);
559 println!("{}", SND_U8_FORMAT);
560 println!("{}", SND_F32P_FORMAT);
561 println!("formaton yuv- {}", YUV420_FORMAT);
562 println!("formaton pal- {}", PAL8_FORMAT);
563 println!("formaton rgb565- {}", RGB565_FORMAT);
564 }
565 }