vcmi/hch/CObjectHandler.h

#ifndef __COBJECTHANDLER_H__
#define __COBJECTHANDLER_H__
#include "../global.h"
#include <string>
#include <vector>
#include <set>
#include <map>
#include "CCreatureHandler.h"

using boost::logic::tribool;
class IGameCallback;
struct BattleResult;
class CCPPObjectScript;
class CGObjectInstance;
class CScript;
class CObjectScript;
class CGHeroInstance;
class CTown;
class CHero;
class CBuilding;
class CSpell;
class CGTownInstance;
class CArtifact;
class CGDefInfo;
class CSpecObjInfo;

class DLL_EXPORT CCastleEvent
{
public:
	std::string name, message;
	std::vector<si32> resources;  //gain / loss of resources
	ui8 players; //players for whom this event can be applied
	ui8 forHuman, forComputer;
	ui32 firstShow; //postpone of first encounter time in days
	ui32 forEvery; //every n days this event will occure
	ui8 bytes[6]; //build specific buildings (raw format, similar to town's)
	si32 gen[7]; //additional creatures in i-th level dwelling

	bool operator<(const CCastleEvent &drugie) const
	{
		return firstShow<drugie.firstShow;
	}

	template <typename Handler> void serialize(Handler &h, const int version)
	{
		h & name & message & resources & players & forHuman & forComputer & firstShow 
			& forEvery & bytes & gen;
	}
};

class CQuest
{
public:
	ui8 missionType; //type of mission: 0 - no mission; 1 - reach level; 2 - reach main statistics values; 3 - win with a certain hero; 4 - win with a certain creature; 5 - collect some atifacts; 6 - have certain troops in army; 7 - collect resources; 8 - be a certain hero; 9 - be a certain playe
	si32 lastDay; //after this day (first day is 0) mission cannot be completed; if -1 - no limit

	ui32 m13489val;
	std::vector<ui32> m2stats;
	std::vector<ui16> m5arts; //artifacts id
	std::vector<std::pair<ui32, ui32> > m6creatures; //pair[cre id, cre count]
	std::vector<ui32> m7resources;

	std::string firstVisitText, nextVisitText, completedText;

	template <typename Handler> void serialize(Handler &h, const int version)
	{
		h & missionType & lastDay & m13489val & m2stats & m5arts & m6creatures & m7resources
			& firstVisitText & nextVisitText & completedText;
	}
};

class DLL_EXPORT IObjectInterface
{
public:
	static IGameCallback *cb;

	IObjectInterface();
	virtual ~IObjectInterface();

	virtual void onHeroVisit(const CGHeroInstance * h) const;
	virtual void onHeroLeave(const CGHeroInstance * h) const;
	virtual void newTurn() const;
	virtual void initObj();
};

class DLL_EXPORT CGObjectInstance : protected IObjectInterface
{
protected:
public:
	mutable std::string hoverName;
	int3 pos; //h3m pos
	si32 ID, subID; //normal ID (this one from OH3 maps ;]) - eg. town=98; hero=34
	si32 id;//number of object in CObjectHandler's vector		
	CGDefInfo * defInfo;
	CSpecObjInfo * info;
	ui8 animPhaseShift;

	ui8 tempOwner;
	ui8 blockVisit; //if non-zero then blocks the tile but is visitable from neighbouring tile

	int getOwner() const;
	void setOwner(int ow);
	int getWidth() const; //returns width of object graphic in tiles
	int getHeight() const; //returns height of object graphic in tiles
	bool visitableAt(int x, int y) const; //returns true if object is visitable at location (x, y) form left top tile of image (x, y in tiles)
	bool blockingAt(int x, int y) const; //returns true if object is blocking location (x, y) form left top tile of image (x, y in tiles)
	bool operator<(const CGObjectInstance & cmp) const;  //screen printing priority comparing
	CGObjectInstance();
	virtual ~CGObjectInstance();
	//CGObjectInstance(const CGObjectInstance & right);
	//CGObjectInstance& operator=(const CGObjectInstance & right);
	virtual const std::string & getHoverText() const;
	//////////////////////////////////////////////////////////////////////////
	void initObj();

	friend class CGameHandler;


	template <typename Handler> void serialize(Handler &h, const int version)
	{
		h & hoverName & pos & ID & subID & id & animPhaseShift & tempOwner & blockVisit;
		//TODO: definfo
	}
};

class  DLL_EXPORT CArmedInstance: public CGObjectInstance
{
public:
	CCreatureSet army; //army
	virtual bool needsLastStack() const; //true if last stack cannot be taken

