Vulkan计算着色器中如何向函数传递无固定大小的缓冲区？

针对你遇到的无界存储缓冲无法直接传递给函数的问题，这里提供几种实用的解决思路：

方法一：用模板化接口块实现通用处理 #

先定义一个通用的存储缓冲接口块模板，再实例化具体的缓冲，函数直接接收模板类型的引用，就能通用处理不同的无界缓冲：

// 通用存储缓冲模板，支持任意数据类型
template <typename T>
buffer StorageBuffer {
    T data[];
};

// 绑定0和1的两个浮点型无界缓冲
layout(binding = 0) buffer StorageBuffer<float> buf1;
layout(binding = 1) buffer StorageBuffer<float> buf2;

// 函数接收通用缓冲引用，直接操作内部的无界数组
void foo(StorageBuffer<float> &buf) {
    uint idx = gl_GlobalInvocationID.x;
    // 这里写你的复杂处理逻辑，比如修改元素值
    buf.data[idx] *= 2.0;
}

void main() {
    foo(buf1);
    foo(buf2);
}

方法二：传递数组指针+运行时长度 #

无界数组可以隐式转为对应类型的指针，再配合CPU端传入的缓冲长度（通过push常量或uniform传递），就能让函数安全处理数组：

layout(binding = 0) buffer storageBuffer1 {
    float data1[];
};

layout(binding = 1) buffer storageBuffer2 {
    float data2[];
};

// 通过push常量传递两个缓冲的元素个数（CPU端提前计算好传入）
layout(push_constant) uniform PushConsts {
    uint len_buf1;
    uint len_buf2;
} push_data;

// 函数接收浮点指针和长度，处理时先判断索引合法性
void foo(float *arr, uint arr_len) {
    uint idx = gl_GlobalInvocationID.x;
    if (idx < arr_len) {
        // 执行你的复杂逻辑
        arr[idx] += 1.0;
    }
}

void main() {
    foo(data1, push_data.len_buf1);
    foo(data2, push_data.len_buf2);
}

方法三：重载函数处理不同缓冲（适合少量缓冲场景） #

如果缓冲数量不多，可以直接针对每个缓冲的接口块类型重载函数：

layout(binding = 0) buffer storageBuffer1 {
    float data1[];
} buf1;

layout(binding = 1) buffer storageBuffer2 {
    float data2[];
} buf2;

// 处理buf1的重载函数
void foo(decltype(buf1) &buf) {
    uint idx = gl_GlobalInvocationID.x;
    buf.data1[idx] *= 0.5;
}

// 处理buf2的重载函数
void foo(decltype(buf2) &buf) {
    uint idx = gl_GlobalInvocationID.x;
    buf.data2[idx] *= 0.5;
}

void main() {
    foo(buf1);
    foo(buf2);
}

为什么直接传float x[]不行？ #

GLSL里函数参数的float x[]语法等价于float x[编译期常量]，必须指定编译时确定的数组大小。而无界数组是运行时才确定长度的，所以编译器会报错。核心解决思路就是绕过编译期数组大小的限制，要么传递包含无界数组的接口块，要么用指针+运行时长度的组合。

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