/* * ConnectionsPlacer.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 "ConnectionsPlacer.h" #include "../CMapGenerator.h" #include "../RmgMap.h" #include "../../TerrainHandler.h" #include "../../mapObjectConstructors/AObjectTypeHandler.h" #include "../../mapObjectConstructors/CObjectClassesHandler.h" #include "../../mapObjects/CGCreature.h" #include "../../mapping/CMapEditManager.h" #include "../RmgObject.h" #include "ObjectManager.h" #include "../Functions.h" #include "RoadPlacer.h" #include "../TileInfo.h" #include "WaterAdopter.h" #include "WaterProxy.h" #include "TownPlacer.h" VCMI_LIB_NAMESPACE_BEGIN void ConnectionsPlacer::process() { collectNeighbourZones(); auto diningPhilosophers = [this](std::function f) { for (auto& c : dConnections) { if (c.getZoneA() != zone.getId() && c.getZoneB() != zone.getId()) continue; auto otherZone = map.getZones().at(c.getZoneB()); auto* cp = otherZone->getModificator(); while (cp) { RecursiveLock lock1(externalAccessMutex, boost::try_to_lock_t{}); RecursiveLock lock2(cp->externalAccessMutex, boost::try_to_lock_t{}); if (lock1.owns_lock() && lock2.owns_lock()) { if (!vstd::contains(dCompleted, c)) { f(c); } break; } } } }; diningPhilosophers([this](const rmg::ZoneConnection& c) { selfSideDirectConnection(c); }); createBorder(); diningPhilosophers([this](const rmg::ZoneConnection& c) { selfSideIndirectConnection(c); }); } void ConnectionsPlacer::init() { DEPENDENCY(WaterAdopter); DEPENDENCY(TownPlacer); POSTFUNCTION(RoadPlacer); POSTFUNCTION(ObjectManager); for(auto c : map.getMapGenOptions().getMapTemplate()->getConnectedZoneIds()) addConnection(c); } void ConnectionsPlacer::addConnection(const rmg::ZoneConnection& connection) { dConnections.push_back(connection); } void ConnectionsPlacer::otherSideConnection(const rmg::ZoneConnection & connection) { dCompleted.push_back(connection); } void ConnectionsPlacer::selfSideDirectConnection(const rmg::ZoneConnection & connection) { bool success = false; auto otherZoneId = (connection.getZoneA() == zone.getId() ? connection.getZoneB() : connection.getZoneA()); auto & otherZone = map.getZones().at(otherZoneId); bool createRoad = shouldGenerateRoad(connection); //1. Try to make direct connection //Do if it's not prohibited by terrain settings const auto * ourTerrain = VLC->terrainTypeHandler->getById(zone.getTerrainType()); const auto * otherTerrain = VLC->terrainTypeHandler->getById(otherZone->getTerrainType()); bool directProhibited = vstd::contains(ourTerrain->prohibitTransitions, otherZone->getTerrainType()) || vstd::contains(otherTerrain->prohibitTransitions, zone.getTerrainType()); auto directConnectionIterator = dNeighbourZones.find(otherZoneId); if (directConnectionIterator != dNeighbourZones.end()) { if (connection.getConnectionType() == rmg::EConnectionType::WIDE) { for (auto borderPos : directConnectionIterator->second) { //TODO: Refactor common code with direct connection int3 potentialPos = zone.areaPossible().nearest(borderPos); assert(borderPos != potentialPos); auto safetyGap = rmg::Area({ potentialPos }); safetyGap.unite(safetyGap.getBorderOutside()); safetyGap.intersect(zone.areaPossible()); if (!safetyGap.empty()) { safetyGap.intersect(otherZone->areaPossible()); if (safetyGap.empty()) { rmg::Area border(zone.getArea().getBorder()); border.unite(otherZone->getArea().getBorder()); auto costFunction = [&border](const int3& s, const int3& d) { return 1.f / (1.f + border.distanceSqr(d)); }; auto ourArea = zone.areaPossible() + zone.freePaths(); auto theirArea = otherZone->areaPossible() + otherZone->freePaths(); theirArea.add(potentialPos); rmg::Path ourPath(ourArea); rmg::Path theirPath(theirArea); ourPath.connect(zone.freePaths()); ourPath = ourPath.search(potentialPos, true, costFunction); theirPath.connect(otherZone->freePaths()); theirPath = theirPath.search(potentialPos, true, costFunction); if (ourPath.valid() && theirPath.valid()) { zone.connectPath(ourPath); otherZone->connectPath(theirPath); otherZone->getModificator()->updateDistances(potentialPos); success = true; break; } } } } } } if (connection.getConnectionType() == rmg::EConnectionType::FICTIVE || connection.getConnectionType() == rmg::EConnectionType::REPULSIVE) { //Fictive or repulsive connections are not real, take no action dCompleted.push_back(connection); return; } float maxDist = -10e6; if(!success && !directProhibited && directConnectionIterator != dNeighbourZones.end()) { int3 guardPos(-1, -1, -1); int3 roadNode; for (auto borderPos : directConnectionIterator->second) { int3 potentialPos = zone.areaPossible().nearest(borderPos); assert(borderPos != potentialPos); //Check if guard pos doesn't touch any 3rd zone. This would create unwanted passage to 3rd zone bool adjacentZone = false; map.foreach_neighbour(potentialPos, [this, &adjacentZone, otherZoneId](int3 & pos) { auto zoneId = map.getZoneID(pos); if (zoneId != zone.getId() && zoneId != otherZoneId) { adjacentZone = true; } }); if (adjacentZone) { continue; } //Take into account distance to objects from both sides float dist = std::min(map.getTileInfo(potentialPos).getNearestObjectDistance(), map.getTileInfo(borderPos).getNearestObjectDistance()); if (dist > 3) //Don't place guards at adjacent tiles { auto safetyGap = rmg::Area({ potentialPos }); safetyGap.unite(safetyGap.getBorderOutside()); safetyGap.intersect(zone.areaPossible()); if (!safetyGap.empty()) { safetyGap.intersect(otherZone->areaPossible()); if (safetyGap.empty()) { float distanceToCenter = zone.getPos().dist2d(potentialPos) * otherZone->getPos().dist2d(potentialPos); auto localDist = (dist - distanceToCenter) * //Prefer close to zone center (std::max(distanceToCenter, dist) / std::min(distanceToCenter, dist)); //Distance to center dominates and is negative, so imbalanced proportions will result in huge penalty if (localDist > maxDist) { maxDist = localDist; guardPos = potentialPos; roadNode = borderPos; } } } } } if(guardPos.valid()) { assert(zone.getModificator()); auto & manager = *zone.getModificator(); auto * monsterType = manager.chooseGuard(connection.getGuardStrength(), true); rmg::Area border(zone.getArea().getBorder()); border.unite(otherZone->getArea().getBorder()); auto costFunction = [&border](const int3 & s, const int3 & d) { return 1.f / (1.f + border.distanceSqr(d)); }; auto ourArea = zone.areaPossible() + zone.freePaths(); auto theirArea = otherZone->areaPossible() + otherZone->freePaths(); theirArea.add(guardPos); rmg::Path ourPath(ourArea); rmg::Path theirPath(theirArea); ourPath.connect(zone.freePaths()); ourPath = ourPath.search(guardPos, true, costFunction); theirPath.connect(otherZone->freePaths()); theirPath = theirPath.search(guardPos, true, costFunction); if(ourPath.valid() && theirPath.valid()) { zone.connectPath(ourPath); otherZone->connectPath(theirPath); if(monsterType) { rmg::Object monster(*monsterType); monster.setPosition(guardPos); manager.placeObject(monster, false, true); //Place objects away from the monster in the other zone, too otherZone->getModificator()->updateDistances(monster); } else { //Update distances from empty passage, too zone.areaPossible().erase(guardPos); zone.freePaths().add(guardPos); map.setOccupied(guardPos, ETileType::FREE); manager.updateDistances(guardPos); otherZone->getModificator()->updateDistances(guardPos); } if (createRoad) { assert(zone.getModificator()); zone.getModificator()->addRoadNode(guardPos); assert(otherZone->getModificator()); otherZone->getModificator()->addRoadNode(roadNode); assert(otherZone->getModificator()); otherZone->getModificator()->otherSideConnection(connection); } success = true; } } } //2. connect via water bool waterMode = map.getMapGenOptions().getWaterContent() != EWaterContent::NONE; if(waterMode && zone.isUnderground() == otherZone->isUnderground()) { if(generator.getZoneWater() && generator.getZoneWater()->getModificator()) { if(generator.getZoneWater()->getModificator()->waterKeepConnection(connection, createRoad)) { assert(otherZone->getModificator()); otherZone->getModificator()->otherSideConnection(connection); success = true; } } } if(success) dCompleted.push_back(connection); } void ConnectionsPlacer::selfSideIndirectConnection(const rmg::ZoneConnection & connection) { bool success = false; auto otherZoneId = (connection.getZoneA() == zone.getId() ? connection.getZoneB() : connection.getZoneA()); auto & otherZone = map.getZones().at(otherZoneId); bool allowRoad = shouldGenerateRoad(connection); //3. place subterrain gates if(zone.isUnderground() != otherZone->isUnderground()) { int3 zShift(0, 0, zone.getPos().z - otherZone->getPos().z); std::scoped_lock doubleLock(zone.areaMutex, otherZone->areaMutex); auto commonArea = zone.areaPossible() * (otherZone->areaPossible() + zShift); if(!commonArea.empty()) { assert(zone.getModificator()); auto & manager = *zone.getModificator(); assert(otherZone->getModificator()); auto & managerOther = *otherZone->getModificator(); auto factory = VLC->objtypeh->getHandlerFor(Obj::SUBTERRANEAN_GATE, 0); auto * gate1 = factory->create(map.mapInstance->cb, nullptr); auto * gate2 = factory->create(map.mapInstance->cb, nullptr); rmg::Object rmgGate1(*gate1); rmg::Object rmgGate2(*gate2); rmgGate1.setTemplate(zone.getTerrainType(), zone.getRand()); rmgGate2.setTemplate(otherZone->getTerrainType(), zone.getRand()); bool guarded1 = manager.addGuard(rmgGate1, connection.getGuardStrength(), true); bool guarded2 = managerOther.addGuard(rmgGate2, connection.getGuardStrength(), true); int minDist = 3; rmg::Path path2(otherZone->area()); rmg::Path path1 = manager.placeAndConnectObject(commonArea, rmgGate1, [this, minDist, &path2, &rmgGate1, &zShift, guarded2, &managerOther, &rmgGate2 ](const int3 & tile) { auto ti = map.getTileInfo(tile); auto otherTi = map.getTileInfo(tile - zShift); float dist = ti.getNearestObjectDistance(); float otherDist = otherTi.getNearestObjectDistance(); if(dist < minDist || otherDist < minDist) return -1.f; //This could fail is accessibleArea is below the map rmg::Area toPlace(rmgGate1.getArea()); toPlace.unite(toPlace.getBorderOutside()); // Add a bit of extra space around toPlace.erase_if([this](const int3 & tile) { return !map.isOnMap(tile); }); toPlace.translate(-zShift); path2 = managerOther.placeAndConnectObject(toPlace, rmgGate2, minDist, guarded2, true, ObjectManager::OptimizeType::NONE); return path2.valid() ? (dist * otherDist) : -1.f; }, guarded1, true, ObjectManager::OptimizeType::DISTANCE); if(path1.valid() && path2.valid()) { zone.connectPath(path1); otherZone->connectPath(path2); manager.placeObject(rmgGate1, guarded1, true, allowRoad); managerOther.placeObject(rmgGate2, guarded2, true, allowRoad); assert(otherZone->getModificator()); otherZone->getModificator()->otherSideConnection(connection); success = true; } } } //4. place monoliths/portals if(!success) { auto factory = VLC->objtypeh->getHandlerFor(Obj::MONOLITH_TWO_WAY, generator.getNextMonlithIndex()); auto * teleport1 = factory->create(map.mapInstance->cb, nullptr); auto * teleport2 = factory->create(map.mapInstance->cb, nullptr); RequiredObjectInfo obj1(teleport1, connection.getGuardStrength(), allowRoad); RequiredObjectInfo obj2(teleport2, connection.getGuardStrength(), allowRoad); zone.getModificator()->addRequiredObject(obj1); otherZone->getModificator()->addRequiredObject(obj2); assert(otherZone->getModificator()); otherZone->getModificator()->otherSideConnection(connection); success = true; } if(success) dCompleted.push_back(connection); } void ConnectionsPlacer::collectNeighbourZones() { auto border = zone.area().getBorderOutside(); for(const auto & i : border) { if(!map.isOnMap(i)) continue; auto zid = map.getZoneID(i); assert(zid != zone.getId()); dNeighbourZones[zid].insert(i); } } bool ConnectionsPlacer::shouldGenerateRoad(const rmg::ZoneConnection& connection) const { return connection.getRoadOption() == rmg::ERoadOption::ROAD_TRUE || (connection.getRoadOption() == rmg::ERoadOption::ROAD_RANDOM && zone.getRand().nextDouble() >= 0.5f); } void ConnectionsPlacer::createBorder() { rmg::Area borderArea(zone.getArea().getBorder()); rmg::Area borderOutsideArea(zone.getArea().getBorderOutside()); auto blockBorder = borderArea.getSubarea([this, &borderOutsideArea](const int3 & t) { auto tile = borderOutsideArea.nearest(t); return map.isOnMap(tile) && map.getZones()[map.getZoneID(tile)]->getType() != ETemplateZoneType::WATER; }); //No border for wide connections for (auto& connection : zone.getConnections()) // We actually placed that connection already { auto otherZone = connection.getOtherZoneId(zone.getId()); if (connection.getConnectionType() == rmg::EConnectionType::WIDE) { auto sharedBorder = borderArea.getSubarea([this, otherZone, &borderOutsideArea](const int3 & t) { auto tile = borderOutsideArea.nearest(t); return map.isOnMap(tile) && map.getZones()[map.getZoneID(tile)]->getId() == otherZone; }); blockBorder.subtract(sharedBorder); } }; Zone::Lock lock(zone.areaMutex); //Protect from erasing same tiles again for(const auto & tile : blockBorder.getTilesVector()) { if(map.isPossible(tile)) { map.setOccupied(tile, ETileType::BLOCKED); zone.areaPossible().erase(tile); } map.foreachDirectNeighbour(tile, [this](int3 &nearbyPos) { if(map.isPossible(nearbyPos) && map.getZoneID(nearbyPos) == zone.getId()) { map.setOccupied(nearbyPos, ETileType::BLOCKED); zone.areaPossible().erase(nearbyPos); } }); } } VCMI_LIB_NAMESPACE_END