mirror of
https://github.com/4jcraft/4jcraft.git
synced 2026-04-24 01:24:07 +00:00
77 lines
2.7 KiB
C++
77 lines
2.7 KiB
C++
#include "../../Platform/stdafx.h"
|
|
#include "../../Headers/net.minecraft.world.entity.h"
|
|
#include "../../Headers/net.minecraft.world.level.h"
|
|
#include "../../Headers/net.minecraft.world.phys.h"
|
|
#include "NearestAttackableTargetGoal.h"
|
|
|
|
SubselectEntitySelector::SubselectEntitySelector(
|
|
NearestAttackableTargetGoal* parent, EntitySelector* subselector) {
|
|
m_parent = parent;
|
|
m_subselector = subselector;
|
|
}
|
|
|
|
SubselectEntitySelector::~SubselectEntitySelector() { delete m_subselector; }
|
|
|
|
bool SubselectEntitySelector::matches(std::shared_ptr<Entity> entity) const {
|
|
if (!entity->instanceof(eTYPE_LIVINGENTITY)) return false;
|
|
if (m_subselector != NULL && !m_subselector->matches(entity)) return false;
|
|
return m_parent->canAttack(std::dynamic_pointer_cast<LivingEntity>(entity),
|
|
false);
|
|
}
|
|
|
|
NearestAttackableTargetGoal::DistComp::DistComp(Entity* source) {
|
|
this->source = source;
|
|
}
|
|
|
|
bool NearestAttackableTargetGoal::DistComp::operator()(
|
|
std::shared_ptr<Entity> e1, std::shared_ptr<Entity> e2) {
|
|
// Should return true if e1 comes before e2 in the sorted list
|
|
double distSqr1 = source->distanceToSqr(e1);
|
|
double distSqr2 = source->distanceToSqr(e2);
|
|
if (distSqr1 < distSqr2) return true;
|
|
if (distSqr1 > distSqr2) return false;
|
|
return true;
|
|
}
|
|
|
|
NearestAttackableTargetGoal::NearestAttackableTargetGoal(
|
|
PathfinderMob* mob, const std::type_info& targetType, int randomInterval,
|
|
bool mustSee, bool mustReach /*= false*/,
|
|
EntitySelector* entitySelector /* =NULL */)
|
|
: TargetGoal(mob, mustSee, mustReach), targetType(targetType) {
|
|
this->randomInterval = randomInterval;
|
|
this->distComp = new DistComp(mob);
|
|
setRequiredControlFlags(TargetGoal::TargetFlag);
|
|
|
|
this->selector = new SubselectEntitySelector(this, entitySelector);
|
|
}
|
|
|
|
NearestAttackableTargetGoal::~NearestAttackableTargetGoal() {
|
|
delete distComp;
|
|
delete selector;
|
|
}
|
|
|
|
bool NearestAttackableTargetGoal::canUse() {
|
|
if (randomInterval > 0 && mob->getRandom()->nextInt(randomInterval) != 0)
|
|
return false;
|
|
double within = getFollowDistance();
|
|
|
|
std::vector<std::shared_ptr<Entity> >* entities =
|
|
mob->level->getEntitiesOfClass(
|
|
targetType, mob->bb->grow(within, 4, within), selector);
|
|
|
|
bool result = false;
|
|
if (entities != NULL && !entities->empty()) {
|
|
std::sort(entities->begin(), entities->end(), *distComp);
|
|
target = std::weak_ptr<LivingEntity>(
|
|
std::dynamic_pointer_cast<LivingEntity>(entities->at(0)));
|
|
result = true;
|
|
}
|
|
|
|
delete entities;
|
|
return result;
|
|
}
|
|
|
|
void NearestAttackableTargetGoal::start() {
|
|
mob->setTarget(target.lock());
|
|
TargetGoal::start();
|
|
} |