;****************************************************************************** ;* SIMD lossless video DSP utils ;* Copyright (c) 2008 Loren Merritt ;* Copyright (c) 2014 Michael Niedermayer ;* ;* 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 pb_ef: times 8 db 14,15 pb_67: times 8 db 6, 7 pb_zzzz2323zzzzabab: db -1,-1,-1,-1, 2, 3, 2, 3,-1,-1,-1,-1,10,11,10,11 pb_zzzzzzzz67676767: db -1,-1,-1,-1,-1,-1,-1,-1, 6, 7, 6, 7, 6, 7, 6, 7 SECTION_TEXT %macro INT16_LOOP 2 ; %1 = a/u (aligned/unaligned), %2 = add/sub movd m4, maskd SPLATW m4, m4 add wq, wq test wq, 2*mmsize - 1 jz %%.tomainloop %%.wordloop: sub wq, 2 %ifidn %2, add mov ax, [srcq+wq] add ax, [dstq+wq] %else mov ax, [src1q+wq] sub ax, [src2q+wq] %endif and ax, maskw mov [dstq+wq], ax test wq, 2*mmsize - 1 jnz %%.wordloop %%.tomainloop: %ifidn %2, add add srcq, wq %else add src1q, wq add src2q, wq %endif add dstq, wq neg wq jz %%.end %%.loop: %ifidn %2, add mov%1 m0, [srcq+wq] mov%1 m1, [dstq+wq] mov%1 m2, [srcq+wq+mmsize] mov%1 m3, [dstq+wq+mmsize] %else mov%1 m0, [src1q+wq] mov%1 m1, [src2q+wq] mov%1 m2, [src1q+wq+mmsize] mov%1 m3, [src2q+wq+mmsize] %endif p%2w m0, m1 p%2w m2, m3 pand m0, m4 pand m2, m4 mov%1 [dstq+wq] , m0 mov%1 [dstq+wq+mmsize], m2 add wq, 2*mmsize jl %%.loop %%.end: RET %endmacro INIT_MMX mmx cglobal add_int16, 4,4,5, dst, src, mask, w INT16_LOOP a, add INIT_XMM sse2 cglobal add_int16, 4,4,5, dst, src, mask, w test srcq, mmsize-1 jnz .unaligned test dstq, mmsize-1 jnz .unaligned INT16_LOOP a, add .unaligned: INT16_LOOP u, add INIT_MMX mmx cglobal diff_int16, 5,5,5, dst, src1, src2, mask, w INT16_LOOP a, sub INIT_XMM sse2 cglobal diff_int16, 5,5,5, dst, src1, src2, mask, w test src1q, mmsize-1 jnz .unaligned test src2q, mmsize-1 jnz .unaligned test dstq, mmsize-1 jnz .unaligned INT16_LOOP a, sub .unaligned: INT16_LOOP u, sub %macro ADD_HFYU_LEFT_LOOP_INT16 2 ; %1 = dst alignment (a/u), %2 = src alignment (a/u) add wq, wq add srcq, wq add dstq, wq neg wq %%.loop: mov%2 m1, [srcq+wq] mova m2, m1 pslld m1, 16 paddw m1, m2 mova m2, m1 pshufb m1, m3 paddw m1, m2 pshufb m0, m5 %if mmsize == 16 mova m2, m1 pshufb m1, m4 paddw m1, m2 %endif paddw m0, m1 pand m0, m7 %ifidn %1, a mova [dstq+wq], m0 %else movq [dstq+wq], m0 movhps [dstq+wq+8], m0 %endif add wq, mmsize jl %%.loop mov eax, mmsize-1 sub eax, wd mov wd, eax shl wd, 8 lea eax, [wd+eax-1] movd m1, eax pshufb m0, m1 movd eax, m0 RET %endmacro ; int add_hfyu_left_prediction_int16(uint16_t *dst, const uint16_t *src, unsigned mask, int w, int left) INIT_MMX ssse3 cglobal add_hfyu_left_prediction_int16, 4,4,8, dst, src, mask, w, left .skip_prologue: mova m5, [pb_67] mova m3, [pb_zzzz2323zzzzabab] movd m0, leftm psllq m0, 48 movd m7, maskm SPLATW m7 ,m7 ADD_HFYU_LEFT_LOOP_INT16 a, a INIT_XMM sse4 cglobal add_hfyu_left_prediction_int16, 4,4,8, dst, src, mask, w, left mova m5, [pb_ef] mova m4, [pb_zzzzzzzz67676767] mova m3, [pb_zzzz2323zzzzabab] movd m0, leftm pslldq m0, 14 movd m7, maskm SPLATW m7 ,m7 test srcq, 15 jnz .src_unaligned test dstq, 15 jnz .dst_unaligned ADD_HFYU_LEFT_LOOP_INT16 a, a .dst_unaligned: ADD_HFYU_LEFT_LOOP_INT16 u, a .src_unaligned: ADD_HFYU_LEFT_LOOP_INT16 u, u ; void add_hfyu_median_prediction_mmxext(uint8_t *dst, const uint8_t *top, const uint8_t *diff, int mask, int w, int *left, int *left_top) INIT_MMX mmxext cglobal add_hfyu_median_prediction_int16, 7,7,0, dst, top, diff, mask, w, left, left_top add wq, wq movd mm6, maskd SPLATW mm6, mm6 movq mm0, [topq] movq mm2, mm0 movd mm4, [left_topq] psllq mm2, 16 movq mm1, mm0 por mm4, mm2 movd mm3, [leftq] psubw mm0, mm4 ; t-tl add dstq, wq add topq, wq add diffq, wq neg wq jmp .skip .loop: movq mm4, [topq+wq] movq mm0, mm4 psllq mm4, 16 por mm4, mm1 movq mm1, mm0 ; t psubw mm0, mm4 ; t-tl .skip: movq mm2, [diffq+wq] %assign i 0 %rep 4 movq mm4, mm0 paddw mm4, mm3 ; t-tl+l pand mm4, mm6 movq mm5, mm3 pmaxsw mm3, mm1 pminsw mm5, mm1 pminsw mm3, mm4 pmaxsw mm3, mm5 ; median paddw mm3, mm2 ; +residual pand mm3, mm6 %if i==0 movq mm7, mm3 psllq mm7, 48 %else movq mm4, mm3 psrlq mm7, 16 psllq mm4, 48 por mm7, mm4 %endif %if i<3 psrlq mm0, 16 psrlq mm1, 16 psrlq mm2, 16 %endif %assign i i+1 %endrep movq [dstq+wq], mm7 add wq, 8 jl .loop movzx r2d, word [dstq-2] mov [leftq], r2d movzx r2d, word [topq-2] mov [left_topq], r2d RET cglobal sub_hfyu_median_prediction_int16, 7,7,0, dst, src1, src2, mask, w, left, left_top add wq, wq movd mm7, maskd SPLATW mm7, mm7 movq mm0, [src1q] movq mm2, [src2q] psllq mm0, 16 psllq mm2, 16 movd mm6, [left_topq] por mm0, mm6 movd mm6, [leftq] por mm2, mm6 xor maskq, maskq .loop: movq mm1, [src1q + maskq] movq mm3, [src2q + maskq] movq mm4, mm2 psubw mm2, mm0 paddw mm2, mm1 pand mm2, mm7 movq mm5, mm4 pmaxsw mm4, mm1 pminsw mm1, mm5 pminsw mm4, mm2 pmaxsw mm4, mm1 psubw mm3, mm4 pand mm3, mm7 movq [dstq + maskq], mm3 add maskq, 8 movq mm0, [src1q + maskq - 2] movq mm2, [src2q + maskq - 2] cmp maskq, wq jb .loop mov maskd, [src1q + wq - 2] mov [left_topq], maskd mov maskd, [src2q + wq - 2] mov [leftq], maskd RET