Reduces MSE from 0.051 to 0.049 (~4% improvement, rounded values).

lib/utils/getTunedTotalCapacity1.ts

@@ -11,16 +11,27 @@ import { CapacityMeshNode } from "lib/types/capacity-mesh-types"
  * @returns The calculated capacity
  */
 export const getTunedTotalCapacity1 = (
-  nodeOrWidth: CapacityMeshNode | { width: number; availableZ?: number[] },
+  nodeOrWidth:
+    | CapacityMeshNode
+    | { width: number; height?: number; availableZ?: number[] },
   maxCapacityFactor = 1,
   opts: { viaDiameter?: number; obstacleMargin?: number } = {},
 ) => {
   const VIA_DIAMETER = opts.viaDiameter ?? 0.3
   const TRACE_WIDTH = 0.15
+  const width = "width" in nodeOrWidth ? nodeOrWidth.width : nodeOrWidth
   const obstacleMargin = opts.obstacleMargin ?? 0.2
+  const height =
+    "height" in nodeOrWidth && typeof nodeOrWidth.height === "number"
+      ? nodeOrWidth.height
+      : width
 
-  const width = "width" in nodeOrWidth ? nodeOrWidth.width : nodeOrWidth
-  const viaLengthAcross = width / (VIA_DIAMETER / 2 + obstacleMargin)
+  // Use the geometric mean so thin rectangular nodes are assigned less capacity
+  // than similarly wide square nodes while preserving previous behavior when only
+  // width is provided.
+  const effectiveNodeSpan = Math.sqrt(width * height)
+  const viaLengthAcross =
+    effectiveNodeSpan / (VIA_DIAMETER / 2 + obstacleMargin)
 
   const tunedTotalCapacity = (viaLengthAcross / 2) ** 1.1 * maxCapacityFactor