Subversion Repositories Kolibri OS

;******************************************************************************

;* MMX/SSE2-optimized functions for the RV30 and RV40 decoders

;* Copyright (C) 2012 Christophe Gisquet 

;*

;* 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