ref: b0612e5369801cbe8e1fa7f2d2dd69b2ba736170
dir: /external/SDL2/src/video/yuv2rgb/yuv_rgb_sse_func.h/
// Copyright 2016 Adrien Descamps // Distributed under BSD 3-Clause License /* You need to define the following macros before including this file: SSE_FUNCTION_NAME STD_FUNCTION_NAME YUV_FORMAT RGB_FORMAT */ /* You may define the following macro, which affects generated code: SSE_ALIGNED */ #ifdef SSE_ALIGNED /* Unaligned instructions seem faster, even on aligned data? */ /* #define LOAD_SI128 _mm_load_si128 #define SAVE_SI128 _mm_stream_si128 */ #define LOAD_SI128 _mm_loadu_si128 #define SAVE_SI128 _mm_storeu_si128 #else #define LOAD_SI128 _mm_loadu_si128 #define SAVE_SI128 _mm_storeu_si128 #endif #define UV2RGB_16(U,V,R1,G1,B1,R2,G2,B2) \ r_tmp = _mm_mullo_epi16(V, _mm_set1_epi16(param->v_r_factor)); \ g_tmp = _mm_add_epi16( \ _mm_mullo_epi16(U, _mm_set1_epi16(param->u_g_factor)), \ _mm_mullo_epi16(V, _mm_set1_epi16(param->v_g_factor))); \ b_tmp = _mm_mullo_epi16(U, _mm_set1_epi16(param->u_b_factor)); \ R1 = _mm_unpacklo_epi16(r_tmp, r_tmp); \ G1 = _mm_unpacklo_epi16(g_tmp, g_tmp); \ B1 = _mm_unpacklo_epi16(b_tmp, b_tmp); \ R2 = _mm_unpackhi_epi16(r_tmp, r_tmp); \ G2 = _mm_unpackhi_epi16(g_tmp, g_tmp); \ B2 = _mm_unpackhi_epi16(b_tmp, b_tmp); \ #define ADD_Y2RGB_16(Y1,Y2,R1,G1,B1,R2,G2,B2) \ Y1 = _mm_mullo_epi16(_mm_sub_epi16(Y1, _mm_set1_epi16(param->y_shift)), _mm_set1_epi16(param->y_factor)); \ Y2 = _mm_mullo_epi16(_mm_sub_epi16(Y2, _mm_set1_epi16(param->y_shift)), _mm_set1_epi16(param->y_factor)); \ \ R1 = _mm_srai_epi16(_mm_add_epi16(R1, Y1), PRECISION); \ G1 = _mm_srai_epi16(_mm_add_epi16(G1, Y1), PRECISION); \ B1 = _mm_srai_epi16(_mm_add_epi16(B1, Y1), PRECISION); \ R2 = _mm_srai_epi16(_mm_add_epi16(R2, Y2), PRECISION); \ G2 = _mm_srai_epi16(_mm_add_epi16(G2, Y2), PRECISION); \ B2 = _mm_srai_epi16(_mm_add_epi16(B2, Y2), PRECISION); \ #define PACK_RGB565_32(R1, R2, G1, G2, B1, B2, RGB1, RGB2, RGB3, RGB4) \ { \ __m128i red_mask, tmp1, tmp2, tmp3, tmp4; \ \ red_mask = _mm_set1_epi16((short)0xF800); \ RGB1 = _mm_and_si128(_mm_unpacklo_epi8(_mm_setzero_si128(), R1), red_mask); \ RGB2 = _mm_and_si128(_mm_unpackhi_epi8(_mm_setzero_si128(), R1), red_mask); \ RGB3 = _mm_and_si128(_mm_unpacklo_epi8(_mm_setzero_si128(), R2), red_mask); \ RGB4 = _mm_and_si128(_mm_unpackhi_epi8(_mm_setzero_si128(), R2), red_mask); \ tmp1 = _mm_slli_epi16(_mm_srli_epi16(_mm_unpacklo_epi8(G1, _mm_setzero_si128()), 2), 5); \ tmp2 = _mm_slli_epi16(_mm_srli_epi16(_mm_unpackhi_epi8(G1, _mm_setzero_si128()), 2), 5); \ tmp3 = _mm_slli_epi16(_mm_srli_epi16(_mm_unpacklo_epi8(G2, _mm_setzero_si128()), 2), 5); \ tmp4 = _mm_slli_epi16(_mm_srli_epi16(_mm_unpackhi_epi8(G2, _mm_setzero_si128()), 2), 5); \ RGB1 = _mm_or_si128(RGB1, tmp1); \ RGB2 = _mm_or_si128(RGB2, tmp2); \ RGB3 = _mm_or_si128(RGB3, tmp3); \ RGB4 = _mm_or_si128(RGB4, tmp4); \ tmp1 = _mm_srli_epi16(_mm_unpacklo_epi8(B1, _mm_setzero_si128()), 3); \ tmp2 = _mm_srli_epi16(_mm_unpackhi_epi8(B1, _mm_setzero_si128()), 3); \ tmp3 = _mm_srli_epi16(_mm_unpacklo_epi8(B2, _mm_setzero_si128()), 3); \ tmp4 = _mm_srli_epi16(_mm_unpackhi_epi8(B2, _mm_setzero_si128()), 3); \ RGB1 = _mm_or_si128(RGB1, tmp1); \ RGB2 = _mm_or_si128(RGB2, tmp2); \ RGB3 = _mm_or_si128(RGB3, tmp3); \ RGB4 = _mm_or_si128(RGB4, tmp4); \ } #define PACK_RGB24_32_STEP1(R1, R2, G1, G2, B1, B2, RGB1, RGB2, RGB3, RGB4, RGB5, RGB6) \ RGB1 = _mm_packus_epi16(_mm_and_si128(R1,_mm_set1_epi16(0xFF)), _mm_and_si128(R2,_mm_set1_epi16(0xFF))); \ RGB2 = _mm_packus_epi16(_mm_and_si128(G1,_mm_set1_epi16(0xFF)), _mm_and_si128(G2,_mm_set1_epi16(0xFF))); \ RGB3 = _mm_packus_epi16(_mm_and_si128(B1,_mm_set1_epi16(0xFF)), _mm_and_si128(B2,_mm_set1_epi16(0xFF))); \ RGB4 = _mm_packus_epi16(_mm_srli_epi16(R1,8), _mm_srli_epi16(R2,8)); \ RGB5 = _mm_packus_epi16(_mm_srli_epi16(G1,8), _mm_srli_epi16(G2,8)); \ RGB6 = _mm_packus_epi16(_mm_srli_epi16(B1,8), _mm_srli_epi16(B2,8)); \ #define PACK_RGB24_32_STEP2(R1, R2, G1, G2, B1, B2, RGB1, RGB2, RGB3, RGB4, RGB5, RGB6) \ R1 = _mm_packus_epi16(_mm_and_si128(RGB1,_mm_set1_epi16(0xFF)), _mm_and_si128(RGB2,_mm_set1_epi16(0xFF))); \ R2 = _mm_packus_epi16(_mm_and_si128(RGB3,_mm_set1_epi16(0xFF)), _mm_and_si128(RGB4,_mm_set1_epi16(0xFF))); \ G1 = _mm_packus_epi16(_mm_and_si128(RGB5,_mm_set1_epi16(0xFF)), _mm_and_si128(RGB6,_mm_set1_epi16(0xFF))); \ G2 = _mm_packus_epi16(_mm_srli_epi16(RGB1,8), _mm_srli_epi16(RGB2,8)); \ B1 = _mm_packus_epi16(_mm_srli_epi16(RGB3,8), _mm_srli_epi16(RGB4,8)); \ B2 = _mm_packus_epi16(_mm_srli_epi16(RGB5,8), _mm_srli_epi16(RGB6,8)); \ #define PACK_RGB24_32(R1, R2, G1, G2, B1, B2, RGB1, RGB2, RGB3, RGB4, RGB5, RGB6) \ PACK_RGB24_32_STEP1(R1, R2, G1, G2, B1, B2, RGB1, RGB2, RGB3, RGB4, RGB5, RGB6) \ PACK_RGB24_32_STEP2(R1, R2, G1, G2, B1, B2, RGB1, RGB2, RGB3, RGB4, RGB5, RGB6) \ PACK_RGB24_32_STEP1(R1, R2, G1, G2, B1, B2, RGB1, RGB2, RGB3, RGB4, RGB5, RGB6) \ PACK_RGB24_32_STEP2(R1, R2, G1, G2, B1, B2, RGB1, RGB2, RGB3, RGB4, RGB5, RGB6) \ PACK_RGB24_32_STEP1(R1, R2, G1, G2, B1, B2, RGB1, RGB2, RGB3, RGB4, RGB5, RGB6) \ #define PACK_RGBA_32(R1, R2, G1, G2, B1, B2, A1, A2, RGB1, RGB2, RGB3, RGB4, RGB5, RGB6, RGB7, RGB8) \ { \ __m128i lo_ab, hi_ab, lo_gr, hi_gr; \ \ lo_ab = _mm_unpacklo_epi8( A1, B1 ); \ hi_ab = _mm_unpackhi_epi8( A1, B1 ); \ lo_gr = _mm_unpacklo_epi8( G1, R1 ); \ hi_gr = _mm_unpackhi_epi8( G1, R1 ); \ RGB1 = _mm_unpacklo_epi16( lo_ab, lo_gr ); \ RGB2 = _mm_unpackhi_epi16( lo_ab, lo_gr ); \ RGB3 = _mm_unpacklo_epi16( hi_ab, hi_gr ); \ RGB4 = _mm_unpackhi_epi16( hi_ab, hi_gr ); \ \ lo_ab = _mm_unpacklo_epi8( A2, B2 ); \ hi_ab = _mm_unpackhi_epi8( A2, B2 ); \ lo_gr = _mm_unpacklo_epi8( G2, R2 ); \ hi_gr = _mm_unpackhi_epi8( G2, R2 ); \ RGB5 = _mm_unpacklo_epi16( lo_ab, lo_gr ); \ RGB6 = _mm_unpackhi_epi16( lo_ab, lo_gr ); \ RGB7 = _mm_unpacklo_epi16( hi_ab, hi_gr ); \ RGB8 = _mm_unpackhi_epi16( hi_ab, hi_gr ); \ } #if RGB_FORMAT == RGB_FORMAT_RGB565 #define PACK_PIXEL \ __m128i rgb_1, rgb_2, rgb_3, rgb_4, rgb_5, rgb_6, rgb_7, rgb_8; \ \ PACK_RGB565_32(r_8_11, r_8_12, g_8_11, g_8_12, b_8_11, b_8_12, rgb_1, rgb_2, rgb_3, rgb_4) \ \ PACK_RGB565_32(r_8_21, r_8_22, g_8_21, g_8_22, b_8_21, b_8_22, rgb_5, rgb_6, rgb_7, rgb_8) \ #elif RGB_FORMAT == RGB_FORMAT_RGB24 #define PACK_PIXEL \ __m128i rgb_1, rgb_2, rgb_3, rgb_4, rgb_5, rgb_6; \ __m128i rgb_7, rgb_8, rgb_9, rgb_10, rgb_11, rgb_12; \ \ PACK_RGB24_32(r_8_11, r_8_12, g_8_11, g_8_12, b_8_11, b_8_12, rgb_1, rgb_2, rgb_3, rgb_4, rgb_5, rgb_6) \ \ PACK_RGB24_32(r_8_21, r_8_22, g_8_21, g_8_22, b_8_21, b_8_22, rgb_7, rgb_8, rgb_9, rgb_10, rgb_11, rgb_12) \ #elif RGB_FORMAT == RGB_FORMAT_RGBA #define PACK_PIXEL \ __m128i rgb_1, rgb_2, rgb_3, rgb_4, rgb_5, rgb_6, rgb_7, rgb_8; \ __m128i rgb_9, rgb_10, rgb_11, rgb_12, rgb_13, rgb_14, rgb_15, rgb_16; \ __m128i a = _mm_set1_epi8((char)0xFF); \ \ PACK_RGBA_32(r_8_11, r_8_12, g_8_11, g_8_12, b_8_11, b_8_12, a, a, rgb_1, rgb_2, rgb_3, rgb_4, rgb_5, rgb_6, rgb_7, rgb_8) \ \ PACK_RGBA_32(r_8_21, r_8_22, g_8_21, g_8_22, b_8_21, b_8_22, a, a, rgb_9, rgb_10, rgb_11, rgb_12, rgb_13, rgb_14, rgb_15, rgb_16) \ #elif RGB_FORMAT == RGB_FORMAT_BGRA #define PACK_PIXEL \ __m128i rgb_1, rgb_2, rgb_3, rgb_4, rgb_5, rgb_6, rgb_7, rgb_8; \ __m128i rgb_9, rgb_10, rgb_11, rgb_12, rgb_13, rgb_14, rgb_15, rgb_16; \ __m128i a = _mm_set1_epi8((char)0xFF); \ \ PACK_RGBA_32(b_8_11, b_8_12, g_8_11, g_8_12, r_8_11, r_8_12, a, a, rgb_1, rgb_2, rgb_3, rgb_4, rgb_5, rgb_6, rgb_7, rgb_8) \ \ PACK_RGBA_32(b_8_21, b_8_22, g_8_21, g_8_22, r_8_21, r_8_22, a, a, rgb_9, rgb_10, rgb_11, rgb_12, rgb_13, rgb_14, rgb_15, rgb_16) \ #elif RGB_FORMAT == RGB_FORMAT_ARGB #define PACK_PIXEL \ __m128i rgb_1, rgb_2, rgb_3, rgb_4, rgb_5, rgb_6, rgb_7, rgb_8; \ __m128i rgb_9, rgb_10, rgb_11, rgb_12, rgb_13, rgb_14, rgb_15, rgb_16; \ __m128i a = _mm_set1_epi8((char)0xFF); \ \ PACK_RGBA_32(a, a, r_8_11, r_8_12, g_8_11, g_8_12, b_8_11, b_8_12, rgb_1, rgb_2, rgb_3, rgb_4, rgb_5, rgb_6, rgb_7, rgb_8) \ \ PACK_RGBA_32(a, a, r_8_21, r_8_22, g_8_21, g_8_22, b_8_21, b_8_22, rgb_9, rgb_10, rgb_11, rgb_12, rgb_13, rgb_14, rgb_15, rgb_16) \ #elif RGB_FORMAT == RGB_FORMAT_ABGR #define PACK_PIXEL \ __m128i rgb_1, rgb_2, rgb_3, rgb_4, rgb_5, rgb_6, rgb_7, rgb_8; \ __m128i rgb_9, rgb_10, rgb_11, rgb_12, rgb_13, rgb_14, rgb_15, rgb_16; \ __m128i a = _mm_set1_epi8((char)0xFF); \ \ PACK_RGBA_32(a, a, b_8_11, b_8_12, g_8_11, g_8_12, r_8_11, r_8_12, rgb_1, rgb_2, rgb_3, rgb_4, rgb_5, rgb_6, rgb_7, rgb_8) \ \ PACK_RGBA_32(a, a, b_8_21, b_8_22, g_8_21, g_8_22, r_8_21, r_8_22, rgb_9, rgb_10, rgb_11, rgb_12, rgb_13, rgb_14, rgb_15, rgb_16) \ #else #error PACK_PIXEL unimplemented #endif #if RGB_FORMAT == RGB_FORMAT_RGB565 #define SAVE_LINE1 \ SAVE_SI128((__m128i*)(rgb_ptr1), rgb_1); \ SAVE_SI128((__m128i*)(rgb_ptr1+16), rgb_2); \ SAVE_SI128((__m128i*)(rgb_ptr1+32), rgb_3); \ SAVE_SI128((__m128i*)(rgb_ptr1+48), rgb_4); \ #define SAVE_LINE2 \ SAVE_SI128((__m128i*)(rgb_ptr2), rgb_5); \ SAVE_SI128((__m128i*)(rgb_ptr2+16), rgb_6); \ SAVE_SI128((__m128i*)(rgb_ptr2+32), rgb_7); \ SAVE_SI128((__m128i*)(rgb_ptr2+48), rgb_8); \ #elif RGB_FORMAT == RGB_FORMAT_RGB24 #define SAVE_LINE1 \ SAVE_SI128((__m128i*)(rgb_ptr1), rgb_1); \ SAVE_SI128((__m128i*)(rgb_ptr1+16), rgb_2); \ SAVE_SI128((__m128i*)(rgb_ptr1+32), rgb_3); \ SAVE_SI128((__m128i*)(rgb_ptr1+48), rgb_4); \ SAVE_SI128((__m128i*)(rgb_ptr1+64), rgb_5); \ SAVE_SI128((__m128i*)(rgb_ptr1+80), rgb_6); \ #define SAVE_LINE2 \ SAVE_SI128((__m128i*)(rgb_ptr2), rgb_7); \ SAVE_SI128((__m128i*)(rgb_ptr2+16), rgb_8); \ SAVE_SI128((__m128i*)(rgb_ptr2+32), rgb_9); \ SAVE_SI128((__m128i*)(rgb_ptr2+48), rgb_10); \ SAVE_SI128((__m128i*)(rgb_ptr2+64), rgb_11); \ SAVE_SI128((__m128i*)(rgb_ptr2+80), rgb_12); \ #elif RGB_FORMAT == RGB_FORMAT_RGBA || RGB_FORMAT == RGB_FORMAT_BGRA || \ RGB_FORMAT == RGB_FORMAT_ARGB || RGB_FORMAT == RGB_FORMAT_ABGR #define SAVE_LINE1 \ SAVE_SI128((__m128i*)(rgb_ptr1), rgb_1); \ SAVE_SI128((__m128i*)(rgb_ptr1+16), rgb_2); \ SAVE_SI128((__m128i*)(rgb_ptr1+32), rgb_3); \ SAVE_SI128((__m128i*)(rgb_ptr1+48), rgb_4); \ SAVE_SI128((__m128i*)(rgb_ptr1+64), rgb_5); \ SAVE_SI128((__m128i*)(rgb_ptr1+80), rgb_6); \ SAVE_SI128((__m128i*)(rgb_ptr1+96), rgb_7); \ SAVE_SI128((__m128i*)(rgb_ptr1+112), rgb_8); \ #define SAVE_LINE2 \ SAVE_SI128((__m128i*)(rgb_ptr2), rgb_9); \ SAVE_SI128((__m128i*)(rgb_ptr2+16), rgb_10); \ SAVE_SI128((__m128i*)(rgb_ptr2+32), rgb_11); \ SAVE_SI128((__m128i*)(rgb_ptr2+48), rgb_12); \ SAVE_SI128((__m128i*)(rgb_ptr2+64), rgb_13); \ SAVE_SI128((__m128i*)(rgb_ptr2+80), rgb_14); \ SAVE_SI128((__m128i*)(rgb_ptr2+96), rgb_15); \ SAVE_SI128((__m128i*)(rgb_ptr2+112), rgb_16); \ #else #error SAVE_LINE unimplemented #endif #if YUV_FORMAT == YUV_FORMAT_420 #define READ_Y(y_ptr) \ y = LOAD_SI128((const __m128i*)(y_ptr)); \ #define READ_UV \ u = LOAD_SI128((const __m128i*)(u_ptr)); \ v = LOAD_SI128((const __m128i*)(v_ptr)); \ #elif YUV_FORMAT == YUV_FORMAT_422 #define READ_Y(y_ptr) \ { \ __m128i y1, y2; \ y1 = _mm_srli_epi16(_mm_slli_epi16(LOAD_SI128((const __m128i*)(y_ptr)), 8), 8); \ y2 = _mm_srli_epi16(_mm_slli_epi16(LOAD_SI128((const __m128i*)(y_ptr+16)), 8), 8); \ y = _mm_packus_epi16(y1, y2); \ } #define READ_UV \ { \ __m128i u1, u2, u3, u4, v1, v2, v3, v4; \ u1 = _mm_srli_epi32(_mm_slli_epi32(LOAD_SI128((const __m128i*)(u_ptr)), 24), 24); \ u2 = _mm_srli_epi32(_mm_slli_epi32(LOAD_SI128((const __m128i*)(u_ptr+16)), 24), 24); \ u3 = _mm_srli_epi32(_mm_slli_epi32(LOAD_SI128((const __m128i*)(u_ptr+32)), 24), 24); \ u4 = _mm_srli_epi32(_mm_slli_epi32(LOAD_SI128((const __m128i*)(u_ptr+48)), 24), 24); \ u = _mm_packus_epi16(_mm_packs_epi32(u1, u2), _mm_packs_epi32(u3, u4)); \ v1 = _mm_srli_epi32(_mm_slli_epi32(LOAD_SI128((const __m128i*)(v_ptr)), 24), 24); \ v2 = _mm_srli_epi32(_mm_slli_epi32(LOAD_SI128((const __m128i*)(v_ptr+16)), 24), 24); \ v3 = _mm_srli_epi32(_mm_slli_epi32(LOAD_SI128((const __m128i*)(v_ptr+32)), 24), 24); \ v4 = _mm_srli_epi32(_mm_slli_epi32(LOAD_SI128((const __m128i*)(v_ptr+48)), 24), 24); \ v = _mm_packus_epi16(_mm_packs_epi32(v1, v2), _mm_packs_epi32(v3, v4)); \ } #elif YUV_FORMAT == YUV_FORMAT_NV12 #define READ_Y(y_ptr) \ y = LOAD_SI128((const __m128i*)(y_ptr)); \ #define READ_UV \ { \ __m128i u1, u2, v1, v2; \ u1 = _mm_srli_epi16(_mm_slli_epi16(LOAD_SI128((const __m128i*)(u_ptr)), 8), 8); \ u2 = _mm_srli_epi16(_mm_slli_epi16(LOAD_SI128((const __m128i*)(u_ptr+16)), 8), 8); \ u = _mm_packus_epi16(u1, u2); \ v1 = _mm_srli_epi16(_mm_slli_epi16(LOAD_SI128((const __m128i*)(v_ptr)), 8), 8); \ v2 = _mm_srli_epi16(_mm_slli_epi16(LOAD_SI128((const __m128i*)(v_ptr+16)), 8), 8); \ v = _mm_packus_epi16(v1, v2); \ } #else #error READ_UV unimplemented #endif #define YUV2RGB_32 \ __m128i r_tmp, g_tmp, b_tmp; \ __m128i r_16_1, g_16_1, b_16_1, r_16_2, g_16_2, b_16_2; \ __m128i r_uv_16_1, g_uv_16_1, b_uv_16_1, r_uv_16_2, g_uv_16_2, b_uv_16_2; \ __m128i y_16_1, y_16_2; \ __m128i y, u, v, u_16, v_16; \ __m128i r_8_11, g_8_11, b_8_11, r_8_21, g_8_21, b_8_21; \ __m128i r_8_12, g_8_12, b_8_12, r_8_22, g_8_22, b_8_22; \ \ READ_UV \ \ /* process first 16 pixels of first line */\ u_16 = _mm_unpacklo_epi8(u, _mm_setzero_si128()); \ v_16 = _mm_unpacklo_epi8(v, _mm_setzero_si128()); \ u_16 = _mm_add_epi16(u_16, _mm_set1_epi16(-128)); \ v_16 = _mm_add_epi16(v_16, _mm_set1_epi16(-128)); \ \ UV2RGB_16(u_16, v_16, r_16_1, g_16_1, b_16_1, r_16_2, g_16_2, b_16_2) \ r_uv_16_1=r_16_1; g_uv_16_1=g_16_1; b_uv_16_1=b_16_1; \ r_uv_16_2=r_16_2; g_uv_16_2=g_16_2; b_uv_16_2=b_16_2; \ \ READ_Y(y_ptr1) \ y_16_1 = _mm_unpacklo_epi8(y, _mm_setzero_si128()); \ y_16_2 = _mm_unpackhi_epi8(y, _mm_setzero_si128()); \ \ ADD_Y2RGB_16(y_16_1, y_16_2, r_16_1, g_16_1, b_16_1, r_16_2, g_16_2, b_16_2) \ \ r_8_11 = _mm_packus_epi16(r_16_1, r_16_2); \ g_8_11 = _mm_packus_epi16(g_16_1, g_16_2); \ b_8_11 = _mm_packus_epi16(b_16_1, b_16_2); \ \ /* process first 16 pixels of second line */\ r_16_1=r_uv_16_1; g_16_1=g_uv_16_1; b_16_1=b_uv_16_1; \ r_16_2=r_uv_16_2; g_16_2=g_uv_16_2; b_16_2=b_uv_16_2; \ \ READ_Y(y_ptr2) \ y_16_1 = _mm_unpacklo_epi8(y, _mm_setzero_si128()); \ y_16_2 = _mm_unpackhi_epi8(y, _mm_setzero_si128()); \ \ ADD_Y2RGB_16(y_16_1, y_16_2, r_16_1, g_16_1, b_16_1, r_16_2, g_16_2, b_16_2) \ \ r_8_21 = _mm_packus_epi16(r_16_1, r_16_2); \ g_8_21 = _mm_packus_epi16(g_16_1, g_16_2); \ b_8_21 = _mm_packus_epi16(b_16_1, b_16_2); \ \ /* process last 16 pixels of first line */\ u_16 = _mm_unpackhi_epi8(u, _mm_setzero_si128()); \ v_16 = _mm_unpackhi_epi8(v, _mm_setzero_si128()); \ u_16 = _mm_add_epi16(u_16, _mm_set1_epi16(-128)); \ v_16 = _mm_add_epi16(v_16, _mm_set1_epi16(-128)); \ \ UV2RGB_16(u_16, v_16, r_16_1, g_16_1, b_16_1, r_16_2, g_16_2, b_16_2) \ r_uv_16_1=r_16_1; g_uv_16_1=g_16_1; b_uv_16_1=b_16_1; \ r_uv_16_2=r_16_2; g_uv_16_2=g_16_2; b_uv_16_2=b_16_2; \ \ READ_Y(y_ptr1+16*y_pixel_stride) \ y_16_1 = _mm_unpacklo_epi8(y, _mm_setzero_si128()); \ y_16_2 = _mm_unpackhi_epi8(y, _mm_setzero_si128()); \ \ ADD_Y2RGB_16(y_16_1, y_16_2, r_16_1, g_16_1, b_16_1, r_16_2, g_16_2, b_16_2) \ \ r_8_12 = _mm_packus_epi16(r_16_1, r_16_2); \ g_8_12 = _mm_packus_epi16(g_16_1, g_16_2); \ b_8_12 = _mm_packus_epi16(b_16_1, b_16_2); \ \ /* process last 16 pixels of second line */\ r_16_1=r_uv_16_1; g_16_1=g_uv_16_1; b_16_1=b_uv_16_1; \ r_16_2=r_uv_16_2; g_16_2=g_uv_16_2; b_16_2=b_uv_16_2; \ \ READ_Y(y_ptr2+16*y_pixel_stride) \ y_16_1 = _mm_unpacklo_epi8(y, _mm_setzero_si128()); \ y_16_2 = _mm_unpackhi_epi8(y, _mm_setzero_si128()); \ \ ADD_Y2RGB_16(y_16_1, y_16_2, r_16_1, g_16_1, b_16_1, r_16_2, g_16_2, b_16_2) \ \ r_8_22 = _mm_packus_epi16(r_16_1, r_16_2); \ g_8_22 = _mm_packus_epi16(g_16_1, g_16_2); \ b_8_22 = _mm_packus_epi16(b_16_1, b_16_2); \ \ void SSE_FUNCTION_NAME(uint32_t width, uint32_t height, const uint8_t *Y, const uint8_t *U, const uint8_t *V, uint32_t Y_stride, uint32_t UV_stride, uint8_t *RGB, uint32_t RGB_stride, YCbCrType yuv_type) { const YUV2RGBParam *const param = &(YUV2RGB[yuv_type]); #if YUV_FORMAT == YUV_FORMAT_420 const int y_pixel_stride = 1; const int uv_pixel_stride = 1; const int uv_x_sample_interval = 2; const int uv_y_sample_interval = 2; #elif YUV_FORMAT == YUV_FORMAT_422 const int y_pixel_stride = 2; const int uv_pixel_stride = 4; const int uv_x_sample_interval = 2; const int uv_y_sample_interval = 1; #elif YUV_FORMAT == YUV_FORMAT_NV12 const int y_pixel_stride = 1; const int uv_pixel_stride = 2; const int uv_x_sample_interval = 2; const int uv_y_sample_interval = 2; #endif #if RGB_FORMAT == RGB_FORMAT_RGB565 const int rgb_pixel_stride = 2; #elif RGB_FORMAT == RGB_FORMAT_RGB24 const int rgb_pixel_stride = 3; #elif RGB_FORMAT == RGB_FORMAT_RGBA || RGB_FORMAT == RGB_FORMAT_BGRA || \ RGB_FORMAT == RGB_FORMAT_ARGB || RGB_FORMAT == RGB_FORMAT_ABGR const int rgb_pixel_stride = 4; #else #error Unknown RGB pixel size #endif if (width >= 32) { uint32_t xpos, ypos; for(ypos=0; ypos<(height-(uv_y_sample_interval-1)); ypos+=uv_y_sample_interval) { const uint8_t *y_ptr1=Y+ypos*Y_stride, *y_ptr2=Y+(ypos+1)*Y_stride, *u_ptr=U+(ypos/uv_y_sample_interval)*UV_stride, *v_ptr=V+(ypos/uv_y_sample_interval)*UV_stride; uint8_t *rgb_ptr1=RGB+ypos*RGB_stride, *rgb_ptr2=RGB+(ypos+1)*RGB_stride; for(xpos=0; xpos<(width-31); xpos+=32) { YUV2RGB_32 { PACK_PIXEL SAVE_LINE1 if (uv_y_sample_interval > 1) { SAVE_LINE2 } } y_ptr1+=32*y_pixel_stride; y_ptr2+=32*y_pixel_stride; u_ptr+=32*uv_pixel_stride/uv_x_sample_interval; v_ptr+=32*uv_pixel_stride/uv_x_sample_interval; rgb_ptr1+=32*rgb_pixel_stride; rgb_ptr2+=32*rgb_pixel_stride; } } /* Catch the last line, if needed */ if (uv_y_sample_interval == 2 && ypos == (height-1)) { const uint8_t *y_ptr=Y+ypos*Y_stride, *u_ptr=U+(ypos/uv_y_sample_interval)*UV_stride, *v_ptr=V+(ypos/uv_y_sample_interval)*UV_stride; uint8_t *rgb_ptr=RGB+ypos*RGB_stride; STD_FUNCTION_NAME(width, 1, y_ptr, u_ptr, v_ptr, Y_stride, UV_stride, rgb_ptr, RGB_stride, yuv_type); } } /* Catch the right column, if needed */ { int converted = (width & ~31); if (converted != width) { const uint8_t *y_ptr=Y+converted*y_pixel_stride, *u_ptr=U+converted*uv_pixel_stride/uv_x_sample_interval, *v_ptr=V+converted*uv_pixel_stride/uv_x_sample_interval; uint8_t *rgb_ptr=RGB+converted*rgb_pixel_stride; STD_FUNCTION_NAME(width-converted, height, y_ptr, u_ptr, v_ptr, Y_stride, UV_stride, rgb_ptr, RGB_stride, yuv_type); } } } #undef SSE_FUNCTION_NAME #undef STD_FUNCTION_NAME #undef YUV_FORMAT #undef RGB_FORMAT #undef SSE_ALIGNED #undef LOAD_SI128 #undef SAVE_SI128 #undef UV2RGB_16 #undef ADD_Y2RGB_16 #undef PACK_RGB24_32_STEP1 #undef PACK_RGB24_32_STEP2 #undef PACK_RGB24_32 #undef PACK_RGBA_32 #undef PACK_PIXEL #undef SAVE_LINE1 #undef SAVE_LINE2 #undef READ_Y #undef READ_UV #undef YUV2RGB_32