Metal纹理转PNG再还原：透明像素异常问题排查

我尝试将Metal纹理转换为PNG（或其他无损压缩格式）保存到磁盘，还原为纹理后希望与原纹理完全一致。目前不透明纹理可正常转换，但透明像素出现异常——比如测试像素[24, 72, 233, 78]转换后变为[48, 127, 255, 78]。通过ImageMagick检查PNG文件，确认问题出在保存环节而非加载环节。

我的纹理格式为bgra8Unorm_srgb，以下是使用的纹理压缩与解压代码：

public struct TextureCompresser {
    let context:CIContext
    let loader:MTKTextureLoader
    
    public init(device:MTLDevice) {
        loader = MTKTextureLoader(device: device)
        context = CIContext(options: [
            CIContextOption.outputPremultiplied: false,
                           ])
    }
    
    public func compress(texture:MTLTexture) -> Data? {
        let options:[CIImageOption:Any] = [CIImageOption.colorSpace:CGColorSpace(name: CGColorSpace.linearSRGB)!]
             
        guard let ciImage = CIImage(mtlTexture: texture, options:options)
            else { return nil }
        
        return  context.pngRepresentation(of: ciImage, format: .BGRA8, colorSpace: CGColorSpace(name: CGColorSpace.sRGB)!)

    }
    
    public func decompress(png:Data) throws -> MTLTexture {
        let usage = MTLTextureUsage(rawValue: MTLTextureUsage.renderTarget.rawValue | MTLTextureUsage.shaderRead.rawValue | MTLTextureUsage.shaderWrite.rawValue)
        
        let options:[MTKTextureLoader.Option : Any] = [
                                MTKTextureLoader.Option.textureUsage:usage.rawValue,
                                MTKTextureLoader.Option.origin:MTKTextureLoader.Origin.flippedVertically.rawValue,
                                MTKTextureLoader.Option.SRGB:NSNumber(value: true),
                                MTKTextureLoader.Option.generateMipmaps:NSNumber(value: false),
                            ]
        
        return try loader.newTexture(data:png,options:options)
    }
}

问题原因与修复方案 #

核心错误点 #

你的代码中色彩空间的转换逻辑不匹配原纹理格式：原纹理是sRGB格式（bgra8Unorm_srgb），但创建CIImage时错误指定了线性sRGB色彩空间，导致透明像素的色彩值被强制从sRGB转换为线性空间，保存PNG时又转回sRGB，两次转换叠加造成了色彩值失真。

修复后的代码 #

public struct TextureCompresser {
    let context: CIContext
    let loader: MTKTextureLoader
    
    public init(device: MTLDevice) {
        loader = MTKTextureLoader(device: device)
        // 保持输出非预乘alpha，匹配PNG的存储规则
        context = CIContext(options: [
            CIContextOption.outputPremultiplied: false
        ])
    }
    
    public func compress(texture: MTLTexture) -> Data? {
        // 使用与原纹理一致的sRGB色彩空间，避免不必要的色彩转换
        let options: [CIImageOption: Any] = [
            CIImageOption.colorSpace: CGColorSpace(name: CGColorSpace.sRGB)!
        ]
        
        guard let ciImage = CIImage(mtlTexture: texture, options: options) else {
            return nil
        }
        
        // 保存时使用原纹理的BGRA8格式和sRGB色彩空间，确保数据无损
        return context.pngRepresentation(
            of: ciImage,
            format: .BGRA8,
            colorSpace: CGColorSpace(name: CGColorSpace.sRGB)!
        )
    }
    
    public func decompress(png: Data) throws -> MTLTexture {
        let usage = MTLTextureUsage(rawValue:
            MTLTextureUsage.renderTarget.rawValue |
            MTLTextureUsage.shaderRead.rawValue |
            MTLTextureUsage.shaderWrite.rawValue
        )
        
        let options: [MTKTextureLoader.Option: Any] = [
            .textureUsage: usage.rawValue,
            .origin: MTLTextureLoader.Origin.flippedVertically.rawValue,
            .SRGB: NSNumber(value: true), // 匹配原纹理的sRGB格式
            .generateMipmaps: NSNumber(value: false)
        ]
        
        return try loader.newTexture(data: png, options: options)
    }
}

关键修改说明 #

1. 创建CIImage时，将色彩空间从linearSRGB改为sRGB，与原纹理的bgra8Unorm_srgb格式完全匹配，避免了多余的色彩空间转换，这是解决透明像素失真的核心。
2. 保存PNG时保持格式和色彩空间与原纹理一致，确保数据完全无损存储。
3. 解压逻辑保留原有的.SRGB选项，保证加载后的纹理格式与原纹理一致。

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