From ecdafb45dd9c416cb0810d6687a20ac97e480ac9 Mon Sep 17 00:00:00 2001
From: eug-vs <eugene@eug-vs.xyz>
Date: Wed, 11 Dec 2024 02:12:22 +0100
Subject: refactor: allow for other solver implementations

---
 src/solver/midpoint.rs | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 src/solver/midpoint.rs

(limited to 'src/solver/midpoint.rs')

diff --git a/src/solver/midpoint.rs b/src/solver/midpoint.rs
new file mode 100644
index 0000000..c5a01c4
--- /dev/null
+++ b/src/solver/midpoint.rs
@@ -0,0 +1,18 @@
+use crate::particle_system::{ParticleSystem, Scalar};
+use super::{PhaseSpace, Solver};
+
+impl Solver for ParticleSystem {
+    fn step(&mut self, dt: Scalar) {
+        let mut state = self.collect_phase_space();
+
+        // Shift to the midpoint
+        self.scatter_phase_space(&&PhaseSpace {
+            0: state.0.clone() + self.compute_derivative().0 * dt / 2.0,
+        });
+
+        state.0 += self.compute_derivative().0 * dt;
+        self.scatter_phase_space(&state);
+
+        self.t += dt;
+    }
+}
-- 
cgit v1.2.3