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.

2322 lines
106KB

  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. void ff_put_pixels8_l2_8(uint8_t *dst, const uint8_t *src1, const uint8_t *src2,
  1131. int dst_stride, int src_stride1, int src_stride2,
  1132. int h)
  1133. {
  1134. put_pixels8_l2_8(dst, src1, src2, dst_stride, src_stride1, src_stride2, h);
  1135. }
  1136. static inline int pix_abs16_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1137. int line_size, int h)
  1138. {
  1139. int s = 0, i;
  1140. for (i = 0; i < h; i++) {
  1141. s += abs(pix1[0] - pix2[0]);
  1142. s += abs(pix1[1] - pix2[1]);
  1143. s += abs(pix1[2] - pix2[2]);
  1144. s += abs(pix1[3] - pix2[3]);
  1145. s += abs(pix1[4] - pix2[4]);
  1146. s += abs(pix1[5] - pix2[5]);
  1147. s += abs(pix1[6] - pix2[6]);
  1148. s += abs(pix1[7] - pix2[7]);
  1149. s += abs(pix1[8] - pix2[8]);
  1150. s += abs(pix1[9] - pix2[9]);
  1151. s += abs(pix1[10] - pix2[10]);
  1152. s += abs(pix1[11] - pix2[11]);
  1153. s += abs(pix1[12] - pix2[12]);
  1154. s += abs(pix1[13] - pix2[13]);
  1155. s += abs(pix1[14] - pix2[14]);
  1156. s += abs(pix1[15] - pix2[15]);
  1157. pix1 += line_size;
  1158. pix2 += line_size;
  1159. }
  1160. return s;
  1161. }
  1162. static int pix_abs16_x2_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1163. int line_size, int h)
  1164. {
  1165. int s = 0, i;
  1166. for (i = 0; i < h; i++) {
  1167. s += abs(pix1[0] - avg2(pix2[0], pix2[1]));
  1168. s += abs(pix1[1] - avg2(pix2[1], pix2[2]));
  1169. s += abs(pix1[2] - avg2(pix2[2], pix2[3]));
  1170. s += abs(pix1[3] - avg2(pix2[3], pix2[4]));
  1171. s += abs(pix1[4] - avg2(pix2[4], pix2[5]));
  1172. s += abs(pix1[5] - avg2(pix2[5], pix2[6]));
  1173. s += abs(pix1[6] - avg2(pix2[6], pix2[7]));
  1174. s += abs(pix1[7] - avg2(pix2[7], pix2[8]));
  1175. s += abs(pix1[8] - avg2(pix2[8], pix2[9]));
  1176. s += abs(pix1[9] - avg2(pix2[9], pix2[10]));
  1177. s += abs(pix1[10] - avg2(pix2[10], pix2[11]));
  1178. s += abs(pix1[11] - avg2(pix2[11], pix2[12]));
  1179. s += abs(pix1[12] - avg2(pix2[12], pix2[13]));
  1180. s += abs(pix1[13] - avg2(pix2[13], pix2[14]));
  1181. s += abs(pix1[14] - avg2(pix2[14], pix2[15]));
  1182. s += abs(pix1[15] - avg2(pix2[15], pix2[16]));
  1183. pix1 += line_size;
  1184. pix2 += line_size;
  1185. }
  1186. return s;
  1187. }
  1188. static int pix_abs16_y2_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1189. int line_size, int h)
  1190. {
  1191. int s = 0, i;
  1192. uint8_t *pix3 = pix2 + line_size;
  1193. for (i = 0; i < h; i++) {
  1194. s += abs(pix1[0] - avg2(pix2[0], pix3[0]));
  1195. s += abs(pix1[1] - avg2(pix2[1], pix3[1]));
  1196. s += abs(pix1[2] - avg2(pix2[2], pix3[2]));
  1197. s += abs(pix1[3] - avg2(pix2[3], pix3[3]));
  1198. s += abs(pix1[4] - avg2(pix2[4], pix3[4]));
  1199. s += abs(pix1[5] - avg2(pix2[5], pix3[5]));
  1200. s += abs(pix1[6] - avg2(pix2[6], pix3[6]));
  1201. s += abs(pix1[7] - avg2(pix2[7], pix3[7]));
  1202. s += abs(pix1[8] - avg2(pix2[8], pix3[8]));
  1203. s += abs(pix1[9] - avg2(pix2[9], pix3[9]));
  1204. s += abs(pix1[10] - avg2(pix2[10], pix3[10]));
  1205. s += abs(pix1[11] - avg2(pix2[11], pix3[11]));
  1206. s += abs(pix1[12] - avg2(pix2[12], pix3[12]));
  1207. s += abs(pix1[13] - avg2(pix2[13], pix3[13]));
  1208. s += abs(pix1[14] - avg2(pix2[14], pix3[14]));
  1209. s += abs(pix1[15] - avg2(pix2[15], pix3[15]));
  1210. pix1 += line_size;
  1211. pix2 += line_size;
  1212. pix3 += line_size;
  1213. }
  1214. return s;
  1215. }
  1216. static int pix_abs16_xy2_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1217. int line_size, int h)
  1218. {
  1219. int s = 0, i;
  1220. uint8_t *pix3 = pix2 + line_size;
  1221. for (i = 0; i < h; i++) {
  1222. s += abs(pix1[0] - avg4(pix2[0], pix2[1], pix3[0], pix3[1]));
  1223. s += abs(pix1[1] - avg4(pix2[1], pix2[2], pix3[1], pix3[2]));
  1224. s += abs(pix1[2] - avg4(pix2[2], pix2[3], pix3[2], pix3[3]));
  1225. s += abs(pix1[3] - avg4(pix2[3], pix2[4], pix3[3], pix3[4]));
  1226. s += abs(pix1[4] - avg4(pix2[4], pix2[5], pix3[4], pix3[5]));
  1227. s += abs(pix1[5] - avg4(pix2[5], pix2[6], pix3[5], pix3[6]));
  1228. s += abs(pix1[6] - avg4(pix2[6], pix2[7], pix3[6], pix3[7]));
  1229. s += abs(pix1[7] - avg4(pix2[7], pix2[8], pix3[7], pix3[8]));
  1230. s += abs(pix1[8] - avg4(pix2[8], pix2[9], pix3[8], pix3[9]));
  1231. s += abs(pix1[9] - avg4(pix2[9], pix2[10], pix3[9], pix3[10]));
  1232. s += abs(pix1[10] - avg4(pix2[10], pix2[11], pix3[10], pix3[11]));
  1233. s += abs(pix1[11] - avg4(pix2[11], pix2[12], pix3[11], pix3[12]));
  1234. s += abs(pix1[12] - avg4(pix2[12], pix2[13], pix3[12], pix3[13]));
  1235. s += abs(pix1[13] - avg4(pix2[13], pix2[14], pix3[13], pix3[14]));
  1236. s += abs(pix1[14] - avg4(pix2[14], pix2[15], pix3[14], pix3[15]));
  1237. s += abs(pix1[15] - avg4(pix2[15], pix2[16], pix3[15], pix3[16]));
  1238. pix1 += line_size;
  1239. pix2 += line_size;
  1240. pix3 += line_size;
  1241. }
  1242. return s;
  1243. }
  1244. static inline int pix_abs8_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1245. int line_size, int h)
  1246. {
  1247. int s = 0, i;
  1248. for (i = 0; i < h; i++) {
  1249. s += abs(pix1[0] - pix2[0]);
  1250. s += abs(pix1[1] - pix2[1]);
  1251. s += abs(pix1[2] - pix2[2]);
  1252. s += abs(pix1[3] - pix2[3]);
  1253. s += abs(pix1[4] - pix2[4]);
  1254. s += abs(pix1[5] - pix2[5]);
  1255. s += abs(pix1[6] - pix2[6]);
  1256. s += abs(pix1[7] - pix2[7]);
  1257. pix1 += line_size;
  1258. pix2 += line_size;
  1259. }
  1260. return s;
  1261. }
  1262. static int pix_abs8_x2_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1263. int line_size, int h)
  1264. {
  1265. int s = 0, i;
  1266. for (i = 0; i < h; i++) {
  1267. s += abs(pix1[0] - avg2(pix2[0], pix2[1]));
  1268. s += abs(pix1[1] - avg2(pix2[1], pix2[2]));
  1269. s += abs(pix1[2] - avg2(pix2[2], pix2[3]));
  1270. s += abs(pix1[3] - avg2(pix2[3], pix2[4]));
  1271. s += abs(pix1[4] - avg2(pix2[4], pix2[5]));
  1272. s += abs(pix1[5] - avg2(pix2[5], pix2[6]));
  1273. s += abs(pix1[6] - avg2(pix2[6], pix2[7]));
  1274. s += abs(pix1[7] - avg2(pix2[7], pix2[8]));
  1275. pix1 += line_size;
  1276. pix2 += line_size;
  1277. }
  1278. return s;
  1279. }
  1280. static int pix_abs8_y2_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1281. int line_size, int h)
  1282. {
  1283. int s = 0, i;
  1284. uint8_t *pix3 = pix2 + line_size;
  1285. for (i = 0; i < h; i++) {
  1286. s += abs(pix1[0] - avg2(pix2[0], pix3[0]));
  1287. s += abs(pix1[1] - avg2(pix2[1], pix3[1]));
  1288. s += abs(pix1[2] - avg2(pix2[2], pix3[2]));
  1289. s += abs(pix1[3] - avg2(pix2[3], pix3[3]));
  1290. s += abs(pix1[4] - avg2(pix2[4], pix3[4]));
  1291. s += abs(pix1[5] - avg2(pix2[5], pix3[5]));
  1292. s += abs(pix1[6] - avg2(pix2[6], pix3[6]));
  1293. s += abs(pix1[7] - avg2(pix2[7], pix3[7]));
  1294. pix1 += line_size;
  1295. pix2 += line_size;
  1296. pix3 += line_size;
  1297. }
  1298. return s;
  1299. }
  1300. static int pix_abs8_xy2_c(MpegEncContext *v, uint8_t *pix1, uint8_t *pix2,
  1301. int line_size, int h)
  1302. {
  1303. int s = 0, i;
  1304. uint8_t *pix3 = pix2 + line_size;
  1305. for (i = 0; i < h; i++) {
  1306. s += abs(pix1[0] - avg4(pix2[0], pix2[1], pix3[0], pix3[1]));
  1307. s += abs(pix1[1] - avg4(pix2[1], pix2[2], pix3[1], pix3[2]));
  1308. s += abs(pix1[2] - avg4(pix2[2], pix2[3], pix3[2], pix3[3]));
  1309. s += abs(pix1[3] - avg4(pix2[3], pix2[4], pix3[3], pix3[4]));
  1310. s += abs(pix1[4] - avg4(pix2[4], pix2[5], pix3[4], pix3[5]));
  1311. s += abs(pix1[5] - avg4(pix2[5], pix2[6], pix3[5], pix3[6]));
  1312. s += abs(pix1[6] - avg4(pix2[6], pix2[7], pix3[6], pix3[7]));
  1313. s += abs(pix1[7] - avg4(pix2[7], pix2[8], pix3[7], pix3[8]));
  1314. pix1 += line_size;
  1315. pix2 += line_size;
  1316. pix3 += line_size;
  1317. }
  1318. return s;
  1319. }
  1320. static int nsse16_c(MpegEncContext *c, uint8_t *s1, uint8_t *s2, int stride, int h)
  1321. {
  1322. int score1 = 0, score2 = 0, x, y;
  1323. for (y = 0; y < h; y++) {
  1324. for (x = 0; x < 16; x++)
  1325. score1 += (s1[x] - s2[x]) * (s1[x] - s2[x]);
  1326. if (y + 1 < h) {
  1327. for (x = 0; x < 15; x++)
  1328. score2 += FFABS(s1[x] - s1[x + stride] -
  1329. s1[x + 1] + s1[x + stride + 1]) -
  1330. FFABS(s2[x] - s2[x + stride] -
  1331. s2[x + 1] + s2[x + stride + 1]);
  1332. }
  1333. s1 += stride;
  1334. s2 += stride;
  1335. }
  1336. if (c)
  1337. return score1 + FFABS(score2) * c->avctx->nsse_weight;
  1338. else
  1339. return score1 + FFABS(score2) * 8;
  1340. }
  1341. static int nsse8_c(MpegEncContext *c, uint8_t *s1, uint8_t *s2, int stride, int h)
  1342. {
  1343. int score1 = 0, score2 = 0, x, y;
  1344. for (y = 0; y < h; y++) {
  1345. for (x = 0; x < 8; x++)
  1346. score1 += (s1[x] - s2[x]) * (s1[x] - s2[x]);
  1347. if (y + 1 < h) {
  1348. for (x = 0; x < 7; x++)
  1349. score2 += FFABS(s1[x] - s1[x + stride] -
  1350. s1[x + 1] + s1[x + stride + 1]) -
  1351. FFABS(s2[x] - s2[x + stride] -
  1352. s2[x + 1] + s2[x + stride + 1]);
  1353. }
  1354. s1 += stride;
  1355. s2 += stride;
  1356. }
  1357. if (c)
  1358. return score1 + FFABS(score2) * c->avctx->nsse_weight;
  1359. else
  1360. return score1 + FFABS(score2) * 8;
  1361. }
  1362. static int try_8x8basis_c(int16_t rem[64], int16_t weight[64],
  1363. int16_t basis[64], int scale)
  1364. {
  1365. int i;
  1366. unsigned int sum = 0;
  1367. for (i = 0; i < 8 * 8; i++) {
  1368. int b = rem[i] + ((basis[i] * scale +
  1369. (1 << (BASIS_SHIFT - RECON_SHIFT - 1))) >>
  1370. (BASIS_SHIFT - RECON_SHIFT));
  1371. int w = weight[i];
  1372. b >>= RECON_SHIFT;
  1373. assert(-512 < b && b < 512);
  1374. sum += (w * b) * (w * b) >> 4;
  1375. }
  1376. return sum >> 2;
  1377. }
  1378. static void add_8x8basis_c(int16_t rem[64], int16_t basis[64], int scale)
  1379. {
  1380. int i;
  1381. for (i = 0; i < 8 * 8; i++)
  1382. rem[i] += (basis[i] * scale +
  1383. (1 << (BASIS_SHIFT - RECON_SHIFT - 1))) >>
  1384. (BASIS_SHIFT - RECON_SHIFT);
  1385. }
  1386. static int zero_cmp(MpegEncContext *s, uint8_t *a, uint8_t *b,
  1387. int stride, int h)
  1388. {
  1389. return 0;
  1390. }
  1391. void ff_set_cmp(DSPContext *c, me_cmp_func *cmp, int type)
  1392. {
  1393. int i;
  1394. memset(cmp, 0, sizeof(void *) * 6);
  1395. for (i = 0; i < 6; i++) {
  1396. switch (type & 0xFF) {
  1397. case FF_CMP_SAD:
  1398. cmp[i] = c->sad[i];
  1399. break;
  1400. case FF_CMP_SATD:
  1401. cmp[i] = c->hadamard8_diff[i];
  1402. break;
  1403. case FF_CMP_SSE:
  1404. cmp[i] = c->sse[i];
  1405. break;
  1406. case FF_CMP_DCT:
  1407. cmp[i] = c->dct_sad[i];
  1408. break;
  1409. case FF_CMP_DCT264:
  1410. cmp[i] = c->dct264_sad[i];
  1411. break;
  1412. case FF_CMP_DCTMAX:
  1413. cmp[i] = c->dct_max[i];
  1414. break;
  1415. case FF_CMP_PSNR:
  1416. cmp[i] = c->quant_psnr[i];
  1417. break;
  1418. case FF_CMP_BIT:
  1419. cmp[i] = c->bit[i];
  1420. break;
  1421. case FF_CMP_RD:
  1422. cmp[i] = c->rd[i];
  1423. break;
  1424. case FF_CMP_VSAD:
  1425. cmp[i] = c->vsad[i];
  1426. break;
  1427. case FF_CMP_VSSE:
  1428. cmp[i] = c->vsse[i];
  1429. break;
  1430. case FF_CMP_ZERO:
  1431. cmp[i] = zero_cmp;
  1432. break;
  1433. case FF_CMP_NSSE:
  1434. cmp[i] = c->nsse[i];
  1435. break;
  1436. default:
  1437. av_log(NULL, AV_LOG_ERROR,
  1438. "internal error in cmp function selection\n");
  1439. }
  1440. }
  1441. }
  1442. #define BUTTERFLY2(o1, o2, i1, i2) \
  1443. o1 = (i1) + (i2); \
  1444. o2 = (i1) - (i2);
  1445. #define BUTTERFLY1(x, y) \
  1446. { \
  1447. int a, b; \
  1448. a = x; \
  1449. b = y; \
  1450. x = a + b; \
  1451. y = a - b; \
  1452. }
  1453. #define BUTTERFLYA(x, y) (FFABS((x) + (y)) + FFABS((x) - (y)))
  1454. static int hadamard8_diff8x8_c(MpegEncContext *s, uint8_t *dst,
  1455. uint8_t *src, int stride, int h)
  1456. {
  1457. int i, temp[64], sum = 0;
  1458. assert(h == 8);
  1459. for (i = 0; i < 8; i++) {
  1460. // FIXME: try pointer walks
  1461. BUTTERFLY2(temp[8 * i + 0], temp[8 * i + 1],
  1462. src[stride * i + 0] - dst[stride * i + 0],
  1463. src[stride * i + 1] - dst[stride * i + 1]);
  1464. BUTTERFLY2(temp[8 * i + 2], temp[8 * i + 3],
  1465. src[stride * i + 2] - dst[stride * i + 2],
  1466. src[stride * i + 3] - dst[stride * i + 3]);
  1467. BUTTERFLY2(temp[8 * i + 4], temp[8 * i + 5],
  1468. src[stride * i + 4] - dst[stride * i + 4],
  1469. src[stride * i + 5] - dst[stride * i + 5]);
  1470. BUTTERFLY2(temp[8 * i + 6], temp[8 * i + 7],
  1471. src[stride * i + 6] - dst[stride * i + 6],
  1472. src[stride * i + 7] - dst[stride * i + 7]);
  1473. BUTTERFLY1(temp[8 * i + 0], temp[8 * i + 2]);
  1474. BUTTERFLY1(temp[8 * i + 1], temp[8 * i + 3]);
  1475. BUTTERFLY1(temp[8 * i + 4], temp[8 * i + 6]);
  1476. BUTTERFLY1(temp[8 * i + 5], temp[8 * i + 7]);
  1477. BUTTERFLY1(temp[8 * i + 0], temp[8 * i + 4]);
  1478. BUTTERFLY1(temp[8 * i + 1], temp[8 * i + 5]);
  1479. BUTTERFLY1(temp[8 * i + 2], temp[8 * i + 6]);
  1480. BUTTERFLY1(temp[8 * i + 3], temp[8 * i + 7]);
  1481. }
  1482. for (i = 0; i < 8; i++) {
  1483. BUTTERFLY1(temp[8 * 0 + i], temp[8 * 1 + i]);
  1484. BUTTERFLY1(temp[8 * 2 + i], temp[8 * 3 + i]);
  1485. BUTTERFLY1(temp[8 * 4 + i], temp[8 * 5 + i]);
  1486. BUTTERFLY1(temp[8 * 6 + i], temp[8 * 7 + i]);
  1487. BUTTERFLY1(temp[8 * 0 + i], temp[8 * 2 + i]);
  1488. BUTTERFLY1(temp[8 * 1 + i], temp[8 * 3 + i]);
  1489. BUTTERFLY1(temp[8 * 4 + i], temp[8 * 6 + i]);
  1490. BUTTERFLY1(temp[8 * 5 + i], temp[8 * 7 + i]);
  1491. sum += BUTTERFLYA(temp[8 * 0 + i], temp[8 * 4 + i]) +
  1492. BUTTERFLYA(temp[8 * 1 + i], temp[8 * 5 + i]) +
  1493. BUTTERFLYA(temp[8 * 2 + i], temp[8 * 6 + i]) +
  1494. BUTTERFLYA(temp[8 * 3 + i], temp[8 * 7 + i]);
  1495. }
  1496. return sum;
  1497. }
  1498. static int hadamard8_intra8x8_c(MpegEncContext *s, uint8_t *src,
  1499. uint8_t *dummy, int stride, int h)
  1500. {
  1501. int i, temp[64], sum = 0;
  1502. assert(h == 8);
  1503. for (i = 0; i < 8; i++) {
  1504. // FIXME: try pointer walks
  1505. BUTTERFLY2(temp[8 * i + 0], temp[8 * i + 1],
  1506. src[stride * i + 0], src[stride * i + 1]);
  1507. BUTTERFLY2(temp[8 * i + 2], temp[8 * i + 3],
  1508. src[stride * i + 2], src[stride * i + 3]);
  1509. BUTTERFLY2(temp[8 * i + 4], temp[8 * i + 5],
  1510. src[stride * i + 4], src[stride * i + 5]);
  1511. BUTTERFLY2(temp[8 * i + 6], temp[8 * i + 7],
  1512. src[stride * i + 6], src[stride * i + 7]);
  1513. BUTTERFLY1(temp[8 * i + 0], temp[8 * i + 2]);
  1514. BUTTERFLY1(temp[8 * i + 1], temp[8 * i + 3]);
  1515. BUTTERFLY1(temp[8 * i + 4], temp[8 * i + 6]);
  1516. BUTTERFLY1(temp[8 * i + 5], temp[8 * i + 7]);
  1517. BUTTERFLY1(temp[8 * i + 0], temp[8 * i + 4]);
  1518. BUTTERFLY1(temp[8 * i + 1], temp[8 * i + 5]);
  1519. BUTTERFLY1(temp[8 * i + 2], temp[8 * i + 6]);
  1520. BUTTERFLY1(temp[8 * i + 3], temp[8 * i + 7]);
  1521. }
  1522. for (i = 0; i < 8; i++) {
  1523. BUTTERFLY1(temp[8 * 0 + i], temp[8 * 1 + i]);
  1524. BUTTERFLY1(temp[8 * 2 + i], temp[8 * 3 + i]);
  1525. BUTTERFLY1(temp[8 * 4 + i], temp[8 * 5 + i]);
  1526. BUTTERFLY1(temp[8 * 6 + i], temp[8 * 7 + i]);
  1527. BUTTERFLY1(temp[8 * 0 + i], temp[8 * 2 + i]);
  1528. BUTTERFLY1(temp[8 * 1 + i], temp[8 * 3 + i]);
  1529. BUTTERFLY1(temp[8 * 4 + i], temp[8 * 6 + i]);
  1530. BUTTERFLY1(temp[8 * 5 + i], temp[8 * 7 + i]);
  1531. sum +=
  1532. BUTTERFLYA(temp[8 * 0 + i], temp[8 * 4 + i])
  1533. + BUTTERFLYA(temp[8 * 1 + i], temp[8 * 5 + i])
  1534. + BUTTERFLYA(temp[8 * 2 + i], temp[8 * 6 + i])
  1535. + BUTTERFLYA(temp[8 * 3 + i], temp[8 * 7 + i]);
  1536. }
  1537. sum -= FFABS(temp[8 * 0] + temp[8 * 4]); // -mean
  1538. return sum;
  1539. }
  1540. static int dct_sad8x8_c(MpegEncContext *s, uint8_t *src1,
  1541. uint8_t *src2, int stride, int h)
  1542. {
  1543. LOCAL_ALIGNED_16(int16_t, temp, [64]);
  1544. assert(h == 8);
  1545. s->dsp.diff_pixels(temp, src1, src2, stride);
  1546. s->dsp.fdct(temp);
  1547. return s->dsp.sum_abs_dctelem(temp);
  1548. }
  1549. #if CONFIG_GPL
  1550. #define DCT8_1D \
  1551. { \
  1552. const int s07 = SRC(0) + SRC(7); \
  1553. const int s16 = SRC(1) + SRC(6); \
  1554. const int s25 = SRC(2) + SRC(5); \
  1555. const int s34 = SRC(3) + SRC(4); \
  1556. const int a0 = s07 + s34; \
  1557. const int a1 = s16 + s25; \
  1558. const int a2 = s07 - s34; \
  1559. const int a3 = s16 - s25; \
  1560. const int d07 = SRC(0) - SRC(7); \
  1561. const int d16 = SRC(1) - SRC(6); \
  1562. const int d25 = SRC(2) - SRC(5); \
  1563. const int d34 = SRC(3) - SRC(4); \
  1564. const int a4 = d16 + d25 + (d07 + (d07 >> 1)); \
  1565. const int a5 = d07 - d34 - (d25 + (d25 >> 1)); \
  1566. const int a6 = d07 + d34 - (d16 + (d16 >> 1)); \
  1567. const int a7 = d16 - d25 + (d34 + (d34 >> 1)); \
  1568. DST(0, a0 + a1); \
  1569. DST(1, a4 + (a7 >> 2)); \
  1570. DST(2, a2 + (a3 >> 1)); \
  1571. DST(3, a5 + (a6 >> 2)); \
  1572. DST(4, a0 - a1); \
  1573. DST(5, a6 - (a5 >> 2)); \
  1574. DST(6, (a2 >> 1) - a3); \
  1575. DST(7, (a4 >> 2) - a7); \
  1576. }
  1577. static int dct264_sad8x8_c(MpegEncContext *s, uint8_t *src1,
  1578. uint8_t *src2, int stride, int h)
  1579. {
  1580. int16_t dct[8][8];
  1581. int i, sum = 0;
  1582. s->dsp.diff_pixels(dct[0], src1, src2, stride);
  1583. #define SRC(x) dct[i][x]
  1584. #define DST(x, v) dct[i][x] = v
  1585. for (i = 0; i < 8; i++)
  1586. DCT8_1D
  1587. #undef SRC
  1588. #undef DST
  1589. #define SRC(x) dct[x][i]
  1590. #define DST(x, v) sum += FFABS(v)
  1591. for (i = 0; i < 8; i++)
  1592. DCT8_1D
  1593. #undef SRC
  1594. #undef DST
  1595. return sum;
  1596. }
  1597. #endif
  1598. static int dct_max8x8_c(MpegEncContext *s, uint8_t *src1,
  1599. uint8_t *src2, int stride, int h)
  1600. {
  1601. LOCAL_ALIGNED_16(int16_t, temp, [64]);
  1602. int sum = 0, i;
  1603. assert(h == 8);
  1604. s->dsp.diff_pixels(temp, src1, src2, stride);
  1605. s->dsp.fdct(temp);
  1606. for (i = 0; i < 64; i++)
  1607. sum = FFMAX(sum, FFABS(temp[i]));
  1608. return sum;
  1609. }
  1610. static int quant_psnr8x8_c(MpegEncContext *s, uint8_t *src1,
  1611. uint8_t *src2, int stride, int h)
  1612. {
  1613. LOCAL_ALIGNED_16(int16_t, temp, [64 * 2]);
  1614. int16_t *const bak = temp + 64;
  1615. int sum = 0, i;
  1616. assert(h == 8);
  1617. s->mb_intra = 0;
  1618. s->dsp.diff_pixels(temp, src1, src2, stride);
  1619. memcpy(bak, temp, 64 * sizeof(int16_t));
  1620. s->block_last_index[0 /* FIXME */] =
  1621. s->fast_dct_quantize(s, temp, 0 /* FIXME */, s->qscale, &i);
  1622. s->dct_unquantize_inter(s, temp, 0, s->qscale);
  1623. ff_simple_idct_8(temp); // FIXME
  1624. for (i = 0; i < 64; i++)
  1625. sum += (temp[i] - bak[i]) * (temp[i] - bak[i]);
  1626. return sum;
  1627. }
  1628. static int rd8x8_c(MpegEncContext *s, uint8_t *src1, uint8_t *src2,
  1629. int stride, int h)
  1630. {
  1631. const uint8_t *scantable = s->intra_scantable.permutated;
  1632. LOCAL_ALIGNED_16(int16_t, temp, [64]);
  1633. LOCAL_ALIGNED_16(uint8_t, lsrc1, [64]);
  1634. LOCAL_ALIGNED_16(uint8_t, lsrc2, [64]);
  1635. int i, last, run, bits, level, distortion, start_i;
  1636. const int esc_length = s->ac_esc_length;
  1637. uint8_t *length, *last_length;
  1638. assert(h == 8);
  1639. copy_block8(lsrc1, src1, 8, stride, 8);
  1640. copy_block8(lsrc2, src2, 8, stride, 8);
  1641. s->dsp.diff_pixels(temp, lsrc1, lsrc2, 8);
  1642. s->block_last_index[0 /* FIXME */] =
  1643. last =
  1644. s->fast_dct_quantize(s, temp, 0 /* FIXME */, s->qscale, &i);
  1645. bits = 0;
  1646. if (s->mb_intra) {
  1647. start_i = 1;
  1648. length = s->intra_ac_vlc_length;
  1649. last_length = s->intra_ac_vlc_last_length;
  1650. bits += s->luma_dc_vlc_length[temp[0] + 256]; // FIXME: chroma
  1651. } else {
  1652. start_i = 0;
  1653. length = s->inter_ac_vlc_length;
  1654. last_length = s->inter_ac_vlc_last_length;
  1655. }
  1656. if (last >= start_i) {
  1657. run = 0;
  1658. for (i = start_i; i < last; i++) {
  1659. int j = scantable[i];
  1660. level = temp[j];
  1661. if (level) {
  1662. level += 64;
  1663. if ((level & (~127)) == 0)
  1664. bits += length[UNI_AC_ENC_INDEX(run, level)];
  1665. else
  1666. bits += esc_length;
  1667. run = 0;
  1668. } else
  1669. run++;
  1670. }
  1671. i = scantable[last];
  1672. level = temp[i] + 64;
  1673. assert(level - 64);
  1674. if ((level & (~127)) == 0) {
  1675. bits += last_length[UNI_AC_ENC_INDEX(run, level)];
  1676. } else
  1677. bits += esc_length;
  1678. }
  1679. if (last >= 0) {
  1680. if (s->mb_intra)
  1681. s->dct_unquantize_intra(s, temp, 0, s->qscale);
  1682. else
  1683. s->dct_unquantize_inter(s, temp, 0, s->qscale);
  1684. }
  1685. s->dsp.idct_add(lsrc2, 8, temp);
  1686. distortion = s->dsp.sse[1](NULL, lsrc2, lsrc1, 8, 8);
  1687. return distortion + ((bits * s->qscale * s->qscale * 109 + 64) >> 7);
  1688. }
  1689. static int bit8x8_c(MpegEncContext *s, uint8_t *src1, uint8_t *src2,
  1690. int stride, int h)
  1691. {
  1692. const uint8_t *scantable = s->intra_scantable.permutated;
  1693. LOCAL_ALIGNED_16(int16_t, temp, [64]);
  1694. int i, last, run, bits, level, start_i;
  1695. const int esc_length = s->ac_esc_length;
  1696. uint8_t *length, *last_length;
  1697. assert(h == 8);
  1698. s->dsp.diff_pixels(temp, src1, src2, stride);
  1699. s->block_last_index[0 /* FIXME */] =
  1700. last =
  1701. s->fast_dct_quantize(s, temp, 0 /* FIXME */, s->qscale, &i);
  1702. bits = 0;
  1703. if (s->mb_intra) {
  1704. start_i = 1;
  1705. length = s->intra_ac_vlc_length;
  1706. last_length = s->intra_ac_vlc_last_length;
  1707. bits += s->luma_dc_vlc_length[temp[0] + 256]; // FIXME: chroma
  1708. } else {
  1709. start_i = 0;
  1710. length = s->inter_ac_vlc_length;
  1711. last_length = s->inter_ac_vlc_last_length;
  1712. }
  1713. if (last >= start_i) {
  1714. run = 0;
  1715. for (i = start_i; i < last; i++) {
  1716. int j = scantable[i];
  1717. level = temp[j];
  1718. if (level) {
  1719. level += 64;
  1720. if ((level & (~127)) == 0)
  1721. bits += length[UNI_AC_ENC_INDEX(run, level)];
  1722. else
  1723. bits += esc_length;
  1724. run = 0;
  1725. } else
  1726. run++;
  1727. }
  1728. i = scantable[last];
  1729. level = temp[i] + 64;
  1730. assert(level - 64);
  1731. if ((level & (~127)) == 0)
  1732. bits += last_length[UNI_AC_ENC_INDEX(run, level)];
  1733. else
  1734. bits += esc_length;
  1735. }
  1736. return bits;
  1737. }
  1738. #define VSAD_INTRA(size) \
  1739. static int vsad_intra ## size ## _c(MpegEncContext *c, \
  1740. uint8_t *s, uint8_t *dummy, \
  1741. int stride, int h) \
  1742. { \
  1743. int score = 0, x, y; \
  1744. \
  1745. for (y = 1; y < h; y++) { \
  1746. for (x = 0; x < size; x += 4) { \
  1747. score += FFABS(s[x] - s[x + stride]) + \
  1748. FFABS(s[x + 1] - s[x + stride + 1]) + \
  1749. FFABS(s[x + 2] - s[x + 2 + stride]) + \
  1750. FFABS(s[x + 3] - s[x + 3 + stride]); \
  1751. } \
  1752. s += stride; \
  1753. } \
  1754. \
  1755. return score; \
  1756. }
  1757. VSAD_INTRA(8)
  1758. VSAD_INTRA(16)
  1759. static int vsad16_c(MpegEncContext *c, uint8_t *s1, uint8_t *s2,
  1760. int stride, int h)
  1761. {
  1762. int score = 0, x, y;
  1763. for (y = 1; y < h; y++) {
  1764. for (x = 0; x < 16; x++)
  1765. score += FFABS(s1[x] - s2[x] - s1[x + stride] + s2[x + stride]);
  1766. s1 += stride;
  1767. s2 += stride;
  1768. }
  1769. return score;
  1770. }
  1771. #define SQ(a) ((a) * (a))
  1772. #define VSSE_INTRA(size) \
  1773. static int vsse_intra ## size ## _c(MpegEncContext *c, \
  1774. uint8_t *s, uint8_t *dummy, \
  1775. int stride, int h) \
  1776. { \
  1777. int score = 0, x, y; \
  1778. \
  1779. for (y = 1; y < h; y++) { \
  1780. for (x = 0; x < size; x += 4) { \
  1781. score += SQ(s[x] - s[x + stride]) + \
  1782. SQ(s[x + 1] - s[x + stride + 1]) + \
  1783. SQ(s[x + 2] - s[x + stride + 2]) + \
  1784. SQ(s[x + 3] - s[x + stride + 3]); \
  1785. } \
  1786. s += stride; \
  1787. } \
  1788. \
  1789. return score; \
  1790. }
  1791. VSSE_INTRA(8)
  1792. VSSE_INTRA(16)
  1793. static int vsse16_c(MpegEncContext *c, uint8_t *s1, uint8_t *s2,
  1794. int stride, int h)
  1795. {
  1796. int score = 0, x, y;
  1797. for (y = 1; y < h; y++) {
  1798. for (x = 0; x < 16; x++)
  1799. score += SQ(s1[x] - s2[x] - s1[x + stride] + s2[x + stride]);
  1800. s1 += stride;
  1801. s2 += stride;
  1802. }
  1803. return score;
  1804. }
  1805. #define WRAPPER8_16_SQ(name8, name16) \
  1806. static int name16(MpegEncContext *s, uint8_t *dst, uint8_t *src, \
  1807. int stride, int h) \
  1808. { \
  1809. int score = 0; \
  1810. \
  1811. score += name8(s, dst, src, stride, 8); \
  1812. score += name8(s, dst + 8, src + 8, stride, 8); \
  1813. if (h == 16) { \
  1814. dst += 8 * stride; \
  1815. src += 8 * stride; \
  1816. score += name8(s, dst, src, stride, 8); \
  1817. score += name8(s, dst + 8, src + 8, stride, 8); \
  1818. } \
  1819. return score; \
  1820. }
  1821. WRAPPER8_16_SQ(hadamard8_diff8x8_c, hadamard8_diff16_c)
  1822. WRAPPER8_16_SQ(hadamard8_intra8x8_c, hadamard8_intra16_c)
  1823. WRAPPER8_16_SQ(dct_sad8x8_c, dct_sad16_c)
  1824. #if CONFIG_GPL
  1825. WRAPPER8_16_SQ(dct264_sad8x8_c, dct264_sad16_c)
  1826. #endif
  1827. WRAPPER8_16_SQ(dct_max8x8_c, dct_max16_c)
  1828. WRAPPER8_16_SQ(quant_psnr8x8_c, quant_psnr16_c)
  1829. WRAPPER8_16_SQ(rd8x8_c, rd16_c)
  1830. WRAPPER8_16_SQ(bit8x8_c, bit16_c)
  1831. static inline uint32_t clipf_c_one(uint32_t a, uint32_t mini,
  1832. uint32_t maxi, uint32_t maxisign)
  1833. {
  1834. if (a > mini)
  1835. return mini;
  1836. else if ((a ^ (1U << 31)) > maxisign)
  1837. return maxi;
  1838. else
  1839. return a;
  1840. }
  1841. static void vector_clipf_c_opposite_sign(float *dst, const float *src,
  1842. float *min, float *max, int len)
  1843. {
  1844. int i;
  1845. uint32_t mini = *(uint32_t *) min;
  1846. uint32_t maxi = *(uint32_t *) max;
  1847. uint32_t maxisign = maxi ^ (1U << 31);
  1848. uint32_t *dsti = (uint32_t *) dst;
  1849. const uint32_t *srci = (const uint32_t *) src;
  1850. for (i = 0; i < len; i += 8) {
  1851. dsti[i + 0] = clipf_c_one(srci[i + 0], mini, maxi, maxisign);
  1852. dsti[i + 1] = clipf_c_one(srci[i + 1], mini, maxi, maxisign);
  1853. dsti[i + 2] = clipf_c_one(srci[i + 2], mini, maxi, maxisign);
  1854. dsti[i + 3] = clipf_c_one(srci[i + 3], mini, maxi, maxisign);
  1855. dsti[i + 4] = clipf_c_one(srci[i + 4], mini, maxi, maxisign);
  1856. dsti[i + 5] = clipf_c_one(srci[i + 5], mini, maxi, maxisign);
  1857. dsti[i + 6] = clipf_c_one(srci[i + 6], mini, maxi, maxisign);
  1858. dsti[i + 7] = clipf_c_one(srci[i + 7], mini, maxi, maxisign);
  1859. }
  1860. }
  1861. static void vector_clipf_c(float *dst, const float *src,
  1862. float min, float max, int len)
  1863. {
  1864. int i;
  1865. if (min < 0 && max > 0) {
  1866. vector_clipf_c_opposite_sign(dst, src, &min, &max, len);
  1867. } else {
  1868. for (i = 0; i < len; i += 8) {
  1869. dst[i] = av_clipf(src[i], min, max);
  1870. dst[i + 1] = av_clipf(src[i + 1], min, max);
  1871. dst[i + 2] = av_clipf(src[i + 2], min, max);
  1872. dst[i + 3] = av_clipf(src[i + 3], min, max);
  1873. dst[i + 4] = av_clipf(src[i + 4], min, max);
  1874. dst[i + 5] = av_clipf(src[i + 5], min, max);
  1875. dst[i + 6] = av_clipf(src[i + 6], min, max);
  1876. dst[i + 7] = av_clipf(src[i + 7], min, max);
  1877. }
  1878. }
  1879. }
  1880. static int32_t scalarproduct_int16_c(const int16_t *v1, const int16_t *v2,
  1881. int order)
  1882. {
  1883. int res = 0;
  1884. while (order--)
  1885. res += *v1++ **v2++;
  1886. return res;
  1887. }
  1888. static void vector_clip_int32_c(int32_t *dst, const int32_t *src, int32_t min,
  1889. int32_t max, unsigned int len)
  1890. {
  1891. do {
  1892. *dst++ = av_clip(*src++, min, max);
  1893. *dst++ = av_clip(*src++, min, max);
  1894. *dst++ = av_clip(*src++, min, max);
  1895. *dst++ = av_clip(*src++, min, max);
  1896. *dst++ = av_clip(*src++, min, max);
  1897. *dst++ = av_clip(*src++, min, max);
  1898. *dst++ = av_clip(*src++, min, max);
  1899. *dst++ = av_clip(*src++, min, max);
  1900. len -= 8;
  1901. } while (len > 0);
  1902. }
  1903. static void jref_idct_put(uint8_t *dest, int line_size, int16_t *block)
  1904. {
  1905. ff_j_rev_dct(block);
  1906. put_pixels_clamped_c(block, dest, line_size);
  1907. }
  1908. static void jref_idct_add(uint8_t *dest, int line_size, int16_t *block)
  1909. {
  1910. ff_j_rev_dct(block);
  1911. add_pixels_clamped_c(block, dest, line_size);
  1912. }
  1913. /* draw the edges of width 'w' of an image of size width, height */
  1914. // FIXME: Check that this is OK for MPEG-4 interlaced.
  1915. static void draw_edges_8_c(uint8_t *buf, int wrap, int width, int height,
  1916. int w, int h, int sides)
  1917. {
  1918. uint8_t *ptr = buf, *last_line;
  1919. int i;
  1920. /* left and right */
  1921. for (i = 0; i < height; i++) {
  1922. memset(ptr - w, ptr[0], w);
  1923. memset(ptr + width, ptr[width - 1], w);
  1924. ptr += wrap;
  1925. }
  1926. /* top and bottom + corners */
  1927. buf -= w;
  1928. last_line = buf + (height - 1) * wrap;
  1929. if (sides & EDGE_TOP)
  1930. for (i = 0; i < h; i++)
  1931. // top
  1932. memcpy(buf - (i + 1) * wrap, buf, width + w + w);
  1933. if (sides & EDGE_BOTTOM)
  1934. for (i = 0; i < h; i++)
  1935. // bottom
  1936. memcpy(last_line + (i + 1) * wrap, last_line, width + w + w);
  1937. }
  1938. static void clear_block_8_c(int16_t *block)
  1939. {
  1940. memset(block, 0, sizeof(int16_t) * 64);
  1941. }
  1942. static void clear_blocks_8_c(int16_t *blocks)
  1943. {
  1944. memset(blocks, 0, sizeof(int16_t) * 6 * 64);
  1945. }
  1946. /* init static data */
  1947. av_cold void ff_dsputil_static_init(void)
  1948. {
  1949. int i;
  1950. for (i = 0; i < 512; i++)
  1951. ff_square_tab[i] = (i - 256) * (i - 256);
  1952. }
  1953. av_cold void ff_dsputil_init(DSPContext *c, AVCodecContext *avctx)
  1954. {
  1955. const unsigned high_bit_depth = avctx->bits_per_raw_sample > 8;
  1956. #if CONFIG_ENCODERS
  1957. if (avctx->bits_per_raw_sample == 10) {
  1958. c->fdct = ff_jpeg_fdct_islow_10;
  1959. c->fdct248 = ff_fdct248_islow_10;
  1960. } else {
  1961. if (avctx->dct_algo == FF_DCT_FASTINT) {
  1962. c->fdct = ff_fdct_ifast;
  1963. c->fdct248 = ff_fdct_ifast248;
  1964. } else if (avctx->dct_algo == FF_DCT_FAAN) {
  1965. c->fdct = ff_faandct;
  1966. c->fdct248 = ff_faandct248;
  1967. } else {
  1968. c->fdct = ff_jpeg_fdct_islow_8; // slow/accurate/default
  1969. c->fdct248 = ff_fdct248_islow_8;
  1970. }
  1971. }
  1972. #endif /* CONFIG_ENCODERS */
  1973. if (avctx->bits_per_raw_sample == 10) {
  1974. c->idct_put = ff_simple_idct_put_10;
  1975. c->idct_add = ff_simple_idct_add_10;
  1976. c->idct = ff_simple_idct_10;
  1977. c->idct_permutation_type = FF_NO_IDCT_PERM;
  1978. } else {
  1979. if (avctx->idct_algo == FF_IDCT_INT) {
  1980. c->idct_put = jref_idct_put;
  1981. c->idct_add = jref_idct_add;
  1982. c->idct = ff_j_rev_dct;
  1983. c->idct_permutation_type = FF_LIBMPEG2_IDCT_PERM;
  1984. } else if (avctx->idct_algo == FF_IDCT_FAAN) {
  1985. c->idct_put = ff_faanidct_put;
  1986. c->idct_add = ff_faanidct_add;
  1987. c->idct = ff_faanidct;
  1988. c->idct_permutation_type = FF_NO_IDCT_PERM;
  1989. } else { // accurate/default
  1990. c->idct_put = ff_simple_idct_put_8;
  1991. c->idct_add = ff_simple_idct_add_8;
  1992. c->idct = ff_simple_idct_8;
  1993. c->idct_permutation_type = FF_NO_IDCT_PERM;
  1994. }
  1995. }
  1996. c->diff_pixels = diff_pixels_c;
  1997. c->put_pixels_clamped = put_pixels_clamped_c;
  1998. c->put_signed_pixels_clamped = put_signed_pixels_clamped_c;
  1999. c->add_pixels_clamped = add_pixels_clamped_c;
  2000. c->sum_abs_dctelem = sum_abs_dctelem_c;
  2001. c->gmc1 = gmc1_c;
  2002. c->gmc = ff_gmc_c;
  2003. c->pix_sum = pix_sum_c;
  2004. c->pix_norm1 = pix_norm1_c;
  2005. c->fill_block_tab[0] = fill_block16_c;
  2006. c->fill_block_tab[1] = fill_block8_c;
  2007. /* TODO [0] 16 [1] 8 */
  2008. c->pix_abs[0][0] = pix_abs16_c;
  2009. c->pix_abs[0][1] = pix_abs16_x2_c;
  2010. c->pix_abs[0][2] = pix_abs16_y2_c;
  2011. c->pix_abs[0][3] = pix_abs16_xy2_c;
  2012. c->pix_abs[1][0] = pix_abs8_c;
  2013. c->pix_abs[1][1] = pix_abs8_x2_c;
  2014. c->pix_abs[1][2] = pix_abs8_y2_c;
  2015. c->pix_abs[1][3] = pix_abs8_xy2_c;
  2016. #define dspfunc(PFX, IDX, NUM) \
  2017. c->PFX ## _pixels_tab[IDX][0] = PFX ## NUM ## _mc00_c; \
  2018. c->PFX ## _pixels_tab[IDX][1] = PFX ## NUM ## _mc10_c; \
  2019. c->PFX ## _pixels_tab[IDX][2] = PFX ## NUM ## _mc20_c; \
  2020. c->PFX ## _pixels_tab[IDX][3] = PFX ## NUM ## _mc30_c; \
  2021. c->PFX ## _pixels_tab[IDX][4] = PFX ## NUM ## _mc01_c; \
  2022. c->PFX ## _pixels_tab[IDX][5] = PFX ## NUM ## _mc11_c; \
  2023. c->PFX ## _pixels_tab[IDX][6] = PFX ## NUM ## _mc21_c; \
  2024. c->PFX ## _pixels_tab[IDX][7] = PFX ## NUM ## _mc31_c; \
  2025. c->PFX ## _pixels_tab[IDX][8] = PFX ## NUM ## _mc02_c; \
  2026. c->PFX ## _pixels_tab[IDX][9] = PFX ## NUM ## _mc12_c; \
  2027. c->PFX ## _pixels_tab[IDX][10] = PFX ## NUM ## _mc22_c; \
  2028. c->PFX ## _pixels_tab[IDX][11] = PFX ## NUM ## _mc32_c; \
  2029. c->PFX ## _pixels_tab[IDX][12] = PFX ## NUM ## _mc03_c; \
  2030. c->PFX ## _pixels_tab[IDX][13] = PFX ## NUM ## _mc13_c; \
  2031. c->PFX ## _pixels_tab[IDX][14] = PFX ## NUM ## _mc23_c; \
  2032. c->PFX ## _pixels_tab[IDX][15] = PFX ## NUM ## _mc33_c
  2033. dspfunc(put_qpel, 0, 16);
  2034. dspfunc(put_qpel, 1, 8);
  2035. dspfunc(put_no_rnd_qpel, 0, 16);
  2036. dspfunc(put_no_rnd_qpel, 1, 8);
  2037. dspfunc(avg_qpel, 0, 16);
  2038. dspfunc(avg_qpel, 1, 8);
  2039. #undef dspfunc
  2040. #define SET_CMP_FUNC(name) \
  2041. c->name[0] = name ## 16_c; \
  2042. c->name[1] = name ## 8x8_c;
  2043. SET_CMP_FUNC(hadamard8_diff)
  2044. c->hadamard8_diff[4] = hadamard8_intra16_c;
  2045. c->hadamard8_diff[5] = hadamard8_intra8x8_c;
  2046. SET_CMP_FUNC(dct_sad)
  2047. SET_CMP_FUNC(dct_max)
  2048. #if CONFIG_GPL
  2049. SET_CMP_FUNC(dct264_sad)
  2050. #endif
  2051. c->sad[0] = pix_abs16_c;
  2052. c->sad[1] = pix_abs8_c;
  2053. c->sse[0] = sse16_c;
  2054. c->sse[1] = sse8_c;
  2055. c->sse[2] = sse4_c;
  2056. SET_CMP_FUNC(quant_psnr)
  2057. SET_CMP_FUNC(rd)
  2058. SET_CMP_FUNC(bit)
  2059. c->vsad[0] = vsad16_c;
  2060. c->vsad[4] = vsad_intra16_c;
  2061. c->vsad[5] = vsad_intra8_c;
  2062. c->vsse[0] = vsse16_c;
  2063. c->vsse[4] = vsse_intra16_c;
  2064. c->vsse[5] = vsse_intra8_c;
  2065. c->nsse[0] = nsse16_c;
  2066. c->nsse[1] = nsse8_c;
  2067. c->bswap_buf = bswap_buf;
  2068. c->bswap16_buf = bswap16_buf;
  2069. c->try_8x8basis = try_8x8basis_c;
  2070. c->add_8x8basis = add_8x8basis_c;
  2071. c->scalarproduct_int16 = scalarproduct_int16_c;
  2072. c->vector_clip_int32 = vector_clip_int32_c;
  2073. c->vector_clipf = vector_clipf_c;
  2074. c->shrink[0] = av_image_copy_plane;
  2075. c->shrink[1] = ff_shrink22;
  2076. c->shrink[2] = ff_shrink44;
  2077. c->shrink[3] = ff_shrink88;
  2078. c->add_pixels8 = add_pixels8_c;
  2079. c->draw_edges = draw_edges_8_c;
  2080. c->clear_block = clear_block_8_c;
  2081. c->clear_blocks = clear_blocks_8_c;
  2082. switch (avctx->bits_per_raw_sample) {
  2083. case 9:
  2084. case 10:
  2085. c->get_pixels = get_pixels_16_c;
  2086. break;
  2087. default:
  2088. c->get_pixels = get_pixels_8_c;
  2089. break;
  2090. }
  2091. if (ARCH_ARM)
  2092. ff_dsputil_init_arm(c, avctx, high_bit_depth);
  2093. if (ARCH_BFIN)
  2094. ff_dsputil_init_bfin(c, avctx, high_bit_depth);
  2095. if (ARCH_PPC)
  2096. ff_dsputil_init_ppc(c, avctx, high_bit_depth);
  2097. if (ARCH_X86)
  2098. ff_dsputil_init_x86(c, avctx, high_bit_depth);
  2099. ff_init_scantable_permutation(c->idct_permutation,
  2100. c->idct_permutation_type);
  2101. }