4jcraft/Minecraft.World/AI/Goals/HurtByTargetGoal.cpp

#include "../../Platform/stdafx.h"
#include "../../Headers/net.minecraft.world.entity.h"
#include "../../Headers/net.minecraft.world.phys.h"
#include "../../Headers/net.minecraft.world.level.h"
#include "HurtByTargetGoal.h"

HurtByTargetGoal::HurtByTargetGoal(Mob* mob, bool alertSameType)
    : TargetGoal(mob, 16, false) {
    this->alertSameType = alertSameType;
    setRequiredControlFlags(TargetGoal::TargetFlag);
}

bool HurtByTargetGoal::canUse() {
    return canAttack(mob->getLastHurtByMob(), false);
}

void HurtByTargetGoal::start() {
    mob->setTarget(mob->getLastHurtByMob());
    oldHurtByMob = mob->getLastHurtByMob();

    if (alertSameType) {
        std::vector<std::shared_ptr<Entity> >* nearby =
            mob->level->getEntitiesOfClass(
                typeid(*mob), AABB::newTemp(mob->x, mob->y, mob->z, mob->x + 1,
                                            mob->y + 1, mob->z + 1)
                                  ->grow(within, 4, within));
        for (AUTO_VAR(it, nearby->begin()); it != nearby->end(); ++it) {
            std::shared_ptr<Mob> other = std::dynamic_pointer_cast<Mob>(*it);
            if (this->mob->shared_from_this() == other) continue;
            if (other->getTarget() != NULL) continue;
            other->setTarget(mob->getLastHurtByMob());
        }
        delete nearby;
    }

    TargetGoal::start();
}

void HurtByTargetGoal::tick() {
    if (mob->getLastHurtByMob() != NULL &&
        mob->getLastHurtByMob() != oldHurtByMob) {
        this->start();
    }
}