iOS：如何从非预乘[UInt8]高效创建图像？

问题背景 #

我手里有一份RGBA格式的[UInt8]颜色数据，想要用最高效的方式创建图像。当前的实现代码如下：

extension CGImage {
  static func fromData(content: [UInt8], width: Int, height: Int) -> CGImage? {
    guard content.count == width * height * 4,
          let space = CGColorSpace(name: CGColorSpace.sRGB) else {
      return nil
    }
    var content = content
    let context = CGContext(
      data: &content,
      width: width,
      height: height,
      bitsPerComponent: 8,
      bytesPerRow: width * 4,
      space: space,
      bitmapInfo: CGImageAlphaInfo.premultipliedLast.rawValue
    )

    let image = context?.makeImage()
    return image
  }
}

现在的问题是：iOS仅支持预乘数据，所以必须使用CGImageAlphaInfo.premultipliedLast，这就需要先做预乘运算。我用Metal处理这个运算，但大图像要耗时约8ms，叠加其他处理后会导致绘制卡顿。想知道有没有其他方式直接从非预乘的[UInt8]数据创建CGImage、CIImage或UIImage？

可行解决方案 #

1. 直接使用非预乘Alpha的CGContext配置 #

iOS并非完全不支持非预乘Alpha，只是需要正确配置bitmapInfo。你可以直接用nonPremultipliedLast创建CGImage，系统会在绘制阶段自动处理预乘，这个内部优化的过程比手动用Metal处理更高效：

extension CGImage {
  static func fromNonPremultipliedData(content: [UInt8], width: Int, height: Int) -> CGImage? {
    guard content.count == width * height * 4,
          let space = CGColorSpace(name: CGColorSpace.sRGB) else {
      return nil
    }
    var content = content
    let context = CGContext(
      data: &content,
      width: width,
      height: height,
      bitsPerComponent: 8,
      bytesPerRow: width * 4,
      space: space,
      bitmapInfo: CGImageAlphaInfo.nonPremultipliedLast.rawValue
    )
    return context?.makeImage()
  }
}

注意：如果后续要将该CGImage用于Metal渲染，可能仍需手动处理预乘，但仅用于UIKit显示的话，系统内部处理的效率会远高于自定义实现。

2. 用CIImage直接包装非预乘数据 #

CIImage支持直接从非预乘像素数据创建实例，无需提前做预乘操作，且能利用Core Image的硬件加速，非常适合大图像场景：

extension CIImage {
  static func fromNonPremultipliedRGBAData(content: [UInt8], width: Int, height: Int) -> CIImage? {
    guard content.count == width * height * 4 else { return nil }
    let data = Data(content)
    let format = CIFormat.RGBA8
    return CIImage(
      bitmapData: data,
      bytesPerRow: width * 4,
      size: CGSize(width: width, height: height),
      format: format,
      colorSpace: CGColorSpace(name: CGColorSpace.sRGB)
    )
  }
}

如果需要转成UIImage直接显示，直接用UIImage(ciImage: ciImage)即可；若需要导出位图数据，可通过CIContext渲染，但仅用于界面显示的话，该方式几乎无额外开销。

3. 优化Metal预乘运算（若必须使用预乘数据） #

如果业务逻辑要求必须使用预乘数据，可通过以下方式优化Metal实现：

• 使用MTLBuffer共享内存，避免数据拷贝
• 将数据直接上传到Metal纹理后再做预乘，比处理原始缓冲区更快
• 将预乘运算与其他图像处理操作合并到同一个Metal Kernel中，减少多次渲染的开销

内容的提问来源于stack exchange，提问作者Son Nguyen