VTK智能着色器替换GLSL-Shader-Replacement实战

VTK 提供了开箱即用的 PBR（基于物理的渲染）与 Phong 光照模型。但在很多高级可视化应用中（如点云高度热力图、医学器官透明剖切线、自定义网格网格线高亮），默认材质无法满足需求。

在传统的 OpenGL 开发中，我们需要重写整个着色器程序（Shader Program）。而 VTK 创新性地提供了 Shader Replacement（着色器模板替换） 机制：允许开发者在 VTK 自动生成的 GLSL 代码中，精准**“插入”或“替换”**自定义代码段。

本文将解构 vtkOpenGLPolyDataMapper 的替换标记原理，并提供一个基于高度热力图的自定义 Shader 实战案例。

---

一、VTK 着色器生成与锚点（Replacement Tags）

vtkOpenGLPolyDataMapper 在编译 OpenGL Shader 时，会使用预定义的锚点标记（Tags）。我们可以将这些标记看作 C++ 代码中的占位符。

1.1 常用 Shader 替换锚点对照表

锚点 Tag 名称	所在 Shader	作用与注入时机	典型用途
//VTK::PositionVC::Impl	Vertex Shader	顶点从 Model 转换到 View Space 之后	自定义顶点平移、波浪形变形
//VTK::Normal::Impl	Vertex/Fragment	法线计算与变换阶段	自定义法线贴图、法线反转
//VTK::Color::Impl	Fragment Shader	决定像素最终输出颜色阶段	热力图映射、自定义渐变色
//VTK::Light::Impl	Fragment Shader	光照方程（Diffuse / Specular）计算	卡通着色 (Toon Shading)、Custom PBR

---

二、AddShaderReplacement 接口详解

void AddShaderReplacement(
    vtkShader::Type type,          // Shader 类型: Vertex, Fragment, Geometry
    const std::string& originalValue, // 被替换的 VTK 锚点 Tag
    bool replaceFirst,             // true 在原 Tag 前插入; false 完全覆盖
    const std::string& replacementValue, // 注入的自定义 GLSL 代码
    bool replaceAllMaster          // 是否应用到所有 Master 模板
);

---

三、实战案例：根据 Z 轴高度映射动态彩虹热力图

下面的代码示范如何通过注入 Fragment Shader，让普通的 Mesh 模型无需构造 vtkLookupTable 就能在 GPU 端根据世界坐标 Z 轴高度输出动态彩虹热力图。

#include <vtkSmartPointer.h>
#include <vtkOpenGLPolyDataMapper.h>
#include <vtkShader.h>
#include <string>

void ApplyHeightMapShader(vtkOpenGLPolyDataMapper* mapper, float minZ, float maxZ) {
    // 1. 在 Fragment Shader 中注入 Uniform 变量定义
    mapper->AddShaderReplacement(
        vtkShader::Fragment,
        "//VTK::System::Dec", // 声明变量的位置
        true, // 在 Tag 前插入
        "uniform float minZ;\n"
        "uniform float maxZ;\n"
        "varying vec3 v_positionMC;\n", // 来自 Vertex Shader 的模型坐标
        false
    );

    // 2. 在 Fragment Shader 中替换颜色计算逻辑
    std::string colorCode = 
        "  float normalizedZ = clamp((v_positionMC.z - minZ) / (maxZ - minZ), 0.0, 1.0);\n"
        "  // 彩虹渐变色插值 (Hue 旋转)\n"
        "  vec3 rainbow = vec3(0.5) + vec3(0.5) * cos(6.28318 * (vec3(1.0, 0.66, 0.33) + normalizedZ));\n"
        "  fragOutput0 = vec4(rainbow, 1.0);\n";

    mapper->AddShaderReplacement(
        vtkShader::Fragment,
        "//VTK::Color::Impl",
        false, // 完全替换默认颜色计算
        colorCode,
        false
    );
}

---

四、着色器替换避坑指南

4.1 避坑策略一：GLSL 变量作用域与声明冲突

注入的代码如果使用了 VTK 已经内置的变量名（如 gl_Position 或 normalVC），会导致 Shader 编译报错（Redefinition Error）。可以在 //VTK::System::Dec 处先观察 VTK 生成的完整 Shader 源码。

4.2 避坑策略二：调试 Shader 编译错误

当注入的 GLSL 代码有语法错误时，VTK 默认只抛出一个通用异常。开启 VTK 的 Shader 调试日志可以打印完整的错误行号：

#include <vtkLogger.h>
// 开启 VTK 内部 Shader 打印选项
vtkLogger::SetStderrVerbosity(vtkLogger::VERBOSITY_INFO);

---

五、总结

通过 vtkOpenGLPolyDataMapper::AddShaderReplacement()，我们无需脱离 VTK 生态去写底层的 OpenGL 框架，就能享受到原生 GLSL 着色器带来的极大定制自由度与 GPU 级性能优势。