1 files changed, 127 insertions, 0 deletions
diff --git a/Minecraft.World/MoveThroughVillageGoal.cpp b/Minecraft.World/MoveThroughVillageGoal.cpp
new file mode 100644
index 00000000..2405374f
--- /dev/null
+++ b/Minecraft.World/MoveThroughVillageGoal.cpp
@@ -0,0 +1,127 @@
+#include "stdafx.h"
+#include "net.minecraft.world.entity.ai.control.h"
+#include "net.minecraft.world.entity.ai.navigation.h"
+#include "net.minecraft.world.entity.ai.util.h"
+#include "net.minecraft.world.entity.ai.village.h"
+#include "net.minecraft.world.entity.h"
+#include "net.minecraft.world.level.h"
+#include "BasicTypeContainers.h"
+#include "MoveThroughVillageGoal.h"
+#include "Path.h"
+
+MoveThroughVillageGoal::MoveThroughVillageGoal(PathfinderMob *mob, float speed, bool onlyAtNight)
+{
+	path = NULL;
+	doorInfo = weak_ptr<DoorInfo>();
+
+	this->mob = mob;
+	this->speed = speed;
+	this->onlyAtNight = onlyAtNight;
+	setRequiredControlFlags(Control::MoveControlFlag);
+}
+
+MoveThroughVillageGoal::~MoveThroughVillageGoal()
+{
+	if(path != NULL) delete path;
+}
+
+bool MoveThroughVillageGoal::canUse()
+{
+	updateVisited();
+
+	if (onlyAtNight && mob->level->isDay()) return false;
+
+	shared_ptr<Village> village = mob->level->villages->getClosestVillage(Mth::floor(mob->x), Mth::floor(mob->y), Mth::floor(mob->z), 0);
+	if (village == NULL) return false;
+
+	shared_ptr<DoorInfo> _doorInfo = getNextDoorInfo(village);
+	if (_doorInfo == NULL) return false;
+	doorInfo = _doorInfo;
+
+	bool oldCanOpenDoors = mob->getNavigation()->canOpenDoors();
+	mob->getNavigation()->setCanOpenDoors(false);
+	delete path;
+
+	path = mob->getNavigation()->createPath(_doorInfo->x, _doorInfo->y, _doorInfo->z);
+	mob->getNavigation()->setCanOpenDoors(oldCanOpenDoors);
+	if (path != NULL) return true;
+
+	Vec3 *pos = RandomPos::getPosTowards(dynamic_pointer_cast<PathfinderMob>(mob->shared_from_this()), 10, 7, Vec3::newTemp(_doorInfo->x, _doorInfo->y, _doorInfo->z));
+	if (pos == NULL) return false;
+	mob->getNavigation()->setCanOpenDoors(false);
+	delete path;
+	path = mob->getNavigation()->createPath(pos->x, pos->y, pos->z);
+	mob->getNavigation()->setCanOpenDoors(oldCanOpenDoors);
+	return path != NULL;
+}
+
+bool MoveThroughVillageGoal::canContinueToUse()
+{
+	if (mob->getNavigation()->isDone()) return false;
+	float dist = mob->bbWidth + 4.f;
+	shared_ptr<DoorInfo> _doorInfo = doorInfo.lock();
+	if( _doorInfo == NULL ) return false;
+
+	return mob->distanceToSqr(_doorInfo->x, _doorInfo->y, _doorInfo->z) > dist * dist;
+}
+
+void MoveThroughVillageGoal::start()
+{
+	mob->getNavigation()->moveTo(path, speed);
+	path = NULL;
+}
+
+void MoveThroughVillageGoal::stop()
+{
+	shared_ptr<DoorInfo> _doorInfo = doorInfo.lock();
+	if( _doorInfo == NULL ) return;
+
+	if (mob->getNavigation()->isDone() || mob->distanceToSqr(_doorInfo->x, _doorInfo->y, _doorInfo->z) < 4 * 4)
+	{
+		visited.push_back(doorInfo);
+	}
+}
+
+shared_ptr<DoorInfo> MoveThroughVillageGoal::getNextDoorInfo(shared_ptr<Village> village)
+{
+	shared_ptr<DoorInfo> closest = nullptr;
+	int closestDistSqr = Integer::MAX_VALUE;
+	vector<shared_ptr<DoorInfo> > *doorInfos = village->getDoorInfos();
+	//for (DoorInfo di : doorInfos)
+	for(AUTO_VAR(it, doorInfos->begin()); it != doorInfos->end(); ++it)
+	{
+		shared_ptr<DoorInfo> di = *it;
+		int distSqr = di->distanceToSqr(Mth::floor(mob->x), Mth::floor(mob->y), Mth::floor(mob->z));
+		if (distSqr < closestDistSqr)
+		{
+			if (hasVisited(di)) continue;
+			closest = di;
+			closestDistSqr = distSqr;
+		}
+	}
+	return closest;
+}
+
+bool MoveThroughVillageGoal::hasVisited(shared_ptr<DoorInfo>di)
+{
+	//for (DoorInfo di2 : visited)
+	for(AUTO_VAR(it, visited.begin()); it != visited.end(); )
+	{
+		shared_ptr<DoorInfo> di2 = (*it).lock();
+		if( di2 == NULL )
+		{
+			it = visited.erase(it);
+		}
+		else
+		{
+			if (di->x == di2->x && di->y == di2->y && di->z == di2->z) return true;
+			++it;
+		}
+	}
+	return false;
+}
+
+void MoveThroughVillageGoal::updateVisited()
+{
+	if (visited.size() > 15) visited.erase(visited.begin());
+}
+\ No newline at end of file