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.

990 lines
33KB

  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 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 "mathops.h"
  27. #include "dsputil.h"
  28. #define BIT_DEPTH 8
  29. #define pixel uint8_t
  30. #define pixel4 uint32_t
  31. #define dctcoef DCTELEM
  32. #define INIT_CLIP uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
  33. #define CLIP(a) cm[a]
  34. #define FUNC(a) a
  35. #define FUNCC(a) a ## _c
  36. #define PIXEL_SPLAT_X4(x) ((x)*0x01010101U)
  37. #define AV_WN4P AV_WN32
  38. #define AV_WN4PA AV_WN32A
  39. static void FUNCC(pred4x4_vertical)(uint8_t *_src, const uint8_t *topright, int _stride){
  40. pixel *src = (pixel*)_src;
  41. int stride = _stride/sizeof(pixel);
  42. const pixel4 a= ((pixel4*)(src-stride))[0];
  43. ((pixel4*)(src+0*stride))[0]= a;
  44. ((pixel4*)(src+1*stride))[0]= a;
  45. ((pixel4*)(src+2*stride))[0]= a;
  46. ((pixel4*)(src+3*stride))[0]= a;
  47. }
  48. static void FUNCC(pred4x4_horizontal)(uint8_t *_src, const uint8_t *topright, int _stride){
  49. pixel *src = (pixel*)_src;
  50. int stride = _stride/sizeof(pixel);
  51. ((pixel4*)(src+0*stride))[0]= PIXEL_SPLAT_X4(src[-1+0*stride]);
  52. ((pixel4*)(src+1*stride))[0]= PIXEL_SPLAT_X4(src[-1+1*stride]);
  53. ((pixel4*)(src+2*stride))[0]= PIXEL_SPLAT_X4(src[-1+2*stride]);
  54. ((pixel4*)(src+3*stride))[0]= PIXEL_SPLAT_X4(src[-1+3*stride]);
  55. }
  56. static void FUNCC(pred4x4_dc)(uint8_t *_src, const uint8_t *topright, int _stride){
  57. pixel *src = (pixel*)_src;
  58. int stride = _stride/sizeof(pixel);
  59. const int dc= ( src[-stride] + src[1-stride] + src[2-stride] + src[3-stride]
  60. + src[-1+0*stride] + src[-1+1*stride] + src[-1+2*stride] + src[-1+3*stride] + 4) >>3;
  61. ((pixel4*)(src+0*stride))[0]=
  62. ((pixel4*)(src+1*stride))[0]=
  63. ((pixel4*)(src+2*stride))[0]=
  64. ((pixel4*)(src+3*stride))[0]= PIXEL_SPLAT_X4(dc);
  65. }
  66. static void FUNCC(pred4x4_left_dc)(uint8_t *_src, const uint8_t *topright, int _stride){
  67. pixel *src = (pixel*)_src;
  68. int stride = _stride/sizeof(pixel);
  69. const int dc= ( src[-1+0*stride] + src[-1+1*stride] + src[-1+2*stride] + src[-1+3*stride] + 2) >>2;
  70. ((pixel4*)(src+0*stride))[0]=
  71. ((pixel4*)(src+1*stride))[0]=
  72. ((pixel4*)(src+2*stride))[0]=
  73. ((pixel4*)(src+3*stride))[0]= PIXEL_SPLAT_X4(dc);
  74. }
  75. static void FUNCC(pred4x4_top_dc)(uint8_t *_src, const uint8_t *topright, int _stride){
  76. pixel *src = (pixel*)_src;
  77. int stride = _stride/sizeof(pixel);
  78. const int dc= ( src[-stride] + src[1-stride] + src[2-stride] + src[3-stride] + 2) >>2;
  79. ((pixel4*)(src+0*stride))[0]=
  80. ((pixel4*)(src+1*stride))[0]=
  81. ((pixel4*)(src+2*stride))[0]=
  82. ((pixel4*)(src+3*stride))[0]= PIXEL_SPLAT_X4(dc);
  83. }
  84. static void FUNCC(pred4x4_128_dc)(uint8_t *_src, const uint8_t *topright, int _stride){
  85. pixel *src = (pixel*)_src;
  86. int stride = _stride/sizeof(pixel);
  87. ((pixel4*)(src+0*stride))[0]=
  88. ((pixel4*)(src+1*stride))[0]=
  89. ((pixel4*)(src+2*stride))[0]=
  90. ((pixel4*)(src+3*stride))[0]= PIXEL_SPLAT_X4(1<<(BIT_DEPTH-1));
  91. }
  92. static void FUNCC(pred4x4_127_dc)(uint8_t *_src, const uint8_t *topright, int _stride){
  93. pixel *src = (pixel*)_src;
  94. int stride = _stride/sizeof(pixel);
  95. ((pixel4*)(src+0*stride))[0]=
  96. ((pixel4*)(src+1*stride))[0]=
  97. ((pixel4*)(src+2*stride))[0]=
  98. ((pixel4*)(src+3*stride))[0]= PIXEL_SPLAT_X4((1<<(BIT_DEPTH-1))-1);
  99. }
  100. static void FUNCC(pred4x4_129_dc)(uint8_t *_src, const uint8_t *topright, int _stride){
  101. pixel *src = (pixel*)_src;
  102. int stride = _stride/sizeof(pixel);
  103. ((pixel4*)(src+0*stride))[0]=
  104. ((pixel4*)(src+1*stride))[0]=
  105. ((pixel4*)(src+2*stride))[0]=
  106. ((pixel4*)(src+3*stride))[0]= PIXEL_SPLAT_X4((1<<(BIT_DEPTH-1))+1);
  107. }
  108. #define LOAD_TOP_RIGHT_EDGE\
  109. const int av_unused t4= topright[0];\
  110. const int av_unused t5= topright[1];\
  111. const int av_unused t6= topright[2];\
  112. const int av_unused t7= topright[3];\
  113. #define LOAD_DOWN_LEFT_EDGE\
  114. const int av_unused l4= src[-1+4*stride];\
  115. const int av_unused l5= src[-1+5*stride];\
  116. const int av_unused l6= src[-1+6*stride];\
  117. const int av_unused l7= src[-1+7*stride];\
  118. #define LOAD_LEFT_EDGE\
  119. const int av_unused l0= src[-1+0*stride];\
  120. const int av_unused l1= src[-1+1*stride];\
  121. const int av_unused l2= src[-1+2*stride];\
  122. const int av_unused l3= src[-1+3*stride];\
  123. #define LOAD_TOP_EDGE\
  124. const int av_unused t0= src[ 0-1*stride];\
  125. const int av_unused t1= src[ 1-1*stride];\
  126. const int av_unused t2= src[ 2-1*stride];\
  127. const int av_unused t3= src[ 3-1*stride];\
  128. static void FUNCC(pred4x4_down_right)(uint8_t *_src, const uint8_t *topright, int _stride){
  129. pixel *src = (pixel*)_src;
  130. int stride = _stride/sizeof(pixel);
  131. const int lt= src[-1-1*stride];
  132. LOAD_TOP_EDGE
  133. LOAD_LEFT_EDGE
  134. src[0+3*stride]=(l3 + 2*l2 + l1 + 2)>>2;
  135. src[0+2*stride]=
  136. src[1+3*stride]=(l2 + 2*l1 + l0 + 2)>>2;
  137. src[0+1*stride]=
  138. src[1+2*stride]=
  139. src[2+3*stride]=(l1 + 2*l0 + lt + 2)>>2;
  140. src[0+0*stride]=
  141. src[1+1*stride]=
  142. src[2+2*stride]=
  143. src[3+3*stride]=(l0 + 2*lt + t0 + 2)>>2;
  144. src[1+0*stride]=
  145. src[2+1*stride]=
  146. src[3+2*stride]=(lt + 2*t0 + t1 + 2)>>2;
  147. src[2+0*stride]=
  148. src[3+1*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  149. src[3+0*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  150. }
  151. static void FUNCC(pred4x4_down_left)(uint8_t *_src, const uint8_t *_topright, int _stride){
  152. pixel *src = (pixel*)_src;
  153. const pixel *topright = (const pixel*)_topright;
  154. int stride = _stride/sizeof(pixel);
  155. LOAD_TOP_EDGE
  156. LOAD_TOP_RIGHT_EDGE
  157. // LOAD_LEFT_EDGE
  158. src[0+0*stride]=(t0 + t2 + 2*t1 + 2)>>2;
  159. src[1+0*stride]=
  160. src[0+1*stride]=(t1 + t3 + 2*t2 + 2)>>2;
  161. src[2+0*stride]=
  162. src[1+1*stride]=
  163. src[0+2*stride]=(t2 + t4 + 2*t3 + 2)>>2;
  164. src[3+0*stride]=
  165. src[2+1*stride]=
  166. src[1+2*stride]=
  167. src[0+3*stride]=(t3 + t5 + 2*t4 + 2)>>2;
  168. src[3+1*stride]=
  169. src[2+2*stride]=
  170. src[1+3*stride]=(t4 + t6 + 2*t5 + 2)>>2;
  171. src[3+2*stride]=
  172. src[2+3*stride]=(t5 + t7 + 2*t6 + 2)>>2;
  173. src[3+3*stride]=(t6 + 3*t7 + 2)>>2;
  174. }
  175. static void FUNCC(pred4x4_vertical_right)(uint8_t *_src, const uint8_t *topright, int _stride){
  176. pixel *src = (pixel*)_src;
  177. int stride = _stride/sizeof(pixel);
  178. const int lt= src[-1-1*stride];
  179. LOAD_TOP_EDGE
  180. LOAD_LEFT_EDGE
  181. src[0+0*stride]=
  182. src[1+2*stride]=(lt + t0 + 1)>>1;
  183. src[1+0*stride]=
  184. src[2+2*stride]=(t0 + t1 + 1)>>1;
  185. src[2+0*stride]=
  186. src[3+2*stride]=(t1 + t2 + 1)>>1;
  187. src[3+0*stride]=(t2 + t3 + 1)>>1;
  188. src[0+1*stride]=
  189. src[1+3*stride]=(l0 + 2*lt + t0 + 2)>>2;
  190. src[1+1*stride]=
  191. src[2+3*stride]=(lt + 2*t0 + t1 + 2)>>2;
  192. src[2+1*stride]=
  193. src[3+3*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  194. src[3+1*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  195. src[0+2*stride]=(lt + 2*l0 + l1 + 2)>>2;
  196. src[0+3*stride]=(l0 + 2*l1 + l2 + 2)>>2;
  197. }
  198. static void FUNCC(pred4x4_vertical_left)(uint8_t *_src, const uint8_t *_topright, int _stride){
  199. pixel *src = (pixel*)_src;
  200. const pixel *topright = (const pixel*)_topright;
  201. int stride = _stride/sizeof(pixel);
  202. LOAD_TOP_EDGE
  203. LOAD_TOP_RIGHT_EDGE
  204. src[0+0*stride]=(t0 + t1 + 1)>>1;
  205. src[1+0*stride]=
  206. src[0+2*stride]=(t1 + t2 + 1)>>1;
  207. src[2+0*stride]=
  208. src[1+2*stride]=(t2 + t3 + 1)>>1;
  209. src[3+0*stride]=
  210. src[2+2*stride]=(t3 + t4+ 1)>>1;
  211. src[3+2*stride]=(t4 + t5+ 1)>>1;
  212. src[0+1*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  213. src[1+1*stride]=
  214. src[0+3*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  215. src[2+1*stride]=
  216. src[1+3*stride]=(t2 + 2*t3 + t4 + 2)>>2;
  217. src[3+1*stride]=
  218. src[2+3*stride]=(t3 + 2*t4 + t5 + 2)>>2;
  219. src[3+3*stride]=(t4 + 2*t5 + t6 + 2)>>2;
  220. }
  221. static void FUNCC(pred4x4_horizontal_up)(uint8_t *_src, const uint8_t *topright, int _stride){
  222. pixel *src = (pixel*)_src;
  223. int stride = _stride/sizeof(pixel);
  224. LOAD_LEFT_EDGE
  225. src[0+0*stride]=(l0 + l1 + 1)>>1;
  226. src[1+0*stride]=(l0 + 2*l1 + l2 + 2)>>2;
  227. src[2+0*stride]=
  228. src[0+1*stride]=(l1 + l2 + 1)>>1;
  229. src[3+0*stride]=
  230. src[1+1*stride]=(l1 + 2*l2 + l3 + 2)>>2;
  231. src[2+1*stride]=
  232. src[0+2*stride]=(l2 + l3 + 1)>>1;
  233. src[3+1*stride]=
  234. src[1+2*stride]=(l2 + 2*l3 + l3 + 2)>>2;
  235. src[3+2*stride]=
  236. src[1+3*stride]=
  237. src[0+3*stride]=
  238. src[2+2*stride]=
  239. src[2+3*stride]=
  240. src[3+3*stride]=l3;
  241. }
  242. static void FUNCC(pred4x4_horizontal_down)(uint8_t *_src, const uint8_t *topright, int _stride){
  243. pixel *src = (pixel*)_src;
  244. int stride = _stride/sizeof(pixel);
  245. const int lt= src[-1-1*stride];
  246. LOAD_TOP_EDGE
  247. LOAD_LEFT_EDGE
  248. src[0+0*stride]=
  249. src[2+1*stride]=(lt + l0 + 1)>>1;
  250. src[1+0*stride]=
  251. src[3+1*stride]=(l0 + 2*lt + t0 + 2)>>2;
  252. src[2+0*stride]=(lt + 2*t0 + t1 + 2)>>2;
  253. src[3+0*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  254. src[0+1*stride]=
  255. src[2+2*stride]=(l0 + l1 + 1)>>1;
  256. src[1+1*stride]=
  257. src[3+2*stride]=(lt + 2*l0 + l1 + 2)>>2;
  258. src[0+2*stride]=
  259. src[2+3*stride]=(l1 + l2+ 1)>>1;
  260. src[1+2*stride]=
  261. src[3+3*stride]=(l0 + 2*l1 + l2 + 2)>>2;
  262. src[0+3*stride]=(l2 + l3 + 1)>>1;
  263. src[1+3*stride]=(l1 + 2*l2 + l3 + 2)>>2;
  264. }
  265. static void FUNCC(pred16x16_vertical)(uint8_t *_src, int _stride){
  266. int i;
  267. pixel *src = (pixel*)_src;
  268. int stride = _stride/sizeof(pixel);
  269. const pixel4 a = ((pixel4*)(src-stride))[0];
  270. const pixel4 b = ((pixel4*)(src-stride))[1];
  271. const pixel4 c = ((pixel4*)(src-stride))[2];
  272. const pixel4 d = ((pixel4*)(src-stride))[3];
  273. for(i=0; i<16; i++){
  274. ((pixel4*)(src+i*stride))[0] = a;
  275. ((pixel4*)(src+i*stride))[1] = b;
  276. ((pixel4*)(src+i*stride))[2] = c;
  277. ((pixel4*)(src+i*stride))[3] = d;
  278. }
  279. }
  280. static void FUNCC(pred16x16_horizontal)(uint8_t *_src, int stride){
  281. int i;
  282. pixel *src = (pixel*)_src;
  283. stride /= sizeof(pixel);
  284. for(i=0; i<16; i++){
  285. ((pixel4*)(src+i*stride))[0] =
  286. ((pixel4*)(src+i*stride))[1] =
  287. ((pixel4*)(src+i*stride))[2] =
  288. ((pixel4*)(src+i*stride))[3] = PIXEL_SPLAT_X4(src[-1+i*stride]);
  289. }
  290. }
  291. #define PREDICT_16x16_DC(v)\
  292. for(i=0; i<16; i++){\
  293. AV_WN4P(src+ 0, v);\
  294. AV_WN4P(src+ 4, v);\
  295. AV_WN4P(src+ 8, v);\
  296. AV_WN4P(src+12, v);\
  297. src += stride;\
  298. }
  299. static void FUNCC(pred16x16_dc)(uint8_t *_src, int stride){
  300. int i, dc=0;
  301. pixel *src = (pixel*)_src;
  302. pixel4 dcsplat;
  303. stride /= sizeof(pixel);
  304. for(i=0;i<16; i++){
  305. dc+= src[-1+i*stride];
  306. }
  307. for(i=0;i<16; i++){
  308. dc+= src[i-stride];
  309. }
  310. dcsplat = PIXEL_SPLAT_X4((dc+16)>>5);
  311. PREDICT_16x16_DC(dcsplat);
  312. }
  313. static void FUNCC(pred16x16_left_dc)(uint8_t *_src, int stride){
  314. int i, dc=0;
  315. pixel *src = (pixel*)_src;
  316. pixel4 dcsplat;
  317. stride /= sizeof(pixel);
  318. for(i=0;i<16; i++){
  319. dc+= src[-1+i*stride];
  320. }
  321. dcsplat = PIXEL_SPLAT_X4((dc+8)>>4);
  322. PREDICT_16x16_DC(dcsplat);
  323. }
  324. static void FUNCC(pred16x16_top_dc)(uint8_t *_src, int stride){
  325. int i, dc=0;
  326. pixel *src = (pixel*)_src;
  327. pixel4 dcsplat;
  328. stride /= sizeof(pixel);
  329. for(i=0;i<16; i++){
  330. dc+= src[i-stride];
  331. }
  332. dcsplat = PIXEL_SPLAT_X4((dc+8)>>4);
  333. PREDICT_16x16_DC(dcsplat);
  334. }
  335. #define PRED16x16_X(n, v) \
  336. static void FUNCC(pred16x16_##n##_dc)(uint8_t *_src, int stride){\
  337. int i;\
  338. pixel *src = (pixel*)_src;\
  339. stride /= sizeof(pixel);\
  340. PREDICT_16x16_DC(PIXEL_SPLAT_X4(v));\
  341. }
  342. PRED16x16_X(127, (1<<(BIT_DEPTH-1))-1);
  343. PRED16x16_X(128, (1<<(BIT_DEPTH-1))+0);
  344. PRED16x16_X(129, (1<<(BIT_DEPTH-1))+1);
  345. static inline void FUNCC(pred16x16_plane_compat)(uint8_t *_src, int _stride, const int svq3, const int rv40){
  346. int i, j, k;
  347. int a;
  348. INIT_CLIP
  349. pixel *src = (pixel*)_src;
  350. int stride = _stride/sizeof(pixel);
  351. const pixel * const src0 = src +7-stride;
  352. const pixel * src1 = src +8*stride-1;
  353. const pixel * src2 = src1-2*stride; // == src+6*stride-1;
  354. int H = src0[1] - src0[-1];
  355. int V = src1[0] - src2[ 0];
  356. for(k=2; k<=8; ++k) {
  357. src1 += stride; src2 -= stride;
  358. H += k*(src0[k] - src0[-k]);
  359. V += k*(src1[0] - src2[ 0]);
  360. }
  361. if(svq3){
  362. H = ( 5*(H/4) ) / 16;
  363. V = ( 5*(V/4) ) / 16;
  364. /* required for 100% accuracy */
  365. i = H; H = V; V = i;
  366. }else if(rv40){
  367. H = ( H + (H>>2) ) >> 4;
  368. V = ( V + (V>>2) ) >> 4;
  369. }else{
  370. H = ( 5*H+32 ) >> 6;
  371. V = ( 5*V+32 ) >> 6;
  372. }
  373. a = 16*(src1[0] + src2[16] + 1) - 7*(V+H);
  374. for(j=16; j>0; --j) {
  375. int b = a;
  376. a += V;
  377. for(i=-16; i<0; i+=4) {
  378. src[16+i] = CLIP((b ) >> 5);
  379. src[17+i] = CLIP((b+ H) >> 5);
  380. src[18+i] = CLIP((b+2*H) >> 5);
  381. src[19+i] = CLIP((b+3*H) >> 5);
  382. b += 4*H;
  383. }
  384. src += stride;
  385. }
  386. }
  387. static void FUNCC(pred16x16_plane)(uint8_t *src, int stride){
  388. FUNCC(pred16x16_plane_compat)(src, stride, 0, 0);
  389. }
  390. static void FUNCC(pred8x8_vertical)(uint8_t *_src, int _stride){
  391. int i;
  392. pixel *src = (pixel*)_src;
  393. int stride = _stride/sizeof(pixel);
  394. const pixel4 a= ((pixel4*)(src-stride))[0];
  395. const pixel4 b= ((pixel4*)(src-stride))[1];
  396. for(i=0; i<8; i++){
  397. ((pixel4*)(src+i*stride))[0]= a;
  398. ((pixel4*)(src+i*stride))[1]= b;
  399. }
  400. }
  401. static void FUNCC(pred8x8_horizontal)(uint8_t *_src, int stride){
  402. int i;
  403. pixel *src = (pixel*)_src;
  404. stride /= sizeof(pixel);
  405. for(i=0; i<8; i++){
  406. ((pixel4*)(src+i*stride))[0]=
  407. ((pixel4*)(src+i*stride))[1]= PIXEL_SPLAT_X4(src[-1+i*stride]);
  408. }
  409. }
  410. #define PRED8x8_X(n, v)\
  411. static void FUNCC(pred8x8_##n##_dc)(uint8_t *_src, int stride){\
  412. int i;\
  413. pixel *src = (pixel*)_src;\
  414. stride /= sizeof(pixel);\
  415. for(i=0; i<8; i++){\
  416. ((pixel4*)(src+i*stride))[0]=\
  417. ((pixel4*)(src+i*stride))[1]= PIXEL_SPLAT_X4(v);\
  418. }\
  419. }
  420. PRED8x8_X(127, (1<<(BIT_DEPTH-1))-1);
  421. PRED8x8_X(128, (1<<(BIT_DEPTH-1))+0);
  422. PRED8x8_X(129, (1<<(BIT_DEPTH-1))+1);
  423. static void FUNCC(pred8x8_left_dc)(uint8_t *_src, int stride){
  424. int i;
  425. int dc0, dc2;
  426. pixel4 dc0splat, dc2splat;
  427. pixel *src = (pixel*)_src;
  428. stride /= sizeof(pixel);
  429. dc0=dc2=0;
  430. for(i=0;i<4; i++){
  431. dc0+= src[-1+i*stride];
  432. dc2+= src[-1+(i+4)*stride];
  433. }
  434. dc0splat = PIXEL_SPLAT_X4((dc0 + 2)>>2);
  435. dc2splat = PIXEL_SPLAT_X4((dc2 + 2)>>2);
  436. for(i=0; i<4; i++){
  437. ((pixel4*)(src+i*stride))[0]=
  438. ((pixel4*)(src+i*stride))[1]= dc0splat;
  439. }
  440. for(i=4; i<8; i++){
  441. ((pixel4*)(src+i*stride))[0]=
  442. ((pixel4*)(src+i*stride))[1]= dc2splat;
  443. }
  444. }
  445. static void FUNCC(pred8x8_top_dc)(uint8_t *_src, int stride){
  446. int i;
  447. int dc0, dc1;
  448. pixel4 dc0splat, dc1splat;
  449. pixel *src = (pixel*)_src;
  450. stride /= sizeof(pixel);
  451. dc0=dc1=0;
  452. for(i=0;i<4; i++){
  453. dc0+= src[i-stride];
  454. dc1+= src[4+i-stride];
  455. }
  456. dc0splat = PIXEL_SPLAT_X4((dc0 + 2)>>2);
  457. dc1splat = PIXEL_SPLAT_X4((dc1 + 2)>>2);
  458. for(i=0; i<4; i++){
  459. ((pixel4*)(src+i*stride))[0]= dc0splat;
  460. ((pixel4*)(src+i*stride))[1]= dc1splat;
  461. }
  462. for(i=4; i<8; i++){
  463. ((pixel4*)(src+i*stride))[0]= dc0splat;
  464. ((pixel4*)(src+i*stride))[1]= dc1splat;
  465. }
  466. }
  467. static void FUNCC(pred8x8_dc)(uint8_t *_src, int stride){
  468. int i;
  469. int dc0, dc1, dc2;
  470. pixel4 dc0splat, dc1splat, dc2splat, dc3splat;
  471. pixel *src = (pixel*)_src;
  472. stride /= sizeof(pixel);
  473. dc0=dc1=dc2=0;
  474. for(i=0;i<4; i++){
  475. dc0+= src[-1+i*stride] + src[i-stride];
  476. dc1+= src[4+i-stride];
  477. dc2+= src[-1+(i+4)*stride];
  478. }
  479. dc0splat = PIXEL_SPLAT_X4((dc0 + 4)>>3);
  480. dc1splat = PIXEL_SPLAT_X4((dc1 + 2)>>2);
  481. dc2splat = PIXEL_SPLAT_X4((dc2 + 2)>>2);
  482. dc3splat = PIXEL_SPLAT_X4((dc1 + dc2 + 4)>>3);
  483. for(i=0; i<4; i++){
  484. ((pixel4*)(src+i*stride))[0]= dc0splat;
  485. ((pixel4*)(src+i*stride))[1]= dc1splat;
  486. }
  487. for(i=4; i<8; i++){
  488. ((pixel4*)(src+i*stride))[0]= dc2splat;
  489. ((pixel4*)(src+i*stride))[1]= dc3splat;
  490. }
  491. }
  492. //the following 4 function should not be optimized!
  493. static void FUNC(pred8x8_mad_cow_dc_l0t)(uint8_t *src, int stride){
  494. FUNCC(pred8x8_top_dc)(src, stride);
  495. FUNCC(pred4x4_dc)(src, NULL, stride);
  496. }
  497. static void FUNC(pred8x8_mad_cow_dc_0lt)(uint8_t *src, int stride){
  498. FUNCC(pred8x8_dc)(src, stride);
  499. FUNCC(pred4x4_top_dc)(src, NULL, stride);
  500. }
  501. static void FUNC(pred8x8_mad_cow_dc_l00)(uint8_t *src, int stride){
  502. FUNCC(pred8x8_left_dc)(src, stride);
  503. FUNCC(pred4x4_128_dc)(src + 4*stride , NULL, stride);
  504. FUNCC(pred4x4_128_dc)(src + 4*stride + 4*sizeof(pixel), NULL, stride);
  505. }
  506. static void FUNC(pred8x8_mad_cow_dc_0l0)(uint8_t *src, int stride){
  507. FUNCC(pred8x8_left_dc)(src, stride);
  508. FUNCC(pred4x4_128_dc)(src , NULL, stride);
  509. FUNCC(pred4x4_128_dc)(src + 4*sizeof(pixel), NULL, stride);
  510. }
  511. static void FUNCC(pred8x8_plane)(uint8_t *_src, int _stride){
  512. int j, k;
  513. int a;
  514. INIT_CLIP
  515. pixel *src = (pixel*)_src;
  516. int stride = _stride/sizeof(pixel);
  517. const pixel * const src0 = src +3-stride;
  518. const pixel * src1 = src +4*stride-1;
  519. const pixel * src2 = src1-2*stride; // == src+2*stride-1;
  520. int H = src0[1] - src0[-1];
  521. int V = src1[0] - src2[ 0];
  522. for(k=2; k<=4; ++k) {
  523. src1 += stride; src2 -= stride;
  524. H += k*(src0[k] - src0[-k]);
  525. V += k*(src1[0] - src2[ 0]);
  526. }
  527. H = ( 17*H+16 ) >> 5;
  528. V = ( 17*V+16 ) >> 5;
  529. a = 16*(src1[0] + src2[8]+1) - 3*(V+H);
  530. for(j=8; j>0; --j) {
  531. int b = a;
  532. a += V;
  533. src[0] = CLIP((b ) >> 5);
  534. src[1] = CLIP((b+ H) >> 5);
  535. src[2] = CLIP((b+2*H) >> 5);
  536. src[3] = CLIP((b+3*H) >> 5);
  537. src[4] = CLIP((b+4*H) >> 5);
  538. src[5] = CLIP((b+5*H) >> 5);
  539. src[6] = CLIP((b+6*H) >> 5);
  540. src[7] = CLIP((b+7*H) >> 5);
  541. src += stride;
  542. }
  543. }
  544. #define SRC(x,y) src[(x)+(y)*stride]
  545. #define PL(y) \
  546. const int l##y = (SRC(-1,y-1) + 2*SRC(-1,y) + SRC(-1,y+1) + 2) >> 2;
  547. #define PREDICT_8x8_LOAD_LEFT \
  548. const int l0 = ((has_topleft ? SRC(-1,-1) : SRC(-1,0)) \
  549. + 2*SRC(-1,0) + SRC(-1,1) + 2) >> 2; \
  550. PL(1) PL(2) PL(3) PL(4) PL(5) PL(6) \
  551. const int l7 av_unused = (SRC(-1,6) + 3*SRC(-1,7) + 2) >> 2
  552. #define PT(x) \
  553. const int t##x = (SRC(x-1,-1) + 2*SRC(x,-1) + SRC(x+1,-1) + 2) >> 2;
  554. #define PREDICT_8x8_LOAD_TOP \
  555. const int t0 = ((has_topleft ? SRC(-1,-1) : SRC(0,-1)) \
  556. + 2*SRC(0,-1) + SRC(1,-1) + 2) >> 2; \
  557. PT(1) PT(2) PT(3) PT(4) PT(5) PT(6) \
  558. const int t7 av_unused = ((has_topright ? SRC(8,-1) : SRC(7,-1)) \
  559. + 2*SRC(7,-1) + SRC(6,-1) + 2) >> 2
  560. #define PTR(x) \
  561. t##x = (SRC(x-1,-1) + 2*SRC(x,-1) + SRC(x+1,-1) + 2) >> 2;
  562. #define PREDICT_8x8_LOAD_TOPRIGHT \
  563. int t8, t9, t10, t11, t12, t13, t14, t15; \
  564. if(has_topright) { \
  565. PTR(8) PTR(9) PTR(10) PTR(11) PTR(12) PTR(13) PTR(14) \
  566. t15 = (SRC(14,-1) + 3*SRC(15,-1) + 2) >> 2; \
  567. } else t8=t9=t10=t11=t12=t13=t14=t15= SRC(7,-1);
  568. #define PREDICT_8x8_LOAD_TOPLEFT \
  569. const int lt = (SRC(-1,0) + 2*SRC(-1,-1) + SRC(0,-1) + 2) >> 2
  570. #define PREDICT_8x8_DC(v) \
  571. int y; \
  572. for( y = 0; y < 8; y++ ) { \
  573. ((pixel4*)src)[0] = \
  574. ((pixel4*)src)[1] = v; \
  575. src += stride; \
  576. }
  577. static void FUNCC(pred8x8l_128_dc)(uint8_t *_src, int has_topleft, int has_topright, int _stride)
  578. {
  579. pixel *src = (pixel*)_src;
  580. int stride = _stride/sizeof(pixel);
  581. PREDICT_8x8_DC(PIXEL_SPLAT_X4(1<<(BIT_DEPTH-1)));
  582. }
  583. static void FUNCC(pred8x8l_left_dc)(uint8_t *_src, int has_topleft, int has_topright, int _stride)
  584. {
  585. pixel *src = (pixel*)_src;
  586. int stride = _stride/sizeof(pixel);
  587. PREDICT_8x8_LOAD_LEFT;
  588. const pixel4 dc = PIXEL_SPLAT_X4((l0+l1+l2+l3+l4+l5+l6+l7+4) >> 3);
  589. PREDICT_8x8_DC(dc);
  590. }
  591. static void FUNCC(pred8x8l_top_dc)(uint8_t *_src, int has_topleft, int has_topright, int _stride)
  592. {
  593. pixel *src = (pixel*)_src;
  594. int stride = _stride/sizeof(pixel);
  595. PREDICT_8x8_LOAD_TOP;
  596. const pixel4 dc = PIXEL_SPLAT_X4((t0+t1+t2+t3+t4+t5+t6+t7+4) >> 3);
  597. PREDICT_8x8_DC(dc);
  598. }
  599. static void FUNCC(pred8x8l_dc)(uint8_t *_src, int has_topleft, int has_topright, int _stride)
  600. {
  601. pixel *src = (pixel*)_src;
  602. int stride = _stride/sizeof(pixel);
  603. PREDICT_8x8_LOAD_LEFT;
  604. PREDICT_8x8_LOAD_TOP;
  605. const pixel4 dc = PIXEL_SPLAT_X4((l0+l1+l2+l3+l4+l5+l6+l7
  606. +t0+t1+t2+t3+t4+t5+t6+t7+8) >> 4);
  607. PREDICT_8x8_DC(dc);
  608. }
  609. static void FUNCC(pred8x8l_horizontal)(uint8_t *_src, int has_topleft, int has_topright, int _stride)
  610. {
  611. pixel *src = (pixel*)_src;
  612. int stride = _stride/sizeof(pixel);
  613. PREDICT_8x8_LOAD_LEFT;
  614. #define ROW(y) ((pixel4*)(src+y*stride))[0] =\
  615. ((pixel4*)(src+y*stride))[1] = PIXEL_SPLAT_X4(l##y)
  616. ROW(0); ROW(1); ROW(2); ROW(3); ROW(4); ROW(5); ROW(6); ROW(7);
  617. #undef ROW
  618. }
  619. static void FUNCC(pred8x8l_vertical)(uint8_t *_src, int has_topleft, int has_topright, int _stride)
  620. {
  621. int y;
  622. pixel *src = (pixel*)_src;
  623. int stride = _stride/sizeof(pixel);
  624. PREDICT_8x8_LOAD_TOP;
  625. src[0] = t0;
  626. src[1] = t1;
  627. src[2] = t2;
  628. src[3] = t3;
  629. src[4] = t4;
  630. src[5] = t5;
  631. src[6] = t6;
  632. src[7] = t7;
  633. for( y = 1; y < 8; y++ ) {
  634. ((pixel4*)(src+y*stride))[0] = ((pixel4*)src)[0];
  635. ((pixel4*)(src+y*stride))[1] = ((pixel4*)src)[1];
  636. }
  637. }
  638. static void FUNCC(pred8x8l_down_left)(uint8_t *_src, int has_topleft, int has_topright, int _stride)
  639. {
  640. pixel *src = (pixel*)_src;
  641. int stride = _stride/sizeof(pixel);
  642. PREDICT_8x8_LOAD_TOP;
  643. PREDICT_8x8_LOAD_TOPRIGHT;
  644. SRC(0,0)= (t0 + 2*t1 + t2 + 2) >> 2;
  645. SRC(0,1)=SRC(1,0)= (t1 + 2*t2 + t3 + 2) >> 2;
  646. SRC(0,2)=SRC(1,1)=SRC(2,0)= (t2 + 2*t3 + t4 + 2) >> 2;
  647. SRC(0,3)=SRC(1,2)=SRC(2,1)=SRC(3,0)= (t3 + 2*t4 + t5 + 2) >> 2;
  648. SRC(0,4)=SRC(1,3)=SRC(2,2)=SRC(3,1)=SRC(4,0)= (t4 + 2*t5 + t6 + 2) >> 2;
  649. SRC(0,5)=SRC(1,4)=SRC(2,3)=SRC(3,2)=SRC(4,1)=SRC(5,0)= (t5 + 2*t6 + t7 + 2) >> 2;
  650. 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;
  651. 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;
  652. 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;
  653. SRC(2,7)=SRC(3,6)=SRC(4,5)=SRC(5,4)=SRC(6,3)=SRC(7,2)= (t9 + 2*t10 + t11 + 2) >> 2;
  654. SRC(3,7)=SRC(4,6)=SRC(5,5)=SRC(6,4)=SRC(7,3)= (t10 + 2*t11 + t12 + 2) >> 2;
  655. SRC(4,7)=SRC(5,6)=SRC(6,5)=SRC(7,4)= (t11 + 2*t12 + t13 + 2) >> 2;
  656. SRC(5,7)=SRC(6,6)=SRC(7,5)= (t12 + 2*t13 + t14 + 2) >> 2;
  657. SRC(6,7)=SRC(7,6)= (t13 + 2*t14 + t15 + 2) >> 2;
  658. SRC(7,7)= (t14 + 3*t15 + 2) >> 2;
  659. }
  660. static void FUNCC(pred8x8l_down_right)(uint8_t *_src, int has_topleft, int has_topright, int _stride)
  661. {
  662. pixel *src = (pixel*)_src;
  663. int stride = _stride/sizeof(pixel);
  664. PREDICT_8x8_LOAD_TOP;
  665. PREDICT_8x8_LOAD_LEFT;
  666. PREDICT_8x8_LOAD_TOPLEFT;
  667. SRC(0,7)= (l7 + 2*l6 + l5 + 2) >> 2;
  668. SRC(0,6)=SRC(1,7)= (l6 + 2*l5 + l4 + 2) >> 2;
  669. SRC(0,5)=SRC(1,6)=SRC(2,7)= (l5 + 2*l4 + l3 + 2) >> 2;
  670. SRC(0,4)=SRC(1,5)=SRC(2,6)=SRC(3,7)= (l4 + 2*l3 + l2 + 2) >> 2;
  671. SRC(0,3)=SRC(1,4)=SRC(2,5)=SRC(3,6)=SRC(4,7)= (l3 + 2*l2 + l1 + 2) >> 2;
  672. SRC(0,2)=SRC(1,3)=SRC(2,4)=SRC(3,5)=SRC(4,6)=SRC(5,7)= (l2 + 2*l1 + l0 + 2) >> 2;
  673. 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;
  674. 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;
  675. 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;
  676. SRC(2,0)=SRC(3,1)=SRC(4,2)=SRC(5,3)=SRC(6,4)=SRC(7,5)= (t0 + 2*t1 + t2 + 2) >> 2;
  677. SRC(3,0)=SRC(4,1)=SRC(5,2)=SRC(6,3)=SRC(7,4)= (t1 + 2*t2 + t3 + 2) >> 2;
  678. SRC(4,0)=SRC(5,1)=SRC(6,2)=SRC(7,3)= (t2 + 2*t3 + t4 + 2) >> 2;
  679. SRC(5,0)=SRC(6,1)=SRC(7,2)= (t3 + 2*t4 + t5 + 2) >> 2;
  680. SRC(6,0)=SRC(7,1)= (t4 + 2*t5 + t6 + 2) >> 2;
  681. SRC(7,0)= (t5 + 2*t6 + t7 + 2) >> 2;
  682. }
  683. static void FUNCC(pred8x8l_vertical_right)(uint8_t *_src, int has_topleft, int has_topright, int _stride)
  684. {
  685. pixel *src = (pixel*)_src;
  686. int stride = _stride/sizeof(pixel);
  687. PREDICT_8x8_LOAD_TOP;
  688. PREDICT_8x8_LOAD_LEFT;
  689. PREDICT_8x8_LOAD_TOPLEFT;
  690. SRC(0,6)= (l5 + 2*l4 + l3 + 2) >> 2;
  691. SRC(0,7)= (l6 + 2*l5 + l4 + 2) >> 2;
  692. SRC(0,4)=SRC(1,6)= (l3 + 2*l2 + l1 + 2) >> 2;
  693. SRC(0,5)=SRC(1,7)= (l4 + 2*l3 + l2 + 2) >> 2;
  694. SRC(0,2)=SRC(1,4)=SRC(2,6)= (l1 + 2*l0 + lt + 2) >> 2;
  695. SRC(0,3)=SRC(1,5)=SRC(2,7)= (l2 + 2*l1 + l0 + 2) >> 2;
  696. SRC(0,1)=SRC(1,3)=SRC(2,5)=SRC(3,7)= (l0 + 2*lt + t0 + 2) >> 2;
  697. SRC(0,0)=SRC(1,2)=SRC(2,4)=SRC(3,6)= (lt + t0 + 1) >> 1;
  698. SRC(1,1)=SRC(2,3)=SRC(3,5)=SRC(4,7)= (lt + 2*t0 + t1 + 2) >> 2;
  699. SRC(1,0)=SRC(2,2)=SRC(3,4)=SRC(4,6)= (t0 + t1 + 1) >> 1;
  700. SRC(2,1)=SRC(3,3)=SRC(4,5)=SRC(5,7)= (t0 + 2*t1 + t2 + 2) >> 2;
  701. SRC(2,0)=SRC(3,2)=SRC(4,4)=SRC(5,6)= (t1 + t2 + 1) >> 1;
  702. SRC(3,1)=SRC(4,3)=SRC(5,5)=SRC(6,7)= (t1 + 2*t2 + t3 + 2) >> 2;
  703. SRC(3,0)=SRC(4,2)=SRC(5,4)=SRC(6,6)= (t2 + t3 + 1) >> 1;
  704. SRC(4,1)=SRC(5,3)=SRC(6,5)=SRC(7,7)= (t2 + 2*t3 + t4 + 2) >> 2;
  705. SRC(4,0)=SRC(5,2)=SRC(6,4)=SRC(7,6)= (t3 + t4 + 1) >> 1;
  706. SRC(5,1)=SRC(6,3)=SRC(7,5)= (t3 + 2*t4 + t5 + 2) >> 2;
  707. SRC(5,0)=SRC(6,2)=SRC(7,4)= (t4 + t5 + 1) >> 1;
  708. SRC(6,1)=SRC(7,3)= (t4 + 2*t5 + t6 + 2) >> 2;
  709. SRC(6,0)=SRC(7,2)= (t5 + t6 + 1) >> 1;
  710. SRC(7,1)= (t5 + 2*t6 + t7 + 2) >> 2;
  711. SRC(7,0)= (t6 + t7 + 1) >> 1;
  712. }
  713. static void FUNCC(pred8x8l_horizontal_down)(uint8_t *_src, int has_topleft, int has_topright, int _stride)
  714. {
  715. pixel *src = (pixel*)_src;
  716. int stride = _stride/sizeof(pixel);
  717. PREDICT_8x8_LOAD_TOP;
  718. PREDICT_8x8_LOAD_LEFT;
  719. PREDICT_8x8_LOAD_TOPLEFT;
  720. SRC(0,7)= (l6 + l7 + 1) >> 1;
  721. SRC(1,7)= (l5 + 2*l6 + l7 + 2) >> 2;
  722. SRC(0,6)=SRC(2,7)= (l5 + l6 + 1) >> 1;
  723. SRC(1,6)=SRC(3,7)= (l4 + 2*l5 + l6 + 2) >> 2;
  724. SRC(0,5)=SRC(2,6)=SRC(4,7)= (l4 + l5 + 1) >> 1;
  725. SRC(1,5)=SRC(3,6)=SRC(5,7)= (l3 + 2*l4 + l5 + 2) >> 2;
  726. SRC(0,4)=SRC(2,5)=SRC(4,6)=SRC(6,7)= (l3 + l4 + 1) >> 1;
  727. SRC(1,4)=SRC(3,5)=SRC(5,6)=SRC(7,7)= (l2 + 2*l3 + l4 + 2) >> 2;
  728. SRC(0,3)=SRC(2,4)=SRC(4,5)=SRC(6,6)= (l2 + l3 + 1) >> 1;
  729. SRC(1,3)=SRC(3,4)=SRC(5,5)=SRC(7,6)= (l1 + 2*l2 + l3 + 2) >> 2;
  730. SRC(0,2)=SRC(2,3)=SRC(4,4)=SRC(6,5)= (l1 + l2 + 1) >> 1;
  731. SRC(1,2)=SRC(3,3)=SRC(5,4)=SRC(7,5)= (l0 + 2*l1 + l2 + 2) >> 2;
  732. SRC(0,1)=SRC(2,2)=SRC(4,3)=SRC(6,4)= (l0 + l1 + 1) >> 1;
  733. SRC(1,1)=SRC(3,2)=SRC(5,3)=SRC(7,4)= (lt + 2*l0 + l1 + 2) >> 2;
  734. SRC(0,0)=SRC(2,1)=SRC(4,2)=SRC(6,3)= (lt + l0 + 1) >> 1;
  735. SRC(1,0)=SRC(3,1)=SRC(5,2)=SRC(7,3)= (l0 + 2*lt + t0 + 2) >> 2;
  736. SRC(2,0)=SRC(4,1)=SRC(6,2)= (t1 + 2*t0 + lt + 2) >> 2;
  737. SRC(3,0)=SRC(5,1)=SRC(7,2)= (t2 + 2*t1 + t0 + 2) >> 2;
  738. SRC(4,0)=SRC(6,1)= (t3 + 2*t2 + t1 + 2) >> 2;
  739. SRC(5,0)=SRC(7,1)= (t4 + 2*t3 + t2 + 2) >> 2;
  740. SRC(6,0)= (t5 + 2*t4 + t3 + 2) >> 2;
  741. SRC(7,0)= (t6 + 2*t5 + t4 + 2) >> 2;
  742. }
  743. static void FUNCC(pred8x8l_vertical_left)(uint8_t *_src, int has_topleft, int has_topright, int _stride)
  744. {
  745. pixel *src = (pixel*)_src;
  746. int stride = _stride/sizeof(pixel);
  747. PREDICT_8x8_LOAD_TOP;
  748. PREDICT_8x8_LOAD_TOPRIGHT;
  749. SRC(0,0)= (t0 + t1 + 1) >> 1;
  750. SRC(0,1)= (t0 + 2*t1 + t2 + 2) >> 2;
  751. SRC(0,2)=SRC(1,0)= (t1 + t2 + 1) >> 1;
  752. SRC(0,3)=SRC(1,1)= (t1 + 2*t2 + t3 + 2) >> 2;
  753. SRC(0,4)=SRC(1,2)=SRC(2,0)= (t2 + t3 + 1) >> 1;
  754. SRC(0,5)=SRC(1,3)=SRC(2,1)= (t2 + 2*t3 + t4 + 2) >> 2;
  755. SRC(0,6)=SRC(1,4)=SRC(2,2)=SRC(3,0)= (t3 + t4 + 1) >> 1;
  756. SRC(0,7)=SRC(1,5)=SRC(2,3)=SRC(3,1)= (t3 + 2*t4 + t5 + 2) >> 2;
  757. SRC(1,6)=SRC(2,4)=SRC(3,2)=SRC(4,0)= (t4 + t5 + 1) >> 1;
  758. SRC(1,7)=SRC(2,5)=SRC(3,3)=SRC(4,1)= (t4 + 2*t5 + t6 + 2) >> 2;
  759. SRC(2,6)=SRC(3,4)=SRC(4,2)=SRC(5,0)= (t5 + t6 + 1) >> 1;
  760. SRC(2,7)=SRC(3,5)=SRC(4,3)=SRC(5,1)= (t5 + 2*t6 + t7 + 2) >> 2;
  761. SRC(3,6)=SRC(4,4)=SRC(5,2)=SRC(6,0)= (t6 + t7 + 1) >> 1;
  762. SRC(3,7)=SRC(4,5)=SRC(5,3)=SRC(6,1)= (t6 + 2*t7 + t8 + 2) >> 2;
  763. SRC(4,6)=SRC(5,4)=SRC(6,2)=SRC(7,0)= (t7 + t8 + 1) >> 1;
  764. SRC(4,7)=SRC(5,5)=SRC(6,3)=SRC(7,1)= (t7 + 2*t8 + t9 + 2) >> 2;
  765. SRC(5,6)=SRC(6,4)=SRC(7,2)= (t8 + t9 + 1) >> 1;
  766. SRC(5,7)=SRC(6,5)=SRC(7,3)= (t8 + 2*t9 + t10 + 2) >> 2;
  767. SRC(6,6)=SRC(7,4)= (t9 + t10 + 1) >> 1;
  768. SRC(6,7)=SRC(7,5)= (t9 + 2*t10 + t11 + 2) >> 2;
  769. SRC(7,6)= (t10 + t11 + 1) >> 1;
  770. SRC(7,7)= (t10 + 2*t11 + t12 + 2) >> 2;
  771. }
  772. static void FUNCC(pred8x8l_horizontal_up)(uint8_t *_src, int has_topleft, int has_topright, int _stride)
  773. {
  774. pixel *src = (pixel*)_src;
  775. int stride = _stride/sizeof(pixel);
  776. PREDICT_8x8_LOAD_LEFT;
  777. SRC(0,0)= (l0 + l1 + 1) >> 1;
  778. SRC(1,0)= (l0 + 2*l1 + l2 + 2) >> 2;
  779. SRC(0,1)=SRC(2,0)= (l1 + l2 + 1) >> 1;
  780. SRC(1,1)=SRC(3,0)= (l1 + 2*l2 + l3 + 2) >> 2;
  781. SRC(0,2)=SRC(2,1)=SRC(4,0)= (l2 + l3 + 1) >> 1;
  782. SRC(1,2)=SRC(3,1)=SRC(5,0)= (l2 + 2*l3 + l4 + 2) >> 2;
  783. SRC(0,3)=SRC(2,2)=SRC(4,1)=SRC(6,0)= (l3 + l4 + 1) >> 1;
  784. SRC(1,3)=SRC(3,2)=SRC(5,1)=SRC(7,0)= (l3 + 2*l4 + l5 + 2) >> 2;
  785. SRC(0,4)=SRC(2,3)=SRC(4,2)=SRC(6,1)= (l4 + l5 + 1) >> 1;
  786. SRC(1,4)=SRC(3,3)=SRC(5,2)=SRC(7,1)= (l4 + 2*l5 + l6 + 2) >> 2;
  787. SRC(0,5)=SRC(2,4)=SRC(4,3)=SRC(6,2)= (l5 + l6 + 1) >> 1;
  788. SRC(1,5)=SRC(3,4)=SRC(5,3)=SRC(7,2)= (l5 + 2*l6 + l7 + 2) >> 2;
  789. SRC(0,6)=SRC(2,5)=SRC(4,4)=SRC(6,3)= (l6 + l7 + 1) >> 1;
  790. SRC(1,6)=SRC(3,5)=SRC(5,4)=SRC(7,3)= (l6 + 3*l7 + 2) >> 2;
  791. SRC(0,7)=SRC(1,7)=SRC(2,6)=SRC(2,7)=SRC(3,6)=
  792. SRC(3,7)=SRC(4,5)=SRC(4,6)=SRC(4,7)=SRC(5,5)=
  793. SRC(5,6)=SRC(5,7)=SRC(6,4)=SRC(6,5)=SRC(6,6)=
  794. SRC(6,7)=SRC(7,4)=SRC(7,5)=SRC(7,6)=SRC(7,7)= l7;
  795. }
  796. #undef PREDICT_8x8_LOAD_LEFT
  797. #undef PREDICT_8x8_LOAD_TOP
  798. #undef PREDICT_8x8_LOAD_TOPLEFT
  799. #undef PREDICT_8x8_LOAD_TOPRIGHT
  800. #undef PREDICT_8x8_DC
  801. #undef PTR
  802. #undef PT
  803. #undef PL
  804. #undef SRC
  805. static void FUNCC(pred4x4_vertical_add)(uint8_t *_pix, const DCTELEM *_block, int stride){
  806. int i;
  807. pixel *pix = (pixel*)_pix;
  808. const dctcoef *block = (const dctcoef*)_block;
  809. stride /= sizeof(pixel);
  810. pix -= stride;
  811. for(i=0; i<4; i++){
  812. pixel v = pix[0];
  813. pix[1*stride]= v += block[0];
  814. pix[2*stride]= v += block[4];
  815. pix[3*stride]= v += block[8];
  816. pix[4*stride]= v + block[12];
  817. pix++;
  818. block++;
  819. }
  820. }
  821. static void FUNCC(pred4x4_horizontal_add)(uint8_t *_pix, const DCTELEM *_block, int stride){
  822. int i;
  823. pixel *pix = (pixel*)_pix;
  824. const dctcoef *block = (const dctcoef*)_block;
  825. stride /= sizeof(pixel);
  826. for(i=0; i<4; i++){
  827. pixel v = pix[-1];
  828. pix[0]= v += block[0];
  829. pix[1]= v += block[1];
  830. pix[2]= v += block[2];
  831. pix[3]= v + block[3];
  832. pix+= stride;
  833. block+= 4;
  834. }
  835. }
  836. static void FUNCC(pred8x8l_vertical_add)(uint8_t *_pix, const DCTELEM *_block, int stride){
  837. int i;
  838. pixel *pix = (pixel*)_pix;
  839. const dctcoef *block = (const dctcoef*)_block;
  840. stride /= sizeof(pixel);
  841. pix -= stride;
  842. for(i=0; i<8; i++){
  843. pixel v = pix[0];
  844. pix[1*stride]= v += block[0];
  845. pix[2*stride]= v += block[8];
  846. pix[3*stride]= v += block[16];
  847. pix[4*stride]= v += block[24];
  848. pix[5*stride]= v += block[32];
  849. pix[6*stride]= v += block[40];
  850. pix[7*stride]= v += block[48];
  851. pix[8*stride]= v + block[56];
  852. pix++;
  853. block++;
  854. }
  855. }
  856. static void FUNCC(pred8x8l_horizontal_add)(uint8_t *_pix, const DCTELEM *_block, int stride){
  857. int i;
  858. pixel *pix = (pixel*)_pix;
  859. const dctcoef *block = (const dctcoef*)_block;
  860. stride /= sizeof(pixel);
  861. for(i=0; i<8; i++){
  862. pixel v = pix[-1];
  863. pix[0]= v += block[0];
  864. pix[1]= v += block[1];
  865. pix[2]= v += block[2];
  866. pix[3]= v += block[3];
  867. pix[4]= v += block[4];
  868. pix[5]= v += block[5];
  869. pix[6]= v += block[6];
  870. pix[7]= v + block[7];
  871. pix+= stride;
  872. block+= 8;
  873. }
  874. }
  875. static void FUNCC(pred16x16_vertical_add)(uint8_t *pix, const int *block_offset, const DCTELEM *block, int stride){
  876. int i;
  877. for(i=0; i<16; i++)
  878. FUNCC(pred4x4_vertical_add)(pix + block_offset[i], block + i*16*sizeof(pixel), stride);
  879. }
  880. static void FUNCC(pred16x16_horizontal_add)(uint8_t *pix, const int *block_offset, const DCTELEM *block, int stride){
  881. int i;
  882. for(i=0; i<16; i++)
  883. FUNCC(pred4x4_horizontal_add)(pix + block_offset[i], block + i*16*sizeof(pixel), stride);
  884. }
  885. static void FUNCC(pred8x8_vertical_add)(uint8_t *pix, const int *block_offset, const DCTELEM *block, int stride){
  886. int i;
  887. for(i=0; i<4; i++)
  888. FUNCC(pred4x4_vertical_add)(pix + block_offset[i], block + i*16*sizeof(pixel), stride);
  889. }
  890. static void FUNCC(pred8x8_horizontal_add)(uint8_t *pix, const int *block_offset, const DCTELEM *block, int stride){
  891. int i;
  892. for(i=0; i<4; i++)
  893. FUNCC(pred4x4_horizontal_add)(pix + block_offset[i], block + i*16*sizeof(pixel), stride);
  894. }