FFmpeg之yuv格式转换(二十)

####先介绍下转换中可能用到的一些函数:其中包括RGB和YUV的互换函数,源于网上流传较多的一些公式,以及经常重复用到的对齐函数

// RGB、YUV互换
int rgbToY(int r, int g, int b) {
    return (((66 * r + 129 * g + 25 * b + 128) >> 8) + 16);
}

int rgbToU(int r, int g, int b) {

    return (((-38 * r - 74 * g + 112 * b + 128) >> 8) + 128);
}

int rgbToV(int r, int g, int b) {
    return (((112 * r - 94 * g - 18 * b + 128) >> 8) + 128);
}

int yuvToR(int y, int u, int v) {
    return (int) ((y & 0xFF) + 1.4075 * ((v & 0xFF) - 128));
}

int yuvToG(int y, int u, int v) {
    return (int) ((y & 0xFF) - 0.3455 * ((u & 0xFF) - 128) - 0.7169 * ((v & 0xFF) - 128));
}

int yuvToB(int y, int u, int v) {
    return (int) ((y & 0xFF) + 1.779 * ((u & 0xFF) - 128));
}

int alignIntToByte(int c) {
    return c & 0xFF;
}

1. RGBA32转换为BGR24

举个例子,对于4x2的图片,RGBA32格式内容为:

R1 G1 B1 A1   R2 G2 B2 A2   R3 G3 B3 A3   R4 G4 B4 A4

R5 G5 B5 A5   R6 G6 B6 A6   R7 G7 B7 A7  R8 G8 B8 A8

那么若需要转化为BGR24,内容将变成:

B1 G1 R1   B2 G2 R2   B3 G3 R3   B4 G4 R4

B5 G5 R5   B6 G6 R6   B7 G7 R7   B8 G8 R8

BGR24内容为3个byte一组,RGBA32内容为4个byte一组。因此,对于第一组RGBA32(R1 G1 B1 A1)和第一组BGR24(B1 G1 R1),其对应关系为:

bgr24[0] = Rgba32[2];
bgr24[1] = Rgba32[1];
bgr24[2] = Rgba32[0];

对应的转换代码:

void rgba32ToBgr24(char *rgba32, char *bgr24, int width, int height) {
    int groupNum = width * height;
    int bgr24Index = 0;
    int rgba32Index = 0;
    for (int i = 0; i < groupNum; i++) {
        *(bgr24 + bgr24Index) = *(rgba32 + rgba32Index + 2);
        *(bgr24 + bgr24Index + 1) = *(rgba32 + rgba32Index + 1);
        *(bgr24 + bgr24Index + 2) = *(rgba32 + rgba32Index);
        bgr24Index += 3;
        rgba32Index += 4;
    }
}

2.BGR24转换为NV21

BGR24的每一组 B G R 数据对应了一个像素点,而NV21是一种YUV数据,YUV数据的每个Y会和对应的UV组成一个像素点,NV21的共用关系是每4个Y(例如下标为:0,1,width,width+1的四个Y)会共用一组UV(对应下标:U:width x height + 1,V:width x height)。所以对于Y而言,需要计算每一组BGR对应的Y值;但是对于NV21UV而言,UV的大小都是各只有Y的一半,所以我们可以在宽度下标和高度下标都为偶数时计算一组UV

void bgr24ToNv21(char *bgr24, char *nv21, int width, int height) {
    int yIndex = 0;
    int uvIndex = width * height;
    int bgrIndex = 0;
    int nv21Length = width * height * 3 / 2;
    for (int j = 0; j < height; ++j) {
        for (int i = 0; i < width; ++i) {
            int b = bgr24[bgrIndex++];
            int g = bgr24[bgrIndex++];
            int r = bgr24[bgrIndex++];
            b = alignIntToByte(b);
            g = alignIntToByte(g);
            r = alignIntToByte(r);
            int y = rgbToY(r, g, b);
            nv21[yIndex++] = static_cast<char>(alignIntToByte(y));
            if ((j & 1) == 0 && ((bgrIndex >> 2) & 1) == 0 && uvIndex < nv21Length - 2) {
                int u = rgbToU(r, g, b);
                int v = rgbToV(r, g, b);
                nv21[uvIndex++] = static_cast<char>(alignIntToByte(v));
                nv21[uvIndex++] = static_cast<char>(alignIntToByte(u));
            }
        }
    }
}

3. NV21转换为BGR24

这是上一个介绍的逆过程,在BGR24转换为NV21时,宽度下标和高度下标都为偶数才进行计算UV;那么若要将NV21转换为BGR24,则我们需要对每组UV进行一次复用。

