1 files changed, 57 insertions, 0 deletions
diff --git a/Minecraft.World/LookAtPlayerGoal.cpp b/Minecraft.World/LookAtPlayerGoal.cpp
new file mode 100644
index 00000000..f2dc10c3
--- /dev/null
+++ b/Minecraft.World/LookAtPlayerGoal.cpp
@@ -0,0 +1,57 @@
+#include "stdafx.h"
+#include "net.minecraft.world.entity.h"
+#include "net.minecraft.world.entity.ai.control.h"
+#include "net.minecraft.world.level.h"
+#include "net.minecraft.world.phys.h"
+#include "LookAtPlayerGoal.h"
+
+LookAtPlayerGoal::LookAtPlayerGoal(Mob *mob, const type_info& lookAtType, float lookDistance) : lookAtType(lookAtType)
+{
+	this->mob = mob;
+	this->lookDistance = lookDistance;
+	this->probability = 0.02f;
+	setRequiredControlFlags(Control::LookControlFlag);
+
+	lookTime = 0;
+}
+
+LookAtPlayerGoal::LookAtPlayerGoal(Mob *mob, const type_info& lookAtType, float lookDistance, float probability) : lookAtType(lookAtType)
+{
+	this->mob = mob;
+	this->lookDistance = lookDistance;
+	this->probability = probability;
+	setRequiredControlFlags(Control::LookControlFlag);
+
+	lookTime = 0;
+}
+
+bool LookAtPlayerGoal::canUse()
+{
+	if (mob->getRandom()->nextFloat() >= probability) return false;
+	if (lookAtType == typeid(Player)) lookAt = mob->level->getNearestPlayer(mob->shared_from_this(), lookDistance);
+	else lookAt = weak_ptr<Entity>(mob->level->getClosestEntityOfClass(lookAtType, mob->bb->grow(lookDistance, 3, lookDistance), mob->shared_from_this()));
+	return lookAt.lock() != NULL;
+}
+
+bool LookAtPlayerGoal::canContinueToUse()
+{
+	if (lookAt.lock() == NULL || !lookAt.lock()->isAlive()) return false;
+	if (mob->distanceToSqr(lookAt.lock()) > lookDistance * lookDistance) return false;
+	return lookTime > 0;
+}
+
+void LookAtPlayerGoal::start()
+{
+	lookTime = 40 + mob->getRandom()->nextInt(40);
+}
+
+void LookAtPlayerGoal::stop()
+{
+	lookAt = weak_ptr<Entity>();
+}
+
+void LookAtPlayerGoal::tick()
+{
+	mob->getLookControl()->setLookAt(lookAt.lock()->x, lookAt.lock()->y + lookAt.lock()->getHeadHeight(), lookAt.lock()->z, 10, mob->getMaxHeadXRot());
+	--lookTime;
+}
+\ No newline at end of file