mirror of https://git.ffmpeg.org/ffmpeg.git
116 lines
3.2 KiB
NASM
116 lines
3.2 KiB
NASM
;******************************************************************************
|
|
;* SIMD-optimized clear block functions
|
|
;* Copyright (c) 2002 Michael Niedermayer
|
|
;* Copyright (c) 2008 Loren Merritt
|
|
;* Copyright (c) 2009 Fiona Glaser
|
|
;*
|
|
;* AVX version by Jokyo Images
|
|
;*
|
|
;* 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 .text
|
|
|
|
;----------------------------------------
|
|
; void ff_clear_block(int16_t *blocks);
|
|
;----------------------------------------
|
|
; %1 = number of xmm registers used
|
|
; %2 = number of inline store loops
|
|
%macro CLEAR_BLOCK 2
|
|
cglobal clear_block, 1, 1, %1, blocks
|
|
ZERO m0, m0, m0
|
|
%assign %%i 0
|
|
%rep %2
|
|
mova [blocksq+mmsize*(0+%%i)], m0
|
|
mova [blocksq+mmsize*(1+%%i)], m0
|
|
mova [blocksq+mmsize*(2+%%i)], m0
|
|
mova [blocksq+mmsize*(3+%%i)], m0
|
|
%assign %%i %%i+4
|
|
%endrep
|
|
RET
|
|
%endmacro
|
|
|
|
INIT_XMM sse
|
|
%define ZERO xorps
|
|
CLEAR_BLOCK 1, 2
|
|
INIT_YMM avx
|
|
CLEAR_BLOCK 1, 1
|
|
|
|
;-----------------------------------------
|
|
; void ff_clear_blocks(int16_t *blocks);
|
|
;-----------------------------------------
|
|
; %1 = number of xmm registers used
|
|
%macro CLEAR_BLOCKS 1
|
|
cglobal clear_blocks, 1, 2, %1, blocks, len
|
|
add blocksq, 768
|
|
mov lenq, -768
|
|
ZERO m0, m0, m0
|
|
.loop:
|
|
mova [blocksq+lenq+mmsize*0], m0
|
|
mova [blocksq+lenq+mmsize*1], m0
|
|
mova [blocksq+lenq+mmsize*2], m0
|
|
mova [blocksq+lenq+mmsize*3], m0
|
|
mova [blocksq+lenq+mmsize*4], m0
|
|
mova [blocksq+lenq+mmsize*5], m0
|
|
mova [blocksq+lenq+mmsize*6], m0
|
|
mova [blocksq+lenq+mmsize*7], m0
|
|
add lenq, mmsize*8
|
|
js .loop
|
|
RET
|
|
%endmacro
|
|
|
|
INIT_XMM sse
|
|
%define ZERO xorps
|
|
CLEAR_BLOCKS 1
|
|
INIT_YMM avx
|
|
CLEAR_BLOCKS 1
|
|
|
|
;-----------------------------------------
|
|
; void ff_fill_block_tab_%1(uint8_t *block, uint8_t value,
|
|
; ptrdiff_t line_size, int h);
|
|
;-----------------------------------------
|
|
%macro FILL_BLOCK_TAB 2
|
|
cglobal fill_block_tab_%1, 4, 5, 1, block, value, stride, h, stride3
|
|
lea stride3q, [strideq + strideq * 2]
|
|
%if cpuflag(avx2)
|
|
movd m0, valued
|
|
vpbroadcastb m0, m0
|
|
%else
|
|
SPLATB_REG m0, value, x
|
|
%endif
|
|
.loop:
|
|
mov%2 [blockq], m0
|
|
mov%2 [blockq + strideq], m0
|
|
mov%2 [blockq + strideq * 2], m0
|
|
mov%2 [blockq + stride3q], m0
|
|
lea blockq, [blockq + strideq * 4]
|
|
sub hd, 4
|
|
jg .loop
|
|
RET
|
|
%endmacro
|
|
|
|
INIT_XMM sse2
|
|
FILL_BLOCK_TAB 8, q
|
|
FILL_BLOCK_TAB 16, a
|
|
%if HAVE_AVX2_EXTERNAL
|
|
INIT_XMM avx2
|
|
FILL_BLOCK_TAB 8, q
|
|
FILL_BLOCK_TAB 16, a
|
|
%endif
|