1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#include "stdafx.h"
#include "net.minecraft.world.entity.h"
#include "net.minecraft.world.level.h"
#include "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(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(dynamic_pointer_cast<LivingEntity>(entity), false);
}
NearestAttackableTargetGoal::DistComp::DistComp(Entity *source)
{
this->source = source;
}
bool NearestAttackableTargetGoal::DistComp::operator() (shared_ptr<Entity> e1, 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 false;
}
NearestAttackableTargetGoal::NearestAttackableTargetGoal(PathfinderMob *mob, const 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();
vector<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 = weak_ptr<LivingEntity>(dynamic_pointer_cast<LivingEntity>(entities->at(0)));
result = true;
}
delete entities;
return result;
}
void NearestAttackableTargetGoal::start()
{
mob->setTarget(target.lock());
TargetGoal::start();
}
|