void nv21ToBgr24(char *nv21, char *bgr24, int width, int height) {
    int bgrLineSize = width * 3;
    //偶数行的bgr数据下标
    int evenLineBgrIndex = 0;
    //奇数行的bgr数据下标
    int oddLineBgrIndex = bgrLineSize;
    //当前一行y数据最左边的下标
    int yLineStart = 0;
    //uv数据的下标
    int uvIndex = width * height;
    //由于NV21的共用关系,每2行做一次转换
    for (int i = 0; i < height; i += 2) {
        for (int widthOffset = 0; widthOffset < width; widthOffset++) {
            char v = nv21[uvIndex];
            char u = nv21[uvIndex + 1];
            char yEven = nv21[yLineStart + widthOffset];
            char yOdd = nv21[yLineStart + width + widthOffset];
            int r, g, b;
            //偶数行YUV转RGB
            r = alignIntToByte(yuvToR(yEven, u, v));
            g = alignIntToByte(yuvToG(yEven, u, v));
            b = alignIntToByte(yuvToB(yEven, u, v));

            bgr24[evenLineBgrIndex++] = (char) b;
            bgr24[evenLineBgrIndex++] = (char) g;
            bgr24[evenLineBgrIndex++] = (char) r;

            //奇数行YUV转RGB
            r = alignIntToByte(yuvToR(yOdd, u, v));
            g = alignIntToByte(yuvToG(yOdd, u, v));
            b = alignIntToByte(yuvToB(yOdd, u, v));

            bgr24[oddLineBgrIndex++] = (char) b;
            bgr24[oddLineBgrIndex++] = (char) g;
            bgr24[oddLineBgrIndex++] = (char) r;
            //每两个y将uv下标增1
            if ((widthOffset & 1) == 1) {
                uvIndex += 2;
            }
        }
        //由于在内层循环中已经做过width * 3次自增,所以外层循环中只需要增加一行
        evenLineBgrIndex += bgrLineSize;
        oddLineBgrIndex += bgrLineSize;
        //y增2行
        yLineStart += (width << 1);
    }
}

4. NV12NV21的互换

NV21NV12只是U与V的数据位置不同,只需要替换U和V的位置即可。因此,将NV21转换为NV12的代码同样适用于NV12转换为NV21。可参考如下代码:

void nv21ToNv12(char *nv21, char *nv12, int width, int height) {
    int ySize = width * height;
    int totalSize = width * height * 3 / 2;
//复制Y
    memcpy(nv12, nv21, ySize);
//UV互换
    for (int uvIndex = ySize; uvIndex < totalSize; uvIndex += 2) {
        *(nv12 + uvIndex) = *(nv21 + uvIndex + 1);
        *(nv12 + uvIndex + 1) = *(nv21 + uvIndex);
    }
}

5. NV21YV12

NV21转化为YV12的过程主要是将其UV数据的交叉排序修改为连续排序。可参考如下代码:

void nv21ToYv12(char *nv21, char *yv12, int width, int height) {
    int ySize = width * height;
    int totalSize = width * height * 3 / 2;
    int i420UIndex = ySize;
    int i420VIndex = ySize * 5 / 4;
    //复制y
    memcpy(yv12, nv21, ySize);
    //复制uv
    for (int uvIndex = ySize; uvIndex < totalSize; uvIndex += 2) {
        *(yv12 + i420UIndex++) = *(nv21 + uvIndex);
        *(yv12 + i420VIndex++) = *(nv21 + uvIndex + 1);
    }
}

6. YUYVNV12

YUYV格式中,两个Y共用一组UV,而NV12是四个Y共用一组UV,因此,这是一个YUV422YUV420的过程,需要舍弃一半的UV,需要注意的是,每一组YUYV是2个像素。可参考如下代码:

void yuyvToNv12(char *yuyv, char *nv12, int width, int height) {
    int ySize = width * height;
    int lineDataSize = width * 2;
    char *nv12UV = nv12 + ySize;
    for (int i = 0; i < height; i++, yuyv += lineDataSize) {
        if ((i & 1) == 0) {
            for (int lineOffset = 0; lineOffset < lineDataSize; lineOffset += 4) {
                //拷贝Y
                *nv12++ = *(yuyv + lineOffset);
                *nv12++ = *(yuyv + lineOffset + 2);
                //拷贝UV
                *nv12UV++ = *(yuyv + lineOffset + 1);
                *nv12UV++ = *(yuyv + lineOffset + 3);
            }
        } else {
            for (int lineOffset = 0; lineOffset < lineDataSize; lineOffset += 4) {
                //拷贝Y
                *nv12++ = *(yuyv + lineOffset);
                *nv12++ = *(yuyv + lineOffset + 2);
            }
        }
    }
}

7. I420YV12的互换

I420YV12只是UV的数据位置不同,因此,I420转换为YV12的代码同样适用于YV12转换为I420。可参考如下代码:

void i420ToYv12(char *i420, char *yv12, int width, int height) {
    int ySize = width * height;
    int uSize = ySize / 4;
    int vSize = uSize;
    //复制Y
    memcpy(yv12, i420, ySize);
    //UV互换
    memcpy(yv12 + ySize + uSize, i420 + ySize, uSize);
    memcpy(yv12 + ySize, i420 + ySize + vSize, vSize);
}

