WebGL Shader性能优化

🚀 WebGL Shader性能优化全指南（结合Cesium实战）

WebGL Shader运行在GPU的SIMD（单指令多数据）架构上，与CPU的分支预测逻辑完全不同。条件语句（if-else、switch）会导致GPU线程束（Warp）分化，降低并行执行效率。以下从底层原理、优化准则、内置API、实战案例四个维度系统讲解Shader优化。

一、为什么条件语句会拖慢Shader？

GPU的核心执行单元是线程束（Warp），通常包含32/64个并行线程。当Shader中出现条件分支时：

若所有线程的分支判断结果一致（如if (czm_time > 10.0)），GPU会并行执行同一分支，性能无损失；
若线程分支结果不一致（如if (gl_FragCoord.x > 100.0)），GPU会串行执行所有分支，未命中分支的线程会被挂起，导致性能下降50%以上；
动态分支（如基于varying变量的判断）是性能杀手，因为每个线程的判断结果可能完全不同。

二、WebGL Shader核心优化准则

1. 用数学函数替代条件分支

GLSL提供了大量无分支条件判断函数，完全可以替代if-else：

条件逻辑	替代方案	性能提升
`if (x > 0.0) {a} else {b}`	`mix(b, a, step(0.0, x))`	✅ 无分支并行执行
`if (x > min && x < max) {a} else {b}`	`mix(b, a, step(min, x) * step(x, max))`	✅ 避免双分支
`clamp(x, 0.0, 1.0)`	`czm_saturate(x)`（Cesium内置）	✅ 硬件级优化

实战案例：优化闪烁效果

// ❌ 低效：条件分支 if (blinkFactor > 0.5) { gl_FragColor = highlightColor; } else { gl_FragColor = baseColor; } // ✅ 高效：无分支混合 gl_FragColor = mix(baseColor, highlightColor, step(0.5, blinkFactor));

2. 减少片元着色器计算量

把计算移到顶点着色器：顶点着色器执行次数是顶点数（通常几千到几万），片元着色器是像素数（通常几百万），尽量在顶点着色器中完成复杂计算，通过varying传递结果；

复用计算结果：避免重复计算相同值，用临时变量缓存：

// ❌ 重复计算 float length = sqrt(dot(vec3(1.0), vec3(1.0))); // ✅ 缓存结果 vec3 v = vec3(1.0); float length = sqrt(dot(v, v));

使用低精度变量：WebGL支持lowp（低精度）、mediump（中精度）、highp（高精度），非关键计算用mediump/lowp减少GPU带宽：
```
// 颜色用mediump足够，无需highp mediump vec4 color = vec4(1.0, 0.0, 0.0, 1.0);
```

3. 高效使用纹理与数据结构

避免重复纹理采样：缓存纹理采样结果，同一像素多次使用时复用：

// ❌ 重复采样 vec4 color1 = texture2D(tex, uv); vec4 color2 = texture2D(tex, uv + vec2(0.1, 0.1)); // ✅ 缓存采样结果 vec4 baseColor = texture2D(tex, uv); vec4 offsetColor = texture2D(tex, uv + vec2(0.1, 0.1));

使用压缩纹理：Cesium支持KTX2/Basis Universal压缩纹理，减少GPU内存占用和带宽；
纹理图集（Texture Atlas）：将多个小纹理合并为一张大纹理，减少纹理切换开销。

4. 内存访问优化

避免随机访问数组：GPU对连续内存访问效率最高，数组索引尽量用连续值，避免基于varying变量的随机索引；
减少varying变量数量：varying变量需要从顶点着色器传递到片元着色器，占用GPU带宽，尽量合并向量：
```
// ❌ 多个单独变量 varying float v1; varying float v2; varying float v3; // ✅ 合并为向量 varying vec3 vData;
```
避免使用discard：discard会打断GPU的深度测试优化，尽量用alpha通道控制透明度替代。

5. 状态管理与着色器复用

