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.

1356 lines
43KB

  1. /*
  2. * H.26L/H.264/AVC/JVT/14496-10/... encoder/decoder
  3. * Copyright (c) 2003-2011 Michael Niedermayer <michaelni@gmx.at>
  4. *
  5. * This file is part of FFmpeg.
  6. *
  7. * FFmpeg is free software; you can redistribute it and/or
  8. * modify it under the terms of the GNU Lesser General Public
  9. * License as published by the Free Software Foundation; either
  10. * version 2.1 of the License, or (at your option) any later version.
  11. *
  12. * FFmpeg is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15. * Lesser General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public
  18. * License along with FFmpeg; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. /**
  22. * @file
  23. * H.264 / AVC / MPEG-4 part10 prediction functions.
  24. * @author Michael Niedermayer <michaelni@gmx.at>
  25. */
  26. #include "libavutil/intreadwrite.h"
  27. #include "mathops.h"
  28. #include "bit_depth_template.c"
  29. static void FUNCC(pred4x4_vertical)(uint8_t *_src, const uint8_t *topright,
  30. ptrdiff_t _stride)
  31. {
  32. pixel *src = (pixel*)_src;
  33. int stride = _stride>>(sizeof(pixel)-1);
  34. const pixel4 a= AV_RN4PA(src-stride);
  35. AV_WN4PA(src+0*stride, a);
  36. AV_WN4PA(src+1*stride, a);
  37. AV_WN4PA(src+2*stride, a);
  38. AV_WN4PA(src+3*stride, a);
  39. }
  40. static void FUNCC(pred4x4_horizontal)(uint8_t *_src, const uint8_t *topright,
  41. ptrdiff_t _stride)
  42. {
  43. pixel *src = (pixel*)_src;
  44. int stride = _stride>>(sizeof(pixel)-1);
  45. AV_WN4PA(src+0*stride, PIXEL_SPLAT_X4(src[-1+0*stride]));
  46. AV_WN4PA(src+1*stride, PIXEL_SPLAT_X4(src[-1+1*stride]));
  47. AV_WN4PA(src+2*stride, PIXEL_SPLAT_X4(src[-1+2*stride]));
  48. AV_WN4PA(src+3*stride, PIXEL_SPLAT_X4(src[-1+3*stride]));
  49. }
  50. static void FUNCC(pred4x4_dc)(uint8_t *_src, const uint8_t *topright,
  51. ptrdiff_t _stride)
  52. {
  53. pixel *src = (pixel*)_src;
  54. int stride = _stride>>(sizeof(pixel)-1);
  55. const int dc= ( src[-stride] + src[1-stride] + src[2-stride] + src[3-stride]
  56. + src[-1+0*stride] + src[-1+1*stride] + src[-1+2*stride] + src[-1+3*stride] + 4) >>3;
  57. const pixel4 a = PIXEL_SPLAT_X4(dc);
  58. AV_WN4PA(src+0*stride, a);
  59. AV_WN4PA(src+1*stride, a);
  60. AV_WN4PA(src+2*stride, a);
  61. AV_WN4PA(src+3*stride, a);
  62. }
  63. static void FUNCC(pred4x4_left_dc)(uint8_t *_src, const uint8_t *topright,
  64. ptrdiff_t _stride)
  65. {
  66. pixel *src = (pixel*)_src;
  67. int stride = _stride>>(sizeof(pixel)-1);
  68. const int dc= ( src[-1+0*stride] + src[-1+1*stride] + src[-1+2*stride] + src[-1+3*stride] + 2) >>2;
  69. const pixel4 a = PIXEL_SPLAT_X4(dc);
  70. AV_WN4PA(src+0*stride, a);
  71. AV_WN4PA(src+1*stride, a);
  72. AV_WN4PA(src+2*stride, a);
  73. AV_WN4PA(src+3*stride, a);
  74. }
  75. static void FUNCC(pred4x4_top_dc)(uint8_t *_src, const uint8_t *topright,
  76. ptrdiff_t _stride)
  77. {
  78. pixel *src = (pixel*)_src;
  79. int stride = _stride>>(sizeof(pixel)-1);
  80. const int dc= ( src[-stride] + src[1-stride] + src[2-stride] + src[3-stride] + 2) >>2;
  81. const pixel4 a = PIXEL_SPLAT_X4(dc);
  82. AV_WN4PA(src+0*stride, a);
  83. AV_WN4PA(src+1*stride, a);
  84. AV_WN4PA(src+2*stride, a);
  85. AV_WN4PA(src+3*stride, a);
  86. }
  87. static void FUNCC(pred4x4_128_dc)(uint8_t *_src, const uint8_t *topright,
  88. ptrdiff_t _stride)
  89. {
  90. pixel *src = (pixel*)_src;
  91. int stride = _stride>>(sizeof(pixel)-1);
  92. const pixel4 a = PIXEL_SPLAT_X4(1<<(BIT_DEPTH-1));
  93. AV_WN4PA(src+0*stride, a);
  94. AV_WN4PA(src+1*stride, a);
  95. AV_WN4PA(src+2*stride, a);
  96. AV_WN4PA(src+3*stride, a);
  97. }
  98. static void FUNCC(pred4x4_127_dc)(uint8_t *_src, const uint8_t *topright,
  99. ptrdiff_t _stride)
  100. {
  101. pixel *src = (pixel*)_src;
  102. int stride = _stride>>(sizeof(pixel)-1);
  103. const pixel4 a = PIXEL_SPLAT_X4((1<<(BIT_DEPTH-1))-1);
  104. AV_WN4PA(src+0*stride, a);
  105. AV_WN4PA(src+1*stride, a);
  106. AV_WN4PA(src+2*stride, a);
  107. AV_WN4PA(src+3*stride, a);
  108. }
  109. static void FUNCC(pred4x4_129_dc)(uint8_t *_src, const uint8_t *topright,
  110. ptrdiff_t _stride)
  111. {
  112. pixel *src = (pixel*)_src;
  113. int stride = _stride>>(sizeof(pixel)-1);
  114. const pixel4 a = PIXEL_SPLAT_X4((1<<(BIT_DEPTH-1))+1);
  115. AV_WN4PA(src+0*stride, a);
  116. AV_WN4PA(src+1*stride, a);
  117. AV_WN4PA(src+2*stride, a);
  118. AV_WN4PA(src+3*stride, a);
  119. }
  120. #define LOAD_TOP_RIGHT_EDGE\
  121. const unsigned av_unused t4 = topright[0];\
  122. const unsigned av_unused t5 = topright[1];\
  123. const unsigned av_unused t6 = topright[2];\
  124. const unsigned av_unused t7 = topright[3];\
  125. #define LOAD_DOWN_LEFT_EDGE\
  126. const unsigned av_unused l4 = src[-1+4*stride];\
  127. const unsigned av_unused l5 = src[-1+5*stride];\
  128. const unsigned av_unused l6 = src[-1+6*stride];\
  129. const unsigned av_unused l7 = src[-1+7*stride];\
  130. #define LOAD_LEFT_EDGE\
  131. const unsigned av_unused l0 = src[-1+0*stride];\
  132. const unsigned av_unused l1 = src[-1+1*stride];\
  133. const unsigned av_unused l2 = src[-1+2*stride];\
  134. const unsigned av_unused l3 = src[-1+3*stride];\
  135. #define LOAD_TOP_EDGE\
  136. const unsigned av_unused t0 = src[ 0-1*stride];\
  137. const unsigned av_unused t1 = src[ 1-1*stride];\
  138. const unsigned av_unused t2 = src[ 2-1*stride];\
  139. const unsigned av_unused t3 = src[ 3-1*stride];\
  140. static void FUNCC(pred4x4_down_right)(uint8_t *_src, const uint8_t *topright,
  141. ptrdiff_t _stride)
  142. {
  143. pixel *src = (pixel*)_src;
  144. int stride = _stride>>(sizeof(pixel)-1);
  145. const int lt= src[-1-1*stride];
  146. LOAD_TOP_EDGE
  147. LOAD_LEFT_EDGE
  148. src[0+3*stride]=(l3 + 2*l2 + l1 + 2)>>2;
  149. src[0+2*stride]=
  150. src[1+3*stride]=(l2 + 2*l1 + l0 + 2)>>2;
  151. src[0+1*stride]=
  152. src[1+2*stride]=
  153. src[2+3*stride]=(l1 + 2*l0 + lt + 2)>>2;
  154. src[0+0*stride]=
  155. src[1+1*stride]=
  156. src[2+2*stride]=
  157. src[3+3*stride]=(l0 + 2*lt + t0 + 2)>>2;
  158. src[1+0*stride]=
  159. src[2+1*stride]=
  160. src[3+2*stride]=(lt + 2*t0 + t1 + 2)>>2;
  161. src[2+0*stride]=
  162. src[3+1*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  163. src[3+0*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  164. }
  165. static void FUNCC(pred4x4_down_left)(uint8_t *_src, const uint8_t *_topright,
  166. ptrdiff_t _stride)
  167. {
  168. pixel *src = (pixel*)_src;
  169. const pixel *topright = (const pixel*)_topright;
  170. int stride = _stride>>(sizeof(pixel)-1);
  171. LOAD_TOP_EDGE
  172. LOAD_TOP_RIGHT_EDGE
  173. // LOAD_LEFT_EDGE
  174. src[0+0*stride]=(t0 + t2 + 2*t1 + 2)>>2;
  175. src[1+0*stride]=
  176. src[0+1*stride]=(t1 + t3 + 2*t2 + 2)>>2;
  177. src[2+0*stride]=
  178. src[1+1*stride]=
  179. src[0+2*stride]=(t2 + t4 + 2*t3 + 2)>>2;
  180. src[3+0*stride]=
  181. src[2+1*stride]=
  182. src[1+2*stride]=
  183. src[0+3*stride]=(t3 + t5 + 2*t4 + 2)>>2;
  184. src[3+1*stride]=
  185. src[2+2*stride]=
  186. src[1+3*stride]=(t4 + t6 + 2*t5 + 2)>>2;
  187. src[3+2*stride]=
  188. src[2+3*stride]=(t5 + t7 + 2*t6 + 2)>>2;
  189. src[3+3*stride]=(t6 + 3*t7 + 2)>>2;
  190. }
  191. static void FUNCC(pred4x4_vertical_right)(uint8_t *_src,
  192. const uint8_t *topright,
  193. ptrdiff_t _stride)
  194. {
  195. pixel *src = (pixel*)_src;
  196. int stride = _stride>>(sizeof(pixel)-1);
  197. const int lt= src[-1-1*stride];
  198. LOAD_TOP_EDGE
  199. LOAD_LEFT_EDGE
  200. src[0+0*stride]=
  201. src[1+2*stride]=(lt + t0 + 1)>>1;
  202. src[1+0*stride]=
  203. src[2+2*stride]=(t0 + t1 + 1)>>1;
  204. src[2+0*stride]=
  205. src[3+2*stride]=(t1 + t2 + 1)>>1;
  206. src[3+0*stride]=(t2 + t3 + 1)>>1;
  207. src[0+1*stride]=
  208. src[1+3*stride]=(l0 + 2*lt + t0 + 2)>>2;
  209. src[1+1*stride]=
  210. src[2+3*stride]=(lt + 2*t0 + t1 + 2)>>2;
  211. src[2+1*stride]=
  212. src[3+3*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  213. src[3+1*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  214. src[0+2*stride]=(lt + 2*l0 + l1 + 2)>>2;
  215. src[0+3*stride]=(l0 + 2*l1 + l2 + 2)>>2;
  216. }
  217. static void FUNCC(pred4x4_vertical_left)(uint8_t *_src,
  218. const uint8_t *_topright,
  219. ptrdiff_t _stride)
  220. {
  221. pixel *src = (pixel*)_src;
  222. const pixel *topright = (const pixel*)_topright;
  223. int stride = _stride>>(sizeof(pixel)-1);
  224. LOAD_TOP_EDGE
  225. LOAD_TOP_RIGHT_EDGE
  226. src[0+0*stride]=(t0 + t1 + 1)>>1;
  227. src[1+0*stride]=
  228. src[0+2*stride]=(t1 + t2 + 1)>>1;
  229. src[2+0*stride]=
  230. src[1+2*stride]=(t2 + t3 + 1)>>1;
  231. src[3+0*stride]=
  232. src[2+2*stride]=(t3 + t4+ 1)>>1;
  233. src[3+2*stride]=(t4 + t5+ 1)>>1;
  234. src[0+1*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  235. src[1+1*stride]=
  236. src[0+3*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  237. src[2+1*stride]=
  238. src[1+3*stride]=(t2 + 2*t3 + t4 + 2)>>2;
  239. src[3+1*stride]=
  240. src[2+3*stride]=(t3 + 2*t4 + t5 + 2)>>2;
  241. src[3+3*stride]=(t4 + 2*t5 + t6 + 2)>>2;
  242. }
  243. static void FUNCC(pred4x4_horizontal_up)(uint8_t *_src, const uint8_t *topright,
  244. ptrdiff_t _stride)
  245. {
  246. pixel *src = (pixel*)_src;
  247. int stride = _stride>>(sizeof(pixel)-1);
  248. LOAD_LEFT_EDGE
  249. src[0+0*stride]=(l0 + l1 + 1)>>1;
  250. src[1+0*stride]=(l0 + 2*l1 + l2 + 2)>>2;
  251. src[2+0*stride]=
  252. src[0+1*stride]=(l1 + l2 + 1)>>1;
  253. src[3+0*stride]=
  254. src[1+1*stride]=(l1 + 2*l2 + l3 + 2)>>2;
  255. src[2+1*stride]=
  256. src[0+2*stride]=(l2 + l3 + 1)>>1;
  257. src[3+1*stride]=
  258. src[1+2*stride]=(l2 + 2*l3 + l3 + 2)>>2;
  259. src[3+2*stride]=
  260. src[1+3*stride]=
  261. src[0+3*stride]=
  262. src[2+2*stride]=
  263. src[2+3*stride]=
  264. src[3+3*stride]=l3;
  265. }
  266. static void FUNCC(pred4x4_horizontal_down)(uint8_t *_src,
  267. const uint8_t *topright,
  268. ptrdiff_t _stride)
  269. {
  270. pixel *src = (pixel*)_src;
  271. int stride = _stride>>(sizeof(pixel)-1);
  272. const int lt= src[-1-1*stride];
  273. LOAD_TOP_EDGE
  274. LOAD_LEFT_EDGE
  275. src[0+0*stride]=
  276. src[2+1*stride]=(lt + l0 + 1)>>1;
  277. src[1+0*stride]=
  278. src[3+1*stride]=(l0 + 2*lt + t0 + 2)>>2;
  279. src[2+0*stride]=(lt + 2*t0 + t1 + 2)>>2;
  280. src[3+0*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  281. src[0+1*stride]=
  282. src[2+2*stride]=(l0 + l1 + 1)>>1;
  283. src[1+1*stride]=
  284. src[3+2*stride]=(lt + 2*l0 + l1 + 2)>>2;
  285. src[0+2*stride]=
  286. src[2+3*stride]=(l1 + l2+ 1)>>1;
  287. src[1+2*stride]=
  288. src[3+3*stride]=(l0 + 2*l1 + l2 + 2)>>2;
  289. src[0+3*stride]=(l2 + l3 + 1)>>1;
  290. src[1+3*stride]=(l1 + 2*l2 + l3 + 2)>>2;
  291. }
  292. static void FUNCC(pred16x16_vertical)(uint8_t *_src, ptrdiff_t _stride)
  293. {
  294. int i;
  295. pixel *src = (pixel*)_src;
  296. int stride = _stride>>(sizeof(pixel)-1);
  297. const pixel4 a = AV_RN4PA(((pixel4*)(src-stride))+0);
  298. const pixel4 b = AV_RN4PA(((pixel4*)(src-stride))+1);
  299. const pixel4 c = AV_RN4PA(((pixel4*)(src-stride))+2);
  300. const pixel4 d = AV_RN4PA(((pixel4*)(src-stride))+3);
  301. for(i=0; i<16; i++){
  302. AV_WN4PA(((pixel4*)(src+i*stride))+0, a);
  303. AV_WN4PA(((pixel4*)(src+i*stride))+1, b);
  304. AV_WN4PA(((pixel4*)(src+i*stride))+2, c);
  305. AV_WN4PA(((pixel4*)(src+i*stride))+3, d);
  306. }
  307. }
  308. static void FUNCC(pred16x16_horizontal)(uint8_t *_src, ptrdiff_t stride)
  309. {
  310. int i;
  311. pixel *src = (pixel*)_src;
  312. stride >>= sizeof(pixel)-1;
  313. for(i=0; i<16; i++){
  314. const pixel4 a = PIXEL_SPLAT_X4(src[-1+i*stride]);
  315. AV_WN4PA(((pixel4*)(src+i*stride))+0, a);
  316. AV_WN4PA(((pixel4*)(src+i*stride))+1, a);
  317. AV_WN4PA(((pixel4*)(src+i*stride))+2, a);
  318. AV_WN4PA(((pixel4*)(src+i*stride))+3, a);
  319. }
  320. }
  321. #define PREDICT_16x16_DC(v)\
  322. for(i=0; i<16; i++){\
  323. AV_WN4PA(src+ 0, v);\
  324. AV_WN4PA(src+ 4, v);\
  325. AV_WN4PA(src+ 8, v);\
  326. AV_WN4PA(src+12, v);\
  327. src += stride;\
  328. }
  329. static void FUNCC(pred16x16_dc)(uint8_t *_src, ptrdiff_t stride)
  330. {
  331. int i, dc=0;
  332. pixel *src = (pixel*)_src;
  333. pixel4 dcsplat;
  334. stride >>= sizeof(pixel)-1;
  335. for(i=0;i<16; i++){
  336. dc+= src[-1+i*stride];
  337. }
  338. for(i=0;i<16; i++){
  339. dc+= src[i-stride];
  340. }
  341. dcsplat = PIXEL_SPLAT_X4((dc+16)>>5);
  342. PREDICT_16x16_DC(dcsplat);
  343. }
  344. static void FUNCC(pred16x16_left_dc)(uint8_t *_src, ptrdiff_t stride)
  345. {
  346. int i, dc=0;
  347. pixel *src = (pixel*)_src;
  348. pixel4 dcsplat;
  349. stride >>= sizeof(pixel)-1;
  350. for(i=0;i<16; i++){
  351. dc+= src[-1+i*stride];
  352. }
  353. dcsplat = PIXEL_SPLAT_X4((dc+8)>>4);
  354. PREDICT_16x16_DC(dcsplat);
  355. }
  356. static void FUNCC(pred16x16_top_dc)(uint8_t *_src, ptrdiff_t stride)
  357. {
  358. int i, dc=0;
  359. pixel *src = (pixel*)_src;
  360. pixel4 dcsplat;
  361. stride >>= sizeof(pixel)-1;
  362. for(i=0;i<16; i++){
  363. dc+= src[i-stride];
  364. }
  365. dcsplat = PIXEL_SPLAT_X4((dc+8)>>4);
  366. PREDICT_16x16_DC(dcsplat);
  367. }
  368. #define PRED16x16_X(n, v) \
  369. static void FUNCC(pred16x16_##n##_dc)(uint8_t *_src, ptrdiff_t stride)\
  370. {\
  371. int i;\
  372. pixel *src = (pixel*)_src;\
  373. stride >>= sizeof(pixel)-1;\
  374. PREDICT_16x16_DC(PIXEL_SPLAT_X4(v));\
  375. }
  376. PRED16x16_X(127, (1<<(BIT_DEPTH-1))-1)
  377. PRED16x16_X(128, (1<<(BIT_DEPTH-1))+0)
  378. PRED16x16_X(129, (1<<(BIT_DEPTH-1))+1)
  379. static inline void FUNCC(pred16x16_plane_compat)(uint8_t *_src,
  380. ptrdiff_t _stride,
  381. const int svq3,
  382. const int rv40)
  383. {
  384. int i, j, k;
  385. int a;
  386. INIT_CLIP
  387. pixel *src = (pixel*)_src;
  388. int stride = _stride>>(sizeof(pixel)-1);
  389. const pixel * const src0 = src +7-stride;
  390. const pixel * src1 = src +8*stride-1;
  391. const pixel * src2 = src1-2*stride; // == src+6*stride-1;
  392. int H = src0[1] - src0[-1];
  393. int V = src1[0] - src2[ 0];
  394. for(k=2; k<=8; ++k) {
  395. src1 += stride; src2 -= stride;
  396. H += k*(src0[k] - src0[-k]);
  397. V += k*(src1[0] - src2[ 0]);
  398. }
  399. if(svq3){
  400. H = ( 5*(H/4) ) / 16;
  401. V = ( 5*(V/4) ) / 16;
  402. /* required for 100% accuracy */
  403. i = H; H = V; V = i;
  404. }else if(rv40){
  405. H = ( H + (H>>2) ) >> 4;
  406. V = ( V + (V>>2) ) >> 4;
  407. }else{
  408. H = ( 5*H+32 ) >> 6;
  409. V = ( 5*V+32 ) >> 6;
  410. }
  411. a = 16*(src1[0] + src2[16] + 1) - 7*(V+H);
  412. for(j=16; j>0; --j) {
  413. int b = a;
  414. a += V;
  415. for(i=-16; i<0; i+=4) {
  416. src[16+i] = CLIP((b ) >> 5);
  417. src[17+i] = CLIP((b+ H) >> 5);
  418. src[18+i] = CLIP((b+2*H) >> 5);
  419. src[19+i] = CLIP((b+3*H) >> 5);
  420. b += 4*H;
  421. }
  422. src += stride;
  423. }
  424. }
  425. static void FUNCC(pred16x16_plane)(uint8_t *src, ptrdiff_t stride)
  426. {
  427. FUNCC(pred16x16_plane_compat)(src, stride, 0, 0);
  428. }
  429. static void FUNCC(pred8x8_vertical)(uint8_t *_src, ptrdiff_t _stride)
  430. {
  431. int i;
  432. pixel *src = (pixel*)_src;
  433. int stride = _stride>>(sizeof(pixel)-1);
  434. const pixel4 a= AV_RN4PA(((pixel4*)(src-stride))+0);
  435. const pixel4 b= AV_RN4PA(((pixel4*)(src-stride))+1);
  436. for(i=0; i<8; i++){
  437. AV_WN4PA(((pixel4*)(src+i*stride))+0, a);
  438. AV_WN4PA(((pixel4*)(src+i*stride))+1, b);
  439. }
  440. }
  441. static void FUNCC(pred8x16_vertical)(uint8_t *_src, ptrdiff_t _stride)
  442. {
  443. int i;
  444. pixel *src = (pixel*)_src;
  445. int stride = _stride>>(sizeof(pixel)-1);
  446. const pixel4 a= AV_RN4PA(((pixel4*)(src-stride))+0);
  447. const pixel4 b= AV_RN4PA(((pixel4*)(src-stride))+1);
  448. for(i=0; i<16; i++){
  449. AV_WN4PA(((pixel4*)(src+i*stride))+0, a);
  450. AV_WN4PA(((pixel4*)(src+i*stride))+1, b);
  451. }
  452. }
  453. static void FUNCC(pred8x8_horizontal)(uint8_t *_src, ptrdiff_t stride)
  454. {
  455. int i;
  456. pixel *src = (pixel*)_src;
  457. stride >>= sizeof(pixel)-1;
  458. for(i=0; i<8; i++){
  459. const pixel4 a = PIXEL_SPLAT_X4(src[-1+i*stride]);
  460. AV_WN4PA(((pixel4*)(src+i*stride))+0, a);
  461. AV_WN4PA(((pixel4*)(src+i*stride))+1, a);
  462. }
  463. }
  464. static void FUNCC(pred8x16_horizontal)(uint8_t *_src, ptrdiff_t stride)
  465. {
  466. int i;
  467. pixel *src = (pixel*)_src;
  468. stride >>= sizeof(pixel)-1;
  469. for(i=0; i<16; i++){
  470. const pixel4 a = PIXEL_SPLAT_X4(src[-1+i*stride]);
  471. AV_WN4PA(((pixel4*)(src+i*stride))+0, a);
  472. AV_WN4PA(((pixel4*)(src+i*stride))+1, a);
  473. }
  474. }
  475. #define PRED8x8_X(n, v)\
  476. static void FUNCC(pred8x8_##n##_dc)(uint8_t *_src, ptrdiff_t stride)\
  477. {\
  478. int i;\
  479. const pixel4 a = PIXEL_SPLAT_X4(v);\
  480. pixel *src = (pixel*)_src;\
  481. stride >>= sizeof(pixel)-1;\
  482. for(i=0; i<8; i++){\
  483. AV_WN4PA(((pixel4*)(src+i*stride))+0, a);\
  484. AV_WN4PA(((pixel4*)(src+i*stride))+1, a);\
  485. }\
  486. }
  487. PRED8x8_X(127, (1<<(BIT_DEPTH-1))-1)
  488. PRED8x8_X(128, (1<<(BIT_DEPTH-1))+0)
  489. PRED8x8_X(129, (1<<(BIT_DEPTH-1))+1)
  490. static void FUNCC(pred8x16_128_dc)(uint8_t *_src, ptrdiff_t stride)
  491. {
  492. FUNCC(pred8x8_128_dc)(_src, stride);
  493. FUNCC(pred8x8_128_dc)(_src+8*stride, stride);
  494. }
  495. static void FUNCC(pred8x8_left_dc)(uint8_t *_src, ptrdiff_t stride)
  496. {
  497. int i;
  498. int dc0, dc2;
  499. pixel4 dc0splat, dc2splat;
  500. pixel *src = (pixel*)_src;
  501. stride >>= sizeof(pixel)-1;
  502. dc0=dc2=0;
  503. for(i=0;i<4; i++){
  504. dc0+= src[-1+i*stride];
  505. dc2+= src[-1+(i+4)*stride];
  506. }
  507. dc0splat = PIXEL_SPLAT_X4((dc0 + 2)>>2);
  508. dc2splat = PIXEL_SPLAT_X4((dc2 + 2)>>2);
  509. for(i=0; i<4; i++){
  510. AV_WN4PA(((pixel4*)(src+i*stride))+0, dc0splat);
  511. AV_WN4PA(((pixel4*)(src+i*stride))+1, dc0splat);
  512. }
  513. for(i=4; i<8; i++){
  514. AV_WN4PA(((pixel4*)(src+i*stride))+0, dc2splat);
  515. AV_WN4PA(((pixel4*)(src+i*stride))+1, dc2splat);
  516. }
  517. }
  518. static void FUNCC(pred8x16_left_dc)(uint8_t *_src, ptrdiff_t stride)
  519. {
  520. FUNCC(pred8x8_left_dc)(_src, stride);
  521. FUNCC(pred8x8_left_dc)(_src+8*stride, stride);
  522. }
  523. static void FUNCC(pred8x8_top_dc)(uint8_t *_src, ptrdiff_t stride)
  524. {
  525. int i;
  526. int dc0, dc1;
  527. pixel4 dc0splat, dc1splat;
  528. pixel *src = (pixel*)_src;
  529. stride >>= sizeof(pixel)-1;
  530. dc0=dc1=0;
  531. for(i=0;i<4; i++){
  532. dc0+= src[i-stride];
  533. dc1+= src[4+i-stride];
  534. }
  535. dc0splat = PIXEL_SPLAT_X4((dc0 + 2)>>2);
  536. dc1splat = PIXEL_SPLAT_X4((dc1 + 2)>>2);
  537. for(i=0; i<4; i++){
  538. AV_WN4PA(((pixel4*)(src+i*stride))+0, dc0splat);
  539. AV_WN4PA(((pixel4*)(src+i*stride))+1, dc1splat);
  540. }
  541. for(i=4; i<8; i++){
  542. AV_WN4PA(((pixel4*)(src+i*stride))+0, dc0splat);
  543. AV_WN4PA(((pixel4*)(src+i*stride))+1, dc1splat);
  544. }
  545. }
  546. static void FUNCC(pred8x16_top_dc)(uint8_t *_src, ptrdiff_t stride)
  547. {
  548. int i;
  549. int dc0, dc1;
  550. pixel4 dc0splat, dc1splat;
  551. pixel *src = (pixel*)_src;
  552. stride >>= sizeof(pixel)-1;
  553. dc0=dc1=0;
  554. for(i=0;i<4; i++){
  555. dc0+= src[i-stride];
  556. dc1+= src[4+i-stride];
  557. }
  558. dc0splat = PIXEL_SPLAT_X4((dc0 + 2)>>2);
  559. dc1splat = PIXEL_SPLAT_X4((dc1 + 2)>>2);
  560. for(i=0; i<16; i++){
  561. AV_WN4PA(((pixel4*)(src+i*stride))+0, dc0splat);
  562. AV_WN4PA(((pixel4*)(src+i*stride))+1, dc1splat);
  563. }
  564. }
  565. static void FUNCC(pred8x8_dc)(uint8_t *_src, ptrdiff_t stride)
  566. {
  567. int i;
  568. int dc0, dc1, dc2;
  569. pixel4 dc0splat, dc1splat, dc2splat, dc3splat;
  570. pixel *src = (pixel*)_src;
  571. stride >>= sizeof(pixel)-1;
  572. dc0=dc1=dc2=0;
  573. for(i=0;i<4; i++){
  574. dc0+= src[-1+i*stride] + src[i-stride];
  575. dc1+= src[4+i-stride];
  576. dc2+= src[-1+(i+4)*stride];
  577. }
  578. dc0splat = PIXEL_SPLAT_X4((dc0 + 4)>>3);
  579. dc1splat = PIXEL_SPLAT_X4((dc1 + 2)>>2);
  580. dc2splat = PIXEL_SPLAT_X4((dc2 + 2)>>2);
  581. dc3splat = PIXEL_SPLAT_X4((dc1 + dc2 + 4)>>3);
  582. for(i=0; i<4; i++){
  583. AV_WN4PA(((pixel4*)(src+i*stride))+0, dc0splat);
  584. AV_WN4PA(((pixel4*)(src+i*stride))+1, dc1splat);
  585. }
  586. for(i=4; i<8; i++){
  587. AV_WN4PA(((pixel4*)(src+i*stride))+0, dc2splat);
  588. AV_WN4PA(((pixel4*)(src+i*stride))+1, dc3splat);
  589. }
  590. }
  591. static void FUNCC(pred8x16_dc)(uint8_t *_src, ptrdiff_t stride)
  592. {
  593. int i;
  594. int dc0, dc1, dc2, dc3, dc4;
  595. pixel4 dc0splat, dc1splat, dc2splat, dc3splat, dc4splat, dc5splat, dc6splat, dc7splat;
  596. pixel *src = (pixel*)_src;
  597. stride >>= sizeof(pixel)-1;
  598. dc0=dc1=dc2=dc3=dc4=0;
  599. for(i=0;i<4; i++){
  600. dc0+= src[-1+i*stride] + src[i-stride];
  601. dc1+= src[4+i-stride];
  602. dc2+= src[-1+(i+4)*stride];
  603. dc3+= src[-1+(i+8)*stride];
  604. dc4+= src[-1+(i+12)*stride];
  605. }
  606. dc0splat = PIXEL_SPLAT_X4((dc0 + 4)>>3);
  607. dc1splat = PIXEL_SPLAT_X4((dc1 + 2)>>2);
  608. dc2splat = PIXEL_SPLAT_X4((dc2 + 2)>>2);
  609. dc3splat = PIXEL_SPLAT_X4((dc1 + dc2 + 4)>>3);
  610. dc4splat = PIXEL_SPLAT_X4((dc3 + 2)>>2);
  611. dc5splat = PIXEL_SPLAT_X4((dc1 + dc3 + 4)>>3);
  612. dc6splat = PIXEL_SPLAT_X4((dc4 + 2)>>2);
  613. dc7splat = PIXEL_SPLAT_X4((dc1 + dc4 + 4)>>3);
  614. for(i=0; i<4; i++){
  615. AV_WN4PA(((pixel4*)(src+i*stride))+0, dc0splat);
  616. AV_WN4PA(((pixel4*)(src+i*stride))+1, dc1splat);
  617. }
  618. for(i=4; i<8; i++){
  619. AV_WN4PA(((pixel4*)(src+i*stride))+0, dc2splat);
  620. AV_WN4PA(((pixel4*)(src+i*stride))+1, dc3splat);
  621. }
  622. for(i=8; i<12; i++){
  623. AV_WN4PA(((pixel4*)(src+i*stride))+0, dc4splat);
  624. AV_WN4PA(((pixel4*)(src+i*stride))+1, dc5splat);
  625. }
  626. for(i=12; i<16; i++){
  627. AV_WN4PA(((pixel4*)(src+i*stride))+0, dc6splat);
  628. AV_WN4PA(((pixel4*)(src+i*stride))+1, dc7splat);
  629. }
  630. }
  631. //the following 4 function should not be optimized!
  632. static void FUNC(pred8x8_mad_cow_dc_l0t)(uint8_t *src, ptrdiff_t stride)
  633. {
  634. FUNCC(pred8x8_top_dc)(src, stride);
  635. FUNCC(pred4x4_dc)(src, NULL, stride);
  636. }
  637. static void FUNC(pred8x16_mad_cow_dc_l0t)(uint8_t *src, ptrdiff_t stride)
  638. {
  639. FUNCC(pred8x16_top_dc)(src, stride);
  640. FUNCC(pred4x4_dc)(src, NULL, stride);
  641. }
  642. static void FUNC(pred8x8_mad_cow_dc_0lt)(uint8_t *src, ptrdiff_t stride)
  643. {
  644. FUNCC(pred8x8_dc)(src, stride);
  645. FUNCC(pred4x4_top_dc)(src, NULL, stride);
  646. }
  647. static void FUNC(pred8x16_mad_cow_dc_0lt)(uint8_t *src, ptrdiff_t stride)
  648. {
  649. FUNCC(pred8x16_dc)(src, stride);
  650. FUNCC(pred4x4_top_dc)(src, NULL, stride);
  651. }
  652. static void FUNC(pred8x8_mad_cow_dc_l00)(uint8_t *src, ptrdiff_t stride)
  653. {
  654. FUNCC(pred8x8_left_dc)(src, stride);
  655. FUNCC(pred4x4_128_dc)(src + 4*stride , NULL, stride);
  656. FUNCC(pred4x4_128_dc)(src + 4*stride + 4*sizeof(pixel), NULL, stride);
  657. }
  658. static void FUNC(pred8x16_mad_cow_dc_l00)(uint8_t *src, ptrdiff_t stride)
  659. {
  660. FUNCC(pred8x16_left_dc)(src, stride);
  661. FUNCC(pred4x4_128_dc)(src + 4*stride , NULL, stride);
  662. FUNCC(pred4x4_128_dc)(src + 4*stride + 4*sizeof(pixel), NULL, stride);
  663. }
  664. static void FUNC(pred8x8_mad_cow_dc_0l0)(uint8_t *src, ptrdiff_t stride)
  665. {
  666. FUNCC(pred8x8_left_dc)(src, stride);
  667. FUNCC(pred4x4_128_dc)(src , NULL, stride);
  668. FUNCC(pred4x4_128_dc)(src + 4*sizeof(pixel), NULL, stride);
  669. }
  670. static void FUNC(pred8x16_mad_cow_dc_0l0)(uint8_t *src, ptrdiff_t stride)
  671. {
  672. FUNCC(pred8x16_left_dc)(src, stride);
  673. FUNCC(pred4x4_128_dc)(src , NULL, stride);
  674. FUNCC(pred4x4_128_dc)(src + 4*sizeof(pixel), NULL, stride);
  675. }
  676. static void FUNCC(pred8x8_plane)(uint8_t *_src, ptrdiff_t _stride)
  677. {
  678. int j, k;
  679. int a;
  680. INIT_CLIP
  681. pixel *src = (pixel*)_src;
  682. int stride = _stride>>(sizeof(pixel)-1);
  683. const pixel * const src0 = src +3-stride;
  684. const pixel * src1 = src +4*stride-1;
  685. const pixel * src2 = src1-2*stride; // == src+2*stride-1;
  686. int H = src0[1] - src0[-1];
  687. int V = src1[0] - src2[ 0];
  688. for(k=2; k<=4; ++k) {
  689. src1 += stride; src2 -= stride;
  690. H += k*(src0[k] - src0[-k]);
  691. V += k*(src1[0] - src2[ 0]);
  692. }
  693. H = ( 17*H+16 ) >> 5;
  694. V = ( 17*V+16 ) >> 5;
  695. a = 16*(src1[0] + src2[8]+1) - 3*(V+H);
  696. for(j=8; j>0; --j) {
  697. int b = a;
  698. a += V;
  699. src[0] = CLIP((b ) >> 5);
  700. src[1] = CLIP((b+ H) >> 5);
  701. src[2] = CLIP((b+2*H) >> 5);
  702. src[3] = CLIP((b+3*H) >> 5);
  703. src[4] = CLIP((b+4*H) >> 5);
  704. src[5] = CLIP((b+5*H) >> 5);
  705. src[6] = CLIP((b+6*H) >> 5);
  706. src[7] = CLIP((b+7*H) >> 5);
  707. src += stride;
  708. }
  709. }
  710. static void FUNCC(pred8x16_plane)(uint8_t *_src, ptrdiff_t _stride)
  711. {
  712. int j, k;
  713. int a;
  714. INIT_CLIP
  715. pixel *src = (pixel*)_src;
  716. int stride = _stride>>(sizeof(pixel)-1);
  717. const pixel * const src0 = src +3-stride;
  718. const pixel * src1 = src +8*stride-1;
  719. const pixel * src2 = src1-2*stride; // == src+6*stride-1;
  720. int H = src0[1] - src0[-1];
  721. int V = src1[0] - src2[ 0];
  722. for (k = 2; k <= 4; ++k) {
  723. src1 += stride; src2 -= stride;
  724. H += k*(src0[k] - src0[-k]);
  725. V += k*(src1[0] - src2[ 0]);
  726. }
  727. for (; k <= 8; ++k) {
  728. src1 += stride; src2 -= stride;
  729. V += k*(src1[0] - src2[0]);
  730. }
  731. H = (17*H+16) >> 5;
  732. V = (5*V+32) >> 6;
  733. a = 16*(src1[0] + src2[8] + 1) - 7*V - 3*H;
  734. for(j=16; j>0; --j) {
  735. int b = a;
  736. a += V;
  737. src[0] = CLIP((b ) >> 5);
  738. src[1] = CLIP((b+ H) >> 5);
  739. src[2] = CLIP((b+2*H) >> 5);
  740. src[3] = CLIP((b+3*H) >> 5);
  741. src[4] = CLIP((b+4*H) >> 5);
  742. src[5] = CLIP((b+5*H) >> 5);
  743. src[6] = CLIP((b+6*H) >> 5);
  744. src[7] = CLIP((b+7*H) >> 5);
  745. src += stride;
  746. }
  747. }
  748. #define SRC(x,y) src[(x)+(y)*stride]
  749. #define PL(y) \
  750. const int l##y = (SRC(-1,y-1) + 2*SRC(-1,y) + SRC(-1,y+1) + 2) >> 2;
  751. #define PREDICT_8x8_LOAD_LEFT \
  752. const int l0 = ((has_topleft ? SRC(-1,-1) : SRC(-1,0)) \
  753. + 2*SRC(-1,0) + SRC(-1,1) + 2) >> 2; \
  754. PL(1) PL(2) PL(3) PL(4) PL(5) PL(6) \
  755. const int l7 av_unused = (SRC(-1,6) + 3*SRC(-1,7) + 2) >> 2
  756. #define PT(x) \
  757. const int t##x = (SRC(x-1,-1) + 2*SRC(x,-1) + SRC(x+1,-1) + 2) >> 2;
  758. #define PREDICT_8x8_LOAD_TOP \
  759. const int t0 = ((has_topleft ? SRC(-1,-1) : SRC(0,-1)) \
  760. + 2*SRC(0,-1) + SRC(1,-1) + 2) >> 2; \
  761. PT(1) PT(2) PT(3) PT(4) PT(5) PT(6) \
  762. const int t7 av_unused = ((has_topright ? SRC(8,-1) : SRC(7,-1)) \
  763. + 2*SRC(7,-1) + SRC(6,-1) + 2) >> 2
  764. #define PTR(x) \
  765. t##x = (SRC(x-1,-1) + 2*SRC(x,-1) + SRC(x+1,-1) + 2) >> 2;
  766. #define PREDICT_8x8_LOAD_TOPRIGHT \
  767. int t8, t9, t10, t11, t12, t13, t14, t15; \
  768. if(has_topright) { \
  769. PTR(8) PTR(9) PTR(10) PTR(11) PTR(12) PTR(13) PTR(14) \
  770. t15 = (SRC(14,-1) + 3*SRC(15,-1) + 2) >> 2; \
  771. } else t8=t9=t10=t11=t12=t13=t14=t15= SRC(7,-1);
  772. #define PREDICT_8x8_LOAD_TOPLEFT \
  773. const int lt = (SRC(-1,0) + 2*SRC(-1,-1) + SRC(0,-1) + 2) >> 2
  774. #define PREDICT_8x8_DC(v) \
  775. int y; \
  776. for( y = 0; y < 8; y++ ) { \
  777. AV_WN4PA(((pixel4*)src)+0, v); \
  778. AV_WN4PA(((pixel4*)src)+1, v); \
  779. src += stride; \
  780. }
  781. static void FUNCC(pred8x8l_128_dc)(uint8_t *_src, int has_topleft,
  782. int has_topright, ptrdiff_t _stride)
  783. {
  784. pixel *src = (pixel*)_src;
  785. int stride = _stride>>(sizeof(pixel)-1);
  786. PREDICT_8x8_DC(PIXEL_SPLAT_X4(1<<(BIT_DEPTH-1)));
  787. }
  788. static void FUNCC(pred8x8l_left_dc)(uint8_t *_src, int has_topleft,
  789. int has_topright, ptrdiff_t _stride)
  790. {
  791. pixel *src = (pixel*)_src;
  792. int stride = _stride>>(sizeof(pixel)-1);
  793. PREDICT_8x8_LOAD_LEFT;
  794. const pixel4 dc = PIXEL_SPLAT_X4((l0+l1+l2+l3+l4+l5+l6+l7+4) >> 3);
  795. PREDICT_8x8_DC(dc);
  796. }
  797. static void FUNCC(pred8x8l_top_dc)(uint8_t *_src, int has_topleft,
  798. int has_topright, ptrdiff_t _stride)
  799. {
  800. pixel *src = (pixel*)_src;
  801. int stride = _stride>>(sizeof(pixel)-1);
  802. PREDICT_8x8_LOAD_TOP;
  803. const pixel4 dc = PIXEL_SPLAT_X4((t0+t1+t2+t3+t4+t5+t6+t7+4) >> 3);
  804. PREDICT_8x8_DC(dc);
  805. }
  806. static void FUNCC(pred8x8l_dc)(uint8_t *_src, int has_topleft,
  807. int has_topright, ptrdiff_t _stride)
  808. {
  809. pixel *src = (pixel*)_src;
  810. int stride = _stride>>(sizeof(pixel)-1);
  811. PREDICT_8x8_LOAD_LEFT;
  812. PREDICT_8x8_LOAD_TOP;
  813. const pixel4 dc = PIXEL_SPLAT_X4((l0+l1+l2+l3+l4+l5+l6+l7
  814. +t0+t1+t2+t3+t4+t5+t6+t7+8) >> 4);
  815. PREDICT_8x8_DC(dc);
  816. }
  817. static void FUNCC(pred8x8l_horizontal)(uint8_t *_src, int has_topleft,
  818. int has_topright, ptrdiff_t _stride)
  819. {
  820. pixel *src = (pixel*)_src;
  821. int stride = _stride>>(sizeof(pixel)-1);
  822. pixel4 a;
  823. PREDICT_8x8_LOAD_LEFT;
  824. #define ROW(y) a = PIXEL_SPLAT_X4(l##y); \
  825. AV_WN4PA(src+y*stride, a); \
  826. AV_WN4PA(src+y*stride+4, a);
  827. ROW(0); ROW(1); ROW(2); ROW(3); ROW(4); ROW(5); ROW(6); ROW(7);
  828. #undef ROW
  829. }
  830. static void FUNCC(pred8x8l_vertical)(uint8_t *_src, int has_topleft,
  831. int has_topright, ptrdiff_t _stride)
  832. {
  833. int y;
  834. pixel *src = (pixel*)_src;
  835. int stride = _stride>>(sizeof(pixel)-1);
  836. pixel4 a, b;
  837. PREDICT_8x8_LOAD_TOP;
  838. src[0] = t0;
  839. src[1] = t1;
  840. src[2] = t2;
  841. src[3] = t3;
  842. src[4] = t4;
  843. src[5] = t5;
  844. src[6] = t6;
  845. src[7] = t7;
  846. a = AV_RN4PA(((pixel4*)src)+0);
  847. b = AV_RN4PA(((pixel4*)src)+1);
  848. for( y = 1; y < 8; y++ ) {
  849. AV_WN4PA(((pixel4*)(src+y*stride))+0, a);
  850. AV_WN4PA(((pixel4*)(src+y*stride))+1, b);
  851. }
  852. }
  853. static void FUNCC(pred8x8l_down_left)(uint8_t *_src, int has_topleft,
  854. int has_topright, ptrdiff_t _stride)
  855. {
  856. pixel *src = (pixel*)_src;
  857. int stride = _stride>>(sizeof(pixel)-1);
  858. PREDICT_8x8_LOAD_TOP;
  859. PREDICT_8x8_LOAD_TOPRIGHT;
  860. SRC(0,0)= (t0 + 2*t1 + t2 + 2) >> 2;
  861. SRC(0,1)=SRC(1,0)= (t1 + 2*t2 + t3 + 2) >> 2;
  862. SRC(0,2)=SRC(1,1)=SRC(2,0)= (t2 + 2*t3 + t4 + 2) >> 2;
  863. SRC(0,3)=SRC(1,2)=SRC(2,1)=SRC(3,0)= (t3 + 2*t4 + t5 + 2) >> 2;
  864. SRC(0,4)=SRC(1,3)=SRC(2,2)=SRC(3,1)=SRC(4,0)= (t4 + 2*t5 + t6 + 2) >> 2;
  865. SRC(0,5)=SRC(1,4)=SRC(2,3)=SRC(3,2)=SRC(4,1)=SRC(5,0)= (t5 + 2*t6 + t7 + 2) >> 2;
  866. SRC(0,6)=SRC(1,5)=SRC(2,4)=SRC(3,3)=SRC(4,2)=SRC(5,1)=SRC(6,0)= (t6 + 2*t7 + t8 + 2) >> 2;
  867. SRC(0,7)=SRC(1,6)=SRC(2,5)=SRC(3,4)=SRC(4,3)=SRC(5,2)=SRC(6,1)=SRC(7,0)= (t7 + 2*t8 + t9 + 2) >> 2;
  868. SRC(1,7)=SRC(2,6)=SRC(3,5)=SRC(4,4)=SRC(5,3)=SRC(6,2)=SRC(7,1)= (t8 + 2*t9 + t10 + 2) >> 2;
  869. SRC(2,7)=SRC(3,6)=SRC(4,5)=SRC(5,4)=SRC(6,3)=SRC(7,2)= (t9 + 2*t10 + t11 + 2) >> 2;
  870. SRC(3,7)=SRC(4,6)=SRC(5,5)=SRC(6,4)=SRC(7,3)= (t10 + 2*t11 + t12 + 2) >> 2;
  871. SRC(4,7)=SRC(5,6)=SRC(6,5)=SRC(7,4)= (t11 + 2*t12 + t13 + 2) >> 2;
  872. SRC(5,7)=SRC(6,6)=SRC(7,5)= (t12 + 2*t13 + t14 + 2) >> 2;
  873. SRC(6,7)=SRC(7,6)= (t13 + 2*t14 + t15 + 2) >> 2;
  874. SRC(7,7)= (t14 + 3*t15 + 2) >> 2;
  875. }
  876. static void FUNCC(pred8x8l_down_right)(uint8_t *_src, int has_topleft,
  877. int has_topright, ptrdiff_t _stride)
  878. {
  879. pixel *src = (pixel*)_src;
  880. int stride = _stride>>(sizeof(pixel)-1);
  881. PREDICT_8x8_LOAD_TOP;
  882. PREDICT_8x8_LOAD_LEFT;
  883. PREDICT_8x8_LOAD_TOPLEFT;
  884. SRC(0,7)= (l7 + 2*l6 + l5 + 2) >> 2;
  885. SRC(0,6)=SRC(1,7)= (l6 + 2*l5 + l4 + 2) >> 2;
  886. SRC(0,5)=SRC(1,6)=SRC(2,7)= (l5 + 2*l4 + l3 + 2) >> 2;
  887. SRC(0,4)=SRC(1,5)=SRC(2,6)=SRC(3,7)= (l4 + 2*l3 + l2 + 2) >> 2;
  888. SRC(0,3)=SRC(1,4)=SRC(2,5)=SRC(3,6)=SRC(4,7)= (l3 + 2*l2 + l1 + 2) >> 2;
  889. SRC(0,2)=SRC(1,3)=SRC(2,4)=SRC(3,5)=SRC(4,6)=SRC(5,7)= (l2 + 2*l1 + l0 + 2) >> 2;
  890. SRC(0,1)=SRC(1,2)=SRC(2,3)=SRC(3,4)=SRC(4,5)=SRC(5,6)=SRC(6,7)= (l1 + 2*l0 + lt + 2) >> 2;
  891. SRC(0,0)=SRC(1,1)=SRC(2,2)=SRC(3,3)=SRC(4,4)=SRC(5,5)=SRC(6,6)=SRC(7,7)= (l0 + 2*lt + t0 + 2) >> 2;
  892. SRC(1,0)=SRC(2,1)=SRC(3,2)=SRC(4,3)=SRC(5,4)=SRC(6,5)=SRC(7,6)= (lt + 2*t0 + t1 + 2) >> 2;
  893. SRC(2,0)=SRC(3,1)=SRC(4,2)=SRC(5,3)=SRC(6,4)=SRC(7,5)= (t0 + 2*t1 + t2 + 2) >> 2;
  894. SRC(3,0)=SRC(4,1)=SRC(5,2)=SRC(6,3)=SRC(7,4)= (t1 + 2*t2 + t3 + 2) >> 2;
  895. SRC(4,0)=SRC(5,1)=SRC(6,2)=SRC(7,3)= (t2 + 2*t3 + t4 + 2) >> 2;
  896. SRC(5,0)=SRC(6,1)=SRC(7,2)= (t3 + 2*t4 + t5 + 2) >> 2;
  897. SRC(6,0)=SRC(7,1)= (t4 + 2*t5 + t6 + 2) >> 2;
  898. SRC(7,0)= (t5 + 2*t6 + t7 + 2) >> 2;
  899. }
  900. static void FUNCC(pred8x8l_vertical_right)(uint8_t *_src, int has_topleft,
  901. int has_topright, ptrdiff_t _stride)
  902. {
  903. pixel *src = (pixel*)_src;
  904. int stride = _stride>>(sizeof(pixel)-1);
  905. PREDICT_8x8_LOAD_TOP;
  906. PREDICT_8x8_LOAD_LEFT;
  907. PREDICT_8x8_LOAD_TOPLEFT;
  908. SRC(0,6)= (l5 + 2*l4 + l3 + 2) >> 2;
  909. SRC(0,7)= (l6 + 2*l5 + l4 + 2) >> 2;
  910. SRC(0,4)=SRC(1,6)= (l3 + 2*l2 + l1 + 2) >> 2;
  911. SRC(0,5)=SRC(1,7)= (l4 + 2*l3 + l2 + 2) >> 2;
  912. SRC(0,2)=SRC(1,4)=SRC(2,6)= (l1 + 2*l0 + lt + 2) >> 2;
  913. SRC(0,3)=SRC(1,5)=SRC(2,7)= (l2 + 2*l1 + l0 + 2) >> 2;
  914. SRC(0,1)=SRC(1,3)=SRC(2,5)=SRC(3,7)= (l0 + 2*lt + t0 + 2) >> 2;
  915. SRC(0,0)=SRC(1,2)=SRC(2,4)=SRC(3,6)= (lt + t0 + 1) >> 1;
  916. SRC(1,1)=SRC(2,3)=SRC(3,5)=SRC(4,7)= (lt + 2*t0 + t1 + 2) >> 2;
  917. SRC(1,0)=SRC(2,2)=SRC(3,4)=SRC(4,6)= (t0 + t1 + 1) >> 1;
  918. SRC(2,1)=SRC(3,3)=SRC(4,5)=SRC(5,7)= (t0 + 2*t1 + t2 + 2) >> 2;
  919. SRC(2,0)=SRC(3,2)=SRC(4,4)=SRC(5,6)= (t1 + t2 + 1) >> 1;
  920. SRC(3,1)=SRC(4,3)=SRC(5,5)=SRC(6,7)= (t1 + 2*t2 + t3 + 2) >> 2;
  921. SRC(3,0)=SRC(4,2)=SRC(5,4)=SRC(6,6)= (t2 + t3 + 1) >> 1;
  922. SRC(4,1)=SRC(5,3)=SRC(6,5)=SRC(7,7)= (t2 + 2*t3 + t4 + 2) >> 2;
  923. SRC(4,0)=SRC(5,2)=SRC(6,4)=SRC(7,6)= (t3 + t4 + 1) >> 1;
  924. SRC(5,1)=SRC(6,3)=SRC(7,5)= (t3 + 2*t4 + t5 + 2) >> 2;
  925. SRC(5,0)=SRC(6,2)=SRC(7,4)= (t4 + t5 + 1) >> 1;
  926. SRC(6,1)=SRC(7,3)= (t4 + 2*t5 + t6 + 2) >> 2;
  927. SRC(6,0)=SRC(7,2)= (t5 + t6 + 1) >> 1;
  928. SRC(7,1)= (t5 + 2*t6 + t7 + 2) >> 2;
  929. SRC(7,0)= (t6 + t7 + 1) >> 1;
  930. }
  931. static void FUNCC(pred8x8l_horizontal_down)(uint8_t *_src, int has_topleft,
  932. int has_topright, ptrdiff_t _stride)
  933. {
  934. pixel *src = (pixel*)_src;
  935. int stride = _stride>>(sizeof(pixel)-1);
  936. PREDICT_8x8_LOAD_TOP;
  937. PREDICT_8x8_LOAD_LEFT;
  938. PREDICT_8x8_LOAD_TOPLEFT;
  939. SRC(0,7)= (l6 + l7 + 1) >> 1;
  940. SRC(1,7)= (l5 + 2*l6 + l7 + 2) >> 2;
  941. SRC(0,6)=SRC(2,7)= (l5 + l6 + 1) >> 1;
  942. SRC(1,6)=SRC(3,7)= (l4 + 2*l5 + l6 + 2) >> 2;
  943. SRC(0,5)=SRC(2,6)=SRC(4,7)= (l4 + l5 + 1) >> 1;
  944. SRC(1,5)=SRC(3,6)=SRC(5,7)= (l3 + 2*l4 + l5 + 2) >> 2;
  945. SRC(0,4)=SRC(2,5)=SRC(4,6)=SRC(6,7)= (l3 + l4 + 1) >> 1;
  946. SRC(1,4)=SRC(3,5)=SRC(5,6)=SRC(7,7)= (l2 + 2*l3 + l4 + 2) >> 2;
  947. SRC(0,3)=SRC(2,4)=SRC(4,5)=SRC(6,6)= (l2 + l3 + 1) >> 1;
  948. SRC(1,3)=SRC(3,4)=SRC(5,5)=SRC(7,6)= (l1 + 2*l2 + l3 + 2) >> 2;
  949. SRC(0,2)=SRC(2,3)=SRC(4,4)=SRC(6,5)= (l1 + l2 + 1) >> 1;
  950. SRC(1,2)=SRC(3,3)=SRC(5,4)=SRC(7,5)= (l0 + 2*l1 + l2 + 2) >> 2;
  951. SRC(0,1)=SRC(2,2)=SRC(4,3)=SRC(6,4)= (l0 + l1 + 1) >> 1;
  952. SRC(1,1)=SRC(3,2)=SRC(5,3)=SRC(7,4)= (lt + 2*l0 + l1 + 2) >> 2;
  953. SRC(0,0)=SRC(2,1)=SRC(4,2)=SRC(6,3)= (lt + l0 + 1) >> 1;
  954. SRC(1,0)=SRC(3,1)=SRC(5,2)=SRC(7,3)= (l0 + 2*lt + t0 + 2) >> 2;
  955. SRC(2,0)=SRC(4,1)=SRC(6,2)= (t1 + 2*t0 + lt + 2) >> 2;
  956. SRC(3,0)=SRC(5,1)=SRC(7,2)= (t2 + 2*t1 + t0 + 2) >> 2;
  957. SRC(4,0)=SRC(6,1)= (t3 + 2*t2 + t1 + 2) >> 2;
  958. SRC(5,0)=SRC(7,1)= (t4 + 2*t3 + t2 + 2) >> 2;
  959. SRC(6,0)= (t5 + 2*t4 + t3 + 2) >> 2;
  960. SRC(7,0)= (t6 + 2*t5 + t4 + 2) >> 2;
  961. }
  962. static void FUNCC(pred8x8l_vertical_left)(uint8_t *_src, int has_topleft,
  963. int has_topright, ptrdiff_t _stride)
  964. {
  965. pixel *src = (pixel*)_src;
  966. int stride = _stride>>(sizeof(pixel)-1);
  967. PREDICT_8x8_LOAD_TOP;
  968. PREDICT_8x8_LOAD_TOPRIGHT;
  969. SRC(0,0)= (t0 + t1 + 1) >> 1;
  970. SRC(0,1)= (t0 + 2*t1 + t2 + 2) >> 2;
  971. SRC(0,2)=SRC(1,0)= (t1 + t2 + 1) >> 1;
  972. SRC(0,3)=SRC(1,1)= (t1 + 2*t2 + t3 + 2) >> 2;
  973. SRC(0,4)=SRC(1,2)=SRC(2,0)= (t2 + t3 + 1) >> 1;
  974. SRC(0,5)=SRC(1,3)=SRC(2,1)= (t2 + 2*t3 + t4 + 2) >> 2;
  975. SRC(0,6)=SRC(1,4)=SRC(2,2)=SRC(3,0)= (t3 + t4 + 1) >> 1;
  976. SRC(0,7)=SRC(1,5)=SRC(2,3)=SRC(3,1)= (t3 + 2*t4 + t5 + 2) >> 2;
  977. SRC(1,6)=SRC(2,4)=SRC(3,2)=SRC(4,0)= (t4 + t5 + 1) >> 1;
  978. SRC(1,7)=SRC(2,5)=SRC(3,3)=SRC(4,1)= (t4 + 2*t5 + t6 + 2) >> 2;
  979. SRC(2,6)=SRC(3,4)=SRC(4,2)=SRC(5,0)= (t5 + t6 + 1) >> 1;
  980. SRC(2,7)=SRC(3,5)=SRC(4,3)=SRC(5,1)= (t5 + 2*t6 + t7 + 2) >> 2;
  981. SRC(3,6)=SRC(4,4)=SRC(5,2)=SRC(6,0)= (t6 + t7 + 1) >> 1;
  982. SRC(3,7)=SRC(4,5)=SRC(5,3)=SRC(6,1)= (t6 + 2*t7 + t8 + 2) >> 2;
  983. SRC(4,6)=SRC(5,4)=SRC(6,2)=SRC(7,0)= (t7 + t8 + 1) >> 1;
  984. SRC(4,7)=SRC(5,5)=SRC(6,3)=SRC(7,1)= (t7 + 2*t8 + t9 + 2) >> 2;
  985. SRC(5,6)=SRC(6,4)=SRC(7,2)= (t8 + t9 + 1) >> 1;
  986. SRC(5,7)=SRC(6,5)=SRC(7,3)= (t8 + 2*t9 + t10 + 2) >> 2;
  987. SRC(6,6)=SRC(7,4)= (t9 + t10 + 1) >> 1;
  988. SRC(6,7)=SRC(7,5)= (t9 + 2*t10 + t11 + 2) >> 2;
  989. SRC(7,6)= (t10 + t11 + 1) >> 1;
  990. SRC(7,7)= (t10 + 2*t11 + t12 + 2) >> 2;
  991. }
  992. static void FUNCC(pred8x8l_horizontal_up)(uint8_t *_src, int has_topleft,
  993. int has_topright, ptrdiff_t _stride)
  994. {
  995. pixel *src = (pixel*)_src;
  996. int stride = _stride>>(sizeof(pixel)-1);
  997. PREDICT_8x8_LOAD_LEFT;
  998. SRC(0,0)= (l0 + l1 + 1) >> 1;
  999. SRC(1,0)= (l0 + 2*l1 + l2 + 2) >> 2;
  1000. SRC(0,1)=SRC(2,0)= (l1 + l2 + 1) >> 1;
  1001. SRC(1,1)=SRC(3,0)= (l1 + 2*l2 + l3 + 2) >> 2;
  1002. SRC(0,2)=SRC(2,1)=SRC(4,0)= (l2 + l3 + 1) >> 1;
  1003. SRC(1,2)=SRC(3,1)=SRC(5,0)= (l2 + 2*l3 + l4 + 2) >> 2;
  1004. SRC(0,3)=SRC(2,2)=SRC(4,1)=SRC(6,0)= (l3 + l4 + 1) >> 1;
  1005. SRC(1,3)=SRC(3,2)=SRC(5,1)=SRC(7,0)= (l3 + 2*l4 + l5 + 2) >> 2;
  1006. SRC(0,4)=SRC(2,3)=SRC(4,2)=SRC(6,1)= (l4 + l5 + 1) >> 1;
  1007. SRC(1,4)=SRC(3,3)=SRC(5,2)=SRC(7,1)= (l4 + 2*l5 + l6 + 2) >> 2;
  1008. SRC(0,5)=SRC(2,4)=SRC(4,3)=SRC(6,2)= (l5 + l6 + 1) >> 1;
  1009. SRC(1,5)=SRC(3,4)=SRC(5,3)=SRC(7,2)= (l5 + 2*l6 + l7 + 2) >> 2;
  1010. SRC(0,6)=SRC(2,5)=SRC(4,4)=SRC(6,3)= (l6 + l7 + 1) >> 1;
  1011. SRC(1,6)=SRC(3,5)=SRC(5,4)=SRC(7,3)= (l6 + 3*l7 + 2) >> 2;
  1012. SRC(0,7)=SRC(1,7)=SRC(2,6)=SRC(2,7)=SRC(3,6)=
  1013. SRC(3,7)=SRC(4,5)=SRC(4,6)=SRC(4,7)=SRC(5,5)=
  1014. SRC(5,6)=SRC(5,7)=SRC(6,4)=SRC(6,5)=SRC(6,6)=
  1015. SRC(6,7)=SRC(7,4)=SRC(7,5)=SRC(7,6)=SRC(7,7)= l7;
  1016. }
  1017. static void FUNCC(pred8x8l_vertical_filter_add)(uint8_t *_src, int16_t *_block, int has_topleft,
  1018. int has_topright, ptrdiff_t _stride)
  1019. {
  1020. int i;
  1021. pixel *src = (pixel*)_src;
  1022. const dctcoef *block = (const dctcoef*)_block;
  1023. pixel pix[8];
  1024. int stride = _stride>>(sizeof(pixel)-1);
  1025. PREDICT_8x8_LOAD_TOP;
  1026. pix[0] = t0;
  1027. pix[1] = t1;
  1028. pix[2] = t2;
  1029. pix[3] = t3;
  1030. pix[4] = t4;
  1031. pix[5] = t5;
  1032. pix[6] = t6;
  1033. pix[7] = t7;
  1034. for(i=0; i<8; i++){
  1035. pixel v = pix[i];
  1036. src[0*stride]= v += block[0];
  1037. src[1*stride]= v += block[8];
  1038. src[2*stride]= v += block[16];
  1039. src[3*stride]= v += block[24];
  1040. src[4*stride]= v += block[32];
  1041. src[5*stride]= v += block[40];
  1042. src[6*stride]= v += block[48];
  1043. src[7*stride]= v + block[56];
  1044. src++;
  1045. block++;
  1046. }
  1047. memset(_block, 0, sizeof(dctcoef) * 64);
  1048. }
  1049. static void FUNCC(pred8x8l_horizontal_filter_add)(uint8_t *_src, int16_t *_block, int has_topleft,
  1050. int has_topright, ptrdiff_t _stride)
  1051. {
  1052. int i;
  1053. pixel *src = (pixel*)_src;
  1054. const dctcoef *block = (const dctcoef*)_block;
  1055. pixel pix[8];
  1056. int stride = _stride>>(sizeof(pixel)-1);
  1057. PREDICT_8x8_LOAD_LEFT;
  1058. pix[0] = l0;
  1059. pix[1] = l1;
  1060. pix[2] = l2;
  1061. pix[3] = l3;
  1062. pix[4] = l4;
  1063. pix[5] = l5;
  1064. pix[6] = l6;
  1065. pix[7] = l7;
  1066. for(i=0; i<8; i++){
  1067. pixel v = pix[i];
  1068. src[0]= v += block[0];
  1069. src[1]= v += block[1];
  1070. src[2]= v += block[2];
  1071. src[3]= v += block[3];
  1072. src[4]= v += block[4];
  1073. src[5]= v += block[5];
  1074. src[6]= v += block[6];
  1075. src[7]= v + block[7];
  1076. src+= stride;
  1077. block+= 8;
  1078. }
  1079. memset(_block, 0, sizeof(dctcoef) * 64);
  1080. }
  1081. #undef PREDICT_8x8_LOAD_LEFT
  1082. #undef PREDICT_8x8_LOAD_TOP
  1083. #undef PREDICT_8x8_LOAD_TOPLEFT
  1084. #undef PREDICT_8x8_LOAD_TOPRIGHT
  1085. #undef PREDICT_8x8_DC
  1086. #undef PTR
  1087. #undef PT
  1088. #undef PL
  1089. #undef SRC
  1090. static void FUNCC(pred4x4_vertical_add)(uint8_t *_pix, int16_t *_block,
  1091. ptrdiff_t stride)
  1092. {
  1093. int i;
  1094. pixel *pix = (pixel*)_pix;
  1095. const dctcoef *block = (const dctcoef*)_block;
  1096. stride >>= sizeof(pixel)-1;
  1097. pix -= stride;
  1098. for(i=0; i<4; i++){
  1099. pixel v = pix[0];
  1100. pix[1*stride]= v += block[0];
  1101. pix[2*stride]= v += block[4];
  1102. pix[3*stride]= v += block[8];
  1103. pix[4*stride]= v + block[12];
  1104. pix++;
  1105. block++;
  1106. }
  1107. memset(_block, 0, sizeof(dctcoef) * 16);
  1108. }
  1109. static void FUNCC(pred4x4_horizontal_add)(uint8_t *_pix, int16_t *_block,
  1110. ptrdiff_t stride)
  1111. {
  1112. int i;
  1113. pixel *pix = (pixel*)_pix;
  1114. const dctcoef *block = (const dctcoef*)_block;
  1115. stride >>= sizeof(pixel)-1;
  1116. for(i=0; i<4; i++){
  1117. pixel v = pix[-1];
  1118. pix[0]= v += block[0];
  1119. pix[1]= v += block[1];
  1120. pix[2]= v += block[2];
  1121. pix[3]= v + block[3];
  1122. pix+= stride;
  1123. block+= 4;
  1124. }
  1125. memset(_block, 0, sizeof(dctcoef) * 16);
  1126. }
  1127. static void FUNCC(pred8x8l_vertical_add)(uint8_t *_pix, int16_t *_block,
  1128. ptrdiff_t stride)
  1129. {
  1130. int i;
  1131. pixel *pix = (pixel*)_pix;
  1132. const dctcoef *block = (const dctcoef*)_block;
  1133. stride >>= sizeof(pixel)-1;
  1134. pix -= stride;
  1135. for(i=0; i<8; i++){
  1136. pixel v = pix[0];
  1137. pix[1*stride]= v += block[0];
  1138. pix[2*stride]= v += block[8];
  1139. pix[3*stride]= v += block[16];
  1140. pix[4*stride]= v += block[24];
  1141. pix[5*stride]= v += block[32];
  1142. pix[6*stride]= v += block[40];
  1143. pix[7*stride]= v += block[48];
  1144. pix[8*stride]= v + block[56];
  1145. pix++;
  1146. block++;
  1147. }
  1148. memset(_block, 0, sizeof(dctcoef) * 64);
  1149. }
  1150. static void FUNCC(pred8x8l_horizontal_add)(uint8_t *_pix, int16_t *_block,
  1151. ptrdiff_t stride)
  1152. {
  1153. int i;
  1154. pixel *pix = (pixel*)_pix;
  1155. const dctcoef *block = (const dctcoef*)_block;
  1156. stride >>= sizeof(pixel)-1;
  1157. for(i=0; i<8; i++){
  1158. pixel v = pix[-1];
  1159. pix[0]= v += block[0];
  1160. pix[1]= v += block[1];
  1161. pix[2]= v += block[2];
  1162. pix[3]= v += block[3];
  1163. pix[4]= v += block[4];
  1164. pix[5]= v += block[5];
  1165. pix[6]= v += block[6];
  1166. pix[7]= v + block[7];
  1167. pix+= stride;
  1168. block+= 8;
  1169. }
  1170. memset(_block, 0, sizeof(dctcoef) * 64);
  1171. }
  1172. static void FUNCC(pred16x16_vertical_add)(uint8_t *pix, const int *block_offset,
  1173. int16_t *block,
  1174. ptrdiff_t stride)
  1175. {
  1176. int i;
  1177. for(i=0; i<16; i++)
  1178. FUNCC(pred4x4_vertical_add)(pix + block_offset[i], block + i*16*sizeof(pixel), stride);
  1179. }
  1180. static void FUNCC(pred16x16_horizontal_add)(uint8_t *pix,
  1181. const int *block_offset,
  1182. int16_t *block,
  1183. ptrdiff_t stride)
  1184. {
  1185. int i;
  1186. for(i=0; i<16; i++)
  1187. FUNCC(pred4x4_horizontal_add)(pix + block_offset[i], block + i*16*sizeof(pixel), stride);
  1188. }
  1189. static void FUNCC(pred8x8_vertical_add)(uint8_t *pix, const int *block_offset,
  1190. int16_t *block, ptrdiff_t stride)
  1191. {
  1192. int i;
  1193. for(i=0; i<4; i++)
  1194. FUNCC(pred4x4_vertical_add)(pix + block_offset[i], block + i*16*sizeof(pixel), stride);
  1195. }
  1196. static void FUNCC(pred8x16_vertical_add)(uint8_t *pix, const int *block_offset,
  1197. int16_t *block, ptrdiff_t stride)
  1198. {
  1199. int i;
  1200. for(i=0; i<4; i++)
  1201. FUNCC(pred4x4_vertical_add)(pix + block_offset[i], block + i*16*sizeof(pixel), stride);
  1202. for(i=4; i<8; i++)
  1203. FUNCC(pred4x4_vertical_add)(pix + block_offset[i+4], block + i*16*sizeof(pixel), stride);
  1204. }
  1205. static void FUNCC(pred8x8_horizontal_add)(uint8_t *pix, const int *block_offset,
  1206. int16_t *block,
  1207. ptrdiff_t stride)
  1208. {
  1209. int i;
  1210. for(i=0; i<4; i++)
  1211. FUNCC(pred4x4_horizontal_add)(pix + block_offset[i], block + i*16*sizeof(pixel), stride);
  1212. }
  1213. static void FUNCC(pred8x16_horizontal_add)(uint8_t *pix,
  1214. const int *block_offset,
  1215. int16_t *block, ptrdiff_t stride)
  1216. {
  1217. int i;
  1218. for(i=0; i<4; i++)
  1219. FUNCC(pred4x4_horizontal_add)(pix + block_offset[i], block + i*16*sizeof(pixel), stride);
  1220. for(i=4; i<8; i++)
  1221. FUNCC(pred4x4_horizontal_add)(pix + block_offset[i+4], block + i*16*sizeof(pixel), stride);
  1222. }