@tscircuit/rectdiff 0.0.14 → 0.0.16

package/tests/board-outline.test.ts

@@ -2,6 +2,7 @@ import { expect, test } from "bun:test"
 import boardWithCutout from "../test-assets/board-with-cutout.json"
 import { RectDiffPipeline } from "../lib/RectDiffPipeline"
 import { getSvgFromGraphicsObject } from "graphics-debug"
+import { makeCapacityMeshNodeWithLayerInfo } from "./fixtures/makeCapacityMeshNodeWithLayerInfo"
 
 test("board outline snapshot", async () => {
   const solver = new RectDiffPipeline({
@@ -11,9 +12,15 @@ test("board outline snapshot", async () => {
   // Run to completion
   solver.solve()
 
-  const viz =
-    solver.rectDiffGridSolverPipeline!.rectDiffSeedingSolver!.visualize()
-  const svg = getSvgFromGraphicsObject(viz)
+  const meshNodesGraphics = makeCapacityMeshNodeWithLayerInfo(
+    solver?.getOutput().meshNodes || [],
+  )
+    .values()
+    .toArray()
+    .flat()
+  const svg = getSvgFromGraphicsObject({
+    rects: meshNodesGraphics,
+  })
 
   await expect(svg).toMatchSvgSnapshot(import.meta.path)
 })

package/tests/fixtures/makeCapacityMeshNodeWithLayerInfo.ts

@@ -0,0 +1,33 @@
+import type { Rect } from "graphics-debug"
+import type { CapacityMeshNode } from "lib/types/capacity-mesh-types"
+import { getColorForZLayer } from "lib/utils/getColorForZLayer"
+
+export const makeCapacityMeshNodeWithLayerInfo = (
+  nodes: CapacityMeshNode[],
+): Map<string, Rect[]> => {
+  const map = new Map<string, Rect[]>()
+
+  for (const node of nodes) {
+    if (!node.availableZ.length) continue
+    const key = node.availableZ.join(",")
+    const colors = getColorForZLayer(node.availableZ)
+    const rect: Rect = {
+      center: node.center,
+      width: node.width,
+      height: node.height,
+      layer: `z${key}`,
+      stroke: "black",
+      fill: node._containsObstacle ? "red" : colors.fill,
+      label: "node",
+    }
+
+    const existing = map.get(key)
+    if (existing) {
+      existing.push(rect)
+    } else {
+      map.set(key, [rect])
+    }
+  }
+
+  return map
+}