diff options
Diffstat (limited to 'sim/src/TeamManager.cpp')
-rw-r--r-- | sim/src/TeamManager.cpp | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/sim/src/TeamManager.cpp b/sim/src/TeamManager.cpp index 7252acb..778cd18 100644 --- a/sim/src/TeamManager.cpp +++ b/sim/src/TeamManager.cpp @@ -8,7 +8,9 @@ #include <entt/entity/registry.hpp> #include <kurator/engine/Context.h> +#include <kurator/sim/components.h> #include <kurator/sim/events.h> +#include <kurator/universe/ShipType.h> namespace kurator @@ -25,6 +27,15 @@ TeamManager::TeamManager() : void +TeamManager::extend(entt::registry& registry) +{ + auto ships = registry.view<universe::ShipType, sim::Team>(); + for (const auto& [entity, type, team] : ships.each()) + add(team.id, entity); +} + + +void TeamManager::add(int team, entt::entity entity) { for (int i = teams.size(); i < team + 1; ++i) |