8. I420转换为YUYV

I420YUYV相比,I420UV只有YUYV的一半,这是一个YUV420YUV422的过程,缺损的数据只能通过复用UV弥补。

void i420ToYuyv(char *i420, char *yuyv, int width, int height) {
    int yuyvLineSize = width * 2;
    int i420YIndex = 0;
    int i420UIndex = width * height;
    int i420VIndex = width * height * 5 / 4;
    int yuyvLineStart = 0;
    for (int i = 0; i < height; i += 2) {
        for (int lineOffset = 0; lineOffset < yuyvLineSize; lineOffset += 4) {
            char u = *(i420 + i420UIndex++);
            char v = *(i420 + i420VIndex++);

            //偶数行数据赋值
            int yuyvOffset = yuyvLineStart + lineOffset;
            *(yuyv + yuyvOffset) = *(i420 + i420YIndex);
            *(yuyv + yuyvOffset + 1) = u;
            *(yuyv + yuyvOffset + 2) = *(i420 + i420YIndex + 1);
            *(yuyv + yuyvOffset + 3) = v;

            //奇数行数据赋值
            int yuyvNextLineOffset = yuyvLineStart + yuyvLineSize + lineOffset;
            *(yuyv + yuyvNextLineOffset) = *(i420 + i420YIndex + width);
            *(yuyv + yuyvNextLineOffset + 1) = u;
            *(yuyv + yuyvNextLineOffset + 2) = *(i420 + i420YIndex + width + 1);
            *(yuyv + yuyvNextLineOffset + 3) = v;

            i420YIndex += 2;
        }
        i420YIndex += width;
        yuyvLineStart += (width << 2);
    }
}

9. I420转换为NV12

I420转换为NV12就是把各自连续的UV修改为交叉存储,示例代码如下

void i420ToNv12(char *i420, char *nv12, int width, int height) {
    int ySize = width * height;
    int totalSize = width * height * 3 / 2;
    int i420UIndex = ySize;
    int i420VIndex = ySize * 5 / 4;
    //复制y
    memcpy(nv12, i420, ySize);
    //复制uv
    for (int uvIndex = ySize; uvIndex < totalSize; uvIndex += 2) {
        *(nv12 + uvIndex) = *(i420 + i420UIndex++);
        *(nv12 + uvIndex + 1) = *(i420 + i420VIndex++);
    }
}

10. I420转换为NV21

I420转换为NV12就是把各自连续的UV修改为交叉存储,并且替换UV的位置。示例代码如下

void i420ToNv21(char *i420, char *nv21, int width, int height) {
    int ySize = width * height;
    int totalSize = width * height * 3 / 2;
    int i420UIndex = ySize;
    int i420VIndex = ySize * 5 / 4;
    //复制y
    memcpy(nv21, i420, ySize);
    //复制uv
    for (int uvIndex = ySize; uvIndex < totalSize; uvIndex += 2) {
        *(nv21 + uvIndex + 1) = *(i420 + i420UIndex++);
        *(nv21 + uvIndex) = *(i420 + i420VIndex++);
    }
}
已标记关键词 清除标记
相关推荐
【为什么还需要学习C++?】 你是否接触很多语言,但从来没有了解过编程语言的本质? 你是否想成为一名资深开发人员,想开发别人做不了的高性能程序? 你是否经常想要窥探大型企业级开发工程的思路,但苦于没有基础只能望洋兴叹?   那么C++就是你个人能力提升,职业之路进阶的不二之选。 【课程特色】 1.课程共19大章节,239课时内容,涵盖数据结构、函数、类、指针、标准库全部知识体系。 2.带你从知识与思想的层面从0构建C++知识框架,分析大型项目实践思路,为你打下坚实的基础。 3.李宁老师结合4大国外顶级C++著作的精华为大家推出的《征服C++11》课程。 【学完后我将达到什么水平?】 1.对C++的各个知识能够熟练配置、开发、部署; 2.吊打一切关于C++的笔试面试题; 3.面向物联网的“嵌入式”和面向大型化的“分布式”开发,掌握职业钥匙,把握行业先机。 【面向人群】 1.希望一站式快速入门的C++初学者; 2.希望快速学习 C++、掌握编程要义、修炼内功的开发者; 3.有志于挑战更高级的开发项目,成为资深开发的工程师。 【课程设计】 本课程包含3大模块 基础篇 本篇主要讲解c++的基础概念,包含数据类型、运算符等基本语法,数组、指针、字符串等基本词法,循环、函数、类等基本句法等。 进阶篇 本篇主要讲解编程中常用的一些技能,包含类的高级技术、类的继承、编译链接和命名空间等。 提升篇: 本篇可以帮助学员更加高效的进行c++开发,其中包含类型转换、文件操作、异常处理、代码重用等内容。
©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页
实付 59.90元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值