Emit events on entity and component creation and destruction

[libs/game.git] / source / game / stage.h

diff --git a/source/game/stage.h b/source/game/stage.h
index 060759696914264ce5c177c0d1674a154b253e52..80e7a4b72c25e51141d43718c6bed196ffdf712c 100644 (file)
--- a/source/game/stage.h
+++ b/source/game/stage.h
@@ -3,18 +3,29 @@
 
 #include <memory>
 #include <msp/time/timedelta.h>
+#include "eventbus.h"
+#include "events.h"
+#include "eventsource.h"
 #include "handle.h"
-#include "root.h"
 
 namespace Msp::Game {
 
+class Root;
 class System;
 
 class Stage
 {
+public:
+       using EventSource = Game::EventSource<Events::EntityCreated, Events::EntityDestroyed,
+               Events::ComponentCreated, Events::ComponentDestroyed>;
+
 private:
        PoolPool pools;
-       Root root;
+       EventBus event_bus;
+       EventSource event_source;
+       /* Use unique_ptr because there's only one root per stage so it's pointless
+       to put it in a pool. */
+       std::unique_ptr<Root> root;
        std::vector<std::unique_ptr<System>> systems;
 
 public:
@@ -22,7 +33,9 @@ public:
        ~Stage();
 
        PoolPool &get_pools() { return pools; }
-       Handle<Root> get_root() { return Handle<Root>::from_object(&root); }
+       EventBus &get_event_bus() { return event_bus; }
+       EventSource &get_event_source() { return event_source; }
+       Handle<Root> get_root() { return Handle<Root>::from_object(root.get()); }
 
        template<typename T, typename F>
        void iterate_objects(const F &);