Subversion Repositories Kolibri OS

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

;* MMX/SSE2/AVX-optimized H.264 deblocking code

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

;* Copyright (C) 2005-2011 x264 project

;*

;* Authors: Loren Merritt 

;*          Fiona Glaser 

;*          Oskar Arvidsson 

;*

;* 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_A1: times 16 db 0xA1

pb_3_1: times 4 db 3, 1

SECTION .text

cextern pb_0

cextern pb_1

cextern pb_3

; expands to [base],...,[base+7*stride]

%define PASS8ROWS(base, base3, stride, stride3) \

    [base], [base+stride], [base+stride*2], [base3], \

    [base3+stride], [base3+stride*2], [base3+stride3], [base3+stride*4]

%define PASS8ROWS(base, base3, stride, stride3, offset) \

    PASS8ROWS(base+offset, base3+offset, stride, stride3)

; in: 8 rows of 4 bytes in %4..%11

; out: 4 rows of 8 bytes in m0..m3

%macro TRANSPOSE4x8_LOAD 11

    movh       m0, %4

    movh       m2, %5

    movh       m1, %6

    movh       m3, %7

    punpckl%1  m0, m2

    punpckl%1  m1, m3

    mova       m2, m0

    punpckl%2  m0, m1

    punpckh%2  m2, m1

    movh       m4, %8

    movh       m6, %9

    movh       m5, %10

    movh       m7, %11

    punpckl%1  m4, m6

    punpckl%1  m5, m7

    mova       m6, m4

    punpckl%2  m4, m5

    punpckh%2  m6, m5

    punpckh%3  m1, m0, m4

    punpckh%3  m3, m2, m6

    punpckl%3  m0, m4

    punpckl%3  m2, m6

%endmacro

; in: 4 rows of 8 bytes in m0..m3

; out: 8 rows of 4 bytes in %1..%8

%macro TRANSPOSE8x4B_STORE 8

    punpckhdq  m4, m0, m0

    punpckhdq  m5, m1, m1

    punpckhdq  m6, m2, m2

    punpcklbw  m0, m1

    punpcklbw  m2, m3

    punpcklwd  m1, m0, m2

    punpckhwd  m0, m2

    movh       %1, m1

    punpckhdq  m1, m1

    movh       %2, m1

    movh       %3, m0

    punpckhdq  m0, m0

    movh       %4, m0

    punpckhdq  m3, m3

    punpcklbw  m4, m5

    punpcklbw  m6, m3

    punpcklwd  m5, m4, m6

    punpckhwd  m4, m6

    movh       %5, m5

    punpckhdq  m5, m5

    movh       %6, m5

    movh       %7, m4

    punpckhdq  m4, m4

    movh       %8, m4

%endmacro

%macro TRANSPOSE4x8B_LOAD 8

    TRANSPOSE4x8_LOAD bw, wd, dq, %1, %2, %3, %4, %5, %6, %7, %8

%endmacro

%macro SBUTTERFLY3 4

    punpckh%1  %4, %2, %3

    punpckl%1  %2, %3

%endmacro

; in: 8 rows of 8 (only the middle 6 pels are used) in %1..%8

; out: 6 rows of 8 in [%9+0*16] .. [%9+5*16]

%macro TRANSPOSE6x8_MEM 9

    RESET_MM_PERMUTATION

    movq  m0, %1

    movq  m1, %2

    movq  m2, %3

    movq  m3, %4

    movq  m4, %5

    movq  m5, %6

    movq  m6, %7

    SBUTTERFLY bw, 0, 1, 7

    SBUTTERFLY bw, 2, 3, 7

    SBUTTERFLY bw, 4, 5, 7

    movq  [%9+0x10], m3

    SBUTTERFLY3 bw, m6, %8, m7

    SBUTTERFLY wd, 0, 2, 3

    SBUTTERFLY wd, 4, 6, 3

    punpckhdq m0, m4

    movq  [%9+0x00], m0

    SBUTTERFLY3 wd, m1, [%9+0x10], m3

    SBUTTERFLY wd, 5, 7, 0

    SBUTTERFLY dq, 1, 5, 0

    SBUTTERFLY dq, 2, 6, 0

    punpckldq m3, m7

    movq  [%9+0x10], m2

    movq  [%9+0x20], m6

    movq  [%9+0x30], m1

    movq  [%9+0x40], m5

    movq  [%9+0x50], m3

    RESET_MM_PERMUTATION

%endmacro

; in: 8 rows of 8 in %1..%8

; out: 8 rows of 8 in %9..%16

%macro TRANSPOSE8x8_MEM 16

    RESET_MM_PERMUTATION

    movq  m0, %1

    movq  m1, %2

    movq  m2, %3

    movq  m3, %4

    movq  m4, %5

    movq  m5, %6

    movq  m6, %7

    SBUTTERFLY bw, 0, 1, 7

    SBUTTERFLY bw, 2, 3, 7

    SBUTTERFLY bw, 4, 5, 7

    SBUTTERFLY3 bw, m6, %8, m7

    movq  %9,  m5

    SBUTTERFLY wd, 0, 2, 5

    SBUTTERFLY wd, 4, 6, 5

    SBUTTERFLY wd, 1, 3, 5

    movq  %11, m6

    movq  m6,  %9

    SBUTTERFLY wd, 6, 7, 5

    SBUTTERFLY dq, 0, 4, 5

    SBUTTERFLY dq, 1, 6, 5

    movq  %9,  m0

    movq  %10, m4

    movq  %13, m1

    movq  %14, m6

    SBUTTERFLY3 dq, m2, %11, m0

    SBUTTERFLY dq, 3, 7, 4

    movq  %11, m2

    movq  %12, m0

    movq  %15, m3

    movq  %16, m7

    RESET_MM_PERMUTATION

