Spaces:
Runtime error
Runtime error
;***************************************************************************** | |
;* x86util.asm | |
;***************************************************************************** | |
;* Copyright (C) 2008-2010 x264 project | |
;* | |
;* Authors: Loren Merritt <[email protected]> | |
;* Holger Lubitz <[email protected]> | |
;* | |
;* This file is part of FFmpeg. | |
;* | |
;* FFmpeg is free software; you can redistribute it and/or | |
;* modify it under the terms of the GNU Lesser General Public | |
;* License as published by the Free Software Foundation; either | |
;* version 2.1 of the License, or (at your option) any later version. | |
;* | |
;* FFmpeg is distributed in the hope that it will be useful, | |
;* but WITHOUT ANY WARRANTY; without even the implied warranty of | |
;* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
;* Lesser General Public License for more details. | |
;* | |
;* You should have received a copy of the GNU Lesser General Public | |
;* License along with FFmpeg; if not, write to the Free Software | |
;* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
;****************************************************************************** | |
%define private_prefix ff | |
%define public_prefix avpriv | |
%define cpuflags_mmxext cpuflags_mmx2 | |
%include "libavutil/x86/x86inc.asm" | |
; expands to [base],...,[base+7*stride] | |
%define PASS8ROWS(base, base3, stride, stride3) \ | |
[base], [base + stride], [base + 2*stride], [base3], \ | |
[base3 + stride], [base3 + 2*stride], [base3 + stride3], [base3 + stride*4] | |
; Interleave low src0 with low src1 and store in src0, | |
; interleave high src0 with high src1 and store in src1. | |
; %1 - types | |
; %2 - index of the register with src0 | |
; %3 - index of the register with src1 | |
; %4 - index of the register for intermediate results | |
; example for %1 - wd: input: src0: x0 x1 x2 x3 z0 z1 z2 z3 | |
; src1: y0 y1 y2 y3 q0 q1 q2 q3 | |
; output: src0: x0 y0 x1 y1 x2 y2 x3 y3 | |
; src1: z0 q0 z1 q1 z2 q2 z3 q3 | |
%macro SBUTTERFLY 4 | |
%ifidn %1, dqqq | |
vperm2i128 m%4, m%2, m%3, q0301 | |
vinserti128 m%2, m%2, xm%3, 1 | |
%elif avx_enabled == 0 | |
mova m%4, m%2 | |
punpckl%1 m%2, m%3 | |
punpckh%1 m%4, m%3 | |
%else | |
punpckh%1 m%4, m%2, m%3 | |
punpckl%1 m%2, m%3 | |
%endif | |
SWAP %3, %4 | |
%endmacro | |
%macro SBUTTERFLY2 4 | |
punpckl%1 m%4, m%2, m%3 | |
punpckh%1 m%2, m%2, m%3 | |
SWAP %2, %4, %3 | |
%endmacro | |
%macro SBUTTERFLYPS 3 | |
unpcklps m%3, m%1, m%2 | |
unpckhps m%1, m%1, m%2 | |
SWAP %1, %3, %2 | |
%endmacro | |
%macro SBUTTERFLYPD 3 | |
movlhps m%3, m%1, m%2 | |
movhlps m%2, m%2, m%1 | |
SWAP %1, %3 | |
%endmacro | |
%macro TRANSPOSE4x4B 5 | |
SBUTTERFLY bw, %1, %2, %5 | |
SBUTTERFLY bw, %3, %4, %5 | |
SBUTTERFLY wd, %1, %3, %5 | |
SBUTTERFLY wd, %2, %4, %5 | |
SWAP %2, %3 | |
%endmacro | |
%macro TRANSPOSE4x4W 5 | |
SBUTTERFLY wd, %1, %2, %5 | |
SBUTTERFLY wd, %3, %4, %5 | |
SBUTTERFLY dq, %1, %3, %5 | |
SBUTTERFLY dq, %2, %4, %5 | |
SWAP %2, %3 | |
%endmacro | |
%macro TRANSPOSE2x4x4B 5 | |
SBUTTERFLY bw, %1, %2, %5 | |
SBUTTERFLY bw, %3, %4, %5 | |
SBUTTERFLY wd, %1, %3, %5 | |
SBUTTERFLY wd, %2, %4, %5 | |
SBUTTERFLY dq, %1, %2, %5 | |
SBUTTERFLY dq, %3, %4, %5 | |
%endmacro | |
%macro TRANSPOSE2x4x4W 5 | |
SBUTTERFLY wd, %1, %2, %5 | |
SBUTTERFLY wd, %3, %4, %5 | |
SBUTTERFLY dq, %1, %3, %5 | |
SBUTTERFLY dq, %2, %4, %5 | |
SBUTTERFLY qdq, %1, %2, %5 | |
SBUTTERFLY qdq, %3, %4, %5 | |
%endmacro | |
%macro TRANSPOSE4x4D 5 | |
SBUTTERFLY dq, %1, %2, %5 | |
SBUTTERFLY dq, %3, %4, %5 | |
SBUTTERFLY qdq, %1, %3, %5 | |
SBUTTERFLY qdq, %2, %4, %5 | |
SWAP %2, %3 | |
%endmacro | |
; identical behavior to TRANSPOSE4x4D, but using SSE1 float ops | |
%macro TRANSPOSE4x4PS 5 | |
SBUTTERFLYPS %1, %2, %5 | |
SBUTTERFLYPS %3, %4, %5 | |
SBUTTERFLYPD %1, %3, %5 | |
SBUTTERFLYPD %2, %4, %5 | |
SWAP %2, %3 | |
%endmacro | |
%macro TRANSPOSE8x4D 9-11 | |
%if ARCH_X86_64 | |
SBUTTERFLY dq, %1, %2, %9 | |
SBUTTERFLY dq, %3, %4, %9 | |
SBUTTERFLY dq, %5, %6, %9 | |
SBUTTERFLY dq, %7, %8, %9 | |
SBUTTERFLY qdq, %1, %3, %9 | |
SBUTTERFLY qdq, %2, %4, %9 | |
SBUTTERFLY qdq, %5, %7, %9 | |
SBUTTERFLY qdq, %6, %8, %9 | |
SWAP %2, %5 | |
SWAP %4, %7 | |
%else | |
; in: m0..m7 | |
; out: m0..m7, unless %11 in which case m2 is in %9 | |
; spills into %9 and %10 | |
movdqa %9, m%7 | |
SBUTTERFLY dq, %1, %2, %7 | |
movdqa %10, m%2 | |
movdqa m%7, %9 | |
SBUTTERFLY dq, %3, %4, %2 | |
SBUTTERFLY dq, %5, %6, %2 | |
SBUTTERFLY dq, %7, %8, %2 | |
SBUTTERFLY qdq, %1, %3, %2 | |
movdqa %9, m%3 | |
movdqa m%2, %10 | |
SBUTTERFLY qdq, %2, %4, %3 | |
SBUTTERFLY qdq, %5, %7, %3 | |
SBUTTERFLY qdq, %6, %8, %3 | |
SWAP %2, %5 | |
SWAP %4, %7 | |
%if %0<11 | |
movdqa m%3, %9 | |
%endif | |
%endif | |
%endmacro | |
%macro TRANSPOSE8x8W 9-11 | |
%if ARCH_X86_64 | |
SBUTTERFLY wd, %1, %2, %9 | |
SBUTTERFLY wd, %3, %4, %9 | |
SBUTTERFLY wd, %5, %6, %9 | |
SBUTTERFLY wd, %7, %8, %9 | |
SBUTTERFLY dq, %1, %3, %9 | |
SBUTTERFLY dq, %2, %4, %9 | |
SBUTTERFLY dq, %5, %7, %9 | |
SBUTTERFLY dq, %6, %8, %9 | |
SBUTTERFLY qdq, %1, %5, %9 | |
SBUTTERFLY qdq, %2, %6, %9 | |
SBUTTERFLY qdq, %3, %7, %9 | |
SBUTTERFLY qdq, %4, %8, %9 | |
SWAP %2, %5 | |
SWAP %4, %7 | |
%else | |
; in: m0..m7, unless %11 in which case m6 is in %9 | |
; out: m0..m7, unless %11 in which case m4 is in %10 | |
; spills into %9 and %10 | |
%if %0<11 | |
movdqa %9, m%7 | |
%endif | |
SBUTTERFLY wd, %1, %2, %7 | |
movdqa %10, m%2 | |
movdqa m%7, %9 | |
SBUTTERFLY wd, %3, %4, %2 | |
SBUTTERFLY wd, %5, %6, %2 | |
SBUTTERFLY wd, %7, %8, %2 | |
SBUTTERFLY dq, %1, %3, %2 | |
movdqa %9, m%3 | |
movdqa m%2, %10 | |
SBUTTERFLY dq, %2, %4, %3 | |
SBUTTERFLY dq, %5, %7, %3 | |
SBUTTERFLY dq, %6, %8, %3 | |
SBUTTERFLY qdq, %1, %5, %3 | |
SBUTTERFLY qdq, %2, %6, %3 | |
movdqa %10, m%2 | |
movdqa m%3, %9 | |
SBUTTERFLY qdq, %3, %7, %2 | |
SBUTTERFLY qdq, %4, %8, %2 | |
SWAP %2, %5 | |
SWAP %4, %7 | |
%if %0<11 | |
movdqa m%5, %10 | |
%endif | |
%endif | |
%endmacro | |
%macro TRANSPOSE16x16W 18-19 | |
; in: m0..m15, unless %19 in which case m6 is in %17 | |
; out: m0..m15, unless %19 in which case m4 is in %18 | |
; spills into %17 and %18 | |
%if %0 < 19 | |
mova %17, m%7 | |
%endif | |
SBUTTERFLY dqqq, %1, %9, %7 | |
SBUTTERFLY dqqq, %2, %10, %7 | |
SBUTTERFLY dqqq, %3, %11, %7 | |
SBUTTERFLY dqqq, %4, %12, %7 | |
SBUTTERFLY dqqq, %5, %13, %7 | |
SBUTTERFLY dqqq, %6, %14, %7 | |
mova %18, m%14 | |
mova m%7, %17 | |
SBUTTERFLY dqqq, %7, %15, %14 | |
SBUTTERFLY dqqq, %8, %16, %14 | |
SBUTTERFLY wd, %1, %2, %14 | |
SBUTTERFLY wd, %3, %4, %14 | |
SBUTTERFLY wd, %5, %6, %14 | |
SBUTTERFLY wd, %7, %8, %14 | |
SBUTTERFLY wd, %9, %10, %14 | |
SBUTTERFLY wd, %11, %12, %14 | |
mova %17, m%12 | |
mova m%14, %18 | |
SBUTTERFLY wd, %13, %14, %12 | |
SBUTTERFLY wd, %15, %16, %12 | |
SBUTTERFLY dq, %1, %3, %12 | |
SBUTTERFLY dq, %2, %4, %12 | |
SBUTTERFLY dq, %5, %7, %12 | |
SBUTTERFLY dq, %6, %8, %12 | |
SBUTTERFLY dq, %9, %11, %12 | |
mova %18, m%11 | |
mova m%12, %17 | |
SBUTTERFLY dq, %10, %12, %11 | |
SBUTTERFLY dq, %13, %15, %11 | |
SBUTTERFLY dq, %14, %16, %11 | |
SBUTTERFLY qdq, %1, %5, %11 | |
SBUTTERFLY qdq, %2, %6, %11 | |
SBUTTERFLY qdq, %3, %7, %11 | |
SBUTTERFLY qdq, %4, %8, %11 | |
SWAP %2, %5 | |
SWAP %4, %7 | |
SBUTTERFLY qdq, %9, %13, %11 | |
SBUTTERFLY qdq, %10, %14, %11 | |
mova m%11, %18 | |
mova %18, m%5 | |
SBUTTERFLY qdq, %11, %15, %5 | |
SBUTTERFLY qdq, %12, %16, %5 | |
%if %0 < 19 | |
mova m%5, %18 | |
%endif | |
SWAP %10, %13 | |
SWAP %12, %15 | |
%endmacro | |
%macro TRANSPOSE_8X8B 8 | |
%if mmsize == 8 | |
%error "This macro does not support mmsize == 8" | |
%endif | |
punpcklbw m%1, m%2 | |
punpcklbw m%3, m%4 | |
punpcklbw m%5, m%6 | |
punpcklbw m%7, m%8 | |
TRANSPOSE4x4W %1, %3, %5, %7, %2 | |
MOVHL m%2, m%1 | |
MOVHL m%4, m%3 | |
MOVHL m%6, m%5 | |
MOVHL m%8, m%7 | |
%endmacro | |
; PABSW macro assumes %1 != %2, while ABS1/2 macros work in-place | |
%macro PABSW 2 | |
%if cpuflag(ssse3) | |
pabsw %1, %2 | |
%elif cpuflag(mmxext) | |
pxor %1, %1 | |
psubw %1, %2 | |
pmaxsw %1, %2 | |
%else | |
pxor %1, %1 | |
pcmpgtw %1, %2 | |
pxor %2, %1 | |
psubw %2, %1 | |
SWAP %1, %2 | |
%endif | |
%endmacro | |
%macro PSIGNW 2 | |
%if cpuflag(ssse3) | |
psignw %1, %2 | |
%else | |
pxor %1, %2 | |
psubw %1, %2 | |
%endif | |
%endmacro | |
%macro ABS1 2 | |
%if cpuflag(ssse3) | |
pabsw %1, %1 | |
%elif cpuflag(mmxext) ; a, tmp | |
pxor %2, %2 | |
psubw %2, %1 | |
pmaxsw %1, %2 | |
%else ; a, tmp | |
pxor %2, %2 | |
pcmpgtw %2, %1 | |
pxor %1, %2 | |
psubw %1, %2 | |
%endif | |
%endmacro | |
%macro ABS2 4 | |
%if cpuflag(ssse3) | |
pabsw %1, %1 | |
pabsw %2, %2 | |
%elif cpuflag(mmxext) ; a, b, tmp0, tmp1 | |
pxor %3, %3 | |
pxor %4, %4 | |
psubw %3, %1 | |
psubw %4, %2 | |
pmaxsw %1, %3 | |
pmaxsw %2, %4 | |
%else ; a, b, tmp0, tmp1 | |
pxor %3, %3 | |
pxor %4, %4 | |
pcmpgtw %3, %1 | |
pcmpgtw %4, %2 | |
pxor %1, %3 | |
pxor %2, %4 | |
psubw %1, %3 | |
psubw %2, %4 | |
%endif | |
%endmacro | |
%macro ABSB 2 ; source mmreg, temp mmreg (unused for SSSE3) | |
%if cpuflag(ssse3) | |
pabsb %1, %1 | |
%else | |
pxor %2, %2 | |
psubb %2, %1 | |
pminub %1, %2 | |
%endif | |
%endmacro | |
%macro ABSB2 4 ; src1, src2, tmp1, tmp2 (tmp1/2 unused for SSSE3) | |
%if cpuflag(ssse3) | |
pabsb %1, %1 | |
pabsb %2, %2 | |
%else | |
pxor %3, %3 | |
pxor %4, %4 | |
psubb %3, %1 | |
psubb %4, %2 | |
pminub %1, %3 | |
pminub %2, %4 | |
%endif | |
%endmacro | |
%macro ABSD2 4 | |
pxor %3, %3 | |
pxor %4, %4 | |
pcmpgtd %3, %1 | |
pcmpgtd %4, %2 | |
pxor %1, %3 | |
pxor %2, %4 | |
psubd %1, %3 | |
psubd %2, %4 | |
%endmacro | |
%macro ABS4 6 | |
ABS2 %1, %2, %5, %6 | |
ABS2 %3, %4, %5, %6 | |
%endmacro | |
%macro SPLATB_LOAD 3 | |
%if cpuflag(ssse3) | |
movd %1, [%2-3] | |
pshufb %1, %3 | |
%else | |
movd %1, [%2-3] ;to avoid crossing a cacheline | |
punpcklbw %1, %1 | |
SPLATW %1, %1, 3 | |
%endif | |
%endmacro | |
%macro SPLATB_REG 3 | |
%if cpuflag(ssse3) | |
movd %1, %2d | |
pshufb %1, %3 | |
%else | |
movd %1, %2d | |
punpcklbw %1, %1 | |
SPLATW %1, %1, 0 | |
%endif | |
%endmacro | |
%macro HADDD 2 ; sum junk | |
%if sizeof%1 == 32 | |
%define %2 xmm%2 | |
vextracti128 %2, %1, 1 | |
%define %1 xmm%1 | |
paddd %1, %2 | |
%endif | |
%if mmsize >= 16 | |
%if cpuflag(xop) && sizeof%1 == 16 | |
vphadddq %1, %1 | |
%endif | |
movhlps %2, %1 | |
paddd %1, %2 | |
%endif | |
%if notcpuflag(xop) || sizeof%1 != 16 | |
%if cpuflag(mmxext) | |
PSHUFLW %2, %1, q0032 | |
%else ; mmx | |
mova %2, %1 | |
psrlq %2, 32 | |
%endif | |
paddd %1, %2 | |
%endif | |
%undef %1 | |
%undef %2 | |
%endmacro | |
%macro HADDW 2 ; reg, tmp | |
%if cpuflag(xop) && sizeof%1 == 16 | |
vphaddwq %1, %1 | |
movhlps %2, %1 | |
paddd %1, %2 | |
%else | |
pmaddwd %1, [pw_1] | |
HADDD %1, %2 | |
%endif | |
%endmacro | |
%macro HADDPS 3 ; dst, src, tmp | |
%if cpuflag(sse3) | |
haddps %1, %1, %2 | |
%else | |
movaps %3, %1 | |
shufps %1, %2, q2020 | |
shufps %3, %2, q3131 | |
addps %1, %3 | |
%endif | |
%endmacro | |
%macro PALIGNR 4-5 | |
%if cpuflag(ssse3) | |
%if %0==5 | |
palignr %1, %2, %3, %4 | |
%else | |
palignr %1, %2, %3 | |
%endif | |
%else ; [dst,] src1, src2, imm, tmp | |
%define %%dst %1 | |
%if %0==5 | |
%ifnidn %1, %2 | |
mova %%dst, %2 | |
%endif | |
%rotate 1 | |
%endif | |
%ifnidn %4, %2 | |
mova %4, %2 | |
%endif | |
%if mmsize==8 | |
psllq %%dst, (8-%3)*8 | |
psrlq %4, %3*8 | |
%else | |
pslldq %%dst, 16-%3 | |
psrldq %4, %3 | |
%endif | |
por %%dst, %4 | |
%endif | |
%endmacro | |
%macro PAVGB 2-4 | |
%if cpuflag(mmxext) | |
pavgb %1, %2 | |
%elif cpuflag(3dnow) | |
pavgusb %1, %2 | |
%elif cpuflag(mmx) | |
movu %3, %2 | |
por %3, %1 | |
pxor %1, %2 | |
pand %1, %4 | |
psrlq %1, 1 | |
psubb %3, %1 | |
SWAP %1, %3 | |
%endif | |
%endmacro | |
%macro PSHUFLW 1+ | |
%if mmsize == 8 | |
pshufw %1 | |
%else | |
pshuflw %1 | |
%endif | |
%endmacro | |
%macro PSWAPD 2 | |
%if cpuflag(mmxext) | |
pshufw %1, %2, q1032 | |
%elif cpuflag(3dnowext) | |
pswapd %1, %2 | |
%elif cpuflag(3dnow) | |
movq %1, %2 | |
psrlq %1, 32 | |
punpckldq %1, %2 | |
%endif | |
%endmacro | |
%macro DEINTB 5 ; mask, reg1, mask, reg2, optional src to fill masks from | |
%ifnum %5 | |
pand m%3, m%5, m%4 ; src .. y6 .. y4 | |
pand m%1, m%5, m%2 ; dst .. y6 .. y4 | |
%else | |
mova m%1, %5 | |
pand m%3, m%1, m%4 ; src .. y6 .. y4 | |
pand m%1, m%1, m%2 ; dst .. y6 .. y4 | |
%endif | |
psrlw m%2, 8 ; dst .. y7 .. y5 | |
psrlw m%4, 8 ; src .. y7 .. y5 | |
%endmacro | |
%macro SUMSUB_BA 3-4 | |
%if %0==3 | |
padd%1 m%2, m%3 | |
padd%1 m%3, m%3 | |
psub%1 m%3, m%2 | |
%else | |
%if avx_enabled == 0 | |
mova m%4, m%2 | |
padd%1 m%2, m%3 | |
psub%1 m%3, m%4 | |
%else | |
padd%1 m%4, m%2, m%3 | |
psub%1 m%3, m%2 | |
SWAP %2, %4 | |
%endif | |
%endif | |
%endmacro | |
%macro SUMSUB_BADC 5-6 | |
%if %0==6 | |
SUMSUB_BA %1, %2, %3, %6 | |
SUMSUB_BA %1, %4, %5, %6 | |
%else | |
padd%1 m%2, m%3 | |
padd%1 m%4, m%5 | |
padd%1 m%3, m%3 | |
padd%1 m%5, m%5 | |
psub%1 m%3, m%2 | |
psub%1 m%5, m%4 | |
%endif | |
%endmacro | |
%macro SUMSUB2_AB 4 | |
%ifnum %3 | |
psub%1 m%4, m%2, m%3 | |
psub%1 m%4, m%3 | |
padd%1 m%2, m%2 | |
padd%1 m%2, m%3 | |
%else | |
mova m%4, m%2 | |
padd%1 m%2, m%2 | |
padd%1 m%2, %3 | |
psub%1 m%4, %3 | |
psub%1 m%4, %3 | |
%endif | |
%endmacro | |
%macro SUMSUB2_BA 4 | |
%if avx_enabled == 0 | |
mova m%4, m%2 | |
padd%1 m%2, m%3 | |
padd%1 m%2, m%3 | |
psub%1 m%3, m%4 | |
psub%1 m%3, m%4 | |
%else | |
padd%1 m%4, m%2, m%3 | |
padd%1 m%4, m%3 | |
psub%1 m%3, m%2 | |
psub%1 m%3, m%2 | |
SWAP %2, %4 | |
%endif | |
%endmacro | |
%macro SUMSUBD2_AB 5 | |
%ifnum %4 | |
psra%1 m%5, m%2, 1 ; %3: %3>>1 | |
psra%1 m%4, m%3, 1 ; %2: %2>>1 | |
padd%1 m%4, m%2 ; %3: %3>>1+%2 | |
psub%1 m%5, m%3 ; %2: %2>>1-%3 | |
SWAP %2, %5 | |
SWAP %3, %4 | |
%else | |
mova %5, m%2 | |
mova %4, m%3 | |
psra%1 m%3, 1 ; %3: %3>>1 | |
psra%1 m%2, 1 ; %2: %2>>1 | |
padd%1 m%3, %5 ; %3: %3>>1+%2 | |
psub%1 m%2, %4 ; %2: %2>>1-%3 | |
%endif | |
%endmacro | |
%macro DCT4_1D 5 | |
%ifnum %5 | |
SUMSUB_BADC w, %4, %1, %3, %2, %5 | |
SUMSUB_BA w, %3, %4, %5 | |
SUMSUB2_AB w, %1, %2, %5 | |
SWAP %1, %3, %4, %5, %2 | |
%else | |
SUMSUB_BADC w, %4, %1, %3, %2 | |
SUMSUB_BA w, %3, %4 | |
mova [%5], m%2 | |
SUMSUB2_AB w, %1, [%5], %2 | |
SWAP %1, %3, %4, %2 | |
%endif | |
%endmacro | |
%macro IDCT4_1D 6-7 | |
%ifnum %6 | |
SUMSUBD2_AB %1, %3, %5, %7, %6 | |
; %3: %3>>1-%5 %5: %3+%5>>1 | |
SUMSUB_BA %1, %4, %2, %7 | |
; %4: %2+%4 %2: %2-%4 | |
SUMSUB_BADC %1, %5, %4, %3, %2, %7 | |
; %5: %2+%4 + (%3+%5>>1) | |
; %4: %2+%4 - (%3+%5>>1) | |
; %3: %2-%4 + (%3>>1-%5) | |
; %2: %2-%4 - (%3>>1-%5) | |
%else | |
%ifidn %1, w | |
SUMSUBD2_AB %1, %3, %5, [%6], [%6+16] | |
%else | |
SUMSUBD2_AB %1, %3, %5, [%6], [%6+32] | |
%endif | |
SUMSUB_BA %1, %4, %2 | |
SUMSUB_BADC %1, %5, %4, %3, %2 | |
%endif | |
SWAP %2, %5, %4 | |
; %2: %2+%4 + (%3+%5>>1) row0 | |
; %3: %2-%4 + (%3>>1-%5) row1 | |
; %4: %2-%4 - (%3>>1-%5) row2 | |
; %5: %2+%4 - (%3+%5>>1) row3 | |
%endmacro | |
%macro LOAD_DIFF 5 | |
%ifidn %3, none | |
movh %1, %4 | |
movh %2, %5 | |
punpcklbw %1, %2 | |
punpcklbw %2, %2 | |
psubw %1, %2 | |
%else | |
movh %1, %4 | |
punpcklbw %1, %3 | |
movh %2, %5 | |
punpcklbw %2, %3 | |
psubw %1, %2 | |
%endif | |
%endmacro | |
%macro STORE_DCT 6 | |
movq [%5+%6+ 0], m%1 | |
movq [%5+%6+ 8], m%2 | |
movq [%5+%6+16], m%3 | |
movq [%5+%6+24], m%4 | |
movhps [%5+%6+32], m%1 | |
movhps [%5+%6+40], m%2 | |
movhps [%5+%6+48], m%3 | |
movhps [%5+%6+56], m%4 | |
%endmacro | |
%macro LOAD_DIFF_8x4P 7-10 r0,r2,0 ; 4x dest, 2x temp, 2x pointer, increment? | |
LOAD_DIFF m%1, m%5, m%7, [%8], [%9] | |
LOAD_DIFF m%2, m%6, m%7, [%8+r1], [%9+r3] | |
LOAD_DIFF m%3, m%5, m%7, [%8+2*r1], [%9+2*r3] | |
LOAD_DIFF m%4, m%6, m%7, [%8+r4], [%9+r5] | |
%if %10 | |
lea %8, [%8+4*r1] | |
lea %9, [%9+4*r3] | |
%endif | |
%endmacro | |
%macro DIFFx2 6-7 | |
movh %3, %5 | |
punpcklbw %3, %4 | |
psraw %1, 6 | |
paddsw %1, %3 | |
movh %3, %6 | |
punpcklbw %3, %4 | |
psraw %2, 6 | |
paddsw %2, %3 | |
packuswb %2, %1 | |
%endmacro | |
%macro STORE_DIFF 4 | |
movh %2, %4 | |
punpcklbw %2, %3 | |
psraw %1, 6 | |
paddsw %1, %2 | |
packuswb %1, %1 | |
movh %4, %1 | |
%endmacro | |
%macro STORE_DIFFx2 8 ; add1, add2, reg1, reg2, zero, shift, source, stride | |
movh %3, [%7] | |
movh %4, [%7+%8] | |
psraw %1, %6 | |
psraw %2, %6 | |
punpcklbw %3, %5 | |
punpcklbw %4, %5 | |
paddw %3, %1 | |
paddw %4, %2 | |
packuswb %3, %5 | |
packuswb %4, %5 | |
movh [%7], %3 | |
movh [%7+%8], %4 | |
%endmacro | |
%macro PMINUB 3 ; dst, src, ignored | |
%if cpuflag(mmxext) | |
pminub %1, %2 | |
%else ; dst, src, tmp | |
mova %3, %1 | |
psubusb %3, %2 | |
psubb %1, %3 | |
%endif | |
%endmacro | |
%macro SPLATW 2-3 0 | |
%if cpuflag(avx2) && %3 == 0 | |
vpbroadcastw %1, %2 | |
%elif mmsize == 16 | |
pshuflw %1, %2, (%3)*0x55 | |
punpcklqdq %1, %1 | |
%elif cpuflag(mmxext) | |
pshufw %1, %2, (%3)*0x55 | |
%else | |
%ifnidn %1, %2 | |
mova %1, %2 | |
%endif | |
%if %3 & 2 | |
punpckhwd %1, %1 | |
%else | |
punpcklwd %1, %1 | |
%endif | |
%if %3 & 1 | |
punpckhwd %1, %1 | |
%else | |
punpcklwd %1, %1 | |
%endif | |
%endif | |
%endmacro | |
%macro SPLATD 1 | |
%if mmsize == 8 | |
punpckldq %1, %1 | |
%elif cpuflag(sse2) | |
pshufd %1, %1, 0 | |
%elif cpuflag(sse) | |
shufps %1, %1, 0 | |
%endif | |
%endmacro | |
%macro CLIPUB 3 ;(dst, min, max) | |
pmaxub %1, %2 | |
pminub %1, %3 | |
%endmacro | |
%macro CLIPW 3 ;(dst, min, max) | |
pmaxsw %1, %2 | |
pminsw %1, %3 | |
%endmacro | |
%macro PMINSD 3 ; dst, src, tmp/unused | |
%if cpuflag(sse4) | |
pminsd %1, %2 | |
%elif cpuflag(sse2) | |
cvtdq2ps %1, %1 | |
minps %1, %2 | |
cvtps2dq %1, %1 | |
%else | |
mova %3, %2 | |
pcmpgtd %3, %1 | |
pxor %1, %2 | |
pand %1, %3 | |
pxor %1, %2 | |
%endif | |
%endmacro | |
%macro PMAXSD 3 ; dst, src, tmp/unused | |
%if cpuflag(sse4) | |
pmaxsd %1, %2 | |
%else | |
mova %3, %1 | |
pcmpgtd %3, %2 | |
pand %1, %3 | |
pandn %3, %2 | |
por %1, %3 | |
%endif | |
%endmacro | |
%macro CLIPD 3-4 | |
%if cpuflag(sse4); src/dst, min, max, unused | |
pminsd %1, %3 | |
pmaxsd %1, %2 | |
%elif cpuflag(sse2) ; src/dst, min (float), max (float), unused | |
cvtdq2ps %1, %1 | |
minps %1, %3 | |
maxps %1, %2 | |
cvtps2dq %1, %1 | |
%else ; src/dst, min, max, tmp | |
PMINSD %1, %3, %4 | |
PMAXSD %1, %2, %4 | |
%endif | |
%endmacro | |
%macro VBROADCASTSS 2 ; dst xmm/ymm, src m32/xmm | |
%if cpuflag(avx2) | |
vbroadcastss %1, %2 | |
%elif cpuflag(avx) | |
%ifnum sizeof%2 ; avx1 register | |
shufps xmm%1, xmm%2, xmm%2, q0000 | |
%if sizeof%1 >= 32 ; mmsize>=32 | |
vinsertf128 %1, %1, xmm%1, 1 | |
%endif | |
%else ; avx1 memory | |
vbroadcastss %1, %2 | |
%endif | |
%else | |
%ifnum sizeof%2 ; sse register | |
shufps %1, %2, %2, q0000 | |
%else ; sse memory | |
movss %1, %2 | |
shufps %1, %1, 0 | |
%endif | |
%endif | |
%endmacro | |
%macro VBROADCASTSD 2 ; dst xmm/ymm, src m64 | |
%if cpuflag(avx) && mmsize == 32 | |
vbroadcastsd %1, %2 | |
%elif cpuflag(sse3) | |
movddup %1, %2 | |
%else ; sse2 | |
movsd %1, %2 | |
movlhps %1, %1 | |
%endif | |
%endmacro | |
%macro VPBROADCASTD 2 ; dst xmm/ymm, src m32/xmm | |
%if cpuflag(avx2) | |
vpbroadcastd %1, %2 | |
%elif cpuflag(avx) && sizeof%1 >= 32 | |
%error vpbroadcastd not possible with ymm on avx1. try vbroadcastss | |
%else | |
%ifnum sizeof%2 ; sse2 register | |
pshufd %1, %2, q0000 | |
%else ; sse memory | |
movd %1, %2 | |
pshufd %1, %1, 0 | |
%endif | |
%endif | |
%endmacro | |
%macro VBROADCASTI128 2 ; dst xmm/ymm, src : 128bits val | |
%if mmsize > 16 | |
vbroadcasti128 %1, %2 | |
%else | |
mova %1, %2 | |
%endif | |
%endmacro | |
%macro SHUFFLE_MASK_W 8 | |
%rep 8 | |
%if %1>=0x80 | |
db %1, %1 | |
%else | |
db %1*2 | |
db %1*2+1 | |
%endif | |
%rotate 1 | |
%endrep | |
%endmacro | |
%macro PMOVSXWD 2; dst, src | |
%if cpuflag(sse4) | |
pmovsxwd %1, %2 | |
%else | |
%ifnidn %1, %2 | |
mova %1, %2 | |
%endif | |
punpcklwd %1, %1 | |
psrad %1, 16 | |
%endif | |
%endmacro | |
; Wrapper for non-FMA version of fmaddps | |
%macro FMULADD_PS 5 | |
%if cpuflag(fma3) || cpuflag(fma4) | |
fmaddps %1, %2, %3, %4 | |
%elifidn %1, %4 | |
mulps %5, %2, %3 | |
addps %1, %4, %5 | |
%else | |
mulps %1, %2, %3 | |
addps %1, %4 | |
%endif | |
%endmacro | |
%macro LSHIFT 2 | |
%if mmsize > 8 | |
pslldq %1, %2 | |
%else | |
psllq %1, 8*(%2) | |
%endif | |
%endmacro | |
%macro RSHIFT 2 | |
%if mmsize > 8 | |
psrldq %1, %2 | |
%else | |
psrlq %1, 8*(%2) | |
%endif | |
%endmacro | |
%macro MOVHL 2 ; dst, src | |
%ifidn %1, %2 | |
punpckhqdq %1, %2 | |
%elif cpuflag(avx) | |
punpckhqdq %1, %2, %2 | |
%elif cpuflag(sse4) | |
pshufd %1, %2, q3232 ; pshufd is slow on some older CPUs, so only use it on more modern ones | |
%else | |
movhlps %1, %2 ; may cause an int/float domain transition and has a dependency on dst | |
%endif | |
%endmacro | |
; Horizontal Sum of Packed Single precision floats | |
; The resulting sum is in all elements. | |
%macro HSUMPS 2 ; dst/src, tmp | |
%if cpuflag(avx) | |
%if sizeof%1>=32 ; avx | |
vperm2f128 %2, %1, %1, (0)*16+(1) | |
addps %1, %2 | |
%endif | |
shufps %2, %1, %1, q1032 | |
addps %1, %2 | |
shufps %2, %1, %1, q0321 | |
addps %1, %2 | |
%else ; this form is a bit faster than the short avx-like emulation. | |
movaps %2, %1 | |
shufps %1, %1, q1032 | |
addps %1, %2 | |
movaps %2, %1 | |
shufps %1, %1, q0321 | |
addps %1, %2 | |
; all %1 members should be equal for as long as float a+b==b+a | |
%endif | |
%endmacro | |
; Emulate blendvps if not available | |
; | |
; src_b is destroyed when using emulation with logical operands | |
; SSE41 blendv instruction is hard coded to use xmm0 as mask | |
%macro BLENDVPS 3 ; dst/src_a, src_b, mask | |
%if cpuflag(avx) | |
blendvps %1, %1, %2, %3 | |
%elif cpuflag(sse4) | |
%ifnidn %3,xmm0 | |
%error sse41 blendvps uses xmm0 as default 3d operand, you used %3 | |
%endif | |
blendvps %1, %2, %3 | |
%else | |
xorps %2, %1 | |
andps %2, %3 | |
xorps %1, %2 | |
%endif | |
%endmacro | |
; Emulate pblendvb if not available | |
; | |
; src_b is destroyed when using emulation with logical operands | |
; SSE41 blendv instruction is hard coded to use xmm0 as mask | |
%macro PBLENDVB 3 ; dst/src_a, src_b, mask | |
%if cpuflag(avx) | |
%if cpuflag(avx) && notcpuflag(avx2) && sizeof%1 >= 32 | |
%error pblendb not possible with ymm on avx1, try blendvps. | |
%endif | |
pblendvb %1, %1, %2, %3 | |
%elif cpuflag(sse4) | |
%ifnidn %3,xmm0 | |
%error sse41 pblendvd uses xmm0 as default 3d operand, you used %3 | |
%endif | |
pblendvb %1, %2, %3 | |
%else | |
pxor %2, %1 | |
pand %2, %3 | |
pxor %1, %2 | |
%endif | |
%endmacro | |