用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;
    }
}