	template <typename Handler> void serialize(Handler &h, const int version)
	{
		h & army;
	}
};

class DLL_EXPORT CGHeroInstance : public CArmedInstance
{
public:
	//////////////////////////////////////////////////////////////////////////

	mutable int moveDir; //format:	123
					//		8 4
					//		765
	mutable ui8 isStanding, tacticFormationEnabled;

	//////////////////////////////////////////////////////////////////////////

	CHero * type;
	ui32 exp; //experience point
	si32 level; //current level of hero
	std::string name; //may be custom
	std::string biography; //if custom
	si32 portrait; //may be custom
	si32 mana; // remaining spell points
	std::vector<si32> primSkills; //0-attack, 1-defence, 2-spell power, 3-knowledge
	std::vector<std::pair<ui8,ui8> > secSkills; //first - ID of skill, second - level of skill (1 - basic, 2 - adv., 3 - expert); if hero has ability (-1, -1) it meansthat it should have default secondary abilities
	si32 movement; //remaining movement points
	si32 identifier; //from the map file
	ui8 sex;
	struct DLL_EXPORT Patrol
	{
		Patrol(){patrolling=false;patrolRadious=-1;};
		bool patrolling;
		int patrolRadious;
	} patrol;
	ui8 inTownGarrison; // if hero is in town garrison 
	CGTownInstance * visitedTown; //set if hero is visiting town or in the town garrison
	std::vector<ui32> artifacts; //hero's artifacts from bag
	std::map<ui16,ui32> artifWorn; //map<position,artifact_id>; positions: 0 - head; 1 - shoulders; 2 - neck; 3 - right hand; 4 - left hand; 5 - torso; 6 - right ring; 7 - left ring; 8 - feet; 9 - misc1; 10 - misc2; 11 - misc3; 12 - misc4; 13 - mach1; 14 - mach2; 15 - mach3; 16 - mach4; 17 - spellbook; 18 - misc5
	std::set<ui32> spells; //known spells (spell IDs)

	//////////////////////////////////////////////////////////////////////////


	template <typename Handler> void serialize(Handler &h, const int version)
	{
		h & exp & level & name & biography & portrait & mana & primSkills & secSkills & movement
			& identifier & sex & inTownGarrison & artifacts & artifWorn & spells;
		//TODO: type
		//TODO: visited town
	}

	//////////////////////////////////////////////////////////////////////////
	const std::string &getBiography() const;
	bool needsLastStack()const;
	unsigned int getTileCost(const EterrainType & ttype, const Eroad & rdtype, const Eriver & rvtype) const;
	unsigned int getLowestCreatureSpeed() const;
	unsigned int getAdditiveMoveBonus() const;
	float getMultiplicativeMoveBonus() const;
	int3 getPosition(bool h3m) const; //h3m=true - returns position of hero object; h3m=false - returns position of hero 'manifestation'
	int getSightDistance() const; //returns sight distance of this hero
	si32 manaLimit() const; //maximum mana value for this hero (basically 10*knowledge)
	bool canWalkOnSea() const;
	int getCurrentLuck() const;
	int getCurrentMorale() const;
	int getPrimSkillLevel(int id) const;
	ui8 getSecSkillLevel(const int & ID) const; //0 - no skill
	int maxMovePoints(bool onLand) const;
	ui32 getArtAtPos(ui16 pos) const; //-1 - no artifact
	void setArtAtPos(ui16 pos, int art);
	const CArtifact * getArt(int pos) const;
	int getSpellSecLevel(int spell) const; //returns level of secondary ability (fire, water, earth, air magic) known to this hero and applicable to given spell; -1 if error
	static int3 convertPosition(int3 src, bool toh3m); //toh3m=true: manifest->h3m; toh3m=false: h3m->manifest

	//////////////////////////////////////////////////////////////////////////

	void initHero(); 
	void initHero(int SUBID); 
	CGHeroInstance();
	virtual ~CGHeroInstance();

	//////////////////////////////////////////////////////////////////////////
	void initObj();
	void onHeroVisit(const CGHeroInstance * h) const;
};

class DLL_EXPORT CGTownInstance : public CArmedInstance
{
public:
	CTown * town;
	std::string name; // name of town
	si32 builded; //how many buildings has been built this turn
	si32 destroyed; //how many buildings has been destroyed this turn
	const CGHeroInstance * garrisonHero, *visitingHero;
	ui32 identifier; //special identifier from h3m (only > RoE maps)
	si32 alignment;
	std::set<si32> forbiddenBuildings, builtBuildings;
	std::vector<ui32> possibleSpells, obligatorySpells;
	std::vector<std::vector<ui32> > spells; //spells[level] -> vector of spells, first will be available in guild