%endmacro

; out: %4 = |%1-%2|>%3

; clobbers: %5

%macro DIFF_GT 5

%if avx_enabled == 0

    mova    %5, %2

    mova    %4, %1

    psubusb %5, %1

    psubusb %4, %2

%else

    psubusb %5, %2, %1

    psubusb %4, %1, %2

%endif

    por     %4, %5

    psubusb %4, %3

%endmacro

; out: %4 = |%1-%2|>%3

; clobbers: %5

%macro DIFF_GT2 5

%if ARCH_X86_64

    psubusb %5, %2, %1

    psubusb %4, %1, %2

%else

    mova    %5, %2

    mova    %4, %1

    psubusb %5, %1

    psubusb %4, %2

%endif

    psubusb %5, %3

    psubusb %4, %3

    pcmpeqb %4, %5

%endmacro

; in: m0=p1 m1=p0 m2=q0 m3=q1 %1=alpha-1 %2=beta-1

; out: m5=beta-1, m7=mask, %3=alpha-1

; clobbers: m4,m6

%macro LOAD_MASK 2-3

    movd     m4, %1

    movd     m5, %2

    SPLATW   m4, m4

    SPLATW   m5, m5

    packuswb m4, m4  ; 16x alpha-1

    packuswb m5, m5  ; 16x beta-1

%if %0>2

    mova     %3, m4

%endif

    DIFF_GT  m1, m2, m4, m7, m6 ; |p0-q0| > alpha-1

    DIFF_GT  m0, m1, m5, m4, m6 ; |p1-p0| > beta-1

    por      m7, m4

    DIFF_GT  m3, m2, m5, m4, m6 ; |q1-q0| > beta-1

    por      m7, m4

    pxor     m6, m6

    pcmpeqb  m7, m6

%endmacro

; in: m0=p1 m1=p0 m2=q0 m3=q1 m7=(tc&mask)

; out: m1=p0' m2=q0'

; clobbers: m0,3-6

%macro DEBLOCK_P0_Q0 0

    pcmpeqb m4, m4

    pxor    m5, m1, m2   ; p0^q0

    pxor    m3, m4

    pand    m5, [pb_1]   ; (p0^q0)&1

    pavgb   m3, m0       ; (p1 - q1 + 256)>>1

    pxor    m4, m1

    pavgb   m3, [pb_3]   ; (((p1 - q1 + 256)>>1)+4)>>1 = 64+2+(p1-q1)>>2

    pavgb   m4, m2       ; (q0 - p0 + 256)>>1

    pavgb   m3, m5

    mova    m6, [pb_A1]

    paddusb m3, m4       ; d+128+33

    psubusb m6, m3

    psubusb m3, [pb_A1]

    pminub  m6, m7

    pminub  m3, m7

    psubusb m1, m6

    psubusb m2, m3

    paddusb m1, m3

    paddusb m2, m6

%endmacro

; in: m1=p0 m2=q0

;     %1=p1 %2=q2 %3=[q2] %4=[q1] %5=tc0 %6=tmp

; out: [q1] = clip( (q2+((p0+q0+1)>>1))>>1, q1-tc0, q1+tc0 )

; clobbers: q2, tmp, tc0

%macro LUMA_Q1 6

    pavgb   %6, m1, m2

    pavgb   %2, %6       ; avg(p2,avg(p0,q0))

    pxor    %6, %3

    pand    %6, [pb_1]   ; (p2^avg(p0,q0))&1

    psubusb %2, %6       ; (p2+((p0+q0+1)>>1))>>1

    psubusb %6, %1, %5

    paddusb %5, %1

    pmaxub  %2, %6

    pminub  %2, %5

    mova    %4, %2

%endmacro

%if ARCH_X86_64

;-----------------------------------------------------------------------------

; void ff_deblock_v_luma(uint8_t *pix, int stride, int alpha, int beta,

;                        int8_t *tc0)

;-----------------------------------------------------------------------------

%macro DEBLOCK_LUMA 0

