在sokol_gfx中处理结构体数组uniform的技巧

在图形编程中，我们经常需要将结构体数组传递给着色器，比如处理多个点光源的情况。本文将以sokol_gfx项目为例，探讨在OpenGL/GLES环境下处理这类问题的正确方法。

常见误区

开发者通常会尝试直接在着色器中定义结构体数组：

struct PointLight {
  float radius;
  vec3 color;
  vec3 position;
};
uniform PointLight[MAX_LIGHTS] lights;

然后在C++代码中定义匹配的结构体，并尝试通过uniform块传递。然而，这种方法在OpenGL/GLES环境下存在诸多限制，特别是当不使用sokol-shdc着色器编译器时。

根本原因

OpenGL/GLES对uniform块的内存布局有严格限制，特别是std140布局规则。这些规则要求：

1. 基本类型必须按特定边界对齐
2. vec3类型在uniform块中会被填充为vec4
3. 结构体数组的布局可能不符合预期

推荐解决方案

方案一：使用结构体数组展开

如果使用sokol-shdc编译器，可以将结构体数组展开为vec4数组：

uniform vs_params {
    PointLight lights[MAX_LIGHTS];
};

编译器会自动处理内存布局问题。但需要注意，sokol-shdc对数组元素类型有限制，仅支持float、vec2、vec4和mat4类型的数组。

方案二：数组结构体(SOA)模式

更可靠的方案是采用"结构体数组"(Structure of Arrays)模式：

uniform vs_params {
    float light_radii[MAX_LIGHTS];
    vec4 light_colors[MAX_LIGHTS];
    vec4 light_positions[MAX_LIGHTS];
};

注意这里使用vec4而非vec3，这是为了满足std140的对齐要求。在C++端，可以这样定义：

struct LightUniforms {
    float radii[MAX_LIGHTS];
    float colors[MAX_LIGHTS][4];
    float positions[MAX_LIGHTS][4];
};

方案三：使用存储缓冲区(Storage Buffer)

较新版本的sokol_gfx支持存储缓冲区，这提供了更灵活的内存布局：

buffer LightBuffer {
    PointLight lights[];
};

存储缓冲区的优势在于：

1. 对数组元素的限制较少
2. 可以动态索引
3. 支持更大的数据量

实现建议

1. 优先考虑使用sokol-shdc编译器，它能自动处理很多底层细节
2. 对于简单场景，SOA模式是最可靠的选择
3. 对于复杂数据结构或大量数据，考虑使用存储缓冲区
4. 始终注意内存对齐要求，特别是在OpenGL/GLES环境下

性能考量

• 频繁更新的uniform数据应该尽量小
• 静态或低频更新的数据适合放入存储缓冲区
• 在移动设备上，注意避免频繁的uniform更新

通过理解这些技术细节和限制，开发者可以更有效地在sokol_gfx项目中处理复杂的数据结构传递问题。