Glslang编译器中的SPIR-V生成缺陷：三元运算符与texelFetch的组合问题

问题概述

在KhronosGroup的glslang编译器项目中，发现了一个关于SPIR-V代码生成的缺陷。当开发者在GLSL着色器代码中将三元条件运算符（a ? b : c）直接作为texelFetch函数的参数使用时，编译器会生成不符合SPIR-V规范的代码。这个缺陷在最新版本的glslangValidator（1.3.280.0）中仍然存在。

技术背景

在GLSL中，texelFetch是一个常用的纹理采样函数，它允许开发者通过整数坐标直接访问纹理中的特定纹素。而三元条件运算符则是GLSL中实现条件选择的便捷方式。正常情况下，这两种语法结构应该能够无缝配合使用。

SPIR-V规范中有一个重要要求：所有OpSampledImage指令必须与其消费结果指令（如OpImageFetch）位于同一个基本块中。这是为了确保着色器程序的正确性和优化可能性。

问题表现

当使用如下代码结构时会出现问题：

const vec4 result = texelFetch(
    sampler2D(globalTexture, globalSampler),
    ivec2(int(pixel.x), globalBuffer.flip == 0 ? int(pixel.y) : 1280 - 1 - int(pixel.y)),
    0
);

编译器生成的SPIR-V代码会将OpSampledImage指令和其消费指令OpImageFetch放置在不同的基本块中，违反了SPIR-V规范。这会导致使用spirv-val验证工具时报告错误：

error: line 67: All OpSampledImage instructions must be in the same block in which their Result <id> are consumed.

解决方案

目前可行的解决方案是将三元运算符的计算提取到单独的变量中，然后再传递给texelFetch函数：

ivec2 coord = ivec2(int(pixel.x), globalBuffer.flip == 0 ? int(pixel.y) : 1280 - 1 - int(pixel.y));
const vec4 result = texelFetch(
    sampler2D(globalTexture, globalSampler),
    coord,
    0
);

这种写法能够生成符合规范的SPIR-V代码，通过验证工具的检查。

技术分析

从生成的SPIR-V代码可以看出，当三元运算符直接作为参数时，编译器会为条件表达式生成控制流结构（OpSelectionMerge和OpBranchConditional），这导致OpSampledImage和OpImageFetch被分隔在不同的基本块中。

而将条件计算提取到单独变量后，条件表达式的计算被提前到texelFetch调用之前，保持了采样操作相关指令在同一基本块中的完整性。

影响范围

这个问题主要影响：

1. 使用复杂条件表达式作为纹理坐标的开发者
2. 依赖自动SPIR-V验证的工具链
3. 追求代码简洁性的开发者

最佳实践建议

基于这个问题，建议开发者在编写GLSL代码时：

1. 避免将复杂表达式（特别是包含控制流的表达式）直接作为纹理采样函数的参数
2. 对于条件纹理坐标计算，先计算坐标再采样
3. 定期使用SPIR-V验证工具检查生成的代码
4. 关注glslang的更新，等待此问题的官方修复

总结

这个glslang编译器的问题展示了高级着色语言到中间表示转换过程中的复杂性。虽然GLSL语法上允许各种表达式组合，但在转换为SPIR-V时需要考虑更多的底层约束。开发者需要了解这些潜在的限制，并采取相应的编码策略来确保生成的代码符合规范。