cglobal deblock_v_luma_8, 5,5,10

    movd    m8, [r4] ; tc0

    lea     r4, [r1*3]

    dec     r2d        ; alpha-1

    neg     r4

    dec     r3d        ; beta-1

    add     r4, r0     ; pix-3*stride

    mova    m0, [r4+r1]   ; p1

    mova    m1, [r4+2*r1] ; p0

    mova    m2, [r0]      ; q0

    mova    m3, [r0+r1]   ; q1

    LOAD_MASK r2d, r3d

    punpcklbw m8, m8

    punpcklbw m8, m8 ; tc = 4x tc0[3], 4x tc0[2], 4x tc0[1], 4x tc0[0]

    pcmpeqb m9, m9

    pcmpeqb m9, m8

    pandn   m9, m7

    pand    m8, m9

    movdqa  m3, [r4] ; p2

    DIFF_GT2 m1, m3, m5, m6, m7 ; |p2-p0| > beta-1

    pand    m6, m9

    psubb   m7, m8, m6

    pand    m6, m8

    LUMA_Q1 m0, m3, [r4], [r4+r1], m6, m4

    movdqa  m4, [r0+2*r1] ; q2

    DIFF_GT2 m2, m4, m5, m6, m3 ; |q2-q0| > beta-1

    pand    m6, m9

    pand    m8, m6

    psubb   m7, m6

    mova    m3, [r0+r1]

    LUMA_Q1 m3, m4, [r0+2*r1], [r0+r1], m8, m6

    DEBLOCK_P0_Q0

    mova    [r4+2*r1], m1

    mova    [r0], m2

    RET

;-----------------------------------------------------------------------------

; void ff_deblock_h_luma(uint8_t *pix, int stride, int alpha, int beta,

;                        int8_t *tc0)

;-----------------------------------------------------------------------------

INIT_MMX cpuname

cglobal deblock_h_luma_8, 5,9,0,0x60+16*WIN64

    movsxd r7,  r1d

    lea    r8,  [r7+r7*2]

    lea    r6,  [r0-4]

    lea    r5,  [r0-4+r8]

%if WIN64

    %define pix_tmp rsp+0x30 ; shadow space + r4

%else

    %define pix_tmp rsp

%endif

    ; transpose 6x16 -> tmp space

    TRANSPOSE6x8_MEM  PASS8ROWS(r6, r5, r7, r8), pix_tmp

    lea    r6, [r6+r7*8]

    lea    r5, [r5+r7*8]

    TRANSPOSE6x8_MEM  PASS8ROWS(r6, r5, r7, r8), pix_tmp+8

    ; vertical filter

    ; alpha, beta, tc0 are still in r2d, r3d, r4

    ; don't backup r6, r5, r7, r8 because deblock_v_luma_sse2 doesn't use them

    lea    r0, [pix_tmp+0x30]

    mov    r1d, 0x10

%if WIN64

    mov    [rsp+0x20], r4

%endif

    call   deblock_v_luma_8

    ; transpose 16x4 -> original space  (only the middle 4 rows were changed by the filter)

    add    r6, 2

    add    r5, 2

    movq   m0, [pix_tmp+0x18]

    movq   m1, [pix_tmp+0x28]

    movq   m2, [pix_tmp+0x38]

    movq   m3, [pix_tmp+0x48]

    TRANSPOSE8x4B_STORE  PASS8ROWS(r6, r5, r7, r8)

    shl    r7,  3

    sub    r6,  r7

    sub    r5,  r7

    shr    r7,  3

    movq   m0, [pix_tmp+0x10]

    movq   m1, [pix_tmp+0x20]

    movq   m2, [pix_tmp+0x30]

    movq   m3, [pix_tmp+0x40]

    TRANSPOSE8x4B_STORE  PASS8ROWS(r6, r5, r7, r8)

    RET

%endmacro

INIT_XMM sse2

DEBLOCK_LUMA

%if HAVE_AVX_EXTERNAL

INIT_XMM avx

DEBLOCK_LUMA

%endif

%else

%macro DEBLOCK_LUMA 2

;-----------------------------------------------------------------------------

; void ff_deblock_v8_luma(uint8_t *pix, int stride, int alpha, int beta,

;                         int8_t *tc0)

;-----------------------------------------------------------------------------

cglobal deblock_%1_luma_8, 5,5,8,2*%2

    lea     r4, [r1*3]

    dec     r2     ; alpha-1

    neg     r4

    dec     r3     ; beta-1

    add     r4, r0 ; pix-3*stride

    mova    m0, [r4+r1]   ; p1

    mova    m1, [r4+2*r1] ; p0

    mova    m2, [r0]      ; q0

    mova    m3, [r0+r1]   ; q1

    LOAD_MASK r2, r3

    mov     r3, r4mp

    pcmpeqb m3, m3

    movd    m4, [r3] ; tc0

    punpcklbw m4, m4

    punpcklbw m4, m4 ; tc = 4x tc0[3], 4x tc0[2], 4x tc0[1], 4x tc0[0]

    mova   [esp+%2], m4 ; tc

    pcmpgtb m4, m3

    mova    m3, [r4] ; p2

    pand    m4, m7

    mova   [esp], m4 ; mask

    DIFF_GT2 m1, m3, m5, m6, m7 ; |p2-p0| > beta-1

    pand    m6, m4

    pand    m4, [esp+%2] ; tc

    psubb   m7, m4, m6

    pand    m6, m4

    LUMA_Q1 m0, m3, [r4], [r4+r1], m6, m4

    mova    m4, [r0+2*r1] ; q2

    DIFF_GT2 m2, m4, m5, m6, m3 ; |q2-q0| > beta-1

    pand    m6, [esp] ; mask

    mova    m5, [esp+%2] ; tc

    psubb   m7, m6

    pand    m5, m6

    mova    m3, [r0+r1]

    LUMA_Q1 m3, m4, [r0+2*r1], [r0+r1], m5, m6

    DEBLOCK_P0_Q0

    mova    [r4+2*r1], m1

    mova    [r0], m2

    RET

