WebGPU 工具分享 - WGSL 代码高亮插件(VSCode)与预处理工具

WGSL 还在积极讨论中,虽然各位大佬不是很满意这个新生儿。
不过,社区已经有了基础的实验性工具(VSCode 插件),并支持了较新的语法。
① WGSL 插件 这个插件支持对文件扩展名为 .wgsl 的源代码文件进行高亮显示。
WebGPU 工具分享 - WGSL 代码高亮插件(VSCode)与预处理工具
文章图片

② WGSL Literal 插件 这个插件允许你在 JavaScript / TypeScript 的模板字符串中进行 wgsl 代码高亮,需要加上模板字符串前置块注释:

const code = /* wgsl */` struct FragmentInput { @location(0) Color: vec3; }; @stage(fragment) fn main(input: FragmentInput) -> @location(0) vec4 { return vec4(input.Color, 1.0); } `;

WebGPU 工具分享 - WGSL 代码高亮插件(VSCode)与预处理工具
文章图片

遗憾的是,截至 2022年3月25日,这两个插件并没有代码格式化功能(几乎没有),也没有代码智能提示功能。
③ WGSL Preprocessor 这是一个 JavaScript / TypeScript 模板字符串 wgsl 预处理函数包,模板字符串除了可插值外,还可以使用前置函数进行预处理。
这个函数包目前只是一个 esm 模块文件,以后不排除会变成更大的 npm 包,由 toji(Brandon Jones)维护。
github.com/toji/wgsl-preprocessor?github.com
这个使得 wgsl 拥有了 glsl 类似的宏定义等语法:
目前支持:
  • #if
  • #elif
  • #else
  • #endif
简单用法:
import { wgsl } from './wgsl-preprocessor.js'; function getDebugShader(sRGB = false) { return wgsl` @stage(fragment) fn main() -> @location(0) vec4 { let color = vec4(1.0, 0.0, 0.0, 1.0); #if ${sRGB} let rgb = pow(color.rgb, vec3(1.0 / 2.2)); return vec4(rgb, color.a); #else return color; #endif }`; } `

为什么没有 #define 宏?
【WebGPU 工具分享 - WGSL 代码高亮插件(VSCode)与预处理工具】因为模板字符串的插值功能已经可以当 #define 宏使用了,你甚至都不需要使用这个字符串预处理函数。
const ambientFactor = 1.0; const sampleCount = 2; const source = ` let ambientFactor = f32(${ambientFactor}); for (var i = 0u; i < ${sampleCount}u; i = i + 1u) { // Etc... } `;

    推荐阅读