@tscircuit/rectdiff 0.0.33 → 0.0.34

package/components/SolverDebugger3d.tsx

@@ -2,6 +2,7 @@ import React, { useCallback, useEffect, useMemo, useRef, useState } from "react"
 import { GenericSolverDebugger } from "@tscircuit/solver-utils/react"
 import type { SimpleRouteJson } from "../lib/types/srj-types"
 import type { CapacityMeshNode } from "../lib/types/capacity-mesh-types"
+import { matchesExactZFilter, parseZFilterInput } from "../lib/utils/z-filter"
 import * as THREE from "three"
 import { OrbitControls } from "three/examples/jsm/controls/OrbitControls.js"
 import type { BaseSolver } from "@tscircuit/solver-utils"
@@ -168,6 +169,7 @@ const ThreeBoardView: React.FC<{
   shrinkBoxes: boolean
   boxShrinkAmount: number
   showBorders: boolean
+  selectedZValues: number[] | null
 }> = ({
   nodes,
   srj,
@@ -181,6 +183,7 @@ const ThreeBoardView: React.FC<{
   shrinkBoxes,
   boxShrinkAmount,
   showBorders,
+  selectedZValues,
 }) => {
   const containerRef = useRef<HTMLDivElement>(null)
   const destroyRef = useRef<() => void>(() => {})
@@ -203,9 +206,17 @@ const ThreeBoardView: React.FC<{
 
   const layerCount = layerNames.length || srj?.layerCount || 1
 
-  const prisms = useMemo(
-    () => buildPrismsFromNodes(nodes, layerCount),
-    [nodes, layerCount],
+  const filteredNodes = useMemo(
+    () =>
+      nodes.filter((node) =>
+        matchesExactZFilter(node.availableZ ?? [], selectedZValues),
+      ),
+    [nodes, selectedZValues],
+  )
+
+  const filteredPrisms = useMemo(
+    () => buildPrismsFromNodes(filteredNodes, layerCount),
+    [filteredNodes, layerCount],
   )
 
   useEffect(() => {
@@ -372,6 +383,8 @@ const ThreeBoardView: React.FC<{
                   .map((name) => zIndexByLayerName.get(name))
                   .filter((z): z is number => typeof z === "number")
 
+          if (!matchesExactZFilter(zs, selectedZValues)) continue
+
           for (const z of zs) {
             if (z < 0 || z >= layerCount) continue
             obstaclesGroup.add(
@@ -389,7 +402,7 @@ const ThreeBoardView: React.FC<{
 
       // Output prisms from nodes (wireframe toggle like the experiment)
       if (showOutput) {
-        for (const p of prisms) {
+        for (const p of filteredPrisms) {
           let box = p
           if (shrinkBoxes && boxShrinkAmount > 0) {
             const s = boxShrinkAmount
@@ -442,7 +455,7 @@ const ThreeBoardView: React.FC<{
             z1: layerCount,
           }
         : (() => {
-            if (prisms.length === 0) {
+            if (filteredPrisms.length === 0) {
               return {
                 minX: -10,
                 maxX: 10,
@@ -456,7 +469,7 @@ const ThreeBoardView: React.FC<{
               minY = Infinity,
               maxX = -Infinity,
               maxY = -Infinity
-            for (const p of prisms) {
+            for (const p of filteredPrisms) {
               minX = Math.min(minX, p.minX)
               maxX = Math.max(maxX, p.maxX)
               minY = Math.min(minY, p.minY)
@@ -517,7 +530,7 @@ const ThreeBoardView: React.FC<{
     }
   }, [
     srj,
-    prisms,
+    filteredPrisms,
     layerCount,
     layerThickness,
     height,
@@ -530,6 +543,7 @@ const ThreeBoardView: React.FC<{
     shrinkBoxes,
     boxShrinkAmount,
     showBorders,
+    selectedZValues,
   ])
 
   return (
@@ -572,10 +586,18 @@ export const SolverDebugger3d: React.FC<SolverDebugger3dProps> = ({
   const [shrinkBoxes, setShrinkBoxes] = useState(true)
   const [boxShrinkAmount, setBoxShrinkAmount] = useState(0.1)
   const [showBorders, setShowBorders] = useState(true)
+  const [zFilterInput, setZFilterInput] = useState("")
 
   // Mesh nodes state - updated when solver completes or during stepping
   const [meshNodes, setMeshNodes] = useState<CapacityMeshNode[]>([])
 
+  const selectedZValues = useMemo(
+    () => parseZFilterInput(zFilterInput),
+    [zFilterInput],
+  )
+  const hasInvalidZFilter =
+    zFilterInput.trim().length > 0 && selectedZValues === null
+
   // Update mesh nodes from solver output
   const updateMeshNodes = useCallback(() => {
     try {
@@ -601,11 +623,16 @@ export const SolverDebugger3d: React.FC<SolverDebugger3dProps> = ({
   // Poll for updates during stepping (GenericSolverDebugger doesn't have onStep)
   useEffect(() => {
     const interval = setInterval(() => {
-      // Only update if solver has output available
-      if (solver.solved || solver.stats?.placed > 0) {
+      if (solver.solved) {
         updateMeshNodes()
+        clearInterval(interval)
+        return
       }
-    }, 100) // Poll every 100ms during active solving
+
+      if (solver.stats?.placed > 0) {
+        updateMeshNodes()
+      }
+    }, 100)
 
     return () => clearInterval(interval)
   }, [updateMeshNodes, solver])
@@ -724,6 +751,32 @@ export const SolverDebugger3d: React.FC<SolverDebugger3dProps> = ({
                 />
                 Wireframe
               </label>
+
+              <label
+                style={{
+                  display: "inline-flex",
+                  gap: 8,
+                  alignItems: "center",
+                  fontSize: 13,
+                }}
+              >
+                <span>Z Filter</span>
+                <input
+                  type="text"
+                  value={zFilterInput}
+                  onChange={(e) => setZFilterInput(e.target.value)}
+                  placeholder="1 or 1,2,3"
+                  style={{
+                    width: 120,
+                    padding: "6px 10px",
+                    borderRadius: 6,
+                    border: `1px solid ${hasInvalidZFilter ? "#dc2626" : "#cbd5e1"}`,
+                    background: "white",
+                    fontSize: 13,
+                  }}
+                  title="Show only exact z matches. Example: 1 matches [1], 1,2,3 matches [1,2,3]."
+                />
+              </label>
             </>
           )}
         </div>
@@ -752,6 +805,7 @@ export const SolverDebugger3d: React.FC<SolverDebugger3dProps> = ({
             shrinkBoxes={shrinkBoxes}
             boxShrinkAmount={boxShrinkAmount}
             showBorders={showBorders}
+            selectedZValues={selectedZValues}
           />
         )}
       </div>

package/lib/fixtures/twoNodeExpansionFixture.ts

@@ -2,7 +2,7 @@ import RBush from "rbush"
 import type { RectDiffExpansionSolverInput } from "../solvers/RectDiffExpansionSolver/RectDiffExpansionSolver"
 import type { SimpleRouteJson } from "../types/srj-types"
 import type { XYRect } from "../rectdiff-types"
-import type { RTreeRect } from "lib/types/capacity-mesh-types"
+import type { RTreeRect } from "../types/capacity-mesh-types"
 import { buildZIndexMap } from "../solvers/RectDiffSeedingSolver/layers"
 
 /**

package/lib/solvers/GapFillSolver/ExpandEdgesToEmptySpaceSolver.ts

@@ -1,5 +1,5 @@
 import { BaseSolver } from "@tscircuit/solver-utils"
-import type { CapacityMeshNode } from "lib/types/capacity-mesh-types"
+import type { CapacityMeshNode } from "../../types/capacity-mesh-types"
 import type { SegmentWithAdjacentEmptySpace } from "./FindSegmentsWithAdjacentEmptySpaceSolver"
 import type { GraphicsObject } from "graphics-debug"
 import RBush from "rbush"
@@ -7,7 +7,7 @@ import { EDGE_MAP, EDGES } from "./edge-constants"
 import { getBoundsFromCorners } from "./getBoundsFromCorners"
 import type { Bounds } from "@tscircuit/math-utils"
 import { midpoint, segmentToBoxMinDistance } from "@tscircuit/math-utils"
-import type { XYRect } from "lib/rectdiff-types"
+import type { XYRect } from "../../rectdiff-types"
 
 const EPS = 1e-4
 

package/lib/solvers/GapFillSolver/FindSegmentsWithAdjacentEmptySpaceSolver.ts

@@ -1,7 +1,7 @@
 import { BaseSolver } from "@tscircuit/solver-utils"
 import Flatbush from "flatbush"
 import type { GraphicsObject, NinePointAnchor } from "graphics-debug"
-import type { CapacityMeshNode } from "lib/types/capacity-mesh-types"
+import type { CapacityMeshNode } from "../../types/capacity-mesh-types"
 import { projectToUncoveredSegments } from "./projectToUncoveredSegments"
 import { EDGES } from "./edge-constants"
 import { visuallyOffsetLine } from "./visuallyOffsetLine"

package/lib/solvers/GapFillSolver/GapFillSolverPipeline.ts

@@ -3,11 +3,11 @@ import {
   definePipelineStep,
   type PipelineStep,
 } from "@tscircuit/solver-utils"
-import type { CapacityMeshNode } from "lib/types/capacity-mesh-types"
+import type { CapacityMeshNode } from "../../types/capacity-mesh-types"
 import type { GraphicsObject } from "graphics-debug"
 import { FindSegmentsWithAdjacentEmptySpaceSolver } from "./FindSegmentsWithAdjacentEmptySpaceSolver"
 import { ExpandEdgesToEmptySpaceSolver } from "./ExpandEdgesToEmptySpaceSolver"
-import type { XYRect } from "lib/rectdiff-types"
+import type { XYRect } from "../../rectdiff-types"
 
 type GapFillSolverInput = {
   meshNodes: CapacityMeshNode[]

package/lib/solvers/RectDiffExpansionSolver/RectDiffExpansionSolver.ts

@@ -1,12 +1,15 @@
 import { BaseSolver } from "@tscircuit/solver-utils"
 import type { GraphicsObject } from "graphics-debug"
-import type { CapacityMeshNode, RTreeRect } from "lib/types/capacity-mesh-types"
+import type {
+  CapacityMeshNode,
+  RTreeRect,
+} from "../../types/capacity-mesh-types"
 import { expandRectFromSeed } from "../../utils/expandRectFromSeed"
 import { finalizeRects } from "../../utils/finalizeRects"
 import { resizeSoftOverlaps } from "../../utils/resizeSoftOverlaps"
 import { rectsToMeshNodes } from "./rectsToMeshNodes"
 import type { XYRect, Candidate3D, Placed3D } from "../../rectdiff-types"
-import type { Obstacle } from "lib/types/srj-types"
+import type { Obstacle } from "../../types/srj-types"
 import RBush from "rbush"
 import { rectToTree } from "../../utils/rectToTree"
 import { sameTreeRect } from "../../utils/sameTreeRect"

package/lib/solvers/RectDiffGridSolverPipeline/RectDiffGridSolverPipeline.ts

@@ -7,11 +7,14 @@ import type {
   Obstacle,
   SimpleRouteConnection,
   SimpleRouteJson,
-} from "lib/types/srj-types"
-import type { GridFill3DOptions, XYRect } from "lib/rectdiff-types"
-import type { CapacityMeshNode, RTreeRect } from "lib/types/capacity-mesh-types"
-import { RectDiffSeedingSolver } from "lib/solvers/RectDiffSeedingSolver/RectDiffSeedingSolver"
-import { RectDiffExpansionSolver } from "lib/solvers/RectDiffExpansionSolver/RectDiffExpansionSolver"
+} from "../../types/srj-types"
+import type { GridFill3DOptions, XYRect } from "../../rectdiff-types"
+import type {
+  CapacityMeshNode,
+  RTreeRect,
+} from "../../types/capacity-mesh-types"
+import { RectDiffSeedingSolver } from "../RectDiffSeedingSolver/RectDiffSeedingSolver"
+import { RectDiffExpansionSolver } from "../RectDiffExpansionSolver/RectDiffExpansionSolver"
 import type { GraphicsObject } from "graphics-debug"
 import RBush from "rbush"
 import { buildObstacleIndexesByLayer } from "./buildObstacleIndexes"

package/lib/solvers/RectDiffGridSolverPipeline/buildObstacleIndexes.ts

@@ -1,14 +1,14 @@
-import type { SimpleRouteJson } from "lib/types/srj-types"
+import type { SimpleRouteJson } from "../../types/srj-types"
 import RBush from "rbush"
-import { computeInverseRects } from "lib/solvers/RectDiffSeedingSolver/computeInverseRects"
+import { computeInverseRects } from "../RectDiffSeedingSolver/computeInverseRects"
 import {
   buildZIndexMap,
   obstacleToXYRect,
   obstacleZs,
-} from "lib/solvers/RectDiffSeedingSolver/layers"
-import type { XYRect } from "lib/rectdiff-types"
-import type { RTreeRect } from "lib/types/capacity-mesh-types"
-import { padRect } from "lib/utils/padRect"
+} from "../RectDiffSeedingSolver/layers"
+import type { XYRect } from "../../rectdiff-types"
+import type { RTreeRect } from "../../types/capacity-mesh-types"
+import { padRect } from "../../utils/padRect"
 
 export const buildObstacleIndexesByLayer = (params: {
   srj: SimpleRouteJson

package/lib/solvers/RectDiffSeedingSolver/RectDiffSeedingSolver.ts

@@ -16,12 +16,12 @@ import { computeCandidates3D } from "./computeCandidates3D"
 import { computeEdgeCandidates3D } from "./computeEdgeCandidates3D"
 import { longestFreeSpanAroundZ } from "./longestFreeSpanAroundZ"
 import { allLayerNode } from "../../utils/buildHardPlacedByLayer"
-import { isFullyOccupiedAtPoint } from "lib/utils/isFullyOccupiedAtPoint"
+import { isFullyOccupiedAtPoint } from "../../utils/isFullyOccupiedAtPoint"
 import { resizeSoftOverlaps } from "../../utils/resizeSoftOverlaps"
-import { getColorForZLayer } from "lib/utils/getColorForZLayer"
+import { getColorForZLayer } from "../../utils/getColorForZLayer"
 import RBush from "rbush"
-import type { RTreeRect } from "lib/types/capacity-mesh-types"
-import { rectToTree } from "lib/utils/rectToTree"
+import type { RTreeRect } from "../../types/capacity-mesh-types"
+import { rectToTree } from "../../utils/rectToTree"
 
 export type RectDiffSeedingSolverInput = {
   simpleRouteJson: SimpleRouteJson

package/lib/solvers/RectDiffSeedingSolver/computeCandidates3D.ts

@@ -3,7 +3,7 @@ import { EPS, distancePointToRectEdges } from "../../utils/rectdiff-geometry"
 import { isFullyOccupiedAtPoint } from "../../utils/isFullyOccupiedAtPoint"
 import { longestFreeSpanAroundZ } from "./longestFreeSpanAroundZ"
 import type RBush from "rbush"
-import type { RTreeRect } from "lib/types/capacity-mesh-types"
+import type { RTreeRect } from "../../types/capacity-mesh-types"
 const quantize = (value: number, precision = 1e-6) =>
   Math.round(value / precision) * precision
 

package/lib/solvers/RectDiffSeedingSolver/computeEdgeCandidates3D.ts

@@ -3,7 +3,7 @@ import { EPS, distancePointToRectEdges } from "../../utils/rectdiff-geometry"
 import { isFullyOccupiedAtPoint } from "../../utils/isFullyOccupiedAtPoint"
 import { longestFreeSpanAroundZ } from "./longestFreeSpanAroundZ"
 import type RBush from "rbush"
-import type { RTreeRect } from "lib/types/capacity-mesh-types"
+import type { RTreeRect } from "../../types/capacity-mesh-types"
 const quantize = (value: number, precision = 1e-6) =>
   Math.round(value / precision) * precision
 

package/lib/solvers/RectDiffSeedingSolver/longestFreeSpanAroundZ.ts

@@ -1,4 +1,4 @@
-import type { RTreeRect } from "lib/types/capacity-mesh-types"
+import type { RTreeRect } from "../../types/capacity-mesh-types"
 import type { XYRect } from "../../rectdiff-types"
 import { clamp, containsPoint } from "../../utils/rectdiff-geometry"
 import type RBush from "rbush"

package/lib/types/capacity-mesh-types.ts

@@ -1,4 +1,4 @@
-import type { XYRect } from "lib/rectdiff-types"
+import type { XYRect } from "../rectdiff-types"
 
 export type CapacityMeshNodeId = string
 

package/lib/utils/expandRectFromSeed.ts

@@ -1,7 +1,7 @@
 import type RBush from "rbush"
 import type { XYRect } from "../rectdiff-types"
 import { EPS, gt, gte, lt, lte, overlaps } from "./rectdiff-geometry"
-import type { RTreeRect } from "lib/types/capacity-mesh-types"
+import type { RTreeRect } from "../types/capacity-mesh-types"
 import { isSelfRect } from "./isSelfRect"
 import {
   searchStripDown,

package/lib/utils/finalizeRects.ts

@@ -1,4 +1,4 @@
-import type { Obstacle } from "lib/types/srj-types"
+import type { Obstacle } from "../types/srj-types"
 import type { Placed3D, Rect3d, XYRect } from "../rectdiff-types"
 import {
   obstacleToXYRect,

package/lib/utils/isFullyOccupiedAtPoint.ts

@@ -1,4 +1,4 @@
-import type { RTreeRect } from "lib/types/capacity-mesh-types"
+import type { RTreeRect } from "../types/capacity-mesh-types"
 import RBush from "rbush"
 
 export type OccupancyParams = {

package/lib/utils/isSelfRect.ts

@@ -1,4 +1,4 @@
-import type { XYRect } from "lib/rectdiff-types"
+import type { XYRect } from "../rectdiff-types"
 
 const EPS = 1e-9
 

package/lib/utils/rectToTree.ts

@@ -1,5 +1,5 @@
-import type { XYRect } from "lib/rectdiff-types"
-import type { RTreeRect } from "lib/types/capacity-mesh-types"
+import type { XYRect } from "../rectdiff-types"
+import type { RTreeRect } from "../types/capacity-mesh-types"
 
 export const rectToTree = (
   rect: XYRect,

package/lib/utils/renderObstacleClearance.ts

@@ -1,4 +1,4 @@
-import type { SimpleRouteJson } from "lib/types/srj-types"
+import type { SimpleRouteJson } from "../types/srj-types"
 import type { GraphicsObject } from "graphics-debug"
 
 /**

package/lib/utils/resizeSoftOverlaps.ts

@@ -1,4 +1,4 @@
-import type { RTreeRect } from "lib/types/capacity-mesh-types"
+import type { RTreeRect } from "../types/capacity-mesh-types"
 import type { Placed3D } from "../rectdiff-types"
 import { overlaps, subtractRect2D, EPS } from "./rectdiff-geometry"
 import type RBush from "rbush"

package/lib/utils/sameTreeRect.ts

@@ -1,4 +1,4 @@
-import type { RTreeRect } from "lib/types/capacity-mesh-types"
+import type { RTreeRect } from "../types/capacity-mesh-types"
 
 export const sameTreeRect = (a: RTreeRect, b: RTreeRect) =>
   a.minX === b.minX &&

package/lib/utils/searchStrip.ts

@@ -1,4 +1,4 @@
-import type { XYRect } from "lib/rectdiff-types"
+import type { XYRect } from "../rectdiff-types"
 
 export const searchStripRight = ({
   rect,

package/lib/utils/z-filter.ts

@@ -0,0 +1,43 @@
+export function normalizeZValues(zValues: number[]): number[] {
+  return [...new Set(zValues)].sort((a, b) => a - b)
+}
+
+export function parseZFilterInput(input: string): number[] | null {
+  const zValues: number[] = []
+  let current = ""
+
+  for (const char of input) {
+    if (char >= "0" && char <= "9") {
+      current += char
+      continue
+    }
+
+    if (char === "," || char === " " || char === "\t" || char === "\n") {
+      if (char === ",") {
+        if (current === "") return null
+        zValues.push(Number(current))
+        current = ""
+      }
+      continue
+    }
+
+    return null
+  }
+
+  if (current !== "") zValues.push(Number(current))
+  if (zValues.length === 0) return null
+
+  return normalizeZValues(zValues)
+}
+
+export function matchesExactZFilter(
+  zValues: number[],
+  selectedZValues: number[] | null,
+): boolean {
+  if (!selectedZValues) return true
+
+  const normalized = normalizeZValues(zValues)
+  if (normalized.length !== selectedZValues.length) return false
+
+  return normalized.every((z, index) => z === selectedZValues[index])
+}

package/package.json

@@ -1,8 +1,12 @@
 {
   "name": "@tscircuit/rectdiff",
-  "version": "0.0.33",
+  "version": "0.0.34",
   "type": "module",
-  "main": "dist/index.js",
+  "main": "lib/index.ts",
+  "types": "lib/index.ts",
+  "exports": {
+    ".": "./lib/index.ts"
+  },
   "scripts": {
     "start": "cosmos",
     "build": "tsup-node ./lib/index.ts --format esm --dts",

package/tests/z-filter.test.ts

@@ -0,0 +1,33 @@
+import { expect, test } from "bun:test"
+import { matchesExactZFilter, parseZFilterInput } from "../lib/utils/z-filter"
+
+test("parseZFilterInput parses a single z", () => {
+  expect(parseZFilterInput("1")).toEqual([1])
+})
+
+test("parseZFilterInput parses multiple z values", () => {
+  expect(parseZFilterInput("1, 2,3")).toEqual([1, 2, 3])
+})
+
+test("parseZFilterInput normalizes duplicates and ordering", () => {
+  expect(parseZFilterInput("3,1,3,2")).toEqual([1, 2, 3])
+})
+
+test("parseZFilterInput returns null for empty input", () => {
+  expect(parseZFilterInput("  ")).toBeNull()
+})
+
+test("parseZFilterInput returns null for invalid input", () => {
+  expect(parseZFilterInput("1,a")).toBeNull()
+})
+
+test("matchesExactZFilter only matches exact single-layer selections", () => {
+  expect(matchesExactZFilter([1], [1])).toBe(true)
+  expect(matchesExactZFilter([1, 2], [1])).toBe(false)
+})
+
+test("matchesExactZFilter only matches exact shared-layer selections", () => {
+  expect(matchesExactZFilter([1, 2, 3], [1, 2, 3])).toBe(true)
+  expect(matchesExactZFilter([1, 2], [1, 2, 3])).toBe(false)
+  expect(matchesExactZFilter([1, 2, 3, 4], [1, 2, 3])).toBe(false)
+})