You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

338 lines
13KB

  1. /*
  2. * Copyright (c) 2015 Kieran Kunhya
  3. *
  4. * This file is part of Libav.
  5. *
  6. * Libav is free software; you can redistribute it and/or
  7. * modify it under the terms of the GNU Lesser General Public
  8. * License as published by the Free Software Foundation; either
  9. * version 2.1 of the License, or (at your option) any later version.
  10. *
  11. * Libav is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14. * Lesser General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU Lesser General Public
  17. * License along with Libav; if not, write to the Free Software
  18. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19. */
  20. #include <stdint.h>
  21. #include "libavutil/attributes.h"
  22. #include "cfhd.h"
  23. #define NB_VLC_TABLE_9 (71 + 3)
  24. #define NB_VLC_TABLE_18 (263 + 1)
  25. static const uint32_t table_9_vlc_bits[NB_VLC_TABLE_9] = {
  26. 0, 0x2, 0xc, 0x1a,
  27. 0x1d, 0x1e, 0x39, 0x3e,
  28. 0x37, 0x7e, 0x6c, 0xe2,
  29. 0xfe, 0xdb, 0xe0, 0x1c3,
  30. 0x1c6, 0x1ff, 0x1fe, 0x1b5,
  31. 0x369, 0x385, 0x71d, 0x6d0,
  32. 0x708, 0x71f, 0xe3d, 0xe39,
  33. 0xe13, 0xe12, 0x1c71, 0x1b45,
  34. 0x1b47, 0x3689, 0x38f2, 0x38e1,
  35. 0x38e0, 0x38f1, 0x3688, 0x6d1b,
  36. 0x71e0, 0x6d19, 0x71e7, 0xe3cd,
  37. 0xda35, 0xda30, 0xe3c3, 0x1b469,
  38. 0x1b462, 0x1c798, 0x1b463, 0x1c799,
  39. 0x38f08, 0x38f09, 0x38f0a, 0x6d1a0,
  40. 0x6d1a3, 0x6d1a1, 0xda345, 0xda344,
  41. 0xe3c2d, 0xe3c2f, 0xe3c2e, 0x38f0b2,
  42. 0x71e160, 0x71e162, 0x71e166, 0x71e161,
  43. 0xe3c2ce, 0xe3c2c6, 0xe3c2c7, 0x1C7859E,
  44. 0x38F0B3F, 0x38F0B3E,
  45. };
  46. static const uint8_t table_9_vlc_len[NB_VLC_TABLE_9] = {
  47. 1, 2, 4, 5, 5, 5, 6, 6,
  48. 6, 7, 7, 8, 8, 8, 8, 9,
  49. 9, 9, 9, 9, 10, 10, 11, 11,
  50. 11, 11, 12, 12, 12, 12, 13, 13,
  51. 13, 14, 14, 14, 14, 14, 14, 15,
  52. 15, 15, 15, 16, 16, 16, 16, 17,
  53. 17, 17, 17, 17, 18, 18, 18, 19,
  54. 19, 19, 20, 20, 20, 20, 20, 22,
  55. 23, 23, 23, 23, 24, 24, 24, 25,
  56. 26, 26,
  57. };
  58. static const uint16_t table_9_vlc_run[NB_VLC_TABLE_9] = {
  59. 1, 1, 1, 1, 12, 1, 32, 160,
  60. 1, 1, 1, 320, 1, 1, 80, 120,
  61. 1, 1, 100, 1, 1, 1, 1, 1,
  62. 1, 1, 1, 1, 1, 1, 1, 1,
  63. 1, 1, 1, 1, 1, 1, 1, 1,
  64. 1, 1, 1, 1, 1, 1, 1, 1,
  65. 1, 1, 1, 1, 1, 1, 1, 1,
  66. 1, 1, 1, 1, 1, 1, 1, 1,
  67. 1, 1, 1, 1, 1, 1, 1, 1,
  68. 1, 1
  69. };
  70. static const uint8_t table_9_vlc_level[NB_VLC_TABLE_9] = {
  71. 0, 1, 2, 3, 0, 4, 0, 0,
  72. 5, 7, 6, 0, 9, 8, 0, 0,
  73. 11, 12, 0, 10, 13, 14, 17, 15,
  74. 16, 18, 22, 21, 20, 19, 25, 23,
  75. 24, 27, 31, 29, 28, 30, 26, 33,
  76. 34, 32, 35, 39, 37, 36, 38, 42,
  77. 40, 43, 41, 44, 45, 46, 47, 48,
  78. 50, 49, 52, 51, 53, 55, 54, 56,
  79. 57, 59, 60, 58, 61, 62, 63, 64,
  80. 64, 64,
  81. };
  82. static const uint32_t table_18_vlc_bits[NB_VLC_TABLE_18] = {
  83. 0, 0x2, 0x7, 0x19,
  84. 0x30, 0x36, 0x6f, 0x63,
  85. 0x69, 0x6b, 0xd1, 0xd4,
  86. 0xdc, 0x189, 0x18a, 0x1a0,
  87. 0x1ab, 0x377, 0x310, 0x316,
  88. 0x343, 0x354, 0x375, 0x623,
  89. 0x684, 0x685, 0x6ab, 0x6ec,
  90. 0xddb, 0xc5c, 0xc5e, 0xc44,
  91. 0xd55, 0xdd1, 0xdd3, 0x1bb5,
  92. 0x188b, 0x18bb, 0x18bf, 0x1aa8,
  93. 0x1ba0, 0x1ba5, 0x1ba4, 0x3115,
  94. 0x3175, 0x317d, 0x3553, 0x3768,
  95. 0x6e87, 0x6ed3, 0x62e8, 0x62f8,
  96. 0x6228, 0x6aa4, 0x6e85, 0xc453,
  97. 0xc5d3, 0xc5f3, 0xdda4, 0xdd08,
  98. 0xdd0c, 0x1bb4b, 0x1bb4a, 0x18ba5,
  99. 0x18be5, 0x1aa95, 0x1aa97, 0x188a4,
  100. 0x1ba13, 0x31748, 0x317c8, 0x35528,
  101. 0x3552c, 0x37424, 0x37434, 0x37436,
  102. 0x62294, 0x62e92, 0x62f92, 0x6aa52,
  103. 0x6aa5a, 0x6e86a, 0x6e86e, 0x6e84a,
  104. 0xc452a, 0xc5d27, 0xc5f26, 0xd54a6,
  105. 0xd54b6, 0xdd096, 0xdd0d6, 0xdd0de,
  106. 0x188a56, 0x18ba4d, 0x18be4e, 0x18be4f,
  107. 0x1aa96e, 0x1ba12e, 0x1ba12f, 0x1ba1af,
  108. 0x1ba1bf, 0x37435d, 0x37437d, 0x317498,
  109. 0x35529c, 0x35529d, 0x3552de, 0x3552df,
  110. 0x62e933, 0x62295d, 0x6aa53d, 0x6aa53f,
  111. 0x6aa53e, 0x6e86b9, 0x6e86f8, 0xd54a79,
  112. 0xc5d265, 0xc452b8, 0xdd0d71, 0xd54a78,
  113. 0xdd0d70, 0xdd0df2, 0xdd0df3, 0x188a5f6,
  114. 0x188a5f5, 0x188a5f4, 0x188a5f3, 0x188a5f2,
  115. 0x188a5f1, 0x188a5f0, 0x188a5ef, 0x188a5ee,
  116. 0x188a5ed, 0x188a5aa, 0x188a5e3, 0x188a5df,
  117. 0x188a589, 0x188a5dd, 0x188a578, 0x188a5e0,
  118. 0x188a588, 0x188a5d6, 0x188a5db, 0x188a5e1,
  119. 0x188a587, 0x188a59a, 0x188a5c4, 0x188a5ec,
  120. 0x188a586, 0x188a573, 0x188a59c, 0x188a5c8,
  121. 0x188a5fb, 0x188a5a1, 0x188a5eb, 0x188a5a8,
  122. 0x188a584, 0x188a5d2, 0x188a599, 0x188a598,
  123. 0x188a583, 0x18ba4c9, 0x188a5d0, 0x188a594,
  124. 0x188a582, 0x188a5cb, 0x188a5d8, 0x188a5e7,
  125. 0x188a581, 0x188a5ea, 0x188a5a9, 0x188a5a6,
  126. 0x188a580, 0x188a5a0, 0x188a59d, 0x188a5c3,
  127. 0x188a57f, 0x188a5c0, 0x188a5de, 0x188a5d4,
  128. 0x188a57e, 0x188a5c2, 0x188a592, 0x188a5cd,
  129. 0x188a57d, 0x188a5a3, 0x188a5e8, 0x188a5a2,
  130. 0x188a57c, 0x188a58e, 0x188a5b3, 0x188a5b2,
  131. 0x188a5b1, 0x188a5b0, 0x188a5af, 0x188a5ae,
  132. 0x188a5ad, 0x188a5ac, 0x188a5ab, 0x188a5da,
  133. 0x188a5e4, 0x188a5e5, 0x188a5d9, 0x188a5b5,
  134. 0x188a5bc, 0x188a5bd, 0x188a5e9, 0x188a5cc,
  135. 0x188a585, 0x188a5d3, 0x188a5e2, 0x188a595,
  136. 0x188a596, 0x188a5b8, 0x188a590, 0x188a5c9,
  137. 0x188a5a4, 0x188a5e6, 0x188a5a5, 0x188a5ce,
  138. 0x188a5bf, 0x188a572, 0x188a59b, 0x188a5be,
  139. 0x188a5c7, 0x188a5ca, 0x188a5d5, 0x188a57b,
  140. 0x188a58d, 0x188a58c, 0x188a58b, 0x188a58a,
  141. 0x18ba4c8, 0x188a5c5, 0x188a5fa, 0x188a5bb,
  142. 0x188a5c1, 0x188a5cf, 0x188a5b9, 0x188a5b6,
  143. 0x188a597, 0x188a5fe, 0x188a5d7, 0x188a5ba,
  144. 0x188a591, 0x188a5c6, 0x188a5dc, 0x188a57a,
  145. 0x188a59f, 0x188a5f9, 0x188a5b4, 0x188a5a7,
  146. 0x188a58f, 0x188a5fd, 0x188a5b7, 0x188a593,
  147. 0x188a59e, 0x188a5f8, 0x188a5ff, 0x188a5fc,
  148. 0x188a579, 0x188a5f7, 0x3114ba2, 0x3114ba3,
  149. };
  150. static const uint8_t table_18_vlc_len[NB_VLC_TABLE_18] = {
  151. 1, 2, 3, 5, 6, 6, 7, 7,
  152. 7, 7, 8, 8, 8, 9, 9, 9,
  153. 9, 10, 10, 10, 10, 10, 10, 11,
  154. 11, 11, 11, 11, 12, 12, 12, 12,
  155. 12, 12, 12, 13, 13, 13, 13, 13,
  156. 13, 13, 13, 14, 14, 14, 14, 14,
  157. 15, 15, 15, 15, 15, 15, 15, 16,
  158. 16, 16, 16, 16, 16, 17, 17, 17,
  159. 17, 17, 17, 17, 17, 18, 18, 18,
  160. 18, 18, 18, 18, 19, 19, 19, 19,
  161. 19, 19, 19, 19, 20, 20, 20, 20,
  162. 20, 20, 20, 20, 21, 21, 21, 21,
  163. 21, 21, 21, 21, 21, 22, 22, 22,
  164. 22, 22, 22, 22, 23, 23, 23, 23,
  165. 23, 23, 23, 24, 24, 24, 24, 24,
  166. 24, 24, 24, 25, 25, 25, 25, 25,
  167. 25, 25, 25, 25, 25, 25, 25, 25,
  168. 25, 25, 25, 25, 25, 25, 25, 25,
  169. 25, 25, 25, 25, 25, 25, 25, 25,
  170. 25, 25, 25, 25, 25, 25, 25, 25,
  171. 25, 25, 25, 25, 25, 25, 25, 25,
  172. 25, 25, 25, 25, 25, 25, 25, 25,
  173. 25, 25, 25, 25, 25, 25, 25, 25,
  174. 25, 25, 25, 25, 25, 25, 25, 25,
  175. 25, 25, 25, 25, 25, 25, 25, 25,
  176. 25, 25, 25, 25, 25, 25, 25, 25,
  177. 25, 25, 25, 25, 25, 25, 25, 25,
  178. 25, 25, 25, 25, 25, 25, 25, 25,
  179. 25, 25, 25, 25, 25, 25, 25, 25,
  180. 25, 25, 25, 25, 25, 25, 25, 25,
  181. 25, 25, 25, 25, 25, 25, 25, 25,
  182. 25, 25, 25, 25, 25, 25, 25, 25,
  183. 25, 25, 25, 25, 25, 25, 26, 26,
  184. };
  185. static const uint16_t table_18_vlc_run[NB_VLC_TABLE_18] = {
  186. 1, 1, 1, 1, 1, 1, 1, 1,
  187. 12, 1, 20, 1, 1, 1, 32, 1,
  188. 1, 1, 1, 1, 60, 1, 1, 1,
  189. 1, 100, 1, 1, 1, 1, 1, 1,
  190. 1, 1, 1, 1, 1, 1, 180, 1,
  191. 1, 320, 1, 1, 1, 1, 1, 1,
  192. 1, 1, 1, 1, 1, 1, 1, 1,
  193. 1, 1, 1, 1, 1, 1, 1, 1,
  194. 1, 1, 1, 1, 1, 1, 1, 1,
  195. 1, 1, 1, 1, 1, 1, 1, 1,
  196. 1, 1, 1, 1, 1, 1, 1, 1,
  197. 1, 1, 1, 1, 1, 1, 1, 1,
  198. 1, 1, 1, 1, 1, 1, 1, 1,
  199. 1, 1, 1, 1, 1, 1, 1, 1,
  200. 1, 1, 1, 1, 1, 1, 1, 1,
  201. 1, 1, 1, 1, 1, 1, 1, 1,
  202. 1, 1, 1, 1, 1, 1, 1, 1,
  203. 1, 1, 1, 1, 1, 1, 1, 1,
  204. 1, 1, 1, 1, 1, 1, 1, 1,
  205. 1, 1, 1, 1, 1, 1, 1, 1,
  206. 1, 1, 1, 1, 1, 1, 1, 1,
  207. 1, 1, 1, 1, 1, 1, 1, 1,
  208. 1, 1, 1, 1, 1, 1, 1, 1,
  209. 1, 1, 1, 1, 1, 1, 1, 1,
  210. 1, 1, 1, 1, 1, 1, 1, 1,
  211. 1, 1, 1, 1, 1, 1, 1, 1,
  212. 1, 1, 1, 1, 1, 1, 1, 1,
  213. 1, 1, 1, 1, 1, 1, 1, 1,
  214. 1, 1, 1, 1, 1, 1, 1, 1,
  215. 1, 1, 1, 1, 1, 1, 1, 1,
  216. 1, 1, 1, 1, 1, 1, 1, 1,
  217. 1, 1, 1, 1, 1, 1, 1, 1,
  218. 1, 1, 1, 1, 1, 1, 1, 2,
  219. };
  220. static const uint8_t table_18_vlc_level[NB_VLC_TABLE_18] = {
  221. 0, 1, 2, 3, 4, 5, 8, 6,
  222. 0, 7, 0, 9, 10, 11, 0, 12,
  223. 13, 18, 14, 15, 0, 16, 17, 19,
  224. 20, 0, 21, 22, 29, 24, 25, 23,
  225. 26, 27, 28, 35, 30, 31, 0, 32,
  226. 33, 0, 34, 36, 37, 38, 39, 40,
  227. 46, 47, 42, 43, 41, 44, 45, 48,
  228. 49, 50, 53, 51, 52, 61, 60, 55,
  229. 56, 57, 58, 54, 59, 62, 63, 64,
  230. 65, 66, 67, 68, 69, 70, 71, 72,
  231. 73, 75, 76, 74, 77, 78, 79, 80,
  232. 81, 82, 83, 84, 85, 86, 87, 88,
  233. 89, 90, 91, 92, 93, 99, 100, 94,
  234. 95, 96, 97, 98, 102, 101, 103, 105,
  235. 104, 106, 107, 111, 109, 108, 113, 110,
  236. 112, 114, 115, 225, 189, 188, 203, 202,
  237. 197, 207, 169, 223, 159, 235, 152, 192,
  238. 179, 201, 172, 149, 178, 120, 219, 150,
  239. 127, 211, 125, 158, 247, 238, 163, 228,
  240. 183, 217, 168, 122, 128, 249, 187, 186,
  241. 136, 181, 255, 230, 135, 233, 222, 145,
  242. 134, 167, 248, 209, 243, 216, 164, 140,
  243. 157, 239, 191, 251, 156, 139, 242, 133,
  244. 162, 213, 165, 212, 227, 198, 236, 234,
  245. 117, 215, 124, 123, 254, 253, 148, 218,
  246. 146, 147, 224, 143, 184, 185, 166, 132,
  247. 129, 250, 151, 119, 193, 176, 245, 229,
  248. 206, 144, 208, 137, 241, 237, 190, 240,
  249. 131, 232, 252, 171, 205, 204, 118, 214,
  250. 180, 126, 182, 175, 141, 138, 177, 153,
  251. 194, 160, 121, 174, 246, 130, 200, 170,
  252. 221, 196, 142, 210, 199, 155, 154, 244,
  253. 220, 195, 161, 231, 173, 226, 116, 255,
  254. };
  255. static int init_vlc_signed(VLC *vlc, CFHD_RL_VLC_ELEM table_rl_vlc[],
  256. unsigned bound,
  257. const uint32_t table_vlc_bits[],
  258. const uint8_t table_vlc_len[],
  259. const uint16_t table_vlc_run[],
  260. const uint8_t table_vlc_level[])
  261. {
  262. uint32_t vlc_bits[NB_VLC_TABLE_18 * 2];
  263. uint8_t vlc_len[NB_VLC_TABLE_18 * 2];
  264. uint16_t vlc_run[NB_VLC_TABLE_18 * 2];
  265. int16_t vlc_level[NB_VLC_TABLE_18 * 2];
  266. unsigned i, j;
  267. int ret;
  268. for (i = 0, j = 0; i < bound; i++, j++) {
  269. vlc_bits[j] = table_vlc_bits[i];
  270. vlc_len[j] = table_vlc_len[i];
  271. vlc_run[j] = table_vlc_run[i];
  272. vlc_level[j] = table_vlc_level[i];
  273. /* Don't include the zero level nor escape bits */
  274. if (table_vlc_level[i] &&
  275. vlc_bits[j] != table_vlc_bits[bound - 1]) {
  276. vlc_bits[j] <<= 1;
  277. vlc_len[j]++;
  278. j++;
  279. vlc_bits[j] = (table_vlc_bits[i] << 1) | 1;
  280. vlc_len[j] = table_vlc_len[i] + 1;
  281. vlc_run[j] = table_vlc_run[i];
  282. vlc_level[j] = -table_vlc_level[i];
  283. }
  284. }
  285. if ((ret = init_vlc(vlc, VLC_BITS, j, vlc_len, 1, 1, vlc_bits, 4, 4, 0)) < 0)
  286. return ret;
  287. for (i = 0; i < (*vlc).table_size; i++) {
  288. int code = (*vlc).table[i][0];
  289. int len = (*vlc).table[i][1];
  290. int level, run;
  291. if (len < 0) { // more bits needed
  292. run = 0;
  293. level = code;
  294. } else {
  295. run = vlc_run[code];
  296. level = vlc_level[code];
  297. }
  298. table_rl_vlc[i].len = len;
  299. table_rl_vlc[i].level = level;
  300. table_rl_vlc[i].run = run;
  301. }
  302. return 0;
  303. }
  304. av_cold int ff_cfhd_init_vlcs(CFHDContext *s)
  305. {
  306. /** Similar to dv.c, generate signed VLC tables **/
  307. int ret = init_vlc_signed(&s->vlc_9, s->table_9_rl_vlc, NB_VLC_TABLE_9,
  308. table_9_vlc_bits, table_9_vlc_len,
  309. table_9_vlc_run, table_9_vlc_level);
  310. if (ret < 0)
  311. return ret;
  312. return init_vlc_signed(&s->vlc_18, s->table_18_rl_vlc, NB_VLC_TABLE_18,
  313. table_18_vlc_bits, table_18_vlc_len,
  314. table_18_vlc_run, table_18_vlc_level);
  315. }