用SSE2实现NV12转RGB32

最近要用到Intel的media SDK播放H264视频，发现intel的SDK转码出来的帧缓冲是NV12像素格式的，需要自己转换成RGB32格式。虽然这种代码网上找找一大堆，但还是自己动动手吧。

google了一下转换公式：

C = Y - 16
D = U - 128
E = V - 128
R = clip(round(1.164383 * C + 1.596027 * E))
G = clip(round(1.164383 * C - (0.391762 * D) - (0.812968 * E)))
B = clip(round(1.164383 * C + 2.017232 * D))

研究了一个小时用SSE2搞定，代码一次调试用过，心情大好，把核心代码贴上来。

 

__m128i c0  =  _mm_setzero_si128();
__m128i c128  =  _mm_set1_epi16( 128 );
__m128i c128_32  =  _mm_set1_epi32( 128 );
__m128i c16  =  _mm_set1_epi16( 16 );
__m128i c255  =  _mm_set1_epi16( 255 );
__m128i c_1_1596  =  _mm_set1_epi32( 0x199012a );
__m128i c_1_2017  =  _mm_set1_epi32( 0x204012a );
__m128i c_0_392  =  _mm_set1_epi32(0xff9c0000) ;
__m128i c_1_813 =  _mm_set1_epi32(0xff30012a) ;

for ( int  y  =   0 ; y  <  src .Height; y ++ )
{
    BYTE *  dest  =  (BYTE * )data.Scan0  +  data.Stride  *  y;
    BYTE *  srcY  =  src.Y  +  src.Pitch  *  y;
    BYTE *  srcUV  =  src.UV  +  src.Pitch  *  (y  /   2 );
     for ( int  x  =   0 ; x  <  src .Width; x  +=   4 )
    {
         // Y0Y1Y2Y30000 - 16
        __m128i Ymm  =  _mm_sub_epi16(_mm_unpacklo_epi8(_mm_cvtsi32_si128( * ( int * )(srcY  +  x)), c0), c16);
         // U0V0U2V20000 - 128
        __m128i UVmm  =  _mm_sub_epi16(_mm_unpacklo_epi8(_mm_cvtsi32_si128( * ( int * )(srcUV  +  x)), c0), c128);
         // U0U0U2U20000
        __m128i Umm  =  _mm_shufflelo_epi16(UVmm, _MM_SHUFFLE( 2 , 2 , 0 , 0 ));
         // V0V0V2V20000
        __m128i Vmm  =  _mm_shufflelo_epi16(UVmm, _MM_SHUFFLE( 3 , 3 , 1 , 1 ));
         // Y0V0Y1V0Y2V2Y3V2
        __m128i YVmm  =  _mm_unpacklo_epi16(Ymm, Vmm);
         // Y0U0Y1U0Y2U2Y3U2
        __m128i YUmm  =  _mm_unpacklo_epi16(Ymm, Umm);

        __m128i Rmm  =  _mm_srai_epi32(_mm_add_epi32(_mm_madd_epi16(YVmm, c_1_1596), c128_32),  8 );
        __m128i Bmm  =  _mm_srai_epi32(_mm_add_epi32(_mm_madd_epi16(YUmm, c_1_2017), c128_32),  8 );
        __m128i Gmm  =  _mm_srai_epi32(_mm_add_epi32(_mm_add_epi32(_mm_madd_epi16(YVmm, c_1_813), _mm_madd_epi16(YUmm, c_0_392)), c128_32),  8 );
        Rmm  =  _mm_slli_epi32(_mm_and_si128(Rmm, _mm_cmpgt_epi32(Rmm, c0)),  16 );
        Bmm  =  _mm_and_si128(Bmm, _mm_cmpgt_epi32(Bmm, c0));
        Gmm  =  _mm_slli_epi32(_mm_min_epi16(_mm_and_si128(Gmm, _mm_cmpgt_epi32(Gmm, c0)), c255),  8 );
         * (__m128i * )dest  =  _mm_or_si128(_mm_min_epi16(_mm_or_si128(Rmm, Bmm), c255), Gmm);
        dest  +=   16 ;
    }
}