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.

1788 lines
73KB

  1. /*
  2. * H.26L/H.264/AVC/JVT/14496-10/... cabac decoding
  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 libavcodec/h264_cabac.c
  23. * H.264 / AVC / MPEG4 part10 cabac decoding.
  24. * @author Michael Niedermayer <michaelni@gmx.at>
  25. */
  26. #define CABAC 1
  27. #include "internal.h"
  28. #include "dsputil.h"
  29. #include "avcodec.h"
  30. #include "h264.h"
  31. #include "h264data.h"
  32. #include "h264_mvpred.h"
  33. #include "golomb.h"
  34. #include "cabac.h"
  35. #if ARCH_X86
  36. #include "x86/h264_i386.h"
  37. #endif
  38. //#undef NDEBUG
  39. #include <assert.h>
  40. /* Cabac pre state table */
  41. static const int8_t cabac_context_init_I[460][2] =
  42. {
  43. /* 0 - 10 */
  44. { 20, -15 }, { 2, 54 }, { 3, 74 }, { 20, -15 },
  45. { 2, 54 }, { 3, 74 }, { -28,127 }, { -23, 104 },
  46. { -6, 53 }, { -1, 54 }, { 7, 51 },
  47. /* 11 - 23 unsused for I */
  48. { 0, 0 }, { 0, 0 }, { 0, 0 }, { 0, 0 },
  49. { 0, 0 }, { 0, 0 }, { 0, 0 }, { 0, 0 },
  50. { 0, 0 }, { 0, 0 }, { 0, 0 }, { 0, 0 },
  51. { 0, 0 },
  52. /* 24- 39 */
  53. { 0, 0 }, { 0, 0 }, { 0, 0 }, { 0, 0 },
  54. { 0, 0 }, { 0, 0 }, { 0, 0 }, { 0, 0 },
  55. { 0, 0 }, { 0, 0 }, { 0, 0 }, { 0, 0 },
  56. { 0, 0 }, { 0, 0 }, { 0, 0 }, { 0, 0 },
  57. /* 40 - 53 */
  58. { 0, 0 }, { 0, 0 }, { 0, 0 }, { 0, 0 },
  59. { 0, 0 }, { 0, 0 }, { 0, 0 }, { 0, 0 },
  60. { 0, 0 }, { 0, 0 }, { 0, 0 }, { 0, 0 },
  61. { 0, 0 }, { 0, 0 },
  62. /* 54 - 59 */
  63. { 0, 0 }, { 0, 0 }, { 0, 0 }, { 0, 0 },
  64. { 0, 0 }, { 0, 0 },
  65. /* 60 - 69 */
  66. { 0, 41 }, { 0, 63 }, { 0, 63 }, { 0, 63 },
  67. { -9, 83 }, { 4, 86 }, { 0, 97 }, { -7, 72 },
  68. { 13, 41 }, { 3, 62 },
  69. /* 70 -> 87 */
  70. { 0, 11 }, { 1, 55 }, { 0, 69 }, { -17, 127 },
  71. { -13, 102 },{ 0, 82 }, { -7, 74 }, { -21, 107 },
  72. { -27, 127 },{ -31, 127 },{ -24, 127 }, { -18, 95 },
  73. { -27, 127 },{ -21, 114 },{ -30, 127 }, { -17, 123 },
  74. { -12, 115 },{ -16, 122 },
  75. /* 88 -> 104 */
  76. { -11, 115 },{ -12, 63 }, { -2, 68 }, { -15, 84 },
  77. { -13, 104 },{ -3, 70 }, { -8, 93 }, { -10, 90 },
  78. { -30, 127 },{ -1, 74 }, { -6, 97 }, { -7, 91 },
  79. { -20, 127 },{ -4, 56 }, { -5, 82 }, { -7, 76 },
  80. { -22, 125 },
  81. /* 105 -> 135 */
  82. { -7, 93 }, { -11, 87 }, { -3, 77 }, { -5, 71 },
  83. { -4, 63 }, { -4, 68 }, { -12, 84 }, { -7, 62 },
  84. { -7, 65 }, { 8, 61 }, { 5, 56 }, { -2, 66 },
  85. { 1, 64 }, { 0, 61 }, { -2, 78 }, { 1, 50 },
  86. { 7, 52 }, { 10, 35 }, { 0, 44 }, { 11, 38 },
  87. { 1, 45 }, { 0, 46 }, { 5, 44 }, { 31, 17 },
  88. { 1, 51 }, { 7, 50 }, { 28, 19 }, { 16, 33 },
  89. { 14, 62 }, { -13, 108 },{ -15, 100 },
  90. /* 136 -> 165 */
  91. { -13, 101 },{ -13, 91 }, { -12, 94 }, { -10, 88 },
  92. { -16, 84 }, { -10, 86 }, { -7, 83 }, { -13, 87 },
  93. { -19, 94 }, { 1, 70 }, { 0, 72 }, { -5, 74 },
  94. { 18, 59 }, { -8, 102 }, { -15, 100 }, { 0, 95 },
  95. { -4, 75 }, { 2, 72 }, { -11, 75 }, { -3, 71 },
  96. { 15, 46 }, { -13, 69 }, { 0, 62 }, { 0, 65 },
  97. { 21, 37 }, { -15, 72 }, { 9, 57 }, { 16, 54 },
  98. { 0, 62 }, { 12, 72 },
  99. /* 166 -> 196 */
  100. { 24, 0 }, { 15, 9 }, { 8, 25 }, { 13, 18 },
  101. { 15, 9 }, { 13, 19 }, { 10, 37 }, { 12, 18 },
  102. { 6, 29 }, { 20, 33 }, { 15, 30 }, { 4, 45 },
  103. { 1, 58 }, { 0, 62 }, { 7, 61 }, { 12, 38 },
  104. { 11, 45 }, { 15, 39 }, { 11, 42 }, { 13, 44 },
  105. { 16, 45 }, { 12, 41 }, { 10, 49 }, { 30, 34 },
  106. { 18, 42 }, { 10, 55 }, { 17, 51 }, { 17, 46 },
  107. { 0, 89 }, { 26, -19 }, { 22, -17 },
  108. /* 197 -> 226 */
  109. { 26, -17 }, { 30, -25 }, { 28, -20 }, { 33, -23 },
  110. { 37, -27 }, { 33, -23 }, { 40, -28 }, { 38, -17 },
  111. { 33, -11 }, { 40, -15 }, { 41, -6 }, { 38, 1 },
  112. { 41, 17 }, { 30, -6 }, { 27, 3 }, { 26, 22 },
  113. { 37, -16 }, { 35, -4 }, { 38, -8 }, { 38, -3 },
  114. { 37, 3 }, { 38, 5 }, { 42, 0 }, { 35, 16 },
  115. { 39, 22 }, { 14, 48 }, { 27, 37 }, { 21, 60 },
  116. { 12, 68 }, { 2, 97 },
  117. /* 227 -> 251 */
  118. { -3, 71 }, { -6, 42 }, { -5, 50 }, { -3, 54 },
  119. { -2, 62 }, { 0, 58 }, { 1, 63 }, { -2, 72 },
  120. { -1, 74 }, { -9, 91 }, { -5, 67 }, { -5, 27 },
  121. { -3, 39 }, { -2, 44 }, { 0, 46 }, { -16, 64 },
  122. { -8, 68 }, { -10, 78 }, { -6, 77 }, { -10, 86 },
  123. { -12, 92 }, { -15, 55 }, { -10, 60 }, { -6, 62 },
  124. { -4, 65 },
  125. /* 252 -> 275 */
  126. { -12, 73 }, { -8, 76 }, { -7, 80 }, { -9, 88 },
  127. { -17, 110 },{ -11, 97 }, { -20, 84 }, { -11, 79 },
  128. { -6, 73 }, { -4, 74 }, { -13, 86 }, { -13, 96 },
  129. { -11, 97 }, { -19, 117 },{ -8, 78 }, { -5, 33 },
  130. { -4, 48 }, { -2, 53 }, { -3, 62 }, { -13, 71 },
  131. { -10, 79 }, { -12, 86 }, { -13, 90 }, { -14, 97 },
  132. /* 276 a bit special (not used, bypass is used instead) */
  133. { 0, 0 },
  134. /* 277 -> 307 */
  135. { -6, 93 }, { -6, 84 }, { -8, 79 }, { 0, 66 },
  136. { -1, 71 }, { 0, 62 }, { -2, 60 }, { -2, 59 },
  137. { -5, 75 }, { -3, 62 }, { -4, 58 }, { -9, 66 },
  138. { -1, 79 }, { 0, 71 }, { 3, 68 }, { 10, 44 },
  139. { -7, 62 }, { 15, 36 }, { 14, 40 }, { 16, 27 },
  140. { 12, 29 }, { 1, 44 }, { 20, 36 }, { 18, 32 },
  141. { 5, 42 }, { 1, 48 }, { 10, 62 }, { 17, 46 },
  142. { 9, 64 }, { -12, 104 },{ -11, 97 },
  143. /* 308 -> 337 */
  144. { -16, 96 }, { -7, 88 }, { -8, 85 }, { -7, 85 },
  145. { -9, 85 }, { -13, 88 }, { 4, 66 }, { -3, 77 },
  146. { -3, 76 }, { -6, 76 }, { 10, 58 }, { -1, 76 },
  147. { -1, 83 }, { -7, 99 }, { -14, 95 }, { 2, 95 },
  148. { 0, 76 }, { -5, 74 }, { 0, 70 }, { -11, 75 },
  149. { 1, 68 }, { 0, 65 }, { -14, 73 }, { 3, 62 },
  150. { 4, 62 }, { -1, 68 }, { -13, 75 }, { 11, 55 },
  151. { 5, 64 }, { 12, 70 },
  152. /* 338 -> 368 */
  153. { 15, 6 }, { 6, 19 }, { 7, 16 }, { 12, 14 },
  154. { 18, 13 }, { 13, 11 }, { 13, 15 }, { 15, 16 },
  155. { 12, 23 }, { 13, 23 }, { 15, 20 }, { 14, 26 },
  156. { 14, 44 }, { 17, 40 }, { 17, 47 }, { 24, 17 },
  157. { 21, 21 }, { 25, 22 }, { 31, 27 }, { 22, 29 },
  158. { 19, 35 }, { 14, 50 }, { 10, 57 }, { 7, 63 },
  159. { -2, 77 }, { -4, 82 }, { -3, 94 }, { 9, 69 },
  160. { -12, 109 },{ 36, -35 }, { 36, -34 },
  161. /* 369 -> 398 */
  162. { 32, -26 }, { 37, -30 }, { 44, -32 }, { 34, -18 },
  163. { 34, -15 }, { 40, -15 }, { 33, -7 }, { 35, -5 },
  164. { 33, 0 }, { 38, 2 }, { 33, 13 }, { 23, 35 },
  165. { 13, 58 }, { 29, -3 }, { 26, 0 }, { 22, 30 },
  166. { 31, -7 }, { 35, -15 }, { 34, -3 }, { 34, 3 },
  167. { 36, -1 }, { 34, 5 }, { 32, 11 }, { 35, 5 },
  168. { 34, 12 }, { 39, 11 }, { 30, 29 }, { 34, 26 },
  169. { 29, 39 }, { 19, 66 },
  170. /* 399 -> 435 */
  171. { 31, 21 }, { 31, 31 }, { 25, 50 },
  172. { -17, 120 }, { -20, 112 }, { -18, 114 }, { -11, 85 },
  173. { -15, 92 }, { -14, 89 }, { -26, 71 }, { -15, 81 },
  174. { -14, 80 }, { 0, 68 }, { -14, 70 }, { -24, 56 },
  175. { -23, 68 }, { -24, 50 }, { -11, 74 }, { 23, -13 },
  176. { 26, -13 }, { 40, -15 }, { 49, -14 }, { 44, 3 },
  177. { 45, 6 }, { 44, 34 }, { 33, 54 }, { 19, 82 },
  178. { -3, 75 }, { -1, 23 }, { 1, 34 }, { 1, 43 },
  179. { 0, 54 }, { -2, 55 }, { 0, 61 }, { 1, 64 },
  180. { 0, 68 }, { -9, 92 },
  181. /* 436 -> 459 */
  182. { -14, 106 }, { -13, 97 }, { -15, 90 }, { -12, 90 },
  183. { -18, 88 }, { -10, 73 }, { -9, 79 }, { -14, 86 },
  184. { -10, 73 }, { -10, 70 }, { -10, 69 }, { -5, 66 },
  185. { -9, 64 }, { -5, 58 }, { 2, 59 }, { 21, -10 },
  186. { 24, -11 }, { 28, -8 }, { 28, -1 }, { 29, 3 },
  187. { 29, 9 }, { 35, 20 }, { 29, 36 }, { 14, 67 }
  188. };
  189. static const int8_t cabac_context_init_PB[3][460][2] =
  190. {
  191. /* i_cabac_init_idc == 0 */
  192. {
  193. /* 0 - 10 */
  194. { 20, -15 }, { 2, 54 }, { 3, 74 }, { 20, -15 },
  195. { 2, 54 }, { 3, 74 }, { -28, 127 }, { -23, 104 },
  196. { -6, 53 }, { -1, 54 }, { 7, 51 },
  197. /* 11 - 23 */
  198. { 23, 33 }, { 23, 2 }, { 21, 0 }, { 1, 9 },
  199. { 0, 49 }, { -37, 118 }, { 5, 57 }, { -13, 78 },
  200. { -11, 65 }, { 1, 62 }, { 12, 49 }, { -4, 73 },
  201. { 17, 50 },
  202. /* 24 - 39 */
  203. { 18, 64 }, { 9, 43 }, { 29, 0 }, { 26, 67 },
  204. { 16, 90 }, { 9, 104 }, { -46, 127 }, { -20, 104 },
  205. { 1, 67 }, { -13, 78 }, { -11, 65 }, { 1, 62 },
  206. { -6, 86 }, { -17, 95 }, { -6, 61 }, { 9, 45 },
  207. /* 40 - 53 */
  208. { -3, 69 }, { -6, 81 }, { -11, 96 }, { 6, 55 },
  209. { 7, 67 }, { -5, 86 }, { 2, 88 }, { 0, 58 },
  210. { -3, 76 }, { -10, 94 }, { 5, 54 }, { 4, 69 },
  211. { -3, 81 }, { 0, 88 },
  212. /* 54 - 59 */
  213. { -7, 67 }, { -5, 74 }, { -4, 74 }, { -5, 80 },
  214. { -7, 72 }, { 1, 58 },
  215. /* 60 - 69 */
  216. { 0, 41 }, { 0, 63 }, { 0, 63 }, { 0, 63 },
  217. { -9, 83 }, { 4, 86 }, { 0, 97 }, { -7, 72 },
  218. { 13, 41 }, { 3, 62 },
  219. /* 70 - 87 */
  220. { 0, 45 }, { -4, 78 }, { -3, 96 }, { -27, 126 },
  221. { -28, 98 }, { -25, 101 }, { -23, 67 }, { -28, 82 },
  222. { -20, 94 }, { -16, 83 }, { -22, 110 }, { -21, 91 },
  223. { -18, 102 }, { -13, 93 }, { -29, 127 }, { -7, 92 },
  224. { -5, 89 }, { -7, 96 }, { -13, 108 }, { -3, 46 },
  225. { -1, 65 }, { -1, 57 }, { -9, 93 }, { -3, 74 },
  226. { -9, 92 }, { -8, 87 }, { -23, 126 }, { 5, 54 },
  227. { 6, 60 }, { 6, 59 }, { 6, 69 }, { -1, 48 },
  228. { 0, 68 }, { -4, 69 }, { -8, 88 },
  229. /* 105 -> 165 */
  230. { -2, 85 }, { -6, 78 }, { -1, 75 }, { -7, 77 },
  231. { 2, 54 }, { 5, 50 }, { -3, 68 }, { 1, 50 },
  232. { 6, 42 }, { -4, 81 }, { 1, 63 }, { -4, 70 },
  233. { 0, 67 }, { 2, 57 }, { -2, 76 }, { 11, 35 },
  234. { 4, 64 }, { 1, 61 }, { 11, 35 }, { 18, 25 },
  235. { 12, 24 }, { 13, 29 }, { 13, 36 }, { -10, 93 },
  236. { -7, 73 }, { -2, 73 }, { 13, 46 }, { 9, 49 },
  237. { -7, 100 }, { 9, 53 }, { 2, 53 }, { 5, 53 },
  238. { -2, 61 }, { 0, 56 }, { 0, 56 }, { -13, 63 },
  239. { -5, 60 }, { -1, 62 }, { 4, 57 }, { -6, 69 },
  240. { 4, 57 }, { 14, 39 }, { 4, 51 }, { 13, 68 },
  241. { 3, 64 }, { 1, 61 }, { 9, 63 }, { 7, 50 },
  242. { 16, 39 }, { 5, 44 }, { 4, 52 }, { 11, 48 },
  243. { -5, 60 }, { -1, 59 }, { 0, 59 }, { 22, 33 },
  244. { 5, 44 }, { 14, 43 }, { -1, 78 }, { 0, 60 },
  245. { 9, 69 },
  246. /* 166 - 226 */
  247. { 11, 28 }, { 2, 40 }, { 3, 44 }, { 0, 49 },
  248. { 0, 46 }, { 2, 44 }, { 2, 51 }, { 0, 47 },
  249. { 4, 39 }, { 2, 62 }, { 6, 46 }, { 0, 54 },
  250. { 3, 54 }, { 2, 58 }, { 4, 63 }, { 6, 51 },
  251. { 6, 57 }, { 7, 53 }, { 6, 52 }, { 6, 55 },
  252. { 11, 45 }, { 14, 36 }, { 8, 53 }, { -1, 82 },
  253. { 7, 55 }, { -3, 78 }, { 15, 46 }, { 22, 31 },
  254. { -1, 84 }, { 25, 7 }, { 30, -7 }, { 28, 3 },
  255. { 28, 4 }, { 32, 0 }, { 34, -1 }, { 30, 6 },
  256. { 30, 6 }, { 32, 9 }, { 31, 19 }, { 26, 27 },
  257. { 26, 30 }, { 37, 20 }, { 28, 34 }, { 17, 70 },
  258. { 1, 67 }, { 5, 59 }, { 9, 67 }, { 16, 30 },
  259. { 18, 32 }, { 18, 35 }, { 22, 29 }, { 24, 31 },
  260. { 23, 38 }, { 18, 43 }, { 20, 41 }, { 11, 63 },
  261. { 9, 59 }, { 9, 64 }, { -1, 94 }, { -2, 89 },
  262. { -9, 108 },
  263. /* 227 - 275 */
  264. { -6, 76 }, { -2, 44 }, { 0, 45 }, { 0, 52 },
  265. { -3, 64 }, { -2, 59 }, { -4, 70 }, { -4, 75 },
  266. { -8, 82 }, { -17, 102 }, { -9, 77 }, { 3, 24 },
  267. { 0, 42 }, { 0, 48 }, { 0, 55 }, { -6, 59 },
  268. { -7, 71 }, { -12, 83 }, { -11, 87 }, { -30, 119 },
  269. { 1, 58 }, { -3, 29 }, { -1, 36 }, { 1, 38 },
  270. { 2, 43 }, { -6, 55 }, { 0, 58 }, { 0, 64 },
  271. { -3, 74 }, { -10, 90 }, { 0, 70 }, { -4, 29 },
  272. { 5, 31 }, { 7, 42 }, { 1, 59 }, { -2, 58 },
  273. { -3, 72 }, { -3, 81 }, { -11, 97 }, { 0, 58 },
  274. { 8, 5 }, { 10, 14 }, { 14, 18 }, { 13, 27 },
  275. { 2, 40 }, { 0, 58 }, { -3, 70 }, { -6, 79 },
  276. { -8, 85 },
  277. /* 276 a bit special (not used, bypass is used instead) */
  278. { 0, 0 },
  279. /* 277 - 337 */
  280. { -13, 106 }, { -16, 106 }, { -10, 87 }, { -21, 114 },
  281. { -18, 110 }, { -14, 98 }, { -22, 110 }, { -21, 106 },
  282. { -18, 103 }, { -21, 107 }, { -23, 108 }, { -26, 112 },
  283. { -10, 96 }, { -12, 95 }, { -5, 91 }, { -9, 93 },
  284. { -22, 94 }, { -5, 86 }, { 9, 67 }, { -4, 80 },
  285. { -10, 85 }, { -1, 70 }, { 7, 60 }, { 9, 58 },
  286. { 5, 61 }, { 12, 50 }, { 15, 50 }, { 18, 49 },
  287. { 17, 54 }, { 10, 41 }, { 7, 46 }, { -1, 51 },
  288. { 7, 49 }, { 8, 52 }, { 9, 41 }, { 6, 47 },
  289. { 2, 55 }, { 13, 41 }, { 10, 44 }, { 6, 50 },
  290. { 5, 53 }, { 13, 49 }, { 4, 63 }, { 6, 64 },
  291. { -2, 69 }, { -2, 59 }, { 6, 70 }, { 10, 44 },
  292. { 9, 31 }, { 12, 43 }, { 3, 53 }, { 14, 34 },
  293. { 10, 38 }, { -3, 52 }, { 13, 40 }, { 17, 32 },
  294. { 7, 44 }, { 7, 38 }, { 13, 50 }, { 10, 57 },
  295. { 26, 43 },
  296. /* 338 - 398 */
  297. { 14, 11 }, { 11, 14 }, { 9, 11 }, { 18, 11 },
  298. { 21, 9 }, { 23, -2 }, { 32, -15 }, { 32, -15 },
  299. { 34, -21 }, { 39, -23 }, { 42, -33 }, { 41, -31 },
  300. { 46, -28 }, { 38, -12 }, { 21, 29 }, { 45, -24 },
  301. { 53, -45 }, { 48, -26 }, { 65, -43 }, { 43, -19 },
  302. { 39, -10 }, { 30, 9 }, { 18, 26 }, { 20, 27 },
  303. { 0, 57 }, { -14, 82 }, { -5, 75 }, { -19, 97 },
  304. { -35, 125 }, { 27, 0 }, { 28, 0 }, { 31, -4 },
  305. { 27, 6 }, { 34, 8 }, { 30, 10 }, { 24, 22 },
  306. { 33, 19 }, { 22, 32 }, { 26, 31 }, { 21, 41 },
  307. { 26, 44 }, { 23, 47 }, { 16, 65 }, { 14, 71 },
  308. { 8, 60 }, { 6, 63 }, { 17, 65 }, { 21, 24 },
  309. { 23, 20 }, { 26, 23 }, { 27, 32 }, { 28, 23 },
  310. { 28, 24 }, { 23, 40 }, { 24, 32 }, { 28, 29 },
  311. { 23, 42 }, { 19, 57 }, { 22, 53 }, { 22, 61 },
  312. { 11, 86 },
  313. /* 399 - 435 */
  314. { 12, 40 }, { 11, 51 }, { 14, 59 },
  315. { -4, 79 }, { -7, 71 }, { -5, 69 }, { -9, 70 },
  316. { -8, 66 }, { -10, 68 }, { -19, 73 }, { -12, 69 },
  317. { -16, 70 }, { -15, 67 }, { -20, 62 }, { -19, 70 },
  318. { -16, 66 }, { -22, 65 }, { -20, 63 }, { 9, -2 },
  319. { 26, -9 }, { 33, -9 }, { 39, -7 }, { 41, -2 },
  320. { 45, 3 }, { 49, 9 }, { 45, 27 }, { 36, 59 },
  321. { -6, 66 }, { -7, 35 }, { -7, 42 }, { -8, 45 },
  322. { -5, 48 }, { -12, 56 }, { -6, 60 }, { -5, 62 },
  323. { -8, 66 }, { -8, 76 },
  324. /* 436 - 459 */
  325. { -5, 85 }, { -6, 81 }, { -10, 77 }, { -7, 81 },
  326. { -17, 80 }, { -18, 73 }, { -4, 74 }, { -10, 83 },
  327. { -9, 71 }, { -9, 67 }, { -1, 61 }, { -8, 66 },
  328. { -14, 66 }, { 0, 59 }, { 2, 59 }, { 21, -13 },
  329. { 33, -14 }, { 39, -7 }, { 46, -2 }, { 51, 2 },
  330. { 60, 6 }, { 61, 17 }, { 55, 34 }, { 42, 62 },
  331. },
  332. /* i_cabac_init_idc == 1 */
  333. {
  334. /* 0 - 10 */
  335. { 20, -15 }, { 2, 54 }, { 3, 74 }, { 20, -15 },
  336. { 2, 54 }, { 3, 74 }, { -28, 127 }, { -23, 104 },
  337. { -6, 53 }, { -1, 54 }, { 7, 51 },
  338. /* 11 - 23 */
  339. { 22, 25 }, { 34, 0 }, { 16, 0 }, { -2, 9 },
  340. { 4, 41 }, { -29, 118 }, { 2, 65 }, { -6, 71 },
  341. { -13, 79 }, { 5, 52 }, { 9, 50 }, { -3, 70 },
  342. { 10, 54 },
  343. /* 24 - 39 */
  344. { 26, 34 }, { 19, 22 }, { 40, 0 }, { 57, 2 },
  345. { 41, 36 }, { 26, 69 }, { -45, 127 }, { -15, 101 },
  346. { -4, 76 }, { -6, 71 }, { -13, 79 }, { 5, 52 },
  347. { 6, 69 }, { -13, 90 }, { 0, 52 }, { 8, 43 },
  348. /* 40 - 53 */
  349. { -2, 69 },{ -5, 82 },{ -10, 96 },{ 2, 59 },
  350. { 2, 75 },{ -3, 87 },{ -3, 100 },{ 1, 56 },
  351. { -3, 74 },{ -6, 85 },{ 0, 59 },{ -3, 81 },
  352. { -7, 86 },{ -5, 95 },
  353. /* 54 - 59 */
  354. { -1, 66 },{ -1, 77 },{ 1, 70 },{ -2, 86 },
  355. { -5, 72 },{ 0, 61 },
  356. /* 60 - 69 */
  357. { 0, 41 }, { 0, 63 }, { 0, 63 }, { 0, 63 },
  358. { -9, 83 }, { 4, 86 }, { 0, 97 }, { -7, 72 },
  359. { 13, 41 }, { 3, 62 },
  360. /* 70 - 104 */
  361. { 13, 15 }, { 7, 51 }, { 2, 80 }, { -39, 127 },
  362. { -18, 91 }, { -17, 96 }, { -26, 81 }, { -35, 98 },
  363. { -24, 102 }, { -23, 97 }, { -27, 119 }, { -24, 99 },
  364. { -21, 110 }, { -18, 102 }, { -36, 127 }, { 0, 80 },
  365. { -5, 89 }, { -7, 94 }, { -4, 92 }, { 0, 39 },
  366. { 0, 65 }, { -15, 84 }, { -35, 127 }, { -2, 73 },
  367. { -12, 104 }, { -9, 91 }, { -31, 127 }, { 3, 55 },
  368. { 7, 56 }, { 7, 55 }, { 8, 61 }, { -3, 53 },
  369. { 0, 68 }, { -7, 74 }, { -9, 88 },
  370. /* 105 -> 165 */
  371. { -13, 103 }, { -13, 91 }, { -9, 89 }, { -14, 92 },
  372. { -8, 76 }, { -12, 87 }, { -23, 110 }, { -24, 105 },
  373. { -10, 78 }, { -20, 112 }, { -17, 99 }, { -78, 127 },
  374. { -70, 127 }, { -50, 127 }, { -46, 127 }, { -4, 66 },
  375. { -5, 78 }, { -4, 71 }, { -8, 72 }, { 2, 59 },
  376. { -1, 55 }, { -7, 70 }, { -6, 75 }, { -8, 89 },
  377. { -34, 119 }, { -3, 75 }, { 32, 20 }, { 30, 22 },
  378. { -44, 127 }, { 0, 54 }, { -5, 61 }, { 0, 58 },
  379. { -1, 60 }, { -3, 61 }, { -8, 67 }, { -25, 84 },
  380. { -14, 74 }, { -5, 65 }, { 5, 52 }, { 2, 57 },
  381. { 0, 61 }, { -9, 69 }, { -11, 70 }, { 18, 55 },
  382. { -4, 71 }, { 0, 58 }, { 7, 61 }, { 9, 41 },
  383. { 18, 25 }, { 9, 32 }, { 5, 43 }, { 9, 47 },
  384. { 0, 44 }, { 0, 51 }, { 2, 46 }, { 19, 38 },
  385. { -4, 66 }, { 15, 38 }, { 12, 42 }, { 9, 34 },
  386. { 0, 89 },
  387. /* 166 - 226 */
  388. { 4, 45 }, { 10, 28 }, { 10, 31 }, { 33, -11 },
  389. { 52, -43 }, { 18, 15 }, { 28, 0 }, { 35, -22 },
  390. { 38, -25 }, { 34, 0 }, { 39, -18 }, { 32, -12 },
  391. { 102, -94 }, { 0, 0 }, { 56, -15 }, { 33, -4 },
  392. { 29, 10 }, { 37, -5 }, { 51, -29 }, { 39, -9 },
  393. { 52, -34 }, { 69, -58 }, { 67, -63 }, { 44, -5 },
  394. { 32, 7 }, { 55, -29 }, { 32, 1 }, { 0, 0 },
  395. { 27, 36 }, { 33, -25 }, { 34, -30 }, { 36, -28 },
  396. { 38, -28 }, { 38, -27 }, { 34, -18 }, { 35, -16 },
  397. { 34, -14 }, { 32, -8 }, { 37, -6 }, { 35, 0 },
  398. { 30, 10 }, { 28, 18 }, { 26, 25 }, { 29, 41 },
  399. { 0, 75 }, { 2, 72 }, { 8, 77 }, { 14, 35 },
  400. { 18, 31 }, { 17, 35 }, { 21, 30 }, { 17, 45 },
  401. { 20, 42 }, { 18, 45 }, { 27, 26 }, { 16, 54 },
  402. { 7, 66 }, { 16, 56 }, { 11, 73 }, { 10, 67 },
  403. { -10, 116 },
  404. /* 227 - 275 */
  405. { -23, 112 }, { -15, 71 }, { -7, 61 }, { 0, 53 },
  406. { -5, 66 }, { -11, 77 }, { -9, 80 }, { -9, 84 },
  407. { -10, 87 }, { -34, 127 }, { -21, 101 }, { -3, 39 },
  408. { -5, 53 }, { -7, 61 }, { -11, 75 }, { -15, 77 },
  409. { -17, 91 }, { -25, 107 }, { -25, 111 }, { -28, 122 },
  410. { -11, 76 }, { -10, 44 }, { -10, 52 }, { -10, 57 },
  411. { -9, 58 }, { -16, 72 }, { -7, 69 }, { -4, 69 },
  412. { -5, 74 }, { -9, 86 }, { 2, 66 }, { -9, 34 },
  413. { 1, 32 }, { 11, 31 }, { 5, 52 }, { -2, 55 },
  414. { -2, 67 }, { 0, 73 }, { -8, 89 }, { 3, 52 },
  415. { 7, 4 }, { 10, 8 }, { 17, 8 }, { 16, 19 },
  416. { 3, 37 }, { -1, 61 }, { -5, 73 }, { -1, 70 },
  417. { -4, 78 },
  418. /* 276 a bit special (not used, bypass is used instead) */
  419. { 0, 0 },
  420. /* 277 - 337 */
  421. { -21, 126 }, { -23, 124 }, { -20, 110 }, { -26, 126 },
  422. { -25, 124 }, { -17, 105 }, { -27, 121 }, { -27, 117 },
  423. { -17, 102 }, { -26, 117 }, { -27, 116 }, { -33, 122 },
  424. { -10, 95 }, { -14, 100 }, { -8, 95 }, { -17, 111 },
  425. { -28, 114 }, { -6, 89 }, { -2, 80 }, { -4, 82 },
  426. { -9, 85 }, { -8, 81 }, { -1, 72 }, { 5, 64 },
  427. { 1, 67 }, { 9, 56 }, { 0, 69 }, { 1, 69 },
  428. { 7, 69 }, { -7, 69 }, { -6, 67 }, { -16, 77 },
  429. { -2, 64 }, { 2, 61 }, { -6, 67 }, { -3, 64 },
  430. { 2, 57 }, { -3, 65 }, { -3, 66 }, { 0, 62 },
  431. { 9, 51 }, { -1, 66 }, { -2, 71 }, { -2, 75 },
  432. { -1, 70 }, { -9, 72 }, { 14, 60 }, { 16, 37 },
  433. { 0, 47 }, { 18, 35 }, { 11, 37 }, { 12, 41 },
  434. { 10, 41 }, { 2, 48 }, { 12, 41 }, { 13, 41 },
  435. { 0, 59 }, { 3, 50 }, { 19, 40 }, { 3, 66 },
  436. { 18, 50 },
  437. /* 338 - 398 */
  438. { 19, -6 }, { 18, -6 }, { 14, 0 }, { 26, -12 },
  439. { 31, -16 }, { 33, -25 }, { 33, -22 }, { 37, -28 },
  440. { 39, -30 }, { 42, -30 }, { 47, -42 }, { 45, -36 },
  441. { 49, -34 }, { 41, -17 }, { 32, 9 }, { 69, -71 },
  442. { 63, -63 }, { 66, -64 }, { 77, -74 }, { 54, -39 },
  443. { 52, -35 }, { 41, -10 }, { 36, 0 }, { 40, -1 },
  444. { 30, 14 }, { 28, 26 }, { 23, 37 }, { 12, 55 },
  445. { 11, 65 }, { 37, -33 }, { 39, -36 }, { 40, -37 },
  446. { 38, -30 }, { 46, -33 }, { 42, -30 }, { 40, -24 },
  447. { 49, -29 }, { 38, -12 }, { 40, -10 }, { 38, -3 },
  448. { 46, -5 }, { 31, 20 }, { 29, 30 }, { 25, 44 },
  449. { 12, 48 }, { 11, 49 }, { 26, 45 }, { 22, 22 },
  450. { 23, 22 }, { 27, 21 }, { 33, 20 }, { 26, 28 },
  451. { 30, 24 }, { 27, 34 }, { 18, 42 }, { 25, 39 },
  452. { 18, 50 }, { 12, 70 }, { 21, 54 }, { 14, 71 },
  453. { 11, 83 },
  454. /* 399 - 435 */
  455. { 25, 32 }, { 21, 49 }, { 21, 54 },
  456. { -5, 85 }, { -6, 81 }, { -10, 77 }, { -7, 81 },
  457. { -17, 80 }, { -18, 73 }, { -4, 74 }, { -10, 83 },
  458. { -9, 71 }, { -9, 67 }, { -1, 61 }, { -8, 66 },
  459. { -14, 66 }, { 0, 59 }, { 2, 59 }, { 17, -10 },
  460. { 32, -13 }, { 42, -9 }, { 49, -5 }, { 53, 0 },
  461. { 64, 3 }, { 68, 10 }, { 66, 27 }, { 47, 57 },
  462. { -5, 71 }, { 0, 24 }, { -1, 36 }, { -2, 42 },
  463. { -2, 52 }, { -9, 57 }, { -6, 63 }, { -4, 65 },
  464. { -4, 67 }, { -7, 82 },
  465. /* 436 - 459 */
  466. { -3, 81 }, { -3, 76 }, { -7, 72 }, { -6, 78 },
  467. { -12, 72 }, { -14, 68 }, { -3, 70 }, { -6, 76 },
  468. { -5, 66 }, { -5, 62 }, { 0, 57 }, { -4, 61 },
  469. { -9, 60 }, { 1, 54 }, { 2, 58 }, { 17, -10 },
  470. { 32, -13 }, { 42, -9 }, { 49, -5 }, { 53, 0 },
  471. { 64, 3 }, { 68, 10 }, { 66, 27 }, { 47, 57 },
  472. },
  473. /* i_cabac_init_idc == 2 */
  474. {
  475. /* 0 - 10 */
  476. { 20, -15 }, { 2, 54 }, { 3, 74 }, { 20, -15 },
  477. { 2, 54 }, { 3, 74 }, { -28, 127 }, { -23, 104 },
  478. { -6, 53 }, { -1, 54 }, { 7, 51 },
  479. /* 11 - 23 */
  480. { 29, 16 }, { 25, 0 }, { 14, 0 }, { -10, 51 },
  481. { -3, 62 }, { -27, 99 }, { 26, 16 }, { -4, 85 },
  482. { -24, 102 }, { 5, 57 }, { 6, 57 }, { -17, 73 },
  483. { 14, 57 },
  484. /* 24 - 39 */
  485. { 20, 40 }, { 20, 10 }, { 29, 0 }, { 54, 0 },
  486. { 37, 42 }, { 12, 97 }, { -32, 127 }, { -22, 117 },
  487. { -2, 74 }, { -4, 85 }, { -24, 102 }, { 5, 57 },
  488. { -6, 93 }, { -14, 88 }, { -6, 44 }, { 4, 55 },
  489. /* 40 - 53 */
  490. { -11, 89 },{ -15, 103 },{ -21, 116 },{ 19, 57 },
  491. { 20, 58 },{ 4, 84 },{ 6, 96 },{ 1, 63 },
  492. { -5, 85 },{ -13, 106 },{ 5, 63 },{ 6, 75 },
  493. { -3, 90 },{ -1, 101 },
  494. /* 54 - 59 */
  495. { 3, 55 },{ -4, 79 },{ -2, 75 },{ -12, 97 },
  496. { -7, 50 },{ 1, 60 },
  497. /* 60 - 69 */
  498. { 0, 41 }, { 0, 63 }, { 0, 63 }, { 0, 63 },
  499. { -9, 83 }, { 4, 86 }, { 0, 97 }, { -7, 72 },
  500. { 13, 41 }, { 3, 62 },
  501. /* 70 - 104 */
  502. { 7, 34 }, { -9, 88 }, { -20, 127 }, { -36, 127 },
  503. { -17, 91 }, { -14, 95 }, { -25, 84 }, { -25, 86 },
  504. { -12, 89 }, { -17, 91 }, { -31, 127 }, { -14, 76 },
  505. { -18, 103 }, { -13, 90 }, { -37, 127 }, { 11, 80 },
  506. { 5, 76 }, { 2, 84 }, { 5, 78 }, { -6, 55 },
  507. { 4, 61 }, { -14, 83 }, { -37, 127 }, { -5, 79 },
  508. { -11, 104 }, { -11, 91 }, { -30, 127 }, { 0, 65 },
  509. { -2, 79 }, { 0, 72 }, { -4, 92 }, { -6, 56 },
  510. { 3, 68 }, { -8, 71 }, { -13, 98 },
  511. /* 105 -> 165 */
  512. { -4, 86 }, { -12, 88 }, { -5, 82 }, { -3, 72 },
  513. { -4, 67 }, { -8, 72 }, { -16, 89 }, { -9, 69 },
  514. { -1, 59 }, { 5, 66 }, { 4, 57 }, { -4, 71 },
  515. { -2, 71 }, { 2, 58 }, { -1, 74 }, { -4, 44 },
  516. { -1, 69 }, { 0, 62 }, { -7, 51 }, { -4, 47 },
  517. { -6, 42 }, { -3, 41 }, { -6, 53 }, { 8, 76 },
  518. { -9, 78 }, { -11, 83 }, { 9, 52 }, { 0, 67 },
  519. { -5, 90 }, { 1, 67 }, { -15, 72 }, { -5, 75 },
  520. { -8, 80 }, { -21, 83 }, { -21, 64 }, { -13, 31 },
  521. { -25, 64 }, { -29, 94 }, { 9, 75 }, { 17, 63 },
  522. { -8, 74 }, { -5, 35 }, { -2, 27 }, { 13, 91 },
  523. { 3, 65 }, { -7, 69 }, { 8, 77 }, { -10, 66 },
  524. { 3, 62 }, { -3, 68 }, { -20, 81 }, { 0, 30 },
  525. { 1, 7 }, { -3, 23 }, { -21, 74 }, { 16, 66 },
  526. { -23, 124 }, { 17, 37 }, { 44, -18 }, { 50, -34 },
  527. { -22, 127 },
  528. /* 166 - 226 */
  529. { 4, 39 }, { 0, 42 }, { 7, 34 }, { 11, 29 },
  530. { 8, 31 }, { 6, 37 }, { 7, 42 }, { 3, 40 },
  531. { 8, 33 }, { 13, 43 }, { 13, 36 }, { 4, 47 },
  532. { 3, 55 }, { 2, 58 }, { 6, 60 }, { 8, 44 },
  533. { 11, 44 }, { 14, 42 }, { 7, 48 }, { 4, 56 },
  534. { 4, 52 }, { 13, 37 }, { 9, 49 }, { 19, 58 },
  535. { 10, 48 }, { 12, 45 }, { 0, 69 }, { 20, 33 },
  536. { 8, 63 }, { 35, -18 }, { 33, -25 }, { 28, -3 },
  537. { 24, 10 }, { 27, 0 }, { 34, -14 }, { 52, -44 },
  538. { 39, -24 }, { 19, 17 }, { 31, 25 }, { 36, 29 },
  539. { 24, 33 }, { 34, 15 }, { 30, 20 }, { 22, 73 },
  540. { 20, 34 }, { 19, 31 }, { 27, 44 }, { 19, 16 },
  541. { 15, 36 }, { 15, 36 }, { 21, 28 }, { 25, 21 },
  542. { 30, 20 }, { 31, 12 }, { 27, 16 }, { 24, 42 },
  543. { 0, 93 }, { 14, 56 }, { 15, 57 }, { 26, 38 },
  544. { -24, 127 },
  545. /* 227 - 275 */
  546. { -24, 115 }, { -22, 82 }, { -9, 62 }, { 0, 53 },
  547. { 0, 59 }, { -14, 85 }, { -13, 89 }, { -13, 94 },
  548. { -11, 92 }, { -29, 127 }, { -21, 100 }, { -14, 57 },
  549. { -12, 67 }, { -11, 71 }, { -10, 77 }, { -21, 85 },
  550. { -16, 88 }, { -23, 104 }, { -15, 98 }, { -37, 127 },
  551. { -10, 82 }, { -8, 48 }, { -8, 61 }, { -8, 66 },
  552. { -7, 70 }, { -14, 75 }, { -10, 79 }, { -9, 83 },
  553. { -12, 92 }, { -18, 108 }, { -4, 79 }, { -22, 69 },
  554. { -16, 75 }, { -2, 58 }, { 1, 58 }, { -13, 78 },
  555. { -9, 83 }, { -4, 81 }, { -13, 99 }, { -13, 81 },
  556. { -6, 38 }, { -13, 62 }, { -6, 58 }, { -2, 59 },
  557. { -16, 73 }, { -10, 76 }, { -13, 86 }, { -9, 83 },
  558. { -10, 87 },
  559. /* 276 a bit special (not used, bypass is used instead) */
  560. { 0, 0 },
  561. /* 277 - 337 */
  562. { -22, 127 }, { -25, 127 }, { -25, 120 }, { -27, 127 },
  563. { -19, 114 }, { -23, 117 }, { -25, 118 }, { -26, 117 },
  564. { -24, 113 }, { -28, 118 }, { -31, 120 }, { -37, 124 },
  565. { -10, 94 }, { -15, 102 }, { -10, 99 }, { -13, 106 },
  566. { -50, 127 }, { -5, 92 }, { 17, 57 }, { -5, 86 },
  567. { -13, 94 }, { -12, 91 }, { -2, 77 }, { 0, 71 },
  568. { -1, 73 }, { 4, 64 }, { -7, 81 }, { 5, 64 },
  569. { 15, 57 }, { 1, 67 }, { 0, 68 }, { -10, 67 },
  570. { 1, 68 }, { 0, 77 }, { 2, 64 }, { 0, 68 },
  571. { -5, 78 }, { 7, 55 }, { 5, 59 }, { 2, 65 },
  572. { 14, 54 }, { 15, 44 }, { 5, 60 }, { 2, 70 },
  573. { -2, 76 }, { -18, 86 }, { 12, 70 }, { 5, 64 },
  574. { -12, 70 }, { 11, 55 }, { 5, 56 }, { 0, 69 },
  575. { 2, 65 }, { -6, 74 }, { 5, 54 }, { 7, 54 },
  576. { -6, 76 }, { -11, 82 }, { -2, 77 }, { -2, 77 },
  577. { 25, 42 },
  578. /* 338 - 398 */
  579. { 17, -13 }, { 16, -9 }, { 17, -12 }, { 27, -21 },
  580. { 37, -30 }, { 41, -40 }, { 42, -41 }, { 48, -47 },
  581. { 39, -32 }, { 46, -40 }, { 52, -51 }, { 46, -41 },
  582. { 52, -39 }, { 43, -19 }, { 32, 11 }, { 61, -55 },
  583. { 56, -46 }, { 62, -50 }, { 81, -67 }, { 45, -20 },
  584. { 35, -2 }, { 28, 15 }, { 34, 1 }, { 39, 1 },
  585. { 30, 17 }, { 20, 38 }, { 18, 45 }, { 15, 54 },
  586. { 0, 79 }, { 36, -16 }, { 37, -14 }, { 37, -17 },
  587. { 32, 1 }, { 34, 15 }, { 29, 15 }, { 24, 25 },
  588. { 34, 22 }, { 31, 16 }, { 35, 18 }, { 31, 28 },
  589. { 33, 41 }, { 36, 28 }, { 27, 47 }, { 21, 62 },
  590. { 18, 31 }, { 19, 26 }, { 36, 24 }, { 24, 23 },
  591. { 27, 16 }, { 24, 30 }, { 31, 29 }, { 22, 41 },
  592. { 22, 42 }, { 16, 60 }, { 15, 52 }, { 14, 60 },
  593. { 3, 78 }, { -16, 123 }, { 21, 53 }, { 22, 56 },
  594. { 25, 61 },
  595. /* 399 - 435 */
  596. { 21, 33 }, { 19, 50 }, { 17, 61 },
  597. { -3, 78 }, { -8, 74 }, { -9, 72 }, { -10, 72 },
  598. { -18, 75 }, { -12, 71 }, { -11, 63 }, { -5, 70 },
  599. { -17, 75 }, { -14, 72 }, { -16, 67 }, { -8, 53 },
  600. { -14, 59 }, { -9, 52 }, { -11, 68 }, { 9, -2 },
  601. { 30, -10 }, { 31, -4 }, { 33, -1 }, { 33, 7 },
  602. { 31, 12 }, { 37, 23 }, { 31, 38 }, { 20, 64 },
  603. { -9, 71 }, { -7, 37 }, { -8, 44 }, { -11, 49 },
  604. { -10, 56 }, { -12, 59 }, { -8, 63 }, { -9, 67 },
  605. { -6, 68 }, { -10, 79 },
  606. /* 436 - 459 */
  607. { -3, 78 }, { -8, 74 }, { -9, 72 }, { -10, 72 },
  608. { -18, 75 }, { -12, 71 }, { -11, 63 }, { -5, 70 },
  609. { -17, 75 }, { -14, 72 }, { -16, 67 }, { -8, 53 },
  610. { -14, 59 }, { -9, 52 }, { -11, 68 }, { 9, -2 },
  611. { 30, -10 }, { 31, -4 }, { 33, -1 }, { 33, 7 },
  612. { 31, 12 }, { 37, 23 }, { 31, 38 }, { 20, 64 },
  613. }
  614. };
  615. void ff_h264_init_cabac_states(H264Context *h) {
  616. MpegEncContext * const s = &h->s;
  617. int i;
  618. /* calculate pre-state */
  619. for( i= 0; i < 460; i++ ) {
  620. int pre;
  621. if( h->slice_type_nos == FF_I_TYPE )
  622. pre = av_clip( ((cabac_context_init_I[i][0] * s->qscale) >>4 ) + cabac_context_init_I[i][1], 1, 126 );
  623. else
  624. pre = av_clip( ((cabac_context_init_PB[h->cabac_init_idc][i][0] * s->qscale) >>4 ) + cabac_context_init_PB[h->cabac_init_idc][i][1], 1, 126 );
  625. if( pre <= 63 )
  626. h->cabac_state[i] = 2 * ( 63 - pre ) + 0;
  627. else
  628. h->cabac_state[i] = 2 * ( pre - 64 ) + 1;
  629. }
  630. }
  631. static int decode_cabac_field_decoding_flag(H264Context *h) {
  632. MpegEncContext * const s = &h->s;
  633. const int mb_x = s->mb_x;
  634. const int mb_y = s->mb_y & ~1;
  635. const int mba_xy = mb_x - 1 + mb_y *s->mb_stride;
  636. const int mbb_xy = mb_x + (mb_y-2)*s->mb_stride;
  637. unsigned int ctx = 0;
  638. if( h->slice_table[mba_xy] == h->slice_num && IS_INTERLACED( s->current_picture.mb_type[mba_xy] ) ) {
  639. ctx += 1;
  640. }
  641. if( h->slice_table[mbb_xy] == h->slice_num && IS_INTERLACED( s->current_picture.mb_type[mbb_xy] ) ) {
  642. ctx += 1;
  643. }
  644. return get_cabac_noinline( &h->cabac, &h->cabac_state[70 + ctx] );
  645. }
  646. static int decode_cabac_intra_mb_type(H264Context *h, int ctx_base, int intra_slice) {
  647. uint8_t *state= &h->cabac_state[ctx_base];
  648. int mb_type;
  649. if(intra_slice){
  650. MpegEncContext * const s = &h->s;
  651. const int mba_xy = h->left_mb_xy[0];
  652. const int mbb_xy = h->top_mb_xy;
  653. int ctx=0;
  654. if( h->slice_table[mba_xy] == h->slice_num && !IS_INTRA4x4( s->current_picture.mb_type[mba_xy] ) )
  655. ctx++;
  656. if( h->slice_table[mbb_xy] == h->slice_num && !IS_INTRA4x4( s->current_picture.mb_type[mbb_xy] ) )
  657. ctx++;
  658. if( get_cabac_noinline( &h->cabac, &state[ctx] ) == 0 )
  659. return 0; /* I4x4 */
  660. state += 2;
  661. }else{
  662. if( get_cabac_noinline( &h->cabac, state ) == 0 )
  663. return 0; /* I4x4 */
  664. }
  665. if( get_cabac_terminate( &h->cabac ) )
  666. return 25; /* PCM */
  667. mb_type = 1; /* I16x16 */
  668. mb_type += 12 * get_cabac_noinline( &h->cabac, &state[1] ); /* cbp_luma != 0 */
  669. if( get_cabac_noinline( &h->cabac, &state[2] ) ) /* cbp_chroma */
  670. mb_type += 4 + 4 * get_cabac_noinline( &h->cabac, &state[2+intra_slice] );
  671. mb_type += 2 * get_cabac_noinline( &h->cabac, &state[3+intra_slice] );
  672. mb_type += 1 * get_cabac_noinline( &h->cabac, &state[3+2*intra_slice] );
  673. return mb_type;
  674. }
  675. static int decode_cabac_mb_type_b( H264Context *h ) {
  676. MpegEncContext * const s = &h->s;
  677. const int mba_xy = h->left_mb_xy[0];
  678. const int mbb_xy = h->top_mb_xy;
  679. int ctx = 0;
  680. int bits;
  681. assert(h->slice_type_nos == FF_B_TYPE);
  682. if( h->slice_table[mba_xy] == h->slice_num && !IS_DIRECT( s->current_picture.mb_type[mba_xy] ) )
  683. ctx++;
  684. if( h->slice_table[mbb_xy] == h->slice_num && !IS_DIRECT( s->current_picture.mb_type[mbb_xy] ) )
  685. ctx++;
  686. if( !get_cabac_noinline( &h->cabac, &h->cabac_state[27+ctx] ) )
  687. return 0; /* B_Direct_16x16 */
  688. if( !get_cabac_noinline( &h->cabac, &h->cabac_state[27+3] ) ) {
  689. return 1 + get_cabac_noinline( &h->cabac, &h->cabac_state[27+5] ); /* B_L[01]_16x16 */
  690. }
  691. bits = get_cabac_noinline( &h->cabac, &h->cabac_state[27+4] ) << 3;
  692. bits|= get_cabac_noinline( &h->cabac, &h->cabac_state[27+5] ) << 2;
  693. bits|= get_cabac_noinline( &h->cabac, &h->cabac_state[27+5] ) << 1;
  694. bits|= get_cabac_noinline( &h->cabac, &h->cabac_state[27+5] );
  695. if( bits < 8 )
  696. return bits + 3; /* B_Bi_16x16 through B_L1_L0_16x8 */
  697. else if( bits == 13 ) {
  698. return decode_cabac_intra_mb_type(h, 32, 0) + 23;
  699. } else if( bits == 14 )
  700. return 11; /* B_L1_L0_8x16 */
  701. else if( bits == 15 )
  702. return 22; /* B_8x8 */
  703. bits= ( bits<<1 ) | get_cabac_noinline( &h->cabac, &h->cabac_state[27+5] );
  704. return bits - 4; /* B_L0_Bi_* through B_Bi_Bi_* */
  705. }
  706. static int decode_cabac_mb_skip( H264Context *h, int mb_x, int mb_y ) {
  707. MpegEncContext * const s = &h->s;
  708. int mba_xy, mbb_xy;
  709. int ctx = 0;
  710. if(FRAME_MBAFF){ //FIXME merge with the stuff in fill_caches?
  711. int mb_xy = mb_x + (mb_y&~1)*s->mb_stride;
  712. mba_xy = mb_xy - 1;
  713. if( (mb_y&1)
  714. && h->slice_table[mba_xy] == h->slice_num
  715. && MB_FIELD == !!IS_INTERLACED( s->current_picture.mb_type[mba_xy] ) )
  716. mba_xy += s->mb_stride;
  717. if( MB_FIELD ){
  718. mbb_xy = mb_xy - s->mb_stride;
  719. if( !(mb_y&1)
  720. && h->slice_table[mbb_xy] == h->slice_num
  721. && IS_INTERLACED( s->current_picture.mb_type[mbb_xy] ) )
  722. mbb_xy -= s->mb_stride;
  723. }else
  724. mbb_xy = mb_x + (mb_y-1)*s->mb_stride;
  725. }else{
  726. int mb_xy = h->mb_xy;
  727. mba_xy = mb_xy - 1;
  728. mbb_xy = mb_xy - (s->mb_stride << FIELD_PICTURE);
  729. }
  730. if( h->slice_table[mba_xy] == h->slice_num && !IS_SKIP( s->current_picture.mb_type[mba_xy] ))
  731. ctx++;
  732. if( h->slice_table[mbb_xy] == h->slice_num && !IS_SKIP( s->current_picture.mb_type[mbb_xy] ))
  733. ctx++;
  734. if( h->slice_type_nos == FF_B_TYPE )
  735. ctx += 13;
  736. return get_cabac_noinline( &h->cabac, &h->cabac_state[11+ctx] );
  737. }
  738. static int decode_cabac_mb_intra4x4_pred_mode( H264Context *h, int pred_mode ) {
  739. int mode = 0;
  740. if( get_cabac( &h->cabac, &h->cabac_state[68] ) )
  741. return pred_mode;
  742. mode += 1 * get_cabac( &h->cabac, &h->cabac_state[69] );
  743. mode += 2 * get_cabac( &h->cabac, &h->cabac_state[69] );
  744. mode += 4 * get_cabac( &h->cabac, &h->cabac_state[69] );
  745. if( mode >= pred_mode )
  746. return mode + 1;
  747. else
  748. return mode;
  749. }
  750. static int decode_cabac_mb_chroma_pre_mode( H264Context *h) {
  751. const int mba_xy = h->left_mb_xy[0];
  752. const int mbb_xy = h->top_mb_xy;
  753. int ctx = 0;
  754. /* No need to test for IS_INTRA4x4 and IS_INTRA16x16, as we set chroma_pred_mode_table to 0 */
  755. if( h->slice_table[mba_xy] == h->slice_num && h->chroma_pred_mode_table[mba_xy] != 0 )
  756. ctx++;
  757. if( h->slice_table[mbb_xy] == h->slice_num && h->chroma_pred_mode_table[mbb_xy] != 0 )
  758. ctx++;
  759. if( get_cabac_noinline( &h->cabac, &h->cabac_state[64+ctx] ) == 0 )
  760. return 0;
  761. if( get_cabac_noinline( &h->cabac, &h->cabac_state[64+3] ) == 0 )
  762. return 1;
  763. if( get_cabac_noinline( &h->cabac, &h->cabac_state[64+3] ) == 0 )
  764. return 2;
  765. else
  766. return 3;
  767. }
  768. static int decode_cabac_mb_cbp_luma( H264Context *h) {
  769. int cbp_b, cbp_a, ctx, cbp = 0;
  770. cbp_a = h->slice_table[h->left_mb_xy[0]] == h->slice_num ? h->left_cbp : -1;
  771. cbp_b = h->slice_table[h->top_mb_xy] == h->slice_num ? h->top_cbp : -1;
  772. ctx = !(cbp_a & 0x02) + 2 * !(cbp_b & 0x04);
  773. cbp |= get_cabac_noinline(&h->cabac, &h->cabac_state[73 + ctx]);
  774. ctx = !(cbp & 0x01) + 2 * !(cbp_b & 0x08);
  775. cbp |= get_cabac_noinline(&h->cabac, &h->cabac_state[73 + ctx]) << 1;
  776. ctx = !(cbp_a & 0x08) + 2 * !(cbp & 0x01);
  777. cbp |= get_cabac_noinline(&h->cabac, &h->cabac_state[73 + ctx]) << 2;
  778. ctx = !(cbp & 0x04) + 2 * !(cbp & 0x02);
  779. cbp |= get_cabac_noinline(&h->cabac, &h->cabac_state[73 + ctx]) << 3;
  780. return cbp;
  781. }
  782. static int decode_cabac_mb_cbp_chroma( H264Context *h) {
  783. int ctx;
  784. int cbp_a, cbp_b;
  785. cbp_a = (h->left_cbp>>4)&0x03;
  786. cbp_b = (h-> top_cbp>>4)&0x03;
  787. ctx = 0;
  788. if( cbp_a > 0 ) ctx++;
  789. if( cbp_b > 0 ) ctx += 2;
  790. if( get_cabac_noinline( &h->cabac, &h->cabac_state[77 + ctx] ) == 0 )
  791. return 0;
  792. ctx = 4;
  793. if( cbp_a == 2 ) ctx++;
  794. if( cbp_b == 2 ) ctx += 2;
  795. return 1 + get_cabac_noinline( &h->cabac, &h->cabac_state[77 + ctx] );
  796. }
  797. static int decode_cabac_mb_dqp( H264Context *h) {
  798. int ctx= h->last_qscale_diff != 0;
  799. int val = 0;
  800. while( get_cabac_noinline( &h->cabac, &h->cabac_state[60 + ctx] ) ) {
  801. ctx= 2+(ctx>>1);
  802. val++;
  803. if(val > 102) //prevent infinite loop
  804. return INT_MIN;
  805. }
  806. if( val&0x01 )
  807. return (val + 1)>>1 ;
  808. else
  809. return -((val + 1)>>1);
  810. }
  811. static int decode_cabac_p_mb_sub_type( H264Context *h ) {
  812. if( get_cabac( &h->cabac, &h->cabac_state[21] ) )
  813. return 0; /* 8x8 */
  814. if( !get_cabac( &h->cabac, &h->cabac_state[22] ) )
  815. return 1; /* 8x4 */
  816. if( get_cabac( &h->cabac, &h->cabac_state[23] ) )
  817. return 2; /* 4x8 */
  818. return 3; /* 4x4 */
  819. }
  820. static int decode_cabac_b_mb_sub_type( H264Context *h ) {
  821. int type;
  822. if( !get_cabac( &h->cabac, &h->cabac_state[36] ) )
  823. return 0; /* B_Direct_8x8 */
  824. if( !get_cabac( &h->cabac, &h->cabac_state[37] ) )
  825. return 1 + get_cabac( &h->cabac, &h->cabac_state[39] ); /* B_L0_8x8, B_L1_8x8 */
  826. type = 3;
  827. if( get_cabac( &h->cabac, &h->cabac_state[38] ) ) {
  828. if( get_cabac( &h->cabac, &h->cabac_state[39] ) )
  829. return 11 + get_cabac( &h->cabac, &h->cabac_state[39] ); /* B_L1_4x4, B_Bi_4x4 */
  830. type += 4;
  831. }
  832. type += 2*get_cabac( &h->cabac, &h->cabac_state[39] );
  833. type += get_cabac( &h->cabac, &h->cabac_state[39] );
  834. return type;
  835. }
  836. static inline int decode_cabac_mb_transform_size( H264Context *h ) {
  837. return get_cabac_noinline( &h->cabac, &h->cabac_state[399 + h->neighbor_transform_size] );
  838. }
  839. static int decode_cabac_mb_ref( H264Context *h, int list, int n ) {
  840. int refa = h->ref_cache[list][scan8[n] - 1];
  841. int refb = h->ref_cache[list][scan8[n] - 8];
  842. int ref = 0;
  843. int ctx = 0;
  844. if( h->slice_type_nos == FF_B_TYPE) {
  845. if( refa > 0 && !h->direct_cache[scan8[n] - 1] )
  846. ctx++;
  847. if( refb > 0 && !h->direct_cache[scan8[n] - 8] )
  848. ctx += 2;
  849. } else {
  850. if( refa > 0 )
  851. ctx++;
  852. if( refb > 0 )
  853. ctx += 2;
  854. }
  855. while( get_cabac( &h->cabac, &h->cabac_state[54+ctx] ) ) {
  856. ref++;
  857. ctx = (ctx>>2)+4;
  858. if(ref >= 32 /*h->ref_list[list]*/){
  859. return -1;
  860. }
  861. }
  862. return ref;
  863. }
  864. static int decode_cabac_mb_mvd( H264Context *h, int list, int n, int l ) {
  865. int amvd = abs( h->mvd_cache[list][scan8[n] - 1][l] ) +
  866. abs( h->mvd_cache[list][scan8[n] - 8][l] );
  867. int ctxbase = (l == 0) ? 40 : 47;
  868. int mvd;
  869. int ctx = (amvd>2) + (amvd>32);
  870. if(!get_cabac(&h->cabac, &h->cabac_state[ctxbase+ctx]))
  871. return 0;
  872. mvd= 1;
  873. ctx= 3;
  874. while( mvd < 9 && get_cabac( &h->cabac, &h->cabac_state[ctxbase+ctx] ) ) {
  875. mvd++;
  876. if( ctx < 6 )
  877. ctx++;
  878. }
  879. if( mvd >= 9 ) {
  880. int k = 3;
  881. while( get_cabac_bypass( &h->cabac ) ) {
  882. mvd += 1 << k;
  883. k++;
  884. if(k>24){
  885. av_log(h->s.avctx, AV_LOG_ERROR, "overflow in decode_cabac_mb_mvd\n");
  886. return INT_MIN;
  887. }
  888. }
  889. while( k-- ) {
  890. if( get_cabac_bypass( &h->cabac ) )
  891. mvd += 1 << k;
  892. }
  893. }
  894. return get_cabac_bypass_sign( &h->cabac, -mvd );
  895. }
  896. static av_always_inline int get_cabac_cbf_ctx( H264Context *h, int cat, int idx, int is_dc ) {
  897. int nza, nzb;
  898. int ctx = 0;
  899. if( is_dc ) {
  900. if( cat == 0 ) {
  901. nza = h->left_cbp&0x100;
  902. nzb = h-> top_cbp&0x100;
  903. } else {
  904. nza = (h->left_cbp>>(6+idx))&0x01;
  905. nzb = (h-> top_cbp>>(6+idx))&0x01;
  906. }
  907. } else {
  908. assert(cat == 1 || cat == 2 || cat == 4);
  909. nza = h->non_zero_count_cache[scan8[idx] - 1];
  910. nzb = h->non_zero_count_cache[scan8[idx] - 8];
  911. }
  912. if( nza > 0 )
  913. ctx++;
  914. if( nzb > 0 )
  915. ctx += 2;
  916. return ctx + 4 * cat;
  917. }
  918. DECLARE_ASM_CONST(1, uint8_t, last_coeff_flag_offset_8x8)[63] = {
  919. 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  920. 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  921. 3, 3, 3, 3, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4,
  922. 5, 5, 5, 5, 6, 6, 6, 6, 7, 7, 7, 7, 8, 8, 8
  923. };
  924. static av_always_inline void decode_cabac_residual_internal( H264Context *h, DCTELEM *block, int cat, int n, const uint8_t *scantable, const uint32_t *qmul, int max_coeff, int is_dc ) {
  925. static const int significant_coeff_flag_offset[2][6] = {
  926. { 105+0, 105+15, 105+29, 105+44, 105+47, 402 },
  927. { 277+0, 277+15, 277+29, 277+44, 277+47, 436 }
  928. };
  929. static const int last_coeff_flag_offset[2][6] = {
  930. { 166+0, 166+15, 166+29, 166+44, 166+47, 417 },
  931. { 338+0, 338+15, 338+29, 338+44, 338+47, 451 }
  932. };
  933. static const int coeff_abs_level_m1_offset[6] = {
  934. 227+0, 227+10, 227+20, 227+30, 227+39, 426
  935. };
  936. static const uint8_t significant_coeff_flag_offset_8x8[2][63] = {
  937. { 0, 1, 2, 3, 4, 5, 5, 4, 4, 3, 3, 4, 4, 4, 5, 5,
  938. 4, 4, 4, 4, 3, 3, 6, 7, 7, 7, 8, 9,10, 9, 8, 7,
  939. 7, 6,11,12,13,11, 6, 7, 8, 9,14,10, 9, 8, 6,11,
  940. 12,13,11, 6, 9,14,10, 9,11,12,13,11,14,10,12 },
  941. { 0, 1, 1, 2, 2, 3, 3, 4, 5, 6, 7, 7, 7, 8, 4, 5,
  942. 6, 9,10,10, 8,11,12,11, 9, 9,10,10, 8,11,12,11,
  943. 9, 9,10,10, 8,11,12,11, 9, 9,10,10, 8,13,13, 9,
  944. 9,10,10, 8,13,13, 9, 9,10,10,14,14,14,14,14 }
  945. };
  946. /* node ctx: 0..3: abslevel1 (with abslevelgt1 == 0).
  947. * 4..7: abslevelgt1 + 3 (and abslevel1 doesn't matter).
  948. * map node ctx => cabac ctx for level=1 */
  949. static const uint8_t coeff_abs_level1_ctx[8] = { 1, 2, 3, 4, 0, 0, 0, 0 };
  950. /* map node ctx => cabac ctx for level>1 */
  951. static const uint8_t coeff_abs_levelgt1_ctx[8] = { 5, 5, 5, 5, 6, 7, 8, 9 };
  952. static const uint8_t coeff_abs_level_transition[2][8] = {
  953. /* update node ctx after decoding a level=1 */
  954. { 1, 2, 3, 3, 4, 5, 6, 7 },
  955. /* update node ctx after decoding a level>1 */
  956. { 4, 4, 4, 4, 5, 6, 7, 7 }
  957. };
  958. int index[64];
  959. int av_unused last;
  960. int coeff_count = 0;
  961. int node_ctx = 0;
  962. uint8_t *significant_coeff_ctx_base;
  963. uint8_t *last_coeff_ctx_base;
  964. uint8_t *abs_level_m1_ctx_base;
  965. #if !ARCH_X86
  966. #define CABAC_ON_STACK
  967. #endif
  968. #ifdef CABAC_ON_STACK
  969. #define CC &cc
  970. CABACContext cc;
  971. cc.range = h->cabac.range;
  972. cc.low = h->cabac.low;
  973. cc.bytestream= h->cabac.bytestream;
  974. #else
  975. #define CC &h->cabac
  976. #endif
  977. /* cat: 0-> DC 16x16 n = 0
  978. * 1-> AC 16x16 n = luma4x4idx
  979. * 2-> Luma4x4 n = luma4x4idx
  980. * 3-> DC Chroma n = iCbCr
  981. * 4-> AC Chroma n = 16 + 4 * iCbCr + chroma4x4idx
  982. * 5-> Luma8x8 n = 4 * luma8x8idx
  983. */
  984. /* read coded block flag */
  985. if( is_dc || cat != 5 ) {
  986. if( get_cabac( CC, &h->cabac_state[85 + get_cabac_cbf_ctx( h, cat, n, is_dc ) ] ) == 0 ) {
  987. if( !is_dc )
  988. h->non_zero_count_cache[scan8[n]] = 0;
  989. #ifdef CABAC_ON_STACK
  990. h->cabac.range = cc.range ;
  991. h->cabac.low = cc.low ;
  992. h->cabac.bytestream= cc.bytestream;
  993. #endif
  994. return;
  995. }
  996. }
  997. significant_coeff_ctx_base = h->cabac_state
  998. + significant_coeff_flag_offset[MB_FIELD][cat];
  999. last_coeff_ctx_base = h->cabac_state
  1000. + last_coeff_flag_offset[MB_FIELD][cat];
  1001. abs_level_m1_ctx_base = h->cabac_state
  1002. + coeff_abs_level_m1_offset[cat];
  1003. if( !is_dc && cat == 5 ) {
  1004. #define DECODE_SIGNIFICANCE( coefs, sig_off, last_off ) \
  1005. for(last= 0; last < coefs; last++) { \
  1006. uint8_t *sig_ctx = significant_coeff_ctx_base + sig_off; \
  1007. if( get_cabac( CC, sig_ctx )) { \
  1008. uint8_t *last_ctx = last_coeff_ctx_base + last_off; \
  1009. index[coeff_count++] = last; \
  1010. if( get_cabac( CC, last_ctx ) ) { \
  1011. last= max_coeff; \
  1012. break; \
  1013. } \
  1014. } \
  1015. }\
  1016. if( last == max_coeff -1 ) {\
  1017. index[coeff_count++] = last;\
  1018. }
  1019. const uint8_t *sig_off = significant_coeff_flag_offset_8x8[MB_FIELD];
  1020. #if ARCH_X86 && HAVE_7REGS && HAVE_EBX_AVAILABLE && !defined(BROKEN_RELOCATIONS)
  1021. coeff_count= decode_significance_8x8_x86(CC, significant_coeff_ctx_base, index, sig_off);
  1022. } else {
  1023. coeff_count= decode_significance_x86(CC, max_coeff, significant_coeff_ctx_base, index);
  1024. #else
  1025. DECODE_SIGNIFICANCE( 63, sig_off[last], last_coeff_flag_offset_8x8[last] );
  1026. } else {
  1027. DECODE_SIGNIFICANCE( max_coeff - 1, last, last );
  1028. #endif
  1029. }
  1030. assert(coeff_count > 0);
  1031. if( is_dc ) {
  1032. if( cat == 0 )
  1033. h->cbp_table[h->mb_xy] |= 0x100;
  1034. else
  1035. h->cbp_table[h->mb_xy] |= 0x40 << n;
  1036. } else {
  1037. if( cat == 5 )
  1038. fill_rectangle(&h->non_zero_count_cache[scan8[n]], 2, 2, 8, coeff_count, 1);
  1039. else {
  1040. assert( cat == 1 || cat == 2 || cat == 4 );
  1041. h->non_zero_count_cache[scan8[n]] = coeff_count;
  1042. }
  1043. }
  1044. do {
  1045. uint8_t *ctx = coeff_abs_level1_ctx[node_ctx] + abs_level_m1_ctx_base;
  1046. int j= scantable[index[--coeff_count]];
  1047. if( get_cabac( CC, ctx ) == 0 ) {
  1048. node_ctx = coeff_abs_level_transition[0][node_ctx];
  1049. if( is_dc ) {
  1050. block[j] = get_cabac_bypass_sign( CC, -1);
  1051. }else{
  1052. block[j] = (get_cabac_bypass_sign( CC, -qmul[j]) + 32) >> 6;
  1053. }
  1054. } else {
  1055. int coeff_abs = 2;
  1056. ctx = coeff_abs_levelgt1_ctx[node_ctx] + abs_level_m1_ctx_base;
  1057. node_ctx = coeff_abs_level_transition[1][node_ctx];
  1058. while( coeff_abs < 15 && get_cabac( CC, ctx ) ) {
  1059. coeff_abs++;
  1060. }
  1061. if( coeff_abs >= 15 ) {
  1062. int j = 0;
  1063. while( get_cabac_bypass( CC ) ) {
  1064. j++;
  1065. }
  1066. coeff_abs=1;
  1067. while( j-- ) {
  1068. coeff_abs += coeff_abs + get_cabac_bypass( CC );
  1069. }
  1070. coeff_abs+= 14;
  1071. }
  1072. if( is_dc ) {
  1073. block[j] = get_cabac_bypass_sign( CC, -coeff_abs );
  1074. }else{
  1075. block[j] = (get_cabac_bypass_sign( CC, -coeff_abs ) * qmul[j] + 32) >> 6;
  1076. }
  1077. }
  1078. } while( coeff_count );
  1079. #ifdef CABAC_ON_STACK
  1080. h->cabac.range = cc.range ;
  1081. h->cabac.low = cc.low ;
  1082. h->cabac.bytestream= cc.bytestream;
  1083. #endif
  1084. }
  1085. #if !CONFIG_SMALL
  1086. static void decode_cabac_residual_dc( H264Context *h, DCTELEM *block, int cat, int n, const uint8_t *scantable, const uint32_t *qmul, int max_coeff ) {
  1087. decode_cabac_residual_internal(h, block, cat, n, scantable, qmul, max_coeff, 1);
  1088. }
  1089. static void decode_cabac_residual_nondc( H264Context *h, DCTELEM *block, int cat, int n, const uint8_t *scantable, const uint32_t *qmul, int max_coeff ) {
  1090. decode_cabac_residual_internal(h, block, cat, n, scantable, qmul, max_coeff, 0);
  1091. }
  1092. #endif
  1093. static void decode_cabac_residual( H264Context *h, DCTELEM *block, int cat, int n, const uint8_t *scantable, const uint32_t *qmul, int max_coeff ) {
  1094. #if CONFIG_SMALL
  1095. decode_cabac_residual_internal(h, block, cat, n, scantable, qmul, max_coeff, cat == 0 || cat == 3);
  1096. #else
  1097. if( cat == 0 || cat == 3 ) decode_cabac_residual_dc(h, block, cat, n, scantable, qmul, max_coeff);
  1098. else decode_cabac_residual_nondc(h, block, cat, n, scantable, qmul, max_coeff);
  1099. #endif
  1100. }
  1101. static inline void compute_mb_neighbors(H264Context *h)
  1102. {
  1103. MpegEncContext * const s = &h->s;
  1104. const int mb_xy = h->mb_xy;
  1105. h->top_mb_xy = mb_xy - s->mb_stride;
  1106. h->left_mb_xy[0] = mb_xy - 1;
  1107. if(FRAME_MBAFF){
  1108. const int pair_xy = s->mb_x + (s->mb_y & ~1)*s->mb_stride;
  1109. const int top_pair_xy = pair_xy - s->mb_stride;
  1110. const int top_mb_field_flag = IS_INTERLACED(s->current_picture.mb_type[top_pair_xy]);
  1111. const int left_mb_field_flag = IS_INTERLACED(s->current_picture.mb_type[pair_xy-1]);
  1112. const int curr_mb_field_flag = MB_FIELD;
  1113. const int bottom = (s->mb_y & 1);
  1114. if (curr_mb_field_flag && (bottom || top_mb_field_flag)){
  1115. h->top_mb_xy -= s->mb_stride;
  1116. }
  1117. if (!left_mb_field_flag == curr_mb_field_flag) {
  1118. h->left_mb_xy[0] = pair_xy - 1;
  1119. }
  1120. } else if (FIELD_PICTURE) {
  1121. h->top_mb_xy -= s->mb_stride;
  1122. }
  1123. return;
  1124. }
  1125. /**
  1126. * decodes a macroblock
  1127. * @returns 0 if OK, AC_ERROR / DC_ERROR / MV_ERROR if an error is noticed
  1128. */
  1129. int ff_h264_decode_mb_cabac(H264Context *h) {
  1130. MpegEncContext * const s = &h->s;
  1131. int mb_xy;
  1132. int mb_type, partition_count, cbp = 0;
  1133. int dct8x8_allowed= h->pps.transform_8x8_mode;
  1134. mb_xy = h->mb_xy = s->mb_x + s->mb_y*s->mb_stride;
  1135. tprintf(s->avctx, "pic:%d mb:%d/%d\n", h->frame_num, s->mb_x, s->mb_y);
  1136. if( h->slice_type_nos != FF_I_TYPE ) {
  1137. int skip;
  1138. /* a skipped mb needs the aff flag from the following mb */
  1139. if( FRAME_MBAFF && s->mb_x==0 && (s->mb_y&1)==0 )
  1140. predict_field_decoding_flag(h);
  1141. if( FRAME_MBAFF && (s->mb_y&1)==1 && h->prev_mb_skipped )
  1142. skip = h->next_mb_skipped;
  1143. else
  1144. skip = decode_cabac_mb_skip( h, s->mb_x, s->mb_y );
  1145. /* read skip flags */
  1146. if( skip ) {
  1147. if( FRAME_MBAFF && (s->mb_y&1)==0 ){
  1148. s->current_picture.mb_type[mb_xy] = MB_TYPE_SKIP;
  1149. h->next_mb_skipped = decode_cabac_mb_skip( h, s->mb_x, s->mb_y+1 );
  1150. if(!h->next_mb_skipped)
  1151. h->mb_mbaff = h->mb_field_decoding_flag = decode_cabac_field_decoding_flag(h);
  1152. }
  1153. decode_mb_skip(h);
  1154. h->cbp_table[mb_xy] = 0;
  1155. h->chroma_pred_mode_table[mb_xy] = 0;
  1156. h->last_qscale_diff = 0;
  1157. return 0;
  1158. }
  1159. }
  1160. if(FRAME_MBAFF){
  1161. if( (s->mb_y&1) == 0 )
  1162. h->mb_mbaff =
  1163. h->mb_field_decoding_flag = decode_cabac_field_decoding_flag(h);
  1164. }
  1165. h->prev_mb_skipped = 0;
  1166. compute_mb_neighbors(h);
  1167. if( h->slice_type_nos == FF_B_TYPE ) {
  1168. mb_type = decode_cabac_mb_type_b( h );
  1169. if( mb_type < 23 ){
  1170. partition_count= b_mb_type_info[mb_type].partition_count;
  1171. mb_type= b_mb_type_info[mb_type].type;
  1172. }else{
  1173. mb_type -= 23;
  1174. goto decode_intra_mb;
  1175. }
  1176. } else if( h->slice_type_nos == FF_P_TYPE ) {
  1177. if( get_cabac_noinline( &h->cabac, &h->cabac_state[14] ) == 0 ) {
  1178. /* P-type */
  1179. if( get_cabac_noinline( &h->cabac, &h->cabac_state[15] ) == 0 ) {
  1180. /* P_L0_D16x16, P_8x8 */
  1181. mb_type= 3 * get_cabac_noinline( &h->cabac, &h->cabac_state[16] );
  1182. } else {
  1183. /* P_L0_D8x16, P_L0_D16x8 */
  1184. mb_type= 2 - get_cabac_noinline( &h->cabac, &h->cabac_state[17] );
  1185. }
  1186. partition_count= p_mb_type_info[mb_type].partition_count;
  1187. mb_type= p_mb_type_info[mb_type].type;
  1188. } else {
  1189. mb_type= decode_cabac_intra_mb_type(h, 17, 0);
  1190. goto decode_intra_mb;
  1191. }
  1192. } else {
  1193. mb_type= decode_cabac_intra_mb_type(h, 3, 1);
  1194. if(h->slice_type == FF_SI_TYPE && mb_type)
  1195. mb_type--;
  1196. assert(h->slice_type_nos == FF_I_TYPE);
  1197. decode_intra_mb:
  1198. partition_count = 0;
  1199. cbp= i_mb_type_info[mb_type].cbp;
  1200. h->intra16x16_pred_mode= i_mb_type_info[mb_type].pred_mode;
  1201. mb_type= i_mb_type_info[mb_type].type;
  1202. }
  1203. if(MB_FIELD)
  1204. mb_type |= MB_TYPE_INTERLACED;
  1205. h->slice_table[ mb_xy ]= h->slice_num;
  1206. if(IS_INTRA_PCM(mb_type)) {
  1207. const uint8_t *ptr;
  1208. // We assume these blocks are very rare so we do not optimize it.
  1209. // FIXME The two following lines get the bitstream position in the cabac
  1210. // decode, I think it should be done by a function in cabac.h (or cabac.c).
  1211. ptr= h->cabac.bytestream;
  1212. if(h->cabac.low&0x1) ptr--;
  1213. if(CABAC_BITS==16){
  1214. if(h->cabac.low&0x1FF) ptr--;
  1215. }
  1216. // The pixels are stored in the same order as levels in h->mb array.
  1217. memcpy(h->mb, ptr, 256); ptr+=256;
  1218. if(CHROMA){
  1219. memcpy(h->mb+128, ptr, 128); ptr+=128;
  1220. }
  1221. ff_init_cabac_decoder(&h->cabac, ptr, h->cabac.bytestream_end - ptr);
  1222. // All blocks are present
  1223. h->cbp_table[mb_xy] = 0x1ef;
  1224. h->chroma_pred_mode_table[mb_xy] = 0;
  1225. // In deblocking, the quantizer is 0
  1226. s->current_picture.qscale_table[mb_xy]= 0;
  1227. // All coeffs are present
  1228. memset(h->non_zero_count[mb_xy], 16, 32);
  1229. s->current_picture.mb_type[mb_xy]= mb_type;
  1230. h->last_qscale_diff = 0;
  1231. return 0;
  1232. }
  1233. if(MB_MBAFF){
  1234. h->ref_count[0] <<= 1;
  1235. h->ref_count[1] <<= 1;
  1236. }
  1237. fill_decode_caches(h, mb_type);
  1238. if( IS_INTRA( mb_type ) ) {
  1239. int i, pred_mode;
  1240. if( IS_INTRA4x4( mb_type ) ) {
  1241. if( dct8x8_allowed && decode_cabac_mb_transform_size( h ) ) {
  1242. mb_type |= MB_TYPE_8x8DCT;
  1243. for( i = 0; i < 16; i+=4 ) {
  1244. int pred = pred_intra_mode( h, i );
  1245. int mode = decode_cabac_mb_intra4x4_pred_mode( h, pred );
  1246. fill_rectangle( &h->intra4x4_pred_mode_cache[ scan8[i] ], 2, 2, 8, mode, 1 );
  1247. }
  1248. } else {
  1249. for( i = 0; i < 16; i++ ) {
  1250. int pred = pred_intra_mode( h, i );
  1251. h->intra4x4_pred_mode_cache[ scan8[i] ] = decode_cabac_mb_intra4x4_pred_mode( h, pred );
  1252. //av_log( s->avctx, AV_LOG_ERROR, "i4x4 pred=%d mode=%d\n", pred, h->intra4x4_pred_mode_cache[ scan8[i] ] );
  1253. }
  1254. }
  1255. ff_h264_write_back_intra_pred_mode(h);
  1256. if( ff_h264_check_intra4x4_pred_mode(h) < 0 ) return -1;
  1257. } else {
  1258. h->intra16x16_pred_mode= ff_h264_check_intra_pred_mode( h, h->intra16x16_pred_mode );
  1259. if( h->intra16x16_pred_mode < 0 ) return -1;
  1260. }
  1261. if(CHROMA){
  1262. h->chroma_pred_mode_table[mb_xy] =
  1263. pred_mode = decode_cabac_mb_chroma_pre_mode( h );
  1264. pred_mode= ff_h264_check_intra_pred_mode( h, pred_mode );
  1265. if( pred_mode < 0 ) return -1;
  1266. h->chroma_pred_mode= pred_mode;
  1267. }
  1268. } else if( partition_count == 4 ) {
  1269. int i, j, sub_partition_count[4], list, ref[2][4];
  1270. if( h->slice_type_nos == FF_B_TYPE ) {
  1271. for( i = 0; i < 4; i++ ) {
  1272. h->sub_mb_type[i] = decode_cabac_b_mb_sub_type( h );
  1273. sub_partition_count[i]= b_sub_mb_type_info[ h->sub_mb_type[i] ].partition_count;
  1274. h->sub_mb_type[i]= b_sub_mb_type_info[ h->sub_mb_type[i] ].type;
  1275. }
  1276. if( IS_DIRECT(h->sub_mb_type[0] | h->sub_mb_type[1] |
  1277. h->sub_mb_type[2] | h->sub_mb_type[3]) ) {
  1278. ff_h264_pred_direct_motion(h, &mb_type);
  1279. h->ref_cache[0][scan8[4]] =
  1280. h->ref_cache[1][scan8[4]] =
  1281. h->ref_cache[0][scan8[12]] =
  1282. h->ref_cache[1][scan8[12]] = PART_NOT_AVAILABLE;
  1283. if( h->ref_count[0] > 1 || h->ref_count[1] > 1 ) {
  1284. for( i = 0; i < 4; i++ )
  1285. if( IS_DIRECT(h->sub_mb_type[i]) )
  1286. fill_rectangle( &h->direct_cache[scan8[4*i]], 2, 2, 8, 1, 1 );
  1287. }
  1288. }
  1289. } else {
  1290. for( i = 0; i < 4; i++ ) {
  1291. h->sub_mb_type[i] = decode_cabac_p_mb_sub_type( h );
  1292. sub_partition_count[i]= p_sub_mb_type_info[ h->sub_mb_type[i] ].partition_count;
  1293. h->sub_mb_type[i]= p_sub_mb_type_info[ h->sub_mb_type[i] ].type;
  1294. }
  1295. }
  1296. for( list = 0; list < h->list_count; list++ ) {
  1297. for( i = 0; i < 4; i++ ) {
  1298. if(IS_DIRECT(h->sub_mb_type[i])) continue;
  1299. if(IS_DIR(h->sub_mb_type[i], 0, list)){
  1300. if( h->ref_count[list] > 1 ){
  1301. ref[list][i] = decode_cabac_mb_ref( h, list, 4*i );
  1302. if(ref[list][i] >= (unsigned)h->ref_count[list]){
  1303. av_log(s->avctx, AV_LOG_ERROR, "Reference %d >= %d\n", ref[list][i], h->ref_count[list]);
  1304. return -1;
  1305. }
  1306. }else
  1307. ref[list][i] = 0;
  1308. } else {
  1309. ref[list][i] = -1;
  1310. }
  1311. h->ref_cache[list][ scan8[4*i]+1 ]=
  1312. h->ref_cache[list][ scan8[4*i]+8 ]=h->ref_cache[list][ scan8[4*i]+9 ]= ref[list][i];
  1313. }
  1314. }
  1315. if(dct8x8_allowed)
  1316. dct8x8_allowed = get_dct8x8_allowed(h);
  1317. for(list=0; list<h->list_count; list++){
  1318. for(i=0; i<4; i++){
  1319. h->ref_cache[list][ scan8[4*i] ]=h->ref_cache[list][ scan8[4*i]+1 ];
  1320. if(IS_DIRECT(h->sub_mb_type[i])){
  1321. fill_rectangle(h->mvd_cache[list][scan8[4*i]], 2, 2, 8, 0, 4);
  1322. continue;
  1323. }
  1324. if(IS_DIR(h->sub_mb_type[i], 0, list) && !IS_DIRECT(h->sub_mb_type[i])){
  1325. const int sub_mb_type= h->sub_mb_type[i];
  1326. const int block_width= (sub_mb_type & (MB_TYPE_16x16|MB_TYPE_16x8)) ? 2 : 1;
  1327. for(j=0; j<sub_partition_count[i]; j++){
  1328. int mpx, mpy;
  1329. int mx, my;
  1330. const int index= 4*i + block_width*j;
  1331. int16_t (* mv_cache)[2]= &h->mv_cache[list][ scan8[index] ];
  1332. int16_t (* mvd_cache)[2]= &h->mvd_cache[list][ scan8[index] ];
  1333. pred_motion(h, index, block_width, list, h->ref_cache[list][ scan8[index] ], &mpx, &mpy);
  1334. mx = mpx + decode_cabac_mb_mvd( h, list, index, 0 );
  1335. my = mpy + decode_cabac_mb_mvd( h, list, index, 1 );
  1336. tprintf(s->avctx, "final mv:%d %d\n", mx, my);
  1337. if(IS_SUB_8X8(sub_mb_type)){
  1338. mv_cache[ 1 ][0]=
  1339. mv_cache[ 8 ][0]= mv_cache[ 9 ][0]= mx;
  1340. mv_cache[ 1 ][1]=
  1341. mv_cache[ 8 ][1]= mv_cache[ 9 ][1]= my;
  1342. mvd_cache[ 1 ][0]=
  1343. mvd_cache[ 8 ][0]= mvd_cache[ 9 ][0]= mx - mpx;
  1344. mvd_cache[ 1 ][1]=
  1345. mvd_cache[ 8 ][1]= mvd_cache[ 9 ][1]= my - mpy;
  1346. }else if(IS_SUB_8X4(sub_mb_type)){
  1347. mv_cache[ 1 ][0]= mx;
  1348. mv_cache[ 1 ][1]= my;
  1349. mvd_cache[ 1 ][0]= mx - mpx;
  1350. mvd_cache[ 1 ][1]= my - mpy;
  1351. }else if(IS_SUB_4X8(sub_mb_type)){
  1352. mv_cache[ 8 ][0]= mx;
  1353. mv_cache[ 8 ][1]= my;
  1354. mvd_cache[ 8 ][0]= mx - mpx;
  1355. mvd_cache[ 8 ][1]= my - mpy;
  1356. }
  1357. mv_cache[ 0 ][0]= mx;
  1358. mv_cache[ 0 ][1]= my;
  1359. mvd_cache[ 0 ][0]= mx - mpx;
  1360. mvd_cache[ 0 ][1]= my - mpy;
  1361. }
  1362. }else{
  1363. uint32_t *p= (uint32_t *)&h->mv_cache[list][ scan8[4*i] ][0];
  1364. uint32_t *pd= (uint32_t *)&h->mvd_cache[list][ scan8[4*i] ][0];
  1365. p[0] = p[1] = p[8] = p[9] = 0;
  1366. pd[0]= pd[1]= pd[8]= pd[9]= 0;
  1367. }
  1368. }
  1369. }
  1370. } else if( IS_DIRECT(mb_type) ) {
  1371. ff_h264_pred_direct_motion(h, &mb_type);
  1372. fill_rectangle(h->mvd_cache[0][scan8[0]], 4, 4, 8, 0, 4);
  1373. fill_rectangle(h->mvd_cache[1][scan8[0]], 4, 4, 8, 0, 4);
  1374. dct8x8_allowed &= h->sps.direct_8x8_inference_flag;
  1375. } else {
  1376. int list, mx, my, i, mpx, mpy;
  1377. if(IS_16X16(mb_type)){
  1378. for(list=0; list<h->list_count; list++){
  1379. if(IS_DIR(mb_type, 0, list)){
  1380. int ref;
  1381. if(h->ref_count[list] > 1){
  1382. ref= decode_cabac_mb_ref(h, list, 0);
  1383. if(ref >= (unsigned)h->ref_count[list]){
  1384. av_log(s->avctx, AV_LOG_ERROR, "Reference %d >= %d\n", ref, h->ref_count[list]);
  1385. return -1;
  1386. }
  1387. }else
  1388. ref=0;
  1389. fill_rectangle(&h->ref_cache[list][ scan8[0] ], 4, 4, 8, ref, 1);
  1390. }else
  1391. fill_rectangle(&h->ref_cache[list][ scan8[0] ], 4, 4, 8, (uint8_t)LIST_NOT_USED, 1); //FIXME factorize and the other fill_rect below too
  1392. }
  1393. for(list=0; list<h->list_count; list++){
  1394. if(IS_DIR(mb_type, 0, list)){
  1395. pred_motion(h, 0, 4, list, h->ref_cache[list][ scan8[0] ], &mpx, &mpy);
  1396. mx = mpx + decode_cabac_mb_mvd( h, list, 0, 0 );
  1397. my = mpy + decode_cabac_mb_mvd( h, list, 0, 1 );
  1398. tprintf(s->avctx, "final mv:%d %d\n", mx, my);
  1399. fill_rectangle(h->mvd_cache[list][ scan8[0] ], 4, 4, 8, pack16to32(mx-mpx,my-mpy), 4);
  1400. fill_rectangle(h->mv_cache[list][ scan8[0] ], 4, 4, 8, pack16to32(mx,my), 4);
  1401. }else
  1402. fill_rectangle(h->mv_cache[list][ scan8[0] ], 4, 4, 8, 0, 4);
  1403. }
  1404. }
  1405. else if(IS_16X8(mb_type)){
  1406. for(list=0; list<h->list_count; list++){
  1407. for(i=0; i<2; i++){
  1408. if(IS_DIR(mb_type, i, list)){
  1409. int ref;
  1410. if(h->ref_count[list] > 1){
  1411. ref= decode_cabac_mb_ref( h, list, 8*i );
  1412. if(ref >= (unsigned)h->ref_count[list]){
  1413. av_log(s->avctx, AV_LOG_ERROR, "Reference %d >= %d\n", ref, h->ref_count[list]);
  1414. return -1;
  1415. }
  1416. }else
  1417. ref=0;
  1418. fill_rectangle(&h->ref_cache[list][ scan8[0] + 16*i ], 4, 2, 8, ref, 1);
  1419. }else
  1420. fill_rectangle(&h->ref_cache[list][ scan8[0] + 16*i ], 4, 2, 8, (LIST_NOT_USED&0xFF), 1);
  1421. }
  1422. }
  1423. for(list=0; list<h->list_count; list++){
  1424. for(i=0; i<2; i++){
  1425. if(IS_DIR(mb_type, i, list)){
  1426. pred_16x8_motion(h, 8*i, list, h->ref_cache[list][scan8[0] + 16*i], &mpx, &mpy);
  1427. mx = mpx + decode_cabac_mb_mvd( h, list, 8*i, 0 );
  1428. my = mpy + decode_cabac_mb_mvd( h, list, 8*i, 1 );
  1429. tprintf(s->avctx, "final mv:%d %d\n", mx, my);
  1430. fill_rectangle(h->mvd_cache[list][ scan8[0] + 16*i ], 4, 2, 8, pack16to32(mx-mpx,my-mpy), 4);
  1431. fill_rectangle(h->mv_cache[list][ scan8[0] + 16*i ], 4, 2, 8, pack16to32(mx,my), 4);
  1432. }else{
  1433. fill_rectangle(h->mvd_cache[list][ scan8[0] + 16*i ], 4, 2, 8, 0, 4);
  1434. fill_rectangle(h-> mv_cache[list][ scan8[0] + 16*i ], 4, 2, 8, 0, 4);
  1435. }
  1436. }
  1437. }
  1438. }else{
  1439. assert(IS_8X16(mb_type));
  1440. for(list=0; list<h->list_count; list++){
  1441. for(i=0; i<2; i++){
  1442. if(IS_DIR(mb_type, i, list)){ //FIXME optimize
  1443. int ref;
  1444. if(h->ref_count[list] > 1){
  1445. ref= decode_cabac_mb_ref( h, list, 4*i );
  1446. if(ref >= (unsigned)h->ref_count[list]){
  1447. av_log(s->avctx, AV_LOG_ERROR, "Reference %d >= %d\n", ref, h->ref_count[list]);
  1448. return -1;
  1449. }
  1450. }else
  1451. ref=0;
  1452. fill_rectangle(&h->ref_cache[list][ scan8[0] + 2*i ], 2, 4, 8, ref, 1);
  1453. }else
  1454. fill_rectangle(&h->ref_cache[list][ scan8[0] + 2*i ], 2, 4, 8, (LIST_NOT_USED&0xFF), 1);
  1455. }
  1456. }
  1457. for(list=0; list<h->list_count; list++){
  1458. for(i=0; i<2; i++){
  1459. if(IS_DIR(mb_type, i, list)){
  1460. pred_8x16_motion(h, i*4, list, h->ref_cache[list][ scan8[0] + 2*i ], &mpx, &mpy);
  1461. mx = mpx + decode_cabac_mb_mvd( h, list, 4*i, 0 );
  1462. my = mpy + decode_cabac_mb_mvd( h, list, 4*i, 1 );
  1463. tprintf(s->avctx, "final mv:%d %d\n", mx, my);
  1464. fill_rectangle(h->mvd_cache[list][ scan8[0] + 2*i ], 2, 4, 8, pack16to32(mx-mpx,my-mpy), 4);
  1465. fill_rectangle(h->mv_cache[list][ scan8[0] + 2*i ], 2, 4, 8, pack16to32(mx,my), 4);
  1466. }else{
  1467. fill_rectangle(h->mvd_cache[list][ scan8[0] + 2*i ], 2, 4, 8, 0, 4);
  1468. fill_rectangle(h-> mv_cache[list][ scan8[0] + 2*i ], 2, 4, 8, 0, 4);
  1469. }
  1470. }
  1471. }
  1472. }
  1473. }
  1474. if( IS_INTER( mb_type ) ) {
  1475. h->chroma_pred_mode_table[mb_xy] = 0;
  1476. write_back_motion( h, mb_type );
  1477. }
  1478. if( !IS_INTRA16x16( mb_type ) ) {
  1479. cbp = decode_cabac_mb_cbp_luma( h );
  1480. if(CHROMA)
  1481. cbp |= decode_cabac_mb_cbp_chroma( h ) << 4;
  1482. }
  1483. h->cbp_table[mb_xy] = h->cbp = cbp;
  1484. if( dct8x8_allowed && (cbp&15) && !IS_INTRA( mb_type ) ) {
  1485. if( decode_cabac_mb_transform_size( h ) )
  1486. mb_type |= MB_TYPE_8x8DCT;
  1487. }
  1488. s->current_picture.mb_type[mb_xy]= mb_type;
  1489. if( cbp || IS_INTRA16x16( mb_type ) ) {
  1490. const uint8_t *scan, *scan8x8, *dc_scan;
  1491. const uint32_t *qmul;
  1492. int dqp;
  1493. if(IS_INTERLACED(mb_type)){
  1494. scan8x8= s->qscale ? h->field_scan8x8 : h->field_scan8x8_q0;
  1495. scan= s->qscale ? h->field_scan : h->field_scan_q0;
  1496. dc_scan= luma_dc_field_scan;
  1497. }else{
  1498. scan8x8= s->qscale ? h->zigzag_scan8x8 : h->zigzag_scan8x8_q0;
  1499. scan= s->qscale ? h->zigzag_scan : h->zigzag_scan_q0;
  1500. dc_scan= luma_dc_zigzag_scan;
  1501. }
  1502. h->last_qscale_diff = dqp = decode_cabac_mb_dqp( h );
  1503. if( dqp == INT_MIN ){
  1504. av_log(h->s.avctx, AV_LOG_ERROR, "cabac decode of qscale diff failed at %d %d\n", s->mb_x, s->mb_y);
  1505. return -1;
  1506. }
  1507. s->qscale += dqp;
  1508. if(((unsigned)s->qscale) > 51){
  1509. if(s->qscale<0) s->qscale+= 52;
  1510. else s->qscale-= 52;
  1511. }
  1512. h->chroma_qp[0] = get_chroma_qp(h, 0, s->qscale);
  1513. h->chroma_qp[1] = get_chroma_qp(h, 1, s->qscale);
  1514. if( IS_INTRA16x16( mb_type ) ) {
  1515. int i;
  1516. //av_log( s->avctx, AV_LOG_ERROR, "INTRA16x16 DC\n" );
  1517. decode_cabac_residual( h, h->mb, 0, 0, dc_scan, NULL, 16);
  1518. if( cbp&15 ) {
  1519. qmul = h->dequant4_coeff[0][s->qscale];
  1520. for( i = 0; i < 16; i++ ) {
  1521. //av_log( s->avctx, AV_LOG_ERROR, "INTRA16x16 AC:%d\n", i );
  1522. decode_cabac_residual(h, h->mb + 16*i, 1, i, scan + 1, qmul, 15);
  1523. }
  1524. } else {
  1525. fill_rectangle(&h->non_zero_count_cache[scan8[0]], 4, 4, 8, 0, 1);
  1526. }
  1527. } else {
  1528. int i8x8, i4x4;
  1529. for( i8x8 = 0; i8x8 < 4; i8x8++ ) {
  1530. if( cbp & (1<<i8x8) ) {
  1531. if( IS_8x8DCT(mb_type) ) {
  1532. decode_cabac_residual(h, h->mb + 64*i8x8, 5, 4*i8x8,
  1533. scan8x8, h->dequant8_coeff[IS_INTRA( mb_type ) ? 0:1][s->qscale], 64);
  1534. } else {
  1535. qmul = h->dequant4_coeff[IS_INTRA( mb_type ) ? 0:3][s->qscale];
  1536. for( i4x4 = 0; i4x4 < 4; i4x4++ ) {
  1537. const int index = 4*i8x8 + i4x4;
  1538. //av_log( s->avctx, AV_LOG_ERROR, "Luma4x4: %d\n", index );
  1539. //START_TIMER
  1540. decode_cabac_residual(h, h->mb + 16*index, 2, index, scan, qmul, 16);
  1541. //STOP_TIMER("decode_residual")
  1542. }
  1543. }
  1544. } else {
  1545. uint8_t * const nnz= &h->non_zero_count_cache[ scan8[4*i8x8] ];
  1546. nnz[0] = nnz[1] = nnz[8] = nnz[9] = 0;
  1547. }
  1548. }
  1549. }
  1550. if( cbp&0x30 ){
  1551. int c;
  1552. for( c = 0; c < 2; c++ ) {
  1553. //av_log( s->avctx, AV_LOG_ERROR, "INTRA C%d-DC\n",c );
  1554. decode_cabac_residual(h, h->mb + 256 + 16*4*c, 3, c, chroma_dc_scan, NULL, 4);
  1555. }
  1556. }
  1557. if( cbp&0x20 ) {
  1558. int c, i;
  1559. for( c = 0; c < 2; c++ ) {
  1560. qmul = h->dequant4_coeff[c+1+(IS_INTRA( mb_type ) ? 0:3)][h->chroma_qp[c]];
  1561. for( i = 0; i < 4; i++ ) {
  1562. const int index = 16 + 4 * c + i;
  1563. //av_log( s->avctx, AV_LOG_ERROR, "INTRA C%d-AC %d\n",c, index - 16 );
  1564. decode_cabac_residual(h, h->mb + 16*index, 4, index, scan + 1, qmul, 15);
  1565. }
  1566. }
  1567. } else {
  1568. uint8_t * const nnz= &h->non_zero_count_cache[0];
  1569. nnz[ scan8[16]+0 ] = nnz[ scan8[16]+1 ] =nnz[ scan8[16]+8 ] =nnz[ scan8[16]+9 ] =
  1570. nnz[ scan8[20]+0 ] = nnz[ scan8[20]+1 ] =nnz[ scan8[20]+8 ] =nnz[ scan8[20]+9 ] = 0;
  1571. }
  1572. } else {
  1573. uint8_t * const nnz= &h->non_zero_count_cache[0];
  1574. fill_rectangle(&nnz[scan8[0]], 4, 4, 8, 0, 1);
  1575. nnz[ scan8[16]+0 ] = nnz[ scan8[16]+1 ] =nnz[ scan8[16]+8 ] =nnz[ scan8[16]+9 ] =
  1576. nnz[ scan8[20]+0 ] = nnz[ scan8[20]+1 ] =nnz[ scan8[20]+8 ] =nnz[ scan8[20]+9 ] = 0;
  1577. h->last_qscale_diff = 0;
  1578. }
  1579. s->current_picture.qscale_table[mb_xy]= s->qscale;
  1580. write_back_non_zero_count(h);
  1581. if(MB_MBAFF){
  1582. h->ref_count[0] >>= 1;
  1583. h->ref_count[1] >>= 1;
  1584. }
  1585. return 0;
  1586. }