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.

233 lines
9.7KB

  1. /*
  2. * H.26L/H.264/AVC/JVT/14496-10/... encoder/decoder
  3. * Copyright (c) 2003 Michael Niedermayer <michaelni@gmx.at>
  4. *
  5. * This file is part of FFmpeg.
  6. *
  7. * FFmpeg is free software; you can redistribute it and/or
  8. * modify it under the terms of the GNU Lesser General Public
  9. * License as published by the Free Software Foundation; either
  10. * version 2.1 of the License, or (at your option) any later version.
  11. *
  12. * FFmpeg is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15. * Lesser General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public
  18. * License along with FFmpeg; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. /**
  22. * @file
  23. * @brief
  24. * H.264 / AVC / MPEG-4 part10 codec data table
  25. * @author Michael Niedermayer <michaelni@gmx.at>
  26. */
  27. #include <stdint.h>
  28. #include "libavutil/avutil.h"
  29. #include "avcodec.h"
  30. #include "h264.h"
  31. #include "h264data.h"
  32. const uint8_t ff_h264_golomb_to_pict_type[5] = {
  33. AV_PICTURE_TYPE_P, AV_PICTURE_TYPE_B, AV_PICTURE_TYPE_I,
  34. AV_PICTURE_TYPE_SP, AV_PICTURE_TYPE_SI
  35. };
  36. const uint8_t ff_h264_golomb_to_intra4x4_cbp[48] = {
  37. 47, 31, 15, 0, 23, 27, 29, 30, 7, 11, 13, 14, 39, 43, 45, 46,
  38. 16, 3, 5, 10, 12, 19, 21, 26, 28, 35, 37, 42, 44, 1, 2, 4,
  39. 8, 17, 18, 20, 24, 6, 9, 22, 25, 32, 33, 34, 36, 40, 38, 41
  40. };
  41. const uint8_t ff_h264_golomb_to_inter_cbp[48] = {
  42. 0, 16, 1, 2, 4, 8, 32, 3, 5, 10, 12, 15, 47, 7, 11, 13,
  43. 14, 6, 9, 31, 35, 37, 42, 44, 33, 34, 36, 40, 39, 43, 45, 46,
  44. 17, 18, 20, 24, 19, 21, 26, 28, 23, 27, 29, 30, 22, 25, 38, 41
  45. };
  46. const uint8_t ff_h264_chroma_dc_scan[4] = {
  47. (0 + 0 * 2) * 16, (1 + 0 * 2) * 16,
  48. (0 + 1 * 2) * 16, (1 + 1 * 2) * 16,
  49. };
  50. const uint8_t ff_h264_chroma422_dc_scan[8] = {
  51. (0 + 0 * 2) * 16, (0 + 1 * 2) * 16,
  52. (1 + 0 * 2) * 16, (0 + 2 * 2) * 16,
  53. (0 + 3 * 2) * 16, (1 + 1 * 2) * 16,
  54. (1 + 2 * 2) * 16, (1 + 3 * 2) * 16,
  55. };
  56. const IMbInfo ff_h264_i_mb_type_info[26] = {
  57. { MB_TYPE_INTRA4x4, -1, -1 },
  58. { MB_TYPE_INTRA16x16, 2, 0 },
  59. { MB_TYPE_INTRA16x16, 1, 0 },
  60. { MB_TYPE_INTRA16x16, 0, 0 },
  61. { MB_TYPE_INTRA16x16, 3, 0 },
  62. { MB_TYPE_INTRA16x16, 2, 16 },
  63. { MB_TYPE_INTRA16x16, 1, 16 },
  64. { MB_TYPE_INTRA16x16, 0, 16 },
  65. { MB_TYPE_INTRA16x16, 3, 16 },
  66. { MB_TYPE_INTRA16x16, 2, 32 },
  67. { MB_TYPE_INTRA16x16, 1, 32 },
  68. { MB_TYPE_INTRA16x16, 0, 32 },
  69. { MB_TYPE_INTRA16x16, 3, 32 },
  70. { MB_TYPE_INTRA16x16, 2, 15 + 0 },
  71. { MB_TYPE_INTRA16x16, 1, 15 + 0 },
  72. { MB_TYPE_INTRA16x16, 0, 15 + 0 },
  73. { MB_TYPE_INTRA16x16, 3, 15 + 0 },
  74. { MB_TYPE_INTRA16x16, 2, 15 + 16 },
  75. { MB_TYPE_INTRA16x16, 1, 15 + 16 },
  76. { MB_TYPE_INTRA16x16, 0, 15 + 16 },
  77. { MB_TYPE_INTRA16x16, 3, 15 + 16 },
  78. { MB_TYPE_INTRA16x16, 2, 15 + 32 },
  79. { MB_TYPE_INTRA16x16, 1, 15 + 32 },
  80. { MB_TYPE_INTRA16x16, 0, 15 + 32 },
  81. { MB_TYPE_INTRA16x16, 3, 15 + 32 },
  82. { MB_TYPE_INTRA_PCM, -1, -1 },
  83. };
  84. const PMbInfo ff_h264_p_mb_type_info[5] = {
  85. { MB_TYPE_16x16 | MB_TYPE_P0L0, 1 },
  86. { MB_TYPE_16x8 | MB_TYPE_P0L0 | MB_TYPE_P1L0, 2 },
  87. { MB_TYPE_8x16 | MB_TYPE_P0L0 | MB_TYPE_P1L0, 2 },
  88. { MB_TYPE_8x8 | MB_TYPE_P0L0 | MB_TYPE_P1L0, 4 },
  89. { MB_TYPE_8x8 | MB_TYPE_P0L0 | MB_TYPE_P1L0 | MB_TYPE_REF0, 4 },
  90. };
  91. const PMbInfo ff_h264_p_sub_mb_type_info[4] = {
  92. { MB_TYPE_16x16 | MB_TYPE_P0L0, 1 },
  93. { MB_TYPE_16x8 | MB_TYPE_P0L0, 2 },
  94. { MB_TYPE_8x16 | MB_TYPE_P0L0, 2 },
  95. { MB_TYPE_8x8 | MB_TYPE_P0L0, 4 },
  96. };
  97. const PMbInfo ff_h264_b_mb_type_info[23] = {
  98. { MB_TYPE_DIRECT2 | MB_TYPE_L0L1, 1, },
  99. { MB_TYPE_16x16 | MB_TYPE_P0L0, 1, },
  100. { MB_TYPE_16x16 | MB_TYPE_P0L1, 1, },
  101. { MB_TYPE_16x16 | MB_TYPE_P0L0 | MB_TYPE_P0L1, 1, },
  102. { MB_TYPE_16x8 | MB_TYPE_P0L0 | MB_TYPE_P1L0, 2, },
  103. { MB_TYPE_8x16 | MB_TYPE_P0L0 | MB_TYPE_P1L0, 2, },
  104. { MB_TYPE_16x8 | MB_TYPE_P0L1 | MB_TYPE_P1L1, 2, },
  105. { MB_TYPE_8x16 | MB_TYPE_P0L1 | MB_TYPE_P1L1, 2, },
  106. { MB_TYPE_16x8 | MB_TYPE_P0L0 | MB_TYPE_P1L1, 2, },
  107. { MB_TYPE_8x16 | MB_TYPE_P0L0 | MB_TYPE_P1L1, 2, },
  108. { MB_TYPE_16x8 | MB_TYPE_P0L1 | MB_TYPE_P1L0, 2, },
  109. { MB_TYPE_8x16 | MB_TYPE_P0L1 | MB_TYPE_P1L0, 2, },
  110. { MB_TYPE_16x8 | MB_TYPE_P0L0 | MB_TYPE_P1L0 | MB_TYPE_P1L1, 2, },
  111. { MB_TYPE_8x16 | MB_TYPE_P0L0 | MB_TYPE_P1L0 | MB_TYPE_P1L1, 2, },
  112. { MB_TYPE_16x8 | MB_TYPE_P0L1 | MB_TYPE_P1L0 | MB_TYPE_P1L1, 2, },
  113. { MB_TYPE_8x16 | MB_TYPE_P0L1 | MB_TYPE_P1L0 | MB_TYPE_P1L1, 2, },
  114. { MB_TYPE_16x8 | MB_TYPE_P0L0 | MB_TYPE_P0L1 | MB_TYPE_P1L0, 2, },
  115. { MB_TYPE_8x16 | MB_TYPE_P0L0 | MB_TYPE_P0L1 | MB_TYPE_P1L0, 2, },
  116. { MB_TYPE_16x8 | MB_TYPE_P0L0 | MB_TYPE_P0L1 | MB_TYPE_P1L1, 2, },
  117. { MB_TYPE_8x16 | MB_TYPE_P0L0 | MB_TYPE_P0L1 | MB_TYPE_P1L1, 2, },
  118. { MB_TYPE_16x8 | MB_TYPE_P0L0 | MB_TYPE_P0L1 | MB_TYPE_P1L0 | MB_TYPE_P1L1, 2, },
  119. { MB_TYPE_8x16 | MB_TYPE_P0L0 | MB_TYPE_P0L1 | MB_TYPE_P1L0 | MB_TYPE_P1L1, 2, },
  120. { MB_TYPE_8x8 | MB_TYPE_P0L0 | MB_TYPE_P0L1 | MB_TYPE_P1L0 | MB_TYPE_P1L1, 4, },
  121. };
  122. const PMbInfo ff_h264_b_sub_mb_type_info[13] = {
  123. { MB_TYPE_DIRECT2, 1, },
  124. { MB_TYPE_16x16 | MB_TYPE_P0L0, 1, },
  125. { MB_TYPE_16x16 | MB_TYPE_P0L1, 1, },
  126. { MB_TYPE_16x16 | MB_TYPE_P0L0 | MB_TYPE_P0L1, 1, },
  127. { MB_TYPE_16x8 | MB_TYPE_P0L0 | MB_TYPE_P1L0, 2, },
  128. { MB_TYPE_8x16 | MB_TYPE_P0L0 | MB_TYPE_P1L0, 2, },
  129. { MB_TYPE_16x8 | MB_TYPE_P0L1 | MB_TYPE_P1L1, 2, },
  130. { MB_TYPE_8x16 | MB_TYPE_P0L1 | MB_TYPE_P1L1, 2, },
  131. { MB_TYPE_16x8 | MB_TYPE_P0L0 | MB_TYPE_P0L1 | MB_TYPE_P1L0 | MB_TYPE_P1L1, 2, },
  132. { MB_TYPE_8x16 | MB_TYPE_P0L0 | MB_TYPE_P0L1 | MB_TYPE_P1L0 | MB_TYPE_P1L1, 2, },
  133. { MB_TYPE_8x8 | MB_TYPE_P0L0 | MB_TYPE_P1L0, 4, },
  134. { MB_TYPE_8x8 | MB_TYPE_P0L1 | MB_TYPE_P1L1, 4, },
  135. { MB_TYPE_8x8 | MB_TYPE_P0L0 | MB_TYPE_P0L1 | MB_TYPE_P1L0 | MB_TYPE_P1L1, 4, },
  136. };
  137. const uint8_t ff_h264_dequant4_coeff_init[6][3] = {
  138. { 10, 13, 16 },
  139. { 11, 14, 18 },
  140. { 13, 16, 20 },
  141. { 14, 18, 23 },
  142. { 16, 20, 25 },
  143. { 18, 23, 29 },
  144. };
  145. const uint8_t ff_h264_dequant8_coeff_init_scan[16] = {
  146. 0, 3, 4, 3, 3, 1, 5, 1, 4, 5, 2, 5, 3, 1, 5, 1
  147. };
  148. const uint8_t ff_h264_dequant8_coeff_init[6][6] = {
  149. { 20, 18, 32, 19, 25, 24 },
  150. { 22, 19, 35, 21, 28, 26 },
  151. { 26, 23, 42, 24, 33, 31 },
  152. { 28, 25, 45, 26, 35, 33 },
  153. { 32, 28, 51, 30, 40, 38 },
  154. { 36, 32, 58, 34, 46, 43 },
  155. };
  156. const uint8_t ff_h264_quant_rem6[QP_MAX_NUM + 1] = {
  157. 0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4, 5, 0, 1, 2,
  158. 3, 4, 5, 0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4, 5,
  159. 0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4, 5, 0, 1, 2,
  160. 3, 4, 5, 0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4, 5,
  161. 0, 1, 2, 3,
  162. };
  163. const uint8_t ff_h264_quant_div6[QP_MAX_NUM + 1] = {
  164. 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 3, 3, 3,
  165. 3, 3, 3, 4, 4, 4, 4, 4, 4, 5, 5, 5, 5, 5, 5, 6, 6, 6, 6, 6, 6,
  166. 7, 7, 7, 7, 7, 7, 8, 8, 8, 8, 8, 8, 9, 9, 9, 9, 9, 9, 10, 10, 10,
  167. 10,10,10,11,11,11,11,11,11,12,12,12,12,12,12,13,13,13, 13, 13, 13,
  168. 14,14,14,14,
  169. };
  170. #define QP(qP, depth) ((qP) + 6 * ((depth) - 8))
  171. #define CHROMA_QP_TABLE_END(d) \
  172. QP(0, d), QP(1, d), QP(2, d), QP(3, d), QP(4, d), QP(5, d), \
  173. QP(6, d), QP(7, d), QP(8, d), QP(9, d), QP(10, d), QP(11, d), \
  174. QP(12, d), QP(13, d), QP(14, d), QP(15, d), QP(16, d), QP(17, d), \
  175. QP(18, d), QP(19, d), QP(20, d), QP(21, d), QP(22, d), QP(23, d), \
  176. QP(24, d), QP(25, d), QP(26, d), QP(27, d), QP(28, d), QP(29, d), \
  177. QP(29, d), QP(30, d), QP(31, d), QP(32, d), QP(32, d), QP(33, d), \
  178. QP(34, d), QP(34, d), QP(35, d), QP(35, d), QP(36, d), QP(36, d), \
  179. QP(37, d), QP(37, d), QP(37, d), QP(38, d), QP(38, d), QP(38, d), \
  180. QP(39, d), QP(39, d), QP(39, d), QP(39, d)
  181. const uint8_t ff_h264_chroma_qp[7][QP_MAX_NUM + 1] = {
  182. { CHROMA_QP_TABLE_END(8) },
  183. { 0, 1, 2, 3, 4, 5,
  184. CHROMA_QP_TABLE_END(9) },
  185. { 0, 1, 2, 3, 4, 5,
  186. 6, 7, 8, 9, 10, 11,
  187. CHROMA_QP_TABLE_END(10) },
  188. { 0, 1, 2, 3, 4, 5,
  189. 6, 7, 8, 9, 10, 11,
  190. 12,13,14,15, 16, 17,
  191. CHROMA_QP_TABLE_END(11) },
  192. { 0, 1, 2, 3, 4, 5,
  193. 6, 7, 8, 9, 10, 11,
  194. 12,13,14,15, 16, 17,
  195. 18,19,20,21, 22, 23,
  196. CHROMA_QP_TABLE_END(12) },
  197. { 0, 1, 2, 3, 4, 5,
  198. 6, 7, 8, 9, 10, 11,
  199. 12,13,14,15, 16, 17,
  200. 18,19,20,21, 22, 23,
  201. 24,25,26,27, 28, 29,
  202. CHROMA_QP_TABLE_END(13) },
  203. { 0, 1, 2, 3, 4, 5,
  204. 6, 7, 8, 9, 10, 11,
  205. 12,13,14,15, 16, 17,
  206. 18,19,20,21, 22, 23,
  207. 24,25,26,27, 28, 29,
  208. 30,31,32,33, 34, 35,
  209. CHROMA_QP_TABLE_END(14) },
  210. };