Moved Modificators to separate folder

2025-06-23 00:28:08 +02:00 · 2023-05-20 10:17:37 +02:00
parent 5f99facd8f
commit cf0515dcd3
36 changed files with 181 additions and 181 deletions
--- a/lib/rmg/modificators/RoadPlacer.cpp
+++ b/lib/rmg/modificators/RoadPlacer.cpp
@ -0,0 +1,147 @@
+/*
+ * RoadPlacer.cpp, part of VCMI engine
+ *
+ * Authors: listed in file AUTHORS in main folder
+ *
+ * License: GNU General Public License v2.0 or later
+ * Full text of license available in license.txt file, in main folder
+ *
+ */
+
+#include "StdInc.h"
+#include "RoadPlacer.h"
+#include "ObjectManager.h"
+#include "ObstaclePlacer.h"
+#include "../Functions.h"
+#include "../CMapGenerator.h"
+#include "../RmgMap.h"
+#include "../threadpool/MapProxy.h"
+#include "../../mapping/CMapEditManager.h"
+#include "../../CModHandler.h"
+#include "../RmgPath.h"
+
+VCMI_LIB_NAMESPACE_BEGIN
+
+void RoadPlacer::process()
+{
+	if(generator.getConfig().defaultRoadType.empty() && generator.getConfig().secondaryRoadType.empty())
+		return; //do not generate roads at all
+	
+	connectRoads();
+}
+
+rmg::Area & RoadPlacer::areaForRoads()
+{
+	return areaRoads;
+}
+
+rmg::Area & RoadPlacer::areaIsolated()
+{
+	return isolated;
+}
+
+const rmg::Area & RoadPlacer::getRoads() const
+{
+	return roads;
+}
+
+bool RoadPlacer::createRoad(const int3 & dst)
+{
+	auto searchArea = zone.areaPossible() + areaRoads + zone.freePaths() - isolated + roads;
+	
+	rmg::Path path(searchArea);
+	path.connect(roads);
+	
+	auto res = path.search(dst, true);
+	if(!res.valid())
+	{
+		res = path.search(dst, false, [](const int3 & src, const int3 & dst)
+		{
+			float weight = dst.dist2dSQ(src);
+			return weight * weight;
+		});
+		if(!res.valid())
+		{
+			logGlobal->warn("Failed to create road");
+			return false;
+		}
+	}
+	roads.unite(res.getPathArea());
+	return true;
+	
+}
+
+void RoadPlacer::drawRoads(bool secondary)
+{
+	if((secondary && generator.getConfig().secondaryRoadType.empty())
+	   || (!secondary && generator.getConfig().defaultRoadType.empty()))
+		return;
+	
+	//RecursiveLock lock(externalAccessMutex);
+	{
+		//FIXME: double lock - unsafe
+		Zone::Lock lock(zone.areaMutex);
+
+		zone.areaPossible().subtract(roads);
+		zone.freePaths().unite(roads);
+	}
+
+	auto tiles = roads.getTilesVector();
+
+	std::string roadName = (secondary ? generator.getConfig().secondaryRoadType : generator.getConfig().defaultRoadType);
+	RoadId roadType(*VLC->modh->identifiers.getIdentifier(CModHandler::scopeGame(), "road", roadName));
+	mapProxy->drawRoads(generator.rand, tiles, roadType);
+}
+
+void RoadPlacer::addRoadNode(const int3& node)
+{
+	RecursiveLock lock(externalAccessMutex);
+	roadNodes.insert(node);
+}
+
+void RoadPlacer::connectRoads()
+{
+	bool noRoadNodes = false;
+	//Assumes objects are already placed
+	if (roadNodes.size() < 2)
+	{
+		//If there are no nodes, draw roads to mines
+		noRoadNodes = true;
+		if (auto* m = zone.getModificator<ObjectManager>())
+		{
+			for(auto * object : m->getMines())
+			{
+				addRoadNode(object->visitablePos());
+			}
+		}
+	}
+
+	if(roadNodes.size() < 2)
+		return;
+	
+	//take any tile from road nodes as destination zone for all other road nodes
+	RecursiveLock lock(externalAccessMutex);
+	if(roads.empty())
+		roads.add(*roadNodes.begin());
+
+	for(const auto & node : roadNodes)
+	{
+		createRoad(node);
+	}
+	
+	//Draw dirt roads if there are only mines
+	drawRoads(noRoadNodes);
+}
+
+char RoadPlacer::dump(const int3 & t)
+{
+	if(roadNodes.count(t))
+		return '@';
+	if(roads.contains(t))
+		return '+';
+	if(isolated.contains(t))
+		return 'i';
+	return Modificator::dump(t);
+}
+
+VCMI_LIB_NAMESPACE_END