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
|
// Rotating cube shader v2 (simplified, no masking)
struct Uniforms {
view_proj: mat4x4<f32>,
inv_view_proj: mat4x4<f32>,
camera_pos_time: vec4<f32>,
params: vec4<f32>,
resolution: vec2<f32>,
aspect_ratio: f32,
_pad: f32,
};
struct ObjectData {
model: mat4x4<f32>,
inv_model: mat4x4<f32>,
color: vec4<f32>,
params: vec4<f32>,
};
@group(0) @binding(0) var<uniform> uniforms: Uniforms;
@group(0) @binding(1) var<storage, read> object: ObjectData;
struct VSOut {
@builtin(position) pos: vec4<f32>,
@location(0) world_pos: vec3<f32>,
@location(1) normal: vec3<f32>,
};
// Cube vertices (hardcoded)
fn get_cube_vertex(vid: u32) -> vec3<f32> {
let positions = array<vec3<f32>, 36>(
// Front face
vec3<f32>(-1, -1, 1), vec3<f32>( 1, -1, 1), vec3<f32>( 1, 1, 1),
vec3<f32>(-1, -1, 1), vec3<f32>( 1, 1, 1), vec3<f32>(-1, 1, 1),
// Back face
vec3<f32>( 1, -1, -1), vec3<f32>(-1, -1, -1), vec3<f32>(-1, 1, -1),
vec3<f32>( 1, -1, -1), vec3<f32>(-1, 1, -1), vec3<f32>( 1, 1, -1),
// Right face
vec3<f32>( 1, -1, 1), vec3<f32>( 1, -1, -1), vec3<f32>( 1, 1, -1),
vec3<f32>( 1, -1, 1), vec3<f32>( 1, 1, -1), vec3<f32>( 1, 1, 1),
// Left face
vec3<f32>(-1, -1, -1), vec3<f32>(-1, -1, 1), vec3<f32>(-1, 1, 1),
vec3<f32>(-1, -1, -1), vec3<f32>(-1, 1, 1), vec3<f32>(-1, 1, -1),
// Top face
vec3<f32>(-1, 1, 1), vec3<f32>( 1, 1, 1), vec3<f32>( 1, 1, -1),
vec3<f32>(-1, 1, 1), vec3<f32>( 1, 1, -1), vec3<f32>(-1, 1, -1),
// Bottom face
vec3<f32>(-1, -1, -1), vec3<f32>( 1, -1, -1), vec3<f32>( 1, -1, 1),
vec3<f32>(-1, -1, -1), vec3<f32>( 1, -1, 1), vec3<f32>(-1, -1, 1)
);
return positions[vid];
}
fn get_cube_normal(vid: u32) -> vec3<f32> {
let face_id = vid / 6u;
let normals = array<vec3<f32>, 6>(
vec3<f32>( 0, 0, 1), // Front
vec3<f32>( 0, 0, -1), // Back
vec3<f32>( 1, 0, 0), // Right
vec3<f32>(-1, 0, 0), // Left
vec3<f32>( 0, 1, 0), // Top
vec3<f32>( 0, -1, 0) // Bottom
);
return normals[face_id];
}
@vertex fn vs_main(@builtin(vertex_index) vid: u32) -> VSOut {
let local_pos = get_cube_vertex(vid);
let local_normal = get_cube_normal(vid);
let world_pos = object.model * vec4<f32>(local_pos, 1.0);
let world_normal = normalize((object.model * vec4<f32>(local_normal, 0.0)).xyz);
let clip_pos = uniforms.view_proj * world_pos;
return VSOut(clip_pos, world_pos.xyz, world_normal);
}
@fragment fn fs_main(@location(0) world_pos: vec3<f32>, @location(1) normal: vec3<f32>) -> @location(0) vec4<f32> {
let N = normalize(normal);
let light_dir = normalize(vec3<f32>(1.0, 1.0, 1.0));
let diffuse = max(dot(N, light_dir), 0.0);
let ambient = 0.3;
let lighting = ambient + diffuse * 0.7;
let color = object.color.rgb * lighting;
return vec4<f32>(color, 1.0);
}
|
