FightGroupOverlay.h

#pragma once
#include "Outpost2DLL/Outpost2DLL.h"
#include <vector>

struct TargetTankCount
{
	map_id tankType;
	map_id weaponType;
	int count;
};

class FightGroupOverlay
{
public:
	FightGroupOverlay(PlayerNum aiPlayerNum, int humanPlayerCount);
	virtual void Initialize(MAP_RECT guardedRect, std::vector<Unit> vehicleFactories);
	bool IsFull();
	
protected:
	const PlayerNum aiPlayerNum;
	const std::size_t humanPlayerCount;
	std::vector<Unit> vehicleFactories;
	
	void AddTank(map_id tankType, map_id weapon);

	void ClearTargetCount();
	void SetTankCounts(const std::vector<TargetTankCount>& targetTankCounts);
	void SetLynxCount(map_id weaponType, int targetCount);
	void SetPantherCount(map_id weaponType, int targetCount);
	void SetTigerCount(map_id weaponType, int targetCount);
	void SetScorpionCount(int targetCount);

	MAP_RECT GetGuardedRect();
	
	int GetFightGroupCount();
	
	void BasicAttack();
	FightGroup GetFightGroup();
	int targetCount;

private:
	MAP_RECT guardedRect;
	BuildingGroup buildingGroup;
	FightGroup fightGroup;
};