	struct StrInfo
	{
		std::map<si32,ui32> creatures; //level - available amount

		template <typename Handler> void serialize(Handler &h, const int version)
		{
			h & creatures;
		}
	} strInfo;
	std::set<CCastleEvent> events;

	//////////////////////////////////////////////////////////////////////////


	template <typename Handler> void serialize(Handler &h, const int version)
	{
		h & name & builded & destroyed & identifier & alignment & forbiddenBuildings & builtBuildings
			& possibleSpells & obligatorySpells & spells & strInfo & events;
		//TODO: town
		//TODO: garrison/visiting hero
	}

	//////////////////////////////////////////////////////////////////////////


	bool needsLastStack() const;
	int getSightDistance() const; //returns sight distance
	int fortLevel() const; //0 - none, 1 - fort, 2 - citadel, 3 - castle
	int hallLevel() const; // -1 - none, 0 - village, 1 - town, 2 - city, 3 - capitol
	int mageGuildLevel() const; // -1 - none, 0 - village, 1 - town, 2 - city, 3 - capitol
	bool creatureDwelling(const int & level, bool upgraded=false) const;
	int getHordeLevel(const int & HID) const; //HID - 0 or 1; returns creature level or -1 if that horde structure is not present
	int creatureGrowth(const int & level) const;
	bool hasFort() const;
	bool hasCapitol() const;
	int dailyIncome() const;
	int spellsAtLevel(int level, bool checkGuild) const; //levels are counted from 1 (1 - 5)

	CGTownInstance();
	virtual ~CGTownInstance();

	//////////////////////////////////////////////////////////////////////////


	void onHeroVisit(const CGHeroInstance * h) const;
	void onHeroLeave(const CGHeroInstance * h) const;
	void initObj();
};

class DLL_EXPORT CGVisitableOPH : public CGObjectInstance //objects visitable only once per hero
{
public:
	std::set<si32> visitors; //ids of heroes who have visited this obj
	si8 ttype; //tree type - used only by trees of knowledge: 0 - give level for free; 1 - take 2000 gold; 2 - take 10 gems
	const std::string & getHoverText() const;

	void onHeroVisit(const CGHeroInstance * h) const;
	void onNAHeroVisit(int heroID, bool alreadyVisited) const;
	void initObj();
	void treeSelected(int heroID, int resType, int resVal, int expVal, ui32 result) const; //handle player's anwer to the Tree of Knowledge dialog
};

class DLL_EXPORT CGEvent : public CGObjectInstance //event objects
{
public:
	bool areGuarders; //true if there are
	CCreatureSet guarders;
	bool isMessage; //true if there is a message
	std::string message;
	unsigned int gainedExp;
	int manaDiff; //amount of gained / lost mana
	int moraleDiff; //morale modifier
	int luckDiff; //luck modifier
	int wood, mercury, ore, sulfur, crystal, gems, gold; //gained / lost resources
	unsigned int attack; //added attack points
	unsigned int defence; //added defence points
	unsigned int power; //added power points
	unsigned int knowledge; //added knowledge points
	std::vector<int> abilities; //gained abilities
	std::vector<int> abilityLevels; //levels of gained abilities
	std::vector<int> artifacts; //gained artifacts
	std::vector<int> spells; //gained spells
	CCreatureSet creatures; //gained creatures
	unsigned char availableFor; //players whom this event is available for
	bool computerActivate; //true if computre player can activate this event
	bool humanActivate; //true if human player can activate this event
};

class DLL_EXPORT CGCreature : public CArmedInstance //creatures on map
{
public:
	ui32 identifier; //unique code for this monster (used in missions)
	ui8 character; //chracter of this set of creatures (0 - the most friendly, 4 - the most hostile)
	std::string message; //message printed for attacking hero
	std::vector<ui32> resources; //[res_id], resources given to hero that has won with monsters
	si32 gainedArtifact; //ID of artifact gained to hero, -1 if none
	ui8 neverFlees; //if true, the troops will never flee
	ui8 notGrowingTeam; //if true, number of units won't grow

	void onHeroVisit(const CGHeroInstance * h) const;
	void endBattle(BattleResult *result) const;
	void initObj();
}; 


class DLL_EXPORT CGSignBottle : public CGObjectInstance //signs and ocean bottles
{
	//TODO: generate default message if sign is 'empty'
public:
	std::string message;
};

