WIP

src/pipelines/agents/agent-generation/agent-compaction.wgsl

@@ -0,0 +1,36 @@
+struct Settings {
+  agentCount: u32,
+  padding0: u32,
+  padding1: u32,
+  padding2: u32,
+};
+
+struct Counters {
+  aliveAgentCount: atomic<u32>,
+  padding0: atomic<u32>,
+  padding1: atomic<u32>,
+};
+
+@group(1) @binding(0) var<uniform> settings: Settings;
+@group(1) @binding(2) var<storage, read_write> counters: Counters;
+@group(1) @binding(3) var<storage, read_write> compactedAgents: array<Agent>;
+
+@compute @workgroup_size(64)
+fn main(
+  @builtin(global_invocation_id) global_id: vec3<u32>,
+  @builtin(num_workgroups) workgroup_count: vec3<u32>
+) {
+  let id = get_id(global_id, workgroup_count);
+
+  if id >= settings.agentCount {
+    return;
+  }
+
+  let agent = agents[id];
+  if agent.colorIndex < 0.0 {
+    return;
+  }
+
+  let compactedIndex = atomicAdd(&counters.aliveAgentCount, 1);
+  compactedAgents[compactedIndex] = agent;
+}