X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fgame%2Fstage.h;h=0c2219bdc18b1d893851ed91467677c5a35fc4b1;hb=e4f03880d49bdbe0c7269be0f40f23b197bcea77;hp=a046947f57ad7d0f49e43621f003ac739a4d5713;hpb=fce6e7846814e54de9d38000d2f5e2295ee2119e;p=libs%2Fgame.git

diff --git a/source/game/stage.h b/source/game/stage.h
index a046947..0c2219b 100644
--- a/source/game/stage.h
+++ b/source/game/stage.h
@@ -8,6 +8,8 @@
 #include "events.h"
 #include "eventsource.h"
 #include "handle.h"
+#include "reflection.h"
+#include "systemscheduler.h"
 
 namespace Msp::Game {
 
@@ -22,6 +24,7 @@ public:
 		Events::ComponentCreated, Events::ComponentDestroyed, Events::CameraChanged>;
 
 private:
+	Reflection::Reflector &reflector;
 	DataFile::Collection &resources;
 	PoolPool pools;
 	EventBus event_bus;
@@ -31,12 +34,15 @@ private:
 	to put it in a pool. */
 	std::unique_ptr<Root> root;
 	std::vector<std::unique_ptr<System>> systems;
+	SystemScheduler scheduler;
 	Handle<Camera> active_camera;
+	bool pending_reschedule = false;
 
 public:
-	Stage(DataFile::Collection &);
+	Stage(Reflection::Reflector &, DataFile::Collection &);
 	~Stage();
 
+	Reflection::Reflector &get_reflector() const { return reflector; }
 	DataFile::Collection &get_resources() const { return resources; }
 	PoolPool &get_pools() { return pools; }
 	EventBus &get_event_bus() { return event_bus; }
@@ -75,8 +81,13 @@ inline void Stage::iterate_objects(const F &func)
 template<typename T, typename... Args>
 inline T &Stage::add_system(Args &&... args)
 {
-	systems.emplace_back(std::make_unique<T>(*this, std::forward<Args>(args)...));
-	return static_cast<T &>(*systems.back());
+	// Ensure that a reflected class exists for scheduling
+	reflector.get_or_create_class<T>();
+
+	auto &sys = systems.emplace_back(std::make_unique<T>(*this, std::forward<Args>(args)...));
+	scheduler.add_system(*sys);
+	pending_reschedule = true;
+	return static_cast<T &>(*sys);
 }
 
 template<typename T>