vcmi/hch/CTownHandler.h

#ifndef CTOWNHANDLER_H
#define CTOWNHANDLER_H

#include "CDefHandler.h"
#include "CCreatureHandler.h"
#include "SDL.h"
#include "../int3.h"
#include <string>
#include <vector>

class CBuilding;
class CSpell;
class CHero;

class CTown
{
public:
	std::string name; //name of type
	std::vector<std::string> names; //names of the town instances
	int bonus; //pic number
};

class CTownHandler
{
	CDefHandler * smallIcons;
public:
	CTownHandler();
	~CTownHandler();
	std::vector<CTown> towns;
	void loadNames();
	SDL_Surface * getPic(int ID, bool fort=true, bool builded=false); //ID=-1 - blank; -2 - border; -3 - random
	static int getTypeByDefName(std::string name);
};

class CTownInstance
{
public:
	int type; //type of town
	int owner; //ID of owner
	int3 pos; //position
	CTown * town;
	std::string name; // name of town
	CCreatureSet garrison;
	int builded; //how many buildings has been built this turn
	int destroyed; //how many buildings has been destroyed this turn
	
	//TODO:
	std::vector<CBuilding *> allBuildings, possibleBuildings, builtBuildings;
	std::vector<int> creatureIncome; //vector by level
	std::vector<int> creaturesLeft; //that can be recruited

	CHero * garrisonHero;

	std::vector<CSpell *> possibleSpells, obligatorySpells, availableSpells;
};

#endif //CTOWNHANDLER_H