;-----------------------------------------------------------------------------

; void ff_deblock_h_luma(uint8_t *pix, int stride, int alpha, int beta,

;                        int8_t *tc0)

;-----------------------------------------------------------------------------

INIT_MMX cpuname

cglobal deblock_h_luma_8, 0,5,8,0x60+12

    mov    r0, r0mp

    mov    r3, r1m

    lea    r4, [r3*3]

    sub    r0, 4

    lea    r1, [r0+r4]

%define pix_tmp esp+12

    ; transpose 6x16 -> tmp space

    TRANSPOSE6x8_MEM  PASS8ROWS(r0, r1, r3, r4), pix_tmp

    lea    r0, [r0+r3*8]

    lea    r1, [r1+r3*8]

    TRANSPOSE6x8_MEM  PASS8ROWS(r0, r1, r3, r4), pix_tmp+8

    ; vertical filter

    lea    r0, [pix_tmp+0x30]

    PUSH   dword r4m

    PUSH   dword r3m

    PUSH   dword r2m

    PUSH   dword 16

    PUSH   dword r0

    call   deblock_%1_luma_8

%ifidn %1, v8

    add    dword [esp   ], 8 ; pix_tmp+0x38

    add    dword [esp+16], 2 ; tc0+2

    call   deblock_%1_luma_8

%endif

    ADD    esp, 20

    ; transpose 16x4 -> original space  (only the middle 4 rows were changed by the filter)

    mov    r0, r0mp

    sub    r0, 2

    movq   m0, [pix_tmp+0x10]

    movq   m1, [pix_tmp+0x20]

    lea    r1, [r0+r4]

    movq   m2, [pix_tmp+0x30]

    movq   m3, [pix_tmp+0x40]

    TRANSPOSE8x4B_STORE  PASS8ROWS(r0, r1, r3, r4)

    lea    r0, [r0+r3*8]

    lea    r1, [r1+r3*8]

    movq   m0, [pix_tmp+0x18]

    movq   m1, [pix_tmp+0x28]

    movq   m2, [pix_tmp+0x38]

    movq   m3, [pix_tmp+0x48]

    TRANSPOSE8x4B_STORE  PASS8ROWS(r0, r1, r3, r4)

    RET

%endmacro ; DEBLOCK_LUMA

INIT_MMX mmxext

DEBLOCK_LUMA v8, 8

INIT_XMM sse2

DEBLOCK_LUMA v, 16

%if HAVE_AVX_EXTERNAL

INIT_XMM avx

DEBLOCK_LUMA v, 16

%endif

%endif ; ARCH

%macro LUMA_INTRA_P012 4 ; p0..p3 in memory

%if ARCH_X86_64

    pavgb t0, p2, p1

    pavgb t1, p0, q0

%else

    mova  t0, p2

    mova  t1, p0

    pavgb t0, p1

    pavgb t1, q0

%endif

    pavgb t0, t1 ; ((p2+p1+1)/2 + (p0+q0+1)/2 + 1)/2

    mova  t5, t1

%if ARCH_X86_64

    paddb t2, p2, p1

    paddb t3, p0, q0

%else

    mova  t2, p2

    mova  t3, p0

    paddb t2, p1

    paddb t3, q0

%endif

    paddb t2, t3

    mova  t3, t2

    mova  t4, t2

    psrlw t2, 1

    pavgb t2, mpb_0

    pxor  t2, t0

    pand  t2, mpb_1

    psubb t0, t2 ; p1' = (p2+p1+p0+q0+2)/4;

%if ARCH_X86_64

    pavgb t1, p2, q1

    psubb t2, p2, q1

%else

    mova  t1, p2

    mova  t2, p2

    pavgb t1, q1

    psubb t2, q1

%endif

    paddb t3, t3

    psubb t3, t2 ; p2+2*p1+2*p0+2*q0+q1

    pand  t2, mpb_1

    psubb t1, t2

    pavgb t1, p1

    pavgb t1, t5 ; (((p2+q1)/2 + p1+1)/2 + (p0+q0+1)/2 + 1)/2

    psrlw t3, 2

    pavgb t3, mpb_0

    pxor  t3, t1

    pand  t3, mpb_1

    psubb t1, t3 ; p0'a = (p2+2*p1+2*p0+2*q0+q1+4)/8

    pxor  t3, p0, q1

    pavgb t2, p0, q1

    pand  t3, mpb_1

    psubb t2, t3

    pavgb t2, p1 ; p0'b = (2*p1+p0+q0+2)/4

    pxor  t1, t2

    pxor  t2, p0

    pand  t1, mask1p

    pand  t2, mask0

    pxor  t1, t2

    pxor  t1, p0

    mova  %1, t1 ; store p0

    mova  t1, %4 ; p3

    paddb t2, t1, p2

    pavgb t1, p2

    pavgb t1, t0 ; (p3+p2+1)/2 + (p2+p1+p0+q0+2)/4

    paddb t2, t2

    paddb t2, t4 ; 2*p3+3*p2+p1+p0+q0

    psrlw t2, 2

    pavgb t2, mpb_0

    pxor  t2, t1

    pand  t2, mpb_1

    psubb t1, t2 ; p2' = (2*p3+3*p2+p1+p0+q0+4)/8

    pxor  t0, p1

    pxor  t1, p2

    pand  t0, mask1p

    pand  t1, mask1p

    pxor  t0, p1

    pxor  t1, p2

    mova  %2, t0 ; store p1

    mova  %3, t1 ; store p2

