如何在WGSL顶点着色器中正确声明并使用常量数组？

背景 #

尝试在WGSL顶点着色器中硬编码顶点渲染单个三角形，用全局let数组TRI_VERTICES存储顶点，通过vertex_index索引取值，代码如下：

let TRI_VERTICES: array<vec4<f32>, 3> = array<vec4<f32>, 3>(
  vec4<f32>(0., 0., 0., 1.0),
  vec4<f32>(0., 1., 0., 1.0),
  vec4<f32>(1., 1., 0., 1.0),
);

@vertex
fn vs_main(
  @builtin(vertex_index) in_vertex_index: u32,
) -> @builtin(position) vec4<f32> {
  return TRI_VERTICES[in_vertex_index];
}

@fragment
fn fs_main(@builtin(position) in: vec4<f32>) -> @location(0) vec4<f32> {
  return vec4<f32>(in.x, in.y, 0.1, 1.0);
}

结合wgpu执行绘制调用：

render_pass.draw(0..3, 0..1);

触发报错：

Shader validation error:
   ┌─ Shader:13:9
   │
13 │   return TRI_VERTICES[in_vertex_index];
   │         ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ naga::Expression [3]


    Entry point vs_main at Vertex is invalid
    Expression [3] is invalid
    The expression [1] may only be indexed by a constant

问题解答 #

1. var<private>声明的数组是否可在vs_main中修改？ #

是的。var<private>默认是可写变量，你可以在vs_main内直接修改数组元素，比如TRI_VERTICES[1] = vec4(0.5, 0.5, 0.0, 1.0);。虽然你的场景里不会去修改它，但从语法语义上，它不属于常量。

2. 能否将TRI_VERTICES声明为真正的常量？ #

可以，但要避开let数组的限制：WGSL中let定义的是编译期常量数组，只能用编译期已知的常量值索引，而vertex_index是运行时动态值，所以直接用let数组会报错。

要实现只读且支持运行时索引的常量数组，有两种可行方案：

• 给var<private>添加const限定符：var<private, const> TRI_VERTICES，这样数组变为只读，无法在着色器内修改，同时允许运行时变量索引；
• 将数组放在常量缓冲区中，用@group(0) @binding(0) const TRI_VERTICES: array<vec4<f32>, 3>声明，这种方式需要在Rust代码中把数组数据上传到GPU常量缓冲区，适合后续需要动态更新或数据量较大的场景。

3. 声明TRI_VERTICES的最优方式是什么？ #

如果是硬编码固定顶点的小型场景，var<private, const>是最合适的选择，代码示例：

var<private, const> TRI_VERTICES: array<vec4<f32>, 3> = array(
  vec4(0., 0., 0., 1.0),
  vec4(0., 1., 0., 1.0),
  vec4(1., 1., 0., 1.0),
);

这个写法的优势：

• 语义上是只读常量，符合存储固定顶点的需求；
• 支持vertex_index这类运行时变量索引；
• 无需额外的GPU缓冲区绑定，数据直接内嵌在着色器中，实现简单高效。

如果后续需要动态修改顶点数据，或数据量较大，再考虑使用常量缓冲区的方案。

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