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.

576 lines
21KB

  1. /*
  2. * H.26L/H.264/AVC/JVT/14496-10/... encoder/decoder
  3. * Copyright (c) 2003 Michael Niedermayer <michaelni@gmx.at>
  4. *
  5. * This file is part of Libav.
  6. *
  7. * Libav 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. * Libav 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 Libav; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. /**
  22. * @file
  23. * H.264 / AVC / MPEG4 part10 prediction functions.
  24. * @author Michael Niedermayer <michaelni@gmx.at>
  25. */
  26. #include "libavutil/attributes.h"
  27. #include "dsputil.h"
  28. #include "h264pred.h"
  29. #define BIT_DEPTH 8
  30. #include "h264pred_template.c"
  31. #undef BIT_DEPTH
  32. #define BIT_DEPTH 9
  33. #include "h264pred_template.c"
  34. #undef BIT_DEPTH
  35. #define BIT_DEPTH 10
  36. #include "h264pred_template.c"
  37. #undef BIT_DEPTH
  38. static void pred4x4_vertical_vp8_c(uint8_t *src, const uint8_t *topright,
  39. ptrdiff_t stride)
  40. {
  41. const unsigned lt = src[-1-1*stride];
  42. LOAD_TOP_EDGE
  43. LOAD_TOP_RIGHT_EDGE
  44. uint32_t v = PACK_4U8((lt + 2*t0 + t1 + 2) >> 2,
  45. (t0 + 2*t1 + t2 + 2) >> 2,
  46. (t1 + 2*t2 + t3 + 2) >> 2,
  47. (t2 + 2*t3 + t4 + 2) >> 2);
  48. AV_WN32A(src+0*stride, v);
  49. AV_WN32A(src+1*stride, v);
  50. AV_WN32A(src+2*stride, v);
  51. AV_WN32A(src+3*stride, v);
  52. }
  53. static void pred4x4_horizontal_vp8_c(uint8_t *src, const uint8_t *topright,
  54. ptrdiff_t stride)
  55. {
  56. const unsigned lt = src[-1-1*stride];
  57. LOAD_LEFT_EDGE
  58. AV_WN32A(src+0*stride, ((lt + 2*l0 + l1 + 2) >> 2)*0x01010101);
  59. AV_WN32A(src+1*stride, ((l0 + 2*l1 + l2 + 2) >> 2)*0x01010101);
  60. AV_WN32A(src+2*stride, ((l1 + 2*l2 + l3 + 2) >> 2)*0x01010101);
  61. AV_WN32A(src+3*stride, ((l2 + 2*l3 + l3 + 2) >> 2)*0x01010101);
  62. }
  63. static void pred4x4_down_left_svq3_c(uint8_t *src, const uint8_t *topright,
  64. ptrdiff_t stride)
  65. {
  66. LOAD_TOP_EDGE
  67. LOAD_LEFT_EDGE
  68. src[0+0*stride]=(l1 + t1)>>1;
  69. src[1+0*stride]=
  70. src[0+1*stride]=(l2 + t2)>>1;
  71. src[2+0*stride]=
  72. src[1+1*stride]=
  73. src[0+2*stride]=
  74. src[3+0*stride]=
  75. src[2+1*stride]=
  76. src[1+2*stride]=
  77. src[0+3*stride]=
  78. src[3+1*stride]=
  79. src[2+2*stride]=
  80. src[1+3*stride]=
  81. src[3+2*stride]=
  82. src[2+3*stride]=
  83. src[3+3*stride]=(l3 + t3)>>1;
  84. }
  85. static void pred4x4_down_left_rv40_c(uint8_t *src, const uint8_t *topright,
  86. ptrdiff_t stride)
  87. {
  88. LOAD_TOP_EDGE
  89. LOAD_TOP_RIGHT_EDGE
  90. LOAD_LEFT_EDGE
  91. LOAD_DOWN_LEFT_EDGE
  92. src[0+0*stride]=(t0 + t2 + 2*t1 + 2 + l0 + l2 + 2*l1 + 2)>>3;
  93. src[1+0*stride]=
  94. src[0+1*stride]=(t1 + t3 + 2*t2 + 2 + l1 + l3 + 2*l2 + 2)>>3;
  95. src[2+0*stride]=
  96. src[1+1*stride]=
  97. src[0+2*stride]=(t2 + t4 + 2*t3 + 2 + l2 + l4 + 2*l3 + 2)>>3;
  98. src[3+0*stride]=
  99. src[2+1*stride]=
  100. src[1+2*stride]=
  101. src[0+3*stride]=(t3 + t5 + 2*t4 + 2 + l3 + l5 + 2*l4 + 2)>>3;
  102. src[3+1*stride]=
  103. src[2+2*stride]=
  104. src[1+3*stride]=(t4 + t6 + 2*t5 + 2 + l4 + l6 + 2*l5 + 2)>>3;
  105. src[3+2*stride]=
  106. src[2+3*stride]=(t5 + t7 + 2*t6 + 2 + l5 + l7 + 2*l6 + 2)>>3;
  107. src[3+3*stride]=(t6 + t7 + 1 + l6 + l7 + 1)>>2;
  108. }
  109. static void pred4x4_down_left_rv40_nodown_c(uint8_t *src,
  110. const uint8_t *topright,
  111. ptrdiff_t stride)
  112. {
  113. LOAD_TOP_EDGE
  114. LOAD_TOP_RIGHT_EDGE
  115. LOAD_LEFT_EDGE
  116. src[0+0*stride]=(t0 + t2 + 2*t1 + 2 + l0 + l2 + 2*l1 + 2)>>3;
  117. src[1+0*stride]=
  118. src[0+1*stride]=(t1 + t3 + 2*t2 + 2 + l1 + l3 + 2*l2 + 2)>>3;
  119. src[2+0*stride]=
  120. src[1+1*stride]=
  121. src[0+2*stride]=(t2 + t4 + 2*t3 + 2 + l2 + 3*l3 + 2)>>3;
  122. src[3+0*stride]=
  123. src[2+1*stride]=
  124. src[1+2*stride]=
  125. src[0+3*stride]=(t3 + t5 + 2*t4 + 2 + l3*4 + 2)>>3;
  126. src[3+1*stride]=
  127. src[2+2*stride]=
  128. src[1+3*stride]=(t4 + t6 + 2*t5 + 2 + l3*4 + 2)>>3;
  129. src[3+2*stride]=
  130. src[2+3*stride]=(t5 + t7 + 2*t6 + 2 + l3*4 + 2)>>3;
  131. src[3+3*stride]=(t6 + t7 + 1 + 2*l3 + 1)>>2;
  132. }
  133. static void pred4x4_vertical_left_rv40(uint8_t *src, const uint8_t *topright,
  134. ptrdiff_t stride,
  135. const int l0, const int l1, const int l2,
  136. const int l3, const int l4)
  137. {
  138. LOAD_TOP_EDGE
  139. LOAD_TOP_RIGHT_EDGE
  140. src[0+0*stride]=(2*t0 + 2*t1 + l1 + 2*l2 + l3 + 4)>>3;
  141. src[1+0*stride]=
  142. src[0+2*stride]=(t1 + t2 + 1)>>1;
  143. src[2+0*stride]=
  144. src[1+2*stride]=(t2 + t3 + 1)>>1;
  145. src[3+0*stride]=
  146. src[2+2*stride]=(t3 + t4+ 1)>>1;
  147. src[3+2*stride]=(t4 + t5+ 1)>>1;
  148. src[0+1*stride]=(t0 + 2*t1 + t2 + l2 + 2*l3 + l4 + 4)>>3;
  149. src[1+1*stride]=
  150. src[0+3*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  151. src[2+1*stride]=
  152. src[1+3*stride]=(t2 + 2*t3 + t4 + 2)>>2;
  153. src[3+1*stride]=
  154. src[2+3*stride]=(t3 + 2*t4 + t5 + 2)>>2;
  155. src[3+3*stride]=(t4 + 2*t5 + t6 + 2)>>2;
  156. }
  157. static void pred4x4_vertical_left_rv40_c(uint8_t *src, const uint8_t *topright,
  158. ptrdiff_t stride)
  159. {
  160. LOAD_LEFT_EDGE
  161. LOAD_DOWN_LEFT_EDGE
  162. pred4x4_vertical_left_rv40(src, topright, stride, l0, l1, l2, l3, l4);
  163. }
  164. static void pred4x4_vertical_left_rv40_nodown_c(uint8_t *src,
  165. const uint8_t *topright,
  166. ptrdiff_t stride)
  167. {
  168. LOAD_LEFT_EDGE
  169. pred4x4_vertical_left_rv40(src, topright, stride, l0, l1, l2, l3, l3);
  170. }
  171. static void pred4x4_vertical_left_vp8_c(uint8_t *src, const uint8_t *topright,
  172. ptrdiff_t stride)
  173. {
  174. LOAD_TOP_EDGE
  175. LOAD_TOP_RIGHT_EDGE
  176. src[0+0*stride]=(t0 + t1 + 1)>>1;
  177. src[1+0*stride]=
  178. src[0+2*stride]=(t1 + t2 + 1)>>1;
  179. src[2+0*stride]=
  180. src[1+2*stride]=(t2 + t3 + 1)>>1;
  181. src[3+0*stride]=
  182. src[2+2*stride]=(t3 + t4 + 1)>>1;
  183. src[0+1*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  184. src[1+1*stride]=
  185. src[0+3*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  186. src[2+1*stride]=
  187. src[1+3*stride]=(t2 + 2*t3 + t4 + 2)>>2;
  188. src[3+1*stride]=
  189. src[2+3*stride]=(t3 + 2*t4 + t5 + 2)>>2;
  190. src[3+2*stride]=(t4 + 2*t5 + t6 + 2)>>2;
  191. src[3+3*stride]=(t5 + 2*t6 + t7 + 2)>>2;
  192. }
  193. static void pred4x4_horizontal_up_rv40_c(uint8_t *src, const uint8_t *topright,
  194. ptrdiff_t stride)
  195. {
  196. LOAD_LEFT_EDGE
  197. LOAD_DOWN_LEFT_EDGE
  198. LOAD_TOP_EDGE
  199. LOAD_TOP_RIGHT_EDGE
  200. src[0+0*stride]=(t1 + 2*t2 + t3 + 2*l0 + 2*l1 + 4)>>3;
  201. src[1+0*stride]=(t2 + 2*t3 + t4 + l0 + 2*l1 + l2 + 4)>>3;
  202. src[2+0*stride]=
  203. src[0+1*stride]=(t3 + 2*t4 + t5 + 2*l1 + 2*l2 + 4)>>3;
  204. src[3+0*stride]=
  205. src[1+1*stride]=(t4 + 2*t5 + t6 + l1 + 2*l2 + l3 + 4)>>3;
  206. src[2+1*stride]=
  207. src[0+2*stride]=(t5 + 2*t6 + t7 + 2*l2 + 2*l3 + 4)>>3;
  208. src[3+1*stride]=
  209. src[1+2*stride]=(t6 + 3*t7 + l2 + 3*l3 + 4)>>3;
  210. src[3+2*stride]=
  211. src[1+3*stride]=(l3 + 2*l4 + l5 + 2)>>2;
  212. src[0+3*stride]=
  213. src[2+2*stride]=(t6 + t7 + l3 + l4 + 2)>>2;
  214. src[2+3*stride]=(l4 + l5 + 1)>>1;
  215. src[3+3*stride]=(l4 + 2*l5 + l6 + 2)>>2;
  216. }
  217. static void pred4x4_horizontal_up_rv40_nodown_c(uint8_t *src,
  218. const uint8_t *topright,
  219. ptrdiff_t stride)
  220. {
  221. LOAD_LEFT_EDGE
  222. LOAD_TOP_EDGE
  223. LOAD_TOP_RIGHT_EDGE
  224. src[0+0*stride]=(t1 + 2*t2 + t3 + 2*l0 + 2*l1 + 4)>>3;
  225. src[1+0*stride]=(t2 + 2*t3 + t4 + l0 + 2*l1 + l2 + 4)>>3;
  226. src[2+0*stride]=
  227. src[0+1*stride]=(t3 + 2*t4 + t5 + 2*l1 + 2*l2 + 4)>>3;
  228. src[3+0*stride]=
  229. src[1+1*stride]=(t4 + 2*t5 + t6 + l1 + 2*l2 + l3 + 4)>>3;
  230. src[2+1*stride]=
  231. src[0+2*stride]=(t5 + 2*t6 + t7 + 2*l2 + 2*l3 + 4)>>3;
  232. src[3+1*stride]=
  233. src[1+2*stride]=(t6 + 3*t7 + l2 + 3*l3 + 4)>>3;
  234. src[3+2*stride]=
  235. src[1+3*stride]=l3;
  236. src[0+3*stride]=
  237. src[2+2*stride]=(t6 + t7 + 2*l3 + 2)>>2;
  238. src[2+3*stride]=
  239. src[3+3*stride]=l3;
  240. }
  241. static void pred4x4_tm_vp8_c(uint8_t *src, const uint8_t *topright,
  242. ptrdiff_t stride)
  243. {
  244. const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP - src[-1-stride];
  245. uint8_t *top = src-stride;
  246. int y;
  247. for (y = 0; y < 4; y++) {
  248. const uint8_t *cm_in = cm + src[-1];
  249. src[0] = cm_in[top[0]];
  250. src[1] = cm_in[top[1]];
  251. src[2] = cm_in[top[2]];
  252. src[3] = cm_in[top[3]];
  253. src += stride;
  254. }
  255. }
  256. static void pred16x16_plane_svq3_c(uint8_t *src, ptrdiff_t stride)
  257. {
  258. pred16x16_plane_compat_8_c(src, stride, 1, 0);
  259. }
  260. static void pred16x16_plane_rv40_c(uint8_t *src, ptrdiff_t stride)
  261. {
  262. pred16x16_plane_compat_8_c(src, stride, 0, 1);
  263. }
  264. static void pred16x16_tm_vp8_c(uint8_t *src, ptrdiff_t stride)
  265. {
  266. const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP - src[-1-stride];
  267. uint8_t *top = src-stride;
  268. int y;
  269. for (y = 0; y < 16; y++) {
  270. const uint8_t *cm_in = cm + src[-1];
  271. src[0] = cm_in[top[0]];
  272. src[1] = cm_in[top[1]];
  273. src[2] = cm_in[top[2]];
  274. src[3] = cm_in[top[3]];
  275. src[4] = cm_in[top[4]];
  276. src[5] = cm_in[top[5]];
  277. src[6] = cm_in[top[6]];
  278. src[7] = cm_in[top[7]];
  279. src[8] = cm_in[top[8]];
  280. src[9] = cm_in[top[9]];
  281. src[10] = cm_in[top[10]];
  282. src[11] = cm_in[top[11]];
  283. src[12] = cm_in[top[12]];
  284. src[13] = cm_in[top[13]];
  285. src[14] = cm_in[top[14]];
  286. src[15] = cm_in[top[15]];
  287. src += stride;
  288. }
  289. }
  290. static void pred8x8_left_dc_rv40_c(uint8_t *src, ptrdiff_t stride)
  291. {
  292. int i;
  293. unsigned dc0;
  294. dc0=0;
  295. for(i=0;i<8; i++)
  296. dc0+= src[-1+i*stride];
  297. dc0= 0x01010101*((dc0 + 4)>>3);
  298. for(i=0; i<8; i++){
  299. ((uint32_t*)(src+i*stride))[0]=
  300. ((uint32_t*)(src+i*stride))[1]= dc0;
  301. }
  302. }
  303. static void pred8x8_top_dc_rv40_c(uint8_t *src, ptrdiff_t stride)
  304. {
  305. int i;
  306. unsigned dc0;
  307. dc0=0;
  308. for(i=0;i<8; i++)
  309. dc0+= src[i-stride];
  310. dc0= 0x01010101*((dc0 + 4)>>3);
  311. for(i=0; i<8; i++){
  312. ((uint32_t*)(src+i*stride))[0]=
  313. ((uint32_t*)(src+i*stride))[1]= dc0;
  314. }
  315. }
  316. static void pred8x8_dc_rv40_c(uint8_t *src, ptrdiff_t stride)
  317. {
  318. int i;
  319. unsigned dc0 = 0;
  320. for(i=0;i<4; i++){
  321. dc0+= src[-1+i*stride] + src[i-stride];
  322. dc0+= src[4+i-stride];
  323. dc0+= src[-1+(i+4)*stride];
  324. }
  325. dc0= 0x01010101*((dc0 + 8)>>4);
  326. for(i=0; i<4; i++){
  327. ((uint32_t*)(src+i*stride))[0]= dc0;
  328. ((uint32_t*)(src+i*stride))[1]= dc0;
  329. }
  330. for(i=4; i<8; i++){
  331. ((uint32_t*)(src+i*stride))[0]= dc0;
  332. ((uint32_t*)(src+i*stride))[1]= dc0;
  333. }
  334. }
  335. static void pred8x8_tm_vp8_c(uint8_t *src, ptrdiff_t stride)
  336. {
  337. const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP - src[-1-stride];
  338. uint8_t *top = src-stride;
  339. int y;
  340. for (y = 0; y < 8; y++) {
  341. const uint8_t *cm_in = cm + src[-1];
  342. src[0] = cm_in[top[0]];
  343. src[1] = cm_in[top[1]];
  344. src[2] = cm_in[top[2]];
  345. src[3] = cm_in[top[3]];
  346. src[4] = cm_in[top[4]];
  347. src[5] = cm_in[top[5]];
  348. src[6] = cm_in[top[6]];
  349. src[7] = cm_in[top[7]];
  350. src += stride;
  351. }
  352. }
  353. /**
  354. * Set the intra prediction function pointers.
  355. */
  356. av_cold void ff_h264_pred_init(H264PredContext *h, int codec_id,
  357. const int bit_depth,
  358. const int chroma_format_idc)
  359. {
  360. #undef FUNC
  361. #undef FUNCC
  362. #define FUNC(a, depth) a ## _ ## depth
  363. #define FUNCC(a, depth) a ## _ ## depth ## _c
  364. #define FUNCD(a) a ## _c
  365. #define H264_PRED(depth) \
  366. if(codec_id != AV_CODEC_ID_RV40){\
  367. if(codec_id == AV_CODEC_ID_VP8) {\
  368. h->pred4x4[VERT_PRED ]= FUNCD(pred4x4_vertical_vp8);\
  369. h->pred4x4[HOR_PRED ]= FUNCD(pred4x4_horizontal_vp8);\
  370. } else {\
  371. h->pred4x4[VERT_PRED ]= FUNCC(pred4x4_vertical , depth);\
  372. h->pred4x4[HOR_PRED ]= FUNCC(pred4x4_horizontal , depth);\
  373. }\
  374. h->pred4x4[DC_PRED ]= FUNCC(pred4x4_dc , depth);\
  375. if(codec_id == AV_CODEC_ID_SVQ3)\
  376. h->pred4x4[DIAG_DOWN_LEFT_PRED ]= FUNCD(pred4x4_down_left_svq3);\
  377. else\
  378. h->pred4x4[DIAG_DOWN_LEFT_PRED ]= FUNCC(pred4x4_down_left , depth);\
  379. h->pred4x4[DIAG_DOWN_RIGHT_PRED]= FUNCC(pred4x4_down_right , depth);\
  380. h->pred4x4[VERT_RIGHT_PRED ]= FUNCC(pred4x4_vertical_right , depth);\
  381. h->pred4x4[HOR_DOWN_PRED ]= FUNCC(pred4x4_horizontal_down , depth);\
  382. if (codec_id == AV_CODEC_ID_VP8) {\
  383. h->pred4x4[VERT_LEFT_PRED ]= FUNCD(pred4x4_vertical_left_vp8);\
  384. } else\
  385. h->pred4x4[VERT_LEFT_PRED ]= FUNCC(pred4x4_vertical_left , depth);\
  386. h->pred4x4[HOR_UP_PRED ]= FUNCC(pred4x4_horizontal_up , depth);\
  387. if(codec_id != AV_CODEC_ID_VP8) {\
  388. h->pred4x4[LEFT_DC_PRED ]= FUNCC(pred4x4_left_dc , depth);\
  389. h->pred4x4[TOP_DC_PRED ]= FUNCC(pred4x4_top_dc , depth);\
  390. h->pred4x4[DC_128_PRED ]= FUNCC(pred4x4_128_dc , depth);\
  391. } else {\
  392. h->pred4x4[TM_VP8_PRED ]= FUNCD(pred4x4_tm_vp8);\
  393. h->pred4x4[DC_127_PRED ]= FUNCC(pred4x4_127_dc , depth);\
  394. h->pred4x4[DC_129_PRED ]= FUNCC(pred4x4_129_dc , depth);\
  395. h->pred4x4[VERT_VP8_PRED ]= FUNCC(pred4x4_vertical , depth);\
  396. h->pred4x4[HOR_VP8_PRED ]= FUNCC(pred4x4_horizontal , depth);\
  397. }\
  398. }else{\
  399. h->pred4x4[VERT_PRED ]= FUNCC(pred4x4_vertical , depth);\
  400. h->pred4x4[HOR_PRED ]= FUNCC(pred4x4_horizontal , depth);\
  401. h->pred4x4[DC_PRED ]= FUNCC(pred4x4_dc , depth);\
  402. h->pred4x4[DIAG_DOWN_LEFT_PRED ]= FUNCD(pred4x4_down_left_rv40);\
  403. h->pred4x4[DIAG_DOWN_RIGHT_PRED]= FUNCC(pred4x4_down_right , depth);\
  404. h->pred4x4[VERT_RIGHT_PRED ]= FUNCC(pred4x4_vertical_right , depth);\
  405. h->pred4x4[HOR_DOWN_PRED ]= FUNCC(pred4x4_horizontal_down , depth);\
  406. h->pred4x4[VERT_LEFT_PRED ]= FUNCD(pred4x4_vertical_left_rv40);\
  407. h->pred4x4[HOR_UP_PRED ]= FUNCD(pred4x4_horizontal_up_rv40);\
  408. h->pred4x4[LEFT_DC_PRED ]= FUNCC(pred4x4_left_dc , depth);\
  409. h->pred4x4[TOP_DC_PRED ]= FUNCC(pred4x4_top_dc , depth);\
  410. h->pred4x4[DC_128_PRED ]= FUNCC(pred4x4_128_dc , depth);\
  411. h->pred4x4[DIAG_DOWN_LEFT_PRED_RV40_NODOWN]= FUNCD(pred4x4_down_left_rv40_nodown);\
  412. h->pred4x4[HOR_UP_PRED_RV40_NODOWN]= FUNCD(pred4x4_horizontal_up_rv40_nodown);\
  413. h->pred4x4[VERT_LEFT_PRED_RV40_NODOWN]= FUNCD(pred4x4_vertical_left_rv40_nodown);\
  414. }\
  415. \
  416. h->pred8x8l[VERT_PRED ]= FUNCC(pred8x8l_vertical , depth);\
  417. h->pred8x8l[HOR_PRED ]= FUNCC(pred8x8l_horizontal , depth);\
  418. h->pred8x8l[DC_PRED ]= FUNCC(pred8x8l_dc , depth);\
  419. h->pred8x8l[DIAG_DOWN_LEFT_PRED ]= FUNCC(pred8x8l_down_left , depth);\
  420. h->pred8x8l[DIAG_DOWN_RIGHT_PRED]= FUNCC(pred8x8l_down_right , depth);\
  421. h->pred8x8l[VERT_RIGHT_PRED ]= FUNCC(pred8x8l_vertical_right , depth);\
  422. h->pred8x8l[HOR_DOWN_PRED ]= FUNCC(pred8x8l_horizontal_down , depth);\
  423. h->pred8x8l[VERT_LEFT_PRED ]= FUNCC(pred8x8l_vertical_left , depth);\
  424. h->pred8x8l[HOR_UP_PRED ]= FUNCC(pred8x8l_horizontal_up , depth);\
  425. h->pred8x8l[LEFT_DC_PRED ]= FUNCC(pred8x8l_left_dc , depth);\
  426. h->pred8x8l[TOP_DC_PRED ]= FUNCC(pred8x8l_top_dc , depth);\
  427. h->pred8x8l[DC_128_PRED ]= FUNCC(pred8x8l_128_dc , depth);\
  428. \
  429. if (chroma_format_idc == 1) {\
  430. h->pred8x8[VERT_PRED8x8 ]= FUNCC(pred8x8_vertical , depth);\
  431. h->pred8x8[HOR_PRED8x8 ]= FUNCC(pred8x8_horizontal , depth);\
  432. } else {\
  433. h->pred8x8[VERT_PRED8x8 ]= FUNCC(pred8x16_vertical , depth);\
  434. h->pred8x8[HOR_PRED8x8 ]= FUNCC(pred8x16_horizontal , depth);\
  435. }\
  436. if (codec_id != AV_CODEC_ID_VP8) {\
  437. if (chroma_format_idc == 1) {\
  438. h->pred8x8[PLANE_PRED8x8]= FUNCC(pred8x8_plane , depth);\
  439. } else {\
  440. h->pred8x8[PLANE_PRED8x8]= FUNCC(pred8x16_plane , depth);\
  441. }\
  442. } else\
  443. h->pred8x8[PLANE_PRED8x8]= FUNCD(pred8x8_tm_vp8);\
  444. if(codec_id != AV_CODEC_ID_RV40 && codec_id != AV_CODEC_ID_VP8){\
  445. if (chroma_format_idc == 1) {\
  446. h->pred8x8[DC_PRED8x8 ]= FUNCC(pred8x8_dc , depth);\
  447. h->pred8x8[LEFT_DC_PRED8x8]= FUNCC(pred8x8_left_dc , depth);\
  448. h->pred8x8[TOP_DC_PRED8x8 ]= FUNCC(pred8x8_top_dc , depth);\
  449. h->pred8x8[ALZHEIMER_DC_L0T_PRED8x8 ]= FUNC(pred8x8_mad_cow_dc_l0t, depth);\
  450. h->pred8x8[ALZHEIMER_DC_0LT_PRED8x8 ]= FUNC(pred8x8_mad_cow_dc_0lt, depth);\
  451. h->pred8x8[ALZHEIMER_DC_L00_PRED8x8 ]= FUNC(pred8x8_mad_cow_dc_l00, depth);\
  452. h->pred8x8[ALZHEIMER_DC_0L0_PRED8x8 ]= FUNC(pred8x8_mad_cow_dc_0l0, depth);\
  453. } else {\
  454. h->pred8x8[DC_PRED8x8 ]= FUNCC(pred8x16_dc , depth);\
  455. h->pred8x8[LEFT_DC_PRED8x8]= FUNCC(pred8x16_left_dc , depth);\
  456. h->pred8x8[TOP_DC_PRED8x8 ]= FUNCC(pred8x16_top_dc , depth);\
  457. h->pred8x8[ALZHEIMER_DC_L0T_PRED8x8 ]= FUNC(pred8x16_mad_cow_dc_l0t, depth);\
  458. h->pred8x8[ALZHEIMER_DC_0LT_PRED8x8 ]= FUNC(pred8x16_mad_cow_dc_0lt, depth);\
  459. h->pred8x8[ALZHEIMER_DC_L00_PRED8x8 ]= FUNC(pred8x16_mad_cow_dc_l00, depth);\
  460. h->pred8x8[ALZHEIMER_DC_0L0_PRED8x8 ]= FUNC(pred8x16_mad_cow_dc_0l0, depth);\
  461. }\
  462. }else{\
  463. h->pred8x8[DC_PRED8x8 ]= FUNCD(pred8x8_dc_rv40);\
  464. h->pred8x8[LEFT_DC_PRED8x8]= FUNCD(pred8x8_left_dc_rv40);\
  465. h->pred8x8[TOP_DC_PRED8x8 ]= FUNCD(pred8x8_top_dc_rv40);\
  466. if (codec_id == AV_CODEC_ID_VP8) {\
  467. h->pred8x8[DC_127_PRED8x8]= FUNCC(pred8x8_127_dc , depth);\
  468. h->pred8x8[DC_129_PRED8x8]= FUNCC(pred8x8_129_dc , depth);\
  469. }\
  470. }\
  471. if (chroma_format_idc == 1) {\
  472. h->pred8x8[DC_128_PRED8x8 ]= FUNCC(pred8x8_128_dc , depth);\
  473. } else {\
  474. h->pred8x8[DC_128_PRED8x8 ]= FUNCC(pred8x16_128_dc , depth);\
  475. }\
  476. \
  477. h->pred16x16[DC_PRED8x8 ]= FUNCC(pred16x16_dc , depth);\
  478. h->pred16x16[VERT_PRED8x8 ]= FUNCC(pred16x16_vertical , depth);\
  479. h->pred16x16[HOR_PRED8x8 ]= FUNCC(pred16x16_horizontal , depth);\
  480. switch(codec_id){\
  481. case AV_CODEC_ID_SVQ3:\
  482. h->pred16x16[PLANE_PRED8x8 ]= FUNCD(pred16x16_plane_svq3);\
  483. break;\
  484. case AV_CODEC_ID_RV40:\
  485. h->pred16x16[PLANE_PRED8x8 ]= FUNCD(pred16x16_plane_rv40);\
  486. break;\
  487. case AV_CODEC_ID_VP8:\
  488. h->pred16x16[PLANE_PRED8x8 ]= FUNCD(pred16x16_tm_vp8);\
  489. h->pred16x16[DC_127_PRED8x8]= FUNCC(pred16x16_127_dc , depth);\
  490. h->pred16x16[DC_129_PRED8x8]= FUNCC(pred16x16_129_dc , depth);\
  491. break;\
  492. default:\
  493. h->pred16x16[PLANE_PRED8x8 ]= FUNCC(pred16x16_plane , depth);\
  494. break;\
  495. }\
  496. h->pred16x16[LEFT_DC_PRED8x8]= FUNCC(pred16x16_left_dc , depth);\
  497. h->pred16x16[TOP_DC_PRED8x8 ]= FUNCC(pred16x16_top_dc , depth);\
  498. h->pred16x16[DC_128_PRED8x8 ]= FUNCC(pred16x16_128_dc , depth);\
  499. \
  500. /* special lossless h/v prediction for h264 */ \
  501. h->pred4x4_add [VERT_PRED ]= FUNCC(pred4x4_vertical_add , depth);\
  502. h->pred4x4_add [ HOR_PRED ]= FUNCC(pred4x4_horizontal_add , depth);\
  503. h->pred8x8l_add [VERT_PRED ]= FUNCC(pred8x8l_vertical_add , depth);\
  504. h->pred8x8l_add [ HOR_PRED ]= FUNCC(pred8x8l_horizontal_add , depth);\
  505. if (chroma_format_idc == 1) {\
  506. h->pred8x8_add [VERT_PRED8x8]= FUNCC(pred8x8_vertical_add , depth);\
  507. h->pred8x8_add [ HOR_PRED8x8]= FUNCC(pred8x8_horizontal_add , depth);\
  508. } else {\
  509. h->pred8x8_add [VERT_PRED8x8]= FUNCC(pred8x16_vertical_add , depth);\
  510. h->pred8x8_add [ HOR_PRED8x8]= FUNCC(pred8x16_horizontal_add , depth);\
  511. }\
  512. h->pred16x16_add[VERT_PRED8x8]= FUNCC(pred16x16_vertical_add , depth);\
  513. h->pred16x16_add[ HOR_PRED8x8]= FUNCC(pred16x16_horizontal_add , depth);\
  514. switch (bit_depth) {
  515. case 9:
  516. H264_PRED(9)
  517. break;
  518. case 10:
  519. H264_PRED(10)
  520. break;
  521. default:
  522. H264_PRED(8)
  523. break;
  524. }
  525. if (ARCH_ARM) ff_h264_pred_init_arm(h, codec_id, bit_depth, chroma_format_idc);
  526. if (ARCH_X86) ff_h264_pred_init_x86(h, codec_id, bit_depth, chroma_format_idc);
  527. }