ffmpeg/libavcodec/x86/cfhdencdsp.asm

433 lines
9.3 KiB
NASM

;******************************************************************************
;* x86-optimized functions for the CFHD encoder
;* Copyright (c) 2021 Paul B Mahol
;*
;* 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
pw_p1_n1: dw 1, -1, 1, -1, 1, -1, 1, -1
pw_n1_p1: dw -1, 1, -1, 1, -1, 1, -1, 1
pw_p5_n11: dw 5, -11, 5, -11, 5, -11, 5, -11
pw_n5_p11: dw -5, 11, -5, 11, -5, 11, -5, 11
pw_p11_n5: dw 11, -5, 11, -5, 11, -5, 11, -5
pw_n11_p5: dw -11, 5, -11, 5, -11, 5, -11, 5
pd_4: times 4 dd 4
pw_n4: times 8 dw -4
cextern pw_m1
cextern pw_1
cextern pw_4
SECTION .text
%if ARCH_X86_64
INIT_XMM sse2
cglobal cfhdenc_horiz_filter, 8, 10, 11, input, low, high, istride, lwidth, hwidth, width, y, x, temp
shl istrideq, 1
shl lwidthq, 1
shl hwidthq, 1
mova m7, [pd_4]
mova m8, [pw_1]
mova m9, [pw_m1]
mova m10,[pw_p1_n1]
movsxdifnidn yq, yd
movsxdifnidn widthq, widthd
neg yq
.looph:
movsx xq, word [inputq]
movsx tempq, word [inputq + 2]
add tempq, xq
movd xm0, tempd
packssdw m0, m0
movd tempd, m0
mov word [lowq], tempw
movsx xq, word [inputq]
imul xq, 5
movsx tempq, word [inputq + 2]
imul tempq, -11
add tempq, xq
movsx xq, word [inputq + 4]
imul xq, 4
add tempq, xq
movsx xq, word [inputq + 6]
imul xq, 4
add tempq, xq
movsx xq, word [inputq + 8]
imul xq, -1
add tempq, xq
movsx xq, word [inputq + 10]
imul xq, -1
add tempq, xq
add tempq, 4
sar tempq, 3
movd xm0, tempd
packssdw m0, m0
movd tempd, m0
mov word [highq], tempw
mov xq, 2
.loopw:
movu m0, [inputq + xq * 2]
movu m1, [inputq + xq * 2 + mmsize]
pmaddwd m0, m8
pmaddwd m1, m8
packssdw m0, m1
movu [lowq+xq], m0
movu m2, [inputq + xq * 2 - 4]
movu m3, [inputq + xq * 2 - 4 + mmsize]
pmaddwd m2, m9
pmaddwd m3, m9
movu m0, [inputq + xq * 2 + 4]
movu m1, [inputq + xq * 2 + 4 + mmsize]
pmaddwd m0, m8
pmaddwd m1, m8
paddd m0, m2
paddd m1, m3
paddd m0, m7
paddd m1, m7
psrad m0, 3
psrad m1, 3
movu m5, [inputq + xq * 2 + 0]
movu m6, [inputq + xq * 2 + mmsize]
pmaddwd m5, m10
pmaddwd m6, m10
paddd m0, m5
paddd m1, m6
packssdw m0, m1
movu [highq+xq], m0
add xq, mmsize
cmp xq, widthq
jl .loopw
add lowq, widthq
add highq, widthq
lea inputq, [inputq + widthq * 2]
movsx xq, word [inputq - 4]
movsx tempq, word [inputq - 2]
add tempq, xq
movd xm0, tempd
packssdw m0, m0
movd tempd, m0
mov word [lowq-2], tempw
movsx tempq, word [inputq - 4]
imul tempq, 11
movsx xq, word [inputq - 2]
imul xq, -5
add tempq, xq
movsx xq, word [inputq - 6]
imul xq, -4
add tempq, xq
movsx xq, word [inputq - 8]
imul xq, -4
add tempq, xq
movsx xq, word [inputq - 10]
add tempq, xq
movsx xq, word [inputq - 12]
add tempq, xq
add tempq, 4
sar tempq, 3
movd xm0, tempd
packssdw m0, m0
movd tempd, m0
mov word [highq-2], tempw
sub inputq, widthq
sub inputq, widthq
sub highq, widthq
sub lowq, widthq
add lowq, lwidthq
add highq, hwidthq
add inputq, istrideq
add yq, 1
jl .looph
RET
%endif
%if ARCH_X86_64
INIT_XMM sse2
cglobal cfhdenc_vert_filter, 8, 11, 14, input, low, high, istride, lwidth, hwidth, width, height, x, y, pos
shl istrideq, 1
shl widthd, 1
sub heightd, 2
xor xq, xq
mova m7, [pd_4]
mova m8, [pw_1]
mova m9, [pw_m1]
mova m10,[pw_p1_n1]
mova m11,[pw_n1_p1]
mova m12,[pw_4]
mova m13,[pw_n4]
.loopw:
mov yq, 2
mov posq, xq
movu m0, [inputq + posq]
add posq, istrideq
movu m1, [inputq + posq]
paddsw m0, m1
movu [lowq + xq], m0
mov posq, xq
movu m0, [inputq + posq]
add posq, istrideq
movu m1, [inputq + posq]
add posq, istrideq
movu m2, [inputq + posq]
add posq, istrideq
movu m3, [inputq + posq]
add posq, istrideq
movu m4, [inputq + posq]
add posq, istrideq
movu m5, [inputq + posq]
mova m6, m0
punpcklwd m0, m1
punpckhwd m1, m6
mova m6, m2
punpcklwd m2, m3
punpckhwd m3, m6
mova m6, m4
punpcklwd m4, m5
punpckhwd m5, m6
pmaddwd m0, [pw_p5_n11]
pmaddwd m1, [pw_n11_p5]
pmaddwd m2, m12
pmaddwd m3, m12
pmaddwd m4, m9
pmaddwd m5, m9
paddd m0, m2
paddd m1, m3
paddd m0, m4
paddd m1, m5
paddd m0, m7
paddd m1, m7
psrad m0, 3
psrad m1, 3
packssdw m0, m1
movu [highq + xq], m0
.looph:
mov posq, istrideq
imul posq, yq
add posq, xq
movu m0, [inputq + posq]
add posq, istrideq
movu m1, [inputq + posq]
paddsw m0, m1
mov posq, lwidthq
imul posq, yq
add posq, xq
movu [lowq + posq], m0
add yq, -2
mov posq, istrideq
imul posq, yq
add posq, xq
movu m0, [inputq + posq]
add posq, istrideq
movu m1, [inputq + posq]
add posq, istrideq
movu m2, [inputq + posq]
add posq, istrideq
movu m3, [inputq + posq]
add posq, istrideq
movu m4, [inputq + posq]
add posq, istrideq
movu m5, [inputq + posq]
add yq, 2
mova m6, m0
punpcklwd m0, m1
punpckhwd m1, m6
mova m6, m2
punpcklwd m2, m3
punpckhwd m3, m6
mova m6, m4
punpcklwd m4, m5
punpckhwd m5, m6
pmaddwd m0, m9
pmaddwd m1, m9
pmaddwd m2, m10
pmaddwd m3, m11
pmaddwd m4, m8
pmaddwd m5, m8
paddd m0, m4
paddd m1, m5
paddd m0, m7
paddd m1, m7
psrad m0, 3
psrad m1, 3
paddd m0, m2
paddd m1, m3
packssdw m0, m1
mov posq, hwidthq
imul posq, yq
add posq, xq
movu [highq + posq], m0
add yq, 2
cmp yq, heightq
jl .looph
mov posq, istrideq
imul posq, yq
add posq, xq
movu m0, [inputq + posq]
add posq, istrideq
movu m1, [inputq + posq]
paddsw m0, m1
mov posq, lwidthq
imul posq, yq
add posq, xq
movu [lowq + posq], m0
sub yq, 4
mov posq, istrideq
imul posq, yq
add posq, xq
movu m0, [inputq + posq]
add posq, istrideq
movu m1, [inputq + posq]
add posq, istrideq
movu m2, [inputq + posq]
add posq, istrideq
movu m3, [inputq + posq]
add posq, istrideq
movu m4, [inputq + posq]
add posq, istrideq
movu m5, [inputq + posq]
add yq, 4
mova m6, m0
punpcklwd m0, m1
punpckhwd m1, m6
mova m6, m2
punpcklwd m2, m3
punpckhwd m3, m6
mova m6, m4
punpcklwd m4, m5
punpckhwd m5, m6
pmaddwd m0, m8
pmaddwd m1, m8
pmaddwd m2, m13
pmaddwd m3, m13
pmaddwd m4, [pw_p11_n5]
pmaddwd m5, [pw_n5_p11]
paddd m4, m2
paddd m5, m3
paddd m4, m0
paddd m5, m1
paddd m4, m7
paddd m5, m7
psrad m4, 3
psrad m5, 3
packssdw m4, m5
mov posq, hwidthq
imul posq, yq
add posq, xq
movu [highq + posq], m4
add xq, mmsize
cmp xq, widthq
jl .loopw
RET
%endif