mirror of
https://github.com/bevyengine/bevy
synced 2024-11-10 07:04:33 +00:00
de3ec47f3f
# Objective - Example `compute_shader_game_of_life` is random and not following the rules of the game of life: at each steps, it randomly reads some pixel of the current step and some of the previous step instead of only from the previous step - Fixes #9353 ## Solution - Adopted from #9678 - Added a switch of the texture displayed every frame otherwise the game of life looks wrong - Added a way to display the texture bigger so that I could manually check everything was right --------- Co-authored-by: Sludge <96552222+SludgePhD@users.noreply.github.com> Co-authored-by: IceSentry <IceSentry@users.noreply.github.com>
71 lines
2.3 KiB
WebGPU Shading Language
71 lines
2.3 KiB
WebGPU Shading Language
// The shader reads the previous frame's state from the `input` texture, and writes the new state of
|
|
// each pixel to the `output` texture. The textures are flipped each step to progress the
|
|
// simulation.
|
|
// Two textures are needed for the game of life as each pixel of step N depends on the state of its
|
|
// neighbors at step N-1.
|
|
|
|
@group(0) @binding(0) var input: texture_storage_2d<r32float, read>;
|
|
|
|
@group(0) @binding(1) var output: texture_storage_2d<r32float, write>;
|
|
|
|
fn hash(value: u32) -> u32 {
|
|
var state = value;
|
|
state = state ^ 2747636419u;
|
|
state = state * 2654435769u;
|
|
state = state ^ state >> 16u;
|
|
state = state * 2654435769u;
|
|
state = state ^ state >> 16u;
|
|
state = state * 2654435769u;
|
|
return state;
|
|
}
|
|
|
|
fn randomFloat(value: u32) -> f32 {
|
|
return f32(hash(value)) / 4294967295.0;
|
|
}
|
|
|
|
@compute @workgroup_size(8, 8, 1)
|
|
fn init(@builtin(global_invocation_id) invocation_id: vec3<u32>, @builtin(num_workgroups) num_workgroups: vec3<u32>) {
|
|
let location = vec2<i32>(i32(invocation_id.x), i32(invocation_id.y));
|
|
|
|
let randomNumber = randomFloat(invocation_id.y << 16u | invocation_id.x);
|
|
let alive = randomNumber > 0.9;
|
|
let color = vec4<f32>(f32(alive));
|
|
|
|
textureStore(output, location, color);
|
|
}
|
|
|
|
fn is_alive(location: vec2<i32>, offset_x: i32, offset_y: i32) -> i32 {
|
|
let value: vec4<f32> = textureLoad(input, location + vec2<i32>(offset_x, offset_y));
|
|
return i32(value.x);
|
|
}
|
|
|
|
fn count_alive(location: vec2<i32>) -> i32 {
|
|
return is_alive(location, -1, -1) +
|
|
is_alive(location, -1, 0) +
|
|
is_alive(location, -1, 1) +
|
|
is_alive(location, 0, -1) +
|
|
is_alive(location, 0, 1) +
|
|
is_alive(location, 1, -1) +
|
|
is_alive(location, 1, 0) +
|
|
is_alive(location, 1, 1);
|
|
}
|
|
|
|
@compute @workgroup_size(8, 8, 1)
|
|
fn update(@builtin(global_invocation_id) invocation_id: vec3<u32>) {
|
|
let location = vec2<i32>(i32(invocation_id.x), i32(invocation_id.y));
|
|
|
|
let n_alive = count_alive(location);
|
|
|
|
var alive: bool;
|
|
if (n_alive == 3) {
|
|
alive = true;
|
|
} else if (n_alive == 2) {
|
|
let currently_alive = is_alive(location, 0, 0);
|
|
alive = bool(currently_alive);
|
|
} else {
|
|
alive = false;
|
|
}
|
|
let color = vec4<f32>(f32(alive));
|
|
|
|
textureStore(output, location, color);
|
|
}
|