%endmacro

%macro LUMA_INTRA_SWAP_PQ 0

    %define q1 m0

    %define q0 m1

    %define p0 m2

    %define p1 m3

    %define p2 q2

    %define mask1p mask1q

%endmacro

%macro DEBLOCK_LUMA_INTRA 1

    %define p1 m0

    %define p0 m1

    %define q0 m2

    %define q1 m3

    %define t0 m4

    %define t1 m5

    %define t2 m6

    %define t3 m7

%if ARCH_X86_64

    %define p2 m8

    %define q2 m9

    %define t4 m10

    %define t5 m11

    %define mask0 m12

    %define mask1p m13

%if WIN64

    %define mask1q [rsp]

%else

    %define mask1q [rsp-24]

%endif

    %define mpb_0 m14

    %define mpb_1 m15

%else

    %define spill(x) [esp+16*x]

    %define p2 [r4+r1]

    %define q2 [r0+2*r1]

    %define t4 spill(0)

    %define t5 spill(1)

    %define mask0 spill(2)

    %define mask1p spill(3)

    %define mask1q spill(4)

    %define mpb_0 [pb_0]

    %define mpb_1 [pb_1]

%endif

;-----------------------------------------------------------------------------

; void ff_deblock_v_luma_intra(uint8_t *pix, int stride, int alpha, int beta)

;-----------------------------------------------------------------------------

%if WIN64

cglobal deblock_%1_luma_intra_8, 4,6,16,0x10

%else

cglobal deblock_%1_luma_intra_8, 4,6,16,ARCH_X86_64*0x50-0x50

%endif

    lea     r4, [r1*4]

    lea     r5, [r1*3] ; 3*stride

    dec     r2d        ; alpha-1

    jl .end

    neg     r4

    dec     r3d        ; beta-1

    jl .end

    add     r4, r0     ; pix-4*stride

    mova    p1, [r4+2*r1]

    mova    p0, [r4+r5]

    mova    q0, [r0]

    mova    q1, [r0+r1]

%if ARCH_X86_64

    pxor    mpb_0, mpb_0

    mova    mpb_1, [pb_1]

    LOAD_MASK r2d, r3d, t5 ; m5=beta-1, t5=alpha-1, m7=mask0

    SWAP    7, 12 ; m12=mask0

    pavgb   t5, mpb_0

    pavgb   t5, mpb_1 ; alpha/4+1

    movdqa  p2, [r4+r1]

    movdqa  q2, [r0+2*r1]

    DIFF_GT2 p0, q0, t5, t0, t3 ; t0 = |p0-q0| > alpha/4+1

    DIFF_GT2 p0, p2, m5, t2, t5 ; mask1 = |p2-p0| > beta-1

    DIFF_GT2 q0, q2, m5, t4, t5 ; t4 = |q2-q0| > beta-1

    pand    t0, mask0

    pand    t4, t0

    pand    t2, t0

    mova    mask1q, t4

    mova    mask1p, t2

%else

    LOAD_MASK r2d, r3d, t5 ; m5=beta-1, t5=alpha-1, m7=mask0

    mova    m4, t5

    mova    mask0, m7

    pavgb   m4, [pb_0]

    pavgb   m4, [pb_1] ; alpha/4+1

    DIFF_GT2 p0, q0, m4, m6, m7 ; m6 = |p0-q0| > alpha/4+1

    pand    m6, mask0

    DIFF_GT2 p0, p2, m5, m4, m7 ; m4 = |p2-p0| > beta-1

    pand    m4, m6

    mova    mask1p, m4

    DIFF_GT2 q0, q2, m5, m4, m7 ; m4 = |q2-q0| > beta-1

    pand    m4, m6

    mova    mask1q, m4

%endif

    LUMA_INTRA_P012 [r4+r5], [r4+2*r1], [r4+r1], [r4]

    LUMA_INTRA_SWAP_PQ

    LUMA_INTRA_P012 [r0], [r0+r1], [r0+2*r1], [r0+r5]

.end:

    RET

INIT_MMX cpuname

%if ARCH_X86_64

;-----------------------------------------------------------------------------

; void ff_deblock_h_luma_intra(uint8_t *pix, int stride, int alpha, int beta)

;-----------------------------------------------------------------------------

cglobal deblock_h_luma_intra_8, 4,9,0,0x80

    movsxd r7,  r1d

    lea    r8,  [r7*3]

    lea    r6,  [r0-4]

    lea    r5,  [r0-4+r8]

