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.

493 lines
19KB

  1. /*
  2. * MMX optimized DSP utils
  3. * Copyright (c) 2000, 2001 Fabrice Bellard
  4. * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
  5. *
  6. * MMX optimization by Nick Kurshev <nickols_k@mail.ru>
  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. #include "config.h"
  25. #include "libavutil/cpu.h"
  26. #include "libavutil/x86/asm.h"
  27. #include "constants.h"
  28. #include "dsputil_x86.h"
  29. #if HAVE_INLINE_ASM
  30. void ff_put_pixels_clamped_mmx(const int16_t *block, uint8_t *pixels,
  31. int line_size)
  32. {
  33. const int16_t *p;
  34. uint8_t *pix;
  35. /* read the pixels */
  36. p = block;
  37. pix = pixels;
  38. /* unrolled loop */
  39. __asm__ volatile (
  40. "movq (%3), %%mm0 \n\t"
  41. "movq 8(%3), %%mm1 \n\t"
  42. "movq 16(%3), %%mm2 \n\t"
  43. "movq 24(%3), %%mm3 \n\t"
  44. "movq 32(%3), %%mm4 \n\t"
  45. "movq 40(%3), %%mm5 \n\t"
  46. "movq 48(%3), %%mm6 \n\t"
  47. "movq 56(%3), %%mm7 \n\t"
  48. "packuswb %%mm1, %%mm0 \n\t"
  49. "packuswb %%mm3, %%mm2 \n\t"
  50. "packuswb %%mm5, %%mm4 \n\t"
  51. "packuswb %%mm7, %%mm6 \n\t"
  52. "movq %%mm0, (%0) \n\t"
  53. "movq %%mm2, (%0, %1) \n\t"
  54. "movq %%mm4, (%0, %1, 2) \n\t"
  55. "movq %%mm6, (%0, %2) \n\t"
  56. :: "r" (pix), "r" ((x86_reg) line_size), "r" ((x86_reg) line_size * 3),
  57. "r" (p)
  58. : "memory");
  59. pix += line_size * 4;
  60. p += 32;
  61. // if here would be an exact copy of the code above
  62. // compiler would generate some very strange code
  63. // thus using "r"
  64. __asm__ volatile (
  65. "movq (%3), %%mm0 \n\t"
  66. "movq 8(%3), %%mm1 \n\t"
  67. "movq 16(%3), %%mm2 \n\t"
  68. "movq 24(%3), %%mm3 \n\t"
  69. "movq 32(%3), %%mm4 \n\t"
  70. "movq 40(%3), %%mm5 \n\t"
  71. "movq 48(%3), %%mm6 \n\t"
  72. "movq 56(%3), %%mm7 \n\t"
  73. "packuswb %%mm1, %%mm0 \n\t"
  74. "packuswb %%mm3, %%mm2 \n\t"
  75. "packuswb %%mm5, %%mm4 \n\t"
  76. "packuswb %%mm7, %%mm6 \n\t"
  77. "movq %%mm0, (%0) \n\t"
  78. "movq %%mm2, (%0, %1) \n\t"
  79. "movq %%mm4, (%0, %1, 2) \n\t"
  80. "movq %%mm6, (%0, %2) \n\t"
  81. :: "r" (pix), "r" ((x86_reg) line_size), "r" ((x86_reg) line_size * 3),
  82. "r" (p)
  83. : "memory");
  84. }
  85. #define put_signed_pixels_clamped_mmx_half(off) \
  86. "movq "#off"(%2), %%mm1 \n\t" \
  87. "movq 16 + "#off"(%2), %%mm2 \n\t" \
  88. "movq 32 + "#off"(%2), %%mm3 \n\t" \
  89. "movq 48 + "#off"(%2), %%mm4 \n\t" \
  90. "packsswb 8 + "#off"(%2), %%mm1 \n\t" \
  91. "packsswb 24 + "#off"(%2), %%mm2 \n\t" \
  92. "packsswb 40 + "#off"(%2), %%mm3 \n\t" \
  93. "packsswb 56 + "#off"(%2), %%mm4 \n\t" \
  94. "paddb %%mm0, %%mm1 \n\t" \
  95. "paddb %%mm0, %%mm2 \n\t" \
  96. "paddb %%mm0, %%mm3 \n\t" \
  97. "paddb %%mm0, %%mm4 \n\t" \
  98. "movq %%mm1, (%0) \n\t" \
  99. "movq %%mm2, (%0, %3) \n\t" \
  100. "movq %%mm3, (%0, %3, 2) \n\t" \
  101. "movq %%mm4, (%0, %1) \n\t"
  102. void ff_put_signed_pixels_clamped_mmx(const int16_t *block, uint8_t *pixels,
  103. int line_size)
  104. {
  105. x86_reg line_skip = line_size;
  106. x86_reg line_skip3;
  107. __asm__ volatile (
  108. "movq "MANGLE(ff_pb_80)", %%mm0 \n\t"
  109. "lea (%3, %3, 2), %1 \n\t"
  110. put_signed_pixels_clamped_mmx_half(0)
  111. "lea (%0, %3, 4), %0 \n\t"
  112. put_signed_pixels_clamped_mmx_half(64)
  113. : "+&r" (pixels), "=&r" (line_skip3)
  114. : "r" (block), "r" (line_skip)
  115. : "memory");
  116. }
  117. void ff_add_pixels_clamped_mmx(const int16_t *block, uint8_t *pixels,
  118. int line_size)
  119. {
  120. const int16_t *p;
  121. uint8_t *pix;
  122. int i;
  123. /* read the pixels */
  124. p = block;
  125. pix = pixels;
  126. MOVQ_ZERO(mm7);
  127. i = 4;
  128. do {
  129. __asm__ volatile (
  130. "movq (%2), %%mm0 \n\t"
  131. "movq 8(%2), %%mm1 \n\t"
  132. "movq 16(%2), %%mm2 \n\t"
  133. "movq 24(%2), %%mm3 \n\t"
  134. "movq %0, %%mm4 \n\t"
  135. "movq %1, %%mm6 \n\t"
  136. "movq %%mm4, %%mm5 \n\t"
  137. "punpcklbw %%mm7, %%mm4 \n\t"
  138. "punpckhbw %%mm7, %%mm5 \n\t"
  139. "paddsw %%mm4, %%mm0 \n\t"
  140. "paddsw %%mm5, %%mm1 \n\t"
  141. "movq %%mm6, %%mm5 \n\t"
  142. "punpcklbw %%mm7, %%mm6 \n\t"
  143. "punpckhbw %%mm7, %%mm5 \n\t"
  144. "paddsw %%mm6, %%mm2 \n\t"
  145. "paddsw %%mm5, %%mm3 \n\t"
  146. "packuswb %%mm1, %%mm0 \n\t"
  147. "packuswb %%mm3, %%mm2 \n\t"
  148. "movq %%mm0, %0 \n\t"
  149. "movq %%mm2, %1 \n\t"
  150. : "+m" (*pix), "+m" (*(pix + line_size))
  151. : "r" (p)
  152. : "memory");
  153. pix += line_size * 2;
  154. p += 16;
  155. } while (--i);
  156. }
  157. #define CLEAR_BLOCKS(name, n) \
  158. void name(int16_t *blocks) \
  159. { \
  160. __asm__ volatile ( \
  161. "pxor %%mm7, %%mm7 \n\t" \
  162. "mov %1, %%"REG_a" \n\t" \
  163. "1: \n\t" \
  164. "movq %%mm7, (%0, %%"REG_a") \n\t" \
  165. "movq %%mm7, 8(%0, %%"REG_a") \n\t" \
  166. "movq %%mm7, 16(%0, %%"REG_a") \n\t" \
  167. "movq %%mm7, 24(%0, %%"REG_a") \n\t" \
  168. "add $32, %%"REG_a" \n\t" \
  169. "js 1b \n\t" \
  170. :: "r"(((uint8_t *) blocks) + 128 * n), \
  171. "i"(-128 * n) \
  172. : "%"REG_a); \
  173. }
  174. CLEAR_BLOCKS(ff_clear_blocks_mmx, 6)
  175. CLEAR_BLOCKS(ff_clear_block_mmx, 1)
  176. void ff_clear_block_sse(int16_t *block)
  177. {
  178. __asm__ volatile (
  179. "xorps %%xmm0, %%xmm0 \n"
  180. "movaps %%xmm0, (%0) \n"
  181. "movaps %%xmm0, 16(%0) \n"
  182. "movaps %%xmm0, 32(%0) \n"
  183. "movaps %%xmm0, 48(%0) \n"
  184. "movaps %%xmm0, 64(%0) \n"
  185. "movaps %%xmm0, 80(%0) \n"
  186. "movaps %%xmm0, 96(%0) \n"
  187. "movaps %%xmm0, 112(%0) \n"
  188. :: "r" (block)
  189. : "memory");
  190. }
  191. void ff_clear_blocks_sse(int16_t *blocks)
  192. {
  193. __asm__ volatile (
  194. "xorps %%xmm0, %%xmm0 \n"
  195. "mov %1, %%"REG_a" \n"
  196. "1: \n"
  197. "movaps %%xmm0, (%0, %%"REG_a") \n"
  198. "movaps %%xmm0, 16(%0, %%"REG_a") \n"
  199. "movaps %%xmm0, 32(%0, %%"REG_a") \n"
  200. "movaps %%xmm0, 48(%0, %%"REG_a") \n"
  201. "movaps %%xmm0, 64(%0, %%"REG_a") \n"
  202. "movaps %%xmm0, 80(%0, %%"REG_a") \n"
  203. "movaps %%xmm0, 96(%0, %%"REG_a") \n"
  204. "movaps %%xmm0, 112(%0, %%"REG_a") \n"
  205. "add $128, %%"REG_a" \n"
  206. "js 1b \n"
  207. :: "r"(((uint8_t *) blocks) + 128 * 6), "i"(-128 * 6)
  208. : "%"REG_a);
  209. }
  210. void ff_add_bytes_mmx(uint8_t *dst, uint8_t *src, int w)
  211. {
  212. x86_reg i = 0;
  213. __asm__ volatile (
  214. "jmp 2f \n\t"
  215. "1: \n\t"
  216. "movq (%1, %0), %%mm0 \n\t"
  217. "movq (%2, %0), %%mm1 \n\t"
  218. "paddb %%mm0, %%mm1 \n\t"
  219. "movq %%mm1, (%2, %0) \n\t"
  220. "movq 8(%1, %0), %%mm0 \n\t"
  221. "movq 8(%2, %0), %%mm1 \n\t"
  222. "paddb %%mm0, %%mm1 \n\t"
  223. "movq %%mm1, 8(%2, %0) \n\t"
  224. "add $16, %0 \n\t"
  225. "2: \n\t"
  226. "cmp %3, %0 \n\t"
  227. "js 1b \n\t"
  228. : "+r" (i)
  229. : "r" (src), "r" (dst), "r" ((x86_reg) w - 15));
  230. for (; i < w; i++)
  231. dst[i + 0] += src[i + 0];
  232. }
  233. /* Draw the edges of width 'w' of an image of size width, height
  234. * this MMX version can only handle w == 8 || w == 16. */
  235. void ff_draw_edges_mmx(uint8_t *buf, int wrap, int width, int height,
  236. int w, int h, int sides)
  237. {
  238. uint8_t *ptr, *last_line;
  239. int i;
  240. last_line = buf + (height - 1) * wrap;
  241. /* left and right */
  242. ptr = buf;
  243. if (w == 8) {
  244. __asm__ volatile (
  245. "1: \n\t"
  246. "movd (%0), %%mm0 \n\t"
  247. "punpcklbw %%mm0, %%mm0 \n\t"
  248. "punpcklwd %%mm0, %%mm0 \n\t"
  249. "punpckldq %%mm0, %%mm0 \n\t"
  250. "movq %%mm0, -8(%0) \n\t"
  251. "movq -8(%0, %2), %%mm1 \n\t"
  252. "punpckhbw %%mm1, %%mm1 \n\t"
  253. "punpckhwd %%mm1, %%mm1 \n\t"
  254. "punpckhdq %%mm1, %%mm1 \n\t"
  255. "movq %%mm1, (%0, %2) \n\t"
  256. "add %1, %0 \n\t"
  257. "cmp %3, %0 \n\t"
  258. "jb 1b \n\t"
  259. : "+r" (ptr)
  260. : "r" ((x86_reg) wrap), "r" ((x86_reg) width),
  261. "r" (ptr + wrap * height));
  262. } else {
  263. __asm__ volatile (
  264. "1: \n\t"
  265. "movd (%0), %%mm0 \n\t"
  266. "punpcklbw %%mm0, %%mm0 \n\t"
  267. "punpcklwd %%mm0, %%mm0 \n\t"
  268. "punpckldq %%mm0, %%mm0 \n\t"
  269. "movq %%mm0, -8(%0) \n\t"
  270. "movq %%mm0, -16(%0) \n\t"
  271. "movq -8(%0, %2), %%mm1 \n\t"
  272. "punpckhbw %%mm1, %%mm1 \n\t"
  273. "punpckhwd %%mm1, %%mm1 \n\t"
  274. "punpckhdq %%mm1, %%mm1 \n\t"
  275. "movq %%mm1, (%0, %2) \n\t"
  276. "movq %%mm1, 8(%0, %2) \n\t"
  277. "add %1, %0 \n\t"
  278. "cmp %3, %0 \n\t"
  279. "jb 1b \n\t"
  280. : "+r" (ptr)
  281. : "r" ((x86_reg) wrap), "r" ((x86_reg) width),
  282. "r" (ptr + wrap * height));
  283. }
  284. /* top and bottom (and hopefully also the corners) */
  285. if (sides & EDGE_TOP) {
  286. for (i = 0; i < h; i += 4) {
  287. ptr = buf - (i + 1) * wrap - w;
  288. __asm__ volatile (
  289. "1: \n\t"
  290. "movq (%1, %0), %%mm0 \n\t"
  291. "movq %%mm0, (%0) \n\t"
  292. "movq %%mm0, (%0, %2) \n\t"
  293. "movq %%mm0, (%0, %2, 2) \n\t"
  294. "movq %%mm0, (%0, %3) \n\t"
  295. "add $8, %0 \n\t"
  296. "cmp %4, %0 \n\t"
  297. "jb 1b \n\t"
  298. : "+r" (ptr)
  299. : "r" ((x86_reg) buf - (x86_reg) ptr - w),
  300. "r" ((x86_reg) - wrap), "r" ((x86_reg) - wrap * 3),
  301. "r" (ptr + width + 2 * w));
  302. }
  303. }
  304. if (sides & EDGE_BOTTOM) {
  305. for (i = 0; i < h; i += 4) {
  306. ptr = last_line + (i + 1) * wrap - w;
  307. __asm__ volatile (
  308. "1: \n\t"
  309. "movq (%1, %0), %%mm0 \n\t"
  310. "movq %%mm0, (%0) \n\t"
  311. "movq %%mm0, (%0, %2) \n\t"
  312. "movq %%mm0, (%0, %2, 2) \n\t"
  313. "movq %%mm0, (%0, %3) \n\t"
  314. "add $8, %0 \n\t"
  315. "cmp %4, %0 \n\t"
  316. "jb 1b \n\t"
  317. : "+r" (ptr)
  318. : "r" ((x86_reg) last_line - (x86_reg) ptr - w),
  319. "r" ((x86_reg) wrap), "r" ((x86_reg) wrap * 3),
  320. "r" (ptr + width + 2 * w));
  321. }
  322. }
  323. }
  324. void ff_gmc_mmx(uint8_t *dst, uint8_t *src,
  325. int stride, int h, int ox, int oy,
  326. int dxx, int dxy, int dyx, int dyy,
  327. int shift, int r, int width, int height)
  328. {
  329. const int w = 8;
  330. const int ix = ox >> (16 + shift);
  331. const int iy = oy >> (16 + shift);
  332. const int oxs = ox >> 4;
  333. const int oys = oy >> 4;
  334. const int dxxs = dxx >> 4;
  335. const int dxys = dxy >> 4;
  336. const int dyxs = dyx >> 4;
  337. const int dyys = dyy >> 4;
  338. const uint16_t r4[4] = { r, r, r, r };
  339. const uint16_t dxy4[4] = { dxys, dxys, dxys, dxys };
  340. const uint16_t dyy4[4] = { dyys, dyys, dyys, dyys };
  341. const uint64_t shift2 = 2 * shift;
  342. int x, y;
  343. const int dxw = (dxx - (1 << (16 + shift))) * (w - 1);
  344. const int dyh = (dyy - (1 << (16 + shift))) * (h - 1);
  345. const int dxh = dxy * (h - 1);
  346. const int dyw = dyx * (w - 1);
  347. if ( // non-constant fullpel offset (3% of blocks)
  348. ((ox ^ (ox + dxw)) | (ox ^ (ox + dxh)) | (ox ^ (ox + dxw + dxh)) |
  349. (oy ^ (oy + dyw)) | (oy ^ (oy + dyh)) | (oy ^ (oy + dyw + dyh))) >> (16 + shift) ||
  350. // uses more than 16 bits of subpel mv (only at huge resolution)
  351. (dxx | dxy | dyx | dyy) & 15 ||
  352. (unsigned) ix >= width - w ||
  353. (unsigned) iy >= height - h) {
  354. // FIXME could still use mmx for some of the rows
  355. ff_gmc_c(dst, src, stride, h, ox, oy, dxx, dxy, dyx, dyy,
  356. shift, r, width, height);
  357. return;
  358. }
  359. src += ix + iy * stride;
  360. __asm__ volatile (
  361. "movd %0, %%mm6 \n\t"
  362. "pxor %%mm7, %%mm7 \n\t"
  363. "punpcklwd %%mm6, %%mm6 \n\t"
  364. "punpcklwd %%mm6, %%mm6 \n\t"
  365. :: "r" (1 << shift));
  366. for (x = 0; x < w; x += 4) {
  367. uint16_t dx4[4] = { oxs - dxys + dxxs * (x + 0),
  368. oxs - dxys + dxxs * (x + 1),
  369. oxs - dxys + dxxs * (x + 2),
  370. oxs - dxys + dxxs * (x + 3) };
  371. uint16_t dy4[4] = { oys - dyys + dyxs * (x + 0),
  372. oys - dyys + dyxs * (x + 1),
  373. oys - dyys + dyxs * (x + 2),
  374. oys - dyys + dyxs * (x + 3) };
  375. for (y = 0; y < h; y++) {
  376. __asm__ volatile (
  377. "movq %0, %%mm4 \n\t"
  378. "movq %1, %%mm5 \n\t"
  379. "paddw %2, %%mm4 \n\t"
  380. "paddw %3, %%mm5 \n\t"
  381. "movq %%mm4, %0 \n\t"
  382. "movq %%mm5, %1 \n\t"
  383. "psrlw $12, %%mm4 \n\t"
  384. "psrlw $12, %%mm5 \n\t"
  385. : "+m" (*dx4), "+m" (*dy4)
  386. : "m" (*dxy4), "m" (*dyy4));
  387. __asm__ volatile (
  388. "movq %%mm6, %%mm2 \n\t"
  389. "movq %%mm6, %%mm1 \n\t"
  390. "psubw %%mm4, %%mm2 \n\t"
  391. "psubw %%mm5, %%mm1 \n\t"
  392. "movq %%mm2, %%mm0 \n\t"
  393. "movq %%mm4, %%mm3 \n\t"
  394. "pmullw %%mm1, %%mm0 \n\t" // (s - dx) * (s - dy)
  395. "pmullw %%mm5, %%mm3 \n\t" // dx * dy
  396. "pmullw %%mm5, %%mm2 \n\t" // (s - dx) * dy
  397. "pmullw %%mm4, %%mm1 \n\t" // dx * (s - dy)
  398. "movd %4, %%mm5 \n\t"
  399. "movd %3, %%mm4 \n\t"
  400. "punpcklbw %%mm7, %%mm5 \n\t"
  401. "punpcklbw %%mm7, %%mm4 \n\t"
  402. "pmullw %%mm5, %%mm3 \n\t" // src[1, 1] * dx * dy
  403. "pmullw %%mm4, %%mm2 \n\t" // src[0, 1] * (s - dx) * dy
  404. "movd %2, %%mm5 \n\t"
  405. "movd %1, %%mm4 \n\t"
  406. "punpcklbw %%mm7, %%mm5 \n\t"
  407. "punpcklbw %%mm7, %%mm4 \n\t"
  408. "pmullw %%mm5, %%mm1 \n\t" // src[1, 0] * dx * (s - dy)
  409. "pmullw %%mm4, %%mm0 \n\t" // src[0, 0] * (s - dx) * (s - dy)
  410. "paddw %5, %%mm1 \n\t"
  411. "paddw %%mm3, %%mm2 \n\t"
  412. "paddw %%mm1, %%mm0 \n\t"
  413. "paddw %%mm2, %%mm0 \n\t"
  414. "psrlw %6, %%mm0 \n\t"
  415. "packuswb %%mm0, %%mm0 \n\t"
  416. "movd %%mm0, %0 \n\t"
  417. : "=m" (dst[x + y * stride])
  418. : "m" (src[0]), "m" (src[1]),
  419. "m" (src[stride]), "m" (src[stride + 1]),
  420. "m" (*r4), "m" (shift2));
  421. src += stride;
  422. }
  423. src += 4 - h * stride;
  424. }
  425. }
  426. void ff_vector_clipf_sse(float *dst, const float *src,
  427. float min, float max, int len)
  428. {
  429. x86_reg i = (len - 16) * 4;
  430. __asm__ volatile (
  431. "movss %3, %%xmm4 \n\t"
  432. "movss %4, %%xmm5 \n\t"
  433. "shufps $0, %%xmm4, %%xmm4 \n\t"
  434. "shufps $0, %%xmm5, %%xmm5 \n\t"
  435. "1: \n\t"
  436. "movaps (%2, %0), %%xmm0 \n\t" // 3/1 on intel
  437. "movaps 16(%2, %0), %%xmm1 \n\t"
  438. "movaps 32(%2, %0), %%xmm2 \n\t"
  439. "movaps 48(%2, %0), %%xmm3 \n\t"
  440. "maxps %%xmm4, %%xmm0 \n\t"
  441. "maxps %%xmm4, %%xmm1 \n\t"
  442. "maxps %%xmm4, %%xmm2 \n\t"
  443. "maxps %%xmm4, %%xmm3 \n\t"
  444. "minps %%xmm5, %%xmm0 \n\t"
  445. "minps %%xmm5, %%xmm1 \n\t"
  446. "minps %%xmm5, %%xmm2 \n\t"
  447. "minps %%xmm5, %%xmm3 \n\t"
  448. "movaps %%xmm0, (%1, %0) \n\t"
  449. "movaps %%xmm1, 16(%1, %0) \n\t"
  450. "movaps %%xmm2, 32(%1, %0) \n\t"
  451. "movaps %%xmm3, 48(%1, %0) \n\t"
  452. "sub $64, %0 \n\t"
  453. "jge 1b \n\t"
  454. : "+&r" (i)
  455. : "r" (dst), "r" (src), "m" (min), "m" (max)
  456. : "memory");
  457. }
  458. #endif /* HAVE_INLINE_ASM */