Metal片段着色器中能否获取帧缓冲及传入Buffer的尺寸？

好问题！我来给你拆解Metal里这两种场景的不同处理方式：

1. 输出帧缓冲纹理的宽高：无需手动传参，可直接获取 #

当你在片段着色器中声明了[[color(0)]]修饰的输出纹理（比如texture2d<half>类型），这个纹理对象本身就内置了获取宽高的方法和属性，完全不用额外传参数。

举个实用的例子：

#include <metal_stdlib>
using namespace metal;

// 假设顶点着色器输出的结构体
struct VertexOut {
    float4 position [[position]];
    float2 texCoord;
};

fragment half4 colorFragment(VertexOut in [[stage_in]],
                              texture2d<half> outputTex [[color(0)]]) {
    // 获取输出帧缓冲的宽高
    uint texWidth = outputTex.get_width();
    uint texHeight = outputTex.get_height();
    
    // 也可以直接用属性访问（适配较新的Metal版本）
    // uint texWidth = outputTex.width;
    // uint texHeight = outputTex.height;
    
    // 举个应用：判断当前像素是否在纹理中心区域
    uint pixelX = uint(in.texCoord.x * texWidth);
    uint pixelY = uint(in.texCoord.y * texHeight);
    bool isInCenter = abs(pixelX - texWidth/2) < 20 && abs(pixelY - texHeight/2) < 20;
    
    return isInCenter ? half4(1.0, 0.0, 0.0, 1.0) : half4(0.0, 1.0, 0.0, 1.0);
}

2. 传入的MTLBuffer长度：必须手动作为参数传入 #

这里要注意：MTLBuffer在着色器里是以指针或数组的形式暴露的，着色器本身无法获取它的总长度——因为GPU只看到内存的起始地址，没有内置的元数据来存储buffer的大小。

所以你得额外把buffer的长度传给着色器，常见的两种做法：

• 把长度放在一个独立的小MTLBuffer中（比如用[[buffer(1)]]修饰）
• 把长度打包到顶点/片段的输入结构体里（如果长度和顶点数据强关联）

示例代码：

#include <metal_stdlib>
using namespace metal;

struct FragmentInput {
    float4 position [[position]];
};

fragment half4 bufferFragment(FragmentInput in [[stage_in]],
                              [[buffer(0)]] device float* dataBuffer,
                              [[buffer(1)]] device uint bufferLength) {
    // 安全遍历buffer，避免内存越界
    float total = 0.0;
    for(uint i=0; i<bufferLength; i++) {
        total += dataBuffer[i];
    }
    float avgValue = total / bufferLength;
    
    // 用平均值生成灰度颜色
    return half4(avgValue, avgValue, avgValue, 1.0);
}

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