Skip to content

docs: Optimize 3d slime mold example #1900

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
reczkok wants to merge 9 commits into
base: release
Choose a base branch
from
Open

docs: Optimize 3d slime mold example #1900

Changes from 1 commit
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
99bef46
chore: Jelly-slider wgsl resolution test (#1884)
cieplypolar Nov 6, 2025
dae310b
better blur and adaptive steps + jitter
reczkok Nov 7, 2025
27cd85c
revert some random files
reczkok Nov 7, 2025
22b03fa
update tests
reczkok Nov 12, 2025
a083cb1
remove unused feature
reczkok Nov 12, 2025
7ab184e
Add double buffering
Nov 21, 2025
c42f99c
Update tests
Nov 21, 2025
621f9a9
Unroll a for loop
Nov 27, 2025
b47b413
Clean up direction updating
Nov 27, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Update tests
  • Loading branch information
Aleksander Katan committed Nov 21, 2025
commit c42f99c8fffa2dff9e7ba9b2b2fb1a8e31a0df24
31 changes: 18 additions & 13 deletions packages/typegpu/tests/examples/individual/slime-mold-3d.test.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -59,21 +59,24 @@ describe('slime mold 3d example', () => {
direction: vec3f,
}

@group(0) @binding(1) var<storage, read_write> agentsData_10: array<Agent_11, 800000>;
@group(0) @binding(1) var<storage, read_write> item_10: array<Agent_11, 800000>;

@group(0) @binding(2) var<storage, read_write> item_12: array<Agent_11, 800000>;

fn wrappedCallback_2(x: u32, _arg_1: u32, _arg_2: u32) {
randSeed_3((f32(x) / 8e+5f));
var pos = ((randInUnitSphere_6() * 64.) + vec3f(128));
var center = vec3f(128);
var dir = normalize((center - pos));
agentsData_10[x] = Agent_11(pos, dir);
item_10[x] = Agent_11(pos, dir);
item_12[x] = Agent_11(pos, dir);
}

struct mainCompute_Input_12 {
struct mainCompute_Input_13 {
@builtin(global_invocation_id) id: vec3u,
}

@compute @workgroup_size(256, 1, 1) fn mainCompute_0(in: mainCompute_Input_12) {
@compute @workgroup_size(256, 1, 1) fn mainCompute_0(in: mainCompute_Input_13) {
if (any(in.id >= sizeUniform_1)) {
return;
}
Expand Down Expand Up @@ -130,14 +133,14 @@ describe('slime mold 3d example', () => {
seed_2(seed);
}

@group(1) @binding(0) var oldState_4: texture_storage_3d<r32float, read>;
@group(1) @binding(1) var oldState_4: texture_storage_3d<r32float, read>;

struct Agent_6 {
position: vec3f,
direction: vec3f,
}

@group(0) @binding(0) var<storage, read_write> agentsData_5: array<Agent_6, 800000>;
@group(1) @binding(0) var<storage, read> oldAgents_5: array<Agent_6>;

fn item_8() -> f32 {
var a = dot(seed_3, vec2f(23.140779495239258, 232.6168975830078));
Expand Down Expand Up @@ -176,7 +179,7 @@ describe('slime mold 3d example', () => {
evaporationRate: f32,
}

@group(0) @binding(1) var<uniform> params_11: Params_12;
@group(0) @binding(0) var<uniform> params_11: Params_12;

struct SenseResult_13 {
weightedDir: vec3f,
Expand Down Expand Up @@ -227,20 +230,22 @@ describe('slime mold 3d example', () => {
return (sign(alignment) * value);
}

@group(1) @binding(1) var newState_18: texture_storage_3d<r32float, write>;
@group(1) @binding(2) var<storage, read_write> newAgents_18: array<Agent_6>;

@group(1) @binding(3) var newState_19: texture_storage_3d<r32float, write>;

struct updateAgents_Input_19 {
struct updateAgents_Input_20 {
@builtin(global_invocation_id) gid: vec3u,
}

@compute @workgroup_size(64) fn updateAgents_0(_arg_0: updateAgents_Input_19) {
@compute @workgroup_size(64) fn updateAgents_0(_arg_0: updateAgents_Input_20) {
if ((_arg_0.gid.x >= 800000u)) {
return;
}
randSeed_1(((f32(_arg_0.gid.x) / 8e+5f) + 0.1f));
var dims = textureDimensions(oldState_4);
var dimsf = vec3f(dims);
var agent = agentsData_5[_arg_0.gid.x];
var agent = oldAgents_5[_arg_0.gid.x];
var random = randFloat01_7();
var direction = normalize(agent.direction);
var senseResult = sense3D_9(agent.position, direction);
Expand Down Expand Up @@ -285,10 +290,10 @@ describe('slime mold 3d example', () => {
var toCenter = normalize((center - newPos));
direction = normalize(((randomDir * 0.3) + (toCenter * 0.7)));
}
agentsData_5[_arg_0.gid.x] = Agent_6(newPos, direction);
newAgents_18[_arg_0.gid.x] = Agent_6(newPos, direction);
var oldState = textureLoad(oldState_4, vec3u(newPos)).x;
var newState = (oldState + 1f);
textureStore(newState_18, vec3u(newPos), vec4f(newState, 0f, 0f, 1f));
textureStore(newState_19, vec3u(newPos), vec4f(newState, 0f, 0f, 1f));
}

struct fullScreenTriangle_Input_1 {
Expand Down