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.

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