summaryrefslogtreecommitdiff
path: root/sim/src/TeamManager.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'sim/src/TeamManager.cpp')
-rw-r--r--sim/src/TeamManager.cpp11
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)