From 617db13cb454a02c1efeed730b596b7c3f51a917 Mon Sep 17 00:00:00 2001
From: CodedSakura <me@codedsakura.dev>
Date: Fri, 31 Jul 2026 19:27:30 +0300
Subject: [PATCH] begin implementing planner

---
 index.html                 |   2 +-
 scripts/metadata.js        |  26 ++++-----
 scripts/planner.js         |   2 +-
 scripts/recipe-selector.js |   7 ---
 scripts/solver.js          | 106 +++++++++++++++++++++++++++++++++++--
 5 files changed, 117 insertions(+), 26 deletions(-)

diff --git a/index.html b/index.html
index 7d702f0..34ff839 100644
--- a/index.html
+++ b/index.html
@@ -126,7 +126,7 @@
   </section>
 </aside>
 <footer>
-  <div>pre-alpha 1.6</div>
+  <div>alpha 1.7</div>
   <div>
     <a href="https://discord.gg/Eq72HwPqfg">Discord server</a>
   </div>
diff --git a/scripts/metadata.js b/scripts/metadata.js
index 8772551..78f909a 100644
--- a/scripts/metadata.js
+++ b/scripts/metadata.js
@@ -9,17 +9,17 @@ function createMetadata(data) {
   return id;
 }
 export function setDomMetadata(domNode, data) {
-  if (domNode.dataset.solverMetadata) {
-    metadata[Number(domNode.dataset.solverMetadata)] = data;
+  if (domNode.dataset.metadata) {
+    metadata[Number(domNode.dataset.metadata)] = data;
     return;
   }
-  domNode.dataset.solverMetadata = createMetadata(data).toString(10);
+  domNode.dataset.metadata = createMetadata(data).toString(10);
 }
 export function setCellMetadata(node, key, value) {
-  if (!node.dataset.solverMetadata) {
-    node.dataset.solverMetadata = createMetadata({ [key]: value }).toString(10);
+  if (!node.dataset.metadata) {
+    node.dataset.metadata = createMetadata({ [key]: value }).toString(10);
   } else {
-    metadata[Number(node.dataset.solverMetadata)][key] = value;
+    metadata[Number(node.dataset.metadata)][key] = value;
   }
   processCellOverlays(node, key, true);
 }
@@ -27,8 +27,8 @@ export function setCellMetadata(node, key, value) {
 export function getMetadata(id) {
   if (typeof id === "object") {
     // assume DOM element
-    if (id.dataset && id.dataset.solverMetadata) {
-      return getMetadata(id.dataset.solverMetadata);
+    if (id.dataset && id.dataset.metadata) {
+      return getMetadata(id.dataset.metadata);
     }
     return null;
   }
@@ -38,21 +38,21 @@ export function getMetadata(id) {
   return metadata[id];
 }
 export function getCellMetadata(node, key) {
-  return metadata[Number(node.dataset.solverMetadata)][key];
+  return metadata[Number(node.dataset.metadata)][key];
 }
 
 export function deleteMetadata(id) {
   if (typeof id === "object") {
     // assume DOM object
-    if (id.dataset && id.dataset.solverMetadata) {
+    if (id.dataset && id.dataset.metadata) {
       clearCellOverlays(id);
-      return deleteMetadata(id.dataset.solverMetadata);
+      return deleteMetadata(id.dataset.metadata);
     }
   }
   delete metadata[id];
 }
 export function clearCellMetadata(node, key) {
-  if (!node.dataset.solverMetadata) return;
-  delete metadata[Number(node.dataset.solverMetadata)][key];
+  if (!node.dataset.metadata) return;
+  delete metadata[Number(node.dataset.metadata)][key];
   processCellOverlays(node, key, false);
 }
diff --git a/scripts/planner.js b/scripts/planner.js
index 9e5d7d1..1cf05b4 100644
--- a/scripts/planner.js
+++ b/scripts/planner.js
@@ -324,7 +324,7 @@ export function getPlannerDom() {
   return document.getElementById("planner");
 }
 export function getPlannerCellsDom() {
-  return document.querySelectorAll("#planner [data-tile]");
+  return [...document.querySelectorAll("#planner [data-tile]")];
 }
 export function getPlannerCellDom(x, y, z) {
   return document.querySelector(`#planner [data-x="${x}"][data-y="${y}"][data-z="${z}"]`);
diff --git a/scripts/recipe-selector.js b/scripts/recipe-selector.js
index e197ed4..c5acf21 100644
--- a/scripts/recipe-selector.js
+++ b/scripts/recipe-selector.js
@@ -88,7 +88,6 @@ function initializeRecipePicker({
     const filter = Filter.fromInput(filterElem);
     const recipeList = new DomList(parent.querySelector(":scope > .select"));
     recipeList.clear();
-    console.log(recipes);
     for (let i = 0; i < recipes.length; i++) {
       const recipe = recipes[i];
       const recipeItem = recipeList.makeItem();
@@ -131,9 +130,6 @@ function initializeRecipePicker({
 }
 
 function rescanDesignRecipes() {
-  console.log("rescan recipes");
-  console.log(configuration);
-
   const multiblockType = getPlannerMultiblockType();
 
   const configs = [
@@ -198,8 +194,6 @@ function rescanSelectionRecipes() {
   const tileMetadata = getAtlasMetadata(selectedCells.length > 0 ? selectedCells[0].dataset.tile : pickerTile);
   const tiles = findFullEntryFromAtlasMetadata(tileMetadata, { all: true });
 
-  console.log(tiles);
-
   if (!tiles[0] || !tiles[0].modules || !tiles[0].modules["ncpf:block_recipes"]) {
     return;
   }
@@ -207,7 +201,6 @@ function rescanSelectionRecipes() {
   const cellRecipes = selectedCells.map(v => getMetadata(v)?.block_recipe);
   const recipeSet = cellRecipes.some(v => v);
   const recipeAllSame = cellRecipes.every(v => v === cellRecipes[0] || checkEquivalence(v, cellRecipes[0]));
-  console.log(">", cellRecipes, recipeSet, recipeAllSame);
 
   const recipes = tiles.flatMap(t => t.modules["ncpf:block_recipes"].recipes);
 
diff --git a/scripts/solver.js b/scripts/solver.js
index 4eeffe5..19f54ae 100644
--- a/scripts/solver.js
+++ b/scripts/solver.js
@@ -1,7 +1,21 @@
-import { emitEvent } from "./events.js";
-import { getPlannerCellDom, getPlannerDom } from "./planner.js";
-import { findFullEntry } from "./ncpf-utils.js";
-import { setCellMetadata, setDomMetadata } from "./metadata.js";
+import { emitEvent, listenTo } from "./events.js";
+import {
+  getPlannerCellDom,
+  getPlannerCellsDom,
+  getPlannerDom,
+  getPlannerMultiblockSize,
+  getPlannerMultiblockType
+} from "./planner.js";
+import { checkEquivalence, findFullEntry, findFullEntryFromAtlasMetadata } from "./ncpf-utils.js";
+import { getMetadata, setCellMetadata, setDomMetadata } from "./metadata.js";
+import { registerLoadCallback } from "./main.js";
+import { getAtlasMetadata } from "./atlas.js";
+import { initialize3dArray } from "./js-utils.js";
+
+registerLoadCallback(() => {
+  listenTo("fileLoad:end", runSolver);
+  listenTo("planner:cell:change", runSolver);
+});
 
 export function resetSolver(design = undefined, designConfiguration) {
   const planner = getPlannerDom();
@@ -34,3 +48,87 @@ export function resetSolver(design = undefined, designConfiguration) {
 
   emitEvent("solver:reset:done");
 }
+
+const sixAxes = [
+  {x: -1, y: 0, z: 0},
+  {x: 0, y: -1, z: 0},
+  {x: 0, y: 0, z: -1},
+  {x: 1, y: 0, z: 0},
+  {x: 0, y: 1, z: 0},
+  {x: 0, y: 0, z: 1},
+]
+function *travelUntil(grid, { x, y, z }, {x: dirX = 0, y: dirY = 0, z: dirZ = 0}, shouldStop = (pos) => false) {
+  x = Number(x);
+  y = Number(y);
+  z = Number(z);
+  while (!shouldStop({x, y, z})
+        && x > 0 && y > 0 && z > 0
+        && x < grid.length - 1 && y < grid[0].length - 1 && z < grid[0][0].length - 1) {
+    x += dirX;
+    y += dirY;
+    z += dirZ;
+    yield grid[x][y][z];
+  }
+}
+
+export function runSolver() {
+  emitEvent("solver:begin")
+
+  const multiblockType = getPlannerMultiblockType();
+
+  const plannerBlocks = getPlannerCellsDom()
+      .map(b => ({ dom: b, ...b.dataset }))
+      .filter(b => b.tile !== "-1")
+      .map(b => (b.metadata ? { ...b, metadata: getMetadata(b.metadata) } : b));
+  const tileMap = Object.fromEntries([...new Set(plannerBlocks.map(b => b.tile))]
+      .map(idx => [idx, getAtlasMetadata(idx)])
+      .map(([idx, md]) => [idx, findFullEntry("blocks", md, { configurationHint: multiblockType })]));
+  plannerBlocks.forEach(v => v.tile = tileMap[v.tile]);
+
+  const plannerBlocksGrid = initialize3dArray(...getPlannerMultiblockSize());
+  plannerBlocks.forEach(b => {
+    plannerBlocksGrid[b.x][b.y][b.z] = b;
+  });
+
+  if (multiblockType === "nuclearcraft:overhaul_sfr") {
+    const fuelCells = plannerBlocks.filter(b => b.tile.modules && b.tile.modules["nuclearcraft:overhaul_sfr:fuel_cell"]);
+    // const neutronSources = plannerBlocks.filter(b => b.tile.modules && b.tile.modules["nuclearcraft:overhaul_sfr:neutron_source"]);
+    // const reflectors = plannerBlocks.filter(b => b.tile.modules && b.tile.modules["nuclearcraft:overhaul_sfr:reflector"]);
+    // const shields = plannerBlocks.filter(b => b.tile.modules && b.tile.modules["nuclearcraft:overhaul_sfr:neutron_shield"]);
+    // const moderators = plannerBlocks.filter(b => b.tile.modules && b.tile.modules["nuclearcraft:overhaul_sfr:moderator"]);
+    // const coolers = plannerBlocks.filter(b => b.tile.modules && b.tile.modules["nuclearcraft:overhaul_sfr:heat_sink"]);
+    // const irradiators = plannerBlocks.filter(b => b.tile.modules && b.tile.modules["nuclearcraft:overhaul_sfr:irradiator"]);
+    // const conductors = plannerBlocks.filter(b => b.tile.modules && b.tile.modules["nuclearcraft:overhaul_sfr:conductor"]);
+
+    fuelCells.forEach(cell => {
+      const recipeList = cell.tile.modules["ncpf:block_recipes"].recipes;
+      cell.recipe = recipeList.find(r => checkEquivalence(r, cell.metadata.block_recipe));
+      cell.axes = sixAxes.map(dir => [...travelUntil(
+          plannerBlocksGrid,
+          cell,
+          dir,
+          ({x, y, z}) => {
+            const block = plannerBlocksGrid[x][y][z];
+            if (block === cell) {
+              return false;
+            }
+            if (!block) {
+              return false;
+            }
+            if (block.tile.modules) {
+              if (block.tile.modules["nuclearcraft:overhaul_sfr:fuel_cell"]) return true;
+              if (block.tile.modules["nuclearcraft:overhaul_sfr:reflector"]) return true;
+            }
+            return false;
+          }
+      )]);
+      if (cell.recipe.modules["nuclearcraft:overhaul_sfr:fuel_stats"].self_priming) {
+        cell.primer = cell;
+      } else if (cell.axes.some(a => (a[a.length - 1]?.tile?.modules ?? {})["nuclearcraft:overhaul_sfr:neutron_source"])) {
+        const primerAxis = cell.axes.find(a => (a[a.length - 1]?.tile?.modules ?? {})["nuclearcraft:overhaul_sfr:neutron_source"]);
+        cell.primer = primerAxis[primerAxis.length - 1];
+      }
+    });
+    console.log(fuelCells);
+  }
+}