mirror of https://github.com/FFmpeg/FFmpeg.git
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.
377 lines
9.2 KiB
377 lines
9.2 KiB
;****************************************************************************** |
|
;* FLAC DSP SIMD optimizations |
|
;* |
|
;* Copyright (C) 2014 Loren Merritt |
|
;* Copyright (C) 2014 James Almer |
|
;* |
|
;* 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 |
|
;****************************************************************************** |
|
|
|
%include "libavutil/x86/x86util.asm" |
|
|
|
SECTION_RODATA |
|
|
|
vector: db 0,1,4,5,8,9,12,13,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,0,1,4,5,8,9,12,13, |
|
|
|
SECTION .text |
|
|
|
%macro PMACSDQL 3 |
|
%if cpuflag(xop) |
|
pmacsdql %1, %2, %3, %1 |
|
%else |
|
pmuldq %2, %3 |
|
paddq %1, %2 |
|
%endif |
|
%endmacro |
|
|
|
%macro LPC_32 3 |
|
INIT_XMM %1 |
|
cglobal flac_lpc_%2, 5,6,5, decoded, coeffs, pred_order, qlevel, len, j |
|
sub lend, pred_orderd |
|
jle .ret |
|
movsxdifnidn pred_orderq, pred_orderd |
|
lea decodedq, [decodedq+pred_orderq*4-8] |
|
lea coeffsq, [coeffsq+pred_orderq*4] |
|
neg pred_orderq |
|
movd m4, qlevelm |
|
ALIGN 16 |
|
.loop_sample: |
|
movd m0, [decodedq+pred_orderq*4+8] |
|
add decodedq, 8 |
|
movd m1, [coeffsq+pred_orderq*4] |
|
pxor m2, m2 |
|
pxor m3, m3 |
|
lea jq, [pred_orderq+1] |
|
test jq, jq |
|
jz .end_order |
|
.loop_order: |
|
PMACSDQL m2, m0, m1 |
|
movd m0, [decodedq+jq*4] |
|
PMACSDQL m3, m1, m0 |
|
movd m1, [coeffsq+jq*4] |
|
inc jq |
|
jl .loop_order |
|
.end_order: |
|
PMACSDQL m2, m0, m1 |
|
%3 m2, m4 |
|
movd m0, [decodedq] |
|
paddd m0, m2 |
|
movd [decodedq], m0 |
|
sub lend, 2 |
|
jl .ret |
|
PMACSDQL m3, m1, m0 |
|
%3 m3, m4 |
|
movd m1, [decodedq+4] |
|
paddd m1, m3 |
|
movd [decodedq+4], m1 |
|
jg .loop_sample |
|
.ret: |
|
RET |
|
%endmacro |
|
|
|
LPC_32 sse4, 16, psrad |
|
LPC_32 sse4, 32, psrlq |
|
%if HAVE_XOP_EXTERNAL |
|
LPC_32 xop, 32, psrlq |
|
%endif |
|
|
|
INIT_XMM sse2 |
|
cglobal flac_wasted_32, 3,3,5, decoded, wasted, len |
|
shl lend, 2 |
|
add decodedq, lenq |
|
neg lenq |
|
movd m4, wastedd |
|
ALIGN 16 |
|
.loop: |
|
mova m0, [decodedq+lenq+mmsize*0] |
|
mova m1, [decodedq+lenq+mmsize*1] |
|
mova m2, [decodedq+lenq+mmsize*2] |
|
mova m3, [decodedq+lenq+mmsize*3] |
|
pslld m0, m4 |
|
pslld m1, m4 |
|
pslld m2, m4 |
|
pslld m3, m4 |
|
mova [decodedq+lenq+mmsize*0], m0 |
|
mova [decodedq+lenq+mmsize*1], m1 |
|
mova [decodedq+lenq+mmsize*2], m2 |
|
mova [decodedq+lenq+mmsize*3], m3 |
|
add lenq, mmsize * 4 |
|
jl .loop |
|
RET |
|
|
|
INIT_XMM sse4 |
|
cglobal flac_wasted_33, 4,4,5, decoded, residuals, wasted, len |
|
shl lend, 2 |
|
lea decodedq, [decodedq+lenq*2] |
|
add residualsq, lenq |
|
neg lenq |
|
movd m4, wastedd |
|
ALIGN 16 |
|
.loop: |
|
pmovsxdq m0, [residualsq+lenq+mmsize*0] |
|
pmovsxdq m1, [residualsq+lenq+mmsize/2] |
|
pmovsxdq m2, [residualsq+lenq+mmsize*1] |
|
pmovsxdq m3, [residualsq+lenq+mmsize*1+mmsize/2] |
|
psllq m0, m4 |
|
psllq m1, m4 |
|
psllq m2, m4 |
|
psllq m3, m4 |
|
mova [decodedq+lenq*2+mmsize*0], m0 |
|
mova [decodedq+lenq*2+mmsize*1], m1 |
|
mova [decodedq+lenq*2+mmsize*2], m2 |
|
mova [decodedq+lenq*2+mmsize*3], m3 |
|
add lenq, mmsize * 2 |
|
jl .loop |
|
RET |
|
|
|
;---------------------------------------------------------------------------------- |
|
;void ff_flac_decorrelate_[lrm]s_16_sse2(uint8_t **out, int32_t **in, int channels, |
|
; int len, int shift); |
|
;---------------------------------------------------------------------------------- |
|
%macro FLAC_DECORRELATE_16 3-4 |
|
cglobal flac_decorrelate_%1_16, 2, 4, 4, out, in0, in1, len |
|
%ifidn %1, indep2 |
|
VBROADCASTI128 m2, [vector] |
|
%endif |
|
%if ARCH_X86_32 |
|
mov lend, lenm |
|
%endif |
|
movd m3, r4m |
|
shl lend, 2 |
|
mov in1q, [in0q + gprsize] |
|
mov in0q, [in0q] |
|
mov outq, [outq] |
|
add in1q, lenq |
|
add in0q, lenq |
|
add outq, lenq |
|
neg lenq |
|
|
|
align 16 |
|
.loop: |
|
mova m0, [in0q + lenq] |
|
mova m1, [in1q + lenq] |
|
%ifidn %1, ms |
|
psrad m2, m1, 1 |
|
psubd m0, m2 |
|
%endif |
|
%ifnidn %1, indep2 |
|
p%4d m2, m0, m1 |
|
packssdw m%2, m%2 |
|
packssdw m%3, m%3 |
|
punpcklwd m%2, m%3 |
|
psllw m%2, m3 |
|
%else |
|
pslld m%2, m3 |
|
pslld m%3, m3 |
|
pshufb m%2, m%2, m2 |
|
pshufb m%3, m%3, m2 |
|
punpcklwd m%2, m%3 |
|
%endif |
|
mova [outq + lenq], m%2 |
|
add lenq, 16 |
|
jl .loop |
|
RET |
|
%endmacro |
|
|
|
INIT_XMM sse2 |
|
FLAC_DECORRELATE_16 ls, 0, 2, sub |
|
FLAC_DECORRELATE_16 rs, 2, 1, add |
|
FLAC_DECORRELATE_16 ms, 2, 0, add |
|
|
|
;---------------------------------------------------------------------------------- |
|
;void ff_flac_decorrelate_[lrm]s_32_sse2(uint8_t **out, int32_t **in, int channels, |
|
; int len, int shift); |
|
;---------------------------------------------------------------------------------- |
|
%macro FLAC_DECORRELATE_32 5 |
|
cglobal flac_decorrelate_%1_32, 2, 4, 4, out, in0, in1, len |
|
%if ARCH_X86_32 |
|
mov lend, lenm |
|
%endif |
|
movd m3, r4m |
|
mov in1q, [in0q + gprsize] |
|
mov in0q, [in0q] |
|
mov outq, [outq] |
|
sub in1q, in0q |
|
|
|
align 16 |
|
.loop: |
|
mova m0, [in0q] |
|
mova m1, [in0q + in1q] |
|
%ifidn %1, ms |
|
psrad m2, m1, 1 |
|
psubd m0, m2 |
|
%endif |
|
p%5d m2, m0, m1 |
|
pslld m%2, m3 |
|
pslld m%3, m3 |
|
|
|
SBUTTERFLY dq, %2, %3, %4 |
|
|
|
mova [outq ], m%2 |
|
mova [outq + mmsize], m%3 |
|
|
|
add in0q, mmsize |
|
add outq, mmsize*2 |
|
sub lend, mmsize/4 |
|
jg .loop |
|
RET |
|
%endmacro |
|
|
|
INIT_XMM sse2 |
|
FLAC_DECORRELATE_32 ls, 0, 2, 1, sub |
|
FLAC_DECORRELATE_32 rs, 2, 1, 0, add |
|
FLAC_DECORRELATE_32 ms, 2, 0, 1, add |
|
|
|
;----------------------------------------------------------------------------------------- |
|
;void ff_flac_decorrelate_indep<ch>_<bps>_<opt>(uint8_t **out, int32_t **in, int channels, |
|
; int len, int shift); |
|
;----------------------------------------------------------------------------------------- |
|
;%1 = bps |
|
;%2 = channels |
|
;%3 = last xmm reg used |
|
;%4 = word/dword (shift instruction) |
|
%macro FLAC_DECORRELATE_INDEP 4 |
|
%define REPCOUNT %2/(32/%1) ; 16bits = channels / 2; 32bits = channels |
|
cglobal flac_decorrelate_indep%2_%1, 2, %2+2, %3+1, out, in0, in1, len, in2, in3, in4, in5, in6, in7 |
|
%if ARCH_X86_32 |
|
%if %2 == 6 |
|
DEFINE_ARGS out, in0, in1, in2, in3, in4, in5 |
|
%define lend dword r3m |
|
%else |
|
mov lend, lenm |
|
%endif |
|
%endif |
|
movd m%3, r4m |
|
|
|
%assign %%i 1 |
|
%rep %2-1 |
|
mov in %+ %%i %+ q, [in0q+%%i*gprsize] |
|
%assign %%i %%i+1 |
|
%endrep |
|
|
|
mov in0q, [in0q] |
|
mov outq, [outq] |
|
|
|
%assign %%i 1 |
|
%rep %2-1 |
|
sub in %+ %%i %+ q, in0q |
|
%assign %%i %%i+1 |
|
%endrep |
|
|
|
align 16 |
|
.loop: |
|
mova m0, [in0q] |
|
|
|
%assign %%i 1 |
|
%rep REPCOUNT-1 |
|
mova m %+ %%i, [in0q + in %+ %%i %+ q] |
|
%assign %%i %%i+1 |
|
%endrep |
|
|
|
%if %1 == 32 |
|
|
|
%if %2 == 8 |
|
TRANSPOSE8x4D 0, 1, 2, 3, 4, 5, 6, 7, 8 |
|
%elif %2 == 6 |
|
SBUTTERFLY dq, 0, 1, 6 |
|
SBUTTERFLY dq, 2, 3, 6 |
|
SBUTTERFLY dq, 4, 5, 6 |
|
|
|
punpcklqdq m6, m0, m2 |
|
punpckhqdq m2, m4 |
|
shufps m4, m0, 0xe4 |
|
punpcklqdq m0, m1, m3 |
|
punpckhqdq m3, m5 |
|
shufps m5, m1, 0xe4 |
|
SWAP 0,6,1,4,5,3 |
|
%elif %2 == 4 |
|
TRANSPOSE4x4D 0, 1, 2, 3, 4 |
|
%else ; %2 == 2 |
|
SBUTTERFLY dq, 0, 1, 2 |
|
%endif |
|
|
|
%else ; %1 == 16 |
|
|
|
%if %2 == 8 |
|
packssdw m0, [in0q + in4q] |
|
packssdw m1, [in0q + in5q] |
|
packssdw m2, [in0q + in6q] |
|
packssdw m3, [in0q + in7q] |
|
TRANSPOSE2x4x4W 0, 1, 2, 3, 4 |
|
%elif %2 == 6 |
|
packssdw m0, [in0q + in3q] |
|
packssdw m1, [in0q + in4q] |
|
packssdw m2, [in0q + in5q] |
|
pshufd m3, m0, q1032 |
|
punpcklwd m0, m1 |
|
punpckhwd m1, m2 |
|
punpcklwd m2, m3 |
|
|
|
shufps m3, m0, m2, q2020 |
|
shufps m0, m1, q2031 |
|
shufps m2, m1, q3131 |
|
shufps m1, m2, m3, q3120 |
|
shufps m3, m0, q0220 |
|
shufps m0, m2, q3113 |
|
SWAP 2, 0, 3 |
|
%else ; %2 == 4 |
|
packssdw m0, [in0q + in2q] |
|
packssdw m1, [in0q + in3q] |
|
SBUTTERFLY wd, 0, 1, 2 |
|
SBUTTERFLY dq, 0, 1, 2 |
|
%endif |
|
|
|
%endif |
|
|
|
%assign %%i 0 |
|
%rep REPCOUNT |
|
psll%4 m %+ %%i, m%3 |
|
%assign %%i %%i+1 |
|
%endrep |
|
|
|
%assign %%i 0 |
|
%rep REPCOUNT |
|
mova [outq + %%i*mmsize], m %+ %%i |
|
%assign %%i %%i+1 |
|
%endrep |
|
|
|
add in0q, mmsize |
|
add outq, mmsize*REPCOUNT |
|
sub lend, mmsize/4 |
|
jg .loop |
|
RET |
|
%endmacro |
|
|
|
INIT_XMM ssse3 |
|
FLAC_DECORRELATE_16 indep2, 0, 1 ; Reuse stereo 16bits macro |
|
FLAC_DECORRELATE_INDEP 32, 2, 3, d |
|
FLAC_DECORRELATE_INDEP 16, 4, 3, w |
|
FLAC_DECORRELATE_INDEP 32, 4, 5, d |
|
FLAC_DECORRELATE_INDEP 16, 6, 4, w |
|
FLAC_DECORRELATE_INDEP 32, 6, 7, d |
|
%if ARCH_X86_64 |
|
FLAC_DECORRELATE_INDEP 16, 8, 5, w |
|
FLAC_DECORRELATE_INDEP 32, 8, 9, d |
|
%endif |
|
|
|
INIT_XMM avx |
|
FLAC_DECORRELATE_INDEP 32, 4, 5, d |
|
FLAC_DECORRELATE_INDEP 32, 6, 7, d |
|
%if ARCH_X86_64 |
|
FLAC_DECORRELATE_INDEP 16, 8, 5, w |
|
FLAC_DECORRELATE_INDEP 32, 8, 9, d |
|
%endif
|
|
|