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.

270 lines
9.1KB

  1. /*
  2. * H26L/H264/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. * H264 / AVC / MPEG4 part10 codec data table
  25. * @author Michael Niedermayer <michaelni@gmx.at>
  26. */
  27. #ifndef AVCODEC_H264DATA_H
  28. #define AVCODEC_H264DATA_H
  29. #include <stdint.h>
  30. #include "libavutil/rational.h"
  31. #include "mpegvideo.h"
  32. #include "h264.h"
  33. static const uint8_t golomb_to_pict_type[5]=
  34. {AV_PICTURE_TYPE_P, AV_PICTURE_TYPE_B, AV_PICTURE_TYPE_I, AV_PICTURE_TYPE_SP, AV_PICTURE_TYPE_SI};
  35. static const uint8_t golomb_to_intra4x4_cbp[48]={
  36. 47, 31, 15, 0, 23, 27, 29, 30, 7, 11, 13, 14, 39, 43, 45, 46,
  37. 16, 3, 5, 10, 12, 19, 21, 26, 28, 35, 37, 42, 44, 1, 2, 4,
  38. 8, 17, 18, 20, 24, 6, 9, 22, 25, 32, 33, 34, 36, 40, 38, 41
  39. };
  40. static const uint8_t golomb_to_inter_cbp[48]={
  41. 0, 16, 1, 2, 4, 8, 32, 3, 5, 10, 12, 15, 47, 7, 11, 13,
  42. 14, 6, 9, 31, 35, 37, 42, 44, 33, 34, 36, 40, 39, 43, 45, 46,
  43. 17, 18, 20, 24, 19, 21, 26, 28, 23, 27, 29, 30, 22, 25, 38, 41
  44. };
  45. static const uint8_t zigzag_scan[16]={
  46. 0+0*4, 1+0*4, 0+1*4, 0+2*4,
  47. 1+1*4, 2+0*4, 3+0*4, 2+1*4,
  48. 1+2*4, 0+3*4, 1+3*4, 2+2*4,
  49. 3+1*4, 3+2*4, 2+3*4, 3+3*4,
  50. };
  51. static const uint8_t field_scan[16]={
  52. 0+0*4, 0+1*4, 1+0*4, 0+2*4,
  53. 0+3*4, 1+1*4, 1+2*4, 1+3*4,
  54. 2+0*4, 2+1*4, 2+2*4, 2+3*4,
  55. 3+0*4, 3+1*4, 3+2*4, 3+3*4,
  56. };
  57. static const uint8_t luma_dc_zigzag_scan[16]={
  58. 0*16 + 0*64, 1*16 + 0*64, 2*16 + 0*64, 0*16 + 2*64,
  59. 3*16 + 0*64, 0*16 + 1*64, 1*16 + 1*64, 2*16 + 1*64,
  60. 1*16 + 2*64, 2*16 + 2*64, 3*16 + 2*64, 0*16 + 3*64,
  61. 3*16 + 1*64, 1*16 + 3*64, 2*16 + 3*64, 3*16 + 3*64,
  62. };
  63. static const uint8_t luma_dc_field_scan[16]={
  64. 0*16 + 0*64, 2*16 + 0*64, 1*16 + 0*64, 0*16 + 2*64,
  65. 2*16 + 2*64, 3*16 + 0*64, 1*16 + 2*64, 3*16 + 2*64,
  66. 0*16 + 1*64, 2*16 + 1*64, 0*16 + 3*64, 2*16 + 3*64,
  67. 1*16 + 1*64, 3*16 + 1*64, 1*16 + 3*64, 3*16 + 3*64,
  68. };
  69. static const uint8_t chroma_dc_scan[4]={
  70. (0+0*2)*16, (1+0*2)*16,
  71. (0+1*2)*16, (1+1*2)*16,
  72. };
  73. static const uint8_t chroma422_dc_scan[8]={
  74. (0+0*2)*16, (0+1*2)*16,
  75. (1+0*2)*16, (0+2*2)*16,
  76. (0+3*2)*16, (1+1*2)*16,
  77. (1+2*2)*16, (1+3*2)*16,
  78. };
  79. // zigzag_scan8x8_cavlc[i] = zigzag_scan8x8[(i/4) + 16*(i%4)]
  80. static const uint8_t zigzag_scan8x8_cavlc[64]={
  81. 0+0*8, 1+1*8, 1+2*8, 2+2*8,
  82. 4+1*8, 0+5*8, 3+3*8, 7+0*8,
  83. 3+4*8, 1+7*8, 5+3*8, 6+3*8,
  84. 2+7*8, 6+4*8, 5+6*8, 7+5*8,
  85. 1+0*8, 2+0*8, 0+3*8, 3+1*8,
  86. 3+2*8, 0+6*8, 4+2*8, 6+1*8,
  87. 2+5*8, 2+6*8, 6+2*8, 5+4*8,
  88. 3+7*8, 7+3*8, 4+7*8, 7+6*8,
  89. 0+1*8, 3+0*8, 0+4*8, 4+0*8,
  90. 2+3*8, 1+5*8, 5+1*8, 5+2*8,
  91. 1+6*8, 3+5*8, 7+1*8, 4+5*8,
  92. 4+6*8, 7+4*8, 5+7*8, 6+7*8,
  93. 0+2*8, 2+1*8, 1+3*8, 5+0*8,
  94. 1+4*8, 2+4*8, 6+0*8, 4+3*8,
  95. 0+7*8, 4+4*8, 7+2*8, 3+6*8,
  96. 5+5*8, 6+5*8, 6+6*8, 7+7*8,
  97. };
  98. static const uint8_t field_scan8x8[64]={
  99. 0+0*8, 0+1*8, 0+2*8, 1+0*8,
  100. 1+1*8, 0+3*8, 0+4*8, 1+2*8,
  101. 2+0*8, 1+3*8, 0+5*8, 0+6*8,
  102. 0+7*8, 1+4*8, 2+1*8, 3+0*8,
  103. 2+2*8, 1+5*8, 1+6*8, 1+7*8,
  104. 2+3*8, 3+1*8, 4+0*8, 3+2*8,
  105. 2+4*8, 2+5*8, 2+6*8, 2+7*8,
  106. 3+3*8, 4+1*8, 5+0*8, 4+2*8,
  107. 3+4*8, 3+5*8, 3+6*8, 3+7*8,
  108. 4+3*8, 5+1*8, 6+0*8, 5+2*8,
  109. 4+4*8, 4+5*8, 4+6*8, 4+7*8,
  110. 5+3*8, 6+1*8, 6+2*8, 5+4*8,
  111. 5+5*8, 5+6*8, 5+7*8, 6+3*8,
  112. 7+0*8, 7+1*8, 6+4*8, 6+5*8,
  113. 6+6*8, 6+7*8, 7+2*8, 7+3*8,
  114. 7+4*8, 7+5*8, 7+6*8, 7+7*8,
  115. };
  116. static const uint8_t field_scan8x8_cavlc[64]={
  117. 0+0*8, 1+1*8, 2+0*8, 0+7*8,
  118. 2+2*8, 2+3*8, 2+4*8, 3+3*8,
  119. 3+4*8, 4+3*8, 4+4*8, 5+3*8,
  120. 5+5*8, 7+0*8, 6+6*8, 7+4*8,
  121. 0+1*8, 0+3*8, 1+3*8, 1+4*8,
  122. 1+5*8, 3+1*8, 2+5*8, 4+1*8,
  123. 3+5*8, 5+1*8, 4+5*8, 6+1*8,
  124. 5+6*8, 7+1*8, 6+7*8, 7+5*8,
  125. 0+2*8, 0+4*8, 0+5*8, 2+1*8,
  126. 1+6*8, 4+0*8, 2+6*8, 5+0*8,
  127. 3+6*8, 6+0*8, 4+6*8, 6+2*8,
  128. 5+7*8, 6+4*8, 7+2*8, 7+6*8,
  129. 1+0*8, 1+2*8, 0+6*8, 3+0*8,
  130. 1+7*8, 3+2*8, 2+7*8, 4+2*8,
  131. 3+7*8, 5+2*8, 4+7*8, 5+4*8,
  132. 6+3*8, 6+5*8, 7+3*8, 7+7*8,
  133. };
  134. typedef struct IMbInfo{
  135. uint16_t type;
  136. uint8_t pred_mode;
  137. uint8_t cbp;
  138. } IMbInfo;
  139. static const IMbInfo i_mb_type_info[26]={
  140. {MB_TYPE_INTRA4x4 , -1, -1},
  141. {MB_TYPE_INTRA16x16, 2, 0},
  142. {MB_TYPE_INTRA16x16, 1, 0},
  143. {MB_TYPE_INTRA16x16, 0, 0},
  144. {MB_TYPE_INTRA16x16, 3, 0},
  145. {MB_TYPE_INTRA16x16, 2, 16},
  146. {MB_TYPE_INTRA16x16, 1, 16},
  147. {MB_TYPE_INTRA16x16, 0, 16},
  148. {MB_TYPE_INTRA16x16, 3, 16},
  149. {MB_TYPE_INTRA16x16, 2, 32},
  150. {MB_TYPE_INTRA16x16, 1, 32},
  151. {MB_TYPE_INTRA16x16, 0, 32},
  152. {MB_TYPE_INTRA16x16, 3, 32},
  153. {MB_TYPE_INTRA16x16, 2, 15+0},
  154. {MB_TYPE_INTRA16x16, 1, 15+0},
  155. {MB_TYPE_INTRA16x16, 0, 15+0},
  156. {MB_TYPE_INTRA16x16, 3, 15+0},
  157. {MB_TYPE_INTRA16x16, 2, 15+16},
  158. {MB_TYPE_INTRA16x16, 1, 15+16},
  159. {MB_TYPE_INTRA16x16, 0, 15+16},
  160. {MB_TYPE_INTRA16x16, 3, 15+16},
  161. {MB_TYPE_INTRA16x16, 2, 15+32},
  162. {MB_TYPE_INTRA16x16, 1, 15+32},
  163. {MB_TYPE_INTRA16x16, 0, 15+32},
  164. {MB_TYPE_INTRA16x16, 3, 15+32},
  165. {MB_TYPE_INTRA_PCM , -1, -1},
  166. };
  167. typedef struct PMbInfo{
  168. uint16_t type;
  169. uint8_t partition_count;
  170. } PMbInfo;
  171. static const PMbInfo p_mb_type_info[5]={
  172. {MB_TYPE_16x16|MB_TYPE_P0L0 , 1},
  173. {MB_TYPE_16x8 |MB_TYPE_P0L0|MB_TYPE_P1L0, 2},
  174. {MB_TYPE_8x16 |MB_TYPE_P0L0|MB_TYPE_P1L0, 2},
  175. {MB_TYPE_8x8 |MB_TYPE_P0L0|MB_TYPE_P1L0, 4},
  176. {MB_TYPE_8x8 |MB_TYPE_P0L0|MB_TYPE_P1L0|MB_TYPE_REF0, 4},
  177. };
  178. static const PMbInfo p_sub_mb_type_info[4]={
  179. {MB_TYPE_16x16|MB_TYPE_P0L0 , 1},
  180. {MB_TYPE_16x8 |MB_TYPE_P0L0 , 2},
  181. {MB_TYPE_8x16 |MB_TYPE_P0L0 , 2},
  182. {MB_TYPE_8x8 |MB_TYPE_P0L0 , 4},
  183. };
  184. static const PMbInfo b_mb_type_info[23]={
  185. {MB_TYPE_DIRECT2|MB_TYPE_L0L1 , 1, },
  186. {MB_TYPE_16x16|MB_TYPE_P0L0 , 1, },
  187. {MB_TYPE_16x16 |MB_TYPE_P0L1 , 1, },
  188. {MB_TYPE_16x16|MB_TYPE_P0L0|MB_TYPE_P0L1 , 1, },
  189. {MB_TYPE_16x8 |MB_TYPE_P0L0 |MB_TYPE_P1L0 , 2, },
  190. {MB_TYPE_8x16 |MB_TYPE_P0L0 |MB_TYPE_P1L0 , 2, },
  191. {MB_TYPE_16x8 |MB_TYPE_P0L1 |MB_TYPE_P1L1, 2, },
  192. {MB_TYPE_8x16 |MB_TYPE_P0L1 |MB_TYPE_P1L1, 2, },
  193. {MB_TYPE_16x8 |MB_TYPE_P0L0 |MB_TYPE_P1L1, 2, },
  194. {MB_TYPE_8x16 |MB_TYPE_P0L0 |MB_TYPE_P1L1, 2, },
  195. {MB_TYPE_16x8 |MB_TYPE_P0L1|MB_TYPE_P1L0 , 2, },
  196. {MB_TYPE_8x16 |MB_TYPE_P0L1|MB_TYPE_P1L0 , 2, },
  197. {MB_TYPE_16x8 |MB_TYPE_P0L0 |MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
  198. {MB_TYPE_8x16 |MB_TYPE_P0L0 |MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
  199. {MB_TYPE_16x8 |MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
  200. {MB_TYPE_8x16 |MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
  201. {MB_TYPE_16x8 |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0 , 2, },
  202. {MB_TYPE_8x16 |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0 , 2, },
  203. {MB_TYPE_16x8 |MB_TYPE_P0L0|MB_TYPE_P0L1 |MB_TYPE_P1L1, 2, },
  204. {MB_TYPE_8x16 |MB_TYPE_P0L0|MB_TYPE_P0L1 |MB_TYPE_P1L1, 2, },
  205. {MB_TYPE_16x8 |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
  206. {MB_TYPE_8x16 |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
  207. {MB_TYPE_8x8 |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 4, },
  208. };
  209. static const PMbInfo b_sub_mb_type_info[13]={
  210. {MB_TYPE_DIRECT2 , 1, },
  211. {MB_TYPE_16x16|MB_TYPE_P0L0 , 1, },
  212. {MB_TYPE_16x16 |MB_TYPE_P0L1 , 1, },
  213. {MB_TYPE_16x16|MB_TYPE_P0L0|MB_TYPE_P0L1 , 1, },
  214. {MB_TYPE_16x8 |MB_TYPE_P0L0 |MB_TYPE_P1L0 , 2, },
  215. {MB_TYPE_8x16 |MB_TYPE_P0L0 |MB_TYPE_P1L0 , 2, },
  216. {MB_TYPE_16x8 |MB_TYPE_P0L1 |MB_TYPE_P1L1, 2, },
  217. {MB_TYPE_8x16 |MB_TYPE_P0L1 |MB_TYPE_P1L1, 2, },
  218. {MB_TYPE_16x8 |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
  219. {MB_TYPE_8x16 |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 2, },
  220. {MB_TYPE_8x8 |MB_TYPE_P0L0 |MB_TYPE_P1L0 , 4, },
  221. {MB_TYPE_8x8 |MB_TYPE_P0L1 |MB_TYPE_P1L1, 4, },
  222. {MB_TYPE_8x8 |MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_P1L0|MB_TYPE_P1L1, 4, },
  223. };
  224. static const uint8_t dequant4_coeff_init[6][3]={
  225. {10,13,16},
  226. {11,14,18},
  227. {13,16,20},
  228. {14,18,23},
  229. {16,20,25},
  230. {18,23,29},
  231. };
  232. static const uint8_t dequant8_coeff_init_scan[16] = {
  233. 0,3,4,3, 3,1,5,1, 4,5,2,5, 3,1,5,1
  234. };
  235. static const uint8_t dequant8_coeff_init[6][6]={
  236. {20,18,32,19,25,24},
  237. {22,19,35,21,28,26},
  238. {26,23,42,24,33,31},
  239. {28,25,45,26,35,33},
  240. {32,28,51,30,40,38},
  241. {36,32,58,34,46,43},
  242. };
  243. #endif /* AVCODEC_H264DATA_H */