janikandClaude Fable 5
6d8634802d
Build PCM package / build (push) Successful in 7s
tests / ubuntu-latest · py3.11 (push) Successful in 53s
tests / ubuntu-latest · py3.13 (push) Successful in 1m43s
tests / archlinux:latest (push) Successful in 39s
tests / debian:12 (push) Successful in 49s
tests / fedora:latest (push) Successful in 1m1s
tests / ubuntu:24.04 (push) Successful in 1m26s
tests / NixOS (FHS wrapper from docs/NIXOS.md) (push) Has been skipped
New EXPERIMENTAL dialog option (default off): after the solve, copper whose |J| is below a threshold (default 10% of the mean |J| over all solved copper cells - mean, not max, since contact-corner spikes would dwarf a max-relative threshold) is vectorized into filled graphic polygons on TRIM_LAYERS (User.5..User.8, configurable), one polygon per region so Edit > Convert can turn one into a rule area by hand. Areas are printed and the polygons saved to low_current_copper.json. The mask -> polygon step is the 0.5 contour of the binary field via contourpy (already in every venv as matplotlib dependency), padded so regions touching the raster edge close, simplified with Douglas-Peucker at 0.4 cells: staircase bevels collapse, one-cell-wide strips survive. Specks under TRIM_MIN_AREA_MM2 are dropped. Explicitly a suggestion, not a safe cut list (docstring, dialog and README all say so): copper carries little current BECAUSE the rest carries it, so removal redistributes |J| - the constant-density optimizer that iterates this to convergence is future work. board_io: the create/delete-with-status-surfaced helpers are now generic (_create_items_checked / _remove_items_checked) and shared between reference-image overlays and trim polygons. Tested on the dev stack (3.13) and the Python 3.9 mac-stack venv, 148 passed each; contourpy 1.3.x has identical API on both. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
81 lines
3.3 KiB
Python
81 lines
3.3 KiB
Python
"""Geometry-in -> results-out pipeline shared by the KiCad entrypoint and
|
|
the offline standalone runner."""
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from . import config, plots, progress, raster, report, solver, trim
|
|
from .errors import UserFacingError
|
|
from .geometry import Problem
|
|
from .solver import Result
|
|
|
|
|
|
def run(problem: Problem, outdir: Path | None, show: bool = True,
|
|
i_test: float | None = None, freq_hz: float = 0.0,
|
|
contact_model: str | None = None, overlay=None,
|
|
trim_pct: float | None = None, trim_push=None) -> Result:
|
|
"""overlay: optional callback(stack, result) run after the solve
|
|
(EXPERIMENTAL in-KiCad overlays); its failures are non-fatal.
|
|
trim_pct: mark copper below this % of the mean |J| (None = off):
|
|
per-layer areas are printed, polygons saved to
|
|
<outdir>/low_current_copper.json and handed to trim_push, an
|
|
optional callback(trim_result) that pushes them into the board
|
|
(failures non-fatal)."""
|
|
if i_test is None:
|
|
i_test = config.TEST_CURRENT_A
|
|
if i_test <= 0:
|
|
raise UserFacingError(f"Test current must be > 0 A (got {i_test:g}).")
|
|
h = raster.choose_cell_size(problem.copper_bbox(), len(problem.layers))
|
|
progress.stage(f"rasterizing {len(problem.layers)} layer(s) at cell "
|
|
f"size {h / 1000:.1f} um ...")
|
|
stack = raster.rasterize_stack(problem, h)
|
|
print(f"grid {stack.shape2d[1]}x{stack.shape2d[0]}x{stack.nlayers}, "
|
|
f"{int(stack.masks.sum())} copper cells, {len(problem.vias)} "
|
|
f"via/pad barrel(s)")
|
|
|
|
e1, e2 = raster.electrode_masks(stack, problem)
|
|
parts1, parts2 = raster.electrode_partition(stack, problem)
|
|
|
|
progress.stage(f"solving @ {i_test:g} A"
|
|
+ (f", {freq_hz:g} Hz" if freq_hz > 0 else " DC") + " ...")
|
|
result = solver.run_solve(problem, stack, e1, e2, i_test, freq_hz,
|
|
contact_model, parts1, parts2)
|
|
for prefix, pcs in (("P", result.part_currents1),
|
|
("N", result.part_currents2)):
|
|
for i, (label, amps) in enumerate(pcs):
|
|
print(f" {prefix}{i + 1} ({label}): {amps:.4g} A "
|
|
f"({100 * amps / i_test:.1f}%)")
|
|
|
|
if outdir is not None:
|
|
outdir.mkdir(parents=True, exist_ok=True)
|
|
report.write_summary(outdir, problem, stack, result)
|
|
print(report.result_line(result, problem, stack))
|
|
|
|
if overlay is not None:
|
|
try:
|
|
overlay(stack, result)
|
|
except Exception as e:
|
|
print(f"overlay push failed: {e}")
|
|
|
|
if trim_pct is not None:
|
|
tr = trim.compute(result, stack, trim_pct)
|
|
print(trim.summary_line(tr))
|
|
if outdir is not None:
|
|
trim.write_json(outdir, tr)
|
|
if trim_push is not None:
|
|
try:
|
|
trim_push(tr)
|
|
except Exception as e:
|
|
print(f"trim push failed: {e}")
|
|
|
|
progress.stage("rendering figures ...")
|
|
figs = [
|
|
(plots.fig_raster(stack, e1, e2, problem, result), "1_raster_map"),
|
|
(plots.fig_potential(result, stack, e1, e2, problem), "2_potential"),
|
|
(plots.fig_current(result, stack, e1, e2, problem),
|
|
"3_current_density"),
|
|
(plots.fig_power(result, stack, e1, e2, problem), "4_power_density"),
|
|
]
|
|
plots.save_and_show(figs, outdir, show=show) # closes the window itself
|
|
return result
|