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.

1052 lines
27KB

  1. /*
  2. * MMX optimized DSP utils
  3. * Copyright (c) 2000, 2001 Gerard Lantau.
  4. *
  5. * This program is free software; you can redistribute it and/or modify
  6. * it under the terms of the GNU General Public License as published by
  7. * the Free Software Foundation; either version 2 of the License, or
  8. * (at your option) any later version.
  9. *
  10. * This program is distributed in the hope that it will be useful,
  11. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. * GNU General Public License for more details.
  14. *
  15. * You should have received a copy of the GNU General Public License
  16. * along with this program; if not, write to the Free Software
  17. * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18. *
  19. * MMX optimization by Nick Kurshev <nickols_k@mail.ru>
  20. */
  21. #include "../dsputil.h"
  22. #include "../simple_idct.h"
  23. #include "../mangle.h"
  24. int mm_flags; /* multimedia extension flags */
  25. int pix_abs16x16_mmx(UINT8 *blk1, UINT8 *blk2, int lx);
  26. int pix_abs16x16_x2_mmx(UINT8 *blk1, UINT8 *blk2, int lx);
  27. int pix_abs16x16_y2_mmx(UINT8 *blk1, UINT8 *blk2, int lx);
  28. int pix_abs16x16_xy2_mmx(UINT8 *blk1, UINT8 *blk2, int lx);
  29. int pix_abs16x16_mmx2(UINT8 *blk1, UINT8 *blk2, int lx);
  30. int pix_abs16x16_x2_mmx2(UINT8 *blk1, UINT8 *blk2, int lx);
  31. int pix_abs16x16_y2_mmx2(UINT8 *blk1, UINT8 *blk2, int lx);
  32. int pix_abs16x16_xy2_mmx2(UINT8 *blk1, UINT8 *blk2, int lx);
  33. int pix_abs8x8_mmx(UINT8 *blk1, UINT8 *blk2, int lx);
  34. int pix_abs8x8_x2_mmx(UINT8 *blk1, UINT8 *blk2, int lx);
  35. int pix_abs8x8_y2_mmx(UINT8 *blk1, UINT8 *blk2, int lx);
  36. int pix_abs8x8_xy2_mmx(UINT8 *blk1, UINT8 *blk2, int lx);
  37. int pix_abs8x8_mmx2(UINT8 *blk1, UINT8 *blk2, int lx);
  38. int pix_abs8x8_x2_mmx2(UINT8 *blk1, UINT8 *blk2, int lx);
  39. int pix_abs8x8_y2_mmx2(UINT8 *blk1, UINT8 *blk2, int lx);
  40. int pix_abs8x8_xy2_mmx2(UINT8 *blk1, UINT8 *blk2, int lx);
  41. /* external functions, from idct_mmx.c */
  42. void ff_mmx_idct(DCTELEM *block);
  43. void ff_mmxext_idct(DCTELEM *block);
  44. /* pixel operations */
  45. static const uint64_t mm_bone __attribute__ ((aligned(8))) = 0x0101010101010101ULL;
  46. static const uint64_t mm_wone __attribute__ ((aligned(8))) = 0x0001000100010001ULL;
  47. static const uint64_t mm_wtwo __attribute__ ((aligned(8))) = 0x0002000200020002ULL;
  48. //static const unsigned short mm_wone[4] __attribute__ ((aligned(8))) = { 0x1, 0x1, 0x1, 0x1 };
  49. //static const unsigned short mm_wtwo[4] __attribute__ ((aligned(8))) = { 0x2, 0x2, 0x2, 0x2 };
  50. #define JUMPALIGN() __asm __volatile (".balign 8"::)
  51. #define MOVQ_ZERO(regd) __asm __volatile ("pxor %%" #regd ", %%" #regd ::)
  52. #ifndef PIC
  53. #define MOVQ_WONE(regd) __asm __volatile ("movq %0, %%" #regd " \n\t" ::"m"(mm_wone))
  54. #define MOVQ_WTWO(regd) __asm __volatile ("movq %0, %%" #regd " \n\t" ::"m"(mm_wtwo))
  55. #define MOVQ_BONE(regd) "movq "MANGLE(mm_bone)", "#regd" \n\t"
  56. #else
  57. // for shared library it's better to use this way for accessing constants
  58. // pcmpeqd -> -1
  59. #define MOVQ_WONE(regd) \
  60. __asm __volatile ( \
  61. "pcmpeqd %%" #regd ", %%" #regd " \n\t" \
  62. "psrlw $15, %%" #regd ::)
  63. #define MOVQ_WTWO(regd) \
  64. __asm __volatile ( \
  65. "pcmpeqd %%" #regd ", %%" #regd " \n\t" \
  66. "psrlw $15, %%" #regd " \n\t" \
  67. "psllw $1, %%" #regd ::)
  68. #define MOVQ_BONE(regd) \
  69. "pcmpeqd " #regd ", " #regd " \n\t" \
  70. "psrlw $15, " #regd " \n\t"\
  71. "packuswb " #regd ", " #regd " \n\t"
  72. #endif
  73. /***********************************/
  74. /* 3Dnow specific */
  75. #define DEF(x) x ## _3dnow
  76. /* for Athlons PAVGUSB is prefered */
  77. #define PAVGB "pavgusb"
  78. #include "dsputil_mmx_avg.h"
  79. #undef DEF
  80. #undef PAVGB
  81. /***********************************/
  82. /* MMX2 specific */
  83. #define DEF(x) x ## _mmx2
  84. /* Introduced only in MMX2 set */
  85. #define PAVGB "pavgb"
  86. #include "dsputil_mmx_avg.h"
  87. #undef DEF
  88. #undef PAVGB
  89. /***********************************/
  90. /* standard MMX */
  91. static void get_pixels_mmx(DCTELEM *block, const UINT8 *pixels, int line_size)
  92. {
  93. asm volatile(
  94. "movl $-128, %%eax \n\t"
  95. "pxor %%mm7, %%mm7 \n\t"
  96. ".balign 16 \n\t"
  97. "1: \n\t"
  98. "movq (%0), %%mm0 \n\t"
  99. "movq (%0, %2), %%mm2 \n\t"
  100. "movq %%mm0, %%mm1 \n\t"
  101. "movq %%mm2, %%mm3 \n\t"
  102. "punpcklbw %%mm7, %%mm0 \n\t"
  103. "punpckhbw %%mm7, %%mm1 \n\t"
  104. "punpcklbw %%mm7, %%mm2 \n\t"
  105. "punpckhbw %%mm7, %%mm3 \n\t"
  106. "movq %%mm0, (%1, %%eax)\n\t"
  107. "movq %%mm1, 8(%1, %%eax)\n\t"
  108. "movq %%mm2, 16(%1, %%eax)\n\t"
  109. "movq %%mm3, 24(%1, %%eax)\n\t"
  110. "addl %3, %0 \n\t"
  111. "addl $32, %%eax \n\t"
  112. "js 1b \n\t"
  113. : "+r" (pixels)
  114. : "r" (block+64), "r" (line_size), "r" (line_size*2)
  115. : "%eax"
  116. );
  117. }
  118. static void diff_pixels_mmx(DCTELEM *block, const UINT8 *s1, const UINT8 *s2, int stride)
  119. {
  120. asm volatile(
  121. "pxor %%mm7, %%mm7 \n\t"
  122. "movl $-128, %%eax \n\t"
  123. ".balign 16 \n\t"
  124. "1: \n\t"
  125. "movq (%0), %%mm0 \n\t"
  126. "movq (%1), %%mm2 \n\t"
  127. "movq %%mm0, %%mm1 \n\t"
  128. "movq %%mm2, %%mm3 \n\t"
  129. "punpcklbw %%mm7, %%mm0 \n\t"
  130. "punpckhbw %%mm7, %%mm1 \n\t"
  131. "punpcklbw %%mm7, %%mm2 \n\t"
  132. "punpckhbw %%mm7, %%mm3 \n\t"
  133. "psubw %%mm2, %%mm0 \n\t"
  134. "psubw %%mm3, %%mm1 \n\t"
  135. "movq %%mm0, (%2, %%eax)\n\t"
  136. "movq %%mm1, 8(%2, %%eax)\n\t"
  137. "addl %3, %0 \n\t"
  138. "addl %3, %1 \n\t"
  139. "addl $16, %%eax \n\t"
  140. "jnz 1b \n\t"
  141. : "+r" (s1), "+r" (s2)
  142. : "r" (block+64), "r" (stride)
  143. : "%eax"
  144. );
  145. }
  146. static void put_pixels_clamped_mmx(const DCTELEM *block, UINT8 *pixels, int line_size)
  147. {
  148. const DCTELEM *p;
  149. UINT8 *pix;
  150. /* read the pixels */
  151. p = block;
  152. pix = pixels;
  153. /* unrolled loop */
  154. __asm __volatile(
  155. "movq %3, %%mm0\n\t"
  156. "movq 8%3, %%mm1\n\t"
  157. "movq 16%3, %%mm2\n\t"
  158. "movq 24%3, %%mm3\n\t"
  159. "movq 32%3, %%mm4\n\t"
  160. "movq 40%3, %%mm5\n\t"
  161. "movq 48%3, %%mm6\n\t"
  162. "movq 56%3, %%mm7\n\t"
  163. "packuswb %%mm1, %%mm0\n\t"
  164. "packuswb %%mm3, %%mm2\n\t"
  165. "packuswb %%mm5, %%mm4\n\t"
  166. "packuswb %%mm7, %%mm6\n\t"
  167. "movq %%mm0, (%0)\n\t"
  168. "movq %%mm2, (%0, %1)\n\t"
  169. "movq %%mm4, (%0, %1, 2)\n\t"
  170. "movq %%mm6, (%0, %2)\n\t"
  171. ::"r" (pix), "r" (line_size), "r" (line_size*3), "m"(*p)
  172. :"memory");
  173. pix += line_size*4;
  174. p += 32;
  175. // if here would be an exact copy of the code above
  176. // compiler would generate some very strange code
  177. // thus using "r"
  178. __asm __volatile(
  179. "movq (%3), %%mm0\n\t"
  180. "movq 8(%3), %%mm1\n\t"
  181. "movq 16(%3), %%mm2\n\t"
  182. "movq 24(%3), %%mm3\n\t"
  183. "movq 32(%3), %%mm4\n\t"
  184. "movq 40(%3), %%mm5\n\t"
  185. "movq 48(%3), %%mm6\n\t"
  186. "movq 56(%3), %%mm7\n\t"
  187. "packuswb %%mm1, %%mm0\n\t"
  188. "packuswb %%mm3, %%mm2\n\t"
  189. "packuswb %%mm5, %%mm4\n\t"
  190. "packuswb %%mm7, %%mm6\n\t"
  191. "movq %%mm0, (%0)\n\t"
  192. "movq %%mm2, (%0, %1)\n\t"
  193. "movq %%mm4, (%0, %1, 2)\n\t"
  194. "movq %%mm6, (%0, %2)\n\t"
  195. ::"r" (pix), "r" (line_size), "r" (line_size*3), "r"(p)
  196. :"memory");
  197. }
  198. static void add_pixels_clamped_mmx(const DCTELEM *block, UINT8 *pixels, int line_size)
  199. {
  200. const DCTELEM *p;
  201. UINT8 *pix;
  202. int i;
  203. /* read the pixels */
  204. p = block;
  205. pix = pixels;
  206. MOVQ_ZERO(mm7);
  207. i = 4;
  208. do {
  209. __asm __volatile(
  210. "movq (%2), %%mm0\n\t"
  211. "movq 8(%2), %%mm1\n\t"
  212. "movq 16(%2), %%mm2\n\t"
  213. "movq 24(%2), %%mm3\n\t"
  214. "movq %0, %%mm4\n\t"
  215. "movq %1, %%mm6\n\t"
  216. "movq %%mm4, %%mm5\n\t"
  217. "punpcklbw %%mm7, %%mm4\n\t"
  218. "punpckhbw %%mm7, %%mm5\n\t"
  219. "paddsw %%mm4, %%mm0\n\t"
  220. "paddsw %%mm5, %%mm1\n\t"
  221. "movq %%mm6, %%mm5\n\t"
  222. "punpcklbw %%mm7, %%mm6\n\t"
  223. "punpckhbw %%mm7, %%mm5\n\t"
  224. "paddsw %%mm6, %%mm2\n\t"
  225. "paddsw %%mm5, %%mm3\n\t"
  226. "packuswb %%mm1, %%mm0\n\t"
  227. "packuswb %%mm3, %%mm2\n\t"
  228. "movq %%mm0, %0\n\t"
  229. "movq %%mm2, %1\n\t"
  230. :"+m"(*pix), "+m"(*(pix+line_size))
  231. :"r"(p)
  232. :"memory");
  233. pix += line_size*2;
  234. p += 16;
  235. } while (--i);
  236. }
  237. static void put_pixels_mmx(UINT8 *block, const UINT8 *pixels, int line_size, int h)
  238. {
  239. asm volatile
  240. (
  241. "lea (%3, %3), %%eax \n\t"
  242. ".balign 8 \n\t"
  243. "1: \n\t"
  244. "movq (%1), %%mm0 \n\t"
  245. "movq (%1, %3), %%mm1 \n\t"
  246. "movq %%mm0, (%2) \n\t"
  247. "movq %%mm1, (%2, %3) \n\t"
  248. "addl %%eax, %1 \n\t"
  249. "addl %%eax, %2 \n\t"
  250. "movq (%1), %%mm0 \n\t"
  251. "movq (%1, %3), %%mm1 \n\t"
  252. "movq %%mm0, (%2) \n\t"
  253. "movq %%mm1, (%2, %3) \n\t"
  254. "addl %%eax, %1 \n\t"
  255. "addl %%eax, %2 \n\t"
  256. "subl $4, %0 \n\t"
  257. "jnz 1b \n\t"
  258. : "+g"(h), "+r" (pixels), "+r" (block)
  259. : "r"(line_size)
  260. : "%eax", "memory"
  261. );
  262. }
  263. static void put_pixels_x2_mmx(UINT8 *block, const UINT8 *pixels, int line_size, int h)
  264. {
  265. UINT8 *p;
  266. const UINT8 *pix;
  267. p = block;
  268. pix = pixels;
  269. MOVQ_ZERO(mm7);
  270. MOVQ_WONE(mm4);
  271. JUMPALIGN();
  272. do {
  273. __asm __volatile(
  274. "movq %1, %%mm0\n\t"
  275. "movq 1%1, %%mm1\n\t"
  276. "movq %%mm0, %%mm2\n\t"
  277. "movq %%mm1, %%mm3\n\t"
  278. "punpcklbw %%mm7, %%mm0\n\t"
  279. "punpcklbw %%mm7, %%mm1\n\t"
  280. "punpckhbw %%mm7, %%mm2\n\t"
  281. "punpckhbw %%mm7, %%mm3\n\t"
  282. "paddusw %%mm1, %%mm0\n\t"
  283. "paddusw %%mm3, %%mm2\n\t"
  284. "paddusw %%mm4, %%mm0\n\t"
  285. "paddusw %%mm4, %%mm2\n\t"
  286. "psrlw $1, %%mm0\n\t"
  287. "psrlw $1, %%mm2\n\t"
  288. "packuswb %%mm2, %%mm0\n\t"
  289. "movq %%mm0, %0\n\t"
  290. :"=m"(*p)
  291. :"m"(*pix)
  292. :"memory");
  293. pix += line_size; p += line_size;
  294. } while (--h);
  295. }
  296. static void put_pixels_y2_mmx(UINT8 *block, const UINT8 *pixels, int line_size, int h)
  297. {
  298. UINT8 *p;
  299. const UINT8 *pix;
  300. p = block;
  301. pix = pixels;
  302. MOVQ_ZERO(mm7);
  303. MOVQ_WONE(mm4);
  304. JUMPALIGN();
  305. do {
  306. __asm __volatile(
  307. "movq %1, %%mm0\n\t"
  308. "movq %2, %%mm1\n\t"
  309. "movq %%mm0, %%mm2\n\t"
  310. "movq %%mm1, %%mm3\n\t"
  311. "punpcklbw %%mm7, %%mm0\n\t"
  312. "punpcklbw %%mm7, %%mm1\n\t"
  313. "punpckhbw %%mm7, %%mm2\n\t"
  314. "punpckhbw %%mm7, %%mm3\n\t"
  315. "paddusw %%mm1, %%mm0\n\t"
  316. "paddusw %%mm3, %%mm2\n\t"
  317. "paddusw %%mm4, %%mm0\n\t"
  318. "paddusw %%mm4, %%mm2\n\t"
  319. "psrlw $1, %%mm0\n\t"
  320. "psrlw $1, %%mm2\n\t"
  321. "packuswb %%mm2, %%mm0\n\t"
  322. "movq %%mm0, %0\n\t"
  323. :"=m"(*p)
  324. :"m"(*pix),
  325. "m"(*(pix+line_size))
  326. :"memory");
  327. pix += line_size;
  328. p += line_size;
  329. } while (--h);
  330. }
  331. static void put_pixels_xy2_mmx(UINT8 *block, const UINT8 *pixels, int line_size, int h)
  332. {
  333. UINT8 *p;
  334. const UINT8 *pix;
  335. p = block;
  336. pix = pixels; // 1s
  337. MOVQ_ZERO(mm7);
  338. MOVQ_WTWO(mm6);
  339. JUMPALIGN();
  340. do {
  341. __asm __volatile(
  342. "movq %1, %%mm0\n\t"
  343. "movq %2, %%mm1\n\t"
  344. "movq 1%1, %%mm4\n\t"
  345. "movq 1%2, %%mm5\n\t"
  346. "movq %%mm0, %%mm2\n\t"
  347. "movq %%mm1, %%mm3\n\t"
  348. "punpcklbw %%mm7, %%mm0\n\t"
  349. "punpcklbw %%mm7, %%mm1\n\t"
  350. "punpckhbw %%mm7, %%mm2\n\t"
  351. "punpckhbw %%mm7, %%mm3\n\t"
  352. "paddusw %%mm1, %%mm0\n\t"
  353. "paddusw %%mm3, %%mm2\n\t"
  354. "movq %%mm4, %%mm1\n\t"
  355. "movq %%mm5, %%mm3\n\t"
  356. "punpcklbw %%mm7, %%mm4\n\t"
  357. "punpcklbw %%mm7, %%mm5\n\t"
  358. "punpckhbw %%mm7, %%mm1\n\t"
  359. "punpckhbw %%mm7, %%mm3\n\t"
  360. "paddusw %%mm5, %%mm4\n\t"
  361. "paddusw %%mm3, %%mm1\n\t"
  362. "paddusw %%mm6, %%mm4\n\t"
  363. "paddusw %%mm6, %%mm1\n\t"
  364. "paddusw %%mm4, %%mm0\n\t"
  365. "paddusw %%mm1, %%mm2\n\t"
  366. "psrlw $2, %%mm0\n\t"
  367. "psrlw $2, %%mm2\n\t"
  368. "packuswb %%mm2, %%mm0\n\t"
  369. "movq %%mm0, %0\n\t"
  370. :"=m"(*p)
  371. :"m"(*pix),
  372. "m"(*(pix+line_size))
  373. :"memory");
  374. pix += line_size;
  375. p += line_size;
  376. } while(--h);
  377. }
  378. static void put_no_rnd_pixels_x2_mmx( UINT8 *block, const UINT8 *pixels, int line_size, int h)
  379. {
  380. UINT8 *p;
  381. const UINT8 *pix;
  382. p = block;
  383. pix = pixels;
  384. MOVQ_ZERO(mm7);
  385. do {
  386. __asm __volatile(
  387. "movq %1, %%mm0\n\t"
  388. "movq 1%1, %%mm1\n\t"
  389. "movq %%mm0, %%mm2\n\t"
  390. "movq %%mm1, %%mm3\n\t"
  391. "punpcklbw %%mm7, %%mm0\n\t"
  392. "punpcklbw %%mm7, %%mm1\n\t"
  393. "punpckhbw %%mm7, %%mm2\n\t"
  394. "punpckhbw %%mm7, %%mm3\n\t"
  395. "paddusw %%mm1, %%mm0\n\t"
  396. "paddusw %%mm3, %%mm2\n\t"
  397. "psrlw $1, %%mm0\n\t"
  398. "psrlw $1, %%mm2\n\t"
  399. "packuswb %%mm2, %%mm0\n\t"
  400. "movq %%mm0, %0\n\t"
  401. :"=m"(*p)
  402. :"m"(*pix)
  403. :"memory");
  404. pix += line_size;
  405. p += line_size;
  406. } while (--h);
  407. }
  408. static void put_no_rnd_pixels_y2_mmx( UINT8 *block, const UINT8 *pixels, int line_size, int h)
  409. {
  410. UINT8 *p;
  411. const UINT8 *pix;
  412. p = block;
  413. pix = pixels;
  414. MOVQ_ZERO(mm7);
  415. JUMPALIGN();
  416. do {
  417. __asm __volatile(
  418. "movq %1, %%mm0\n\t"
  419. "movq %2, %%mm1\n\t"
  420. "movq %%mm0, %%mm2\n\t"
  421. "movq %%mm1, %%mm3\n\t"
  422. "punpcklbw %%mm7, %%mm0\n\t"
  423. "punpcklbw %%mm7, %%mm1\n\t"
  424. "punpckhbw %%mm7, %%mm2\n\t"
  425. "punpckhbw %%mm7, %%mm3\n\t"
  426. "paddusw %%mm1, %%mm0\n\t"
  427. "paddusw %%mm3, %%mm2\n\t"
  428. "psrlw $1, %%mm0\n\t"
  429. "psrlw $1, %%mm2\n\t"
  430. "packuswb %%mm2, %%mm0\n\t"
  431. "movq %%mm0, %0\n\t"
  432. :"=m"(*p)
  433. :"m"(*pix),
  434. "m"(*(pix+line_size))
  435. :"memory");
  436. pix += line_size;
  437. p += line_size;
  438. } while(--h);
  439. }
  440. static void put_no_rnd_pixels_xy2_mmx( UINT8 *block, const UINT8 *pixels, int line_size, int h)
  441. {
  442. UINT8 *p;
  443. const UINT8 *pix;
  444. p = block;
  445. pix = pixels;
  446. MOVQ_ZERO(mm7);
  447. MOVQ_WONE(mm6);
  448. JUMPALIGN();
  449. do {
  450. __asm __volatile(
  451. "movq %1, %%mm0\n\t"
  452. "movq %2, %%mm1\n\t"
  453. "movq 1%1, %%mm4\n\t"
  454. "movq 1%2, %%mm5\n\t"
  455. "movq %%mm0, %%mm2\n\t"
  456. "movq %%mm1, %%mm3\n\t"
  457. "punpcklbw %%mm7, %%mm0\n\t"
  458. "punpcklbw %%mm7, %%mm1\n\t"
  459. "punpckhbw %%mm7, %%mm2\n\t"
  460. "punpckhbw %%mm7, %%mm3\n\t"
  461. "paddusw %%mm1, %%mm0\n\t"
  462. "paddusw %%mm3, %%mm2\n\t"
  463. "movq %%mm4, %%mm1\n\t"
  464. "movq %%mm5, %%mm3\n\t"
  465. "punpcklbw %%mm7, %%mm4\n\t"
  466. "punpcklbw %%mm7, %%mm5\n\t"
  467. "punpckhbw %%mm7, %%mm1\n\t"
  468. "punpckhbw %%mm7, %%mm3\n\t"
  469. "paddusw %%mm5, %%mm4\n\t"
  470. "paddusw %%mm3, %%mm1\n\t"
  471. "paddusw %%mm6, %%mm4\n\t"
  472. "paddusw %%mm6, %%mm1\n\t"
  473. "paddusw %%mm4, %%mm0\n\t"
  474. "paddusw %%mm1, %%mm2\n\t"
  475. "psrlw $2, %%mm0\n\t"
  476. "psrlw $2, %%mm2\n\t"
  477. "packuswb %%mm2, %%mm0\n\t"
  478. "movq %%mm0, %0\n\t"
  479. :"=m"(*p)
  480. :"m"(*pix),
  481. "m"(*(pix+line_size))
  482. :"memory");
  483. pix += line_size;
  484. p += line_size;
  485. } while(--h);
  486. }
  487. static void avg_pixels_mmx(UINT8 *block, const UINT8 *pixels, int line_size, int h)
  488. {
  489. UINT8 *p;
  490. const UINT8 *pix;
  491. p = block;
  492. pix = pixels;
  493. MOVQ_ZERO(mm7);
  494. MOVQ_WONE(mm6);
  495. JUMPALIGN();
  496. do {
  497. __asm __volatile(
  498. "movq %0, %%mm0\n\t"
  499. "movq %1, %%mm1\n\t"
  500. "movq %%mm0, %%mm2\n\t"
  501. "movq %%mm1, %%mm3\n\t"
  502. "punpcklbw %%mm7, %%mm0\n\t"
  503. "punpcklbw %%mm7, %%mm1\n\t"
  504. "punpckhbw %%mm7, %%mm2\n\t"
  505. "punpckhbw %%mm7, %%mm3\n\t"
  506. "paddusw %%mm1, %%mm0\n\t"
  507. "paddusw %%mm3, %%mm2\n\t"
  508. "paddusw %%mm6, %%mm0\n\t"
  509. "paddusw %%mm6, %%mm2\n\t"
  510. "psrlw $1, %%mm0\n\t"
  511. "psrlw $1, %%mm2\n\t"
  512. "packuswb %%mm2, %%mm0\n\t"
  513. "movq %%mm0, %0\n\t"
  514. :"+m"(*p)
  515. :"m"(*pix)
  516. :"memory");
  517. pix += line_size;
  518. p += line_size;
  519. }
  520. while (--h);
  521. }
  522. static void avg_pixels_x2_mmx( UINT8 *block, const UINT8 *pixels, int line_size, int h)
  523. {
  524. UINT8 *p;
  525. const UINT8 *pix;
  526. p = block;
  527. pix = pixels;
  528. MOVQ_ZERO(mm7);
  529. MOVQ_WONE(mm6);
  530. JUMPALIGN();
  531. do {
  532. __asm __volatile(
  533. "movq %1, %%mm1\n\t"
  534. "movq %0, %%mm0\n\t"
  535. "movq 1%1, %%mm4\n\t"
  536. "movq %%mm0, %%mm2\n\t"
  537. "movq %%mm1, %%mm3\n\t"
  538. "movq %%mm4, %%mm5\n\t"
  539. "punpcklbw %%mm7, %%mm1\n\t"
  540. "punpckhbw %%mm7, %%mm3\n\t"
  541. "punpcklbw %%mm7, %%mm4\n\t"
  542. "punpckhbw %%mm7, %%mm5\n\t"
  543. "punpcklbw %%mm7, %%mm0\n\t"
  544. "punpckhbw %%mm7, %%mm2\n\t"
  545. "paddusw %%mm4, %%mm1\n\t"
  546. "paddusw %%mm5, %%mm3\n\t"
  547. "paddusw %%mm6, %%mm1\n\t"
  548. "paddusw %%mm6, %%mm3\n\t"
  549. "psrlw $1, %%mm1\n\t"
  550. "psrlw $1, %%mm3\n\t"
  551. "paddusw %%mm6, %%mm0\n\t"
  552. "paddusw %%mm6, %%mm2\n\t"
  553. "paddusw %%mm1, %%mm0\n\t"
  554. "paddusw %%mm3, %%mm2\n\t"
  555. "psrlw $1, %%mm0\n\t"
  556. "psrlw $1, %%mm2\n\t"
  557. "packuswb %%mm2, %%mm0\n\t"
  558. "movq %%mm0, %0\n\t"
  559. :"+m"(*p)
  560. :"m"(*pix)
  561. :"memory");
  562. pix += line_size;
  563. p += line_size;
  564. } while (--h);
  565. }
  566. static void avg_pixels_y2_mmx( UINT8 *block, const UINT8 *pixels, int line_size, int h)
  567. {
  568. UINT8 *p;
  569. const UINT8 *pix;
  570. p = block;
  571. pix = pixels;
  572. MOVQ_ZERO(mm7);
  573. MOVQ_WONE(mm6);
  574. JUMPALIGN();
  575. do {
  576. __asm __volatile(
  577. "movq %1, %%mm1\n\t"
  578. "movq %0, %%mm0\n\t"
  579. "movq %2, %%mm4\n\t"
  580. "movq %%mm0, %%mm2\n\t"
  581. "movq %%mm1, %%mm3\n\t"
  582. "movq %%mm4, %%mm5\n\t"
  583. "punpcklbw %%mm7, %%mm1\n\t"
  584. "punpckhbw %%mm7, %%mm3\n\t"
  585. "punpcklbw %%mm7, %%mm4\n\t"
  586. "punpckhbw %%mm7, %%mm5\n\t"
  587. "punpcklbw %%mm7, %%mm0\n\t"
  588. "punpckhbw %%mm7, %%mm2\n\t"
  589. "paddusw %%mm4, %%mm1\n\t"
  590. "paddusw %%mm5, %%mm3\n\t"
  591. "paddusw %%mm6, %%mm1\n\t"
  592. "paddusw %%mm6, %%mm3\n\t"
  593. "psrlw $1, %%mm1\n\t"
  594. "psrlw $1, %%mm3\n\t"
  595. "paddusw %%mm6, %%mm0\n\t"
  596. "paddusw %%mm6, %%mm2\n\t"
  597. "paddusw %%mm1, %%mm0\n\t"
  598. "paddusw %%mm3, %%mm2\n\t"
  599. "psrlw $1, %%mm0\n\t"
  600. "psrlw $1, %%mm2\n\t"
  601. "packuswb %%mm2, %%mm0\n\t"
  602. "movq %%mm0, %0\n\t"
  603. :"+m"(*p)
  604. :"m"(*pix), "m"(*(pix+line_size))
  605. :"memory");
  606. pix += line_size;
  607. p += line_size ;
  608. } while(--h);
  609. }
  610. static void avg_pixels_xy2_mmx( UINT8 *block, const UINT8 *pixels, int line_size, int h)
  611. {
  612. UINT8 *p;
  613. const UINT8 *pix;
  614. p = block;
  615. pix = pixels;
  616. MOVQ_ZERO(mm7);
  617. // this doesn't seem to be used offten - so
  618. // the inside usage of mm_wone is not optimized
  619. MOVQ_WTWO(mm6);
  620. do {
  621. __asm __volatile(
  622. "movq %1, %%mm0\n\t"
  623. "movq %2, %%mm1\n\t"
  624. "movq 1%1, %%mm4\n\t"
  625. "movq 1%2, %%mm5\n\t"
  626. "movq %%mm0, %%mm2\n\t"
  627. "movq %%mm1, %%mm3\n\t"
  628. "punpcklbw %%mm7, %%mm0\n\t"
  629. "punpcklbw %%mm7, %%mm1\n\t"
  630. "punpckhbw %%mm7, %%mm2\n\t"
  631. "punpckhbw %%mm7, %%mm3\n\t"
  632. "paddusw %%mm1, %%mm0\n\t"
  633. "paddusw %%mm3, %%mm2\n\t"
  634. "movq %%mm4, %%mm1\n\t"
  635. "movq %%mm5, %%mm3\n\t"
  636. "punpcklbw %%mm7, %%mm4\n\t"
  637. "punpcklbw %%mm7, %%mm5\n\t"
  638. "punpckhbw %%mm7, %%mm1\n\t"
  639. "punpckhbw %%mm7, %%mm3\n\t"
  640. "paddusw %%mm5, %%mm4\n\t"
  641. "paddusw %%mm3, %%mm1\n\t"
  642. "paddusw %%mm6, %%mm4\n\t"
  643. "paddusw %%mm6, %%mm1\n\t"
  644. "paddusw %%mm4, %%mm0\n\t"
  645. "paddusw %%mm1, %%mm2\n\t"
  646. "movq %3, %%mm5\n\t"
  647. "psrlw $2, %%mm0\n\t"
  648. "movq %0, %%mm1\n\t"
  649. "psrlw $2, %%mm2\n\t"
  650. "movq %%mm1, %%mm3\n\t"
  651. "punpcklbw %%mm7, %%mm1\n\t"
  652. "punpckhbw %%mm7, %%mm3\n\t"
  653. "paddusw %%mm1, %%mm0\n\t"
  654. "paddusw %%mm3, %%mm2\n\t"
  655. "paddusw %%mm5, %%mm0\n\t"
  656. "paddusw %%mm5, %%mm2\n\t"
  657. "psrlw $1, %%mm0\n\t"
  658. "psrlw $1, %%mm2\n\t"
  659. "packuswb %%mm2, %%mm0\n\t"
  660. "movq %%mm0, %0\n\t"
  661. :"+m"(*p)
  662. :"m"(*pix),
  663. "m"(*(pix+line_size)), "m"(mm_wone)
  664. :"memory");
  665. pix += line_size;
  666. p += line_size ;
  667. } while(--h);
  668. }
  669. static void avg_no_rnd_pixels_mmx( UINT8 *block, const UINT8 *pixels, int line_size, int h)
  670. {
  671. UINT8 *p;
  672. const UINT8 *pix;
  673. p = block;
  674. pix = pixels;
  675. MOVQ_ZERO(mm7);
  676. do {
  677. __asm __volatile(
  678. "movq %1, %%mm0\n\t"
  679. "movq %0, %%mm1\n\t"
  680. "movq %%mm0, %%mm2\n\t"
  681. "movq %%mm1, %%mm3\n\t"
  682. "punpcklbw %%mm7, %%mm0\n\t"
  683. "punpcklbw %%mm7, %%mm1\n\t"
  684. "punpckhbw %%mm7, %%mm2\n\t"
  685. "punpckhbw %%mm7, %%mm3\n\t"
  686. "paddusw %%mm1, %%mm0\n\t"
  687. "paddusw %%mm3, %%mm2\n\t"
  688. "psrlw $1, %%mm0\n\t"
  689. "psrlw $1, %%mm2\n\t"
  690. "packuswb %%mm2, %%mm0\n\t"
  691. "movq %%mm0, %0\n\t"
  692. :"+m"(*p)
  693. :"m"(*pix)
  694. :"memory");
  695. pix += line_size;
  696. p += line_size ;
  697. } while (--h);
  698. }
  699. static void avg_no_rnd_pixels_x2_mmx( UINT8 *block, const UINT8 *pixels, int line_size, int h)
  700. {
  701. UINT8 *p;
  702. const UINT8 *pix;
  703. p = block;
  704. pix = pixels;
  705. MOVQ_ZERO(mm7);
  706. do {
  707. __asm __volatile(
  708. "movq %1, %%mm0\n\t"
  709. "movq 1%1, %%mm1\n\t"
  710. "movq %0, %%mm4\n\t"
  711. "movq %%mm0, %%mm2\n\t"
  712. "movq %%mm1, %%mm3\n\t"
  713. "movq %%mm4, %%mm5\n\t"
  714. "punpcklbw %%mm7, %%mm0\n\t"
  715. "punpcklbw %%mm7, %%mm1\n\t"
  716. "punpckhbw %%mm7, %%mm2\n\t"
  717. "punpckhbw %%mm7, %%mm3\n\t"
  718. "punpcklbw %%mm7, %%mm4\n\t"
  719. "punpckhbw %%mm7, %%mm5\n\t"
  720. "paddusw %%mm1, %%mm0\n\t"
  721. "paddusw %%mm3, %%mm2\n\t"
  722. "psrlw $1, %%mm0\n\t"
  723. "psrlw $1, %%mm2\n\t"
  724. "paddusw %%mm4, %%mm0\n\t"
  725. "paddusw %%mm5, %%mm2\n\t"
  726. "psrlw $1, %%mm0\n\t"
  727. "psrlw $1, %%mm2\n\t"
  728. "packuswb %%mm2, %%mm0\n\t"
  729. "movq %%mm0, %0\n\t"
  730. :"+m"(*p)
  731. :"m"(*pix)
  732. :"memory");
  733. pix += line_size;
  734. p += line_size;
  735. } while (--h);
  736. }
  737. static void avg_no_rnd_pixels_y2_mmx( UINT8 *block, const UINT8 *pixels, int line_size, int h)
  738. {
  739. UINT8 *p;
  740. const UINT8 *pix;
  741. p = block;
  742. pix = pixels;
  743. MOVQ_ZERO(mm7);
  744. do {
  745. __asm __volatile(
  746. "movq %1, %%mm0\n\t"
  747. "movq %2, %%mm1\n\t"
  748. "movq %0, %%mm4\n\t"
  749. "movq %%mm0, %%mm2\n\t"
  750. "movq %%mm1, %%mm3\n\t"
  751. "movq %%mm4, %%mm5\n\t"
  752. "punpcklbw %%mm7, %%mm0\n\t"
  753. "punpcklbw %%mm7, %%mm1\n\t"
  754. "punpckhbw %%mm7, %%mm2\n\t"
  755. "punpckhbw %%mm7, %%mm3\n\t"
  756. "punpcklbw %%mm7, %%mm4\n\t"
  757. "punpckhbw %%mm7, %%mm5\n\t"
  758. "paddusw %%mm1, %%mm0\n\t"
  759. "paddusw %%mm3, %%mm2\n\t"
  760. "psrlw $1, %%mm0\n\t"
  761. "psrlw $1, %%mm2\n\t"
  762. "paddusw %%mm4, %%mm0\n\t"
  763. "paddusw %%mm5, %%mm2\n\t"
  764. "psrlw $1, %%mm0\n\t"
  765. "psrlw $1, %%mm2\n\t"
  766. "packuswb %%mm2, %%mm0\n\t"
  767. "movq %%mm0, %0\n\t"
  768. :"+m"(*p)
  769. :"m"(*pix), "m"(*(pix+line_size))
  770. :"memory");
  771. pix += line_size;
  772. p += line_size ;
  773. } while(--h);
  774. }
  775. static void avg_no_rnd_pixels_xy2_mmx( UINT8 *block, const UINT8 *pixels, int line_size, int h)
  776. {
  777. UINT8 *p;
  778. const UINT8 *pix;
  779. p = block;
  780. pix = pixels;
  781. MOVQ_ZERO(mm7);
  782. MOVQ_WONE(mm6);
  783. JUMPALIGN();
  784. do {
  785. __asm __volatile(
  786. "movq %1, %%mm0\n\t"
  787. "movq %2, %%mm1\n\t"
  788. "movq 1%1, %%mm4\n\t"
  789. "movq 1%2, %%mm5\n\t"
  790. "movq %%mm0, %%mm2\n\t"
  791. "movq %%mm1, %%mm3\n\t"
  792. "punpcklbw %%mm7, %%mm0\n\t"
  793. "punpcklbw %%mm7, %%mm1\n\t"
  794. "punpckhbw %%mm7, %%mm2\n\t"
  795. "punpckhbw %%mm7, %%mm3\n\t"
  796. "paddusw %%mm1, %%mm0\n\t"
  797. "paddusw %%mm3, %%mm2\n\t"
  798. "movq %%mm4, %%mm1\n\t"
  799. "movq %%mm5, %%mm3\n\t"
  800. "punpcklbw %%mm7, %%mm4\n\t"
  801. "punpcklbw %%mm7, %%mm5\n\t"
  802. "punpckhbw %%mm7, %%mm1\n\t"
  803. "punpckhbw %%mm7, %%mm3\n\t"
  804. "paddusw %%mm5, %%mm4\n\t"
  805. "paddusw %%mm3, %%mm1\n\t"
  806. "paddusw %%mm6, %%mm4\n\t"
  807. "paddusw %%mm6, %%mm1\n\t"
  808. "paddusw %%mm4, %%mm0\n\t"
  809. "paddusw %%mm1, %%mm2\n\t"
  810. "movq %0, %%mm1\n\t"
  811. "psrlw $2, %%mm0\n\t"
  812. "movq %%mm1, %%mm3\n\t"
  813. "psrlw $2, %%mm2\n\t"
  814. "punpcklbw %%mm7, %%mm1\n\t"
  815. "punpckhbw %%mm7, %%mm3\n\t"
  816. "paddusw %%mm1, %%mm0\n\t"
  817. "paddusw %%mm3, %%mm2\n\t"
  818. "psrlw $1, %%mm0\n\t"
  819. "psrlw $1, %%mm2\n\t"
  820. "packuswb %%mm2, %%mm0\n\t"
  821. "movq %%mm0, %0\n\t"
  822. :"+m"(*p)
  823. :"m"(*pix),
  824. "m"(*(pix+line_size))
  825. :"memory");
  826. pix += line_size;
  827. p += line_size;
  828. } while(--h);
  829. }
  830. static void clear_blocks_mmx(DCTELEM *blocks)
  831. {
  832. asm volatile(
  833. "pxor %%mm7, %%mm7 \n\t"
  834. "movl $-128*6, %%eax \n\t"
  835. "1: \n\t"
  836. "movq %%mm7, (%0, %%eax) \n\t"
  837. "movq %%mm7, 8(%0, %%eax) \n\t"
  838. "movq %%mm7, 16(%0, %%eax) \n\t"
  839. "movq %%mm7, 24(%0, %%eax) \n\t"
  840. "addl $32, %%eax \n\t"
  841. " js 1b \n\t"
  842. : : "r" (((int)blocks)+128*6)
  843. : "%eax"
  844. );
  845. }
  846. #if 0
  847. static void just_return() { return; }
  848. #endif
  849. #ifndef TESTCPU_MAIN
  850. void dsputil_init_mmx(void)
  851. {
  852. mm_flags = mm_support();
  853. #if 1
  854. printf("libavcodec: CPU flags:");
  855. if (mm_flags & MM_MMX)
  856. printf(" mmx");
  857. if (mm_flags & MM_MMXEXT)
  858. printf(" mmxext");
  859. if (mm_flags & MM_3DNOW)
  860. printf(" 3dnow");
  861. if (mm_flags & MM_SSE)
  862. printf(" sse");
  863. if (mm_flags & MM_SSE2)
  864. printf(" sse2");
  865. printf("\n");
  866. #endif
  867. if (mm_flags & MM_MMX) {
  868. get_pixels = get_pixels_mmx;
  869. diff_pixels = diff_pixels_mmx;
  870. put_pixels_clamped = put_pixels_clamped_mmx;
  871. add_pixels_clamped = add_pixels_clamped_mmx;
  872. clear_blocks= clear_blocks_mmx;
  873. pix_abs16x16 = pix_abs16x16_mmx;
  874. pix_abs16x16_x2 = pix_abs16x16_x2_mmx;
  875. pix_abs16x16_y2 = pix_abs16x16_y2_mmx;
  876. pix_abs16x16_xy2 = pix_abs16x16_xy2_mmx;
  877. pix_abs8x8 = pix_abs8x8_mmx;
  878. pix_abs8x8_x2 = pix_abs8x8_x2_mmx;
  879. pix_abs8x8_y2 = pix_abs8x8_y2_mmx;
  880. pix_abs8x8_xy2= pix_abs8x8_xy2_mmx;
  881. av_fdct = fdct_mmx;
  882. put_pixels_tab[0] = put_pixels_mmx;
  883. put_pixels_tab[1] = put_pixels_x2_mmx;
  884. put_pixels_tab[2] = put_pixels_y2_mmx;
  885. put_pixels_tab[3] = put_pixels_xy2_mmx;
  886. put_no_rnd_pixels_tab[0] = put_pixels_mmx;
  887. put_no_rnd_pixels_tab[1] = put_no_rnd_pixels_x2_mmx;
  888. put_no_rnd_pixels_tab[2] = put_no_rnd_pixels_y2_mmx;
  889. put_no_rnd_pixels_tab[3] = put_no_rnd_pixels_xy2_mmx;
  890. avg_pixels_tab[0] = avg_pixels_mmx;
  891. avg_pixels_tab[1] = avg_pixels_x2_mmx;
  892. avg_pixels_tab[2] = avg_pixels_y2_mmx;
  893. avg_pixels_tab[3] = avg_pixels_xy2_mmx;
  894. avg_no_rnd_pixels_tab[0] = avg_no_rnd_pixels_mmx;
  895. avg_no_rnd_pixels_tab[1] = avg_no_rnd_pixels_x2_mmx;
  896. avg_no_rnd_pixels_tab[2] = avg_no_rnd_pixels_y2_mmx;
  897. avg_no_rnd_pixels_tab[3] = avg_no_rnd_pixels_xy2_mmx;
  898. if (mm_flags & MM_MMXEXT) {
  899. pix_abs16x16 = pix_abs16x16_mmx2;
  900. pix_abs16x16_x2 = pix_abs16x16_x2_mmx2;
  901. pix_abs16x16_y2 = pix_abs16x16_y2_mmx2;
  902. pix_abs16x16_xy2= pix_abs16x16_xy2_mmx2;
  903. pix_abs8x8 = pix_abs8x8_mmx2;
  904. pix_abs8x8_x2 = pix_abs8x8_x2_mmx2;
  905. pix_abs8x8_y2 = pix_abs8x8_y2_mmx2;
  906. pix_abs8x8_xy2= pix_abs8x8_xy2_mmx2;
  907. put_pixels_tab[1] = put_pixels_x2_mmx2;
  908. put_pixels_tab[2] = put_pixels_y2_mmx2;
  909. put_no_rnd_pixels_tab[1] = put_no_rnd_pixels_x2_mmx2;
  910. put_no_rnd_pixels_tab[2] = put_no_rnd_pixels_y2_mmx2;
  911. avg_pixels_tab[0] = avg_pixels_mmx2;
  912. avg_pixels_tab[1] = avg_pixels_x2_mmx2;
  913. avg_pixels_tab[2] = avg_pixels_y2_mmx2;
  914. avg_pixels_tab[3] = avg_pixels_xy2_mmx2;
  915. } else if (mm_flags & MM_3DNOW) {
  916. put_pixels_tab[1] = put_pixels_x2_3dnow;
  917. put_pixels_tab[2] = put_pixels_y2_3dnow;
  918. put_no_rnd_pixels_tab[1] = put_no_rnd_pixels_x2_3dnow;
  919. put_no_rnd_pixels_tab[2] = put_no_rnd_pixels_y2_3dnow;
  920. avg_pixels_tab[0] = avg_pixels_3dnow;
  921. avg_pixels_tab[1] = avg_pixels_x2_3dnow;
  922. avg_pixels_tab[2] = avg_pixels_y2_3dnow;
  923. avg_pixels_tab[3] = avg_pixels_xy2_3dnow;
  924. }
  925. /* idct */
  926. if (mm_flags & MM_MMXEXT) {
  927. ff_idct = ff_mmxext_idct;
  928. } else {
  929. ff_idct = ff_mmx_idct;
  930. }
  931. #ifdef SIMPLE_IDCT
  932. // ff_idct = simple_idct;
  933. ff_idct = simple_idct_mmx;
  934. #endif
  935. }
  936. #if 0
  937. // for speed testing
  938. get_pixels = just_return;
  939. put_pixels_clamped = just_return;
  940. add_pixels_clamped = just_return;
  941. pix_abs16x16 = just_return;
  942. pix_abs16x16_x2 = just_return;
  943. pix_abs16x16_y2 = just_return;
  944. pix_abs16x16_xy2 = just_return;
  945. put_pixels_tab[0] = just_return;
  946. put_pixels_tab[1] = just_return;
  947. put_pixels_tab[2] = just_return;
  948. put_pixels_tab[3] = just_return;
  949. put_no_rnd_pixels_tab[0] = just_return;
  950. put_no_rnd_pixels_tab[1] = just_return;
  951. put_no_rnd_pixels_tab[2] = just_return;
  952. put_no_rnd_pixels_tab[3] = just_return;
  953. avg_pixels_tab[0] = just_return;
  954. avg_pixels_tab[1] = just_return;
  955. avg_pixels_tab[2] = just_return;
  956. avg_pixels_tab[3] = just_return;
  957. avg_no_rnd_pixels_tab[0] = just_return;
  958. avg_no_rnd_pixels_tab[1] = just_return;
  959. avg_no_rnd_pixels_tab[2] = just_return;
  960. avg_no_rnd_pixels_tab[3] = just_return;
  961. //av_fdct = just_return;
  962. //ff_idct = just_return;
  963. #endif
  964. }
  965. /* remove any non bit exact operation (testing purpose). NOTE that
  966. this function should be kept as small as possible because it is
  967. always difficult to test automatically non bit exact cases. */
  968. void dsputil_set_bit_exact_mmx(void)
  969. {
  970. if (mm_flags & MM_MMX) {
  971. if (mm_flags & MM_MMXEXT) {
  972. put_no_rnd_pixels_tab[1] = put_no_rnd_pixels_x2_mmx;
  973. put_no_rnd_pixels_tab[2] = put_no_rnd_pixels_y2_mmx;
  974. avg_pixels_tab[3] = avg_pixels_xy2_mmx;
  975. } else if (mm_flags & MM_3DNOW) {
  976. put_no_rnd_pixels_tab[1] = put_no_rnd_pixels_x2_mmx;
  977. put_no_rnd_pixels_tab[2] = put_no_rnd_pixels_y2_mmx;
  978. avg_pixels_tab[3] = avg_pixels_xy2_mmx;
  979. }
  980. }
  981. }
  982. #endif