1 files changed, 85 insertions, 0 deletions
diff --git a/Minecraft.World/FollowOwnerGoal.cpp b/Minecraft.World/FollowOwnerGoal.cpp
new file mode 100644
index 00000000..faba2226
--- /dev/null
+++ b/Minecraft.World/FollowOwnerGoal.cpp
@@ -0,0 +1,85 @@
+#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.level.h"
+#include "net.minecraft.world.phys.h"
+#include "FollowOwnerGoal.h"
+
+FollowOwnerGoal::FollowOwnerGoal(TamableAnimal *tamable, float speed, float startDistance, float stopDistance)
+{
+	owner = weak_ptr<Mob>();
+	timeToRecalcPath = 0;
+	oldAvoidWater = false;
+
+	this->tamable = tamable;
+	this->level = tamable->level;
+	this->speed = speed;
+	this->navigation = tamable->getNavigation();
+	this->startDistance = startDistance;
+	this->stopDistance = stopDistance;
+	setRequiredControlFlags(Control::MoveControlFlag | Control::LookControlFlag);
+}
+
+bool FollowOwnerGoal::canUse()
+{
+	shared_ptr<Mob> owner = tamable->getOwner();
+	if (owner == NULL) return false;
+	if (tamable->isSitting()) return false;
+	if (tamable->distanceToSqr(owner) < startDistance * startDistance) return false;
+	this->owner = weak_ptr<Mob>(owner);
+	return true;
+}
+
+bool FollowOwnerGoal::canContinueToUse()
+{
+	return owner.lock() != NULL && !navigation->isDone() && tamable->distanceToSqr(owner.lock()) > stopDistance * stopDistance && !tamable->isSitting();
+}
+
+void FollowOwnerGoal::start()
+{
+	timeToRecalcPath = 0;
+	oldAvoidWater = tamable->getNavigation()->getAvoidWater();
+	tamable->getNavigation()->setAvoidWater(false);
+}
+
+void FollowOwnerGoal::stop()
+{
+	owner = weak_ptr<Mob>();
+	navigation->stop();
+	tamable->getNavigation()->setAvoidWater(oldAvoidWater);
+}
+
+void FollowOwnerGoal::tick()
+{
+	tamable->getLookControl()->setLookAt(owner.lock(), 10, tamable->getMaxHeadXRot());
+	if (tamable->isSitting()) return;
+
+	if (--timeToRecalcPath > 0) return;
+	timeToRecalcPath = 10;
+
+	if (navigation->moveTo(owner.lock(), speed)) return;
+	if (tamable->distanceToSqr(owner.lock()) < TeleportDistance * TeleportDistance) return;
+
+	// find a good spawn position nearby the owner
+	int sx = Mth::floor(owner.lock()->x) - 2;
+	int sz = Mth::floor(owner.lock()->z) - 2;
+	int y = Mth::floor(owner.lock()->bb->y0);
+	for (int x = 0; x <= 4; x++)
+	{
+		for (int z = 0; z <= 4; z++)
+		{
+			if (x >= 1 && z >= 1 && x <= 3 && z <= 3)
+			{
+				continue;
+			}
+			if (level->isTopSolidBlocking(sx + x, y - 1, sz + z) && !level->isSolidBlockingTile(sx + x, y, sz + z) && !level->isSolidBlockingTile(sx + x, y + 1, sz + z))
+			{
+				tamable->moveTo(sx + x + .5f, y, sz + z + .5f, tamable->yRot, tamable->xRot);
+				navigation->stop();
+				return;
+			}
+		}
+	}
+}
+\ No newline at end of file