%if WIN64

    %define pix_tmp rsp+0x20 ; shadow space

%else

    %define pix_tmp rsp

%endif

    ; transpose 8x16 -> tmp space

    TRANSPOSE8x8_MEM  PASS8ROWS(r6, r5, r7, r8), PASS8ROWS(pix_tmp, pix_tmp+0x30, 0x10, 0x30)

    lea    r6, [r6+r7*8]

    lea    r5, [r5+r7*8]

    TRANSPOSE8x8_MEM  PASS8ROWS(r6, r5, r7, r8), PASS8ROWS(pix_tmp+8, pix_tmp+0x38, 0x10, 0x30)

    lea    r0,  [pix_tmp+0x40]

    mov    r1,  0x10

    call   deblock_v_luma_intra_8

    ; transpose 16x6 -> original space (but we can't write only 6 pixels, so really 16x8)

    lea    r5, [r6+r8]

    TRANSPOSE8x8_MEM  PASS8ROWS(pix_tmp+8, pix_tmp+0x38, 0x10, 0x30), PASS8ROWS(r6, r5, r7, r8)

    shl    r7,  3

    sub    r6,  r7

    sub    r5,  r7

    shr    r7,  3

    TRANSPOSE8x8_MEM  PASS8ROWS(pix_tmp, pix_tmp+0x30, 0x10, 0x30), PASS8ROWS(r6, r5, r7, r8)

    RET

%else

cglobal deblock_h_luma_intra_8, 2,4,8,0x80

    lea    r3,  [r1*3]

    sub    r0,  4

    lea    r2,  [r0+r3]

    %define pix_tmp rsp

    ; transpose 8x16 -> tmp space

    TRANSPOSE8x8_MEM  PASS8ROWS(r0, r2, r1, r3), PASS8ROWS(pix_tmp, pix_tmp+0x30, 0x10, 0x30)

    lea    r0,  [r0+r1*8]

    lea    r2,  [r2+r1*8]

    TRANSPOSE8x8_MEM  PASS8ROWS(r0, r2, r1, r3), PASS8ROWS(pix_tmp+8, pix_tmp+0x38, 0x10, 0x30)

    lea    r0,  [pix_tmp+0x40]

    PUSH   dword r3m

    PUSH   dword r2m

    PUSH   dword 16

    PUSH   r0

    call   deblock_%1_luma_intra_8

%ifidn %1, v8

    add    dword [rsp], 8 ; pix_tmp+8

    call   deblock_%1_luma_intra_8

%endif

    ADD    esp, 16

    mov    r1,  r1m

    mov    r0,  r0mp

    lea    r3,  [r1*3]

    sub    r0,  4

    lea    r2,  [r0+r3]

    ; transpose 16x6 -> original space (but we can't write only 6 pixels, so really 16x8)

    TRANSPOSE8x8_MEM  PASS8ROWS(pix_tmp, pix_tmp+0x30, 0x10, 0x30), PASS8ROWS(r0, r2, r1, r3)

    lea    r0,  [r0+r1*8]

    lea    r2,  [r2+r1*8]

    TRANSPOSE8x8_MEM  PASS8ROWS(pix_tmp+8, pix_tmp+0x38, 0x10, 0x30), PASS8ROWS(r0, r2, r1, r3)

    RET

%endif ; ARCH_X86_64

%endmacro ; DEBLOCK_LUMA_INTRA

INIT_XMM sse2

DEBLOCK_LUMA_INTRA v

%if HAVE_AVX_EXTERNAL

INIT_XMM avx

DEBLOCK_LUMA_INTRA v

%endif

%if ARCH_X86_64 == 0

INIT_MMX mmxext

DEBLOCK_LUMA_INTRA v8

%endif

INIT_MMX mmxext

%macro CHROMA_V_START 0

    dec    r2d      ; alpha-1

    dec    r3d      ; beta-1

    mov    t5, r0

    sub    t5, r1

    sub    t5, r1

%endmacro

%macro CHROMA_H_START 0

    dec    r2d

    dec    r3d

    sub    r0, 2

    lea    t6, [r1*3]

    mov    t5, r0

    add    r0, t6

%endmacro

%define t5 r5

%define t6 r6

;-----------------------------------------------------------------------------

; void ff_deblock_v_chroma(uint8_t *pix, int stride, int alpha, int beta,

;                          int8_t *tc0)

;-----------------------------------------------------------------------------

cglobal deblock_v_chroma_8, 5,6

    CHROMA_V_START

    movq  m0, [t5]

    movq  m1, [t5+r1]

    movq  m2, [r0]

    movq  m3, [r0+r1]

    call ff_chroma_inter_body_mmxext

    movq  [t5+r1], m1

    movq  [r0], m2

    RET

;-----------------------------------------------------------------------------

; void ff_deblock_h_chroma(uint8_t *pix, int stride, int alpha, int beta,

;                          int8_t *tc0)

;-----------------------------------------------------------------------------

cglobal deblock_h_chroma_8, 5,7

%if ARCH_X86_64

    ; This could use the red zone on 64 bit unix to avoid the stack pointer

    ; readjustment, but valgrind assumes the red zone is clobbered on

    ; function calls and returns.

    sub   rsp, 16

    %define buf0 [rsp]

    %define buf1 [rsp+8]

%else

    %define buf0 r0m

    %define buf1 r2m

%endif

    CHROMA_H_START

    TRANSPOSE4x8_LOAD  bw, wd, dq, PASS8ROWS(t5, r0, r1, t6)

    movq  buf0, m0

    movq  buf1, m3

    LOAD_MASK  r2d, r3d

    movd       m6, [r4] ; tc0

    punpcklbw  m6, m6

    pand       m7, m6

    DEBLOCK_P0_Q0

    movq  m0, buf0

    movq  m3, buf1

    TRANSPOSE8x4B_STORE PASS8ROWS(t5, r0, r1, t6)

%if ARCH_X86_64

    add   rsp, 16

%endif

    RET

ALIGN 16

ff_chroma_inter_body_mmxext:

    LOAD_MASK  r2d, r3d

    movd       m6, [r4] ; tc0

    punpcklbw  m6, m6

    pand       m7, m6

    DEBLOCK_P0_Q0

    ret

; in: %1=p0 %2=p1 %3=q1

; out: p0 = (p0 + q1 + 2*p1 + 2) >> 2

%macro CHROMA_INTRA_P0 3

    movq    m4, %1

    pxor    m4, %3

    pand    m4, [pb_1] ; m4 = (p0^q1)&1

    pavgb   %1, %3

    psubusb %1, m4

    pavgb   %1, %2             ; dst = avg(p1, avg(p0,q1) - ((p0^q1)&1))

%endmacro

%define t5 r4

%define t6 r5

;------------------------------------------------------------------------------

; void ff_deblock_v_chroma_intra(uint8_t *pix, int stride, int alpha, int beta)

;------------------------------------------------------------------------------

cglobal deblock_v_chroma_intra_8, 4,5

    CHROMA_V_START

    movq  m0, [t5]

    movq  m1, [t5+r1]

    movq  m2, [r0]

    movq  m3, [r0+r1]

    call ff_chroma_intra_body_mmxext

    movq  [t5+r1], m1

    movq  [r0], m2

    RET

;------------------------------------------------------------------------------

; void ff_deblock_h_chroma_intra(uint8_t *pix, int stride, int alpha, int beta)

;------------------------------------------------------------------------------

cglobal deblock_h_chroma_intra_8, 4,6

    CHROMA_H_START

    TRANSPOSE4x8_LOAD  bw, wd, dq, PASS8ROWS(t5, r0, r1, t6)

    call ff_chroma_intra_body_mmxext

    TRANSPOSE8x4B_STORE PASS8ROWS(t5, r0, r1, t6)

    RET

ALIGN 16

ff_chroma_intra_body_mmxext:

    LOAD_MASK r2d, r3d

    movq   m5, m1

    movq   m6, m2

    CHROMA_INTRA_P0  m1, m0, m3

    CHROMA_INTRA_P0  m2, m3, m0

    psubb  m1, m5

    psubb  m2, m6

    pand   m1, m7

    pand   m2, m7

    paddb  m1, m5

    paddb  m2, m6

    ret

;-----------------------------------------------------------------------------

; void ff_h264_loop_filter_strength(int16_t bs[2][4][4], uint8_t nnz[40],

;                                   int8_t ref[2][40], int16_t mv[2][40][2],

;                                   int bidir,    int edges,    int step,

;                                   int mask_mv0, int mask_mv1, int field);

;

; bidir    is 0 or 1

; edges    is 1 or 4

; step     is 1 or 2

; mask_mv0 is 0 or 3

; mask_mv1 is 0 or 1

; field    is 0 or 1

;-----------------------------------------------------------------------------

%macro loop_filter_strength_iteration 7 ; edges, step, mask_mv,

                                        ; dir, d_idx, mask_dir, bidir

%define edgesd    %1

%define stepd     %2

%define mask_mvd  %3

%define dir       %4

%define d_idx     %5

%define mask_dir  %6

%define bidir     %7

    xor          b_idxd, b_idxd ; for (b_idx = 0; b_idx < edges; b_idx += step)

%%.b_idx_loop:

%if mask_dir == 0

    pxor             m0, m0

%endif

    test         b_idxd, dword mask_mvd

    jnz %%.skip_loop_iter                       ; if (!(b_idx & mask_mv))

%if bidir == 1

    movd             m2, [refq+b_idxq+d_idx+12] ; { ref0[bn] }

    punpckldq        m2, [refq+b_idxq+d_idx+52] ; { ref0[bn], ref1[bn] }

    pshufw           m0, [refq+b_idxq+12], 0x44 ; { ref0[b],  ref0[b]  }

    pshufw           m1, [refq+b_idxq+52], 0x44 ; { ref1[b],  ref1[b]  }

    pshufw           m3, m2, 0x4E               ; { ref1[bn], ref0[bn] }

    psubb            m0, m2                     ; { ref0[b] != ref0[bn],

                                                ;   ref0[b] != ref1[bn] }

    psubb            m1, m3                     ; { ref1[b] != ref1[bn],

                                                ;   ref1[b] != ref0[bn] }

    por              m0, m1

    mova             m1, [mvq+b_idxq*4+(d_idx+12)*4]

    mova             m2, [mvq+b_idxq*4+(d_idx+12)*4+mmsize]

    mova             m3, m1

    mova             m4, m2

    psubw            m1, [mvq+b_idxq*4+12*4]

    psubw            m2, [mvq+b_idxq*4+12*4+mmsize]

    psubw            m3, [mvq+b_idxq*4+52*4]

    psubw            m4, [mvq+b_idxq*4+52*4+mmsize]

    packsswb         m1, m2

    packsswb         m3, m4

    paddb            m1, m6

    paddb            m3, m6

    psubusb          m1, m5 ; abs(mv[b] - mv[bn]) >= limit

    psubusb          m3, m5

    packsswb         m1, m3

    por              m0, m1

    mova             m1, [mvq+b_idxq*4+(d_idx+52)*4]

    mova             m2, [mvq+b_idxq*4+(d_idx+52)*4+mmsize]

    mova             m3, m1

    mova             m4, m2

    psubw            m1, [mvq+b_idxq*4+12*4]

    psubw            m2, [mvq+b_idxq*4+12*4+mmsize]

    psubw            m3, [mvq+b_idxq*4+52*4]

    psubw            m4, [mvq+b_idxq*4+52*4+mmsize]

    packsswb         m1, m2

    packsswb         m3, m4

    paddb            m1, m6

    paddb            m3, m6

    psubusb          m1, m5 ; abs(mv[b] - mv[bn]) >= limit

    psubusb          m3, m5

    packsswb         m1, m3

    pshufw           m1, m1, 0x4E

    por              m0, m1

    pshufw           m1, m0, 0x4E

    pminub           m0, m1

%else ; bidir == 0

    movd             m0, [refq+b_idxq+12]

    psubb            m0, [refq+b_idxq+d_idx+12] ; ref[b] != ref[bn]

    mova             m1, [mvq+b_idxq*4+12*4]

    mova             m2, [mvq+b_idxq*4+12*4+mmsize]

    psubw            m1, [mvq+b_idxq*4+(d_idx+12)*4]

    psubw            m2, [mvq+b_idxq*4+(d_idx+12)*4+mmsize]

    packsswb         m1, m2

    paddb            m1, m6

    psubusb          m1, m5 ; abs(mv[b] - mv[bn]) >= limit

    packsswb         m1, m1

    por              m0, m1

%endif ; bidir == 1/0

%%.skip_loop_iter:

    movd             m1, [nnzq+b_idxq+12]

    por              m1, [nnzq+b_idxq+d_idx+12] ; nnz[b] || nnz[bn]

    pminub           m1, m7

    pminub           m0, m7

    psllw            m1, 1

    pxor             m2, m2

    pmaxub           m1, m0

    punpcklbw        m1, m2

    movq [bsq+b_idxq+32*dir], m1

    add          b_idxd, dword stepd

    cmp          b_idxd, dword edgesd

    jl %%.b_idx_loop

%endmacro

INIT_MMX mmxext

cglobal h264_loop_filter_strength, 9, 9, 0, bs, nnz, ref, mv, bidir, edges, \

                                            step, mask_mv0, mask_mv1, field

%define b_idxq bidirq

%define b_idxd bidird

    cmp    dword fieldm, 0

    mova             m7, [pb_1]

    mova             m5, [pb_3]

    je .nofield

    mova             m5, [pb_3_1]

.nofield:

    mova             m6, m5

    paddb            m5, m5

    shl     dword stepd, 3

    shl    dword edgesd, 3

%if ARCH_X86_32

%define mask_mv0d mask_mv0m

%define mask_mv1d mask_mv1m

%endif

    shl dword mask_mv1d, 3

    shl dword mask_mv0d, 3

    cmp    dword bidird, 0

    jne .bidir

    loop_filter_strength_iteration edgesd, stepd, mask_mv1d, 1, -8,  0, 0

    loop_filter_strength_iteration     32,     8, mask_mv0d, 0, -1, -1, 0

    mova             m0, [bsq+mmsize*0]

    mova             m1, [bsq+mmsize*1]

    mova             m2, [bsq+mmsize*2]

    mova             m3, [bsq+mmsize*3]

    TRANSPOSE4x4W 0, 1, 2, 3, 4

    mova  [bsq+mmsize*0], m0

    mova  [bsq+mmsize*1], m1

    mova  [bsq+mmsize*2], m2

    mova  [bsq+mmsize*3], m3

    RET

.bidir:

    loop_filter_strength_iteration edgesd, stepd, mask_mv1d, 1, -8,  0, 1

    loop_filter_strength_iteration     32,     8, mask_mv0d, 0, -1, -1, 1

    mova             m0, [bsq+mmsize*0]

    mova             m1, [bsq+mmsize*1]

    mova             m2, [bsq+mmsize*2]

    mova             m3, [bsq+mmsize*3]

    TRANSPOSE4x4W 0, 1, 2, 3, 4

    mova  [bsq+mmsize*0], m0

    mova  [bsq+mmsize*1], m1

    mova  [bsq+mmsize*2], m2

    mova  [bsq+mmsize*3], m3

    RET