C++编写一个函数把图像从RGB转成I420

在C++中，将图像从RGB格式转换为I420格式涉及到图像数据的重排和可能的颜色空间转换。I420是一种YUV色彩空间的存储格式，通常用于视频压缩和处理，其中Y表示亮度信息，U和V表示色度信息。Y分量的数据通常是每像素一个字节，而U和V分量的数据则按照每两个像素共享一个字节的方式存储。

下面是一个简化的C++函数示例，用于将RGB图像转换为I420格式。请注意，这个示例没有进行任何错误检查，并假设输入图像的尺寸是偶数，因为I420格式要求宽度和高度都是偶数。

#include <cstdint>
#include <algorithm>

void ConvertRGBtoI420(const uint8_t* rgb_image, int width, int height,
                      uint8_t* y_plane, uint8_t* u_plane, uint8_t* v_plane) {
    const int y_size = width * height;
    const int uv_size = (width / 2) * (height / 2);
    uint8_t* u_base = u_plane;
    uint8_t* v_base = v_plane;

    for (int i = 0; i < height; ++i) {
        for (int j = 0; j < width; ++j) {
            int index = i * width * 3 + j * 3; // RGB image data index
            int y_index = i * width + j;       // Y plane index

            // Convert RGB to Y
            y_plane[y_index] = std::min(255, static_cast<int>(0.299 * rgb_image[index] + 0.587 * rgb_image[index + 1] + 0.114 * rgb_image[index + 2]));

            // Fill U and V values. Note: this is a simplified version without chroma subsampling.
            if (i % 2 == 0 && j % 2 == 0) {
                int uv_index = (i / 2) * (width / 2) + (j / 2);
                u_base[uv_index] = std::min(255, static_cast<int>(-0.147 * rgb_image[index] - 0.289 * rgb_image[index + 1] + 0.436 * rgb_image[index + 2]));
                v_base[uv_index] = std::min(255, static_cast<int>(0.615 * rgb_image[index] - 0.515 * rgb_image[index + 1] - 0.100 * rgb_image[index + 2]));
            }
        }
    }
}

在上面的代码中，我们首先遍历RGB图像的每个像素，将其转换为Y分量，并存储在Y平面上。对于U和V分量，我们只在每隔一行和每隔一个像素的位置存储它们的值，这模拟了4:2:0的采样模式。请注意，这里的RGB到YUV的转换公式是简化的，并且可能需要根据你的具体应用场景进行调整。