#pragma once

#include <random>
#include <vector>

#include <entt/entity/registry.hpp>


namespace kurator
{
namespace sim
{


class TeamManager
{
public:
	using Team = std::vector<entt::entity>;
	TeamManager();
	void add(int team, entt::entity entity);
	Team get(int team) const;
	entt::entity random(int team);
	void clear(entt::registry& registry);
private:
	std::vector<Team> teams;
	std::mt19937 generator;
};


}  // namespace sim
}  // namespace kurator