1
0
mirror of https://github.com/vcmi/vcmi.git synced 2024-11-28 08:48:48 +02:00
vcmi/lib/spells/effects/UnitEffect.cpp
2022-12-18 20:23:19 +02:00

299 lines
7.5 KiB
C++

/*
* UnitEffect.cpp, part of VCMI engine
*
* Authors: listed in file AUTHORS in main folder
*
* License: GNU General Public License v2.0 or later
* Full text of license available in license.txt file, in main folder
*
*/
#include "StdInc.h"
#include "UnitEffect.h"
#include "../ISpellMechanics.h"
#include "../../NetPacksBase.h"
#include "../../battle/CBattleInfoCallback.h"
#include "../../battle/Unit.h"
#include "../../serializer/JsonSerializeFormat.h"
VCMI_LIB_NAMESPACE_BEGIN
namespace spells
{
namespace effects
{
UnitEffect::UnitEffect()
: Effect(),
chainLength(0),
chainFactor(0.0),
ignoreImmunity(false)
{
}
UnitEffect::~UnitEffect() = default;
void UnitEffect::adjustTargetTypes(std::vector<TargetType> & types) const
{
}
void UnitEffect::adjustAffectedHexes(std::set<BattleHex> & hexes, const Mechanics * m, const Target & spellTarget) const
{
for(auto & destnation : spellTarget)
hexes.insert(destnation.hexValue);
}
bool UnitEffect::applicable(Problem & problem, const Mechanics * m) const
{
//stack effect is applicable in general if there is at least one smart target
auto mainFilter = std::bind(&UnitEffect::getStackFilter, this, m, true, _1);
auto predicate = std::bind(&UnitEffect::eraseByImmunityFilter, this, m, _1);
auto targets = m->battle()->battleGetUnitsIf(mainFilter);
vstd::erase_if(targets, predicate);
if(targets.empty())
{
MetaString text;
text.addTxt(MetaString::GENERAL_TXT, 185);
problem.add(std::move(text), Problem::NORMAL);
return false;
}
return true;
}
bool UnitEffect::applicable(Problem & problem, const Mechanics * m, const EffectTarget & target) const
{
//stack effect is applicable if it affects at least one smart target
//assume target correctly transformed, just reapply smart filter
for(auto & item : target)
if(item.unitValue)
if(getStackFilter(m, true, item.unitValue))
return true;
return false;
}
bool UnitEffect::getStackFilter(const Mechanics * m, bool alwaysSmart, const battle::Unit * s) const
{
return isValidTarget(m, s) && isSmartTarget(m, s, alwaysSmart);
}
bool UnitEffect::eraseByImmunityFilter(const Mechanics * m, const battle::Unit * s) const
{
return !isReceptive(m, s);
}
EffectTarget UnitEffect::filterTarget(const Mechanics * m, const EffectTarget & target) const
{
EffectTarget res;
vstd::copy_if(target, std::back_inserter(res), [this, m](const Destination & d)
{
return d.unitValue && isValidTarget(m, d.unitValue) && isReceptive(m, d.unitValue);
});
return res;
}
EffectTarget UnitEffect::transformTarget(const Mechanics * m, const Target & aimPoint, const Target & spellTarget) const
{
if(chainLength > 1)
return transformTargetByChain(m, aimPoint, spellTarget);
else
return transformTargetByRange(m, aimPoint, spellTarget);
}
EffectTarget UnitEffect::transformTargetByRange(const Mechanics * m, const Target & aimPoint, const Target & spellTarget) const
{
auto mainFilter = std::bind(&UnitEffect::getStackFilter, this, m, false, _1);
Target spellTargetCopy(spellTarget);
// make sure that we have valid target with valid aim, even if spell have invalid range configured
// TODO: check than spell range is actually not valid
// also hackfix for banned creature massive spells
// FIXME: potentially breaking change: aimPoint may NOT be in Target - example: frost ring
if(!aimPoint.empty() && spellTarget.empty())
spellTargetCopy.insert(spellTargetCopy.begin(), Destination(aimPoint.front()));
std::set<const battle::Unit *> targets;
if(m->isMassive())
{
//ignore spellTarget and add all stacks
auto units = m->battle()->battleGetUnitsIf(mainFilter);
for(auto unit : units)
targets.insert(unit);
}
else
{
//process each tile
for(const Destination & dest : spellTargetCopy)
{
if(dest.unitValue)
{
if(mainFilter(dest.unitValue))
targets.insert(dest.unitValue);
}
else if(dest.hexValue.isValid())
{
//select one unit on tile, prefer alive one
const battle::Unit * targetOnTile = nullptr;
auto predicate = [&](const battle::Unit * unit)
{
return unit->coversPos(dest.hexValue) && mainFilter(unit);
};
auto units = m->battle()->battleGetUnitsIf(predicate);
for(auto unit : units)
{
if(unit->alive())
{
targetOnTile = unit;
break;
}
}
if(targetOnTile == nullptr && !units.empty())
targetOnTile = units.front();
if(targetOnTile)
targets.insert(targetOnTile);
}
else
{
logGlobal->debug("Invalid destination in spell Target");
}
}
}
auto predicate = std::bind(&UnitEffect::eraseByImmunityFilter, this, m, _1);
vstd::erase_if(targets, predicate);
if(m->alwaysHitFirstTarget())
{
if(!aimPoint.empty() && aimPoint.front().unitValue)
targets.insert(aimPoint.front().unitValue);
else
logGlobal->error("Spell-like attack with no primary target.");
}
EffectTarget effectTarget;
for(auto s : targets)
effectTarget.push_back(Destination(s));
return effectTarget;
}
EffectTarget UnitEffect::transformTargetByChain(const Mechanics * m, const Target & aimPoint, const Target & spellTarget) const
{
EffectTarget byRange = transformTargetByRange(m, aimPoint, spellTarget);
if(byRange.empty())
{
return EffectTarget();
}
const Destination & mainDestination = byRange.front();
if(!mainDestination.hexValue.isValid())
{
return EffectTarget();
}
std::set<BattleHex> possibleHexes;
auto possibleTargets = m->battle()->battleGetUnitsIf([&](const battle::Unit * unit) -> bool
{
return isValidTarget(m, unit);
});
for(auto unit : possibleTargets)
{
for(auto hex : battle::Unit::getHexes(unit->getPosition(), unit->doubleWide(), unit->unitSide()))
possibleHexes.insert(hex);
}
BattleHex destHex = mainDestination.hexValue;
EffectTarget effectTarget;
for(int32_t targetIndex = 0; targetIndex < chainLength; ++targetIndex)
{
auto unit = m->battle()->battleGetUnitByPos(destHex, true);
if(!unit)
break;
if(m->alwaysHitFirstTarget() && targetIndex == 0)
effectTarget.emplace_back(unit);
else if(isReceptive(m, unit) && isValidTarget(m, unit))
effectTarget.emplace_back(unit);
else
effectTarget.emplace_back();
for(auto hex : battle::Unit::getHexes(unit->getPosition(), unit->doubleWide(), unit->unitSide()))
possibleHexes.erase(hex);
if(possibleHexes.empty())
break;
destHex = BattleHex::getClosestTile(unit->unitSide(), destHex, possibleHexes);
}
return effectTarget;
}
bool UnitEffect::isValidTarget(const Mechanics * m, const battle::Unit * unit) const
{
// TODO: override in rising effect
// TODO: check absolute immunity here
return unit->isValidTarget(false);
}
bool UnitEffect::isReceptive(const Mechanics * m, const battle::Unit * unit) const
{
if(ignoreImmunity)
{
//ignore all immunities, except specific absolute immunity(VCMI addition)
//SPELL_IMMUNITY absolute case
std::stringstream cachingStr;
cachingStr << "type_" << Bonus::SPELL_IMMUNITY << "subtype_" << m->getSpellIndex() << "addInfo_1";
return !unit->hasBonus(Selector::typeSubtypeInfo(Bonus::SPELL_IMMUNITY, m->getSpellIndex(), 1), cachingStr.str());
}
else
{
return m->isReceptive(unit);
}
}
bool UnitEffect::isSmartTarget(const Mechanics * m, const battle::Unit * unit, bool alwaysSmart) const
{
const bool smart = m->isSmart() || alwaysSmart;
const bool ignoreOwner = !smart;
return ignoreOwner || m->ownerMatches(unit);
}
void UnitEffect::serializeJsonEffect(JsonSerializeFormat & handler)
{
handler.serializeBool("ignoreImmunity", ignoreImmunity);
handler.serializeInt("chainLength", chainLength, 0);
handler.serializeFloat("chainFactor", chainFactor, 0);
serializeJsonUnitEffect(handler);
}
}
}
VCMI_LIB_NAMESPACE_END