1 files changed, 38 insertions, 0 deletions
diff --git a/Minecraft.World/MoveTowardsRestrictionGoal.cpp b/Minecraft.World/MoveTowardsRestrictionGoal.cpp
new file mode 100644
index 00000000..a1848d46
--- /dev/null
+++ b/Minecraft.World/MoveTowardsRestrictionGoal.cpp
@@ -0,0 +1,38 @@
+#include "stdafx.h"
+#include "net.minecraft.world.entity.ai.control.h"
+#include "net.minecraft.world.entity.ai.util.h"
+#include "net.minecraft.world.entity.ai.navigation.h"
+#include "net.minecraft.world.entity.h"
+#include "net.minecraft.world.level.h"
+#include "MoveTowardsRestrictionGoal.h"
+
+MoveTowardsRestrictionGoal::MoveTowardsRestrictionGoal(PathfinderMob *mob, float speed)
+{
+	wantedX = wantedY = wantedZ = 0.0;
+
+	this->mob = mob;
+	this->speed = speed;
+	setRequiredControlFlags(Control::MoveControlFlag);
+}
+
+bool MoveTowardsRestrictionGoal::canUse()
+{
+	if (mob->isWithinRestriction()) return false;
+	Pos *towards = mob->getRestrictCenter();
+	Vec3 *pos = RandomPos::getPosTowards(dynamic_pointer_cast<PathfinderMob>(mob->shared_from_this()), 16, 7, Vec3::newTemp(towards->x, towards->y, towards->z));
+	if (pos == NULL) return false;
+	wantedX = pos->x;
+	wantedY = pos->y;
+	wantedZ = pos->z;
+	return true;
+}
+
+bool MoveTowardsRestrictionGoal::canContinueToUse()
+{
+	return !mob->getNavigation()->isDone();
+}
+
+void MoveTowardsRestrictionGoal::start()
+{
+	mob->getNavigation()->moveTo(wantedX, wantedY, wantedZ, speed);
+}
+\ No newline at end of file