RGB24轉yuv420 高效率 且顏色沒有失真

前面使用dshow獲取到的屏幕截圖是一段rgb24的數據，需要將其轉換成yuv420格式才能進行編碼等操作。

網上找了很多資料，發現能用的並不多，這裏總結一下。

1.雷大神寫的 bmp轉yuv：

http://blog.csdn.net/leixiaohua1020/article/details/13506099

這個是讀取bmp文件然後轉換成Yuv的，經過簡單修改就可以將輸入換成我們前面獲得的rbg數據，運行後發現轉換的效果很好，但是效率實在太低了，非常耗時。

由於本人對圖片的組成和原理方面不懂，想要優化這個似乎太難了，只能繼續上網找別的算法。

2.

http://blog.csdn.net/wishfly/article/details/50766370

這個效率挺高的，可惜轉換出的圖片顏色偏黃。。。

代碼就幾行 很簡單。。  然而並不知道怎麼改。。

3.

http://blog.csdn.net/wuzongman/article/details/8126012

果然功夫不負有心人，這個可以用。

轉換完之後發現圖像居然是垂直倒立的，於是參考了雷大神的代碼，從中找到了倒轉的代碼，將2者結合結果非常完美。

整合後的代碼代碼如下：

#include "rgb2yuv.h"

Rgb2YUV::Rgb2YUV()
{
    table_init();
}

//表的初始化
void Rgb2YUV::table_init()
{
    int i;
    for(i = 0; i < COLORSIZE; i++)
    {
        Y_R[i] = (i * 1224) >> 12; //Y對應的查表數組0.2988
        Y_G[i] = (i * 2404) >> 12;  //0.5869
        Y_B[i] = (i * 469)  >> 12; //0.1162
        U_R[i] = (i * 692)  >> 12; //U對應的查表數組0.1688
        U_G[i] = (i * 1356) >> 12;  //0.3312
        U_B[i] = i /*(* 2048) */>> 1; //0.5
     //   V_R[i] = (i * 2048) >> 12; ///V對應的查表數組
        V_G[i] = (i * 1731) >> 12;  //0.4184
        V_B[i] = (i * 334)  >> 12; //0.0816
    }
}

void Rgb2YUV::RGB2YUV420(uint8_t *rgbBufIn, uint8_t *yuvBufOut , int nWidth, int nHeight)
{
    int pix = 0;
    int pixP4;

    RGB *in = (RGB *)rgbBufIn; //需要計算的原始數據
//    unsigned char out[XSIZE*YSIZE * 3/2]; //計算後的結果

    int IMGSIZE = nWidth * nHeight;

    for(int y = 0; y < nHeight; y++) //line
    {
        for(int x=0;x < nWidth;x++)//pixf
        {

//           RGB rgbByte = in[pix]; (這樣取的數據經過轉換後是一個垂直倒立的圖像)
            RGB rgbByte = in[(nHeight-y-1)*nWidth+x]; //取得垂直方向上鏡像的位置即可解決倒立問題

           //首先執行顏色互換 -- 沒有這個的話  得到的YUV圖像顏色不對
           uint8_t temp = rgbByte.r; //順序調整
           rgbByte.r = rgbByte.b;
           rgbByte.b = temp;

            int     i = Y_R[rgbByte.r] + Y_G[rgbByte.g] + Y_B[rgbByte.b];
            yuvBufOut[pix]= i; //Y
            if((x%2==1)&&(y%2==1))
            {
                pixP4=(nWidth>>1) *(y>>1) + (x>>1);
                i=U_B[rgbByte.b] - U_R[rgbByte.r] - U_G[rgbByte.g]+128;

                yuvBufOut[pixP4+IMGSIZE]     = i;
                /*+  U_B[in[pix+1].b] - U_R[in[pix+1].r] - U_G[in[pix+1].g]
                +U_B[in[pix+XSIZE].b] - U_R[in[pix+XSIZE].r] - U_G[in[pix+XSIZE].g]
                +U_B[in[pix+1+XSIZE].b] - U_R[in[pix+1+XSIZE].r] - U_G[in[pix+1+XSIZE].g] )/4*/
                //U
                i=U_B[rgbByte.r] - V_G[rgbByte.g] - V_B[rgbByte.b]+128;

                yuvBufOut[pixP4 + 5 * IMGSIZE /4] = i;
                /*+U_B[in[pix+1].r] - V_G[in[pix+1].g] - V_B[in[pix+1].b]
                +U_B[in[pix+XSIZE].r] - V_G[in[pix+XSIZE].g] - V_B[in[pix+XSIZE].b]
                +U_B[in[pix+1+XSIZE].r] - V_G[in[pix+1+XSIZE].g] - V_B[in[pix+1+XSIZE].b])/4*/
                //V
            }

            pix++;
        }

    }
}

完整代碼下載地址：

http://download.csdn.net/detail/qq214517703/9612782