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.

2447 lines
110KB

  1. /*
  2. * DSP utils
  3. * Copyright (c) 2000, 2001 Fabrice Bellard
  4. * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
  5. *
  6. * gmc & q-pel & 32/64 bit based MC by Michael Niedermayer <michaelni@gmx.at>
  7. *
  8. * This file is part of Libav.
  9. *
  10. * Libav is free software; you can redistribute it and/or
  11. * modify it under the terms of the GNU Lesser General Public
  12. * License as published by the Free Software Foundation; either
  13. * version 2.1 of the License, or (at your option) any later version.
  14. *
  15. * Libav is distributed in the hope that it will be useful,
  16. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  18. * Lesser General Public License for more details.
  19. *
  20. * You should have received a copy of the GNU Lesser General Public
  21. * License along with Libav; if not, write to the Free Software
  22. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  23. */
  24. /**
  25. * @file
  26. * DSP utils
  27. */
  28. #include "libavutil/attributes.h"
  29. #include "libavutil/imgutils.h"
  30. #include "avcodec.h"
  31. #include "copy_block.h"
  32. #include "dct.h"
  33. #include "dsputil.h"
  34. #include "simple_idct.h"
  35. #include "faandct.h"
  36. #include "faanidct.h"
  37. #include "imgconvert.h"
  38. #include "mathops.h"
  39. #include "mpegvideo.h"
  40. #include "config.h"
  41. uint32_t ff_square_tab[512] = { 0, };
  42. #define BIT_DEPTH 16
  43. #include "dsputilenc_template.c"
  44. #undef BIT_DEPTH
  45. #define BIT_DEPTH 8
  46. #include "hpel_template.c"
  47. #include "tpel_template.c"
  48. #include "dsputil_template.c"
  49. #include "dsputilenc_template.c"
  50. /* Input permutation for the simple_idct_mmx */
  51. static const uint8_t simple_mmx_permutation[64] = {
  52. 0x00, 0x08, 0x04, 0x09, 0x01, 0x0C, 0x05, 0x0D,
  53. 0x10, 0x18, 0x14, 0x19, 0x11, 0x1C, 0x15, 0x1D,
  54. 0x20, 0x28, 0x24, 0x29, 0x21, 0x2C, 0x25, 0x2D,
  55. 0x12, 0x1A, 0x16, 0x1B, 0x13, 0x1E, 0x17, 0x1F,
  56. 0x02, 0x0A, 0x06, 0x0B, 0x03, 0x0E, 0x07, 0x0F,
  57. 0x30, 0x38, 0x34, 0x39, 0x31, 0x3C, 0x35, 0x3D,
  58. 0x22, 0x2A, 0x26, 0x2B, 0x23, 0x2E, 0x27, 0x2F,
  59. 0x32, 0x3A, 0x36, 0x3B, 0x33, 0x3E, 0x37, 0x3F,
  60. };
  61. static const uint8_t idct_sse2_row_perm[8] = { 0, 4, 1, 5, 2, 6, 3, 7 };
  62. av_cold void ff_init_scantable(uint8_t *permutation, ScanTable *st,
  63. const uint8_t *src_scantable)
  64. {
  65. int i, end;
  66. st->scantable = src_scantable;
  67. for (i = 0; i < 64; i++) {
  68. int j = src_scantable[i];
  69. st->permutated[i] = permutation[j];
  70. }
  71. end = -1;
  72. for (i = 0; i < 64; i++) {
  73. int j = st->permutated[i];
  74. if (j > end)
  75. end = j;
  76. st->raster_end[i] = end;
  77. }
  78. }
  79. av_cold void ff_init_scantable_permutation(uint8_t *idct_permutation,
  80. int idct_permutation_type)
  81. {
  82. int i;
  83. switch (idct_permutation_type) {
  84. case FF_NO_IDCT_PERM:
  85. for (i = 0; i < 64; i++)
  86. idct_permutation[i] = i;
  87. break;
  88. case FF_LIBMPEG2_IDCT_PERM:
  89. for (i = 0; i < 64; i++)
  90. idct_permutation[i] = (i & 0x38) | ((i & 6) >> 1) | ((i & 1) << 2);
  91. break;
  92. case FF_SIMPLE_IDCT_PERM:
  93. for (i = 0; i < 64; i++)
  94. idct_permutation[i] = simple_mmx_permutation[i];
  95. break;
  96. case FF_TRANSPOSE_IDCT_PERM:
  97. for (i = 0; i < 64; i++)
  98. idct_permutation[i] = ((i & 7) << 3) | (i >> 3);
  99. break;
  100. case FF_PARTTRANS_IDCT_PERM:
  101. for (i = 0; i < 64; i++)
  102. idct_permutation[i] = (i & 0x24) | ((i & 3) << 3) | ((i >> 3) & 3);
  103. break;
  104. case FF_SSE2_IDCT_PERM:
  105. for (i = 0; i < 64; i++)
  106. idct_permutation[i] = (i & 0x38) | idct_sse2_row_perm[i & 7];
  107. break;
  108. default:
  109. av_log(NULL, AV_LOG_ERROR,
  110. "Internal error, IDCT permutation not set\n");
  111. }
  112. }
  113. static int pix_sum_c(uint8_t *pix, int line_size)
  114. {
  115. int s = 0, i, j;
  116. for (i = 0; i < 16; i++) {
  117. for (j = 0; j < 16; j += 8) {
  118. s += pix[0];
  119. s += pix[1];
  120. s += pix[2];
  121. s += pix[3];
  122. s += pix[4];
  123. s += pix[5];
  124. s += pix[6];
  125. s += pix[7];
  126. pix += 8;
  127. }
  128. pix += line_size - 16;
  129. }
  130. return s;
  131. }
  132. static int pix_norm1_c(uint8_t *pix, int line_size)
  133. {
  134. int s = 0, i, j;
  135. uint32_t *sq = ff_square_tab + 256;
  136. for (i = 0; i < 16; i++) {
  137. for (j = 0; j < 16; j += 8) {
  138. #if 0
  139. s += sq[pix[0]];
  140. s += sq[pix[1]];
  141. s += sq[pix[2]];
  142. s += sq[pix[3]];
  143. s += sq[pix[4]];
  144. s += sq[pix[5]];
  145. s += sq[pix[6]];
  146. s += sq[pix[7]];
  147. #else
  148. #if HAVE_FAST_64BIT
  149. register uint64_t x = *(uint64_t *) pix;
  150. s += sq[x & 0xff];
  151. s += sq[(x >> 8) & 0xff];
  152. s += sq[(x >> 16) & 0xff];
  153. s += sq[(x >> 24) & 0xff];
  154. s += sq[(x >> 32) & 0xff];
  155. s += sq[(x >> 40) & 0xff];
  156. s += sq[(x >> 48) & 0xff];
  157. s += sq[(x >> 56) & 0xff];
  158. #else
  159. register uint32_t x = *(uint32_t *) pix;
  160. s += sq[x & 0xff];
  161. s += sq[(x >> 8) & 0xff];
  162. s += sq[(x >> 16) & 0xff];
  163. s += sq[(x >> 24) & 0xff];
  164. x = *(uint32_t *) (pix + 4);
  165. s += sq[x & 0xff];
  166. s += sq[(x >> 8) & 0xff];
  167. s += sq[(x >> 16) & 0xff];
  168. s += sq[(x >> 24) & 0xff];
  169. #endif
  170. #endif
  171. pix += 8;
  172. }
  173. pix += line_size - 16;
  174. }
  175. return s;
  176. }
  177. static void bswap_buf(uint32_t *dst, const uint32_t *src, int w)
  178. {
  179. int i;
  180. for (i = 0; i + 8 <= w; i += 8) {
  181. dst[i + 0] = av_bswap32(src[i + 0]);
  182. dst[i + 1] = av_bswap32(src[i + 1]);
  183. dst[i + 2] = av_bswap32(src[i + 2]);
  184. dst[i + 3] = av_bswap32(src[i + 3]);
  185. dst[i + 4] = av_bswap32(src[i + 4]);
  186. dst[i + 5] = av_bswap32(src[i + 5]);
  187. dst[i + 6] = av_bswap32(src[i + 6]);
  188. dst[i + 7] = av_bswap32(src[i + 7]);
  189. }
  190. for (; i < w; i++)
  191. dst[i + 0] = av_bswap32(src[i + 0]);
  192. }
  193. static void bswap16_buf(uint16_t *dst, const uint16_t *src, int len)
  194. {
  195. while (len--)
  196. *dst++ = av_bswap16(*src++);
  197. }
  198. static int sse4_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  199. int line_size, int h)
  200. {
  201. int s = 0, i;
  202. uint32_t *sq = ff_square_tab + 256;
  203. for (i = 0; i < h; i++) {
  204. s += sq[pix1[0] - pix2[0]];
  205. s += sq[pix1[1] - pix2[1]];
  206. s += sq[pix1[2] - pix2[2]];
  207. s += sq[pix1[3] - pix2[3]];
  208. pix1 += line_size;
  209. pix2 += line_size;
  210. }
  211. return s;
  212. }
  213. static int sse8_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  214. int line_size, int h)
  215. {
  216. int s = 0, i;
  217. uint32_t *sq = ff_square_tab + 256;
  218. for (i = 0; i < h; i++) {
  219. s += sq[pix1[0] - pix2[0]];
  220. s += sq[pix1[1] - pix2[1]];
  221. s += sq[pix1[2] - pix2[2]];
  222. s += sq[pix1[3] - pix2[3]];
  223. s += sq[pix1[4] - pix2[4]];
  224. s += sq[pix1[5] - pix2[5]];
  225. s += sq[pix1[6] - pix2[6]];
  226. s += sq[pix1[7] - pix2[7]];
  227. pix1 += line_size;
  228. pix2 += line_size;
  229. }
  230. return s;
  231. }
  232. static int sse16_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  233. int line_size, int h)
  234. {
  235. int s = 0, i;
  236. uint32_t *sq = ff_square_tab + 256;
  237. for (i = 0; i < h; i++) {
  238. s += sq[pix1[0] - pix2[0]];
  239. s += sq[pix1[1] - pix2[1]];
  240. s += sq[pix1[2] - pix2[2]];
  241. s += sq[pix1[3] - pix2[3]];
  242. s += sq[pix1[4] - pix2[4]];
  243. s += sq[pix1[5] - pix2[5]];
  244. s += sq[pix1[6] - pix2[6]];
  245. s += sq[pix1[7] - pix2[7]];
  246. s += sq[pix1[8] - pix2[8]];
  247. s += sq[pix1[9] - pix2[9]];
  248. s += sq[pix1[10] - pix2[10]];
  249. s += sq[pix1[11] - pix2[11]];
  250. s += sq[pix1[12] - pix2[12]];
  251. s += sq[pix1[13] - pix2[13]];
  252. s += sq[pix1[14] - pix2[14]];
  253. s += sq[pix1[15] - pix2[15]];
  254. pix1 += line_size;
  255. pix2 += line_size;
  256. }
  257. return s;
  258. }
  259. static void diff_pixels_c(int16_t *restrict block, const uint8_t *s1,
  260. const uint8_t *s2, int stride)
  261. {
  262. int i;
  263. /* read the pixels */
  264. for (i = 0; i < 8; i++) {
  265. block[0] = s1[0] - s2[0];
  266. block[1] = s1[1] - s2[1];
  267. block[2] = s1[2] - s2[2];
  268. block[3] = s1[3] - s2[3];
  269. block[4] = s1[4] - s2[4];
  270. block[5] = s1[5] - s2[5];
  271. block[6] = s1[6] - s2[6];
  272. block[7] = s1[7] - s2[7];
  273. s1 += stride;
  274. s2 += stride;
  275. block += 8;
  276. }
  277. }
  278. static void put_pixels_clamped_c(const int16_t *block, uint8_t *restrict pixels,
  279. int line_size)
  280. {
  281. int i;
  282. /* read the pixels */
  283. for (i = 0; i < 8; i++) {
  284. pixels[0] = av_clip_uint8(block[0]);
  285. pixels[1] = av_clip_uint8(block[1]);
  286. pixels[2] = av_clip_uint8(block[2]);
  287. pixels[3] = av_clip_uint8(block[3]);
  288. pixels[4] = av_clip_uint8(block[4]);
  289. pixels[5] = av_clip_uint8(block[5]);
  290. pixels[6] = av_clip_uint8(block[6]);
  291. pixels[7] = av_clip_uint8(block[7]);
  292. pixels += line_size;
  293. block += 8;
  294. }
  295. }
  296. static void put_signed_pixels_clamped_c(const int16_t *block,
  297. uint8_t *restrict pixels,
  298. int line_size)
  299. {
  300. int i, j;
  301. for (i = 0; i < 8; i++) {
  302. for (j = 0; j < 8; j++) {
  303. if (*block < -128)
  304. *pixels = 0;
  305. else if (*block > 127)
  306. *pixels = 255;
  307. else
  308. *pixels = (uint8_t) (*block + 128);
  309. block++;
  310. pixels++;
  311. }
  312. pixels += (line_size - 8);
  313. }
  314. }
  315. static void add_pixels8_c(uint8_t *restrict pixels, int16_t *block,
  316. int line_size)
  317. {
  318. int i;
  319. for (i = 0; i < 8; i++) {
  320. pixels[0] += block[0];
  321. pixels[1] += block[1];
  322. pixels[2] += block[2];
  323. pixels[3] += block[3];
  324. pixels[4] += block[4];
  325. pixels[5] += block[5];
  326. pixels[6] += block[6];
  327. pixels[7] += block[7];
  328. pixels += line_size;
  329. block += 8;
  330. }
  331. }
  332. static void add_pixels_clamped_c(const int16_t *block, uint8_t *restrict pixels,
  333. int line_size)
  334. {
  335. int i;
  336. /* read the pixels */
  337. for (i = 0; i < 8; i++) {
  338. pixels[0] = av_clip_uint8(pixels[0] + block[0]);
  339. pixels[1] = av_clip_uint8(pixels[1] + block[1]);
  340. pixels[2] = av_clip_uint8(pixels[2] + block[2]);
  341. pixels[3] = av_clip_uint8(pixels[3] + block[3]);
  342. pixels[4] = av_clip_uint8(pixels[4] + block[4]);
  343. pixels[5] = av_clip_uint8(pixels[5] + block[5]);
  344. pixels[6] = av_clip_uint8(pixels[6] + block[6]);
  345. pixels[7] = av_clip_uint8(pixels[7] + block[7]);
  346. pixels += line_size;
  347. block += 8;
  348. }
  349. }
  350. static int sum_abs_dctelem_c(int16_t *block)
  351. {
  352. int sum = 0, i;
  353. for (i = 0; i < 64; i++)
  354. sum += FFABS(block[i]);
  355. return sum;
  356. }
  357. static void fill_block16_c(uint8_t *block, uint8_t value, int line_size, int h)
  358. {
  359. int i;
  360. for (i = 0; i < h; i++) {
  361. memset(block, value, 16);
  362. block += line_size;
  363. }
  364. }
  365. static void fill_block8_c(uint8_t *block, uint8_t value, int line_size, int h)
  366. {
  367. int i;
  368. for (i = 0; i < h; i++) {
  369. memset(block, value, 8);
  370. block += line_size;
  371. }
  372. }
  373. #define avg2(a, b) ((a + b + 1) >> 1)
  374. #define avg4(a, b, c, d) ((a + b + c + d + 2) >> 2)
  375. static void gmc1_c(uint8_t *dst, uint8_t *src, int stride, int h,
  376. int x16, int y16, int rounder)
  377. {
  378. const int A = (16 - x16) * (16 - y16);
  379. const int B = (x16) * (16 - y16);
  380. const int C = (16 - x16) * (y16);
  381. const int D = (x16) * (y16);
  382. int i;
  383. for (i = 0; i < h; i++) {
  384. dst[0] = (A * src[0] + B * src[1] + C * src[stride + 0] + D * src[stride + 1] + rounder) >> 8;
  385. dst[1] = (A * src[1] + B * src[2] + C * src[stride + 1] + D * src[stride + 2] + rounder) >> 8;
  386. dst[2] = (A * src[2] + B * src[3] + C * src[stride + 2] + D * src[stride + 3] + rounder) >> 8;
  387. dst[3] = (A * src[3] + B * src[4] + C * src[stride + 3] + D * src[stride + 4] + rounder) >> 8;
  388. dst[4] = (A * src[4] + B * src[5] + C * src[stride + 4] + D * src[stride + 5] + rounder) >> 8;
  389. dst[5] = (A * src[5] + B * src[6] + C * src[stride + 5] + D * src[stride + 6] + rounder) >> 8;
  390. dst[6] = (A * src[6] + B * src[7] + C * src[stride + 6] + D * src[stride + 7] + rounder) >> 8;
  391. dst[7] = (A * src[7] + B * src[8] + C * src[stride + 7] + D * src[stride + 8] + rounder) >> 8;
  392. dst += stride;
  393. src += stride;
  394. }
  395. }
  396. void ff_gmc_c(uint8_t *dst, uint8_t *src, int stride, int h, int ox, int oy,
  397. int dxx, int dxy, int dyx, int dyy, int shift, int r,
  398. int width, int height)
  399. {
  400. int y, vx, vy;
  401. const int s = 1 << shift;
  402. width--;
  403. height--;
  404. for (y = 0; y < h; y++) {
  405. int x;
  406. vx = ox;
  407. vy = oy;
  408. for (x = 0; x < 8; x++) { // FIXME: optimize
  409. int index;
  410. int src_x = vx >> 16;
  411. int src_y = vy >> 16;
  412. int frac_x = src_x & (s - 1);
  413. int frac_y = src_y & (s - 1);
  414. src_x >>= shift;
  415. src_y >>= shift;
  416. if ((unsigned) src_x < width) {
  417. if ((unsigned) src_y < height) {
  418. index = src_x + src_y * stride;
  419. dst[y * stride + x] =
  420. ((src[index] * (s - frac_x) +
  421. src[index + 1] * frac_x) * (s - frac_y) +
  422. (src[index + stride] * (s - frac_x) +
  423. src[index + stride + 1] * frac_x) * frac_y +
  424. r) >> (shift * 2);
  425. } else {
  426. index = src_x + av_clip(src_y, 0, height) * stride;
  427. dst[y * stride + x] =
  428. ((src[index] * (s - frac_x) +
  429. src[index + 1] * frac_x) * s +
  430. r) >> (shift * 2);
  431. }
  432. } else {
  433. if ((unsigned) src_y < height) {
  434. index = av_clip(src_x, 0, width) + src_y * stride;
  435. dst[y * stride + x] =
  436. ((src[index] * (s - frac_y) +
  437. src[index + stride] * frac_y) * s +
  438. r) >> (shift * 2);
  439. } else {
  440. index = av_clip(src_x, 0, width) +
  441. av_clip(src_y, 0, height) * stride;
  442. dst[y * stride + x] = src[index];
  443. }
  444. }
  445. vx += dxx;
  446. vy += dyx;
  447. }
  448. ox += dxy;
  449. oy += dyy;
  450. }
  451. }
  452. #define QPEL_MC(r, OPNAME, RND, OP) \
  453. static void OPNAME ## mpeg4_qpel8_h_lowpass(uint8_t *dst, uint8_t *src, \
  454. int dstStride, int srcStride, \
  455. int h) \
  456. { \
  457. const uint8_t *cm = ff_crop_tab + MAX_NEG_CROP; \
  458. int i; \
  459. \
  460. for (i = 0; i < h; i++) { \
  461. OP(dst[0], (src[0] + src[1]) * 20 - (src[0] + src[2]) * 6 + (src[1] + src[3]) * 3 - (src[2] + src[4])); \
  462. OP(dst[1], (src[1] + src[2]) * 20 - (src[0] + src[3]) * 6 + (src[0] + src[4]) * 3 - (src[1] + src[5])); \
  463. OP(dst[2], (src[2] + src[3]) * 20 - (src[1] + src[4]) * 6 + (src[0] + src[5]) * 3 - (src[0] + src[6])); \
  464. OP(dst[3], (src[3] + src[4]) * 20 - (src[2] + src[5]) * 6 + (src[1] + src[6]) * 3 - (src[0] + src[7])); \
  465. OP(dst[4], (src[4] + src[5]) * 20 - (src[3] + src[6]) * 6 + (src[2] + src[7]) * 3 - (src[1] + src[8])); \
  466. OP(dst[5], (src[5] + src[6]) * 20 - (src[4] + src[7]) * 6 + (src[3] + src[8]) * 3 - (src[2] + src[8])); \
  467. OP(dst[6], (src[6] + src[7]) * 20 - (src[5] + src[8]) * 6 + (src[4] + src[8]) * 3 - (src[3] + src[7])); \
  468. OP(dst[7], (src[7] + src[8]) * 20 - (src[6] + src[8]) * 6 + (src[5] + src[7]) * 3 - (src[4] + src[6])); \
  469. dst += dstStride; \
  470. src += srcStride; \
  471. } \
  472. } \
  473. \
  474. static void OPNAME ## mpeg4_qpel8_v_lowpass(uint8_t *dst, uint8_t *src, \
  475. int dstStride, int srcStride) \
  476. { \
  477. const uint8_t *cm = ff_crop_tab + MAX_NEG_CROP; \
  478. const int w = 8; \
  479. int i; \
  480. \
  481. for (i = 0; i < w; i++) { \
  482. const int src0 = src[0 * srcStride]; \
  483. const int src1 = src[1 * srcStride]; \
  484. const int src2 = src[2 * srcStride]; \
  485. const int src3 = src[3 * srcStride]; \
  486. const int src4 = src[4 * srcStride]; \
  487. const int src5 = src[5 * srcStride]; \
  488. const int src6 = src[6 * srcStride]; \
  489. const int src7 = src[7 * srcStride]; \
  490. const int src8 = src[8 * srcStride]; \
  491. OP(dst[0 * dstStride], (src0 + src1) * 20 - (src0 + src2) * 6 + (src1 + src3) * 3 - (src2 + src4)); \
  492. OP(dst[1 * dstStride], (src1 + src2) * 20 - (src0 + src3) * 6 + (src0 + src4) * 3 - (src1 + src5)); \
  493. OP(dst[2 * dstStride], (src2 + src3) * 20 - (src1 + src4) * 6 + (src0 + src5) * 3 - (src0 + src6)); \
  494. OP(dst[3 * dstStride], (src3 + src4) * 20 - (src2 + src5) * 6 + (src1 + src6) * 3 - (src0 + src7)); \
  495. OP(dst[4 * dstStride], (src4 + src5) * 20 - (src3 + src6) * 6 + (src2 + src7) * 3 - (src1 + src8)); \
  496. OP(dst[5 * dstStride], (src5 + src6) * 20 - (src4 + src7) * 6 + (src3 + src8) * 3 - (src2 + src8)); \
  497. OP(dst[6 * dstStride], (src6 + src7) * 20 - (src5 + src8) * 6 + (src4 + src8) * 3 - (src3 + src7)); \
  498. OP(dst[7 * dstStride], (src7 + src8) * 20 - (src6 + src8) * 6 + (src5 + src7) * 3 - (src4 + src6)); \
  499. dst++; \
  500. src++; \
  501. } \
  502. } \
  503. \
  504. static void OPNAME ## mpeg4_qpel16_h_lowpass(uint8_t *dst, uint8_t *src, \
  505. int dstStride, int srcStride, \
  506. int h) \
  507. { \
  508. const uint8_t *cm = ff_crop_tab + MAX_NEG_CROP; \
  509. int i; \
  510. \
  511. for (i = 0; i < h; i++) { \
  512. OP(dst[0], (src[0] + src[1]) * 20 - (src[0] + src[2]) * 6 + (src[1] + src[3]) * 3 - (src[2] + src[4])); \
  513. OP(dst[1], (src[1] + src[2]) * 20 - (src[0] + src[3]) * 6 + (src[0] + src[4]) * 3 - (src[1] + src[5])); \
  514. OP(dst[2], (src[2] + src[3]) * 20 - (src[1] + src[4]) * 6 + (src[0] + src[5]) * 3 - (src[0] + src[6])); \
  515. OP(dst[3], (src[3] + src[4]) * 20 - (src[2] + src[5]) * 6 + (src[1] + src[6]) * 3 - (src[0] + src[7])); \
  516. OP(dst[4], (src[4] + src[5]) * 20 - (src[3] + src[6]) * 6 + (src[2] + src[7]) * 3 - (src[1] + src[8])); \
  517. OP(dst[5], (src[5] + src[6]) * 20 - (src[4] + src[7]) * 6 + (src[3] + src[8]) * 3 - (src[2] + src[9])); \
  518. OP(dst[6], (src[6] + src[7]) * 20 - (src[5] + src[8]) * 6 + (src[4] + src[9]) * 3 - (src[3] + src[10])); \
  519. OP(dst[7], (src[7] + src[8]) * 20 - (src[6] + src[9]) * 6 + (src[5] + src[10]) * 3 - (src[4] + src[11])); \
  520. OP(dst[8], (src[8] + src[9]) * 20 - (src[7] + src[10]) * 6 + (src[6] + src[11]) * 3 - (src[5] + src[12])); \
  521. OP(dst[9], (src[9] + src[10]) * 20 - (src[8] + src[11]) * 6 + (src[7] + src[12]) * 3 - (src[6] + src[13])); \
  522. OP(dst[10], (src[10] + src[11]) * 20 - (src[9] + src[12]) * 6 + (src[8] + src[13]) * 3 - (src[7] + src[14])); \
  523. OP(dst[11], (src[11] + src[12]) * 20 - (src[10] + src[13]) * 6 + (src[9] + src[14]) * 3 - (src[8] + src[15])); \
  524. OP(dst[12], (src[12] + src[13]) * 20 - (src[11] + src[14]) * 6 + (src[10] + src[15]) * 3 - (src[9] + src[16])); \
  525. OP(dst[13], (src[13] + src[14]) * 20 - (src[12] + src[15]) * 6 + (src[11] + src[16]) * 3 - (src[10] + src[16])); \
  526. OP(dst[14], (src[14] + src[15]) * 20 - (src[13] + src[16]) * 6 + (src[12] + src[16]) * 3 - (src[11] + src[15])); \
  527. OP(dst[15], (src[15] + src[16]) * 20 - (src[14] + src[16]) * 6 + (src[13] + src[15]) * 3 - (src[12] + src[14])); \
  528. dst += dstStride; \
  529. src += srcStride; \
  530. } \
  531. } \
  532. \
  533. static void OPNAME ## mpeg4_qpel16_v_lowpass(uint8_t *dst, uint8_t *src, \
  534. int dstStride, int srcStride) \
  535. { \
  536. const uint8_t *cm = ff_crop_tab + MAX_NEG_CROP; \
  537. const int w = 16; \
  538. int i; \
  539. \
  540. for (i = 0; i < w; i++) { \
  541. const int src0 = src[0 * srcStride]; \
  542. const int src1 = src[1 * srcStride]; \
  543. const int src2 = src[2 * srcStride]; \
  544. const int src3 = src[3 * srcStride]; \
  545. const int src4 = src[4 * srcStride]; \
  546. const int src5 = src[5 * srcStride]; \
  547. const int src6 = src[6 * srcStride]; \
  548. const int src7 = src[7 * srcStride]; \
  549. const int src8 = src[8 * srcStride]; \
  550. const int src9 = src[9 * srcStride]; \
  551. const int src10 = src[10 * srcStride]; \
  552. const int src11 = src[11 * srcStride]; \
  553. const int src12 = src[12 * srcStride]; \
  554. const int src13 = src[13 * srcStride]; \
  555. const int src14 = src[14 * srcStride]; \
  556. const int src15 = src[15 * srcStride]; \
  557. const int src16 = src[16 * srcStride]; \
  558. OP(dst[0 * dstStride], (src0 + src1) * 20 - (src0 + src2) * 6 + (src1 + src3) * 3 - (src2 + src4)); \
  559. OP(dst[1 * dstStride], (src1 + src2) * 20 - (src0 + src3) * 6 + (src0 + src4) * 3 - (src1 + src5)); \
  560. OP(dst[2 * dstStride], (src2 + src3) * 20 - (src1 + src4) * 6 + (src0 + src5) * 3 - (src0 + src6)); \
  561. OP(dst[3 * dstStride], (src3 + src4) * 20 - (src2 + src5) * 6 + (src1 + src6) * 3 - (src0 + src7)); \
  562. OP(dst[4 * dstStride], (src4 + src5) * 20 - (src3 + src6) * 6 + (src2 + src7) * 3 - (src1 + src8)); \
  563. OP(dst[5 * dstStride], (src5 + src6) * 20 - (src4 + src7) * 6 + (src3 + src8) * 3 - (src2 + src9)); \
  564. OP(dst[6 * dstStride], (src6 + src7) * 20 - (src5 + src8) * 6 + (src4 + src9) * 3 - (src3 + src10)); \
  565. OP(dst[7 * dstStride], (src7 + src8) * 20 - (src6 + src9) * 6 + (src5 + src10) * 3 - (src4 + src11)); \
  566. OP(dst[8 * dstStride], (src8 + src9) * 20 - (src7 + src10) * 6 + (src6 + src11) * 3 - (src5 + src12)); \
  567. OP(dst[9 * dstStride], (src9 + src10) * 20 - (src8 + src11) * 6 + (src7 + src12) * 3 - (src6 + src13)); \
  568. OP(dst[10 * dstStride], (src10 + src11) * 20 - (src9 + src12) * 6 + (src8 + src13) * 3 - (src7 + src14)); \
  569. OP(dst[11 * dstStride], (src11 + src12) * 20 - (src10 + src13) * 6 + (src9 + src14) * 3 - (src8 + src15)); \
  570. OP(dst[12 * dstStride], (src12 + src13) * 20 - (src11 + src14) * 6 + (src10 + src15) * 3 - (src9 + src16)); \
  571. OP(dst[13 * dstStride], (src13 + src14) * 20 - (src12 + src15) * 6 + (src11 + src16) * 3 - (src10 + src16)); \
  572. OP(dst[14 * dstStride], (src14 + src15) * 20 - (src13 + src16) * 6 + (src12 + src16) * 3 - (src11 + src15)); \
  573. OP(dst[15 * dstStride], (src15 + src16) * 20 - (src14 + src16) * 6 + (src13 + src15) * 3 - (src12 + src14)); \
  574. dst++; \
  575. src++; \
  576. } \
  577. } \
  578. \
  579. static void OPNAME ## qpel8_mc10_c(uint8_t *dst, uint8_t *src, \
  580. ptrdiff_t stride) \
  581. { \
  582. uint8_t half[64]; \
  583. \
  584. put ## RND ## mpeg4_qpel8_h_lowpass(half, src, 8, stride, 8); \
  585. OPNAME ## pixels8_l2_8(dst, src, half, stride, stride, 8, 8); \
  586. } \
  587. \
  588. static void OPNAME ## qpel8_mc20_c(uint8_t *dst, uint8_t *src, \
  589. ptrdiff_t stride) \
  590. { \
  591. OPNAME ## mpeg4_qpel8_h_lowpass(dst, src, stride, stride, 8); \
  592. } \
  593. \
  594. static void OPNAME ## qpel8_mc30_c(uint8_t *dst, uint8_t *src, \
  595. ptrdiff_t stride) \
  596. { \
  597. uint8_t half[64]; \
  598. \
  599. put ## RND ## mpeg4_qpel8_h_lowpass(half, src, 8, stride, 8); \
  600. OPNAME ## pixels8_l2_8(dst, src + 1, half, stride, stride, 8, 8); \
  601. } \
  602. \
  603. static void OPNAME ## qpel8_mc01_c(uint8_t *dst, uint8_t *src, \
  604. ptrdiff_t stride) \
  605. { \
  606. uint8_t full[16 * 9]; \
  607. uint8_t half[64]; \
  608. \
  609. copy_block9(full, src, 16, stride, 9); \
  610. put ## RND ## mpeg4_qpel8_v_lowpass(half, full, 8, 16); \
  611. OPNAME ## pixels8_l2_8(dst, full, half, stride, 16, 8, 8); \
  612. } \
  613. \
  614. static void OPNAME ## qpel8_mc02_c(uint8_t *dst, uint8_t *src, \
  615. ptrdiff_t stride) \
  616. { \
  617. uint8_t full[16 * 9]; \
  618. \
  619. copy_block9(full, src, 16, stride, 9); \
  620. OPNAME ## mpeg4_qpel8_v_lowpass(dst, full, stride, 16); \
  621. } \
  622. \
  623. static void OPNAME ## qpel8_mc03_c(uint8_t *dst, uint8_t *src, \
  624. ptrdiff_t stride) \
  625. { \
  626. uint8_t full[16 * 9]; \
  627. uint8_t half[64]; \
  628. \
  629. copy_block9(full, src, 16, stride, 9); \
  630. put ## RND ## mpeg4_qpel8_v_lowpass(half, full, 8, 16); \
  631. OPNAME ## pixels8_l2_8(dst, full + 16, half, stride, 16, 8, 8); \
  632. } \
  633. \
  634. void ff_ ## OPNAME ## qpel8_mc11_old_c(uint8_t *dst, uint8_t *src, \
  635. ptrdiff_t stride) \
  636. { \
  637. uint8_t full[16 * 9]; \
  638. uint8_t halfH[72]; \
  639. uint8_t halfV[64]; \
  640. uint8_t halfHV[64]; \
  641. \
  642. copy_block9(full, src, 16, stride, 9); \
  643. put ## RND ## mpeg4_qpel8_h_lowpass(halfH, full, 8, 16, 9); \
  644. put ## RND ## mpeg4_qpel8_v_lowpass(halfV, full, 8, 16); \
  645. put ## RND ## mpeg4_qpel8_v_lowpass(halfHV, halfH, 8, 8); \
  646. OPNAME ## pixels8_l4_8(dst, full, halfH, halfV, halfHV, \
  647. stride, 16, 8, 8, 8, 8); \
  648. } \
  649. \
  650. static void OPNAME ## qpel8_mc11_c(uint8_t *dst, uint8_t *src, \
  651. ptrdiff_t stride) \
  652. { \
  653. uint8_t full[16 * 9]; \
  654. uint8_t halfH[72]; \
  655. uint8_t halfHV[64]; \
  656. \
  657. copy_block9(full, src, 16, stride, 9); \
  658. put ## RND ## mpeg4_qpel8_h_lowpass(halfH, full, 8, 16, 9); \
  659. put ## RND ## pixels8_l2_8(halfH, halfH, full, 8, 8, 16, 9); \
  660. put ## RND ## mpeg4_qpel8_v_lowpass(halfHV, halfH, 8, 8); \
  661. OPNAME ## pixels8_l2_8(dst, halfH, halfHV, stride, 8, 8, 8); \
  662. } \
  663. \
  664. void ff_ ## OPNAME ## qpel8_mc31_old_c(uint8_t *dst, uint8_t *src, \
  665. ptrdiff_t stride) \
  666. { \
  667. uint8_t full[16 * 9]; \
  668. uint8_t halfH[72]; \
  669. uint8_t halfV[64]; \
  670. uint8_t halfHV[64]; \
  671. \
  672. copy_block9(full, src, 16, stride, 9); \
  673. put ## RND ## mpeg4_qpel8_h_lowpass(halfH, full, 8, 16, 9); \
  674. put ## RND ## mpeg4_qpel8_v_lowpass(halfV, full + 1, 8, 16); \
  675. put ## RND ## mpeg4_qpel8_v_lowpass(halfHV, halfH, 8, 8); \
  676. OPNAME ## pixels8_l4_8(dst, full + 1, halfH, halfV, halfHV, \
  677. stride, 16, 8, 8, 8, 8); \
  678. } \
  679. \
  680. static void OPNAME ## qpel8_mc31_c(uint8_t *dst, uint8_t *src, \
  681. ptrdiff_t stride) \
  682. { \
  683. uint8_t full[16 * 9]; \
  684. uint8_t halfH[72]; \
  685. uint8_t halfHV[64]; \
  686. \
  687. copy_block9(full, src, 16, stride, 9); \
  688. put ## RND ## mpeg4_qpel8_h_lowpass(halfH, full, 8, 16, 9); \
  689. put ## RND ## pixels8_l2_8(halfH, halfH, full + 1, 8, 8, 16, 9); \
  690. put ## RND ## mpeg4_qpel8_v_lowpass(halfHV, halfH, 8, 8); \
  691. OPNAME ## pixels8_l2_8(dst, halfH, halfHV, stride, 8, 8, 8); \
  692. } \
  693. \
  694. void ff_ ## OPNAME ## qpel8_mc13_old_c(uint8_t *dst, uint8_t *src, \
  695. ptrdiff_t stride) \
  696. { \
  697. uint8_t full[16 * 9]; \
  698. uint8_t halfH[72]; \
  699. uint8_t halfV[64]; \
  700. uint8_t halfHV[64]; \
  701. \
  702. copy_block9(full, src, 16, stride, 9); \
  703. put ## RND ## mpeg4_qpel8_h_lowpass(halfH, full, 8, 16, 9); \
  704. put ## RND ## mpeg4_qpel8_v_lowpass(halfV, full, 8, 16); \
  705. put ## RND ## mpeg4_qpel8_v_lowpass(halfHV, halfH, 8, 8); \
  706. OPNAME ## pixels8_l4_8(dst, full + 16, halfH + 8, halfV, halfHV, \
  707. stride, 16, 8, 8, 8, 8); \
  708. } \
  709. \
  710. static void OPNAME ## qpel8_mc13_c(uint8_t *dst, uint8_t *src, \
  711. ptrdiff_t stride) \
  712. { \
  713. uint8_t full[16 * 9]; \
  714. uint8_t halfH[72]; \
  715. uint8_t halfHV[64]; \
  716. \
  717. copy_block9(full, src, 16, stride, 9); \
  718. put ## RND ## mpeg4_qpel8_h_lowpass(halfH, full, 8, 16, 9); \
  719. put ## RND ## pixels8_l2_8(halfH, halfH, full, 8, 8, 16, 9); \
  720. put ## RND ## mpeg4_qpel8_v_lowpass(halfHV, halfH, 8, 8); \
  721. OPNAME ## pixels8_l2_8(dst, halfH + 8, halfHV, stride, 8, 8, 8); \
  722. } \
  723. \
  724. void ff_ ## OPNAME ## qpel8_mc33_old_c(uint8_t *dst, uint8_t *src, \
  725. ptrdiff_t stride) \
  726. { \
  727. uint8_t full[16 * 9]; \
  728. uint8_t halfH[72]; \
  729. uint8_t halfV[64]; \
  730. uint8_t halfHV[64]; \
  731. \
  732. copy_block9(full, src, 16, stride, 9); \
  733. put ## RND ## mpeg4_qpel8_h_lowpass(halfH, full, 8, 16, 9); \
  734. put ## RND ## mpeg4_qpel8_v_lowpass(halfV, full + 1, 8, 16); \
  735. put ## RND ## mpeg4_qpel8_v_lowpass(halfHV, halfH, 8, 8); \
  736. OPNAME ## pixels8_l4_8(dst, full + 17, halfH + 8, halfV, halfHV, \
  737. stride, 16, 8, 8, 8, 8); \
  738. } \
  739. \
  740. static void OPNAME ## qpel8_mc33_c(uint8_t *dst, uint8_t *src, \
  741. ptrdiff_t stride) \
  742. { \
  743. uint8_t full[16 * 9]; \
  744. uint8_t halfH[72]; \
  745. uint8_t halfHV[64]; \
  746. \
  747. copy_block9(full, src, 16, stride, 9); \
  748. put ## RND ## mpeg4_qpel8_h_lowpass(halfH, full, 8, 16, 9); \
  749. put ## RND ## pixels8_l2_8(halfH, halfH, full + 1, 8, 8, 16, 9); \
  750. put ## RND ## mpeg4_qpel8_v_lowpass(halfHV, halfH, 8, 8); \
  751. OPNAME ## pixels8_l2_8(dst, halfH + 8, halfHV, stride, 8, 8, 8); \
  752. } \
  753. \
  754. static void OPNAME ## qpel8_mc21_c(uint8_t *dst, uint8_t *src, \
  755. ptrdiff_t stride) \
  756. { \
  757. uint8_t halfH[72]; \
  758. uint8_t halfHV[64]; \
  759. \
  760. put ## RND ## mpeg4_qpel8_h_lowpass(halfH, src, 8, stride, 9); \
  761. put ## RND ## mpeg4_qpel8_v_lowpass(halfHV, halfH, 8, 8); \
  762. OPNAME ## pixels8_l2_8(dst, halfH, halfHV, stride, 8, 8, 8); \
  763. } \
  764. \
  765. static void OPNAME ## qpel8_mc23_c(uint8_t *dst, uint8_t *src, \
  766. ptrdiff_t stride) \
  767. { \
  768. uint8_t halfH[72]; \
  769. uint8_t halfHV[64]; \
  770. \
  771. put ## RND ## mpeg4_qpel8_h_lowpass(halfH, src, 8, stride, 9); \
  772. put ## RND ## mpeg4_qpel8_v_lowpass(halfHV, halfH, 8, 8); \
  773. OPNAME ## pixels8_l2_8(dst, halfH + 8, halfHV, stride, 8, 8, 8); \
  774. } \
  775. \
  776. void ff_ ## OPNAME ## qpel8_mc12_old_c(uint8_t *dst, uint8_t *src, \
  777. ptrdiff_t stride) \
  778. { \
  779. uint8_t full[16 * 9]; \
  780. uint8_t halfH[72]; \
  781. uint8_t halfV[64]; \
  782. uint8_t halfHV[64]; \
  783. \
  784. copy_block9(full, src, 16, stride, 9); \
  785. put ## RND ## mpeg4_qpel8_h_lowpass(halfH, full, 8, 16, 9); \
  786. put ## RND ## mpeg4_qpel8_v_lowpass(halfV, full, 8, 16); \
  787. put ## RND ## mpeg4_qpel8_v_lowpass(halfHV, halfH, 8, 8); \
  788. OPNAME ## pixels8_l2_8(dst, halfV, halfHV, stride, 8, 8, 8); \
  789. } \
  790. \
  791. static void OPNAME ## qpel8_mc12_c(uint8_t *dst, uint8_t *src, \
  792. ptrdiff_t stride) \
  793. { \
  794. uint8_t full[16 * 9]; \
  795. uint8_t halfH[72]; \
  796. \
  797. copy_block9(full, src, 16, stride, 9); \
  798. put ## RND ## mpeg4_qpel8_h_lowpass(halfH, full, 8, 16, 9); \
  799. put ## RND ## pixels8_l2_8(halfH, halfH, full, 8, 8, 16, 9); \
  800. OPNAME ## mpeg4_qpel8_v_lowpass(dst, halfH, stride, 8); \
  801. } \
  802. \
  803. void ff_ ## OPNAME ## qpel8_mc32_old_c(uint8_t *dst, uint8_t *src, \
  804. ptrdiff_t stride) \
  805. { \
  806. uint8_t full[16 * 9]; \
  807. uint8_t halfH[72]; \
  808. uint8_t halfV[64]; \
  809. uint8_t halfHV[64]; \
  810. \
  811. copy_block9(full, src, 16, stride, 9); \
  812. put ## RND ## mpeg4_qpel8_h_lowpass(halfH, full, 8, 16, 9); \
  813. put ## RND ## mpeg4_qpel8_v_lowpass(halfV, full + 1, 8, 16); \
  814. put ## RND ## mpeg4_qpel8_v_lowpass(halfHV, halfH, 8, 8); \
  815. OPNAME ## pixels8_l2_8(dst, halfV, halfHV, stride, 8, 8, 8); \
  816. } \
  817. \
  818. static void OPNAME ## qpel8_mc32_c(uint8_t *dst, uint8_t *src, \
  819. ptrdiff_t stride) \
  820. { \
  821. uint8_t full[16 * 9]; \
  822. uint8_t halfH[72]; \
  823. \
  824. copy_block9(full, src, 16, stride, 9); \
  825. put ## RND ## mpeg4_qpel8_h_lowpass(halfH, full, 8, 16, 9); \
  826. put ## RND ## pixels8_l2_8(halfH, halfH, full + 1, 8, 8, 16, 9); \
  827. OPNAME ## mpeg4_qpel8_v_lowpass(dst, halfH, stride, 8); \
  828. } \
  829. \
  830. static void OPNAME ## qpel8_mc22_c(uint8_t *dst, uint8_t *src, \
  831. ptrdiff_t stride) \
  832. { \
  833. uint8_t halfH[72]; \
  834. \
  835. put ## RND ## mpeg4_qpel8_h_lowpass(halfH, src, 8, stride, 9); \
  836. OPNAME ## mpeg4_qpel8_v_lowpass(dst, halfH, stride, 8); \
  837. } \
  838. \
  839. static void OPNAME ## qpel16_mc10_c(uint8_t *dst, uint8_t *src, \
  840. ptrdiff_t stride) \
  841. { \
  842. uint8_t half[256]; \
  843. \
  844. put ## RND ## mpeg4_qpel16_h_lowpass(half, src, 16, stride, 16); \
  845. OPNAME ## pixels16_l2_8(dst, src, half, stride, stride, 16, 16); \
  846. } \
  847. \
  848. static void OPNAME ## qpel16_mc20_c(uint8_t *dst, uint8_t *src, \
  849. ptrdiff_t stride) \
  850. { \
  851. OPNAME ## mpeg4_qpel16_h_lowpass(dst, src, stride, stride, 16); \
  852. } \
  853. \
  854. static void OPNAME ## qpel16_mc30_c(uint8_t *dst, uint8_t *src, \
  855. ptrdiff_t stride) \
  856. { \
  857. uint8_t half[256]; \
  858. \
  859. put ## RND ## mpeg4_qpel16_h_lowpass(half, src, 16, stride, 16); \
  860. OPNAME ## pixels16_l2_8(dst, src + 1, half, stride, stride, 16, 16); \
  861. } \
  862. \
  863. static void OPNAME ## qpel16_mc01_c(uint8_t *dst, uint8_t *src, \
  864. ptrdiff_t stride) \
  865. { \
  866. uint8_t full[24 * 17]; \
  867. uint8_t half[256]; \
  868. \
  869. copy_block17(full, src, 24, stride, 17); \
  870. put ## RND ## mpeg4_qpel16_v_lowpass(half, full, 16, 24); \
  871. OPNAME ## pixels16_l2_8(dst, full, half, stride, 24, 16, 16); \
  872. } \
  873. \
  874. static void OPNAME ## qpel16_mc02_c(uint8_t *dst, uint8_t *src, \
  875. ptrdiff_t stride) \
  876. { \
  877. uint8_t full[24 * 17]; \
  878. \
  879. copy_block17(full, src, 24, stride, 17); \
  880. OPNAME ## mpeg4_qpel16_v_lowpass(dst, full, stride, 24); \
  881. } \
  882. \
  883. static void OPNAME ## qpel16_mc03_c(uint8_t *dst, uint8_t *src, \
  884. ptrdiff_t stride) \
  885. { \
  886. uint8_t full[24 * 17]; \
  887. uint8_t half[256]; \
  888. \
  889. copy_block17(full, src, 24, stride, 17); \
  890. put ## RND ## mpeg4_qpel16_v_lowpass(half, full, 16, 24); \
  891. OPNAME ## pixels16_l2_8(dst, full + 24, half, stride, 24, 16, 16); \
  892. } \
  893. \
  894. void ff_ ## OPNAME ## qpel16_mc11_old_c(uint8_t *dst, uint8_t *src, \
  895. ptrdiff_t stride) \
  896. { \
  897. uint8_t full[24 * 17]; \
  898. uint8_t halfH[272]; \
  899. uint8_t halfV[256]; \
  900. uint8_t halfHV[256]; \
  901. \
  902. copy_block17(full, src, 24, stride, 17); \
  903. put ## RND ## mpeg4_qpel16_h_lowpass(halfH, full, 16, 24, 17); \
  904. put ## RND ## mpeg4_qpel16_v_lowpass(halfV, full, 16, 24); \
  905. put ## RND ## mpeg4_qpel16_v_lowpass(halfHV, halfH, 16, 16); \
  906. OPNAME ## pixels16_l4_8(dst, full, halfH, halfV, halfHV, \
  907. stride, 24, 16, 16, 16, 16); \
  908. } \
  909. \
  910. static void OPNAME ## qpel16_mc11_c(uint8_t *dst, uint8_t *src, \
  911. ptrdiff_t stride) \
  912. { \
  913. uint8_t full[24 * 17]; \
  914. uint8_t halfH[272]; \
  915. uint8_t halfHV[256]; \
  916. \
  917. copy_block17(full, src, 24, stride, 17); \
  918. put ## RND ## mpeg4_qpel16_h_lowpass(halfH, full, 16, 24, 17); \
  919. put ## RND ## pixels16_l2_8(halfH, halfH, full, 16, 16, 24, 17); \
  920. put ## RND ## mpeg4_qpel16_v_lowpass(halfHV, halfH, 16, 16); \
  921. OPNAME ## pixels16_l2_8(dst, halfH, halfHV, stride, 16, 16, 16); \
  922. } \
  923. \
  924. void ff_ ## OPNAME ## qpel16_mc31_old_c(uint8_t *dst, uint8_t *src, \
  925. ptrdiff_t stride) \
  926. { \
  927. uint8_t full[24 * 17]; \
  928. uint8_t halfH[272]; \
  929. uint8_t halfV[256]; \
  930. uint8_t halfHV[256]; \
  931. \
  932. copy_block17(full, src, 24, stride, 17); \
  933. put ## RND ## mpeg4_qpel16_h_lowpass(halfH, full, 16, 24, 17); \
  934. put ## RND ## mpeg4_qpel16_v_lowpass(halfV, full + 1, 16, 24); \
  935. put ## RND ## mpeg4_qpel16_v_lowpass(halfHV, halfH, 16, 16); \
  936. OPNAME ## pixels16_l4_8(dst, full + 1, halfH, halfV, halfHV, \
  937. stride, 24, 16, 16, 16, 16); \
  938. } \
  939. \
  940. static void OPNAME ## qpel16_mc31_c(uint8_t *dst, uint8_t *src, \
  941. ptrdiff_t stride) \
  942. { \
  943. uint8_t full[24 * 17]; \
  944. uint8_t halfH[272]; \
  945. uint8_t halfHV[256]; \
  946. \
  947. copy_block17(full, src, 24, stride, 17); \
  948. put ## RND ## mpeg4_qpel16_h_lowpass(halfH, full, 16, 24, 17); \
  949. put ## RND ## pixels16_l2_8(halfH, halfH, full + 1, 16, 16, 24, 17); \
  950. put ## RND ## mpeg4_qpel16_v_lowpass(halfHV, halfH, 16, 16); \
  951. OPNAME ## pixels16_l2_8(dst, halfH, halfHV, stride, 16, 16, 16); \
  952. } \
  953. \
  954. void ff_ ## OPNAME ## qpel16_mc13_old_c(uint8_t *dst, uint8_t *src, \
  955. ptrdiff_t stride) \
  956. { \
  957. uint8_t full[24 * 17]; \
  958. uint8_t halfH[272]; \
  959. uint8_t halfV[256]; \
  960. uint8_t halfHV[256]; \
  961. \
  962. copy_block17(full, src, 24, stride, 17); \
  963. put ## RND ## mpeg4_qpel16_h_lowpass(halfH, full, 16, 24, 17); \
  964. put ## RND ## mpeg4_qpel16_v_lowpass(halfV, full, 16, 24); \
  965. put ## RND ## mpeg4_qpel16_v_lowpass(halfHV, halfH, 16, 16); \
  966. OPNAME ## pixels16_l4_8(dst, full + 24, halfH + 16, halfV, halfHV, \
  967. stride, 24, 16, 16, 16, 16); \
  968. } \
  969. \
  970. static void OPNAME ## qpel16_mc13_c(uint8_t *dst, uint8_t *src, \
  971. ptrdiff_t stride) \
  972. { \
  973. uint8_t full[24 * 17]; \
  974. uint8_t halfH[272]; \
  975. uint8_t halfHV[256]; \
  976. \
  977. copy_block17(full, src, 24, stride, 17); \
  978. put ## RND ## mpeg4_qpel16_h_lowpass(halfH, full, 16, 24, 17); \
  979. put ## RND ## pixels16_l2_8(halfH, halfH, full, 16, 16, 24, 17); \
  980. put ## RND ## mpeg4_qpel16_v_lowpass(halfHV, halfH, 16, 16); \
  981. OPNAME ## pixels16_l2_8(dst, halfH + 16, halfHV, stride, 16, 16, 16); \
  982. } \
  983. \
  984. void ff_ ## OPNAME ## qpel16_mc33_old_c(uint8_t *dst, uint8_t *src, \
  985. ptrdiff_t stride) \
  986. { \
  987. uint8_t full[24 * 17]; \
  988. uint8_t halfH[272]; \
  989. uint8_t halfV[256]; \
  990. uint8_t halfHV[256]; \
  991. \
  992. copy_block17(full, src, 24, stride, 17); \
  993. put ## RND ## mpeg4_qpel16_h_lowpass(halfH, full, 16, 24, 17); \
  994. put ## RND ## mpeg4_qpel16_v_lowpass(halfV, full + 1, 16, 24); \
  995. put ## RND ## mpeg4_qpel16_v_lowpass(halfHV, halfH, 16, 16); \
  996. OPNAME ## pixels16_l4_8(dst, full + 25, halfH + 16, halfV, halfHV, \
  997. stride, 24, 16, 16, 16, 16); \
  998. } \
  999. \
  1000. static void OPNAME ## qpel16_mc33_c(uint8_t *dst, uint8_t *src, \
  1001. ptrdiff_t stride) \
  1002. { \
  1003. uint8_t full[24 * 17]; \
  1004. uint8_t halfH[272]; \
  1005. uint8_t halfHV[256]; \
  1006. \
  1007. copy_block17(full, src, 24, stride, 17); \
  1008. put ## RND ## mpeg4_qpel16_h_lowpass(halfH, full, 16, 24, 17); \
  1009. put ## RND ## pixels16_l2_8(halfH, halfH, full + 1, 16, 16, 24, 17); \
  1010. put ## RND ## mpeg4_qpel16_v_lowpass(halfHV, halfH, 16, 16); \
  1011. OPNAME ## pixels16_l2_8(dst, halfH + 16, halfHV, stride, 16, 16, 16); \
  1012. } \
  1013. \
  1014. static void OPNAME ## qpel16_mc21_c(uint8_t *dst, uint8_t *src, \
  1015. ptrdiff_t stride) \
  1016. { \
  1017. uint8_t halfH[272]; \
  1018. uint8_t halfHV[256]; \
  1019. \
  1020. put ## RND ## mpeg4_qpel16_h_lowpass(halfH, src, 16, stride, 17); \
  1021. put ## RND ## mpeg4_qpel16_v_lowpass(halfHV, halfH, 16, 16); \
  1022. OPNAME ## pixels16_l2_8(dst, halfH, halfHV, stride, 16, 16, 16); \
  1023. } \
  1024. \
  1025. static void OPNAME ## qpel16_mc23_c(uint8_t *dst, uint8_t *src, \
  1026. ptrdiff_t stride) \
  1027. { \
  1028. uint8_t halfH[272]; \
  1029. uint8_t halfHV[256]; \
  1030. \
  1031. put ## RND ## mpeg4_qpel16_h_lowpass(halfH, src, 16, stride, 17); \
  1032. put ## RND ## mpeg4_qpel16_v_lowpass(halfHV, halfH, 16, 16); \
  1033. OPNAME ## pixels16_l2_8(dst, halfH + 16, halfHV, stride, 16, 16, 16); \
  1034. } \
  1035. \
  1036. void ff_ ## OPNAME ## qpel16_mc12_old_c(uint8_t *dst, uint8_t *src, \
  1037. ptrdiff_t stride) \
  1038. { \
  1039. uint8_t full[24 * 17]; \
  1040. uint8_t halfH[272]; \
  1041. uint8_t halfV[256]; \
  1042. uint8_t halfHV[256]; \
  1043. \
  1044. copy_block17(full, src, 24, stride, 17); \
  1045. put ## RND ## mpeg4_qpel16_h_lowpass(halfH, full, 16, 24, 17); \
  1046. put ## RND ## mpeg4_qpel16_v_lowpass(halfV, full, 16, 24); \
  1047. put ## RND ## mpeg4_qpel16_v_lowpass(halfHV, halfH, 16, 16); \
  1048. OPNAME ## pixels16_l2_8(dst, halfV, halfHV, stride, 16, 16, 16); \
  1049. } \
  1050. \
  1051. static void OPNAME ## qpel16_mc12_c(uint8_t *dst, uint8_t *src, \
  1052. ptrdiff_t stride) \
  1053. { \
  1054. uint8_t full[24 * 17]; \
  1055. uint8_t halfH[272]; \
  1056. \
  1057. copy_block17(full, src, 24, stride, 17); \
  1058. put ## RND ## mpeg4_qpel16_h_lowpass(halfH, full, 16, 24, 17); \
  1059. put ## RND ## pixels16_l2_8(halfH, halfH, full, 16, 16, 24, 17); \
  1060. OPNAME ## mpeg4_qpel16_v_lowpass(dst, halfH, stride, 16); \
  1061. } \
  1062. \
  1063. void ff_ ## OPNAME ## qpel16_mc32_old_c(uint8_t *dst, uint8_t *src, \
  1064. ptrdiff_t stride) \
  1065. { \
  1066. uint8_t full[24 * 17]; \
  1067. uint8_t halfH[272]; \
  1068. uint8_t halfV[256]; \
  1069. uint8_t halfHV[256]; \
  1070. \
  1071. copy_block17(full, src, 24, stride, 17); \
  1072. put ## RND ## mpeg4_qpel16_h_lowpass(halfH, full, 16, 24, 17); \
  1073. put ## RND ## mpeg4_qpel16_v_lowpass(halfV, full + 1, 16, 24); \
  1074. put ## RND ## mpeg4_qpel16_v_lowpass(halfHV, halfH, 16, 16); \
  1075. OPNAME ## pixels16_l2_8(dst, halfV, halfHV, stride, 16, 16, 16); \
  1076. } \
  1077. \
  1078. static void OPNAME ## qpel16_mc32_c(uint8_t *dst, uint8_t *src, \
  1079. ptrdiff_t stride) \
  1080. { \
  1081. uint8_t full[24 * 17]; \
  1082. uint8_t halfH[272]; \
  1083. \
  1084. copy_block17(full, src, 24, stride, 17); \
  1085. put ## RND ## mpeg4_qpel16_h_lowpass(halfH, full, 16, 24, 17); \
  1086. put ## RND ## pixels16_l2_8(halfH, halfH, full + 1, 16, 16, 24, 17); \
  1087. OPNAME ## mpeg4_qpel16_v_lowpass(dst, halfH, stride, 16); \
  1088. } \
  1089. \
  1090. static void OPNAME ## qpel16_mc22_c(uint8_t *dst, uint8_t *src, \
  1091. ptrdiff_t stride) \
  1092. { \
  1093. uint8_t halfH[272]; \
  1094. \
  1095. put ## RND ## mpeg4_qpel16_h_lowpass(halfH, src, 16, stride, 17); \
  1096. OPNAME ## mpeg4_qpel16_v_lowpass(dst, halfH, stride, 16); \
  1097. }
  1098. #define op_avg(a, b) a = (((a) + cm[((b) + 16) >> 5] + 1) >> 1)
  1099. #define op_avg_no_rnd(a, b) a = (((a) + cm[((b) + 15) >> 5]) >> 1)
  1100. #define op_put(a, b) a = cm[((b) + 16) >> 5]
  1101. #define op_put_no_rnd(a, b) a = cm[((b) + 15) >> 5]
  1102. QPEL_MC(0, put_, _, op_put)
  1103. QPEL_MC(1, put_no_rnd_, _no_rnd_, op_put_no_rnd)
  1104. QPEL_MC(0, avg_, _, op_avg)
  1105. #undef op_avg
  1106. #undef op_put
  1107. #undef op_put_no_rnd
  1108. void ff_put_pixels8x8_c(uint8_t *dst, uint8_t *src, ptrdiff_t stride)
  1109. {
  1110. put_pixels8_8_c(dst, src, stride, 8);
  1111. }
  1112. void ff_avg_pixels8x8_c(uint8_t *dst, uint8_t *src, ptrdiff_t stride)
  1113. {
  1114. avg_pixels8_8_c(dst, src, stride, 8);
  1115. }
  1116. void ff_put_pixels16x16_c(uint8_t *dst, uint8_t *src, ptrdiff_t stride)
  1117. {
  1118. put_pixels16_8_c(dst, src, stride, 16);
  1119. }
  1120. void ff_avg_pixels16x16_c(uint8_t *dst, uint8_t *src, ptrdiff_t stride)
  1121. {
  1122. avg_pixels16_8_c(dst, src, stride, 16);
  1123. }
  1124. #define put_qpel8_mc00_c ff_put_pixels8x8_c
  1125. #define avg_qpel8_mc00_c ff_avg_pixels8x8_c
  1126. #define put_qpel16_mc00_c ff_put_pixels16x16_c
  1127. #define avg_qpel16_mc00_c ff_avg_pixels16x16_c
  1128. #define put_no_rnd_qpel8_mc00_c ff_put_pixels8x8_c
  1129. #define put_no_rnd_qpel16_mc00_c ff_put_pixels16x16_c
  1130. static void wmv2_mspel8_h_lowpass(uint8_t *dst, uint8_t *src,
  1131. int dstStride, int srcStride, int h)
  1132. {
  1133. const uint8_t *cm = ff_crop_tab + MAX_NEG_CROP;
  1134. int i;
  1135. for (i = 0; i < h; i++) {
  1136. dst[0] = cm[(9 * (src[0] + src[1]) - (src[-1] + src[2]) + 8) >> 4];
  1137. dst[1] = cm[(9 * (src[1] + src[2]) - (src[0] + src[3]) + 8) >> 4];
  1138. dst[2] = cm[(9 * (src[2] + src[3]) - (src[1] + src[4]) + 8) >> 4];
  1139. dst[3] = cm[(9 * (src[3] + src[4]) - (src[2] + src[5]) + 8) >> 4];
  1140. dst[4] = cm[(9 * (src[4] + src[5]) - (src[3] + src[6]) + 8) >> 4];
  1141. dst[5] = cm[(9 * (src[5] + src[6]) - (src[4] + src[7]) + 8) >> 4];
  1142. dst[6] = cm[(9 * (src[6] + src[7]) - (src[5] + src[8]) + 8) >> 4];
  1143. dst[7] = cm[(9 * (src[7] + src[8]) - (src[6] + src[9]) + 8) >> 4];
  1144. dst += dstStride;
  1145. src += srcStride;
  1146. }
  1147. }
  1148. static void wmv2_mspel8_v_lowpass(uint8_t *dst, uint8_t *src,
  1149. int dstStride, int srcStride, int w)
  1150. {
  1151. const uint8_t *cm = ff_crop_tab + MAX_NEG_CROP;
  1152. int i;
  1153. for (i = 0; i < w; i++) {
  1154. const int src_1 = src[-srcStride];
  1155. const int src0 = src[0];
  1156. const int src1 = src[srcStride];
  1157. const int src2 = src[2 * srcStride];
  1158. const int src3 = src[3 * srcStride];
  1159. const int src4 = src[4 * srcStride];
  1160. const int src5 = src[5 * srcStride];
  1161. const int src6 = src[6 * srcStride];
  1162. const int src7 = src[7 * srcStride];
  1163. const int src8 = src[8 * srcStride];
  1164. const int src9 = src[9 * srcStride];
  1165. dst[0 * dstStride] = cm[(9 * (src0 + src1) - (src_1 + src2) + 8) >> 4];
  1166. dst[1 * dstStride] = cm[(9 * (src1 + src2) - (src0 + src3) + 8) >> 4];
  1167. dst[2 * dstStride] = cm[(9 * (src2 + src3) - (src1 + src4) + 8) >> 4];
  1168. dst[3 * dstStride] = cm[(9 * (src3 + src4) - (src2 + src5) + 8) >> 4];
  1169. dst[4 * dstStride] = cm[(9 * (src4 + src5) - (src3 + src6) + 8) >> 4];
  1170. dst[5 * dstStride] = cm[(9 * (src5 + src6) - (src4 + src7) + 8) >> 4];
  1171. dst[6 * dstStride] = cm[(9 * (src6 + src7) - (src5 + src8) + 8) >> 4];
  1172. dst[7 * dstStride] = cm[(9 * (src7 + src8) - (src6 + src9) + 8) >> 4];
  1173. src++;
  1174. dst++;
  1175. }
  1176. }
  1177. static void put_mspel8_mc10_c(uint8_t *dst, uint8_t *src, ptrdiff_t stride)
  1178. {
  1179. uint8_t half[64];
  1180. wmv2_mspel8_h_lowpass(half, src, 8, stride, 8);
  1181. put_pixels8_l2_8(dst, src, half, stride, stride, 8, 8);
  1182. }
  1183. static void put_mspel8_mc20_c(uint8_t *dst, uint8_t *src, ptrdiff_t stride)
  1184. {
  1185. wmv2_mspel8_h_lowpass(dst, src, stride, stride, 8);
  1186. }
  1187. static void put_mspel8_mc30_c(uint8_t *dst, uint8_t *src, ptrdiff_t stride)
  1188. {
  1189. uint8_t half[64];
  1190. wmv2_mspel8_h_lowpass(half, src, 8, stride, 8);
  1191. put_pixels8_l2_8(dst, src + 1, half, stride, stride, 8, 8);
  1192. }
  1193. static void put_mspel8_mc02_c(uint8_t *dst, uint8_t *src, ptrdiff_t stride)
  1194. {
  1195. wmv2_mspel8_v_lowpass(dst, src, stride, stride, 8);
  1196. }
  1197. static void put_mspel8_mc12_c(uint8_t *dst, uint8_t *src, ptrdiff_t stride)
  1198. {
  1199. uint8_t halfH[88];
  1200. uint8_t halfV[64];
  1201. uint8_t halfHV[64];
  1202. wmv2_mspel8_h_lowpass(halfH, src - stride, 8, stride, 11);
  1203. wmv2_mspel8_v_lowpass(halfV, src, 8, stride, 8);
  1204. wmv2_mspel8_v_lowpass(halfHV, halfH + 8, 8, 8, 8);
  1205. put_pixels8_l2_8(dst, halfV, halfHV, stride, 8, 8, 8);
  1206. }
  1207. static void put_mspel8_mc32_c(uint8_t *dst, uint8_t *src, ptrdiff_t stride)
  1208. {
  1209. uint8_t halfH[88];
  1210. uint8_t halfV[64];
  1211. uint8_t halfHV[64];
  1212. wmv2_mspel8_h_lowpass(halfH, src - stride, 8, stride, 11);
  1213. wmv2_mspel8_v_lowpass(halfV, src + 1, 8, stride, 8);
  1214. wmv2_mspel8_v_lowpass(halfHV, halfH + 8, 8, 8, 8);
  1215. put_pixels8_l2_8(dst, halfV, halfHV, stride, 8, 8, 8);
  1216. }
  1217. static void put_mspel8_mc22_c(uint8_t *dst, uint8_t *src, ptrdiff_t stride)
  1218. {
  1219. uint8_t halfH[88];
  1220. wmv2_mspel8_h_lowpass(halfH, src - stride, 8, stride, 11);
  1221. wmv2_mspel8_v_lowpass(dst, halfH + 8, stride, 8, 8);
  1222. }
  1223. static inline int pix_abs16_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1224. int line_size, int h)
  1225. {
  1226. int s = 0, i;
  1227. for (i = 0; i < h; i++) {
  1228. s += abs(pix1[0] - pix2[0]);
  1229. s += abs(pix1[1] - pix2[1]);
  1230. s += abs(pix1[2] - pix2[2]);
  1231. s += abs(pix1[3] - pix2[3]);
  1232. s += abs(pix1[4] - pix2[4]);
  1233. s += abs(pix1[5] - pix2[5]);
  1234. s += abs(pix1[6] - pix2[6]);
  1235. s += abs(pix1[7] - pix2[7]);
  1236. s += abs(pix1[8] - pix2[8]);
  1237. s += abs(pix1[9] - pix2[9]);
  1238. s += abs(pix1[10] - pix2[10]);
  1239. s += abs(pix1[11] - pix2[11]);
  1240. s += abs(pix1[12] - pix2[12]);
  1241. s += abs(pix1[13] - pix2[13]);
  1242. s += abs(pix1[14] - pix2[14]);
  1243. s += abs(pix1[15] - pix2[15]);
  1244. pix1 += line_size;
  1245. pix2 += line_size;
  1246. }
  1247. return s;
  1248. }
  1249. static int pix_abs16_x2_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1250. int line_size, int h)
  1251. {
  1252. int s = 0, i;
  1253. for (i = 0; i < h; i++) {
  1254. s += abs(pix1[0] - avg2(pix2[0], pix2[1]));
  1255. s += abs(pix1[1] - avg2(pix2[1], pix2[2]));
  1256. s += abs(pix1[2] - avg2(pix2[2], pix2[3]));
  1257. s += abs(pix1[3] - avg2(pix2[3], pix2[4]));
  1258. s += abs(pix1[4] - avg2(pix2[4], pix2[5]));
  1259. s += abs(pix1[5] - avg2(pix2[5], pix2[6]));
  1260. s += abs(pix1[6] - avg2(pix2[6], pix2[7]));
  1261. s += abs(pix1[7] - avg2(pix2[7], pix2[8]));
  1262. s += abs(pix1[8] - avg2(pix2[8], pix2[9]));
  1263. s += abs(pix1[9] - avg2(pix2[9], pix2[10]));
  1264. s += abs(pix1[10] - avg2(pix2[10], pix2[11]));
  1265. s += abs(pix1[11] - avg2(pix2[11], pix2[12]));
  1266. s += abs(pix1[12] - avg2(pix2[12], pix2[13]));
  1267. s += abs(pix1[13] - avg2(pix2[13], pix2[14]));
  1268. s += abs(pix1[14] - avg2(pix2[14], pix2[15]));
  1269. s += abs(pix1[15] - avg2(pix2[15], pix2[16]));
  1270. pix1 += line_size;
  1271. pix2 += line_size;
  1272. }
  1273. return s;
  1274. }
  1275. static int pix_abs16_y2_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1276. int line_size, int h)
  1277. {
  1278. int s = 0, i;
  1279. uint8_t *pix3 = pix2 + line_size;
  1280. for (i = 0; i < h; i++) {
  1281. s += abs(pix1[0] - avg2(pix2[0], pix3[0]));
  1282. s += abs(pix1[1] - avg2(pix2[1], pix3[1]));
  1283. s += abs(pix1[2] - avg2(pix2[2], pix3[2]));
  1284. s += abs(pix1[3] - avg2(pix2[3], pix3[3]));
  1285. s += abs(pix1[4] - avg2(pix2[4], pix3[4]));
  1286. s += abs(pix1[5] - avg2(pix2[5], pix3[5]));
  1287. s += abs(pix1[6] - avg2(pix2[6], pix3[6]));
  1288. s += abs(pix1[7] - avg2(pix2[7], pix3[7]));
  1289. s += abs(pix1[8] - avg2(pix2[8], pix3[8]));
  1290. s += abs(pix1[9] - avg2(pix2[9], pix3[9]));
  1291. s += abs(pix1[10] - avg2(pix2[10], pix3[10]));
  1292. s += abs(pix1[11] - avg2(pix2[11], pix3[11]));
  1293. s += abs(pix1[12] - avg2(pix2[12], pix3[12]));
  1294. s += abs(pix1[13] - avg2(pix2[13], pix3[13]));
  1295. s += abs(pix1[14] - avg2(pix2[14], pix3[14]));
  1296. s += abs(pix1[15] - avg2(pix2[15], pix3[15]));
  1297. pix1 += line_size;
  1298. pix2 += line_size;
  1299. pix3 += line_size;
  1300. }
  1301. return s;
  1302. }
  1303. static int pix_abs16_xy2_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1304. int line_size, int h)
  1305. {
  1306. int s = 0, i;
  1307. uint8_t *pix3 = pix2 + line_size;
  1308. for (i = 0; i < h; i++) {
  1309. s += abs(pix1[0] - avg4(pix2[0], pix2[1], pix3[0], pix3[1]));
  1310. s += abs(pix1[1] - avg4(pix2[1], pix2[2], pix3[1], pix3[2]));
  1311. s += abs(pix1[2] - avg4(pix2[2], pix2[3], pix3[2], pix3[3]));
  1312. s += abs(pix1[3] - avg4(pix2[3], pix2[4], pix3[3], pix3[4]));
  1313. s += abs(pix1[4] - avg4(pix2[4], pix2[5], pix3[4], pix3[5]));
  1314. s += abs(pix1[5] - avg4(pix2[5], pix2[6], pix3[5], pix3[6]));
  1315. s += abs(pix1[6] - avg4(pix2[6], pix2[7], pix3[6], pix3[7]));
  1316. s += abs(pix1[7] - avg4(pix2[7], pix2[8], pix3[7], pix3[8]));
  1317. s += abs(pix1[8] - avg4(pix2[8], pix2[9], pix3[8], pix3[9]));
  1318. s += abs(pix1[9] - avg4(pix2[9], pix2[10], pix3[9], pix3[10]));
  1319. s += abs(pix1[10] - avg4(pix2[10], pix2[11], pix3[10], pix3[11]));
  1320. s += abs(pix1[11] - avg4(pix2[11], pix2[12], pix3[11], pix3[12]));
  1321. s += abs(pix1[12] - avg4(pix2[12], pix2[13], pix3[12], pix3[13]));
  1322. s += abs(pix1[13] - avg4(pix2[13], pix2[14], pix3[13], pix3[14]));
  1323. s += abs(pix1[14] - avg4(pix2[14], pix2[15], pix3[14], pix3[15]));
  1324. s += abs(pix1[15] - avg4(pix2[15], pix2[16], pix3[15], pix3[16]));
  1325. pix1 += line_size;
  1326. pix2 += line_size;
  1327. pix3 += line_size;
  1328. }
  1329. return s;
  1330. }
  1331. static inline int pix_abs8_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1332. int line_size, int h)
  1333. {
  1334. int s = 0, i;
  1335. for (i = 0; i < h; i++) {
  1336. s += abs(pix1[0] - pix2[0]);
  1337. s += abs(pix1[1] - pix2[1]);
  1338. s += abs(pix1[2] - pix2[2]);
  1339. s += abs(pix1[3] - pix2[3]);
  1340. s += abs(pix1[4] - pix2[4]);
  1341. s += abs(pix1[5] - pix2[5]);
  1342. s += abs(pix1[6] - pix2[6]);
  1343. s += abs(pix1[7] - pix2[7]);
  1344. pix1 += line_size;
  1345. pix2 += line_size;
  1346. }
  1347. return s;
  1348. }
  1349. static int pix_abs8_x2_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1350. int line_size, int h)
  1351. {
  1352. int s = 0, i;
  1353. for (i = 0; i < h; i++) {
  1354. s += abs(pix1[0] - avg2(pix2[0], pix2[1]));
  1355. s += abs(pix1[1] - avg2(pix2[1], pix2[2]));
  1356. s += abs(pix1[2] - avg2(pix2[2], pix2[3]));
  1357. s += abs(pix1[3] - avg2(pix2[3], pix2[4]));
  1358. s += abs(pix1[4] - avg2(pix2[4], pix2[5]));
  1359. s += abs(pix1[5] - avg2(pix2[5], pix2[6]));
  1360. s += abs(pix1[6] - avg2(pix2[6], pix2[7]));
  1361. s += abs(pix1[7] - avg2(pix2[7], pix2[8]));
  1362. pix1 += line_size;
  1363. pix2 += line_size;
  1364. }
  1365. return s;
  1366. }
  1367. static int pix_abs8_y2_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1368. int line_size, int h)
  1369. {
  1370. int s = 0, i;
  1371. uint8_t *pix3 = pix2 + line_size;
  1372. for (i = 0; i < h; i++) {
  1373. s += abs(pix1[0] - avg2(pix2[0], pix3[0]));
  1374. s += abs(pix1[1] - avg2(pix2[1], pix3[1]));
  1375. s += abs(pix1[2] - avg2(pix2[2], pix3[2]));
  1376. s += abs(pix1[3] - avg2(pix2[3], pix3[3]));
  1377. s += abs(pix1[4] - avg2(pix2[4], pix3[4]));
  1378. s += abs(pix1[5] - avg2(pix2[5], pix3[5]));
  1379. s += abs(pix1[6] - avg2(pix2[6], pix3[6]));
  1380. s += abs(pix1[7] - avg2(pix2[7], pix3[7]));
  1381. pix1 += line_size;
  1382. pix2 += line_size;
  1383. pix3 += line_size;
  1384. }
  1385. return s;
  1386. }
  1387. static int pix_abs8_xy2_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1388. int line_size, int h)
  1389. {
  1390. int s = 0, i;
  1391. uint8_t *pix3 = pix2 + line_size;
  1392. for (i = 0; i < h; i++) {
  1393. s += abs(pix1[0] - avg4(pix2[0], pix2[1], pix3[0], pix3[1]));
  1394. s += abs(pix1[1] - avg4(pix2[1], pix2[2], pix3[1], pix3[2]));
  1395. s += abs(pix1[2] - avg4(pix2[2], pix2[3], pix3[2], pix3[3]));
  1396. s += abs(pix1[3] - avg4(pix2[3], pix2[4], pix3[3], pix3[4]));
  1397. s += abs(pix1[4] - avg4(pix2[4], pix2[5], pix3[4], pix3[5]));
  1398. s += abs(pix1[5] - avg4(pix2[5], pix2[6], pix3[5], pix3[6]));
  1399. s += abs(pix1[6] - avg4(pix2[6], pix2[7], pix3[6], pix3[7]));
  1400. s += abs(pix1[7] - avg4(pix2[7], pix2[8], pix3[7], pix3[8]));
  1401. pix1 += line_size;
  1402. pix2 += line_size;
  1403. pix3 += line_size;
  1404. }
  1405. return s;
  1406. }
  1407. static int nsse16_c(MpegEncContext *c, uint8_t *s1, uint8_t *s2, int stride, int h)
  1408. {
  1409. int score1 = 0, score2 = 0, x, y;
  1410. for (y = 0; y < h; y++) {
  1411. for (x = 0; x < 16; x++)
  1412. score1 += (s1[x] - s2[x]) * (s1[x] - s2[x]);
  1413. if (y + 1 < h) {
  1414. for (x = 0; x < 15; x++)
  1415. score2 += FFABS(s1[x] - s1[x + stride] -
  1416. s1[x + 1] + s1[x + stride + 1]) -
  1417. FFABS(s2[x] - s2[x + stride] -
  1418. s2[x + 1] + s2[x + stride + 1]);
  1419. }
  1420. s1 += stride;
  1421. s2 += stride;
  1422. }
  1423. if (c)
  1424. return score1 + FFABS(score2) * c->avctx->nsse_weight;
  1425. else
  1426. return score1 + FFABS(score2) * 8;
  1427. }
  1428. static int nsse8_c(MpegEncContext *c, uint8_t *s1, uint8_t *s2, int stride, int h)
  1429. {
  1430. int score1 = 0, score2 = 0, x, y;
  1431. for (y = 0; y < h; y++) {
  1432. for (x = 0; x < 8; x++)
  1433. score1 += (s1[x] - s2[x]) * (s1[x] - s2[x]);
  1434. if (y + 1 < h) {
  1435. for (x = 0; x < 7; x++)
  1436. score2 += FFABS(s1[x] - s1[x + stride] -
  1437. s1[x + 1] + s1[x + stride + 1]) -
  1438. FFABS(s2[x] - s2[x + stride] -
  1439. s2[x + 1] + s2[x + stride + 1]);
  1440. }
  1441. s1 += stride;
  1442. s2 += stride;
  1443. }
  1444. if (c)
  1445. return score1 + FFABS(score2) * c->avctx->nsse_weight;
  1446. else
  1447. return score1 + FFABS(score2) * 8;
  1448. }
  1449. static int try_8x8basis_c(int16_t rem[64], int16_t weight[64],
  1450. int16_t basis[64], int scale)
  1451. {
  1452. int i;
  1453. unsigned int sum = 0;
  1454. for (i = 0; i < 8 * 8; i++) {
  1455. int b = rem[i] + ((basis[i] * scale +
  1456. (1 << (BASIS_SHIFT - RECON_SHIFT - 1))) >>
  1457. (BASIS_SHIFT - RECON_SHIFT));
  1458. int w = weight[i];
  1459. b >>= RECON_SHIFT;
  1460. assert(-512 < b && b < 512);
  1461. sum += (w * b) * (w * b) >> 4;
  1462. }
  1463. return sum >> 2;
  1464. }
  1465. static void add_8x8basis_c(int16_t rem[64], int16_t basis[64], int scale)
  1466. {
  1467. int i;
  1468. for (i = 0; i < 8 * 8; i++)
  1469. rem[i] += (basis[i] * scale +
  1470. (1 << (BASIS_SHIFT - RECON_SHIFT - 1))) >>
  1471. (BASIS_SHIFT - RECON_SHIFT);
  1472. }
  1473. static int zero_cmp(MpegEncContext *s, uint8_t *a, uint8_t *b,
  1474. int stride, int h)
  1475. {
  1476. return 0;
  1477. }
  1478. void ff_set_cmp(DSPContext *c, me_cmp_func *cmp, int type)
  1479. {
  1480. int i;
  1481. memset(cmp, 0, sizeof(void *) * 6);
  1482. for (i = 0; i < 6; i++) {
  1483. switch (type & 0xFF) {
  1484. case FF_CMP_SAD:
  1485. cmp[i] = c->sad[i];
  1486. break;
  1487. case FF_CMP_SATD:
  1488. cmp[i] = c->hadamard8_diff[i];
  1489. break;
  1490. case FF_CMP_SSE:
  1491. cmp[i] = c->sse[i];
  1492. break;
  1493. case FF_CMP_DCT:
  1494. cmp[i] = c->dct_sad[i];
  1495. break;
  1496. case FF_CMP_DCT264:
  1497. cmp[i] = c->dct264_sad[i];
  1498. break;
  1499. case FF_CMP_DCTMAX:
  1500. cmp[i] = c->dct_max[i];
  1501. break;
  1502. case FF_CMP_PSNR:
  1503. cmp[i] = c->quant_psnr[i];
  1504. break;
  1505. case FF_CMP_BIT:
  1506. cmp[i] = c->bit[i];
  1507. break;
  1508. case FF_CMP_RD:
  1509. cmp[i] = c->rd[i];
  1510. break;
  1511. case FF_CMP_VSAD:
  1512. cmp[i] = c->vsad[i];
  1513. break;
  1514. case FF_CMP_VSSE:
  1515. cmp[i] = c->vsse[i];
  1516. break;
  1517. case FF_CMP_ZERO:
  1518. cmp[i] = zero_cmp;
  1519. break;
  1520. case FF_CMP_NSSE:
  1521. cmp[i] = c->nsse[i];
  1522. break;
  1523. default:
  1524. av_log(NULL, AV_LOG_ERROR,
  1525. "internal error in cmp function selection\n");
  1526. }
  1527. }
  1528. }
  1529. #define BUTTERFLY2(o1, o2, i1, i2) \
  1530. o1 = (i1) + (i2); \
  1531. o2 = (i1) - (i2);
  1532. #define BUTTERFLY1(x, y) \
  1533. { \
  1534. int a, b; \
  1535. a = x; \
  1536. b = y; \
  1537. x = a + b; \
  1538. y = a - b; \
  1539. }
  1540. #define BUTTERFLYA(x, y) (FFABS((x) + (y)) + FFABS((x) - (y)))
  1541. static int hadamard8_diff8x8_c(MpegEncContext *s, uint8_t *dst,
  1542. uint8_t *src, int stride, int h)
  1543. {
  1544. int i, temp[64], sum = 0;
  1545. assert(h == 8);
  1546. for (i = 0; i < 8; i++) {
  1547. // FIXME: try pointer walks
  1548. BUTTERFLY2(temp[8 * i + 0], temp[8 * i + 1],
  1549. src[stride * i + 0] - dst[stride * i + 0],
  1550. src[stride * i + 1] - dst[stride * i + 1]);
  1551. BUTTERFLY2(temp[8 * i + 2], temp[8 * i + 3],
  1552. src[stride * i + 2] - dst[stride * i + 2],
  1553. src[stride * i + 3] - dst[stride * i + 3]);
  1554. BUTTERFLY2(temp[8 * i + 4], temp[8 * i + 5],
  1555. src[stride * i + 4] - dst[stride * i + 4],
  1556. src[stride * i + 5] - dst[stride * i + 5]);
  1557. BUTTERFLY2(temp[8 * i + 6], temp[8 * i + 7],
  1558. src[stride * i + 6] - dst[stride * i + 6],
  1559. src[stride * i + 7] - dst[stride * i + 7]);
  1560. BUTTERFLY1(temp[8 * i + 0], temp[8 * i + 2]);
  1561. BUTTERFLY1(temp[8 * i + 1], temp[8 * i + 3]);
  1562. BUTTERFLY1(temp[8 * i + 4], temp[8 * i + 6]);
  1563. BUTTERFLY1(temp[8 * i + 5], temp[8 * i + 7]);
  1564. BUTTERFLY1(temp[8 * i + 0], temp[8 * i + 4]);
  1565. BUTTERFLY1(temp[8 * i + 1], temp[8 * i + 5]);
  1566. BUTTERFLY1(temp[8 * i + 2], temp[8 * i + 6]);
  1567. BUTTERFLY1(temp[8 * i + 3], temp[8 * i + 7]);
  1568. }
  1569. for (i = 0; i < 8; i++) {
  1570. BUTTERFLY1(temp[8 * 0 + i], temp[8 * 1 + i]);
  1571. BUTTERFLY1(temp[8 * 2 + i], temp[8 * 3 + i]);
  1572. BUTTERFLY1(temp[8 * 4 + i], temp[8 * 5 + i]);
  1573. BUTTERFLY1(temp[8 * 6 + i], temp[8 * 7 + i]);
  1574. BUTTERFLY1(temp[8 * 0 + i], temp[8 * 2 + i]);
  1575. BUTTERFLY1(temp[8 * 1 + i], temp[8 * 3 + i]);
  1576. BUTTERFLY1(temp[8 * 4 + i], temp[8 * 6 + i]);
  1577. BUTTERFLY1(temp[8 * 5 + i], temp[8 * 7 + i]);
  1578. sum += BUTTERFLYA(temp[8 * 0 + i], temp[8 * 4 + i]) +
  1579. BUTTERFLYA(temp[8 * 1 + i], temp[8 * 5 + i]) +
  1580. BUTTERFLYA(temp[8 * 2 + i], temp[8 * 6 + i]) +
  1581. BUTTERFLYA(temp[8 * 3 + i], temp[8 * 7 + i]);
  1582. }
  1583. return sum;
  1584. }
  1585. static int hadamard8_intra8x8_c(MpegEncContext *s, uint8_t *src,
  1586. uint8_t *dummy, int stride, int h)
  1587. {
  1588. int i, temp[64], sum = 0;
  1589. assert(h == 8);
  1590. for (i = 0; i < 8; i++) {
  1591. // FIXME: try pointer walks
  1592. BUTTERFLY2(temp[8 * i + 0], temp[8 * i + 1],
  1593. src[stride * i + 0], src[stride * i + 1]);
  1594. BUTTERFLY2(temp[8 * i + 2], temp[8 * i + 3],
  1595. src[stride * i + 2], src[stride * i + 3]);
  1596. BUTTERFLY2(temp[8 * i + 4], temp[8 * i + 5],
  1597. src[stride * i + 4], src[stride * i + 5]);
  1598. BUTTERFLY2(temp[8 * i + 6], temp[8 * i + 7],
  1599. src[stride * i + 6], src[stride * i + 7]);
  1600. BUTTERFLY1(temp[8 * i + 0], temp[8 * i + 2]);
  1601. BUTTERFLY1(temp[8 * i + 1], temp[8 * i + 3]);
  1602. BUTTERFLY1(temp[8 * i + 4], temp[8 * i + 6]);
  1603. BUTTERFLY1(temp[8 * i + 5], temp[8 * i + 7]);
  1604. BUTTERFLY1(temp[8 * i + 0], temp[8 * i + 4]);
  1605. BUTTERFLY1(temp[8 * i + 1], temp[8 * i + 5]);
  1606. BUTTERFLY1(temp[8 * i + 2], temp[8 * i + 6]);
  1607. BUTTERFLY1(temp[8 * i + 3], temp[8 * i + 7]);
  1608. }
  1609. for (i = 0; i < 8; i++) {
  1610. BUTTERFLY1(temp[8 * 0 + i], temp[8 * 1 + i]);
  1611. BUTTERFLY1(temp[8 * 2 + i], temp[8 * 3 + i]);
  1612. BUTTERFLY1(temp[8 * 4 + i], temp[8 * 5 + i]);
  1613. BUTTERFLY1(temp[8 * 6 + i], temp[8 * 7 + i]);
  1614. BUTTERFLY1(temp[8 * 0 + i], temp[8 * 2 + i]);
  1615. BUTTERFLY1(temp[8 * 1 + i], temp[8 * 3 + i]);
  1616. BUTTERFLY1(temp[8 * 4 + i], temp[8 * 6 + i]);
  1617. BUTTERFLY1(temp[8 * 5 + i], temp[8 * 7 + i]);
  1618. sum +=
  1619. BUTTERFLYA(temp[8 * 0 + i], temp[8 * 4 + i])
  1620. + BUTTERFLYA(temp[8 * 1 + i], temp[8 * 5 + i])
  1621. + BUTTERFLYA(temp[8 * 2 + i], temp[8 * 6 + i])
  1622. + BUTTERFLYA(temp[8 * 3 + i], temp[8 * 7 + i]);
  1623. }
  1624. sum -= FFABS(temp[8 * 0] + temp[8 * 4]); // -mean
  1625. return sum;
  1626. }
  1627. static int dct_sad8x8_c(MpegEncContext *s, uint8_t *src1,
  1628. uint8_t *src2, int stride, int h)
  1629. {
  1630. LOCAL_ALIGNED_16(int16_t, temp, [64]);
  1631. assert(h == 8);
  1632. s->dsp.diff_pixels(temp, src1, src2, stride);
  1633. s->dsp.fdct(temp);
  1634. return s->dsp.sum_abs_dctelem(temp);
  1635. }
  1636. #if CONFIG_GPL
  1637. #define DCT8_1D \
  1638. { \
  1639. const int s07 = SRC(0) + SRC(7); \
  1640. const int s16 = SRC(1) + SRC(6); \
  1641. const int s25 = SRC(2) + SRC(5); \
  1642. const int s34 = SRC(3) + SRC(4); \
  1643. const int a0 = s07 + s34; \
  1644. const int a1 = s16 + s25; \
  1645. const int a2 = s07 - s34; \
  1646. const int a3 = s16 - s25; \
  1647. const int d07 = SRC(0) - SRC(7); \
  1648. const int d16 = SRC(1) - SRC(6); \
  1649. const int d25 = SRC(2) - SRC(5); \
  1650. const int d34 = SRC(3) - SRC(4); \
  1651. const int a4 = d16 + d25 + (d07 + (d07 >> 1)); \
  1652. const int a5 = d07 - d34 - (d25 + (d25 >> 1)); \
  1653. const int a6 = d07 + d34 - (d16 + (d16 >> 1)); \
  1654. const int a7 = d16 - d25 + (d34 + (d34 >> 1)); \
  1655. DST(0, a0 + a1); \
  1656. DST(1, a4 + (a7 >> 2)); \
  1657. DST(2, a2 + (a3 >> 1)); \
  1658. DST(3, a5 + (a6 >> 2)); \
  1659. DST(4, a0 - a1); \
  1660. DST(5, a6 - (a5 >> 2)); \
  1661. DST(6, (a2 >> 1) - a3); \
  1662. DST(7, (a4 >> 2) - a7); \
  1663. }
  1664. static int dct264_sad8x8_c(MpegEncContext *s, uint8_t *src1,
  1665. uint8_t *src2, int stride, int h)
  1666. {
  1667. int16_t dct[8][8];
  1668. int i, sum = 0;
  1669. s->dsp.diff_pixels(dct[0], src1, src2, stride);
  1670. #define SRC(x) dct[i][x]
  1671. #define DST(x, v) dct[i][x] = v
  1672. for (i = 0; i < 8; i++)
  1673. DCT8_1D
  1674. #undef SRC
  1675. #undef DST
  1676. #define SRC(x) dct[x][i]
  1677. #define DST(x, v) sum += FFABS(v)
  1678. for (i = 0; i < 8; i++)
  1679. DCT8_1D
  1680. #undef SRC
  1681. #undef DST
  1682. return sum;
  1683. }
  1684. #endif
  1685. static int dct_max8x8_c(MpegEncContext *s, uint8_t *src1,
  1686. uint8_t *src2, int stride, int h)
  1687. {
  1688. LOCAL_ALIGNED_16(int16_t, temp, [64]);
  1689. int sum = 0, i;
  1690. assert(h == 8);
  1691. s->dsp.diff_pixels(temp, src1, src2, stride);
  1692. s->dsp.fdct(temp);
  1693. for (i = 0; i < 64; i++)
  1694. sum = FFMAX(sum, FFABS(temp[i]));
  1695. return sum;
  1696. }
  1697. static int quant_psnr8x8_c(MpegEncContext *s, uint8_t *src1,
  1698. uint8_t *src2, int stride, int h)
  1699. {
  1700. LOCAL_ALIGNED_16(int16_t, temp, [64 * 2]);
  1701. int16_t *const bak = temp + 64;
  1702. int sum = 0, i;
  1703. assert(h == 8);
  1704. s->mb_intra = 0;
  1705. s->dsp.diff_pixels(temp, src1, src2, stride);
  1706. memcpy(bak, temp, 64 * sizeof(int16_t));
  1707. s->block_last_index[0 /* FIXME */] =
  1708. s->fast_dct_quantize(s, temp, 0 /* FIXME */, s->qscale, &i);
  1709. s->dct_unquantize_inter(s, temp, 0, s->qscale);
  1710. ff_simple_idct_8(temp); // FIXME
  1711. for (i = 0; i < 64; i++)
  1712. sum += (temp[i] - bak[i]) * (temp[i] - bak[i]);
  1713. return sum;
  1714. }
  1715. static int rd8x8_c(MpegEncContext *s, uint8_t *src1, uint8_t *src2,
  1716. int stride, int h)
  1717. {
  1718. const uint8_t *scantable = s->intra_scantable.permutated;
  1719. LOCAL_ALIGNED_16(int16_t, temp, [64]);
  1720. LOCAL_ALIGNED_16(uint8_t, lsrc1, [64]);
  1721. LOCAL_ALIGNED_16(uint8_t, lsrc2, [64]);
  1722. int i, last, run, bits, level, distortion, start_i;
  1723. const int esc_length = s->ac_esc_length;
  1724. uint8_t *length, *last_length;
  1725. assert(h == 8);
  1726. copy_block8(lsrc1, src1, 8, stride, 8);
  1727. copy_block8(lsrc2, src2, 8, stride, 8);
  1728. s->dsp.diff_pixels(temp, lsrc1, lsrc2, 8);
  1729. s->block_last_index[0 /* FIXME */] =
  1730. last =
  1731. s->fast_dct_quantize(s, temp, 0 /* FIXME */, s->qscale, &i);
  1732. bits = 0;
  1733. if (s->mb_intra) {
  1734. start_i = 1;
  1735. length = s->intra_ac_vlc_length;
  1736. last_length = s->intra_ac_vlc_last_length;
  1737. bits += s->luma_dc_vlc_length[temp[0] + 256]; // FIXME: chroma
  1738. } else {
  1739. start_i = 0;
  1740. length = s->inter_ac_vlc_length;
  1741. last_length = s->inter_ac_vlc_last_length;
  1742. }
  1743. if (last >= start_i) {
  1744. run = 0;
  1745. for (i = start_i; i < last; i++) {
  1746. int j = scantable[i];
  1747. level = temp[j];
  1748. if (level) {
  1749. level += 64;
  1750. if ((level & (~127)) == 0)
  1751. bits += length[UNI_AC_ENC_INDEX(run, level)];
  1752. else
  1753. bits += esc_length;
  1754. run = 0;
  1755. } else
  1756. run++;
  1757. }
  1758. i = scantable[last];
  1759. level = temp[i] + 64;
  1760. assert(level - 64);
  1761. if ((level & (~127)) == 0) {
  1762. bits += last_length[UNI_AC_ENC_INDEX(run, level)];
  1763. } else
  1764. bits += esc_length;
  1765. }
  1766. if (last >= 0) {
  1767. if (s->mb_intra)
  1768. s->dct_unquantize_intra(s, temp, 0, s->qscale);
  1769. else
  1770. s->dct_unquantize_inter(s, temp, 0, s->qscale);
  1771. }
  1772. s->dsp.idct_add(lsrc2, 8, temp);
  1773. distortion = s->dsp.sse[1](NULL, lsrc2, lsrc1, 8, 8);
  1774. return distortion + ((bits * s->qscale * s->qscale * 109 + 64) >> 7);
  1775. }
  1776. static int bit8x8_c(MpegEncContext *s, uint8_t *src1, uint8_t *src2,
  1777. int stride, int h)
  1778. {
  1779. const uint8_t *scantable = s->intra_scantable.permutated;
  1780. LOCAL_ALIGNED_16(int16_t, temp, [64]);
  1781. int i, last, run, bits, level, start_i;
  1782. const int esc_length = s->ac_esc_length;
  1783. uint8_t *length, *last_length;
  1784. assert(h == 8);
  1785. s->dsp.diff_pixels(temp, src1, src2, stride);
  1786. s->block_last_index[0 /* FIXME */] =
  1787. last =
  1788. s->fast_dct_quantize(s, temp, 0 /* FIXME */, s->qscale, &i);
  1789. bits = 0;
  1790. if (s->mb_intra) {
  1791. start_i = 1;
  1792. length = s->intra_ac_vlc_length;
  1793. last_length = s->intra_ac_vlc_last_length;
  1794. bits += s->luma_dc_vlc_length[temp[0] + 256]; // FIXME: chroma
  1795. } else {
  1796. start_i = 0;
  1797. length = s->inter_ac_vlc_length;
  1798. last_length = s->inter_ac_vlc_last_length;
  1799. }
  1800. if (last >= start_i) {
  1801. run = 0;
  1802. for (i = start_i; i < last; i++) {
  1803. int j = scantable[i];
  1804. level = temp[j];
  1805. if (level) {
  1806. level += 64;
  1807. if ((level & (~127)) == 0)
  1808. bits += length[UNI_AC_ENC_INDEX(run, level)];
  1809. else
  1810. bits += esc_length;
  1811. run = 0;
  1812. } else
  1813. run++;
  1814. }
  1815. i = scantable[last];
  1816. level = temp[i] + 64;
  1817. assert(level - 64);
  1818. if ((level & (~127)) == 0)
  1819. bits += last_length[UNI_AC_ENC_INDEX(run, level)];
  1820. else
  1821. bits += esc_length;
  1822. }
  1823. return bits;
  1824. }
  1825. #define VSAD_INTRA(size) \
  1826. static int vsad_intra ## size ## _c(MpegEncContext *c, \
  1827. uint8_t *s, uint8_t *dummy, \
  1828. int stride, int h) \
  1829. { \
  1830. int score = 0, x, y; \
  1831. \
  1832. for (y = 1; y < h; y++) { \
  1833. for (x = 0; x < size; x += 4) { \
  1834. score += FFABS(s[x] - s[x + stride]) + \
  1835. FFABS(s[x + 1] - s[x + stride + 1]) + \
  1836. FFABS(s[x + 2] - s[x + 2 + stride]) + \
  1837. FFABS(s[x + 3] - s[x + 3 + stride]); \
  1838. } \
  1839. s += stride; \
  1840. } \
  1841. \
  1842. return score; \
  1843. }
  1844. VSAD_INTRA(8)
  1845. VSAD_INTRA(16)
  1846. static int vsad16_c(MpegEncContext *c, uint8_t *s1, uint8_t *s2,
  1847. int stride, int h)
  1848. {
  1849. int score = 0, x, y;
  1850. for (y = 1; y < h; y++) {
  1851. for (x = 0; x < 16; x++)
  1852. score += FFABS(s1[x] - s2[x] - s1[x + stride] + s2[x + stride]);
  1853. s1 += stride;
  1854. s2 += stride;
  1855. }
  1856. return score;
  1857. }
  1858. #define SQ(a) ((a) * (a))
  1859. #define VSSE_INTRA(size) \
  1860. static int vsse_intra ## size ## _c(MpegEncContext *c, \
  1861. uint8_t *s, uint8_t *dummy, \
  1862. int stride, int h) \
  1863. { \
  1864. int score = 0, x, y; \
  1865. \
  1866. for (y = 1; y < h; y++) { \
  1867. for (x = 0; x < size; x += 4) { \
  1868. score += SQ(s[x] - s[x + stride]) + \
  1869. SQ(s[x + 1] - s[x + stride + 1]) + \
  1870. SQ(s[x + 2] - s[x + stride + 2]) + \
  1871. SQ(s[x + 3] - s[x + stride + 3]); \
  1872. } \
  1873. s += stride; \
  1874. } \
  1875. \
  1876. return score; \
  1877. }
  1878. VSSE_INTRA(8)
  1879. VSSE_INTRA(16)
  1880. static int vsse16_c(MpegEncContext *c, uint8_t *s1, uint8_t *s2,
  1881. int stride, int h)
  1882. {
  1883. int score = 0, x, y;
  1884. for (y = 1; y < h; y++) {
  1885. for (x = 0; x < 16; x++)
  1886. score += SQ(s1[x] - s2[x] - s1[x + stride] + s2[x + stride]);
  1887. s1 += stride;
  1888. s2 += stride;
  1889. }
  1890. return score;
  1891. }
  1892. #define WRAPPER8_16_SQ(name8, name16) \
  1893. static int name16(MpegEncContext *s, uint8_t *dst, uint8_t *src, \
  1894. int stride, int h) \
  1895. { \
  1896. int score = 0; \
  1897. \
  1898. score += name8(s, dst, src, stride, 8); \
  1899. score += name8(s, dst + 8, src + 8, stride, 8); \
  1900. if (h == 16) { \
  1901. dst += 8 * stride; \
  1902. src += 8 * stride; \
  1903. score += name8(s, dst, src, stride, 8); \
  1904. score += name8(s, dst + 8, src + 8, stride, 8); \
  1905. } \
  1906. return score; \
  1907. }
  1908. WRAPPER8_16_SQ(hadamard8_diff8x8_c, hadamard8_diff16_c)
  1909. WRAPPER8_16_SQ(hadamard8_intra8x8_c, hadamard8_intra16_c)
  1910. WRAPPER8_16_SQ(dct_sad8x8_c, dct_sad16_c)
  1911. #if CONFIG_GPL
  1912. WRAPPER8_16_SQ(dct264_sad8x8_c, dct264_sad16_c)
  1913. #endif
  1914. WRAPPER8_16_SQ(dct_max8x8_c, dct_max16_c)
  1915. WRAPPER8_16_SQ(quant_psnr8x8_c, quant_psnr16_c)
  1916. WRAPPER8_16_SQ(rd8x8_c, rd16_c)
  1917. WRAPPER8_16_SQ(bit8x8_c, bit16_c)
  1918. static inline uint32_t clipf_c_one(uint32_t a, uint32_t mini,
  1919. uint32_t maxi, uint32_t maxisign)
  1920. {
  1921. if (a > mini)
  1922. return mini;
  1923. else if ((a ^ (1U << 31)) > maxisign)
  1924. return maxi;
  1925. else
  1926. return a;
  1927. }
  1928. static void vector_clipf_c_opposite_sign(float *dst, const float *src,
  1929. float *min, float *max, int len)
  1930. {
  1931. int i;
  1932. uint32_t mini = *(uint32_t *) min;
  1933. uint32_t maxi = *(uint32_t *) max;
  1934. uint32_t maxisign = maxi ^ (1U << 31);
  1935. uint32_t *dsti = (uint32_t *) dst;
  1936. const uint32_t *srci = (const uint32_t *) src;
  1937. for (i = 0; i < len; i += 8) {
  1938. dsti[i + 0] = clipf_c_one(srci[i + 0], mini, maxi, maxisign);
  1939. dsti[i + 1] = clipf_c_one(srci[i + 1], mini, maxi, maxisign);
  1940. dsti[i + 2] = clipf_c_one(srci[i + 2], mini, maxi, maxisign);
  1941. dsti[i + 3] = clipf_c_one(srci[i + 3], mini, maxi, maxisign);
  1942. dsti[i + 4] = clipf_c_one(srci[i + 4], mini, maxi, maxisign);
  1943. dsti[i + 5] = clipf_c_one(srci[i + 5], mini, maxi, maxisign);
  1944. dsti[i + 6] = clipf_c_one(srci[i + 6], mini, maxi, maxisign);
  1945. dsti[i + 7] = clipf_c_one(srci[i + 7], mini, maxi, maxisign);
  1946. }
  1947. }
  1948. static void vector_clipf_c(float *dst, const float *src,
  1949. float min, float max, int len)
  1950. {
  1951. int i;
  1952. if (min < 0 && max > 0) {
  1953. vector_clipf_c_opposite_sign(dst, src, &min, &max, len);
  1954. } else {
  1955. for (i = 0; i < len; i += 8) {
  1956. dst[i] = av_clipf(src[i], min, max);
  1957. dst[i + 1] = av_clipf(src[i + 1], min, max);
  1958. dst[i + 2] = av_clipf(src[i + 2], min, max);
  1959. dst[i + 3] = av_clipf(src[i + 3], min, max);
  1960. dst[i + 4] = av_clipf(src[i + 4], min, max);
  1961. dst[i + 5] = av_clipf(src[i + 5], min, max);
  1962. dst[i + 6] = av_clipf(src[i + 6], min, max);
  1963. dst[i + 7] = av_clipf(src[i + 7], min, max);
  1964. }
  1965. }
  1966. }
  1967. static int32_t scalarproduct_int16_c(const int16_t *v1, const int16_t *v2,
  1968. int order)
  1969. {
  1970. int res = 0;
  1971. while (order--)
  1972. res += *v1++ **v2++;
  1973. return res;
  1974. }
  1975. static int32_t scalarproduct_and_madd_int16_c(int16_t *v1, const int16_t *v2,
  1976. const int16_t *v3,
  1977. int order, int mul)
  1978. {
  1979. int res = 0;
  1980. while (order--) {
  1981. res += *v1 * *v2++;
  1982. *v1++ += mul * *v3++;
  1983. }
  1984. return res;
  1985. }
  1986. static void vector_clip_int32_c(int32_t *dst, const int32_t *src, int32_t min,
  1987. int32_t max, unsigned int len)
  1988. {
  1989. do {
  1990. *dst++ = av_clip(*src++, min, max);
  1991. *dst++ = av_clip(*src++, min, max);
  1992. *dst++ = av_clip(*src++, min, max);
  1993. *dst++ = av_clip(*src++, min, max);
  1994. *dst++ = av_clip(*src++, min, max);
  1995. *dst++ = av_clip(*src++, min, max);
  1996. *dst++ = av_clip(*src++, min, max);
  1997. *dst++ = av_clip(*src++, min, max);
  1998. len -= 8;
  1999. } while (len > 0);
  2000. }
  2001. static void jref_idct_put(uint8_t *dest, int line_size, int16_t *block)
  2002. {
  2003. ff_j_rev_dct(block);
  2004. put_pixels_clamped_c(block, dest, line_size);
  2005. }
  2006. static void jref_idct_add(uint8_t *dest, int line_size, int16_t *block)
  2007. {
  2008. ff_j_rev_dct(block);
  2009. add_pixels_clamped_c(block, dest, line_size);
  2010. }
  2011. /* draw the edges of width 'w' of an image of size width, height */
  2012. // FIXME: Check that this is OK for MPEG-4 interlaced.
  2013. static void draw_edges_8_c(uint8_t *buf, int wrap, int width, int height,
  2014. int w, int h, int sides)
  2015. {
  2016. uint8_t *ptr = buf, *last_line;
  2017. int i;
  2018. /* left and right */
  2019. for (i = 0; i < height; i++) {
  2020. memset(ptr - w, ptr[0], w);
  2021. memset(ptr + width, ptr[width - 1], w);
  2022. ptr += wrap;
  2023. }
  2024. /* top and bottom + corners */
  2025. buf -= w;
  2026. last_line = buf + (height - 1) * wrap;
  2027. if (sides & EDGE_TOP)
  2028. for (i = 0; i < h; i++)
  2029. // top
  2030. memcpy(buf - (i + 1) * wrap, buf, width + w + w);
  2031. if (sides & EDGE_BOTTOM)
  2032. for (i = 0; i < h; i++)
  2033. // bottom
  2034. memcpy(last_line + (i + 1) * wrap, last_line, width + w + w);
  2035. }
  2036. static void clear_block_8_c(int16_t *block)
  2037. {
  2038. memset(block, 0, sizeof(int16_t) * 64);
  2039. }
  2040. static void clear_blocks_8_c(int16_t *blocks)
  2041. {
  2042. memset(blocks, 0, sizeof(int16_t) * 6 * 64);
  2043. }
  2044. /* init static data */
  2045. av_cold void ff_dsputil_static_init(void)
  2046. {
  2047. int i;
  2048. for (i = 0; i < 512; i++)
  2049. ff_square_tab[i] = (i - 256) * (i - 256);
  2050. }
  2051. av_cold void ff_dsputil_init(DSPContext *c, AVCodecContext *avctx)
  2052. {
  2053. const unsigned high_bit_depth = avctx->bits_per_raw_sample > 8;
  2054. #if CONFIG_ENCODERS
  2055. if (avctx->bits_per_raw_sample == 10) {
  2056. c->fdct = ff_jpeg_fdct_islow_10;
  2057. c->fdct248 = ff_fdct248_islow_10;
  2058. } else {
  2059. if (avctx->dct_algo == FF_DCT_FASTINT) {
  2060. c->fdct = ff_fdct_ifast;
  2061. c->fdct248 = ff_fdct_ifast248;
  2062. } else if (avctx->dct_algo == FF_DCT_FAAN) {
  2063. c->fdct = ff_faandct;
  2064. c->fdct248 = ff_faandct248;
  2065. } else {
  2066. c->fdct = ff_jpeg_fdct_islow_8; // slow/accurate/default
  2067. c->fdct248 = ff_fdct248_islow_8;
  2068. }
  2069. }
  2070. #endif /* CONFIG_ENCODERS */
  2071. if (avctx->bits_per_raw_sample == 10) {
  2072. c->idct_put = ff_simple_idct_put_10;
  2073. c->idct_add = ff_simple_idct_add_10;
  2074. c->idct = ff_simple_idct_10;
  2075. c->idct_permutation_type = FF_NO_IDCT_PERM;
  2076. } else {
  2077. if (avctx->idct_algo == FF_IDCT_INT) {
  2078. c->idct_put = jref_idct_put;
  2079. c->idct_add = jref_idct_add;
  2080. c->idct = ff_j_rev_dct;
  2081. c->idct_permutation_type = FF_LIBMPEG2_IDCT_PERM;
  2082. } else if (avctx->idct_algo == FF_IDCT_FAAN) {
  2083. c->idct_put = ff_faanidct_put;
  2084. c->idct_add = ff_faanidct_add;
  2085. c->idct = ff_faanidct;
  2086. c->idct_permutation_type = FF_NO_IDCT_PERM;
  2087. } else { // accurate/default
  2088. c->idct_put = ff_simple_idct_put_8;
  2089. c->idct_add = ff_simple_idct_add_8;
  2090. c->idct = ff_simple_idct_8;
  2091. c->idct_permutation_type = FF_NO_IDCT_PERM;
  2092. }
  2093. }
  2094. c->diff_pixels = diff_pixels_c;
  2095. c->put_pixels_clamped = put_pixels_clamped_c;
  2096. c->put_signed_pixels_clamped = put_signed_pixels_clamped_c;
  2097. c->add_pixels_clamped = add_pixels_clamped_c;
  2098. c->sum_abs_dctelem = sum_abs_dctelem_c;
  2099. c->gmc1 = gmc1_c;
  2100. c->gmc = ff_gmc_c;
  2101. c->pix_sum = pix_sum_c;
  2102. c->pix_norm1 = pix_norm1_c;
  2103. c->fill_block_tab[0] = fill_block16_c;
  2104. c->fill_block_tab[1] = fill_block8_c;
  2105. /* TODO [0] 16 [1] 8 */
  2106. c->pix_abs[0][0] = pix_abs16_c;
  2107. c->pix_abs[0][1] = pix_abs16_x2_c;
  2108. c->pix_abs[0][2] = pix_abs16_y2_c;
  2109. c->pix_abs[0][3] = pix_abs16_xy2_c;
  2110. c->pix_abs[1][0] = pix_abs8_c;
  2111. c->pix_abs[1][1] = pix_abs8_x2_c;
  2112. c->pix_abs[1][2] = pix_abs8_y2_c;
  2113. c->pix_abs[1][3] = pix_abs8_xy2_c;
  2114. #define dspfunc(PFX, IDX, NUM) \
  2115. c->PFX ## _pixels_tab[IDX][0] = PFX ## NUM ## _mc00_c; \
  2116. c->PFX ## _pixels_tab[IDX][1] = PFX ## NUM ## _mc10_c; \
  2117. c->PFX ## _pixels_tab[IDX][2] = PFX ## NUM ## _mc20_c; \
  2118. c->PFX ## _pixels_tab[IDX][3] = PFX ## NUM ## _mc30_c; \
  2119. c->PFX ## _pixels_tab[IDX][4] = PFX ## NUM ## _mc01_c; \
  2120. c->PFX ## _pixels_tab[IDX][5] = PFX ## NUM ## _mc11_c; \
  2121. c->PFX ## _pixels_tab[IDX][6] = PFX ## NUM ## _mc21_c; \
  2122. c->PFX ## _pixels_tab[IDX][7] = PFX ## NUM ## _mc31_c; \
  2123. c->PFX ## _pixels_tab[IDX][8] = PFX ## NUM ## _mc02_c; \
  2124. c->PFX ## _pixels_tab[IDX][9] = PFX ## NUM ## _mc12_c; \
  2125. c->PFX ## _pixels_tab[IDX][10] = PFX ## NUM ## _mc22_c; \
  2126. c->PFX ## _pixels_tab[IDX][11] = PFX ## NUM ## _mc32_c; \
  2127. c->PFX ## _pixels_tab[IDX][12] = PFX ## NUM ## _mc03_c; \
  2128. c->PFX ## _pixels_tab[IDX][13] = PFX ## NUM ## _mc13_c; \
  2129. c->PFX ## _pixels_tab[IDX][14] = PFX ## NUM ## _mc23_c; \
  2130. c->PFX ## _pixels_tab[IDX][15] = PFX ## NUM ## _mc33_c
  2131. dspfunc(put_qpel, 0, 16);
  2132. dspfunc(put_qpel, 1, 8);
  2133. dspfunc(put_no_rnd_qpel, 0, 16);
  2134. dspfunc(put_no_rnd_qpel, 1, 8);
  2135. dspfunc(avg_qpel, 0, 16);
  2136. dspfunc(avg_qpel, 1, 8);
  2137. #undef dspfunc
  2138. c->put_mspel_pixels_tab[0] = ff_put_pixels8x8_c;
  2139. c->put_mspel_pixels_tab[1] = put_mspel8_mc10_c;
  2140. c->put_mspel_pixels_tab[2] = put_mspel8_mc20_c;
  2141. c->put_mspel_pixels_tab[3] = put_mspel8_mc30_c;
  2142. c->put_mspel_pixels_tab[4] = put_mspel8_mc02_c;
  2143. c->put_mspel_pixels_tab[5] = put_mspel8_mc12_c;
  2144. c->put_mspel_pixels_tab[6] = put_mspel8_mc22_c;
  2145. c->put_mspel_pixels_tab[7] = put_mspel8_mc32_c;
  2146. #define SET_CMP_FUNC(name) \
  2147. c->name[0] = name ## 16_c; \
  2148. c->name[1] = name ## 8x8_c;
  2149. SET_CMP_FUNC(hadamard8_diff)
  2150. c->hadamard8_diff[4] = hadamard8_intra16_c;
  2151. c->hadamard8_diff[5] = hadamard8_intra8x8_c;
  2152. SET_CMP_FUNC(dct_sad)
  2153. SET_CMP_FUNC(dct_max)
  2154. #if CONFIG_GPL
  2155. SET_CMP_FUNC(dct264_sad)
  2156. #endif
  2157. c->sad[0] = pix_abs16_c;
  2158. c->sad[1] = pix_abs8_c;
  2159. c->sse[0] = sse16_c;
  2160. c->sse[1] = sse8_c;
  2161. c->sse[2] = sse4_c;
  2162. SET_CMP_FUNC(quant_psnr)
  2163. SET_CMP_FUNC(rd)
  2164. SET_CMP_FUNC(bit)
  2165. c->vsad[0] = vsad16_c;
  2166. c->vsad[4] = vsad_intra16_c;
  2167. c->vsad[5] = vsad_intra8_c;
  2168. c->vsse[0] = vsse16_c;
  2169. c->vsse[4] = vsse_intra16_c;
  2170. c->vsse[5] = vsse_intra8_c;
  2171. c->nsse[0] = nsse16_c;
  2172. c->nsse[1] = nsse8_c;
  2173. c->bswap_buf = bswap_buf;
  2174. c->bswap16_buf = bswap16_buf;
  2175. c->try_8x8basis = try_8x8basis_c;
  2176. c->add_8x8basis = add_8x8basis_c;
  2177. c->scalarproduct_and_madd_int16 = scalarproduct_and_madd_int16_c;
  2178. c->scalarproduct_int16 = scalarproduct_int16_c;
  2179. c->vector_clip_int32 = vector_clip_int32_c;
  2180. c->vector_clipf = vector_clipf_c;
  2181. c->shrink[0] = av_image_copy_plane;
  2182. c->shrink[1] = ff_shrink22;
  2183. c->shrink[2] = ff_shrink44;
  2184. c->shrink[3] = ff_shrink88;
  2185. c->add_pixels8 = add_pixels8_c;
  2186. c->draw_edges = draw_edges_8_c;
  2187. c->clear_block = clear_block_8_c;
  2188. c->clear_blocks = clear_blocks_8_c;
  2189. switch (avctx->bits_per_raw_sample) {
  2190. case 9:
  2191. case 10:
  2192. c->get_pixels = get_pixels_16_c;
  2193. break;
  2194. default:
  2195. c->get_pixels = get_pixels_8_c;
  2196. break;
  2197. }
  2198. if (ARCH_ARM)
  2199. ff_dsputil_init_arm(c, avctx, high_bit_depth);
  2200. if (ARCH_BFIN)
  2201. ff_dsputil_init_bfin(c, avctx, high_bit_depth);
  2202. if (ARCH_PPC)
  2203. ff_dsputil_init_ppc(c, avctx, high_bit_depth);
  2204. if (ARCH_X86)
  2205. ff_dsputil_init_x86(c, avctx, high_bit_depth);
  2206. ff_init_scantable_permutation(c->idct_permutation,
  2207. c->idct_permutation_type);
  2208. }