Fixed!

Author: iwoplaza

apps/typegpu-docs/src/examples/simulation/gravity/compute.ts

@@ -142,29 +142,28 @@ export const computeGravityShader = tgpu['~unstable'].computeFn({
     destroyed: computeLayout.$.inState[currentId].destroyed,
   });
 
-  if (current.destroyed === 1) {
-    return;
-  }
+  if (current.destroyed === 0) {
+    for (let i = 0; i < computeLayout.$.celestialBodiesCount; i++) {
+      const other = computeLayout.$.inState[i];
 
-  for (let i = 0; i < computeLayout.$.celestialBodiesCount; i++) {
-    const other = computeLayout.$.inState[i];
+      if (d.u32(i) === input.gid.x || other.destroyed === 1) {
+        continue;
+      }
 
-    if (d.u32(i) === input.gid.x || other.destroyed === 1) {
-      continue;
-    }
+      const dist = std.max(
+        radiusOf(current) + radiusOf(other),
+        std.distance(current.position, other.position),
+      );
+      const gravityForce = (current.mass * other.mass) / dist / dist;
 
-    const dist = std.max(
-      radiusOf(current) + radiusOf(other),
-      std.distance(current.position, other.position),
-    );
-    const gravityForce = (current.mass * other.mass) / dist / dist;
+      const direction = std.normalize(other.position.sub(current.position));
+      current.velocity = current.velocity.add(
+        direction.mul((gravityForce / current.mass) * dt),
+      );
+    }
 
-    const direction = std.normalize(other.position.sub(current.position));
-    current.velocity = current.velocity.add(
-      direction.mul((gravityForce / current.mass) * dt),
-    );
+    current.position = current.position.add(current.velocity.mul(dt));
   }
 
-  current.position = current.position.add(current.velocity.mul(dt));
   computeLayout.$.outState[input.gid.x] = CelestialBody(current);
 });

packages/typegpu/tests/examples/individual/gravity.test.ts

@@ -131,20 +131,19 @@ describe('gravity example', () => {
         let dt = (time_1.passed * time_1.multiplier);
         let currentId = input.gid.x;
         var current = CelestialBody_4(inState_3[currentId].destroyed, inState_3[currentId].position, inState_3[currentId].velocity, inState_3[currentId].mass, inState_3[currentId].radiusMultiplier, inState_3[currentId].collisionBehavior, inState_3[currentId].textureIndex, inState_3[currentId].ambientLightFactor);
-        if ((current.destroyed == 1)) {
-          return;
-        }
-        for (var i = 0; (i < celestialBodiesCount_5); i++) {
-          let other = (&inState_3[i]);
-          if (((u32(i) == input.gid.x) || ((*other).destroyed == 1))) {
-            continue;
+        if ((current.destroyed == 0)) {
+          for (var i = 0; (i < celestialBodiesCount_5); i++) {
+            let other = (&inState_3[i]);
+            if (((u32(i) == input.gid.x) || ((*other).destroyed == 1))) {
+              continue;
+            }
+            let dist = max((radiusOf_6(current) + radiusOf_6((*other))), distance(current.position, (*other).position));
+            let gravityForce = (((current.mass * (*other).mass) / dist) / dist);
+            var direction = normalize(((*other).position - current.position));
+            current.velocity = (current.velocity + (direction * ((gravityForce / current.mass) * dt)));
           }
-          let dist = max((radiusOf_6(current) + radiusOf_6((*other))), distance(current.position, (*other).position));
-          let gravityForce = (((current.mass * (*other).mass) / dist) / dist);
-          var direction = normalize(((*other).position - current.position));
-          current.velocity = (current.velocity + (direction * ((gravityForce / current.mass) * dt)));
+          current.position = (current.position + (current.velocity * dt));
         }
-        current.position = (current.position + (current.velocity * dt));
         outState_7[input.gid.x] = current;
       }