nihav-qt/src/codecs/mace.rs

   1 use nihav_core::codecs::*;
   2 use std::str::FromStr;
   3
   4 const SND_U8P_FORMAT: NASoniton = NASoniton { bits: 8, be: false, packed: false, planar: true, float: false, signed: false };
   5
   6 #[derive(Clone,Copy,Default)]
   7 struct ChannelPredictor {
   8     index:      usize,
   9     prev:       i16,
  10     pprev:      i16,
  11     scale:      i16,
  12     level:      i16,
  13 }
  14
  15 fn clip(val: i32) -> i16 {
  16     if val > 0x7FFF { 0x7FFF }
  17     else if val < -0x8000 { -0x7FFF }
  18     else { val as i16 }
  19 }
  20
  21 fn to_sample(val: i32) -> u8 {
  22     ((val >> 8) + 128).max(0).min(255) as u8
  23 }
  24
  25 impl ChannelPredictor {
  26     fn get_quant(&mut self, idx: usize, middle: bool) -> i16 {
  27         let pred;
  28         let tab_idx = (self.index >> 4) & 0x7F;
  29         if !middle {
  30             if idx < 4 {
  31                 pred = QUANT_TAB0[tab_idx][idx];
  32             } else {
  33                 pred = -1 - QUANT_TAB0[tab_idx][8 - 1 - idx];
  34             }
  35             self.index -= self.index >> 5;
  36             self.index = ((self.index as isize) + (STEP_TAB0[idx] as isize)).max(0) as usize;
  37         } else {
  38             if idx < 2 {
  39                 pred = QUANT_TAB1[tab_idx][idx];
  40             } else {
  41                 pred = -1 - QUANT_TAB1[tab_idx][4 - 1 - idx];
  42             }
  43             self.index -= self.index >> 5;
  44             self.index = ((self.index as isize) + (STEP_TAB1[idx] as isize)).max(0) as usize;
  45         }
  46
  47         pred
  48     }
  49     fn pred_mace3(&mut self, idx: usize, middle: bool) -> u8 {
  50         let pred = self.get_quant(idx, middle);
  51         let cur = clip(i32::from(pred) + i32::from(self.level));
  52         self.level = cur - (cur >> 3);
  53         to_sample(i32::from(cur))
  54     }
  55     fn pred_mace6(&mut self, idx: usize, middle: bool) -> (u8, u8) {
  56         let pred = self.get_quant(idx, middle);
  57         if (self.prev ^ pred) >= 0 {
  58             self.scale = self.scale.saturating_add(506);
  59         } else {
  60             self.scale = clip(i32::from(self.scale) - 314);
  61         }
  62         let mut cur = clip(i32::from(pred) + i32::from(self.level));
  63
  64         self.level = ((i32::from(cur) * i32::from(self.scale)) >> 15) as i16;
  65         cur >>= 1;
  66
  67         let diff = (i32::from(self.pprev) - i32::from(cur)) >> 2;
  68         let s0 = i32::from(self.pprev) + i32::from(self.prev) - diff;
  69         let s1 = i32::from(self.prev) + i32::from(cur) + diff;
  70
  71         self.pprev = self.prev;
  72         self.prev  = cur;
  73
  74         (to_sample(s0), to_sample(s1))
  75     }
  76 }
  77
  78 struct MaceDecoder {
  79     ainfo:      NAAudioInfo,
  80     chmap:      NAChannelMap,
  81     ch_pred:    [ChannelPredictor; 2],
  82     is_mace6:   bool,
  83 }
  84
  85 impl MaceDecoder {
  86     fn new(is_mace6: bool) -> Self {
  87         Self {
  88             ainfo:      NAAudioInfo::new(0, 1, SND_U8P_FORMAT, 1),
  89             chmap:      NAChannelMap::new(),
  90             ch_pred:    [ChannelPredictor::default(); 2],
  91             is_mace6,
  92         }
  93     }
  94 }
  95
  96 impl NADecoder for MaceDecoder {
  97     fn init(&mut self, _supp: &mut NADecoderSupport, info: NACodecInfoRef) -> DecoderResult<()> {
  98         if let NACodecTypeInfo::Audio(ainfo) = info.get_properties() {
  99             let channels = ainfo.get_channels() as usize;
 100             validate!(channels == 2 || channels == 1);
 101             self.ainfo = NAAudioInfo::new(ainfo.get_sample_rate(), channels as u8, SND_U8P_FORMAT, 1);
 102             self.chmap = NAChannelMap::from_str(if channels == 1 { "C" } else { "L,R" }).unwrap();
 103             Ok(())
 104         } else {
 105             Err(DecoderError::InvalidData)
 106         }
 107     }
 108     fn decode(&mut self, _supp: &mut NADecoderSupport, pkt: &NAPacket) -> DecoderResult<NAFrameRef> {
 109         let info = pkt.get_stream().get_info();
 110         if let NACodecTypeInfo::Audio(_) = info.get_properties() {
 111             let pktbuf = pkt.get_buffer();
 112             let channels = self.chmap.num_channels();
 113             let nsamples = pktbuf.len() * (if self.is_mace6 { 6 } else { 3 }) / channels;
 114             let abuf = alloc_audio_buffer(self.ainfo, nsamples, self.chmap.clone())?;
 115             let mut adata = abuf.get_abuf_u8().unwrap();
 116             let mut off = [adata.get_offset(0), adata.get_offset(1)];
 117             let dst = adata.get_data_mut().unwrap();
 118
 119             if !self.is_mace6 {
 120                 for src in pktbuf.chunks(channels) {
 121                     for ch in 0..channels {
 122                         let val = src[ch];
 123                         let idx0 = val & 7;
 124                         let idx1 = (val >> 3) & 3;
 125                         let idx2 = val >> 5;
 126                         dst[off[ch]]     = self.ch_pred[ch].pred_mace3(idx0 as usize, false);
 127                         dst[off[ch] + 1] = self.ch_pred[ch].pred_mace3(idx1 as usize, true);
 128                         dst[off[ch] + 2] = self.ch_pred[ch].pred_mace3(idx2 as usize, false);
 129                         off[ch] += 3;
 130                     }
 131                 }
 132             } else {
 133                 for src in pktbuf.chunks(channels) {
 134                     for ch in 0..channels {
 135                         let val = src[ch];
 136                         let idx0 = val >> 5;
 137                         let idx1 = (val >> 3) & 3;
 138                         let idx2 = val & 7;
 139                         let (s0, s1) = self.ch_pred[ch].pred_mace6(idx0 as usize, false);
 140                         dst[off[ch]]     = s0;
 141                         dst[off[ch] + 1] = s1;
 142                         let (s0, s1) = self.ch_pred[ch].pred_mace6(idx1 as usize, true);
 143                         dst[off[ch] + 2] = s0;
 144                         dst[off[ch] + 3] = s1;
 145                         let (s0, s1) = self.ch_pred[ch].pred_mace6(idx2 as usize, false);
 146                         dst[off[ch] + 4] = s0;
 147                         dst[off[ch] + 5] = s1;
 148                         off[ch] += 6;
 149                     }
 150                 }
 151             }
 152
 153             let mut frm = NAFrame::new_from_pkt(pkt, info.replace_info(NACodecTypeInfo::Audio(self.ainfo)), abuf);
 154             frm.set_duration(Some(nsamples as u64));
 155             frm.set_keyframe(false);
 156             Ok(frm.into_ref())
 157         } else {
 158             Err(DecoderError::InvalidData)
 159         }
 160     }
 161     fn flush(&mut self) {
 162     }
 163 }
 164
 165 impl NAOptionHandler for MaceDecoder {
 166     fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
 167     fn set_options(&mut self, _options: &[NAOption]) { }
 168     fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
 169 }
 170
 171 pub fn get_decoder_3() -> Box<dyn NADecoder + Send> {
 172     Box::new(MaceDecoder::new(false))
 173 }
 174
 175 pub fn get_decoder_6() -> Box<dyn NADecoder + Send> {
 176     Box::new(MaceDecoder::new(true))
 177 }
 178
 179 #[cfg(test)]
 180 mod test {
 181     use nihav_core::codecs::RegisteredDecoders;
 182     use nihav_core::demuxers::RegisteredDemuxers;
 183     use nihav_codec_support::test::dec_video::*;
 184     use crate::qt_register_all_codecs;
 185     use nihav_commonfmt::generic_register_all_demuxers;
 186     #[test]
 187     fn test_mace3() {
 188         let mut dmx_reg = RegisteredDemuxers::new();
 189         generic_register_all_demuxers(&mut dmx_reg);
 190         let mut dec_reg = RegisteredDecoders::new();
 191         qt_register_all_codecs(&mut dec_reg);
 192
 193         test_decoding("mov", "mace-3", "assets/QT/surge-1-8-MAC3.mov", None, &dmx_reg, &dec_reg,
 194                       ExpectedTestResult::MD5([0x2df88db3, 0xa6167019, 0x6d4c64e7, 0xc89da2a5]));
 195     }
 196     #[test]
 197     fn test_mace6() {
 198         let mut dmx_reg = RegisteredDemuxers::new();
 199         generic_register_all_demuxers(&mut dmx_reg);
 200         let mut dec_reg = RegisteredDecoders::new();
 201         qt_register_all_codecs(&mut dec_reg);
 202
 203         test_decoding("mov", "mace-6", "assets/QT/surge-1-8-MAC6.mov", None, &dmx_reg, &dec_reg,
 204                       ExpectedTestResult::MD5([0xc32857e2, 0xc1ea1ce8, 0x2d77dacf, 0xef504f1f]));
 205     }
 206 }
 207
 208 const STEP_TAB0: [i16; 8] = [ -13, 8, 76, 222, 222, 76, 8, -13 ];
 209 const STEP_TAB1: [i16; 4] = [ -18, 140, 140, -18 ];
 210 const QUANT_TAB0: [[i16; 4]; 128] = [
 211     [    37,   116,   206,   330 ], [    39,   121,   216,   346 ],
 212     [    41,   127,   225,   361 ], [    42,   132,   235,   377 ],
 213     [    44,   137,   245,   392 ], [    46,   144,   256,   410 ],
 214     [    48,   150,   267,   428 ], [    51,   157,   280,   449 ],
 215     [    53,   165,   293,   470 ], [    55,   172,   306,   490 ],
 216     [    58,   179,   319,   511 ], [    60,   187,   333,   534 ],
 217     [    63,   195,   348,   557 ], [    66,   205,   364,   583 ],
 218     [    69,   214,   380,   609 ], [    72,   223,   396,   635 ],
 219     [    75,   233,   414,   663 ], [    79,   244,   433,   694 ],
 220     [    82,   254,   453,   725 ], [    86,   265,   472,   756 ],
 221     [    90,   278,   495,   792 ], [    94,   290,   516,   826 ],
 222     [    98,   303,   538,   862 ], [   102,   316,   562,   901 ],
 223     [   107,   331,   588,   942 ], [   112,   345,   614,   983 ],
 224     [   117,   361,   641,  1027 ], [   122,   377,   670,  1074 ],
 225     [   127,   394,   701,  1123 ], [   133,   411,   732,  1172 ],
 226     [   139,   430,   764,  1224 ], [   145,   449,   799,  1280 ],
 227     [   152,   469,   835,  1337 ], [   159,   490,   872,  1397 ],
 228     [   166,   512,   911,  1459 ], [   173,   535,   951,  1523 ],
 229     [   181,   558,   993,  1590 ], [   189,   584,  1038,  1663 ],
 230     [   197,   610,  1085,  1738 ], [   206,   637,  1133,  1815 ],
 231     [   215,   665,  1183,  1895 ], [   225,   695,  1237,  1980 ],
 232     [   235,   726,  1291,  2068 ], [   246,   759,  1349,  2161 ],
 233     [   257,   792,  1409,  2257 ], [   268,   828,  1472,  2357 ],
 234     [   280,   865,  1538,  2463 ], [   293,   903,  1606,  2572 ],
 235     [   306,   944,  1678,  2688 ], [   319,   986,  1753,  2807 ],
 236     [   334,  1030,  1832,  2933 ], [   349,  1076,  1914,  3065 ],
 237     [   364,  1124,  1999,  3202 ], [   380,  1174,  2088,  3344 ],
 238     [   398,  1227,  2182,  3494 ], [   415,  1281,  2278,  3649 ],
 239     [   434,  1339,  2380,  3811 ], [   453,  1398,  2486,  3982 ],
 240     [   473,  1461,  2598,  4160 ], [   495,  1526,  2714,  4346 ],
 241     [   517,  1594,  2835,  4540 ], [   540,  1665,  2961,  4741 ],
 242     [   564,  1740,  3093,  4953 ], [   589,  1818,  3232,  5175 ],
 243     [   615,  1898,  3375,  5405 ], [   643,  1984,  3527,  5647 ],
 244     [   671,  2072,  3683,  5898 ], [   701,  2164,  3848,  6161 ],
 245     [   733,  2261,  4020,  6438 ], [   766,  2362,  4199,  6724 ],
 246     [   800,  2467,  4386,  7024 ], [   836,  2578,  4583,  7339 ],
 247     [   873,  2692,  4786,  7664 ], [   912,  2813,  5001,  8008 ],
 248     [   952,  2938,  5223,  8364 ], [   995,  3070,  5457,  8739 ],
 249     [  1039,  3207,  5701,  9129 ], [  1086,  3350,  5956,  9537 ],
 250     [  1134,  3499,  6220,  9960 ], [  1185,  3655,  6497, 10404 ],
 251     [  1238,  3818,  6788, 10869 ], [  1293,  3989,  7091, 11355 ],
 252     [  1351,  4166,  7407, 11861 ], [  1411,  4352,  7738, 12390 ],
 253     [  1474,  4547,  8084, 12946 ], [  1540,  4750,  8444, 13522 ],
 254     [  1609,  4962,  8821, 14126 ], [  1680,  5183,  9215, 14756 ],
 255     [  1756,  5415,  9626, 15415 ], [  1834,  5657, 10057, 16104 ],
 256     [  1916,  5909, 10505, 16822 ], [  2001,  6173, 10975, 17574 ],
 257     [  2091,  6448, 11463, 18356 ], [  2184,  6736, 11974, 19175 ],
 258     [  2282,  7037, 12510, 20032 ], [  2383,  7351, 13068, 20926 ],
 259     [  2490,  7679, 13652, 21861 ], [  2601,  8021, 14260, 22834 ],
 260     [  2717,  8380, 14897, 23854 ], [  2838,  8753, 15561, 24918 ],
 261     [  2965,  9144, 16256, 26031 ], [  3097,  9553, 16982, 27193 ],
 262     [  3236,  9979, 17740, 28407 ], [  3380, 10424, 18532, 29675 ],
 263     [  3531, 10890, 19359, 31000 ], [  3688, 11375, 20222, 32382 ],
 264     [  3853, 11883, 21125, 32767 ], [  4025, 12414, 22069, 32767 ],
 265     [  4205, 12967, 23053, 32767 ], [  4392, 13546, 24082, 32767 ],
 266     [  4589, 14151, 25157, 32767 ], [  4793, 14783, 26280, 32767 ],
 267     [  5007, 15442, 27452, 32767 ], [  5231, 16132, 28678, 32767 ],
 268     [  5464, 16851, 29957, 32767 ], [  5708, 17603, 31294, 32767 ],
 269     [  5963, 18389, 32691, 32767 ], [  6229, 19210, 32767, 32767 ],
 270     [  6507, 20067, 32767, 32767 ], [  6797, 20963, 32767, 32767 ],
 271     [  7101, 21899, 32767, 32767 ], [  7418, 22876, 32767, 32767 ],
 272     [  7749, 23897, 32767, 32767 ], [  8095, 24964, 32767, 32767 ],
 273     [  8456, 26078, 32767, 32767 ], [  8833, 27242, 32767, 32767 ],
 274     [  9228, 28457, 32767, 32767 ], [  9639, 29727, 32767, 32767 ]
 275 ];
 276 const QUANT_TAB1: [[i16; 2]; 128] = [
 277     [    64,   216 ], [    67,   226 ], [    70,   236 ], [    74,   246 ],
 278     [    77,   257 ], [    80,   268 ], [    84,   280 ], [    88,   294 ],
 279     [    92,   307 ], [    96,   321 ], [   100,   334 ], [   104,   350 ],
 280     [   109,   365 ], [   114,   382 ], [   119,   399 ], [   124,   416 ],
 281     [   130,   434 ], [   136,   454 ], [   142,   475 ], [   148,   495 ],
 282     [   155,   519 ], [   162,   541 ], [   169,   564 ], [   176,   590 ],
 283     [   185,   617 ], [   193,   644 ], [   201,   673 ], [   210,   703 ],
 284     [   220,   735 ], [   230,   767 ], [   240,   801 ], [   251,   838 ],
 285     [   262,   876 ], [   274,   914 ], [   286,   955 ], [   299,   997 ],
 286     [   312,  1041 ], [   326,  1089 ], [   341,  1138 ], [   356,  1188 ],
 287     [   372,  1241 ], [   388,  1297 ], [   406,  1354 ], [   424,  1415 ],
 288     [   443,  1478 ], [   462,  1544 ], [   483,  1613 ], [   505,  1684 ],
 289     [   527,  1760 ], [   551,  1838 ], [   576,  1921 ], [   601,  2007 ],
 290     [   628,  2097 ], [   656,  2190 ], [   686,  2288 ], [   716,  2389 ],
 291     [   748,  2496 ], [   781,  2607 ], [   816,  2724 ], [   853,  2846 ],
 292     [   891,  2973 ], [   930,  3104 ], [   972,  3243 ], [  1016,  3389 ],
 293     [  1061,  3539 ], [  1108,  3698 ], [  1158,  3862 ], [  1209,  4035 ],
 294     [  1264,  4216 ], [  1320,  4403 ], [  1379,  4599 ], [  1441,  4806 ],
 295     [  1505,  5019 ], [  1572,  5244 ], [  1642,  5477 ], [  1715,  5722 ],
 296     [  1792,  5978 ], [  1872,  6245 ], [  1955,  6522 ], [  2043,  6813 ],
 297     [  2134,  7118 ], [  2229,  7436 ], [  2329,  7767 ], [  2432,  8114 ],
 298     [  2541,  8477 ], [  2655,  8854 ], [  2773,  9250 ], [  2897,  9663 ],
 299     [  3026, 10094 ], [  3162, 10546 ], [  3303, 11016 ], [  3450, 11508 ],
 300     [  3604, 12020 ], [  3765, 12556 ], [  3933, 13118 ], [  4108, 13703 ],
 301     [  4292, 14315 ], [  4483, 14953 ], [  4683, 15621 ], [  4892, 16318 ],
 302     [  5111, 17046 ], [  5339, 17807 ], [  5577, 18602 ], [  5826, 19433 ],
 303     [  6086, 20300 ], [  6358, 21205 ], [  6642, 22152 ], [  6938, 23141 ],
 304     [  7248, 24173 ], [  7571, 25252 ], [  7909, 26380 ], [  8262, 27557 ],
 305     [  8631, 28786 ], [  9016, 30072 ], [  9419, 31413 ], [  9839, 32767 ],
 306     [ 10278, 32767 ], [ 10737, 32767 ], [ 11216, 32767 ], [ 11717, 32767 ],
 307     [ 12240, 32767 ], [ 12786, 32767 ], [ 13356, 32767 ], [ 13953, 32767 ],
 308     [ 14576, 32767 ], [ 15226, 32767 ], [ 15906, 32767 ], [ 16615, 32767 ]
 309 ];