Add MapInfoCallback and EditorCallback

2025-11-25 22:42:04 +02:00 · 2025-05-26 22:52:34 +02:00
parent 507c305a5a
commit b2e51a5fb1
10 changed files with 456 additions and 106 deletions
--- a/lib/callback/MapInfoCallback.cpp
+++ b/lib/callback/MapInfoCallback.cpp
@@ -0,0 +1,120 @@
+#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