source/game/stage.h

   1 #ifndef MSP_GAME_STAGE_H_
   2 #define MSP_GAME_STAGE_H_
   3
   4 #include <memory>
   5 #include <msp/time/timedelta.h>
   6 #include "eventbus.h"
   7 #include "events.h"
   8 #include "eventsource.h"
   9 #include "handle.h"
  10
  11 namespace Msp::Game {
  12
  13 class Root;
  14 class System;
  15
  16 class Stage
  17 {
  18 public:
  19         using EventSource = Game::EventSource<Events::EntityCreated, Events::EntityDestroyed,
  20                 Events::ComponentCreated, Events::ComponentDestroyed>;
  21
  22 private:
  23         PoolPool pools;
  24         EventBus event_bus;
  25         EventSource event_source;
  26         /* Use unique_ptr because there's only one root per stage so it's pointless
  27         to put it in a pool. */
  28         std::unique_ptr<Root> root;
  29         std::vector<std::unique_ptr<System>> systems;
  30
  31 public:
  32         Stage();
  33         ~Stage();
  34
  35         PoolPool &get_pools() { return pools; }
  36         EventBus &get_event_bus() { return event_bus; }
  37         EventSource &get_event_source() { return event_source; }
  38         Handle<Root> get_root() { return Handle<Root>::from_object(root.get()); }
  39
  40         template<typename T, typename F>
  41         void iterate_objects(const F &);
  42
  43         template<typename T, typename... Args>
  44         T &add_system(Args &&...);
  45
  46         const std::vector<std::unique_ptr<System>> &get_systems() const { return systems; }
  47
  48         template<typename T>
  49         T *get_system() const;
  50
  51         void tick(Time::TimeDelta);
  52 };
  53
  54 template<typename T, typename F>
  55 inline void Stage::iterate_objects(const F &func)
  56 {
  57         pools.get_pool<T>().iterate_objects(func);
  58 }
  59
  60 template<typename T, typename... Args>
  61 inline T &Stage::add_system(Args &&... args)
  62 {
  63         systems.emplace_back(std::make_unique<T>(*this, std::forward<Args>(args)...));
  64         return static_cast<T &>(*systems.back());
  65 }
  66
  67 template<typename T>
  68 inline T *Stage::get_system() const
  69 {
  70         for(const auto &s: systems)
  71                 if(T *ts = dynamic_cast<T *>(s.get()))
  72                         return ts;
  73         return nullptr;
  74 }
  75
  76 } // namespace Msp::Game
  77
  78 #endif