WGSL Code Generation

Complete code generation from ShaderM monad to WGSL modules.

Features:

• Automatic binding assignment for storage buffers
• Workgroup variable declarations
• Struct definitions
• Function generation with attributes
• Extension and diagnostic directives
• Complete module assembly

Usage:

def myShader : ShaderM Unit := do
  -- ... shader code ...

let module := generateModule "myKernel" {x := 256, y := 1, z := 1} [] myShader
IO.println module.toWGSL

structure Hesper.WGSL.CodeGen.FunctionParam : Type

Function parameter with optional built-in binding

• name : String
• ty : WGSLType
• builtin : Option BuiltinBinding

def Hesper.WGSL.CodeGen.FunctionParam.toWGSL (param : FunctionParam) : String

Generate WGSL parameter declaration

Equations

• One or more equations did not get rendered due to their size.

structure Hesper.WGSL.CodeGen.FunctionDecl : Type

Function declaration with attributes, parameters, and body

• name : String
• attributes : List String
• params : List FunctionParam
• returnType : Option WGSLType
• body : List Stmt

def Hesper.WGSL.CodeGen.FunctionDecl.toWGSL (func : FunctionDecl) : String

Generate complete WGSL function

Equations

• One or more equations did not get rendered due to their size.

structure Hesper.WGSL.CodeGen.ShaderModule : Type

Complete shader module with all declarations

• extensions : List String
• diagnostics : List (String × String)
• structs : List StructDef
• storageBuffers : List StorageBuffer
• workgroupVars : List WorkgroupVar
• functions : List FunctionDecl

def Hesper.WGSL.CodeGen.ShaderModule.storageBufferDecl (buf : StorageBuffer) : String

Generate storage buffer declaration with binding

Equations

• One or more equations did not get rendered due to their size.

def Hesper.WGSL.CodeGen.ShaderModule.workgroupVarDecl (wvar : WorkgroupVar) : String

Generate workgroup variable declaration

Equations

• Hesper.WGSL.CodeGen.ShaderModule.workgroupVarDecl wvar = toString "var<workgroup> " ++ toString wvar.name ++ toString ": " ++ toString wvar.type.toWGSL ++ toString ";\n"

def Hesper.WGSL.CodeGen.ShaderModule.structDecl (struct : StructDef) : String

Generate struct definition

Equations

• One or more equations did not get rendered due to their size.

def Hesper.WGSL.CodeGen.ShaderModule.toWGSL (module : ShaderModule) : String

Generate complete WGSL module

Equations

• One or more equations did not get rendered due to their size.

def Hesper.WGSL.CodeGen.generateComputeModule (funcName : String) (workgroupSize : WorkgroupSize) (extensions : List String := []) (computation : Monad.ShaderM Unit) : ShaderModule

Generate a compute shader module from ShaderM monad

Equations

• One or more equations did not get rendered due to their size.

def Hesper.WGSL.CodeGen.generateComputeModuleWithDiagnostics (funcName : String := "main") (workgroupSize : WorkgroupSize := { x := 256 }) (extensions : List String := []) (diagnostics : List (String × String) := []) (computation : Monad.ShaderM Unit) : ShaderModule

Generate compute shader module with diagnostics support

Equations

• One or more equations did not get rendered due to their size.

def Hesper.WGSL.CodeGen.generateWGSL (funcName : String := "main") (workgroupSize : WorkgroupSize := { x := 256 }) (extensions : List String := []) (diagnostics : List (String × String) := []) (computation : Monad.ShaderM Unit) : String

Convenience function: Generate WGSL string from ShaderM computation

Equations

• One or more equations did not get rendered due to their size.

def Hesper.WGSL.CodeGen.generateWGSLSimple (computation : Monad.ShaderM Unit) : String

Generate WGSL with default parameters

Equations

• Hesper.WGSL.CodeGen.generateWGSLSimple computation = Hesper.WGSL.CodeGen.generateWGSL "main" { x := 256 } [] [] computation