YUV420圖像旋轉90算法的優化

        在做android攝像頭捕獲時，發現從android攝像頭出來的原始視是逆時針旋轉了90度的，所以須要把它順時針旋轉90。android視頻支持的是NV21格式，它是一種YUV420的格式。當然，始果你用的是android sdk的話，當中image就提供這個能力。可是我是在ndk下開發，沒有找到對應的功能（假設你知道請告訴我）。

         我本想用開源的圖像處理庫（opencv）做旋轉，可是opencv僅僅能處理bmp的圖像。這種話，須要先把NV21轉換成BMP32。然后再做旋轉。所以要操作兩次，效率肯定低。最后也沒找到好的方法(假設你知道一次用opencv做YUV420格式的旋轉，請你告訴我)。所以僅僅有自己寫一個。

先從網上搜索了一個：

void CTool::YUV420spRotate90(uchar *des, uchar *src,int width,int height)
{
    int wh = width * height;
    //旋轉Y
    int k = 0;
    for(int i = 0; i < width; i++) {
        for(int j = 0; j < height; j++)
        {
            des[k] = src[width * j + i];
            k++;
        }
    }

    for(int i = 0; i < width; i += 2) {
        for(int j = 0; j < height / 2; j++)
        {
            des[k] = src[wh+ width * j + i];
            des[k+1] = src[wh + width * j + i + 1];
            k+=2;
        }
    }
}

在android上執行了一下，發現視頻很卡。所以僅僅有自己優化了。這段代碼里有乘法，優化的重點就是去掉乘法用加法取代：

void CTool::YUV420spRotate90(uchar *dst, const uchar *src, int srcWidth, int srcHeight)
{
    static int nWidth = 0, nHeight = 0;
    static int wh = 0;
    static int uvHeight = 0;
    if(srcWidth != nWidth || srcHeight != nHeight)
    {
        nWidth = srcWidth;
        nHeight = srcHeight;
        wh = srcWidth * srcHeight;
        uvHeight = srcHeight >> 1;//uvHeight = height / 2
    }

    //旋轉Y
    int k = 0;
    for(int i = 0; i < srcWidth; i++) {
        int nPos = 0;
        for(int j = 0; j < srcHeight; j++) {
            dst[k] = src[nPos + i];
            k++;
            nPos += srcWidth;
        }
    }

    for(int i = 0; i < srcWidth; i+=2){
        int nPos = wh;
        for(int j = 0; j < uvHeight; j++) {
            dst[k] = src[nPos + i];
            dst[k + 1] = src[nPos + i + 1];
            k += 2;
            nPos += srcWidth;
        }
    }
    return;
}

逆時針旋轉90度：

void CTool::YUV420spRotateNegative90(uchar *dst, const uchar *src, int srcWidth, int height)
{
    static int nWidth = 0, nHeight = 0;
    static int wh = 0;
    static int uvHeight = 0;
    if(srcWidth != nWidth || height != nHeight)
    {
        nWidth = srcWidth;
        nHeight = height;
        wh = srcWidth * height;
        uvHeight = height >> 1;//uvHeight = height / 2
    }

    //旋轉Y
    int k = 0;
    for(int i = 0; i < srcWidth; i++){
        int nPos = srcWidth - 1;
        for(int j = 0; j < height; j++)
        {
            dst[k] = src[nPos - i];
            k++;
            nPos += srcWidth;
        }
    }

    for(int i = 0; i < srcWidth; i+=2){
        int nPos = wh + srcWidth - 1;
        for(int j = 0; j < uvHeight; j++) {
            dst[k] = src[nPos - i - 1];
            dst[k + 1] = src[nPos - i];
            k += 2;
            nPos += srcWidth;
        }
    }

    return;
}

 編譯執行，捕獲視頻顯示很流暢 :)