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.

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