在 GPUImage 中实现 ColorConversion-CSDN博客

0x00 背景

常见的颜色空间有 RGB 和 YCbCr。YCbCr 能够提供比 RGB 更好的压缩比，因为只需要保证 Y 分量精度够高，Cb 和 Cr 进行适当压缩不影响最后的体感质量，所以在视频相关领域中被广泛使用。

在 OpenCV 中，有方便的 cvtColor 函数可以执行该转换，但在 iOS 中则没有相应的方法，只能手撸。幸运的是转换公式是现成的，可以参考这里: http://www.equasys.de/colorconversion.html , 里面有很多套公式，通过对 GPUImage 代码的研读，可以推算出 iOS 中一般采用的是 HDTV 那一套(BT.709)。

0x01 CPU 方式

有了公式就能进行转换了，最简单的办法就是进行 pixel-wise 的操作，但即便这个简单的需求发现资料也挺少，这里就简单提一下。

首先需要将 Image 转换为 Bitmap，Bitmap 有几种不同的 Pixel format，最常用的是 kCGImageAlphaPremultipliedLast 代表 RGB 分量已经预乘了 alpha，并且 alpha 在最高位地址。当我们取出一个像素点时（32bits integer），拿pixel 的信息可以拆分为两步:

拿到 pixel 的 value。首先可以通过 bitmapcreate+drawimage 的形式拿到 rawData 的 pointer。然后用 i 遍历 height，j 遍历 width，并通过：

uint *pixel_data =  rawData + i * width + j
复制代码

即可拿到单像素的值，value 格式为 RGBA8888

然后结合 iOS big endian的特性，通过如下的宏拿到各个 channel 的分量

#define Mask8(x) ( (x) & 0xFF )
#define R(x) ( Mask8(x) )
#define G(x) ( Mask8(x >> 8 ) )
#define B(x) ( Mask8(x >> 16) )
复制代码

能够进行 pixel-wise operation 理论上来说就可以转换了，但 CPU 方式在做 pixel-wise operation 往往不如 GPU，所以 CPU 方式在此不继续展开。

0x02 GPU 方式

iOS 上进行 GPU 编程有几个选择，比如 OpenGL ES 和 metal。但 GPUImage 良好的架构，当仁不让的成为第一选择。

核心的思想就是将 pixel-wise 的转换放到 shader 中实现，GPUImage 内部包含了一个 YUV-> RGB 的转换流程，通过研究其实现结合0x01中介绍的公式可以得出转换的 shader

PS:

OpenGL 的 mat 时 column major，所以从公式中拿出矩阵来乘的时候需要进行一次 transpose
iOS 的颜色空间 RGB 分别是[0,1]而不是[0,255] ，所以公式中的16 128 128 需要分别转换为 16.0/255.0 0.5 0.5

RGB->YUV:

varying mediump vec2 textureCoordinate;
uniform sampler2D inputImageTexture;
void main()
{
    mediump vec3 yuv;
    mediump vec3 rgb;
    rgb = texture2D(inputImageTexture, textureCoordinate).rgb;
    yuv = mat3(0.256, -.148, .439,
                .504, -.291, -.368,
                0.098, .439,-.071 ) * rgb + vec3(16.0 / 255.0, 0.5, 0.5);
    gl_FragColor = vec4(yuv.r,0,0, 1);
}
复制代码

YUV->RGB:

varying mediump vec2 textureCoordinate;
uniform sampler2D inputImageTexture;
void main()
{
    mediump vec3 yuv;
    mediump vec3 rgb;
    yuv = texture2D(inputImageTexture, textureCoordinate).rgb - vec3(16.0/255.0, 0.5, 0.5);

    rgb = mat3(1.164, 1.164, 1.164,
                0, -.392, 2.017,
                1.596, -.813,0 ) * yuv;
    gl_FragColor = vec4(rgb, 1);
}
复制代码

最后分别为这两个 shader 配套建立一个 custom filter 即可。