dec_sse2.c revision 1e7bf8805bd030c19924a5306837ecd72c295751
1a2415724fb3466168b2af5b08bd94ba732c0e753Vikas 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// 8a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora// SSE2 version of some decoding functions (idct, loop filtering). 9466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// 10466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Author: somnath@google.com (Somnath Banerjee) 11466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// cduvivier@google.com (Christian Duvivier) 12466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 13a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora#include "./dsp.h" 14a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora 151e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#if defined(__cplusplus) || defined(c_plusplus) 161e7bf8805bd030c19924a5306837ecd72c295751Vikas Aroraextern "C" { 171e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#endif 181e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 19a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora#if defined(WEBP_USE_SSE2) 20466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 21466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#include <emmintrin.h> 22a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora#include "../dec/vp8i.h" 23466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 24a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 25466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Transforms (Paragraph 14.4) 26466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 27466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void TransformSSE2(const int16_t* in, uint8_t* dst, int do_two) { 28466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // This implementation makes use of 16-bit fixed point versions of two 29466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // multiply constants: 30466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // K1 = sqrt(2) * cos (pi/8) ~= 85627 / 2^16 31466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // K2 = sqrt(2) * sin (pi/8) ~= 35468 / 2^16 32466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 33466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // To be able to use signed 16-bit integers, we use the following trick to 34466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // have constants within range: 35466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // - Associated constants are obtained by subtracting the 16-bit fixed point 36466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // version of one: 37466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // k = K - (1 << 16) => K = k + (1 << 16) 38466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // K1 = 85267 => k1 = 20091 39466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // K2 = 35468 => k2 = -30068 40466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // - The multiplication of a variable by a constant become the sum of the 41466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // variable and the multiplication of that variable by the associated 42466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // constant: 43466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // (x * K) >> 16 = (x * (k + (1 << 16))) >> 16 = ((x * k ) >> 16) + x 44466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i k1 = _mm_set1_epi16(20091); 45466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i k2 = _mm_set1_epi16(-30068); 46466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i T0, T1, T2, T3; 47466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 48466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load and concatenate the transform coefficients (we'll do two transforms 49466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // in parallel). In the case of only one transform, the second half of the 50466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // vectors will just contain random value we'll never use nor store. 51466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i in0, in1, in2, in3; 52466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 53466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in0 = _mm_loadl_epi64((__m128i*)&in[0]); 54466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in1 = _mm_loadl_epi64((__m128i*)&in[4]); 55466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in2 = _mm_loadl_epi64((__m128i*)&in[8]); 56466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in3 = _mm_loadl_epi64((__m128i*)&in[12]); 57466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 x x x x 58466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a01 a11 a21 a31 x x x x 59466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 x x x x 60466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a03 a13 a23 a33 x x x x 61466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora if (do_two) { 62466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inB0 = _mm_loadl_epi64((__m128i*)&in[16]); 63466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inB1 = _mm_loadl_epi64((__m128i*)&in[20]); 64466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inB2 = _mm_loadl_epi64((__m128i*)&in[24]); 65466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i inB3 = _mm_loadl_epi64((__m128i*)&in[28]); 66466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in0 = _mm_unpacklo_epi64(in0, inB0); 67466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in1 = _mm_unpacklo_epi64(in1, inB1); 68466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in2 = _mm_unpacklo_epi64(in2, inB2); 69466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora in3 = _mm_unpacklo_epi64(in3, inB3); 70466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 b00 b10 b20 b30 71466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a01 a11 a21 a31 b01 b11 b21 b31 72466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 b02 b12 b22 b32 73466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a03 a13 a23 a33 b03 b13 b23 b33 74466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 75466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 76466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 77466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Vertical pass and subsequent transpose. 78466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 79466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // First pass, c and d calculations are longer because of the "trick" 80466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // multiplications. 81466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a = _mm_add_epi16(in0, in2); 82466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b = _mm_sub_epi16(in0, in2); 83466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // c = MUL(in1, K2) - MUL(in3, K1) = MUL(in1, k2) - MUL(in3, k1) + in1 - in3 84466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c1 = _mm_mulhi_epi16(in1, k2); 85466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c2 = _mm_mulhi_epi16(in3, k1); 86466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c3 = _mm_sub_epi16(in1, in3); 87466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c4 = _mm_sub_epi16(c1, c2); 88466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c = _mm_add_epi16(c3, c4); 89466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // d = MUL(in1, K1) + MUL(in3, K2) = MUL(in1, k1) + MUL(in3, k2) + in1 + in3 90466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d1 = _mm_mulhi_epi16(in1, k1); 91466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d2 = _mm_mulhi_epi16(in3, k2); 92466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d3 = _mm_add_epi16(in1, in3); 93466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d4 = _mm_add_epi16(d1, d2); 94466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d = _mm_add_epi16(d3, d4); 95466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 96466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Second pass. 97466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp0 = _mm_add_epi16(a, d); 98466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp1 = _mm_add_epi16(b, c); 99466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp2 = _mm_sub_epi16(b, c); 100466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp3 = _mm_sub_epi16(a, d); 101466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 102466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Transpose the two 4x4. 103466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a01 a02 a03 b00 b01 b02 b03 104466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a10 a11 a12 a13 b10 b11 b12 b13 105466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a20 a21 a22 a23 b20 b21 b22 b23 106466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a30 a31 a32 a33 b30 b31 b32 b33 107466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_0 = _mm_unpacklo_epi16(tmp0, tmp1); 108466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_1 = _mm_unpacklo_epi16(tmp2, tmp3); 109466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_2 = _mm_unpackhi_epi16(tmp0, tmp1); 110466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_3 = _mm_unpackhi_epi16(tmp2, tmp3); 111466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a01 a11 a02 a12 a03 a13 112466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a20 a30 a21 a31 a22 a32 a23 a33 113466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b00 b10 b01 b11 b02 b12 b03 b13 114466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b20 b30 b21 b31 b22 b32 b23 b33 115466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_0 = _mm_unpacklo_epi32(transpose0_0, transpose0_1); 116466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_1 = _mm_unpacklo_epi32(transpose0_2, transpose0_3); 117466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_2 = _mm_unpackhi_epi32(transpose0_0, transpose0_1); 118466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_3 = _mm_unpackhi_epi32(transpose0_2, transpose0_3); 119466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 a01 a11 a21 a31 120466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b00 b10 b20 b30 b01 b11 b21 b31 121466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 a03 a13 a23 a33 122466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b02 b12 a22 b32 b03 b13 b23 b33 123466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T0 = _mm_unpacklo_epi64(transpose1_0, transpose1_1); 124466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T1 = _mm_unpackhi_epi64(transpose1_0, transpose1_1); 125466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T2 = _mm_unpacklo_epi64(transpose1_2, transpose1_3); 126466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T3 = _mm_unpackhi_epi64(transpose1_2, transpose1_3); 127466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 b00 b10 b20 b30 128466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a01 a11 a21 a31 b01 b11 b21 b31 129466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 b02 b12 b22 b32 130466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a03 a13 a23 a33 b03 b13 b23 b33 131466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 132466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 133466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Horizontal pass and subsequent transpose. 134466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 135466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // First pass, c and d calculations are longer because of the "trick" 136466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // multiplications. 137466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i four = _mm_set1_epi16(4); 138466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i dc = _mm_add_epi16(T0, four); 139466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a = _mm_add_epi16(dc, T2); 140466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i b = _mm_sub_epi16(dc, T2); 141466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // c = MUL(T1, K2) - MUL(T3, K1) = MUL(T1, k2) - MUL(T3, k1) + T1 - T3 142466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c1 = _mm_mulhi_epi16(T1, k2); 143466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c2 = _mm_mulhi_epi16(T3, k1); 144466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c3 = _mm_sub_epi16(T1, T3); 145466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c4 = _mm_sub_epi16(c1, c2); 146466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i c = _mm_add_epi16(c3, c4); 147466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // d = MUL(T1, K1) + MUL(T3, K2) = MUL(T1, k1) + MUL(T3, k2) + T1 + T3 148466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d1 = _mm_mulhi_epi16(T1, k1); 149466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d2 = _mm_mulhi_epi16(T3, k2); 150466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d3 = _mm_add_epi16(T1, T3); 151466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d4 = _mm_add_epi16(d1, d2); 152466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i d = _mm_add_epi16(d3, d4); 153466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 154466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Second pass. 155466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp0 = _mm_add_epi16(a, d); 156466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp1 = _mm_add_epi16(b, c); 157466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp2 = _mm_sub_epi16(b, c); 158466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i tmp3 = _mm_sub_epi16(a, d); 159466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i shifted0 = _mm_srai_epi16(tmp0, 3); 160466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i shifted1 = _mm_srai_epi16(tmp1, 3); 161466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i shifted2 = _mm_srai_epi16(tmp2, 3); 162466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i shifted3 = _mm_srai_epi16(tmp3, 3); 163466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 164466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Transpose the two 4x4. 165466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a01 a02 a03 b00 b01 b02 b03 166466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a10 a11 a12 a13 b10 b11 b12 b13 167466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a20 a21 a22 a23 b20 b21 b22 b23 168466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a30 a31 a32 a33 b30 b31 b32 b33 169466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_0 = _mm_unpacklo_epi16(shifted0, shifted1); 170466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_1 = _mm_unpacklo_epi16(shifted2, shifted3); 171466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_2 = _mm_unpackhi_epi16(shifted0, shifted1); 172466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose0_3 = _mm_unpackhi_epi16(shifted2, shifted3); 173466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a01 a11 a02 a12 a03 a13 174466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a20 a30 a21 a31 a22 a32 a23 a33 175466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b00 b10 b01 b11 b02 b12 b03 b13 176466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b20 b30 b21 b31 b22 b32 b23 b33 177466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_0 = _mm_unpacklo_epi32(transpose0_0, transpose0_1); 178466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_1 = _mm_unpacklo_epi32(transpose0_2, transpose0_3); 179466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_2 = _mm_unpackhi_epi32(transpose0_0, transpose0_1); 180466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i transpose1_3 = _mm_unpackhi_epi32(transpose0_2, transpose0_3); 181466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 a01 a11 a21 a31 182466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b00 b10 b20 b30 b01 b11 b21 b31 183466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 a03 a13 a23 a33 184466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // b02 b12 a22 b32 b03 b13 b23 b33 185466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T0 = _mm_unpacklo_epi64(transpose1_0, transpose1_1); 186466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T1 = _mm_unpackhi_epi64(transpose1_0, transpose1_1); 187466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T2 = _mm_unpacklo_epi64(transpose1_2, transpose1_3); 188466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora T3 = _mm_unpackhi_epi64(transpose1_2, transpose1_3); 189466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a00 a10 a20 a30 b00 b10 b20 b30 190466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a01 a11 a21 a31 b01 b11 b21 b31 191466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a02 a12 a22 a32 b02 b12 b22 b32 192466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // a03 a13 a23 a33 b03 b13 b23 b33 193466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 194466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 195466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Add inverse transform to 'dst' and store. 196466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { 1971e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const __m128i zero = _mm_setzero_si128(); 198466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load the reference(s). 199466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i dst0, dst1, dst2, dst3; 200466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora if (do_two) { 201466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load eight bytes/pixels per line. 202466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst0 = _mm_loadl_epi64((__m128i*)&dst[0 * BPS]); 203466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst1 = _mm_loadl_epi64((__m128i*)&dst[1 * BPS]); 204466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst2 = _mm_loadl_epi64((__m128i*)&dst[2 * BPS]); 205466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst3 = _mm_loadl_epi64((__m128i*)&dst[3 * BPS]); 206466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } else { 207466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load four bytes/pixels per line. 208466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst0 = _mm_cvtsi32_si128(*(int*)&dst[0 * BPS]); 209466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst1 = _mm_cvtsi32_si128(*(int*)&dst[1 * BPS]); 210466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst2 = _mm_cvtsi32_si128(*(int*)&dst[2 * BPS]); 211466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst3 = _mm_cvtsi32_si128(*(int*)&dst[3 * BPS]); 212466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 213466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Convert to 16b. 214466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst0 = _mm_unpacklo_epi8(dst0, zero); 215466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst1 = _mm_unpacklo_epi8(dst1, zero); 216466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst2 = _mm_unpacklo_epi8(dst2, zero); 217466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst3 = _mm_unpacklo_epi8(dst3, zero); 218466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Add the inverse transform(s). 219466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst0 = _mm_add_epi16(dst0, T0); 220466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst1 = _mm_add_epi16(dst1, T1); 221466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst2 = _mm_add_epi16(dst2, T2); 222466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst3 = _mm_add_epi16(dst3, T3); 223466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Unsigned saturate to 8b. 224466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst0 = _mm_packus_epi16(dst0, dst0); 225466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst1 = _mm_packus_epi16(dst1, dst1); 226466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst2 = _mm_packus_epi16(dst2, dst2); 227466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora dst3 = _mm_packus_epi16(dst3, dst3); 228466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Store the results. 229466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora if (do_two) { 230466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Store eight bytes/pixels per line. 231466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storel_epi64((__m128i*)&dst[0 * BPS], dst0); 232466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storel_epi64((__m128i*)&dst[1 * BPS], dst1); 233466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storel_epi64((__m128i*)&dst[2 * BPS], dst2); 234466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storel_epi64((__m128i*)&dst[3 * BPS], dst3); 235466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } else { 236466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Store four bytes/pixels per line. 237466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *((int32_t *)&dst[0 * BPS]) = _mm_cvtsi128_si32(dst0); 238466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *((int32_t *)&dst[1 * BPS]) = _mm_cvtsi128_si32(dst1); 239466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *((int32_t *)&dst[2 * BPS]) = _mm_cvtsi128_si32(dst2); 240466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *((int32_t *)&dst[3 * BPS]) = _mm_cvtsi128_si32(dst3); 241466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 242466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 243466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 244466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 245a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 246466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Loop Filter (Paragraph 15) 247466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 248466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Compute abs(p - q) = subs(p - q) OR subs(q - p) 249466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#define MM_ABS(p, q) _mm_or_si128( \ 250466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_subs_epu8((q), (p)), \ 251466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_subs_epu8((p), (q))) 252466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 253466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Shift each byte of "a" by N bits while preserving by the sign bit. 254466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// 255466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// It first shifts the lower bytes of the words and then the upper bytes and 256466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// then merges the results together. 257466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#define SIGNED_SHIFT_N(a, N) { \ 258466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i t = a; \ 259466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t = _mm_slli_epi16(t, 8); \ 260466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t = _mm_srai_epi16(t, N); \ 261466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t = _mm_srli_epi16(t, 8); \ 262466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora \ 263466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora a = _mm_srai_epi16(a, N + 8); \ 264466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora a = _mm_slli_epi16(a, 8); \ 265466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora \ 266466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora a = _mm_or_si128(t, a); \ 267466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 268466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 269466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#define FLIP_SIGN_BIT2(a, b) { \ 270466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora a = _mm_xor_si128(a, sign_bit); \ 271466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora b = _mm_xor_si128(b, sign_bit); \ 272466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 273466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 274466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#define FLIP_SIGN_BIT4(a, b, c, d) { \ 275466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora FLIP_SIGN_BIT2(a, b); \ 276466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora FLIP_SIGN_BIT2(c, d); \ 277466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 278466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 279466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#define GET_NOTHEV(p1, p0, q0, q1, hev_thresh, not_hev) { \ 280466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i zero = _mm_setzero_si128(); \ 2811e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const __m128i t_1 = MM_ABS(p1, p0); \ 2821e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const __m128i t_2 = MM_ABS(q1, q0); \ 283466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora \ 284466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i h = _mm_set1_epi8(hev_thresh); \ 2851e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const __m128i t_3 = _mm_subs_epu8(t_1, h); /* abs(p1 - p0) - hev_tresh */ \ 2861e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const __m128i t_4 = _mm_subs_epu8(t_2, h); /* abs(q1 - q0) - hev_tresh */ \ 287466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora \ 2881e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora not_hev = _mm_or_si128(t_3, t_4); \ 289466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora not_hev = _mm_cmpeq_epi8(not_hev, zero); /* not_hev <= t1 && not_hev <= t2 */\ 290466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 291466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 292466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#define GET_BASE_DELTA(p1, p0, q0, q1, o) { \ 293466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i qp0 = _mm_subs_epi8(q0, p0); /* q0 - p0 */ \ 294466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora o = _mm_subs_epi8(p1, q1); /* p1 - q1 */ \ 295466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora o = _mm_adds_epi8(o, qp0); /* p1 - q1 + 1 * (q0 - p0) */ \ 296466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora o = _mm_adds_epi8(o, qp0); /* p1 - q1 + 2 * (q0 - p0) */ \ 297466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora o = _mm_adds_epi8(o, qp0); /* p1 - q1 + 3 * (q0 - p0) */ \ 298466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 299466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 300466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#define DO_SIMPLE_FILTER(p0, q0, fl) { \ 301466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i three = _mm_set1_epi8(3); \ 302466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i four = _mm_set1_epi8(4); \ 303466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i v3 = _mm_adds_epi8(fl, three); \ 304466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i v4 = _mm_adds_epi8(fl, four); \ 305466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora \ 306466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora /* Do +4 side */ \ 307466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora SIGNED_SHIFT_N(v4, 3); /* v4 >> 3 */ \ 308466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora q0 = _mm_subs_epi8(q0, v4); /* q0 -= v4 */ \ 309466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora \ 310466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora /* Now do +3 side */ \ 311466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora SIGNED_SHIFT_N(v3, 3); /* v3 >> 3 */ \ 312466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora p0 = _mm_adds_epi8(p0, v3); /* p0 += v3 */ \ 313466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 314466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 315466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Updates values of 2 pixels at MB edge during complex filtering. 316466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Update operations: 3171e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// q = q - delta and p = p + delta; where delta = [(a_hi >> 7), (a_lo >> 7)] 318466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#define UPDATE_2PIXELS(pi, qi, a_lo, a_hi) { \ 319466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a_lo7 = _mm_srai_epi16(a_lo, 7); \ 320466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a_hi7 = _mm_srai_epi16(a_hi, 7); \ 3211e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora const __m128i delta = _mm_packs_epi16(a_lo7, a_hi7); \ 3221e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora pi = _mm_adds_epi8(pi, delta); \ 3231e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora qi = _mm_subs_epi8(qi, delta); \ 324466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 325466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 326466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void NeedsFilter(const __m128i* p1, const __m128i* p0, const __m128i* q0, 327466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i* q1, int thresh, __m128i *mask) { 328466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i t1 = MM_ABS(*p1, *q1); // abs(p1 - q1) 329466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *mask = _mm_set1_epi8(0xFE); 330466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = _mm_and_si128(t1, *mask); // set lsb of each byte to zero 331466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = _mm_srli_epi16(t1, 1); // abs(p1 - q1) / 2 332466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 333466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *mask = MM_ABS(*p0, *q0); // abs(p0 - q0) 334466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *mask = _mm_adds_epu8(*mask, *mask); // abs(p0 - q0) * 2 335466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *mask = _mm_adds_epu8(*mask, t1); // abs(p0 - q0) * 2 + abs(p1 - q1) / 2 336466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 337466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = _mm_set1_epi8(thresh); 338466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *mask = _mm_subs_epu8(*mask, t1); // mask <= thresh 339466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *mask = _mm_cmpeq_epi8(*mask, _mm_setzero_si128()); 340466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 341466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 342a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 343466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Edge filtering functions 344466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 345466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Applies filter on 2 pixels (p0 and q0) 346a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arorastatic WEBP_INLINE void DoFilter2(const __m128i* p1, __m128i* p0, __m128i* q0, 347a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora const __m128i* q1, int thresh) { 348466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i a, mask; 349466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i sign_bit = _mm_set1_epi8(0x80); 350466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i p1s = _mm_xor_si128(*p1, sign_bit); 351466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i q1s = _mm_xor_si128(*q1, sign_bit); 352466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 353466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora NeedsFilter(p1, p0, q0, q1, thresh, &mask); 354466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 355466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // convert to signed values 356466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora FLIP_SIGN_BIT2(*p0, *q0); 357466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 358466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora GET_BASE_DELTA(p1s, *p0, *q0, q1s, a); 359466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora a = _mm_and_si128(a, mask); // mask filter values we don't care about 360466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora DO_SIMPLE_FILTER(*p0, *q0, a); 361466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 362466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // unoffset 363466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora FLIP_SIGN_BIT2(*p0, *q0); 364466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 365466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 366466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Applies filter on 4 pixels (p1, p0, q0 and q1) 367a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arorastatic WEBP_INLINE void DoFilter4(__m128i* p1, __m128i *p0, 368a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora __m128i* q0, __m128i* q1, 369a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora const __m128i* mask, int hev_thresh) { 370466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i not_hev; 371466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i t1, t2, t3; 372466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i sign_bit = _mm_set1_epi8(0x80); 373466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 374466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // compute hev mask 375466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora GET_NOTHEV(*p1, *p0, *q0, *q1, hev_thresh, not_hev); 376466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 377466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // convert to signed values 378466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora FLIP_SIGN_BIT4(*p1, *p0, *q0, *q1); 379466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 380466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = _mm_subs_epi8(*p1, *q1); // p1 - q1 381466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = _mm_andnot_si128(not_hev, t1); // hev(p1 - q1) 382466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t2 = _mm_subs_epi8(*q0, *p0); // q0 - p0 383466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = _mm_adds_epi8(t1, t2); // hev(p1 - q1) + 1 * (q0 - p0) 384466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = _mm_adds_epi8(t1, t2); // hev(p1 - q1) + 2 * (q0 - p0) 385466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = _mm_adds_epi8(t1, t2); // hev(p1 - q1) + 3 * (q0 - p0) 386466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = _mm_and_si128(t1, *mask); // mask filter values we don't care about 387466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 388466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Do +4 side 389466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t2 = _mm_set1_epi8(4); 390466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t2 = _mm_adds_epi8(t1, t2); // 3 * (q0 - p0) + (p1 - q1) + 4 391466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora SIGNED_SHIFT_N(t2, 3); // (3 * (q0 - p0) + hev(p1 - q1) + 4) >> 3 392466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t3 = t2; // save t2 393466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *q0 = _mm_subs_epi8(*q0, t2); // q0 -= t2 394466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 395466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Now do +3 side 396466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t2 = _mm_set1_epi8(3); 397466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t2 = _mm_adds_epi8(t1, t2); // +3 instead of +4 398466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora SIGNED_SHIFT_N(t2, 3); // (3 * (q0 - p0) + hev(p1 - q1) + 3) >> 3 399466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *p0 = _mm_adds_epi8(*p0, t2); // p0 += t2 400466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 401466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t2 = _mm_set1_epi8(1); 402466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t3 = _mm_adds_epi8(t3, t2); 403466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora SIGNED_SHIFT_N(t3, 1); // (3 * (q0 - p0) + hev(p1 - q1) + 4) >> 4 404466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 405466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t3 = _mm_and_si128(not_hev, t3); // if !hev 406466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *q1 = _mm_subs_epi8(*q1, t3); // q1 -= t3 407466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *p1 = _mm_adds_epi8(*p1, t3); // p1 += t3 408466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 409466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // unoffset 410466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora FLIP_SIGN_BIT4(*p1, *p0, *q0, *q1); 411466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 412466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 413466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Applies filter on 6 pixels (p2, p1, p0, q0, q1 and q2) 414a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arorastatic WEBP_INLINE void DoFilter6(__m128i *p2, __m128i* p1, __m128i *p0, 415a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora __m128i* q0, __m128i* q1, __m128i *q2, 416a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora const __m128i* mask, int hev_thresh) { 417466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i a, not_hev; 418466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i sign_bit = _mm_set1_epi8(0x80); 419466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 420466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // compute hev mask 421466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora GET_NOTHEV(*p1, *p0, *q0, *q1, hev_thresh, not_hev); 422466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 423466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // convert to signed values 424466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora FLIP_SIGN_BIT4(*p1, *p0, *q0, *q1); 425466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora FLIP_SIGN_BIT2(*p2, *q2); 426466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 427466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora GET_BASE_DELTA(*p1, *p0, *q0, *q1, a); 428466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 429466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { // do simple filter on pixels with hev 430466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i m = _mm_andnot_si128(not_hev, *mask); 431466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i f = _mm_and_si128(a, m); 432466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora DO_SIMPLE_FILTER(*p0, *q0, f); 433466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 434466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora { // do strong filter on pixels with not hev 435466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i zero = _mm_setzero_si128(); 436466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i nine = _mm_set1_epi16(0x0900); 437466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i sixty_three = _mm_set1_epi16(63); 438466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 439466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i m = _mm_and_si128(not_hev, *mask); 440466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i f = _mm_and_si128(a, m); 441466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i f_lo = _mm_unpacklo_epi8(zero, f); 442466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i f_hi = _mm_unpackhi_epi8(zero, f); 443466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 444466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i f9_lo = _mm_mulhi_epi16(f_lo, nine); // Filter (lo) * 9 445466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i f9_hi = _mm_mulhi_epi16(f_hi, nine); // Filter (hi) * 9 446466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i f18_lo = _mm_add_epi16(f9_lo, f9_lo); // Filter (lo) * 18 447466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i f18_hi = _mm_add_epi16(f9_hi, f9_hi); // Filter (hi) * 18 448466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 449466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a2_lo = _mm_add_epi16(f9_lo, sixty_three); // Filter * 9 + 63 450466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a2_hi = _mm_add_epi16(f9_hi, sixty_three); // Filter * 9 + 63 451466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 452466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a1_lo = _mm_add_epi16(f18_lo, sixty_three); // F... * 18 + 63 453466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a1_hi = _mm_add_epi16(f18_hi, sixty_three); // F... * 18 + 63 454466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 455466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a0_lo = _mm_add_epi16(f18_lo, a2_lo); // Filter * 27 + 63 456466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i a0_hi = _mm_add_epi16(f18_hi, a2_hi); // Filter * 27 + 63 457466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 458466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora UPDATE_2PIXELS(*p2, *q2, a2_lo, a2_hi); 459466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora UPDATE_2PIXELS(*p1, *q1, a1_lo, a1_hi); 460466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora UPDATE_2PIXELS(*p0, *q0, a0_lo, a0_hi); 461466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 462466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 463466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // unoffset 464466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora FLIP_SIGN_BIT4(*p1, *p0, *q0, *q1); 465466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora FLIP_SIGN_BIT2(*p2, *q2); 466466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 467466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 468466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// reads 8 rows across a vertical edge. 469466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// 470466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// TODO(somnath): Investigate _mm_shuffle* also see if it can be broken into 471466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// two Load4x4() to avoid code duplication. 472a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arorastatic WEBP_INLINE void Load8x4(const uint8_t* b, int stride, 473a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora __m128i* p, __m128i* q) { 474466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i t1, t2; 475466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 476466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load 0th, 1st, 4th and 5th rows 477466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i r0 = _mm_cvtsi32_si128(*((int*)&b[0 * stride])); // 03 02 01 00 478466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i r1 = _mm_cvtsi32_si128(*((int*)&b[1 * stride])); // 13 12 11 10 479466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i r4 = _mm_cvtsi32_si128(*((int*)&b[4 * stride])); // 43 42 41 40 480466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i r5 = _mm_cvtsi32_si128(*((int*)&b[5 * stride])); // 53 52 51 50 481466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 482466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora r0 = _mm_unpacklo_epi32(r0, r4); // 43 42 41 40 03 02 01 00 483466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora r1 = _mm_unpacklo_epi32(r1, r5); // 53 52 51 50 13 12 11 10 484466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 485466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // t1 = 53 43 52 42 51 41 50 40 13 03 12 02 11 01 10 00 486466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = _mm_unpacklo_epi8(r0, r1); 487466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 488466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load 2nd, 3rd, 6th and 7th rows 489466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora r0 = _mm_cvtsi32_si128(*((int*)&b[2 * stride])); // 23 22 21 22 490466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora r1 = _mm_cvtsi32_si128(*((int*)&b[3 * stride])); // 33 32 31 30 491466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora r4 = _mm_cvtsi32_si128(*((int*)&b[6 * stride])); // 63 62 61 60 492466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora r5 = _mm_cvtsi32_si128(*((int*)&b[7 * stride])); // 73 72 71 70 493466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 494466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora r0 = _mm_unpacklo_epi32(r0, r4); // 63 62 61 60 23 22 21 20 495466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora r1 = _mm_unpacklo_epi32(r1, r5); // 73 72 71 70 33 32 31 30 496466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 497466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // t2 = 73 63 72 62 71 61 70 60 33 23 32 22 31 21 30 20 498466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t2 = _mm_unpacklo_epi8(r0, r1); 499466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 500466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // t1 = 33 23 13 03 32 22 12 02 31 21 11 01 30 20 10 00 501466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // t2 = 73 63 53 43 72 62 52 42 71 61 51 41 70 60 50 40 502466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora r0 = t1; 503466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = _mm_unpacklo_epi16(t1, t2); 504466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t2 = _mm_unpackhi_epi16(r0, t2); 505466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 506466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // *p = 71 61 51 41 31 21 11 01 70 60 50 40 30 20 10 00 507466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // *q = 73 63 53 43 33 23 13 03 72 62 52 42 32 22 12 02 508466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *p = _mm_unpacklo_epi32(t1, t2); 509466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *q = _mm_unpackhi_epi32(t1, t2); 510466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 511466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 512a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arorastatic WEBP_INLINE void Load16x4(const uint8_t* r0, const uint8_t* r8, 513a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora int stride, 514a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora __m128i* p1, __m128i* p0, 515a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora __m128i* q0, __m128i* q1) { 516466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i t1, t2; 517466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Assume the pixels around the edge (|) are numbered as follows 518466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 00 01 | 02 03 519466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 10 11 | 12 13 520466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // ... | ... 521466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // e0 e1 | e2 e3 522466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // f0 f1 | f2 f3 523466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // 524466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // r0 is pointing to the 0th row (00) 525466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // r8 is pointing to the 8th row (80) 526466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 527466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load 528466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // p1 = 71 61 51 41 31 21 11 01 70 60 50 40 30 20 10 00 529466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // q0 = 73 63 53 43 33 23 13 03 72 62 52 42 32 22 12 02 530466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // p0 = f1 e1 d1 c1 b1 a1 91 81 f0 e0 d0 c0 b0 a0 90 80 531466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // q1 = f3 e3 d3 c3 b3 a3 93 83 f2 e2 d2 c2 b2 a2 92 82 532466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Load8x4(r0, stride, p1, q0); 533466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Load8x4(r8, stride, p0, q1); 534466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 535466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = *p1; 536466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t2 = *q0; 537466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // p1 = f0 e0 d0 c0 b0 a0 90 80 70 60 50 40 30 20 10 00 538466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // p0 = f1 e1 d1 c1 b1 a1 91 81 71 61 51 41 31 21 11 01 539466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // q0 = f2 e2 d2 c2 b2 a2 92 82 72 62 52 42 32 22 12 02 540466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // q1 = f3 e3 d3 c3 b3 a3 93 83 73 63 53 43 33 23 13 03 541466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *p1 = _mm_unpacklo_epi64(t1, *p0); 542466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *p0 = _mm_unpackhi_epi64(t1, *p0); 543466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *q0 = _mm_unpacklo_epi64(t2, *q1); 544466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *q1 = _mm_unpackhi_epi64(t2, *q1); 545466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 546466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 547a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arorastatic WEBP_INLINE void Store4x4(__m128i* x, uint8_t* dst, int stride) { 548466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int i; 549466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora for (i = 0; i < 4; ++i, dst += stride) { 550466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *((int32_t*)dst) = _mm_cvtsi128_si32(*x); 551466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *x = _mm_srli_si128(*x, 4); 552466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 553466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 554466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 555466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Transpose back and store 556a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arorastatic WEBP_INLINE void Store16x4(uint8_t* r0, uint8_t* r8, int stride, 557a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora __m128i* p1, __m128i* p0, 558a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora __m128i* q0, __m128i* q1) { 559466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i t1; 560466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 561466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // p0 = 71 70 61 60 51 50 41 40 31 30 21 20 11 10 01 00 562466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // p1 = f1 f0 e1 e0 d1 d0 c1 c0 b1 b0 a1 a0 91 90 81 80 563466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = *p0; 564466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *p0 = _mm_unpacklo_epi8(*p1, t1); 565466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *p1 = _mm_unpackhi_epi8(*p1, t1); 566466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 567466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // q0 = 73 72 63 62 53 52 43 42 33 32 23 22 13 12 03 02 568466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // q1 = f3 f2 e3 e2 d3 d2 c3 c2 b3 b2 a3 a2 93 92 83 82 569466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = *q0; 570466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *q0 = _mm_unpacklo_epi8(t1, *q1); 571466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *q1 = _mm_unpackhi_epi8(t1, *q1); 572466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 573466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // p0 = 33 32 31 30 23 22 21 20 13 12 11 10 03 02 01 00 574466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // q0 = 73 72 71 70 63 62 61 60 53 52 51 50 43 42 41 40 575466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = *p0; 576466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *p0 = _mm_unpacklo_epi16(t1, *q0); 577466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *q0 = _mm_unpackhi_epi16(t1, *q0); 578466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 579466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // p1 = b3 b2 b1 b0 a3 a2 a1 a0 93 92 91 90 83 82 81 80 580466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // q1 = f3 f2 f1 f0 e3 e2 e1 e0 d3 d2 d1 d0 c3 c2 c1 c0 581466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora t1 = *p1; 582466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *p1 = _mm_unpacklo_epi16(t1, *q1); 583466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora *q1 = _mm_unpackhi_epi16(t1, *q1); 584466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 585466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Store4x4(p0, r0, stride); 586466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora r0 += 4 * stride; 587466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Store4x4(q0, r0, stride); 588466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 589466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Store4x4(p1, r8, stride); 590466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora r8 += 4 * stride; 591466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Store4x4(q1, r8, stride); 592466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 593466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 594a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 595466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Simple In-loop filtering (Paragraph 15.2) 596466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 597466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void SimpleVFilter16SSE2(uint8_t* p, int stride, int thresh) { 598466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load 599466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i p1 = _mm_loadu_si128((__m128i*)&p[-2 * stride]); 600466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i p0 = _mm_loadu_si128((__m128i*)&p[-stride]); 601466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i q0 = _mm_loadu_si128((__m128i*)&p[0]); 602466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i q1 = _mm_loadu_si128((__m128i*)&p[stride]); 603466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 604466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora DoFilter2(&p1, &p0, &q0, &q1, thresh); 605466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 606466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Store 607466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&p[-stride], p0); 608466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)p, q0); 609466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 610466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 611466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void SimpleHFilter16SSE2(uint8_t* p, int stride, int thresh) { 612466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i p1, p0, q0, q1; 613466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 614466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora p -= 2; // beginning of p1 615466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 616466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Load16x4(p, p + 8 * stride, stride, &p1, &p0, &q0, &q1); 617466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora DoFilter2(&p1, &p0, &q0, &q1, thresh); 618466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Store16x4(p, p + 8 * stride, stride, &p1, &p0, &q0, &q1); 619466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 620466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 621466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void SimpleVFilter16iSSE2(uint8_t* p, int stride, int thresh) { 622466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int k; 623466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora for (k = 3; k > 0; --k) { 624466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora p += 4 * stride; 625466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora SimpleVFilter16SSE2(p, stride, thresh); 626466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 627466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 628466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 629466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void SimpleHFilter16iSSE2(uint8_t* p, int stride, int thresh) { 630466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int k; 631466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora for (k = 3; k > 0; --k) { 632466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora p += 4; 633466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora SimpleHFilter16SSE2(p, stride, thresh); 634466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 635466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 636466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 637a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 638466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// Complex In-loop filtering (Paragraph 15.3) 639466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 640466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#define MAX_DIFF1(p3, p2, p1, p0, m) { \ 641466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora m = MM_ABS(p3, p2); \ 642466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora m = _mm_max_epu8(m, MM_ABS(p2, p1)); \ 643466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora m = _mm_max_epu8(m, MM_ABS(p1, p0)); \ 644466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 645466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 646466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#define MAX_DIFF2(p3, p2, p1, p0, m) { \ 647466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora m = _mm_max_epu8(m, MM_ABS(p3, p2)); \ 648466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora m = _mm_max_epu8(m, MM_ABS(p2, p1)); \ 649466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora m = _mm_max_epu8(m, MM_ABS(p1, p0)); \ 650466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 651466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 652466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#define LOAD_H_EDGES4(p, stride, e1, e2, e3, e4) { \ 653466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora e1 = _mm_loadu_si128((__m128i*)&(p)[0 * stride]); \ 654466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora e2 = _mm_loadu_si128((__m128i*)&(p)[1 * stride]); \ 655466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora e3 = _mm_loadu_si128((__m128i*)&(p)[2 * stride]); \ 656466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora e4 = _mm_loadu_si128((__m128i*)&(p)[3 * stride]); \ 657466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 658466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 659466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#define LOADUV_H_EDGE(p, u, v, stride) { \ 660466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora p = _mm_loadl_epi64((__m128i*)&(u)[(stride)]); \ 661466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora p = _mm_unpacklo_epi64(p, _mm_loadl_epi64((__m128i*)&(v)[(stride)])); \ 662466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 663466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 664466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#define LOADUV_H_EDGES4(u, v, stride, e1, e2, e3, e4) { \ 665466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora LOADUV_H_EDGE(e1, u, v, 0 * stride); \ 666466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora LOADUV_H_EDGE(e2, u, v, 1 * stride); \ 667466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora LOADUV_H_EDGE(e3, u, v, 2 * stride); \ 668466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora LOADUV_H_EDGE(e4, u, v, 3 * stride); \ 669466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 670466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 671466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#define STOREUV(p, u, v, stride) { \ 672466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storel_epi64((__m128i*)&u[(stride)], p); \ 673466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora p = _mm_srli_si128(p, 8); \ 674466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storel_epi64((__m128i*)&v[(stride)], p); \ 675466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 676466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 677466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#define COMPLEX_FL_MASK(p1, p0, q0, q1, thresh, ithresh, mask) { \ 678466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i fl_yes; \ 679466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora const __m128i it = _mm_set1_epi8(ithresh); \ 680466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora mask = _mm_subs_epu8(mask, it); \ 681466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora mask = _mm_cmpeq_epi8(mask, _mm_setzero_si128()); \ 682466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora NeedsFilter(&p1, &p0, &q0, &q1, thresh, &fl_yes); \ 683466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora mask = _mm_and_si128(mask, fl_yes); \ 684466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 685466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 686466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// on macroblock edges 687466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void VFilter16SSE2(uint8_t* p, int stride, 688466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int thresh, int ithresh, int hev_thresh) { 689466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i t1; 690466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i mask; 691466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i p2, p1, p0, q0, q1, q2; 692466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 693466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load p3, p2, p1, p0 694466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora LOAD_H_EDGES4(p - 4 * stride, stride, t1, p2, p1, p0); 695466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF1(t1, p2, p1, p0, mask); 696466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 697466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load q0, q1, q2, q3 698466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora LOAD_H_EDGES4(p, stride, q0, q1, q2, t1); 699466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF2(t1, q2, q1, q0, mask); 700466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 701466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora COMPLEX_FL_MASK(p1, p0, q0, q1, thresh, ithresh, mask); 702466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora DoFilter6(&p2, &p1, &p0, &q0, &q1, &q2, &mask, hev_thresh); 703466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 704466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Store 705466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&p[-3 * stride], p2); 706466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&p[-2 * stride], p1); 707466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&p[-1 * stride], p0); 708466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&p[0 * stride], q0); 709466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&p[1 * stride], q1); 710466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&p[2 * stride], q2); 711466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 712466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 713466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void HFilter16SSE2(uint8_t* p, int stride, 714466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int thresh, int ithresh, int hev_thresh) { 715466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i mask; 716466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i p3, p2, p1, p0, q0, q1, q2, q3; 717466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 718466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora uint8_t* const b = p - 4; 719466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Load16x4(b, b + 8 * stride, stride, &p3, &p2, &p1, &p0); // p3, p2, p1, p0 720466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF1(p3, p2, p1, p0, mask); 721466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 722466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Load16x4(p, p + 8 * stride, stride, &q0, &q1, &q2, &q3); // q0, q1, q2, q3 723466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF2(q3, q2, q1, q0, mask); 724466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 725466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora COMPLEX_FL_MASK(p1, p0, q0, q1, thresh, ithresh, mask); 726466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora DoFilter6(&p2, &p1, &p0, &q0, &q1, &q2, &mask, hev_thresh); 727466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 728466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Store16x4(b, b + 8 * stride, stride, &p3, &p2, &p1, &p0); 729466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Store16x4(p, p + 8 * stride, stride, &q0, &q1, &q2, &q3); 730466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 731466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 732466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// on three inner edges 733466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void VFilter16iSSE2(uint8_t* p, int stride, 734466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int thresh, int ithresh, int hev_thresh) { 735466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int k; 736466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i mask; 737466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i t1, t2, p1, p0, q0, q1; 738466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 739466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora for (k = 3; k > 0; --k) { 740466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load p3, p2, p1, p0 741466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora LOAD_H_EDGES4(p, stride, t2, t1, p1, p0); 742466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF1(t2, t1, p1, p0, mask); 743466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 744466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora p += 4 * stride; 745466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 746466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load q0, q1, q2, q3 747466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora LOAD_H_EDGES4(p, stride, q0, q1, t1, t2); 748466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF2(t2, t1, q1, q0, mask); 749466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 750466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora COMPLEX_FL_MASK(p1, p0, q0, q1, thresh, ithresh, mask); 751466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora DoFilter4(&p1, &p0, &q0, &q1, &mask, hev_thresh); 752466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 753466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Store 754466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&p[-2 * stride], p1); 755466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&p[-1 * stride], p0); 756466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&p[0 * stride], q0); 757466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora _mm_storeu_si128((__m128i*)&p[1 * stride], q1); 758466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 759466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 760466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 761466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void HFilter16iSSE2(uint8_t* p, int stride, 762466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int thresh, int ithresh, int hev_thresh) { 763466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int k; 764466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora uint8_t* b; 765466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i mask; 766466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i t1, t2, p1, p0, q0, q1; 767466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 768466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora for (k = 3; k > 0; --k) { 769466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora b = p; 770466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Load16x4(b, b + 8 * stride, stride, &t2, &t1, &p1, &p0); // p3, p2, p1, p0 771466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF1(t2, t1, p1, p0, mask); 772466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 773466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora b += 4; // beginning of q0 774466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Load16x4(b, b + 8 * stride, stride, &q0, &q1, &t1, &t2); // q0, q1, q2, q3 775466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF2(t2, t1, q1, q0, mask); 776466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 777466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora COMPLEX_FL_MASK(p1, p0, q0, q1, thresh, ithresh, mask); 778466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora DoFilter4(&p1, &p0, &q0, &q1, &mask, hev_thresh); 779466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 780466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora b -= 2; // beginning of p1 781466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Store16x4(b, b + 8 * stride, stride, &p1, &p0, &q0, &q1); 782466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 783466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora p += 4; 784466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora } 785466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 786466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 787466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora// 8-pixels wide variant, for chroma filtering 788466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void VFilter8SSE2(uint8_t* u, uint8_t* v, int stride, 789466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int thresh, int ithresh, int hev_thresh) { 790466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i mask; 791466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i t1, p2, p1, p0, q0, q1, q2; 792466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 793466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load p3, p2, p1, p0 794466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora LOADUV_H_EDGES4(u - 4 * stride, v - 4 * stride, stride, t1, p2, p1, p0); 795466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF1(t1, p2, p1, p0, mask); 796466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 797466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load q0, q1, q2, q3 798466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora LOADUV_H_EDGES4(u, v, stride, q0, q1, q2, t1); 799466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF2(t1, q2, q1, q0, mask); 800466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 801466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora COMPLEX_FL_MASK(p1, p0, q0, q1, thresh, ithresh, mask); 802466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora DoFilter6(&p2, &p1, &p0, &q0, &q1, &q2, &mask, hev_thresh); 803466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 804466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Store 805466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora STOREUV(p2, u, v, -3 * stride); 806466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora STOREUV(p1, u, v, -2 * stride); 807466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora STOREUV(p0, u, v, -1 * stride); 808466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora STOREUV(q0, u, v, 0 * stride); 809466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora STOREUV(q1, u, v, 1 * stride); 810466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora STOREUV(q2, u, v, 2 * stride); 811466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 812466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 813466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void HFilter8SSE2(uint8_t* u, uint8_t* v, int stride, 814466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int thresh, int ithresh, int hev_thresh) { 815466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i mask; 816466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i p3, p2, p1, p0, q0, q1, q2, q3; 817466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 818466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora uint8_t* const tu = u - 4; 819466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora uint8_t* const tv = v - 4; 820466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Load16x4(tu, tv, stride, &p3, &p2, &p1, &p0); // p3, p2, p1, p0 821466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF1(p3, p2, p1, p0, mask); 822466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 823466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Load16x4(u, v, stride, &q0, &q1, &q2, &q3); // q0, q1, q2, q3 824466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF2(q3, q2, q1, q0, mask); 825466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 826466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora COMPLEX_FL_MASK(p1, p0, q0, q1, thresh, ithresh, mask); 827466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora DoFilter6(&p2, &p1, &p0, &q0, &q1, &q2, &mask, hev_thresh); 828466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 829466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Store16x4(tu, tv, stride, &p3, &p2, &p1, &p0); 830466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Store16x4(u, v, stride, &q0, &q1, &q2, &q3); 831466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 832466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 833466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void VFilter8iSSE2(uint8_t* u, uint8_t* v, int stride, 834466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int thresh, int ithresh, int hev_thresh) { 835466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i mask; 836466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i t1, t2, p1, p0, q0, q1; 837466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 838466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load p3, p2, p1, p0 839466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora LOADUV_H_EDGES4(u, v, stride, t2, t1, p1, p0); 840466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF1(t2, t1, p1, p0, mask); 841466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 842466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora u += 4 * stride; 843466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora v += 4 * stride; 844466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 845466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Load q0, q1, q2, q3 846466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora LOADUV_H_EDGES4(u, v, stride, q0, q1, t1, t2); 847466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF2(t2, t1, q1, q0, mask); 848466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 849466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora COMPLEX_FL_MASK(p1, p0, q0, q1, thresh, ithresh, mask); 850466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora DoFilter4(&p1, &p0, &q0, &q1, &mask, hev_thresh); 851466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 852466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora // Store 853466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora STOREUV(p1, u, v, -2 * stride); 854466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora STOREUV(p0, u, v, -1 * stride); 855466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora STOREUV(q0, u, v, 0 * stride); 856466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora STOREUV(q1, u, v, 1 * stride); 857466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 858466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 859466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arorastatic void HFilter8iSSE2(uint8_t* u, uint8_t* v, int stride, 860466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora int thresh, int ithresh, int hev_thresh) { 861466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i mask; 862466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora __m128i t1, t2, p1, p0, q0, q1; 863466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Load16x4(u, v, stride, &t2, &t1, &p1, &p0); // p3, p2, p1, p0 864466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF1(t2, t1, p1, p0, mask); 865466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 866466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora u += 4; // beginning of q0 867466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora v += 4; 868466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Load16x4(u, v, stride, &q0, &q1, &t1, &t2); // q0, q1, q2, q3 869466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora MAX_DIFF2(t2, t1, q1, q0, mask); 870466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 871466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora COMPLEX_FL_MASK(p1, p0, q0, q1, thresh, ithresh, mask); 872466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora DoFilter4(&p1, &p0, &q0, &q1, &mask, hev_thresh); 873466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 874466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora u -= 2; // beginning of p1 875466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora v -= 2; 876466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora Store16x4(u, v, stride, &p1, &p0, &q0, &q1); 877466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 878466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 8791e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#endif // WEBP_USE_SSE2 8801e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 8811e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora//------------------------------------------------------------------------------ 8821e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora// Entry point 8831e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora 884466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Aroraextern void VP8DspInitSSE2(void); 885466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 886466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Aroravoid VP8DspInitSSE2(void) { 8871e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#if defined(WEBP_USE_SSE2) 888466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8Transform = TransformSSE2; 889466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 890466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8VFilter16 = VFilter16SSE2; 891466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8HFilter16 = HFilter16SSE2; 892466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8VFilter8 = VFilter8SSE2; 893466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8HFilter8 = HFilter8SSE2; 894466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8VFilter16i = VFilter16iSSE2; 895466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8HFilter16i = HFilter16iSSE2; 896466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8VFilter8i = VFilter8iSSE2; 897466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8HFilter8i = HFilter8iSSE2; 898466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 899466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8SimpleVFilter16 = SimpleVFilter16SSE2; 900466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8SimpleHFilter16 = SimpleHFilter16SSE2; 901466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8SimpleVFilter16i = SimpleVFilter16iSSE2; 902466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora VP8SimpleHFilter16i = SimpleHFilter16iSSE2; 9031e7bf8805bd030c19924a5306837ecd72c295751Vikas Arora#endif // WEBP_USE_SSE2 904466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} 905466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora 906466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#if defined(__cplusplus) || defined(c_plusplus) 907466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora} // extern "C" 908466727975bcc57c0c5597bcd0747a2fe4777b303Vikas Arora#endif 909