12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028 |
- ;*****************************************************************************
- ;* 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 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
- ;******************************************************************************
- %define private_prefix ff
- %define public_prefix avpriv
- %define cpuflags_mmxext cpuflags_mmx2
- %include "libavutil/x86/x86inc.asm"
- ; expands to [base],...,[base+7*stride]
- %define PASS8ROWS(base, base3, stride, stride3) \
- [base], [base + stride], [base + 2*stride], [base3], \
- [base3 + stride], [base3 + 2*stride], [base3 + stride3], [base3 + stride*4]
- ; Interleave low src0 with low src1 and store in src0,
- ; interleave high src0 with high src1 and store in src1.
- ; %1 - types
- ; %2 - index of the register with src0
- ; %3 - index of the register with src1
- ; %4 - index of the register for intermediate results
- ; example for %1 - wd: input: src0: x0 x1 x2 x3 z0 z1 z2 z3
- ; src1: y0 y1 y2 y3 q0 q1 q2 q3
- ; output: src0: x0 y0 x1 y1 x2 y2 x3 y3
- ; src1: z0 q0 z1 q1 z2 q2 z3 q3
- %macro SBUTTERFLY 4
- %ifidn %1, dqqq
- vperm2i128 m%4, m%2, m%3, q0301
- vinserti128 m%2, m%2, xm%3, 1
- %elif avx_enabled == 0
- mova m%4, m%2
- punpckl%1 m%2, m%3
- punpckh%1 m%4, m%3
- %else
- punpckh%1 m%4, m%2, m%3
- punpckl%1 m%2, m%3
- %endif
- SWAP %3, %4
- %endmacro
- %macro SBUTTERFLY2 4
- punpckl%1 m%4, m%2, m%3
- punpckh%1 m%2, m%2, m%3
- SWAP %2, %4, %3
- %endmacro
- %macro SBUTTERFLYPS 3
- unpcklps m%3, m%1, m%2
- unpckhps m%1, m%1, m%2
- SWAP %1, %3, %2
- %endmacro
- %macro SBUTTERFLYPD 3
- movlhps m%3, m%1, m%2
- movhlps m%2, m%2, m%1
- SWAP %1, %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 TRANSPOSE2x4x4B 5
- SBUTTERFLY bw, %1, %2, %5
- SBUTTERFLY bw, %3, %4, %5
- SBUTTERFLY wd, %1, %3, %5
- SBUTTERFLY wd, %2, %4, %5
- SBUTTERFLY dq, %1, %2, %5
- SBUTTERFLY dq, %3, %4, %5
- %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
- ; identical behavior to TRANSPOSE4x4D, but using SSE1 float ops
- %macro TRANSPOSE4x4PS 5
- SBUTTERFLYPS %1, %2, %5
- SBUTTERFLYPS %3, %4, %5
- SBUTTERFLYPD %1, %3, %5
- SBUTTERFLYPD %2, %4, %5
- SWAP %2, %3
- %endmacro
- %macro TRANSPOSE8x4D 9-11
- %if ARCH_X86_64
- SBUTTERFLY dq, %1, %2, %9
- SBUTTERFLY dq, %3, %4, %9
- SBUTTERFLY dq, %5, %6, %9
- SBUTTERFLY dq, %7, %8, %9
- SBUTTERFLY qdq, %1, %3, %9
- SBUTTERFLY qdq, %2, %4, %9
- SBUTTERFLY qdq, %5, %7, %9
- SBUTTERFLY qdq, %6, %8, %9
- SWAP %2, %5
- SWAP %4, %7
- %else
- ; in: m0..m7
- ; out: m0..m7, unless %11 in which case m2 is in %9
- ; spills into %9 and %10
- movdqa %9, m%7
- SBUTTERFLY dq, %1, %2, %7
- movdqa %10, m%2
- movdqa m%7, %9
- SBUTTERFLY dq, %3, %4, %2
- SBUTTERFLY dq, %5, %6, %2
- SBUTTERFLY dq, %7, %8, %2
- SBUTTERFLY qdq, %1, %3, %2
- movdqa %9, m%3
- movdqa m%2, %10
- SBUTTERFLY qdq, %2, %4, %3
- SBUTTERFLY qdq, %5, %7, %3
- SBUTTERFLY qdq, %6, %8, %3
- SWAP %2, %5
- SWAP %4, %7
- %if %0<11
- movdqa m%3, %9
- %endif
- %endif
- %endmacro
- %macro TRANSPOSE8x8W 9-11
- %if 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
- %macro TRANSPOSE16x16W 18-19
- ; in: m0..m15, unless %19 in which case m6 is in %17
- ; out: m0..m15, unless %19 in which case m4 is in %18
- ; spills into %17 and %18
- %if %0 < 19
- mova %17, m%7
- %endif
- SBUTTERFLY dqqq, %1, %9, %7
- SBUTTERFLY dqqq, %2, %10, %7
- SBUTTERFLY dqqq, %3, %11, %7
- SBUTTERFLY dqqq, %4, %12, %7
- SBUTTERFLY dqqq, %5, %13, %7
- SBUTTERFLY dqqq, %6, %14, %7
- mova %18, m%14
- mova m%7, %17
- SBUTTERFLY dqqq, %7, %15, %14
- SBUTTERFLY dqqq, %8, %16, %14
- SBUTTERFLY wd, %1, %2, %14
- SBUTTERFLY wd, %3, %4, %14
- SBUTTERFLY wd, %5, %6, %14
- SBUTTERFLY wd, %7, %8, %14
- SBUTTERFLY wd, %9, %10, %14
- SBUTTERFLY wd, %11, %12, %14
- mova %17, m%12
- mova m%14, %18
- SBUTTERFLY wd, %13, %14, %12
- SBUTTERFLY wd, %15, %16, %12
- SBUTTERFLY dq, %1, %3, %12
- SBUTTERFLY dq, %2, %4, %12
- SBUTTERFLY dq, %5, %7, %12
- SBUTTERFLY dq, %6, %8, %12
- SBUTTERFLY dq, %9, %11, %12
- mova %18, m%11
- mova m%12, %17
- SBUTTERFLY dq, %10, %12, %11
- SBUTTERFLY dq, %13, %15, %11
- SBUTTERFLY dq, %14, %16, %11
- SBUTTERFLY qdq, %1, %5, %11
- SBUTTERFLY qdq, %2, %6, %11
- SBUTTERFLY qdq, %3, %7, %11
- SBUTTERFLY qdq, %4, %8, %11
- SWAP %2, %5
- SWAP %4, %7
- SBUTTERFLY qdq, %9, %13, %11
- SBUTTERFLY qdq, %10, %14, %11
- mova m%11, %18
- mova %18, m%5
- SBUTTERFLY qdq, %11, %15, %5
- SBUTTERFLY qdq, %12, %16, %5
- %if %0 < 19
- mova m%5, %18
- %endif
- SWAP %10, %13
- SWAP %12, %15
- %endmacro
- %macro TRANSPOSE_8X8B 8
- %if mmsize == 8
- %error "This macro does not support mmsize == 8"
- %endif
- punpcklbw m%1, m%2
- punpcklbw m%3, m%4
- punpcklbw m%5, m%6
- punpcklbw m%7, m%8
- TRANSPOSE4x4W %1, %3, %5, %7, %2
- MOVHL m%2, m%1
- MOVHL m%4, m%3
- MOVHL m%6, m%5
- MOVHL m%8, m%7
- %endmacro
- ; PABSW macro assumes %1 != %2, while ABS1/2 macros work in-place
- %macro PABSW 2
- %if cpuflag(ssse3)
- pabsw %1, %2
- %elif cpuflag(mmxext)
- pxor %1, %1
- psubw %1, %2
- pmaxsw %1, %2
- %else
- pxor %1, %1
- pcmpgtw %1, %2
- pxor %2, %1
- psubw %2, %1
- SWAP %1, %2
- %endif
- %endmacro
- %macro PSIGNW 2
- %if cpuflag(ssse3)
- psignw %1, %2
- %else
- pxor %1, %2
- psubw %1, %2
- %endif
- %endmacro
- %macro ABS1 2
- %if cpuflag(ssse3)
- pabsw %1, %1
- %elif cpuflag(mmxext) ; a, tmp
- pxor %2, %2
- psubw %2, %1
- pmaxsw %1, %2
- %else ; a, tmp
- pxor %2, %2
- pcmpgtw %2, %1
- pxor %1, %2
- psubw %1, %2
- %endif
- %endmacro
- %macro ABS2 4
- %if cpuflag(ssse3)
- pabsw %1, %1
- pabsw %2, %2
- %elif cpuflag(mmxext) ; a, b, tmp0, tmp1
- pxor %3, %3
- pxor %4, %4
- psubw %3, %1
- psubw %4, %2
- pmaxsw %1, %3
- pmaxsw %2, %4
- %else ; 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
- %endif
- %endmacro
- %macro ABSB 2 ; source mmreg, temp mmreg (unused for SSSE3)
- %if cpuflag(ssse3)
- pabsb %1, %1
- %else
- pxor %2, %2
- psubb %2, %1
- pminub %1, %2
- %endif
- %endmacro
- %macro ABSB2 4 ; src1, src2, tmp1, tmp2 (tmp1/2 unused for SSSE3)
- %if cpuflag(ssse3)
- pabsb %1, %1
- pabsb %2, %2
- %else
- pxor %3, %3
- pxor %4, %4
- psubb %3, %1
- psubb %4, %2
- pminub %1, %3
- pminub %2, %4
- %endif
- %endmacro
- %macro ABSD2 4
- pxor %3, %3
- pxor %4, %4
- pcmpgtd %3, %1
- pcmpgtd %4, %2
- pxor %1, %3
- pxor %2, %4
- psubd %1, %3
- psubd %2, %4
- %endmacro
- %macro ABS4 6
- ABS2 %1, %2, %5, %6
- ABS2 %3, %4, %5, %6
- %endmacro
- %macro SPLATB_LOAD 3
- %if cpuflag(ssse3)
- movd %1, [%2-3]
- pshufb %1, %3
- %else
- movd %1, [%2-3] ;to avoid crossing a cacheline
- punpcklbw %1, %1
- SPLATW %1, %1, 3
- %endif
- %endmacro
- %macro SPLATB_REG 3
- %if cpuflag(ssse3)
- movd %1, %2d
- pshufb %1, %3
- %else
- movd %1, %2d
- punpcklbw %1, %1
- SPLATW %1, %1, 0
- %endif
- %endmacro
- %macro HADDD 2 ; sum junk
- %if sizeof%1 == 32
- %define %2 xmm%2
- vextracti128 %2, %1, 1
- %define %1 xmm%1
- paddd %1, %2
- %endif
- %if mmsize >= 16
- %if cpuflag(xop) && sizeof%1 == 16
- vphadddq %1, %1
- %endif
- movhlps %2, %1
- paddd %1, %2
- %endif
- %if notcpuflag(xop) || sizeof%1 != 16
- %if cpuflag(mmxext)
- PSHUFLW %2, %1, q0032
- %else ; mmx
- mova %2, %1
- psrlq %2, 32
- %endif
- paddd %1, %2
- %endif
- %undef %1
- %undef %2
- %endmacro
- %macro HADDW 2 ; reg, tmp
- %if cpuflag(xop) && sizeof%1 == 16
- vphaddwq %1, %1
- movhlps %2, %1
- paddd %1, %2
- %else
- pmaddwd %1, [pw_1]
- HADDD %1, %2
- %endif
- %endmacro
- %macro HADDPS 3 ; dst, src, tmp
- %if cpuflag(sse3)
- haddps %1, %1, %2
- %else
- movaps %3, %1
- shufps %1, %2, q2020
- shufps %3, %2, q3131
- addps %1, %3
- %endif
- %endmacro
- %macro PALIGNR 4-5
- %if cpuflag(ssse3)
- %if %0==5
- palignr %1, %2, %3, %4
- %else
- palignr %1, %2, %3
- %endif
- %else ; [dst,] src1, src2, imm, tmp
- %define %%dst %1
- %if %0==5
- %ifnidn %1, %2
- mova %%dst, %2
- %endif
- %rotate 1
- %endif
- %ifnidn %4, %2
- mova %4, %2
- %endif
- %if mmsize==8
- psllq %%dst, (8-%3)*8
- psrlq %4, %3*8
- %else
- pslldq %%dst, 16-%3
- psrldq %4, %3
- %endif
- por %%dst, %4
- %endif
- %endmacro
- %macro PAVGB 2-4
- %if cpuflag(mmxext)
- pavgb %1, %2
- %elif cpuflag(3dnow)
- pavgusb %1, %2
- %elif cpuflag(mmx)
- movu %3, %2
- por %3, %1
- pxor %1, %2
- pand %1, %4
- psrlq %1, 1
- psubb %3, %1
- SWAP %1, %3
- %endif
- %endmacro
- %macro PSHUFLW 1+
- %if mmsize == 8
- pshufw %1
- %else
- pshuflw %1
- %endif
- %endmacro
- %macro PSWAPD 2
- %if cpuflag(mmxext)
- pshufw %1, %2, q1032
- %elif cpuflag(3dnowext)
- pswapd %1, %2
- %elif cpuflag(3dnow)
- movq %1, %2
- psrlq %1, 32
- punpckldq %1, %2
- %endif
- %endmacro
- %macro DEINTB 5 ; mask, reg1, mask, reg2, optional src to fill masks from
- %ifnum %5
- pand m%3, m%5, m%4 ; src .. y6 .. y4
- pand m%1, m%5, m%2 ; dst .. y6 .. y4
- %else
- mova m%1, %5
- pand m%3, m%1, m%4 ; src .. y6 .. y4
- pand m%1, m%1, m%2 ; dst .. y6 .. y4
- %endif
- psrlw m%2, 8 ; dst .. y7 .. y5
- psrlw m%4, 8 ; src .. y7 .. y5
- %endmacro
- %macro SUMSUB_BA 3-4
- %if %0==3
- padd%1 m%2, m%3
- padd%1 m%3, m%3
- psub%1 m%3, m%2
- %else
- %if avx_enabled == 0
- mova m%4, m%2
- padd%1 m%2, m%3
- psub%1 m%3, m%4
- %else
- padd%1 m%4, m%2, m%3
- psub%1 m%3, m%2
- SWAP %2, %4
- %endif
- %endif
- %endmacro
- %macro SUMSUB_BADC 5-6
- %if %0==6
- SUMSUB_BA %1, %2, %3, %6
- SUMSUB_BA %1, %4, %5, %6
- %else
- padd%1 m%2, m%3
- padd%1 m%4, m%5
- padd%1 m%3, m%3
- padd%1 m%5, m%5
- psub%1 m%3, m%2
- psub%1 m%5, m%4
- %endif
- %endmacro
- %macro SUMSUB2_AB 4
- %ifnum %3
- psub%1 m%4, m%2, m%3
- psub%1 m%4, m%3
- padd%1 m%2, m%2
- padd%1 m%2, m%3
- %else
- mova m%4, m%2
- padd%1 m%2, m%2
- padd%1 m%2, %3
- psub%1 m%4, %3
- psub%1 m%4, %3
- %endif
- %endmacro
- %macro SUMSUB2_BA 4
- %if avx_enabled == 0
- mova m%4, m%2
- padd%1 m%2, m%3
- padd%1 m%2, m%3
- psub%1 m%3, m%4
- psub%1 m%3, m%4
- %else
- padd%1 m%4, m%2, m%3
- padd%1 m%4, m%3
- psub%1 m%3, m%2
- psub%1 m%3, m%2
- SWAP %2, %4
- %endif
- %endmacro
- %macro SUMSUBD2_AB 5
- %ifnum %4
- psra%1 m%5, m%2, 1 ; %3: %3>>1
- psra%1 m%4, m%3, 1 ; %2: %2>>1
- padd%1 m%4, m%2 ; %3: %3>>1+%2
- psub%1 m%5, m%3 ; %2: %2>>1-%3
- SWAP %2, %5
- SWAP %3, %4
- %else
- mova %5, m%2
- mova %4, m%3
- psra%1 m%3, 1 ; %3: %3>>1
- psra%1 m%2, 1 ; %2: %2>>1
- padd%1 m%3, %5 ; %3: %3>>1+%2
- psub%1 m%2, %4 ; %2: %2>>1-%3
- %endif
- %endmacro
- %macro DCT4_1D 5
- %ifnum %5
- SUMSUB_BADC w, %4, %1, %3, %2, %5
- SUMSUB_BA w, %3, %4, %5
- SUMSUB2_AB w, %1, %2, %5
- SWAP %1, %3, %4, %5, %2
- %else
- SUMSUB_BADC w, %4, %1, %3, %2
- SUMSUB_BA w, %3, %4
- mova [%5], m%2
- SUMSUB2_AB w, %1, [%5], %2
- SWAP %1, %3, %4, %2
- %endif
- %endmacro
- %macro IDCT4_1D 6-7
- %ifnum %6
- SUMSUBD2_AB %1, %3, %5, %7, %6
- ; %3: %3>>1-%5 %5: %3+%5>>1
- SUMSUB_BA %1, %4, %2, %7
- ; %4: %2+%4 %2: %2-%4
- SUMSUB_BADC %1, %5, %4, %3, %2, %7
- ; %5: %2+%4 + (%3+%5>>1)
- ; %4: %2+%4 - (%3+%5>>1)
- ; %3: %2-%4 + (%3>>1-%5)
- ; %2: %2-%4 - (%3>>1-%5)
- %else
- %ifidn %1, w
- SUMSUBD2_AB %1, %3, %5, [%6], [%6+16]
- %else
- SUMSUBD2_AB %1, %3, %5, [%6], [%6+32]
- %endif
- SUMSUB_BA %1, %4, %2
- SUMSUB_BADC %1, %5, %4, %3, %2
- %endif
- SWAP %2, %5, %4
- ; %2: %2+%4 + (%3+%5>>1) row0
- ; %3: %2-%4 + (%3>>1-%5) row1
- ; %4: %2-%4 - (%3>>1-%5) row2
- ; %5: %2+%4 - (%3+%5>>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]
- psraw %1, %6
- psraw %2, %6
- punpcklbw %3, %5
- punpcklbw %4, %5
- paddw %3, %1
- paddw %4, %2
- packuswb %3, %5
- packuswb %4, %5
- movh [%7], %3
- movh [%7+%8], %4
- %endmacro
- %macro PMINUB 3 ; dst, src, ignored
- %if cpuflag(mmxext)
- pminub %1, %2
- %else ; dst, src, tmp
- mova %3, %1
- psubusb %3, %2
- psubb %1, %3
- %endif
- %endmacro
- %macro SPLATW 2-3 0
- %if cpuflag(avx2) && %3 == 0
- vpbroadcastw %1, %2
- %elif mmsize == 16
- pshuflw %1, %2, (%3)*0x55
- punpcklqdq %1, %1
- %elif cpuflag(mmxext)
- pshufw %1, %2, (%3)*0x55
- %else
- %ifnidn %1, %2
- mova %1, %2
- %endif
- %if %3 & 2
- punpckhwd %1, %1
- %else
- punpcklwd %1, %1
- %endif
- %if %3 & 1
- punpckhwd %1, %1
- %else
- punpcklwd %1, %1
- %endif
- %endif
- %endmacro
- %macro SPLATD 1
- %if mmsize == 8
- punpckldq %1, %1
- %elif cpuflag(sse2)
- pshufd %1, %1, 0
- %elif cpuflag(sse)
- shufps %1, %1, 0
- %endif
- %endmacro
- %macro CLIPUB 3 ;(dst, min, max)
- pmaxub %1, %2
- pminub %1, %3
- %endmacro
- %macro CLIPW 3 ;(dst, min, max)
- pmaxsw %1, %2
- pminsw %1, %3
- %endmacro
- %macro PMINSD 3 ; dst, src, tmp/unused
- %if cpuflag(sse4)
- pminsd %1, %2
- %elif cpuflag(sse2)
- cvtdq2ps %1, %1
- minps %1, %2
- cvtps2dq %1, %1
- %else
- mova %3, %2
- pcmpgtd %3, %1
- pxor %1, %2
- pand %1, %3
- pxor %1, %2
- %endif
- %endmacro
- %macro PMAXSD 3 ; dst, src, tmp/unused
- %if cpuflag(sse4)
- pmaxsd %1, %2
- %else
- mova %3, %1
- pcmpgtd %3, %2
- pand %1, %3
- pandn %3, %2
- por %1, %3
- %endif
- %endmacro
- %macro CLIPD 3-4
- %if cpuflag(sse4); src/dst, min, max, unused
- pminsd %1, %3
- pmaxsd %1, %2
- %elif cpuflag(sse2) ; src/dst, min (float), max (float), unused
- cvtdq2ps %1, %1
- minps %1, %3
- maxps %1, %2
- cvtps2dq %1, %1
- %else ; src/dst, min, max, tmp
- PMINSD %1, %3, %4
- PMAXSD %1, %2, %4
- %endif
- %endmacro
- %macro VBROADCASTSS 2 ; dst xmm/ymm, src m32/xmm
- %if cpuflag(avx2)
- vbroadcastss %1, %2
- %elif cpuflag(avx)
- %ifnum sizeof%2 ; avx1 register
- shufps xmm%1, xmm%2, xmm%2, q0000
- %if sizeof%1 >= 32 ; mmsize>=32
- vinsertf128 %1, %1, xmm%1, 1
- %endif
- %else ; avx1 memory
- vbroadcastss %1, %2
- %endif
- %else
- %ifnum sizeof%2 ; sse register
- shufps %1, %2, %2, q0000
- %else ; sse memory
- movss %1, %2
- shufps %1, %1, 0
- %endif
- %endif
- %endmacro
- %macro VBROADCASTSD 2 ; dst xmm/ymm, src m64
- %if cpuflag(avx) && mmsize == 32
- vbroadcastsd %1, %2
- %elif cpuflag(sse3)
- movddup %1, %2
- %else ; sse2
- movsd %1, %2
- movlhps %1, %1
- %endif
- %endmacro
- %macro VPBROADCASTD 2 ; dst xmm/ymm, src m32/xmm
- %if cpuflag(avx2)
- vpbroadcastd %1, %2
- %elif cpuflag(avx) && sizeof%1 >= 32
- %error vpbroadcastd not possible with ymm on avx1. try vbroadcastss
- %else
- %ifnum sizeof%2 ; sse2 register
- pshufd %1, %2, q0000
- %else ; sse memory
- movd %1, %2
- pshufd %1, %1, 0
- %endif
- %endif
- %endmacro
- %macro VBROADCASTI128 2 ; dst xmm/ymm, src : 128bits val
- %if mmsize > 16
- vbroadcasti128 %1, %2
- %else
- mova %1, %2
- %endif
- %endmacro
- %macro SHUFFLE_MASK_W 8
- %rep 8
- %if %1>=0x80
- db %1, %1
- %else
- db %1*2
- db %1*2+1
- %endif
- %rotate 1
- %endrep
- %endmacro
- %macro PMOVSXWD 2; dst, src
- %if cpuflag(sse4)
- pmovsxwd %1, %2
- %else
- %ifnidn %1, %2
- mova %1, %2
- %endif
- punpcklwd %1, %1
- psrad %1, 16
- %endif
- %endmacro
- ; Wrapper for non-FMA version of fmaddps
- %macro FMULADD_PS 5
- %if cpuflag(fma3) || cpuflag(fma4)
- fmaddps %1, %2, %3, %4
- %elifidn %1, %4
- mulps %5, %2, %3
- addps %1, %4, %5
- %else
- mulps %1, %2, %3
- addps %1, %4
- %endif
- %endmacro
- %macro LSHIFT 2
- %if mmsize > 8
- pslldq %1, %2
- %else
- psllq %1, 8*(%2)
- %endif
- %endmacro
- %macro RSHIFT 2
- %if mmsize > 8
- psrldq %1, %2
- %else
- psrlq %1, 8*(%2)
- %endif
- %endmacro
- %macro MOVHL 2 ; dst, src
- %ifidn %1, %2
- punpckhqdq %1, %2
- %elif cpuflag(avx)
- punpckhqdq %1, %2, %2
- %elif cpuflag(sse4)
- pshufd %1, %2, q3232 ; pshufd is slow on some older CPUs, so only use it on more modern ones
- %else
- movhlps %1, %2 ; may cause an int/float domain transition and has a dependency on dst
- %endif
- %endmacro
- ; Horizontal Sum of Packed Single precision floats
- ; The resulting sum is in all elements.
- %macro HSUMPS 2 ; dst/src, tmp
- %if cpuflag(avx)
- %if sizeof%1>=32 ; avx
- vperm2f128 %2, %1, %1, (0)*16+(1)
- addps %1, %2
- %endif
- shufps %2, %1, %1, q1032
- addps %1, %2
- shufps %2, %1, %1, q0321
- addps %1, %2
- %else ; this form is a bit faster than the short avx-like emulation.
- movaps %2, %1
- shufps %1, %1, q1032
- addps %1, %2
- movaps %2, %1
- shufps %1, %1, q0321
- addps %1, %2
- ; all %1 members should be equal for as long as float a+b==b+a
- %endif
- %endmacro
- ; Emulate blendvps if not available
- ;
- ; src_b is destroyed when using emulation with logical operands
- ; SSE41 blendv instruction is hard coded to use xmm0 as mask
- %macro BLENDVPS 3 ; dst/src_a, src_b, mask
- %if cpuflag(avx)
- blendvps %1, %1, %2, %3
- %elif cpuflag(sse4)
- %ifnidn %3,xmm0
- %error sse41 blendvps uses xmm0 as default 3d operand, you used %3
- %endif
- blendvps %1, %2, %3
- %else
- xorps %2, %1
- andps %2, %3
- xorps %1, %2
- %endif
- %endmacro
- ; Emulate pblendvb if not available
- ;
- ; src_b is destroyed when using emulation with logical operands
- ; SSE41 blendv instruction is hard coded to use xmm0 as mask
- %macro PBLENDVB 3 ; dst/src_a, src_b, mask
- %if cpuflag(avx)
- %if cpuflag(avx) && notcpuflag(avx2) && sizeof%1 >= 32
- %error pblendb not possible with ymm on avx1, try blendvps.
- %endif
- pblendvb %1, %1, %2, %3
- %elif cpuflag(sse4)
- %ifnidn %3,xmm0
- %error sse41 pblendvd uses xmm0 as default 3d operand, you used %3
- %endif
- pblendvb %1, %2, %3
- %else
- pxor %2, %1
- pand %2, %3
- pxor %1, %2
- %endif
- %endmacro
|