X-Git-Url: https://git.localhorst.tv/?a=blobdiff_plain;f=src%2Fworld%2Fsim.cpp;h=8b4c2140c2ba96b672c9dd08ac7987223b52c9f9;hb=ed2711d42a7815657bf0653c25b8b9be8b7f1368;hp=46454fd2952197311b72db9bfeace430a44bebdc;hpb=c5556cf5f6813887a3503433c021ccd2e7fae865;p=blobs.git

diff --git a/src/world/sim.cpp b/src/world/sim.cpp
index 46454fd..8b4c214 100644
--- a/src/world/sim.cpp
+++ b/src/world/sim.cpp
@@ -1,6 +1,8 @@
 #include "Simulation.hpp"
 
 #include "Body.hpp"
+#include "Planet.hpp"
+#include "Sun.hpp"
 
 
 namespace blobs {
@@ -8,7 +10,9 @@ namespace world {
 
 Simulation::Simulation(Body &r)
 : root(r)
-, all_bodies()
+, bodies()
+, planets()
+, suns()
 , time(0.0) {
 	AddBody(r);
 }
@@ -19,14 +23,25 @@ Simulation::~Simulation() {
 
 void Simulation::AddBody(Body &b) {
 	b.SetSimulation(*this);
-	all_bodies.push_back(&b);
+	bodies.insert(&b);
+}
+
+void Simulation::AddPlanet(Planet &p) {
+	AddBody(p);
+	planets.insert(&p);
+}
+
+void Simulation::AddSun(Sun &s) {
+	AddBody(s);
+	suns.insert(&s);
 }
 
 void Simulation::Tick() {
 	constexpr double dt = 0.01666666666666666666666666666666;
 	time += dt;
-	for (auto body : all_bodies) {
+	for (auto body : bodies) {
 		body->Rotation(body->Rotation() + dt * body->AngularMomentum() / body->Inertia());
+		body->Cache();
 	}
 }