1 ;*****************************************************************************
3 ;*****************************************************************************
4 ;* Copyright (C) 2008-2010 x264 project
6 ;* Authors: Loren Merritt <lorenm@u.washington.edu>
7 ;* Holger Lubitz <holger@lubitz.org>
9 ;* This file is part of FFmpeg.
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.
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.
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 ;******************************************************************************
26 %define private_prefix ff
27 %define public_prefix avpriv
28 %define cpuflags_mmxext cpuflags_mmx2
30 %include "libavutil/x86/x86inc.asm"
32 ; expands to [base],...,[base+7*stride]
33 %define PASS8ROWS(base, base3, stride, stride3) \
34 [base], [base + stride], [base + 2*stride], [base3], \
35 [base3 + stride], [base3 + 2*stride], [base3 + stride3], [base3 + stride*4]
39 vperm2i128 m%4, m%2, m%3, q0301
40 vinserti128 m%2, m%2, xm%3, 1
41 %elif avx_enabled == 0
46 punpckh%1 m%4, m%2, m%3
53 punpckl%1 m%4, m%2, m%3
54 punpckh%1 m%2, m%2, m%3
59 unpcklps m%3, m%1, m%2
60 unpckhps m%1, m%1, m%2
64 %macro TRANSPOSE4x4B 5
65 SBUTTERFLY bw, %1, %2, %5
66 SBUTTERFLY bw, %3, %4, %5
67 SBUTTERFLY wd, %1, %3, %5
68 SBUTTERFLY wd, %2, %4, %5
72 %macro TRANSPOSE4x4W 5
73 SBUTTERFLY wd, %1, %2, %5
74 SBUTTERFLY wd, %3, %4, %5
75 SBUTTERFLY dq, %1, %3, %5
76 SBUTTERFLY dq, %2, %4, %5
80 %macro TRANSPOSE2x4x4B 5
81 SBUTTERFLY bw, %1, %2, %5
82 SBUTTERFLY bw, %3, %4, %5
83 SBUTTERFLY wd, %1, %3, %5
84 SBUTTERFLY wd, %2, %4, %5
85 SBUTTERFLY dq, %1, %2, %5
86 SBUTTERFLY dq, %3, %4, %5
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
98 %macro TRANSPOSE4x4D 5
99 SBUTTERFLY dq, %1, %2, %5
100 SBUTTERFLY dq, %3, %4, %5
101 SBUTTERFLY qdq, %1, %3, %5
102 SBUTTERFLY qdq, %2, %4, %5
106 ; identical behavior to TRANSPOSE4x4D, but using SSE1 float ops
107 %macro TRANSPOSE4x4PS 5
108 SBUTTERFLYPS %1, %2, %5
109 SBUTTERFLYPS %3, %4, %5
110 movlhps m%5, m%1, m%3
113 movlhps m%5, m%2, m%4
118 %macro TRANSPOSE8x4D 9-11
120 SBUTTERFLY dq, %1, %2, %9
121 SBUTTERFLY dq, %3, %4, %9
122 SBUTTERFLY dq, %5, %6, %9
123 SBUTTERFLY dq, %7, %8, %9
124 SBUTTERFLY qdq, %1, %3, %9
125 SBUTTERFLY qdq, %2, %4, %9
126 SBUTTERFLY qdq, %5, %7, %9
127 SBUTTERFLY qdq, %6, %8, %9
132 ; out: m0..m7, unless %11 in which case m2 is in %9
133 ; spills into %9 and %10
135 SBUTTERFLY dq, %1, %2, %7
138 SBUTTERFLY dq, %3, %4, %2
139 SBUTTERFLY dq, %5, %6, %2
140 SBUTTERFLY dq, %7, %8, %2
141 SBUTTERFLY qdq, %1, %3, %2
144 SBUTTERFLY qdq, %2, %4, %3
145 SBUTTERFLY qdq, %5, %7, %3
146 SBUTTERFLY qdq, %6, %8, %3
155 %macro TRANSPOSE8x8W 9-11
157 SBUTTERFLY wd, %1, %2, %9
158 SBUTTERFLY wd, %3, %4, %9
159 SBUTTERFLY wd, %5, %6, %9
160 SBUTTERFLY wd, %7, %8, %9
161 SBUTTERFLY dq, %1, %3, %9
162 SBUTTERFLY dq, %2, %4, %9
163 SBUTTERFLY dq, %5, %7, %9
164 SBUTTERFLY dq, %6, %8, %9
165 SBUTTERFLY qdq, %1, %5, %9
166 SBUTTERFLY qdq, %2, %6, %9
167 SBUTTERFLY qdq, %3, %7, %9
168 SBUTTERFLY qdq, %4, %8, %9
172 ; in: m0..m7, unless %11 in which case m6 is in %9
173 ; out: m0..m7, unless %11 in which case m4 is in %10
174 ; spills into %9 and %10
178 SBUTTERFLY wd, %1, %2, %7
181 SBUTTERFLY wd, %3, %4, %2
182 SBUTTERFLY wd, %5, %6, %2
183 SBUTTERFLY wd, %7, %8, %2
184 SBUTTERFLY dq, %1, %3, %2
187 SBUTTERFLY dq, %2, %4, %3
188 SBUTTERFLY dq, %5, %7, %3
189 SBUTTERFLY dq, %6, %8, %3
190 SBUTTERFLY qdq, %1, %5, %3
191 SBUTTERFLY qdq, %2, %6, %3
194 SBUTTERFLY qdq, %3, %7, %2
195 SBUTTERFLY qdq, %4, %8, %2
204 %macro TRANSPOSE16x16W 18-19
205 ; in: m0..m15, unless %19 in which case m6 is in %17
206 ; out: m0..m15, unless %19 in which case m4 is in %18
207 ; spills into %17 and %18
212 SBUTTERFLY dqqq, %1, %9, %7
213 SBUTTERFLY dqqq, %2, %10, %7
214 SBUTTERFLY dqqq, %3, %11, %7
215 SBUTTERFLY dqqq, %4, %12, %7
216 SBUTTERFLY dqqq, %5, %13, %7
217 SBUTTERFLY dqqq, %6, %14, %7
220 SBUTTERFLY dqqq, %7, %15, %14
221 SBUTTERFLY dqqq, %8, %16, %14
223 SBUTTERFLY wd, %1, %2, %14
224 SBUTTERFLY wd, %3, %4, %14
225 SBUTTERFLY wd, %5, %6, %14
226 SBUTTERFLY wd, %7, %8, %14
227 SBUTTERFLY wd, %9, %10, %14
228 SBUTTERFLY wd, %11, %12, %14
231 SBUTTERFLY wd, %13, %14, %12
232 SBUTTERFLY wd, %15, %16, %12
234 SBUTTERFLY dq, %1, %3, %12
235 SBUTTERFLY dq, %2, %4, %12
236 SBUTTERFLY dq, %5, %7, %12
237 SBUTTERFLY dq, %6, %8, %12
238 SBUTTERFLY dq, %9, %11, %12
241 SBUTTERFLY dq, %10, %12, %11
242 SBUTTERFLY dq, %13, %15, %11
243 SBUTTERFLY dq, %14, %16, %11
245 SBUTTERFLY qdq, %1, %5, %11
246 SBUTTERFLY qdq, %2, %6, %11
247 SBUTTERFLY qdq, %3, %7, %11
248 SBUTTERFLY qdq, %4, %8, %11
253 SBUTTERFLY qdq, %9, %13, %11
254 SBUTTERFLY qdq, %10, %14, %11
257 SBUTTERFLY qdq, %11, %15, %5
258 SBUTTERFLY qdq, %12, %16, %5
268 %macro TRANSPOSE_8X8B 8
270 %error "This macro does not support mmsize == 8"
276 TRANSPOSE4x4W %1, %3, %5, %7, %2
283 ; PABSW macro assumes %1 != %2, while ABS1/2 macros work in-place
287 %elif cpuflag(mmxext)
312 %elif cpuflag(mmxext) ; a, tmp
328 %elif cpuflag(mmxext) ; a, b, tmp0, tmp1
335 %else ; a, b, tmp0, tmp1
347 %macro ABSB 2 ; source mmreg, temp mmreg (unused for ssse3)
357 %macro ABSB2 4 ; src1, src2, tmp1, tmp2 (tmp1/2 unused for SSSE3)
392 movd %1, [%2-3] ;to avoid crossing a cacheline
409 %macro HADDD 2 ; sum junk
412 vextracti128 %2, %1, 1
417 %if cpuflag(xop) && sizeof%1 == 16
423 %if notcpuflag(xop) || sizeof%1 != 16
425 PSHUFLW %2, %1, q0032
436 %macro HADDW 2 ; reg, tmp
437 %if cpuflag(xop) && sizeof%1 == 16
447 %macro HADDPS 3 ; dst, src, tmp
461 palignr %1, %2, %3, %4
465 %elif cpuflag(mmx) ; [dst,] src1, src2, imm, tmp
477 psllq %%dst, (8-%3)*8
514 %elif cpuflag(3dnowext)
523 %macro DEINTB 5 ; mask, reg1, mask, reg2, optional src to fill masks from
525 pand m%3, m%5, m%4 ; src .. y6 .. y4
526 pand m%1, m%5, m%2 ; dst .. y6 .. y4
529 pand m%3, m%1, m%4 ; src .. y6 .. y4
530 pand m%1, m%1, m%2 ; dst .. y6 .. y4
532 psrlw m%2, 8 ; dst .. y7 .. y5
533 psrlw m%4, 8 ; src .. y7 .. y5
554 %macro SUMSUB_BADC 5-6
556 SUMSUB_BA %1, %2, %3, %6
557 SUMSUB_BA %1, %4, %5, %6
601 psra%1 m%5, m%2, 1 ; %3: %3>>1
602 psra%1 m%4, m%3, 1 ; %2: %2>>1
603 padd%1 m%4, m%2 ; %3: %3>>1+%2
604 psub%1 m%5, m%3 ; %2: %2>>1-%3
610 psra%1 m%3, 1 ; %3: %3>>1
611 psra%1 m%2, 1 ; %2: %2>>1
612 padd%1 m%3, %5 ; %3: %3>>1+%2
613 psub%1 m%2, %4 ; %2: %2>>1-%3
619 SUMSUB_BADC w, %4, %1, %3, %2, %5
620 SUMSUB_BA w, %3, %4, %5
621 SUMSUB2_AB w, %1, %2, %5
622 SWAP %1, %3, %4, %5, %2
624 SUMSUB_BADC w, %4, %1, %3, %2
627 SUMSUB2_AB w, %1, [%5], %2
634 SUMSUBD2_AB %1, %3, %5, %7, %6
635 ; %3: %3>>1-%5 %5: %3+%5>>1
636 SUMSUB_BA %1, %4, %2, %7
637 ; %4: %2+%4 %2: %2-%4
638 SUMSUB_BADC %1, %5, %4, %3, %2, %7
639 ; %5: %2+%4 + (%3+%5>>1)
640 ; %4: %2+%4 - (%3+%5>>1)
641 ; %3: %2-%4 + (%3>>1-%5)
642 ; %2: %2-%4 - (%3>>1-%5)
645 SUMSUBD2_AB %1, %3, %5, [%6], [%6+16]
647 SUMSUBD2_AB %1, %3, %5, [%6], [%6+32]
650 SUMSUB_BADC %1, %5, %4, %3, %2
653 ; %2: %2+%4 + (%3+%5>>1) row0
654 ; %3: %2-%4 + (%3>>1-%5) row1
655 ; %4: %2-%4 - (%3>>1-%5) row2
656 ; %5: %2+%4 - (%3+%5>>1) row3
681 movhps [%5+%6+32], m%1
682 movhps [%5+%6+40], m%2
683 movhps [%5+%6+48], m%3
684 movhps [%5+%6+56], m%4
687 %macro LOAD_DIFF_8x4P 7-10 r0,r2,0 ; 4x dest, 2x temp, 2x pointer, increment?
688 LOAD_DIFF m%1, m%5, m%7, [%8], [%9]
689 LOAD_DIFF m%2, m%6, m%7, [%8+r1], [%9+r3]
690 LOAD_DIFF m%3, m%5, m%7, [%8+2*r1], [%9+2*r3]
691 LOAD_DIFF m%4, m%6, m%7, [%8+r4], [%9+r5]
719 %macro STORE_DIFFx2 8 ; add1, add2, reg1, reg2, zero, shift, source, stride
734 %macro PMINUB 3 ; dst, src, ignored
737 %else ; dst, src, tmp
745 %if cpuflag(avx2) && %3 == 0
748 pshuflw %1, %2, (%3)*0x55
750 %elif cpuflag(mmxext)
751 pshufw %1, %2, (%3)*0x55
779 %macro CLIPUB 3 ;(dst, min, max)
784 %macro CLIPW 3 ;(dst, min, max)
789 %macro PMINSD_MMX 3 ; dst, src, tmp
797 %macro PMAXSD_MMX 3 ; dst, src, tmp
805 %macro CLIPD_MMX 3-4 ; src/dst, min, max, tmp
806 PMINSD_MMX %1, %3, %4
807 PMAXSD_MMX %1, %2, %4
810 %macro CLIPD_SSE2 3-4 ; src/dst, min (float), max (float), unused
817 %macro CLIPD_SSE41 3-4 ; src/dst, min, max, unused
822 %macro VBROADCASTSS 2 ; dst xmm/ymm, src m32
831 %macro VBROADCASTSD 2 ; dst xmm/ymm, src m64
832 %if cpuflag(avx) && mmsize == 32
842 %macro SHUFFLE_MASK_W 8
854 %macro PMOVSXWD 2; dst, src
866 ; Wrapper for non-FMA version of fmaddps
868 %if cpuflag(fma3) || cpuflag(fma4)
869 fmaddps %1, %2, %3, %4
895 %macro MOVHL 2 ; dst, src
899 punpckhqdq %1, %2, %2
901 pshufd %1, %2, q3232 ; pshufd is slow on some older CPUs, so only use it on more modern ones
903 movhlps %1, %2 ; may cause an int/float domain transition and has a dependency on dst