#pragma once

#include <array>
#include <memory>
#include <vector>

#include <kurator/campaign/Loadout.h>
#include <kurator/universe/Repository.h>

#include "Scene.h"
#include "SceneFrame.h"
#include "Session.h"


namespace kurator
{


struct Group
{
	campaign::Loadout loadout;
	std::array<int, 6> counts = {0};
};


class ScenarioEditor : public Scene
{
public:
	explicit ScenarioEditor(std::shared_ptr<Session> _session);
	void update() override;
	void draw() const override;
	void start_battle();
private:
	void regenerate();
	std::shared_ptr<Session> session;
	std::shared_ptr<universe::Repository> repository;
	std::shared_ptr<Scene> battle;
	SceneFrame frame;
	float radius;
	std::vector<Group> groups;
};


}  // namespace kurator