feat: add full system update

Author: art049

benches/particle_simulation.rs

@@ -35,6 +35,17 @@ fn aos_apply_gravity(c: &mut Criterion) {
     });
 }
 
+fn aos_full_update(c: &mut Criterion) {
+    c.bench_function("aos_full_update", |b| {
+        let mut system = aos::ParticleSystem::new(MEDIUM);
+        let gravity = Vec3::new(0.0, -9.81, 0.0);
+        let dt = 0.016;
+        b.iter(|| {
+            black_box(system.update(black_box(gravity), black_box(dt)));
+        });
+    });
+}
+
 // ============================================================================
 // Structure of Arrays - Cache Friendly
 // ============================================================================
@@ -67,13 +78,26 @@ fn soa_apply_gravity(c: &mut Criterion) {
     });
 }
 
+fn soa_full_update(c: &mut Criterion) {
+    c.bench_function("soa_full_update", |b| {
+        let mut system = soa::ParticleSystem::new(MEDIUM);
+        let gravity = Vec3::new(0.0, -9.81, 0.0);
+        let dt = 0.016;
+        b.iter(|| {
+            black_box(system.update(black_box(gravity), black_box(dt)));
+        });
+    });
+}
+
 criterion_group!(
     benches,
     aos_update_positions,
     aos_kinetic_energy,
     aos_apply_gravity,
+    aos_full_update,
     soa_update_positions,
     soa_kinetic_energy,
-    soa_apply_gravity
+    soa_apply_gravity,
+    soa_full_update
 );
 criterion_main!(benches);

src/aos.rs

@@ -71,4 +71,13 @@ impl ParticleSystem {
             particle.velocity = particle.velocity.add(&gravity.scale(dt));
         }
     }
+
+    /// Full simulation update: apply forces, update positions, and compute energy
+    /// This simulates a typical physics frame with multiple passes over the data
+    #[inline(never)]
+    pub fn update(&mut self, gravity: Vec3, dt: f32) -> f32 {
+        self.apply_gravity(gravity, dt);
+        self.update_positions(dt);
+        self.compute_kinetic_energy()
+    }
 }

src/soa.rs

@@ -61,4 +61,13 @@ impl ParticleSystem {
             *velocity = velocity.add(&gravity.scale(dt));
         }
     }
+
+    /// Full simulation update: apply forces, update positions, and compute energy
+    /// This simulates a typical physics frame with multiple passes over the data
+    #[inline(never)]
+    pub fn update(&mut self, gravity: Vec3, dt: f32) -> f32 {
+        self.apply_gravity(gravity, dt);
+        self.update_positions(dt);
+        self.compute_kinetic_energy()
+    }
 }