Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 27 additions & 1 deletion lib/RectDiffPipeline.ts
Original file line number Diff line number Diff line change
Expand Up @@ -4,12 +4,13 @@ import {
type PipelineStep,
} from "@tscircuit/solver-utils"
import type { SimpleRouteJson } from "./types/srj-types"
import type { GridFill3DOptions } from "./rectdiff-types"
import type { GridFill3DOptions, XYRect } from "./rectdiff-types"
import type { CapacityMeshNode } from "./types/capacity-mesh-types"
import type { GraphicsObject } from "graphics-debug"
import { GapFillSolverPipeline } from "./solvers/GapFillSolver/GapFillSolverPipeline"
import { RectDiffGridSolverPipeline } from "./solvers/RectDiffGridSolverPipeline/RectDiffGridSolverPipeline"
import { createBaseVisualization } from "./rectdiff-visualization"
import { computeInverseRects } from "./solvers/RectDiffSeedingSolver/computeInverseRects"

export interface RectDiffPipelineInput {
simpleRouteJson: SimpleRouteJson
Expand All @@ -19,6 +20,7 @@ export interface RectDiffPipelineInput {
export class RectDiffPipeline extends BasePipelineSolver<RectDiffPipelineInput> {
rectDiffGridSolverPipeline?: RectDiffGridSolverPipeline
gapFillSolver?: GapFillSolverPipeline
boardVoidRects: XYRect[] | undefined

override pipelineDef: PipelineStep<any>[] = [
definePipelineStep(
Expand All @@ -28,6 +30,7 @@ export class RectDiffPipeline extends BasePipelineSolver<RectDiffPipelineInput>
{
simpleRouteJson: rectDiffPipeline.inputProblem.simpleRouteJson,
gridOptions: rectDiffPipeline.inputProblem.gridOptions,
boardVoidRects: rectDiffPipeline.boardVoidRects,
},
],
),
Expand All @@ -39,11 +42,34 @@ export class RectDiffPipeline extends BasePipelineSolver<RectDiffPipelineInput>
meshNodes:
rectDiffPipeline.rectDiffGridSolverPipeline?.getOutput()
.meshNodes ?? [],
boardVoid: {
boardVoidRects: rectDiffPipeline.boardVoidRects || [],
layerCount:
rectDiffPipeline.inputProblem.simpleRouteJson.layerCount || 0,
},
},
],
),
]

override _setup(): void {
if (this.inputProblem.simpleRouteJson.outline) {
this.boardVoidRects = computeInverseRects(
{
x: this.inputProblem.simpleRouteJson.bounds.minX,
y: this.inputProblem.simpleRouteJson.bounds.minY,
width:
this.inputProblem.simpleRouteJson.bounds.maxX -
this.inputProblem.simpleRouteJson.bounds.minX,
height:
this.inputProblem.simpleRouteJson.bounds.maxY -
this.inputProblem.simpleRouteJson.bounds.minY,
},
this.inputProblem.simpleRouteJson.outline ?? [],
)
}
}

override getConstructorParams() {
return [this.inputProblem]
}
Expand Down
38 changes: 32 additions & 6 deletions lib/solvers/GapFillSolver/ExpandEdgesToEmptySpaceSolver.ts
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,19 @@ 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"

const EPS = 1e-4

export type ExpandEdgesToEmptySpaceSolverInput = {
inputMeshNodes: CapacityMeshNode[]
segmentsWithAdjacentEmptySpace: Array<SegmentWithAdjacentEmptySpace>
boardVoid?: {
boardVoidRects: XYRect[]
layerCount: number
}
}

