8394990b27
Add 31bit pairs before unpacking in x86 block error code BUG=webm:1210 Change-Id: I5ca8c7f7775585a17fe09d6bbfc25e1f2955eb0a
116 lines
3.2 KiB
NASM
116 lines
3.2 KiB
NASM
;
|
|
; Copyright (c) 2010 The WebM project authors. All Rights Reserved.
|
|
;
|
|
; Use of this source code is governed by a BSD-style license
|
|
; that can be found in the LICENSE file in the root of the source
|
|
; tree. An additional intellectual property rights grant can be found
|
|
; in the file PATENTS. All contributing project authors may
|
|
; be found in the AUTHORS file in the root of the source tree.
|
|
;
|
|
|
|
%define private_prefix vp9
|
|
|
|
%include "third_party/x86inc/x86inc.asm"
|
|
%include "vpx_dsp/x86/bitdepth_conversion_sse2.asm"
|
|
|
|
SECTION .text
|
|
|
|
; int64_t vp9_block_error(int16_t *coeff, int16_t *dqcoeff, intptr_t block_size,
|
|
; int64_t *ssz)
|
|
|
|
INIT_XMM sse2
|
|
cglobal block_error, 3, 3, 8, uqc, dqc, size, ssz
|
|
pxor m4, m4 ; sse accumulator
|
|
pxor m6, m6 ; ssz accumulator
|
|
pxor m5, m5 ; dedicated zero register
|
|
.loop:
|
|
LOAD_TRAN_LOW 2, uqcq, 0
|
|
LOAD_TRAN_LOW 0, dqcq, 0
|
|
LOAD_TRAN_LOW 3, uqcq, 8
|
|
LOAD_TRAN_LOW 1, dqcq, 8
|
|
INCREMENT_ELEMENTS_TRAN_LOW uqcq, 16
|
|
INCREMENT_ELEMENTS_TRAN_LOW dqcq, 16
|
|
sub sizeq, 16
|
|
psubw m0, m2
|
|
psubw m1, m3
|
|
; individual errors are max. 15bit+sign, so squares are 30bit, and
|
|
; thus the sum of 2 should fit in a 31bit integer (+ unused sign bit)
|
|
pmaddwd m0, m0
|
|
pmaddwd m1, m1
|
|
pmaddwd m2, m2
|
|
pmaddwd m3, m3
|
|
; the sum of 2 31bit integers will fit in a 32bit unsigned integer
|
|
paddd m0, m1
|
|
paddd m2, m3
|
|
; accumulate in 64bit
|
|
punpckldq m7, m0, m5
|
|
punpckhdq m0, m5
|
|
paddq m4, m7
|
|
punpckldq m7, m2, m5
|
|
paddq m4, m0
|
|
punpckhdq m2, m5
|
|
paddq m6, m7
|
|
paddq m6, m2
|
|
jg .loop
|
|
|
|
; accumulate horizontally and store in return value
|
|
movhlps m5, m4
|
|
movhlps m7, m6
|
|
paddq m4, m5
|
|
paddq m6, m7
|
|
%if ARCH_X86_64
|
|
movq rax, m4
|
|
movq [sszq], m6
|
|
%else
|
|
mov eax, sszm
|
|
pshufd m5, m4, 0x1
|
|
movq [eax], m6
|
|
movd eax, m4
|
|
movd edx, m5
|
|
%endif
|
|
RET
|
|
|
|
; Compute the sum of squared difference between two tran_low_t vectors.
|
|
; Vectors are converted (if necessary) to int16_t for calculations.
|
|
; int64_t vp9_block_error_fp(tran_low_t *coeff, tran_low_t *dqcoeff,
|
|
; intptr_t block_size)
|
|
|
|
INIT_XMM sse2
|
|
cglobal block_error_fp, 3, 3, 6, uqc, dqc, size
|
|
pxor m4, m4 ; sse accumulator
|
|
pxor m5, m5 ; dedicated zero register
|
|
.loop:
|
|
LOAD_TRAN_LOW 2, uqcq, 0
|
|
LOAD_TRAN_LOW 0, dqcq, 0
|
|
LOAD_TRAN_LOW 3, uqcq, 8
|
|
LOAD_TRAN_LOW 1, dqcq, 8
|
|
INCREMENT_ELEMENTS_TRAN_LOW uqcq, 16
|
|
INCREMENT_ELEMENTS_TRAN_LOW dqcq, 16
|
|
sub sizeq, 16
|
|
psubw m0, m2
|
|
psubw m1, m3
|
|
; individual errors are max. 15bit+sign, so squares are 30bit, and
|
|
; thus the sum of 2 should fit in a 31bit integer (+ unused sign bit)
|
|
pmaddwd m0, m0
|
|
pmaddwd m1, m1
|
|
; the sum of 2 31bit integers will fit in a 32bit unsigned integer
|
|
paddd m0, m1
|
|
; accumulate in 64bit
|
|
punpckldq m3, m0, m5
|
|
punpckhdq m0, m5
|
|
paddq m4, m3
|
|
paddq m4, m0
|
|
jnz .loop
|
|
|
|
; accumulate horizontally and store in return value
|
|
movhlps m5, m4
|
|
paddq m4, m5
|
|
%if ARCH_X86_64
|
|
movq rax, m4
|
|
%else
|
|
pshufd m5, m4, 0x1
|
|
movd eax, m4
|
|
movd edx, m5
|
|
%endif
|
|
RET
|