减少着色器切换：Cesium中尽量复用相同的Fabric材质定义，避免频繁创建新的Material实例；
批量处理绘制：用Primitive替代Entity批量渲染相同类型的几何体，减少Draw Call；
避免动态修改Uniform：Uniform修改会触发GPU状态更新，尽量将动态参数合并为一个纹理或数组。

三、WebGL/Cesium 关键内置API详解

Cesium封装了大量硬件级优化的内置函数，优先使用这些函数比手动实现效率高10-100倍：

1. WebGL标准内置函数（GPU加速）

函数	作用	优势
`dot(a, b)`	向量点积	硬件指令级实现，比手动计算`a.xb.x + a.yb.y`快5倍
`cross(a, b)`	向量叉积	同上，GPU原生支持
`normalize(v)`	向量归一化	避免手动计算`sqrt(dot(v, v))`的精度误差
`mix(a, b, t)`	线性混合	硬件优化的插值，比手动计算`a(1-t)+bt`更稳定
`step(edge, x)`	阶跃函数	返回`x>edge?1.0:0.0`，无分支条件判断
`smoothstep(edge0, edge1, x)`	平滑阶跃	带缓动的阶跃函数，用于边缘平滑过渡

2. Cesium专属内置函数（`czm_`前缀）

Cesium针对GIS场景优化的函数，完全兼容WebGL 1.0/2.0：

函数	作用	实战场景
`czm_saturate(x)`	等价于`clamp(x, 0.0, 1.0)`	颜色值、透明度归一化
`czm_modelViewProjection`	模型视图投影矩阵	将顶点坐标转换为屏幕坐标
`czm_inverseModelViewProjection`	逆投影矩阵	屏幕坐标转世界坐标（如鼠标拾取）
`czm_viewport`	视口参数（x,y,width,height）	计算纹理坐标、屏幕空间效果
`czm_time`	全局时间变量（秒）	动画、闪烁效果
`czm_sunDirectionEC`	太阳方向（相机坐标系）	光照计算、阴影效果
`czm_translateRelativeToEye(distance)`	相对相机平移	实现跟随相机的UI元素
`czm_rayMarch(ray, step, maxSteps)`	光线步进算法	体积云、水体渲染

实战案例：用Cesium内置函数优化光照计算

// ❌ 手动计算漫反射 vec3 normal = normalize(vNormal); vec3 sunDir = normalize(czm_sunDirectionEC); float diffuse = max(dot(normal, sunDir), 0.0); // ✅ 用Cesium内置函数简化 float diffuse = czm_saturate(dot(vNormal, czm_sunDirectionEC));

四、Cesium Shader优化实战案例

1. 优化前：动态分支导致的闪烁效果

// ❌ 低效：基于varying变量的动态分支 varying float vDistance; uniform float threshold; void main() { if (vDistance > threshold) { gl_FragColor = vec4(1.0, 0.0, 0.0, 1.0); } else { gl_FragColor = vec4(0.0, 1.0, 0.0, 1.0); } }

2. 优化后：无分支混合

// ✅ 高效：用step函数替代条件分支 varying float vDistance; uniform float threshold; void main() { vec4 color1 = vec4(1.0, 0.0, 0.0, 1.0); vec4 color2 = vec4(0.0, 1.0, 0.0, 1.0); // 无分支混合颜色 gl_FragColor = mix(color2, color1, step(threshold, vDistance)); }

3. 极致优化：用纹理采样替代分支

对于复杂的多分支逻辑，可将分支结果预渲染到纹理中，通过纹理采样替代条件判断：

// ✅ 极致高效：纹理采样替代多分支 uniform sampler2D colorMap; varying float vIndex; void main() { // 用索引采样纹理，完全无分支 gl_FragColor = texture2D(colorMap, vec2(vIndex, 0.5)); }

五、性能分析工具

Chrome DevTools：
- Performance面板：查看GPU时间线，定位Shader执行瓶颈；
- WebGL Inspector：捕获WebGL帧，分析Shader的指令数、内存占用；
Cesium Sandcastle：
- 官方在线编辑器（https://sandcastle.cesium.com/），支持实时测试Shader性能；
RenderDoc：
- 专业GPU调试工具，可查看Shader的汇编代码、线程执行情况。