|
- ;******************************************************************************
- ;* MMX/SSE2-optimized functions for the RV30 and RV40 decoders
- ;* Copyright (C) 2012 Christophe Gisquet <christophe.gisquet@gmail.com>
- ;*
- ;* 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_row_coeffs: times 4 dw 13
- times 4 dw 17
- times 4 dw 7
- pd_512: times 2 dd 0x200
- pw_col_coeffs: dw 13, 13, 13, -13
- dw 17, 7, 7, -17
- dw 13, -13, 13, 13
- dw -7, 17, -17, -7
-
- SECTION .text
-
- %macro IDCT_DC_NOROUND 1
- imul %1, 13*13*3
- sar %1, 11
- %endmacro
-
- %macro IDCT_DC_ROUND 1
- imul %1, 13*13
- add %1, 0x200
- sar %1, 10
- %endmacro
-
- %macro rv34_idct 1
- cglobal rv34_idct_%1, 1, 2, 0
- movsx r1, word [r0]
- IDCT_DC r1
- movd m0, r1d
- pshufw m0, m0, 0
- movq [r0+ 0], m0
- movq [r0+ 8], m0
- movq [r0+16], m0
- movq [r0+24], m0
- REP_RET
- %endmacro
-
- INIT_MMX mmxext
- %define IDCT_DC IDCT_DC_ROUND
- rv34_idct dc
- %define IDCT_DC IDCT_DC_NOROUND
- rv34_idct dc_noround
-
- ; ff_rv34_idct_dc_add_mmx(uint8_t *dst, int stride, int dc);
- INIT_MMX mmx
- cglobal rv34_idct_dc_add, 3, 3
- ; calculate DC
- IDCT_DC_ROUND r2
- pxor m1, m1
- movd m0, r2d
- psubw m1, m0
- packuswb m0, m0
- packuswb m1, m1
- punpcklbw m0, m0
- punpcklbw m1, m1
- punpcklwd m0, m0
- punpcklwd m1, m1
-
- ; add DC
- lea r2, [r0+r1*2]
- movh m2, [r0]
- movh m3, [r0+r1]
- movh m4, [r2]
- movh m5, [r2+r1]
- paddusb m2, m0
- paddusb m3, m0
- paddusb m4, m0
- paddusb m5, m0
- psubusb m2, m1
- psubusb m3, m1
- psubusb m4, m1
- psubusb m5, m1
- movh [r0], m2
- movh [r0+r1], m3
- movh [r2], m4
- movh [r2+r1], m5
- RET
-
- ; Load coeffs and perform row transform
- ; Output: coeffs in mm[0467], rounder in mm5
- %macro ROW_TRANSFORM 1
- pxor mm7, mm7
- mova mm0, [%1+ 0*8]
- mova mm1, [%1+ 1*8]
- mova mm2, [%1+ 2*8]
- mova mm3, [%1+ 3*8]
- mova [%1+ 0*8], mm7
- mova [%1+ 1*8], mm7
- mova [%1+ 2*8], mm7
- mova [%1+ 3*8], mm7
- mova mm4, mm0
- mova mm6, [pw_row_coeffs+ 0]
- paddsw mm0, mm2 ; b0 + b2
- psubsw mm4, mm2 ; b0 - b2
- pmullw mm0, mm6 ; *13 = z0
- pmullw mm4, mm6 ; *13 = z1
- mova mm5, mm1
- pmullw mm1, [pw_row_coeffs+ 8] ; b1*17
- pmullw mm5, [pw_row_coeffs+16] ; b1* 7
- mova mm7, mm3
- pmullw mm3, [pw_row_coeffs+ 8] ; b3*17
- pmullw mm7, [pw_row_coeffs+16] ; b3* 7
- paddsw mm1, mm7 ; z3 = b1*17 + b3* 7
- psubsw mm5, mm3 ; z2 = b1* 7 - b3*17
- mova mm7, mm0
- mova mm6, mm4
- paddsw mm0, mm1 ; z0 + z3
- psubsw mm7, mm1 ; z0 - z3
- paddsw mm4, mm5 ; z1 + z2
- psubsw mm6, mm5 ; z1 - z2
- mova mm5, [pd_512] ; 0x200
- %endmacro
-
- ; ff_rv34_idct_add_mmxext(uint8_t *dst, ptrdiff_t stride, int16_t *block);
- %macro COL_TRANSFORM 4
- pshufw mm3, %2, 0xDD ; col. 1,3,1,3
- pshufw %2, %2, 0x88 ; col. 0,2,0,2
- pmaddwd %2, %3 ; 13*c0+13*c2 | 13*c0-13*c2 = z0 | z1
- pmaddwd mm3, %4 ; 17*c1+ 7*c3 | 7*c1-17*c3 = z3 | z2
- paddd %2, mm5
- pshufw mm1, %2, 01001110b ; z1 | z0
- pshufw mm2, mm3, 01001110b ; z2 | z3
- paddd %2, mm3 ; z0+z3 | z1+z2
- psubd mm1, mm2 ; z1-z2 | z0-z3
- movd mm3, %1
- psrad %2, 10
- pxor mm2, mm2
- psrad mm1, 10
- punpcklbw mm3, mm2
- packssdw %2, mm1
- paddw %2, mm3
- packuswb %2, %2
- movd %1, %2
- %endmacro
- INIT_MMX mmxext
- cglobal rv34_idct_add, 3,3,0, d, s, b
- ROW_TRANSFORM bq
- COL_TRANSFORM [dq], mm0, [pw_col_coeffs+ 0], [pw_col_coeffs+ 8]
- mova mm0, [pw_col_coeffs+ 0]
- COL_TRANSFORM [dq+sq], mm4, mm0, [pw_col_coeffs+ 8]
- mova mm4, [pw_col_coeffs+ 8]
- lea dq, [dq + 2*sq]
- COL_TRANSFORM [dq], mm6, mm0, mm4
- COL_TRANSFORM [dq+sq], mm7, mm0, mm4
- ret
-
- ; ff_rv34_idct_dc_add_sse4(uint8_t *dst, int stride, int dc);
- INIT_XMM sse4
- cglobal rv34_idct_dc_add, 3, 3, 6
- ; load data
- IDCT_DC_ROUND r2
- pxor m1, m1
-
- ; calculate DC
- movd m0, r2d
- lea r2, [r0+r1*2]
- movd m2, [r0]
- movd m3, [r0+r1]
- pshuflw m0, m0, 0
- movd m4, [r2]
- movd m5, [r2+r1]
- punpcklqdq m0, m0
- punpckldq m2, m3
- punpckldq m4, m5
- punpcklbw m2, m1
- punpcklbw m4, m1
- paddw m2, m0
- paddw m4, m0
- packuswb m2, m4
- movd [r0], m2
- pextrd [r0+r1], m2, 1
- pextrd [r2], m2, 2
- pextrd [r2+r1], m2, 3
- RET
|
