1 files changed, 61 insertions, 0 deletions
diff --git a/Minecraft.World/OcelotAttackGoal.cpp b/Minecraft.World/OcelotAttackGoal.cpp
new file mode 100644
index 00000000..d9e0f29b
--- /dev/null
+++ b/Minecraft.World/OcelotAttackGoal.cpp
@@ -0,0 +1,61 @@
+#include "stdafx.h"
+#include "net.minecraft.world.entity.ai.control.h"
+#include "net.minecraft.world.entity.ai.navigation.h"
+#include "net.minecraft.world.entity.h"
+#include "net.minecraft.world.entity.animal.h"
+#include "net.minecraft.world.phys.h"
+#include "OcelotAttackGoal.h"
+
+OcelotAttackGoal::OcelotAttackGoal(Mob *mob)
+{
+	target = weak_ptr<LivingEntity>();
+	attackTime = 0;
+	speed = 0;
+	trackTarget = false;
+
+	this->mob = mob;
+	this->level = mob->level;
+	setRequiredControlFlags(Control::MoveControlFlag | Control::LookControlFlag);
+}
+
+bool OcelotAttackGoal::canUse()
+{
+	shared_ptr<LivingEntity> bestTarget = mob->getTarget();
+	if (bestTarget == NULL) return false;
+	target = weak_ptr<LivingEntity>(bestTarget);
+	return true;
+}
+
+bool OcelotAttackGoal::canContinueToUse()
+{
+	if (target.lock() == NULL || !target.lock()->isAlive()) return false;
+	if (mob->distanceToSqr(target.lock()) > 15 * 15) return false;
+	return !mob->getNavigation()->isDone() || canUse();
+}
+
+void OcelotAttackGoal::stop()
+{
+	target = weak_ptr<Mob>();
+	mob->getNavigation()->stop();
+}
+
+void OcelotAttackGoal::tick()
+{
+	mob->getLookControl()->setLookAt(target.lock(), 30, 30);
+
+	double meleeRadiusSqr = (mob->bbWidth * 2) * (mob->bbWidth * 2);
+	double distSqr = mob->distanceToSqr(target.lock()->x, target.lock()->bb->y0, target.lock()->z);
+
+	double speedModifier = Ocelot::WALK_SPEED_MOD;
+	if (distSqr > meleeRadiusSqr && distSqr < 4 * 4) speedModifier = Ocelot::SPRINT_SPEED_MOD;
+	else if (distSqr < 15 * 15) speedModifier = Ocelot::SNEAK_SPEED_MOD;
+
+	mob->getNavigation()->moveTo(target.lock(), speedModifier);
+
+	attackTime = max(attackTime - 1, 0);
+
+	if (distSqr > meleeRadiusSqr) return;
+	if (attackTime > 0) return;
+	attackTime = 20;
+	mob->doHurtTarget(target.lock());
+}
+\ No newline at end of file