windjammer 0.48.0

A simple language inspired by Go, Ruby, and Elixir that transpiles to Rust - 80% of Rust's power with 20% of the complexity
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130

#![cfg(any(
    not(any(
        feature = "parser_tests",
        feature = "analyzer_tests",
        feature = "codegen_tests",
        feature = "interpreter_tests",
        feature = "conformance_tests",
        feature = "integration_tests",
    )),
    feature = "parser_tests",
))]

//! TDD tests for WJSL → WGSL codegen
//!
//! Tests the transpiler that converts WJSL AST to WGSL source code.

use windjammer::wjsl::transpile_wjsl;

#[test]
fn test_transpile_vertex_shader() {
    let source = r#"
@vertex
fn main(@builtin(vertex_index) idx: u32) -> @builtin(position) vec4 {
    return vec4(0.0, 0.0, 0.0, 1.0);
}
"#;
    let wgsl = transpile_wjsl(source).unwrap();
    assert!(wgsl.contains("@vertex"));
    assert!(wgsl.contains("@builtin(vertex_index)"));
    assert!(wgsl.contains("@builtin(position)"));
    assert!(wgsl.contains("vec4<f32>"));
    assert!(wgsl.contains("return vec4(0.0, 0.0, 0.0, 1.0)"));
}

#[test]
fn test_transpile_uniforms() {
    let source = r#"
struct CameraUniforms {
    view_matrix: mat4x4,
    proj_matrix: mat4x4,
}

@group(0) @binding(0) uniform camera: CameraUniforms;
@group(0) @binding(1) uniform params: vec2<f32>;

@vertex
fn main(@builtin(vertex_index) idx: u32) -> @builtin(position) vec4 {
    return vec4(0.0, 0.0, 0.0, 1.0);
}
"#;
    let wgsl = transpile_wjsl(source).unwrap();
    assert!(wgsl.contains("struct CameraUniforms"));
    assert!(wgsl.contains("view_matrix"));
    assert!(wgsl.contains("mat4x4<f32>"));
    assert!(wgsl.contains("@group(0) @binding(0) var<uniform> camera"));
    assert!(wgsl.contains("@group(0) @binding(1) var<uniform> params"));
    assert!(wgsl.contains("vec2<f32>"));
}

#[test]
fn test_transpile_compute_shader() {
    let source = r#"
@compute @workgroup_size(8, 8, 1)
fn main(@builtin(global_invocation_id) id: vec3<u32>) {
    if (id.x >= 128u || id.y >= 128u) { return; }
}
"#;
    let wgsl = transpile_wjsl(source).unwrap();
    assert!(wgsl.contains("@compute"));
    assert!(wgsl.contains("@workgroup_size(8, 8, 1)"));
    assert!(wgsl.contains("@builtin(global_invocation_id)"));
    assert!(wgsl.contains("vec3<u32>"));
}

#[test]
fn test_transpile_storage_bindings() {
    let source = r#"
@group(0) @binding(0) storage read svo_nodes: array<u32>;
@group(0) @binding(1) storage read_write gbuffer: array<vec4>;

@compute @workgroup_size(8, 8, 1)
fn main(@builtin(global_invocation_id) id: vec3<u32>) {
    return;
}
"#;
    let wgsl = transpile_wjsl(source).unwrap();
    assert!(wgsl.contains("var<storage, read> svo_nodes"));
    assert!(wgsl.contains("var<storage, read_write> gbuffer"));
    assert!(wgsl.contains("array<u32>"));
    assert!(wgsl.contains("array<vec4<f32>>"));
}

#[test]
fn test_transpile_texture_and_sampler() {
    let source = r#"
@group(0) @binding(0) texture_2d albedo_map: texture_2d<f32>;
@group(0) @binding(1) sampler tex_sampler: sampler;

@fragment
fn main(@location(0) uv: vec2) -> @location(0) vec4 {
    return vec4(0.0, 0.0, 0.0, 1.0);
}
"#;
    let wgsl = transpile_wjsl(source).unwrap();
    assert!(wgsl.contains("var albedo_map: texture_2d<f32>"));
    assert!(wgsl.contains("var tex_sampler: sampler"));
    assert!(wgsl.contains("@fragment"));
    assert!(wgsl.contains("@location(0)"));
}

#[test]
fn test_transpile_helper_functions() {
    let source = r#"
fn tonemap_aces(color: vec3) -> vec3 {
    let a = 2.51;
    return clamp(color * a, vec3(0.0), vec3(1.0));
}

@fragment
fn main(@location(0) color: vec4) -> @location(0) vec4 {
    let lit = tonemap_aces(color.rgb);
    return vec4(lit, 1.0);
}
"#;
    let wgsl = transpile_wjsl(source).unwrap();
    assert!(wgsl.contains("fn tonemap_aces"));
    assert!(wgsl.contains("color: vec3<f32>"));
    assert!(wgsl.contains("-> vec3<f32>"));
    assert!(wgsl.contains("tonemap_aces(color.rgb)"));
}