Swift向Metal片段着色器Uniform传递数组数据失败，求正确实现方法

嘿，我之前踩过一模一样的坑！问题出在你Swift里用的是动态数组[Float]，这玩意儿的内存布局和Metal里的固定大小数组完全不兼容，导致Metal根本读不到你要传的0.5。

问题根源 #

Swift的[Float]是动态数组，它本质上是一个包含三个部分的结构体：指向实际数据的指针、数组元素个数、当前容量。你把它放进Uniform结构体里，传给Metal的其实是这三个元数据的值，不是你要的0.5这个Float本身！而Metal那边的float test[1]是直接在结构体里占4字节的连续内存，自然读不到正确的数值。

正确的实现方法 #

解决思路很简单：把Swift里的动态数组换成固定大小的数组类型，让它的内存布局和Metal的数组完全对齐。这里分两种情况，看你用的Swift版本：

方法1：Swift 5.9+ 用原生固定大小数组

Swift 5.9开始支持了固定大小数组语法，直接用分号声明长度：

// Swift端修改后的结构体
struct Uniforms {
    var test: [Float; 1] // 固定大小为1的Float数组，注意是分号！
}

// 初始化Uniforms
let floatArray: [Float; 1] = [0.5]
let uniforms = Uniforms(test: floatArray)

// 传递Uniform的代码不变
commandEncoder.setFragmentBytes(&uniforms, length: MemoryLayout<Uniforms>.stride, index: 0)

方法2：旧版本Swift 用SIMD类型

如果你的Swift版本低于5.9，可以用SIMD类型来替代，Metal和Swift的SIMD类型内存布局是完全兼容的：

// Swift端修改后的结构体
struct Uniforms {
    var test: SIMD1<Float> // 对应Metal的float[1]
}

// 初始化Uniforms
let uniforms = Uniforms(test: SIMD1<Float>(0.5))

// 传递代码不变
commandEncoder.setFragmentBytes(&uniforms, length: MemoryLayout<Uniforms>.stride, index: 0)

Metal端代码无需修改 #

你原来的Metal结构体和片段着色器代码完全没问题，保持原样即可：

struct Uniforms {
    float test[1];
};

fragment float4 fragmentShader(Uniforms uniforms [[buffer(0)]]) {
    float testColor = 1.0;
    for (int i = 0; i < 1; i++) {
        testColor *= uniforms.test[i];
    }
    float a = 1.0 - testColor;
    return float4(1.0, 0.0, 0.0, a);
}

现在运行的话，你应该能看到半透明的红色（alpha值是0.5），说明0.5已经正确传递到Metal里了。

额外注意点 #

如果以后要传更长的数组，比如长度为3的Float数组：

• Swift端用[Float; 3]或者SIMD3<Float>
• Metal端用float test[3]
  这样两者的内存布局会完全匹配，不会再出现传值失败的问题。

备注：内容来源于stack exchange，提问作者persec