improve mse by 4% to 0.049 from 0.051 (rounded values)

Author: 0hmX

lib/utils/getTunedTotalCapacity1.ts

@@ -11,7 +11,9 @@ 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 } = {},
 ) => {
@@ -20,7 +22,17 @@ export const getTunedTotalCapacity1 = (
   const obstacleMargin = opts.obstacleMargin ?? 0.2
 
   const width = "width" in nodeOrWidth ? nodeOrWidth.width : nodeOrWidth
-  const viaLengthAcross = width / (VIA_DIAMETER / 2 + obstacleMargin)
+  const height =
+    "height" in nodeOrWidth && typeof nodeOrWidth.height === "number"
+      ? nodeOrWidth.height
+      : width
+
+  // 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
 
@@ -30,7 +42,6 @@ export const getTunedTotalCapacity1 = (
 
   return tunedTotalCapacity
 }
-
 /**
  * Calculate the optimal subdivision depth to reach a target minimum capacity
  * @param initialWidth The initial width of the top-level node