mirror of https://git.ffmpeg.org/ffmpeg.git
1555 lines
49 KiB
NASM
1555 lines
49 KiB
NASM
;******************************************************************************
|
|
;* VP9 Intra prediction SIMD optimizations
|
|
;*
|
|
;* Copyright (c) 2013 Ronald S. Bultje <rsbultje gmail com>
|
|
;*
|
|
;* Parts based on:
|
|
;* H.264 intra prediction asm optimizations
|
|
;* Copyright (c) 2010 Fiona Glaser
|
|
;* Copyright (c) 2010 Holger Lubitz
|
|
;* Copyright (c) 2010 Loren Merritt
|
|
;* Copyright (c) 2010 Ronald S. Bultje
|
|
;*
|
|
;* 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 32
|
|
|
|
pw_m256: times 16 dw -256
|
|
pw_m255: times 16 dw -255
|
|
pw_512: times 8 dw 512
|
|
pw_1024: times 8 dw 1024
|
|
pw_2048: times 8 dw 2048
|
|
pw_4096: times 8 dw 4096
|
|
pw_8192: times 8 dw 8192
|
|
|
|
pb_4x3_4x2_4x1_4x0: times 4 db 3
|
|
times 4 db 2
|
|
times 4 db 1
|
|
times 4 db 0
|
|
pb_8x1_8x0: times 8 db 1
|
|
times 8 db 0
|
|
pb_8x3_8x2: times 8 db 3
|
|
times 8 db 2
|
|
pb_0to5_2x7: db 0, 1, 2, 3, 4, 5, 7, 7
|
|
times 8 db -1
|
|
pb_0to6_9x7: db 0, 1, 2, 3, 4, 5, 6
|
|
times 9 db 7
|
|
pb_1to6_10x7: db 1, 2, 3, 4, 5, 6
|
|
times 10 db 7
|
|
pb_2to6_3x7:
|
|
pb_2to6_11x7: db 2, 3, 4, 5, 6
|
|
times 11 db 7
|
|
pb_1toE_2xF: db 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 15
|
|
pb_2toE_3xF: db 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 15, 15
|
|
pb_13456_3xm1: db 1, 3, 4, 5, 6
|
|
times 3 db -1
|
|
pb_6012_4xm1: db 6, 0, 1, 2
|
|
times 4 db -1
|
|
pb_6xm1_246_8toE: times 6 db -1
|
|
db 2, 4, 6, 8, 9, 10, 11, 12, 13, 14
|
|
pb_6xm1_BDF_0to6: times 6 db -1
|
|
db 11, 13, 15, 0, 1, 2, 3, 4, 5, 6
|
|
pb_02468ACE_13579BDF: db 0, 2, 4, 6, 8, 10, 12, 14, 1, 3, 5, 7, 9, 11, 13, 15
|
|
pb_7to1_9x0: db 7, 6, 5, 4
|
|
pb_3to1_5x0: db 3, 2, 1
|
|
times 9 db 0
|
|
pb_Fto0: db 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0
|
|
|
|
pb_2: times 32 db 2
|
|
pb_15: times 16 db 15
|
|
|
|
cextern pb_1
|
|
cextern pb_3
|
|
|
|
SECTION .text
|
|
|
|
; dc_NxN(uint8_t *dst, ptrdiff_t stride, const uint8_t *l, const uint8_t *a)
|
|
|
|
INIT_MMX ssse3
|
|
cglobal vp9_ipred_dc_4x4, 4, 4, 0, dst, stride, l, a
|
|
movd m0, [lq]
|
|
punpckldq m0, [aq]
|
|
pxor m1, m1
|
|
psadbw m0, m1
|
|
pmulhrsw m0, [pw_4096]
|
|
pshufb m0, m1
|
|
movd [dstq+strideq*0], m0
|
|
movd [dstq+strideq*1], m0
|
|
lea dstq, [dstq+strideq*2]
|
|
movd [dstq+strideq*0], m0
|
|
movd [dstq+strideq*1], m0
|
|
RET
|
|
|
|
INIT_MMX ssse3
|
|
cglobal vp9_ipred_dc_8x8, 4, 4, 0, dst, stride, l, a
|
|
movq m0, [lq]
|
|
movq m1, [aq]
|
|
DEFINE_ARGS dst, stride, stride3
|
|
lea stride3q, [strideq*3]
|
|
pxor m2, m2
|
|
psadbw m0, m2
|
|
psadbw m1, m2
|
|
paddw m0, m1
|
|
pmulhrsw m0, [pw_2048]
|
|
pshufb m0, m2
|
|
movq [dstq+strideq*0], m0
|
|
movq [dstq+strideq*1], m0
|
|
movq [dstq+strideq*2], m0
|
|
movq [dstq+stride3q ], m0
|
|
lea dstq, [dstq+strideq*4]
|
|
movq [dstq+strideq*0], m0
|
|
movq [dstq+strideq*1], m0
|
|
movq [dstq+strideq*2], m0
|
|
movq [dstq+stride3q ], m0
|
|
RET
|
|
|
|
INIT_XMM ssse3
|
|
cglobal vp9_ipred_dc_16x16, 4, 4, 3, dst, stride, l, a
|
|
mova m0, [lq]
|
|
mova m1, [aq]
|
|
DEFINE_ARGS dst, stride, stride3, cnt
|
|
lea stride3q, [strideq*3]
|
|
pxor m2, m2
|
|
psadbw m0, m2
|
|
psadbw m1, m2
|
|
paddw m0, m1
|
|
movhlps m1, m0
|
|
paddw m0, m1
|
|
pmulhrsw m0, [pw_1024]
|
|
pshufb m0, m2
|
|
mov cntd, 4
|
|
.loop:
|
|
mova [dstq+strideq*0], m0
|
|
mova [dstq+strideq*1], m0
|
|
mova [dstq+strideq*2], m0
|
|
mova [dstq+stride3q ], m0
|
|
lea dstq, [dstq+strideq*4]
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
|
|
INIT_XMM ssse3
|
|
cglobal vp9_ipred_dc_32x32, 4, 4, 5, dst, stride, l, a
|
|
mova m0, [lq]
|
|
mova m1, [lq+16]
|
|
mova m2, [aq]
|
|
mova m3, [aq+16]
|
|
DEFINE_ARGS dst, stride, stride3, cnt
|
|
lea stride3q, [strideq*3]
|
|
pxor m4, m4
|
|
psadbw m0, m4
|
|
psadbw m1, m4
|
|
psadbw m2, m4
|
|
psadbw m3, m4
|
|
paddw m0, m1
|
|
paddw m2, m3
|
|
paddw m0, m2
|
|
movhlps m1, m0
|
|
paddw m0, m1
|
|
pmulhrsw m0, [pw_512]
|
|
pshufb m0, m4
|
|
mov cntd, 8
|
|
.loop:
|
|
mova [dstq+strideq*0+ 0], m0
|
|
mova [dstq+strideq*0+16], m0
|
|
mova [dstq+strideq*1+ 0], m0
|
|
mova [dstq+strideq*1+16], m0
|
|
mova [dstq+strideq*2+ 0], m0
|
|
mova [dstq+strideq*2+16], m0
|
|
mova [dstq+stride3q + 0], m0
|
|
mova [dstq+stride3q +16], m0
|
|
lea dstq, [dstq+strideq*4]
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
|
|
%if HAVE_AVX2_EXTERNAL
|
|
INIT_YMM avx2
|
|
cglobal vp9_ipred_dc_32x32, 4, 4, 3, dst, stride, l, a
|
|
mova m0, [lq]
|
|
mova m1, [aq]
|
|
DEFINE_ARGS dst, stride, stride3, cnt
|
|
lea stride3q, [strideq*3]
|
|
pxor m2, m2
|
|
psadbw m0, m2
|
|
psadbw m1, m2
|
|
paddw m0, m1
|
|
vextracti128 xm1, m0, 1
|
|
paddw xm0, xm1
|
|
movhlps xm1, xm0
|
|
paddw xm0, xm1
|
|
pmulhrsw xm0, [pw_512]
|
|
vpbroadcastb m0, xm0
|
|
mov cntd, 4
|
|
.loop:
|
|
mova [dstq+strideq*0], m0
|
|
mova [dstq+strideq*1], m0
|
|
mova [dstq+strideq*2], m0
|
|
mova [dstq+stride3q ], m0
|
|
lea dstq, [dstq+strideq*4]
|
|
mova [dstq+strideq*0], m0
|
|
mova [dstq+strideq*1], m0
|
|
mova [dstq+strideq*2], m0
|
|
mova [dstq+stride3q ], m0
|
|
lea dstq, [dstq+strideq*4]
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
%endif
|
|
|
|
; dc_top/left_NxN(uint8_t *dst, ptrdiff_t stride, const uint8_t *l, const uint8_t *a)
|
|
|
|
%macro DC_1D_FUNCS 2 ; dir (top or left), arg (a or l)
|
|
INIT_MMX ssse3
|
|
cglobal vp9_ipred_dc_%1_4x4, 4, 4, 0, dst, stride, l, a
|
|
movd m0, [%2q]
|
|
pxor m1, m1
|
|
psadbw m0, m1
|
|
pmulhrsw m0, [pw_8192]
|
|
pshufb m0, m1
|
|
movd [dstq+strideq*0], m0
|
|
movd [dstq+strideq*1], m0
|
|
lea dstq, [dstq+strideq*2]
|
|
movd [dstq+strideq*0], m0
|
|
movd [dstq+strideq*1], m0
|
|
RET
|
|
|
|
INIT_MMX ssse3
|
|
cglobal vp9_ipred_dc_%1_8x8, 4, 4, 0, dst, stride, l, a
|
|
movq m0, [%2q]
|
|
DEFINE_ARGS dst, stride, stride3
|
|
lea stride3q, [strideq*3]
|
|
pxor m1, m1
|
|
psadbw m0, m1
|
|
pmulhrsw m0, [pw_4096]
|
|
pshufb m0, m1
|
|
movq [dstq+strideq*0], m0
|
|
movq [dstq+strideq*1], m0
|
|
movq [dstq+strideq*2], m0
|
|
movq [dstq+stride3q ], m0
|
|
lea dstq, [dstq+strideq*4]
|
|
movq [dstq+strideq*0], m0
|
|
movq [dstq+strideq*1], m0
|
|
movq [dstq+strideq*2], m0
|
|
movq [dstq+stride3q ], m0
|
|
RET
|
|
|
|
INIT_XMM ssse3
|
|
cglobal vp9_ipred_dc_%1_16x16, 4, 4, 3, dst, stride, l, a
|
|
mova m0, [%2q]
|
|
DEFINE_ARGS dst, stride, stride3, cnt
|
|
lea stride3q, [strideq*3]
|
|
pxor m2, m2
|
|
psadbw m0, m2
|
|
movhlps m1, m0
|
|
paddw m0, m1
|
|
pmulhrsw m0, [pw_2048]
|
|
pshufb m0, m2
|
|
mov cntd, 4
|
|
.loop:
|
|
mova [dstq+strideq*0], m0
|
|
mova [dstq+strideq*1], m0
|
|
mova [dstq+strideq*2], m0
|
|
mova [dstq+stride3q ], m0
|
|
lea dstq, [dstq+strideq*4]
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
|
|
INIT_XMM ssse3
|
|
cglobal vp9_ipred_dc_%1_32x32, 4, 4, 3, dst, stride, l, a
|
|
mova m0, [%2q]
|
|
mova m1, [%2q+16]
|
|
DEFINE_ARGS dst, stride, stride3, cnt
|
|
lea stride3q, [strideq*3]
|
|
pxor m2, m2
|
|
psadbw m0, m2
|
|
psadbw m1, m2
|
|
paddw m0, m1
|
|
movhlps m1, m0
|
|
paddw m0, m1
|
|
pmulhrsw m0, [pw_1024]
|
|
pshufb m0, m2
|
|
mov cntd, 8
|
|
.loop:
|
|
mova [dstq+strideq*0+ 0], m0
|
|
mova [dstq+strideq*0+16], m0
|
|
mova [dstq+strideq*1+ 0], m0
|
|
mova [dstq+strideq*1+16], m0
|
|
mova [dstq+strideq*2+ 0], m0
|
|
mova [dstq+strideq*2+16], m0
|
|
mova [dstq+stride3q + 0], m0
|
|
mova [dstq+stride3q +16], m0
|
|
lea dstq, [dstq+strideq*4]
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
|
|
%if HAVE_AVX2_EXTERNAL
|
|
INIT_YMM avx2
|
|
cglobal vp9_ipred_dc_%1_32x32, 4, 4, 3, dst, stride, l, a
|
|
mova m0, [%2q]
|
|
DEFINE_ARGS dst, stride, stride3, cnt
|
|
lea stride3q, [strideq*3]
|
|
pxor m2, m2
|
|
psadbw m0, m2
|
|
vextracti128 xm1, m0, 1
|
|
paddw xm0, xm1
|
|
movhlps xm1, xm0
|
|
paddw xm0, xm1
|
|
pmulhrsw xm0, [pw_1024]
|
|
vpbroadcastb m0, xm0
|
|
mov cntd, 4
|
|
.loop:
|
|
mova [dstq+strideq*0], m0
|
|
mova [dstq+strideq*1], m0
|
|
mova [dstq+strideq*2], m0
|
|
mova [dstq+stride3q ], m0
|
|
lea dstq, [dstq+strideq*4]
|
|
mova [dstq+strideq*0], m0
|
|
mova [dstq+strideq*1], m0
|
|
mova [dstq+strideq*2], m0
|
|
mova [dstq+stride3q ], m0
|
|
lea dstq, [dstq+strideq*4]
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
%endif
|
|
%endmacro
|
|
|
|
DC_1D_FUNCS top, a
|
|
DC_1D_FUNCS left, l
|
|
|
|
; v
|
|
|
|
INIT_MMX mmx
|
|
cglobal vp9_ipred_v_8x8, 4, 4, 0, dst, stride, l, a
|
|
movq m0, [aq]
|
|
DEFINE_ARGS dst, stride, stride3
|
|
lea stride3q, [strideq*3]
|
|
movq [dstq+strideq*0], m0
|
|
movq [dstq+strideq*1], m0
|
|
movq [dstq+strideq*2], m0
|
|
movq [dstq+stride3q ], m0
|
|
lea dstq, [dstq+strideq*4]
|
|
movq [dstq+strideq*0], m0
|
|
movq [dstq+strideq*1], m0
|
|
movq [dstq+strideq*2], m0
|
|
movq [dstq+stride3q ], m0
|
|
RET
|
|
|
|
INIT_XMM sse2
|
|
cglobal vp9_ipred_v_16x16, 4, 4, 1, dst, stride, l, a
|
|
mova m0, [aq]
|
|
DEFINE_ARGS dst, stride, stride3, cnt
|
|
lea stride3q, [strideq*3]
|
|
mov cntd, 4
|
|
.loop:
|
|
mova [dstq+strideq*0], m0
|
|
mova [dstq+strideq*1], m0
|
|
mova [dstq+strideq*2], m0
|
|
mova [dstq+stride3q ], m0
|
|
lea dstq, [dstq+strideq*4]
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
|
|
INIT_XMM sse2
|
|
cglobal vp9_ipred_v_32x32, 4, 4, 2, dst, stride, l, a
|
|
mova m0, [aq]
|
|
mova m1, [aq+16]
|
|
DEFINE_ARGS dst, stride, stride3, cnt
|
|
lea stride3q, [strideq*3]
|
|
mov cntd, 8
|
|
.loop:
|
|
mova [dstq+strideq*0+ 0], m0
|
|
mova [dstq+strideq*0+16], m1
|
|
mova [dstq+strideq*1+ 0], m0
|
|
mova [dstq+strideq*1+16], m1
|
|
mova [dstq+strideq*2+ 0], m0
|
|
mova [dstq+strideq*2+16], m1
|
|
mova [dstq+stride3q + 0], m0
|
|
mova [dstq+stride3q +16], m1
|
|
lea dstq, [dstq+strideq*4]
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
|
|
%if HAVE_AVX2_EXTERNAL
|
|
INIT_YMM avx2
|
|
cglobal vp9_ipred_v_32x32, 4, 4, 1, dst, stride, l, a
|
|
mova m0, [aq]
|
|
DEFINE_ARGS dst, stride, stride3, cnt
|
|
lea stride3q, [strideq*3]
|
|
mov cntd, 4
|
|
.loop:
|
|
mova [dstq+strideq*0], m0
|
|
mova [dstq+strideq*1], m0
|
|
mova [dstq+strideq*2], m0
|
|
mova [dstq+stride3q ], m0
|
|
lea dstq, [dstq+strideq*4]
|
|
mova [dstq+strideq*0], m0
|
|
mova [dstq+strideq*1], m0
|
|
mova [dstq+strideq*2], m0
|
|
mova [dstq+stride3q ], m0
|
|
lea dstq, [dstq+strideq*4]
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
%endif
|
|
|
|
; h
|
|
|
|
INIT_XMM ssse3
|
|
cglobal vp9_ipred_h_4x4, 3, 4, 1, dst, stride, l, stride3
|
|
movd m0, [lq]
|
|
pshufb m0, [pb_4x3_4x2_4x1_4x0]
|
|
lea stride3q, [strideq*3]
|
|
movd [dstq+strideq*0], m0
|
|
psrldq m0, 4
|
|
movd [dstq+strideq*1], m0
|
|
psrldq m0, 4
|
|
movd [dstq+strideq*2], m0
|
|
psrldq m0, 4
|
|
movd [dstq+stride3q ], m0
|
|
RET
|
|
|
|
%macro H_XMM_FUNCS 1
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_h_8x8, 3, 5, 4, dst, stride, l, stride3, cnt
|
|
mova m2, [pb_8x1_8x0]
|
|
mova m3, [pb_8x3_8x2]
|
|
lea stride3q, [strideq*3]
|
|
mov cntq, 1
|
|
.loop:
|
|
movd m0, [lq+cntq*4]
|
|
pshufb m1, m0, m3
|
|
pshufb m0, m2
|
|
movq [dstq+strideq*0], m1
|
|
movhps [dstq+strideq*1], m1
|
|
movq [dstq+strideq*2], m0
|
|
movhps [dstq+stride3q ], m0
|
|
lea dstq, [dstq+strideq*4]
|
|
dec cntq
|
|
jge .loop
|
|
RET
|
|
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_h_16x16, 3, 5, 8, dst, stride, l, stride3, cnt
|
|
mova m5, [pb_1]
|
|
mova m6, [pb_2]
|
|
mova m7, [pb_3]
|
|
pxor m4, m4
|
|
lea stride3q, [strideq*3]
|
|
mov cntq, 3
|
|
.loop:
|
|
movd m3, [lq+cntq*4]
|
|
pshufb m0, m3, m7
|
|
pshufb m1, m3, m6
|
|
mova [dstq+strideq*0], m0
|
|
mova [dstq+strideq*1], m1
|
|
pshufb m2, m3, m5
|
|
pshufb m3, m4
|
|
mova [dstq+strideq*2], m2
|
|
mova [dstq+stride3q ], m3
|
|
lea dstq, [dstq+strideq*4]
|
|
dec cntq
|
|
jge .loop
|
|
RET
|
|
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_h_32x32, 3, 5, 8, dst, stride, l, stride3, cnt
|
|
mova m5, [pb_1]
|
|
mova m6, [pb_2]
|
|
mova m7, [pb_3]
|
|
pxor m4, m4
|
|
lea stride3q, [strideq*3]
|
|
mov cntq, 7
|
|
.loop:
|
|
movd m3, [lq+cntq*4]
|
|
pshufb m0, m3, m7
|
|
pshufb m1, m3, m6
|
|
mova [dstq+strideq*0+ 0], m0
|
|
mova [dstq+strideq*0+16], m0
|
|
mova [dstq+strideq*1+ 0], m1
|
|
mova [dstq+strideq*1+16], m1
|
|
pshufb m2, m3, m5
|
|
pshufb m3, m4
|
|
mova [dstq+strideq*2+ 0], m2
|
|
mova [dstq+strideq*2+16], m2
|
|
mova [dstq+stride3q + 0], m3
|
|
mova [dstq+stride3q +16], m3
|
|
lea dstq, [dstq+strideq*4]
|
|
dec cntq
|
|
jge .loop
|
|
RET
|
|
%endmacro
|
|
|
|
H_XMM_FUNCS ssse3
|
|
H_XMM_FUNCS avx
|
|
|
|
%if HAVE_AVX2_EXTERNAL
|
|
INIT_YMM avx2
|
|
cglobal vp9_ipred_h_32x32, 3, 5, 8, dst, stride, l, stride3, cnt
|
|
mova m5, [pb_1]
|
|
mova m6, [pb_2]
|
|
mova m7, [pb_3]
|
|
pxor m4, m4
|
|
lea stride3q, [strideq*3]
|
|
mov cntq, 7
|
|
.loop:
|
|
movd xm3, [lq+cntq*4]
|
|
vinserti128 m3, m3, xm3, 1
|
|
pshufb m0, m3, m7
|
|
pshufb m1, m3, m6
|
|
mova [dstq+strideq*0], m0
|
|
mova [dstq+strideq*1], m1
|
|
pshufb m2, m3, m5
|
|
pshufb m3, m4
|
|
mova [dstq+strideq*2], m2
|
|
mova [dstq+stride3q ], m3
|
|
lea dstq, [dstq+strideq*4]
|
|
dec cntq
|
|
jge .loop
|
|
RET
|
|
%endif
|
|
|
|
; tm
|
|
|
|
INIT_MMX ssse3
|
|
cglobal vp9_ipred_tm_4x4, 4, 4, 0, dst, stride, l, a
|
|
pxor m1, m1
|
|
pinsrw m2, [aq-1], 0
|
|
movd m0, [aq]
|
|
DEFINE_ARGS dst, stride, l, cnt
|
|
mova m3, [pw_m256]
|
|
mova m4, [pw_m255]
|
|
pshufb m2, m3
|
|
punpcklbw m0, m1
|
|
psubw m0, m2
|
|
mov cntq, 1
|
|
.loop:
|
|
pinsrw m2, [lq+cntq*2], 0
|
|
pshufb m1, m2, m4
|
|
pshufb m2, m3
|
|
paddw m1, m0
|
|
paddw m2, m0
|
|
packuswb m1, m1
|
|
packuswb m2, m2
|
|
movd [dstq+strideq*0], m1
|
|
movd [dstq+strideq*1], m2
|
|
lea dstq, [dstq+strideq*2]
|
|
dec cntq
|
|
jge .loop
|
|
RET
|
|
|
|
%macro TM_XMM_FUNCS 1
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_tm_8x8, 4, 4, 5, dst, stride, l, a
|
|
pxor m1, m1
|
|
pinsrw m2, [aq-1], 0
|
|
movh m0, [aq]
|
|
DEFINE_ARGS dst, stride, l, cnt
|
|
mova m3, [pw_m256]
|
|
mova m4, [pw_m255]
|
|
pshufb m2, m3
|
|
punpcklbw m0, m1
|
|
psubw m0, m2
|
|
mov cntq, 3
|
|
.loop:
|
|
pinsrw m2, [lq+cntq*2], 0
|
|
pshufb m1, m2, m4
|
|
pshufb m2, m3
|
|
paddw m1, m0
|
|
paddw m2, m0
|
|
packuswb m1, m2
|
|
movh [dstq+strideq*0], m1
|
|
movhps [dstq+strideq*1], m1
|
|
lea dstq, [dstq+strideq*2]
|
|
dec cntq
|
|
jge .loop
|
|
RET
|
|
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_tm_16x16, 4, 4, 8, dst, stride, l, a
|
|
pxor m3, m3
|
|
pinsrw m2, [aq-1], 0
|
|
mova m0, [aq]
|
|
DEFINE_ARGS dst, stride, l, cnt
|
|
mova m4, [pw_m256]
|
|
mova m5, [pw_m255]
|
|
pshufb m2, m4
|
|
punpckhbw m1, m0, m3
|
|
punpcklbw m0, m3
|
|
psubw m1, m2
|
|
psubw m0, m2
|
|
mov cntq, 7
|
|
.loop:
|
|
pinsrw m7, [lq+cntq*2], 0
|
|
pshufb m3, m7, m5
|
|
pshufb m7, m4
|
|
paddw m2, m3, m0
|
|
paddw m3, m1
|
|
paddw m6, m7, m0
|
|
paddw m7, m1
|
|
packuswb m2, m3
|
|
packuswb m6, m7
|
|
mova [dstq+strideq*0], m2
|
|
mova [dstq+strideq*1], m6
|
|
lea dstq, [dstq+strideq*2]
|
|
dec cntq
|
|
jge .loop
|
|
RET
|
|
|
|
%if ARCH_X86_64
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_tm_32x32, 4, 4, 14, dst, stride, l, a
|
|
pxor m5, m5
|
|
pinsrw m4, [aq-1], 0
|
|
mova m0, [aq]
|
|
mova m2, [aq+16]
|
|
DEFINE_ARGS dst, stride, l, cnt
|
|
mova m8, [pw_m256]
|
|
mova m9, [pw_m255]
|
|
pshufb m4, m8
|
|
punpckhbw m1, m0, m5
|
|
punpckhbw m3, m2, m5
|
|
punpcklbw m0, m5
|
|
punpcklbw m2, m5
|
|
psubw m1, m4
|
|
psubw m0, m4
|
|
psubw m3, m4
|
|
psubw m2, m4
|
|
mov cntq, 15
|
|
.loop:
|
|
pinsrw m13, [lq+cntq*2], 0
|
|
pshufb m7, m13, m9
|
|
pshufb m13, m8
|
|
paddw m4, m7, m0
|
|
paddw m5, m7, m1
|
|
paddw m6, m7, m2
|
|
paddw m7, m3
|
|
paddw m10, m13, m0
|
|
paddw m11, m13, m1
|
|
paddw m12, m13, m2
|
|
paddw m13, m3
|
|
packuswb m4, m5
|
|
packuswb m6, m7
|
|
packuswb m10, m11
|
|
packuswb m12, m13
|
|
mova [dstq+strideq*0+ 0], m4
|
|
mova [dstq+strideq*0+16], m6
|
|
mova [dstq+strideq*1+ 0], m10
|
|
mova [dstq+strideq*1+16], m12
|
|
lea dstq, [dstq+strideq*2]
|
|
dec cntq
|
|
jge .loop
|
|
RET
|
|
%endif
|
|
%endmacro
|
|
|
|
TM_XMM_FUNCS ssse3
|
|
TM_XMM_FUNCS avx
|
|
|
|
%if HAVE_AVX2_EXTERNAL
|
|
INIT_YMM avx2
|
|
cglobal vp9_ipred_tm_32x32, 4, 4, 8, dst, stride, l, a
|
|
pxor m3, m3
|
|
pinsrw xm2, [aq-1], 0
|
|
vinserti128 m2, m2, xm2, 1
|
|
mova m0, [aq]
|
|
DEFINE_ARGS dst, stride, l, cnt
|
|
mova m4, [pw_m256]
|
|
mova m5, [pw_m255]
|
|
pshufb m2, m4
|
|
punpckhbw m1, m0, m3
|
|
punpcklbw m0, m3
|
|
psubw m1, m2
|
|
psubw m0, m2
|
|
mov cntq, 15
|
|
.loop:
|
|
pinsrw xm7, [lq+cntq*2], 0
|
|
vinserti128 m7, m7, xm7, 1
|
|
pshufb m3, m7, m5
|
|
pshufb m7, m4
|
|
paddw m2, m3, m0
|
|
paddw m3, m1
|
|
paddw m6, m7, m0
|
|
paddw m7, m1
|
|
packuswb m2, m3
|
|
packuswb m6, m7
|
|
mova [dstq+strideq*0], m2
|
|
mova [dstq+strideq*1], m6
|
|
lea dstq, [dstq+strideq*2]
|
|
dec cntq
|
|
jge .loop
|
|
RET
|
|
%endif
|
|
|
|
; dl
|
|
|
|
%macro LOWPASS 4 ; left [dst], center, right, tmp
|
|
pxor m%4, m%1, m%3
|
|
pand m%4, [pb_1]
|
|
pavgb m%1, m%3
|
|
psubusb m%1, m%4
|
|
pavgb m%1, m%2
|
|
%endmacro
|
|
|
|
INIT_MMX ssse3
|
|
cglobal vp9_ipred_dl_4x4, 4, 4, 0, dst, stride, l, a
|
|
movq m1, [aq]
|
|
pshufb m0, m1, [pb_0to5_2x7]
|
|
pshufb m2, m1, [pb_2to6_3x7]
|
|
psrlq m1, 8
|
|
LOWPASS 0, 1, 2, 3
|
|
|
|
pshufw m1, m0, q3321
|
|
movd [dstq+strideq*0], m0
|
|
movd [dstq+strideq*2], m1
|
|
psrlq m0, 8
|
|
psrlq m1, 8
|
|
add dstq, strideq
|
|
movd [dstq+strideq*0], m0
|
|
movd [dstq+strideq*2], m1
|
|
RET
|
|
|
|
%macro DL_XMM_FUNCS 1
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_dl_8x8, 4, 4, 4, dst, stride, stride5, a
|
|
movq m0, [aq]
|
|
lea stride5q, [strideq*5]
|
|
pshufb m1, m0, [pb_1to6_10x7]
|
|
psrldq m2, m1, 1
|
|
shufps m0, m1, q3210
|
|
LOWPASS 0, 1, 2, 3
|
|
|
|
pshufd m1, m0, q3321
|
|
movq [dstq+strideq*0], m0
|
|
movq [dstq+strideq*4], m1
|
|
psrldq m0, 1
|
|
psrldq m1, 1
|
|
movq [dstq+strideq*1], m0
|
|
movq [dstq+stride5q ], m1
|
|
lea dstq, [dstq+strideq*2]
|
|
psrldq m0, 1
|
|
psrldq m1, 1
|
|
movq [dstq+strideq*0], m0
|
|
movq [dstq+strideq*4], m1
|
|
psrldq m0, 1
|
|
psrldq m1, 1
|
|
movq [dstq+strideq*1], m0
|
|
movq [dstq+stride5q ], m1
|
|
RET
|
|
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_dl_16x16, 4, 4, 6, dst, stride, l, a
|
|
mova m5, [pb_1toE_2xF]
|
|
mova m0, [aq]
|
|
pshufb m1, m0, m5
|
|
pshufb m2, m1, m5
|
|
pshufb m4, m0, [pb_15]
|
|
LOWPASS 0, 1, 2, 3
|
|
DEFINE_ARGS dst, stride, cnt, stride9
|
|
lea stride9q, [strideq*3]
|
|
mov cntd, 4
|
|
lea stride9q, [stride9q*3]
|
|
|
|
.loop:
|
|
movhlps m4, m0
|
|
mova [dstq+strideq*0], m0
|
|
pshufb m0, m5
|
|
mova [dstq+strideq*8], m4
|
|
movhlps m4, m0
|
|
mova [dstq+strideq*1], m0
|
|
pshufb m0, m5
|
|
mova [dstq+stride9q ], m4
|
|
lea dstq, [dstq+strideq*2]
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_dl_32x32, 4, 5, 8, dst, stride, cnt, a, dst16
|
|
mova m5, [pb_1toE_2xF]
|
|
mova m0, [aq]
|
|
mova m1, [aq+16]
|
|
palignr m2, m1, m0, 1
|
|
palignr m3, m1, m0, 2
|
|
LOWPASS 0, 2, 3, 4
|
|
pshufb m2, m1, m5
|
|
pshufb m3, m2, m5
|
|
pshufb m6, m1, [pb_15]
|
|
LOWPASS 1, 2, 3, 4
|
|
mova m7, m6
|
|
lea dst16q, [dstq +strideq*8]
|
|
mov cntd, 8
|
|
lea dst16q, [dst16q+strideq*8]
|
|
.loop:
|
|
movhlps m7, m1
|
|
mova [dstq +strideq*0+ 0], m0
|
|
mova [dstq +strideq*0+16], m1
|
|
movhps [dstq+strideq*8+ 0], m0
|
|
movq [dstq +strideq*8+ 8], m1
|
|
mova [dstq +strideq*8+16], m7
|
|
mova [dst16q+strideq*0+ 0], m1
|
|
mova [dst16q+strideq*0+16], m6
|
|
mova [dst16q+strideq*8+ 0], m7
|
|
mova [dst16q+strideq*8+16], m6
|
|
%if cpuflag(avx)
|
|
vpalignr m0, m1, m0, 1
|
|
pshufb m1, m5
|
|
%else
|
|
palignr m2, m1, m0, 1
|
|
pshufb m1, m5
|
|
mova m0, m2
|
|
%endif
|
|
add dstq, strideq
|
|
add dst16q, strideq
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
%endmacro
|
|
|
|
DL_XMM_FUNCS ssse3
|
|
DL_XMM_FUNCS avx
|
|
|
|
; dr
|
|
|
|
INIT_MMX ssse3
|
|
cglobal vp9_ipred_dr_4x4, 4, 4, 0, dst, stride, l, a
|
|
movd m0, [lq]
|
|
punpckldq m0, [aq-1]
|
|
movd m1, [aq+3]
|
|
DEFINE_ARGS dst, stride, stride3
|
|
lea stride3q, [strideq*3]
|
|
palignr m1, m0, 1
|
|
psrlq m2, m1, 8
|
|
LOWPASS 0, 1, 2, 3
|
|
|
|
movd [dstq+stride3q ], m0
|
|
psrlq m0, 8
|
|
movd [dstq+strideq*2], m0
|
|
psrlq m0, 8
|
|
movd [dstq+strideq*1], m0
|
|
psrlq m0, 8
|
|
movd [dstq+strideq*0], m0
|
|
RET
|
|
|
|
%macro DR_XMM_FUNCS 1
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_dr_8x8, 4, 4, 4, dst, stride, l, a
|
|
movq m1, [lq]
|
|
movhps m1, [aq-1]
|
|
movd m2, [aq+7]
|
|
DEFINE_ARGS dst, stride, stride3
|
|
lea stride3q, [strideq*3]
|
|
pslldq m0, m1, 1
|
|
palignr m2, m1, 1
|
|
LOWPASS 0, 1, 2, 3
|
|
|
|
movhps [dstq+strideq*0], m0
|
|
pslldq m0, 1
|
|
movhps [dstq+strideq*1], m0
|
|
pslldq m0, 1
|
|
movhps [dstq+strideq*2], m0
|
|
pslldq m0, 1
|
|
movhps [dstq+stride3q ], m0
|
|
pslldq m0, 1
|
|
lea dstq, [dstq+strideq*4]
|
|
movhps [dstq+strideq*0], m0
|
|
pslldq m0, 1
|
|
movhps [dstq+strideq*1], m0
|
|
pslldq m0, 1
|
|
movhps [dstq+strideq*2], m0
|
|
pslldq m0, 1
|
|
movhps [dstq+stride3q ], m0
|
|
RET
|
|
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_dr_16x16, 4, 4, 6, dst, stride, l, a
|
|
mova m1, [lq]
|
|
movu m2, [aq-1]
|
|
movd m4, [aq+15]
|
|
DEFINE_ARGS dst, stride, stride9, cnt
|
|
lea stride9q, [strideq *3]
|
|
mov cntd, 4
|
|
lea stride9q, [stride9q*3]
|
|
palignr m4, m2, 1
|
|
palignr m3, m2, m1, 15
|
|
LOWPASS 3, 2, 4, 5
|
|
pslldq m0, m1, 1
|
|
palignr m2, m1, 1
|
|
LOWPASS 0, 1, 2, 4
|
|
|
|
.loop:
|
|
mova [dstq+strideq*0 ], m3
|
|
movhps [dstq+strideq*8+0], m0
|
|
movq [dstq+strideq*8+8], m3
|
|
palignr m3, m0, 15
|
|
pslldq m0, 1
|
|
mova [dstq+strideq*1 ], m3
|
|
movhps [dstq+stride9q +0], m0
|
|
movq [dstq+stride9q +8], m3
|
|
palignr m3, m0, 15
|
|
pslldq m0, 1
|
|
lea dstq, [dstq+strideq*2]
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_dr_32x32, 4, 4, 8, dst, stride, l, a
|
|
mova m1, [lq]
|
|
mova m2, [lq+16]
|
|
movu m3, [aq-1]
|
|
movu m4, [aq+15]
|
|
movd m5, [aq+31]
|
|
DEFINE_ARGS dst, stride, stride8, cnt
|
|
lea stride8q, [strideq*8]
|
|
palignr m5, m4, 1
|
|
palignr m6, m4, m3, 15
|
|
LOWPASS 5, 4, 6, 7
|
|
palignr m4, m3, 1
|
|
palignr m6, m3, m2, 15
|
|
LOWPASS 4, 3, 6, 7
|
|
palignr m3, m2, 1
|
|
palignr m6, m2, m1, 15
|
|
LOWPASS 3, 2, 6, 7
|
|
palignr m2, m1, 1
|
|
pslldq m0, m1, 1
|
|
LOWPASS 2, 1, 0, 6
|
|
mov cntd, 16
|
|
|
|
; out=m2/m3/m4/m5
|
|
.loop:
|
|
mova [dstq+stride8q*0+ 0], m4
|
|
mova [dstq+stride8q*0+16], m5
|
|
mova [dstq+stride8q*2+ 0], m3
|
|
mova [dstq+stride8q*2+16], m4
|
|
palignr m5, m4, 15
|
|
palignr m4, m3, 15
|
|
palignr m3, m2, 15
|
|
pslldq m2, 1
|
|
add dstq, strideq
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
%endmacro
|
|
|
|
DR_XMM_FUNCS ssse3
|
|
DR_XMM_FUNCS avx
|
|
|
|
; vl
|
|
|
|
INIT_MMX ssse3
|
|
cglobal vp9_ipred_vl_4x4, 4, 4, 0, dst, stride, l, a
|
|
movq m0, [aq]
|
|
psrlq m1, m0, 8
|
|
psrlq m2, m1, 8
|
|
LOWPASS 2, 1, 0, 3
|
|
pavgb m1, m0
|
|
movd [dstq+strideq*0], m1
|
|
movd [dstq+strideq*1], m2
|
|
lea dstq, [dstq+strideq*2]
|
|
psrlq m1, 8
|
|
psrlq m2, 8
|
|
movd [dstq+strideq*0], m1
|
|
movd [dstq+strideq*1], m2
|
|
RET
|
|
|
|
%macro VL_XMM_FUNCS 1
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_vl_8x8, 4, 4, 4, dst, stride, l, a
|
|
movq m0, [aq]
|
|
pshufb m0, [pb_0to6_9x7]
|
|
DEFINE_ARGS dst, stride, stride3
|
|
lea stride3q, [strideq*3]
|
|
psrldq m1, m0, 1
|
|
psrldq m2, m0, 2
|
|
LOWPASS 2, 1, 0, 3
|
|
pavgb m1, m0
|
|
|
|
movq [dstq+strideq*0], m1
|
|
movq [dstq+strideq*1], m2
|
|
psrldq m1, 1
|
|
psrldq m2, 1
|
|
movq [dstq+strideq*2], m1
|
|
movq [dstq+stride3q ], m2
|
|
lea dstq, [dstq+strideq*4]
|
|
psrldq m1, 1
|
|
psrldq m2, 1
|
|
movq [dstq+strideq*0], m1
|
|
movq [dstq+strideq*1], m2
|
|
psrldq m1, 1
|
|
psrldq m2, 1
|
|
movq [dstq+strideq*2], m1
|
|
movq [dstq+stride3q ], m2
|
|
RET
|
|
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_vl_16x16, 4, 4, 5, dst, stride, l, a
|
|
mova m0, [aq]
|
|
mova m4, [pb_1toE_2xF]
|
|
DEFINE_ARGS dst, stride, stride3, cnt
|
|
lea stride3q, [strideq*3]
|
|
pshufb m1, m0, m4
|
|
pshufb m2, m1, m4
|
|
LOWPASS 2, 1, 0, 3
|
|
pavgb m1, m0
|
|
mov cntd, 4
|
|
.loop:
|
|
mova [dstq+strideq*0], m1
|
|
mova [dstq+strideq*1], m2
|
|
pshufb m1, m4
|
|
pshufb m2, m4
|
|
mova [dstq+strideq*2], m1
|
|
mova [dstq+stride3q ], m2
|
|
pshufb m1, m4
|
|
pshufb m2, m4
|
|
lea dstq, [dstq+strideq*4]
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_vl_32x32, 4, 4, 7, dst, stride, l, a
|
|
mova m0, [aq]
|
|
mova m5, [aq+16]
|
|
mova m4, [pb_1toE_2xF]
|
|
DEFINE_ARGS dst, stride, dst16, cnt
|
|
palignr m2, m5, m0, 1
|
|
palignr m3, m5, m0, 2
|
|
lea dst16q, [dstq +strideq*8]
|
|
LOWPASS 3, 2, 0, 6
|
|
pavgb m2, m0
|
|
pshufb m0, m5, m4
|
|
pshufb m1, m0, m4
|
|
lea dst16q, [dst16q+strideq*8]
|
|
LOWPASS 1, 0, 5, 6
|
|
pavgb m0, m5
|
|
pshufb m5, [pb_15]
|
|
mov cntd, 8
|
|
|
|
.loop:
|
|
%macro %%write 3
|
|
mova [dstq+stride%1+ 0], %2
|
|
mova [dstq+stride%1+16], %3
|
|
movhps [dst16q+stride%1 ], %2
|
|
movu [dst16q+stride%1+ 8], %3
|
|
movq [dst16q+stride%1+24], m5
|
|
%if cpuflag(avx)
|
|
palignr %2, %3, %2, 1
|
|
pshufb %3, m4
|
|
%else
|
|
palignr m6, %3, %2, 1
|
|
pshufb %3, m4
|
|
mova %2, m6
|
|
%endif
|
|
%endmacro
|
|
|
|
%%write q*0, m2, m0
|
|
%%write q*1, m3, m1
|
|
lea dstq, [dstq +strideq*2]
|
|
lea dst16q, [dst16q+strideq*2]
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
%endmacro
|
|
|
|
VL_XMM_FUNCS ssse3
|
|
VL_XMM_FUNCS avx
|
|
|
|
; vr
|
|
|
|
INIT_MMX ssse3
|
|
cglobal vp9_ipred_vr_4x4, 4, 4, 0, dst, stride, l, a
|
|
movq m1, [aq-1]
|
|
punpckldq m2, [lq]
|
|
movd m0, [aq]
|
|
DEFINE_ARGS dst, stride, stride3
|
|
lea stride3q, [strideq*3]
|
|
pavgb m0, m1
|
|
palignr m1, m2, 5
|
|
psrlq m2, m1, 8
|
|
psllq m3, m1, 8
|
|
LOWPASS 2, 1, 3, 4
|
|
|
|
; ABCD <- for the following predictor:
|
|
; EFGH
|
|
; IABC | m0 contains ABCDxxxx
|
|
; JEFG | m2 contains xJIEFGHx
|
|
|
|
punpckldq m0, m2
|
|
pshufb m2, [pb_13456_3xm1]
|
|
movd [dstq+strideq*0], m0
|
|
pshufb m0, [pb_6012_4xm1]
|
|
movd [dstq+stride3q ], m2
|
|
psrlq m2, 8
|
|
movd [dstq+strideq*2], m0
|
|
movd [dstq+strideq*1], m2
|
|
RET
|
|
|
|
%macro VR_XMM_FUNCS 1
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_vr_8x8, 4, 4, 5, dst, stride, l, a
|
|
movu m1, [aq-1]
|
|
movhps m2, [lq]
|
|
movq m0, [aq]
|
|
DEFINE_ARGS dst, stride, stride3
|
|
lea stride3q, [strideq*3]
|
|
pavgb m0, m1
|
|
palignr m1, m2, 9
|
|
pslldq m2, m1, 1
|
|
pslldq m3, m1, 2
|
|
LOWPASS 1, 2, 3, 4
|
|
|
|
; ABCDEFGH <- for the following predictor:
|
|
; IJKLMNOP
|
|
; QABCDEFG | m0 contains ABCDEFGHxxxxxxxx
|
|
; RIJKLMNO | m1 contains xxVUTSRQIJKLMNOP
|
|
; SQABCDEF
|
|
; TRIJKLMN
|
|
; USQABCDE
|
|
; VTRIJKLM
|
|
|
|
punpcklqdq m0, m1 ; ABCDEFGHxxVUTSRQ
|
|
movq [dstq+strideq*0], m0
|
|
pshufb m0, [pb_6xm1_BDF_0to6] ; xxxxxxUSQABCDEFG
|
|
movhps [dstq+strideq*1], m1
|
|
pshufb m1, [pb_6xm1_246_8toE] ; xxxxxxVTRIJKLMNO
|
|
movhps [dstq+strideq*2], m0
|
|
pslldq m0, 1
|
|
movhps [dstq+stride3q ], m1
|
|
lea dstq, [dstq+strideq*4]
|
|
pslldq m1, 1
|
|
movhps [dstq+strideq*0], m0
|
|
pslldq m0, 1
|
|
movhps [dstq+strideq*1], m1
|
|
pslldq m1, 1
|
|
movhps [dstq+strideq*2], m0
|
|
movhps [dstq+stride3q ], m1
|
|
RET
|
|
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_vr_16x16, 4, 4, 6, dst, stride, l, a
|
|
mova m0, [aq]
|
|
movu m1, [aq-1]
|
|
mova m2, [lq]
|
|
DEFINE_ARGS dst, stride, stride3, cnt
|
|
lea stride3q, [strideq*3]
|
|
palignr m3, m1, m2, 15
|
|
LOWPASS 3, 1, 0, 4
|
|
pavgb m0, m1
|
|
palignr m1, m2, 1
|
|
pslldq m4, m2, 1
|
|
LOWPASS 1, 2, 4, 5
|
|
pshufb m1, [pb_02468ACE_13579BDF]
|
|
mov cntd, 4
|
|
|
|
.loop:
|
|
movlhps m2, m1
|
|
mova [dstq+strideq*0], m0
|
|
mova [dstq+strideq*1], m3
|
|
palignr m4, m0, m1, 15
|
|
palignr m5, m3, m2, 15
|
|
mova [dstq+strideq*2], m4
|
|
mova [dstq+stride3q ], m5
|
|
lea dstq, [dstq+strideq*4]
|
|
palignr m0, m1, 14
|
|
palignr m3, m2, 14
|
|
pslldq m1, 2
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
|
|
%if ARCH_X86_64
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_vr_32x32, 4, 4, 9, dst, stride, l, a
|
|
mova m0, [aq]
|
|
mova m2, [aq+16]
|
|
movu m1, [aq-1]
|
|
palignr m3, m2, m0, 15
|
|
palignr m4, m2, m0, 14
|
|
LOWPASS 4, 3, 2, 5
|
|
pavgb m3, m2
|
|
mova m2, [lq+16]
|
|
palignr m5, m1, m2, 15
|
|
LOWPASS 5, 1, 0, 6
|
|
pavgb m0, m1
|
|
mova m6, [lq]
|
|
palignr m1, m2, 1
|
|
palignr m7, m2, m6, 15
|
|
LOWPASS 1, 2, 7, 8
|
|
palignr m2, m6, 1
|
|
pslldq m7, m6, 1
|
|
LOWPASS 2, 6, 7, 8
|
|
pshufb m1, [pb_02468ACE_13579BDF]
|
|
pshufb m2, [pb_02468ACE_13579BDF]
|
|
DEFINE_ARGS dst, stride, dst16, cnt
|
|
lea dst16q, [dstq +strideq*8]
|
|
lea dst16q, [dst16q+strideq*8]
|
|
SBUTTERFLY qdq, 2, 1, 6
|
|
mov cntd, 8
|
|
|
|
.loop:
|
|
; even lines (0, 2, 4, ...): m1 | m0, m3
|
|
; odd lines (1, 3, 5, ...): m2 | m5, m4
|
|
%macro %%write 4
|
|
mova [dstq+stride%1+ 0], %3
|
|
mova [dstq+stride%1+16], %4
|
|
movhps [dst16q+stride%1 ], %2
|
|
movu [dst16q+stride%1+ 8], %3
|
|
movq [dst16q+stride%1+24], %4
|
|
palignr %4, %3, 15
|
|
palignr %3, %2, 15
|
|
pslldq %2, 1
|
|
%endmacro
|
|
|
|
%%write q*0, m1, m0, m3
|
|
%%write q*1, m2, m5, m4
|
|
lea dstq, [dstq +strideq*2]
|
|
lea dst16q, [dst16q+strideq*2]
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
%endif
|
|
%endmacro
|
|
|
|
VR_XMM_FUNCS ssse3
|
|
VR_XMM_FUNCS avx
|
|
|
|
; hd
|
|
|
|
INIT_MMX ssse3
|
|
cglobal vp9_ipred_hd_4x4, 4, 4, 0, dst, stride, l, a
|
|
movd m0, [lq]
|
|
punpckldq m0, [aq-1]
|
|
DEFINE_ARGS dst, stride, stride3
|
|
lea stride3q, [strideq*3]
|
|
psrlq m1, m0, 8
|
|
psrlq m2, m1, 8
|
|
LOWPASS 2, 1, 0, 3
|
|
pavgb m1, m0
|
|
|
|
; DHIJ <- for the following predictor:
|
|
; CGDH
|
|
; BFCG | m1 contains ABCDxxxx
|
|
; AEBF | m2 contains EFGHIJxx
|
|
|
|
punpcklbw m1, m2
|
|
punpckhdq m0, m1, m2
|
|
|
|
; m1 contains AEBFCGDH
|
|
; m0 contains CGDHIJxx
|
|
|
|
movd [dstq+stride3q ], m1
|
|
movd [dstq+strideq*1], m0
|
|
psrlq m1, 16
|
|
psrlq m0, 16
|
|
movd [dstq+strideq*2], m1
|
|
movd [dstq+strideq*0], m0
|
|
RET
|
|
|
|
%macro HD_XMM_FUNCS 1
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_hd_8x8, 4, 4, 4, dst, stride, l, a
|
|
movq m0, [lq]
|
|
movhps m0, [aq-1]
|
|
DEFINE_ARGS dst, stride, stride3, dst4
|
|
lea stride3q, [strideq*3]
|
|
lea dst4q, [dstq+strideq*4]
|
|
psrldq m1, m0, 1
|
|
psrldq m2, m1, 1
|
|
LOWPASS 2, 1, 0, 3
|
|
pavgb m1, m0
|
|
|
|
; HPQRSTUV <- for the following predictor
|
|
; GOHPQRST
|
|
; FNGOHPQR | m1 contains ABCDEFGHxxxxxxxx
|
|
; EMFNGOHP | m2 contains IJKLMNOPQRSTUVxx
|
|
; DLEMFNGO
|
|
; CKDLEMFN
|
|
; BJCKDLEM
|
|
; AIBJCKDL
|
|
|
|
punpcklbw m1, m2
|
|
movhlps m2, m2
|
|
|
|
; m1 contains AIBJCKDLEMFNGOHP
|
|
; m2 contains QRSTUVxxxxxxxxxx
|
|
|
|
movhps [dstq +stride3q ], m1
|
|
movq [dst4q+stride3q ], m1
|
|
palignr m3, m2, m1, 2
|
|
movhps [dstq +strideq*2], m3
|
|
movq [dst4q+strideq*2], m3
|
|
palignr m3, m2, m1, 4
|
|
movhps [dstq +strideq*1], m3
|
|
movq [dst4q+strideq*1], m3
|
|
palignr m2, m1, 6
|
|
movhps [dstq +strideq*0], m2
|
|
movq [dst4q+strideq*0], m2
|
|
RET
|
|
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_hd_16x16, 4, 6, 7, dst, stride, l, a
|
|
mova m0, [lq]
|
|
movu m3, [aq-1]
|
|
DEFINE_ARGS dst, stride, stride4, dst4, dst8, dst12
|
|
lea stride4q, [strideq*4]
|
|
lea dst4q, [dstq +stride4q]
|
|
lea dst8q, [dst4q+stride4q]
|
|
lea dst12q, [dst8q+stride4q]
|
|
psrldq m4, m3, 1
|
|
psrldq m5, m3, 2
|
|
LOWPASS 5, 4, 3, 6
|
|
palignr m1, m3, m0, 1
|
|
palignr m2, m3, m0, 2
|
|
LOWPASS 2, 1, 0, 6
|
|
pavgb m1, m0
|
|
SBUTTERFLY bw, 1, 2, 6
|
|
|
|
; I PROBABLY INVERTED L0 ad L16 here
|
|
; m1, m2, m5
|
|
.loop:
|
|
sub stride4q, strideq
|
|
movhps [dstq +stride4q +0], m2
|
|
movq [dstq +stride4q +8], m5
|
|
mova [dst4q+stride4q ], m2
|
|
movhps [dst8q+stride4q +0], m1
|
|
movq [dst8q+stride4q +8], m2
|
|
mova [dst12q+stride4q ], m1
|
|
%if cpuflag(avx)
|
|
palignr m1, m2, m1, 2
|
|
palignr m2, m5, m2, 2
|
|
%else
|
|
palignr m3, m2, m1, 2
|
|
palignr m0, m5, m2, 2
|
|
mova m1, m3
|
|
mova m2, m0
|
|
%endif
|
|
psrldq m5, 2
|
|
jg .loop
|
|
RET
|
|
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_hd_32x32, 4, 6, 8, dst, stride, l, a
|
|
mova m0, [lq]
|
|
mova m1, [lq+16]
|
|
movu m2, [aq-1]
|
|
movu m3, [aq+15]
|
|
DEFINE_ARGS dst, stride, stride8, dst8, dst16, dst24
|
|
lea stride8q, [strideq*8]
|
|
lea dst8q, [dstq +stride8q]
|
|
lea dst16q, [dst8q +stride8q]
|
|
lea dst24q, [dst16q+stride8q]
|
|
psrldq m4, m3, 1
|
|
psrldq m5, m3, 2
|
|
LOWPASS 5, 4, 3, 6
|
|
palignr m4, m3, m2, 2
|
|
palignr m3, m2, 1
|
|
LOWPASS 4, 3, 2, 6
|
|
palignr m3, m2, m1, 2
|
|
palignr m2, m1, 1
|
|
LOWPASS 3, 2, 1, 6
|
|
pavgb m2, m1
|
|
palignr m6, m1, m0, 1
|
|
palignr m1, m0, 2
|
|
LOWPASS 1, 6, 0, 7
|
|
pavgb m0, m1
|
|
SBUTTERFLY bw, 2, 3, 6
|
|
SBUTTERFLY bw, 0, 1, 6
|
|
|
|
; m0, m1, m2, m3, m4, m5
|
|
.loop:
|
|
sub stride8q, strideq
|
|
mova [dstq +stride8q+ 0], m3
|
|
mova [dstq +stride8q+16], m4
|
|
mova [dst8q +stride8q+ 0], m2
|
|
mova [dst8q +stride8q+16], m3
|
|
mova [dst16q+stride8q+ 0], m1
|
|
mova [dst16q+stride8q+16], m2
|
|
mova [dst24q+stride8q+ 0], m0
|
|
mova [dst24q+stride8q+16], m1
|
|
%if cpuflag(avx)
|
|
palignr m0, m1, m0, 2
|
|
palignr m1, m2, m1, 2
|
|
palignr m2, m3, m2, 2
|
|
palignr m3, m4, m3, 2
|
|
palignr m4, m5, m4, 2
|
|
psrldq m5, 2
|
|
%else
|
|
psrldq m6, m5, 2
|
|
palignr m5, m4, 2
|
|
palignr m4, m3, 2
|
|
palignr m3, m2, 2
|
|
palignr m2, m1, 2
|
|
palignr m1, m0, 2
|
|
mova m0, m1
|
|
mova m1, m2
|
|
mova m2, m3
|
|
mova m3, m4
|
|
mova m4, m5
|
|
mova m5, m6
|
|
%endif
|
|
jg .loop
|
|
RET
|
|
%endmacro
|
|
|
|
HD_XMM_FUNCS ssse3
|
|
HD_XMM_FUNCS avx
|
|
|
|
INIT_MMX ssse3
|
|
cglobal vp9_ipred_hu_4x4, 3, 3, 0, dst, stride, l
|
|
movd m0, [lq]
|
|
pshufb m0, [pb_3to1_5x0]
|
|
psrlq m1, m0, 8
|
|
psrlq m2, m1, 8
|
|
LOWPASS 2, 1, 0, 3
|
|
pavgb m1, m0
|
|
DEFINE_ARGS dst, stride, stride3
|
|
lea stride3q, [strideq*3]
|
|
SBUTTERFLY bw, 1, 2, 0
|
|
palignr m2, m1, 2
|
|
movd [dstq+strideq*0], m1
|
|
movd [dstq+strideq*1], m2
|
|
punpckhdq m1, m1
|
|
punpckhdq m2, m2
|
|
movd [dstq+strideq*2], m1
|
|
movd [dstq+stride3q ], m2
|
|
RET
|
|
|
|
%macro HU_XMM_FUNCS 1
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_hu_8x8, 3, 4, 4, dst, stride, l
|
|
movq m0, [lq]
|
|
pshufb m0, [pb_7to1_9x0]
|
|
psrldq m1, m0, 1
|
|
psrldq m2, m1, 1
|
|
LOWPASS 2, 1, 0, 3
|
|
pavgb m1, m0
|
|
DEFINE_ARGS dst, stride, stride3, dst4
|
|
lea stride3q, [strideq*3]
|
|
lea dst4q, [dstq+strideq*4]
|
|
SBUTTERFLY bw, 1, 2, 0
|
|
movq [dstq +strideq*0], m1
|
|
movhps [dst4q+strideq*0], m1
|
|
palignr m0, m2, m1, 2
|
|
movq [dstq +strideq*1], m0
|
|
movhps [dst4q+strideq*1], m0
|
|
palignr m0, m2, m1, 4
|
|
movq [dstq +strideq*2], m0
|
|
movhps [dst4q+strideq*2], m0
|
|
palignr m2, m1, 6
|
|
movq [dstq +stride3q ], m2
|
|
movhps [dst4q+stride3q ], m2
|
|
RET
|
|
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_hu_16x16, 3, 4, 5, dst, stride, l
|
|
mova m0, [lq]
|
|
pshufb m0, [pb_Fto0]
|
|
mova m3, [pb_2toE_3xF]
|
|
pshufb m1, m0, [pb_1toE_2xF]
|
|
pshufb m2, m0, m3
|
|
LOWPASS 2, 1, 0, 4
|
|
pavgb m1, m0
|
|
DEFINE_ARGS dst, stride, stride9, cnt
|
|
lea stride9q, [strideq *3]
|
|
mov cntd, 4
|
|
lea stride9q, [stride9q*3]
|
|
SBUTTERFLY bw, 1, 2, 0
|
|
|
|
.loop:
|
|
mova [dstq+strideq*0], m1
|
|
mova [dstq+strideq*8], m2
|
|
palignr m0, m2, m1, 2
|
|
pshufb m2, m3
|
|
mova [dstq+strideq*1], m0
|
|
mova [dstq+stride9q ], m2
|
|
palignr m1, m2, m0, 2
|
|
pshufb m2, m3
|
|
lea dstq, [dstq+strideq*2]
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
|
|
INIT_XMM %1
|
|
cglobal vp9_ipred_hu_32x32, 3, 7, 7, dst, stride, l
|
|
mova m0, [lq]
|
|
mova m1, [lq+16]
|
|
mova m2, [pb_Fto0]
|
|
mova m4, [pb_2toE_3xF]
|
|
pshufb m0, m2
|
|
pshufb m1, m2
|
|
palignr m2, m0, m1, 1
|
|
palignr m3, m0, m1, 2
|
|
LOWPASS 3, 2, 1, 5
|
|
pavgb m2, m1
|
|
pshufb m1, m0, m4
|
|
pshufb m5, m0, [pb_1toE_2xF]
|
|
LOWPASS 1, 5, 0, 6
|
|
pavgb m0, m5
|
|
DEFINE_ARGS dst, stride, cnt, stride0, dst8, dst16, dst24
|
|
mov cntd, 8
|
|
xor stride0q, stride0q
|
|
lea dst8q, [dstq +strideq*8]
|
|
lea dst16q, [dst8q +strideq*8]
|
|
lea dst24q, [dst16q+strideq*8]
|
|
SBUTTERFLY bw, 0, 1, 5
|
|
SBUTTERFLY bw, 2, 3, 5
|
|
pshufb m6, m1, [pb_15]
|
|
|
|
.loop:
|
|
mova [dstq +stride0q+ 0], m2
|
|
mova [dstq +stride0q+16], m3
|
|
mova [dst8q +stride0q+ 0], m3
|
|
mova [dst8q +stride0q+16], m0
|
|
mova [dst16q+stride0q+ 0], m0
|
|
mova [dst16q+stride0q+16], m1
|
|
mova [dst24q+stride0q+ 0], m1
|
|
mova [dst24q+stride0q+16], m6
|
|
%if cpuflag(avx)
|
|
palignr m2, m3, m2, 2
|
|
palignr m3, m0, m3, 2
|
|
palignr m0, m1, m0, 2
|
|
pshufb m1, m4
|
|
%else
|
|
pshufb m5, m1, m4
|
|
palignr m1, m0, 2
|
|
palignr m0, m3, 2
|
|
palignr m3, m2, 2
|
|
mova m2, m3
|
|
mova m3, m0
|
|
mova m0, m1
|
|
mova m1, m5
|
|
%endif
|
|
add stride0q, strideq
|
|
dec cntd
|
|
jg .loop
|
|
RET
|
|
%endmacro
|
|
|
|
HU_XMM_FUNCS ssse3
|
|
HU_XMM_FUNCS avx
|
|
|
|
; FIXME 127, 128, 129 ?
|