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.

1442 lines
43KB

  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. * This file is part of FFmpeg.
  7. *
  8. * FFmpeg is free software; you can redistribute it and/or
  9. * modify it under the terms of the GNU Lesser General Public
  10. * License as published by the Free Software Foundation; either
  11. * version 2.1 of the License, or (at your option) any later version.
  12. *
  13. * FFmpeg is distributed in the hope that it will be useful,
  14. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16. * Lesser General Public License for more details.
  17. *
  18. * You should have received a copy of the GNU Lesser General Public
  19. * License along with FFmpeg; if not, write to the Free Software
  20. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21. *
  22. * MMX optimization by Nick Kurshev <nickols_k@mail.ru>
  23. */
  24. #include "libavutil/x86_cpu.h"
  25. #include "libavcodec/dsputil.h"
  26. #include "libavcodec/mpegvideo.h"
  27. #include "dsputil_mmx.h"
  28. static void get_pixels_mmx(DCTELEM *block, const uint8_t *pixels, int line_size)
  29. {
  30. __asm__ volatile(
  31. "mov $-128, %%"REG_a" \n\t"
  32. "pxor %%mm7, %%mm7 \n\t"
  33. ASMALIGN(4)
  34. "1: \n\t"
  35. "movq (%0), %%mm0 \n\t"
  36. "movq (%0, %2), %%mm2 \n\t"
  37. "movq %%mm0, %%mm1 \n\t"
  38. "movq %%mm2, %%mm3 \n\t"
  39. "punpcklbw %%mm7, %%mm0 \n\t"
  40. "punpckhbw %%mm7, %%mm1 \n\t"
  41. "punpcklbw %%mm7, %%mm2 \n\t"
  42. "punpckhbw %%mm7, %%mm3 \n\t"
  43. "movq %%mm0, (%1, %%"REG_a") \n\t"
  44. "movq %%mm1, 8(%1, %%"REG_a") \n\t"
  45. "movq %%mm2, 16(%1, %%"REG_a") \n\t"
  46. "movq %%mm3, 24(%1, %%"REG_a") \n\t"
  47. "add %3, %0 \n\t"
  48. "add $32, %%"REG_a" \n\t"
  49. "js 1b \n\t"
  50. : "+r" (pixels)
  51. : "r" (block+64), "r" ((x86_reg)line_size), "r" ((x86_reg)line_size*2)
  52. : "%"REG_a
  53. );
  54. }
  55. static void get_pixels_sse2(DCTELEM *block, const uint8_t *pixels, int line_size)
  56. {
  57. __asm__ volatile(
  58. "pxor %%xmm7, %%xmm7 \n\t"
  59. "movq (%0), %%xmm0 \n\t"
  60. "movq (%0, %2), %%xmm1 \n\t"
  61. "movq (%0, %2,2), %%xmm2 \n\t"
  62. "movq (%0, %3), %%xmm3 \n\t"
  63. "lea (%0,%2,4), %0 \n\t"
  64. "punpcklbw %%xmm7, %%xmm0 \n\t"
  65. "punpcklbw %%xmm7, %%xmm1 \n\t"
  66. "punpcklbw %%xmm7, %%xmm2 \n\t"
  67. "punpcklbw %%xmm7, %%xmm3 \n\t"
  68. "movdqa %%xmm0, (%1) \n\t"
  69. "movdqa %%xmm1, 16(%1) \n\t"
  70. "movdqa %%xmm2, 32(%1) \n\t"
  71. "movdqa %%xmm3, 48(%1) \n\t"
  72. "movq (%0), %%xmm0 \n\t"
  73. "movq (%0, %2), %%xmm1 \n\t"
  74. "movq (%0, %2,2), %%xmm2 \n\t"
  75. "movq (%0, %3), %%xmm3 \n\t"
  76. "punpcklbw %%xmm7, %%xmm0 \n\t"
  77. "punpcklbw %%xmm7, %%xmm1 \n\t"
  78. "punpcklbw %%xmm7, %%xmm2 \n\t"
  79. "punpcklbw %%xmm7, %%xmm3 \n\t"
  80. "movdqa %%xmm0, 64(%1) \n\t"
  81. "movdqa %%xmm1, 80(%1) \n\t"
  82. "movdqa %%xmm2, 96(%1) \n\t"
  83. "movdqa %%xmm3, 112(%1) \n\t"
  84. : "+r" (pixels)
  85. : "r" (block), "r" ((x86_reg)line_size), "r" ((x86_reg)line_size*3)
  86. );
  87. }
  88. static inline void diff_pixels_mmx(DCTELEM *block, const uint8_t *s1, const uint8_t *s2, int stride)
  89. {
  90. __asm__ volatile(
  91. "pxor %%mm7, %%mm7 \n\t"
  92. "mov $-128, %%"REG_a" \n\t"
  93. ASMALIGN(4)
  94. "1: \n\t"
  95. "movq (%0), %%mm0 \n\t"
  96. "movq (%1), %%mm2 \n\t"
  97. "movq %%mm0, %%mm1 \n\t"
  98. "movq %%mm2, %%mm3 \n\t"
  99. "punpcklbw %%mm7, %%mm0 \n\t"
  100. "punpckhbw %%mm7, %%mm1 \n\t"
  101. "punpcklbw %%mm7, %%mm2 \n\t"
  102. "punpckhbw %%mm7, %%mm3 \n\t"
  103. "psubw %%mm2, %%mm0 \n\t"
  104. "psubw %%mm3, %%mm1 \n\t"
  105. "movq %%mm0, (%2, %%"REG_a") \n\t"
  106. "movq %%mm1, 8(%2, %%"REG_a") \n\t"
  107. "add %3, %0 \n\t"
  108. "add %3, %1 \n\t"
  109. "add $16, %%"REG_a" \n\t"
  110. "jnz 1b \n\t"
  111. : "+r" (s1), "+r" (s2)
  112. : "r" (block+64), "r" ((x86_reg)stride)
  113. : "%"REG_a
  114. );
  115. }
  116. static int pix_sum16_mmx(uint8_t * pix, int line_size){
  117. const int h=16;
  118. int sum;
  119. x86_reg index= -line_size*h;
  120. __asm__ volatile(
  121. "pxor %%mm7, %%mm7 \n\t"
  122. "pxor %%mm6, %%mm6 \n\t"
  123. "1: \n\t"
  124. "movq (%2, %1), %%mm0 \n\t"
  125. "movq (%2, %1), %%mm1 \n\t"
  126. "movq 8(%2, %1), %%mm2 \n\t"
  127. "movq 8(%2, %1), %%mm3 \n\t"
  128. "punpcklbw %%mm7, %%mm0 \n\t"
  129. "punpckhbw %%mm7, %%mm1 \n\t"
  130. "punpcklbw %%mm7, %%mm2 \n\t"
  131. "punpckhbw %%mm7, %%mm3 \n\t"
  132. "paddw %%mm0, %%mm1 \n\t"
  133. "paddw %%mm2, %%mm3 \n\t"
  134. "paddw %%mm1, %%mm3 \n\t"
  135. "paddw %%mm3, %%mm6 \n\t"
  136. "add %3, %1 \n\t"
  137. " js 1b \n\t"
  138. "movq %%mm6, %%mm5 \n\t"
  139. "psrlq $32, %%mm6 \n\t"
  140. "paddw %%mm5, %%mm6 \n\t"
  141. "movq %%mm6, %%mm5 \n\t"
  142. "psrlq $16, %%mm6 \n\t"
  143. "paddw %%mm5, %%mm6 \n\t"
  144. "movd %%mm6, %0 \n\t"
  145. "andl $0xFFFF, %0 \n\t"
  146. : "=&r" (sum), "+r" (index)
  147. : "r" (pix - index), "r" ((x86_reg)line_size)
  148. );
  149. return sum;
  150. }
  151. static int pix_norm1_mmx(uint8_t *pix, int line_size) {
  152. int tmp;
  153. __asm__ volatile (
  154. "movl $16,%%ecx\n"
  155. "pxor %%mm0,%%mm0\n"
  156. "pxor %%mm7,%%mm7\n"
  157. "1:\n"
  158. "movq (%0),%%mm2\n" /* mm2 = pix[0-7] */
  159. "movq 8(%0),%%mm3\n" /* mm3 = pix[8-15] */
  160. "movq %%mm2,%%mm1\n" /* mm1 = mm2 = pix[0-7] */
  161. "punpckhbw %%mm0,%%mm1\n" /* mm1 = [pix4-7] */
  162. "punpcklbw %%mm0,%%mm2\n" /* mm2 = [pix0-3] */
  163. "movq %%mm3,%%mm4\n" /* mm4 = mm3 = pix[8-15] */
  164. "punpckhbw %%mm0,%%mm3\n" /* mm3 = [pix12-15] */
  165. "punpcklbw %%mm0,%%mm4\n" /* mm4 = [pix8-11] */
  166. "pmaddwd %%mm1,%%mm1\n" /* mm1 = (pix0^2+pix1^2,pix2^2+pix3^2) */
  167. "pmaddwd %%mm2,%%mm2\n" /* mm2 = (pix4^2+pix5^2,pix6^2+pix7^2) */
  168. "pmaddwd %%mm3,%%mm3\n"
  169. "pmaddwd %%mm4,%%mm4\n"
  170. "paddd %%mm1,%%mm2\n" /* mm2 = (pix0^2+pix1^2+pix4^2+pix5^2,
  171. pix2^2+pix3^2+pix6^2+pix7^2) */
  172. "paddd %%mm3,%%mm4\n"
  173. "paddd %%mm2,%%mm7\n"
  174. "add %2, %0\n"
  175. "paddd %%mm4,%%mm7\n"
  176. "dec %%ecx\n"
  177. "jnz 1b\n"
  178. "movq %%mm7,%%mm1\n"
  179. "psrlq $32, %%mm7\n" /* shift hi dword to lo */
  180. "paddd %%mm7,%%mm1\n"
  181. "movd %%mm1,%1\n"
  182. : "+r" (pix), "=r"(tmp) : "r" ((x86_reg)line_size) : "%ecx" );
  183. return tmp;
  184. }
  185. static int sse8_mmx(void *v, uint8_t * pix1, uint8_t * pix2, int line_size, int h) {
  186. int tmp;
  187. __asm__ volatile (
  188. "movl %4,%%ecx\n"
  189. "shr $1,%%ecx\n"
  190. "pxor %%mm0,%%mm0\n" /* mm0 = 0 */
  191. "pxor %%mm7,%%mm7\n" /* mm7 holds the sum */
  192. "1:\n"
  193. "movq (%0),%%mm1\n" /* mm1 = pix1[0][0-7] */
  194. "movq (%1),%%mm2\n" /* mm2 = pix2[0][0-7] */
  195. "movq (%0,%3),%%mm3\n" /* mm3 = pix1[1][0-7] */
  196. "movq (%1,%3),%%mm4\n" /* mm4 = pix2[1][0-7] */
  197. /* todo: mm1-mm2, mm3-mm4 */
  198. /* algo: subtract mm1 from mm2 with saturation and vice versa */
  199. /* OR the results to get absolute difference */
  200. "movq %%mm1,%%mm5\n"
  201. "movq %%mm3,%%mm6\n"
  202. "psubusb %%mm2,%%mm1\n"
  203. "psubusb %%mm4,%%mm3\n"
  204. "psubusb %%mm5,%%mm2\n"
  205. "psubusb %%mm6,%%mm4\n"
  206. "por %%mm1,%%mm2\n"
  207. "por %%mm3,%%mm4\n"
  208. /* now convert to 16-bit vectors so we can square them */
  209. "movq %%mm2,%%mm1\n"
  210. "movq %%mm4,%%mm3\n"
  211. "punpckhbw %%mm0,%%mm2\n"
  212. "punpckhbw %%mm0,%%mm4\n"
  213. "punpcklbw %%mm0,%%mm1\n" /* mm1 now spread over (mm1,mm2) */
  214. "punpcklbw %%mm0,%%mm3\n" /* mm4 now spread over (mm3,mm4) */
  215. "pmaddwd %%mm2,%%mm2\n"
  216. "pmaddwd %%mm4,%%mm4\n"
  217. "pmaddwd %%mm1,%%mm1\n"
  218. "pmaddwd %%mm3,%%mm3\n"
  219. "lea (%0,%3,2), %0\n" /* pix1 += 2*line_size */
  220. "lea (%1,%3,2), %1\n" /* pix2 += 2*line_size */
  221. "paddd %%mm2,%%mm1\n"
  222. "paddd %%mm4,%%mm3\n"
  223. "paddd %%mm1,%%mm7\n"
  224. "paddd %%mm3,%%mm7\n"
  225. "decl %%ecx\n"
  226. "jnz 1b\n"
  227. "movq %%mm7,%%mm1\n"
  228. "psrlq $32, %%mm7\n" /* shift hi dword to lo */
  229. "paddd %%mm7,%%mm1\n"
  230. "movd %%mm1,%2\n"
  231. : "+r" (pix1), "+r" (pix2), "=r"(tmp)
  232. : "r" ((x86_reg)line_size) , "m" (h)
  233. : "%ecx");
  234. return tmp;
  235. }
  236. static int sse16_mmx(void *v, uint8_t * pix1, uint8_t * pix2, int line_size, int h) {
  237. int tmp;
  238. __asm__ volatile (
  239. "movl %4,%%ecx\n"
  240. "pxor %%mm0,%%mm0\n" /* mm0 = 0 */
  241. "pxor %%mm7,%%mm7\n" /* mm7 holds the sum */
  242. "1:\n"
  243. "movq (%0),%%mm1\n" /* mm1 = pix1[0-7] */
  244. "movq (%1),%%mm2\n" /* mm2 = pix2[0-7] */
  245. "movq 8(%0),%%mm3\n" /* mm3 = pix1[8-15] */
  246. "movq 8(%1),%%mm4\n" /* mm4 = pix2[8-15] */
  247. /* todo: mm1-mm2, mm3-mm4 */
  248. /* algo: subtract mm1 from mm2 with saturation and vice versa */
  249. /* OR the results to get absolute difference */
  250. "movq %%mm1,%%mm5\n"
  251. "movq %%mm3,%%mm6\n"
  252. "psubusb %%mm2,%%mm1\n"
  253. "psubusb %%mm4,%%mm3\n"
  254. "psubusb %%mm5,%%mm2\n"
  255. "psubusb %%mm6,%%mm4\n"
  256. "por %%mm1,%%mm2\n"
  257. "por %%mm3,%%mm4\n"
  258. /* now convert to 16-bit vectors so we can square them */
  259. "movq %%mm2,%%mm1\n"
  260. "movq %%mm4,%%mm3\n"
  261. "punpckhbw %%mm0,%%mm2\n"
  262. "punpckhbw %%mm0,%%mm4\n"
  263. "punpcklbw %%mm0,%%mm1\n" /* mm1 now spread over (mm1,mm2) */
  264. "punpcklbw %%mm0,%%mm3\n" /* mm4 now spread over (mm3,mm4) */
  265. "pmaddwd %%mm2,%%mm2\n"
  266. "pmaddwd %%mm4,%%mm4\n"
  267. "pmaddwd %%mm1,%%mm1\n"
  268. "pmaddwd %%mm3,%%mm3\n"
  269. "add %3,%0\n"
  270. "add %3,%1\n"
  271. "paddd %%mm2,%%mm1\n"
  272. "paddd %%mm4,%%mm3\n"
  273. "paddd %%mm1,%%mm7\n"
  274. "paddd %%mm3,%%mm7\n"
  275. "decl %%ecx\n"
  276. "jnz 1b\n"
  277. "movq %%mm7,%%mm1\n"
  278. "psrlq $32, %%mm7\n" /* shift hi dword to lo */
  279. "paddd %%mm7,%%mm1\n"
  280. "movd %%mm1,%2\n"
  281. : "+r" (pix1), "+r" (pix2), "=r"(tmp)
  282. : "r" ((x86_reg)line_size) , "m" (h)
  283. : "%ecx");
  284. return tmp;
  285. }
  286. static int sse16_sse2(void *v, uint8_t * pix1, uint8_t * pix2, int line_size, int h) {
  287. int tmp;
  288. __asm__ volatile (
  289. "shr $1,%2\n"
  290. "pxor %%xmm0,%%xmm0\n" /* mm0 = 0 */
  291. "pxor %%xmm7,%%xmm7\n" /* mm7 holds the sum */
  292. "1:\n"
  293. "movdqu (%0),%%xmm1\n" /* mm1 = pix1[0][0-15] */
  294. "movdqu (%1),%%xmm2\n" /* mm2 = pix2[0][0-15] */
  295. "movdqu (%0,%4),%%xmm3\n" /* mm3 = pix1[1][0-15] */
  296. "movdqu (%1,%4),%%xmm4\n" /* mm4 = pix2[1][0-15] */
  297. /* todo: mm1-mm2, mm3-mm4 */
  298. /* algo: subtract mm1 from mm2 with saturation and vice versa */
  299. /* OR the results to get absolute difference */
  300. "movdqa %%xmm1,%%xmm5\n"
  301. "movdqa %%xmm3,%%xmm6\n"
  302. "psubusb %%xmm2,%%xmm1\n"
  303. "psubusb %%xmm4,%%xmm3\n"
  304. "psubusb %%xmm5,%%xmm2\n"
  305. "psubusb %%xmm6,%%xmm4\n"
  306. "por %%xmm1,%%xmm2\n"
  307. "por %%xmm3,%%xmm4\n"
  308. /* now convert to 16-bit vectors so we can square them */
  309. "movdqa %%xmm2,%%xmm1\n"
  310. "movdqa %%xmm4,%%xmm3\n"
  311. "punpckhbw %%xmm0,%%xmm2\n"
  312. "punpckhbw %%xmm0,%%xmm4\n"
  313. "punpcklbw %%xmm0,%%xmm1\n" /* mm1 now spread over (mm1,mm2) */
  314. "punpcklbw %%xmm0,%%xmm3\n" /* mm4 now spread over (mm3,mm4) */
  315. "pmaddwd %%xmm2,%%xmm2\n"
  316. "pmaddwd %%xmm4,%%xmm4\n"
  317. "pmaddwd %%xmm1,%%xmm1\n"
  318. "pmaddwd %%xmm3,%%xmm3\n"
  319. "lea (%0,%4,2), %0\n" /* pix1 += 2*line_size */
  320. "lea (%1,%4,2), %1\n" /* pix2 += 2*line_size */
  321. "paddd %%xmm2,%%xmm1\n"
  322. "paddd %%xmm4,%%xmm3\n"
  323. "paddd %%xmm1,%%xmm7\n"
  324. "paddd %%xmm3,%%xmm7\n"
  325. "decl %2\n"
  326. "jnz 1b\n"
  327. "movdqa %%xmm7,%%xmm1\n"
  328. "psrldq $8, %%xmm7\n" /* shift hi qword to lo */
  329. "paddd %%xmm1,%%xmm7\n"
  330. "movdqa %%xmm7,%%xmm1\n"
  331. "psrldq $4, %%xmm7\n" /* shift hi dword to lo */
  332. "paddd %%xmm1,%%xmm7\n"
  333. "movd %%xmm7,%3\n"
  334. : "+r" (pix1), "+r" (pix2), "+r"(h), "=r"(tmp)
  335. : "r" ((x86_reg)line_size));
  336. return tmp;
  337. }
  338. static int hf_noise8_mmx(uint8_t * pix1, int line_size, int h) {
  339. int tmp;
  340. __asm__ volatile (
  341. "movl %3,%%ecx\n"
  342. "pxor %%mm7,%%mm7\n"
  343. "pxor %%mm6,%%mm6\n"
  344. "movq (%0),%%mm0\n"
  345. "movq %%mm0, %%mm1\n"
  346. "psllq $8, %%mm0\n"
  347. "psrlq $8, %%mm1\n"
  348. "psrlq $8, %%mm0\n"
  349. "movq %%mm0, %%mm2\n"
  350. "movq %%mm1, %%mm3\n"
  351. "punpcklbw %%mm7,%%mm0\n"
  352. "punpcklbw %%mm7,%%mm1\n"
  353. "punpckhbw %%mm7,%%mm2\n"
  354. "punpckhbw %%mm7,%%mm3\n"
  355. "psubw %%mm1, %%mm0\n"
  356. "psubw %%mm3, %%mm2\n"
  357. "add %2,%0\n"
  358. "movq (%0),%%mm4\n"
  359. "movq %%mm4, %%mm1\n"
  360. "psllq $8, %%mm4\n"
  361. "psrlq $8, %%mm1\n"
  362. "psrlq $8, %%mm4\n"
  363. "movq %%mm4, %%mm5\n"
  364. "movq %%mm1, %%mm3\n"
  365. "punpcklbw %%mm7,%%mm4\n"
  366. "punpcklbw %%mm7,%%mm1\n"
  367. "punpckhbw %%mm7,%%mm5\n"
  368. "punpckhbw %%mm7,%%mm3\n"
  369. "psubw %%mm1, %%mm4\n"
  370. "psubw %%mm3, %%mm5\n"
  371. "psubw %%mm4, %%mm0\n"
  372. "psubw %%mm5, %%mm2\n"
  373. "pxor %%mm3, %%mm3\n"
  374. "pxor %%mm1, %%mm1\n"
  375. "pcmpgtw %%mm0, %%mm3\n\t"
  376. "pcmpgtw %%mm2, %%mm1\n\t"
  377. "pxor %%mm3, %%mm0\n"
  378. "pxor %%mm1, %%mm2\n"
  379. "psubw %%mm3, %%mm0\n"
  380. "psubw %%mm1, %%mm2\n"
  381. "paddw %%mm0, %%mm2\n"
  382. "paddw %%mm2, %%mm6\n"
  383. "add %2,%0\n"
  384. "1:\n"
  385. "movq (%0),%%mm0\n"
  386. "movq %%mm0, %%mm1\n"
  387. "psllq $8, %%mm0\n"
  388. "psrlq $8, %%mm1\n"
  389. "psrlq $8, %%mm0\n"
  390. "movq %%mm0, %%mm2\n"
  391. "movq %%mm1, %%mm3\n"
  392. "punpcklbw %%mm7,%%mm0\n"
  393. "punpcklbw %%mm7,%%mm1\n"
  394. "punpckhbw %%mm7,%%mm2\n"
  395. "punpckhbw %%mm7,%%mm3\n"
  396. "psubw %%mm1, %%mm0\n"
  397. "psubw %%mm3, %%mm2\n"
  398. "psubw %%mm0, %%mm4\n"
  399. "psubw %%mm2, %%mm5\n"
  400. "pxor %%mm3, %%mm3\n"
  401. "pxor %%mm1, %%mm1\n"
  402. "pcmpgtw %%mm4, %%mm3\n\t"
  403. "pcmpgtw %%mm5, %%mm1\n\t"
  404. "pxor %%mm3, %%mm4\n"
  405. "pxor %%mm1, %%mm5\n"
  406. "psubw %%mm3, %%mm4\n"
  407. "psubw %%mm1, %%mm5\n"
  408. "paddw %%mm4, %%mm5\n"
  409. "paddw %%mm5, %%mm6\n"
  410. "add %2,%0\n"
  411. "movq (%0),%%mm4\n"
  412. "movq %%mm4, %%mm1\n"
  413. "psllq $8, %%mm4\n"
  414. "psrlq $8, %%mm1\n"
  415. "psrlq $8, %%mm4\n"
  416. "movq %%mm4, %%mm5\n"
  417. "movq %%mm1, %%mm3\n"
  418. "punpcklbw %%mm7,%%mm4\n"
  419. "punpcklbw %%mm7,%%mm1\n"
  420. "punpckhbw %%mm7,%%mm5\n"
  421. "punpckhbw %%mm7,%%mm3\n"
  422. "psubw %%mm1, %%mm4\n"
  423. "psubw %%mm3, %%mm5\n"
  424. "psubw %%mm4, %%mm0\n"
  425. "psubw %%mm5, %%mm2\n"
  426. "pxor %%mm3, %%mm3\n"
  427. "pxor %%mm1, %%mm1\n"
  428. "pcmpgtw %%mm0, %%mm3\n\t"
  429. "pcmpgtw %%mm2, %%mm1\n\t"
  430. "pxor %%mm3, %%mm0\n"
  431. "pxor %%mm1, %%mm2\n"
  432. "psubw %%mm3, %%mm0\n"
  433. "psubw %%mm1, %%mm2\n"
  434. "paddw %%mm0, %%mm2\n"
  435. "paddw %%mm2, %%mm6\n"
  436. "add %2,%0\n"
  437. "subl $2, %%ecx\n"
  438. " jnz 1b\n"
  439. "movq %%mm6, %%mm0\n"
  440. "punpcklwd %%mm7,%%mm0\n"
  441. "punpckhwd %%mm7,%%mm6\n"
  442. "paddd %%mm0, %%mm6\n"
  443. "movq %%mm6,%%mm0\n"
  444. "psrlq $32, %%mm6\n"
  445. "paddd %%mm6,%%mm0\n"
  446. "movd %%mm0,%1\n"
  447. : "+r" (pix1), "=r"(tmp)
  448. : "r" ((x86_reg)line_size) , "g" (h-2)
  449. : "%ecx");
  450. return tmp;
  451. }
  452. static int hf_noise16_mmx(uint8_t * pix1, int line_size, int h) {
  453. int tmp;
  454. uint8_t * pix= pix1;
  455. __asm__ volatile (
  456. "movl %3,%%ecx\n"
  457. "pxor %%mm7,%%mm7\n"
  458. "pxor %%mm6,%%mm6\n"
  459. "movq (%0),%%mm0\n"
  460. "movq 1(%0),%%mm1\n"
  461. "movq %%mm0, %%mm2\n"
  462. "movq %%mm1, %%mm3\n"
  463. "punpcklbw %%mm7,%%mm0\n"
  464. "punpcklbw %%mm7,%%mm1\n"
  465. "punpckhbw %%mm7,%%mm2\n"
  466. "punpckhbw %%mm7,%%mm3\n"
  467. "psubw %%mm1, %%mm0\n"
  468. "psubw %%mm3, %%mm2\n"
  469. "add %2,%0\n"
  470. "movq (%0),%%mm4\n"
  471. "movq 1(%0),%%mm1\n"
  472. "movq %%mm4, %%mm5\n"
  473. "movq %%mm1, %%mm3\n"
  474. "punpcklbw %%mm7,%%mm4\n"
  475. "punpcklbw %%mm7,%%mm1\n"
  476. "punpckhbw %%mm7,%%mm5\n"
  477. "punpckhbw %%mm7,%%mm3\n"
  478. "psubw %%mm1, %%mm4\n"
  479. "psubw %%mm3, %%mm5\n"
  480. "psubw %%mm4, %%mm0\n"
  481. "psubw %%mm5, %%mm2\n"
  482. "pxor %%mm3, %%mm3\n"
  483. "pxor %%mm1, %%mm1\n"
  484. "pcmpgtw %%mm0, %%mm3\n\t"
  485. "pcmpgtw %%mm2, %%mm1\n\t"
  486. "pxor %%mm3, %%mm0\n"
  487. "pxor %%mm1, %%mm2\n"
  488. "psubw %%mm3, %%mm0\n"
  489. "psubw %%mm1, %%mm2\n"
  490. "paddw %%mm0, %%mm2\n"
  491. "paddw %%mm2, %%mm6\n"
  492. "add %2,%0\n"
  493. "1:\n"
  494. "movq (%0),%%mm0\n"
  495. "movq 1(%0),%%mm1\n"
  496. "movq %%mm0, %%mm2\n"
  497. "movq %%mm1, %%mm3\n"
  498. "punpcklbw %%mm7,%%mm0\n"
  499. "punpcklbw %%mm7,%%mm1\n"
  500. "punpckhbw %%mm7,%%mm2\n"
  501. "punpckhbw %%mm7,%%mm3\n"
  502. "psubw %%mm1, %%mm0\n"
  503. "psubw %%mm3, %%mm2\n"
  504. "psubw %%mm0, %%mm4\n"
  505. "psubw %%mm2, %%mm5\n"
  506. "pxor %%mm3, %%mm3\n"
  507. "pxor %%mm1, %%mm1\n"
  508. "pcmpgtw %%mm4, %%mm3\n\t"
  509. "pcmpgtw %%mm5, %%mm1\n\t"
  510. "pxor %%mm3, %%mm4\n"
  511. "pxor %%mm1, %%mm5\n"
  512. "psubw %%mm3, %%mm4\n"
  513. "psubw %%mm1, %%mm5\n"
  514. "paddw %%mm4, %%mm5\n"
  515. "paddw %%mm5, %%mm6\n"
  516. "add %2,%0\n"
  517. "movq (%0),%%mm4\n"
  518. "movq 1(%0),%%mm1\n"
  519. "movq %%mm4, %%mm5\n"
  520. "movq %%mm1, %%mm3\n"
  521. "punpcklbw %%mm7,%%mm4\n"
  522. "punpcklbw %%mm7,%%mm1\n"
  523. "punpckhbw %%mm7,%%mm5\n"
  524. "punpckhbw %%mm7,%%mm3\n"
  525. "psubw %%mm1, %%mm4\n"
  526. "psubw %%mm3, %%mm5\n"
  527. "psubw %%mm4, %%mm0\n"
  528. "psubw %%mm5, %%mm2\n"
  529. "pxor %%mm3, %%mm3\n"
  530. "pxor %%mm1, %%mm1\n"
  531. "pcmpgtw %%mm0, %%mm3\n\t"
  532. "pcmpgtw %%mm2, %%mm1\n\t"
  533. "pxor %%mm3, %%mm0\n"
  534. "pxor %%mm1, %%mm2\n"
  535. "psubw %%mm3, %%mm0\n"
  536. "psubw %%mm1, %%mm2\n"
  537. "paddw %%mm0, %%mm2\n"
  538. "paddw %%mm2, %%mm6\n"
  539. "add %2,%0\n"
  540. "subl $2, %%ecx\n"
  541. " jnz 1b\n"
  542. "movq %%mm6, %%mm0\n"
  543. "punpcklwd %%mm7,%%mm0\n"
  544. "punpckhwd %%mm7,%%mm6\n"
  545. "paddd %%mm0, %%mm6\n"
  546. "movq %%mm6,%%mm0\n"
  547. "psrlq $32, %%mm6\n"
  548. "paddd %%mm6,%%mm0\n"
  549. "movd %%mm0,%1\n"
  550. : "+r" (pix1), "=r"(tmp)
  551. : "r" ((x86_reg)line_size) , "g" (h-2)
  552. : "%ecx");
  553. return tmp + hf_noise8_mmx(pix+8, line_size, h);
  554. }
  555. static int nsse16_mmx(void *p, uint8_t * pix1, uint8_t * pix2, int line_size, int h) {
  556. MpegEncContext *c = p;
  557. int score1, score2;
  558. if(c) score1 = c->dsp.sse[0](c, pix1, pix2, line_size, h);
  559. else score1 = sse16_mmx(c, pix1, pix2, line_size, h);
  560. score2= hf_noise16_mmx(pix1, line_size, h) - hf_noise16_mmx(pix2, line_size, h);
  561. if(c) return score1 + FFABS(score2)*c->avctx->nsse_weight;
  562. else return score1 + FFABS(score2)*8;
  563. }
  564. static int nsse8_mmx(void *p, uint8_t * pix1, uint8_t * pix2, int line_size, int h) {
  565. MpegEncContext *c = p;
  566. int score1= sse8_mmx(c, pix1, pix2, line_size, h);
  567. int score2= hf_noise8_mmx(pix1, line_size, h) - hf_noise8_mmx(pix2, line_size, h);
  568. if(c) return score1 + FFABS(score2)*c->avctx->nsse_weight;
  569. else return score1 + FFABS(score2)*8;
  570. }
  571. static int vsad_intra16_mmx(void *v, uint8_t * pix, uint8_t * dummy, int line_size, int h) {
  572. int tmp;
  573. assert( (((int)pix) & 7) == 0);
  574. assert((line_size &7) ==0);
  575. #define SUM(in0, in1, out0, out1) \
  576. "movq (%0), %%mm2\n"\
  577. "movq 8(%0), %%mm3\n"\
  578. "add %2,%0\n"\
  579. "movq %%mm2, " #out0 "\n"\
  580. "movq %%mm3, " #out1 "\n"\
  581. "psubusb " #in0 ", %%mm2\n"\
  582. "psubusb " #in1 ", %%mm3\n"\
  583. "psubusb " #out0 ", " #in0 "\n"\
  584. "psubusb " #out1 ", " #in1 "\n"\
  585. "por %%mm2, " #in0 "\n"\
  586. "por %%mm3, " #in1 "\n"\
  587. "movq " #in0 ", %%mm2\n"\
  588. "movq " #in1 ", %%mm3\n"\
  589. "punpcklbw %%mm7, " #in0 "\n"\
  590. "punpcklbw %%mm7, " #in1 "\n"\
  591. "punpckhbw %%mm7, %%mm2\n"\
  592. "punpckhbw %%mm7, %%mm3\n"\
  593. "paddw " #in1 ", " #in0 "\n"\
  594. "paddw %%mm3, %%mm2\n"\
  595. "paddw %%mm2, " #in0 "\n"\
  596. "paddw " #in0 ", %%mm6\n"
  597. __asm__ volatile (
  598. "movl %3,%%ecx\n"
  599. "pxor %%mm6,%%mm6\n"
  600. "pxor %%mm7,%%mm7\n"
  601. "movq (%0),%%mm0\n"
  602. "movq 8(%0),%%mm1\n"
  603. "add %2,%0\n"
  604. "jmp 2f\n"
  605. "1:\n"
  606. SUM(%%mm4, %%mm5, %%mm0, %%mm1)
  607. "2:\n"
  608. SUM(%%mm0, %%mm1, %%mm4, %%mm5)
  609. "subl $2, %%ecx\n"
  610. "jnz 1b\n"
  611. "movq %%mm6,%%mm0\n"
  612. "psrlq $32, %%mm6\n"
  613. "paddw %%mm6,%%mm0\n"
  614. "movq %%mm0,%%mm6\n"
  615. "psrlq $16, %%mm0\n"
  616. "paddw %%mm6,%%mm0\n"
  617. "movd %%mm0,%1\n"
  618. : "+r" (pix), "=r"(tmp)
  619. : "r" ((x86_reg)line_size) , "m" (h)
  620. : "%ecx");
  621. return tmp & 0xFFFF;
  622. }
  623. #undef SUM
  624. static int vsad_intra16_mmx2(void *v, uint8_t * pix, uint8_t * dummy, int line_size, int h) {
  625. int tmp;
  626. assert( (((int)pix) & 7) == 0);
  627. assert((line_size &7) ==0);
  628. #define SUM(in0, in1, out0, out1) \
  629. "movq (%0), " #out0 "\n"\
  630. "movq 8(%0), " #out1 "\n"\
  631. "add %2,%0\n"\
  632. "psadbw " #out0 ", " #in0 "\n"\
  633. "psadbw " #out1 ", " #in1 "\n"\
  634. "paddw " #in1 ", " #in0 "\n"\
  635. "paddw " #in0 ", %%mm6\n"
  636. __asm__ volatile (
  637. "movl %3,%%ecx\n"
  638. "pxor %%mm6,%%mm6\n"
  639. "pxor %%mm7,%%mm7\n"
  640. "movq (%0),%%mm0\n"
  641. "movq 8(%0),%%mm1\n"
  642. "add %2,%0\n"
  643. "jmp 2f\n"
  644. "1:\n"
  645. SUM(%%mm4, %%mm5, %%mm0, %%mm1)
  646. "2:\n"
  647. SUM(%%mm0, %%mm1, %%mm4, %%mm5)
  648. "subl $2, %%ecx\n"
  649. "jnz 1b\n"
  650. "movd %%mm6,%1\n"
  651. : "+r" (pix), "=r"(tmp)
  652. : "r" ((x86_reg)line_size) , "m" (h)
  653. : "%ecx");
  654. return tmp;
  655. }
  656. #undef SUM
  657. static int vsad16_mmx(void *v, uint8_t * pix1, uint8_t * pix2, int line_size, int h) {
  658. int tmp;
  659. assert( (((int)pix1) & 7) == 0);
  660. assert( (((int)pix2) & 7) == 0);
  661. assert((line_size &7) ==0);
  662. #define SUM(in0, in1, out0, out1) \
  663. "movq (%0),%%mm2\n"\
  664. "movq (%1)," #out0 "\n"\
  665. "movq 8(%0),%%mm3\n"\
  666. "movq 8(%1)," #out1 "\n"\
  667. "add %3,%0\n"\
  668. "add %3,%1\n"\
  669. "psubb " #out0 ", %%mm2\n"\
  670. "psubb " #out1 ", %%mm3\n"\
  671. "pxor %%mm7, %%mm2\n"\
  672. "pxor %%mm7, %%mm3\n"\
  673. "movq %%mm2, " #out0 "\n"\
  674. "movq %%mm3, " #out1 "\n"\
  675. "psubusb " #in0 ", %%mm2\n"\
  676. "psubusb " #in1 ", %%mm3\n"\
  677. "psubusb " #out0 ", " #in0 "\n"\
  678. "psubusb " #out1 ", " #in1 "\n"\
  679. "por %%mm2, " #in0 "\n"\
  680. "por %%mm3, " #in1 "\n"\
  681. "movq " #in0 ", %%mm2\n"\
  682. "movq " #in1 ", %%mm3\n"\
  683. "punpcklbw %%mm7, " #in0 "\n"\
  684. "punpcklbw %%mm7, " #in1 "\n"\
  685. "punpckhbw %%mm7, %%mm2\n"\
  686. "punpckhbw %%mm7, %%mm3\n"\
  687. "paddw " #in1 ", " #in0 "\n"\
  688. "paddw %%mm3, %%mm2\n"\
  689. "paddw %%mm2, " #in0 "\n"\
  690. "paddw " #in0 ", %%mm6\n"
  691. __asm__ volatile (
  692. "movl %4,%%ecx\n"
  693. "pxor %%mm6,%%mm6\n"
  694. "pcmpeqw %%mm7,%%mm7\n"
  695. "psllw $15, %%mm7\n"
  696. "packsswb %%mm7, %%mm7\n"
  697. "movq (%0),%%mm0\n"
  698. "movq (%1),%%mm2\n"
  699. "movq 8(%0),%%mm1\n"
  700. "movq 8(%1),%%mm3\n"
  701. "add %3,%0\n"
  702. "add %3,%1\n"
  703. "psubb %%mm2, %%mm0\n"
  704. "psubb %%mm3, %%mm1\n"
  705. "pxor %%mm7, %%mm0\n"
  706. "pxor %%mm7, %%mm1\n"
  707. "jmp 2f\n"
  708. "1:\n"
  709. SUM(%%mm4, %%mm5, %%mm0, %%mm1)
  710. "2:\n"
  711. SUM(%%mm0, %%mm1, %%mm4, %%mm5)
  712. "subl $2, %%ecx\n"
  713. "jnz 1b\n"
  714. "movq %%mm6,%%mm0\n"
  715. "psrlq $32, %%mm6\n"
  716. "paddw %%mm6,%%mm0\n"
  717. "movq %%mm0,%%mm6\n"
  718. "psrlq $16, %%mm0\n"
  719. "paddw %%mm6,%%mm0\n"
  720. "movd %%mm0,%2\n"
  721. : "+r" (pix1), "+r" (pix2), "=r"(tmp)
  722. : "r" ((x86_reg)line_size) , "m" (h)
  723. : "%ecx");
  724. return tmp & 0x7FFF;
  725. }
  726. #undef SUM
  727. static int vsad16_mmx2(void *v, uint8_t * pix1, uint8_t * pix2, int line_size, int h) {
  728. int tmp;
  729. assert( (((int)pix1) & 7) == 0);
  730. assert( (((int)pix2) & 7) == 0);
  731. assert((line_size &7) ==0);
  732. #define SUM(in0, in1, out0, out1) \
  733. "movq (%0)," #out0 "\n"\
  734. "movq (%1),%%mm2\n"\
  735. "movq 8(%0)," #out1 "\n"\
  736. "movq 8(%1),%%mm3\n"\
  737. "add %3,%0\n"\
  738. "add %3,%1\n"\
  739. "psubb %%mm2, " #out0 "\n"\
  740. "psubb %%mm3, " #out1 "\n"\
  741. "pxor %%mm7, " #out0 "\n"\
  742. "pxor %%mm7, " #out1 "\n"\
  743. "psadbw " #out0 ", " #in0 "\n"\
  744. "psadbw " #out1 ", " #in1 "\n"\
  745. "paddw " #in1 ", " #in0 "\n"\
  746. "paddw " #in0 ", %%mm6\n"
  747. __asm__ volatile (
  748. "movl %4,%%ecx\n"
  749. "pxor %%mm6,%%mm6\n"
  750. "pcmpeqw %%mm7,%%mm7\n"
  751. "psllw $15, %%mm7\n"
  752. "packsswb %%mm7, %%mm7\n"
  753. "movq (%0),%%mm0\n"
  754. "movq (%1),%%mm2\n"
  755. "movq 8(%0),%%mm1\n"
  756. "movq 8(%1),%%mm3\n"
  757. "add %3,%0\n"
  758. "add %3,%1\n"
  759. "psubb %%mm2, %%mm0\n"
  760. "psubb %%mm3, %%mm1\n"
  761. "pxor %%mm7, %%mm0\n"
  762. "pxor %%mm7, %%mm1\n"
  763. "jmp 2f\n"
  764. "1:\n"
  765. SUM(%%mm4, %%mm5, %%mm0, %%mm1)
  766. "2:\n"
  767. SUM(%%mm0, %%mm1, %%mm4, %%mm5)
  768. "subl $2, %%ecx\n"
  769. "jnz 1b\n"
  770. "movd %%mm6,%2\n"
  771. : "+r" (pix1), "+r" (pix2), "=r"(tmp)
  772. : "r" ((x86_reg)line_size) , "m" (h)
  773. : "%ecx");
  774. return tmp;
  775. }
  776. #undef SUM
  777. static void diff_bytes_mmx(uint8_t *dst, uint8_t *src1, uint8_t *src2, int w){
  778. x86_reg i=0;
  779. __asm__ volatile(
  780. "1: \n\t"
  781. "movq (%2, %0), %%mm0 \n\t"
  782. "movq (%1, %0), %%mm1 \n\t"
  783. "psubb %%mm0, %%mm1 \n\t"
  784. "movq %%mm1, (%3, %0) \n\t"
  785. "movq 8(%2, %0), %%mm0 \n\t"
  786. "movq 8(%1, %0), %%mm1 \n\t"
  787. "psubb %%mm0, %%mm1 \n\t"
  788. "movq %%mm1, 8(%3, %0) \n\t"
  789. "add $16, %0 \n\t"
  790. "cmp %4, %0 \n\t"
  791. " jb 1b \n\t"
  792. : "+r" (i)
  793. : "r"(src1), "r"(src2), "r"(dst), "r"((x86_reg)w-15)
  794. );
  795. for(; i<w; i++)
  796. dst[i+0] = src1[i+0]-src2[i+0];
  797. }
  798. static void sub_hfyu_median_prediction_mmx2(uint8_t *dst, uint8_t *src1, uint8_t *src2, int w, int *left, int *left_top){
  799. x86_reg i=0;
  800. uint8_t l, lt;
  801. __asm__ volatile(
  802. "1: \n\t"
  803. "movq -1(%1, %0), %%mm0 \n\t" // LT
  804. "movq (%1, %0), %%mm1 \n\t" // T
  805. "movq -1(%2, %0), %%mm2 \n\t" // L
  806. "movq (%2, %0), %%mm3 \n\t" // X
  807. "movq %%mm2, %%mm4 \n\t" // L
  808. "psubb %%mm0, %%mm2 \n\t"
  809. "paddb %%mm1, %%mm2 \n\t" // L + T - LT
  810. "movq %%mm4, %%mm5 \n\t" // L
  811. "pmaxub %%mm1, %%mm4 \n\t" // max(T, L)
  812. "pminub %%mm5, %%mm1 \n\t" // min(T, L)
  813. "pminub %%mm2, %%mm4 \n\t"
  814. "pmaxub %%mm1, %%mm4 \n\t"
  815. "psubb %%mm4, %%mm3 \n\t" // dst - pred
  816. "movq %%mm3, (%3, %0) \n\t"
  817. "add $8, %0 \n\t"
  818. "cmp %4, %0 \n\t"
  819. " jb 1b \n\t"
  820. : "+r" (i)
  821. : "r"(src1), "r"(src2), "r"(dst), "r"((x86_reg)w)
  822. );
  823. l= *left;
  824. lt= *left_top;
  825. dst[0]= src2[0] - mid_pred(l, src1[0], (l + src1[0] - lt)&0xFF);
  826. *left_top= src1[w-1];
  827. *left = src2[w-1];
  828. }
  829. #define DIFF_PIXELS_1(m,a,t,p1,p2)\
  830. "mov"#m" "#p1", "#a" \n\t"\
  831. "mov"#m" "#p2", "#t" \n\t"\
  832. "punpcklbw "#a", "#t" \n\t"\
  833. "punpcklbw "#a", "#a" \n\t"\
  834. "psubw "#t", "#a" \n\t"\
  835. #define DIFF_PIXELS_8(m0,m1,mm,p1,p2,stride,temp) {\
  836. uint8_t *p1b=p1, *p2b=p2;\
  837. __asm__ volatile(\
  838. DIFF_PIXELS_1(m0, mm##0, mm##7, (%1), (%2))\
  839. DIFF_PIXELS_1(m0, mm##1, mm##7, (%1,%3), (%2,%3))\
  840. DIFF_PIXELS_1(m0, mm##2, mm##7, (%1,%3,2), (%2,%3,2))\
  841. "add %4, %1 \n\t"\
  842. "add %4, %2 \n\t"\
  843. DIFF_PIXELS_1(m0, mm##3, mm##7, (%1), (%2))\
  844. DIFF_PIXELS_1(m0, mm##4, mm##7, (%1,%3), (%2,%3))\
  845. DIFF_PIXELS_1(m0, mm##5, mm##7, (%1,%3,2), (%2,%3,2))\
  846. DIFF_PIXELS_1(m0, mm##6, mm##7, (%1,%4), (%2,%4))\
  847. "mov"#m1" "#mm"0, %0 \n\t"\
  848. DIFF_PIXELS_1(m0, mm##7, mm##0, (%1,%3,4), (%2,%3,4))\
  849. "mov"#m1" %0, "#mm"0 \n\t"\
  850. : "+m"(temp), "+r"(p1b), "+r"(p2b)\
  851. : "r"((x86_reg)stride), "r"((x86_reg)stride*3)\
  852. );\
  853. }
  854. //the "+m"(temp) is needed as gcc 2.95 sometimes fails to compile "=m"(temp)
  855. #define DIFF_PIXELS_4x8(p1,p2,stride,temp) DIFF_PIXELS_8(d, q, %%mm, p1, p2, stride, temp)
  856. #define DIFF_PIXELS_8x8(p1,p2,stride,temp) DIFF_PIXELS_8(q, dqa, %%xmm, p1, p2, stride, temp)
  857. #define LBUTTERFLY2(a1,b1,a2,b2)\
  858. "paddw " #b1 ", " #a1 " \n\t"\
  859. "paddw " #b2 ", " #a2 " \n\t"\
  860. "paddw " #b1 ", " #b1 " \n\t"\
  861. "paddw " #b2 ", " #b2 " \n\t"\
  862. "psubw " #a1 ", " #b1 " \n\t"\
  863. "psubw " #a2 ", " #b2 " \n\t"
  864. #define HADAMARD8(m0, m1, m2, m3, m4, m5, m6, m7)\
  865. LBUTTERFLY2(m0, m1, m2, m3)\
  866. LBUTTERFLY2(m4, m5, m6, m7)\
  867. LBUTTERFLY2(m0, m2, m1, m3)\
  868. LBUTTERFLY2(m4, m6, m5, m7)\
  869. LBUTTERFLY2(m0, m4, m1, m5)\
  870. LBUTTERFLY2(m2, m6, m3, m7)\
  871. #define HADAMARD48 HADAMARD8(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm6, %%mm7)
  872. #define MMABS_MMX(a,z)\
  873. "pxor " #z ", " #z " \n\t"\
  874. "pcmpgtw " #a ", " #z " \n\t"\
  875. "pxor " #z ", " #a " \n\t"\
  876. "psubw " #z ", " #a " \n\t"
  877. #define MMABS_MMX2(a,z)\
  878. "pxor " #z ", " #z " \n\t"\
  879. "psubw " #a ", " #z " \n\t"\
  880. "pmaxsw " #z ", " #a " \n\t"
  881. #define MMABS_SSSE3(a,z)\
  882. "pabsw " #a ", " #a " \n\t"
  883. #define MMABS_SUM(a,z, sum)\
  884. MMABS(a,z)\
  885. "paddusw " #a ", " #sum " \n\t"
  886. #define MMABS_SUM_8x8_NOSPILL\
  887. MMABS(%%xmm0, %%xmm8)\
  888. MMABS(%%xmm1, %%xmm9)\
  889. MMABS_SUM(%%xmm2, %%xmm8, %%xmm0)\
  890. MMABS_SUM(%%xmm3, %%xmm9, %%xmm1)\
  891. MMABS_SUM(%%xmm4, %%xmm8, %%xmm0)\
  892. MMABS_SUM(%%xmm5, %%xmm9, %%xmm1)\
  893. MMABS_SUM(%%xmm6, %%xmm8, %%xmm0)\
  894. MMABS_SUM(%%xmm7, %%xmm9, %%xmm1)\
  895. "paddusw %%xmm1, %%xmm0 \n\t"
  896. #ifdef ARCH_X86_64
  897. #define MMABS_SUM_8x8_SSE2 MMABS_SUM_8x8_NOSPILL
  898. #else
  899. #define MMABS_SUM_8x8_SSE2\
  900. "movdqa %%xmm7, (%1) \n\t"\
  901. MMABS(%%xmm0, %%xmm7)\
  902. MMABS(%%xmm1, %%xmm7)\
  903. MMABS_SUM(%%xmm2, %%xmm7, %%xmm0)\
  904. MMABS_SUM(%%xmm3, %%xmm7, %%xmm1)\
  905. MMABS_SUM(%%xmm4, %%xmm7, %%xmm0)\
  906. MMABS_SUM(%%xmm5, %%xmm7, %%xmm1)\
  907. MMABS_SUM(%%xmm6, %%xmm7, %%xmm0)\
  908. "movdqa (%1), %%xmm2 \n\t"\
  909. MMABS_SUM(%%xmm2, %%xmm7, %%xmm1)\
  910. "paddusw %%xmm1, %%xmm0 \n\t"
  911. #endif
  912. /* FIXME: HSUM_* saturates at 64k, while an 8x8 hadamard or dct block can get up to
  913. * about 100k on extreme inputs. But that's very unlikely to occur in natural video,
  914. * and it's even more unlikely to not have any alternative mvs/modes with lower cost. */
  915. #define HSUM_MMX(a, t, dst)\
  916. "movq "#a", "#t" \n\t"\
  917. "psrlq $32, "#a" \n\t"\
  918. "paddusw "#t", "#a" \n\t"\
  919. "movq "#a", "#t" \n\t"\
  920. "psrlq $16, "#a" \n\t"\
  921. "paddusw "#t", "#a" \n\t"\
  922. "movd "#a", "#dst" \n\t"\
  923. #define HSUM_MMX2(a, t, dst)\
  924. "pshufw $0x0E, "#a", "#t" \n\t"\
  925. "paddusw "#t", "#a" \n\t"\
  926. "pshufw $0x01, "#a", "#t" \n\t"\
  927. "paddusw "#t", "#a" \n\t"\
  928. "movd "#a", "#dst" \n\t"\
  929. #define HSUM_SSE2(a, t, dst)\
  930. "movhlps "#a", "#t" \n\t"\
  931. "paddusw "#t", "#a" \n\t"\
  932. "pshuflw $0x0E, "#a", "#t" \n\t"\
  933. "paddusw "#t", "#a" \n\t"\
  934. "pshuflw $0x01, "#a", "#t" \n\t"\
  935. "paddusw "#t", "#a" \n\t"\
  936. "movd "#a", "#dst" \n\t"\
  937. #define HADAMARD8_DIFF_MMX(cpu) \
  938. static int hadamard8_diff_##cpu(void *s, uint8_t *src1, uint8_t *src2, int stride, int h){\
  939. DECLARE_ALIGNED_8(uint64_t, temp[13]);\
  940. int sum;\
  941. \
  942. assert(h==8);\
  943. \
  944. DIFF_PIXELS_4x8(src1, src2, stride, temp[0]);\
  945. \
  946. __asm__ volatile(\
  947. HADAMARD48\
  948. \
  949. "movq %%mm7, 96(%1) \n\t"\
  950. \
  951. TRANSPOSE4(%%mm0, %%mm1, %%mm2, %%mm3, %%mm7)\
  952. STORE4(8, 0(%1), %%mm0, %%mm3, %%mm7, %%mm2)\
  953. \
  954. "movq 96(%1), %%mm7 \n\t"\
  955. TRANSPOSE4(%%mm4, %%mm5, %%mm6, %%mm7, %%mm0)\
  956. STORE4(8, 64(%1), %%mm4, %%mm7, %%mm0, %%mm6)\
  957. \
  958. : "=r" (sum)\
  959. : "r"(temp)\
  960. );\
  961. \
  962. DIFF_PIXELS_4x8(src1+4, src2+4, stride, temp[4]);\
  963. \
  964. __asm__ volatile(\
  965. HADAMARD48\
  966. \
  967. "movq %%mm7, 96(%1) \n\t"\
  968. \
  969. TRANSPOSE4(%%mm0, %%mm1, %%mm2, %%mm3, %%mm7)\
  970. STORE4(8, 32(%1), %%mm0, %%mm3, %%mm7, %%mm2)\
  971. \
  972. "movq 96(%1), %%mm7 \n\t"\
  973. TRANSPOSE4(%%mm4, %%mm5, %%mm6, %%mm7, %%mm0)\
  974. "movq %%mm7, %%mm5 \n\t"/*FIXME remove*/\
  975. "movq %%mm6, %%mm7 \n\t"\
  976. "movq %%mm0, %%mm6 \n\t"\
  977. \
  978. LOAD4(8, 64(%1), %%mm0, %%mm1, %%mm2, %%mm3)\
  979. \
  980. HADAMARD48\
  981. "movq %%mm7, 64(%1) \n\t"\
  982. MMABS(%%mm0, %%mm7)\
  983. MMABS(%%mm1, %%mm7)\
  984. MMABS_SUM(%%mm2, %%mm7, %%mm0)\
  985. MMABS_SUM(%%mm3, %%mm7, %%mm1)\
  986. MMABS_SUM(%%mm4, %%mm7, %%mm0)\
  987. MMABS_SUM(%%mm5, %%mm7, %%mm1)\
  988. MMABS_SUM(%%mm6, %%mm7, %%mm0)\
  989. "movq 64(%1), %%mm2 \n\t"\
  990. MMABS_SUM(%%mm2, %%mm7, %%mm1)\
  991. "paddusw %%mm1, %%mm0 \n\t"\
  992. "movq %%mm0, 64(%1) \n\t"\
  993. \
  994. LOAD4(8, 0(%1), %%mm0, %%mm1, %%mm2, %%mm3)\
  995. LOAD4(8, 32(%1), %%mm4, %%mm5, %%mm6, %%mm7)\
  996. \
  997. HADAMARD48\
  998. "movq %%mm7, (%1) \n\t"\
  999. MMABS(%%mm0, %%mm7)\
  1000. MMABS(%%mm1, %%mm7)\
  1001. MMABS_SUM(%%mm2, %%mm7, %%mm0)\
  1002. MMABS_SUM(%%mm3, %%mm7, %%mm1)\
  1003. MMABS_SUM(%%mm4, %%mm7, %%mm0)\
  1004. MMABS_SUM(%%mm5, %%mm7, %%mm1)\
  1005. MMABS_SUM(%%mm6, %%mm7, %%mm0)\
  1006. "movq (%1), %%mm2 \n\t"\
  1007. MMABS_SUM(%%mm2, %%mm7, %%mm1)\
  1008. "paddusw 64(%1), %%mm0 \n\t"\
  1009. "paddusw %%mm1, %%mm0 \n\t"\
  1010. \
  1011. HSUM(%%mm0, %%mm1, %0)\
  1012. \
  1013. : "=r" (sum)\
  1014. : "r"(temp)\
  1015. );\
  1016. return sum&0xFFFF;\
  1017. }\
  1018. WRAPPER8_16_SQ(hadamard8_diff_##cpu, hadamard8_diff16_##cpu)
  1019. #define HADAMARD8_DIFF_SSE2(cpu) \
  1020. static int hadamard8_diff_##cpu(void *s, uint8_t *src1, uint8_t *src2, int stride, int h){\
  1021. DECLARE_ALIGNED_16(uint64_t, temp[4]);\
  1022. int sum;\
  1023. \
  1024. assert(h==8);\
  1025. \
  1026. DIFF_PIXELS_8x8(src1, src2, stride, temp[0]);\
  1027. \
  1028. __asm__ volatile(\
  1029. HADAMARD8(%%xmm0, %%xmm1, %%xmm2, %%xmm3, %%xmm4, %%xmm5, %%xmm6, %%xmm7)\
  1030. TRANSPOSE8(%%xmm0, %%xmm1, %%xmm2, %%xmm3, %%xmm4, %%xmm5, %%xmm6, %%xmm7, (%1))\
  1031. HADAMARD8(%%xmm0, %%xmm5, %%xmm7, %%xmm3, %%xmm6, %%xmm4, %%xmm2, %%xmm1)\
  1032. MMABS_SUM_8x8\
  1033. HSUM_SSE2(%%xmm0, %%xmm1, %0)\
  1034. : "=r" (sum)\
  1035. : "r"(temp)\
  1036. );\
  1037. return sum&0xFFFF;\
  1038. }\
  1039. WRAPPER8_16_SQ(hadamard8_diff_##cpu, hadamard8_diff16_##cpu)
  1040. #define MMABS(a,z) MMABS_MMX(a,z)
  1041. #define HSUM(a,t,dst) HSUM_MMX(a,t,dst)
  1042. HADAMARD8_DIFF_MMX(mmx)
  1043. #undef MMABS
  1044. #undef HSUM
  1045. #define MMABS(a,z) MMABS_MMX2(a,z)
  1046. #define MMABS_SUM_8x8 MMABS_SUM_8x8_SSE2
  1047. #define HSUM(a,t,dst) HSUM_MMX2(a,t,dst)
  1048. HADAMARD8_DIFF_MMX(mmx2)
  1049. HADAMARD8_DIFF_SSE2(sse2)
  1050. #undef MMABS
  1051. #undef MMABS_SUM_8x8
  1052. #undef HSUM
  1053. #ifdef HAVE_SSSE3
  1054. #define MMABS(a,z) MMABS_SSSE3(a,z)
  1055. #define MMABS_SUM_8x8 MMABS_SUM_8x8_NOSPILL
  1056. HADAMARD8_DIFF_SSE2(ssse3)
  1057. #undef MMABS
  1058. #undef MMABS_SUM_8x8
  1059. #endif
  1060. #define DCT_SAD4(m,mm,o)\
  1061. "mov"#m" "#o"+ 0(%1), "#mm"2 \n\t"\
  1062. "mov"#m" "#o"+16(%1), "#mm"3 \n\t"\
  1063. "mov"#m" "#o"+32(%1), "#mm"4 \n\t"\
  1064. "mov"#m" "#o"+48(%1), "#mm"5 \n\t"\
  1065. MMABS_SUM(mm##2, mm##6, mm##0)\
  1066. MMABS_SUM(mm##3, mm##7, mm##1)\
  1067. MMABS_SUM(mm##4, mm##6, mm##0)\
  1068. MMABS_SUM(mm##5, mm##7, mm##1)\
  1069. #define DCT_SAD_MMX\
  1070. "pxor %%mm0, %%mm0 \n\t"\
  1071. "pxor %%mm1, %%mm1 \n\t"\
  1072. DCT_SAD4(q, %%mm, 0)\
  1073. DCT_SAD4(q, %%mm, 8)\
  1074. DCT_SAD4(q, %%mm, 64)\
  1075. DCT_SAD4(q, %%mm, 72)\
  1076. "paddusw %%mm1, %%mm0 \n\t"\
  1077. HSUM(%%mm0, %%mm1, %0)
  1078. #define DCT_SAD_SSE2\
  1079. "pxor %%xmm0, %%xmm0 \n\t"\
  1080. "pxor %%xmm1, %%xmm1 \n\t"\
  1081. DCT_SAD4(dqa, %%xmm, 0)\
  1082. DCT_SAD4(dqa, %%xmm, 64)\
  1083. "paddusw %%xmm1, %%xmm0 \n\t"\
  1084. HSUM(%%xmm0, %%xmm1, %0)
  1085. #define DCT_SAD_FUNC(cpu) \
  1086. static int sum_abs_dctelem_##cpu(DCTELEM *block){\
  1087. int sum;\
  1088. __asm__ volatile(\
  1089. DCT_SAD\
  1090. :"=r"(sum)\
  1091. :"r"(block)\
  1092. );\
  1093. return sum&0xFFFF;\
  1094. }
  1095. #define DCT_SAD DCT_SAD_MMX
  1096. #define HSUM(a,t,dst) HSUM_MMX(a,t,dst)
  1097. #define MMABS(a,z) MMABS_MMX(a,z)
  1098. DCT_SAD_FUNC(mmx)
  1099. #undef MMABS
  1100. #undef HSUM
  1101. #define HSUM(a,t,dst) HSUM_MMX2(a,t,dst)
  1102. #define MMABS(a,z) MMABS_MMX2(a,z)
  1103. DCT_SAD_FUNC(mmx2)
  1104. #undef HSUM
  1105. #undef DCT_SAD
  1106. #define DCT_SAD DCT_SAD_SSE2
  1107. #define HSUM(a,t,dst) HSUM_SSE2(a,t,dst)
  1108. DCT_SAD_FUNC(sse2)
  1109. #undef MMABS
  1110. #ifdef HAVE_SSSE3
  1111. #define MMABS(a,z) MMABS_SSSE3(a,z)
  1112. DCT_SAD_FUNC(ssse3)
  1113. #undef MMABS
  1114. #endif
  1115. #undef HSUM
  1116. #undef DCT_SAD
  1117. static int ssd_int8_vs_int16_mmx(const int8_t *pix1, const int16_t *pix2, int size){
  1118. int sum;
  1119. x86_reg i=size;
  1120. __asm__ volatile(
  1121. "pxor %%mm4, %%mm4 \n"
  1122. "1: \n"
  1123. "sub $8, %0 \n"
  1124. "movq (%2,%0), %%mm2 \n"
  1125. "movq (%3,%0,2), %%mm0 \n"
  1126. "movq 8(%3,%0,2), %%mm1 \n"
  1127. "punpckhbw %%mm2, %%mm3 \n"
  1128. "punpcklbw %%mm2, %%mm2 \n"
  1129. "psraw $8, %%mm3 \n"
  1130. "psraw $8, %%mm2 \n"
  1131. "psubw %%mm3, %%mm1 \n"
  1132. "psubw %%mm2, %%mm0 \n"
  1133. "pmaddwd %%mm1, %%mm1 \n"
  1134. "pmaddwd %%mm0, %%mm0 \n"
  1135. "paddd %%mm1, %%mm4 \n"
  1136. "paddd %%mm0, %%mm4 \n"
  1137. "jg 1b \n"
  1138. "movq %%mm4, %%mm3 \n"
  1139. "psrlq $32, %%mm3 \n"
  1140. "paddd %%mm3, %%mm4 \n"
  1141. "movd %%mm4, %1 \n"
  1142. :"+r"(i), "=r"(sum)
  1143. :"r"(pix1), "r"(pix2)
  1144. );
  1145. return sum;
  1146. }
  1147. #define PHADDD(a, t)\
  1148. "movq "#a", "#t" \n\t"\
  1149. "psrlq $32, "#a" \n\t"\
  1150. "paddd "#t", "#a" \n\t"
  1151. /*
  1152. pmulhw: dst[0-15]=(src[0-15]*dst[0-15])[16-31]
  1153. pmulhrw: dst[0-15]=(src[0-15]*dst[0-15] + 0x8000)[16-31]
  1154. pmulhrsw: dst[0-15]=(src[0-15]*dst[0-15] + 0x4000)[15-30]
  1155. */
  1156. #define PMULHRW(x, y, s, o)\
  1157. "pmulhw " #s ", "#x " \n\t"\
  1158. "pmulhw " #s ", "#y " \n\t"\
  1159. "paddw " #o ", "#x " \n\t"\
  1160. "paddw " #o ", "#y " \n\t"\
  1161. "psraw $1, "#x " \n\t"\
  1162. "psraw $1, "#y " \n\t"
  1163. #define DEF(x) x ## _mmx
  1164. #define SET_RND MOVQ_WONE
  1165. #define SCALE_OFFSET 1
  1166. #include "dsputil_mmx_qns_template.c"
  1167. #undef DEF
  1168. #undef SET_RND
  1169. #undef SCALE_OFFSET
  1170. #undef PMULHRW
  1171. #define DEF(x) x ## _3dnow
  1172. #define SET_RND(x)
  1173. #define SCALE_OFFSET 0
  1174. #define PMULHRW(x, y, s, o)\
  1175. "pmulhrw " #s ", "#x " \n\t"\
  1176. "pmulhrw " #s ", "#y " \n\t"
  1177. #include "dsputil_mmx_qns_template.c"
  1178. #undef DEF
  1179. #undef SET_RND
  1180. #undef SCALE_OFFSET
  1181. #undef PMULHRW
  1182. #ifdef HAVE_SSSE3
  1183. #undef PHADDD
  1184. #define DEF(x) x ## _ssse3
  1185. #define SET_RND(x)
  1186. #define SCALE_OFFSET -1
  1187. #define PHADDD(a, t)\
  1188. "pshufw $0x0E, "#a", "#t" \n\t"\
  1189. "paddd "#t", "#a" \n\t" /* faster than phaddd on core2 */
  1190. #define PMULHRW(x, y, s, o)\
  1191. "pmulhrsw " #s ", "#x " \n\t"\
  1192. "pmulhrsw " #s ", "#y " \n\t"
  1193. #include "dsputil_mmx_qns_template.c"
  1194. #undef DEF
  1195. #undef SET_RND
  1196. #undef SCALE_OFFSET
  1197. #undef PMULHRW
  1198. #undef PHADDD
  1199. #endif //HAVE_SSSE3
  1200. /* FLAC specific */
  1201. void ff_flac_compute_autocorr_sse2(const int32_t *data, int len, int lag,
  1202. double *autoc);
  1203. void dsputilenc_init_mmx(DSPContext* c, AVCodecContext *avctx)
  1204. {
  1205. if (mm_flags & FF_MM_MMX) {
  1206. const int dct_algo = avctx->dct_algo;
  1207. if(dct_algo==FF_DCT_AUTO || dct_algo==FF_DCT_MMX){
  1208. if(mm_flags & FF_MM_SSE2){
  1209. c->fdct = ff_fdct_sse2;
  1210. }else if(mm_flags & FF_MM_MMXEXT){
  1211. c->fdct = ff_fdct_mmx2;
  1212. }else{
  1213. c->fdct = ff_fdct_mmx;
  1214. }
  1215. }
  1216. c->get_pixels = get_pixels_mmx;
  1217. c->diff_pixels = diff_pixels_mmx;
  1218. c->pix_sum = pix_sum16_mmx;
  1219. c->diff_bytes= diff_bytes_mmx;
  1220. c->sum_abs_dctelem= sum_abs_dctelem_mmx;
  1221. c->hadamard8_diff[0]= hadamard8_diff16_mmx;
  1222. c->hadamard8_diff[1]= hadamard8_diff_mmx;
  1223. c->pix_norm1 = pix_norm1_mmx;
  1224. c->sse[0] = (mm_flags & FF_MM_SSE2) ? sse16_sse2 : sse16_mmx;
  1225. c->sse[1] = sse8_mmx;
  1226. c->vsad[4]= vsad_intra16_mmx;
  1227. c->nsse[0] = nsse16_mmx;
  1228. c->nsse[1] = nsse8_mmx;
  1229. if(!(avctx->flags & CODEC_FLAG_BITEXACT)){
  1230. c->vsad[0] = vsad16_mmx;
  1231. }
  1232. if(!(avctx->flags & CODEC_FLAG_BITEXACT)){
  1233. c->try_8x8basis= try_8x8basis_mmx;
  1234. }
  1235. c->add_8x8basis= add_8x8basis_mmx;
  1236. c->ssd_int8_vs_int16 = ssd_int8_vs_int16_mmx;
  1237. if (mm_flags & FF_MM_MMXEXT) {
  1238. c->sum_abs_dctelem= sum_abs_dctelem_mmx2;
  1239. c->hadamard8_diff[0]= hadamard8_diff16_mmx2;
  1240. c->hadamard8_diff[1]= hadamard8_diff_mmx2;
  1241. c->vsad[4]= vsad_intra16_mmx2;
  1242. if(!(avctx->flags & CODEC_FLAG_BITEXACT)){
  1243. c->vsad[0] = vsad16_mmx2;
  1244. }
  1245. c->sub_hfyu_median_prediction= sub_hfyu_median_prediction_mmx2;
  1246. }
  1247. if(mm_flags & FF_MM_SSE2){
  1248. c->get_pixels = get_pixels_sse2;
  1249. c->sum_abs_dctelem= sum_abs_dctelem_sse2;
  1250. c->hadamard8_diff[0]= hadamard8_diff16_sse2;
  1251. c->hadamard8_diff[1]= hadamard8_diff_sse2;
  1252. if (ENABLE_FLAC_ENCODER)
  1253. c->flac_compute_autocorr = ff_flac_compute_autocorr_sse2;
  1254. }
  1255. #ifdef HAVE_SSSE3
  1256. if(mm_flags & FF_MM_SSSE3){
  1257. if(!(avctx->flags & CODEC_FLAG_BITEXACT)){
  1258. c->try_8x8basis= try_8x8basis_ssse3;
  1259. }
  1260. c->add_8x8basis= add_8x8basis_ssse3;
  1261. c->sum_abs_dctelem= sum_abs_dctelem_ssse3;
  1262. c->hadamard8_diff[0]= hadamard8_diff16_ssse3;
  1263. c->hadamard8_diff[1]= hadamard8_diff_ssse3;
  1264. }
  1265. #endif
  1266. if(mm_flags & FF_MM_3DNOW){
  1267. if(!(avctx->flags & CODEC_FLAG_BITEXACT)){
  1268. c->try_8x8basis= try_8x8basis_3dnow;
  1269. }
  1270. c->add_8x8basis= add_8x8basis_3dnow;
  1271. }
  1272. }
  1273. dsputil_init_pix_mmx(c, avctx);
  1274. }