Metal Compute Shader复制纹理后Alpha值减半问题排查与解决

我实现了一个无操作的Metal Compute Shader，代码如下：

#include <metal_stdlib>
using namespace metal;

kernel void doNothing(
  texture2d<half, access::read> inTexture [[ texture (0) ]],
  texture2d<half, access::read_write> outTexture [[ texture (1) ]],
  uint2 gid [[ thread_position_in_grid ]]
) {
  outTexture.write(inTexture.read(gid), gid);
}

预期该着色器仅将inTexture的像素复制到outTexture，但实际像素复制正常，Alpha值却减半。输入纹理由默认设备的纹理加载器从CGImage创建，输出纹理使用.rgba8Unorm_srgb像素类型、与输入纹理相同的宽高且禁用mipmap。请问该问题的原因是什么，如何修复？

原因分析 #

核心问题是sRGB纹理的自动颜色空间转换错误作用到了Alpha通道：

• 输入纹理从CGImage加载后，Metal会默认将其转换为线性颜色空间的rgba16Float格式，对应着色器里的half类型，此时RGB和Alpha都是0-1的线性值。
• 输出纹理用的是.rgba8Unorm_srgb格式，Metal在写入这类纹理时，会自动把线性空间的RGB转成sRGB（这步是对的，所以RGB显示正常），但它错误地把Alpha通道也套用了同样的sRGB转换曲线。
• 这种转换会压缩线性Alpha的数值，你看到的“减半”就是线性Alpha被错误编码后的直观结果。

修复方案 #

有两种靠谱的修复方式：

方式一：手动拆分通道处理，保留Alpha线性 #

修改Shader代码，只对RGB通道做线性到sRGB的转换，Alpha直接原样保留：

#include <metal_stdlib>
using namespace metal;

kernel void doNothing(
  texture2d<half, access::read> inTexture [[ texture (0) ]],
  texture2d<half, access::read_write> outTexture [[ texture (1) ]],
  uint2 gid [[ thread_position_in_grid ]]
) {
  half4 inColor = inTexture.read(gid);
  // 仅转换RGB通道，Alpha保持线性
  half4 outColor = half4(linear_to_srgb(inColor.rgb), inColor.a);
  outTexture.write(outColor, gid);
}

方式二：修改输出纹理格式（如果不需要sRGB） #

如果你的场景不需要sRGB颜色空间，直接把输出纹理的像素格式改成.rgba8Unorm（非sRGB），这样Metal不会做任何颜色空间转换，Alpha就能正常复制：

// Swift示例：创建输出纹理时调整pixelFormat
let textureDescriptor = MTLTextureDescriptor.texture2DDescriptor(
  pixelFormat: .rgba8Unorm,
  width: inputTexture.width,
  height: inputTexture.height,
  mipmapped: false
)
let outputTexture = device.makeTexture(descriptor: textureDescriptor)!

注意 #

如果必须用sRGB格式的输出纹理，优先选方式一，因为它完全符合sRGB纹理的规范——RGB通道用sRGB空间，Alpha通道保持线性。

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