class DLL_EXPORT CGSeerHut : public CGObjectInstance, public CQuest
{
public:
	char rewardType; //type of reward: 0 - no reward; 1 - experience; 2 - mana points; 3 - morale bonus; 4 - luck bonus; 5 - resources; 6 - main ability bonus (attak, defence etd.); 7 - secondary ability gain; 8 - artifact; 9 - spell; 10 - creature
	//for reward 1
	int r1exp;
	//for reward 2
	int r2mana;
	//for reward 3
	int r3morale;
	//for reward 4
	int r4luck;
	//for reward 5
	unsigned char r5type; //0 - wood, 1 - mercury, 2 - ore, 3 - sulfur, 4 - crystal, 5 - gems, 6 - gold
	int r5amount;
	//for reward 6
	unsigned char r6type; //0 - attack, 1 - defence, 2 - power, 3 - knowledge
	int r6amount;
	//for reward 7
	int r7ability; //ability id
	unsigned char r7level; //1 - basic, 2 - advanced, 3 - expert
	//for reward 8
	int r8art;//artifact id
	//for reward 9
	int r9spell;//spell id
	//for reward 10
	int r10creature; //creature id
	int r10amount;
};

class DLL_EXPORT CGWitchHut : public CGObjectInstance
{
public:
	std::vector<int> allowedAbilities;
};


class DLL_EXPORT CGScholar : public CGObjectInstance
{
public:
	ui8 bonusType; //255 - random, 0 - primary skill, 1 - secondary skill, 2 - spell

	ui8 r0type;
	ui32 r1; //Ability ID
	ui32 r2; //Spell ID
};

class DLL_EXPORT CGGarrison : public CArmedInstance
{
public:
	bool removableUnits;
};

class DLL_EXPORT CGArtifact : public CArmedInstance
{
public:
	std::string message;
	ui32 spell; //if it's spell scroll
	void onHeroVisit(const CGHeroInstance * h) const;
	void initObj();	
};

class DLL_EXPORT CGResource : public CArmedInstance
{
public:
	int amount; //0 if random
	std::string message;

	void onHeroVisit(const CGHeroInstance * h) const;
	void initObj();
};

class DLL_EXPORT CGPickable : public CGObjectInstance //campfire, treasure chest
{
	ui32 type, val1, val2;

	void onHeroVisit(const CGHeroInstance * h) const;
	void initObj();
	void chosen(int which, int heroID) const;
};

class DLL_EXPORT CGShrine : public CGObjectInstance
{
public:
	unsigned char spell; //number of spell or 255 if random
};

class DLL_EXPORT CGPandoraBox : public CArmedInstance
{
public:
	std::string message;

	//gained things:
	unsigned int gainedExp;
	int manaDiff;
	int moraleDiff;
	int luckDiff;
	int wood, mercury, ore, sulfur, crystal, gems, gold;
	int attack, defence, power, knowledge;
	std::vector<int> abilities;
	std::vector<int> abilityLevels;
	std::vector<int> artifacts;
	std::vector<int> spells;
	CCreatureSet creatures;
};

class DLL_EXPORT CGQuestGuard : public CGObjectInstance, public CQuest
{
public:
};

class DLL_EXPORT CGMine : public CArmedInstance
{
public: 
	void onHeroVisit(const CGHeroInstance * h) const;
	void newTurn() const;
	void initObj();	
};

class DLL_EXPORT CGVisitableOPW : public CGObjectInstance //objects visitable OPW
{
public:
	ui8 visited; //true if object has been visited this week

	void onHeroVisit(const CGHeroInstance * h) const;
	void newTurn() const;
};

class DLL_EXPORT CGTeleport : public CGObjectInstance //teleports and subterranean gates
{
public:
	static std::map<int,std::map<int, std::vector<int> > > objs; //map[ID][subID] => vector of ids
	void onHeroVisit(const CGHeroInstance * h) const;
	void initObj();	
};

class DLL_EXPORT CObjectHandler
{
public:
	std::vector<std::string> names; //vector of objects; i-th object in vector has subnumber i
	std::vector<int> cregens; //type 17. dwelling subid -> creature ID
	void loadObjects();

	std::vector<std::string> creGens; //names of creatures' generators
	std::vector<std::string> advobtxt;
	std::vector<std::string> xtrainfo;
	std::vector<std::string> restypes;
	std::vector<std::pair<std::string,std::string> > mines; //first - name; second - event description
};



#endif // __COBJECTHANDLER_H__