mirror of https://github.com/FFmpeg/FFmpeg.git
Timothy Gu
9 years ago
parent
f1e7c42f08
commit
e3461197b1
3 changed files with 320 additions and 295 deletions
@ -0,0 +1,317 @@ |
||||
;****************************************************************************** |
||||
;* VC1 loopfilter optimizations |
||||
;* Copyright (c) 2009 David Conrad |
||||
;* |
||||
;* 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" |
||||
|
||||
cextern pw_4 |
||||
cextern pw_5 |
||||
|
||||
section .text |
||||
|
||||
; dst_low, dst_high (src), zero |
||||
; zero-extends one vector from 8 to 16 bits |
||||
%macro UNPACK_8TO16 4 |
||||
mova m%2, m%3 |
||||
punpckh%1 m%3, m%4 |
||||
punpckl%1 m%2, m%4 |
||||
%endmacro |
||||
|
||||
%macro STORE_4_WORDS 6 |
||||
%if cpuflag(sse4) |
||||
pextrw %1, %5, %6+0 |
||||
pextrw %2, %5, %6+1 |
||||
pextrw %3, %5, %6+2 |
||||
pextrw %4, %5, %6+3 |
||||
%else |
||||
movd %6d, %5 |
||||
%if mmsize==16 |
||||
psrldq %5, 4 |
||||
%else |
||||
psrlq %5, 32 |
||||
%endif |
||||
mov %1, %6w |
||||
shr %6, 16 |
||||
mov %2, %6w |
||||
movd %6d, %5 |
||||
mov %3, %6w |
||||
shr %6, 16 |
||||
mov %4, %6w |
||||
%endif |
||||
%endmacro |
||||
|
||||
; in: p1 p0 q0 q1, clobbers p0 |
||||
; out: p1 = (2*(p1 - q1) - 5*(p0 - q0) + 4) >> 3 |
||||
%macro VC1_LOOP_FILTER_A0 4 |
||||
psubw %1, %4 |
||||
psubw %2, %3 |
||||
paddw %1, %1 |
||||
pmullw %2, [pw_5] |
||||
psubw %1, %2 |
||||
paddw %1, [pw_4] |
||||
psraw %1, 3 |
||||
%endmacro |
||||
|
||||
; in: p0 q0 a0 a1 a2 |
||||
; m0 m1 m7 m6 m5 |
||||
; %1: size |
||||
; out: m0=p0' m1=q0' |
||||
%macro VC1_FILTER 1 |
||||
PABSW m4, m7 |
||||
PABSW m3, m6 |
||||
PABSW m2, m5 |
||||
mova m6, m4 |
||||
pminsw m3, m2 |
||||
pcmpgtw m6, m3 ; if (a2 < a0 || a1 < a0) |
||||
psubw m3, m4 |
||||
pmullw m3, [pw_5] ; 5*(a3 - a0) |
||||
PABSW m2, m3 |
||||
psraw m2, 3 ; abs(d/8) |
||||
pxor m7, m3 ; d_sign ^= a0_sign |
||||
|
||||
pxor m5, m5 |
||||
movd m3, r2d |
||||
%if %1 > 4 |
||||
punpcklbw m3, m3 |
||||
%endif |
||||
punpcklbw m3, m5 |
||||
pcmpgtw m3, m4 ; if (a0 < pq) |
||||
pand m6, m3 |
||||
|
||||
mova m3, m0 |
||||
psubw m3, m1 |
||||
PABSW m4, m3 |
||||
psraw m4, 1 |
||||
pxor m3, m7 ; d_sign ^ clip_sign |
||||
psraw m3, 15 |
||||
pminsw m2, m4 ; min(d, clip) |
||||
pcmpgtw m4, m5 |
||||
pand m6, m4 ; filt3 (C return value) |
||||
|
||||
; each set of 4 pixels is not filtered if the 3rd is not |
||||
%if mmsize==16 |
||||
pshuflw m4, m6, 0xaa |
||||
%if %1 > 4 |
||||
pshufhw m4, m4, 0xaa |
||||
%endif |
||||
%else |
||||
pshufw m4, m6, 0xaa |
||||
%endif |
||||
pandn m3, m4 |
||||
pand m2, m6 |
||||
pand m3, m2 ; d final |
||||
|
||||
psraw m7, 15 |
||||
pxor m3, m7 |
||||
psubw m3, m7 |
||||
psubw m0, m3 |
||||
paddw m1, m3 |
||||
packuswb m0, m0 |
||||
packuswb m1, m1 |
||||
%endmacro |
||||
|
||||
; 1st param: size of filter |
||||
; 2nd param: mov suffix equivalent to the filter size |
||||
%macro VC1_V_LOOP_FILTER 2 |
||||
pxor m5, m5 |
||||
mov%2 m6, [r4] |
||||
mov%2 m4, [r4+r1] |
||||
mov%2 m7, [r4+2*r1] |
||||
mov%2 m0, [r4+r3] |
||||
punpcklbw m6, m5 |
||||
punpcklbw m4, m5 |
||||
punpcklbw m7, m5 |
||||
punpcklbw m0, m5 |
||||
|
||||
VC1_LOOP_FILTER_A0 m6, m4, m7, m0 |
||||
mov%2 m1, [r0] |
||||
mov%2 m2, [r0+r1] |
||||
punpcklbw m1, m5 |
||||
punpcklbw m2, m5 |
||||
mova m4, m0 |
||||
VC1_LOOP_FILTER_A0 m7, m4, m1, m2 |
||||
mov%2 m3, [r0+2*r1] |
||||
mov%2 m4, [r0+r3] |
||||
punpcklbw m3, m5 |
||||
punpcklbw m4, m5 |
||||
mova m5, m1 |
||||
VC1_LOOP_FILTER_A0 m5, m2, m3, m4 |
||||
|
||||
VC1_FILTER %1 |
||||
mov%2 [r4+r3], m0 |
||||
mov%2 [r0], m1 |
||||
%endmacro |
||||
|
||||
; 1st param: size of filter |
||||
; NOTE: UNPACK_8TO16 this number of 8 bit numbers are in half a register |
||||
; 2nd (optional) param: temp register to use for storing words |
||||
%macro VC1_H_LOOP_FILTER 1-2 |
||||
%if %1 == 4 |
||||
movq m0, [r0 -4] |
||||
movq m1, [r0+ r1-4] |
||||
movq m2, [r0+2*r1-4] |
||||
movq m3, [r0+ r3-4] |
||||
TRANSPOSE4x4B 0, 1, 2, 3, 4 |
||||
%else |
||||
movq m0, [r0 -4] |
||||
movq m4, [r0+ r1-4] |
||||
movq m1, [r0+2*r1-4] |
||||
movq m5, [r0+ r3-4] |
||||
movq m2, [r4 -4] |
||||
movq m6, [r4+ r1-4] |
||||
movq m3, [r4+2*r1-4] |
||||
movq m7, [r4+ r3-4] |
||||
punpcklbw m0, m4 |
||||
punpcklbw m1, m5 |
||||
punpcklbw m2, m6 |
||||
punpcklbw m3, m7 |
||||
TRANSPOSE4x4W 0, 1, 2, 3, 4 |
||||
%endif |
||||
pxor m5, m5 |
||||
|
||||
UNPACK_8TO16 bw, 6, 0, 5 |
||||
UNPACK_8TO16 bw, 7, 1, 5 |
||||
VC1_LOOP_FILTER_A0 m6, m0, m7, m1 |
||||
UNPACK_8TO16 bw, 4, 2, 5 |
||||
mova m0, m1 ; m0 = p0 |
||||
VC1_LOOP_FILTER_A0 m7, m1, m4, m2 |
||||
UNPACK_8TO16 bw, 1, 3, 5 |
||||
mova m5, m4 |
||||
VC1_LOOP_FILTER_A0 m5, m2, m1, m3 |
||||
SWAP 1, 4 ; m1 = q0 |
||||
|
||||
VC1_FILTER %1 |
||||
punpcklbw m0, m1 |
||||
%if %0 > 1 |
||||
STORE_4_WORDS [r0-1], [r0+r1-1], [r0+2*r1-1], [r0+r3-1], m0, %2 |
||||
%if %1 > 4 |
||||
psrldq m0, 4 |
||||
STORE_4_WORDS [r4-1], [r4+r1-1], [r4+2*r1-1], [r4+r3-1], m0, %2 |
||||
%endif |
||||
%else |
||||
STORE_4_WORDS [r0-1], [r0+r1-1], [r0+2*r1-1], [r0+r3-1], m0, 0 |
||||
STORE_4_WORDS [r4-1], [r4+r1-1], [r4+2*r1-1], [r4+r3-1], m0, 4 |
||||
%endif |
||||
%endmacro |
||||
|
||||
|
||||
%macro START_V_FILTER 0 |
||||
mov r4, r0 |
||||
lea r3, [4*r1] |
||||
sub r4, r3 |
||||
lea r3, [r1+2*r1] |
||||
imul r2, 0x01010101 |
||||
%endmacro |
||||
|
||||
%macro START_H_FILTER 1 |
||||
lea r3, [r1+2*r1] |
||||
%if %1 > 4 |
||||
lea r4, [r0+4*r1] |
||||
%endif |
||||
imul r2, 0x01010101 |
||||
%endmacro |
||||
|
||||
%macro VC1_LF 0 |
||||
cglobal vc1_v_loop_filter_internal |
||||
VC1_V_LOOP_FILTER 4, d |
||||
ret |
||||
|
||||
cglobal vc1_h_loop_filter_internal |
||||
VC1_H_LOOP_FILTER 4, r4 |
||||
ret |
||||
|
||||
; void ff_vc1_v_loop_filter4_mmxext(uint8_t *src, int stride, int pq) |
||||
cglobal vc1_v_loop_filter4, 3,5,0 |
||||
START_V_FILTER |
||||
call vc1_v_loop_filter_internal |
||||
RET |
||||
|
||||
; void ff_vc1_h_loop_filter4_mmxext(uint8_t *src, int stride, int pq) |
||||
cglobal vc1_h_loop_filter4, 3,5,0 |
||||
START_H_FILTER 4 |
||||
call vc1_h_loop_filter_internal |
||||
RET |
||||
|
||||
; void ff_vc1_v_loop_filter8_mmxext(uint8_t *src, int stride, int pq) |
||||
cglobal vc1_v_loop_filter8, 3,5,0 |
||||
START_V_FILTER |
||||
call vc1_v_loop_filter_internal |
||||
add r4, 4 |
||||
add r0, 4 |
||||
call vc1_v_loop_filter_internal |
||||
RET |
||||
|
||||
; void ff_vc1_h_loop_filter8_mmxext(uint8_t *src, int stride, int pq) |
||||
cglobal vc1_h_loop_filter8, 3,5,0 |
||||
START_H_FILTER 4 |
||||
call vc1_h_loop_filter_internal |
||||
lea r0, [r0+4*r1] |
||||
call vc1_h_loop_filter_internal |
||||
RET |
||||
%endmacro |
||||
|
||||
INIT_MMX mmxext |
||||
VC1_LF |
||||
|
||||
INIT_XMM sse2 |
||||
; void ff_vc1_v_loop_filter8_sse2(uint8_t *src, int stride, int pq) |
||||
cglobal vc1_v_loop_filter8, 3,5,8 |
||||
START_V_FILTER |
||||
VC1_V_LOOP_FILTER 8, q |
||||
RET |
||||
|
||||
; void ff_vc1_h_loop_filter8_sse2(uint8_t *src, int stride, int pq) |
||||
cglobal vc1_h_loop_filter8, 3,6,8 |
||||
START_H_FILTER 8 |
||||
VC1_H_LOOP_FILTER 8, r5 |
||||
RET |
||||
|
||||
INIT_MMX ssse3 |
||||
; void ff_vc1_v_loop_filter4_ssse3(uint8_t *src, int stride, int pq) |
||||
cglobal vc1_v_loop_filter4, 3,5,0 |
||||
START_V_FILTER |
||||
VC1_V_LOOP_FILTER 4, d |
||||
RET |
||||
|
||||
; void ff_vc1_h_loop_filter4_ssse3(uint8_t *src, int stride, int pq) |
||||
cglobal vc1_h_loop_filter4, 3,5,0 |
||||
START_H_FILTER 4 |
||||
VC1_H_LOOP_FILTER 4, r4 |
||||
RET |
||||
|
||||
INIT_XMM ssse3 |
||||
; void ff_vc1_v_loop_filter8_ssse3(uint8_t *src, int stride, int pq) |
||||
cglobal vc1_v_loop_filter8, 3,5,8 |
||||
START_V_FILTER |
||||
VC1_V_LOOP_FILTER 8, q |
||||
RET |
||||
|
||||
; void ff_vc1_h_loop_filter8_ssse3(uint8_t *src, int stride, int pq) |
||||
cglobal vc1_h_loop_filter8, 3,6,8 |
||||
START_H_FILTER 8 |
||||
VC1_H_LOOP_FILTER 8, r5 |
||||
RET |
||||
|
||||
INIT_XMM sse4 |
||||
; void ff_vc1_h_loop_filter8_sse4(uint8_t *src, int stride, int pq) |
||||
cglobal vc1_h_loop_filter8, 3,5,8 |
||||
START_H_FILTER 8 |
||||
VC1_H_LOOP_FILTER 8 |
||||
RET |
||||
Loading…
Reference in new issue