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.

918 lines
19KB

  1. ;*****************************************************************************
  2. ;* x86util.asm
  3. ;*****************************************************************************
  4. ;* Copyright (C) 2008-2010 x264 project
  5. ;*
  6. ;* Authors: Loren Merritt <lorenm@u.washington.edu>
  7. ;* Holger Lubitz <holger@lubitz.org>
  8. ;*
  9. ;* This file is part of FFmpeg.
  10. ;*
  11. ;* FFmpeg is free software; you can redistribute it and/or
  12. ;* modify it under the terms of the GNU Lesser General Public
  13. ;* License as published by the Free Software Foundation; either
  14. ;* version 2.1 of the License, or (at your option) any later version.
  15. ;*
  16. ;* FFmpeg is distributed in the hope that it will be useful,
  17. ;* but WITHOUT ANY WARRANTY; without even the implied warranty of
  18. ;* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  19. ;* Lesser General Public License for more details.
  20. ;*
  21. ;* You should have received a copy of the GNU Lesser General Public
  22. ;* License along with FFmpeg; if not, write to the Free Software
  23. ;* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  24. ;******************************************************************************
  25. %define private_prefix ff
  26. %define public_prefix avpriv
  27. %define cpuflags_mmxext cpuflags_mmx2
  28. %include "libavutil/x86/x86inc.asm"
  29. ; expands to [base],...,[base+7*stride]
  30. %define PASS8ROWS(base, base3, stride, stride3) \
  31. [base], [base + stride], [base + 2*stride], [base3], \
  32. [base3 + stride], [base3 + 2*stride], [base3 + stride3], [base3 + stride*4]
  33. ; Interleave low src0 with low src1 and store in src0,
  34. ; interleave high src0 with high src1 and store in src1.
  35. ; %1 - types
  36. ; %2 - index of the register with src0
  37. ; %3 - index of the register with src1
  38. ; %4 - index of the register for intermediate results
  39. ; example for %1 - wd: input: src0: x0 x1 x2 x3 z0 z1 z2 z3
  40. ; src1: y0 y1 y2 y3 q0 q1 q2 q3
  41. ; output: src0: x0 y0 x1 y1 x2 y2 x3 y3
  42. ; src1: z0 q0 z1 q1 z2 q2 z3 q3
  43. %macro SBUTTERFLY 4
  44. %ifidn %1, dqqq
  45. vperm2i128 m%4, m%2, m%3, q0301
  46. vinserti128 m%2, m%2, xm%3, 1
  47. %elif avx_enabled == 0
  48. mova m%4, m%2
  49. punpckl%1 m%2, m%3
  50. punpckh%1 m%4, m%3
  51. %else
  52. punpckh%1 m%4, m%2, m%3
  53. punpckl%1 m%2, m%3
  54. %endif
  55. SWAP %3, %4
  56. %endmacro
  57. %macro SBUTTERFLY2 4
  58. punpckl%1 m%4, m%2, m%3
  59. punpckh%1 m%2, m%2, m%3
  60. SWAP %2, %4, %3
  61. %endmacro
  62. %macro SBUTTERFLYPS 3
  63. unpcklps m%3, m%1, m%2
  64. unpckhps m%1, m%1, m%2
  65. SWAP %1, %3, %2
  66. %endmacro
  67. %macro TRANSPOSE4x4B 5
  68. SBUTTERFLY bw, %1, %2, %5
  69. SBUTTERFLY bw, %3, %4, %5
  70. SBUTTERFLY wd, %1, %3, %5
  71. SBUTTERFLY wd, %2, %4, %5
  72. SWAP %2, %3
  73. %endmacro
  74. %macro TRANSPOSE4x4W 5
  75. SBUTTERFLY wd, %1, %2, %5
  76. SBUTTERFLY wd, %3, %4, %5
  77. SBUTTERFLY dq, %1, %3, %5
  78. SBUTTERFLY dq, %2, %4, %5
  79. SWAP %2, %3
  80. %endmacro
  81. %macro TRANSPOSE2x4x4B 5
  82. SBUTTERFLY bw, %1, %2, %5
  83. SBUTTERFLY bw, %3, %4, %5
  84. SBUTTERFLY wd, %1, %3, %5
  85. SBUTTERFLY wd, %2, %4, %5
  86. SBUTTERFLY dq, %1, %2, %5
  87. SBUTTERFLY dq, %3, %4, %5
  88. %endmacro
  89. %macro TRANSPOSE2x4x4W 5
  90. SBUTTERFLY wd, %1, %2, %5
  91. SBUTTERFLY wd, %3, %4, %5
  92. SBUTTERFLY dq, %1, %3, %5
  93. SBUTTERFLY dq, %2, %4, %5
  94. SBUTTERFLY qdq, %1, %2, %5
  95. SBUTTERFLY qdq, %3, %4, %5
  96. %endmacro
  97. %macro TRANSPOSE4x4D 5
  98. SBUTTERFLY dq, %1, %2, %5
  99. SBUTTERFLY dq, %3, %4, %5
  100. SBUTTERFLY qdq, %1, %3, %5
  101. SBUTTERFLY qdq, %2, %4, %5
  102. SWAP %2, %3
  103. %endmacro
  104. ; identical behavior to TRANSPOSE4x4D, but using SSE1 float ops
  105. %macro TRANSPOSE4x4PS 5
  106. SBUTTERFLYPS %1, %2, %5
  107. SBUTTERFLYPS %3, %4, %5
  108. movlhps m%5, m%1, m%3
  109. movhlps m%3, m%1
  110. SWAP %5, %1
  111. movlhps m%5, m%2, m%4
  112. movhlps m%4, m%2
  113. SWAP %5, %2, %3
  114. %endmacro
  115. %macro TRANSPOSE8x4D 9-11
  116. %if ARCH_X86_64
  117. SBUTTERFLY dq, %1, %2, %9
  118. SBUTTERFLY dq, %3, %4, %9
  119. SBUTTERFLY dq, %5, %6, %9
  120. SBUTTERFLY dq, %7, %8, %9
  121. SBUTTERFLY qdq, %1, %3, %9
  122. SBUTTERFLY qdq, %2, %4, %9
  123. SBUTTERFLY qdq, %5, %7, %9
  124. SBUTTERFLY qdq, %6, %8, %9
  125. SWAP %2, %5
  126. SWAP %4, %7
  127. %else
  128. ; in: m0..m7
  129. ; out: m0..m7, unless %11 in which case m2 is in %9
  130. ; spills into %9 and %10
  131. movdqa %9, m%7
  132. SBUTTERFLY dq, %1, %2, %7
  133. movdqa %10, m%2
  134. movdqa m%7, %9
  135. SBUTTERFLY dq, %3, %4, %2
  136. SBUTTERFLY dq, %5, %6, %2
  137. SBUTTERFLY dq, %7, %8, %2
  138. SBUTTERFLY qdq, %1, %3, %2
  139. movdqa %9, m%3
  140. movdqa m%2, %10
  141. SBUTTERFLY qdq, %2, %4, %3
  142. SBUTTERFLY qdq, %5, %7, %3
  143. SBUTTERFLY qdq, %6, %8, %3
  144. SWAP %2, %5
  145. SWAP %4, %7
  146. %if %0<11
  147. movdqa m%3, %9
  148. %endif
  149. %endif
  150. %endmacro
  151. %macro TRANSPOSE8x8W 9-11
  152. %if ARCH_X86_64
  153. SBUTTERFLY wd, %1, %2, %9
  154. SBUTTERFLY wd, %3, %4, %9
  155. SBUTTERFLY wd, %5, %6, %9
  156. SBUTTERFLY wd, %7, %8, %9
  157. SBUTTERFLY dq, %1, %3, %9
  158. SBUTTERFLY dq, %2, %4, %9
  159. SBUTTERFLY dq, %5, %7, %9
  160. SBUTTERFLY dq, %6, %8, %9
  161. SBUTTERFLY qdq, %1, %5, %9
  162. SBUTTERFLY qdq, %2, %6, %9
  163. SBUTTERFLY qdq, %3, %7, %9
  164. SBUTTERFLY qdq, %4, %8, %9
  165. SWAP %2, %5
  166. SWAP %4, %7
  167. %else
  168. ; in: m0..m7, unless %11 in which case m6 is in %9
  169. ; out: m0..m7, unless %11 in which case m4 is in %10
  170. ; spills into %9 and %10
  171. %if %0<11
  172. movdqa %9, m%7
  173. %endif
  174. SBUTTERFLY wd, %1, %2, %7
  175. movdqa %10, m%2
  176. movdqa m%7, %9
  177. SBUTTERFLY wd, %3, %4, %2
  178. SBUTTERFLY wd, %5, %6, %2
  179. SBUTTERFLY wd, %7, %8, %2
  180. SBUTTERFLY dq, %1, %3, %2
  181. movdqa %9, m%3
  182. movdqa m%2, %10
  183. SBUTTERFLY dq, %2, %4, %3
  184. SBUTTERFLY dq, %5, %7, %3
  185. SBUTTERFLY dq, %6, %8, %3
  186. SBUTTERFLY qdq, %1, %5, %3
  187. SBUTTERFLY qdq, %2, %6, %3
  188. movdqa %10, m%2
  189. movdqa m%3, %9
  190. SBUTTERFLY qdq, %3, %7, %2
  191. SBUTTERFLY qdq, %4, %8, %2
  192. SWAP %2, %5
  193. SWAP %4, %7
  194. %if %0<11
  195. movdqa m%5, %10
  196. %endif
  197. %endif
  198. %endmacro
  199. %macro TRANSPOSE16x16W 18-19
  200. ; in: m0..m15, unless %19 in which case m6 is in %17
  201. ; out: m0..m15, unless %19 in which case m4 is in %18
  202. ; spills into %17 and %18
  203. %if %0 < 19
  204. mova %17, m%7
  205. %endif
  206. SBUTTERFLY dqqq, %1, %9, %7
  207. SBUTTERFLY dqqq, %2, %10, %7
  208. SBUTTERFLY dqqq, %3, %11, %7
  209. SBUTTERFLY dqqq, %4, %12, %7
  210. SBUTTERFLY dqqq, %5, %13, %7
  211. SBUTTERFLY dqqq, %6, %14, %7
  212. mova %18, m%14
  213. mova m%7, %17
  214. SBUTTERFLY dqqq, %7, %15, %14
  215. SBUTTERFLY dqqq, %8, %16, %14
  216. SBUTTERFLY wd, %1, %2, %14
  217. SBUTTERFLY wd, %3, %4, %14
  218. SBUTTERFLY wd, %5, %6, %14
  219. SBUTTERFLY wd, %7, %8, %14
  220. SBUTTERFLY wd, %9, %10, %14
  221. SBUTTERFLY wd, %11, %12, %14
  222. mova %17, m%12
  223. mova m%14, %18
  224. SBUTTERFLY wd, %13, %14, %12
  225. SBUTTERFLY wd, %15, %16, %12
  226. SBUTTERFLY dq, %1, %3, %12
  227. SBUTTERFLY dq, %2, %4, %12
  228. SBUTTERFLY dq, %5, %7, %12
  229. SBUTTERFLY dq, %6, %8, %12
  230. SBUTTERFLY dq, %9, %11, %12
  231. mova %18, m%11
  232. mova m%12, %17
  233. SBUTTERFLY dq, %10, %12, %11
  234. SBUTTERFLY dq, %13, %15, %11
  235. SBUTTERFLY dq, %14, %16, %11
  236. SBUTTERFLY qdq, %1, %5, %11
  237. SBUTTERFLY qdq, %2, %6, %11
  238. SBUTTERFLY qdq, %3, %7, %11
  239. SBUTTERFLY qdq, %4, %8, %11
  240. SWAP %2, %5
  241. SWAP %4, %7
  242. SBUTTERFLY qdq, %9, %13, %11
  243. SBUTTERFLY qdq, %10, %14, %11
  244. mova m%11, %18
  245. mova %18, m%5
  246. SBUTTERFLY qdq, %11, %15, %5
  247. SBUTTERFLY qdq, %12, %16, %5
  248. %if %0 < 19
  249. mova m%5, %18
  250. %endif
  251. SWAP %10, %13
  252. SWAP %12, %15
  253. %endmacro
  254. %macro TRANSPOSE_8X8B 8
  255. %if mmsize == 8
  256. %error "This macro does not support mmsize == 8"
  257. %endif
  258. punpcklbw m%1, m%2
  259. punpcklbw m%3, m%4
  260. punpcklbw m%5, m%6
  261. punpcklbw m%7, m%8
  262. TRANSPOSE4x4W %1, %3, %5, %7, %2
  263. MOVHL m%2, m%1
  264. MOVHL m%4, m%3
  265. MOVHL m%6, m%5
  266. MOVHL m%8, m%7
  267. %endmacro
  268. ; PABSW macro assumes %1 != %2, while ABS1/2 macros work in-place
  269. %macro PABSW 2
  270. %if cpuflag(ssse3)
  271. pabsw %1, %2
  272. %elif cpuflag(mmxext)
  273. pxor %1, %1
  274. psubw %1, %2
  275. pmaxsw %1, %2
  276. %else
  277. pxor %1, %1
  278. pcmpgtw %1, %2
  279. pxor %2, %1
  280. psubw %2, %1
  281. SWAP %1, %2
  282. %endif
  283. %endmacro
  284. %macro PSIGNW 2
  285. %if cpuflag(ssse3)
  286. psignw %1, %2
  287. %else
  288. pxor %1, %2
  289. psubw %1, %2
  290. %endif
  291. %endmacro
  292. %macro ABS1 2
  293. %if cpuflag(ssse3)
  294. pabsw %1, %1
  295. %elif cpuflag(mmxext) ; a, tmp
  296. pxor %2, %2
  297. psubw %2, %1
  298. pmaxsw %1, %2
  299. %else ; a, tmp
  300. pxor %2, %2
  301. pcmpgtw %2, %1
  302. pxor %1, %2
  303. psubw %1, %2
  304. %endif
  305. %endmacro
  306. %macro ABS2 4
  307. %if cpuflag(ssse3)
  308. pabsw %1, %1
  309. pabsw %2, %2
  310. %elif cpuflag(mmxext) ; a, b, tmp0, tmp1
  311. pxor %3, %3
  312. pxor %4, %4
  313. psubw %3, %1
  314. psubw %4, %2
  315. pmaxsw %1, %3
  316. pmaxsw %2, %4
  317. %else ; a, b, tmp0, tmp1
  318. pxor %3, %3
  319. pxor %4, %4
  320. pcmpgtw %3, %1
  321. pcmpgtw %4, %2
  322. pxor %1, %3
  323. pxor %2, %4
  324. psubw %1, %3
  325. psubw %2, %4
  326. %endif
  327. %endmacro
  328. %macro ABSB 2 ; source mmreg, temp mmreg (unused for ssse3)
  329. %if cpuflag(ssse3)
  330. pabsb %1, %1
  331. %else
  332. pxor %2, %2
  333. psubb %2, %1
  334. pminub %1, %2
  335. %endif
  336. %endmacro
  337. %macro ABSB2 4 ; src1, src2, tmp1, tmp2 (tmp1/2 unused for SSSE3)
  338. %if cpuflag(ssse3)
  339. pabsb %1, %1
  340. pabsb %2, %2
  341. %else
  342. pxor %3, %3
  343. pxor %4, %4
  344. psubb %3, %1
  345. psubb %4, %2
  346. pminub %1, %3
  347. pminub %2, %4
  348. %endif
  349. %endmacro
  350. %macro ABSD2_MMX 4
  351. pxor %3, %3
  352. pxor %4, %4
  353. pcmpgtd %3, %1
  354. pcmpgtd %4, %2
  355. pxor %1, %3
  356. pxor %2, %4
  357. psubd %1, %3
  358. psubd %2, %4
  359. %endmacro
  360. %macro ABS4 6
  361. ABS2 %1, %2, %5, %6
  362. ABS2 %3, %4, %5, %6
  363. %endmacro
  364. %macro SPLATB_LOAD 3
  365. %if cpuflag(ssse3)
  366. movd %1, [%2-3]
  367. pshufb %1, %3
  368. %else
  369. movd %1, [%2-3] ;to avoid crossing a cacheline
  370. punpcklbw %1, %1
  371. SPLATW %1, %1, 3
  372. %endif
  373. %endmacro
  374. %macro SPLATB_REG 3
  375. %if cpuflag(ssse3)
  376. movd %1, %2d
  377. pshufb %1, %3
  378. %else
  379. movd %1, %2d
  380. punpcklbw %1, %1
  381. SPLATW %1, %1, 0
  382. %endif
  383. %endmacro
  384. %macro HADDD 2 ; sum junk
  385. %if sizeof%1 == 32
  386. %define %2 xmm%2
  387. vextracti128 %2, %1, 1
  388. %define %1 xmm%1
  389. paddd %1, %2
  390. %endif
  391. %if mmsize >= 16
  392. %if cpuflag(xop) && sizeof%1 == 16
  393. vphadddq %1, %1
  394. %endif
  395. movhlps %2, %1
  396. paddd %1, %2
  397. %endif
  398. %if notcpuflag(xop) || sizeof%1 != 16
  399. %if cpuflag(mmxext)
  400. PSHUFLW %2, %1, q0032
  401. %else ; mmx
  402. mova %2, %1
  403. psrlq %2, 32
  404. %endif
  405. paddd %1, %2
  406. %endif
  407. %undef %1
  408. %undef %2
  409. %endmacro
  410. %macro HADDW 2 ; reg, tmp
  411. %if cpuflag(xop) && sizeof%1 == 16
  412. vphaddwq %1, %1
  413. movhlps %2, %1
  414. paddd %1, %2
  415. %else
  416. pmaddwd %1, [pw_1]
  417. HADDD %1, %2
  418. %endif
  419. %endmacro
  420. %macro HADDPS 3 ; dst, src, tmp
  421. %if cpuflag(sse3)
  422. haddps %1, %1, %2
  423. %else
  424. movaps %3, %1
  425. shufps %1, %2, q2020
  426. shufps %3, %2, q3131
  427. addps %1, %3
  428. %endif
  429. %endmacro
  430. %macro PALIGNR 4-5
  431. %if cpuflag(ssse3)
  432. %if %0==5
  433. palignr %1, %2, %3, %4
  434. %else
  435. palignr %1, %2, %3
  436. %endif
  437. %elif cpuflag(mmx) ; [dst,] src1, src2, imm, tmp
  438. %define %%dst %1
  439. %if %0==5
  440. %ifnidn %1, %2
  441. mova %%dst, %2
  442. %endif
  443. %rotate 1
  444. %endif
  445. %ifnidn %4, %2
  446. mova %4, %2
  447. %endif
  448. %if mmsize==8
  449. psllq %%dst, (8-%3)*8
  450. psrlq %4, %3*8
  451. %else
  452. pslldq %%dst, 16-%3
  453. psrldq %4, %3
  454. %endif
  455. por %%dst, %4
  456. %endif
  457. %endmacro
  458. %macro PAVGB 2-4
  459. %if cpuflag(mmxext)
  460. pavgb %1, %2
  461. %elif cpuflag(3dnow)
  462. pavgusb %1, %2
  463. %elif cpuflag(mmx)
  464. movu %3, %2
  465. por %3, %1
  466. pxor %1, %2
  467. pand %1, %4
  468. psrlq %1, 1
  469. psubb %3, %1
  470. SWAP %1, %3
  471. %endif
  472. %endmacro
  473. %macro PSHUFLW 1+
  474. %if mmsize == 8
  475. pshufw %1
  476. %else
  477. pshuflw %1
  478. %endif
  479. %endmacro
  480. %macro PSWAPD 2
  481. %if cpuflag(mmxext)
  482. pshufw %1, %2, q1032
  483. %elif cpuflag(3dnowext)
  484. pswapd %1, %2
  485. %elif cpuflag(3dnow)
  486. movq %1, %2
  487. psrlq %1, 32
  488. punpckldq %1, %2
  489. %endif
  490. %endmacro
  491. %macro DEINTB 5 ; mask, reg1, mask, reg2, optional src to fill masks from
  492. %ifnum %5
  493. pand m%3, m%5, m%4 ; src .. y6 .. y4
  494. pand m%1, m%5, m%2 ; dst .. y6 .. y4
  495. %else
  496. mova m%1, %5
  497. pand m%3, m%1, m%4 ; src .. y6 .. y4
  498. pand m%1, m%1, m%2 ; dst .. y6 .. y4
  499. %endif
  500. psrlw m%2, 8 ; dst .. y7 .. y5
  501. psrlw m%4, 8 ; src .. y7 .. y5
  502. %endmacro
  503. %macro SUMSUB_BA 3-4
  504. %if %0==3
  505. padd%1 m%2, m%3
  506. padd%1 m%3, m%3
  507. psub%1 m%3, m%2
  508. %else
  509. %if avx_enabled == 0
  510. mova m%4, m%2
  511. padd%1 m%2, m%3
  512. psub%1 m%3, m%4
  513. %else
  514. padd%1 m%4, m%2, m%3
  515. psub%1 m%3, m%2
  516. SWAP %2, %4
  517. %endif
  518. %endif
  519. %endmacro
  520. %macro SUMSUB_BADC 5-6
  521. %if %0==6
  522. SUMSUB_BA %1, %2, %3, %6
  523. SUMSUB_BA %1, %4, %5, %6
  524. %else
  525. padd%1 m%2, m%3
  526. padd%1 m%4, m%5
  527. padd%1 m%3, m%3
  528. padd%1 m%5, m%5
  529. psub%1 m%3, m%2
  530. psub%1 m%5, m%4
  531. %endif
  532. %endmacro
  533. %macro SUMSUB2_AB 4
  534. %ifnum %3
  535. psub%1 m%4, m%2, m%3
  536. psub%1 m%4, m%3
  537. padd%1 m%2, m%2
  538. padd%1 m%2, m%3
  539. %else
  540. mova m%4, m%2
  541. padd%1 m%2, m%2
  542. padd%1 m%2, %3
  543. psub%1 m%4, %3
  544. psub%1 m%4, %3
  545. %endif
  546. %endmacro
  547. %macro SUMSUB2_BA 4
  548. %if avx_enabled == 0
  549. mova m%4, m%2
  550. padd%1 m%2, m%3
  551. padd%1 m%2, m%3
  552. psub%1 m%3, m%4
  553. psub%1 m%3, m%4
  554. %else
  555. padd%1 m%4, m%2, m%3
  556. padd%1 m%4, m%3
  557. psub%1 m%3, m%2
  558. psub%1 m%3, m%2
  559. SWAP %2, %4
  560. %endif
  561. %endmacro
  562. %macro SUMSUBD2_AB 5
  563. %ifnum %4
  564. psra%1 m%5, m%2, 1 ; %3: %3>>1
  565. psra%1 m%4, m%3, 1 ; %2: %2>>1
  566. padd%1 m%4, m%2 ; %3: %3>>1+%2
  567. psub%1 m%5, m%3 ; %2: %2>>1-%3
  568. SWAP %2, %5
  569. SWAP %3, %4
  570. %else
  571. mova %5, m%2
  572. mova %4, m%3
  573. psra%1 m%3, 1 ; %3: %3>>1
  574. psra%1 m%2, 1 ; %2: %2>>1
  575. padd%1 m%3, %5 ; %3: %3>>1+%2
  576. psub%1 m%2, %4 ; %2: %2>>1-%3
  577. %endif
  578. %endmacro
  579. %macro DCT4_1D 5
  580. %ifnum %5
  581. SUMSUB_BADC w, %4, %1, %3, %2, %5
  582. SUMSUB_BA w, %3, %4, %5
  583. SUMSUB2_AB w, %1, %2, %5
  584. SWAP %1, %3, %4, %5, %2
  585. %else
  586. SUMSUB_BADC w, %4, %1, %3, %2
  587. SUMSUB_BA w, %3, %4
  588. mova [%5], m%2
  589. SUMSUB2_AB w, %1, [%5], %2
  590. SWAP %1, %3, %4, %2
  591. %endif
  592. %endmacro
  593. %macro IDCT4_1D 6-7
  594. %ifnum %6
  595. SUMSUBD2_AB %1, %3, %5, %7, %6
  596. ; %3: %3>>1-%5 %5: %3+%5>>1
  597. SUMSUB_BA %1, %4, %2, %7
  598. ; %4: %2+%4 %2: %2-%4
  599. SUMSUB_BADC %1, %5, %4, %3, %2, %7
  600. ; %5: %2+%4 + (%3+%5>>1)
  601. ; %4: %2+%4 - (%3+%5>>1)
  602. ; %3: %2-%4 + (%3>>1-%5)
  603. ; %2: %2-%4 - (%3>>1-%5)
  604. %else
  605. %ifidn %1, w
  606. SUMSUBD2_AB %1, %3, %5, [%6], [%6+16]
  607. %else
  608. SUMSUBD2_AB %1, %3, %5, [%6], [%6+32]
  609. %endif
  610. SUMSUB_BA %1, %4, %2
  611. SUMSUB_BADC %1, %5, %4, %3, %2
  612. %endif
  613. SWAP %2, %5, %4
  614. ; %2: %2+%4 + (%3+%5>>1) row0
  615. ; %3: %2-%4 + (%3>>1-%5) row1
  616. ; %4: %2-%4 - (%3>>1-%5) row2
  617. ; %5: %2+%4 - (%3+%5>>1) row3
  618. %endmacro
  619. %macro LOAD_DIFF 5
  620. %ifidn %3, none
  621. movh %1, %4
  622. movh %2, %5
  623. punpcklbw %1, %2
  624. punpcklbw %2, %2
  625. psubw %1, %2
  626. %else
  627. movh %1, %4
  628. punpcklbw %1, %3
  629. movh %2, %5
  630. punpcklbw %2, %3
  631. psubw %1, %2
  632. %endif
  633. %endmacro
  634. %macro STORE_DCT 6
  635. movq [%5+%6+ 0], m%1
  636. movq [%5+%6+ 8], m%2
  637. movq [%5+%6+16], m%3
  638. movq [%5+%6+24], m%4
  639. movhps [%5+%6+32], m%1
  640. movhps [%5+%6+40], m%2
  641. movhps [%5+%6+48], m%3
  642. movhps [%5+%6+56], m%4
  643. %endmacro
  644. %macro LOAD_DIFF_8x4P 7-10 r0,r2,0 ; 4x dest, 2x temp, 2x pointer, increment?
  645. LOAD_DIFF m%1, m%5, m%7, [%8], [%9]
  646. LOAD_DIFF m%2, m%6, m%7, [%8+r1], [%9+r3]
  647. LOAD_DIFF m%3, m%5, m%7, [%8+2*r1], [%9+2*r3]
  648. LOAD_DIFF m%4, m%6, m%7, [%8+r4], [%9+r5]
  649. %if %10
  650. lea %8, [%8+4*r1]
  651. lea %9, [%9+4*r3]
  652. %endif
  653. %endmacro
  654. %macro DIFFx2 6-7
  655. movh %3, %5
  656. punpcklbw %3, %4
  657. psraw %1, 6
  658. paddsw %1, %3
  659. movh %3, %6
  660. punpcklbw %3, %4
  661. psraw %2, 6
  662. paddsw %2, %3
  663. packuswb %2, %1
  664. %endmacro
  665. %macro STORE_DIFF 4
  666. movh %2, %4
  667. punpcklbw %2, %3
  668. psraw %1, 6
  669. paddsw %1, %2
  670. packuswb %1, %1
  671. movh %4, %1
  672. %endmacro
  673. %macro STORE_DIFFx2 8 ; add1, add2, reg1, reg2, zero, shift, source, stride
  674. movh %3, [%7]
  675. movh %4, [%7+%8]
  676. psraw %1, %6
  677. psraw %2, %6
  678. punpcklbw %3, %5
  679. punpcklbw %4, %5
  680. paddw %3, %1
  681. paddw %4, %2
  682. packuswb %3, %5
  683. packuswb %4, %5
  684. movh [%7], %3
  685. movh [%7+%8], %4
  686. %endmacro
  687. %macro PMINUB 3 ; dst, src, ignored
  688. %if cpuflag(mmxext)
  689. pminub %1, %2
  690. %else ; dst, src, tmp
  691. mova %3, %1
  692. psubusb %3, %2
  693. psubb %1, %3
  694. %endif
  695. %endmacro
  696. %macro SPLATW 2-3 0
  697. %if cpuflag(avx2) && %3 == 0
  698. vpbroadcastw %1, %2
  699. %elif mmsize == 16
  700. pshuflw %1, %2, (%3)*0x55
  701. punpcklqdq %1, %1
  702. %elif cpuflag(mmxext)
  703. pshufw %1, %2, (%3)*0x55
  704. %else
  705. %ifnidn %1, %2
  706. mova %1, %2
  707. %endif
  708. %if %3 & 2
  709. punpckhwd %1, %1
  710. %else
  711. punpcklwd %1, %1
  712. %endif
  713. %if %3 & 1
  714. punpckhwd %1, %1
  715. %else
  716. punpcklwd %1, %1
  717. %endif
  718. %endif
  719. %endmacro
  720. %macro SPLATD 1
  721. %if mmsize == 8
  722. punpckldq %1, %1
  723. %elif cpuflag(sse2)
  724. pshufd %1, %1, 0
  725. %elif cpuflag(sse)
  726. shufps %1, %1, 0
  727. %endif
  728. %endmacro
  729. %macro CLIPUB 3 ;(dst, min, max)
  730. pmaxub %1, %2
  731. pminub %1, %3
  732. %endmacro
  733. %macro CLIPW 3 ;(dst, min, max)
  734. pmaxsw %1, %2
  735. pminsw %1, %3
  736. %endmacro
  737. %macro PMINSD_MMX 3 ; dst, src, tmp
  738. mova %3, %2
  739. pcmpgtd %3, %1
  740. pxor %1, %2
  741. pand %1, %3
  742. pxor %1, %2
  743. %endmacro
  744. %macro PMAXSD_MMX 3 ; dst, src, tmp
  745. mova %3, %1
  746. pcmpgtd %3, %2
  747. pand %1, %3
  748. pandn %3, %2
  749. por %1, %3
  750. %endmacro
  751. %macro CLIPD_MMX 3-4 ; src/dst, min, max, tmp
  752. PMINSD_MMX %1, %3, %4
  753. PMAXSD_MMX %1, %2, %4
  754. %endmacro
  755. %macro CLIPD_SSE2 3-4 ; src/dst, min (float), max (float), unused
  756. cvtdq2ps %1, %1
  757. minps %1, %3
  758. maxps %1, %2
  759. cvtps2dq %1, %1
  760. %endmacro
  761. %macro CLIPD_SSE41 3-4 ; src/dst, min, max, unused
  762. pminsd %1, %3
  763. pmaxsd %1, %2
  764. %endmacro
  765. %macro VBROADCASTSS 2 ; dst xmm/ymm, src m32
  766. %if cpuflag(avx)
  767. vbroadcastss %1, %2
  768. %else ; sse
  769. %ifnidn %1, %2
  770. movss %1, %2
  771. %endif
  772. shufps %1, %1, 0
  773. %endif
  774. %endmacro
  775. %macro VBROADCASTSD 2 ; dst xmm/ymm, src m64
  776. %if cpuflag(avx) && mmsize == 32
  777. vbroadcastsd %1, %2
  778. %elif cpuflag(sse3)
  779. movddup %1, %2
  780. %else ; sse2
  781. movsd %1, %2
  782. movlhps %1, %1
  783. %endif
  784. %endmacro
  785. %macro SHUFFLE_MASK_W 8
  786. %rep 8
  787. %if %1>=0x80
  788. db %1, %1
  789. %else
  790. db %1*2
  791. db %1*2+1
  792. %endif
  793. %rotate 1
  794. %endrep
  795. %endmacro
  796. %macro PMOVSXWD 2; dst, src
  797. %if cpuflag(sse4)
  798. pmovsxwd %1, %2
  799. %else
  800. %ifnidn %1, %2
  801. mova %1, %2
  802. %endif
  803. punpcklwd %1, %1
  804. psrad %1, 16
  805. %endif
  806. %endmacro
  807. ; Wrapper for non-FMA version of fmaddps
  808. %macro FMULADD_PS 5
  809. %if cpuflag(fma3) || cpuflag(fma4)
  810. fmaddps %1, %2, %3, %4
  811. %elifidn %1, %4
  812. mulps %5, %2, %3
  813. addps %1, %4, %5
  814. %else
  815. mulps %1, %2, %3
  816. addps %1, %4
  817. %endif
  818. %endmacro
  819. %macro LSHIFT 2
  820. %if mmsize > 8
  821. pslldq %1, %2
  822. %else
  823. psllq %1, 8*(%2)
  824. %endif
  825. %endmacro
  826. %macro RSHIFT 2
  827. %if mmsize > 8
  828. psrldq %1, %2
  829. %else
  830. psrlq %1, 8*(%2)
  831. %endif
  832. %endmacro
  833. %macro MOVHL 2 ; dst, src
  834. %ifidn %1, %2
  835. punpckhqdq %1, %2
  836. %elif cpuflag(avx)
  837. punpckhqdq %1, %2, %2
  838. %elif cpuflag(sse4)
  839. pshufd %1, %2, q3232 ; pshufd is slow on some older CPUs, so only use it on more modern ones
  840. %else
  841. movhlps %1, %2 ; may cause an int/float domain transition and has a dependency on dst
  842. %endif
  843. %endmacro