export interface ExpandedSegment {
segment: SegmentWithAdjacentEmptySpace
newNode: CapacityMeshNode
Expand All @@ -28,15 +38,31 @@ export class ExpandEdgesToEmptySpaceSolver extends BaseSolver {

rectSpatialIndex: RBush<CapacityMeshNode>

constructor(
private input: {
inputMeshNodes: CapacityMeshNode[]
segmentsWithAdjacentEmptySpace: Array<SegmentWithAdjacentEmptySpace>
},
) {
constructor(private input: ExpandEdgesToEmptySpaceSolverInput) {
super()
this.unprocessedSegments = [...this.input.segmentsWithAdjacentEmptySpace]
this.rectSpatialIndex = new RBush<CapacityMeshNode>()
// create fake bound for the boardVoidRects
this.rectSpatialIndex.load(
this.input.boardVoid?.boardVoidRects.map((rect, index) => ({
capacityMeshNodeId: `void-rect-${index}`,
center: {
x: rect.x + rect.width / 2,
y: rect.y + rect.height / 2,
},
width: rect.width,
height: rect.height,
availableZ: Array.from(
{ length: this.input.boardVoid?.layerCount || 0 },
(_, i) => i,
),
layer: "void",
minX: rect.x,
minY: rect.y,
maxX: rect.x + rect.width,
maxY: rect.y + rect.height,
})) || [],
)
this.rectSpatialIndex.load(
this.input.inputMeshNodes.map((n) => ({
...n,
Expand Down
13 changes: 10 additions & 3 deletions lib/solvers/GapFillSolver/GapFillSolverPipeline.ts
Original file line number Diff line number Diff line change
Expand Up @@ -3,15 +3,21 @@ import {
definePipelineStep,
type PipelineStep,
} from "@tscircuit/solver-utils"
import type { SimpleRouteJson } from "lib/types/srj-types"
import type { CapacityMeshNode } from "lib/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"

export class GapFillSolverPipeline extends BasePipelineSolver<{
type GapFillSolverInput = {
meshNodes: CapacityMeshNode[]
}> {
boardVoid?: {
boardVoidRects: XYRect[]
layerCount: number
}
}

export class GapFillSolverPipeline extends BasePipelineSolver<GapFillSolverInput> {
findSegmentsWithAdjacentEmptySpaceSolver?: FindSegmentsWithAdjacentEmptySpaceSolver
expandEdgesToEmptySpaceSolver?: ExpandEdgesToEmptySpaceSolver

Expand Down Expand Up @@ -39,6 +45,7 @@ export class GapFillSolverPipeline extends BasePipelineSolver<{
segmentsWithAdjacentEmptySpace:
gapFillPipeline.findSegmentsWithAdjacentEmptySpaceSolver!.getOutput()
.segmentsWithAdjacentEmptySpace,
boardVoid: gapFillPipeline.inputProblem.boardVoid,
},
],
{
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -10,26 +10,26 @@ import { RectDiffSeedingSolver } from "lib/solvers/RectDiffSeedingSolver/RectDif
import { RectDiffExpansionSolver } from "lib/solvers/RectDiffExpansionSolver/RectDiffExpansionSolver"
import type { GraphicsObject } from "graphics-debug"
import RBush from "rbush"
import { buildObstacleIndexes } from "./buildObstacleIndexes"
import { buildObstacleIndexesByLayer } from "./buildObstacleIndexes"

export type RectDiffGridSolverPipelineInput = {
simpleRouteJson: SimpleRouteJson
gridOptions?: Partial<GridFill3DOptions>
boardVoidRects?: XYRect[]
}

export class RectDiffGridSolverPipeline extends BasePipelineSolver<RectDiffGridSolverPipelineInput> {
rectDiffSeedingSolver?: RectDiffSeedingSolver
rectDiffExpansionSolver?: RectDiffExpansionSolver
private boardVoidRects?: XYRect[]
private obstacleIndexByLayer: Array<RBush<RTreeRect>>

constructor(inputProblem: RectDiffGridSolverPipelineInput) {
super(inputProblem)
const { obstacleIndexByLayer, boardVoidRects } = buildObstacleIndexes(
inputProblem.simpleRouteJson,
)
const { obstacleIndexByLayer } = buildObstacleIndexesByLayer({
srj: inputProblem.simpleRouteJson,
boardVoidRects: inputProblem.boardVoidRects,
})
this.obstacleIndexByLayer = obstacleIndexByLayer
this.boardVoidRects = boardVoidRects
}

override pipelineDef: PipelineStep<any>[] = [
Expand All @@ -41,7 +41,7 @@ export class RectDiffGridSolverPipeline extends BasePipelineSolver<RectDiffGridS
simpleRouteJson: pipeline.inputProblem.simpleRouteJson,
gridOptions: pipeline.inputProblem.gridOptions,
obstacleIndexByLayer: pipeline.obstacleIndexByLayer,
boardVoidRects: pipeline.boardVoidRects,
boardVoidRects: pipeline.inputProblem.boardVoidRects,
},
],
),
Expand All @@ -52,7 +52,7 @@ export class RectDiffGridSolverPipeline extends BasePipelineSolver<RectDiffGridS
{
initialSnapshot: {
...pipeline.rectDiffSeedingSolver!.getOutput(),
boardVoidRects: pipeline.boardVoidRects ?? [],
boardVoidRects: pipeline.inputProblem.boardVoidRects ?? [],
},
obstacleIndexByLayer: pipeline.obstacleIndexByLayer,
},
Expand Down
15 changes: 7 additions & 8 deletions lib/solvers/RectDiffGridSolverPipeline/buildObstacleIndexes.ts
Original file line number Diff line number Diff line change
Expand Up @@ -9,12 +9,13 @@ import {
import type { XYRect } from "lib/rectdiff-types"
import type { RTreeRect } from "lib/types/capacity-mesh-types"

export const buildObstacleIndexes = (
srj: SimpleRouteJson,
): {
export const buildObstacleIndexesByLayer = (params: {
srj: SimpleRouteJson
boardVoidRects?: XYRect[]
}): {
obstacleIndexByLayer: Array<RBush<RTreeRect>>
boardVoidRects: XYRect[]
} => {
const { srj, boardVoidRects } = params
const { layerNames, zIndexByName } = buildZIndexMap(srj)
const layerCount = Math.max(1, layerNames.length, srj.layerCount || 1)
const bounds: XYRect = {
Expand All @@ -39,10 +40,8 @@ export const buildObstacleIndexes = (
obstacleIndexByLayer[z]?.insert(treeRect)
}

let boardVoidRects: XYRect[] = []
if (srj.outline && srj.outline.length > 2) {
boardVoidRects = computeInverseRects(bounds, srj.outline as any)
for (const voidRect of boardVoidRects) {
for (const voidRect of boardVoidRects ?? []) {
for (let z = 0; z < layerCount; z++) insertObstacle(voidRect, z)
}
}
Expand All @@ -66,5 +65,5 @@ export const buildObstacleIndexes = (
for (const z of zLayers) insertObstacle(rect, z)
}

return { obstacleIndexByLayer, boardVoidRects }
return { obstacleIndexByLayer }
}
17 changes: 1 addition & 16 deletions tests/__snapshots__/board-outline.snap.svg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading