mirror of
				https://github.com/nyanmisaka/ffmpeg-rockchip.git
				synced 2025-10-31 12:36:41 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			447 lines
		
	
	
		
			9.3 KiB
		
	
	
	
		
			NASM
		
	
	
	
	
	
			
		
		
	
	
			447 lines
		
	
	
		
			9.3 KiB
		
	
	
	
		
			NASM
		
	
	
	
	
	
| ;*****************************************************************************
 | |
| ;* x86util.asm
 | |
| ;*****************************************************************************
 | |
| ;* Copyright (C) 2008-2010 x264 project
 | |
| ;*
 | |
| ;* Authors: Loren Merritt <lorenm@u.washington.edu>
 | |
| ;*          Holger Lubitz <holger@lubitz.org>
 | |
| ;*
 | |
| ;* This file is part of Libav.
 | |
| ;*
 | |
| ;* Libav 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.
 | |
| ;*
 | |
| ;* Libav 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 Libav; if not, write to the Free Software
 | |
| ;* 51, Inc., Foundation Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 | |
| ;******************************************************************************
 | |
| 
 | |
| %macro SBUTTERFLY 4
 | |
|     mova      m%4, m%2
 | |
|     punpckl%1 m%2, m%3
 | |
|     punpckh%1 m%4, m%3
 | |
|     SWAP %3, %4
 | |
| %endmacro
 | |
| 
 | |
| %macro SBUTTERFLY2 4
 | |
|     mova      m%4, m%2
 | |
|     punpckh%1 m%2, m%3
 | |
|     punpckl%1 m%4, m%3
 | |
|     SWAP %2, %4, %3
 | |
| %endmacro
 | |
| 
 | |
| %macro TRANSPOSE4x4B 5
 | |
|     SBUTTERFLY bw, %1, %2, %5
 | |
|     SBUTTERFLY bw, %3, %4, %5
 | |
|     SBUTTERFLY wd, %1, %3, %5
 | |
|     SBUTTERFLY wd, %2, %4, %5
 | |
|     SWAP %2, %3
 | |
| %endmacro
 | |
| 
 | |
| %macro TRANSPOSE4x4W 5
 | |
|     SBUTTERFLY wd, %1, %2, %5
 | |
|     SBUTTERFLY wd, %3, %4, %5
 | |
|     SBUTTERFLY dq, %1, %3, %5
 | |
|     SBUTTERFLY dq, %2, %4, %5
 | |
|     SWAP %2, %3
 | |
| %endmacro
 | |
| 
 | |
| %macro TRANSPOSE2x4x4W 5
 | |
|     SBUTTERFLY wd,  %1, %2, %5
 | |
|     SBUTTERFLY wd,  %3, %4, %5
 | |
|     SBUTTERFLY dq,  %1, %3, %5
 | |
|     SBUTTERFLY dq,  %2, %4, %5
 | |
|     SBUTTERFLY qdq, %1, %2, %5
 | |
|     SBUTTERFLY qdq, %3, %4, %5
 | |
| %endmacro
 | |
| 
 | |
| %macro TRANSPOSE4x4D 5
 | |
|     SBUTTERFLY dq,  %1, %2, %5
 | |
|     SBUTTERFLY dq,  %3, %4, %5
 | |
|     SBUTTERFLY qdq, %1, %3, %5
 | |
|     SBUTTERFLY qdq, %2, %4, %5
 | |
|     SWAP %2, %3
 | |
| %endmacro
 | |
| 
 | |
| %macro TRANSPOSE8x8W 9-11
 | |
| %ifdef ARCH_X86_64
 | |
|     SBUTTERFLY wd,  %1, %2, %9
 | |
|     SBUTTERFLY wd,  %3, %4, %9
 | |
|     SBUTTERFLY wd,  %5, %6, %9
 | |
|     SBUTTERFLY wd,  %7, %8, %9
 | |
|     SBUTTERFLY dq,  %1, %3, %9
 | |
|     SBUTTERFLY dq,  %2, %4, %9
 | |
|     SBUTTERFLY dq,  %5, %7, %9
 | |
|     SBUTTERFLY dq,  %6, %8, %9
 | |
|     SBUTTERFLY qdq, %1, %5, %9
 | |
|     SBUTTERFLY qdq, %2, %6, %9
 | |
|     SBUTTERFLY qdq, %3, %7, %9
 | |
|     SBUTTERFLY qdq, %4, %8, %9
 | |
|     SWAP %2, %5
 | |
|     SWAP %4, %7
 | |
| %else
 | |
| ; in:  m0..m7, unless %11 in which case m6 is in %9
 | |
| ; out: m0..m7, unless %11 in which case m4 is in %10
 | |
| ; spills into %9 and %10
 | |
| %if %0<11
 | |
|     movdqa %9, m%7
 | |
| %endif
 | |
|     SBUTTERFLY wd,  %1, %2, %7
 | |
|     movdqa %10, m%2
 | |
|     movdqa m%7, %9
 | |
|     SBUTTERFLY wd,  %3, %4, %2
 | |
|     SBUTTERFLY wd,  %5, %6, %2
 | |
|     SBUTTERFLY wd,  %7, %8, %2
 | |
|     SBUTTERFLY dq,  %1, %3, %2
 | |
|     movdqa %9, m%3
 | |
|     movdqa m%2, %10
 | |
|     SBUTTERFLY dq,  %2, %4, %3
 | |
|     SBUTTERFLY dq,  %5, %7, %3
 | |
|     SBUTTERFLY dq,  %6, %8, %3
 | |
|     SBUTTERFLY qdq, %1, %5, %3
 | |
|     SBUTTERFLY qdq, %2, %6, %3
 | |
|     movdqa %10, m%2
 | |
|     movdqa m%3, %9
 | |
|     SBUTTERFLY qdq, %3, %7, %2
 | |
|     SBUTTERFLY qdq, %4, %8, %2
 | |
|     SWAP %2, %5
 | |
|     SWAP %4, %7
 | |
| %if %0<11
 | |
|     movdqa m%5, %10
 | |
| %endif
 | |
| %endif
 | |
| %endmacro
 | |
| 
 | |
| ; PABSW macros assume %1 != %2, while ABS1/2 macros work in-place
 | |
| %macro PABSW_MMX 2
 | |
|     pxor       %1, %1
 | |
|     pcmpgtw    %1, %2
 | |
|     pxor       %2, %1
 | |
|     psubw      %2, %1
 | |
|     SWAP       %1, %2
 | |
| %endmacro
 | |
| 
 | |
| %macro PSIGNW_MMX 2
 | |
|     pxor       %1, %2
 | |
|     psubw      %1, %2
 | |
| %endmacro
 | |
| 
 | |
| %macro PABSW_MMX2 2
 | |
|     pxor    %1, %1
 | |
|     psubw   %1, %2
 | |
|     pmaxsw  %1, %2
 | |
| %endmacro
 | |
| 
 | |
| %macro PABSW_SSSE3 2
 | |
|     pabsw      %1, %2
 | |
| %endmacro
 | |
| 
 | |
| %macro PSIGNW_SSSE3 2
 | |
|     psignw     %1, %2
 | |
| %endmacro
 | |
| 
 | |
| %macro ABS1_MMX 2    ; a, tmp
 | |
|     pxor       %2, %2
 | |
|     pcmpgtw    %2, %1
 | |
|     pxor       %1, %2
 | |
|     psubw      %1, %2
 | |
| %endmacro
 | |
| 
 | |
| %macro ABS2_MMX 4    ; a, b, tmp0, tmp1
 | |
|     pxor       %3, %3
 | |
|     pxor       %4, %4
 | |
|     pcmpgtw    %3, %1
 | |
|     pcmpgtw    %4, %2
 | |
|     pxor       %1, %3
 | |
|     pxor       %2, %4
 | |
|     psubw      %1, %3
 | |
|     psubw      %2, %4
 | |
| %endmacro
 | |
| 
 | |
| %macro ABS1_MMX2 2   ; a, tmp
 | |
|     pxor    %2, %2
 | |
|     psubw   %2, %1
 | |
|     pmaxsw  %1, %2
 | |
| %endmacro
 | |
| 
 | |
| %macro ABS2_MMX2 4   ; a, b, tmp0, tmp1
 | |
|     pxor    %3, %3
 | |
|     pxor    %4, %4
 | |
|     psubw   %3, %1
 | |
|     psubw   %4, %2
 | |
|     pmaxsw  %1, %3
 | |
|     pmaxsw  %2, %4
 | |
| %endmacro
 | |
| 
 | |
| %macro ABS1_SSSE3 2
 | |
|     pabsw   %1, %1
 | |
| %endmacro
 | |
| 
 | |
| %macro ABS2_SSSE3 4
 | |
|     pabsw   %1, %1
 | |
|     pabsw   %2, %2
 | |
| %endmacro
 | |
| 
 | |
| %macro ABSB_MMX 2
 | |
|     pxor    %2, %2
 | |
|     psubb   %2, %1
 | |
|     pminub  %1, %2
 | |
| %endmacro
 | |
| 
 | |
| %macro ABSB2_MMX 4
 | |
|     pxor    %3, %3
 | |
|     pxor    %4, %4
 | |
|     psubb   %3, %1
 | |
|     psubb   %4, %2
 | |
|     pminub  %1, %3
 | |
|     pminub  %2, %4
 | |
| %endmacro
 | |
| 
 | |
| %macro ABSB_SSSE3 2
 | |
|     pabsb   %1, %1
 | |
| %endmacro
 | |
| 
 | |
| %macro ABSB2_SSSE3 4
 | |
|     pabsb   %1, %1
 | |
|     pabsb   %2, %2
 | |
| %endmacro
 | |
| 
 | |
| %macro ABS4 6
 | |
|     ABS2 %1, %2, %5, %6
 | |
|     ABS2 %3, %4, %5, %6
 | |
| %endmacro
 | |
| 
 | |
| %define ABS1 ABS1_MMX
 | |
| %define ABS2 ABS2_MMX
 | |
| %define ABSB ABSB_MMX
 | |
| %define ABSB2 ABSB2_MMX
 | |
| 
 | |
| %macro SPLATB_MMX 3
 | |
|     movd      %1, [%2-3] ;to avoid crossing a cacheline
 | |
|     punpcklbw %1, %1
 | |
| %if mmsize==16
 | |
|     pshuflw   %1, %1, 0xff
 | |
|     punpcklqdq %1, %1
 | |
| %else
 | |
|     pshufw    %1, %1, 0xff
 | |
| %endif
 | |
| %endmacro
 | |
| 
 | |
| %macro SPLATB_SSSE3 3
 | |
|     movd      %1, [%2-3]
 | |
|     pshufb    %1, %3
 | |
| %endmacro
 | |
| 
 | |
| %macro PALIGNR_MMX 4
 | |
|     %ifnidn %4, %2
 | |
|     mova    %4, %2
 | |
|     %endif
 | |
|     %if mmsize == 8
 | |
|     psllq   %1, (8-%3)*8
 | |
|     psrlq   %4, %3*8
 | |
|     %else
 | |
|     pslldq  %1, 16-%3
 | |
|     psrldq  %4, %3
 | |
|     %endif
 | |
|     por     %1, %4
 | |
| %endmacro
 | |
| 
 | |
| %macro PALIGNR_SSSE3 4
 | |
|     palignr %1, %2, %3
 | |
| %endmacro
 | |
| 
 | |
| %macro DEINTB 5 ; mask, reg1, mask, reg2, optional src to fill masks from
 | |
| %ifnum %5
 | |
|     mova   m%1, m%5
 | |
|     mova   m%3, m%5
 | |
| %else
 | |
|     mova   m%1, %5
 | |
|     mova   m%3, m%1
 | |
| %endif
 | |
|     pand   m%1, m%2 ; dst .. y6 .. y4
 | |
|     pand   m%3, m%4 ; src .. y6 .. y4
 | |
|     psrlw  m%2, 8   ; dst .. y7 .. y5
 | |
|     psrlw  m%4, 8   ; src .. y7 .. y5
 | |
| %endmacro
 | |
| 
 | |
| %macro SUMSUB_BA 2-3
 | |
| %if %0==2
 | |
|     paddw   %1, %2
 | |
|     paddw   %2, %2
 | |
|     psubw   %2, %1
 | |
| %else
 | |
|     mova    %3, %1
 | |
|     paddw   %1, %2
 | |
|     psubw   %2, %3
 | |
| %endif
 | |
| %endmacro
 | |
| 
 | |
| %macro SUMSUB_BADC 4-5
 | |
| %if %0==5
 | |
|     SUMSUB_BA %1, %2, %5
 | |
|     SUMSUB_BA %3, %4, %5
 | |
| %else
 | |
|     paddw   %1, %2
 | |
|     paddw   %3, %4
 | |
|     paddw   %2, %2
 | |
|     paddw   %4, %4
 | |
|     psubw   %2, %1
 | |
|     psubw   %4, %3
 | |
| %endif
 | |
| %endmacro
 | |
| 
 | |
| %macro SUMSUB2_AB 3
 | |
|     mova    %3, %1
 | |
|     paddw   %1, %1
 | |
|     paddw   %1, %2
 | |
|     psubw   %3, %2
 | |
|     psubw   %3, %2
 | |
| %endmacro
 | |
| 
 | |
| %macro SUMSUB2_BA 3
 | |
|     mova    m%3, m%1
 | |
|     paddw   m%1, m%2
 | |
|     paddw   m%1, m%2
 | |
|     psubw   m%2, m%3
 | |
|     psubw   m%2, m%3
 | |
| %endmacro
 | |
| 
 | |
| %macro SUMSUBD2_AB 4
 | |
|     mova    %4, %1
 | |
|     mova    %3, %2
 | |
|     psraw   %2, 1  ; %2: %2>>1
 | |
|     psraw   %1, 1  ; %1: %1>>1
 | |
|     paddw   %2, %4 ; %2: %2>>1+%1
 | |
|     psubw   %1, %3 ; %1: %1>>1-%2
 | |
| %endmacro
 | |
| 
 | |
| %macro DCT4_1D 5
 | |
| %ifnum %5
 | |
|     SUMSUB_BADC m%4, m%1, m%3, m%2; m%5
 | |
|     SUMSUB_BA   m%3, m%4, m%5
 | |
|     SUMSUB2_AB  m%1, m%2, m%5
 | |
|     SWAP %1, %3, %4, %5, %2
 | |
| %else
 | |
|     SUMSUB_BADC m%4, m%1, m%3, m%2
 | |
|     SUMSUB_BA   m%3, m%4
 | |
|     mova       [%5], m%2
 | |
|     SUMSUB2_AB m%1, [%5], m%2
 | |
|     SWAP %1, %3, %4, %2
 | |
| %endif
 | |
| %endmacro
 | |
| 
 | |
| %macro IDCT4_1D 5-6
 | |
| %ifnum %5
 | |
|     SUMSUBD2_AB m%2, m%4, m%6, m%5
 | |
|     ; %2: %2>>1-%4 %4: %2+%4>>1
 | |
|     SUMSUB_BA   m%3, m%1, m%6
 | |
|     ; %3: %1+%3 %1: %1-%3
 | |
|     SUMSUB_BADC m%4, m%3, m%2, m%1, m%6
 | |
|     ; %4: %1+%3 + (%2+%4>>1)
 | |
|     ; %3: %1+%3 - (%2+%4>>1)
 | |
|     ; %2: %1-%3 + (%2>>1-%4)
 | |
|     ; %1: %1-%3 - (%2>>1-%4)
 | |
| %else
 | |
|     SUMSUBD2_AB m%2, m%4, [%5], [%5+16]
 | |
|     SUMSUB_BA   m%3, m%1
 | |
|     SUMSUB_BADC m%4, m%3, m%2, m%1
 | |
| %endif
 | |
|     SWAP %1, %4, %3
 | |
|     ; %1: %1+%3 + (%2+%4>>1) row0
 | |
|     ; %2: %1-%3 + (%2>>1-%4) row1
 | |
|     ; %3: %1-%3 - (%2>>1-%4) row2
 | |
|     ; %4: %1+%3 - (%2+%4>>1) row3
 | |
| %endmacro
 | |
| 
 | |
| 
 | |
| %macro LOAD_DIFF 5
 | |
| %ifidn %3, none
 | |
|     movh       %1, %4
 | |
|     movh       %2, %5
 | |
|     punpcklbw  %1, %2
 | |
|     punpcklbw  %2, %2
 | |
|     psubw      %1, %2
 | |
| %else
 | |
|     movh       %1, %4
 | |
|     punpcklbw  %1, %3
 | |
|     movh       %2, %5
 | |
|     punpcklbw  %2, %3
 | |
|     psubw      %1, %2
 | |
| %endif
 | |
| %endmacro
 | |
| 
 | |
| %macro STORE_DCT 6
 | |
|     movq   [%5+%6+ 0], m%1
 | |
|     movq   [%5+%6+ 8], m%2
 | |
|     movq   [%5+%6+16], m%3
 | |
|     movq   [%5+%6+24], m%4
 | |
|     movhps [%5+%6+32], m%1
 | |
|     movhps [%5+%6+40], m%2
 | |
|     movhps [%5+%6+48], m%3
 | |
|     movhps [%5+%6+56], m%4
 | |
| %endmacro
 | |
| 
 | |
| %macro LOAD_DIFF_8x4P 7-10 r0,r2,0 ; 4x dest, 2x temp, 2x pointer, increment?
 | |
|     LOAD_DIFF m%1, m%5, m%7, [%8],      [%9]
 | |
|     LOAD_DIFF m%2, m%6, m%7, [%8+r1],   [%9+r3]
 | |
|     LOAD_DIFF m%3, m%5, m%7, [%8+2*r1], [%9+2*r3]
 | |
|     LOAD_DIFF m%4, m%6, m%7, [%8+r4],   [%9+r5]
 | |
| %if %10
 | |
|     lea %8, [%8+4*r1]
 | |
|     lea %9, [%9+4*r3]
 | |
| %endif
 | |
| %endmacro
 | |
| 
 | |
| %macro DIFFx2 6-7
 | |
|     movh       %3, %5
 | |
|     punpcklbw  %3, %4
 | |
|     psraw      %1, 6
 | |
|     paddsw     %1, %3
 | |
|     movh       %3, %6
 | |
|     punpcklbw  %3, %4
 | |
|     psraw      %2, 6
 | |
|     paddsw     %2, %3
 | |
|     packuswb   %2, %1
 | |
| %endmacro
 | |
| 
 | |
| %macro STORE_DIFF 4
 | |
|     movh       %2, %4
 | |
|     punpcklbw  %2, %3
 | |
|     psraw      %1, 6
 | |
|     paddsw     %1, %2
 | |
|     packuswb   %1, %1
 | |
|     movh       %4, %1
 | |
| %endmacro
 | |
| 
 | |
| %macro STORE_DIFFx2 8 ; add1, add2, reg1, reg2, zero, shift, source, stride
 | |
|     movh       %3, [%7]
 | |
|     movh       %4, [%7+%8]
 | |
|     punpcklbw  %3, %5
 | |
|     punpcklbw  %4, %5
 | |
|     psraw      %1, %6
 | |
|     psraw      %2, %6
 | |
|     paddw      %3, %1
 | |
|     paddw      %4, %2
 | |
|     packuswb   %3, %5
 | |
|     packuswb   %4, %5
 | |
|     movh     [%7], %3
 | |
|     movh  [%7+%8], %4
 | |
| %endmacro
 | |
| 
 | |
| %macro PMINUB_MMX 3 ; dst, src, tmp
 | |
|     mova     %3, %1
 | |
|     psubusb  %3, %2
 | |
|     psubb    %1, %3
 | |
| %endmacro
 | |
| 
 | |
| %macro PMINUB_MMXEXT 3 ; dst, src, ignored
 | |
|     pminub   %1, %2
 | |
| %endmacro
 | 
