使用UIGraphicsBeginImageContext生成新图时颜色失真问题求助

核心原因 #

问题出在色彩空间不匹配：系统默认的绘图上下文（包括UIGraphicsBeginImageContext和MetalPetal默认配置）会使用线性RGB色彩空间，而多数图片采用的是带gamma校正的sRGB色彩空间，两者的自动转换会导致画面亮度异常。

解决方案 #

1. 替换为UIGraphicsImageRenderer（推荐） #

UIGraphicsImageRenderer会自动匹配原图的色彩空间和缩放比例，避免手动配置出错：

// originalImageURI is a local path
guard let originalImage = normalizedImageOrientation(image: originalImageURI) else { ... }

let renderer = UIGraphicsImageRenderer(size: originalImage.size, format: UIGraphicsImageRendererFormat(for: originalImage.scale))
let combinedImage = renderer.image { context in
    originalImage.draw(at: .zero)
}

2. 修复UIGraphicsBeginImageContext的色彩空间配置 #

如果必须使用旧API，需手动指定sRGB色彩空间：

// originalImageURI is a local path
guard let originalImage = normalizedImageOrientation(image: originalImageURI) else { ... }

// 指定sRGB色彩空间
guard let colorSpace = CGColorSpace(name: CGColorSpace.sRGB) else { ... }
UIGraphicsBeginImageContextWithOptions(originalImage.size, false, originalImage.scale)
if let context = UIGraphicsGetCurrentContext() {
    // 确保上下文背景透明（适配带alpha通道的原图）
    context.setFillColor(UIColor.clear.cgColor)
    context.fill(CGRect(origin: .zero, size: originalImage.size))
    originalImage.draw(at: .zero)
}
let combinedImage = UIGraphicsGetImageFromCurrentImageContext()
UIGraphicsEndImageContext()

3. 修正MetalPetal的色彩空间传递 #

创建MTIImage和生成CGImage时，明确指定原图的色彩空间：

guard let originalImage = normalizedImageOrientation(image: originalImageURI),
      let cgImage = originalImage.cgImage else { ... }

// 保留原图色彩空间，无则 fallback 到 sRGB
let targetColorSpace = cgImage.colorSpace ?? CGColorSpace(name: CGColorSpace.sRGB)!
let mtiImage = MTIImage(cgImage: cgImage, colorSpace: targetColorSpace)

guard let device = MTLCreateSystemDefaultDevice(),
      let context = try? MTIContext(device: device, options: MTIContextOptions()) else { ... }

// 生成CGImage时指定色彩空间和alpha信息
let outputOptions = MTICGImageCreationOptions(
    colorSpace: targetColorSpace,
    alphaInfo: cgImage.alphaInfo
)
let combinedCGImage = try context.makeCGImage(from: mtiImage, options: outputOptions)
let resultImage = UIImage(
    cgImage: combinedCGImage,
    scale: originalImage.scale,
    orientation: originalImage.imageOrientation
)

额外注意点 #

• 确认normalizedImageOrientation方法未修改原图的色彩空间或gamma值；
• 优先用真机测试（模拟器的色彩渲染逻辑与真机存在差异）。

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