15a50414796e9a458925c7a13a15055d02406bf43Vikas Arora// Copyright 2011 Google Inc. All Rights Reserved. 2466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// 3466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// This code is licensed under the same terms as WebM: 4466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Software License Agreement: http://www.webmproject.org/license/software/ 5466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Additional IP Rights Grant: http://www.webmproject.org/license/additional/ 6466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// ----------------------------------------------------------------------------- 7466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// 85a50414796e9a458925c7a13a15055d02406bf43Vikas Arora// SSE2 version of speed-critical encoding functions. 9466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// 10466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Author: Christian Duvivier (cduvivier@google.com) 11466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 125a50414796e9a458925c7a13a15055d02406bf43Vikas Arora#include "./dsp.h" 135a50414796e9a458925c7a13a15055d02406bf43Vikas Arora 145a50414796e9a458925c7a13a15055d02406bf43Vikas Arora#if defined(WEBP_USE_SSE2) 155a50414796e9a458925c7a13a15055d02406bf43Vikas Arora#include <stdlib.h> // for abs() 16466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#include <emmintrin.h> 17466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 185a50414796e9a458925c7a13a15055d02406bf43Vikas Arora#include "../enc/vp8enci.h" 19466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 20466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#if defined(__cplusplus) || defined(c_plusplus) 21466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Aroraextern "C" { 22466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#endif 23466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 245a50414796e9a458925c7a13a15055d02406bf43Vikas Arora//------------------------------------------------------------------------------ 25466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Compute susceptibility based on DCT-coeff histograms: 26466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// the higher, the "easier" the macroblock is to compress. 27466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 28466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic int CollectHistogramSSE2(const uint8_t* ref, const uint8_t* pred, 29466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int start_block, int end_block) { 30466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int histo[MAX_COEFF_THRESH + 1] = { 0 }; 31466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int16_t out[16]; 32466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int j, k; 33466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i max_coeff_thresh = _mm_set1_epi16(MAX_COEFF_THRESH); 34466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora for (j = start_block; j < end_block; ++j) { 355a50414796e9a458925c7a13a15055d02406bf43Vikas Arora VP8FTransform(ref + VP8DspScan[j], pred + VP8DspScan[j], out); 36466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 37466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Convert coefficients to bin (within out[]). 38466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 39466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load. 40466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i out0 = _mm_loadu_si128((__m128i*)&out[0]); 41466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i out1 = _mm_loadu_si128((__m128i*)&out[8]); 42466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // sign(out) = out >> 15 (0x0000 if positive, 0xffff if negative) 43466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i sign0 = _mm_srai_epi16(out0, 15); 44466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i sign1 = _mm_srai_epi16(out1, 15); 45466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // abs(out) = (out ^ sign) - sign 46466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i xor0 = _mm_xor_si128(out0, sign0); 47466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i xor1 = _mm_xor_si128(out1, sign1); 48466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i abs0 = _mm_sub_epi16(xor0, sign0); 49466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i abs1 = _mm_sub_epi16(xor1, sign1); 50466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // v = abs(out) >> 2 51466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i v0 = _mm_srai_epi16(abs0, 2); 52466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i v1 = _mm_srai_epi16(abs1, 2); 53466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // bin = min(v, MAX_COEFF_THRESH) 54466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i bin0 = _mm_min_epi16(v0, max_coeff_thresh); 55466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i bin1 = _mm_min_epi16(v1, max_coeff_thresh); 56466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Store. 57466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&out[0], bin0); 58466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&out[8], bin1); 59466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 60466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 61466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Use bin to update histogram. 62466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora for (k = 0; k < 16; ++k) { 63466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora histo[out[k]]++; 64466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 65466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 66466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 67466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora return VP8GetAlpha(histo); 68466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 69466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 705a50414796e9a458925c7a13a15055d02406bf43Vikas Arora//------------------------------------------------------------------------------ 71466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Transforms (Paragraph 14.4) 72466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 73466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Does one or two inverse transforms. 74466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void ITransformSSE2(const uint8_t* ref, const int16_t* in, uint8_t* dst, 75466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int do_two) { 76466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // This implementation makes use of 16-bit fixed point versions of two 77466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // multiply constants: 78466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // K1 = sqrt(2) * cos (pi/8) ~= 85627 / 2^16 79466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // K2 = sqrt(2) * sin (pi/8) ~= 35468 / 2^16 80466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 81466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // To be able to use signed 16-bit integers, we use the following trick to 82466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // have constants within range: 83466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // - Associated constants are obtained by subtracting the 16-bit fixed point 84466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // version of one: 85466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // k = K - (1 << 16) => K = k + (1 << 16) 86466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // K1 = 85267 => k1 = 20091 87466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // K2 = 35468 => k2 = -30068 88466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // - The multiplication of a variable by a constant become the sum of the 89466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // variable and the multiplication of that variable by the associated 90466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // constant: 91466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // (x * K) >> 16 = (x * (k + (1 << 16))) >> 16 = ((x * k ) >> 16) + x 92466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i k1 = _mm_set1_epi16(20091); 93466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i k2 = _mm_set1_epi16(-30068); 94466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i T0, T1, T2, T3; 95466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 96466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load and concatenate the transform coefficients (we'll do two inverse 97466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // transforms in parallel). In the case of only one inverse transform, the 98466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // second half of the vectors will just contain random value we'll never 99466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // use nor store. 100466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i in0, in1, in2, in3; 101466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 102466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in0 = _mm_loadl_epi64((__m128i*)&in[0]); 103466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in1 = _mm_loadl_epi64((__m128i*)&in[4]); 104466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in2 = _mm_loadl_epi64((__m128i*)&in[8]); 105466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in3 = _mm_loadl_epi64((__m128i*)&in[12]); 106466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 x x x x 107466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a01 a11 a21 a31 x x x x 108466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 x x x x 109466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a03 a13 a23 a33 x x x x 110466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora if (do_two) { 111466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inB0 = _mm_loadl_epi64((__m128i*)&in[16]); 112466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inB1 = _mm_loadl_epi64((__m128i*)&in[20]); 113466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inB2 = _mm_loadl_epi64((__m128i*)&in[24]); 114466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inB3 = _mm_loadl_epi64((__m128i*)&in[28]); 115466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in0 = _mm_unpacklo_epi64(in0, inB0); 116466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in1 = _mm_unpacklo_epi64(in1, inB1); 117466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in2 = _mm_unpacklo_epi64(in2, inB2); 118466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in3 = _mm_unpacklo_epi64(in3, inB3); 119466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 b00 b10 b20 b30 120466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a01 a11 a21 a31 b01 b11 b21 b31 121466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 b02 b12 b22 b32 122466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a03 a13 a23 a33 b03 b13 b23 b33 123466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 124466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 125466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 126466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Vertical pass and subsequent transpose. 127466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 128466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // First pass, c and d calculations are longer because of the "trick" 129466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // multiplications. 130466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a = _mm_add_epi16(in0, in2); 131466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b = _mm_sub_epi16(in0, in2); 132466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // c = MUL(in1, K2) - MUL(in3, K1) = MUL(in1, k2) - MUL(in3, k1) + in1 - in3 133466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c1 = _mm_mulhi_epi16(in1, k2); 134466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c2 = _mm_mulhi_epi16(in3, k1); 135466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c3 = _mm_sub_epi16(in1, in3); 136466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c4 = _mm_sub_epi16(c1, c2); 137466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c = _mm_add_epi16(c3, c4); 138466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // d = MUL(in1, K1) + MUL(in3, K2) = MUL(in1, k1) + MUL(in3, k2) + in1 + in3 139466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d1 = _mm_mulhi_epi16(in1, k1); 140466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d2 = _mm_mulhi_epi16(in3, k2); 141466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d3 = _mm_add_epi16(in1, in3); 142466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d4 = _mm_add_epi16(d1, d2); 143466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d = _mm_add_epi16(d3, d4); 144466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 145466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Second pass. 146466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp0 = _mm_add_epi16(a, d); 147466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp1 = _mm_add_epi16(b, c); 148466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp2 = _mm_sub_epi16(b, c); 149466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp3 = _mm_sub_epi16(a, d); 150466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 151466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Transpose the two 4x4. 152466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a01 a02 a03 b00 b01 b02 b03 153466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a10 a11 a12 a13 b10 b11 b12 b13 154466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a20 a21 a22 a23 b20 b21 b22 b23 155466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a30 a31 a32 a33 b30 b31 b32 b33 156466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_0 = _mm_unpacklo_epi16(tmp0, tmp1); 157466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_1 = _mm_unpacklo_epi16(tmp2, tmp3); 158466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_2 = _mm_unpackhi_epi16(tmp0, tmp1); 159466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_3 = _mm_unpackhi_epi16(tmp2, tmp3); 160466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a01 a11 a02 a12 a03 a13 161466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a20 a30 a21 a31 a22 a32 a23 a33 162466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b00 b10 b01 b11 b02 b12 b03 b13 163466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b20 b30 b21 b31 b22 b32 b23 b33 164466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_0 = _mm_unpacklo_epi32(transpose0_0, transpose0_1); 165466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_1 = _mm_unpacklo_epi32(transpose0_2, transpose0_3); 166466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_2 = _mm_unpackhi_epi32(transpose0_0, transpose0_1); 167466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_3 = _mm_unpackhi_epi32(transpose0_2, transpose0_3); 168466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 a01 a11 a21 a31 169466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b00 b10 b20 b30 b01 b11 b21 b31 170466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 a03 a13 a23 a33 171466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b02 b12 a22 b32 b03 b13 b23 b33 172466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T0 = _mm_unpacklo_epi64(transpose1_0, transpose1_1); 173466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T1 = _mm_unpackhi_epi64(transpose1_0, transpose1_1); 174466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T2 = _mm_unpacklo_epi64(transpose1_2, transpose1_3); 175466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T3 = _mm_unpackhi_epi64(transpose1_2, transpose1_3); 176466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 b00 b10 b20 b30 177466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a01 a11 a21 a31 b01 b11 b21 b31 178466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 b02 b12 b22 b32 179466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a03 a13 a23 a33 b03 b13 b23 b33 180466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 181466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 182466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Horizontal pass and subsequent transpose. 183466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 184466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // First pass, c and d calculations are longer because of the "trick" 185466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // multiplications. 186466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i four = _mm_set1_epi16(4); 187466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i dc = _mm_add_epi16(T0, four); 188466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a = _mm_add_epi16(dc, T2); 189466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b = _mm_sub_epi16(dc, T2); 190466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // c = MUL(T1, K2) - MUL(T3, K1) = MUL(T1, k2) - MUL(T3, k1) + T1 - T3 191466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c1 = _mm_mulhi_epi16(T1, k2); 192466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c2 = _mm_mulhi_epi16(T3, k1); 193466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c3 = _mm_sub_epi16(T1, T3); 194466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c4 = _mm_sub_epi16(c1, c2); 195466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c = _mm_add_epi16(c3, c4); 196466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // d = MUL(T1, K1) + MUL(T3, K2) = MUL(T1, k1) + MUL(T3, k2) + T1 + T3 197466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d1 = _mm_mulhi_epi16(T1, k1); 198466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d2 = _mm_mulhi_epi16(T3, k2); 199466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d3 = _mm_add_epi16(T1, T3); 200466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d4 = _mm_add_epi16(d1, d2); 201466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d = _mm_add_epi16(d3, d4); 202466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 203466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Second pass. 204466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp0 = _mm_add_epi16(a, d); 205466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp1 = _mm_add_epi16(b, c); 206466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp2 = _mm_sub_epi16(b, c); 207466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp3 = _mm_sub_epi16(a, d); 208466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i shifted0 = _mm_srai_epi16(tmp0, 3); 209466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i shifted1 = _mm_srai_epi16(tmp1, 3); 210466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i shifted2 = _mm_srai_epi16(tmp2, 3); 211466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i shifted3 = _mm_srai_epi16(tmp3, 3); 212466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 213466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Transpose the two 4x4. 214466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a01 a02 a03 b00 b01 b02 b03 215466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a10 a11 a12 a13 b10 b11 b12 b13 216466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a20 a21 a22 a23 b20 b21 b22 b23 217466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a30 a31 a32 a33 b30 b31 b32 b33 218466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_0 = _mm_unpacklo_epi16(shifted0, shifted1); 219466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_1 = _mm_unpacklo_epi16(shifted2, shifted3); 220466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_2 = _mm_unpackhi_epi16(shifted0, shifted1); 221466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_3 = _mm_unpackhi_epi16(shifted2, shifted3); 222466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a01 a11 a02 a12 a03 a13 223466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a20 a30 a21 a31 a22 a32 a23 a33 224466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b00 b10 b01 b11 b02 b12 b03 b13 225466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b20 b30 b21 b31 b22 b32 b23 b33 226466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_0 = _mm_unpacklo_epi32(transpose0_0, transpose0_1); 227466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_1 = _mm_unpacklo_epi32(transpose0_2, transpose0_3); 228466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_2 = _mm_unpackhi_epi32(transpose0_0, transpose0_1); 229466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_3 = _mm_unpackhi_epi32(transpose0_2, transpose0_3); 230466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 a01 a11 a21 a31 231466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b00 b10 b20 b30 b01 b11 b21 b31 232466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 a03 a13 a23 a33 233466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b02 b12 a22 b32 b03 b13 b23 b33 234466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T0 = _mm_unpacklo_epi64(transpose1_0, transpose1_1); 235466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T1 = _mm_unpackhi_epi64(transpose1_0, transpose1_1); 236466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T2 = _mm_unpacklo_epi64(transpose1_2, transpose1_3); 237466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T3 = _mm_unpackhi_epi64(transpose1_2, transpose1_3); 238466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 b00 b10 b20 b30 239466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a01 a11 a21 a31 b01 b11 b21 b31 240466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 b02 b12 b22 b32 241466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a03 a13 a23 a33 b03 b13 b23 b33 242466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 243466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 244466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Add inverse transform to 'ref' and store. 245466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 246466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i zero = _mm_set1_epi16(0); 247466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load the reference(s). 248466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i ref0, ref1, ref2, ref3; 249466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora if (do_two) { 250466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load eight bytes/pixels per line. 251466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref0 = _mm_loadl_epi64((__m128i*)&ref[0 * BPS]); 252466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref1 = _mm_loadl_epi64((__m128i*)&ref[1 * BPS]); 253466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref2 = _mm_loadl_epi64((__m128i*)&ref[2 * BPS]); 254466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref3 = _mm_loadl_epi64((__m128i*)&ref[3 * BPS]); 255466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } else { 256466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load four bytes/pixels per line. 257466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref0 = _mm_cvtsi32_si128(*(int*)&ref[0 * BPS]); 258466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref1 = _mm_cvtsi32_si128(*(int*)&ref[1 * BPS]); 259466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref2 = _mm_cvtsi32_si128(*(int*)&ref[2 * BPS]); 260466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref3 = _mm_cvtsi32_si128(*(int*)&ref[3 * BPS]); 261466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 262466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Convert to 16b. 263466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref0 = _mm_unpacklo_epi8(ref0, zero); 264466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref1 = _mm_unpacklo_epi8(ref1, zero); 265466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref2 = _mm_unpacklo_epi8(ref2, zero); 266466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref3 = _mm_unpacklo_epi8(ref3, zero); 267466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Add the inverse transform(s). 268466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref0 = _mm_add_epi16(ref0, T0); 269466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref1 = _mm_add_epi16(ref1, T1); 270466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref2 = _mm_add_epi16(ref2, T2); 271466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref3 = _mm_add_epi16(ref3, T3); 272466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Unsigned saturate to 8b. 273466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref0 = _mm_packus_epi16(ref0, ref0); 274466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref1 = _mm_packus_epi16(ref1, ref1); 275466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref2 = _mm_packus_epi16(ref2, ref2); 276466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora ref3 = _mm_packus_epi16(ref3, ref3); 277466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Store the results. 278466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora if (do_two) { 279466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Store eight bytes/pixels per line. 280466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storel_epi64((__m128i*)&dst[0 * BPS], ref0); 281466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storel_epi64((__m128i*)&dst[1 * BPS], ref1); 282466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storel_epi64((__m128i*)&dst[2 * BPS], ref2); 283466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storel_epi64((__m128i*)&dst[3 * BPS], ref3); 284466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } else { 285466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Store four bytes/pixels per line. 286466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *((int32_t *)&dst[0 * BPS]) = _mm_cvtsi128_si32(ref0); 287466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *((int32_t *)&dst[1 * BPS]) = _mm_cvtsi128_si32(ref1); 288466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *((int32_t *)&dst[2 * BPS]) = _mm_cvtsi128_si32(ref2); 289466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *((int32_t *)&dst[3 * BPS]) = _mm_cvtsi128_si32(ref3); 290466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 291466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 292466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 293466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 294466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void FTransformSSE2(const uint8_t* src, const uint8_t* ref, 295466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int16_t* out) { 296466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i zero = _mm_setzero_si128(); 297466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i seven = _mm_set1_epi16(7); 298466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i k7500 = _mm_set1_epi32(7500); 299466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i k14500 = _mm_set1_epi32(14500); 300466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i k51000 = _mm_set1_epi32(51000); 301466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i k12000_plus_one = _mm_set1_epi32(12000 + (1 << 16)); 302466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i k5352_2217 = _mm_set_epi16(5352, 2217, 5352, 2217, 303466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 5352, 2217, 5352, 2217); 304466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i k2217_5352 = _mm_set_epi16(2217, -5352, 2217, -5352, 305466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 2217, -5352, 2217, -5352); 306466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 307466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i v01, v32; 308466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 309466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Difference between src and ref and initial transpose. 310466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 311466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load src and convert to 16b. 312466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i src0 = _mm_loadl_epi64((__m128i*)&src[0 * BPS]); 313466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i src1 = _mm_loadl_epi64((__m128i*)&src[1 * BPS]); 314466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i src2 = _mm_loadl_epi64((__m128i*)&src[2 * BPS]); 315466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i src3 = _mm_loadl_epi64((__m128i*)&src[3 * BPS]); 316466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i src_0 = _mm_unpacklo_epi8(src0, zero); 317466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i src_1 = _mm_unpacklo_epi8(src1, zero); 318466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i src_2 = _mm_unpacklo_epi8(src2, zero); 319466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i src_3 = _mm_unpacklo_epi8(src3, zero); 320466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load ref and convert to 16b. 321466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i ref0 = _mm_loadl_epi64((__m128i*)&ref[0 * BPS]); 322466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i ref1 = _mm_loadl_epi64((__m128i*)&ref[1 * BPS]); 323466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i ref2 = _mm_loadl_epi64((__m128i*)&ref[2 * BPS]); 324466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i ref3 = _mm_loadl_epi64((__m128i*)&ref[3 * BPS]); 325466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i ref_0 = _mm_unpacklo_epi8(ref0, zero); 326466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i ref_1 = _mm_unpacklo_epi8(ref1, zero); 327466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i ref_2 = _mm_unpacklo_epi8(ref2, zero); 328466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i ref_3 = _mm_unpacklo_epi8(ref3, zero); 329466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Compute difference. 330466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i diff0 = _mm_sub_epi16(src_0, ref_0); 331466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i diff1 = _mm_sub_epi16(src_1, ref_1); 332466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i diff2 = _mm_sub_epi16(src_2, ref_2); 333466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i diff3 = _mm_sub_epi16(src_3, ref_3); 334466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 335466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Transpose. 336466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 00 01 02 03 0 0 0 0 337466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 10 11 12 13 0 0 0 0 338466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 20 21 22 23 0 0 0 0 339466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 30 31 32 33 0 0 0 0 340466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_0 = _mm_unpacklo_epi16(diff0, diff1); 341466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_1 = _mm_unpacklo_epi16(diff2, diff3); 342466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 00 10 01 11 02 12 03 13 343466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 20 30 21 31 22 32 23 33 344466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i v23 = _mm_unpackhi_epi32(transpose0_0, transpose0_1); 345466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora v01 = _mm_unpacklo_epi32(transpose0_0, transpose0_1); 346466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora v32 = _mm_shuffle_epi32(v23, _MM_SHUFFLE(1, 0, 3, 2)); 347466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 a03 a13 a23 a33 348466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 a01 a11 a21 a31 349466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a03 a13 a23 a33 a02 a12 a22 a32 350466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 351466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 352466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // First pass and subsequent transpose. 353466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 354466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Same operations are done on the (0,3) and (1,2) pairs. 355466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b0 = (a0 + a3) << 3 356466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b1 = (a1 + a2) << 3 357466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b3 = (a0 - a3) << 3 358466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b2 = (a1 - a2) << 3 359466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a01 = _mm_add_epi16(v01, v32); 360466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a32 = _mm_sub_epi16(v01, v32); 361466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b01 = _mm_slli_epi16(a01, 3); 362466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b32 = _mm_slli_epi16(a32, 3); 363466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b11 = _mm_unpackhi_epi64(b01, b01); 364466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b22 = _mm_unpackhi_epi64(b32, b32); 365466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 366466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // e0 = b0 + b1 367466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // e2 = b0 - b1 368466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i e0 = _mm_add_epi16(b01, b11); 369466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i e2 = _mm_sub_epi16(b01, b11); 370466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i e02 = _mm_unpacklo_epi64(e0, e2); 371466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 372466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // e1 = (b3 * 5352 + b2 * 2217 + 14500) >> 12 373466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // e3 = (b3 * 2217 - b2 * 5352 + 7500) >> 12 374466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b23 = _mm_unpacklo_epi16(b22, b32); 375466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c1 = _mm_madd_epi16(b23, k5352_2217); 376466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c3 = _mm_madd_epi16(b23, k2217_5352); 377466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d1 = _mm_add_epi32(c1, k14500); 378466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d3 = _mm_add_epi32(c3, k7500); 379466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i e1 = _mm_srai_epi32(d1, 12); 380466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i e3 = _mm_srai_epi32(d3, 12); 381466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i e13 = _mm_packs_epi32(e1, e3); 382466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 383466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Transpose. 384466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 00 01 02 03 20 21 22 23 385466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 10 11 12 13 30 31 32 33 386466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_0 = _mm_unpacklo_epi16(e02, e13); 387466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_1 = _mm_unpackhi_epi16(e02, e13); 388466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 00 10 01 11 02 12 03 13 389466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 20 30 21 31 22 32 23 33 390466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i v23 = _mm_unpackhi_epi32(transpose0_0, transpose0_1); 391466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora v01 = _mm_unpacklo_epi32(transpose0_0, transpose0_1); 392466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora v32 = _mm_shuffle_epi32(v23, _MM_SHUFFLE(1, 0, 3, 2)); 393466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 02 12 22 32 03 13 23 33 394466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 00 10 20 30 01 11 21 31 395466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 03 13 23 33 02 12 22 32 396466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 397466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 398466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Second pass 399466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 400466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Same operations are done on the (0,3) and (1,2) pairs. 401466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a0 = v0 + v3 402466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a1 = v1 + v2 403466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a3 = v0 - v3 404466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a2 = v1 - v2 405466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a01 = _mm_add_epi16(v01, v32); 406466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a32 = _mm_sub_epi16(v01, v32); 407466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a11 = _mm_unpackhi_epi64(a01, a01); 408466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a22 = _mm_unpackhi_epi64(a32, a32); 409466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 410466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // d0 = (a0 + a1 + 7) >> 4; 411466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // d2 = (a0 - a1 + 7) >> 4; 412466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b0 = _mm_add_epi16(a01, a11); 413466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b2 = _mm_sub_epi16(a01, a11); 414466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c0 = _mm_add_epi16(b0, seven); 415466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c2 = _mm_add_epi16(b2, seven); 416466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d0 = _mm_srai_epi16(c0, 4); 417466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d2 = _mm_srai_epi16(c2, 4); 418466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 419466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // f1 = ((b3 * 5352 + b2 * 2217 + 12000) >> 16) 420466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // f3 = ((b3 * 2217 - b2 * 5352 + 51000) >> 16) 421466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b23 = _mm_unpacklo_epi16(a22, a32); 422466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c1 = _mm_madd_epi16(b23, k5352_2217); 423466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c3 = _mm_madd_epi16(b23, k2217_5352); 424466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d1 = _mm_add_epi32(c1, k12000_plus_one); 425466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d3 = _mm_add_epi32(c3, k51000); 426466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i e1 = _mm_srai_epi32(d1, 16); 427466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i e3 = _mm_srai_epi32(d3, 16); 428466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i f1 = _mm_packs_epi32(e1, e1); 429466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i f3 = _mm_packs_epi32(e3, e3); 430466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // f1 = f1 + (a3 != 0); 431466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // The compare will return (0xffff, 0) for (==0, !=0). To turn that into the 432466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // desired (0, 1), we add one earlier through k12000_plus_one. 433466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i g1 = _mm_add_epi16(f1, _mm_cmpeq_epi16(a32, zero)); 434466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 435466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storel_epi64((__m128i*)&out[ 0], d0); 436466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storel_epi64((__m128i*)&out[ 4], g1); 437466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storel_epi64((__m128i*)&out[ 8], d2); 438466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storel_epi64((__m128i*)&out[12], f3); 439466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 440466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 441466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 4425a50414796e9a458925c7a13a15055d02406bf43Vikas Arora//------------------------------------------------------------------------------ 443466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Metric 444466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 445466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic int SSE4x4SSE2(const uint8_t* a, const uint8_t* b) { 446466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i zero = _mm_set1_epi16(0); 447466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 448466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load values. 449466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a0 = _mm_loadl_epi64((__m128i*)&a[BPS * 0]); 450466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a1 = _mm_loadl_epi64((__m128i*)&a[BPS * 1]); 451466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a2 = _mm_loadl_epi64((__m128i*)&a[BPS * 2]); 452466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a3 = _mm_loadl_epi64((__m128i*)&a[BPS * 3]); 453466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b0 = _mm_loadl_epi64((__m128i*)&b[BPS * 0]); 454466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b1 = _mm_loadl_epi64((__m128i*)&b[BPS * 1]); 455466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b2 = _mm_loadl_epi64((__m128i*)&b[BPS * 2]); 456466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b3 = _mm_loadl_epi64((__m128i*)&b[BPS * 3]); 457466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 458466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Combine pair of lines and convert to 16b. 459466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a01 = _mm_unpacklo_epi32(a0, a1); 460466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a23 = _mm_unpacklo_epi32(a2, a3); 461466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b01 = _mm_unpacklo_epi32(b0, b1); 462466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b23 = _mm_unpacklo_epi32(b2, b3); 463466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a01s = _mm_unpacklo_epi8(a01, zero); 464466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a23s = _mm_unpacklo_epi8(a23, zero); 465466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b01s = _mm_unpacklo_epi8(b01, zero); 466466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b23s = _mm_unpacklo_epi8(b23, zero); 467466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 468466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Compute differences; (a-b)^2 = (abs(a-b))^2 = (sat8(a-b) + sat8(b-a))^2 469466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // TODO(cduvivier): Dissassemble and figure out why this is fastest. We don't 470466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // need absolute values, there is no need to do calculation 471466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // in 8bit as we are already in 16bit, ... Yet this is what 472466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // benchmarks the fastest! 473466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d0 = _mm_subs_epu8(a01s, b01s); 474466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d1 = _mm_subs_epu8(b01s, a01s); 475466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d2 = _mm_subs_epu8(a23s, b23s); 476466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d3 = _mm_subs_epu8(b23s, a23s); 477466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 478466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Square and add them all together. 479466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i madd0 = _mm_madd_epi16(d0, d0); 480466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i madd1 = _mm_madd_epi16(d1, d1); 481466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i madd2 = _mm_madd_epi16(d2, d2); 482466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i madd3 = _mm_madd_epi16(d3, d3); 483466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i sum0 = _mm_add_epi32(madd0, madd1); 484466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i sum1 = _mm_add_epi32(madd2, madd3); 485466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i sum2 = _mm_add_epi32(sum0, sum1); 486466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int32_t tmp[4]; 487466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)tmp, sum2); 488466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora return (tmp[3] + tmp[2] + tmp[1] + tmp[0]); 489466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 490466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 4915a50414796e9a458925c7a13a15055d02406bf43Vikas Arora//------------------------------------------------------------------------------ 492466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Texture distortion 493466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// 494466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// We try to match the spectral content (weighted) between source and 495466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// reconstructed samples. 496466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 497466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Hadamard transform 498466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Returns the difference between the weighted sum of the absolute value of 499466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// transformed coefficients. 500466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic int TTransformSSE2(const uint8_t* inA, const uint8_t* inB, 501466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const uint16_t* const w) { 502466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int32_t sum[4]; 503466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i tmp_0, tmp_1, tmp_2, tmp_3; 504466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i zero = _mm_setzero_si128(); 505466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i one = _mm_set1_epi16(1); 506466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i three = _mm_set1_epi16(3); 507466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 508466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load, combine and tranpose inputs. 509466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 510466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inA_0 = _mm_loadl_epi64((__m128i*)&inA[BPS * 0]); 511466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inA_1 = _mm_loadl_epi64((__m128i*)&inA[BPS * 1]); 512466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inA_2 = _mm_loadl_epi64((__m128i*)&inA[BPS * 2]); 513466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inA_3 = _mm_loadl_epi64((__m128i*)&inA[BPS * 3]); 514466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inB_0 = _mm_loadl_epi64((__m128i*)&inB[BPS * 0]); 515466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inB_1 = _mm_loadl_epi64((__m128i*)&inB[BPS * 1]); 516466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inB_2 = _mm_loadl_epi64((__m128i*)&inB[BPS * 2]); 517466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inB_3 = _mm_loadl_epi64((__m128i*)&inB[BPS * 3]); 518466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 519466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Combine inA and inB (we'll do two transforms in parallel). 520466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inAB_0 = _mm_unpacklo_epi8(inA_0, inB_0); 521466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inAB_1 = _mm_unpacklo_epi8(inA_1, inB_1); 522466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inAB_2 = _mm_unpacklo_epi8(inA_2, inB_2); 523466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inAB_3 = _mm_unpacklo_epi8(inA_3, inB_3); 524466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 b00 a01 b01 a02 b03 a03 b03 0 0 0 0 0 0 0 0 525466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a10 b10 a11 b11 a12 b12 a13 b13 0 0 0 0 0 0 0 0 526466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a20 b20 a21 b21 a22 b22 a23 b23 0 0 0 0 0 0 0 0 527466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a30 b30 a31 b31 a32 b32 a33 b33 0 0 0 0 0 0 0 0 528466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 529466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Transpose the two 4x4, discarding the filling zeroes. 530466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_0 = _mm_unpacklo_epi8(inAB_0, inAB_2); 531466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_1 = _mm_unpacklo_epi8(inAB_1, inAB_3); 532466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a20 b00 b20 a01 a21 b01 b21 a02 a22 b02 b22 a03 a23 b03 b23 533466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a10 a30 b10 b30 a11 a31 b11 b31 a12 a32 b12 b32 a13 a33 b13 b33 534466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_0 = _mm_unpacklo_epi8(transpose0_0, transpose0_1); 535466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_1 = _mm_unpackhi_epi8(transpose0_0, transpose0_1); 536466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 b00 b10 b20 b30 a01 a11 a21 a31 b01 b11 b21 b31 537466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 b02 b12 b22 b32 a03 a13 a23 a33 b03 b13 b23 b33 538466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 539466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Convert to 16b. 540466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora tmp_0 = _mm_unpacklo_epi8(transpose1_0, zero); 541466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora tmp_1 = _mm_unpackhi_epi8(transpose1_0, zero); 542466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora tmp_2 = _mm_unpacklo_epi8(transpose1_1, zero); 543466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora tmp_3 = _mm_unpackhi_epi8(transpose1_1, zero); 544466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 b00 b10 b20 b30 545466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a01 a11 a21 a31 b01 b11 b21 b31 546466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 b02 b12 b22 b32 547466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a03 a13 a23 a33 b03 b13 b23 b33 548466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 549466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 550466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Horizontal pass and subsequent transpose. 551466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 552466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Calculate a and b (two 4x4 at once). 553466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a0 = _mm_slli_epi16(_mm_add_epi16(tmp_0, tmp_2), 2); 554466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a1 = _mm_slli_epi16(_mm_add_epi16(tmp_1, tmp_3), 2); 555466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a2 = _mm_slli_epi16(_mm_sub_epi16(tmp_1, tmp_3), 2); 556466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a3 = _mm_slli_epi16(_mm_sub_epi16(tmp_0, tmp_2), 2); 557466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b0_extra = (a0 != 0); 558466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b0_extra = _mm_andnot_si128(_mm_cmpeq_epi16 (a0, zero), one); 559466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b0_base = _mm_add_epi16(a0, a1); 560466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b1 = _mm_add_epi16(a3, a2); 561466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b2 = _mm_sub_epi16(a3, a2); 562466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b3 = _mm_sub_epi16(a0, a1); 563466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b0 = _mm_add_epi16(b0_base, b0_extra); 564466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a01 a02 a03 b00 b01 b02 b03 565466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a10 a11 a12 a13 b10 b11 b12 b13 566466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a20 a21 a22 a23 b20 b21 b22 b23 567466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a30 a31 a32 a33 b30 b31 b32 b33 568466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 569466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Transpose the two 4x4. 570466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_0 = _mm_unpacklo_epi16(b0, b1); 571466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_1 = _mm_unpacklo_epi16(b2, b3); 572466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_2 = _mm_unpackhi_epi16(b0, b1); 573466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_3 = _mm_unpackhi_epi16(b2, b3); 574466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a01 a11 a02 a12 a03 a13 575466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a20 a30 a21 a31 a22 a32 a23 a33 576466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b00 b10 b01 b11 b02 b12 b03 b13 577466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b20 b30 b21 b31 b22 b32 b23 b33 578466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_0 = _mm_unpacklo_epi32(transpose0_0, transpose0_1); 579466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_1 = _mm_unpacklo_epi32(transpose0_2, transpose0_3); 580466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_2 = _mm_unpackhi_epi32(transpose0_0, transpose0_1); 581466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_3 = _mm_unpackhi_epi32(transpose0_2, transpose0_3); 582466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 a01 a11 a21 a31 583466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b00 b10 b20 b30 b01 b11 b21 b31 584466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 a03 a13 a23 a33 585466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b02 b12 a22 b32 b03 b13 b23 b33 586466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora tmp_0 = _mm_unpacklo_epi64(transpose1_0, transpose1_1); 587466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora tmp_1 = _mm_unpackhi_epi64(transpose1_0, transpose1_1); 588466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora tmp_2 = _mm_unpacklo_epi64(transpose1_2, transpose1_3); 589466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora tmp_3 = _mm_unpackhi_epi64(transpose1_2, transpose1_3); 590466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 b00 b10 b20 b30 591466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a01 a11 a21 a31 b01 b11 b21 b31 592466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 b02 b12 b22 b32 593466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a03 a13 a23 a33 b03 b13 b23 b33 594466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 595466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 596466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Vertical pass and difference of weighted sums. 597466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 598466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load all inputs. 599466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // TODO(cduvivier): Make variable declarations and allocations aligned so 600466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // we can use _mm_load_si128 instead of _mm_loadu_si128. 601466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i w_0 = _mm_loadu_si128((__m128i*)&w[0]); 602466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i w_8 = _mm_loadu_si128((__m128i*)&w[8]); 603466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 604466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Calculate a and b (two 4x4 at once). 605466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a0 = _mm_add_epi16(tmp_0, tmp_2); 606466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a1 = _mm_add_epi16(tmp_1, tmp_3); 607466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a2 = _mm_sub_epi16(tmp_1, tmp_3); 608466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a3 = _mm_sub_epi16(tmp_0, tmp_2); 609466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b0 = _mm_add_epi16(a0, a1); 610466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b1 = _mm_add_epi16(a3, a2); 611466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b2 = _mm_sub_epi16(a3, a2); 612466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b3 = _mm_sub_epi16(a0, a1); 613466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 614466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Separate the transforms of inA and inB. 615466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i A_b0 = _mm_unpacklo_epi64(b0, b1); 616466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i A_b2 = _mm_unpacklo_epi64(b2, b3); 617466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i B_b0 = _mm_unpackhi_epi64(b0, b1); 618466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i B_b2 = _mm_unpackhi_epi64(b2, b3); 619466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 620466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 621466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // sign(b) = b >> 15 (0x0000 if positive, 0xffff if negative) 622466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i sign_A_b0 = _mm_srai_epi16(A_b0, 15); 623466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i sign_A_b2 = _mm_srai_epi16(A_b2, 15); 624466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i sign_B_b0 = _mm_srai_epi16(B_b0, 15); 625466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i sign_B_b2 = _mm_srai_epi16(B_b2, 15); 626466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 627466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b = abs(b) = (b ^ sign) - sign 628466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora A_b0 = _mm_xor_si128(A_b0, sign_A_b0); 629466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora A_b2 = _mm_xor_si128(A_b2, sign_A_b2); 630466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora B_b0 = _mm_xor_si128(B_b0, sign_B_b0); 631466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora B_b2 = _mm_xor_si128(B_b2, sign_B_b2); 632466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora A_b0 = _mm_sub_epi16(A_b0, sign_A_b0); 633466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora A_b2 = _mm_sub_epi16(A_b2, sign_A_b2); 634466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora B_b0 = _mm_sub_epi16(B_b0, sign_B_b0); 635466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora B_b2 = _mm_sub_epi16(B_b2, sign_B_b2); 636466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 637466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 638466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b = abs(b) + 3 639466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora A_b0 = _mm_add_epi16(A_b0, three); 640466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora A_b2 = _mm_add_epi16(A_b2, three); 641466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora B_b0 = _mm_add_epi16(B_b0, three); 642466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora B_b2 = _mm_add_epi16(B_b2, three); 643466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 644466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // abs((b + (b<0) + 3) >> 3) = (abs(b) + 3) >> 3 645466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b = (abs(b) + 3) >> 3 646466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora A_b0 = _mm_srai_epi16(A_b0, 3); 647466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora A_b2 = _mm_srai_epi16(A_b2, 3); 648466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora B_b0 = _mm_srai_epi16(B_b0, 3); 649466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora B_b2 = _mm_srai_epi16(B_b2, 3); 650466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 651466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // weighted sums 652466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora A_b0 = _mm_madd_epi16(A_b0, w_0); 653466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora A_b2 = _mm_madd_epi16(A_b2, w_8); 654466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora B_b0 = _mm_madd_epi16(B_b0, w_0); 655466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora B_b2 = _mm_madd_epi16(B_b2, w_8); 656466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora A_b0 = _mm_add_epi32(A_b0, A_b2); 657466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora B_b0 = _mm_add_epi32(B_b0, B_b2); 658466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 659466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // difference of weighted sums 660466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora A_b0 = _mm_sub_epi32(A_b0, B_b0); 661466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&sum[0], A_b0); 662466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 663466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora return sum[0] + sum[1] + sum[2] + sum[3]; 664466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 665466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 666466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic int Disto4x4SSE2(const uint8_t* const a, const uint8_t* const b, 667466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const uint16_t* const w) { 668466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const int diff_sum = TTransformSSE2(a, b, w); 669466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora return (abs(diff_sum) + 8) >> 4; 670466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 671466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 672466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic int Disto16x16SSE2(const uint8_t* const a, const uint8_t* const b, 673466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const uint16_t* const w) { 674466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int D = 0; 675466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int x, y; 676466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora for (y = 0; y < 16 * BPS; y += 4 * BPS) { 677466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora for (x = 0; x < 16; x += 4) { 678466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora D += Disto4x4SSE2(a + x + y, b + x + y, w); 679466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 680466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 681466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora return D; 682466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 683466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 684466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 6855a50414796e9a458925c7a13a15055d02406bf43Vikas Arora//------------------------------------------------------------------------------ 686466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Quantization 687466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// 688466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 689466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Simple quantization 690466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic int QuantizeBlockSSE2(int16_t in[16], int16_t out[16], 691466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int n, const VP8Matrix* const mtx) { 692466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i max_coeff_2047 = _mm_set1_epi16(2047); 693466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i zero = _mm_set1_epi16(0); 694466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i sign0, sign8; 695466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i coeff0, coeff8; 696466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i out0, out8; 697466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i packed_out; 698466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 699466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load all inputs. 700466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // TODO(cduvivier): Make variable declarations and allocations aligned so that 701466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // we can use _mm_load_si128 instead of _mm_loadu_si128. 702466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i in0 = _mm_loadu_si128((__m128i*)&in[0]); 703466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i in8 = _mm_loadu_si128((__m128i*)&in[8]); 704466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i sharpen0 = _mm_loadu_si128((__m128i*)&mtx->sharpen_[0]); 705466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i sharpen8 = _mm_loadu_si128((__m128i*)&mtx->sharpen_[8]); 706466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i iq0 = _mm_loadu_si128((__m128i*)&mtx->iq_[0]); 707466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i iq8 = _mm_loadu_si128((__m128i*)&mtx->iq_[8]); 708466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i bias0 = _mm_loadu_si128((__m128i*)&mtx->bias_[0]); 709466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i bias8 = _mm_loadu_si128((__m128i*)&mtx->bias_[8]); 710466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i q0 = _mm_loadu_si128((__m128i*)&mtx->q_[0]); 711466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i q8 = _mm_loadu_si128((__m128i*)&mtx->q_[8]); 712466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i zthresh0 = _mm_loadu_si128((__m128i*)&mtx->zthresh_[0]); 713466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i zthresh8 = _mm_loadu_si128((__m128i*)&mtx->zthresh_[8]); 714466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 715466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // sign(in) = in >> 15 (0x0000 if positive, 0xffff if negative) 716466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora sign0 = _mm_srai_epi16(in0, 15); 717466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora sign8 = _mm_srai_epi16(in8, 15); 718466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 719466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // coeff = abs(in) = (in ^ sign) - sign 720466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora coeff0 = _mm_xor_si128(in0, sign0); 721466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora coeff8 = _mm_xor_si128(in8, sign8); 722466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora coeff0 = _mm_sub_epi16(coeff0, sign0); 723466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora coeff8 = _mm_sub_epi16(coeff8, sign8); 724466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 725466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // coeff = abs(in) + sharpen 726466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora coeff0 = _mm_add_epi16(coeff0, sharpen0); 727466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora coeff8 = _mm_add_epi16(coeff8, sharpen8); 728466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 729466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // if (coeff > 2047) coeff = 2047 730466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora coeff0 = _mm_min_epi16(coeff0, max_coeff_2047); 731466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora coeff8 = _mm_min_epi16(coeff8, max_coeff_2047); 732466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 733466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // out = (coeff * iQ + B) >> QFIX; 734466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 735466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // doing calculations with 32b precision (QFIX=17) 736466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // out = (coeff * iQ) 737466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i coeff_iQ0H = _mm_mulhi_epu16(coeff0, iq0); 738466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i coeff_iQ0L = _mm_mullo_epi16(coeff0, iq0); 739466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i coeff_iQ8H = _mm_mulhi_epu16(coeff8, iq8); 740466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i coeff_iQ8L = _mm_mullo_epi16(coeff8, iq8); 741466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i out_00 = _mm_unpacklo_epi16(coeff_iQ0L, coeff_iQ0H); 742466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i out_04 = _mm_unpackhi_epi16(coeff_iQ0L, coeff_iQ0H); 743466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i out_08 = _mm_unpacklo_epi16(coeff_iQ8L, coeff_iQ8H); 744466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i out_12 = _mm_unpackhi_epi16(coeff_iQ8L, coeff_iQ8H); 745466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // expand bias from 16b to 32b 746466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i bias_00 = _mm_unpacklo_epi16(bias0, zero); 747466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i bias_04 = _mm_unpackhi_epi16(bias0, zero); 748466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i bias_08 = _mm_unpacklo_epi16(bias8, zero); 749466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i bias_12 = _mm_unpackhi_epi16(bias8, zero); 750466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // out = (coeff * iQ + B) 751466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out_00 = _mm_add_epi32(out_00, bias_00); 752466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out_04 = _mm_add_epi32(out_04, bias_04); 753466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out_08 = _mm_add_epi32(out_08, bias_08); 754466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out_12 = _mm_add_epi32(out_12, bias_12); 755466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // out = (coeff * iQ + B) >> QFIX; 756466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out_00 = _mm_srai_epi32(out_00, QFIX); 757466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out_04 = _mm_srai_epi32(out_04, QFIX); 758466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out_08 = _mm_srai_epi32(out_08, QFIX); 759466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out_12 = _mm_srai_epi32(out_12, QFIX); 760466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // pack result as 16b 761466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out0 = _mm_packs_epi32(out_00, out_04); 762466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out8 = _mm_packs_epi32(out_08, out_12); 763466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 764466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 765466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // get sign back (if (sign[j]) out_n = -out_n) 766466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out0 = _mm_xor_si128(out0, sign0); 767466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out8 = _mm_xor_si128(out8, sign8); 768466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out0 = _mm_sub_epi16(out0, sign0); 769466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out8 = _mm_sub_epi16(out8, sign8); 770466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 771466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // in = out * Q 772466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in0 = _mm_mullo_epi16(out0, q0); 773466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in8 = _mm_mullo_epi16(out8, q8); 774466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 775466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // if (coeff <= mtx->zthresh_) {in=0; out=0;} 776466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 777466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i cmp0 = _mm_cmpgt_epi16(coeff0, zthresh0); 778466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i cmp8 = _mm_cmpgt_epi16(coeff8, zthresh8); 779466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in0 = _mm_and_si128(in0, cmp0); 780466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in8 = _mm_and_si128(in8, cmp8); 781466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&in[0], in0); 782466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&in[8], in8); 783466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out0 = _mm_and_si128(out0, cmp0); 784466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out8 = _mm_and_si128(out8, cmp8); 785466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 786466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 787466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // zigzag the output before storing it. 788466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 789466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // The zigzag pattern can almost be reproduced with a small sequence of 790466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // shuffles. After it, we only need to swap the 7th (ending up in third 791466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // position instead of twelfth) and 8th values. 792466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 793466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i outZ0, outZ8; 794466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora outZ0 = _mm_shufflehi_epi16(out0, _MM_SHUFFLE(2, 1, 3, 0)); 795466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora outZ0 = _mm_shuffle_epi32 (outZ0, _MM_SHUFFLE(3, 1, 2, 0)); 796466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora outZ0 = _mm_shufflehi_epi16(outZ0, _MM_SHUFFLE(3, 1, 0, 2)); 797466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora outZ8 = _mm_shufflelo_epi16(out8, _MM_SHUFFLE(3, 0, 2, 1)); 798466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora outZ8 = _mm_shuffle_epi32 (outZ8, _MM_SHUFFLE(3, 1, 2, 0)); 799466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora outZ8 = _mm_shufflelo_epi16(outZ8, _MM_SHUFFLE(1, 3, 2, 0)); 800466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&out[0], outZ0); 801466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&out[8], outZ8); 802466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora packed_out = _mm_packs_epi16(outZ0, outZ8); 803466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 804466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 805466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const int16_t outZ_12 = out[12]; 806466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const int16_t outZ_3 = out[3]; 807466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out[3] = outZ_12; 808466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora out[12] = outZ_3; 809466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 810466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 811466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // detect if all 'out' values are zeroes or not 812466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 813466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int32_t tmp[4]; 814466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)tmp, packed_out); 815466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora if (n) { 816466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora tmp[0] &= ~0xff; 817466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 818466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora return (tmp[3] || tmp[2] || tmp[1] || tmp[0]); 819466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 820466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 821466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 822466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Aroraextern void VP8EncDspInitSSE2(void); 823466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Aroravoid VP8EncDspInitSSE2(void) { 824466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8CollectHistogram = CollectHistogramSSE2; 825466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8EncQuantizeBlock = QuantizeBlockSSE2; 826466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8ITransform = ITransformSSE2; 827466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8FTransform = FTransformSSE2; 828466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8SSE4x4 = SSE4x4SSE2; 829466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8TDisto4x4 = Disto4x4SSE2; 830466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8TDisto16x16 = Disto16x16SSE2; 831466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 832466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 833466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#if defined(__cplusplus) || defined(c_plusplus) 834466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} // extern "C" 835466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#endif 836466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 8375a50414796e9a458925c7a13a15055d02406bf43Vikas Arora#endif // WEBP_USE_SSE2 838