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.

468 lines
18KB

  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. /* Draw the edges of width 'w' of an image of size width, height
  212. * this MMX version can only handle w == 8 || w == 16. */
  213. void ff_draw_edges_mmx(uint8_t *buf, int wrap, int width, int height,
  214. int w, int h, int sides)
  215. {
  216. uint8_t *ptr, *last_line;
  217. int i;
  218. last_line = buf + (height - 1) * wrap;
  219. /* left and right */
  220. ptr = buf;
  221. if (w == 8) {
  222. __asm__ volatile (
  223. "1: \n\t"
  224. "movd (%0), %%mm0 \n\t"
  225. "punpcklbw %%mm0, %%mm0 \n\t"
  226. "punpcklwd %%mm0, %%mm0 \n\t"
  227. "punpckldq %%mm0, %%mm0 \n\t"
  228. "movq %%mm0, -8(%0) \n\t"
  229. "movq -8(%0, %2), %%mm1 \n\t"
  230. "punpckhbw %%mm1, %%mm1 \n\t"
  231. "punpckhwd %%mm1, %%mm1 \n\t"
  232. "punpckhdq %%mm1, %%mm1 \n\t"
  233. "movq %%mm1, (%0, %2) \n\t"
  234. "add %1, %0 \n\t"
  235. "cmp %3, %0 \n\t"
  236. "jb 1b \n\t"
  237. : "+r" (ptr)
  238. : "r" ((x86_reg) wrap), "r" ((x86_reg) width),
  239. "r" (ptr + wrap * height));
  240. } else {
  241. __asm__ volatile (
  242. "1: \n\t"
  243. "movd (%0), %%mm0 \n\t"
  244. "punpcklbw %%mm0, %%mm0 \n\t"
  245. "punpcklwd %%mm0, %%mm0 \n\t"
  246. "punpckldq %%mm0, %%mm0 \n\t"
  247. "movq %%mm0, -8(%0) \n\t"
  248. "movq %%mm0, -16(%0) \n\t"
  249. "movq -8(%0, %2), %%mm1 \n\t"
  250. "punpckhbw %%mm1, %%mm1 \n\t"
  251. "punpckhwd %%mm1, %%mm1 \n\t"
  252. "punpckhdq %%mm1, %%mm1 \n\t"
  253. "movq %%mm1, (%0, %2) \n\t"
  254. "movq %%mm1, 8(%0, %2) \n\t"
  255. "add %1, %0 \n\t"
  256. "cmp %3, %0 \n\t"
  257. "jb 1b \n\t"
  258. : "+r" (ptr)
  259. : "r" ((x86_reg) wrap), "r" ((x86_reg) width),
  260. "r" (ptr + wrap * height));
  261. }
  262. /* top and bottom (and hopefully also the corners) */
  263. if (sides & EDGE_TOP) {
  264. for (i = 0; i < h; i += 4) {
  265. ptr = buf - (i + 1) * wrap - w;
  266. __asm__ volatile (
  267. "1: \n\t"
  268. "movq (%1, %0), %%mm0 \n\t"
  269. "movq %%mm0, (%0) \n\t"
  270. "movq %%mm0, (%0, %2) \n\t"
  271. "movq %%mm0, (%0, %2, 2) \n\t"
  272. "movq %%mm0, (%0, %3) \n\t"
  273. "add $8, %0 \n\t"
  274. "cmp %4, %0 \n\t"
  275. "jb 1b \n\t"
  276. : "+r" (ptr)
  277. : "r" ((x86_reg) buf - (x86_reg) ptr - w),
  278. "r" ((x86_reg) - wrap), "r" ((x86_reg) - wrap * 3),
  279. "r" (ptr + width + 2 * w));
  280. }
  281. }
  282. if (sides & EDGE_BOTTOM) {
  283. for (i = 0; i < h; i += 4) {
  284. ptr = last_line + (i + 1) * wrap - w;
  285. __asm__ volatile (
  286. "1: \n\t"
  287. "movq (%1, %0), %%mm0 \n\t"
  288. "movq %%mm0, (%0) \n\t"
  289. "movq %%mm0, (%0, %2) \n\t"
  290. "movq %%mm0, (%0, %2, 2) \n\t"
  291. "movq %%mm0, (%0, %3) \n\t"
  292. "add $8, %0 \n\t"
  293. "cmp %4, %0 \n\t"
  294. "jb 1b \n\t"
  295. : "+r" (ptr)
  296. : "r" ((x86_reg) last_line - (x86_reg) ptr - w),
  297. "r" ((x86_reg) wrap), "r" ((x86_reg) wrap * 3),
  298. "r" (ptr + width + 2 * w));
  299. }
  300. }
  301. }
  302. void ff_gmc_mmx(uint8_t *dst, uint8_t *src,
  303. int stride, int h, int ox, int oy,
  304. int dxx, int dxy, int dyx, int dyy,
  305. int shift, int r, int width, int height)
  306. {
  307. const int w = 8;
  308. const int ix = ox >> (16 + shift);
  309. const int iy = oy >> (16 + shift);
  310. const int oxs = ox >> 4;
  311. const int oys = oy >> 4;
  312. const int dxxs = dxx >> 4;
  313. const int dxys = dxy >> 4;
  314. const int dyxs = dyx >> 4;
  315. const int dyys = dyy >> 4;
  316. const uint16_t r4[4] = { r, r, r, r };
  317. const uint16_t dxy4[4] = { dxys, dxys, dxys, dxys };
  318. const uint16_t dyy4[4] = { dyys, dyys, dyys, dyys };
  319. const uint64_t shift2 = 2 * shift;
  320. int x, y;
  321. const int dxw = (dxx - (1 << (16 + shift))) * (w - 1);
  322. const int dyh = (dyy - (1 << (16 + shift))) * (h - 1);
  323. const int dxh = dxy * (h - 1);
  324. const int dyw = dyx * (w - 1);
  325. if ( // non-constant fullpel offset (3% of blocks)
  326. ((ox ^ (ox + dxw)) | (ox ^ (ox + dxh)) | (ox ^ (ox + dxw + dxh)) |
  327. (oy ^ (oy + dyw)) | (oy ^ (oy + dyh)) | (oy ^ (oy + dyw + dyh))) >> (16 + shift) ||
  328. // uses more than 16 bits of subpel mv (only at huge resolution)
  329. (dxx | dxy | dyx | dyy) & 15 ||
  330. (unsigned) ix >= width - w ||
  331. (unsigned) iy >= height - h) {
  332. // FIXME could still use mmx for some of the rows
  333. ff_gmc_c(dst, src, stride, h, ox, oy, dxx, dxy, dyx, dyy,
  334. shift, r, width, height);
  335. return;
  336. }
  337. src += ix + iy * stride;
  338. __asm__ volatile (
  339. "movd %0, %%mm6 \n\t"
  340. "pxor %%mm7, %%mm7 \n\t"
  341. "punpcklwd %%mm6, %%mm6 \n\t"
  342. "punpcklwd %%mm6, %%mm6 \n\t"
  343. :: "r" (1 << shift));
  344. for (x = 0; x < w; x += 4) {
  345. uint16_t dx4[4] = { oxs - dxys + dxxs * (x + 0),
  346. oxs - dxys + dxxs * (x + 1),
  347. oxs - dxys + dxxs * (x + 2),
  348. oxs - dxys + dxxs * (x + 3) };
  349. uint16_t dy4[4] = { oys - dyys + dyxs * (x + 0),
  350. oys - dyys + dyxs * (x + 1),
  351. oys - dyys + dyxs * (x + 2),
  352. oys - dyys + dyxs * (x + 3) };
  353. for (y = 0; y < h; y++) {
  354. __asm__ volatile (
  355. "movq %0, %%mm4 \n\t"
  356. "movq %1, %%mm5 \n\t"
  357. "paddw %2, %%mm4 \n\t"
  358. "paddw %3, %%mm5 \n\t"
  359. "movq %%mm4, %0 \n\t"
  360. "movq %%mm5, %1 \n\t"
  361. "psrlw $12, %%mm4 \n\t"
  362. "psrlw $12, %%mm5 \n\t"
  363. : "+m" (*dx4), "+m" (*dy4)
  364. : "m" (*dxy4), "m" (*dyy4));
  365. __asm__ volatile (
  366. "movq %%mm6, %%mm2 \n\t"
  367. "movq %%mm6, %%mm1 \n\t"
  368. "psubw %%mm4, %%mm2 \n\t"
  369. "psubw %%mm5, %%mm1 \n\t"
  370. "movq %%mm2, %%mm0 \n\t"
  371. "movq %%mm4, %%mm3 \n\t"
  372. "pmullw %%mm1, %%mm0 \n\t" // (s - dx) * (s - dy)
  373. "pmullw %%mm5, %%mm3 \n\t" // dx * dy
  374. "pmullw %%mm5, %%mm2 \n\t" // (s - dx) * dy
  375. "pmullw %%mm4, %%mm1 \n\t" // dx * (s - dy)
  376. "movd %4, %%mm5 \n\t"
  377. "movd %3, %%mm4 \n\t"
  378. "punpcklbw %%mm7, %%mm5 \n\t"
  379. "punpcklbw %%mm7, %%mm4 \n\t"
  380. "pmullw %%mm5, %%mm3 \n\t" // src[1, 1] * dx * dy
  381. "pmullw %%mm4, %%mm2 \n\t" // src[0, 1] * (s - dx) * dy
  382. "movd %2, %%mm5 \n\t"
  383. "movd %1, %%mm4 \n\t"
  384. "punpcklbw %%mm7, %%mm5 \n\t"
  385. "punpcklbw %%mm7, %%mm4 \n\t"
  386. "pmullw %%mm5, %%mm1 \n\t" // src[1, 0] * dx * (s - dy)
  387. "pmullw %%mm4, %%mm0 \n\t" // src[0, 0] * (s - dx) * (s - dy)
  388. "paddw %5, %%mm1 \n\t"
  389. "paddw %%mm3, %%mm2 \n\t"
  390. "paddw %%mm1, %%mm0 \n\t"
  391. "paddw %%mm2, %%mm0 \n\t"
  392. "psrlw %6, %%mm0 \n\t"
  393. "packuswb %%mm0, %%mm0 \n\t"
  394. "movd %%mm0, %0 \n\t"
  395. : "=m" (dst[x + y * stride])
  396. : "m" (src[0]), "m" (src[1]),
  397. "m" (src[stride]), "m" (src[stride + 1]),
  398. "m" (*r4), "m" (shift2));
  399. src += stride;
  400. }
  401. src += 4 - h * stride;
  402. }
  403. }
  404. void ff_vector_clipf_sse(float *dst, const float *src,
  405. float min, float max, int len)
  406. {
  407. x86_reg i = (len - 16) * 4;
  408. __asm__ volatile (
  409. "movss %3, %%xmm4 \n\t"
  410. "movss %4, %%xmm5 \n\t"
  411. "shufps $0, %%xmm4, %%xmm4 \n\t"
  412. "shufps $0, %%xmm5, %%xmm5 \n\t"
  413. "1: \n\t"
  414. "movaps (%2, %0), %%xmm0 \n\t" // 3/1 on intel
  415. "movaps 16(%2, %0), %%xmm1 \n\t"
  416. "movaps 32(%2, %0), %%xmm2 \n\t"
  417. "movaps 48(%2, %0), %%xmm3 \n\t"
  418. "maxps %%xmm4, %%xmm0 \n\t"
  419. "maxps %%xmm4, %%xmm1 \n\t"
  420. "maxps %%xmm4, %%xmm2 \n\t"
  421. "maxps %%xmm4, %%xmm3 \n\t"
  422. "minps %%xmm5, %%xmm0 \n\t"
  423. "minps %%xmm5, %%xmm1 \n\t"
  424. "minps %%xmm5, %%xmm2 \n\t"
  425. "minps %%xmm5, %%xmm3 \n\t"
  426. "movaps %%xmm0, (%1, %0) \n\t"
  427. "movaps %%xmm1, 16(%1, %0) \n\t"
  428. "movaps %%xmm2, 32(%1, %0) \n\t"
  429. "movaps %%xmm3, 48(%1, %0) \n\t"
  430. "sub $64, %0 \n\t"
  431. "jge 1b \n\t"
  432. : "+&r" (i)
  433. : "r" (dst), "r" (src), "m" (min), "m" (max)
  434. : "memory");
  435. }
  436. #endif /* HAVE_INLINE_ASM */