vcmi/lib/callback/MapInfoCallback.cpp

#include "StdInc.h"
#include "MapInfoCallback.h"
#include "../constants/EntityIdentifiers.h"
#include "../mapObjects/CGObjectInstance.h"
#include "../mapObjects/CGHeroInstance.h"
#include "../mapObjects/CGTownInstance.h"
#include "../mapObjects/MiscObjects.h"
#include "../StartInfo.h"
#include "../mapping/CMap.h"
#include "../spells/CSpellHandler.h"

VCMI_LIB_NAMESPACE_BEGIN

MapInfoCallback::~MapInfoCallback() = default;

const CGObjectInstance * MapInfoCallback::getObj(ObjectInstanceID objid, bool verbose) const
{
	if(!objid.hasValue())
	{
		if(verbose)
			logGlobal->error("Cannot get object with id %d. No such object", objid.getNum());
		return nullptr;
	}

	const CGObjectInstance * ret = getMapConstPtr()->getObject(objid);
	if(!ret && verbose)
	{
		logGlobal->error("Cannot get object with id %d. Object was removed", objid.getNum());
		return nullptr;
	}

	return ret;
}

const CGHeroInstance * MapInfoCallback::getHero(ObjectInstanceID objid) const
{
	const CGObjectInstance * obj = getObj(objid, false);
	return dynamic_cast<const CGHeroInstance *>(obj);
}

const CGTownInstance * MapInfoCallback::getTown(ObjectInstanceID objid) const
{
	const CGObjectInstance * obj = getObj(objid, false);
	return dynamic_cast<const CGTownInstance*>(obj);
}

PlayerColor MapInfoCallback::getOwner(ObjectInstanceID heroID) const
{
	const CGObjectInstance * obj = getObj(heroID);
	if(!obj)
	{
		logGlobal->error("MapInfoCallback::getOwner(heroID): No such object for heroID: %d", heroID.num);
		return PlayerColor::CANNOT_DETERMINE;
	}

	return obj->tempOwner;
}

const CArtifactInstance * MapInfoCallback::getArtInstance(ArtifactInstanceID aid) const
{
	return getMapConstPtr()->getArtifactInstance(aid);
}

const CGObjectInstance * MapInfoCallback::getObjInstance(ObjectInstanceID oid) const
{
	return getMapConstPtr()->getObject((oid));
}

bool MapInfoCallback::isInTheMap(const int3 & pos) const
{
	return getMapConstPtr()->isInTheMap(pos);
}

bool MapInfoCallback::isAllowed(SpellID id) const
{
	return getMapConstPtr()->allowedSpells.count(id) != 0;
}

bool MapInfoCallback::isAllowed(ArtifactID id) const
{
	return getMapConstPtr()->allowedArtifact.count(id) != 0;
}

bool MapInfoCallback::isAllowed(SecondarySkill id) const
{
	return getMapConstPtr()->allowedAbilities.count(id) != 0;
}

int3 MapInfoCallback::getMapSize() const
{
	return int3(getMapConstPtr()->width, getMapConstPtr()->height, getMapConstPtr()->twoLevel ? 2 : 1);
}

void MapInfoCallback::getAllowedSpells(std::vector<SpellID> & out, std::optional<ui16> level)
{
	for (auto const & spellID : getMapConstPtr()->allowedSpells)
	{
		const auto * spell = spellID.toEntity(LIBRARY);

		if (!isAllowed(spellID))
			continue;

		if (level.has_value() && spell->getLevel() != level)
			continue;

		out.push_back(spellID);
	}
}

const IGameSettings & MapInfoCallback::getSettings() const
{
	return getMapConstPtr()->getSettings();
}

const CMapHeader * MapInfoCallback::getMapHeader() const
{
	return getMapConstPtr();
}

VCMI_LIB_NAMESPACE_END