From dea7f604f90afbd61a4fccee0646274c7ea0c0dd Mon Sep 17 00:00:00 2001
From: barrulus <b@rry.im>
Date: Mon, 11 Aug 2025 18:21:33 +0100
Subject: [PATCH] Enabled larger maps with more burgs

---
 index.html                  |  2 +-
 main.js                     | 54 ++++++++++++++++++++++++++++++++-----
 modules/burgs-and-states.js |  2 +-
 3 files changed, 49 insertions(+), 9 deletions(-)

diff --git a/index.html b/index.html
index 3c0ba0f3..78dc5507 100644
--- a/index.html
+++ b/index.html
@@ -1713,7 +1713,7 @@
               </td>
               <td>Towns number</td>
               <td>
-                <input id="manorsInput" data-stored="manors" type="range" min="0" max="1000" step="1" value="1000" />
+                <input id="manorsInput" data-stored="manors" type="range" min="0" max="100000" step="1" value="100000" />
               </td>
               <td>
                 <output id="manorsOutput" data-stored="manors" value="auto"></output>
diff --git a/main.js b/main.js
index 848ff3c5..bd6ed474 100644
--- a/main.js
+++ b/main.js
@@ -249,7 +249,7 @@ document.addEventListener("DOMContentLoaded", async () => {
     hideLoading();
     await checkLoadParameters();
   }
-  restoreDefaultEvents(); // apply default viewbox events
+  restoreDefaultEvents; // apply default viewbox events
   initiateAutosave();
 });
 
@@ -647,10 +647,11 @@ async function generate(options) {
     Rivers.generate();
     Biomes.define();
 
-    rankCells();
+    rankCells(); // original suitability
     Cultures.generate();
     Cultures.expand();
     BurgsAndStates.generate();
+    updateSuitabilityFromHubs(); // <-- recalculate suitability based on distance from hubs
     Routes.generate();
     Religions.generate();
     BurgsAndStates.defineStateForms();
@@ -1172,9 +1173,10 @@ function rankCells() {
   cells.s = new Int16Array(cells.i.length); // cell suitability array
   cells.pop = new Float32Array(cells.i.length); // cell population array
 
-  const flMean = d3.median(cells.fl.filter(f => f)) || 0;
-  const flMax = d3.max(cells.fl) + d3.max(cells.conf); // to normalize flux
-  const areaMean = d3.mean(cells.area); // to adjust population by cell area
+  // --- ORIGINAL LOGIC RESTORED ---
+  const flMean = d3.median(cells.fl?.filter(f => f)) || 0;
+  const flMax = d3.max(cells.fl || []) + d3.max(cells.conf || []); // to normalize flux
+  const areaMean = d3.mean(cells.area);
 
   for (const i of cells.i) {
     if (cells.h[i] < 20) continue; // no population in water
@@ -1186,14 +1188,14 @@ function rankCells() {
     if (cells.t[i] === 1) {
       if (cells.r[i]) s += 15; // estuary is valued
       const feature = features[cells.f[cells.haven[i]]];
-      if (feature.type === "lake") {
+      if (feature && feature.type === "lake") {
         if (feature.group === "freshwater") s += 30;
         else if (feature.group == "salt") s += 10;
         else if (feature.group == "frozen") s += 1;
         else if (feature.group == "dry") s -= 5;
         else if (feature.group == "sinkhole") s -= 5;
         else if (feature.group == "lava") s -= 30;
-      } else {
+      } else if (feature) {
         s += 5; // ocean coast is valued
         if (cells.harbor[i] === 1) s += 20; // safe sea harbor is valued
       }
@@ -1207,6 +1209,44 @@ function rankCells() {
   TIME && console.timeEnd("rankCells");
 }
 
+// --- NEW FUNCTION: Update suitability based on distance from capitals/ports ---
+function updateSuitabilityFromHubs() {
+  TIME && console.time("updateSuitabilityFromHubs");
+  const {cells} = pack;
+  if (!pack.burgs) return;
+
+  const burgs = pack.burgs.filter(b => b.i && !b.removed);
+  const hubs = burgs.filter(b => b.capital || b.port).map(b => b.cell);
+  if (!hubs.length) return;
+
+  const cellCoords = cells.p;
+  const maxSuitability = 100;
+  const decayRate = 0.02;
+  const areaMean = d3.mean(cells.area);
+
+  function minDistToHub(cellId) {
+    const [x, y] = cellCoords[cellId];
+    let minDist = Infinity;
+    for (const hubCell of hubs) {
+      const [hx, hy] = cellCoords[hubCell];
+      const dist = Math.hypot(x - hx, y - hy);
+      if (dist < minDist) minDist = dist;
+    }
+    return minDist;
+  }
+
+  for (const i of cells.i) {
+    if (cells.h[i] < 20) continue;
+    const dist = minDistToHub(i);
+    let s = maxSuitability * Math.exp(-decayRate * dist);
+    s *= +biomesData.habitability[cells.biome[i]] / 100;
+    cells.s[i] = s;
+    cells.pop[i] = s > 0 ? (s * cells.area[i]) / areaMean : 0;
+  }
+
+  TIME && console.timeEnd("updateSuitabilityFromHubs");
+}
+
 // show map stats on generation complete
 function showStatistics() {
   const heightmap = byId("templateInput").value;
diff --git a/modules/burgs-and-states.js b/modules/burgs-and-states.js
index 8e3879c1..8c7242e3 100644
--- a/modules/burgs-and-states.js
+++ b/modules/burgs-and-states.js
@@ -120,7 +120,7 @@ window.BurgsAndStates = (() => {
         .sort((a, b) => score[b] - score[a]); // filtered and sorted array of indexes
 
       const desiredNumber =
-        manorsInput.value == 1000
+        manorsInput.value == 100000
           ? rn(sorted.length / 5 / (grid.points.length / 10000) ** 0.8)
           : manorsInput.valueAsNumber;
       const burgsNumber = Math.min(desiredNumber, sorted.length); // towns to generate