1 files changed, 54 insertions, 0 deletions
diff --git a/Minecraft.World/SwellGoal.cpp b/Minecraft.World/SwellGoal.cpp
new file mode 100644
index 00000000..9065ba16
--- /dev/null
+++ b/Minecraft.World/SwellGoal.cpp
@@ -0,0 +1,54 @@
+#include "stdafx.h"
+#include "net.minecraft.world.entity.ai.control.h"
+#include "net.minecraft.world.entity.ai.sensing.h"
+#include "net.minecraft.world.entity.ai.navigation.h"
+#include "net.minecraft.world.entity.monster.h"
+#include "SwellGoal.h"
+
+SwellGoal::SwellGoal(Creeper *creeper)
+{
+	target = weak_ptr<Mob>();
+
+	this->creeper = creeper;
+	setRequiredControlFlags(Control::MoveControlFlag);
+}
+
+bool SwellGoal::canUse()
+{
+	shared_ptr<Mob> target = creeper->getTarget();
+	return creeper->getSwellDir() > 0 || (target != NULL && (creeper->distanceToSqr(target) < 3 * 3));
+}
+
+void SwellGoal::start()
+{
+	creeper->getNavigation()->stop();
+	target = weak_ptr<Mob>(creeper->getTarget());
+}
+
+void SwellGoal::stop()
+{
+	target = weak_ptr<Mob>();
+}
+
+void SwellGoal::tick()
+{
+	if (target.lock() == NULL)
+	{
+		creeper->setSwellDir(-1);
+		return;
+	}
+
+	if (creeper->distanceToSqr(target.lock()) > 7 * 7)
+	{
+		creeper->setSwellDir(-1);
+		return;
+	}
+
+	if (!creeper->getSensing()->canSee(target.lock()))
+	{
+		creeper->setSwellDir(-1);
+		return;
+	}
+
+	creeper->setSwellDir(1);
+}