Compare commits
210 commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
736107f7a4 | ||
|
|
e3674de08d | ||
|
|
516a9d337f | ||
|
|
2451b517e8 | ||
|
|
bdd4008ab9 | ||
|
|
6f006e2782 | ||
|
|
8df709f55f | ||
|
|
985e263cfe | ||
|
|
5ccfd7a259 | ||
|
|
bcc4aeefa7 | ||
|
|
0c15111463 | ||
|
|
1a0f519720 | ||
|
|
3d0ddc3dc8 | ||
|
|
575046a60c | ||
|
|
8de776616c | ||
|
|
e19dec20b6 | ||
|
|
97513df5d9 | ||
|
|
2ce55ebdca | ||
|
|
f3c95a42d4 | ||
|
|
8a2599f5f4 | ||
|
|
2647709215 | ||
|
|
87413855bf | ||
|
|
af4fb2668f | ||
|
|
a47a694ba0 | ||
|
|
dd8ad98f13 | ||
|
|
a1ea416269 | ||
|
|
a7f0324506 | ||
|
|
6ab4640099 | ||
|
|
04a088c3e8 | ||
|
|
df4895abfa | ||
|
|
42dfd1be7f | ||
|
|
fd6880640d | ||
|
|
6a8bd8dc3f | ||
|
|
46df184358 | ||
|
|
c2679260a0 | ||
|
|
9d2d635eee | ||
|
|
75905f7d0c | ||
|
|
7073b6e33f | ||
|
|
718de2b338 | ||
|
|
379cf273cb | ||
|
|
da6d8349fa | ||
|
|
1f5a9261e1 | ||
|
|
3ae6b9be61 | ||
|
|
a6434f16cd | ||
|
|
63faf4280d | ||
|
|
7fb8215a1e | ||
|
|
81ae51d4be | ||
|
|
8ffa7c1b76 | ||
|
|
be8371c7bc | ||
|
|
1a854b1812 | ||
|
|
091ee84910 | ||
|
|
fd63c44314 | ||
|
|
cbe8dfa252 | ||
|
|
67ce4af957 | ||
|
|
eadd250ee3 | ||
|
|
5f5bbccd05 | ||
|
|
ea2664219d | ||
|
|
be25b860a9 | ||
|
|
e42b7462c9 | ||
|
|
754c9557e5 | ||
|
|
d7efa57732 | ||
|
|
689ce748db | ||
|
|
8e337c0506 | ||
|
|
b23cafbfd4 | ||
|
|
78ffb61aee | ||
|
|
58d5784903 | ||
|
|
1ecb7be6f9 | ||
|
|
ef9d61ffd5 | ||
|
|
79f555a465 | ||
|
|
501882ea3d | ||
|
|
e3a6716187 | ||
|
|
8950a593f8 | ||
|
|
344825c5da | ||
|
|
f1b3ab5e72 | ||
|
|
2092b86431 | ||
|
|
885ce36fd3 | ||
|
|
edaf246b9d | ||
|
|
b1d0260c70 | ||
|
|
a2c6e2d64a | ||
|
|
1d5f4c8193 | ||
|
|
30878adfb1 | ||
|
|
66811af966 | ||
|
|
ec85d6c169 | ||
|
|
f447b12571 | ||
|
|
93fd764482 | ||
|
|
26e85279dc | ||
|
|
1ed127e3b3 | ||
|
|
1ee6b6587a | ||
|
|
e98f3f3ace | ||
|
|
6f8d4bb999 | ||
|
|
4aab344a18 | ||
|
|
21218239e4 | ||
|
|
a1d6ebf79f | ||
|
|
b0274a93c0 | ||
|
|
0150c318bb | ||
|
|
6de138bf7c | ||
|
|
224a666219 | ||
|
|
cee355ff57 | ||
|
|
4c3815b25a | ||
|
|
ef3b5d5e1c | ||
|
|
552f30c15d | ||
|
|
f721692bf3 | ||
|
|
04c4b3ff0c | ||
|
|
227d4ed1cd | ||
|
|
ea4c28e307 | ||
|
|
51ef4882a1 | ||
|
|
df75a2fddb | ||
|
|
11325b213b | ||
|
|
74fb384c4c | ||
|
|
9af0713445 | ||
|
|
09e3731b74 | ||
|
|
37b6b8f8d2 | ||
|
|
07362c592f | ||
|
|
2f5f7719c6 | ||
|
|
cb1d3eb3fb | ||
|
|
4ee5c51f22 | ||
|
|
53788354e8 | ||
|
|
10962ae2f4 | ||
|
|
a19d307a7d | ||
|
|
165e101dda | ||
|
|
187c44555c | ||
|
|
36da1fd68b | ||
|
|
9624e46147 | ||
|
|
a35125b123 | ||
|
|
31af2b260c | ||
|
|
dd49698df9 | ||
|
|
ef2b53325c | ||
|
|
5d3cd4694d | ||
|
|
3631871a40 | ||
|
|
b710462419 | ||
|
|
313aa7dd26 | ||
|
|
1ce02a9d25 | ||
|
|
0ae13de322 | ||
|
|
c212663bb2 | ||
|
|
2a9c91b025 | ||
|
|
2d4c40c0f7 | ||
|
|
a11f144c67 | ||
|
|
e78e939a13 | ||
|
|
4ea1b26293 | ||
|
|
cc59a6567e | ||
|
|
a2d1429036 | ||
|
|
f793f12edb | ||
|
|
78f5bf965f | ||
|
|
84f7e5d25b | ||
|
|
ba689e632e | ||
|
|
4400dc361e | ||
|
|
fdaf3c7e93 | ||
|
|
6e12adb07e | ||
|
|
cff22b9e08 | ||
|
|
f711c1d91c | ||
|
|
61e591b5b8 | ||
|
|
95da482033 | ||
|
|
d2143bdf4d | ||
|
|
5f1350d4f4 | ||
|
|
5ff40e0ad1 | ||
|
|
7d21dafd6a | ||
|
|
301601e81d | ||
|
|
3e47e7c2da | ||
|
|
ba4cafa3a4 | ||
|
|
2f0a21abf9 | ||
|
|
eb5c01ddd0 | ||
|
|
6e7337cca5 | ||
|
|
eb20595e00 | ||
|
|
e4dcbe177f | ||
|
|
8cb91dabdb | ||
|
|
91b99a0452 | ||
|
|
09c9d26728 | ||
|
|
21ca5f7f5c | ||
|
|
a39af853c8 | ||
|
|
b69e9fded4 | ||
|
|
60674ab5b3 | ||
|
|
bdbdf7f586 | ||
|
|
a1b8cbf861 | ||
|
|
58142cb0c7 | ||
|
|
08c4091e57 | ||
|
|
860fa4c53b | ||
|
|
b2729a46ac | ||
|
|
572486aa25 | ||
|
|
0799cc55ad | ||
|
|
9f2f1f97f9 | ||
|
|
cb188ac593 | ||
|
|
a5087636ab | ||
|
|
fb1d392831 | ||
|
|
8f7b2893dc | ||
|
|
e696c09eac | ||
|
|
3bc92876b4 | ||
|
|
34fae0a7c2 | ||
|
|
2eb88e8127 | ||
|
|
95728fb33c | ||
|
|
0920af4149 | ||
|
|
7a95a0dde7 | ||
|
|
88642a1803 | ||
|
|
56d55fda5d | ||
|
|
467e482bf4 | ||
|
|
dd8507d202 | ||
|
|
d43308c5cc | ||
|
|
c29802c2b9 | ||
|
|
a92177904e | ||
|
|
0148db6249 | ||
|
|
5178eba26f | ||
|
|
ddd0641649 | ||
|
|
e349c2c08b | ||
|
|
96fde32c0b | ||
|
|
86276490eb | ||
|
|
070494a1c3 | ||
|
|
1d0ba4da70 | ||
|
|
3561817903 | ||
|
|
03f2ec0a30 | ||
|
|
732c58f70b | ||
|
|
4bac77d0b4 |
2
.gitignore
vendored
|
|
@ -3,3 +3,5 @@ gerbonara_test_failures
|
||||||
__pycache__
|
__pycache__
|
||||||
.tox
|
.tox
|
||||||
docs/_build/
|
docs/_build/
|
||||||
|
build
|
||||||
|
dist
|
||||||
|
|
|
||||||
|
|
@ -14,53 +14,54 @@ build:archlinux:
|
||||||
GIT_SUBMODULE_STRATEGY: none
|
GIT_SUBMODULE_STRATEGY: none
|
||||||
script:
|
script:
|
||||||
- git config --global --add safe.directory "$CI_PROJECT_DIR"
|
- git config --global --add safe.directory "$CI_PROJECT_DIR"
|
||||||
- pip3 install --user wheel setuptools
|
- uv build
|
||||||
- python3 setup.py sdist bdist_wheel
|
|
||||||
artifacts:
|
artifacts:
|
||||||
name: "gerbolyze-$CI_COMMIT_REF_NAME-gerbonara"
|
name: "gerbolyze-$CI_COMMIT_REF_NAME-gerbonara"
|
||||||
paths:
|
paths:
|
||||||
- dist/*
|
- dist/*
|
||||||
|
|
||||||
test:archlinux:
|
# FIXME: disable tests since (a) currenty kicad-cli is broken (aborts on start), and the workaround of using an older
|
||||||
stage: test
|
# version from the KiCad project's kicad-cli containers does not work in gitlab CI. Pain.
|
||||||
image: "registry.gitlab.com/gerbolyze/build-containers/archlinux:latest"
|
#test:archlinux:
|
||||||
script:
|
# stage: test
|
||||||
- git clone --depth 1 https://gitlab.com/kicad/libraries/kicad-symbols
|
# image: "registry.gitlab.com/gerbolyze/build-containers/archlinux:latest"
|
||||||
- git clone --depth 1 https://gitlab.com/kicad/libraries/kicad-footprints
|
# script:
|
||||||
- env KICAD_SYMBOLS=kicad-symbols KICAD_FOOTPRINTS=kicad-footprints pytest -o 'testpaths=gerbonara/tests' -o 'norecursedirs=*'
|
# - git clone --depth 1 https://gitlab.com/kicad/libraries/kicad-symbols
|
||||||
dependencies:
|
# - git clone --depth 1 https://gitlab.com/kicad/libraries/kicad-footprints
|
||||||
- build:archlinux
|
# - env KICAD_SYMBOLS=kicad-symbols KICAD_FOOTPRINTS=kicad-footprints pytest -o 'testpaths=gerbonara/tests' -o 'norecursedirs=*'
|
||||||
cache:
|
# dependencies:
|
||||||
key: test-image-cache
|
# - build:archlinux
|
||||||
paths:
|
# cache:
|
||||||
- gerbonara/tests/image_cache/*.svg
|
# key: test-image-cache
|
||||||
- gerbonara/tests/image_cache/*.png
|
# paths:
|
||||||
artifacts:
|
# - gerbonara/tests/image_cache/*.svg
|
||||||
name: "gerbolyze-$CI_COMMIT_REF_NAME-gerbonara"
|
# - gerbonara/tests/image_cache/*.png
|
||||||
when: on_failure
|
# artifacts:
|
||||||
paths:
|
# name: "gerbolyze-$CI_COMMIT_REF_NAME-gerbonara"
|
||||||
- gerbonara_test_failures/*
|
# when: on_failure
|
||||||
|
# paths:
|
||||||
test:ubuntu2204:
|
# - gerbonara_test_failures/*
|
||||||
stage: test
|
#
|
||||||
image: "registry.gitlab.com/gerbolyze/build-containers/ubuntu:22.04"
|
#test:ubuntu-rolling:
|
||||||
script:
|
# stage: test
|
||||||
- python3 -m pip install pytest beautifulsoup4 pillow numpy slugify lxml click scipy
|
# image: "registry.gitlab.com/gerbolyze/build-containers/ubuntu:rolling"
|
||||||
- git clone --depth 1 https://gitlab.com/kicad/libraries/kicad-symbols
|
# script:
|
||||||
- git clone --depth 1 https://gitlab.com/kicad/libraries/kicad-footprints
|
# - python3 -m pip install --break-system-packages pytest beautifulsoup4 pillow numpy slugify lxml click scipy
|
||||||
- env KICAD_SYMBOLS=kicad-symbols KICAD_FOOTPRINTS=kicad-footprints python3 -m pytest -o 'testpaths=gerbonara/tests' -o 'norecursedirs=*'
|
# - git clone --depth 1 https://gitlab.com/kicad/libraries/kicad-symbols
|
||||||
dependencies:
|
# - git clone --depth 1 https://gitlab.com/kicad/libraries/kicad-footprints
|
||||||
- build:archlinux
|
# - env KICAD_SYMBOLS=kicad-symbols KICAD_FOOTPRINTS=kicad-footprints python3 -m pytest -o 'testpaths=gerbonara/tests' -o 'norecursedirs=*'
|
||||||
cache:
|
# dependencies:
|
||||||
key: test-image-cache
|
# - build:archlinux
|
||||||
paths:
|
# cache:
|
||||||
- gerbonara/tests/image_cache/*.svg
|
# key: test-image-cache
|
||||||
- gerbonara/tests/image_cache/*.png
|
# paths:
|
||||||
artifacts:
|
# - gerbonara/tests/image_cache/*.svg
|
||||||
name: "gerbolyze-$CI_COMMIT_REF_NAME-gerbonara"
|
# - gerbonara/tests/image_cache/*.png
|
||||||
when: on_failure
|
# artifacts:
|
||||||
paths:
|
# name: "gerbolyze-$CI_COMMIT_REF_NAME-gerbonara"
|
||||||
- gerbonara_test_failures/*
|
# when: on_failure
|
||||||
|
# paths:
|
||||||
|
# - gerbonara_test_failures/*
|
||||||
|
|
||||||
docs:archlinux:
|
docs:archlinux:
|
||||||
stage: test
|
stage: test
|
||||||
|
|
@ -83,7 +84,7 @@ publish:gerbonara:
|
||||||
cache: {}
|
cache: {}
|
||||||
script:
|
script:
|
||||||
- export TWINE_USERNAME TWINE_PASSWORD
|
- export TWINE_USERNAME TWINE_PASSWORD
|
||||||
- pip3 install --user twine rich
|
- pip3 install --user --break-system-packages twine rich
|
||||||
- twine upload dist/*
|
- twine upload dist/*
|
||||||
dependencies:
|
dependencies:
|
||||||
- build:archlinux
|
- build:archlinux
|
||||||
|
|
|
||||||
|
|
@ -60,6 +60,7 @@ layers, or whole board stacks (:py:class:`~.layers.LayerStack`) to SVG.
|
||||||
|
|
||||||
``gerbonara render`` renders one or more Gerber or Excellon files as a single SVG file. It can read single files,
|
``gerbonara render`` renders one or more Gerber or Excellon files as a single SVG file. It can read single files,
|
||||||
directorys of files, and ZIP files. To read directories or zips, it applies gerbonara's layer filename matching rules.
|
directorys of files, and ZIP files. To read directories or zips, it applies gerbonara's layer filename matching rules.
|
||||||
|
These built-in rules should work with common settings in a wide variety of CAD tools.
|
||||||
|
|
||||||
.. option:: --warnings [default|ignore|once]
|
.. option:: --warnings [default|ignore|once]
|
||||||
|
|
||||||
|
|
|
||||||
BIN
docs/ex-mask-islands.png
Normal file
|
After Width: | Height: | Size: 243 KiB |
64
docs/examples.rst
Normal file
|
|
@ -0,0 +1,64 @@
|
||||||
|
.. _examples-doc:
|
||||||
|
|
||||||
|
Examples
|
||||||
|
========
|
||||||
|
|
||||||
|
Solder mask rings
|
||||||
|
-----------------
|
||||||
|
|
||||||
|
This example script takes a board exported with a more recent KiCad version, and removes solder mask everywhere, but
|
||||||
|
leaves a thin ring of solder mask around every pad. Might be useful for some artsy boards.
|
||||||
|
|
||||||
|
.. image:: ex-mask-islands.png
|
||||||
|
|
||||||
|
.. code-block:: python
|
||||||
|
from gerbonara import *
|
||||||
|
from shapely import *
|
||||||
|
|
||||||
|
stack = layers.LayerStack.open('gerber')
|
||||||
|
# Let's work in mm here. Gerbonara will take care to convert units when the file is in US customary units.
|
||||||
|
(x1, y1), (x2, y2) = stack.bounding_box(unit=utils.MM)
|
||||||
|
|
||||||
|
for l in [stack['bottom mask'], stack['top mask']]:
|
||||||
|
|
||||||
|
# The solder mask gerber layer by convention is "negative". That is, a "dark" polarity (drawn) Gerber primitive
|
||||||
|
# will result in an opening in the solder mask. Conversely, an empty gerber file would lead to the entire board
|
||||||
|
# being covered in solder mask.
|
||||||
|
#
|
||||||
|
# Here, we add a rectangle covering the entire board so the entire board is *free* of solder mask.
|
||||||
|
|
||||||
|
new = [graphic_objects.Region(
|
||||||
|
[(x1, y1), (x1, y2), (x2, y2), (x2, y1), (x1, y1)],
|
||||||
|
unit=utils.MM,
|
||||||
|
polarity_dark=True)]
|
||||||
|
|
||||||
|
# Iterate through all objects on the solder mask layer. In later KiCad versions, everything on the solder mask
|
||||||
|
# layer is exported as a Gerber region, which is a really bad idea, but makes things easy for us here.
|
||||||
|
for obj in l.objects:
|
||||||
|
if isinstance(obj, gerbonara.graphic_objects.Region):
|
||||||
|
regions = []
|
||||||
|
else:
|
||||||
|
regions = [gerbonara.graphic_objects.Region.from_arc_poly(prim.to_arc_poly())
|
||||||
|
for prim in obj.to_primitives(unit=gerbonara.utils.MM)]
|
||||||
|
|
||||||
|
for obj in regions:
|
||||||
|
# Convert the region to a shapely line string
|
||||||
|
ls = LineString(obj.outline).normalize()
|
||||||
|
|
||||||
|
# Ask shapely to offset the line string by 1 mm
|
||||||
|
out = ls.offset_curve(obj.unit(1, 'mm'))
|
||||||
|
|
||||||
|
# For negative offsets, this operation can result in an object being split up into multiple parts, so we
|
||||||
|
# might get back a MultiLineString instead of a LineString.
|
||||||
|
for ls in (out.geoms if hasattr(out, 'geoms') else [out]):
|
||||||
|
|
||||||
|
# Convert the resulting shapely object back to a Gerber region.
|
||||||
|
new.append(graphic_objects.Region(
|
||||||
|
unit=obj.unit,
|
||||||
|
polarity_dark=not obj.polarity_dark,
|
||||||
|
outline=list(ls.coords)))
|
||||||
|
|
||||||
|
# Append the new objects to the original layer data
|
||||||
|
l.objects = new + l.objects
|
||||||
|
# Write the modified layer stack to a new Gerber directory
|
||||||
|
stack.save_to_directory('output-gerbers')
|
||||||
|
|
@ -48,6 +48,7 @@ Features
|
||||||
|
|
||||||
cli
|
cli
|
||||||
api-concepts
|
api-concepts
|
||||||
|
examples
|
||||||
file-api
|
file-api
|
||||||
object-api
|
object-api
|
||||||
apertures
|
apertures
|
||||||
|
|
@ -70,10 +71,12 @@ Then, you are ready to read and write gerber files:
|
||||||
|
|
||||||
from gerbonara import LayerStack
|
from gerbonara import LayerStack
|
||||||
|
|
||||||
stack = LayerStack.from_directory('output/gerber')
|
stack = LayerStack.open('output/gerber')
|
||||||
w, h = stack.outline.size('mm')
|
w, h = stack.outline.size('mm')
|
||||||
print(f'Board size is {w:.1f} mm x {h:.1f} mm')
|
print(f'Board size is {w:.1f} mm x {h:.1f} mm')
|
||||||
|
|
||||||
|
You can find some more elaborate examples in this doc's :ref:`Examples section<examples-doc>`.
|
||||||
|
|
||||||
Command-Line Interface
|
Command-Line Interface
|
||||||
======================
|
======================
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -10,7 +10,7 @@ from gerbonara.utils import MM
|
||||||
from gerbonara.utils import rotate_point
|
from gerbonara.utils import rotate_point
|
||||||
|
|
||||||
def highlight_outline(input_dir, output_dir):
|
def highlight_outline(input_dir, output_dir):
|
||||||
stack = LayerStack.from_directory(input_dir)
|
stack = LayerStack.open(input_dir)
|
||||||
|
|
||||||
outline = []
|
outline = []
|
||||||
for obj in stack.outline.objects:
|
for obj in stack.outline.objects:
|
||||||
|
|
@ -28,7 +28,6 @@ def highlight_outline(input_dir, output_dir):
|
||||||
marker_nx, marker_ny = math.sin(marker_angle), math.cos(marker_angle)
|
marker_nx, marker_ny = math.sin(marker_angle), math.cos(marker_angle)
|
||||||
|
|
||||||
ap = CircleAperture(0.1, unit=MM)
|
ap = CircleAperture(0.1, unit=MM)
|
||||||
stack['top silk'].apertures.append(ap)
|
|
||||||
|
|
||||||
for line in outline:
|
for line in outline:
|
||||||
cx, cy = (line.x1 + line.x2)/2, (line.y1 + line.y2)/2
|
cx, cy = (line.x1 + line.x2)/2, (line.y1 + line.y2)/2
|
||||||
|
|
|
||||||
|
|
@ -7,5 +7,5 @@ if __name__ == '__main__':
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
import gerbonara
|
import gerbonara
|
||||||
print(gerbonara.LayerStack.from_directory(args.input))
|
print(gerbonara.LayerStack.open(args.input))
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -2,6 +2,7 @@
|
||||||
|
|
||||||
import math
|
import math
|
||||||
|
|
||||||
|
from gerbonara.utils import MM
|
||||||
from gerbonara.graphic_objects import Arc
|
from gerbonara.graphic_objects import Arc
|
||||||
from gerbonara.graphic_objects import rotate_point
|
from gerbonara.graphic_objects import rotate_point
|
||||||
|
|
||||||
|
|
@ -22,7 +23,8 @@ def approx_test():
|
||||||
x1, y1 = rotate_point(0, -1, start_angle*eps)
|
x1, y1 = rotate_point(0, -1, start_angle*eps)
|
||||||
x2, y2 = rotate_point(x1, y1, sweep_angle*eps*(-1 if clockwise else 1))
|
x2, y2 = rotate_point(x1, y1, sweep_angle*eps*(-1 if clockwise else 1))
|
||||||
|
|
||||||
arc = Arc(x1+cx, y1+cy, x2+cx, y2+cy, -x1, -y1, clockwise=clockwise, aperture=None, polarity_dark=True)
|
arc = Arc(x1+cx, y1+cy, x2+cx, y2+cy, -x1, -y1, clockwise=clockwise, aperture=None,
|
||||||
|
polarity_dark=True, unit=MM)
|
||||||
lines = arc.approximate(max_error=max_error)
|
lines = arc.approximate(max_error=max_error)
|
||||||
|
|
||||||
print(f'<path style="fill: {color}; stroke: none;" d="M {cx} {cy} L {lines[0].x1} {lines[0].y1}', end=' ')
|
print(f'<path style="fill: {color}; stroke: none;" d="M {cx} {cy} L {lines[0].x1} {lines[0].y1}', end=' ')
|
||||||
|
|
|
||||||
|
|
@ -1,9 +0,0 @@
|
||||||
#!/usr/bin/env python3
|
|
||||||
|
|
||||||
import click
|
|
||||||
|
|
||||||
from .cli import cli
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
cli()
|
|
||||||
|
|
||||||
|
|
@ -1,231 +0,0 @@
|
||||||
#!/usr/bin/env python
|
|
||||||
# -*- coding: utf-8 -*-
|
|
||||||
|
|
||||||
# Copyright 2021 Jan Sebastian Götte <gerbonara@jaseg.de>
|
|
||||||
|
|
||||||
from dataclasses import dataclass, field, replace
|
|
||||||
import operator
|
|
||||||
import re
|
|
||||||
import ast
|
|
||||||
import copy
|
|
||||||
import math
|
|
||||||
|
|
||||||
from . import primitive as ap
|
|
||||||
from .expression import *
|
|
||||||
from ..utils import MM
|
|
||||||
|
|
||||||
# we make our own here instead of using math.degrees to make sure this works with expressions, too.
|
|
||||||
def rad_to_deg(x):
|
|
||||||
return (x / math.pi) * 180
|
|
||||||
|
|
||||||
def _map_expression(node):
|
|
||||||
if isinstance(node, ast.Num):
|
|
||||||
return ConstantExpression(node.n)
|
|
||||||
|
|
||||||
elif isinstance(node, ast.BinOp):
|
|
||||||
op_map = {ast.Add: operator.add, ast.Sub: operator.sub, ast.Mult: operator.mul, ast.Div: operator.truediv}
|
|
||||||
return OperatorExpression(op_map[type(node.op)], _map_expression(node.left), _map_expression(node.right))
|
|
||||||
|
|
||||||
elif isinstance(node, ast.UnaryOp):
|
|
||||||
if type(node.op) == ast.UAdd:
|
|
||||||
return _map_expression(node.operand)
|
|
||||||
else:
|
|
||||||
return OperatorExpression(operator.sub, ConstantExpression(0), _map_expression(node.operand))
|
|
||||||
|
|
||||||
elif isinstance(node, ast.Name):
|
|
||||||
return VariableExpression(int(node.id[3:])) # node.id has format var[0-9]+
|
|
||||||
|
|
||||||
else:
|
|
||||||
raise SyntaxError('Invalid aperture macro expression')
|
|
||||||
|
|
||||||
def _parse_expression(expr):
|
|
||||||
expr = expr.lower().replace('x', '*')
|
|
||||||
expr = re.sub(r'\$([0-9]+)', r'var\1', expr)
|
|
||||||
try:
|
|
||||||
parsed = ast.parse(expr, mode='eval').body
|
|
||||||
except SyntaxError as e:
|
|
||||||
raise SyntaxError('Invalid aperture macro expression') from e
|
|
||||||
return _map_expression(parsed)
|
|
||||||
|
|
||||||
@dataclass(frozen=True, slots=True)
|
|
||||||
class ApertureMacro:
|
|
||||||
name: str = field(default=None, hash=False, compare=False)
|
|
||||||
primitives: tuple = ()
|
|
||||||
variables: tuple = ()
|
|
||||||
comments: tuple = field(default=(), hash=False, compare=False)
|
|
||||||
|
|
||||||
def __post_init__(self):
|
|
||||||
if self.name is None or re.match(r'GNX[0-9A-F]{16}', self.name):
|
|
||||||
# We can't use field(default_factory=...) here because that factory doesn't get a reference to the instance.
|
|
||||||
self._reset_name()
|
|
||||||
|
|
||||||
def _reset_name(self):
|
|
||||||
object.__setattr__(self, 'name', f'GNX{hash(self)&0xffffffffffffffff:016X}')
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def parse_macro(kls, macro_name, body, unit):
|
|
||||||
comments = []
|
|
||||||
variables = {}
|
|
||||||
primitives = []
|
|
||||||
|
|
||||||
blocks = body.split('*')
|
|
||||||
for block in blocks:
|
|
||||||
if not (block := block.strip()): # empty block
|
|
||||||
continue
|
|
||||||
|
|
||||||
if block.startswith('0 '): # comment
|
|
||||||
comments.append(block[2:])
|
|
||||||
continue
|
|
||||||
|
|
||||||
block = re.sub(r'\s', '', block)
|
|
||||||
|
|
||||||
if block[0] == '$': # variable definition
|
|
||||||
name, expr = block.partition('=')
|
|
||||||
number = int(name[1:])
|
|
||||||
if number in variables:
|
|
||||||
raise SyntaxError(f'Re-definition of aperture macro variable {number} inside macro')
|
|
||||||
variables[number] = _parse_expression(expr)
|
|
||||||
|
|
||||||
else: # primitive
|
|
||||||
primitive, *args = block.split(',')
|
|
||||||
args = [ _parse_expression(arg) for arg in args ]
|
|
||||||
primitives.append(ap.PRIMITIVE_CLASSES[int(primitive)].from_arglist(unit, args))
|
|
||||||
|
|
||||||
variables = [variables.get(i+1) for i in range(max(variables.keys(), default=0))]
|
|
||||||
return kls(macro_name, tuple(primitives), tuple(variables), tuple(comments))
|
|
||||||
|
|
||||||
def __str__(self):
|
|
||||||
return f'<Aperture macro {self.name}, variables {str(self.variables)}, primitives {self.primitives}>'
|
|
||||||
|
|
||||||
def __repr__(self):
|
|
||||||
return str(self)
|
|
||||||
|
|
||||||
def dilated(self, offset, unit=MM):
|
|
||||||
new_primitives = []
|
|
||||||
for primitive in self.primitives:
|
|
||||||
try:
|
|
||||||
if primitive.exposure.calculate():
|
|
||||||
new_primitives += primitive.dilated(offset, unit)
|
|
||||||
except IndexError:
|
|
||||||
warnings.warn('Cannot dilate aperture macro primitive with exposure value computed from macro variable.')
|
|
||||||
pass
|
|
||||||
return replace(self, primitives=tuple(new_primitives))
|
|
||||||
|
|
||||||
def to_gerber(self, unit=None):
|
|
||||||
""" Serialize this macro's content (without the name) into Gerber using the given file unit """
|
|
||||||
comments = [ str(c) for c in self.comments ]
|
|
||||||
variable_defs = [ f'${var}={str(expr)[1:-1]}' for var, expr in enumerate(self.variables, start=1) if expr is not None ]
|
|
||||||
primitive_defs = [ prim.to_gerber(unit) for prim in self.primitives ]
|
|
||||||
return '*\n'.join(comments + variable_defs + primitive_defs)
|
|
||||||
|
|
||||||
def to_graphic_primitives(self, offset, rotation, parameters : [float], unit=None, polarity_dark=True):
|
|
||||||
variables = {i: v for i, v in enumerate(self.variables, start=1) if v is not None}
|
|
||||||
for number, value in enumerate(parameters, start=1):
|
|
||||||
if number in variables:
|
|
||||||
raise SyntaxError(f'Re-definition of aperture macro variable {number} through parameter {value}')
|
|
||||||
variables[number] = value
|
|
||||||
|
|
||||||
for primitive in self.primitives:
|
|
||||||
yield from primitive.to_graphic_primitives(offset, rotation, variables, unit, polarity_dark)
|
|
||||||
|
|
||||||
def rotated(self, angle):
|
|
||||||
# aperture macro primitives use degree counter-clockwise, our API uses radians clockwise
|
|
||||||
return replace(self, primitives=tuple(
|
|
||||||
replace(primitive, rotation=primitive.rotation - rad_to_deg(angle)) for primitive in self.primitives))
|
|
||||||
|
|
||||||
def scaled(self, scale):
|
|
||||||
return replace(self, primitives=tuple(
|
|
||||||
primitive.scaled(scale) for primitive in self.primitives))
|
|
||||||
|
|
||||||
|
|
||||||
var = VariableExpression
|
|
||||||
deg_per_rad = 180 / math.pi
|
|
||||||
|
|
||||||
class GenericMacros:
|
|
||||||
|
|
||||||
_generic_hole = lambda n: (ap.Circle('mm', 0, var(n), 0, 0),)
|
|
||||||
|
|
||||||
# NOTE: All generic macros have rotation values specified in **clockwise radians** like the rest of the user-facing
|
|
||||||
# API.
|
|
||||||
circle = ApertureMacro('GNC', (
|
|
||||||
ap.Circle('mm', 1, var(1), 0, 0, var(4) * -deg_per_rad),
|
|
||||||
*_generic_hole(2)))
|
|
||||||
|
|
||||||
rect = ApertureMacro('GNR', (
|
|
||||||
ap.CenterLine('mm', 1, var(1), var(2), 0, 0, var(5) * -deg_per_rad),
|
|
||||||
*_generic_hole(3)))
|
|
||||||
|
|
||||||
# params: width, height, corner radius, *hole, rotation
|
|
||||||
rounded_rect = ApertureMacro('GRR', (
|
|
||||||
ap.CenterLine('mm', 1, var(1)-2*var(3), var(2), 0, 0, var(6) * -deg_per_rad),
|
|
||||||
ap.CenterLine('mm', 1, var(1), var(2)-2*var(3), 0, 0, var(6) * -deg_per_rad),
|
|
||||||
ap.Circle('mm', 1, var(3)*2, +(var(1)/2-var(3)), +(var(2)/2-var(3)), var(6) * -deg_per_rad),
|
|
||||||
ap.Circle('mm', 1, var(3)*2, +(var(1)/2-var(3)), -(var(2)/2-var(3)), var(6) * -deg_per_rad),
|
|
||||||
ap.Circle('mm', 1, var(3)*2, -(var(1)/2-var(3)), +(var(2)/2-var(3)), var(6) * -deg_per_rad),
|
|
||||||
ap.Circle('mm', 1, var(3)*2, -(var(1)/2-var(3)), -(var(2)/2-var(3)), var(6) * -deg_per_rad),
|
|
||||||
*_generic_hole(4)))
|
|
||||||
|
|
||||||
# params: width, height, length difference between narrow side (top) and wide side (bottom), *hole, rotation
|
|
||||||
isosceles_trapezoid = ApertureMacro('GTR', (
|
|
||||||
ap.Outline('mm', 1, 4,
|
|
||||||
(var(1)/-2, var(2)/-2,
|
|
||||||
var(1)/-2+var(3)/2, var(2)/2,
|
|
||||||
var(1)/2-var(3)/2, var(2)/2,
|
|
||||||
var(1)/2, var(2)/-2,
|
|
||||||
var(1)/-2, var(2)/-2,),
|
|
||||||
var(6) * -deg_per_rad),
|
|
||||||
*_generic_hole(4)))
|
|
||||||
|
|
||||||
# params: width, height, length difference between narrow side (top) and wide side (bottom), margin, *hole, rotation
|
|
||||||
rounded_isosceles_trapezoid = ApertureMacro('GRTR', (
|
|
||||||
ap.Outline('mm', 1, 4,
|
|
||||||
(var(1)/-2, var(2)/-2,
|
|
||||||
var(1)/-2+var(3)/2, var(2)/2,
|
|
||||||
var(1)/2-var(3)/2, var(2)/2,
|
|
||||||
var(1)/2, var(2)/-2,
|
|
||||||
var(1)/-2, var(2)/-2,),
|
|
||||||
var(6) * -deg_per_rad),
|
|
||||||
ap.VectorLine('mm', 1, var(4)*2,
|
|
||||||
var(1)/-2, var(2)/-2,
|
|
||||||
var(1)/-2+var(3)/2, var(2)/2,),
|
|
||||||
ap.VectorLine('mm', 1, var(4)*2,
|
|
||||||
var(1)/-2+var(3)/2, var(2)/2,
|
|
||||||
var(1)/2-var(3)/2, var(2)/2,),
|
|
||||||
ap.VectorLine('mm', 1, var(4)*2,
|
|
||||||
var(1)/2-var(3)/2, var(2)/2,
|
|
||||||
var(1)/2, var(2)/-2,),
|
|
||||||
ap.VectorLine('mm', 1, var(4)*2,
|
|
||||||
var(1)/2, var(2)/-2,
|
|
||||||
var(1)/-2, var(2)/-2,),
|
|
||||||
ap.Circle('mm', 1, var(4)*2,
|
|
||||||
var(1)/-2, var(2)/-2,),
|
|
||||||
ap.Circle('mm', 1, var(4)*2,
|
|
||||||
var(1)/-2+var(3)/2, var(2)/2,),
|
|
||||||
ap.Circle('mm', 1, var(4)*2,
|
|
||||||
var(1)/2-var(3)/2, var(2)/2,),
|
|
||||||
ap.Circle('mm', 1, var(4)*2,
|
|
||||||
var(1)/2, var(2)/-2,),
|
|
||||||
*_generic_hole(5)))
|
|
||||||
|
|
||||||
# w must be larger than h
|
|
||||||
# params: width, height, *hole, rotation
|
|
||||||
obround = ApertureMacro('GNO', (
|
|
||||||
ap.CenterLine('mm', 1, var(1)-var(2), var(2), 0, 0, var(5) * -deg_per_rad),
|
|
||||||
ap.Circle('mm', 1, var(2), +(var(1)-var(2))/2, 0, var(5) * -deg_per_rad),
|
|
||||||
ap.Circle('mm', 1, var(2), -(var(1)-var(2))/2, 0, var(5) * -deg_per_rad),
|
|
||||||
*_generic_hole(3) ))
|
|
||||||
|
|
||||||
polygon = ApertureMacro('GNP', (
|
|
||||||
ap.Polygon('mm', 1, var(2), 0, 0, var(1), var(3) * -deg_per_rad),
|
|
||||||
ap.Circle('mm', 0, var(4), 0, 0)))
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
import sys
|
|
||||||
#for line in sys.stdin:
|
|
||||||
#expr = _parse_expression(line.strip())
|
|
||||||
#print(expr, '->', expr.optimized())
|
|
||||||
|
|
||||||
for primitive in parse_macro(sys.stdin.read(), 'mm'):
|
|
||||||
print(primitive)
|
|
||||||
|
|
||||||
|
|
@ -1,221 +0,0 @@
|
||||||
from .sexp import *
|
|
||||||
from .sexp_mapper import *
|
|
||||||
import time
|
|
||||||
|
|
||||||
from dataclasses import field
|
|
||||||
import math
|
|
||||||
import uuid
|
|
||||||
from contextlib import contextmanager
|
|
||||||
from itertools import cycle
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('color')
|
|
||||||
class Color:
|
|
||||||
r: int = None
|
|
||||||
g: int = None
|
|
||||||
b: int = None
|
|
||||||
a: int = None
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('stroke')
|
|
||||||
class Stroke:
|
|
||||||
width: Named(float) = 0.254
|
|
||||||
type: Named(AtomChoice(Atom.dash, Atom.dot, Atom.dash_dot_dot, Atom.dash_dot, Atom.default, Atom.solid)) = Atom.default
|
|
||||||
color: Color = None
|
|
||||||
|
|
||||||
|
|
||||||
class Dasher:
|
|
||||||
def __init__(self, obj):
|
|
||||||
if obj.stroke:
|
|
||||||
w, t = obj.stroke.width, obj.stroke.type
|
|
||||||
else:
|
|
||||||
w = obj.width or 0
|
|
||||||
t = Atom.solid
|
|
||||||
|
|
||||||
self.width = w
|
|
||||||
gap = 4*w
|
|
||||||
dot = 0
|
|
||||||
dash = 11*w
|
|
||||||
self.pattern = {
|
|
||||||
Atom.dash: [dash, gap],
|
|
||||||
Atom.dot: [dot, gap],
|
|
||||||
Atom.dash_dot_dot: [dash, gap, dot, gap, dot, gap],
|
|
||||||
Atom.dash_dot: [dash, gap, dot, gap],
|
|
||||||
Atom.default: [1e99],
|
|
||||||
Atom.solid: [1e99]}[t]
|
|
||||||
self.solid = t in (Atom.default, Atom.solid)
|
|
||||||
self.start_x, self.start_y = None, None
|
|
||||||
self.cur_x, self.cur_y = None, None
|
|
||||||
self.segments = []
|
|
||||||
|
|
||||||
def move(self, x, y):
|
|
||||||
if self.cur_x is None:
|
|
||||||
self.start_x, self.start_y = x, y
|
|
||||||
self.cur_x, self.cur_y = x, y
|
|
||||||
|
|
||||||
def line(self, x, y):
|
|
||||||
if x is None or y is None:
|
|
||||||
raise ValueError('line() called before move()')
|
|
||||||
self.segments.append((self.cur_x, self.cur_y, x, y))
|
|
||||||
self.cur_x, self.cur_y = x, y
|
|
||||||
|
|
||||||
def close(self):
|
|
||||||
self.segments.append((self.cur_x, self.cur_y, self.start_x, self.start_y))
|
|
||||||
self.cur_x, self.cur_y = None, None
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def _interpolate(x1, y1, x2, y2, length):
|
|
||||||
dx, dy = x2-x1, y2-y1
|
|
||||||
total = math.hypot(dx, dy)
|
|
||||||
if total == 0:
|
|
||||||
return x2, y2
|
|
||||||
frac = length / total
|
|
||||||
return x1 + dx*frac, y1 + dy*frac
|
|
||||||
|
|
||||||
def __iter__(self):
|
|
||||||
it = iter(self.segments)
|
|
||||||
segment_remaining, segment_pos = 0, 0
|
|
||||||
|
|
||||||
if self.width is None or self.width < 1e-3:
|
|
||||||
return
|
|
||||||
|
|
||||||
for length, stroked in cycle(zip(self.pattern, cycle([True, False]))):
|
|
||||||
length = max(1e-12, length)
|
|
||||||
import sys
|
|
||||||
while length > 0:
|
|
||||||
if segment_remaining == 0:
|
|
||||||
try:
|
|
||||||
x1, y1, x2, y2 = next(it)
|
|
||||||
except StopIteration:
|
|
||||||
return
|
|
||||||
dx, dy = x2-x1, y2-y1
|
|
||||||
lx, ly = x1, y1
|
|
||||||
segment_remaining = math.hypot(dx, dy)
|
|
||||||
segment_pos = 0
|
|
||||||
|
|
||||||
if segment_remaining > length:
|
|
||||||
segment_pos += length
|
|
||||||
ix, iy = self._interpolate(x1, y1, x2, y2, segment_pos)
|
|
||||||
segment_remaining -= length
|
|
||||||
if stroked:
|
|
||||||
yield lx, ly, ix, iy
|
|
||||||
lx, ly = ix, iy
|
|
||||||
break
|
|
||||||
|
|
||||||
else:
|
|
||||||
length -= segment_remaining
|
|
||||||
segment_remaining = 0
|
|
||||||
if stroked:
|
|
||||||
yield lx, ly, x2, y2
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('xy')
|
|
||||||
class XYCoord:
|
|
||||||
x: float = 0
|
|
||||||
y: float = 0
|
|
||||||
|
|
||||||
def isclose(self, other, tol=1e-6):
|
|
||||||
return math.isclose(self.x, other.x, tol) and math.isclose(self.y, other.y, tol)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('pts')
|
|
||||||
class PointList:
|
|
||||||
xy : List(XYCoord) = field(default_factory=list)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('xyz')
|
|
||||||
class XYZCoord:
|
|
||||||
x: float = 0
|
|
||||||
y: float = 0
|
|
||||||
z: float = 0
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('at')
|
|
||||||
class AtPos(XYCoord):
|
|
||||||
x: float = 0 # in millimeter
|
|
||||||
y: float = 0 # in millimeter
|
|
||||||
rotation: int = 0 # in degrees, can only be 0, 90, 180 or 270.
|
|
||||||
unlocked: Flag() = False
|
|
||||||
|
|
||||||
def __before_sexp__(self):
|
|
||||||
self.rotation = int(round(self.rotation % 360))
|
|
||||||
|
|
||||||
@property
|
|
||||||
def rotation_rad(self):
|
|
||||||
return math.radians(self.rotation)
|
|
||||||
|
|
||||||
@rotation_rad.setter
|
|
||||||
def rotation_rad(self, value):
|
|
||||||
self.rotation = math.degrees(value)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('font')
|
|
||||||
class FontSpec:
|
|
||||||
face: Named(str) = None
|
|
||||||
size: Rename(XYCoord) = field(default_factory=lambda: XYCoord(1.27, 1.27))
|
|
||||||
thickness: Named(float) = None
|
|
||||||
bold: Flag() = False
|
|
||||||
italic: Flag() = False
|
|
||||||
line_spacing: Named(float) = None
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('justify')
|
|
||||||
class Justify:
|
|
||||||
h: AtomChoice(Atom.left, Atom.right) = None
|
|
||||||
v: AtomChoice(Atom.top, Atom.bottom) = None
|
|
||||||
mirror: Flag() = False
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('effects')
|
|
||||||
class TextEffect:
|
|
||||||
font: FontSpec = field(default_factory=FontSpec)
|
|
||||||
justify: OmitDefault(Justify) = field(default_factory=Justify)
|
|
||||||
hide: Flag() = False
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('tstamp')
|
|
||||||
class Timestamp:
|
|
||||||
value: str = field(default_factory=uuid.uuid4)
|
|
||||||
|
|
||||||
def __after_parse__(self, parent):
|
|
||||||
self.value = str(self.value)
|
|
||||||
|
|
||||||
def before_sexp(self):
|
|
||||||
self.value = Atom(str(self.value))
|
|
||||||
|
|
||||||
def bump(self):
|
|
||||||
self.value = uuid.uuid4()
|
|
||||||
|
|
||||||
@sexp_type('tedit')
|
|
||||||
class EditTime:
|
|
||||||
value: str = field(default_factory=time.time)
|
|
||||||
|
|
||||||
def __after_parse__(self, parent):
|
|
||||||
self.value = int(str(self.value), 16)
|
|
||||||
|
|
||||||
def __before_sexp__(self):
|
|
||||||
self.value = Atom(f'{int(self.value):08X}')
|
|
||||||
|
|
||||||
def bump(self):
|
|
||||||
self.value = time.time()
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
class Foo:
|
|
||||||
pass
|
|
||||||
|
|
||||||
foo = Foo()
|
|
||||||
foo.stroke = troke(0.01, Atom.dash_dot_dot)
|
|
||||||
d = Dasher(foo)
|
|
||||||
#d = Dasher(Stroke(0.01, Atom.solid))
|
|
||||||
d.move(1, 1)
|
|
||||||
d.line(1, 2)
|
|
||||||
d.line(3, 2)
|
|
||||||
d.line(3, 1)
|
|
||||||
d.close()
|
|
||||||
|
|
||||||
print('<?xml version="1.0" standalone="no"?>')
|
|
||||||
print('<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">')
|
|
||||||
print('<svg version="1.1" width="4cm" height="3cm" viewBox="0 0 4 3" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink">')
|
|
||||||
for x1, y1, x2, y2 in d:
|
|
||||||
print(f'<path fill="none" stroke="black" stroke-width="0.01" stroke-linecap="round" d="M {x1},{y1} L {x2},{y2}"/>')
|
|
||||||
print('</svg>')
|
|
||||||
|
|
@ -1,230 +0,0 @@
|
||||||
|
|
||||||
import string
|
|
||||||
import math
|
|
||||||
|
|
||||||
from .sexp import *
|
|
||||||
from .base_types import *
|
|
||||||
from .primitives import *
|
|
||||||
|
|
||||||
from ... import graphic_objects as go
|
|
||||||
from ... import apertures as ap
|
|
||||||
from ...newstroke import Newstroke
|
|
||||||
from ...utils import rotate_point, MM
|
|
||||||
|
|
||||||
@sexp_type('layer')
|
|
||||||
class TextLayer:
|
|
||||||
layer: str = ''
|
|
||||||
knockout: Flag() = False
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('gr_text')
|
|
||||||
class Text:
|
|
||||||
text: str = ''
|
|
||||||
at: AtPos = field(default_factory=AtPos)
|
|
||||||
layer: TextLayer = field(default_factory=TextLayer)
|
|
||||||
tstamp: Timestamp = None
|
|
||||||
effects: TextEffect = field(default_factory=TextEffect)
|
|
||||||
|
|
||||||
def render(self, variables={}):
|
|
||||||
if not self.effects or self.effects.hide or not self.effects.font:
|
|
||||||
return
|
|
||||||
|
|
||||||
font = Newstroke.load()
|
|
||||||
line_width = self.effects.font.thickness
|
|
||||||
text = string.Template(self.text).safe_substitute(variables)
|
|
||||||
strokes = list(font.render(text, size=self.effects.font.size.y))
|
|
||||||
min_x = min(x for st in strokes for x, y in st)
|
|
||||||
min_y = min(y for st in strokes for x, y in st)
|
|
||||||
max_x = max(x for st in strokes for x, y in st)
|
|
||||||
max_y = max(y for st in strokes for x, y in st)
|
|
||||||
w = max_x - min_x
|
|
||||||
h = max_y - min_y
|
|
||||||
|
|
||||||
offx = -min_x + {
|
|
||||||
None: -w/2,
|
|
||||||
Atom.right: -w,
|
|
||||||
Atom.left: 0
|
|
||||||
}[self.effects.justify.h if self.effects.justify else None]
|
|
||||||
|
|
||||||
offy = {
|
|
||||||
None: self.effects.font.size.y/2,
|
|
||||||
Atom.top: self.effects.font.size.y,
|
|
||||||
Atom.bottom: 0
|
|
||||||
}[self.effects.justify.v if self.effects.justify else None]
|
|
||||||
|
|
||||||
aperture = ap.CircleAperture(line_width or 0.2, unit=MM)
|
|
||||||
for stroke in strokes:
|
|
||||||
out = []
|
|
||||||
|
|
||||||
for x, y in stroke:
|
|
||||||
x, y = x+offx, y+offy
|
|
||||||
x, y = rotate_point(x, y, math.radians(self.at.rotation or 0))
|
|
||||||
x, y = x+self.at.x, y+self.at.y
|
|
||||||
out.append((x, y))
|
|
||||||
|
|
||||||
for p1, p2 in zip(out[:-1], out[1:]):
|
|
||||||
yield go.Line(*p1, *p2, aperture=aperture, unit=MM)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('gr_text_box')
|
|
||||||
class TextBox:
|
|
||||||
locked: Flag() = False
|
|
||||||
text: str = ''
|
|
||||||
start: Named(XYCoord) = None
|
|
||||||
end: Named(XYCoord) = None
|
|
||||||
pts: PointList = field(default_factory=PointList)
|
|
||||||
angle: OmitDefault(Named(float)) = 0.0
|
|
||||||
layer: Named(str) = ""
|
|
||||||
tstamp: Timestamp = None
|
|
||||||
effects: TextEffect = field(default_factory=TextEffect)
|
|
||||||
stroke: Stroke = field(default_factory=Stroke)
|
|
||||||
render_cache: RenderCache = None
|
|
||||||
|
|
||||||
def render(self, variables={}):
|
|
||||||
text = string.Template(self.text).safe_substitute(variables)
|
|
||||||
if text != self.text:
|
|
||||||
raise ValueError('Rendering of vector font text with variables not yet supported')
|
|
||||||
|
|
||||||
if not render_cache or not render_cache.polygons:
|
|
||||||
raise ValueError('Vector font text with empty render cache')
|
|
||||||
|
|
||||||
for poly in render_cache.polygons:
|
|
||||||
reg = go.Region([(p.x, p.y) for p in poly.pts.xy], unit=MM)
|
|
||||||
|
|
||||||
if self.stroke:
|
|
||||||
if self.stroke.type not in (None, Atom.default, Atom.solid):
|
|
||||||
raise ValueError('Dashed strokes are not supported on vector text')
|
|
||||||
|
|
||||||
yield from reg.outline_objects(aperture=ap.CircleAperture(self.stroke.width, unit=MM))
|
|
||||||
|
|
||||||
yield reg
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('gr_line')
|
|
||||||
class Line:
|
|
||||||
start: Rename(XYCoord) = None
|
|
||||||
end: Rename(XYCoord) = None
|
|
||||||
angle: Named(float) = None # wat
|
|
||||||
layer: Named(str) = None
|
|
||||||
width: Named(float) = None
|
|
||||||
tstamp: Timestamp = None
|
|
||||||
|
|
||||||
def render(self, variables=None):
|
|
||||||
if self.angle:
|
|
||||||
raise NotImplementedError('Angles on lines are not implemented. Please raise an issue and provide an example file.')
|
|
||||||
|
|
||||||
aperture = ap.CircleAperture(self.width, unit=MM)
|
|
||||||
yield go.Line(self.start.x, self.start.y, self.end.x, self.end.y, aperture=aperture, unit=MM)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('fill')
|
|
||||||
class FillMode:
|
|
||||||
# Needed for compatibility with weird files
|
|
||||||
fill: AtomChoice(Atom.solid, Atom.yes, Atom.no, Atom.none) = False
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def __map__(self, obj, parent=None):
|
|
||||||
return obj[1] in (Atom.solid, Atom.yes)
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def __sexp__(self, value):
|
|
||||||
yield [Atom.fill, Atom.solid if value else Atom.none]
|
|
||||||
|
|
||||||
@sexp_type('gr_rect')
|
|
||||||
class Rectangle:
|
|
||||||
start: Rename(XYCoord) = None
|
|
||||||
end: Rename(XYCoord) = None
|
|
||||||
layer: Named(str) = None
|
|
||||||
width: Named(float) = None
|
|
||||||
fill: FillMode = False
|
|
||||||
tstamp: Timestamp = None
|
|
||||||
|
|
||||||
def render(self, variables=None):
|
|
||||||
rect = go.Region.from_rectangle(self.start.x, self.start.y,
|
|
||||||
self.end.x-self.start.x, self.end.y-self.start.y,
|
|
||||||
unit=MM)
|
|
||||||
|
|
||||||
if self.fill:
|
|
||||||
yield rect
|
|
||||||
|
|
||||||
if self.width:
|
|
||||||
yield from rect.outline_objects(aperture=ap.CircleAperture(self.width, unit=MM))
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('gr_circle')
|
|
||||||
class Circle:
|
|
||||||
center: Rename(XYCoord) = None
|
|
||||||
end: Rename(XYCoord) = None
|
|
||||||
layer: Named(str) = None
|
|
||||||
width: Named(float) = None
|
|
||||||
fill: FillMode = False
|
|
||||||
tstamp: Timestamp = None
|
|
||||||
|
|
||||||
def render(self, variables=None):
|
|
||||||
r = math.dist((self.center.x, self.center.y), (self.end.x, self.end.y))
|
|
||||||
aperture = ap.CircleAperture(self.width or 0, unit=MM)
|
|
||||||
arc = go.Arc.from_circle(self.center.x, self.center.y, r, aperture=aperture, unit=MM)
|
|
||||||
|
|
||||||
if self.width:
|
|
||||||
yield arc
|
|
||||||
|
|
||||||
if self.fill:
|
|
||||||
yield arc.to_region()
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('gr_arc')
|
|
||||||
class Arc:
|
|
||||||
start: Rename(XYCoord) = None
|
|
||||||
mid: Rename(XYCoord) = None
|
|
||||||
end: Rename(XYCoord) = None
|
|
||||||
layer: Named(str) = None
|
|
||||||
width: Named(float) = None
|
|
||||||
tstamp: Timestamp = None
|
|
||||||
|
|
||||||
def render(self, variables=None):
|
|
||||||
if not self.width:
|
|
||||||
return
|
|
||||||
|
|
||||||
cx, cy = self.mid.x, self.mid.y
|
|
||||||
x1, y1 = self.start.x, self.start.y
|
|
||||||
x2, y2 = self.end.x, self.end.y
|
|
||||||
yield go.Arc(x1, y1, x2, y2, cx-x1, cy-y1, aperture=ap.CircleAperture(self.width or 0, unit=MM), clockwise=True, unit=MM)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('gr_poly')
|
|
||||||
class Polygon:
|
|
||||||
pts: PointList = field(default_factory=PointList)
|
|
||||||
layer: Named(str) = None
|
|
||||||
width: Named(float) = None
|
|
||||||
fill: FillMode = True
|
|
||||||
tstamp: Timestamp = None
|
|
||||||
|
|
||||||
def render(self, variables=None):
|
|
||||||
reg = go.Region([(pt.x, pt.y) for pt in self.pts.xy], unit=MM)
|
|
||||||
|
|
||||||
if self.width and self.width >= 0.005:
|
|
||||||
yield from reg.outline_objects(aperture=ap.CircleAperture(self.width, unit=MM))
|
|
||||||
|
|
||||||
if self.fill:
|
|
||||||
yield reg
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('gr_curve')
|
|
||||||
class Curve:
|
|
||||||
pts: PointList = field(default_factory=PointList)
|
|
||||||
layer: Named(str) = None
|
|
||||||
width: Named(float) = None
|
|
||||||
tstamp: Timestamp = None
|
|
||||||
|
|
||||||
def render(self, variables=None):
|
|
||||||
raise NotImplementedError('Bezier rendering is not yet supported. Please raise an issue and provide an example file.')
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('gr_bbox')
|
|
||||||
class AnnotationBBox:
|
|
||||||
start: Rename(XYCoord) = None
|
|
||||||
end: Rename(XYCoord) = None
|
|
||||||
|
|
||||||
def render(self, variables=None):
|
|
||||||
return []
|
|
||||||
|
|
||||||
|
|
@ -1,97 +0,0 @@
|
||||||
|
|
||||||
import enum
|
|
||||||
|
|
||||||
from .sexp import *
|
|
||||||
from .base_types import *
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('hatch')
|
|
||||||
class Hatch:
|
|
||||||
style: AtomChoice(Atom.none, Atom.edge, Atom.full) = Atom.edge
|
|
||||||
pitch: float = 0.5
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('connect_pads')
|
|
||||||
class PadConnection:
|
|
||||||
type: AtomChoice(Atom.thru_hole_only, Atom.full, Atom.no) = None
|
|
||||||
clearance: Named(float) = 0
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('keepout')
|
|
||||||
class ZoneKeepout:
|
|
||||||
tracks_allowed: Named(YesNoAtom(yes=Atom.allowed, no=Atom.not_allowed), name='tracks') = True
|
|
||||||
vias_allowed: Named(YesNoAtom(yes=Atom.allowed, no=Atom.not_allowed), name='vias') = True
|
|
||||||
pads_allowed: Named(YesNoAtom(yes=Atom.allowed, no=Atom.not_allowed), name='pads') = True
|
|
||||||
copperpour_allowed: Named(YesNoAtom(yes=Atom.allowed, no=Atom.not_allowed), name='copperpour') = True
|
|
||||||
footprints_allowed: Named(YesNoAtom(yes=Atom.allowed, no=Atom.not_allowed), name='footprints') = True
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('smoothing')
|
|
||||||
class ZoneSmoothing:
|
|
||||||
style: AtomChoice(Atom.chamfer, Atom.fillet) = Atom.chamfer
|
|
||||||
radius: Named(float) = None
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('fill')
|
|
||||||
class ZoneFill:
|
|
||||||
yes: Flag() = False
|
|
||||||
mode: Flag(atom=Atom.hatched) = False
|
|
||||||
thermal_gap: Named(float) = 0.508
|
|
||||||
thermal_bridge_width: Named(float) = 0.508
|
|
||||||
smoothing: ZoneSmoothing = None
|
|
||||||
island_removal_node: Named(int) = None
|
|
||||||
islan_area_min: Named(float) = None
|
|
||||||
hatch_thickness: Named(float) = None
|
|
||||||
hatch_gap: Named(float) = None
|
|
||||||
hatch_orientation: Named(int) = None
|
|
||||||
hatch_smoothing_level: Named(int) = None
|
|
||||||
hatch_smoothing_value: Named(float) = None
|
|
||||||
hatch_border_algorithm: Named(int) = None
|
|
||||||
hatch_min_hole_area: Named(float) = None
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('filled_polygon')
|
|
||||||
class FillPolygon:
|
|
||||||
layer: Named(str) = ""
|
|
||||||
pts: PointList = field(default_factory=PointList)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('fill_segments')
|
|
||||||
class FillSegment:
|
|
||||||
layer: Named(str) = ""
|
|
||||||
pts: PointList = field(default_factory=PointList)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('zone')
|
|
||||||
class Zone:
|
|
||||||
net: Named(int) = 0
|
|
||||||
net_name: Named(str) = ""
|
|
||||||
layer: Named(str) = None
|
|
||||||
layers: Named(Array(str)) = None
|
|
||||||
tstamp: Timestamp = None
|
|
||||||
name: Named(str) = None
|
|
||||||
hatch: Hatch = None
|
|
||||||
priority: OmitDefault(Named(int)) = 0
|
|
||||||
connect_pads: PadConnection = field(default_factory=PadConnection)
|
|
||||||
min_thickness: Named(float) = 0.254
|
|
||||||
filled_areas_thickness: Flag() = True
|
|
||||||
keepouts: List(ZoneKeepout) = field(default_factory=list)
|
|
||||||
fill: ZoneFill = field(default_factory=ZoneFill)
|
|
||||||
polygon: Named(PointList) = field(default_factory=PointList)
|
|
||||||
fill_polygons: List(FillPolygon) = field(default_factory=list)
|
|
||||||
fill_segments: List(FillSegment) = field(default_factory=list)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('polygon')
|
|
||||||
class RenderCachePolygon:
|
|
||||||
pts: PointList = field(default_factory=PointList)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('render_cache')
|
|
||||||
class RenderCache:
|
|
||||||
text: str = None
|
|
||||||
rotation: int = 0
|
|
||||||
polygons: List(RenderCachePolygon) = field(default_factory=list)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -1,289 +0,0 @@
|
||||||
|
|
||||||
from dataclasses import MISSING
|
|
||||||
from .sexp import *
|
|
||||||
|
|
||||||
|
|
||||||
SEXP_END = type('SEXP_END', (), {})
|
|
||||||
|
|
||||||
|
|
||||||
class AtomChoice:
|
|
||||||
def __init__(self, *choices):
|
|
||||||
self.choices = choices
|
|
||||||
|
|
||||||
def __contains__(self, value):
|
|
||||||
return value in self.choices
|
|
||||||
|
|
||||||
def __atoms__(self):
|
|
||||||
return self.choices
|
|
||||||
|
|
||||||
def __map__(self, obj, parent=None):
|
|
||||||
obj, = obj
|
|
||||||
if obj not in self:
|
|
||||||
raise TypeError(f'Invalid atom {obj} for {type(self)}, valid choices are: {", ".join(map(str, self.choices))}')
|
|
||||||
return obj
|
|
||||||
|
|
||||||
def __sexp__(self, value):
|
|
||||||
yield value
|
|
||||||
|
|
||||||
|
|
||||||
class Flag:
|
|
||||||
def __init__(self, atom=None, invert=None):
|
|
||||||
self.atom, self.invert = atom, invert
|
|
||||||
|
|
||||||
def __bind_field__(self, field):
|
|
||||||
if self.atom is None:
|
|
||||||
self.atom = Atom(field.name)
|
|
||||||
if self.invert is None:
|
|
||||||
self.invert = bool(field.default)
|
|
||||||
|
|
||||||
def __atoms__(self):
|
|
||||||
return [self.atom]
|
|
||||||
|
|
||||||
def __map__(self, obj, parent=None):
|
|
||||||
return not self.invert
|
|
||||||
|
|
||||||
def __sexp__(self, value):
|
|
||||||
if bool(value) == (not self.invert):
|
|
||||||
yield self.atom
|
|
||||||
|
|
||||||
|
|
||||||
def sexp(t, v):
|
|
||||||
if v is None:
|
|
||||||
return []
|
|
||||||
elif t in (int, float, str, Atom):
|
|
||||||
return [t(v)]
|
|
||||||
elif hasattr(t, '__sexp__'):
|
|
||||||
return list(t.__sexp__(v))
|
|
||||||
elif isinstance(t, list):
|
|
||||||
t, = t
|
|
||||||
return [sexp(t, elem) for elem in v]
|
|
||||||
else:
|
|
||||||
raise TypeError(f'Python type {t} has no defined s-expression serialization')
|
|
||||||
|
|
||||||
|
|
||||||
def map_sexp(t, v, parent=None):
|
|
||||||
if t is not Atom and hasattr(t, '__map__'):
|
|
||||||
return t.__map__(v, parent=parent)
|
|
||||||
elif t in (int, float, str, Atom):
|
|
||||||
v, = v
|
|
||||||
if not isinstance(v, t):
|
|
||||||
types = set({type(v), t})
|
|
||||||
if types == {int, float} or types == {str, Atom}:
|
|
||||||
v = t(v)
|
|
||||||
else:
|
|
||||||
raise TypeError(f'Cannot map s-expression value {v} of type {type(v)} to Python type {t}')
|
|
||||||
return v
|
|
||||||
elif isinstance(t, list):
|
|
||||||
t, = t
|
|
||||||
return [map_sexp(t, elem, parent=parent) for elem in v]
|
|
||||||
else:
|
|
||||||
raise TypeError(f'Python type {t} has no defined s-expression deserialization')
|
|
||||||
|
|
||||||
|
|
||||||
class WrapperType:
|
|
||||||
def __init__(self, next_type):
|
|
||||||
self.next_type = next_type
|
|
||||||
|
|
||||||
def __bind_field__(self, field):
|
|
||||||
self.field = field
|
|
||||||
getattr(self.next_type, '__bind_field__', lambda x: None)(field)
|
|
||||||
|
|
||||||
def __atoms__(self):
|
|
||||||
if hasattr(self, 'name_atom'):
|
|
||||||
return [self.name_atom]
|
|
||||||
elif self.next_type is Atom:
|
|
||||||
return []
|
|
||||||
else:
|
|
||||||
return getattr(self.next_type, '__atoms__', lambda: [])()
|
|
||||||
|
|
||||||
class Named(WrapperType):
|
|
||||||
def __init__(self, next_type, name=None):
|
|
||||||
super().__init__(next_type)
|
|
||||||
self.name_atom = Atom(name) if name else None
|
|
||||||
|
|
||||||
def __bind_field__(self, field):
|
|
||||||
if self.next_type is not Atom:
|
|
||||||
getattr(self.next_type, '__bind_field__', lambda x: None)(field)
|
|
||||||
if self.name_atom is None:
|
|
||||||
self.name_atom = Atom(field.name)
|
|
||||||
|
|
||||||
def __map__(self, obj, parent=None):
|
|
||||||
k, *obj = obj
|
|
||||||
if self.next_type in (int, float, str, Atom) or isinstance(self.next_type, AtomChoice):
|
|
||||||
return map_sexp(self.next_type, [*obj], parent=parent)
|
|
||||||
else:
|
|
||||||
return map_sexp(self.next_type, obj, parent=parent)
|
|
||||||
|
|
||||||
def __sexp__(self, value):
|
|
||||||
value = sexp(self.next_type, value)
|
|
||||||
if value is not None:
|
|
||||||
yield [self.name_atom, *value]
|
|
||||||
|
|
||||||
|
|
||||||
class Rename(WrapperType):
|
|
||||||
def __init__(self, next_type, name=None):
|
|
||||||
super().__init__(next_type)
|
|
||||||
self.name_atom = Atom(name) if name else None
|
|
||||||
|
|
||||||
def __bind_field__(self, field):
|
|
||||||
if self.name_atom is None:
|
|
||||||
self.name_atom = Atom(field.name)
|
|
||||||
|
|
||||||
def __map__(self, obj, parent=None):
|
|
||||||
return map_sexp(self.next_type, obj, parent=parent)
|
|
||||||
|
|
||||||
def __sexp__(self, value):
|
|
||||||
value, = sexp(self.next_type, value)
|
|
||||||
if self.next_type in (str, float, int, Atom):
|
|
||||||
yield [self.name_atom, *value]
|
|
||||||
else:
|
|
||||||
key, *rest = value
|
|
||||||
yield [self.name_atom, *rest]
|
|
||||||
|
|
||||||
|
|
||||||
class OmitDefault(WrapperType):
|
|
||||||
def __bind_field__(self, field):
|
|
||||||
getattr(self.next_type, '__bind_field__', lambda x: None)(field)
|
|
||||||
if field.default_factory != MISSING:
|
|
||||||
self.default = field.default_factory()
|
|
||||||
else:
|
|
||||||
self.default = field.default
|
|
||||||
|
|
||||||
def __map__(self, obj, parent=None):
|
|
||||||
return map_sexp(self.next_type, obj, parent=parent)
|
|
||||||
|
|
||||||
def __sexp__(self, value):
|
|
||||||
if value != self.default:
|
|
||||||
yield from sexp(self.next_type, value)
|
|
||||||
|
|
||||||
|
|
||||||
class YesNoAtom:
|
|
||||||
def __init__(self, yes=Atom.yes, no=Atom.no):
|
|
||||||
self.yes, self.no = yes, no
|
|
||||||
|
|
||||||
def __map__(self, value, parent=None):
|
|
||||||
value, = value
|
|
||||||
return value == self.yes
|
|
||||||
|
|
||||||
def __sexp__(self, value):
|
|
||||||
yield self.yes if value else self.no
|
|
||||||
|
|
||||||
|
|
||||||
class Wrap(WrapperType):
|
|
||||||
def __map__(self, value, parent=None):
|
|
||||||
value, = value
|
|
||||||
return map_sexp(self.next_type, value, parent=parent)
|
|
||||||
|
|
||||||
def __sexp__(self, value):
|
|
||||||
for inner in sexp(self.next_type, value):
|
|
||||||
yield [inner]
|
|
||||||
|
|
||||||
|
|
||||||
class Array(WrapperType):
|
|
||||||
def __map__(self, value, parent=None):
|
|
||||||
return [map_sexp(self.next_type, [elem], parent=parent) for elem in value]
|
|
||||||
|
|
||||||
def __sexp__(self, value):
|
|
||||||
for e in value:
|
|
||||||
yield from sexp(self.next_type, e)
|
|
||||||
|
|
||||||
|
|
||||||
class List(WrapperType):
|
|
||||||
def __bind_field__(self, field):
|
|
||||||
self.attr = field.name
|
|
||||||
|
|
||||||
def __map__(self, value, parent):
|
|
||||||
l = getattr(parent, self.attr, [])
|
|
||||||
mapped = map_sexp(self.next_type, value, parent=parent)
|
|
||||||
l.append(mapped)
|
|
||||||
setattr(parent, self.attr, l)
|
|
||||||
|
|
||||||
def __sexp__(self, value):
|
|
||||||
for elem in value:
|
|
||||||
yield from sexp(self.next_type, elem)
|
|
||||||
|
|
||||||
|
|
||||||
class _SexpTemplate:
|
|
||||||
@staticmethod
|
|
||||||
def __atoms__(kls):
|
|
||||||
return [kls.name_atom]
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def __map__(kls, value, *args, parent=None, **kwargs):
|
|
||||||
positional = iter(kls.positional)
|
|
||||||
inst = kls(*args, **kwargs)
|
|
||||||
|
|
||||||
for v in value[1:]: # skip key
|
|
||||||
if isinstance(v, Atom) and v in kls.keys:
|
|
||||||
name, etype = kls.keys[v]
|
|
||||||
mapped = map_sexp(etype, [v], parent=inst)
|
|
||||||
if mapped is not None:
|
|
||||||
setattr(inst, name, mapped)
|
|
||||||
|
|
||||||
elif isinstance(v, list):
|
|
||||||
name, etype = kls.keys[v[0]]
|
|
||||||
mapped = map_sexp(etype, v, parent=inst)
|
|
||||||
if mapped is not None:
|
|
||||||
setattr(inst, name, mapped)
|
|
||||||
|
|
||||||
else:
|
|
||||||
try:
|
|
||||||
pos_key = next(positional)
|
|
||||||
setattr(inst, pos_key.name, v)
|
|
||||||
except StopIteration:
|
|
||||||
raise TypeError(f'Unhandled positional argument {v!r} while parsing {kls}')
|
|
||||||
|
|
||||||
getattr(inst, '__after_parse__', lambda x: None)(parent)
|
|
||||||
return inst
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def __sexp__(kls, value):
|
|
||||||
getattr(value, '__before_sexp__', lambda: None)()
|
|
||||||
|
|
||||||
out = [kls.name_atom]
|
|
||||||
for f in fields(kls):
|
|
||||||
if f.type is SEXP_END:
|
|
||||||
break
|
|
||||||
out += sexp(f.type, getattr(value, f.name))
|
|
||||||
yield out
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def parse(kls, data, *args, **kwargs):
|
|
||||||
return kls.__map__(parse_sexp(data), *args, **kwargs)
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def sexp(self):
|
|
||||||
return next(self.__sexp__(self))
|
|
||||||
|
|
||||||
|
|
||||||
def sexp_type(name=None):
|
|
||||||
def register(cls):
|
|
||||||
cls = dataclass(cls)
|
|
||||||
cls.name_atom = Atom(name) if name is not None else None
|
|
||||||
for key in '__sexp__', '__map__', '__atoms__', 'parse':
|
|
||||||
if not hasattr(cls, key):
|
|
||||||
setattr(cls, key, classmethod(getattr(_SexpTemplate, key)))
|
|
||||||
|
|
||||||
if not hasattr(cls, 'sexp'):
|
|
||||||
setattr(cls, 'sexp', getattr(_SexpTemplate, 'sexp'))
|
|
||||||
cls.positional = []
|
|
||||||
cls.keys = {}
|
|
||||||
for f in fields(cls):
|
|
||||||
f_type = f.type
|
|
||||||
if f_type is SEXP_END:
|
|
||||||
break
|
|
||||||
|
|
||||||
if hasattr(f_type, '__bind_field__'):
|
|
||||||
f_type.__bind_field__(f)
|
|
||||||
|
|
||||||
atoms = getattr(f_type, '__atoms__', lambda: [])
|
|
||||||
atoms = list(atoms())
|
|
||||||
for atom in atoms:
|
|
||||||
cls.keys[atom] = (f.name, f_type)
|
|
||||||
if not atoms:
|
|
||||||
cls.positional.append(f)
|
|
||||||
|
|
||||||
return cls
|
|
||||||
return register
|
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -1,446 +0,0 @@
|
||||||
"""
|
|
||||||
Library for processing KiCad's symbol files.
|
|
||||||
"""
|
|
||||||
|
|
||||||
import json
|
|
||||||
import string
|
|
||||||
import math
|
|
||||||
import re
|
|
||||||
import sys
|
|
||||||
import itertools
|
|
||||||
from fnmatch import fnmatch
|
|
||||||
from collections import defaultdict
|
|
||||||
from dataclasses import field
|
|
||||||
from pathlib import Path
|
|
||||||
from typing import Any, Dict, List, Optional, Tuple
|
|
||||||
|
|
||||||
from .sexp import *
|
|
||||||
from .sexp_mapper import *
|
|
||||||
from .base_types import *
|
|
||||||
|
|
||||||
|
|
||||||
PIN_ETYPE = AtomChoice(Atom.input, Atom.output, Atom.bidirectional, Atom.tri_state, Atom.passive, Atom.free,
|
|
||||||
Atom.unspecified, Atom.power_in, Atom.power_out, Atom.open_collector, Atom.open_emitter,
|
|
||||||
Atom.no_connect)
|
|
||||||
|
|
||||||
|
|
||||||
PIN_STYLE = AtomChoice(Atom.line, Atom.inverted, Atom.clock, Atom.inverted_clock, Atom.input_low, Atom.clock_low,
|
|
||||||
Atom.output_low, Atom.edge_clock_high, Atom.non_logic)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('alternate')
|
|
||||||
class AltFunction:
|
|
||||||
name: str = None
|
|
||||||
etype: PIN_ETYPE = Atom.unspecified
|
|
||||||
shape: PIN_STYLE = Atom.line
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('__styled_text')
|
|
||||||
class StyledText:
|
|
||||||
value: str = None
|
|
||||||
effects: TextEffect = field(default_factory=TextEffect)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('pin')
|
|
||||||
class Pin:
|
|
||||||
etype: PIN_ETYPE = Atom.unspecified
|
|
||||||
style: PIN_STYLE = Atom.line
|
|
||||||
at: AtPos = field(default_factory=AtPos)
|
|
||||||
length: Named(float) = 2.54
|
|
||||||
hide: Flag() = False
|
|
||||||
name: Rename(StyledText) = field(default_factory=StyledText)
|
|
||||||
number: Rename(StyledText) = field(default_factory=StyledText)
|
|
||||||
alternates: List(AltFunction) = field(default_factory=list)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def direction(self):
|
|
||||||
return {0: 'R', 90: 'U', 180: 'L', 270: 'D'}.get(self.at.rotation, 'R')
|
|
||||||
|
|
||||||
@direction.setter
|
|
||||||
def direction(self, value):
|
|
||||||
self.at.rotation = {0: 'R', 90: 'U', 180: 'L', 270: 'D'}[value[0].upper()]
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('fill')
|
|
||||||
class Fill:
|
|
||||||
type: Named(AtomChoice(Atom.none, Atom.outline, Atom.background)) = Atom.none
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('circle')
|
|
||||||
class Circle:
|
|
||||||
center: Rename(XYCoord) = field(default_factory=XYCoord)
|
|
||||||
radius: Named(float) = 0.0
|
|
||||||
stroke: Stroke = field(default_factory=Stroke)
|
|
||||||
fill: Fill = field(default_factory=Fill)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('arc')
|
|
||||||
class Arc:
|
|
||||||
start: Rename(XYCoord) = field(default_factory=XYCoord)
|
|
||||||
mid: Rename(XYCoord) = field(default_factory=XYCoord)
|
|
||||||
end: Rename(XYCoord) = field(default_factory=XYCoord)
|
|
||||||
stroke: Stroke = field(default_factory=Stroke)
|
|
||||||
fill: Fill = field(default_factory=Fill)
|
|
||||||
|
|
||||||
# TODO add function to calculate center, bounding box
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('polyline')
|
|
||||||
class Polyline:
|
|
||||||
pts: PointList = field(default_factory=PointList)
|
|
||||||
stroke: Stroke = field(default_factory=Stroke)
|
|
||||||
fill: Fill = field(default_factory=Fill)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def points(self):
|
|
||||||
return self.pts.xy
|
|
||||||
|
|
||||||
@points.setter
|
|
||||||
def points(self, value):
|
|
||||||
self.pts.xy = value
|
|
||||||
|
|
||||||
@property
|
|
||||||
def closed(self):
|
|
||||||
# if the last and first point are the same, we consider the polyline closed
|
|
||||||
# a closed triangle will have 4 points (A-B-C-A) stored in the list of points
|
|
||||||
return len(self.points) > 3 and self.points[0] == self.points[-1]
|
|
||||||
|
|
||||||
@property
|
|
||||||
def bbox(self):
|
|
||||||
if not self.points:
|
|
||||||
return (0.0, 0.0, 0.0, 0.0)
|
|
||||||
|
|
||||||
return (min(p.x for p in self.points),
|
|
||||||
min(p.y for p in self.points),
|
|
||||||
max(p.x for p in self.points),
|
|
||||||
max(p.y for p in self.points))
|
|
||||||
|
|
||||||
def as_rectangle(self):
|
|
||||||
(maxx, maxy, minx, miny) = self.get_boundingbox()
|
|
||||||
return Rectangle(
|
|
||||||
minx,
|
|
||||||
maxy,
|
|
||||||
maxx,
|
|
||||||
miny,
|
|
||||||
self.stroke_width,
|
|
||||||
self.stroke_color,
|
|
||||||
self.fill_type,
|
|
||||||
self.fill_color,
|
|
||||||
unit=self.unit,
|
|
||||||
demorgan=self.demorgan,
|
|
||||||
)
|
|
||||||
|
|
||||||
def get_center_of_boundingbox(self):
|
|
||||||
(maxx, maxy, minx, miny) = self.get_boundingbox()
|
|
||||||
return ((minx + maxx) / 2, ((miny + maxy) / 2))
|
|
||||||
|
|
||||||
def is_rectangle(self):
|
|
||||||
# a rectangle has 5 points and is closed
|
|
||||||
if len(self.points) != 5 or not self.is_closed():
|
|
||||||
return False
|
|
||||||
|
|
||||||
# construct lines between the points
|
|
||||||
p0 = self.points[0]
|
|
||||||
for p1_idx in range(1, len(self.points)):
|
|
||||||
p1 = self.points[p1_idx]
|
|
||||||
dx = p1.x - p0.x
|
|
||||||
dy = p1.y - p0.y
|
|
||||||
if dx != 0 and dy != 0:
|
|
||||||
# if a line is neither horizontal or vertical its not
|
|
||||||
# part of a rectangle
|
|
||||||
return False
|
|
||||||
# select next point
|
|
||||||
p0 = p1
|
|
||||||
|
|
||||||
return True
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('at')
|
|
||||||
class TextPos(XYCoord):
|
|
||||||
x: float = 0 # in millimeter
|
|
||||||
y: float = 0 # in millimeter
|
|
||||||
rotation: int = 0 # in degrees
|
|
||||||
|
|
||||||
def __after_parse__(self, parent):
|
|
||||||
self.rotation = self.rotation / 10
|
|
||||||
|
|
||||||
def __before_sexp__(self):
|
|
||||||
self.rotation = round((self.rotation % 360) * 10)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def rotation_rad(self):
|
|
||||||
return math.radians(self.rotation)
|
|
||||||
|
|
||||||
@rotation_rad.setter
|
|
||||||
def rotation_rad(self, value):
|
|
||||||
self.rotation = math.degrees(value)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('text')
|
|
||||||
class Text:
|
|
||||||
text: str = None
|
|
||||||
at: TextPos = field(default_factory=TextPos)
|
|
||||||
rotation: float = None
|
|
||||||
effects: TextEffect = field(default_factory=TextEffect)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('rectangle')
|
|
||||||
class Rectangle:
|
|
||||||
"""
|
|
||||||
Some v6 symbols use rectangles, newer ones encode them as polylines.
|
|
||||||
At some point in time we can most likely remove this class since its not used anymore
|
|
||||||
"""
|
|
||||||
|
|
||||||
start: Rename(XYCoord) = None
|
|
||||||
end: Rename(XYCoord) = None
|
|
||||||
stroke: Stroke = field(default_factory=Stroke)
|
|
||||||
fill: Fill = field(default_factory=Fill)
|
|
||||||
|
|
||||||
def as_polyline(self):
|
|
||||||
x1, y1 = self.start
|
|
||||||
x2, y2 = self.end
|
|
||||||
return Polyline([Point(x1, y1), Point(x2, y1), Point(x2, y2), Point(x1, y2), Point(x1, y1)],
|
|
||||||
self.stroke, self.fill)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('property')
|
|
||||||
class Property:
|
|
||||||
name: str = None
|
|
||||||
value: str = None
|
|
||||||
id: Named(int) = None
|
|
||||||
at: AtPos = field(default_factory=AtPos)
|
|
||||||
effects: TextEffect = field(default_factory=TextEffect)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('pin_numbers')
|
|
||||||
class PinNumberSpec:
|
|
||||||
hide: Flag() = False
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('pin_names')
|
|
||||||
class PinNameSpec:
|
|
||||||
offset: OmitDefault(Named(float)) = 0.508
|
|
||||||
hide: Flag() = False
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('symbol')
|
|
||||||
class Unit:
|
|
||||||
name: str = None
|
|
||||||
circles: List(Circle) = field(default_factory=list)
|
|
||||||
arcs: List(Arc) = field(default_factory=list)
|
|
||||||
polylines: List(Polyline) = field(default_factory=list)
|
|
||||||
rectangles: List(Rectangle) = field(default_factory=list)
|
|
||||||
texts: List(Text) = field(default_factory=list)
|
|
||||||
pins: List(Pin) = field(default_factory=list)
|
|
||||||
unit_name: Named(str) = None
|
|
||||||
_ : SEXP_END = None
|
|
||||||
global_units: list = field(default_factory=list)
|
|
||||||
unit_global: Flag() = False
|
|
||||||
style_global: Flag() = False
|
|
||||||
demorgan_style: int = 1
|
|
||||||
unit_index: int = 1
|
|
||||||
symbol = None
|
|
||||||
|
|
||||||
def __after_parse__(self, parent):
|
|
||||||
self.symbol = parent
|
|
||||||
|
|
||||||
if not (m := re.fullmatch(r'(.*)_([0-9]+)_([0-9]+)', self.name)):
|
|
||||||
raise FormatError(f'Invalid unit name "{self.name}"')
|
|
||||||
sym_name, unit_index, demorgan_style = m.groups()
|
|
||||||
if sym_name != self.symbol.name:
|
|
||||||
raise FormatError(f'Unit name "{self.name}" does not match symbol name "{self.symbol.name}"')
|
|
||||||
self.demorgan_style = int(demorgan_style)
|
|
||||||
self.unit_index = int(unit_index)
|
|
||||||
self.style_global = self._demorgan_style == 0
|
|
||||||
self.unit_global = self.unit_index == 0
|
|
||||||
|
|
||||||
def __before_sexp__(self):
|
|
||||||
self.name = f'{self.symbol.name}_{self.unit_index}_{self.demorgan_style}'
|
|
||||||
|
|
||||||
def __getattr__(self, name):
|
|
||||||
if name.startswith('all_'):
|
|
||||||
name = name[4:]
|
|
||||||
return itertools.chain(getattr(self.global_units, name, []), getattr(self, name, []))
|
|
||||||
|
|
||||||
def pin_stacks(self):
|
|
||||||
stacks = defaultdict(lambda: set())
|
|
||||||
for pin in self.all_pins():
|
|
||||||
stacks[(pin.at.x, pin.at.y)].add(pin)
|
|
||||||
return stacks
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('symbol')
|
|
||||||
class Symbol:
|
|
||||||
name: str = None
|
|
||||||
extends: Named(str) = None
|
|
||||||
power: Wrap(Flag()) = False
|
|
||||||
pin_numbers: OmitDefault(PinNumberSpec) = field(default_factory=PinNumberSpec)
|
|
||||||
pin_names: OmitDefault(PinNameSpec) = field(default_factory=PinNameSpec)
|
|
||||||
in_bom: Named(YesNoAtom()) = True
|
|
||||||
on_board: Named(YesNoAtom()) = True
|
|
||||||
properties: List(Property) = field(default_factory=list)
|
|
||||||
raw_units: List(Unit) = field(default_factory=list)
|
|
||||||
_ : SEXP_END = None
|
|
||||||
styles: {str: {str: Unit}} = None
|
|
||||||
global_units: {str: {str: Unit}} = None
|
|
||||||
library = None
|
|
||||||
|
|
||||||
def __after_parse__(self, parent):
|
|
||||||
self.library = parent
|
|
||||||
|
|
||||||
self.global_units = {}
|
|
||||||
self.styles = {}
|
|
||||||
|
|
||||||
if self.extends:
|
|
||||||
self.in_bom = None
|
|
||||||
self.on_board = None
|
|
||||||
|
|
||||||
self.properties = {prop.name: prop for prop in self.properties}
|
|
||||||
if (prop := self.properties.get('ki_fp_filters')):
|
|
||||||
prop.value = prop.value.split() if prop.value else []
|
|
||||||
|
|
||||||
for unit in self.raw_units:
|
|
||||||
if unit.unit_global or unit.style_global:
|
|
||||||
d = self.global_units.get(unit.demorgan_style, {})
|
|
||||||
d[unit.name] = unit
|
|
||||||
self.global_units[unit.demorgan_style] = d
|
|
||||||
|
|
||||||
for other in self.raw_units:
|
|
||||||
if other.unit_global or other.style_global or other == unit:
|
|
||||||
continue
|
|
||||||
if not (unit.unit_global or other.name == unit.name):
|
|
||||||
continue
|
|
||||||
if not (unit.style_global or other.demorgan_style == unit.demorgan_style):
|
|
||||||
continue
|
|
||||||
other.global_units.append(unit)
|
|
||||||
|
|
||||||
else:
|
|
||||||
d = self.styles.get(unit.demorgan_style, {})
|
|
||||||
d[unit.name] = unit
|
|
||||||
self.styles[unit.demorgan_style] = d
|
|
||||||
|
|
||||||
def __before_sexp__(self):
|
|
||||||
self.raw_units = ([unit for style in self.global_units.values() for unit in style.values()] +
|
|
||||||
[unit for style in self.styles.values() for unit in style.values()])
|
|
||||||
if (prop := self.properties.get('ki_fp_filters')):
|
|
||||||
if not isinstance(prop.value, str):
|
|
||||||
prop.value = ' '.join(prop.value)
|
|
||||||
self.properties = list(self.properties.values())
|
|
||||||
|
|
||||||
def default_properties(self):
|
|
||||||
for i, (name, value, hide) in enumerate([
|
|
||||||
('Reference', 'U', False),
|
|
||||||
('Value', None, False),
|
|
||||||
('Footprint', None, True),
|
|
||||||
('Datasheet', None, True),
|
|
||||||
('ki_locked', None, True),
|
|
||||||
('ki_keywords', None, True),
|
|
||||||
('ki_description', None, True),
|
|
||||||
('ki_fp_filters', None, False),
|
|
||||||
]):
|
|
||||||
self.properties[name] = Property(name=name, value=value, id=i, effects=TextEffect(hide=hide))
|
|
||||||
|
|
||||||
def units(self, demorgan_style=None):
|
|
||||||
if self.extends:
|
|
||||||
return self.library[self.extends].units(demorgan_style)
|
|
||||||
else:
|
|
||||||
return self.styles.get(demorgan_style or 'default', {})
|
|
||||||
|
|
||||||
def get_center_rectangle(self, units):
|
|
||||||
# return a polyline for the requested unit that is a rectangle
|
|
||||||
# and is closest to the center
|
|
||||||
candidates = {}
|
|
||||||
# building a dict with floats as keys.. there needs to be a rule against that^^
|
|
||||||
pl_rects = [i.as_polyline() for i in self.rectangles]
|
|
||||||
pl_rects.extend(pl for pl in self.polylines if pl.is_rectangle())
|
|
||||||
for pl in pl_rects:
|
|
||||||
if pl.unit in units:
|
|
||||||
# extract the center, calculate the distance to origin
|
|
||||||
(x, y) = pl.get_center_of_boundingbox()
|
|
||||||
dist = math.sqrt(x * x + y * y)
|
|
||||||
candidates[dist] = pl
|
|
||||||
|
|
||||||
if candidates:
|
|
||||||
# sort the list return the first (smallest) item
|
|
||||||
return candidates[sorted(candidates.keys())[0]]
|
|
||||||
return None
|
|
||||||
|
|
||||||
def is_graphic_symbol(self):
|
|
||||||
return self.extends is None and (
|
|
||||||
not self.pins or self.get_property("Reference").value == "#SYM"
|
|
||||||
)
|
|
||||||
|
|
||||||
def pins_by_name(self, demorgan_style=None):
|
|
||||||
pins = defaultdict(lambda: set())
|
|
||||||
for unit in self.units(demorgan_style):
|
|
||||||
for pin in unit.all_pins:
|
|
||||||
pins[pin.name].add(pin)
|
|
||||||
return pins
|
|
||||||
|
|
||||||
def pins_by_number(self, demorgan_style=None):
|
|
||||||
pins = defaultdict(lambda: set())
|
|
||||||
for unit in self.units(demorgan_style):
|
|
||||||
for pin in unit.all_pins:
|
|
||||||
pins[pin.number].add(pin)
|
|
||||||
return pins
|
|
||||||
|
|
||||||
def __getattr__(self, name):
|
|
||||||
if name.startswith('all_'):
|
|
||||||
return itertools.chain(getattr(unit, name) for unit in self.raw_units)
|
|
||||||
|
|
||||||
def filter_pins(self, name=None, direction=None, electrical_type=None):
|
|
||||||
for pin in self.all_pins:
|
|
||||||
if name and not fnmatch(pin.name, name):
|
|
||||||
continue
|
|
||||||
if direction and not pin.direction in direction:
|
|
||||||
continue
|
|
||||||
if electrical_type and not pin.etype in electical_type:
|
|
||||||
continue
|
|
||||||
yield pin
|
|
||||||
|
|
||||||
def heuristically_small(self):
|
|
||||||
""" Heuristically try to determine whether this is a "small" component like a resistor, capacitor, LED, diode,
|
|
||||||
or transistor etc. When we have at most two pins, or there is no filled rectangle as symbol outline and we have
|
|
||||||
3 or 4 pins, we assume this is a small symbol.
|
|
||||||
"""
|
|
||||||
if len(self.all_pins) <= 2:
|
|
||||||
return True
|
|
||||||
if len(self.all_pins) > 4:
|
|
||||||
return False
|
|
||||||
return bool(self.get_center_rectangle(range(self.unit_count)))
|
|
||||||
|
|
||||||
|
|
||||||
SUPPORTED_FILE_FORMAT_VERSIONS = [20211014, 20220914]
|
|
||||||
@sexp_type('kicad_symbol_lib')
|
|
||||||
class Library:
|
|
||||||
_version: Named(int, name='version') = 20211014
|
|
||||||
generator: Named(Atom) = Atom.kicad_library_utils
|
|
||||||
symbols: List(Symbol) = field(default_factory=list)
|
|
||||||
_ : SEXP_END = None
|
|
||||||
original_filename: str = None
|
|
||||||
|
|
||||||
@property
|
|
||||||
def version(self):
|
|
||||||
return self._version
|
|
||||||
|
|
||||||
@version.setter
|
|
||||||
def version(self, value):
|
|
||||||
if value not in SUPPORTED_FILE_FORMAT_VERSIONS:
|
|
||||||
raise FormatError(f'File format version {value} is not supported. Supported versions are {", ".join(map(str, SUPPORTED_FILE_FORMAT_VERSIONS))}.')
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def open(cls, filename: str):
|
|
||||||
with open(filename) as f:
|
|
||||||
return cls.parse(f.read())
|
|
||||||
|
|
||||||
def write(self, filename=None):
|
|
||||||
with open(filename or self.original_filename, 'w') as f:
|
|
||||||
f.write(build_sexp(sexp(self)))
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
if len(sys.argv) >= 2:
|
|
||||||
a = Library.open(sys.argv[1])
|
|
||||||
print(build_sexp(sexp(a)))
|
|
||||||
else:
|
|
||||||
print("pass a .kicad_sym file please")
|
|
||||||
|
|
@ -1,739 +0,0 @@
|
||||||
|
|
||||||
import sys
|
|
||||||
import re
|
|
||||||
import math
|
|
||||||
import string
|
|
||||||
import itertools
|
|
||||||
from copy import copy, deepcopy
|
|
||||||
import warnings
|
|
||||||
import importlib.resources
|
|
||||||
|
|
||||||
from ..utils import MM, rotate_point
|
|
||||||
from .primitives import *
|
|
||||||
from ..graphic_objects import Region
|
|
||||||
from ..apertures import RectangleAperture, CircleAperture, ApertureMacroInstance
|
|
||||||
from ..aperture_macros.parse import ApertureMacro, VariableExpression
|
|
||||||
from ..aperture_macros import primitive as amp
|
|
||||||
from .kicad import footprints as kfp
|
|
||||||
from . import data as package_data
|
|
||||||
|
|
||||||
|
|
||||||
class ProtoBoard(Board):
|
|
||||||
def __init__(self, w, h, content, margin=None, corner_radius=None, mounting_hole_dia=None, mounting_hole_offset=None, unit=MM):
|
|
||||||
corner_radius = corner_radius or unit(1.5, MM)
|
|
||||||
super().__init__(w, h, corner_radius, unit=unit)
|
|
||||||
self.margin = margin or unit(2, MM)
|
|
||||||
self.content = content
|
|
||||||
|
|
||||||
if mounting_hole_dia:
|
|
||||||
mounting_hole_offset = mounting_hole_offset or mounting_hole_dia*2
|
|
||||||
ko = mounting_hole_offset*2
|
|
||||||
|
|
||||||
self.add(Hole(mounting_hole_offset, mounting_hole_offset, mounting_hole_dia, unit=unit))
|
|
||||||
self.add(Hole(w-mounting_hole_offset, mounting_hole_offset, mounting_hole_dia, unit=unit))
|
|
||||||
self.add(Hole(mounting_hole_offset, h-mounting_hole_offset, mounting_hole_dia, unit=unit))
|
|
||||||
self.add(Hole(w-mounting_hole_offset, h-mounting_hole_offset, mounting_hole_dia, unit=unit))
|
|
||||||
|
|
||||||
self.keepouts.append(((0, 0), (ko, ko)))
|
|
||||||
self.keepouts.append(((w-ko, 0), (w, ko)))
|
|
||||||
self.keepouts.append(((0, h-ko), (ko, h)))
|
|
||||||
self.keepouts.append(((w-ko, h-ko), (w, h)))
|
|
||||||
|
|
||||||
self.generate()
|
|
||||||
|
|
||||||
def generate(self, unit=MM):
|
|
||||||
bbox = ((self.margin, self.margin), (self.w-self.margin, self.h-self.margin))
|
|
||||||
bbox = unit.convert_bounds_from(self.unit, bbox)
|
|
||||||
for obj in self.content.generate(bbox, (True, True, True, True), unit):
|
|
||||||
self.add(obj, keepout_errors='skip')
|
|
||||||
|
|
||||||
|
|
||||||
class PropLayout:
|
|
||||||
def __init__(self, content, direction, proportions):
|
|
||||||
self.content = list(content)
|
|
||||||
if direction not in ('h', 'v'):
|
|
||||||
raise ValueError('direction must be one of "h", or "v".')
|
|
||||||
self.direction = direction
|
|
||||||
self.proportions = list(proportions)
|
|
||||||
if len(content) != len(proportions):
|
|
||||||
raise ValueError('proportions and content must have same length')
|
|
||||||
|
|
||||||
def generate(self, bbox, border_text, unit=MM):
|
|
||||||
for i, (bbox, child) in enumerate(self.layout_2d(bbox, unit)):
|
|
||||||
first = bool(i == 0)
|
|
||||||
last = bool(i == len(self.content)-1)
|
|
||||||
yield from child.generate(bbox, (
|
|
||||||
border_text[0] and (last or self.direction == 'h'),
|
|
||||||
border_text[1] and (last or self.direction == 'v'),
|
|
||||||
border_text[2] and (first or self.direction == 'h'),
|
|
||||||
border_text[3] and (first or self.direction == 'v'),
|
|
||||||
), unit)
|
|
||||||
|
|
||||||
def fit_size(self, w, h, unit=MM):
|
|
||||||
widths = []
|
|
||||||
heights = []
|
|
||||||
for ((x_min, y_min), (x_max, y_max)), child in self.layout_2d(((0, 0), (w, h)), unit):
|
|
||||||
if not isinstance(child, EmptyProtoArea):
|
|
||||||
widths.append(x_max - x_min)
|
|
||||||
heights.append(y_max - y_min)
|
|
||||||
if self.direction == 'h':
|
|
||||||
return sum(widths), max(heights, default=0)
|
|
||||||
else:
|
|
||||||
return max(widths, default=0), sum(heights)
|
|
||||||
|
|
||||||
def layout_2d(self, bbox, unit=MM):
|
|
||||||
(x, y), (w, h) = bbox
|
|
||||||
w, h = w-x, h-y
|
|
||||||
|
|
||||||
actual_l = 0
|
|
||||||
target_l = 0
|
|
||||||
|
|
||||||
for l, child in zip(self.layout(w if self.direction == 'h' else h, unit), self.content):
|
|
||||||
this_x, this_y = x, y
|
|
||||||
this_w, this_h = w, h
|
|
||||||
target_l += l
|
|
||||||
|
|
||||||
if self.direction == 'h':
|
|
||||||
this_w = target_l - actual_l
|
|
||||||
else:
|
|
||||||
this_h = target_l - actual_l
|
|
||||||
|
|
||||||
this_w, this_h = child.fit_size(this_w, this_h, unit)
|
|
||||||
|
|
||||||
if self.direction == 'h':
|
|
||||||
x += this_w
|
|
||||||
actual_l += this_w
|
|
||||||
this_h = h
|
|
||||||
else:
|
|
||||||
y += this_h
|
|
||||||
actual_l += this_h
|
|
||||||
this_w = w
|
|
||||||
|
|
||||||
yield ((this_x, this_y), (this_x+this_w, this_y+this_h)), child
|
|
||||||
|
|
||||||
def layout(self, length, unit=MM):
|
|
||||||
out = [ eval_value(value, MM(length, unit)) for value in self.proportions ]
|
|
||||||
total_length = sum(value for value in out if value is not None)
|
|
||||||
if length - total_length < -1e-6:
|
|
||||||
raise ValueError(f'Proportions sum to {total_length} mm, which is greater than the available space of {length} mm.')
|
|
||||||
|
|
||||||
leftover = length - total_length
|
|
||||||
sum_props = sum( (value or 1.0) for value in self.proportions if not isinstance(value, str) )
|
|
||||||
return [ unit(leftover * (value or 1.0) / sum_props if not isinstance(value, str) else calculated, MM)
|
|
||||||
for value, calculated in zip(self.proportions, out) ]
|
|
||||||
|
|
||||||
@property
|
|
||||||
def single_sided(self):
|
|
||||||
return all(elem.single_sided for elem in self.content)
|
|
||||||
|
|
||||||
def __str__(self):
|
|
||||||
children = ', '.join( f'{elem}:{width}' for elem, width in zip(self.content, self.proportions))
|
|
||||||
return f'PropLayout[{self.direction.upper()}]({children})'
|
|
||||||
|
|
||||||
|
|
||||||
class TwoSideLayout:
|
|
||||||
def __init__(self, top, bottom):
|
|
||||||
self.top, self.bottom = top, bottom
|
|
||||||
|
|
||||||
if not top.single_sided or not bottom.single_sided:
|
|
||||||
warnings.warn('Two-sided pattern used on one side of a TwoSideLayout')
|
|
||||||
|
|
||||||
def fit_size(self, w, h, unit=MM):
|
|
||||||
w1, h1 = self.top.fit_size(w, h, unit)
|
|
||||||
w2, h2 = self.bottom.fit_size(w, h, unit)
|
|
||||||
if isinstance(self.top, EmptyProtoArea):
|
|
||||||
if isinstance(self.bottom, EmptyProtoArea):
|
|
||||||
return w1, h1
|
|
||||||
return w2, h2
|
|
||||||
if isinstance(self.bottom, EmptyProtoArea):
|
|
||||||
return w1, h1
|
|
||||||
return max(w1, w2), max(h1, h2)
|
|
||||||
|
|
||||||
def generate(self, bbox, border_text, unit=MM):
|
|
||||||
yield from self.top.generate(bbox, border_text, unit)
|
|
||||||
for obj in self.bottom.generate(bbox, border_text, unit):
|
|
||||||
obj.side = 'bottom'
|
|
||||||
yield obj
|
|
||||||
|
|
||||||
|
|
||||||
def numeric(start=1):
|
|
||||||
def gen():
|
|
||||||
nonlocal start
|
|
||||||
for i in itertools.count(start):
|
|
||||||
yield str(i)
|
|
||||||
|
|
||||||
return gen
|
|
||||||
|
|
||||||
|
|
||||||
def alphabetic(case='upper'):
|
|
||||||
if case not in ('lower', 'upper'):
|
|
||||||
raise ValueError('case must be one of "lower" or "upper".')
|
|
||||||
|
|
||||||
index = string.ascii_lowercase if case == 'lower' else string.ascii_uppercase
|
|
||||||
|
|
||||||
def gen():
|
|
||||||
nonlocal index
|
|
||||||
|
|
||||||
for i in itertools.count():
|
|
||||||
if i<26:
|
|
||||||
yield index[i]
|
|
||||||
continue
|
|
||||||
|
|
||||||
i -= 26
|
|
||||||
if i<26*26:
|
|
||||||
yield index[i//26] + index[i%26]
|
|
||||||
continue
|
|
||||||
|
|
||||||
i -= 26*26
|
|
||||||
if i<26*26*26:
|
|
||||||
yield index[i//(26*26)] + index[(i//26)%26] + index[i%26]
|
|
||||||
|
|
||||||
else:
|
|
||||||
raise ValueError('row/column index out of range')
|
|
||||||
|
|
||||||
return gen
|
|
||||||
|
|
||||||
|
|
||||||
class PatternProtoArea:
|
|
||||||
def __init__(self, pitch_x, pitch_y=None, obj=None, numbers=True, font_size=None, font_stroke=None, number_x_gen=alphabetic(), number_y_gen=numeric(), interval_x=5, interval_y=None, margin=0, unit=MM):
|
|
||||||
self.pitch_x = pitch_x
|
|
||||||
self.pitch_y = pitch_y or pitch_x
|
|
||||||
self.margin = margin
|
|
||||||
self.obj = obj
|
|
||||||
self.unit = unit
|
|
||||||
self.numbers = numbers
|
|
||||||
self.font_size = font_size or unit(1.0, MM)
|
|
||||||
self.font_stroke = font_stroke or unit(0.2, MM)
|
|
||||||
self.interval_x = interval_x
|
|
||||||
self.interval_y = interval_y or (1 if MM(self.pitch_y, unit) >= 2.0 else 5)
|
|
||||||
self.number_x_gen, self.number_y_gen = number_x_gen, number_y_gen
|
|
||||||
|
|
||||||
def fit_size(self, w, h, unit=MM):
|
|
||||||
(min_x, min_y), (max_x, max_y) = self.fit_rect(((0, 0), (max(0, w-2*self.margin), max(0, h-2*self.margin))))
|
|
||||||
return max_x-min_x + 2*self.margin, max_y-min_y + 2*self.margin
|
|
||||||
|
|
||||||
def fit_rect(self, bbox, unit=MM):
|
|
||||||
(x, y), (w, h) = bbox
|
|
||||||
x, y = x+self.margin, y+self.margin
|
|
||||||
w, h = w-x-self.margin, h-y-self.margin
|
|
||||||
|
|
||||||
w_mod = round((w + 5e-7) % unit(self.pitch_x, self.unit), 6)
|
|
||||||
h_mod = round((h + 5e-7) % unit(self.pitch_y, self.unit), 6)
|
|
||||||
w_fit, h_fit = round(w - w_mod, 6), round(h - h_mod, 6)
|
|
||||||
|
|
||||||
x = x + (w-w_fit)/2
|
|
||||||
y = y + (h-h_fit)/2
|
|
||||||
return (x, y), (x+w_fit, y+h_fit)
|
|
||||||
|
|
||||||
def generate(self, bbox, border_text, unit=MM):
|
|
||||||
(x, y), (w, h) = bbox
|
|
||||||
w, h = w-x, h-y
|
|
||||||
|
|
||||||
n_x = int(w//unit(self.pitch_x, self.unit))
|
|
||||||
n_y = int(h//unit(self.pitch_y, self.unit))
|
|
||||||
off_x = (w % unit(self.pitch_x, self.unit)) / 2
|
|
||||||
off_y = (h % unit(self.pitch_y, self.unit)) / 2
|
|
||||||
|
|
||||||
if self.numbers:
|
|
||||||
for i, lno_i in list(zip(range(n_y), self.number_y_gen())):
|
|
||||||
if i == 0 or i == n_y - 1 or (i+1) % self.interval_y == 0:
|
|
||||||
t_y = off_y + y + (n_y - 1 - i + 0.5) * self.pitch_y
|
|
||||||
|
|
||||||
if border_text[3]:
|
|
||||||
t_x = x + off_x
|
|
||||||
yield Text(t_x, t_y, lno_i, self.font_size, self.font_stroke, 'right', 'middle', unit=self.unit)
|
|
||||||
if not self.single_sided:
|
|
||||||
yield Text(t_x, t_y, lno_i, self.font_size, self.font_stroke, 'right', 'middle', side='bottom', unit=self.unit)
|
|
||||||
|
|
||||||
if border_text[1]:
|
|
||||||
t_x = x + w - off_x
|
|
||||||
yield Text(t_x, t_y, lno_i, self.font_size, self.font_stroke, 'left', 'middle', unit=self.unit)
|
|
||||||
if not self.single_sided:
|
|
||||||
yield Text(t_x, t_y, lno_i, self.font_size, self.font_stroke, 'left', 'middle', side='bottom', unit=self.unit)
|
|
||||||
|
|
||||||
for i, lno_i in zip(range(n_x), self.number_x_gen()):
|
|
||||||
if i == 0 or i == n_x - 1 or (i+1) % self.interval_x == 0:
|
|
||||||
t_x = off_x + x + (i + 0.5) * self.pitch_x
|
|
||||||
|
|
||||||
if border_text[2]:
|
|
||||||
t_y = y + off_y
|
|
||||||
yield Text(t_x, t_y, lno_i, self.font_size, self.font_stroke, 'center', 'top', unit=self.unit)
|
|
||||||
if not self.single_sided:
|
|
||||||
yield Text(t_x, t_y, lno_i, self.font_size, self.font_stroke, 'center', 'top', side='bottom', unit=self.unit)
|
|
||||||
|
|
||||||
if border_text[0]:
|
|
||||||
t_y = y + h - off_y
|
|
||||||
yield Text(t_x, t_y, lno_i, self.font_size, self.font_stroke, 'center', 'bottom', unit=self.unit)
|
|
||||||
if not self.single_sided:
|
|
||||||
yield Text(t_x, t_y, lno_i, self.font_size, self.font_stroke, 'center', 'bottom', side='bottom', unit=self.unit)
|
|
||||||
|
|
||||||
|
|
||||||
for i in range(n_x):
|
|
||||||
for j in range(n_y):
|
|
||||||
if hasattr(self.obj, 'inst'):
|
|
||||||
inst = self.obj.inst(i, j, i == n_x-1, j == n_y-1)
|
|
||||||
if not inst:
|
|
||||||
continue
|
|
||||||
else:
|
|
||||||
inst = copy(self.obj)
|
|
||||||
|
|
||||||
inst.x = inst.unit(off_x + x, unit) + (i + 0.5) * inst.unit(self.pitch_x, self.unit)
|
|
||||||
inst.y = inst.unit(off_y + y, unit) + (j + 0.5) * inst.unit(self.pitch_y, self.unit)
|
|
||||||
yield inst
|
|
||||||
|
|
||||||
@property
|
|
||||||
def single_sided(self):
|
|
||||||
return self.obj.single_sided
|
|
||||||
|
|
||||||
|
|
||||||
class EmptyProtoArea:
|
|
||||||
def __init__(self, copper_fill=False):
|
|
||||||
self.copper_fill = copper_fill
|
|
||||||
|
|
||||||
def fit_size(self, w, h, unit=MM):
|
|
||||||
return w, h
|
|
||||||
|
|
||||||
def generate(self, bbox, border_text, unit=MM):
|
|
||||||
if self.copper_fill:
|
|
||||||
(min_x, min_y), (max_x, max_y) = bbox
|
|
||||||
group = ObjectGroup(0, 0, top_copper=[Region([(min_x, min_y), (max_x, min_y), (max_x, max_y), (min_x, max_y)],
|
|
||||||
unit=unit, polarity_dark=True)])
|
|
||||||
group.bounding_box = lambda *args, **kwargs: None
|
|
||||||
yield group
|
|
||||||
|
|
||||||
@property
|
|
||||||
def single_sided(self):
|
|
||||||
return True
|
|
||||||
|
|
||||||
|
|
||||||
class ManhattanPads(ObjectGroup):
|
|
||||||
def __init__(self, w, h=None, gap=0.2, unit=MM):
|
|
||||||
super().__init__(0, 0)
|
|
||||||
h = h or w
|
|
||||||
self.gap = gap
|
|
||||||
self.unit = unit
|
|
||||||
|
|
||||||
p = (w-2*gap)/2
|
|
||||||
q = (h-2*gap)/2
|
|
||||||
small_ap = RectangleAperture(p, q, unit=unit)
|
|
||||||
|
|
||||||
s = min(w, h) / 2 / math.sqrt(2)
|
|
||||||
large_ap = RectangleAperture(s, s, unit=unit).rotated(math.pi/4)
|
|
||||||
large_ap_neg = RectangleAperture(s+2*gap, s+2*gap, unit=unit).rotated(math.pi/4)
|
|
||||||
|
|
||||||
a = gap/2 + p/2
|
|
||||||
b = gap/2 + q/2
|
|
||||||
|
|
||||||
self.top_copper.append(Flash(-a, -b, aperture=small_ap, unit=unit))
|
|
||||||
self.top_copper.append(Flash(-a, b, aperture=small_ap, unit=unit))
|
|
||||||
self.top_copper.append(Flash( a, -b, aperture=small_ap, unit=unit))
|
|
||||||
self.top_copper.append(Flash( a, b, aperture=small_ap, unit=unit))
|
|
||||||
self.top_copper.append(Flash(0, 0, aperture=large_ap_neg, polarity_dark=False, unit=unit))
|
|
||||||
self.top_copper.append(Flash(0, 0, aperture=large_ap, unit=unit))
|
|
||||||
self.top_mask = self.top_copper
|
|
||||||
|
|
||||||
|
|
||||||
class RFGroundProto(ObjectGroup):
|
|
||||||
def __init__(self, pitch=None, drill=None, clearance=None, via_dia=None, via_drill=None, pad_dia=None, trace_width=None, unit=MM):
|
|
||||||
super().__init__(0, 0)
|
|
||||||
self.unit = unit
|
|
||||||
self.pitch = pitch = pitch or unit(2.54, MM)
|
|
||||||
self.drill = drill = drill or unit(0.9, MM)
|
|
||||||
self.clearance = clearance = clearance or unit(0.3, MM)
|
|
||||||
self.via_drill = via_drill = via_drill or unit(0.4, MM)
|
|
||||||
self.via_dia = via_dia = via_dia or unit(0.8, MM)
|
|
||||||
|
|
||||||
if pad_dia is None:
|
|
||||||
self.trace_width = trace_width = trace_width or unit(0.3, MM)
|
|
||||||
pad_dia = pitch - trace_width - 2*clearance
|
|
||||||
elif trace_width is None:
|
|
||||||
trace_width = pitch - pad_dia - 2*clearance
|
|
||||||
self.pad_dia = pad_dia
|
|
||||||
|
|
||||||
via_ap = RectangleAperture(via_dia, via_dia, unit=unit).rotated(math.pi/4)
|
|
||||||
pad_ap = CircleAperture(pad_dia, unit=unit)
|
|
||||||
pad_neg_ap = CircleAperture(pad_dia+2*clearance, unit=unit)
|
|
||||||
ground_ap = RectangleAperture(pitch + unit(0.01, MM), pitch + unit(0.01, MM), unit=unit)
|
|
||||||
pad_drill = ExcellonTool(drill, plated=True, unit=unit)
|
|
||||||
via_drill = ExcellonTool(via_drill, plated=True, unit=unit)
|
|
||||||
|
|
||||||
self.top_copper.append(Flash(0, 0, aperture=ground_ap, unit=unit))
|
|
||||||
self.top_copper.append(Flash(0, 0, aperture=pad_neg_ap, polarity_dark=False, unit=unit))
|
|
||||||
self.top_copper.append(Flash(0, 0, aperture=pad_ap, unit=unit))
|
|
||||||
self.top_mask.append(Flash(0, 0, aperture=pad_ap, unit=unit))
|
|
||||||
self.top_copper.append(Flash(pitch/2, pitch/2, aperture=via_ap, unit=unit))
|
|
||||||
self.top_mask.append(Flash(pitch/2, pitch/2, aperture=via_ap, unit=unit))
|
|
||||||
self.drill_pth.append(Flash(0, 0, aperture=pad_drill, unit=unit))
|
|
||||||
self.drill_pth.append(Flash(pitch/2, pitch/2, aperture=via_drill, unit=unit))
|
|
||||||
|
|
||||||
self.bottom_copper = self.top_copper
|
|
||||||
self.bottom_mask = self.top_mask
|
|
||||||
|
|
||||||
def inst(self, x, y, border_x, border_y):
|
|
||||||
inst = copy(self)
|
|
||||||
if border_x or border_y:
|
|
||||||
inst.drill_pth = inst.drill_pth[:-1]
|
|
||||||
inst.top_copper = inst.bottom_copper = inst.top_copper[:-1]
|
|
||||||
inst.top_mask = inst.bottom_mask = inst.top_mask[:-1]
|
|
||||||
return inst
|
|
||||||
|
|
||||||
|
|
||||||
class THTFlowerProto(ObjectGroup):
|
|
||||||
def __init__(self, pitch=None, drill=None, diameter=None, unit=MM):
|
|
||||||
super().__init__(0, 0, unit=unit)
|
|
||||||
self.pitch = pitch = pitch or unit(2.54, MM)
|
|
||||||
drill = drill or unit(0.9, MM)
|
|
||||||
diameter = diameter or unit(2.0, MM)
|
|
||||||
|
|
||||||
p = pitch / 2
|
|
||||||
self.objects.append(THTPad.circle(-p, 0, drill, diameter, paste=False, unit=unit))
|
|
||||||
self.objects.append(THTPad.circle( p, 0, drill, diameter, paste=False, unit=unit))
|
|
||||||
self.objects.append(THTPad.circle(0, -p, drill, diameter, paste=False, unit=unit))
|
|
||||||
self.objects.append(THTPad.circle(0, p, drill, diameter, paste=False, unit=unit))
|
|
||||||
|
|
||||||
middle_ap = CircleAperture(diameter, unit=unit)
|
|
||||||
self.top_copper.append(Flash(0, 0, aperture=middle_ap, unit=unit))
|
|
||||||
self.bottom_copper = self.top_mask = self.bottom_mask = self.top_copper
|
|
||||||
|
|
||||||
def inst(self, x, y, border_x, border_y):
|
|
||||||
if (x % 2 == 0) and (y % 2 == 0):
|
|
||||||
return copy(self)
|
|
||||||
|
|
||||||
if (x % 2 == 1) and (y % 2 == 1):
|
|
||||||
return copy(self)
|
|
||||||
|
|
||||||
return None
|
|
||||||
|
|
||||||
def bounding_box(self, unit=MM):
|
|
||||||
x, y, rotation = self.abs_pos
|
|
||||||
p = self.pitch/2
|
|
||||||
return unit.convert_bounds_from(self.unit, ((x-p, y-p), (x+p, y+p)))
|
|
||||||
|
|
||||||
class PoweredProto(ObjectGroup):
|
|
||||||
""" Cell primitive for "powered" THT breadboards. This cell type is based on regular THT pads in a 100 mil grid, but
|
|
||||||
adds small SMD pads diagonally between the THT pads. These SMD pads are interconnected with traces and vias in such
|
|
||||||
a way that every second one is inter-linked, forming two fully connected grids. Next to every THT pad you have one
|
|
||||||
pad of each grid, so this layout is awesome for distributing power throughout the board.
|
|
||||||
|
|
||||||
This design is based on one that Yajima Manufacturing Akizuki Denshi, Akihabara's finest electronics store sells for
|
|
||||||
next to nothing. Sadly, they don't ship internationally and they don't even have an English website, but if you ever
|
|
||||||
are in Akihabara, Tokyo, Japan I can *highly* recommend a visit. The ones Yajima make are better than what this will
|
|
||||||
produce since the Yajima ones use a two-colored silkscreen to visually distinguish the two power pad grids.
|
|
||||||
|
|
||||||
Links:
|
|
||||||
Akizuki Denshi product page: https://akizukidenshi.com/catalog/g/gP-07214/
|
|
||||||
Yajima Manufacturing Corporation website: http://www.yajima-works.co.jp/index.html
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self, pitch=None, drill=None, clearance=None, power_pad_dia=None, via_size=None, trace_width=None, unit=MM):
|
|
||||||
super().__init__(0, 0)
|
|
||||||
self.unit = unit
|
|
||||||
self.pitch = pitch = pitch or unit(2.54, MM)
|
|
||||||
self.drill = drill = drill or unit(0.9, MM)
|
|
||||||
self.clearance = clearance = clearance or unit(0.3, MM)
|
|
||||||
self.trace_width = trace_width = trace_width or unit(0.3, MM)
|
|
||||||
self.via_size = via_size = via_size or unit(0.4, MM)
|
|
||||||
|
|
||||||
main_pad_dia = pitch - trace_width - 2*clearance
|
|
||||||
power_pad_dia_max = math.sqrt(2)*pitch - main_pad_dia - 2*clearance
|
|
||||||
if power_pad_dia is None:
|
|
||||||
power_pad_dia = power_pad_dia_max - clearance # reduce some more to give the user more room
|
|
||||||
elif power_pad_dia > power_pad_dia_max:
|
|
||||||
warnings.warn(f'Power pad diameter {power_pad_dia} > {power_pad_dia_max} violates pad-to-pad clearance')
|
|
||||||
self.power_pad_dia = power_pad_dia
|
|
||||||
|
|
||||||
main_ap = CircleAperture(main_pad_dia, unit=unit)
|
|
||||||
power_ap = CircleAperture(self.power_pad_dia, unit=unit)
|
|
||||||
|
|
||||||
for l in [self.top_copper, self.bottom_copper]:
|
|
||||||
l.append(Flash(0, 0, aperture=main_ap, unit=unit))
|
|
||||||
|
|
||||||
l.append(Flash(-pitch/2, -pitch/2, aperture=power_ap, unit=unit))
|
|
||||||
l.append(Flash(-pitch/2, pitch/2, aperture=power_ap, unit=unit))
|
|
||||||
l.append(Flash( pitch/2, -pitch/2, aperture=power_ap, unit=unit))
|
|
||||||
l.append(Flash( pitch/2, pitch/2, aperture=power_ap, unit=unit))
|
|
||||||
|
|
||||||
self.drill_pth.append(Flash(0, 0, ExcellonTool(drill, plated=True, unit=unit), unit=unit))
|
|
||||||
self.drill_pth.append(Flash(-pitch/2, -pitch/2, ExcellonTool(via_size, plated=True, unit=unit), unit=unit))
|
|
||||||
|
|
||||||
self.top_mask = copy(self.top_copper)
|
|
||||||
self.bottom_mask = copy(self.bottom_copper)
|
|
||||||
|
|
||||||
self.line_ap = CircleAperture(trace_width, unit=unit)
|
|
||||||
self.top_copper.append(Line(-pitch/2, -pitch/2, -pitch/2, pitch/2, aperture=self.line_ap, unit=unit))
|
|
||||||
self.top_copper.append(Line(pitch/2, -pitch/2, pitch/2, pitch/2, aperture=self.line_ap, unit=unit))
|
|
||||||
self.bottom_copper.append(Line(-pitch/2, -pitch/2, pitch/2, -pitch/2, aperture=self.line_ap, unit=unit))
|
|
||||||
self.bottom_copper.append(Line(-pitch/2, pitch/2, pitch/2, pitch/2, aperture=self.line_ap, unit=unit))
|
|
||||||
|
|
||||||
def inst(self, x, y, border_x, border_y):
|
|
||||||
inst = copy(self)
|
|
||||||
if (x + y) % 2 == 0:
|
|
||||||
inst.drill_pth = inst.drill_pth[:-1]
|
|
||||||
|
|
||||||
c = self.power_pad_dia/2 + self.clearance
|
|
||||||
p = self.pitch/2
|
|
||||||
|
|
||||||
if x == 1:
|
|
||||||
inst.top_silk = [Line(-p, -p+c, -p, p-c, aperture=self.line_ap, unit=self.unit)]
|
|
||||||
elif x % 2 == 0:
|
|
||||||
inst.top_silk = [Line(p, -p+c, p, p-c, aperture=self.line_ap, unit=self.unit)]
|
|
||||||
|
|
||||||
if y == 0:
|
|
||||||
inst.bottom_silk = [Line(-p+c, -p, p-c, -p, aperture=self.line_ap, unit=self.unit)]
|
|
||||||
elif y % 2 == 1:
|
|
||||||
inst.bottom_silk = [Line(-p+c, p, p-c, p, aperture=self.line_ap, unit=self.unit)]
|
|
||||||
|
|
||||||
return inst
|
|
||||||
|
|
||||||
def bounding_box(self, unit=MM):
|
|
||||||
x, y, rotation = self.abs_pos
|
|
||||||
p = self.pitch/2
|
|
||||||
return unit.convert_bounds_from(self.unit, ((x-p, y-p), (x+p, y+p)))
|
|
||||||
|
|
||||||
|
|
||||||
class SpikyProto(ObjectGroup):
|
|
||||||
""" Cell primitive for the "spiky" protoboard designed by @electroniceel and published on github at the URL below.
|
|
||||||
This layout has small-ish standard THT pads, but in between these pads it puts a grid of SMD pads that are designed
|
|
||||||
for easy solder bridging to allow for the construction of traces from solder bridging.
|
|
||||||
|
|
||||||
Github URL: https://github.com/electroniceel/protoboard
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self, pitch=None, drill=None, clearance=None, power_pad_dia=None, via_size=None, trace_width=None, unit=MM):
|
|
||||||
super().__init__(0, 0, unit=unit)
|
|
||||||
res = importlib.resources.files(package_data)
|
|
||||||
|
|
||||||
self.fp_center = kfp.Footprint.load(res.joinpath('center-pad-spikes.kicad_mod').read_text(encoding='utf-8'))
|
|
||||||
self.corner_pad = kfp.FootprintInstance(1.27, 1.27, self.fp_center, unit=MM)
|
|
||||||
|
|
||||||
self.pad = kfp.Footprint.load(res.joinpath('tht-0.8.kicad_mod').read_text(encoding='utf-8'))
|
|
||||||
self.center_pad = kfp.FootprintInstance(0, 0, self.pad, unit=MM)
|
|
||||||
|
|
||||||
self.fp_between = kfp.Footprint.load(res.joinpath('pad-between-spiked.kicad_mod').read_text(encoding='utf-8'))
|
|
||||||
self.right_pad = kfp.FootprintInstance(1.27, 0, self.fp_between, unit=MM)
|
|
||||||
self.top_pad = kfp.FootprintInstance(0, 1.27, self.fp_between, rotation=math.pi/2, unit=MM)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def objects(self):
|
|
||||||
return [x for x in (self.center_pad, self.corner_pad, self.right_pad, self.top_pad) if x is not None]
|
|
||||||
|
|
||||||
@objects.setter
|
|
||||||
def objects(self, value):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def inst(self, x, y, border_x, border_y):
|
|
||||||
inst = copy(self)
|
|
||||||
|
|
||||||
if border_x:
|
|
||||||
inst.corner_pad = inst.right_pad = None
|
|
||||||
|
|
||||||
if border_y:
|
|
||||||
inst.corner_pad = inst.top_pad = None
|
|
||||||
|
|
||||||
return inst
|
|
||||||
|
|
||||||
|
|
||||||
class AlioCell(ObjectGroup):
|
|
||||||
""" Cell primitive for the ALio protoboard designed by arief ibrahim adha and published on hackaday.io at the URL
|
|
||||||
below. Similar to electroniceel's spiky protoboard, this layout has small-ish standard THT pads, but in between
|
|
||||||
these pads it puts a grid of SMD pads that are designed for easy solder bridging to allow for the construction of
|
|
||||||
traces from solder bridging.
|
|
||||||
|
|
||||||
Hackaday.io URL: https://hackaday.io/project/28570/
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self, pitch=None, drill=None, clearance=None, link_pad_width=None, link_trace_width=None, via_size=None, unit=MM):
|
|
||||||
super().__init__(0, 0, unit=unit)
|
|
||||||
self.pitch = pitch or unit(2.54, MM)
|
|
||||||
self.drill = drill or unit(0.9, MM)
|
|
||||||
self.clearance = clearance or unit(0.3, MM)
|
|
||||||
self.link_pad_width = link_pad_width or unit(1.1, MM)
|
|
||||||
self.link_trace_width = link_trace_width or unit(0.5, MM)
|
|
||||||
self.via_size = via_size or unit(0.4, MM)
|
|
||||||
self.border_x, self.border_y = False, False
|
|
||||||
self.inst_x, self.inst_y = None, None
|
|
||||||
|
|
||||||
@property
|
|
||||||
def single_sided(self):
|
|
||||||
return False
|
|
||||||
|
|
||||||
def inst(self, x, y, border_x, border_y):
|
|
||||||
inst = copy(self)
|
|
||||||
inst.border_x, inst.border_y = border_x, border_y
|
|
||||||
inst.inst_x, inst.inst_y = x, y
|
|
||||||
return inst
|
|
||||||
|
|
||||||
def bounding_box(self, unit):
|
|
||||||
x, y, rotation = self.abs_pos
|
|
||||||
# FIXME hack
|
|
||||||
return self.unit.convert_bounds_to(unit, ((x-self.pitch/2, y-self.pitch/2), (x+self.pitch/2, y+self.pitch/2)))
|
|
||||||
|
|
||||||
def render(self, layer_stack, cache=None):
|
|
||||||
x, y, rotation = self.abs_pos
|
|
||||||
def xf(fe):
|
|
||||||
fe = copy(fe)
|
|
||||||
fe.rotate(rotation)
|
|
||||||
fe.offset(x, y, self.unit)
|
|
||||||
return fe
|
|
||||||
|
|
||||||
var = VariableExpression
|
|
||||||
# parameters: [1: total height = pad width, 2: pitch, 3: trace width, 4: corner radius, 5: rotation, 6: clearance]
|
|
||||||
alio_main_macro = ApertureMacro('ALIOM', (
|
|
||||||
amp.CenterLine(MM, 1, var(2)-var(6), var(2)-var(3)-2*var(6), 0, 0, var(5)),
|
|
||||||
amp.Outline(MM, 0, 5, (
|
|
||||||
-var(2)/2, -var(2)/2,
|
|
||||||
-var(2)/2, -(var(7)-var(8)),
|
|
||||||
-var(7), -(var(7)-var(8)),
|
|
||||||
-(var(7)-var(8)), -var(7),
|
|
||||||
-(var(7)-var(8)), -var(2)/2,
|
|
||||||
-var(2)/2, -var(2)/2,
|
|
||||||
), var(5)),
|
|
||||||
amp.Outline(MM, 0, 5, (
|
|
||||||
-var(2)/2, var(2)/2,
|
|
||||||
-var(2)/2, (var(7)-var(8)),
|
|
||||||
-var(7), (var(7)-var(8)),
|
|
||||||
-(var(7)-var(8)), var(7),
|
|
||||||
-(var(7)-var(8)), var(2)/2,
|
|
||||||
-var(2)/2, var(2)/2,
|
|
||||||
), var(5)),
|
|
||||||
amp.Outline(MM, 0, 5, (
|
|
||||||
var(2)/2, -var(2)/2,
|
|
||||||
var(2)/2, -(var(7)-var(8)),
|
|
||||||
var(7), -(var(7)-var(8)),
|
|
||||||
(var(7)-var(8)), -var(7),
|
|
||||||
(var(7)-var(8)), -var(2)/2,
|
|
||||||
var(2)/2, -var(2)/2,
|
|
||||||
), var(5)),
|
|
||||||
amp.Outline(MM, 0, 5, (
|
|
||||||
var(2)/2, var(2)/2,
|
|
||||||
var(2)/2, (var(7)-var(8)),
|
|
||||||
var(7), (var(7)-var(8)),
|
|
||||||
(var(7)-var(8)), var(7),
|
|
||||||
(var(7)-var(8)), var(2)/2,
|
|
||||||
var(2)/2, var(2)/2,
|
|
||||||
), var(5)),
|
|
||||||
amp.Circle(MM, 0, 2*var(8), -var(7), -var(7), var(5)),
|
|
||||||
amp.Circle(MM, 0, 2*var(8), -var(7), var(7), var(5)),
|
|
||||||
amp.Circle(MM, 0, 2*var(8), var(7), -var(7), var(5)),
|
|
||||||
amp.Circle(MM, 0, 2*var(8), var(7), var(7), var(5)),
|
|
||||||
), (
|
|
||||||
None, # 1
|
|
||||||
None, # 2
|
|
||||||
None, # 3
|
|
||||||
None, # 4
|
|
||||||
None, # 5
|
|
||||||
None, # 6
|
|
||||||
var(2)/2 - var(1)/2 + var(4), # 7
|
|
||||||
var(4)+var(6), # 8
|
|
||||||
))
|
|
||||||
corner_radius = (self.link_pad_width - self.link_trace_width)/3
|
|
||||||
main_ap = ApertureMacroInstance(alio_main_macro, (self.link_pad_width, # 1
|
|
||||||
self.pitch, # 2
|
|
||||||
self.link_trace_width, # 3
|
|
||||||
corner_radius, # 4
|
|
||||||
rotation, # 5
|
|
||||||
self.clearance), unit=MM) # 6
|
|
||||||
main_ap_90 = ApertureMacroInstance(alio_main_macro, (self.link_pad_width, # 1
|
|
||||||
self.pitch, # 2
|
|
||||||
self.link_trace_width, # 3
|
|
||||||
corner_radius, # 4
|
|
||||||
rotation-90, # 5
|
|
||||||
self.clearance), unit=MM) # 6
|
|
||||||
main_drill = ExcellonTool(self.drill, plated=True, unit=self.unit)
|
|
||||||
via_drill = ExcellonTool(self.via_size, plated=True, unit=self.unit)
|
|
||||||
|
|
||||||
# parameters: [1: total height = pad width, 2: total width, 3: trace width, 4: corner radius, 5: rotation]
|
|
||||||
alio_macro = ApertureMacro('ALIOP', (
|
|
||||||
amp.CenterLine(MM, 1, var(1)-2*var(4), var(1), 0, 0, var(5)),
|
|
||||||
amp.CenterLine(MM, 1, var(1), var(1)-2*var(4), 0, 0, var(5)),
|
|
||||||
amp.Circle(MM, 1, 2*var(4), -var(1)/2+var(4), -var(1)/2+var(4), var(5)),
|
|
||||||
amp.Circle(MM, 1, 2*var(4), -var(1)/2+var(4), var(1)/2-var(4), var(5)),
|
|
||||||
amp.Circle(MM, 1, 2*var(4), var(1)/2-var(4), -var(1)/2+var(4), var(5)),
|
|
||||||
amp.Circle(MM, 1, 2*var(4), var(1)/2-var(4), var(1)/2-var(4), var(5)),
|
|
||||||
amp.CenterLine(MM, 1, var(2), var(3), -var(2)/2 + var(1)/2, 0, var(5)),
|
|
||||||
))
|
|
||||||
alio_dark = ApertureMacroInstance(alio_macro, (self.link_pad_width, # 1
|
|
||||||
self.pitch-self.clearance, # 2
|
|
||||||
self.link_trace_width, # 3
|
|
||||||
corner_radius, # 4
|
|
||||||
rotation), unit=MM) # 5
|
|
||||||
alio_dark_90 = ApertureMacroInstance(alio_macro, (self.link_pad_width, # 1
|
|
||||||
self.pitch-self.clearance, # 2
|
|
||||||
self.link_trace_width, # 3
|
|
||||||
corner_radius, # 4
|
|
||||||
rotation+90), unit=MM) # 5
|
|
||||||
|
|
||||||
# all layers are identical here
|
|
||||||
for side, use in (('top', 'copper'), ('top', 'mask'), ('bottom', 'copper'), ('bottom', 'mask')):
|
|
||||||
if side == 'top':
|
|
||||||
layer_stack[side, use].objects.insert(0, xf(Flash(0, 0, aperture=main_ap, unit=self.unit)))
|
|
||||||
if not self.border_y:
|
|
||||||
layer_stack[side, use].objects.append(xf(Flash(self.pitch/2, self.pitch/2, aperture=alio_dark, unit=self.unit)))
|
|
||||||
else:
|
|
||||||
layer_stack[side, use].objects.insert(0, xf(Flash(0, 0, aperture=main_ap_90, unit=self.unit)))
|
|
||||||
if not self.border_x:
|
|
||||||
layer_stack[side, use].objects.append(xf(Flash(self.pitch/2, self.pitch/2, aperture=alio_dark_90, unit=self.unit)))
|
|
||||||
|
|
||||||
layer_stack.drill_pth.append(Flash(x, y, aperture=main_drill, unit=self.unit))
|
|
||||||
if not (self.border_x or self.border_y):
|
|
||||||
layer_stack.drill_pth.append(xf(Flash(self.pitch/2, self.pitch/2, aperture=via_drill, unit=self.unit)))
|
|
||||||
|
|
||||||
|
|
||||||
def convert_to_mm(value, unit):
|
|
||||||
unitl = unit.lower()
|
|
||||||
if unitl == 'mm':
|
|
||||||
return value
|
|
||||||
elif unitl == 'cm':
|
|
||||||
return value*10
|
|
||||||
elif unitl == 'in':
|
|
||||||
return value*25.4
|
|
||||||
elif unitl == 'mil':
|
|
||||||
return value/1000*25.4
|
|
||||||
else:
|
|
||||||
raise ValueError(f'Invalid unit {unit}, allowed units are mm, cm, in, and mil.')
|
|
||||||
|
|
||||||
|
|
||||||
_VALUE_RE = re.compile('([0-9]*\.?[0-9]+)(cm|mm|in|mil|%)')
|
|
||||||
def eval_value(value, total_length=None):
|
|
||||||
if not isinstance(value, str):
|
|
||||||
return None
|
|
||||||
|
|
||||||
m = _VALUE_RE.match(value.lower())
|
|
||||||
number, unit = m.groups()
|
|
||||||
if unit == '%':
|
|
||||||
if total_length is None:
|
|
||||||
raise ValueError('Percentages are not allowed for this value')
|
|
||||||
return total_length * float(number) / 100
|
|
||||||
return convert_to_mm(float(number), unit)
|
|
||||||
|
|
||||||
|
|
||||||
def _demo():
|
|
||||||
#pattern1 = PatternProtoArea(2.54, obj=THTPad.circle(0, 0, 0.9, 1.8, paste=False))
|
|
||||||
#pattern1 = PatternProtoArea(2.54, 2.54, obj=SpikyProto())
|
|
||||||
#pattern2 = PatternProtoArea(1.2, 2.0, obj=SMDPad.rect(0, 0, 1.0, 1.8, paste=False))
|
|
||||||
#pattern3 = PatternProtoArea(2.54, 1.27, obj=SMDPad.rect(0, 0, 2.3, 1.0, paste=False))
|
|
||||||
#pattern3 = EmptyProtoArea(copper_fill=True)
|
|
||||||
#stack = TwoSideLayout(pattern2, pattern3)
|
|
||||||
#pattern2 = PatternProtoArea(2.54, obj=PoweredProto(), margin=1)
|
|
||||||
pattern = PatternProtoArea(2.54, obj=RFGroundProto())
|
|
||||||
#stack = PropLayout([pattern2, pattern3], 'h', [0.5, 0.5])
|
|
||||||
#pattern = PropLayout([pattern1, stack], 'h', [0.5, 0.5])
|
|
||||||
#pattern = PatternProtoArea(2.54, obj=ManhattanPads(2.54))
|
|
||||||
#pattern = PatternProtoArea(2.54*1.5, obj=THTFlowerProto())
|
|
||||||
#pattern = PatternProtoArea(2.54, obj=THTPad.circle(0, 0, 0.9, 1.8, paste=False))
|
|
||||||
#pattern = PatternProtoArea(2.54, obj=PoweredProto())
|
|
||||||
#pattern = PatternProtoArea(2.54, obj=AlioCell(), margin=2)
|
|
||||||
pb = ProtoBoard(50, 47, pattern, mounting_hole_dia=3.2, mounting_hole_offset=5)
|
|
||||||
#pb = ProtoBoard(10, 10, pattern1)
|
|
||||||
print(pb.pretty_svg())
|
|
||||||
pb.layer_stack().save_to_directory('/tmp/testdir')
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
_demo()
|
|
||||||
#cnt = alphabetic()()
|
|
||||||
#for _ in range(32):
|
|
||||||
# for _ in range(26):
|
|
||||||
# print(f'{next(cnt):>2}', end=' ', file=sys.stderr)
|
|
||||||
# print(file=sys.stderr)
|
|
||||||
|
|
||||||
|
|
@ -1,99 +0,0 @@
|
||||||
#!/usr/bin/env python
|
|
||||||
|
|
||||||
from pathlib import Path
|
|
||||||
import unicodedata
|
|
||||||
import re
|
|
||||||
import ast
|
|
||||||
from functools import lru_cache
|
|
||||||
from importlib.resources import files
|
|
||||||
|
|
||||||
from . import data
|
|
||||||
|
|
||||||
|
|
||||||
STROKE_FONT_SCALE = 1/21
|
|
||||||
FONT_OFFSET = -10
|
|
||||||
DEFAULT_SPACE_WIDTH = 0.6
|
|
||||||
DEFAULT_CHAR_GAP = 0.2
|
|
||||||
|
|
||||||
_dec = lambda c: ord(c)-ord('R')
|
|
||||||
|
|
||||||
|
|
||||||
class Newstroke:
|
|
||||||
def __init__(self, newstroke_cpp=None):
|
|
||||||
if newstroke_cpp is None:
|
|
||||||
newstroke_cpp = files(data).joinpath('newstroke_font.cpp').read_bytes()
|
|
||||||
self.glyphs = dict(self.load_font(newstroke_cpp))
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
@lru_cache
|
|
||||||
def load(kls):
|
|
||||||
return kls()
|
|
||||||
|
|
||||||
def render(self, text, size=1.0, space_width=DEFAULT_SPACE_WIDTH, char_gap=DEFAULT_CHAR_GAP):
|
|
||||||
text = unicodedata.normalize('NFC', text)
|
|
||||||
missing_glyph = self.glyphs['?']
|
|
||||||
x = 0
|
|
||||||
for c in text:
|
|
||||||
if c == ' ':
|
|
||||||
x += space_width*size
|
|
||||||
continue
|
|
||||||
|
|
||||||
width, strokes = self.glyphs.get(c, missing_glyph)
|
|
||||||
glyph_w = max(width, max(x for st in strokes for x, _y in st))
|
|
||||||
|
|
||||||
for st in strokes:
|
|
||||||
yield self.transform_stroke(st, translate=(x, 0), scale=(size, size))
|
|
||||||
|
|
||||||
x += glyph_w*size
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def transform_stroke(kls, stroke, translate, scale):
|
|
||||||
dx, dy = translate
|
|
||||||
sx, sy = scale
|
|
||||||
return [(x*sx+dx, y*sy+dy) for x, y in stroke]
|
|
||||||
|
|
||||||
|
|
||||||
def load_font(self, newstroke_cpp):
|
|
||||||
e = []
|
|
||||||
for char, (width, strokes) in self.load_glyphs(newstroke_cpp):
|
|
||||||
yield char, (width, strokes)
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def decode_stroke(kls, stroke, start_x):
|
|
||||||
for i in range(0, len(stroke), 2):
|
|
||||||
x = (stroke[i]-0x52-start_x)*STROKE_FONT_SCALE
|
|
||||||
y = (stroke[i+1]-0x52+FONT_OFFSET)*STROKE_FONT_SCALE
|
|
||||||
yield (x, y)
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def decode_glyph(kls, data):
|
|
||||||
start_x, end_x = data[0]-0x52, data[1]-0x52
|
|
||||||
width = end_x - start_x
|
|
||||||
|
|
||||||
strokes = tuple(tuple(kls.decode_stroke(st, start_x)) for st in data[2:].split(b' R'))
|
|
||||||
return width*STROKE_FONT_SCALE, strokes
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def load_glyphs(kls, newstroke_cpp):
|
|
||||||
it = iter(newstroke_cpp.splitlines())
|
|
||||||
|
|
||||||
for line in it:
|
|
||||||
if re.search(rb'char.*\*', line):
|
|
||||||
break
|
|
||||||
|
|
||||||
charcode = 0x20
|
|
||||||
for line in it:
|
|
||||||
if (match := re.search(rb'".*"', line)):
|
|
||||||
yield chr(charcode), kls.decode_glyph(match.group(0)[1:-1].replace(b'\\\\', b'\\'))
|
|
||||||
charcode += 1
|
|
||||||
else:
|
|
||||||
if b'}' in line:
|
|
||||||
break
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
import time
|
|
||||||
t1 = time.time()
|
|
||||||
Newstroke()
|
|
||||||
t2 = time.time()
|
|
||||||
print((t2-t1)*1000)
|
|
||||||
|
|
@ -1,63 +0,0 @@
|
||||||
|
|
||||||
import os
|
|
||||||
from pathlib import Path
|
|
||||||
|
|
||||||
import pytest
|
|
||||||
|
|
||||||
from .image_support import ImageDifference, run_cargo_cmd
|
|
||||||
|
|
||||||
def pytest_assertrepr_compare(op, left, right):
|
|
||||||
if isinstance(left, ImageDifference) or isinstance(right, ImageDifference):
|
|
||||||
diff = left if isinstance(left, ImageDifference) else right
|
|
||||||
return [
|
|
||||||
f'Image difference assertion failed.',
|
|
||||||
f' Calculated difference: {diff}',
|
|
||||||
f' Histogram: {diff.histogram}', ]
|
|
||||||
|
|
||||||
# store report in node object so tmp_gbr can determine if the test failed.
|
|
||||||
@pytest.hookimpl(tryfirst=True, hookwrapper=True)
|
|
||||||
def pytest_runtest_makereport(item, call):
|
|
||||||
outcome = yield
|
|
||||||
rep = outcome.get_result()
|
|
||||||
setattr(item, f'rep_{rep.when}', rep)
|
|
||||||
|
|
||||||
fail_dir = Path('gerbonara_test_failures')
|
|
||||||
def pytest_sessionstart(session):
|
|
||||||
if not hasattr(session.config, 'workerinput'): # on worker
|
|
||||||
return
|
|
||||||
|
|
||||||
# on coordinator
|
|
||||||
for f in chain(fail_dir.glob('*.gbr'), fail_dir.glob('*.png')):
|
|
||||||
f.unlink()
|
|
||||||
|
|
||||||
try:
|
|
||||||
run_cargo_cmd('resvg', '--help')
|
|
||||||
except FileNotFoundError:
|
|
||||||
pytest.exit('resvg binary not found, aborting test.', 2)
|
|
||||||
|
|
||||||
def pytest_addoption(parser):
|
|
||||||
parser.addoption('--kicad-symbol-library', nargs='*', help='Run symbol library tests on given symbol libraries. May be given multiple times.')
|
|
||||||
parser.addoption('--kicad-footprint-files', nargs='*', help='Run footprint library tests on given footprint files. May be given multiple times.')
|
|
||||||
|
|
||||||
def pytest_generate_tests(metafunc):
|
|
||||||
if 'kicad_library_file' in metafunc.fixturenames:
|
|
||||||
if not (library_files := metafunc.config.getoption('symbol_library', None)):
|
|
||||||
if (lib_dir := os.environ.get('KICAD_SYMBOLS')):
|
|
||||||
lib_dir = Path(lib_dir).expanduser()
|
|
||||||
if not lib_dir.is_dir():
|
|
||||||
raise ValueError(f'Path "{lib_dir}" given by KICAD_SYMBOLS environment variable does not exist or is not a directory.')
|
|
||||||
library_files = list(lib_dir.glob('*.kicad_sym'))
|
|
||||||
else:
|
|
||||||
raise ValueError('Either --kicad-symbol-library command line parameter or KICAD_SYMBOLS environment variable must be given.')
|
|
||||||
metafunc.parametrize('kicad_library_file', library_files, ids=list(map(str, library_files)))
|
|
||||||
|
|
||||||
if 'kicad_mod_file' in metafunc.fixturenames:
|
|
||||||
if not (mod_files := metafunc.config.getoption('footprint_files', None)):
|
|
||||||
if (lib_dir := os.environ.get('KICAD_FOOTPRINTS')):
|
|
||||||
lib_dir = Path(lib_dir).expanduser()
|
|
||||||
if not lib_dir.is_dir():
|
|
||||||
raise ValueError(f'Path "{lib_dir}" given by KICAD_FOOTPRINTS environment variable does not exist or is not a directory.')
|
|
||||||
mod_files = list(lib_dir.glob('**/*.kicad_mod'))
|
|
||||||
else:
|
|
||||||
raise ValueError('Either --kicad-footprint-files command line parameter or KICAD_FOOTPRINTS environment variable must be given.')
|
|
||||||
metafunc.parametrize('kicad_mod_file', mod_files, ids=list(map(str, mod_files)))
|
|
||||||
|
|
@ -1,308 +0,0 @@
|
||||||
#!/usr/bin/env python
|
|
||||||
# -*- coding: utf-8 -*-
|
|
||||||
#
|
|
||||||
# Copyright 2022 Jan Sebastian Götte <gerbonara@jaseg.de>
|
|
||||||
#
|
|
||||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
# you may not use this file except in compliance with the License.
|
|
||||||
# You may obtain a copy of the License at
|
|
||||||
#
|
|
||||||
# http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
#
|
|
||||||
# Unless required by applicable law or agreed to in writing, software
|
|
||||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
# See the License for the specific language governing permissions and
|
|
||||||
# limitations under the License.
|
|
||||||
#
|
|
||||||
# Based on https://github.com/tracespace/tracespace
|
|
||||||
#
|
|
||||||
|
|
||||||
import subprocess
|
|
||||||
from pathlib import Path
|
|
||||||
import tempfile
|
|
||||||
import textwrap
|
|
||||||
import os
|
|
||||||
from functools import total_ordering
|
|
||||||
import shutil
|
|
||||||
import bs4
|
|
||||||
from contextlib import contextmanager
|
|
||||||
import hashlib
|
|
||||||
|
|
||||||
import numpy as np
|
|
||||||
from PIL import Image
|
|
||||||
|
|
||||||
cachedir = Path(__file__).parent / 'image_cache'
|
|
||||||
cachedir.mkdir(exist_ok=True)
|
|
||||||
|
|
||||||
@total_ordering
|
|
||||||
class ImageDifference:
|
|
||||||
def __init__(self, value, histogram):
|
|
||||||
self.value = value
|
|
||||||
self.histogram = histogram
|
|
||||||
|
|
||||||
def __float__(self):
|
|
||||||
return float(self.value)
|
|
||||||
|
|
||||||
def __eq__(self, other):
|
|
||||||
return float(self) == float(other)
|
|
||||||
|
|
||||||
def __lt__(self, other):
|
|
||||||
return float(self) < float(other)
|
|
||||||
|
|
||||||
def __str__(self):
|
|
||||||
return str(float(self))
|
|
||||||
|
|
||||||
@total_ordering
|
|
||||||
class Histogram:
|
|
||||||
def __init__(self, value, size):
|
|
||||||
self.value, self.size = value, size
|
|
||||||
|
|
||||||
def __eq__(self, other):
|
|
||||||
other = np.array(other)
|
|
||||||
other[other == None] = self.value[other == None]
|
|
||||||
return (self.value == other).all()
|
|
||||||
|
|
||||||
def __lt__(self, other):
|
|
||||||
other = np.array(other)
|
|
||||||
other[other == None] = self.value[other == None]
|
|
||||||
return (self.value <= other).all()
|
|
||||||
|
|
||||||
def __getitem__(self, index):
|
|
||||||
return self.value[index]
|
|
||||||
|
|
||||||
def __str__(self):
|
|
||||||
return f'{list(self.value)} size={self.size}'
|
|
||||||
|
|
||||||
|
|
||||||
def run_cargo_cmd(cmd, args, **kwargs):
|
|
||||||
if cmd.upper() in os.environ:
|
|
||||||
return subprocess.run([os.environ[cmd.upper()], *args], **kwargs)
|
|
||||||
|
|
||||||
try:
|
|
||||||
return subprocess.run([cmd, *args], **kwargs)
|
|
||||||
|
|
||||||
except FileNotFoundError:
|
|
||||||
return subprocess.run([str(Path.home() / '.cargo' / 'bin' / cmd), *args], **kwargs)
|
|
||||||
|
|
||||||
def svg_to_png(in_svg, out_png, dpi=100, bg=None):
|
|
||||||
params = f'{dpi}{bg}'.encode()
|
|
||||||
digest = hashlib.blake2b(Path(in_svg).read_bytes() + params).hexdigest()
|
|
||||||
cachefile = cachedir / f'{digest}.png'
|
|
||||||
|
|
||||||
if not cachefile.is_file():
|
|
||||||
bg = 'black' if bg is None else bg
|
|
||||||
run_cargo_cmd('resvg', ['--background', bg, '--dpi', str(dpi), in_svg, cachefile], check=True, stdout=subprocess.DEVNULL)
|
|
||||||
|
|
||||||
shutil.copy(cachefile, out_png)
|
|
||||||
|
|
||||||
to_gerbv_svg_units = lambda val, unit='mm': val*72 if unit == 'inch' else val/25.4*72
|
|
||||||
|
|
||||||
def gerbv_export(in_gbr, out_svg, export_format='svg', origin=(0, 0), size=(6, 6), fg='#ffffff', bg='#000000', override_unit_spec=None):
|
|
||||||
params = f'{origin}{size}{fg}{bg}'.encode()
|
|
||||||
digest = hashlib.blake2b(Path(in_gbr).read_bytes() + params).hexdigest()
|
|
||||||
cachefile = cachedir / f'{digest}.svg'
|
|
||||||
|
|
||||||
if not cachefile.is_file():
|
|
||||||
print(f'Building cache for {Path(in_gbr).name}')
|
|
||||||
# NOTE: gerbv seems to always export 'clear' polarity apertures as white, irrespective of --foreground, --background
|
|
||||||
# and project file color settings.
|
|
||||||
# TODO: File issue upstream.
|
|
||||||
with tempfile.NamedTemporaryFile('w') as f:
|
|
||||||
if override_unit_spec:
|
|
||||||
units, zeros, digits = override_unit_spec
|
|
||||||
print(f'{Path(in_gbr).name}: overriding excellon unit spec to {units=} {zeros=} {digits=}')
|
|
||||||
units = 0 if units == 'inch' else 1
|
|
||||||
zeros = {None: 0, 'leading': 1, 'trailing': 2}[zeros]
|
|
||||||
unit_spec = textwrap.dedent(f'''(cons 'attribs (list
|
|
||||||
(list 'autodetect 'Boolean 0)
|
|
||||||
(list 'zero_suppression 'Enum {zeros})
|
|
||||||
(list 'units 'Enum {units})
|
|
||||||
(list 'digits 'Integer {digits})
|
|
||||||
))''')
|
|
||||||
else:
|
|
||||||
unit_spec = ''
|
|
||||||
|
|
||||||
r, g, b = int(fg[1:3], 16), int(fg[3:5], 16), int(fg[5:], 16)
|
|
||||||
color = f"(cons 'color #({r*257} {g*257} {b*257}))"
|
|
||||||
f.write(f'''(gerbv-file-version! "2.0A")(define-layer! 0 (cons 'filename "{in_gbr}"){unit_spec}{color})''')
|
|
||||||
f.flush()
|
|
||||||
if override_unit_spec:
|
|
||||||
shutil.copy(f.name, '/tmp/foo.gbv')
|
|
||||||
|
|
||||||
x, y = origin
|
|
||||||
w, h = size
|
|
||||||
cmd = ['gerbv', '-x', export_format,
|
|
||||||
'--border=0',
|
|
||||||
f'--origin={x:.6f}x{y:.6f}', f'--window_inch={w:.6f}x{h:.6f}',
|
|
||||||
f'--background={bg}',
|
|
||||||
f'--foreground={fg}',
|
|
||||||
'-o', str(cachefile), '-p', f.name]
|
|
||||||
subprocess.run(cmd, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
||||||
else:
|
|
||||||
print(f'Re-using cache for {Path(in_gbr).name}')
|
|
||||||
shutil.copy(cachefile, out_svg)
|
|
||||||
|
|
||||||
def kicad_fp_export(mod_file, out_svg):
|
|
||||||
mod_file = Path(mod_file)
|
|
||||||
if mod_file.suffix.lower() != '.kicad_mod':
|
|
||||||
raise ValueError("KiCad footprint file must have .kicad_mod extension for kicad-cli to do it's thing")
|
|
||||||
|
|
||||||
params = f'(noparams)'.encode()
|
|
||||||
digest = hashlib.blake2b(mod_file.read_bytes() + params).hexdigest()
|
|
||||||
cachefile = cachedir / f'{digest}.svg'
|
|
||||||
|
|
||||||
if not cachefile.is_file():
|
|
||||||
print(f'Building cache for {mod_file.name}')
|
|
||||||
|
|
||||||
with tempfile.TemporaryDirectory() as tmpdir:
|
|
||||||
pretty_dir = mod_file.parent
|
|
||||||
fp_name = mod_file.name[:-len('.kicad_mod')]
|
|
||||||
cmd = ['kicad-cli', 'fp', 'export', 'svg', '--output', tmpdir, '--footprint', fp_name, pretty_dir]
|
|
||||||
subprocess.run(cmd, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
||||||
out_file = Path(tmpdir) / f'{fp_name}.svg'
|
|
||||||
shutil.copy(out_file, cachefile)
|
|
||||||
else:
|
|
||||||
print(f'Re-using cache for {mod_file.name}')
|
|
||||||
shutil.copy(cachefile, out_svg)
|
|
||||||
|
|
||||||
@contextmanager
|
|
||||||
def svg_soup(filename):
|
|
||||||
with open(filename, 'r') as f:
|
|
||||||
soup = bs4.BeautifulSoup(f.read(), 'xml')
|
|
||||||
|
|
||||||
yield soup
|
|
||||||
|
|
||||||
with open(filename, 'w') as f:
|
|
||||||
f.write(str(soup))
|
|
||||||
|
|
||||||
def cleanup_gerbv_svg(soup):
|
|
||||||
width = soup.svg["width"]
|
|
||||||
height = soup.svg["height"]
|
|
||||||
width = width[:-2] if width.endswith('pt') else width
|
|
||||||
height = height[:-2] if height.endswith('pt') else height
|
|
||||||
soup.svg['width'] = f'{float(width)/72*25.4:.4f}mm'
|
|
||||||
soup.svg['height'] = f'{float(height)/72*25.4:.4f}mm'
|
|
||||||
for group in soup.find_all('g'):
|
|
||||||
# gerbv uses Cairo's SVG canvas. Cairo's SVG canvas is kind of broken. It has no support for unit
|
|
||||||
# handling at all, which means the output files just end up being in pixels at 72 dpi. Further, it
|
|
||||||
# seems gerbv's aperture macro rendering interacts poorly with Cairo's SVG export. gerbv renders
|
|
||||||
# aperture macros into a new surface, which for some reason gets clipped by Cairo to the given
|
|
||||||
# canvas size. This is just wrong, so we just nuke the clip path from these SVG groups here.
|
|
||||||
#
|
|
||||||
# Apart from being graphically broken, this additionally causes very bad rendering performance.
|
|
||||||
del group['clip-path']
|
|
||||||
|
|
||||||
def gerber_difference(reference, actual, diff_out=None, svg_transform=None, size=(10,10), ref_unit_spec=None):
|
|
||||||
with tempfile.NamedTemporaryFile(suffix='.svg') as act_svg,\
|
|
||||||
tempfile.NamedTemporaryFile(suffix='.svg') as ref_svg:
|
|
||||||
|
|
||||||
gerbv_export(reference, ref_svg.name, size=size, export_format='svg', override_unit_spec=ref_unit_spec)
|
|
||||||
gerbv_export(actual, act_svg.name, size=size, export_format='svg')
|
|
||||||
|
|
||||||
with svg_soup(ref_svg.name) as soup:
|
|
||||||
if svg_transform is not None:
|
|
||||||
svg = soup.svg
|
|
||||||
children = list(svg.children)
|
|
||||||
g = soup.new_tag('g', attrs={'transform': svg_transform})
|
|
||||||
for c in children:
|
|
||||||
g.append(c.extract())
|
|
||||||
svg.append(g)
|
|
||||||
|
|
||||||
cleanup_gerbv_svg(soup)
|
|
||||||
|
|
||||||
with svg_soup(act_svg.name) as soup:
|
|
||||||
cleanup_gerbv_svg(soup)
|
|
||||||
|
|
||||||
return svg_difference(ref_svg.name, act_svg.name, diff_out=diff_out)
|
|
||||||
|
|
||||||
def gerber_difference_merge(ref1, ref2, actual, diff_out=None, composite_out=None, svg_transform1=None, svg_transform2=None, size=(10,10)):
|
|
||||||
with tempfile.NamedTemporaryFile(suffix='.svg') as act_svg,\
|
|
||||||
tempfile.NamedTemporaryFile(suffix='.svg') as ref1_svg,\
|
|
||||||
tempfile.NamedTemporaryFile(suffix='.svg') as ref2_svg:
|
|
||||||
|
|
||||||
gerbv_export(ref1, ref1_svg.name, size=size, export_format='svg')
|
|
||||||
gerbv_export(ref2, ref2_svg.name, size=size, export_format='svg')
|
|
||||||
gerbv_export(actual, act_svg.name, size=size, export_format='svg')
|
|
||||||
for var in ['ref1_svg', 'ref2_svg', 'act_svg']:
|
|
||||||
print(f'=== {var} ===')
|
|
||||||
print(Path(locals()[var].name).read_text().splitlines()[1])
|
|
||||||
|
|
||||||
with svg_soup(ref1_svg.name) as soup1:
|
|
||||||
if svg_transform1 is not None:
|
|
||||||
svg = soup1.svg
|
|
||||||
children = list(svg.children)
|
|
||||||
g = soup1.new_tag('g', attrs={'transform': svg_transform1})
|
|
||||||
for c in children:
|
|
||||||
g.append(c.extract())
|
|
||||||
svg.append(g)
|
|
||||||
cleanup_gerbv_svg(soup1)
|
|
||||||
|
|
||||||
with svg_soup(ref2_svg.name) as soup2:
|
|
||||||
if svg_transform2 is not None:
|
|
||||||
svg = soup2.svg
|
|
||||||
children = list(svg.children)
|
|
||||||
g = soup2.new_tag('g', attrs={'transform': svg_transform2})
|
|
||||||
for c in children:
|
|
||||||
g.append(c.extract())
|
|
||||||
svg.append(g)
|
|
||||||
cleanup_gerbv_svg(soup2)
|
|
||||||
|
|
||||||
defs1 = soup1.find('defs')
|
|
||||||
if not defs1:
|
|
||||||
defs1 = soup1.new_tag('defs')
|
|
||||||
soup1.find('svg').insert(0, defs1)
|
|
||||||
|
|
||||||
defs2 = soup2.find('defs')
|
|
||||||
if defs2:
|
|
||||||
defs2 = defs2.extract()
|
|
||||||
# explicitly convert .contents into list here and below because else bs4 stumbles over itself
|
|
||||||
# iterating because we modify the tree in the loop body.
|
|
||||||
for c in list(defs2.contents):
|
|
||||||
if hasattr(c, 'attrs'):
|
|
||||||
c['id'] = 'gn-merge-b-' + c.attrs.get('id', str(id(c)))
|
|
||||||
defs1.append(c)
|
|
||||||
|
|
||||||
for use in soup2.find_all('use', recursive=True):
|
|
||||||
if (href := use.get('xlink:href', '')).startswith('#'):
|
|
||||||
use['xlink:href'] = f'#gn-merge-b-{href[1:]}'
|
|
||||||
|
|
||||||
svg1 = soup1.find('svg')
|
|
||||||
for c in list(soup2.find('svg').contents):
|
|
||||||
if hasattr(c, 'attrs'):
|
|
||||||
c['id'] = 'gn-merge-b-' + c.attrs.get('id', str(id(c)))
|
|
||||||
svg1.append(c)
|
|
||||||
|
|
||||||
if composite_out:
|
|
||||||
shutil.copyfile(ref1_svg.name, composite_out)
|
|
||||||
|
|
||||||
with svg_soup(act_svg.name) as soup:
|
|
||||||
cleanup_gerbv_svg(soup)
|
|
||||||
|
|
||||||
return svg_difference(ref1_svg.name, act_svg.name, diff_out=diff_out)
|
|
||||||
|
|
||||||
def svg_difference(reference, actual, diff_out=None, background=None, dpi=100):
|
|
||||||
with tempfile.NamedTemporaryFile(suffix='-ref.png') as ref_png,\
|
|
||||||
tempfile.NamedTemporaryFile(suffix='-act.png') as act_png:
|
|
||||||
|
|
||||||
svg_to_png(reference, ref_png.name, bg=background, dpi=dpi)
|
|
||||||
svg_to_png(actual, act_png.name, bg=background, dpi=dpi)
|
|
||||||
|
|
||||||
return image_difference(ref_png.name, act_png.name, diff_out=diff_out)
|
|
||||||
|
|
||||||
def image_difference(reference, actual, diff_out=None):
|
|
||||||
ref = np.array(Image.open(reference)).astype(float)
|
|
||||||
out = np.array(Image.open(actual)).astype(float)
|
|
||||||
|
|
||||||
ref, out = ref.mean(axis=2), out.mean(axis=2) # convert to grayscale
|
|
||||||
# TODO blur images here before comparison to mitigate aliasing issue
|
|
||||||
delta = np.abs(out - ref).astype(float) / 255
|
|
||||||
if diff_out:
|
|
||||||
Image.fromarray((delta*255).astype(np.uint8), mode='L').save(diff_out)
|
|
||||||
|
|
||||||
hist, _bins = np.histogram(delta, bins=10, range=(0, 1))
|
|
||||||
return (ImageDifference(delta.mean(), hist),
|
|
||||||
ImageDifference(delta.max(), hist),
|
|
||||||
Histogram(hist, out.size))
|
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -1,9 +0,0 @@
|
||||||
%FSLAX46Y46*%
|
|
||||||
%MOMM*%
|
|
||||||
%ADD10C,0.150000*%
|
|
||||||
%ADD11C,0.100000*%
|
|
||||||
%ADD12C,0.600000*%
|
|
||||||
%ADD13C,0.120000*%
|
|
||||||
|
|
||||||
%LPD*%
|
|
||||||
G54D10*X168523809Y-90902380D02*X168523809Y-89902380D01*X168285714Y-89902380D01*X168142857Y-89950000D01*X168047619Y-90045238D01*X168000000Y-90140476D01*X167952380Y-90330952D01*X167952380Y-90473809D01*X168000000Y-90664285D01*X168047619Y-90759523D01*X168142857Y-90854761D01*X168285714Y-90902380D01*X168523809Y-90902380D01*X167571428Y-90616666D02*X167095238Y-90616666D01*X167666666Y-90902380D02*X167333333Y-89902380D01*X167000000Y-90902380D01*X166809523Y-89902380D02*X166238095Y-89902380D01*X166523809Y-90902380D02*X166523809Y-89902380D01*X165904761Y-90378571D02*X165571428Y-90378571D01*X165428571Y-90902380D02*X165904761Y-90902380D01*X165904761Y-89902380D01*X165428571Y-89902380D01*X168509523Y-78304761D02*X168366666Y-78352380D01*X168128571Y-78352380D01*X168033333Y-78304761D01*X167985714Y-78257142D01*X167938095Y-78161904D01*X167938095Y-78066666D01*X167985714Y-77971428D01*X168033333Y-77923809D01*X168128571Y-77876190D01*X168319047Y-77828571D01*X168414285Y-77780952D01*X168461904Y-77733333D01*X168509523Y-77638095D01*X168509523Y-77542857D01*X168461904Y-77447619D01*X168414285Y-77400000D01*X168319047Y-77352380D01*X168080952Y-77352380D01*X167938095Y-77400000D01*X167509523Y-78352380D02*X167509523Y-77352380D01*X166938095Y-78352380D01*X166938095Y-77352380D01*G54D11*G36*X168500000Y-89450000D02*G01X128500000Y-89450000D01*X128500000Y-78950000D01*X168500000Y-78950000D01*X168500000Y-89450000D01*G37*X168500000Y-89450000D02*X128500000Y-89450000D01*X128500000Y-78950000D01*X168500000Y-78950000D01*X168500000Y-89450000D01*G54D12*X131250000Y-58357142D02*X130678571Y-58357142D01*X130392857Y-58500000D01*X130107142Y-58785714D01*X129964285Y-59357142D01*X129964285Y-60357142D01*X130107142Y-60928571D01*X130392857Y-61214285D01*X130678571Y-61357142D01*X131250000Y-61357142D01*X131535714Y-61214285D01*X131821428Y-60928571D01*X131964285Y-60357142D01*X131964285Y-59357142D01*X131821428Y-58785714D01*X131535714Y-58500000D01*X131250000Y-58357142D01*X128678571Y-58357142D02*X128678571Y-60785714D01*X128535714Y-61071428D01*X128392857Y-61214285D01*X128107142Y-61357142D01*X127535714Y-61357142D01*X127250000Y-61214285D01*X127107142Y-61071428D01*X126964285Y-60785714D01*X126964285Y-58357142D01*X125964285Y-58357142D02*X124250000Y-58357142D01*X125107142Y-61357142D02*X125107142Y-58357142D01*X150071428Y-61357142D02*X150071428Y-58357142D01*X148642857Y-61357142D02*X148642857Y-58357142D01*X146928571Y-61357142D01*X146928571Y-58357142D01*G54D13*X117000000Y-76450000D02*X117000000Y-77150000D01*X118200000Y-77150000D02*X118200000Y-76450000D01*G54D10*X120242857Y-77157142D02*X120290476Y-77204761D01*X120433333Y-77252380D01*X120528571Y-77252380D01*X120671428Y-77204761D01*X120766666Y-77109523D01*X120814285Y-77014285D01*X120861904Y-76823809D01*X120861904Y-76680952D01*X120814285Y-76490476D01*X120766666Y-76395238D01*X120671428Y-76300000D01*X120528571Y-76252380D01*X120433333Y-76252380D01*X120290476Y-76300000D01*X120242857Y-76347619D01*X119290476Y-77252380D02*X119861904Y-77252380D01*X119576190Y-77252380D02*X119576190Y-76252380D01*X119671428Y-76395238D01*X119766666Y-76490476D01*X119861904Y-76538095D01*X118338095Y-77252380D02*X118909523Y-77252380D01*X118623809Y-77252380D02*X118623809Y-76252380D01*X118719047Y-76395238D01*X118814285Y-76490476D01*X118909523Y-76538095D01*X0Y0D02*M00*
|
|
||||||
|
|
@ -1,59 +0,0 @@
|
||||||
|
|
||||||
from itertools import zip_longest
|
|
||||||
import re
|
|
||||||
|
|
||||||
from ..cad.kicad.sexp import build_sexp
|
|
||||||
from ..cad.kicad.sexp_mapper import sexp
|
|
||||||
from ..cad.kicad.symbols import Library
|
|
||||||
|
|
||||||
|
|
||||||
def test_parse(kicad_library_file):
|
|
||||||
Library.open(kicad_library_file)
|
|
||||||
|
|
||||||
|
|
||||||
def test_round_trip(kicad_library_file):
|
|
||||||
print('========== Stage 1 load ==========')
|
|
||||||
orig_lib = Library.open(kicad_library_file)
|
|
||||||
print('========== Stage 1 save ==========')
|
|
||||||
stage1_sexp = build_sexp(orig_lib.sexp())
|
|
||||||
|
|
||||||
print('========== Stage 2 load ==========')
|
|
||||||
reparsed_lib = Library.parse(stage1_sexp)
|
|
||||||
print('========== Stage 2 save ==========')
|
|
||||||
stage2_sexp = build_sexp(reparsed_lib.sexp())
|
|
||||||
print('========== Checks ==========')
|
|
||||||
|
|
||||||
for stage1, stage2 in zip_longest(stage1_sexp.splitlines(), stage2_sexp.splitlines()):
|
|
||||||
assert stage1 == stage2
|
|
||||||
|
|
||||||
original = re.sub(r'\(', '\n(', re.sub(r'\s+', ' ', kicad_library_file.read_text()))
|
|
||||||
original = re.sub(r'\) \)', '))', original)
|
|
||||||
original = re.sub(r'\) \)', '))', original)
|
|
||||||
original = re.sub(r'\) \)', '))', original)
|
|
||||||
original = re.sub(r'\) \)', '))', original)
|
|
||||||
stage1 = re.sub(r'\(', '\n(', re.sub(r'\s+', ' ', stage1_sexp))
|
|
||||||
for original, stage1 in zip_longest(original.splitlines(), stage1.splitlines()):
|
|
||||||
if original.startswith('(version'):
|
|
||||||
continue
|
|
||||||
|
|
||||||
original, stage1 = original.strip(), stage1.strip()
|
|
||||||
if original != stage1:
|
|
||||||
if any(original.startswith(f'({foo}') for foo in ['arc', 'circle', 'rectangle', 'polyline', 'text']):
|
|
||||||
# These files have symbols with graphic primitives in non-standard order
|
|
||||||
return
|
|
||||||
|
|
||||||
if original.startswith('(offset') and stage1.startswith('(offset'):
|
|
||||||
# Some symbol files contain ints where floats should be.
|
|
||||||
return
|
|
||||||
|
|
||||||
if original.startswith('(symbol') and stage1.startswith('(symbol'):
|
|
||||||
# Re-export can change symbol order. This is ok.
|
|
||||||
return
|
|
||||||
|
|
||||||
if original.startswith('(at') and stage1.startswith('(at'):
|
|
||||||
# There is some disagreement as to whether rotation angles are ints or floats, and the spec doesn't say.
|
|
||||||
return
|
|
||||||
|
|
||||||
assert original == stage1
|
|
||||||
|
|
||||||
|
|
||||||
72
pyproject.toml
Normal file
|
|
@ -0,0 +1,72 @@
|
||||||
|
[project]
|
||||||
|
name = "gerbonara"
|
||||||
|
version = "1.6.2"
|
||||||
|
description = "Tools to handle Gerber and Excellon files in Python"
|
||||||
|
readme = "README.md"
|
||||||
|
license = "Apache-2.0"
|
||||||
|
requires-python = ">=3.12"
|
||||||
|
dependencies = ["click", "rtree", "quart"]
|
||||||
|
|
||||||
|
authors = [
|
||||||
|
{ name = "jaseg" },
|
||||||
|
{ name = "XenGi" },
|
||||||
|
]
|
||||||
|
|
||||||
|
maintainers = [
|
||||||
|
{ name = "Gerbonara maintainers", email = "gerbonara@jaseg.de" },
|
||||||
|
]
|
||||||
|
|
||||||
|
keywords = ["gerber", "excellon", "pcb", "RS274x", "EDA"]
|
||||||
|
|
||||||
|
classifiers = [
|
||||||
|
"Development Status :: 5 - Production/Stable",
|
||||||
|
"Environment :: Console",
|
||||||
|
"Intended Audience :: Developers",
|
||||||
|
"Intended Audience :: Information Technology",
|
||||||
|
"Intended Audience :: Manufacturing",
|
||||||
|
"Intended Audience :: Science/Research",
|
||||||
|
"License :: OSI Approved :: Apache Software License",
|
||||||
|
"Topic :: Artistic Software",
|
||||||
|
"Topic :: Multimedia :: Graphics",
|
||||||
|
"Topic :: Printing",
|
||||||
|
"Topic :: Scientific/Engineering",
|
||||||
|
"Topic :: Scientific/Engineering :: Electronic Design Automation (EDA)",
|
||||||
|
"Topic :: Scientific/Engineering :: Image Processing",
|
||||||
|
"Topic :: Utilities",
|
||||||
|
]
|
||||||
|
|
||||||
|
[project.urls]
|
||||||
|
Homepage = "https://jaseg.de/projects/gerbonara/"
|
||||||
|
Documentation = "https://gerbolyze.gitlab.io/gerbonara/"
|
||||||
|
Source = "https://git.jaseg.de/gerbonara.git"
|
||||||
|
Tracker = "https://gitlab.com/gerbolyze/gerbonara/issues"
|
||||||
|
|
||||||
|
[project.scripts]
|
||||||
|
gerbonara = "gerbonara.cli:cli"
|
||||||
|
protoserve = "gerbonara.cad.protoserve:main"
|
||||||
|
|
||||||
|
[dependency-groups]
|
||||||
|
dev = [
|
||||||
|
"pytest",
|
||||||
|
"pytest-xdist",
|
||||||
|
"numpy",
|
||||||
|
"scipy",
|
||||||
|
"tqdm",
|
||||||
|
"beautifulsoup4",
|
||||||
|
"lxml",
|
||||||
|
"pillow"
|
||||||
|
]
|
||||||
|
|
||||||
|
[build-system]
|
||||||
|
requires = ["uv-build"]
|
||||||
|
build-backend = "uv_build"
|
||||||
|
|
||||||
|
[tool.pytest]
|
||||||
|
testpaths = ["tests"]
|
||||||
|
norecursedirs = ["*"]
|
||||||
|
kicad_symbols_tag = "9.0.6"
|
||||||
|
kicad_footprints_tag = "9.0.6"
|
||||||
|
kicad_source_tag = "9.0.6"
|
||||||
|
# Tag to use for container for footprint svg export
|
||||||
|
# For a list of available tags, see https://hub.docker.com/r/kicad/kicad/tags
|
||||||
|
kicad_container_tag = "9.0.6-full"
|
||||||
|
|
@ -1,3 +0,0 @@
|
||||||
[pytest]
|
|
||||||
testpaths = gerbonara/tests
|
|
||||||
norecursedirs=*
|
|
||||||
65
setup.py
|
|
@ -1,65 +0,0 @@
|
||||||
#!/usr/bin/env python3
|
|
||||||
|
|
||||||
from pathlib import Path
|
|
||||||
from setuptools import setup, find_packages
|
|
||||||
import subprocess
|
|
||||||
|
|
||||||
def version():
|
|
||||||
try:
|
|
||||||
res = subprocess.run(['git', 'describe', '--tags', '--match', 'v*'], capture_output=True, check=True, text=True)
|
|
||||||
version, _, _rest = res.stdout.strip()[1:].partition('-')
|
|
||||||
return version
|
|
||||||
except:
|
|
||||||
subprocess.run(['git', 'describe', '--tags', '--match', 'v*'])
|
|
||||||
raise
|
|
||||||
|
|
||||||
setup(
|
|
||||||
name='gerbonara',
|
|
||||||
version=version(),
|
|
||||||
author='jaseg, XenGi',
|
|
||||||
author_email='gerbonara@jaseg.de',
|
|
||||||
description='Tools to handle Gerber and Excellon files in Python',
|
|
||||||
long_description=Path('README.md').read_text(),
|
|
||||||
long_description_content_type='text/markdown',
|
|
||||||
url='https://gitlab.com/gerbolyze/gerbonara',
|
|
||||||
project_urls={
|
|
||||||
'Documentation': 'https://gerbolyze.gitlab.io/gerbonara/',
|
|
||||||
# 'Funding': 'https://donate.pypi.org',
|
|
||||||
# 'Say Thanks!': 'http://saythanks.io/to/example',
|
|
||||||
'Source': 'https://gitlab.com/gerbolyze/gerbonara',
|
|
||||||
'Tracker': 'https://gitlab.com/gerbolyze/gerbonara/issues',
|
|
||||||
},
|
|
||||||
packages=find_packages(exclude=['tests']),
|
|
||||||
install_requires=['click'],
|
|
||||||
entry_points={
|
|
||||||
'console_scripts': [
|
|
||||||
'gerbonara = gerbonara.cli:cli',
|
|
||||||
],
|
|
||||||
},
|
|
||||||
classifiers=[
|
|
||||||
'Development Status :: 4 - Beta',
|
|
||||||
#'Development Status :: 5 - Production/Stable',
|
|
||||||
'Environment :: Console',
|
|
||||||
'Intended Audience :: Developers',
|
|
||||||
'Intended Audience :: Information Technology',
|
|
||||||
'Intended Audience :: Manufacturing',
|
|
||||||
'Intended Audience :: Science/Research',
|
|
||||||
'License :: OSI Approved :: Apache Software License',
|
|
||||||
'Natural Language :: English',
|
|
||||||
'Operating System :: POSIX :: Linux',
|
|
||||||
'Programming Language :: Python :: 3',
|
|
||||||
'Programming Language :: Python :: 3 :: Only',
|
|
||||||
'Programming Language :: Python :: 3.10',
|
|
||||||
'Programming Language :: Python :: 3.11',
|
|
||||||
'Topic :: Artistic Software',
|
|
||||||
'Topic :: Multimedia :: Graphics',
|
|
||||||
'Topic :: Printing',
|
|
||||||
'Topic :: Scientific/Engineering',
|
|
||||||
'Topic :: Scientific/Engineering :: Electronic Design Automation (EDA)',
|
|
||||||
'Topic :: Scientific/Engineering :: Image Processing',
|
|
||||||
'Topic :: Utilities',
|
|
||||||
'Typing :: Typed',
|
|
||||||
],
|
|
||||||
keywords='gerber excellon pcb',
|
|
||||||
python_requires='>=3.10',
|
|
||||||
)
|
|
||||||
|
|
@ -30,5 +30,6 @@ from .excellon import ExcellonFile
|
||||||
from .ipc356 import Netlist
|
from .ipc356 import Netlist
|
||||||
from .layers import LayerStack
|
from .layers import LayerStack
|
||||||
from .utils import MM, Inch
|
from .utils import MM, Inch
|
||||||
|
from importlib.metadata import version
|
||||||
|
|
||||||
__version__ = '1.0.2'
|
__version__ = version('gerbonara')
|
||||||
42
src/gerbonara/__main__.py
Normal file
|
|
@ -0,0 +1,42 @@
|
||||||
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
|
import click
|
||||||
|
from zipfile import is_zipfile
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from .layers import LayerStack
|
||||||
|
from .rs274x import GerberFile
|
||||||
|
|
||||||
|
|
||||||
|
@click.group()
|
||||||
|
def cli():
|
||||||
|
pass
|
||||||
|
|
||||||
|
@cli.command(help='Render a folder or zip of Gerber and Excellon files to a pretty, semi-photorealistic SVG.')
|
||||||
|
@click.option('-t' ,'--top', help='Render board top side.', is_flag=True)
|
||||||
|
@click.option('-b' ,'--bottom', help='Render board bottom side.', is_flag=True)
|
||||||
|
@click.argument('input_zip_or_dir', type=click.Path(exists=True, path_type=Path))
|
||||||
|
@click.argument('output_svg', required=False, default='-', type=click.File('w'))
|
||||||
|
def pretty(input_zip_or_dir, output_svg, top, bottom):
|
||||||
|
if (bool(top) + bool(bottom)) != 1:
|
||||||
|
raise click.UsageError('Excactly one of --top or --bottom must be given when rendering a dir or zip of gerbers.')
|
||||||
|
|
||||||
|
stack = LayerStack.open(input_zip_or_dir, lazy=True)
|
||||||
|
print(f'Loaded {stack}')
|
||||||
|
|
||||||
|
svg = stack.to_pretty_svg(side=('top' if top else 'bottom'))
|
||||||
|
|
||||||
|
output_svg.write(str(svg))
|
||||||
|
|
||||||
|
@cli.command(help='Render an individual Gerber or Excellon file to SVG')
|
||||||
|
@click.option('-f', '--foreground', default='black', help='Foreground color')
|
||||||
|
@click.option('-b', '--background', default='white', help='Background color used for "clear" areas.')
|
||||||
|
@click.argument('input_gerber', type=click.Path(exists=True, dir_okay=False, path_type=Path))
|
||||||
|
@click.argument('output_svg', required=False, default='-', type=click.File('w'))
|
||||||
|
def render(input_gerber, output_svg, foreground, background):
|
||||||
|
layer = GerberFile.open(input_gerber)
|
||||||
|
output_svg.write(str(layer.to_svg(fg=foreground, bg=background)))
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
cli()
|
||||||
|
|
||||||
18
src/gerbonara/aperture_macros/__init__.py
Normal file
|
|
@ -0,0 +1,18 @@
|
||||||
|
|
||||||
|
from .parse import ApertureMacro, GenericMacros
|
||||||
|
from .expression import (Expression,
|
||||||
|
UnitExpression,
|
||||||
|
ConstantExpression,
|
||||||
|
VariableExpression,
|
||||||
|
ParameterExpression,
|
||||||
|
NegatedExpression,
|
||||||
|
OperatorExpression)
|
||||||
|
from .primitive import (Comment,
|
||||||
|
Circle,
|
||||||
|
VectorLine,
|
||||||
|
CenterLine,
|
||||||
|
Outline,
|
||||||
|
Polygon,
|
||||||
|
Moire,
|
||||||
|
Thermal)
|
||||||
|
|
||||||
|
|
@ -7,6 +7,7 @@ from dataclasses import dataclass
|
||||||
import operator
|
import operator
|
||||||
import re
|
import re
|
||||||
import ast
|
import ast
|
||||||
|
import math
|
||||||
|
|
||||||
from ..utils import LengthUnit, MM, Inch, MILLIMETERS_PER_INCH
|
from ..utils import LengthUnit, MM, Inch, MILLIMETERS_PER_INCH
|
||||||
|
|
||||||
|
|
@ -33,7 +34,7 @@ class Expression:
|
||||||
def calculate(self, variable_binding={}, unit=None):
|
def calculate(self, variable_binding={}, unit=None):
|
||||||
expr = self.converted(unit).optimized(variable_binding)
|
expr = self.converted(unit).optimized(variable_binding)
|
||||||
if not isinstance(expr, ConstantExpression):
|
if not isinstance(expr, ConstantExpression):
|
||||||
raise IndexError(f'Cannot fully resolve expression due to unresolved variables: {expr} with variables {variable_binding}')
|
raise IndexError(f'Cannot fully resolve expression due to unresolved parameters: residual expression {expr} under parameters {variable_binding}')
|
||||||
return expr.value
|
return expr.value
|
||||||
|
|
||||||
def __add__(self, other):
|
def __add__(self, other):
|
||||||
|
|
@ -61,11 +62,18 @@ class Expression:
|
||||||
return expr(other) / self
|
return expr(other) / self
|
||||||
|
|
||||||
def __neg__(self):
|
def __neg__(self):
|
||||||
return 0 - self
|
return NegatedExpression(self).optimized()
|
||||||
|
|
||||||
def __pos__(self):
|
def __pos__(self):
|
||||||
return self
|
return self
|
||||||
|
|
||||||
|
def parameters(self):
|
||||||
|
return tuple()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def _operator(self):
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True, slots=True)
|
@dataclass(frozen=True, slots=True)
|
||||||
class UnitExpression(Expression):
|
class UnitExpression(Expression):
|
||||||
|
|
@ -79,8 +87,8 @@ class UnitExpression(Expression):
|
||||||
object.__setattr__(self, 'expr', expr)
|
object.__setattr__(self, 'expr', expr)
|
||||||
object.__setattr__(self, 'unit', unit)
|
object.__setattr__(self, 'unit', unit)
|
||||||
|
|
||||||
def to_gerber(self, unit=None):
|
def to_gerber(self, register_variable=None, unit=None):
|
||||||
return self.converted(unit).optimized().to_gerber()
|
return self.converted(unit).optimized().to_gerber(register_variable)
|
||||||
|
|
||||||
def __eq__(self, other):
|
def __eq__(self, other):
|
||||||
return type(other) == type(self) and \
|
return type(other) == type(self) and \
|
||||||
|
|
@ -147,6 +155,10 @@ class UnitExpression(Expression):
|
||||||
def __pos__(self):
|
def __pos__(self):
|
||||||
return self
|
return self
|
||||||
|
|
||||||
|
def parameters(self):
|
||||||
|
return self.expr.parameters()
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True, slots=True)
|
@dataclass(frozen=True, slots=True)
|
||||||
class ConstantExpression(Expression):
|
class ConstantExpression(Expression):
|
||||||
value: float
|
value: float
|
||||||
|
|
@ -155,14 +167,44 @@ class ConstantExpression(Expression):
|
||||||
return float(self.value)
|
return float(self.value)
|
||||||
|
|
||||||
def __eq__(self, other):
|
def __eq__(self, other):
|
||||||
return type(self) == type(other) and self.value == other.value
|
try:
|
||||||
|
return math.isclose(self.value, float(other), abs_tol=1e-9)
|
||||||
|
except TypeError:
|
||||||
|
return False
|
||||||
|
|
||||||
def to_gerber(self, _unit=None):
|
def to_gerber(self, register_variable=None, unit=None):
|
||||||
|
if self == 0: # Avoid producing "-0" for negative floating point zeros
|
||||||
|
return '0'
|
||||||
return f'{self.value:.6f}'.rstrip('0').rstrip('.')
|
return f'{self.value:.6f}'.rstrip('0').rstrip('.')
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True, slots=True)
|
@dataclass(frozen=True, slots=True)
|
||||||
class VariableExpression(Expression):
|
class VariableExpression(Expression):
|
||||||
|
''' An expression that encapsulates some other complex expression and will replace all occurences of it with a newly
|
||||||
|
allocated variable at export time.
|
||||||
|
'''
|
||||||
|
expr: Expression
|
||||||
|
|
||||||
|
def optimized(self, variable_binding={}):
|
||||||
|
opt = self.expr.optimized(variable_binding)
|
||||||
|
if isinstance(opt, OperatorExpression):
|
||||||
|
return self
|
||||||
|
else:
|
||||||
|
return opt
|
||||||
|
|
||||||
|
def __eq__(self, other):
|
||||||
|
return type(self) == type(other) and self.expr == other.expr
|
||||||
|
|
||||||
|
def to_gerber(self, register_variable=None, unit=None):
|
||||||
|
if register_variable is None:
|
||||||
|
return self.expr.to_gerber(None, unit)
|
||||||
|
else:
|
||||||
|
num = register_variable(self.expr.converted(unit).optimized())
|
||||||
|
return f'${num}'
|
||||||
|
|
||||||
|
@dataclass(frozen=True, slots=True)
|
||||||
|
class ParameterExpression(Expression):
|
||||||
|
''' An expression that refers to a macro variable or parameter '''
|
||||||
number: int
|
number: int
|
||||||
|
|
||||||
def optimized(self, variable_binding={}):
|
def optimized(self, variable_binding={}):
|
||||||
|
|
@ -174,9 +216,50 @@ class VariableExpression(Expression):
|
||||||
return type(self) == type(other) and \
|
return type(self) == type(other) and \
|
||||||
self.number == other.number
|
self.number == other.number
|
||||||
|
|
||||||
def to_gerber(self, _unit=None):
|
def to_gerber(self, register_variable=None, unit=None):
|
||||||
return f'${self.number}'
|
return f'${self.number}'
|
||||||
|
|
||||||
|
def parameters(self):
|
||||||
|
yield self
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass(frozen=True, slots=True)
|
||||||
|
class NegatedExpression(Expression):
|
||||||
|
value: Expression
|
||||||
|
|
||||||
|
def optimized(self, variable_binding={}):
|
||||||
|
match self.value.optimized(variable_binding):
|
||||||
|
# -(-x) == x
|
||||||
|
case NegatedExpression(inner_value):
|
||||||
|
return inner_value
|
||||||
|
# -(x) == -x
|
||||||
|
case ConstantExpression(inner_value):
|
||||||
|
return ConstantExpression(-inner_value)
|
||||||
|
# -(x-y) == y-x
|
||||||
|
case OperatorExpression(operator.sub, l, r):
|
||||||
|
return OperatorExpression(operator.sub, r, l)
|
||||||
|
# Round very small values and negative floating point zeros to a (positive) zero
|
||||||
|
case 0:
|
||||||
|
return expr(0)
|
||||||
|
# Default case
|
||||||
|
case x:
|
||||||
|
return NegatedExpression(x)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def _operator(self):
|
||||||
|
return self.value._operator
|
||||||
|
|
||||||
|
def __eq__(self, other):
|
||||||
|
return type(self) == type(other) and \
|
||||||
|
self.value == other.value
|
||||||
|
|
||||||
|
def to_gerber(self, register_variable=None, unit=None):
|
||||||
|
val_str = self.value.to_gerber(register_variable, unit)
|
||||||
|
if isinstance(self.value, (VariableExpression, ParameterExpression)):
|
||||||
|
return f'-{val_str}'
|
||||||
|
else:
|
||||||
|
return f'-({val_str})'
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True, slots=True)
|
@dataclass(frozen=True, slots=True)
|
||||||
class OperatorExpression(Expression):
|
class OperatorExpression(Expression):
|
||||||
|
|
@ -195,22 +278,88 @@ class OperatorExpression(Expression):
|
||||||
self.l == other.l and \
|
self.l == other.l and \
|
||||||
self.r == other.r
|
self.r == other.r
|
||||||
|
|
||||||
|
@property
|
||||||
|
def _operator(self):
|
||||||
|
return self.op
|
||||||
|
|
||||||
def optimized(self, variable_binding={}):
|
def optimized(self, variable_binding={}):
|
||||||
l = self.l.optimized(variable_binding)
|
l = self.l.optimized(variable_binding)
|
||||||
r = self.r.optimized(variable_binding)
|
r = self.r.optimized(variable_binding)
|
||||||
|
|
||||||
#if self.op in (operator.add, operator.mul):
|
match (l, self.op, r):
|
||||||
# if id(r) < id(l):
|
case (ConstantExpression(), op, ConstantExpression()):
|
||||||
# l, r = r, l
|
return ConstantExpression(self.op(float(l), float(r)))
|
||||||
|
|
||||||
if isinstance(l, ConstantExpression) and isinstance(r, ConstantExpression):
|
# Minimize operations with neutral elements and zeros
|
||||||
return ConstantExpression(self.op(float(l), float(r)))
|
# 0 + x == x
|
||||||
|
case (0, operator.add, r):
|
||||||
|
return r
|
||||||
|
# x + 0 == x
|
||||||
|
case (l, operator.add, 0):
|
||||||
|
return l
|
||||||
|
# 0 * x == 0
|
||||||
|
case (0, operator.mul, r):
|
||||||
|
return expr(0)
|
||||||
|
# x * 0 == 0
|
||||||
|
case (l, operator.mul, 0):
|
||||||
|
return expr(0)
|
||||||
|
# x * 1 == x
|
||||||
|
case (l, operator.mul, 1):
|
||||||
|
return l
|
||||||
|
# 1 * x == x
|
||||||
|
case (1, operator.mul, r):
|
||||||
|
return r
|
||||||
|
# x * -1 == -x
|
||||||
|
case (l, operator.mul, -1):
|
||||||
|
rv = -l
|
||||||
|
# -1 * x == -x
|
||||||
|
case (-1, operator.mul, r):
|
||||||
|
rv = -r
|
||||||
|
# x - 0 == x
|
||||||
|
case (l, operator.sub, 0):
|
||||||
|
return l
|
||||||
|
# 0 - x == -x (unary minus)
|
||||||
|
case (0, operator.sub, r):
|
||||||
|
rv = -r
|
||||||
|
# x - x == 0
|
||||||
|
case (l, operator.sub, r) if l == r:
|
||||||
|
return expr(0)
|
||||||
|
# x - -y == x + y
|
||||||
|
case (l, operator.sub, NegatedExpression(r)):
|
||||||
|
rv = (l + r)
|
||||||
|
# x / 1 == x
|
||||||
|
case (l, operator.truediv, 1):
|
||||||
|
return l
|
||||||
|
# x / -1 == -x
|
||||||
|
case (l, operator.truediv, -1):
|
||||||
|
rv = -l
|
||||||
|
# x / x == 1
|
||||||
|
case (l, operator.truediv, r) if l == r:
|
||||||
|
return expr(1)
|
||||||
|
# -x [*/] -y == x [*/] y
|
||||||
|
case (NegatedExpression(l), (operator.truediv | operator.mul) as op, NegatedExpression(r)):
|
||||||
|
rv = op(l, r)
|
||||||
|
# -x [*/] y == -(x [*/] y)
|
||||||
|
case (NegatedExpression(l), (operator.truediv | operator.mul) as op, r):
|
||||||
|
rv = NegatedExpression(op(l, r))
|
||||||
|
# x [*/] -y == -(x [*/] y)
|
||||||
|
case (l, (operator.truediv | operator.mul) as op, NegatedExpression(r)):
|
||||||
|
rv = NegatedExpression(op(l, r))
|
||||||
|
# x + -y == x - y
|
||||||
|
case (l, operator.add, NegatedExpression(r)):
|
||||||
|
rv = l-r
|
||||||
|
# -x + y == y - x
|
||||||
|
case (NegatedExpression(l), operator.add, r):
|
||||||
|
rv = r-l
|
||||||
|
|
||||||
return OperatorExpression(self.op, l, r)
|
case _: # default
|
||||||
|
return OperatorExpression(self.op, l, r)
|
||||||
def to_gerber(self, unit=None):
|
|
||||||
lval = self.l.to_gerber(unit)
|
return expr(rv).optimized(variable_binding)
|
||||||
rval = self.r.to_gerber(unit)
|
|
||||||
|
def to_gerber(self, register_variable=None, unit=None):
|
||||||
|
lval = self.l.to_gerber(register_variable, unit)
|
||||||
|
rval = self.r.to_gerber(register_variable, unit)
|
||||||
|
|
||||||
if isinstance(self.l, OperatorExpression):
|
if isinstance(self.l, OperatorExpression):
|
||||||
lval = f'({lval})'
|
lval = f'({lval})'
|
||||||
|
|
@ -224,3 +373,7 @@ class OperatorExpression(Expression):
|
||||||
|
|
||||||
return f'{lval}{op}{rval}'
|
return f'{lval}{op}{rval}'
|
||||||
|
|
||||||
|
def parameters(self):
|
||||||
|
yield from self.l.parameters()
|
||||||
|
yield from self.r.parameters()
|
||||||
|
|
||||||
463
src/gerbonara/aperture_macros/parse.py
Normal file
|
|
@ -0,0 +1,463 @@
|
||||||
|
#!/usr/bin/env python
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
# Copyright 2021 Jan Sebastian Götte <gerbonara@jaseg.de>
|
||||||
|
|
||||||
|
from dataclasses import dataclass, field, replace, fields
|
||||||
|
import operator
|
||||||
|
import re
|
||||||
|
import ast
|
||||||
|
import copy
|
||||||
|
import warnings
|
||||||
|
import math
|
||||||
|
|
||||||
|
from . import primitive as ap
|
||||||
|
from .expression import *
|
||||||
|
from ..apertures import ApertureMacroInstance
|
||||||
|
from ..utils import MM
|
||||||
|
|
||||||
|
# we make our own here instead of using math.degrees to make sure this works with expressions, too.
|
||||||
|
def rad_to_deg(x):
|
||||||
|
return (x / math.pi) * 180
|
||||||
|
|
||||||
|
def _map_expression(node, variables={}, parameters=set()):
|
||||||
|
if isinstance(node, ast.Constant):
|
||||||
|
return ConstantExpression(node.value)
|
||||||
|
|
||||||
|
elif isinstance(node, ast.BinOp):
|
||||||
|
op_map = {ast.Add: operator.add, ast.Sub: operator.sub, ast.Mult: operator.mul, ast.Div: operator.truediv}
|
||||||
|
return OperatorExpression(op_map[type(node.op)],
|
||||||
|
_map_expression(node.left, variables, parameters),
|
||||||
|
_map_expression(node.right, variables, parameters))
|
||||||
|
|
||||||
|
elif isinstance(node, ast.UnaryOp):
|
||||||
|
if type(node.op) == ast.UAdd:
|
||||||
|
return _map_expression(node.operand, variables, parameters)
|
||||||
|
else:
|
||||||
|
return NegatedExpression(_map_expression(node.operand, variables, parameters))
|
||||||
|
|
||||||
|
elif isinstance(node, ast.Name):
|
||||||
|
num = int(node.id[3:]) # node.id has format var[0-9]+
|
||||||
|
if num in variables:
|
||||||
|
return VariableExpression(variables[num])
|
||||||
|
else:
|
||||||
|
parameters.add(num)
|
||||||
|
return ParameterExpression(num)
|
||||||
|
|
||||||
|
else:
|
||||||
|
raise SyntaxError('Invalid aperture macro expression')
|
||||||
|
|
||||||
|
def _parse_expression(expr, variables, parameters):
|
||||||
|
expr = expr.lower().replace('x', '*')
|
||||||
|
expr = re.sub(r'\$([0-9]+)', r'var\1', expr)
|
||||||
|
try:
|
||||||
|
parsed = ast.parse(expr, mode='eval').body
|
||||||
|
except SyntaxError as e:
|
||||||
|
raise SyntaxError('Invalid aperture macro expression') from e
|
||||||
|
return _map_expression(parsed, variables, parameters)
|
||||||
|
|
||||||
|
@dataclass(frozen=True, slots=True)
|
||||||
|
class ApertureMacro:
|
||||||
|
""" Definition of an aperture macro in a Gerber file.
|
||||||
|
|
||||||
|
An aperture macro is a collection of shape primitives that are flashed all at once. The properties of these
|
||||||
|
primitives such as their relative position and size can be given explicitly, or can be given as a basic
|
||||||
|
arithmetic expression (so +/-/*/:, no higher functions) based on parameters. After the macro is defined in the
|
||||||
|
Gerber file, it is *bound* to a particular set of parameter values in an aperture definition. One macro can be
|
||||||
|
used by zero, or by multiple aperture definitions. To flash a macro, you must first bind it in an aperture
|
||||||
|
definition, which can then be flash'ed.
|
||||||
|
|
||||||
|
Gerbonara calls these apertures that bind a macro :py:class:`~..apertures.ApertureMacroInst`. You can bind a
|
||||||
|
macro to a set of parameters by calling it:
|
||||||
|
|
||||||
|
.. code-block: python
|
||||||
|
|
||||||
|
# am is some instance of ApertureMacro
|
||||||
|
aperture_def = am(1, 2, 3)
|
||||||
|
gerber.objects.append(Flash(x=12, y=34, aperture=aperture_def))
|
||||||
|
|
||||||
|
Internally, the aperture macro API uses millimeters though most functions allow you to pass an unit parameter.
|
||||||
|
|
||||||
|
When you want to programmatically create aperture macros, we recommend using :py:meth:`~.ApertureMacro.map` on a
|
||||||
|
dataclass-like class definition. Have a look at this code from :py:class:`~.GenericMacros`:
|
||||||
|
|
||||||
|
.. code-block: python
|
||||||
|
|
||||||
|
@ApertureMacro.map('GNR')
|
||||||
|
class rect:
|
||||||
|
w: float # width
|
||||||
|
h: float # height
|
||||||
|
hole_dia: float = 0
|
||||||
|
rotation: float = 0
|
||||||
|
|
||||||
|
def draw(self):
|
||||||
|
yield ap.CenterLine('mm', 1, self.w, self.h, 0, 0, self.rotation * -deg_per_rad)
|
||||||
|
yield ap.Circle('mm', 0, self.hole_dia, 0, 0)
|
||||||
|
|
||||||
|
# rect now is an instance of ApertureMacro
|
||||||
|
|
||||||
|
After this, you can bind this macro to an aperture by calling it. When you use this dataclass-like syntax,
|
||||||
|
keyword arguments are supported, and default values work like with normal dataclasses:
|
||||||
|
|
||||||
|
.. code-block: python
|
||||||
|
|
||||||
|
# returns an instance of ApertureMacroInstance containing the given parameters
|
||||||
|
my_rect = GenericMacros.rect(w=12, h=34)
|
||||||
|
|
||||||
|
gerber.objects.append(Flash(x=12, y=34, aperture=my_rect))
|
||||||
|
|
||||||
|
.. important::
|
||||||
|
Use your own programmatically defined aperture macros sparingly. While support is getting better, many
|
||||||
|
tools, including the expensive, commercial tools that PCB manufacturers use, still have bugs when handling
|
||||||
|
aperture macros. When using advanced macros with many primitives or with complex, embedded arithmetic
|
||||||
|
expressions, make sure to carefully check the manufacturing files provided by your PCB fab.
|
||||||
|
|
||||||
|
gerbonara currently handles embedded arithmetic expressions by *always* calculating them out since we have
|
||||||
|
recently seen high-end commercial tooling failing at issues as basic as operator precedence. This increases
|
||||||
|
file sizes very very slightly, but it makes sure that you get correct results.
|
||||||
|
|
||||||
|
This means that you can use gerbonara to calculate out aperture macros and hard-bake their values into the
|
||||||
|
gerber source. This can be useful if you have a file that includes complex macros that some manufacturer's
|
||||||
|
tooling can't handle on its own.
|
||||||
|
"""
|
||||||
|
|
||||||
|
name: str = field(default=None, hash=False, compare=False)
|
||||||
|
num_parameters: int = 0
|
||||||
|
primitives: tuple = ()
|
||||||
|
comments: tuple = field(default=(), hash=False, compare=False)
|
||||||
|
_param_dataclass: object = field(default=None, hash=False, compare=False)
|
||||||
|
|
||||||
|
def __post_init__(self):
|
||||||
|
if self.name is None or re.match(r'GNX[0-9A-F]{16}', self.name):
|
||||||
|
# We can't use field(default_factory=...) here because that factory doesn't get a reference to the instance.
|
||||||
|
self._reset_name()
|
||||||
|
|
||||||
|
def _reset_name(self):
|
||||||
|
object.__setattr__(self, 'name', f'GNX{hash(self)&0xffffffffffffffff:016X}')
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def map(our_kls, macro_name=None):
|
||||||
|
def wrapper(kls):
|
||||||
|
nonlocal our_kls, macro_name
|
||||||
|
dc = dataclass(kls)
|
||||||
|
|
||||||
|
# Construct a mock instance of the dataclass with every field bound to its correpsonding ParameterExpression,
|
||||||
|
# then draw() it to get a list of bound macro primitives.
|
||||||
|
primitives = tuple(dc(*[ParameterExpression(i+1) for i in range(len(fields(dc)))]).draw())
|
||||||
|
name = macro_name if macro_name else f'GNM{kls.__name__}'
|
||||||
|
|
||||||
|
# Python allows a lot more unicode in class names than the Gerber spec allows in aperture macro names
|
||||||
|
if not re.fullmatch('[._$a-zA-Z][._$a-zA-Z0-9]{0,126}', name):
|
||||||
|
raise ValueError(f'Name {name!r} is invalid as an aperture macro name')
|
||||||
|
|
||||||
|
return our_kls(
|
||||||
|
name = name,
|
||||||
|
num_parameters = len(fields(dc)),
|
||||||
|
primitives = primitives,
|
||||||
|
comments = [l.strip() for l in dc.__doc__.strip().splitlines()],
|
||||||
|
_param_dataclass = dc)
|
||||||
|
return wrapper
|
||||||
|
|
||||||
|
def __call__(self, *args, unit=MM, **kwargs):
|
||||||
|
if self._param_dataclass:
|
||||||
|
# Above, in map(), we construct the dataclass with the ParameterExpression(i) as params to draw the macro
|
||||||
|
# primitives. Here, we construct it with the user's supplied concrete numeric parameters instead, and then
|
||||||
|
# extract a list of these parameters. This should work great as long as the user doesn't get too fancy with
|
||||||
|
# dataclass metaprogramming hackery.
|
||||||
|
bound = self._param_dataclass(*args, **kwargs)
|
||||||
|
return ApertureMacroInstance(macro=self, parameters=tuple(getattr(bound, f.name) or 0 for f in fields(bound)), unit=unit)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def parse_macro(kls, macro_name, body, unit):
|
||||||
|
comments = []
|
||||||
|
variables = {}
|
||||||
|
parameters = set()
|
||||||
|
primitives = []
|
||||||
|
|
||||||
|
blocks = body.split('*')
|
||||||
|
for block in blocks:
|
||||||
|
if not (block := block.strip()): # empty block
|
||||||
|
continue
|
||||||
|
|
||||||
|
if block.startswith('0 '): # comment
|
||||||
|
comments.append(block[2:])
|
||||||
|
continue
|
||||||
|
|
||||||
|
block = re.sub(r'\s', '', block)
|
||||||
|
|
||||||
|
if block[0] == '$': # variable definition
|
||||||
|
try:
|
||||||
|
name, _, expr = block.partition('=')
|
||||||
|
number = int(name[1:])
|
||||||
|
if number in variables:
|
||||||
|
warnings.warn(f'Re-definition of aperture macro variable ${number} inside aperture macro "{macro_name}". Previous definition of ${number} was ${variables[number]}.')
|
||||||
|
variables[number] = _parse_expression(expr, variables, parameters)
|
||||||
|
except Exception as e:
|
||||||
|
raise SyntaxError(f'Error parsing variable definition {block!r}') from e
|
||||||
|
|
||||||
|
else: # primitive
|
||||||
|
primitive, *args = block.split(',')
|
||||||
|
args = [ _parse_expression(arg, variables, parameters) for arg in args ]
|
||||||
|
try:
|
||||||
|
primitives.append(ap.PRIMITIVE_CLASSES[int(primitive)].from_arglist(unit, args))
|
||||||
|
except KeyError as e:
|
||||||
|
raise SyntaxError(f'Unknown aperture macro primitive code {int(primitive)}')
|
||||||
|
|
||||||
|
return kls(macro_name, max(parameters, default=0), tuple(primitives), tuple(comments))
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return f'<Aperture macro {self.name}, primitives {self.primitives}>'
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return str(self)
|
||||||
|
|
||||||
|
def dilated(self, offset, unit=MM):
|
||||||
|
new_primitives = []
|
||||||
|
for primitive in self.primitives:
|
||||||
|
try:
|
||||||
|
if primitive.exposure.calculate():
|
||||||
|
new_primitives += primitive.dilated(offset, unit)
|
||||||
|
except IndexError:
|
||||||
|
warnings.warn('Cannot dilate aperture macro primitive with exposure value computed from macro variable.')
|
||||||
|
pass
|
||||||
|
return replace(self, primitives=tuple(new_primitives))
|
||||||
|
|
||||||
|
def substitute_params(self, params, unit=None, macro_name=None):
|
||||||
|
params = dict(enumerate(params, start=1))
|
||||||
|
return replace(self,
|
||||||
|
num_parameters=0,
|
||||||
|
name=macro_name,
|
||||||
|
primitives=tuple(p.substitute_params(params, unit) for p in self.primitives),
|
||||||
|
comments=(f'Fully substituted instance of {self.name} macro',
|
||||||
|
f'Original parameters: {"X".join(map(str, params.values())) if params else "none"}'))
|
||||||
|
|
||||||
|
def to_gerber(self, settings):
|
||||||
|
""" Serialize this macro's content (without the name) into Gerber using the given file unit """
|
||||||
|
comments = [ f'0 {c.replace("*", "_").replace("%", "_")}' for c in self.comments ]
|
||||||
|
|
||||||
|
subexpression_variables = {}
|
||||||
|
def register_variable(expr):
|
||||||
|
expr_str = expr.to_gerber(register_variable, settings.unit)
|
||||||
|
if expr_str not in subexpression_variables:
|
||||||
|
subexpression_variables[expr_str] = self.num_parameters + 1 + len(subexpression_variables)
|
||||||
|
return subexpression_variables[expr_str]
|
||||||
|
|
||||||
|
primitive_defs = [prim.to_gerber(register_variable, settings) for prim in self.primitives]
|
||||||
|
variable_defs = [f'${num}={expr_str}' for expr_str, num in subexpression_variables.items()]
|
||||||
|
return '*\n'.join(comments + variable_defs + primitive_defs)
|
||||||
|
|
||||||
|
def to_graphic_primitives(self, offset, rotation, parameters : [float], unit=None, polarity_dark=True):
|
||||||
|
parameters = dict(enumerate(parameters, start=1))
|
||||||
|
for primitive in self.primitives:
|
||||||
|
yield from primitive.to_graphic_primitives(offset, rotation, parameters, unit, polarity_dark)
|
||||||
|
|
||||||
|
def rotated(self, angle):
|
||||||
|
# aperture macro primitives use degree counter-clockwise, our API uses radians clockwise
|
||||||
|
return replace(self, primitives=tuple(
|
||||||
|
replace(primitive, rotation=primitive.rotation - rad_to_deg(angle)) for primitive in self.primitives))
|
||||||
|
|
||||||
|
def scaled(self, scale):
|
||||||
|
return replace(self, primitives=tuple(
|
||||||
|
primitive.scaled(scale) for primitive in self.primitives))
|
||||||
|
|
||||||
|
|
||||||
|
var = ParameterExpression
|
||||||
|
deg_per_rad = 180 / math.pi
|
||||||
|
|
||||||
|
class GenericMacros:
|
||||||
|
"""NOTE:
|
||||||
|
All generic macros have rotation values specified in **clockwise radians** like the rest of the user-facing API.
|
||||||
|
"""
|
||||||
|
|
||||||
|
@ApertureMacro.map('GNC')
|
||||||
|
class circle:
|
||||||
|
""" Filled circle macro with an optional round hole
|
||||||
|
|
||||||
|
:param float diameter: Diameter of the circle
|
||||||
|
:param hole_dia: Diameter of the hole (optional)
|
||||||
|
"""
|
||||||
|
diameter: float
|
||||||
|
hole_dia: float = 0
|
||||||
|
|
||||||
|
def draw(self):
|
||||||
|
yield ap.Circle('mm', 1, self.diameter, 0, 0)
|
||||||
|
yield ap.Circle('mm', 0, self.hole_dia, 0, 0)
|
||||||
|
|
||||||
|
@ApertureMacro.map('GNR')
|
||||||
|
class rect:
|
||||||
|
""" Axis-aligned rectangle with an optional round center hole.
|
||||||
|
|
||||||
|
:param float w: Width
|
||||||
|
:param float h: Height
|
||||||
|
:param float hole_dia: Diameter of the round hole (optional)
|
||||||
|
:param float rotation: Rotation in clockwise radians (optional)
|
||||||
|
"""
|
||||||
|
w: float # width
|
||||||
|
h: float # height
|
||||||
|
hole_dia: float = 0
|
||||||
|
rotation: float = 0
|
||||||
|
|
||||||
|
def draw(self):
|
||||||
|
yield ap.CenterLine('mm', 1, self.w, self.h, 0, 0, self.rotation * -deg_per_rad)
|
||||||
|
yield ap.Circle('mm', 0, self.hole_dia, 0, 0)
|
||||||
|
|
||||||
|
@ApertureMacro.map('GRR')
|
||||||
|
class rounded_rect:
|
||||||
|
""" Rectangle with circular arc corners and an optional round center hole.
|
||||||
|
|
||||||
|
:param float w: Width
|
||||||
|
:param float h: Height
|
||||||
|
:param float r: Corner radius
|
||||||
|
:param float hole_dia: Diameter of the round hole (optional)
|
||||||
|
:param float rotation: Rotation in clockwise radians (optional)
|
||||||
|
"""
|
||||||
|
w: float # width
|
||||||
|
h: float # height
|
||||||
|
r: float # Corner radius
|
||||||
|
hole_dia: float = 0
|
||||||
|
rotation: float = 0
|
||||||
|
|
||||||
|
def draw(self):
|
||||||
|
yield ap.CenterLine('mm', 1, self.w-2*self.r, self.h, 0, 0, self.rotation * -deg_per_rad)
|
||||||
|
yield ap.CenterLine('mm', 1, self.w, self.h-2*self.r, 0, 0, self.rotation * -deg_per_rad)
|
||||||
|
yield ap.Circle('mm', 1, self.r*2, +(self.w/2-self.r), +(self.h/2-self.r), self.rotation * -deg_per_rad)
|
||||||
|
yield ap.Circle('mm', 1, self.r*2, +(self.w/2-self.r), -(self.h/2-self.r), self.rotation * -deg_per_rad)
|
||||||
|
yield ap.Circle('mm', 1, self.r*2, -(self.w/2-self.r), +(self.h/2-self.r), self.rotation * -deg_per_rad)
|
||||||
|
yield ap.Circle('mm', 1, self.r*2, -(self.w/2-self.r), -(self.h/2-self.r), self.rotation * -deg_per_rad)
|
||||||
|
yield ap.Circle('mm', 0, self.hole_dia, 0, 0)
|
||||||
|
|
||||||
|
@ApertureMacro.map('GTR')
|
||||||
|
class isosceles_trapezoid:
|
||||||
|
""" Isosceles trapezoid with a wider bottom edge and narrower top edge, with an optional round center hole.
|
||||||
|
|
||||||
|
:param float w: Width of the bottom (wider) edge
|
||||||
|
:param float h: Height
|
||||||
|
:param float d: Length difference between bottom and top edges; top width = w - d
|
||||||
|
:param float hole_dia: Diameter of the round hole (optional)
|
||||||
|
:param float rotation: Rotation in clockwise radians (optional)
|
||||||
|
"""
|
||||||
|
w: float # width
|
||||||
|
h: float # height
|
||||||
|
d: float # length difference between narrow side (top) and wide side (bottom)
|
||||||
|
hole_dia: float = 0
|
||||||
|
rotation: float = 0
|
||||||
|
|
||||||
|
def draw(self):
|
||||||
|
yield ap.Outline('mm', 1, 4,
|
||||||
|
(self.w/-2, self.h/-2,
|
||||||
|
self.w/-2+self.d/2, self.h/2,
|
||||||
|
self.w/2-self.d/2, self.h/2,
|
||||||
|
self.w/2, self.h/-2,
|
||||||
|
self.w/-2, self.h/-2,),
|
||||||
|
self.rotation * -deg_per_rad)
|
||||||
|
yield ap.Circle('mm', 0, self.hole_dia, 0, 0)
|
||||||
|
|
||||||
|
@ApertureMacro.map('GRTR')
|
||||||
|
class rounded_isosceles_trapezoid:
|
||||||
|
""" Isosceles trapezoid with rounded corners and an optional round center hole. Unlike the rounded rectangle, the shape is defined by first defining a non-rounded trapezoid, which is then offet to the outside by the given margin.
|
||||||
|
|
||||||
|
:param float w: Width of the bottom (wider) edge
|
||||||
|
:param float h: Height
|
||||||
|
:param float d: Length difference between bottom and top edges; top width = w - d
|
||||||
|
:param float margin: Corner rounding radius
|
||||||
|
:param float hole_dia: Diameter of the round hole (optional)
|
||||||
|
:param float rotation: Rotation in clockwise radians (optional)
|
||||||
|
"""
|
||||||
|
w: float
|
||||||
|
h: float
|
||||||
|
d: float # length difference between narrow side (top) and wide side (bottom)
|
||||||
|
margin: float
|
||||||
|
hole_dia: float = 0
|
||||||
|
rotation: float = 0
|
||||||
|
|
||||||
|
def draw(self):
|
||||||
|
rot = self.rotation * -deg_per_rad
|
||||||
|
yield ap.Outline('mm', 1, 4,
|
||||||
|
(self.w/-2, self.h/-2,
|
||||||
|
self.w/-2+self.d/2, self.h/2,
|
||||||
|
self.w/2-self.d/2, self.h/2,
|
||||||
|
self.w/2, self.h/-2,
|
||||||
|
self.w/-2, self.h/-2,),
|
||||||
|
rot)
|
||||||
|
|
||||||
|
yield ap.VectorLine('mm', 1, self.margin*2,
|
||||||
|
self.w/-2, self.h/-2,
|
||||||
|
self.w/-2+self.d/2, self.h/2,
|
||||||
|
rot)
|
||||||
|
yield ap.VectorLine('mm', 1, self.margin*2,
|
||||||
|
self.w/-2+self.d/2, self.h/2,
|
||||||
|
self.w/2-self.d/2, self.h/2,
|
||||||
|
rot)
|
||||||
|
yield ap.VectorLine('mm', 1, self.margin*2,
|
||||||
|
self.w/2-self.d/2, self.h/2,
|
||||||
|
self.w/2, self.h/-2,
|
||||||
|
rot)
|
||||||
|
yield ap.VectorLine('mm', 1, self.margin*2,
|
||||||
|
self.w/2, self.h/-2,
|
||||||
|
self.w/-2, self.h/-2,
|
||||||
|
rot)
|
||||||
|
|
||||||
|
yield ap.Circle('mm', 1, self.margin*2,
|
||||||
|
self.w/-2, self.h/-2,
|
||||||
|
rot)
|
||||||
|
yield ap.Circle('mm', 1, self.margin*2,
|
||||||
|
self.w/-2+self.d/2, self.h/2,
|
||||||
|
rot)
|
||||||
|
yield ap.Circle('mm', 1, self.margin*2,
|
||||||
|
self.w/2-self.d/2, self.h/2,
|
||||||
|
rot)
|
||||||
|
yield ap.Circle('mm', 1, self.margin*2,
|
||||||
|
self.w/2, self.h/-2,
|
||||||
|
rot)
|
||||||
|
|
||||||
|
yield ap.Circle('mm', 0, self.hole_dia, 0, 0)
|
||||||
|
|
||||||
|
@ApertureMacro.map('GNO')
|
||||||
|
class obround:
|
||||||
|
""" Rectangle with semicircular end caps (stadium shape), with an optional round center hole. The long axis is along the X axis when rotation is zero.
|
||||||
|
|
||||||
|
:param float w: Total width including end caps; must satisfy w >= h
|
||||||
|
:param float h: Height, equal to the end cap diameter
|
||||||
|
:param float hole_dia: Diameter of the round hole (optional)
|
||||||
|
:param float rotation: Rotation in clockwise radians (optional)
|
||||||
|
"""
|
||||||
|
w: float
|
||||||
|
h: float
|
||||||
|
hole_dia: float = 0
|
||||||
|
rotation: float = 0
|
||||||
|
|
||||||
|
def draw(self):
|
||||||
|
rot = self.rotation * -deg_per_rad
|
||||||
|
yield ap.CenterLine('mm', 1, self.w - self.h, self.h, 0, 0, rot)
|
||||||
|
yield ap.Circle('mm', 1, self.h, +(self.w-self.h)/2, 0, rot)
|
||||||
|
yield ap.Circle('mm', 1, self.h, -(self.w-self.h)/2, 0, rot)
|
||||||
|
yield ap.Circle('mm', 0, self.hole_dia, 0, 0)
|
||||||
|
|
||||||
|
@ApertureMacro.map('GNP')
|
||||||
|
class polygon:
|
||||||
|
""" Regular n-sided polygon with an optional round center hole.
|
||||||
|
|
||||||
|
:param int n: Number of sides
|
||||||
|
:param float diameter: Diameter of the circumscribed circle
|
||||||
|
:param float hole_dia: Diameter of the round hole (optional)
|
||||||
|
:param float rotation: Rotation in clockwise radians (optional)
|
||||||
|
"""
|
||||||
|
n: int
|
||||||
|
diameter: float
|
||||||
|
hole_dia: float = 0
|
||||||
|
rotation: float = 0
|
||||||
|
|
||||||
|
def draw(self):
|
||||||
|
yield ap.Polygon('mm', 1, self.diameter, 0, 0, self.n, self.rotation * -deg_per_rad)
|
||||||
|
yield ap.Circle('mm', 0, self.hole_dia, 0, 0)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
import sys
|
||||||
|
#for line in sys.stdin:
|
||||||
|
#expr = _parse_expression(line.strip())
|
||||||
|
#print(expr, '->', expr.optimized())
|
||||||
|
|
||||||
|
for primitive in parse_macro(sys.stdin.read(), 'mm'):
|
||||||
|
print(primitive)
|
||||||
|
|
||||||
|
|
@ -7,7 +7,7 @@
|
||||||
import warnings
|
import warnings
|
||||||
import contextlib
|
import contextlib
|
||||||
import math
|
import math
|
||||||
from dataclasses import dataclass, fields
|
from dataclasses import dataclass, fields, replace
|
||||||
|
|
||||||
from .expression import Expression, UnitExpression, ConstantExpression, expr
|
from .expression import Expression, UnitExpression, ConstantExpression, expr
|
||||||
|
|
||||||
|
|
@ -34,7 +34,6 @@ def rad_to_deg(a):
|
||||||
@dataclass(frozen=True, slots=True)
|
@dataclass(frozen=True, slots=True)
|
||||||
class Primitive:
|
class Primitive:
|
||||||
unit: LengthUnit
|
unit: LengthUnit
|
||||||
exposure : Expression
|
|
||||||
|
|
||||||
def __post_init__(self):
|
def __post_init__(self):
|
||||||
for field in fields(self):
|
for field in fields(self):
|
||||||
|
|
@ -46,9 +45,16 @@ class Primitive:
|
||||||
elif field.type == Expression:
|
elif field.type == Expression:
|
||||||
object.__setattr__(self, field.name, expr(getattr(self, field.name)))
|
object.__setattr__(self, field.name, expr(getattr(self, field.name)))
|
||||||
|
|
||||||
def to_gerber(self, unit=None):
|
def to_gerber(self, register_variable=None, settings=None):
|
||||||
return f'{self.code},' + ','.join(
|
return f'{self.code},' + ','.join(
|
||||||
getattr(self, field.name).to_gerber(unit) for field in fields(self) if field.name != 'unit')
|
getattr(self, field.name).optimized().to_gerber(register_variable, settings.unit)
|
||||||
|
for field in fields(self) if issubclass(field.type, Expression))
|
||||||
|
|
||||||
|
def substitute_params(self, binding, unit):
|
||||||
|
out = replace(self, unit=unit, **{
|
||||||
|
field.name: getattr(self, field.name).calculate(binding, unit)
|
||||||
|
for field in fields(self) if issubclass(field.type, Expression)})
|
||||||
|
return out
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
attrs = ','.join(str(getattr(self, name)).strip('<>') for name in type(self).__annotations__)
|
attrs = ','.join(str(getattr(self, name)).strip('<>') for name in type(self).__annotations__)
|
||||||
|
|
@ -61,6 +67,11 @@ class Primitive:
|
||||||
def from_arglist(kls, unit, arglist):
|
def from_arglist(kls, unit, arglist):
|
||||||
return kls(unit, *arglist)
|
return kls(unit, *arglist)
|
||||||
|
|
||||||
|
def parameters(self):
|
||||||
|
for field in fields(self):
|
||||||
|
if issubclass(field.type, Expression):
|
||||||
|
yield from getattr(self, field.name).parameters()
|
||||||
|
|
||||||
class Calculator:
|
class Calculator:
|
||||||
def __init__(self, instance, variable_binding={}, unit=None):
|
def __init__(self, instance, variable_binding={}, unit=None):
|
||||||
self.instance = instance
|
self.instance = instance
|
||||||
|
|
@ -83,18 +94,29 @@ class Primitive:
|
||||||
@dataclass(frozen=True, slots=True)
|
@dataclass(frozen=True, slots=True)
|
||||||
class Circle(Primitive):
|
class Circle(Primitive):
|
||||||
code = 1
|
code = 1
|
||||||
|
exposure : Expression
|
||||||
diameter : UnitExpression
|
diameter : UnitExpression
|
||||||
# center x/y
|
# center x/y
|
||||||
x : UnitExpression
|
x : UnitExpression = 0
|
||||||
y : UnitExpression
|
y : UnitExpression = 0
|
||||||
rotation : Expression = 0
|
rotation : Expression = 0
|
||||||
|
|
||||||
def to_graphic_primitives(self, offset, rotation, variable_binding={}, unit=None, polarity_dark=True):
|
def to_graphic_primitives(self, offset, rotation, variable_binding={}, unit=None, polarity_dark=True):
|
||||||
with self.Calculator(self, variable_binding, unit) as calc:
|
with self.Calculator(self, variable_binding, unit) as calc:
|
||||||
x, y = rotate_point(calc.x, calc.y, -(deg_to_rad(calc.rotation) + rotation), 0, 0)
|
x, y = rotate_point(calc.x, calc.y, -(deg_to_rad(calc.rotation) + rotation), 0, 0)
|
||||||
x, y = x+offset[0], y+offset[1]
|
x, y = x+offset[0], y+offset[1]
|
||||||
|
|
||||||
|
if math.isclose(calc.diameter, 0):
|
||||||
|
return []
|
||||||
|
|
||||||
return [ gp.Circle(x, y, calc.diameter/2, polarity_dark=(bool(calc.exposure) == polarity_dark)) ]
|
return [ gp.Circle(x, y, calc.diameter/2, polarity_dark=(bool(calc.exposure) == polarity_dark)) ]
|
||||||
|
|
||||||
|
def substitute_params(self, binding, unit):
|
||||||
|
with self.Calculator(self, binding, unit) as calc:
|
||||||
|
x, y = rotate_point(calc.x, calc.y, -deg_to_rad(calc.rotation), 0, 0)
|
||||||
|
new = Circle(unit, self.exposure, calc.diameter, x, y)
|
||||||
|
return new
|
||||||
|
|
||||||
def dilated(self, offset, unit):
|
def dilated(self, offset, unit):
|
||||||
return replace(self, diameter=self.diameter + UnitExpression(offset, unit))
|
return replace(self, diameter=self.diameter + UnitExpression(offset, unit))
|
||||||
|
|
||||||
|
|
@ -106,6 +128,7 @@ class Circle(Primitive):
|
||||||
@dataclass(frozen=True, slots=True)
|
@dataclass(frozen=True, slots=True)
|
||||||
class VectorLine(Primitive):
|
class VectorLine(Primitive):
|
||||||
code = 20
|
code = 20
|
||||||
|
exposure : Expression
|
||||||
width : UnitExpression
|
width : UnitExpression
|
||||||
start_x : UnitExpression
|
start_x : UnitExpression
|
||||||
start_y : UnitExpression
|
start_y : UnitExpression
|
||||||
|
|
@ -125,9 +148,18 @@ class VectorLine(Primitive):
|
||||||
center_x, center_y = center_x+offset[0], center_y+offset[1]
|
center_x, center_y = center_x+offset[0], center_y+offset[1]
|
||||||
rotation += deg_to_rad(calc.rotation) + math.atan2(delta_y, delta_x)
|
rotation += deg_to_rad(calc.rotation) + math.atan2(delta_y, delta_x)
|
||||||
|
|
||||||
|
if math.isclose(calc.width, 0):
|
||||||
|
return []
|
||||||
|
|
||||||
return [ gp.Rectangle(center_x, center_y, length, calc.width, rotation=rotation,
|
return [ gp.Rectangle(center_x, center_y, length, calc.width, rotation=rotation,
|
||||||
polarity_dark=(bool(calc.exposure) == polarity_dark)) ]
|
polarity_dark=(bool(calc.exposure) == polarity_dark)) ]
|
||||||
|
|
||||||
|
def substitute_params(self, binding, unit):
|
||||||
|
with self.Calculator(self, binding, unit) as calc:
|
||||||
|
x1, y1 = rotate_point(calc.start_x, calc.start_y, -deg_to_rad(calc.rotation), 0, 0)
|
||||||
|
x2, y2 = rotate_point(calc.end_x, calc.end_y, -deg_to_rad(calc.rotation), 0, 0)
|
||||||
|
return VectorLine(unit, calc.exposure, calc.width, x1, y1, x2, y2)
|
||||||
|
|
||||||
def dilated(self, offset, unit):
|
def dilated(self, offset, unit):
|
||||||
return replace(self, width=self.width + UnitExpression(2*offset, unit))
|
return replace(self, width=self.width + UnitExpression(2*offset, unit))
|
||||||
|
|
||||||
|
|
@ -142,6 +174,7 @@ class VectorLine(Primitive):
|
||||||
@dataclass(frozen=True, slots=True)
|
@dataclass(frozen=True, slots=True)
|
||||||
class CenterLine(Primitive):
|
class CenterLine(Primitive):
|
||||||
code = 21
|
code = 21
|
||||||
|
exposure : Expression
|
||||||
width : UnitExpression
|
width : UnitExpression
|
||||||
height : UnitExpression
|
height : UnitExpression
|
||||||
# center x/y
|
# center x/y
|
||||||
|
|
@ -156,8 +189,17 @@ class CenterLine(Primitive):
|
||||||
x, y = x+offset[0], y+offset[1]
|
x, y = x+offset[0], y+offset[1]
|
||||||
w, h = calc.width, calc.height
|
w, h = calc.width, calc.height
|
||||||
|
|
||||||
|
if math.isclose(calc.width, 0) or math.isclose(calc.height, 0):
|
||||||
|
return []
|
||||||
|
|
||||||
return [ gp.Rectangle(x, y, w, h, rotation, polarity_dark=(bool(calc.exposure) == polarity_dark)) ]
|
return [ gp.Rectangle(x, y, w, h, rotation, polarity_dark=(bool(calc.exposure) == polarity_dark)) ]
|
||||||
|
|
||||||
|
def substitute_params(self, binding, unit):
|
||||||
|
with self.Calculator(self, binding, unit) as calc:
|
||||||
|
x1, y1 = rotate_point(calc.x, calc.y-calc.height/2, -deg_to_rad(calc.rotation), 0, 0)
|
||||||
|
x2, y2 = rotate_point(calc.x, calc.y+calc.height/2, -deg_to_rad(calc.rotation), 0, 0)
|
||||||
|
return VectorLine(unit, calc.exposure, calc.width, x1, y1, x2, y2)
|
||||||
|
|
||||||
def dilated(self, offset, unit):
|
def dilated(self, offset, unit):
|
||||||
return replace(self, width=self.width + UnitExpression(2*offset, unit))
|
return replace(self, width=self.width + UnitExpression(2*offset, unit))
|
||||||
|
|
||||||
|
|
@ -172,6 +214,7 @@ class CenterLine(Primitive):
|
||||||
@dataclass(frozen=True, slots=True)
|
@dataclass(frozen=True, slots=True)
|
||||||
class Polygon(Primitive):
|
class Polygon(Primitive):
|
||||||
code = 5
|
code = 5
|
||||||
|
exposure : Expression
|
||||||
n_vertices : Expression
|
n_vertices : Expression
|
||||||
# center x/y
|
# center x/y
|
||||||
x : UnitExpression
|
x : UnitExpression
|
||||||
|
|
@ -184,7 +227,8 @@ class Polygon(Primitive):
|
||||||
rotation += deg_to_rad(calc.rotation)
|
rotation += deg_to_rad(calc.rotation)
|
||||||
x, y = rotate_point(calc.x, calc.y, -rotation, 0, 0)
|
x, y = rotate_point(calc.x, calc.y, -rotation, 0, 0)
|
||||||
x, y = x+offset[0], y+offset[1]
|
x, y = x+offset[0], y+offset[1]
|
||||||
return [ gp.ArcPoly.from_regular_polygon(calc.x, calc.y, calc.diameter/2, calc.n_vertices, rotation,
|
print('xy', calc.x, calc.y)
|
||||||
|
return [ gp.ArcPoly.from_regular_polygon(x, y, calc.diameter/2, int(calc.n_vertices), rotation,
|
||||||
polarity_dark=(bool(calc.exposure) == polarity_dark)) ]
|
polarity_dark=(bool(calc.exposure) == polarity_dark)) ]
|
||||||
|
|
||||||
def dilated(self, offset, unit):
|
def dilated(self, offset, unit):
|
||||||
|
|
@ -197,9 +241,60 @@ class Polygon(Primitive):
|
||||||
y=self.y * UnitExpression(scale))
|
y=self.y * UnitExpression(scale))
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass(frozen=True, slots=True)
|
||||||
|
class Moire(Primitive):
|
||||||
|
""" Deprecated, but still found in some really old gerber files. """
|
||||||
|
code = 6
|
||||||
|
# center x/y
|
||||||
|
x : UnitExpression
|
||||||
|
y : UnitExpression
|
||||||
|
d_outer : UnitExpression
|
||||||
|
line_thickness : UnitExpression
|
||||||
|
gap_w : UnitExpression
|
||||||
|
num_circles : Expression
|
||||||
|
crosshair_thickness : UnitExpression = 0
|
||||||
|
crosshair_length : UnitExpression =0
|
||||||
|
rotation : Expression = 0
|
||||||
|
|
||||||
|
def to_graphic_primitives(self, offset, rotation, variable_binding={}, unit=None, polarity_dark=True):
|
||||||
|
with self.Calculator(self, variable_binding, unit) as calc:
|
||||||
|
rotation += deg_to_rad(calc.rotation)
|
||||||
|
x, y = rotate_point(calc.x, calc.y, -rotation, 0, 0)
|
||||||
|
x, y = x+offset[0], y+offset[1]
|
||||||
|
|
||||||
|
if math.isclose(calc.d_outer, 0):
|
||||||
|
return []
|
||||||
|
|
||||||
|
pitch = calc.line_thickness + calc.gap_w
|
||||||
|
for i in range(int(round(calc.num_circles))):
|
||||||
|
yield gp.Circle(x, y, calc.d_outer/2 - i*pitch, polarity_dark=True)
|
||||||
|
yield gp.Circle(x, y, calc.d_inner/2 - i*pitch - calc.line_thickness, polarity_dark=False)
|
||||||
|
|
||||||
|
if math.isclose(calc.crosshair_thickness, 0, abs_tol=1e-6) or\
|
||||||
|
math.isclose(calc.crosshair_length, 0, abs_tol=1e-6):
|
||||||
|
return
|
||||||
|
|
||||||
|
yield gp.Rectangle(x, y, crosshair_length, crosshair_thickness, rotation=rotation, polarity_dark=True)
|
||||||
|
yield gp.Rectangle(x, y, crosshair_thickness, crosshair_length, rotation=rotation, polarity_dark=True)
|
||||||
|
|
||||||
|
def dilate(self, offset, unit):
|
||||||
|
# I'd rather print a warning and produce graphically slightly incorrect output in these few cases here than
|
||||||
|
# producing macros that may evaluate to primitives with negative values.
|
||||||
|
warnings.warn('Attempted dilation of macro aperture thermal primitive. This is not supported.')
|
||||||
|
|
||||||
|
def scale(self, scale):
|
||||||
|
return replace(self,
|
||||||
|
d_outer=self.d_outer * UnitExpression(scale),
|
||||||
|
d_inner=self.d_inner * UnitExpression(scale),
|
||||||
|
gap_w=self.gap_w * UnitExpression(scale),
|
||||||
|
x=self.x * UnitExpression(scale),
|
||||||
|
y=self.y * UnitExpression(scale))
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True, slots=True)
|
@dataclass(frozen=True, slots=True)
|
||||||
class Thermal(Primitive):
|
class Thermal(Primitive):
|
||||||
code = 7
|
code = 7
|
||||||
|
# Note: Thermal primitives according to spec don't have an exposure variable
|
||||||
# center x/y
|
# center x/y
|
||||||
x : UnitExpression
|
x : UnitExpression
|
||||||
y : UnitExpression
|
y : UnitExpression
|
||||||
|
|
@ -214,13 +309,16 @@ class Thermal(Primitive):
|
||||||
x, y = rotate_point(calc.x, calc.y, -rotation, 0, 0)
|
x, y = rotate_point(calc.x, calc.y, -rotation, 0, 0)
|
||||||
x, y = x+offset[0], y+offset[1]
|
x, y = x+offset[0], y+offset[1]
|
||||||
|
|
||||||
dark = (bool(calc.exposure) == polarity_dark)
|
dark = True
|
||||||
|
|
||||||
|
if math.isclose(calc.d_outer, 0):
|
||||||
|
return []
|
||||||
|
|
||||||
return [
|
return [
|
||||||
gp.Circle(x, y, calc.d_outer/2, polarity_dark=dark),
|
gp.Circle(x, y, calc.d_outer/2, polarity_dark=dark),
|
||||||
gp.Circle(x, y, calc.d_inner/2, polarity_dark=not dark),
|
gp.Circle(x, y, calc.d_inner/2, polarity_dark=not dark),
|
||||||
gp.Rectangle(x, y, d_outer, gap_w, rotation=rotation, polarity_dark=not dark),
|
gp.Rectangle(x, y, calc.d_outer, calc.gap_w, rotation=rotation, polarity_dark=not dark),
|
||||||
gp.Rectangle(x, y, gap_w, d_outer, rotation=rotation, polarity_dark=not dark),
|
gp.Rectangle(x, y, calc.gap_w, calc.d_outer, rotation=rotation, polarity_dark=not dark),
|
||||||
]
|
]
|
||||||
|
|
||||||
def dilate(self, offset, unit):
|
def dilate(self, offset, unit):
|
||||||
|
|
@ -240,6 +338,7 @@ class Thermal(Primitive):
|
||||||
@dataclass(frozen=True, slots=True)
|
@dataclass(frozen=True, slots=True)
|
||||||
class Outline(Primitive):
|
class Outline(Primitive):
|
||||||
code = 4
|
code = 4
|
||||||
|
exposure : Expression
|
||||||
length: Expression
|
length: Expression
|
||||||
coords: tuple
|
coords: tuple
|
||||||
rotation: Expression = 0
|
rotation: Expression = 0
|
||||||
|
|
@ -276,21 +375,24 @@ class Outline(Primitive):
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return f'<Outline {len(self.coords)} points>'
|
return f'<Outline {len(self.coords)} points>'
|
||||||
|
|
||||||
def to_gerber(self, unit=None):
|
def to_gerber(self, register_variable=None, settings=None):
|
||||||
# Calculate out rotation since at least gerbv mis-renders Outlines with rotation other than zero.
|
|
||||||
rotation = self.rotation.optimized()
|
rotation = self.rotation.optimized()
|
||||||
coords = self.coords
|
coords = ','.join(coord.optimized().to_gerber(register_variable, settings.unit) for coord in self.coords)
|
||||||
if isinstance(rotation, ConstantExpression):
|
return f'{self.code},{self.exposure.optimized().to_gerber(register_variable)},{len(self.coords)//2-1},{coords},{rotation.to_gerber(register_variable)}'
|
||||||
rotation = math.radians(rotation.value)
|
|
||||||
# This will work even with variables in x and y, we just need to pass in cx and cy as UnitExpressions
|
def substitute_params(self, binding, unit):
|
||||||
unit_zero = UnitExpression(expr(0), MM)
|
with self.Calculator(self, binding, unit) as calc:
|
||||||
coords = [ rotate_point(x, y, -rotation, cx=unit_zero, cy=unit_zero) for x, y in self.points ]
|
rotation = calc.rotation
|
||||||
|
coords = [ rotate_point(x.calculate(binding, unit), y.calculate(binding, unit), -deg_to_rad(rotation), 0, 0)
|
||||||
|
for x, y in self.points ]
|
||||||
coords = [ e for point in coords for e in point ]
|
coords = [ e for point in coords for e in point ]
|
||||||
|
return Outline(unit, calc.exposure, calc.length, coords)
|
||||||
|
|
||||||
rotation = ConstantExpression(0)
|
def parameters(self):
|
||||||
|
yield from Primitive.parameters(self)
|
||||||
|
|
||||||
coords = ','.join(coord.to_gerber(unit) for coord in coords)
|
for expr in self.coords:
|
||||||
return f'{self.code},{self.exposure.to_gerber()},{len(self.coords)//2-1},{coords},{rotation.to_gerber()}'
|
yield from expr.parameters()
|
||||||
|
|
||||||
def to_graphic_primitives(self, offset, rotation, variable_binding={}, unit=None, polarity_dark=True):
|
def to_graphic_primitives(self, offset, rotation, variable_binding={}, unit=None, polarity_dark=True):
|
||||||
with self.Calculator(self, variable_binding, unit) as calc:
|
with self.Calculator(self, variable_binding, unit) as calc:
|
||||||
|
|
@ -298,6 +400,10 @@ class Outline(Primitive):
|
||||||
bound_coords = [ rotate_point(calc(x), calc(y), -rotation, 0, 0) for x, y in self.points ]
|
bound_coords = [ rotate_point(calc(x), calc(y), -rotation, 0, 0) for x, y in self.points ]
|
||||||
bound_coords = [ (x+offset[0], y+offset[1]) for x, y in bound_coords ]
|
bound_coords = [ (x+offset[0], y+offset[1]) for x, y in bound_coords ]
|
||||||
bound_radii = [None] * len(bound_coords)
|
bound_radii = [None] * len(bound_coords)
|
||||||
|
|
||||||
|
if len(bound_coords) < 3:
|
||||||
|
return []
|
||||||
|
|
||||||
return [gp.ArcPoly(bound_coords, bound_radii, polarity_dark=(bool(calc.exposure) == polarity_dark))]
|
return [gp.ArcPoly(bound_coords, bound_radii, polarity_dark=(bool(calc.exposure) == polarity_dark))]
|
||||||
|
|
||||||
def dilated(self, offset, unit):
|
def dilated(self, offset, unit):
|
||||||
|
|
@ -313,7 +419,7 @@ class Comment:
|
||||||
code = 0
|
code = 0
|
||||||
comment: str
|
comment: str
|
||||||
|
|
||||||
def to_gerber(self, unit=None):
|
def to_gerber(self, register_variable=None, settings=None):
|
||||||
return f'0 {self.comment}'
|
return f'0 {self.comment}'
|
||||||
|
|
||||||
def dilated(self, offset, unit):
|
def dilated(self, offset, unit):
|
||||||
|
|
@ -331,6 +437,7 @@ PRIMITIVE_CLASSES = {
|
||||||
CenterLine,
|
CenterLine,
|
||||||
Outline,
|
Outline,
|
||||||
Polygon,
|
Polygon,
|
||||||
|
Moire,
|
||||||
Thermal,
|
Thermal,
|
||||||
]},
|
]},
|
||||||
# alternative codes
|
# alternative codes
|
||||||
|
|
@ -16,11 +16,11 @@
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
#
|
#
|
||||||
|
|
||||||
|
import warnings
|
||||||
import math
|
import math
|
||||||
from dataclasses import dataclass, replace, field, fields, InitVar, KW_ONLY
|
from dataclasses import dataclass, replace, field, fields, InitVar, KW_ONLY
|
||||||
from functools import lru_cache
|
from functools import lru_cache
|
||||||
|
|
||||||
from .aperture_macros.parse import GenericMacros
|
|
||||||
from .utils import LengthUnit, MM, Inch, sum_bounds
|
from .utils import LengthUnit, MM, Inch, sum_bounds
|
||||||
|
|
||||||
from . import graphic_primitives as gp
|
from . import graphic_primitives as gp
|
||||||
|
|
@ -150,6 +150,8 @@ class ExcellonTool(Aperture):
|
||||||
# Internal use, for layer dilation.
|
# Internal use, for layer dilation.
|
||||||
def dilated(self, offset, unit=MM):
|
def dilated(self, offset, unit=MM):
|
||||||
offset = unit(offset, self.unit)
|
offset = unit(offset, self.unit)
|
||||||
|
if math.isclose(offset, 0, abs_tol=1e-6):
|
||||||
|
return self
|
||||||
return replace(self, diameter=self.diameter+2*offset)
|
return replace(self, diameter=self.diameter+2*offset)
|
||||||
|
|
||||||
@lru_cache()
|
@lru_cache()
|
||||||
|
|
@ -157,7 +159,8 @@ class ExcellonTool(Aperture):
|
||||||
return self
|
return self
|
||||||
|
|
||||||
def to_macro(self, rotation=0):
|
def to_macro(self, rotation=0):
|
||||||
return ApertureMacroInstance(GenericMacros.circle, self._params(unit=MM))
|
from .aperture_macros.parse import GenericMacros
|
||||||
|
return ApertureMacroInstance(GenericMacros.circle, self._params(unit=MM), unit=MM)
|
||||||
|
|
||||||
def _params(self, unit=None):
|
def _params(self, unit=None):
|
||||||
return (self.unit.convert_to(unit, self.diameter),)
|
return (self.unit.convert_to(unit, self.diameter),)
|
||||||
|
|
@ -188,6 +191,8 @@ class CircleAperture(Aperture):
|
||||||
|
|
||||||
def dilated(self, offset, unit=MM):
|
def dilated(self, offset, unit=MM):
|
||||||
offset = self.unit(offset, unit)
|
offset = self.unit(offset, unit)
|
||||||
|
if math.isclose(offset, 0, abs_tol=1e-6):
|
||||||
|
return self
|
||||||
return replace(self, diameter=self.diameter+2*offset, hole_dia=None)
|
return replace(self, diameter=self.diameter+2*offset, hole_dia=None)
|
||||||
|
|
||||||
@lru_cache()
|
@lru_cache()
|
||||||
|
|
@ -200,7 +205,9 @@ class CircleAperture(Aperture):
|
||||||
hole_dia=None if self.hole_dia is None else self.hole_dia*scale)
|
hole_dia=None if self.hole_dia is None else self.hole_dia*scale)
|
||||||
|
|
||||||
def to_macro(self, rotation=0):
|
def to_macro(self, rotation=0):
|
||||||
return ApertureMacroInstance(GenericMacros.circle, self._params(unit=MM))
|
from .aperture_macros.parse import GenericMacros
|
||||||
|
return GenericMacros.circle(MM(self.diameter, self.unit),
|
||||||
|
MM(self.hole_dia, self.unit))
|
||||||
|
|
||||||
def _params(self, unit=None):
|
def _params(self, unit=None):
|
||||||
return _strip_right(
|
return _strip_right(
|
||||||
|
|
@ -235,13 +242,15 @@ class RectangleAperture(Aperture):
|
||||||
|
|
||||||
def dilated(self, offset, unit=MM):
|
def dilated(self, offset, unit=MM):
|
||||||
offset = self.unit(offset, unit)
|
offset = self.unit(offset, unit)
|
||||||
|
if math.isclose(offset, 0, abs_tol=1e-6):
|
||||||
|
return self
|
||||||
return replace(self, w=self.w+2*offset, h=self.h+2*offset, hole_dia=None)
|
return replace(self, w=self.w+2*offset, h=self.h+2*offset, hole_dia=None)
|
||||||
|
|
||||||
@lru_cache()
|
@lru_cache()
|
||||||
def rotated(self, angle=0):
|
def rotated(self, angle=0):
|
||||||
if math.isclose(angle % math.pi, 0):
|
if math.isclose(angle % math.pi, 0, abs_tol=1e-6):
|
||||||
return self
|
return self
|
||||||
elif math.isclose(angle % math.pi, math.pi/2):
|
elif math.isclose(angle % math.pi, math.pi/2, abs_tol=1e-6):
|
||||||
return replace(self, w=self.h, h=self.w, hole_dia=self.hole_dia)
|
return replace(self, w=self.h, h=self.w, hole_dia=self.hole_dia)
|
||||||
else: # odd angle
|
else: # odd angle
|
||||||
return self.to_macro(angle)
|
return self.to_macro(angle)
|
||||||
|
|
@ -253,12 +262,11 @@ class RectangleAperture(Aperture):
|
||||||
hole_dia=None if self.hole_dia is None else self.hole_dia*scale)
|
hole_dia=None if self.hole_dia is None else self.hole_dia*scale)
|
||||||
|
|
||||||
def to_macro(self, rotation=0):
|
def to_macro(self, rotation=0):
|
||||||
return ApertureMacroInstance(GenericMacros.rect,
|
from .aperture_macros.parse import GenericMacros
|
||||||
(MM(self.w, self.unit),
|
return GenericMacros.rect(MM(self.w, self.unit),
|
||||||
MM(self.h, self.unit),
|
MM(self.h, self.unit),
|
||||||
MM(self.hole_dia, self.unit) or 0,
|
MM(self.hole_dia, self.unit),
|
||||||
0,
|
rotation)
|
||||||
rotation))
|
|
||||||
|
|
||||||
def _params(self, unit=None):
|
def _params(self, unit=None):
|
||||||
return _strip_right(
|
return _strip_right(
|
||||||
|
|
@ -295,6 +303,8 @@ class ObroundAperture(Aperture):
|
||||||
|
|
||||||
def dilated(self, offset, unit=MM):
|
def dilated(self, offset, unit=MM):
|
||||||
offset = self.unit(offset, unit)
|
offset = self.unit(offset, unit)
|
||||||
|
if math.isclose(offset, 0, abs_tol=1e-6):
|
||||||
|
return self
|
||||||
return replace(self, w=self.w+2*offset, h=self.h+2*offset, hole_dia=None)
|
return replace(self, w=self.w+2*offset, h=self.h+2*offset, hole_dia=None)
|
||||||
|
|
||||||
@lru_cache()
|
@lru_cache()
|
||||||
|
|
@ -320,12 +330,11 @@ class ObroundAperture(Aperture):
|
||||||
rotation -= -math.pi/2
|
rotation -= -math.pi/2
|
||||||
inst = replace(self, w=self.h, h=self.w, hole_dia=self.hole_dia)
|
inst = replace(self, w=self.h, h=self.w, hole_dia=self.hole_dia)
|
||||||
|
|
||||||
return ApertureMacroInstance(GenericMacros.obround,
|
from .aperture_macros.parse import GenericMacros
|
||||||
(MM(inst.w, self.unit),
|
return GenericMacros.obround(MM(inst.w, self.unit),
|
||||||
MM(inst.h, self.unit),
|
MM(inst.h, self.unit),
|
||||||
MM(inst.hole_dia, self.unit) or 0,
|
MM(inst.hole_dia, self.unit) or 0,
|
||||||
0,
|
rotation)
|
||||||
rotation))
|
|
||||||
|
|
||||||
def _params(self, unit=None):
|
def _params(self, unit=None):
|
||||||
return _strip_right(
|
return _strip_right(
|
||||||
|
|
@ -362,6 +371,8 @@ class PolygonAperture(Aperture):
|
||||||
|
|
||||||
def dilated(self, offset, unit=MM):
|
def dilated(self, offset, unit=MM):
|
||||||
offset = self.unit(offset, unit)
|
offset = self.unit(offset, unit)
|
||||||
|
if math.isclose(offset, 0, abs_tol=1e-6):
|
||||||
|
return self
|
||||||
return replace(self, diameter=self.diameter+2*offset, hole_dia=None)
|
return replace(self, diameter=self.diameter+2*offset, hole_dia=None)
|
||||||
|
|
||||||
flash = _flash_hole
|
flash = _flash_hole
|
||||||
|
|
@ -379,16 +390,22 @@ class PolygonAperture(Aperture):
|
||||||
hole_dia=None if self.hole_dia is None else self.hole_dia*scale)
|
hole_dia=None if self.hole_dia is None else self.hole_dia*scale)
|
||||||
|
|
||||||
def to_macro(self):
|
def to_macro(self):
|
||||||
return ApertureMacroInstance(GenericMacros.polygon, self._params(MM))
|
from .aperture_macros.parse import GenericMacros
|
||||||
|
return GenericMacros.polygon(self.n_vertices,
|
||||||
|
MM(self.diameter, self.unit),
|
||||||
|
MM(self.hole_dia, self.unit),
|
||||||
|
self.rotation)
|
||||||
|
|
||||||
def _params(self, unit=None):
|
def _params(self, unit=None):
|
||||||
rotation = self.rotation % (2*math.pi / self.n_vertices)
|
rotation = self.rotation % (2*math.pi / self.n_vertices)
|
||||||
if math.isclose(rotation, 0, abs_tol=1e-6):
|
if math.isclose(rotation, 0, abs_tol=1e-6):
|
||||||
rotation = None
|
rotation = None
|
||||||
|
else:
|
||||||
|
rotation = math.degrees(rotation)
|
||||||
|
|
||||||
if self.hole_dia is not None:
|
if self.hole_dia is not None:
|
||||||
return self.unit.convert_to(unit, self.diameter), self.n_vertices, rotation, self.unit.convert_to(unit, self.hole_dia)
|
return self.unit.convert_to(unit, self.diameter), self.n_vertices, rotation, self.unit.convert_to(unit, self.hole_dia)
|
||||||
elif rotation is not None and not math.isclose(rotation, 0):
|
elif rotation is not None and not math.isclose(rotation, 0, abs_tol=1e-6):
|
||||||
return self.unit.convert_to(unit, self.diameter), self.n_vertices, rotation
|
return self.unit.convert_to(unit, self.diameter), self.n_vertices, rotation
|
||||||
else:
|
else:
|
||||||
return self.unit.convert_to(unit, self.diameter), self.n_vertices
|
return self.unit.convert_to(unit, self.diameter), self.n_vertices
|
||||||
|
|
@ -418,11 +435,13 @@ class ApertureMacroInstance(Aperture):
|
||||||
return out
|
return out
|
||||||
|
|
||||||
def dilated(self, offset, unit=MM):
|
def dilated(self, offset, unit=MM):
|
||||||
|
if math.isclose(offset, 0, abs_tol=1e-6):
|
||||||
|
return self
|
||||||
return replace(self, macro=self.macro.dilated(offset, unit))
|
return replace(self, macro=self.macro.dilated(offset, unit))
|
||||||
|
|
||||||
@lru_cache()
|
@lru_cache()
|
||||||
def rotated(self, angle=0.0):
|
def rotated(self, angle=0.0):
|
||||||
if math.isclose(angle % (2*math.pi), 0):
|
if math.isclose(angle % (2*math.pi), 0, abs_tol=1e-6):
|
||||||
return self
|
return self
|
||||||
else:
|
else:
|
||||||
return self.to_macro(angle)
|
return self.to_macro(angle)
|
||||||
|
|
@ -433,9 +452,17 @@ class ApertureMacroInstance(Aperture):
|
||||||
def scaled(self, scale):
|
def scaled(self, scale):
|
||||||
return replace(self, macro=self.macro.scaled(scale))
|
return replace(self, macro=self.macro.scaled(scale))
|
||||||
|
|
||||||
|
def calculate_out(self, unit=None, macro_name=None):
|
||||||
|
return replace(self,
|
||||||
|
parameters=tuple(),
|
||||||
|
macro=self.macro.substitute_params(self._params(unit), unit, macro_name))
|
||||||
|
|
||||||
def _params(self, unit=None):
|
def _params(self, unit=None):
|
||||||
# We ignore "unit" here as we convert the actual macro, not this instantiation.
|
# We ignore "unit" here as we convert the actual macro, not this instantiation.
|
||||||
# We do this because here we do not have information about which parameter has which physical units.
|
# We do this because here we do not have information about which parameter has which physical units.
|
||||||
return tuple(self.parameters)
|
parameters = self.parameters
|
||||||
|
if len(parameters) > self.macro.num_parameters:
|
||||||
|
warnings.warn(f'Aperture definition using macro {self.macro.name} has more parameters than the macro uses.')
|
||||||
|
parameters = parameters[:self.macro.num_parameters]
|
||||||
|
return tuple(parameters)
|
||||||
|
|
||||||
138
src/gerbonara/cad/breakout.py
Normal file
|
|
@ -0,0 +1,138 @@
|
||||||
|
|
||||||
|
from dataclasses import dataclass
|
||||||
|
|
||||||
|
from ..utils import MM
|
||||||
|
from .primitives import *
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class PadRing(Positioned):
|
||||||
|
w: int
|
||||||
|
h: int
|
||||||
|
pitch: float = 2.54
|
||||||
|
clearance: float = 0.2
|
||||||
|
rows: int = 2
|
||||||
|
trace_width: float = 0.4
|
||||||
|
drill_dia: float = 0.9
|
||||||
|
stagger: bool = False
|
||||||
|
|
||||||
|
def ports(self):
|
||||||
|
x, y, rotation = self.abs_pos
|
||||||
|
|
||||||
|
x += self.pitch/2
|
||||||
|
y += self.pitch/2
|
||||||
|
|
||||||
|
x += self.pitch * self.rows
|
||||||
|
y += self.pitch * self.rows
|
||||||
|
|
||||||
|
pad_dia = self.pitch - 2*self.clearance - self.trace_width
|
||||||
|
offset = pad_dia/2 - self.trace_width/2
|
||||||
|
|
||||||
|
for i in range(1, self.w):
|
||||||
|
yield (x+self.pitch/2 + i*self.pitch, y+offset)
|
||||||
|
yield (x+self.pitch/2 + i*self.pitch, y+(self.h+1)*self.pitch-offset)
|
||||||
|
|
||||||
|
for i in range(0, self.w):
|
||||||
|
yield (x + (i+1)*self.pitch, y+offset)
|
||||||
|
yield (x + (i+1)*self.pitch, y+(self.h+1)*self.pitch-offset)
|
||||||
|
|
||||||
|
for i in range(1, self.h):
|
||||||
|
yield (x+offset, y+self.pitch/2 + i*self.pitch)
|
||||||
|
yield (x+(self.w+1)*self.pitch-offset, y+self.pitch/2 + i*self.pitch)
|
||||||
|
|
||||||
|
for i in range(0, self.h):
|
||||||
|
yield (x+offset, y + (i+1)*self.pitch)
|
||||||
|
yield (x+(self.w+1)*self.pitch-offset, y + (i+1)*self.pitch)
|
||||||
|
|
||||||
|
|
||||||
|
def generate(self, bbox, border_text, unit=MM):
|
||||||
|
x, y, rotation = self.abs_pos
|
||||||
|
|
||||||
|
x += self.pitch/2
|
||||||
|
y += self.pitch/2
|
||||||
|
|
||||||
|
x += self.pitch * self.rows
|
||||||
|
y += self.pitch * self.rows
|
||||||
|
|
||||||
|
pad_dia = self.pitch - 2*self.clearance - self.trace_width
|
||||||
|
|
||||||
|
for i in range(self.w + 2 + 2*(self.rows-1)):
|
||||||
|
for j in range(self.rows):
|
||||||
|
yield THTPad.circle(x + (i - (self.rows - 1))*self.pitch, y - j*self.pitch, self.drill_dia, pad_dia, paste=False)
|
||||||
|
yield THTPad.circle(x + (i - (self.rows - 1))*self.pitch, y + (self.h + 1 + j)*self.pitch, self.drill_dia, pad_dia, paste=False)
|
||||||
|
|
||||||
|
if self.rows >= 2 and 1 <= i < self.w:
|
||||||
|
yield Trace(self.trace_width, start=(x+i*self.pitch, y-self.pitch), end=(x+(i + 0.5)*self.pitch, y+pad_dia/2 - self.trace_width/2))
|
||||||
|
yield Trace(self.trace_width, start=(x+i*self.pitch, y+(self.h+2)*self.pitch), end=(x+(i + 0.5)*self.pitch, y+(self.h+1)*self.pitch -pad_dia/2 + self.trace_width/2), orientation=('cw',))
|
||||||
|
|
||||||
|
for i in range(1, self.h+1):
|
||||||
|
for j in range(self.rows):
|
||||||
|
yield THTPad.circle(x - j*self.pitch, y + i*self.pitch, self.drill_dia, pad_dia, paste=False)
|
||||||
|
yield THTPad.circle(x + (self.w + 1 + j)*self.pitch, y + i*self.pitch, self.drill_dia, pad_dia, paste=False)
|
||||||
|
|
||||||
|
for i in range(1, self.h):
|
||||||
|
yield (x+offset, y+self.pitch/2 + i*self.pitch)
|
||||||
|
yield (x+(self.w+1)*self.pitch-offset, y+self.pitch/2 + i*self.pitch)
|
||||||
|
|
||||||
|
|
||||||
|
def generate(self, bbox, border_text, unit=MM):
|
||||||
|
x, y, rotation = self.abs_pos
|
||||||
|
|
||||||
|
x += self.pitch/2
|
||||||
|
y += self.pitch/2
|
||||||
|
|
||||||
|
x += self.pitch * self.rows
|
||||||
|
y += self.pitch * self.rows
|
||||||
|
|
||||||
|
pad_dia = self.pitch - 2*self.clearance - self.trace_width
|
||||||
|
|
||||||
|
for i in range(self.w + 2 + 2*(self.rows-1)):
|
||||||
|
for j in range(self.rows):
|
||||||
|
yield THTPad.circle(x + (i - (self.rows - 1))*self.pitch, y - j*self.pitch, self.drill_dia, pad_dia, paste=False)
|
||||||
|
yield THTPad.circle(x + (i - (self.rows - 1))*self.pitch, y + (self.h + 1 + j)*self.pitch, self.drill_dia, pad_dia, paste=False)
|
||||||
|
|
||||||
|
if self.rows >= 2 and 1 <= i < self.w:
|
||||||
|
yield Trace(self.trace_width, start=(x+i*self.pitch, y-self.pitch), end=(x+(i + 0.5)*self.pitch, y+pad_dia/2 - self.trace_width/2))
|
||||||
|
yield Trace(self.trace_width, start=(x+i*self.pitch, y+(self.h+2)*self.pitch), end=(x+(i + 0.5)*self.pitch, y+(self.h+1)*self.pitch -pad_dia/2 + self.trace_width/2), orientation=('cw',))
|
||||||
|
|
||||||
|
for i in range(1, self.h+1):
|
||||||
|
for j in range(self.rows):
|
||||||
|
yield THTPad.circle(x - j*self.pitch, y + i*self.pitch, self.drill_dia, pad_dia, paste=False)
|
||||||
|
yield THTPad.circle(x + (self.w + 1 + j)*self.pitch, y + i*self.pitch, self.drill_dia, pad_dia, paste=False)
|
||||||
|
|
||||||
|
if self.rows >= 2 and i < self.h:
|
||||||
|
yield Trace(self.trace_width,
|
||||||
|
start=(
|
||||||
|
x-self.pitch,
|
||||||
|
y+i*self.pitch),
|
||||||
|
end=(
|
||||||
|
x+pad_dia/2 - self.trace_width/2,
|
||||||
|
y+(i + 0.5)*self.pitch),
|
||||||
|
orientation=('cw',))
|
||||||
|
yield Trace(self.trace_width,
|
||||||
|
start=(
|
||||||
|
x+(self.w+2)*self.pitch,
|
||||||
|
y+i*self.pitch),
|
||||||
|
end=(
|
||||||
|
x+(self.w+1)*self.pitch -pad_dia/2 + self.trace_width/2,
|
||||||
|
y+(i + 0.5)*self.pitch))
|
||||||
|
|
||||||
|
|
||||||
|
def _breakout_demo():
|
||||||
|
b = Board(100, 80)
|
||||||
|
|
||||||
|
ring = PadRing(5, 5, 8, 12)
|
||||||
|
for obj in ring.generate(None, None):
|
||||||
|
b.add(obj)
|
||||||
|
|
||||||
|
for x, y in ring.ports():
|
||||||
|
b.add(Trace(0.1, start=(23, 27), end=(x, y)))
|
||||||
|
|
||||||
|
with open('/tmp/test.svg', 'w') as f:
|
||||||
|
f.write(str(b.pretty_svg()))
|
||||||
|
b.layer_stack().save_to_directory('/tmp/testdir')
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
_breakout_demo()
|
||||||
|
|
||||||
1
src/gerbonara/cad/kicad/__init__.py
Normal file
|
|
@ -0,0 +1 @@
|
||||||
|
|
||||||
630
src/gerbonara/cad/kicad/base_types.py
Normal file
|
|
@ -0,0 +1,630 @@
|
||||||
|
import string
|
||||||
|
import time
|
||||||
|
from dataclasses import field, replace
|
||||||
|
import math
|
||||||
|
import uuid
|
||||||
|
from contextlib import contextmanager
|
||||||
|
from itertools import cycle
|
||||||
|
|
||||||
|
from .sexp import *
|
||||||
|
from .sexp_mapper import *
|
||||||
|
from ...newstroke import Newstroke
|
||||||
|
from ...utils import rotate_point, sum_bounds, Tag, MM
|
||||||
|
from ...layers import LayerStack
|
||||||
|
from ... import apertures as ap
|
||||||
|
from ... import graphic_objects as go
|
||||||
|
|
||||||
|
|
||||||
|
LAYER_MAP_K2G = {
|
||||||
|
'F.Cu': ('top', 'copper'),
|
||||||
|
'B.Cu': ('bottom', 'copper'),
|
||||||
|
'F.SilkS': ('top', 'silk'),
|
||||||
|
'B.SilkS': ('bottom', 'silk'),
|
||||||
|
'F.Paste': ('top', 'paste'),
|
||||||
|
'B.Paste': ('bottom', 'paste'),
|
||||||
|
'F.Mask': ('top', 'mask'),
|
||||||
|
'B.Mask': ('bottom', 'mask'),
|
||||||
|
'B.CrtYd': ('bottom', 'courtyard'),
|
||||||
|
'F.CrtYd': ('top', 'courtyard'),
|
||||||
|
'B.Fab': ('bottom', 'fabrication'),
|
||||||
|
'F.Fab': ('top', 'fabrication'),
|
||||||
|
'B.Adhes': ('bottom', 'adhesive'),
|
||||||
|
'F.Adhes': ('top', 'adhesive'),
|
||||||
|
'Dwgs.User': ('mechanical', 'drawings'),
|
||||||
|
'Cmts.User': ('mechanical', 'comments'),
|
||||||
|
'Edge.Cuts': ('mechanical', 'outline'),
|
||||||
|
}
|
||||||
|
|
||||||
|
LAYER_MAP_G2K = {v: k for k, v in LAYER_MAP_K2G.items()}
|
||||||
|
|
||||||
|
|
||||||
|
class BBoxMixin:
|
||||||
|
def bounding_box(self, unit=MM):
|
||||||
|
if not hasattr(self, '_bounding_box'):
|
||||||
|
(min_x, min_y), (max_x, max_y) = sum_bounds(fe.bounding_box(unit) for fe in self.render())
|
||||||
|
# Convert back from gerbonara's coordinates to kicad coordinates.
|
||||||
|
self._bounding_box = (min_x, -max_y), (max_x, -min_y)
|
||||||
|
|
||||||
|
return self._bounding_box
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('uuid')
|
||||||
|
class UUID:
|
||||||
|
value: str = field(default_factory=uuid.uuid4)
|
||||||
|
|
||||||
|
def __deepcopy__(self, memo):
|
||||||
|
return UUID()
|
||||||
|
|
||||||
|
def __after_parse__(self, parent):
|
||||||
|
self.value = str(self.value)
|
||||||
|
|
||||||
|
def before_sexp(self):
|
||||||
|
self.value = str(self.value)
|
||||||
|
|
||||||
|
def bump(self):
|
||||||
|
self.value = uuid.uuid4()
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('group')
|
||||||
|
class Group:
|
||||||
|
locked: Flag() = False
|
||||||
|
name: str = ""
|
||||||
|
id: Named(str) = None
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
members: Named(Array(str)) = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('color')
|
||||||
|
class Color:
|
||||||
|
r: int = None
|
||||||
|
g: int = None
|
||||||
|
b: int = None
|
||||||
|
a: float = None
|
||||||
|
|
||||||
|
def __bool__(self):
|
||||||
|
return self.r or self.b or self.g or not math.isclose(self.a, 0, abs_tol=1e-3)
|
||||||
|
|
||||||
|
def svg(self, default=None):
|
||||||
|
if default and not self:
|
||||||
|
return default
|
||||||
|
|
||||||
|
return f'rgba({self.r} {self.g} {self.b} {self.a})'
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('stroke')
|
||||||
|
class Stroke:
|
||||||
|
width: Named(float) = 0.254
|
||||||
|
type: Named(AtomChoice(Atom.dash, Atom.dot, Atom.dash_dot_dot, Atom.dash_dot, Atom.default, Atom.solid)) = Atom.default
|
||||||
|
color: Color = None
|
||||||
|
|
||||||
|
def svg_color(self, default=None):
|
||||||
|
if self.color:
|
||||||
|
return self.color.svg(default)
|
||||||
|
else:
|
||||||
|
return default
|
||||||
|
|
||||||
|
def svg_attrs(self, default_color=None):
|
||||||
|
w = self.width
|
||||||
|
if not (color := self.color or default_color):
|
||||||
|
return {}
|
||||||
|
|
||||||
|
attrs = {'stroke': color,
|
||||||
|
'stroke_linecap': 'round',
|
||||||
|
'stroke_linejoin': 'round',
|
||||||
|
'stroke_width': self.width or 0.254}
|
||||||
|
|
||||||
|
if self.type not in (Atom.default, Atom.solid):
|
||||||
|
attrs['stroke_dasharray'] = {
|
||||||
|
Atom.dash: f'{w*5:.3f},{w*5:.3f}',
|
||||||
|
Atom.dot: f'{w*2:.3f},{w*2:.3f}',
|
||||||
|
Atom.dash_dot: f'{w*5:.3f},{w*3:.3f}{w:.3f},{w*3:.3f}',
|
||||||
|
Atom.dash_dot_dot: f'{w*5:.3f},{w*3:.3f}{w:.3f},{w*3:.3f}{w:.3f},{w*3:.3f}',
|
||||||
|
}[self.type]
|
||||||
|
|
||||||
|
return attrs
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('fill')
|
||||||
|
class Fill:
|
||||||
|
type: Named(AtomChoice(Atom.none, Atom.outline, Atom.background, Atom.color)) = Atom.none
|
||||||
|
color: Color = None
|
||||||
|
|
||||||
|
|
||||||
|
class WidthMixin:
|
||||||
|
def __post_init__(self):
|
||||||
|
if self.width is not None:
|
||||||
|
self.stroke = Stroke(self.width)
|
||||||
|
|
||||||
|
|
||||||
|
class Dasher:
|
||||||
|
def __init__(self, obj):
|
||||||
|
if obj.stroke:
|
||||||
|
w = obj.stroke.width if obj.stroke.width not in (None, 0, 0.0) else 0.254
|
||||||
|
t = obj.stroke.type
|
||||||
|
else:
|
||||||
|
w = obj.width or 0
|
||||||
|
t = Atom.solid
|
||||||
|
|
||||||
|
self.width = w
|
||||||
|
gap = 4*w
|
||||||
|
dot = 0
|
||||||
|
dash = 11*w
|
||||||
|
self.pattern = {
|
||||||
|
Atom.dash: [dash, gap],
|
||||||
|
Atom.dot: [dot, gap],
|
||||||
|
Atom.dash_dot_dot: [dash, gap, dot, gap, dot, gap],
|
||||||
|
Atom.dash_dot: [dash, gap, dot, gap],
|
||||||
|
Atom.default: [1e99],
|
||||||
|
Atom.solid: [1e99]}[t]
|
||||||
|
self.solid = t in (Atom.default, Atom.solid)
|
||||||
|
self.start_x, self.start_y = None, None
|
||||||
|
self.cur_x, self.cur_y = None, None
|
||||||
|
self.segments = []
|
||||||
|
|
||||||
|
def move(self, x, y):
|
||||||
|
if self.cur_x is None:
|
||||||
|
self.start_x, self.start_y = x, y
|
||||||
|
self.cur_x, self.cur_y = x, y
|
||||||
|
|
||||||
|
def line(self, x, y):
|
||||||
|
if x is None or y is None:
|
||||||
|
raise ValueError('line() called before move()')
|
||||||
|
self.segments.append((self.cur_x, self.cur_y, x, y))
|
||||||
|
self.cur_x, self.cur_y = x, y
|
||||||
|
|
||||||
|
def close(self):
|
||||||
|
self.segments.append((self.cur_x, self.cur_y, self.start_x, self.start_y))
|
||||||
|
self.cur_x, self.cur_y = None, None
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _interpolate(x1, y1, x2, y2, length):
|
||||||
|
dx, dy = x2-x1, y2-y1
|
||||||
|
total = math.hypot(dx, dy)
|
||||||
|
if total == 0:
|
||||||
|
return x2, y2
|
||||||
|
frac = length / total
|
||||||
|
return x1 + dx*frac, y1 + dy*frac
|
||||||
|
|
||||||
|
def __iter__(self):
|
||||||
|
it = iter(self.segments)
|
||||||
|
segment_remaining, segment_pos = 0, 0
|
||||||
|
|
||||||
|
if self.width is None or self.width < 1e-3:
|
||||||
|
return
|
||||||
|
|
||||||
|
for length, stroked in cycle(zip(self.pattern, cycle([True, False]))):
|
||||||
|
length = max(1e-12, length)
|
||||||
|
while length > 0:
|
||||||
|
if segment_remaining == 0:
|
||||||
|
try:
|
||||||
|
x1, y1, x2, y2 = next(it)
|
||||||
|
except StopIteration:
|
||||||
|
return
|
||||||
|
dx, dy = x2-x1, y2-y1
|
||||||
|
lx, ly = x1, y1
|
||||||
|
segment_remaining = math.hypot(dx, dy)
|
||||||
|
segment_pos = 0
|
||||||
|
|
||||||
|
if segment_remaining > length:
|
||||||
|
segment_pos += length
|
||||||
|
ix, iy = self._interpolate(x1, y1, x2, y2, segment_pos)
|
||||||
|
segment_remaining -= length
|
||||||
|
if stroked:
|
||||||
|
yield lx, ly, ix, iy
|
||||||
|
lx, ly = ix, iy
|
||||||
|
break
|
||||||
|
|
||||||
|
else:
|
||||||
|
length -= segment_remaining
|
||||||
|
segment_remaining = 0
|
||||||
|
if stroked:
|
||||||
|
yield lx, ly, x2, y2
|
||||||
|
|
||||||
|
def svg(self, **kwargs):
|
||||||
|
if 'fill' not in kwargs:
|
||||||
|
kwargs['fill'] = 'none'
|
||||||
|
if 'stroke' not in kwargs:
|
||||||
|
kwargs['stroke'] = 'black'
|
||||||
|
if 'stroke_width' not in kwargs:
|
||||||
|
kwargs['stroke_width'] = 0.254
|
||||||
|
if 'stroke_linecap' not in kwargs:
|
||||||
|
kwargs['stroke_linecap'] = 'round'
|
||||||
|
|
||||||
|
d = ' '.join(f'M {x1:.3f} {y1:.3f} L {x2:.3f} {y2:.3f}' for x1, y1, x2, y2 in self)
|
||||||
|
return Tag('path', d=d, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('xy')
|
||||||
|
class XYCoord:
|
||||||
|
x: float = 0
|
||||||
|
y: float = 0
|
||||||
|
|
||||||
|
def __init__(self, x=None, y=None):
|
||||||
|
if x is None:
|
||||||
|
self.x, self.y = None, None
|
||||||
|
elif isinstance(x, XYCoord):
|
||||||
|
self.x, self.y = x.x, x.y
|
||||||
|
elif isinstance(x, (tuple, list)):
|
||||||
|
self.x, self.y = x
|
||||||
|
elif hasattr(x, 'abs_pos'):
|
||||||
|
self.x, self.y, _1, _2 = x.abs_pos
|
||||||
|
elif hasattr(x, 'at'):
|
||||||
|
self.x, self.y = x.at.x, x.at.y
|
||||||
|
else:
|
||||||
|
self.x, self.y = x, y
|
||||||
|
|
||||||
|
def __iter__(self):
|
||||||
|
return iter((self.x, self.y))
|
||||||
|
|
||||||
|
def __getitem__(self, index):
|
||||||
|
return (self.x, self.y)[index]
|
||||||
|
|
||||||
|
def __setitem__(self, index, value):
|
||||||
|
if index == 0:
|
||||||
|
self.x = value
|
||||||
|
elif index == 1:
|
||||||
|
self.y = value
|
||||||
|
else:
|
||||||
|
raise IndexError(f'Invalid 2D point coordinate index {index}')
|
||||||
|
|
||||||
|
def within_distance(self, x, y, dist):
|
||||||
|
return math.dist((x, y), (self.x, self.y)) < dist
|
||||||
|
|
||||||
|
def isclose(self, other, tol=1e-3):
|
||||||
|
return math.isclose(self.x, other.x, tol) and math.isclose(self.y, other.y, tol)
|
||||||
|
|
||||||
|
def with_offset(self, x=0, y=0):
|
||||||
|
return replace(self, x=self.x+x, y=self.y+y)
|
||||||
|
|
||||||
|
def with_rotation(self, angle, cx=0, cy=0):
|
||||||
|
x, y = rotate_point(self.x, self.y, angle, cx, cy)
|
||||||
|
return replace(self, x=x, y=y)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('pts')
|
||||||
|
class PointList:
|
||||||
|
@classmethod
|
||||||
|
def __map__(kls, obj, parent=None, path=''):
|
||||||
|
_tag, *values = obj
|
||||||
|
return [map_sexp(XYCoord, elem, parent=parent, path=path) for elem in values]
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def __sexp__(kls, value):
|
||||||
|
yield [kls.name_atom, *(e for elem in value for e in elem.__sexp__(elem))]
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('arc')
|
||||||
|
class Arc:
|
||||||
|
start: Rename(XYCoord) = None
|
||||||
|
mid: Rename(XYCoord) = None
|
||||||
|
end: Rename(XYCoord) = None
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('pts')
|
||||||
|
class ArcPointList:
|
||||||
|
@classmethod
|
||||||
|
def __map__(kls, obj, parent=None, path=''):
|
||||||
|
_tag, *values = obj
|
||||||
|
return [map_sexp((XYCoord if elem[0] == 'xy' else Arc), elem, parent=parent, path=path) for elem in values]
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def __sexp__(kls, value):
|
||||||
|
yield [kls.name_atom, *(e for elem in value for e in elem.__sexp__(elem))]
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('net')
|
||||||
|
class Net:
|
||||||
|
index: int = 0
|
||||||
|
name: str = ''
|
||||||
|
|
||||||
|
|
||||||
|
class NetMixin:
|
||||||
|
def reset_net(self):
|
||||||
|
self.net = Net()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def net_index(self):
|
||||||
|
if self.net is None:
|
||||||
|
return 0
|
||||||
|
return self.net.index
|
||||||
|
|
||||||
|
@property
|
||||||
|
def net_name(self):
|
||||||
|
if self.net is None:
|
||||||
|
return ''
|
||||||
|
return self.net.name
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('xyz')
|
||||||
|
class XYZCoord:
|
||||||
|
x: float = 0
|
||||||
|
y: float = 0
|
||||||
|
z: float = 0
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('at')
|
||||||
|
class AtPos(XYCoord):
|
||||||
|
x: float = 0 # in millimeter
|
||||||
|
y: float = 0 # in millimeter
|
||||||
|
rotation: int = 0 # in degrees, can only be 0, 90, 180 or 270.
|
||||||
|
unlocked: Flag() = True
|
||||||
|
|
||||||
|
def __before_sexp__(self):
|
||||||
|
self.rotation = int(round(self.rotation % 360))
|
||||||
|
|
||||||
|
@property
|
||||||
|
def rotation_rad(self):
|
||||||
|
return math.radians(self.rotation)
|
||||||
|
|
||||||
|
@rotation_rad.setter
|
||||||
|
def rotation_rad(self, value):
|
||||||
|
self.rotation = math.degrees(value)
|
||||||
|
|
||||||
|
def with_rotation(self, angle, cx=0, cy=0):
|
||||||
|
obj = super().with_rotation(angle, cx, cy)
|
||||||
|
return replace(obj, rotation=self.rotation + angle)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('font')
|
||||||
|
class FontSpec:
|
||||||
|
face: Named(str) = None
|
||||||
|
size: Rename(XYCoord) = field(default_factory=lambda: XYCoord(1.27, 1.27))
|
||||||
|
thickness: Named(float) = None
|
||||||
|
bold: OmitDefault(Named(LegacyCompatibleFlag())) = False
|
||||||
|
italic: OmitDefault(Named(LegacyCompatibleFlag())) = False
|
||||||
|
line_spacing: Named(float) = None
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('justify')
|
||||||
|
class Justify:
|
||||||
|
h: AtomChoice(Atom.left, Atom.right) = None
|
||||||
|
v: AtomChoice(Atom.top, Atom.bottom) = None
|
||||||
|
mirror: Flag() = False
|
||||||
|
|
||||||
|
@property
|
||||||
|
def h_str(self):
|
||||||
|
if self.h is None:
|
||||||
|
return 'center'
|
||||||
|
else:
|
||||||
|
return str(self.h)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def v_str(self):
|
||||||
|
if self.v is None:
|
||||||
|
return 'middle'
|
||||||
|
else:
|
||||||
|
return str(self.v)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('effects')
|
||||||
|
class TextEffect:
|
||||||
|
font: FontSpec = field(default_factory=FontSpec)
|
||||||
|
justify: OmitDefault(Justify) = field(default_factory=Justify)
|
||||||
|
hide: OmitDefault(Named(LegacyCompatibleFlag())) = False
|
||||||
|
|
||||||
|
|
||||||
|
class TextMixin:
|
||||||
|
@property
|
||||||
|
def size(self):
|
||||||
|
return self.effects.font.size.y or 1.27
|
||||||
|
|
||||||
|
@size.setter
|
||||||
|
def size(self, value):
|
||||||
|
self.effects.font.size.x = self.effects.font.size.y = value
|
||||||
|
|
||||||
|
@property
|
||||||
|
def line_width(self):
|
||||||
|
return self.effects.font.thickness or 0.254
|
||||||
|
|
||||||
|
@line_width.setter
|
||||||
|
def line_width(self, value):
|
||||||
|
self.effects.font.thickness = value
|
||||||
|
|
||||||
|
def bounding_box(self, default=None):
|
||||||
|
if not self.text or not self.text.strip():
|
||||||
|
return default
|
||||||
|
|
||||||
|
lines = list(self.render())
|
||||||
|
x1 = min(min(l.x1, l.x2) for l in lines)
|
||||||
|
y1 = min(min(l.y1, l.y2) for l in lines)
|
||||||
|
x2 = max(max(l.x1, l.x2) for l in lines)
|
||||||
|
y2 = max(max(l.y1, l.y2) for l in lines)
|
||||||
|
r = self.effects.font.thickness/2
|
||||||
|
return (x1-r, -(y1-r)), (x2+r, -(y2+r))
|
||||||
|
|
||||||
|
def svg_path_data(self):
|
||||||
|
for line in self.render():
|
||||||
|
yield f'M {line.x1:.3f} {line.y1:.3f} L {line.x2:.3f} {line.y2:.3f}'
|
||||||
|
|
||||||
|
@property
|
||||||
|
def default_v_align(self):
|
||||||
|
return 'bottom'
|
||||||
|
|
||||||
|
@property
|
||||||
|
def h_align(self):
|
||||||
|
return 'left' if self.effects.justify.h else 'center'
|
||||||
|
|
||||||
|
@property
|
||||||
|
def mirrored(self):
|
||||||
|
return False, False
|
||||||
|
|
||||||
|
def to_svg(self, color='black', variables={}):
|
||||||
|
if not self.effects or self.effects.hide or not self.effects.font:
|
||||||
|
return
|
||||||
|
|
||||||
|
font = Newstroke.load()
|
||||||
|
text = string.Template(self.text).safe_substitute(variables)
|
||||||
|
aperture = ap.CircleAperture(self.line_width or 0.2, unit=MM)
|
||||||
|
rot = self.rotation
|
||||||
|
h_align = self.h_align
|
||||||
|
mx, my = self.mirrored
|
||||||
|
if rot in (90, 270):
|
||||||
|
h_align = {'left': 'right', 'right': 'left'}.get(h_align, h_align)
|
||||||
|
rot = (rot+180)%360
|
||||||
|
elif rot == 180:
|
||||||
|
rot = 0
|
||||||
|
h_align = {'left': 'right', 'right': 'left'}.get(h_align, h_align)
|
||||||
|
|
||||||
|
if my and rot in (0, 180):
|
||||||
|
h_align = {'left': 'right', 'right': 'left'}.get(h_align, h_align)
|
||||||
|
rot = (rot+180)%360
|
||||||
|
if mx and rot in (90, 270):
|
||||||
|
h_align = {'left': 'right', 'right': 'left'}.get(h_align, h_align)
|
||||||
|
rot = (rot+180)%360
|
||||||
|
if rot == 180:
|
||||||
|
rot = 0
|
||||||
|
h_align = {'left': 'right', 'right': 'left'}.get(h_align, h_align)
|
||||||
|
if rot == 90:
|
||||||
|
rot = 270
|
||||||
|
h_align = {'left': 'right', 'right': 'left'}.get(h_align, h_align)
|
||||||
|
|
||||||
|
yield font.render_svg(text,
|
||||||
|
size=self.size or 1.27,
|
||||||
|
h_align=h_align,
|
||||||
|
v_align=self.effects.justify.v or self.default_v_align,
|
||||||
|
stroke=color,
|
||||||
|
stroke_width=f'{self.line_width:.3f}',
|
||||||
|
scale=(1,1),
|
||||||
|
rotation=0,
|
||||||
|
transform=f'translate({self.at.x:.3f} {self.at.y:.3f}) rotate({rot})',
|
||||||
|
)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def _text_offset(self):
|
||||||
|
return (0, 0)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def rotation(self):
|
||||||
|
return self.at.rotation
|
||||||
|
|
||||||
|
def render(self, variables={}):
|
||||||
|
if not self.effects or self.effects.hide or not self.effects.font:
|
||||||
|
return
|
||||||
|
|
||||||
|
font = Newstroke.load()
|
||||||
|
text = string.Template(self.text).safe_substitute(variables)
|
||||||
|
aperture = ap.CircleAperture(self.line_width or 0.2, unit=MM)
|
||||||
|
for stroke in font.render(text,
|
||||||
|
x0=self.at.x, y=self.at.y,
|
||||||
|
size=self.size or 1.27,
|
||||||
|
h_align=self.effects.justify.h_str,
|
||||||
|
v_align=self.effects.justify.v_str,
|
||||||
|
rotation=self.at.rotation,
|
||||||
|
):
|
||||||
|
|
||||||
|
points = []
|
||||||
|
for x, y in stroke:
|
||||||
|
x, y = x+offx, y+offy
|
||||||
|
x, y = rotate_point(x, y, math.radians(-rot or 0))
|
||||||
|
x, y = x+self.at.x, y+self.at.y
|
||||||
|
points.append((x, -y))
|
||||||
|
|
||||||
|
for p1, p2 in zip(points[:-1], points[1:]):
|
||||||
|
yield go.Line(*p1, *p2, aperture=aperture, unit=MM)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('tstamp')
|
||||||
|
class Timestamp:
|
||||||
|
value: str = field(default_factory=uuid.uuid4)
|
||||||
|
|
||||||
|
def __deepcopy__(self, memo):
|
||||||
|
return Timestamp()
|
||||||
|
|
||||||
|
def __after_parse__(self, parent):
|
||||||
|
self.value = str(self.value)
|
||||||
|
|
||||||
|
def before_sexp(self):
|
||||||
|
self.value = Atom(str(self.value))
|
||||||
|
|
||||||
|
def bump(self):
|
||||||
|
self.value = uuid.uuid4()
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('tedit')
|
||||||
|
class EditTime:
|
||||||
|
value: str = field(default_factory=time.time)
|
||||||
|
|
||||||
|
def __deepcopy__(self, memo):
|
||||||
|
return EditTime()
|
||||||
|
|
||||||
|
def __after_parse__(self, parent):
|
||||||
|
self.value = int(str(self.value), 16)
|
||||||
|
|
||||||
|
def __before_sexp__(self):
|
||||||
|
self.value = Atom(f'{int(self.value):08X}')
|
||||||
|
|
||||||
|
def bump(self):
|
||||||
|
self.value = time.time()
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('paper')
|
||||||
|
class PageSettings:
|
||||||
|
page_format: str = 'A4'
|
||||||
|
width: float = None
|
||||||
|
height: float = None
|
||||||
|
portrait: Flag() = False
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('property')
|
||||||
|
class Property:
|
||||||
|
key: str = ''
|
||||||
|
value: str = ''
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('property')
|
||||||
|
class DrawnProperty(TextMixin):
|
||||||
|
key: str = None
|
||||||
|
value: str = None
|
||||||
|
id: Named(int) = None
|
||||||
|
at: AtPos = None
|
||||||
|
unlocked: OmitDefault(Named(YesNoAtom())) = True
|
||||||
|
layer: Named(str) = None
|
||||||
|
hide: OmitDefault(Named(YesNoAtom())) = False
|
||||||
|
uuid: UUID = None
|
||||||
|
tstamp: Timestamp = None
|
||||||
|
effects: OmitDefault(TextEffect) = field(default_factory=TextEffect)
|
||||||
|
_ : SEXP_END = None
|
||||||
|
parent: object = None
|
||||||
|
|
||||||
|
def __after_parse(self, parent=None):
|
||||||
|
self.parent = parent
|
||||||
|
|
||||||
|
# Alias value for text mixin
|
||||||
|
@property
|
||||||
|
def text(self):
|
||||||
|
return self.value
|
||||||
|
|
||||||
|
@text.setter
|
||||||
|
def text(self, value):
|
||||||
|
self.value = value
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('chamfer')
|
||||||
|
class Chamfer:
|
||||||
|
top_left: Flag() = False
|
||||||
|
top_right: Flag() = False
|
||||||
|
bottom_left: Flag() = False
|
||||||
|
bottom_right: Flag() = False
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
class Foo:
|
||||||
|
pass
|
||||||
|
|
||||||
|
foo = Foo()
|
||||||
|
foo.stroke = troke(0.01, Atom.dash_dot_dot)
|
||||||
|
d = Dasher(foo)
|
||||||
|
#d = Dasher(Stroke(0.01, Atom.solid))
|
||||||
|
d.move(1, 1)
|
||||||
|
d.line(1, 2)
|
||||||
|
d.line(3, 2)
|
||||||
|
d.line(3, 1)
|
||||||
|
d.close()
|
||||||
|
|
||||||
|
print('<?xml version="1.0" standalone="no"?>')
|
||||||
|
print('<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">')
|
||||||
|
print('<svg version="1.1" width="4cm" height="3cm" viewBox="0 0 4 3" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink">')
|
||||||
|
for x1, y1, x2, y2 in d:
|
||||||
|
print(f'<path fill="none" stroke="black" stroke-width="0.01" stroke-linecap="round" d="M {x1},{y1} L {x2},{y2}"/>')
|
||||||
|
print('</svg>')
|
||||||
|
|
@ -2,6 +2,7 @@
|
||||||
Library for handling KiCad's footprint files (`*.kicad_mod`).
|
Library for handling KiCad's footprint files (`*.kicad_mod`).
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
import re
|
||||||
import copy
|
import copy
|
||||||
import enum
|
import enum
|
||||||
import string
|
import string
|
||||||
|
|
@ -11,7 +12,7 @@ import time
|
||||||
import fnmatch
|
import fnmatch
|
||||||
from itertools import chain
|
from itertools import chain
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from dataclasses import field
|
from dataclasses import field, replace
|
||||||
|
|
||||||
from .sexp import *
|
from .sexp import *
|
||||||
from .base_types import *
|
from .base_types import *
|
||||||
|
|
@ -20,6 +21,7 @@ from . import graphical_primitives as gr
|
||||||
|
|
||||||
from ..primitives import Positioned
|
from ..primitives import Positioned
|
||||||
|
|
||||||
|
from ... import __version__
|
||||||
from ... import graphic_primitives as gp
|
from ... import graphic_primitives as gp
|
||||||
from ... import graphic_objects as go
|
from ... import graphic_objects as go
|
||||||
from ... import apertures as ap
|
from ... import apertures as ap
|
||||||
|
|
@ -30,18 +32,22 @@ from ...aperture_macros.parse import GenericMacros, ApertureMacro
|
||||||
from ...aperture_macros import primitive as amp
|
from ...aperture_macros import primitive as amp
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('property')
|
class _MISSING:
|
||||||
class Property:
|
pass
|
||||||
key: str = ''
|
|
||||||
value: str = ''
|
|
||||||
|
|
||||||
|
def angle_difference(a, b):
|
||||||
|
return (b - a + math.pi) % (2*math.pi) - math.pi
|
||||||
|
|
||||||
@sexp_type('attr')
|
@sexp_type('attr')
|
||||||
class Attribute:
|
class Attribute:
|
||||||
type: AtomChoice(Atom.smd, Atom.through_hole) = None
|
type: AtomChoice(Atom.smd, Atom.through_hole) = None
|
||||||
board_only: Flag() = False
|
board_only: Flag() = False
|
||||||
|
virtual: Flag() = False # prior to 20208026
|
||||||
exclude_from_pos_files: Flag() = False
|
exclude_from_pos_files: Flag() = False
|
||||||
exclude_from_bom: Flag() = False
|
exclude_from_bom: Flag() = False
|
||||||
|
allow_missing_courtyard: Flag() = False
|
||||||
|
allow_soldermask_bridges: Flag() = False
|
||||||
|
dnp: Flag() = False
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('fp_text')
|
@sexp_type('fp_text')
|
||||||
|
|
@ -49,8 +55,9 @@ class Text:
|
||||||
type: AtomChoice(Atom.reference, Atom.value, Atom.user) = Atom.user
|
type: AtomChoice(Atom.reference, Atom.value, Atom.user) = Atom.user
|
||||||
text: str = ""
|
text: str = ""
|
||||||
at: AtPos = field(default_factory=AtPos)
|
at: AtPos = field(default_factory=AtPos)
|
||||||
unlocked: Flag() = False
|
unlocked: OmitDefault(Named(YesNoAtom())) = False
|
||||||
layer: Named(str) = None
|
layer: Named(str) = None
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
hide: Flag() = False
|
hide: Flag() = False
|
||||||
effects: TextEffect = field(default_factory=TextEffect)
|
effects: TextEffect = field(default_factory=TextEffect)
|
||||||
tstamp: Timestamp = None
|
tstamp: Timestamp = None
|
||||||
|
|
@ -67,12 +74,15 @@ class TextBox:
|
||||||
locked: Flag() = False
|
locked: Flag() = False
|
||||||
text: str = None
|
text: str = None
|
||||||
start: Rename(XYCoord) = None
|
start: Rename(XYCoord) = None
|
||||||
end: Named(XYCoord) = None
|
end: Rename(XYCoord) = None
|
||||||
|
margins: Rename(gr.Margins) = None
|
||||||
pts: PointList = None
|
pts: PointList = None
|
||||||
angle: Named(float) = 0.0
|
angle: Named(float) = 0.0
|
||||||
layer: Named(str) = None
|
layer: Named(str) = None
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
tstamp: Timestamp = None
|
tstamp: Timestamp = None
|
||||||
effects: TextEffect = field(default_factory=TextEffect)
|
effects: TextEffect = field(default_factory=TextEffect)
|
||||||
|
border: Named(YesNoAtom()) = False
|
||||||
stroke: Stroke = field(default_factory=Stroke)
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
render_cache: RenderCache = None
|
render_cache: RenderCache = None
|
||||||
|
|
||||||
|
|
@ -85,18 +95,23 @@ class Line:
|
||||||
start: Rename(XYCoord) = None
|
start: Rename(XYCoord) = None
|
||||||
end: Rename(XYCoord) = None
|
end: Rename(XYCoord) = None
|
||||||
layer: Named(str) = None
|
layer: Named(str) = None
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
width: Named(float) = None
|
width: Named(float) = None
|
||||||
stroke: Stroke = None
|
stroke: Stroke = None
|
||||||
locked: Flag() = False
|
locked: Flag() = False
|
||||||
tstamp: Timestamp = None
|
tstamp: Timestamp = None
|
||||||
|
|
||||||
|
def to_graphical_primitive(self, flip=False):
|
||||||
|
# FIXME flip
|
||||||
|
return gr.Line(self.start, self.end, self.layer, self.width, self.stroke, self.tstamp)
|
||||||
|
|
||||||
def render(self, variables=None, cache=None):
|
def render(self, variables=None, cache=None):
|
||||||
dasher = Dasher(self)
|
dasher = Dasher(self)
|
||||||
dasher.move(self.start.x, self.start.y)
|
dasher.move(self.start.x, self.start.y)
|
||||||
dasher.line(self.end.x, self.end.y)
|
dasher.line(self.end.x, self.end.y)
|
||||||
|
|
||||||
for x1, y1, x2, y2 in dasher:
|
for x1, y1, x2, y2 in dasher:
|
||||||
yield go.Line(x1, y1, x2, y2, aperture=ap.CircleAperture(dasher.width, unit=MM), unit=MM)
|
yield go.Line(x1, -y1, x2, -y2, aperture=ap.CircleAperture(dasher.width, unit=MM), unit=MM)
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('fp_rect')
|
@sexp_type('fp_rect')
|
||||||
|
|
@ -104,9 +119,10 @@ class Rectangle:
|
||||||
start: Rename(XYCoord) = None
|
start: Rename(XYCoord) = None
|
||||||
end: Rename(XYCoord) = None
|
end: Rename(XYCoord) = None
|
||||||
layer: Named(str) = None
|
layer: Named(str) = None
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
width: Named(float) = None
|
width: Named(float) = None
|
||||||
stroke: Stroke = None
|
stroke: Stroke = None
|
||||||
fill: Named(AtomChoice(Atom.solid, Atom.none)) = None
|
fill: gr.FillMode = None
|
||||||
locked: Flag() = False
|
locked: Flag() = False
|
||||||
tstamp: Timestamp = None
|
tstamp: Timestamp = None
|
||||||
|
|
||||||
|
|
@ -118,7 +134,7 @@ class Rectangle:
|
||||||
w, h = x2-x1, y2-y1
|
w, h = x2-x1, y2-y1
|
||||||
|
|
||||||
if self.fill == Atom.solid:
|
if self.fill == Atom.solid:
|
||||||
yield go.Region.from_rectangle(x1, y1, w, h, unit=MM)
|
yield go.Region.from_rectangle(x1, -y1, w, h, unit=MM)
|
||||||
|
|
||||||
dasher = Dasher(self)
|
dasher = Dasher(self)
|
||||||
dasher.move(x1, y1)
|
dasher.move(x1, y1)
|
||||||
|
|
@ -129,7 +145,7 @@ class Rectangle:
|
||||||
|
|
||||||
aperture = ap.CircleAperture(dasher.width, unit=MM)
|
aperture = ap.CircleAperture(dasher.width, unit=MM)
|
||||||
for x1, y1, x2, y2 in dasher:
|
for x1, y1, x2, y2 in dasher:
|
||||||
yield go.Line(x1, y1, x2, y2, aperture=aperture, unit=MM)
|
yield go.Line(x1, -y1, x2, -y2, aperture=aperture, unit=MM)
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('fp_circle')
|
@sexp_type('fp_circle')
|
||||||
|
|
@ -137,9 +153,10 @@ class Circle:
|
||||||
center: Rename(XYCoord) = None
|
center: Rename(XYCoord) = None
|
||||||
end: Rename(XYCoord) = None
|
end: Rename(XYCoord) = None
|
||||||
layer: Named(str) = None
|
layer: Named(str) = None
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
width: Named(float) = None
|
width: Named(float) = None
|
||||||
stroke: Stroke = None
|
stroke: Stroke = None
|
||||||
fill: Named(AtomChoice(Atom.solid, Atom.none)) = None
|
fill: gr.FillMode = None
|
||||||
locked: Flag() = False
|
locked: Flag() = False
|
||||||
tstamp: Timestamp = None
|
tstamp: Timestamp = None
|
||||||
|
|
||||||
|
|
@ -150,7 +167,7 @@ class Circle:
|
||||||
dasher = Dasher(self)
|
dasher = Dasher(self)
|
||||||
aperture = ap.CircleAperture(dasher.width or 0, unit=MM)
|
aperture = ap.CircleAperture(dasher.width or 0, unit=MM)
|
||||||
|
|
||||||
circle = go.Arc.from_circle(x, y, r, aperture=aperture, unit=MM)
|
circle = go.Arc.from_circle(x, -y, r, aperture=aperture, unit=MM)
|
||||||
|
|
||||||
if self.fill == Atom.solid:
|
if self.fill == Atom.solid:
|
||||||
yield circle.to_region()
|
yield circle.to_region()
|
||||||
|
|
@ -164,19 +181,25 @@ class Circle:
|
||||||
|
|
||||||
aperture = ap.CircleAperture(dasher.width, unit=MM)
|
aperture = ap.CircleAperture(dasher.width, unit=MM)
|
||||||
for x1, y1, x2, y2 in dasher:
|
for x1, y1, x2, y2 in dasher:
|
||||||
yield go.Line(x1, y1, x2, y2, aperture=aperture, unit=MM)
|
yield go.Line(x1, -y1, x2, -y2, aperture=aperture, unit=MM)
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('fp_arc')
|
@sexp_type('fp_arc')
|
||||||
class Arc:
|
class Arc:
|
||||||
start: Rename(XYCoord) = None
|
start: Rename(XYCoord) = None
|
||||||
mid: Rename(XYCoord) = None
|
mid: Rename(XYCoord) = None
|
||||||
end: Rename(XYCoord) = None
|
end: Rename(XYCoord) = None
|
||||||
layer: Named(str) = None
|
|
||||||
width: Named(float) = None
|
width: Named(float) = None
|
||||||
|
angle: Named(float) = None
|
||||||
stroke: Stroke = None
|
stroke: Stroke = None
|
||||||
|
layer: Named(str) = None
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
locked: Flag() = False
|
locked: Flag() = False
|
||||||
tstamp: Timestamp = None
|
tstamp: Timestamp = None
|
||||||
|
|
||||||
|
def to_graphical_primitive(self, flip=False):
|
||||||
|
# FIXME flip
|
||||||
|
return gr.Arc(self.start, self.mid, self.end, self.layer, self.width, self.stroke, self.tstamp)
|
||||||
|
|
||||||
def render(self, variables=None, cache=None):
|
def render(self, variables=None, cache=None):
|
||||||
mx, my = self.mid.x, self.mid.y
|
mx, my = self.mid.x, self.mid.y
|
||||||
|
|
@ -188,7 +211,7 @@ class Arc:
|
||||||
if math.isclose(x1, x2, abs_tol=1e-6) and math.isclose(y1, y2, abs_tol=1e-6):
|
if math.isclose(x1, x2, abs_tol=1e-6) and math.isclose(y1, y2, abs_tol=1e-6):
|
||||||
cx = (x1 + mx) / 2
|
cx = (x1 + mx) / 2
|
||||||
cy = (y1 + my) / 2
|
cy = (y1 + my) / 2
|
||||||
arc = go.Arc(x1, y1, x2, y2, cx-x1, cy-y1, clockwise=True, aperture=aperture, unit=MM)
|
arc = go.Arc(x1, -y1, x2, -y2, cx-x1, -(cy-y1), clockwise=True, aperture=aperture, unit=MM)
|
||||||
if dasher.solid:
|
if dasher.solid:
|
||||||
yield arc
|
yield arc
|
||||||
|
|
||||||
|
|
@ -198,7 +221,7 @@ class Arc:
|
||||||
dasher.segments.append((line.x1, line.y1, line.x2, line.y2))
|
dasher.segments.append((line.x1, line.y1, line.x2, line.y2))
|
||||||
|
|
||||||
for line in dasher:
|
for line in dasher:
|
||||||
yield go.Line(x1, y1, x2, y2, aperture=ap.CircleAperture(dasher.width, unit=MM), unit=MM)
|
yield go.Line(x1, -y1, x2, -y2, aperture=ap.CircleAperture(dasher.width, unit=MM), unit=MM)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
# https://stackoverflow.com/questions/56224824/how-do-i-find-the-circumcenter-of-the-triangle-using-python-without-external-lib
|
# https://stackoverflow.com/questions/56224824/how-do-i-find-the-circumcenter-of-the-triangle-using-python-without-external-lib
|
||||||
|
|
@ -207,7 +230,7 @@ class Arc:
|
||||||
cy = ((x1 * x1 + y1 * y1) * (mx - x2) + (x2 * x2 + y2 * y2) * (x1 - mx) + (mx * mx + my * my) * (x2 - x1)) / d
|
cy = ((x1 * x1 + y1 * y1) * (mx - x2) + (x2 * x2 + y2 * y2) * (x1 - mx) + (mx * mx + my * my) * (x2 - x1)) / d
|
||||||
|
|
||||||
# KiCad only has clockwise arcs.
|
# KiCad only has clockwise arcs.
|
||||||
arc = go.Arc(x1, y1, x2, y2, cx-x1, cy-y1, clockwise=False, aperture=aperture, unit=MM)
|
arc = go.Arc(x1, -y1, x2, -y2, cx-x1, -(cy-y1), clockwise=True, aperture=aperture, unit=MM)
|
||||||
if dasher.solid:
|
if dasher.solid:
|
||||||
yield arc
|
yield arc
|
||||||
|
|
||||||
|
|
@ -217,41 +240,44 @@ class Arc:
|
||||||
dasher.segments.append((line.x1, line.y1, line.x2, line.y2))
|
dasher.segments.append((line.x1, line.y1, line.x2, line.y2))
|
||||||
|
|
||||||
for line in dasher:
|
for line in dasher:
|
||||||
yield go.Line(x1, y1, x2, y2, aperture=ap.CircleAperture(dasher.width, unit=MM), unit=MM)
|
yield go.Line(x1, -y1, x2, -y2, aperture=ap.CircleAperture(dasher.width, unit=MM), unit=MM)
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('fp_poly')
|
@sexp_type('fp_poly')
|
||||||
class Polygon:
|
class Polygon:
|
||||||
pts: PointList = field(default_factory=PointList)
|
pts: PointList = field(default_factory=list)
|
||||||
layer: Named(str) = None
|
layer: Named(str) = None
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
width: Named(float) = None
|
width: Named(float) = None
|
||||||
stroke: Stroke = None
|
stroke: Stroke = None
|
||||||
fill: Named(AtomChoice(Atom.solid, Atom.none)) = None
|
fill: gr.FillMode = None
|
||||||
locked: Flag() = False
|
locked: Flag() = False
|
||||||
tstamp: Timestamp = None
|
tstamp: Timestamp = None
|
||||||
|
|
||||||
def render(self, variables=None, cache=None):
|
def render(self, variables=None, cache=None):
|
||||||
if len(self.pts.xy) < 2:
|
if len(self.pts) < 2:
|
||||||
return
|
return
|
||||||
|
|
||||||
dasher = Dasher(self)
|
dasher = Dasher(self)
|
||||||
start = self.pts.xy[0]
|
start = self.pts[0]
|
||||||
dasher.move(start.x, start.y)
|
dasher.move(start.x, start.y)
|
||||||
for point in self.pts.xy[1:]:
|
for point in self.pts[1:]:
|
||||||
dasher.line(point.x, point.y)
|
dasher.line(point.x, point.y)
|
||||||
|
|
||||||
aperture = ap.CircleAperture(dasher.width, unit=MM)
|
if dasher.width > 0:
|
||||||
for x1, y1, x2, y2 in dasher:
|
aperture = ap.CircleAperture(dasher.width, unit=MM)
|
||||||
yield go.Line(x1, y1, x2, y2, aperture=aperture, unit=MM)
|
for x1, y1, x2, y2 in dasher:
|
||||||
|
yield go.Line(x1, -y1, x2, -y2, aperture=aperture, unit=MM)
|
||||||
|
|
||||||
if self.fill == Atom.solid:
|
if self.fill == Atom.solid:
|
||||||
yield go.Region([(pt.x, pt.y) for pt in self.pts.xy], unit=MM)
|
yield go.Region([(pt.x, -pt.y) for pt in self.pts], unit=MM)
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('fp_curve')
|
@sexp_type('fp_curve')
|
||||||
class Curve:
|
class Curve:
|
||||||
pts: PointList = field(default_factory=PointList)
|
pts: PointList = field(default_factory=list)
|
||||||
layer: Named(str) = None
|
layer: Named(str) = None
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
width: Named(float) = None
|
width: Named(float) = None
|
||||||
stroke: Stroke = None
|
stroke: Stroke = None
|
||||||
locked: Flag() = False
|
locked: Flag() = False
|
||||||
|
|
@ -261,46 +287,6 @@ class Curve:
|
||||||
raise NotImplementedError('Bezier rendering is not yet supported. Please raise an issue and provide an example file.')
|
raise NotImplementedError('Bezier rendering is not yet supported. Please raise an issue and provide an example file.')
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('format')
|
|
||||||
class DimensionFormat:
|
|
||||||
prefix: Named(str) = None
|
|
||||||
suffix: Named(str) = None
|
|
||||||
units: Named(int) = 3
|
|
||||||
units_format: Named(int) = 0
|
|
||||||
precision: Named(int) = 3
|
|
||||||
override_value: Named(str) = None
|
|
||||||
suppress_zeros: Flag() = False
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('style')
|
|
||||||
class DimensionStyle:
|
|
||||||
thickness: Named(float) = None
|
|
||||||
arrow_length: Named(float) = None
|
|
||||||
text_position_mode: Named(int) = 0
|
|
||||||
extension_height: Named(float) = None
|
|
||||||
text_frame: Named(int) = 0
|
|
||||||
extension_offset: Named(str) = None
|
|
||||||
keep_text_aligned: Flag() = False
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('dimension')
|
|
||||||
class Dimension:
|
|
||||||
locked: Flag() = False
|
|
||||||
type: AtomChoice(Atom.aligned, Atom.leader, Atom.center, Atom.orthogonal, Atom.radial) = None
|
|
||||||
layer: Named(str) = None
|
|
||||||
tstamp: Timestamp = None
|
|
||||||
pts: PointList = field(default_factory=PointList)
|
|
||||||
height: Named(float) = None
|
|
||||||
orientation: Named(int) = 0
|
|
||||||
leader_length: Named(float) = None
|
|
||||||
gr_text: Named(Text) = None
|
|
||||||
format: DimensionFormat = field(default_factory=DimensionFormat)
|
|
||||||
style: DimensionStyle = field(default_factory=DimensionStyle)
|
|
||||||
|
|
||||||
def render(self, variables=None, cache=None):
|
|
||||||
raise NotImplementedError()
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('drill')
|
@sexp_type('drill')
|
||||||
class Drill:
|
class Drill:
|
||||||
oval: Flag() = False
|
oval: Flag() = False
|
||||||
|
|
@ -309,12 +295,6 @@ class Drill:
|
||||||
offset: Rename(XYCoord) = None
|
offset: Rename(XYCoord) = None
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('net')
|
|
||||||
class NetDef:
|
|
||||||
number: int = None
|
|
||||||
name: str = None
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('options')
|
@sexp_type('options')
|
||||||
class CustomPadOptions:
|
class CustomPadOptions:
|
||||||
clearance: Named(AtomChoice(Atom.outline, Atom.convexhull)) = Atom.outline
|
clearance: Named(AtomChoice(Atom.outline, Atom.convexhull)) = Atom.outline
|
||||||
|
|
@ -331,7 +311,7 @@ class CustomPadPrimitives:
|
||||||
polygons: List(gr.Polygon) = field(default_factory=list)
|
polygons: List(gr.Polygon) = field(default_factory=list)
|
||||||
curves: List(gr.Curve) = field(default_factory=list)
|
curves: List(gr.Curve) = field(default_factory=list)
|
||||||
width: Named(float) = None
|
width: Named(float) = None
|
||||||
fill: Named(YesNoAtom()) = True
|
fill: gr.FillMode = True
|
||||||
|
|
||||||
def all(self):
|
def all(self):
|
||||||
yield from self.lines
|
yield from self.lines
|
||||||
|
|
@ -342,36 +322,33 @@ class CustomPadPrimitives:
|
||||||
yield from self.curves
|
yield from self.curves
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('chamfer')
|
|
||||||
class Chamfer:
|
|
||||||
top_left: Flag() = False
|
|
||||||
top_right: Flag() = False
|
|
||||||
bottom_left: Flag() = False
|
|
||||||
bottom_right: Flag() = False
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('pad')
|
@sexp_type('pad')
|
||||||
class Pad:
|
class Pad(NetMixin):
|
||||||
number: str = None
|
number: str = None
|
||||||
type: AtomChoice(Atom.thru_hole, Atom.smd, Atom.connect, Atom.np_thru_hole) = None
|
type: AtomChoice(Atom.thru_hole, Atom.smd, Atom.connect, Atom.np_thru_hole) = None
|
||||||
shape: AtomChoice(Atom.circle, Atom.rect, Atom.oval, Atom.trapezoid, Atom.roundrect, Atom.custom) = None
|
shape: AtomChoice(Atom.circle, Atom.rect, Atom.oval, Atom.trapezoid, Atom.roundrect, Atom.custom) = None
|
||||||
at: AtPos = field(default_factory=AtPos)
|
at: AtPos = field(default_factory=AtPos)
|
||||||
locked: Wrap(Flag()) = False
|
locked: Flag() = False
|
||||||
size: Rename(XYCoord) = field(default_factory=XYCoord)
|
size: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
drill: Drill = None
|
drill: Drill = None
|
||||||
layers: Named(Array(str)) = field(default_factory=list)
|
layers: Named(Array(str)) = field(default_factory=list)
|
||||||
properties: List(Property) = field(default_factory=list)
|
properties: List(Property) = field(default_factory=list)
|
||||||
remove_unused_layers: Wrap(Flag()) = False
|
remove_unused_layers: Named(YesNoAtom()) = False
|
||||||
keep_end_layers: Wrap(Flag()) = False
|
keep_end_layers: Named(YesNoAtom()) = False
|
||||||
|
zone_layer_connections: Named(Array(str)) = field(default_factory=list)
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
rect_delta: Rename(XYCoord) = None
|
rect_delta: Rename(XYCoord) = None
|
||||||
roundrect_rratio: Named(float) = None
|
roundrect_rratio: Named(float) = None
|
||||||
thermal_bridge_angle: Named(int) = 45
|
thermal_bridge_angle: Named(int) = 45
|
||||||
|
thermal_bridge_width: Named(float) = 0.5
|
||||||
chamfer_ratio: Named(float) = None
|
chamfer_ratio: Named(float) = None
|
||||||
chamfer: Chamfer = None
|
chamfer: Chamfer = None
|
||||||
net: NetDef = None
|
net: Net = None
|
||||||
tstamp: Timestamp = None
|
tstamp: Timestamp = None
|
||||||
pin_function: Named(str) = None
|
pin_function: Named(str) = None
|
||||||
pintype: Named(str) = None
|
pintype: Named(str) = None
|
||||||
|
pinfunction: Named(str) = None
|
||||||
|
teardrops: gr.TeardropSpec = None
|
||||||
die_length: Named(float) = None
|
die_length: Named(float) = None
|
||||||
solder_mask_margin: Named(float) = None
|
solder_mask_margin: Named(float) = None
|
||||||
solder_paste_margin: Named(float) = None
|
solder_paste_margin: Named(float) = None
|
||||||
|
|
@ -381,7 +358,41 @@ class Pad:
|
||||||
thermal_width: Named(float) = None
|
thermal_width: Named(float) = None
|
||||||
thermal_gap: Named(float) = None
|
thermal_gap: Named(float) = None
|
||||||
options: OmitDefault(CustomPadOptions) = None
|
options: OmitDefault(CustomPadOptions) = None
|
||||||
|
padstack: gr.PadStack = None
|
||||||
primitives: OmitDefault(CustomPadPrimitives) = None
|
primitives: OmitDefault(CustomPadPrimitives) = None
|
||||||
|
_: SEXP_END = None
|
||||||
|
footprint: object = field(repr=False, default=None)
|
||||||
|
|
||||||
|
def __after_parse__(self, parent=None):
|
||||||
|
self.layers = unfuck_layers(self.layers)
|
||||||
|
|
||||||
|
def __before_sexp__(self):
|
||||||
|
self.layers = fuck_layers(self.layers)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def abs_pos(self):
|
||||||
|
if self.footprint:
|
||||||
|
px, py, pr = self.footprint.at.x, self.footprint.at.y, self.footprint.at.rotation
|
||||||
|
else:
|
||||||
|
px, py, pr = 0, 0, 0
|
||||||
|
|
||||||
|
x, y = rotate_point(self.at.x, self.at.y, math.radians(pr))
|
||||||
|
return x+px, y+py, self.at.rotation, False
|
||||||
|
|
||||||
|
@property
|
||||||
|
def layer_mask(self):
|
||||||
|
return layer_mask(self.layers)
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.at = self.at.with_offset(x, y)
|
||||||
|
|
||||||
|
def find_connected_footprints(self, **filters):
|
||||||
|
""" Find footprints connected to the same net as this pad """
|
||||||
|
return self.footprint.board.find_footprints(net=self.net.name, **filters)
|
||||||
|
|
||||||
|
def find_same_net(self, include_vias=True):
|
||||||
|
""" Find traces and vias of the same net as this pad. """
|
||||||
|
return self.footprint.board.find_traces(self.net.name, include_vias=include_vias)
|
||||||
|
|
||||||
def render(self, variables=None, margin=None, cache=None):
|
def render(self, variables=None, margin=None, cache=None):
|
||||||
#if self.type in (Atom.connect, Atom.np_thru_hole):
|
#if self.type in (Atom.connect, Atom.np_thru_hole):
|
||||||
|
|
@ -401,7 +412,7 @@ class Pad:
|
||||||
else:
|
else:
|
||||||
aperture = self.aperture(margin)
|
aperture = self.aperture(margin)
|
||||||
|
|
||||||
yield go.Flash(self.at.x+ox, self.at.y+oy, aperture, unit=MM)
|
yield go.Flash(self.at.x+ox, -(self.at.y+oy), aperture, unit=MM)
|
||||||
|
|
||||||
def aperture(self, margin=None):
|
def aperture(self, margin=None):
|
||||||
rotation = math.radians(self.at.rotation)
|
rotation = math.radians(self.at.rotation)
|
||||||
|
|
@ -412,16 +423,16 @@ class Pad:
|
||||||
|
|
||||||
elif self.shape == Atom.rect:
|
elif self.shape == Atom.rect:
|
||||||
if margin > 0:
|
if margin > 0:
|
||||||
return ap.ApertureMacroInstance(GenericMacros.rounded_rect,
|
return GenericMacros.rounded_rect(self.size.x+2*margin,
|
||||||
(self.size.x+2*margin, self.size.y+2*margin,
|
self.size.y+2*margin,
|
||||||
margin,
|
margin,
|
||||||
0, 0, # no hole
|
0, # no hole
|
||||||
rotation), unit=MM)
|
rotation)
|
||||||
else:
|
else:
|
||||||
return ap.RectangleAperture(self.size.x+2*margin, self.size.y+2*margin, unit=MM).rotated(rotation)
|
return ap.RectangleAperture(self.size.x+2*margin, self.size.y+2*margin, unit=MM).rotated(-rotation)
|
||||||
|
|
||||||
elif self.shape == Atom.oval:
|
elif self.shape == Atom.oval:
|
||||||
return ap.ObroundAperture(self.size.x+2*margin, self.size.y+2*margin, unit=MM).rotated(rotation)
|
return ap.ObroundAperture(self.size.x+2*margin, self.size.y+2*margin, unit=MM).rotated(-rotation)
|
||||||
|
|
||||||
elif self.shape == Atom.trapezoid:
|
elif self.shape == Atom.trapezoid:
|
||||||
# KiCad's trapezoid aperture "rect_delta" param is just weird to the point that I think it's probably
|
# KiCad's trapezoid aperture "rect_delta" param is just weird to the point that I think it's probably
|
||||||
|
|
@ -443,30 +454,31 @@ class Pad:
|
||||||
# Note: KiCad already uses MM units, so no conversion needed here.
|
# Note: KiCad already uses MM units, so no conversion needed here.
|
||||||
|
|
||||||
alpha = math.atan(y / dy) if dy > 0 else 0
|
alpha = math.atan(y / dy) if dy > 0 else 0
|
||||||
return ap.ApertureMacroInstance(GenericMacros.isosceles_trapezoid,
|
return GenericMacros.isosceles_trapezoid(x+dy+2*margin*math.cos(alpha),
|
||||||
(x+dy+2*margin*math.cos(alpha), y+2*margin,
|
y+2*margin,
|
||||||
2*dy,
|
2*dy,
|
||||||
0, 0, # no hole
|
0, # no hole
|
||||||
rotation), unit=MM)
|
-rotation + math.pi)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
return ap.ApertureMacroInstance(GenericMacros.rounded_isosceles_trapezoid,
|
return GenericMacros.rounded_isosceles_trapezoid(x+dy,
|
||||||
(x+dy, y,
|
y,
|
||||||
2*dy, margin,
|
2*dy,
|
||||||
0, 0, # no hole
|
margin,
|
||||||
rotation), unit=MM)
|
0, # no hole
|
||||||
|
-rotation + math.pi)
|
||||||
|
|
||||||
elif self.shape == Atom.roundrect:
|
elif self.shape == Atom.roundrect:
|
||||||
x, y = self.size.x, self.size.y
|
x, y = self.size.x, self.size.y
|
||||||
r = min(x, y) * self.roundrect_rratio
|
r = min(x, y) * self.roundrect_rratio
|
||||||
if margin > -r:
|
if margin > -r:
|
||||||
return ap.ApertureMacroInstance(GenericMacros.rounded_rect,
|
return GenericMacros.rounded_rect(x+2*margin,
|
||||||
(x+2*margin, y+2*margin,
|
y+2*margin,
|
||||||
r+margin,
|
r+margin,
|
||||||
0, 0, # no hole
|
0, # no hole
|
||||||
rotation), unit=MM)
|
rotation)
|
||||||
else:
|
else:
|
||||||
return ap.RectangleAperture(x+margin, y+margin, unit=MM).rotated(rotation)
|
return ap.RectangleAperture(x+margin, y+margin, unit=MM).rotated(-rotation)
|
||||||
|
|
||||||
elif self.shape == Atom.custom:
|
elif self.shape == Atom.custom:
|
||||||
primitives = []
|
primitives = []
|
||||||
|
|
@ -475,7 +487,7 @@ class Pad:
|
||||||
for obj in self.primitives.all():
|
for obj in self.primitives.all():
|
||||||
for gn_obj in obj.render():
|
for gn_obj in obj.render():
|
||||||
if margin and isinstance(gn_obj, (go.Line, go.Arc)):
|
if margin and isinstance(gn_obj, (go.Line, go.Arc)):
|
||||||
gn_obj = gn_obj.dilated(margin)
|
gn_obj = replace(gn_obj, aperture=gn_obj.aperture.dilated(margin))
|
||||||
|
|
||||||
if isinstance(gn_obj, go.Region) and margin > 0:
|
if isinstance(gn_obj, go.Region) and margin > 0:
|
||||||
for line in gn_obj.outline_objects(ap.CircleAperture(2*margin, unit=MM)):
|
for line in gn_obj.outline_objects(ap.CircleAperture(2*margin, unit=MM)):
|
||||||
|
|
@ -508,7 +520,7 @@ class Pad:
|
||||||
elif self.options.anchor == Atom.circle and self.size.x > 0:
|
elif self.options.anchor == Atom.circle and self.size.x > 0:
|
||||||
primitives.append(amp.Circle(MM, 1, self.size.x+2*margin, 0, 0, 0))
|
primitives.append(amp.Circle(MM, 1, self.size.x+2*margin, 0, 0, 0))
|
||||||
|
|
||||||
macro = ApertureMacro(primitives=tuple(primitives)).rotated(rotation)
|
macro = ApertureMacro(primitives=tuple(primitives)).rotated(-rotation)
|
||||||
return ap.ApertureMacroInstance(macro, unit=MM)
|
return ap.ApertureMacroInstance(macro, unit=MM)
|
||||||
|
|
||||||
def render_drill(self):
|
def render_drill(self):
|
||||||
|
|
@ -533,38 +545,40 @@ class Pad:
|
||||||
dy = 0
|
dy = 0
|
||||||
|
|
||||||
aperture = ap.ExcellonTool(min(dia, w), plated=plated, unit=MM)
|
aperture = ap.ExcellonTool(min(dia, w), plated=plated, unit=MM)
|
||||||
l = go.Line(ox-dx, oy-dy, ox+dx, oy+dy, aperture=aperture, unit=MM)
|
l = go.Line(ox-dx, -(oy-dy), ox+dx, -(oy+dy), aperture=aperture, unit=MM)
|
||||||
l.rotate(math.radians(self.at.rotation))
|
l.rotate(math.radians(self.at.rotation))
|
||||||
l.offset(self.at.x, self.at.y)
|
l.offset(self.at.x, -self.at.y)
|
||||||
yield l
|
yield l
|
||||||
|
|
||||||
else:
|
else:
|
||||||
aperture = ap.ExcellonTool(self.drill.diameter, plated=plated, unit=MM)
|
aperture = ap.ExcellonTool(self.drill.diameter, plated=plated, unit=MM)
|
||||||
yield go.Flash(self.at.x, self.at.y, aperture=aperture, unit=MM)
|
yield go.Flash(self.at.x, -self.at.y, aperture=aperture, unit=MM)
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('group')
|
|
||||||
class Group:
|
|
||||||
name: str = ""
|
|
||||||
id: Named(str) = ""
|
|
||||||
members: Named(List(str)) = field(default_factory=list)
|
|
||||||
|
|
||||||
|
|
||||||
@sexp_type('model')
|
@sexp_type('model')
|
||||||
class Model:
|
class Model:
|
||||||
name: str = ''
|
name: str = ''
|
||||||
|
hide: Flag() = False
|
||||||
at: Named(XYZCoord) = field(default_factory=XYZCoord)
|
at: Named(XYZCoord) = field(default_factory=XYZCoord)
|
||||||
offset: Named(XYZCoord) = field(default_factory=XYZCoord)
|
offset: Named(XYZCoord) = field(default_factory=XYZCoord)
|
||||||
|
opacity: Named(float) = None
|
||||||
scale: Named(XYZCoord) = field(default_factory=XYZCoord)
|
scale: Named(XYZCoord) = field(default_factory=XYZCoord)
|
||||||
rotate: Named(XYZCoord) = field(default_factory=XYZCoord)
|
rotate: Named(XYZCoord) = field(default_factory=XYZCoord)
|
||||||
|
|
||||||
|
|
||||||
SUPPORTED_FILE_FORMAT_VERSIONS = [20210108, 20211014, 20221018]
|
@sexp_type('component_classes')
|
||||||
|
class FootprintComponentClasses:
|
||||||
|
classes: List(Named(str, name='class')) = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
SUPPORTED_FILE_FORMAT_VERSIONS = [20210108, 20211014, 20221018, 20230517]
|
||||||
@sexp_type('footprint')
|
@sexp_type('footprint')
|
||||||
class Footprint:
|
class Footprint:
|
||||||
name: str = None
|
name: str = None
|
||||||
_version: Named(int, name='version') = 20210108
|
_version: Named(int, name='version') = 20221018
|
||||||
generator: Named(Atom) = Atom.kicad_library_utils
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
generator: Named(str) = Atom.gerbonara
|
||||||
|
generator_version: Named(str) = __version__
|
||||||
locked: Flag() = False
|
locked: Flag() = False
|
||||||
placed: Flag() = False
|
placed: Flag() = False
|
||||||
layer: Named(str) = 'F.Cu'
|
layer: Named(str) = 'F.Cu'
|
||||||
|
|
@ -573,20 +587,24 @@ class Footprint:
|
||||||
at: AtPos = field(default_factory=AtPos)
|
at: AtPos = field(default_factory=AtPos)
|
||||||
descr: Named(str) = None
|
descr: Named(str) = None
|
||||||
tags: Named(str) = None
|
tags: Named(str) = None
|
||||||
properties: List(Property) = field(default_factory=list)
|
properties: List(DrawnProperty) = field(default_factory=list)
|
||||||
|
component_classes: FootprintComponentClasses = None
|
||||||
path: Named(str) = None
|
path: Named(str) = None
|
||||||
|
sheetname: Named(str) = None
|
||||||
|
sheetfile: Named(str) = None
|
||||||
autoplace_cost90: Named(float) = None
|
autoplace_cost90: Named(float) = None
|
||||||
autoplace_cost180: Named(float) = None
|
autoplace_cost180: Named(float) = None
|
||||||
solder_mask_margin: Named(float) = None
|
solder_mask_margin: Named(float) = None
|
||||||
|
solder_paste_margin_ratio: Named(float) = None
|
||||||
solder_paste_margin: Named(float) = None
|
solder_paste_margin: Named(float) = None
|
||||||
solder_paste_ratio: Named(float) = None
|
solder_paste_ratio: Named(float) = None
|
||||||
clearance: Named(float) = None
|
clearance: Named(float) = None
|
||||||
zone_connect: Named(int) = None
|
zone_connect: Named(int) = None
|
||||||
thermal_width: Named(float) = None
|
thermal_width: Named(float) = None
|
||||||
thermal_gap: Named(float) = None
|
thermal_gap: Named(float) = None
|
||||||
attributes: List(Attribute) = field(default_factory=list)
|
attributes: Attribute = field(default_factory=Attribute)
|
||||||
private_layers: Named(str) = None
|
private_layers: Named(str) = None
|
||||||
net_tie_pad_groups: Named(str) = None
|
net_tie_pad_groups: Named(Array(str)) = None
|
||||||
texts: List(Text) = field(default_factory=list)
|
texts: List(Text) = field(default_factory=list)
|
||||||
text_boxes: List(TextBox) = field(default_factory=list)
|
text_boxes: List(TextBox) = field(default_factory=list)
|
||||||
lines: List(Line) = field(default_factory=list)
|
lines: List(Line) = field(default_factory=list)
|
||||||
|
|
@ -595,14 +613,98 @@ class Footprint:
|
||||||
arcs: List(Arc) = field(default_factory=list)
|
arcs: List(Arc) = field(default_factory=list)
|
||||||
polygons: List(Polygon) = field(default_factory=list)
|
polygons: List(Polygon) = field(default_factory=list)
|
||||||
curves: List(Curve) = field(default_factory=list)
|
curves: List(Curve) = field(default_factory=list)
|
||||||
dimensions: List(Dimension) = field(default_factory=list)
|
dimensions: List(gr.Dimension) = field(default_factory=list)
|
||||||
pads: List(Pad) = field(default_factory=list)
|
pads: List(Pad) = field(default_factory=list)
|
||||||
zones: List(Zone) = field(default_factory=list)
|
zones: List(Zone) = field(default_factory=list)
|
||||||
groups: List(Group) = field(default_factory=list)
|
groups: List(Group) = field(default_factory=list)
|
||||||
|
embedded_fonts: Named(YesNoAtom()) = False
|
||||||
models: List(Model) = field(default_factory=list)
|
models: List(Model) = field(default_factory=list)
|
||||||
_ : SEXP_END = None
|
_ : SEXP_END = None
|
||||||
original_filename: str = None
|
original_filename: str = None
|
||||||
_bounding_box: tuple = None
|
board: object = field(repr=False, default=None)
|
||||||
|
|
||||||
|
def __after_parse__(self, parent):
|
||||||
|
for pad in self.pads:
|
||||||
|
pad.footprint = self
|
||||||
|
|
||||||
|
def property_value(self, key, default=_MISSING):
|
||||||
|
for prop in self.properties:
|
||||||
|
if prop.key == key:
|
||||||
|
return prop.value
|
||||||
|
|
||||||
|
if default is not _MISSING:
|
||||||
|
return default
|
||||||
|
|
||||||
|
raise IndexError(f'Footprint has no property named "{key}"')
|
||||||
|
|
||||||
|
def set_property(self, key, value, x=0, y=0, rotation=0, layer='F.Fab', hide=True, effects=None):
|
||||||
|
for prop in self.properties:
|
||||||
|
if prop.key == key:
|
||||||
|
old_value, prop.value = prop.value, value
|
||||||
|
return old_value
|
||||||
|
|
||||||
|
if effects is None:
|
||||||
|
effects = TextEffect()
|
||||||
|
|
||||||
|
self.properties.append(DrawnProperty(key, value,
|
||||||
|
at=AtPos(x, y, rotation, unlocked=True),
|
||||||
|
layer=layer,
|
||||||
|
hide=hide,
|
||||||
|
effects=effects))
|
||||||
|
|
||||||
|
def make_standard_properties(self):
|
||||||
|
if not self.property_value('Reference', None):
|
||||||
|
self.set_property('Reference', 'REF**', 0, 0, 0, 'F.SilkS')
|
||||||
|
|
||||||
|
if not self.property_value('Value', None):
|
||||||
|
self.set_property('Value', self.name or 'VAL**', 0, 0, 0, hide=False)
|
||||||
|
|
||||||
|
if not self.property_value('Footprint', None):
|
||||||
|
self.set_property('Footprint', '', 0, 0, 0)
|
||||||
|
|
||||||
|
if not self.property_value('Datasheet', None):
|
||||||
|
self.set_property('Datasheet', '', 0, 0, 0)
|
||||||
|
|
||||||
|
if not self.property_value('Description', None):
|
||||||
|
self.set_property('Description', self.descr or '', 0, 0, 0)
|
||||||
|
|
||||||
|
def reset_nets(self):
|
||||||
|
for pad in self.pads:
|
||||||
|
pad.reset_net()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def pads_by_number(self):
|
||||||
|
return {(int(pad.number) if pad.number.isnumeric() else pad.number): pad for pad in self.pads if pad.number}
|
||||||
|
|
||||||
|
def find_pads(self, number=None, net=None):
|
||||||
|
for pad in self.pads:
|
||||||
|
if number is not None and pad.number == str(number):
|
||||||
|
yield pad
|
||||||
|
elif isinstance(net, str) and fnmatch.fnmatch(pad.net.name, net):
|
||||||
|
yield pad
|
||||||
|
elif net is not None and pad.net.number == net:
|
||||||
|
yield pad
|
||||||
|
|
||||||
|
def pad(self, number=None, net=None):
|
||||||
|
candidates = list(self.find_pads(number=number, net=net))
|
||||||
|
if not candidates:
|
||||||
|
raise IndexError(f'No such pad "{number or net}"')
|
||||||
|
|
||||||
|
if len(candidates) > 1:
|
||||||
|
raise IndexError(f'Ambiguous pad "{number or net}", {len(candidates)} matching pads.')
|
||||||
|
|
||||||
|
return candidates[0]
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.at = self.at.with_offset(x, y)
|
||||||
|
|
||||||
|
def copy_placement(self, template):
|
||||||
|
# Fix up rotation of pads - KiCad saves each pad's rotation in *absolute* coordinates, not relative to the
|
||||||
|
# footprint. Because we overwrite the footprint's rotation below, we have to first fix all pads to match the
|
||||||
|
# new rotation.
|
||||||
|
self.rotate(math.radians(template.at.rotation - self.at.rotation))
|
||||||
|
self.at = copy.copy(template.at)
|
||||||
|
self.side = template.side
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def version(self):
|
def version(self):
|
||||||
|
|
@ -613,9 +715,36 @@ class Footprint:
|
||||||
if value not in SUPPORTED_FILE_FORMAT_VERSIONS:
|
if value not in SUPPORTED_FILE_FORMAT_VERSIONS:
|
||||||
raise FormatError(f'File format version {value} is not supported. Supported versions are {", ".join(map(str, SUPPORTED_FILE_FORMAT_VERSIONS))}.')
|
raise FormatError(f'File format version {value} is not supported. Supported versions are {", ".join(map(str, SUPPORTED_FILE_FORMAT_VERSIONS))}.')
|
||||||
|
|
||||||
|
@property
|
||||||
|
def reference(self):
|
||||||
|
return self.property_value('Reference')
|
||||||
|
|
||||||
|
@reference.setter
|
||||||
|
def reference(self, value):
|
||||||
|
self.set_property('Reference', value)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def parsed_reference(self):
|
||||||
|
ref = self.reference
|
||||||
|
if (m := re.match(r'^.*[^0-9]([0-9]+)$', ref)):
|
||||||
|
return m.group(0), int(m.group(1))
|
||||||
|
else:
|
||||||
|
return ref
|
||||||
|
|
||||||
|
@property
|
||||||
|
def value(self):
|
||||||
|
return self.property_value('Value')
|
||||||
|
|
||||||
|
@value.setter
|
||||||
|
def value(self, value):
|
||||||
|
self.set_property('Value', value)
|
||||||
|
|
||||||
def write(self, filename=None):
|
def write(self, filename=None):
|
||||||
with open(filename or self.original_filename, 'w') as f:
|
with open(filename or self.original_filename, 'w') as f:
|
||||||
f.write(build_sexp(sexp(self)))
|
f.write(self.serialize())
|
||||||
|
|
||||||
|
def serialize(self):
|
||||||
|
return build_sexp(sexp(type(self), self)[0])
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def open_pretty(kls, pretty_dir, fp_name, *args, **kwargs):
|
def open_pretty(kls, pretty_dir, fp_name, *args, **kwargs):
|
||||||
|
|
@ -638,11 +767,111 @@ class Footprint:
|
||||||
def load(kls, data, *args, **kwargs):
|
def load(kls, data, *args, **kwargs):
|
||||||
return kls.parse(data, *args, **kwargs)
|
return kls.parse(data, *args, **kwargs)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def side(self):
|
||||||
|
return 'front' if self.layer == 'F.Cu' else 'back'
|
||||||
|
|
||||||
|
@side.setter
|
||||||
|
def side(self, value):
|
||||||
|
if value not in ('front', 'back'):
|
||||||
|
raise ValueError(f'side must be either "front" or "back", not {side!r}')
|
||||||
|
|
||||||
|
if self.side != value:
|
||||||
|
self.flip()
|
||||||
|
|
||||||
|
def flip(self):
|
||||||
|
def flip_layer(name):
|
||||||
|
if name.startswith('F.'):
|
||||||
|
return f'B.{name[2:]}'
|
||||||
|
elif name.startswith('B.'):
|
||||||
|
return f'F.{name[2:]}'
|
||||||
|
else:
|
||||||
|
return name
|
||||||
|
|
||||||
|
self.layer = flip_layer(self.layer)
|
||||||
|
for obj in self.objects():
|
||||||
|
if getattr(obj, 'layer', None) is not None:
|
||||||
|
obj.layer = flip_layer(obj.layer)
|
||||||
|
|
||||||
|
if hasattr(obj, 'layers'):
|
||||||
|
obj.layers = [flip_layer(name) for name in obj.layers]
|
||||||
|
|
||||||
|
for obj in chain(self.texts, self.text_boxes):
|
||||||
|
obj.effects.justify.mirror = not obj.effects.justify.mirror
|
||||||
|
|
||||||
|
for obj in self.properties:
|
||||||
|
if obj.layer is not None:
|
||||||
|
obj.effects.justify.mirror = not obj.effects.justify.mirror
|
||||||
|
obj.layer = flip_layer(obj.layer)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def single_sided(self):
|
def single_sided(self):
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
|
||||||
|
def face(self, direction, pad=None, net=None):
|
||||||
|
if not net and not pad:
|
||||||
|
pad = '1'
|
||||||
|
|
||||||
def objects(self, text=False, pads=True):
|
candidates = list(self.find_pads(net=net, number=pad))
|
||||||
|
if len(candidates) == 0:
|
||||||
|
raise KeyError(f'Reference pad "{net or pad}" not found.')
|
||||||
|
|
||||||
|
if len(candidates) > 1:
|
||||||
|
raise KeyError(f'Reference pad "{net or pad}" is ambiguous, {len(candidates)} matching pads found.')
|
||||||
|
|
||||||
|
pad = candidates[0]
|
||||||
|
pad_angle = math.atan2(pad.at.y, pad.at.x)
|
||||||
|
|
||||||
|
target_angle = {
|
||||||
|
'right': 0,
|
||||||
|
'top right': math.pi/4,
|
||||||
|
'top': math.pi/2,
|
||||||
|
'top left': 3*math.pi/4,
|
||||||
|
'left': math.pi,
|
||||||
|
'bottom left': -3*math.pi/4,
|
||||||
|
'bottom': -math.pi/2,
|
||||||
|
'bottom right': -math.pi/4}.get(direction, direction)
|
||||||
|
|
||||||
|
delta = angle_difference(target_angle, pad_angle)
|
||||||
|
adj = round(delta / (math.pi/2)) * math.pi/2
|
||||||
|
self.set_rotation(adj)
|
||||||
|
|
||||||
|
def rotate(self, angle=None, cx=None, cy=None, **reference_pad):
|
||||||
|
""" Rotate this footprint by the given angle in radians, counter-clockwise. When (cx, cy) are given, rotate
|
||||||
|
around the given coordinates in the global coordinate space. Otherwise rotate around the footprint's origin. """
|
||||||
|
if (cx, cy) != (None, None):
|
||||||
|
x, y = self.at.x-cx, self.at.y-cy
|
||||||
|
self.at.x = math.cos(-angle)*x - math.sin(-angle)*y + cx
|
||||||
|
self.at.y = math.sin(-angle)*x + math.cos(-angle)*y + cy
|
||||||
|
|
||||||
|
self.at.rotation = (self.at.rotation + math.degrees(angle)) % 360
|
||||||
|
|
||||||
|
for pad in self.pads:
|
||||||
|
pad.at.rotation = (pad.at.rotation + math.degrees(angle)) % 360
|
||||||
|
|
||||||
|
for prop in self.properties:
|
||||||
|
if prop.at is not None:
|
||||||
|
prop.at.rotation = (prop.at.rotation + math.degrees(angle)) % 360
|
||||||
|
|
||||||
|
for text in self.texts:
|
||||||
|
text.at.rotation = (text.at.rotation + math.degrees(angle)) % 360
|
||||||
|
|
||||||
|
def set_rotation(self, angle):
|
||||||
|
old_deg = self.at.rotation
|
||||||
|
new_deg = self.at.rotation = -math.degrees(angle)
|
||||||
|
delta = new_deg - old_deg
|
||||||
|
|
||||||
|
for pad in self.pads:
|
||||||
|
pad.at.rotation = (pad.at.rotation + delta) % 360
|
||||||
|
|
||||||
|
for prop in self.properties:
|
||||||
|
if prop.at is not None:
|
||||||
|
prop.at.rotation = (prop.at.rotation + delta) % 360
|
||||||
|
|
||||||
|
for text in self.texts:
|
||||||
|
text.at.rotation = (text.at.rotation + delta) % 360
|
||||||
|
|
||||||
|
def objects(self, text=False, pads=True, groups=True, zones=True):
|
||||||
return chain(
|
return chain(
|
||||||
(self.texts if text else []),
|
(self.texts if text else []),
|
||||||
(self.text_boxes if text else []),
|
(self.text_boxes if text else []),
|
||||||
|
|
@ -653,15 +882,19 @@ class Footprint:
|
||||||
self.polygons,
|
self.polygons,
|
||||||
self.curves,
|
self.curves,
|
||||||
(self.dimensions if text else []),
|
(self.dimensions if text else []),
|
||||||
(self.pads if pads else []))
|
(self.pads if pads else []),
|
||||||
|
(self.zones if zones else []),
|
||||||
|
self.groups if groups else [])
|
||||||
|
|
||||||
def render(self, layer_stack, layer_map, x=0, y=0, rotation=0, text=False, side=None, variables={}, cache=None):
|
def render(self, layer_stack, layer_map=None, x=0, y=0, rotation=0, text=False, flip=False, variables={}, cache=None):
|
||||||
x += self.at.x
|
x += self.at.x
|
||||||
y += self.at.y
|
y += self.at.y
|
||||||
rotation += math.radians(self.at.rotation)
|
rotation += math.radians(self.at.rotation)
|
||||||
flip = (side != 'top') if side else (self.layer != 'F.Cu')
|
|
||||||
|
|
||||||
for obj in self.objects(pads=False, text=text):
|
if layer_map is None:
|
||||||
|
layer_map = {kc_id: gn_id for kc_id, gn_id in LAYER_MAP_K2G.items() if gn_id in layer_stack}
|
||||||
|
|
||||||
|
for obj in self.objects(pads=False, text=text, zones=False, groups=False):
|
||||||
if not (layer := layer_map.get(obj.layer)):
|
if not (layer := layer_map.get(obj.layer)):
|
||||||
continue
|
continue
|
||||||
|
|
||||||
|
|
@ -715,38 +948,13 @@ class Footprint:
|
||||||
layer_stack.drill_pth.append(fe)
|
layer_stack.drill_pth.append(fe)
|
||||||
|
|
||||||
def bounding_box(self, unit=MM):
|
def bounding_box(self, unit=MM):
|
||||||
if not self._bounding_box:
|
if not hasattr(self, '_bounding_box'):
|
||||||
stack = LayerStack()
|
stack = LayerStack()
|
||||||
layer_map = {kc_id: gn_id for kc_id, gn_id in LAYER_MAP_K2G.items() if gn_id in stack}
|
self.render(stack, layer_map=None, x=0, y=0, rotation=0, flip=False, text=False, variables={})
|
||||||
self.render(stack, layer_map, x=0, y=0, rotation=0, side='top', text=False, variables={})
|
|
||||||
self._bounding_box = stack.bounding_box(unit)
|
self._bounding_box = stack.bounding_box(unit)
|
||||||
return self._bounding_box
|
return self._bounding_box
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
LAYER_MAP_K2G = {
|
|
||||||
'F.Cu': ('top', 'copper'),
|
|
||||||
'B.Cu': ('bottom', 'copper'),
|
|
||||||
'F.SilkS': ('top', 'silk'),
|
|
||||||
'B.SilkS': ('bottom', 'silk'),
|
|
||||||
'F.Paste': ('top', 'paste'),
|
|
||||||
'B.Paste': ('bottom', 'paste'),
|
|
||||||
'F.Mask': ('top', 'mask'),
|
|
||||||
'B.Mask': ('bottom', 'mask'),
|
|
||||||
'B.CrtYd': ('bottom', 'courtyard'),
|
|
||||||
'F.CrtYd': ('top', 'courtyard'),
|
|
||||||
'B.Fab': ('bottom', 'fabrication'),
|
|
||||||
'F.Fab': ('top', 'fabrication'),
|
|
||||||
'B.Adhes': ('bottom', 'adhesive'),
|
|
||||||
'F.Adhes': ('top', 'adhesive'),
|
|
||||||
'Dwgs.User': ('mechanical', 'drawings'),
|
|
||||||
'Cmts.User': ('mechanical', 'comments'),
|
|
||||||
'Edge.Cuts': ('mechanical', 'outline'),
|
|
||||||
}
|
|
||||||
|
|
||||||
LAYER_MAP_G2K = {v: k for k, v in LAYER_MAP_K2G.items()}
|
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class FootprintInstance(Positioned):
|
class FootprintInstance(Positioned):
|
||||||
sexp: Footprint = None
|
sexp: Footprint = None
|
||||||
|
|
@ -756,7 +964,7 @@ class FootprintInstance(Positioned):
|
||||||
variables: dict = field(default_factory=lambda: {})
|
variables: dict = field(default_factory=lambda: {})
|
||||||
|
|
||||||
def render(self, layer_stack, cache=None):
|
def render(self, layer_stack, cache=None):
|
||||||
x, y, rotation = self.abs_pos
|
x, y, rotation, flip= self.abs_pos
|
||||||
x, y = MM(x, self.unit), MM(y, self.unit)
|
x, y = MM(x, self.unit), MM(y, self.unit)
|
||||||
|
|
||||||
variables = dict(self.variables)
|
variables = dict(self.variables)
|
||||||
|
|
@ -767,17 +975,16 @@ class FootprintInstance(Positioned):
|
||||||
if self.value is not None:
|
if self.value is not None:
|
||||||
variables['VALUE'] = str(self.value)
|
variables['VALUE'] = str(self.value)
|
||||||
|
|
||||||
layer_map = {kc_id: gn_id for kc_id, gn_id in LAYER_MAP_K2G.items() if gn_id in layer_stack}
|
self.sexp.render(layer_stack, layer_map=None,
|
||||||
|
|
||||||
self.sexp.render(layer_stack, layer_map,
|
|
||||||
x=x, y=y, rotation=rotation,
|
x=x, y=y, rotation=rotation,
|
||||||
side=self.side,
|
flip=flip,
|
||||||
text=(not self.hide_text),
|
text=(not self.hide_text),
|
||||||
variables=variables, cache=cache)
|
variables=variables, cache=cache)
|
||||||
|
|
||||||
def bounding_box(self, unit=MM):
|
def bounding_box(self, unit=MM):
|
||||||
return offset_bounds(self.sexp.bounding_box(unit), unit(self.x, self.unit), unit(self.y, self.unit))
|
return offset_bounds(self.sexp.bounding_box(unit), unit(self.x, self.unit), unit(self.y, self.unit))
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
import sys
|
import sys
|
||||||
from ...layers import LayerStack
|
from ...layers import LayerStack
|
||||||
462
src/gerbonara/cad/kicad/graphical_primitives.py
Normal file
|
|
@ -0,0 +1,462 @@
|
||||||
|
|
||||||
|
import string
|
||||||
|
import math
|
||||||
|
import base64
|
||||||
|
import textwrap
|
||||||
|
|
||||||
|
from .sexp import *
|
||||||
|
from .base_types import *
|
||||||
|
from .primitives import *
|
||||||
|
|
||||||
|
from ... import graphic_objects as go
|
||||||
|
from ... import apertures as ap
|
||||||
|
from ...newstroke import Newstroke
|
||||||
|
from ...utils import rotate_point, MM, arc_bounds
|
||||||
|
|
||||||
|
@sexp_type('layer')
|
||||||
|
class TextLayer:
|
||||||
|
layer: str = ''
|
||||||
|
knockout: Flag() = False
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('gr_text')
|
||||||
|
class Text(TextMixin, BBoxMixin):
|
||||||
|
locked: Flag() = False
|
||||||
|
text: str = ''
|
||||||
|
at: AtPos = field(default_factory=AtPos)
|
||||||
|
layer: TextLayer = field(default_factory=TextLayer)
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
tstamp: Timestamp = None
|
||||||
|
effects: TextEffect = field(default_factory=TextEffect)
|
||||||
|
render_cache: RenderCache = None
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.at = self.at.with_offset(x, y)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('gr_text_box')
|
||||||
|
class TextBox(BBoxMixin):
|
||||||
|
locked: Flag() = False
|
||||||
|
text: str = ''
|
||||||
|
start: Named(XYCoord) = None
|
||||||
|
end: Named(XYCoord) = None
|
||||||
|
margins: Margins = None
|
||||||
|
pts: PointList = field(default_factory=list)
|
||||||
|
angle: OmitDefault(Named(float)) = 0.0
|
||||||
|
layer: Named(str) = ""
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
tstamp: Timestamp = None
|
||||||
|
effects: TextEffect = field(default_factory=TextEffect)
|
||||||
|
border: Named(YesNoAtom()) = False
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
render_cache: RenderCache = None
|
||||||
|
|
||||||
|
def render(self, variables={}):
|
||||||
|
text = string.Template(self.text).safe_substitute(variables)
|
||||||
|
if text != self.text:
|
||||||
|
raise ValueError('Rendering of vector font text with variables not yet supported')
|
||||||
|
|
||||||
|
if not render_cache or not render_cache.polygons:
|
||||||
|
raise ValueError('Vector font text with empty render cache')
|
||||||
|
|
||||||
|
for poly in render_cache.polygons:
|
||||||
|
reg = go.Region([(p.x, -p.y) for p in poly.pts], unit=MM)
|
||||||
|
|
||||||
|
if self.stroke:
|
||||||
|
if self.stroke.type not in (None, Atom.default, Atom.solid):
|
||||||
|
raise ValueError('Dashed strokes are not supported on vector text')
|
||||||
|
|
||||||
|
yield from reg.outline_objects(aperture=ap.CircleAperture(self.stroke.width, unit=MM))
|
||||||
|
|
||||||
|
yield reg
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.start = self.start.with_offset(x, y)
|
||||||
|
self.end = self.end.with_offset(x, y)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('gr_line')
|
||||||
|
class Line(WidthMixin):
|
||||||
|
locked: Flag() = False
|
||||||
|
start: Rename(XYCoord) = None
|
||||||
|
end: Rename(XYCoord) = None
|
||||||
|
angle: Named(float) = None # wat
|
||||||
|
layer: Named(str) = None
|
||||||
|
width: Named(float) = None
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
tstamp: Timestamp = None
|
||||||
|
|
||||||
|
def rotate(self, angle, cx=None, cy=None):
|
||||||
|
self.start.x, self.start.y = rotate_point(self.start.x, self.start.y, angle, cx, cy)
|
||||||
|
self.end.x, self.end.y = rotate_point(self.end.x, self.end.y, angle, cx, cy)
|
||||||
|
|
||||||
|
def render(self, variables=None):
|
||||||
|
if self.angle:
|
||||||
|
raise NotImplementedError('Angles on lines are not implemented. Please raise an issue and provide an example file.')
|
||||||
|
|
||||||
|
dasher = Dasher(self)
|
||||||
|
dasher.move(self.start.x, self.start.y)
|
||||||
|
dasher.line(self.end.x, self.end.y)
|
||||||
|
|
||||||
|
for x1, y1, x2, y2 in dasher:
|
||||||
|
yield go.Line(x1, -y1, x2, -y2, aperture=ap.CircleAperture(dasher.width, unit=MM), unit=MM)
|
||||||
|
# FIXME render all primitives using dasher, maybe share code w/ fp_ prefix primitives
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.start = self.start.with_offset(x, y)
|
||||||
|
self.end = self.end.with_offset(x, y)
|
||||||
|
|
||||||
|
def bounding_box(self, unit=MM):
|
||||||
|
x_min, x_max = min(self.start.x, self.end.x), max(self.start.x, self.end.x)
|
||||||
|
y_min, y_max = min(self.start.y, self.end.y), max(self.start.y, self.end.y)
|
||||||
|
w = self.stroke.width if self.stroke else self.width
|
||||||
|
return (x_min-w, y_max-w), (x_max+w, y_max+w)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('target')
|
||||||
|
class Target(WidthMixin):
|
||||||
|
shape: AtomChoice(Atom.x, Atom.plus) = 'plus'
|
||||||
|
at: AtPos = field(default_factory=AtPos)
|
||||||
|
size: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
width: Named(float) = None
|
||||||
|
layer: Named(str) = None
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
tstamp: Timestamp = None
|
||||||
|
|
||||||
|
def render(self, variables=None):
|
||||||
|
raise NotImplementedError('Target objects are not implemented yet')
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('fill')
|
||||||
|
class FillMode:
|
||||||
|
# Needed for compatibility with weird files
|
||||||
|
fill: AtomChoice(Atom.solid, Atom.yes, Atom.no, Atom.none) = False
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def __map__(kls, obj, parent=None, path=''):
|
||||||
|
return obj[1] in (Atom.solid, Atom.yes)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def __sexp__(kls, value):
|
||||||
|
yield [Atom.fill, Atom.solid if value else Atom.none]
|
||||||
|
|
||||||
|
@sexp_type('gr_rect')
|
||||||
|
class Rectangle(BBoxMixin, WidthMixin):
|
||||||
|
locked: Flag() = False
|
||||||
|
start: Rename(XYCoord) = None
|
||||||
|
end: Rename(XYCoord) = None
|
||||||
|
layer: Named(str) = None
|
||||||
|
width: Named(float) = None
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
fill: FillMode = False
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
tstamp: Timestamp = None
|
||||||
|
|
||||||
|
def render(self, variables=None):
|
||||||
|
rect = go.Region.from_rectangle(self.start.x, -self.start.y,
|
||||||
|
self.end.x-self.start.x, -(self.end.y-self.start.y),
|
||||||
|
unit=MM)
|
||||||
|
|
||||||
|
if self.fill:
|
||||||
|
yield rect
|
||||||
|
|
||||||
|
if (w := self.stroke.width if self.stroke else self.width):
|
||||||
|
# FIXME stroke support
|
||||||
|
yield from rect.outline_objects(aperture=ap.CircleAperture(w, unit=MM))
|
||||||
|
|
||||||
|
@property
|
||||||
|
def top_left(self):
|
||||||
|
return ((min(self.start.x, self.end.x), min(self.start.y, self.end.y)),
|
||||||
|
(max(self.start.x, self.end.x), max(self.start.y, self.end.y)))
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.start = self.start.with_offset(x, y)
|
||||||
|
self.end = self.end.with_offset(x, y)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('gr_circle')
|
||||||
|
class Circle(BBoxMixin, WidthMixin):
|
||||||
|
locked: Flag() = False
|
||||||
|
center: Rename(XYCoord) = None
|
||||||
|
end: Rename(XYCoord) = None
|
||||||
|
layer: Named(str) = None
|
||||||
|
width: Named(float) = None
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
fill: FillMode = False
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
tstamp: Timestamp = None
|
||||||
|
|
||||||
|
def render(self, variables=None):
|
||||||
|
r = math.dist((self.center.x, -self.center.y), (self.end.x, -self.end.y))
|
||||||
|
w = self.stroke.width if self.stroke else self.width
|
||||||
|
aperture = ap.CircleAperture(w or 0, unit=MM)
|
||||||
|
arc = go.Arc.from_circle(self.center.x, -self.center.y, r, aperture=aperture, unit=MM)
|
||||||
|
|
||||||
|
if w:
|
||||||
|
# FIXME stroke support
|
||||||
|
yield arc
|
||||||
|
|
||||||
|
if self.fill:
|
||||||
|
yield arc.to_region()
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.center = self.center.with_offset(x, y)
|
||||||
|
self.end = self.end.with_offset(x, y)
|
||||||
|
|
||||||
|
def rotate(self, angle, cx=0, cy=0):
|
||||||
|
self.center = self.center.with_rotation(angle, cx, cy)
|
||||||
|
self.end = self.end.with_rotation(angle, cx, cy)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('gr_arc')
|
||||||
|
class Arc(WidthMixin, BBoxMixin):
|
||||||
|
locked: Flag() = False
|
||||||
|
start: Rename(XYCoord) = None
|
||||||
|
mid: Rename(XYCoord) = None
|
||||||
|
end: Rename(XYCoord) = None
|
||||||
|
angle: Named(float) = None
|
||||||
|
layer: Named(str) = None
|
||||||
|
width: Named(float) = None
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
tstamp: Timestamp = None
|
||||||
|
_: SEXP_END = None
|
||||||
|
center: XYCoord = None
|
||||||
|
|
||||||
|
def __post_init__(self):
|
||||||
|
self.start = XYCoord(self.start)
|
||||||
|
self.end = XYCoord(self.end)
|
||||||
|
if self.mid or self.center is None:
|
||||||
|
self.mid = XYCoord(self.mid)
|
||||||
|
elif self.center:
|
||||||
|
self.mid = center_arc_to_kicad_mid(XYCoord(self.center), self.start, self.end)
|
||||||
|
self.center = None
|
||||||
|
|
||||||
|
def render(self, variables=None):
|
||||||
|
if not (w := self.stroke.width if self.stroke else self.width):
|
||||||
|
return
|
||||||
|
|
||||||
|
aperture = ap.CircleAperture(w, unit=MM)
|
||||||
|
x1, y1 = self.start.x, self.start.y
|
||||||
|
x2, y2 = self.end.x, self.end.y
|
||||||
|
(cx, cy), _r, clockwise = kicad_mid_to_center_arc(self.mid, self.start, self.end)
|
||||||
|
yield go.Arc(x1, -y1, x2, -y2, cx-x1, -(cy-y1), aperture=aperture, clockwise=not clockwise, unit=MM)
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.start = self.start.with_offset(x, y)
|
||||||
|
self.mid = self.mid.with_offset(x, y)
|
||||||
|
self.end = self.end.with_offset(x, y)
|
||||||
|
|
||||||
|
def rotate(self, angle, cx=None, cy=None):
|
||||||
|
self.start.x, self.start.y = rotate_point(self.start.x, self.start.y, angle, cx, cy)
|
||||||
|
self.mid.x, self.mid.y = rotate_point(self.mid.x, self.mid.y, angle, cx, cy)
|
||||||
|
self.end.x, self.end.y = rotate_point(self.end.x, self.end.y, angle, cx, cy)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('gr_poly')
|
||||||
|
class Polygon(BBoxMixin, WidthMixin):
|
||||||
|
pts: ArcPointList = field(default_factory=list)
|
||||||
|
layer: Named(str) = None
|
||||||
|
width: Named(float) = None
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
fill: FillMode = True
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
tstamp: Timestamp = None
|
||||||
|
|
||||||
|
def render(self, variables=None):
|
||||||
|
points = []
|
||||||
|
centers = []
|
||||||
|
for point_or_arc in self.pts:
|
||||||
|
if points:
|
||||||
|
centers.append((None, (None, None)))
|
||||||
|
|
||||||
|
if isinstance(point_or_arc, XYCoord):
|
||||||
|
points.append((point_or_arc.x, -point_or_arc.y))
|
||||||
|
|
||||||
|
else: # base_types.Arc
|
||||||
|
points.append((point_or_arc.start.x, -point_or_arc.start.y))
|
||||||
|
points.append((point_or_arc.end.x, -point_or_arc.end.y))
|
||||||
|
(cx, cy), _r, clockwise = kicad_mid_to_center_arc(point_or_arc.mid, point_or_arc.start, point_or_arc.end)
|
||||||
|
centers.append((not clockwise, (cx, -cy)))
|
||||||
|
|
||||||
|
reg = go.Region(points, centers, unit=MM)
|
||||||
|
reg.close()
|
||||||
|
|
||||||
|
w = self.stroke.width if self.stroke else self.width
|
||||||
|
# FIXME stroke support
|
||||||
|
if w and w >= 0.005:
|
||||||
|
yield from reg.outline_objects(aperture=ap.CircleAperture(w, unit=MM))
|
||||||
|
|
||||||
|
if self.fill:
|
||||||
|
yield reg
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.pts = [pt.with_offset(x, y) for pt in self.pts]
|
||||||
|
|
||||||
|
def rotate(self, angle, cx=0, cy=0):
|
||||||
|
self.pts = [pt.with_rotation(angle, cx, cy) for pt in self.pts]
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('gr_curve')
|
||||||
|
class Curve(BBoxMixin, WidthMixin):
|
||||||
|
locked: Flag() = False
|
||||||
|
pts: PointList = field(default_factory=list)
|
||||||
|
layer: Named(str) = None
|
||||||
|
width: Named(float) = None
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
tstamp: Timestamp = None
|
||||||
|
|
||||||
|
def render(self, variables=None):
|
||||||
|
raise NotImplementedError('Bezier rendering is not yet supported. Please raise an issue and provide an example file.')
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.pts =[pt.with_offset(x, y) for pt in self.pts]
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('gr_bbox')
|
||||||
|
class AnnotationBBox:
|
||||||
|
start: Rename(XYCoord) = None
|
||||||
|
end: Rename(XYCoord) = None
|
||||||
|
width: Named(float) = None
|
||||||
|
fill: FillMode = False
|
||||||
|
|
||||||
|
def render(self, variables=None):
|
||||||
|
return []
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.start = self.start.with_offset(x, y)
|
||||||
|
self.end = self.end.with_offset(x, y)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('format')
|
||||||
|
class DimensionFormat:
|
||||||
|
prefix: Named(str) = None
|
||||||
|
suffix: Named(str) = None
|
||||||
|
units: Named(int) = 2
|
||||||
|
units_format: Named(int) = 1
|
||||||
|
precision: Named(int) = 7
|
||||||
|
override_value: Named(str) = None
|
||||||
|
suppress_zeros: Flag() = False
|
||||||
|
suppress_zeroes: Flag() = False
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('style')
|
||||||
|
class DimensionStyle:
|
||||||
|
thickness: Named(float) = 0.1
|
||||||
|
arrow_length: Named(float) = 1.27
|
||||||
|
text_position_mode: Named(int) = 0
|
||||||
|
arrow_direction: Named(AtomChoice(Atom.inward, Atom.outward)) = None
|
||||||
|
extension_height: Named(float) = None
|
||||||
|
text_frame: Named(float) = None
|
||||||
|
extension_offset: Named(float) = None
|
||||||
|
keep_text_aligned: Flag() = False
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('data')
|
||||||
|
class Base64Blob:
|
||||||
|
@classmethod
|
||||||
|
def __map__(kls, obj, parent=None, path=''):
|
||||||
|
_data, *content = obj
|
||||||
|
for x in content[:10]:
|
||||||
|
print(str(x))
|
||||||
|
return base64.b64decode(''.join(map(str, content)))
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def __sexp__(kls, value):
|
||||||
|
encoded = base64.b64encode(value).decode()
|
||||||
|
yield [Atom.data, *textwrap.wrap(encoded, 76)]
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('image')
|
||||||
|
class Image:
|
||||||
|
at: AtPos = field(default_factory=AtPos)
|
||||||
|
scale: Named(float) = None
|
||||||
|
layer: Named(str) = None
|
||||||
|
locked: Flag() = False
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
data: Base64Blob = ''
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.at = self.at.with_offset(x, y)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('dimension')
|
||||||
|
class Dimension:
|
||||||
|
value: float = None
|
||||||
|
locked: Flag() = False
|
||||||
|
dimension_type: Named(AtomChoice(Atom.aligned, Atom.leader, Atom.center, Atom.orthogonal, Atom.radial), name='type') = Atom.aligned
|
||||||
|
layer: Named(str) = 'Dwgs.User'
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
tstamp: Timestamp = field(default_factory=Timestamp)
|
||||||
|
pts: PointList = field(default_factory=list)
|
||||||
|
height: Named(float) = None
|
||||||
|
width: Named(float) = None
|
||||||
|
orientation: Named(int) = None
|
||||||
|
leader_length: Named(float) = None
|
||||||
|
gr_text: Text = None
|
||||||
|
dimension_format: OmitDefault(DimensionFormat) = field(default_factory=DimensionFormat)
|
||||||
|
dimension_style: OmitDefault(DimensionStyle) = field(default_factory=DimensionStyle)
|
||||||
|
|
||||||
|
def render(self, variables=None):
|
||||||
|
raise NotImplementedError('Dimension rendering is not yet supported. Please raise an issue.')
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.pts = [pt.with_offset(x, y) for pt in self.pts]
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('options')
|
||||||
|
class PadStackLayerOptions:
|
||||||
|
anchor: AtomChoice(Atom.rect, Atom.circle) = Atom.circle
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('primitives')
|
||||||
|
class PadStackPrimitives:
|
||||||
|
vectors: Rename(Line, name='gr_vector') = field(default_factory=list)
|
||||||
|
lines: List(Line) = field(default_factory=list)
|
||||||
|
bboxes: List(AnnotationBBox) = field(default_factory=list)
|
||||||
|
arcs: List(Arc) = field(default_factory=list)
|
||||||
|
circles: List(Circle) = field(default_factory=list)
|
||||||
|
curves: List(Curve) = field(default_factory=list)
|
||||||
|
polygons:List(Polygon) = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('layer')
|
||||||
|
class PadStackLayer:
|
||||||
|
layer: str = ''
|
||||||
|
shape: Named(AtomChoice(Atom.circle, Atom.rect, Atom.oval, Atom.trapezoid, Atom.roundrect, Atom.custom)) = Atom.circle
|
||||||
|
size: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
rect_delta: Rename(XYCoord) = None
|
||||||
|
offset: Rename(XYCoord) = None
|
||||||
|
roundrect_rratio: Named(float) = None
|
||||||
|
chamfer_ratio: Named(float) = None
|
||||||
|
chamfer: Chamfer = None
|
||||||
|
primitives: PadStackPrimitives = None
|
||||||
|
options: PadStackLayerOptions = None
|
||||||
|
thermal_bridge_angle: Named(float) = None
|
||||||
|
thermal_gap: Named(float) = None
|
||||||
|
thermal_bridge_width: Named(float) = None
|
||||||
|
clearance: Named(float) = None
|
||||||
|
zone_connect: Named(int) = None
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('padstack')
|
||||||
|
class PadStack:
|
||||||
|
mode: Named(AtomChoice('front_inner_back', 'custom')) = Atom.front_inner_back
|
||||||
|
layers: List(PadStackLayer) = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('teardrops')
|
||||||
|
class TeardropSpec:
|
||||||
|
best_length_ratio: Named(float) = 1.0
|
||||||
|
max_length: Named(float) = 2.0
|
||||||
|
best_width_ratio: Named(float) = 1.0
|
||||||
|
max_width: Named(float) = 2.0
|
||||||
|
curve_points: Named(int) = 0
|
||||||
|
filter_ratio: Named(float) = 0.9
|
||||||
|
enabled: Named(YesNoAtom()) = True
|
||||||
|
allow_two_segments: Named(YesNoAtom()) = True
|
||||||
|
prefer_zone_connections: Named(YesNoAtom()) = True
|
||||||
|
|
||||||
|
|
||||||
848
src/gerbonara/cad/kicad/pcb.py
Normal file
|
|
@ -0,0 +1,848 @@
|
||||||
|
"""
|
||||||
|
Library for handling KiCad's PCB files (`*.kicad_mod`).
|
||||||
|
"""
|
||||||
|
|
||||||
|
import math
|
||||||
|
from pathlib import Path
|
||||||
|
from dataclasses import field, KW_ONLY, fields
|
||||||
|
from itertools import chain
|
||||||
|
import re
|
||||||
|
import fnmatch
|
||||||
|
import functools
|
||||||
|
|
||||||
|
from .sexp import *
|
||||||
|
from .base_types import *
|
||||||
|
from .primitives import *
|
||||||
|
from .footprints import Footprint, Pad
|
||||||
|
from . import graphical_primitives as gr
|
||||||
|
import rtree.index
|
||||||
|
|
||||||
|
from .. import primitives as cad_pr
|
||||||
|
|
||||||
|
from ... import graphic_primitives as gp
|
||||||
|
from ... import graphic_objects as go
|
||||||
|
from ... import apertures as ap
|
||||||
|
from ...layers import LayerStack
|
||||||
|
from ...newstroke import Newstroke
|
||||||
|
from ...utils import MM, rotate_point
|
||||||
|
|
||||||
|
|
||||||
|
def match_filter(f, value):
|
||||||
|
if isinstance(f, str) and re.fullmatch(f, value):
|
||||||
|
return True
|
||||||
|
return value in f
|
||||||
|
|
||||||
|
def gn_side_to_kicad(side, layer='Cu'):
|
||||||
|
if side == 'top':
|
||||||
|
return f'F.{layer}'
|
||||||
|
elif side == 'bottom':
|
||||||
|
return f'B.{layer}'
|
||||||
|
elif side.startswith('inner'):
|
||||||
|
return f'In{int(side[5:])}.{layer}'
|
||||||
|
else:
|
||||||
|
raise ValueError(f'Cannot parse gerbonara side name "{side}"')
|
||||||
|
|
||||||
|
def gn_layer_to_kicad(layer, flip=False):
|
||||||
|
side = 'B' if flip else 'F'
|
||||||
|
if layer == 'silk':
|
||||||
|
return f'{side}.SilkS'
|
||||||
|
elif layer == 'mask':
|
||||||
|
return f'{side}.Mask'
|
||||||
|
elif layer == 'paste':
|
||||||
|
return f'{side}.Paste'
|
||||||
|
elif layer == 'copper':
|
||||||
|
return f'{side}.Cu'
|
||||||
|
else:
|
||||||
|
raise ValueError('Cannot translate gerbonara layer name "{layer}" to KiCad')
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('general')
|
||||||
|
class GeneralSection:
|
||||||
|
thickness: Named(float) = 1.60
|
||||||
|
legacy_teardrops: Named(YesNoAtom()) = False
|
||||||
|
drawings: Named(int) = None
|
||||||
|
tracks: Named(int) = None
|
||||||
|
zones: Named(int) = None
|
||||||
|
modules: Named(int) = None
|
||||||
|
nets: Named(int) = None
|
||||||
|
links: Named(int) = None
|
||||||
|
no_connects: Named(int) = None
|
||||||
|
area: Named(Array(float)) = None
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('layers')
|
||||||
|
class LayerSettings:
|
||||||
|
index: int = 0
|
||||||
|
canonical_name: str = None
|
||||||
|
layer_type: AtomChoice(Atom.jumper, Atom.mixed, Atom.power, Atom.signal, Atom.user, Atom.auxiliary) = Atom.signal
|
||||||
|
custom_name: str = None
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('layer')
|
||||||
|
class LayerStackupSettings:
|
||||||
|
dielectric: Flag() = False
|
||||||
|
name: str = None
|
||||||
|
index: int = None
|
||||||
|
layer_type: Named(str, name='type') = ''
|
||||||
|
color: Color = None
|
||||||
|
thickness: Named(float) = None
|
||||||
|
material: Named(str) = None
|
||||||
|
epsilon_r: Named(float) = None
|
||||||
|
loss_tangent: Named(float) = None
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('stackup')
|
||||||
|
class StackupSettings:
|
||||||
|
layers: List(LayerStackupSettings) = field(default_factory=list)
|
||||||
|
copper_finish: Named(str) = None
|
||||||
|
dielectric_constraints: Named(YesNoAtom()) = None
|
||||||
|
edge_connector: Named(AtomChoice(Atom.yes, Atom.bevelled)) = None
|
||||||
|
castellated_pads: Named(YesNoAtom()) = None
|
||||||
|
edge_plating: Named(YesNoAtom()) = None
|
||||||
|
|
||||||
|
@sexp_type('setup')
|
||||||
|
class BoardSetup:
|
||||||
|
@classmethod
|
||||||
|
def __map__(kls, obj, parent=None, path=''):
|
||||||
|
return obj
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def __sexp__(kls, value):
|
||||||
|
yield value
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('segment')
|
||||||
|
class TrackSegment(BBoxMixin):
|
||||||
|
start: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
end: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
width: Named(float) = 0.5
|
||||||
|
locked: Flag() = False
|
||||||
|
layer: Named(str) = 'F.Cu'
|
||||||
|
extra_layers: Named(Array(str), name='layers') = field(default_factory=list)
|
||||||
|
solder_mask_margin: Named(float) = None
|
||||||
|
net: Named(int) = 0
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
tstamp: Timestamp = None
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def from_footprint_line(kls, line, flip=False):
|
||||||
|
# FIXME flip
|
||||||
|
return kls(line.start, line.end, line.width or line.stroke.width, line.layer, line.locked, tstamp=line.tstamp)
|
||||||
|
|
||||||
|
def __post_init__(self):
|
||||||
|
self.start = XYCoord(self.start)
|
||||||
|
self.end = XYCoord(self.end)
|
||||||
|
|
||||||
|
def __after_parse__(self, parent):
|
||||||
|
if self.extra_layers:
|
||||||
|
self.layer, *self.extra_layers = self.extra_layers
|
||||||
|
|
||||||
|
def __before_sexp__(self):
|
||||||
|
if self.extra_layers:
|
||||||
|
self.extra_layers.insert(0, self.layer)
|
||||||
|
self.layer = None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def layer_mask(self):
|
||||||
|
return layer_mask([self.layer])
|
||||||
|
|
||||||
|
def render(self, variables=None, cache=None):
|
||||||
|
if not self.width:
|
||||||
|
return
|
||||||
|
|
||||||
|
aperture = ap.CircleAperture(self.width, unit=MM)
|
||||||
|
yield go.Line(self.start.x, -self.start.y, self.end.x, -self.end.y, aperture=aperture, unit=MM)
|
||||||
|
|
||||||
|
def rotate(self, angle, cx=None, cy=None):
|
||||||
|
if cx is None or cy is None:
|
||||||
|
cx, cy = self.start.x, self.start.y
|
||||||
|
|
||||||
|
self.start.x, self.start.y = rotate_point(self.start.x, self.start.y, angle, cx, cy)
|
||||||
|
self.end.x, self.end.y = rotate_point(self.end.x, self.end.y, angle, cx, cy)
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.start = self.start.with_offset(x, y)
|
||||||
|
self.end = self.end.with_offset(x, y)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('arc')
|
||||||
|
class TrackArc(BBoxMixin):
|
||||||
|
start: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
mid: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
end: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
width: Named(float) = 0.5
|
||||||
|
layer: Named(str) = 'F.Cu'
|
||||||
|
locked: Flag() = False
|
||||||
|
net: Named(int) = 0
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
tstamp: Timestamp = None
|
||||||
|
_: SEXP_END = None
|
||||||
|
center: XYCoord = None
|
||||||
|
|
||||||
|
def __post_init__(self):
|
||||||
|
self.start = XYCoord(self.start)
|
||||||
|
self.end = XYCoord(self.end)
|
||||||
|
self.mid = XYCoord(self.mid) if self.center is None else center_arc_to_kicad_mid(XYCoord(self.center), self.start, self.end)
|
||||||
|
self.center = None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def layer_mask(self):
|
||||||
|
return layer_mask([self.layer])
|
||||||
|
|
||||||
|
def render(self, variables=None, cache=None):
|
||||||
|
if not self.width:
|
||||||
|
return
|
||||||
|
|
||||||
|
aperture = ap.CircleAperture(self.width, unit=MM)
|
||||||
|
cx, cy = self.mid.x, self.mid.y
|
||||||
|
x1, y1 = self.start.x, self.start.y
|
||||||
|
x2, y2 = self.end.x, self.end.y
|
||||||
|
yield go.Arc(x1, -y1, x2, -y2, cx-x1, -(cy-y1), aperture=aperture, clockwise=True, unit=MM)
|
||||||
|
|
||||||
|
def rotate(self, angle, cx=None, cy=None):
|
||||||
|
self.start.x, self.start.y = rotate_point(self.start.x, self.start.y, angle, cx, cy)
|
||||||
|
self.mid.x, self.mid.y = rotate_point(self.mid.x, self.mid.y, angle, cx, cy)
|
||||||
|
self.end.x, self.end.y = rotate_point(self.end.x, self.end.y, angle, cx, cy)
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.start = self.start.with_offset(x, y)
|
||||||
|
self.mid = self.mid.with_offset(x, y)
|
||||||
|
self.end = self.end.with_offset(x, y)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('tenting')
|
||||||
|
class Tenting:
|
||||||
|
front: Flag() = False
|
||||||
|
back: Flag() = False
|
||||||
|
none: Flag() = False
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('via')
|
||||||
|
class Via(BBoxMixin):
|
||||||
|
via_type: AtomChoice(Atom.blind, Atom.micro) = None
|
||||||
|
locked: Flag() = False
|
||||||
|
at: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
size: Named(float) = 0.8
|
||||||
|
drill: Named(float) = 0.4
|
||||||
|
layers: Named(Array(str)) = field(default_factory=lambda: ['F.Cu', 'B.Cu'])
|
||||||
|
teardrops: gr.TeardropSpec = None
|
||||||
|
tenting: Tenting = None
|
||||||
|
padstack: gr.PadStack = None
|
||||||
|
remove_unused_layers: Flag() = False
|
||||||
|
keep_end_layers: Flag() = False
|
||||||
|
free: Named(YesNoAtom()) = False
|
||||||
|
zone_layer_connections: Named(Array(str)) = field(default_factory=list)
|
||||||
|
net: Named(int) = 0
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
tstamp: Timestamp = None
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def from_pad(kls, pad):
|
||||||
|
if pad.type != Atom.thru_hole or pad.shape != Atom.circle:
|
||||||
|
raise ValueError('Can only convert circular through-hole pads to vias.')
|
||||||
|
|
||||||
|
if pad.drill and (pad.drill.oval or pad.drill.offset):
|
||||||
|
raise ValueError('Can only convert pads with centered, circular drills to vias.')
|
||||||
|
|
||||||
|
x, y, rot, _flip = pad.abs_pos
|
||||||
|
return kls(locked=pad.locked,
|
||||||
|
at=XYCoord(x, y),
|
||||||
|
size=max(pad.size.x, pad.size.y),
|
||||||
|
drill=pad.drill.diameter if pad.drill else 0,
|
||||||
|
layers=[l for l in pad.layers if l.endswith('.Cu')],
|
||||||
|
free=True,
|
||||||
|
net=pad.net.number if pad.net else 0,
|
||||||
|
tstamp=pad.tstamp)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def abs_pos(self):
|
||||||
|
return self.at.x, self.at.y, 0, False
|
||||||
|
|
||||||
|
@property
|
||||||
|
def layer_mask(self):
|
||||||
|
return layer_mask(self.layers)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def width(self):
|
||||||
|
return self.size
|
||||||
|
|
||||||
|
def __post_init__(self):
|
||||||
|
self.at = XYCoord(self.at)
|
||||||
|
|
||||||
|
def render_drill(self):
|
||||||
|
aperture = ap.ExcellonTool(self.drill, plated=True, unit=MM)
|
||||||
|
yield go.Flash(self.at.x, -self.at.y, aperture=aperture, unit=MM)
|
||||||
|
|
||||||
|
def render(self, variables=None, cache=None):
|
||||||
|
aperture = ap.CircleAperture(self.size, unit=MM)
|
||||||
|
yield go.Flash(self.at.x, -self.at.y, aperture, unit=MM)
|
||||||
|
|
||||||
|
def rotate(self, angle, cx=None, cy=None):
|
||||||
|
if cx is None or cy is None:
|
||||||
|
return
|
||||||
|
|
||||||
|
self.at.x, self.at.y = rotate_point(self.at.x, self.at.y, angle, cx, cy)
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.at = self.at.with_offset(x, y)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('net_class')
|
||||||
|
class LegacyNetclass:
|
||||||
|
name: str = ''
|
||||||
|
description: str = ''
|
||||||
|
clearance: Named(float) = None
|
||||||
|
trace_width: Named(float) = None
|
||||||
|
via_dia: Named(float) = None
|
||||||
|
via_drill: Named(float) = None
|
||||||
|
uvia_dia: Named(float) = None
|
||||||
|
uvia_drill: Named(float) = None
|
||||||
|
diff_pair_width: Named(float) = None
|
||||||
|
diff_pair_gap: Named(float) = None
|
||||||
|
nets: Rename(List(Named(str)), name='add_net') = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('generated')
|
||||||
|
class GeneratedPatterns:
|
||||||
|
type: Named(Atom) = ''
|
||||||
|
name: Named(str) = ''
|
||||||
|
layer: Named(str) = ''
|
||||||
|
locked: Flag() = False
|
||||||
|
members: Named(Array(Atom), name='members') = field(default_factory=list)
|
||||||
|
_ : SEXP_END = None
|
||||||
|
params: dict = field(default_factory=dict)
|
||||||
|
|
||||||
|
def __catchall__(self, sexp_value, path=''):
|
||||||
|
key, value = sexp_value
|
||||||
|
self.params[key] = value
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def __sexp__(kls, value):
|
||||||
|
return [kls.name_atom,
|
||||||
|
['type', value.type],
|
||||||
|
['name', value.name],
|
||||||
|
['layer', value.layer],
|
||||||
|
['locked', ('true' if value.locked else 'false')],
|
||||||
|
*[[k, v] for k, v in value.params.items()],
|
||||||
|
['members', *value.members]]
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
SUPPORTED_FILE_FORMAT_VERSIONS = [20200119, 20200512, 20210108, 20211014, 20220621, 20221018, 20230517, 20240706, 20240922, 20241229]
|
||||||
|
@sexp_type('kicad_pcb')
|
||||||
|
class Board:
|
||||||
|
_version: Named(int, name='version') = 20230517
|
||||||
|
generator: Named(str) = Atom.gerbonara
|
||||||
|
generator_version: Named(str) = Atom.gerbonara
|
||||||
|
legacy_generator: Named(Array(str), name='host') = None
|
||||||
|
general: GeneralSection = None
|
||||||
|
paper: PageSettings = None
|
||||||
|
legacy_page: Rename(PageSettings, 'page') = None
|
||||||
|
title_block: TitleBlock = None
|
||||||
|
layers: Named(Array(Untagged(LayerSettings))) = field(default_factory=list)
|
||||||
|
setup: BoardSetup = field(default_factory=BoardSetup)
|
||||||
|
properties: List(Property) = field(default_factory=list)
|
||||||
|
nets: List(Net) = field(default_factory=list)
|
||||||
|
legacy_netclasses: List(LegacyNetclass) = field(default_factory=list)
|
||||||
|
footprints: List(Footprint) = field(default_factory=list)
|
||||||
|
legacy_footprints: Rename(List(Footprint), 'module') = field(default_factory=list)
|
||||||
|
# Graphical elements
|
||||||
|
texts: List(gr.Text) = field(default_factory=list)
|
||||||
|
text_boxes: List(gr.TextBox) = field(default_factory=list)
|
||||||
|
lines: List(gr.Line) = field(default_factory=list)
|
||||||
|
targets: List(gr.Target) = field(default_factory=list)
|
||||||
|
rectangles: List(gr.Rectangle) = field(default_factory=list)
|
||||||
|
circles: List(gr.Circle) = field(default_factory=list)
|
||||||
|
arcs: List(gr.Arc) = field(default_factory=list)
|
||||||
|
polygons: List(gr.Polygon) = field(default_factory=list)
|
||||||
|
curves: List(gr.Curve) = field(default_factory=list)
|
||||||
|
dimensions: List(gr.Dimension) = field(default_factory=list)
|
||||||
|
images: List(gr.Image) = field(default_factory=list)
|
||||||
|
# Tracks
|
||||||
|
track_segments: List(TrackSegment) = field(default_factory=list)
|
||||||
|
track_arcs: List(TrackArc) = field(default_factory=list)
|
||||||
|
vias: List(Via) = field(default_factory=list)
|
||||||
|
# Other stuff
|
||||||
|
zones: List(Zone) = field(default_factory=list)
|
||||||
|
groups: List(Group) = field(default_factory=list)
|
||||||
|
generated_patterns: List(GeneratedPatterns) = field(default_factory=list)
|
||||||
|
embedded_fonts: Named(YesNoAtom()) = False
|
||||||
|
|
||||||
|
_ : SEXP_END = None
|
||||||
|
original_filename: str = None
|
||||||
|
_trace_index: rtree.index.Index = None
|
||||||
|
_trace_index_map: dict = None
|
||||||
|
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def empty_board(kls, inner_layers=0, **kwargs):
|
||||||
|
if 'setup' not in kwargs:
|
||||||
|
kwargs['setup'] = None
|
||||||
|
b = Board(**kwargs)
|
||||||
|
b.init_default_layers(inner_layers)
|
||||||
|
b.__after_parse__(None)
|
||||||
|
return b
|
||||||
|
|
||||||
|
|
||||||
|
def init_default_layers(self, inner_layers=0):
|
||||||
|
inner = [(i, f'In{i}.Cu', 'signal', None) for i in range(1, inner_layers+1)]
|
||||||
|
self.layers = [LayerSettings(idx, name, Atom(ltype)) for idx, name, ltype, cname in [
|
||||||
|
(0, 'F.Cu', 'signal', None),
|
||||||
|
*inner,
|
||||||
|
(31, 'B.Cu', 'signal', None),
|
||||||
|
(32, 'B.Adhes', 'user', 'B.Adhesive'),
|
||||||
|
(33, 'F.Adhes', 'user', 'F.Adhesive'),
|
||||||
|
(34, 'B.Paste', 'user', None),
|
||||||
|
(35, 'F.Paste', 'user', None),
|
||||||
|
(36, 'B.SilkS', 'user', 'B.Silkscreen'),
|
||||||
|
(37, 'F.SilkS', 'user', 'F.Silkscreen'),
|
||||||
|
(38, 'B.Mask', 'user', None),
|
||||||
|
(39, 'F.Mask', 'user', None),
|
||||||
|
(40, 'Dwgs.User', 'user', 'User.Drawings'),
|
||||||
|
(41, 'Cmts.User', 'user', 'User.Comments'),
|
||||||
|
(42, 'Eco1.User', 'user', 'User.Eco1'),
|
||||||
|
(43, 'Eco2.User', 'user', 'User.Eco2'),
|
||||||
|
(44, 'Edge.Cuts', 'user', None),
|
||||||
|
(45, 'Margin', 'user', None),
|
||||||
|
(46, 'B.CrtYd', 'user', 'B.Courtyard'),
|
||||||
|
(47, 'F.CrtYd', 'user', 'F.Courtyard'),
|
||||||
|
(48, 'B.Fab', 'user', None),
|
||||||
|
(49, 'F.Fab', 'user', None),
|
||||||
|
(50, 'User.1', 'auxiliary', None),
|
||||||
|
(51, 'User.2', 'auxiliary', None),
|
||||||
|
(52, 'User.3', 'auxiliary', None),
|
||||||
|
(53, 'User.4', 'auxiliary', None),
|
||||||
|
(54, 'User.5', 'auxiliary', None),
|
||||||
|
(55, 'User.6', 'auxiliary', None),
|
||||||
|
(56, 'User.7', 'auxiliary', None),
|
||||||
|
(57, 'User.8', 'auxiliary', None),
|
||||||
|
(58, 'User.9', 'auxiliary', None)]]
|
||||||
|
|
||||||
|
|
||||||
|
def rebuild_trace_index(self):
|
||||||
|
idx = self._trace_index = rtree.index.Index()
|
||||||
|
id_map = self._trace_index_map = {}
|
||||||
|
for obj in chain(self.track_segments, self.track_arcs):
|
||||||
|
for i, field in enumerate(('start', 'end')):
|
||||||
|
obj_id = id(obj) + i
|
||||||
|
coord = getattr(obj, field)
|
||||||
|
id_map[obj_id] = obj, field, obj.width, obj.layer_mask
|
||||||
|
idx.insert(obj_id, (coord.x, coord.y, coord.x, coord.y))
|
||||||
|
|
||||||
|
for fp in self.footprints:
|
||||||
|
for pad in fp.pads:
|
||||||
|
obj_id = id(pad)
|
||||||
|
id_map[obj_id] = pad, 'at', 0, pad.layer_mask
|
||||||
|
idx.insert(obj_id, (pad.at.x, pad.at.y, pad.at.x, pad.at.y))
|
||||||
|
|
||||||
|
for via in self.vias:
|
||||||
|
obj_id = id(via)
|
||||||
|
id_map[obj_id] = via, 'at', via.size, via.layer_mask
|
||||||
|
idx.insert(obj_id, (via.at.x, via.at.y, via.at.x, via.at.y))
|
||||||
|
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _require_trace_index(fun):
|
||||||
|
@functools.wraps(fun)
|
||||||
|
def wrapper(self, *args, **kwargs):
|
||||||
|
if self._trace_index is None:
|
||||||
|
self.rebuild_trace_index()
|
||||||
|
|
||||||
|
return fun(self, *args, **kwargs)
|
||||||
|
return wrapper
|
||||||
|
|
||||||
|
|
||||||
|
@_require_trace_index
|
||||||
|
def query_trace_index_nearest(self, point, layers='*.Cu', n=1):
|
||||||
|
layers = layer_mask(layers)
|
||||||
|
|
||||||
|
x, y = point
|
||||||
|
for obj_id in self._trace_index.nearest((x, y, x, y), n):
|
||||||
|
entry = obj, attr, size, mask = self._trace_index_map[obj_id]
|
||||||
|
if layers & mask:
|
||||||
|
yield entry
|
||||||
|
|
||||||
|
|
||||||
|
@_require_trace_index
|
||||||
|
def query_trace_index_tolerance(self, point, layers='*.Cu', tol=10e-6):
|
||||||
|
layers = layer_mask(layers)
|
||||||
|
|
||||||
|
x, y = point
|
||||||
|
for obj_id in self._trace_index.intersection((x-tol, y-tol, x+tol, y+tol)):
|
||||||
|
entry = obj, attr, size, mask = self._trace_index_map[obj_id]
|
||||||
|
attr = getattr(obj, attr)
|
||||||
|
if layers & mask and math.dist((attr.x, attr.y), (x, y)) <= tol:
|
||||||
|
yield entry
|
||||||
|
|
||||||
|
|
||||||
|
def find_connected_traces(self, obj, layers='*.Cu', tol=10e-6):
|
||||||
|
search_frontier = []
|
||||||
|
visited = set()
|
||||||
|
def enqueue(obj):
|
||||||
|
visited.add(id(obj))
|
||||||
|
|
||||||
|
if isinstance(obj, (TrackSegment, TrackArc)):
|
||||||
|
search_frontier.append((obj.start, obj.width, obj.layer_mask))
|
||||||
|
search_frontier.append((obj.end, obj.width, obj.layer_mask))
|
||||||
|
|
||||||
|
elif isinstance(obj, Via):
|
||||||
|
search_frontier.append((obj.at, obj.size, obj.layer_mask))
|
||||||
|
|
||||||
|
elif isinstance(obj, Pad):
|
||||||
|
search_frontier.append((obj.at, max(obj.size.x, obj.size.y), obj.layer_mask))
|
||||||
|
|
||||||
|
elif isinstance(obj, (Footprint)):
|
||||||
|
for pad in obj.pads:
|
||||||
|
search_frontier.append((pad.at, max(pad.size.x, pad.size.y), pad.layer_mask))
|
||||||
|
|
||||||
|
else:
|
||||||
|
raise TypeError(f'Finding connected traces for {type(obj)} objects is not (yet) supported.')
|
||||||
|
|
||||||
|
enqueue(obj)
|
||||||
|
yield obj
|
||||||
|
|
||||||
|
filter_layers = layer_mask(layers)
|
||||||
|
while search_frontier:
|
||||||
|
coord, size, layers = search_frontier.pop()
|
||||||
|
x, y = coord.x, coord.y
|
||||||
|
|
||||||
|
# First, find all bounding box intersections
|
||||||
|
found = []
|
||||||
|
for cand, attr, cand_size, cand_mask in self.query_trace_index_tolerance((x, y), layers&filter_layers, size):
|
||||||
|
cand_coord = getattr(cand, attr)
|
||||||
|
dist = math.dist((x, y), (cand_coord.x, cand_coord.y))
|
||||||
|
if dist <= size/2 + cand_size/2 and layers&cand_mask:
|
||||||
|
found.append((dist, cand))
|
||||||
|
|
||||||
|
if not found:
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Second, filter to match only objects that are within tolerance of closest
|
||||||
|
min_dist = min(e[0] for e in found)
|
||||||
|
for dist, cand in found:
|
||||||
|
if dist < min_dist+tol and id(cand) not in visited:
|
||||||
|
enqueue(cand)
|
||||||
|
yield cand
|
||||||
|
|
||||||
|
|
||||||
|
def __after_parse__(self, parent):
|
||||||
|
self.properties = {prop.key: prop.value for prop in self.properties}
|
||||||
|
|
||||||
|
for fp in self.footprints:
|
||||||
|
fp.board = self
|
||||||
|
|
||||||
|
self.nets = {net.index: net.name for net in self.nets}
|
||||||
|
if self.legacy_page:
|
||||||
|
self.paper, self.legacy_page = self.legacy_page, None
|
||||||
|
|
||||||
|
|
||||||
|
def __before_sexp__(self):
|
||||||
|
self.properties = [Property(key, value) for key, value in self.properties.items()]
|
||||||
|
self.nets = [Net(index, name) for index, name in self.nets.items()]
|
||||||
|
|
||||||
|
|
||||||
|
def remove(self, obj):
|
||||||
|
match obj:
|
||||||
|
case gr.Text():
|
||||||
|
self.texts.remove(obj)
|
||||||
|
case gr.TextBox():
|
||||||
|
self.text_boxes.remove(obj)
|
||||||
|
case gr.Line():
|
||||||
|
self.lines.remove(obj)
|
||||||
|
case gr.Rectangle():
|
||||||
|
self.rectangles.remove(obj)
|
||||||
|
case gr.Circle():
|
||||||
|
self.circles.remove(obj)
|
||||||
|
case gr.Arc():
|
||||||
|
self.arcs.remove(obj)
|
||||||
|
case gr.Polygon():
|
||||||
|
self.polygons.remove(obj)
|
||||||
|
case gr.Curve():
|
||||||
|
self.curves.remove(obj)
|
||||||
|
case gr.Dimension():
|
||||||
|
self.dimensions.remove(obj)
|
||||||
|
case gr.Image():
|
||||||
|
self.images.remove(obj)
|
||||||
|
case TrackSegment():
|
||||||
|
self.track_segments.remove(obj)
|
||||||
|
case TrackArc():
|
||||||
|
self.track_arcs.remove(obj)
|
||||||
|
case Via():
|
||||||
|
self.vias.remove(obj)
|
||||||
|
case Zone():
|
||||||
|
self.zones.remove(obj)
|
||||||
|
case Group():
|
||||||
|
self.groups.remove(obj)
|
||||||
|
case Footprint():
|
||||||
|
self.footprints.remove(obj)
|
||||||
|
case _:
|
||||||
|
raise TypeError('Can only remove KiCad objects, cannot map generic gerbonara.cad objects for removal')
|
||||||
|
|
||||||
|
|
||||||
|
def remove_many(self, iterable):
|
||||||
|
iterable = {id(obj) for obj in iterable}
|
||||||
|
for field in fields(self):
|
||||||
|
if field.default_factory is list and field.name not in ('nets', 'properties'):
|
||||||
|
setattr(self, field.name, [obj for obj in getattr(self, field.name) if id(obj) not in iterable])
|
||||||
|
|
||||||
|
|
||||||
|
def add(self, obj):
|
||||||
|
match obj:
|
||||||
|
case gr.Text():
|
||||||
|
self.texts.append(obj)
|
||||||
|
case gr.TextBox():
|
||||||
|
self.text_boxes.append(obj)
|
||||||
|
case gr.Line():
|
||||||
|
self.lines.append(obj)
|
||||||
|
case gr.Rectangle():
|
||||||
|
self.rectangles.append(obj)
|
||||||
|
case gr.Circle():
|
||||||
|
self.circles.append(obj)
|
||||||
|
case gr.Arc():
|
||||||
|
self.arcs.append(obj)
|
||||||
|
case gr.Polygon():
|
||||||
|
self.polygons.append(obj)
|
||||||
|
case gr.Curve():
|
||||||
|
self.curves.append(obj)
|
||||||
|
case gr.Dimension():
|
||||||
|
self.dimensions.append(obj)
|
||||||
|
case gr.Image():
|
||||||
|
self.images.append(obj)
|
||||||
|
case TrackSegment():
|
||||||
|
self.track_segments.append(obj)
|
||||||
|
case TrackArc():
|
||||||
|
self.track_arcs.append(obj)
|
||||||
|
case Via():
|
||||||
|
self.vias.append(obj)
|
||||||
|
case Zone():
|
||||||
|
self.zones.append(obj)
|
||||||
|
case Group():
|
||||||
|
self.groups.append(obj)
|
||||||
|
case Footprint():
|
||||||
|
self.footprints.append(obj)
|
||||||
|
obj.board = self
|
||||||
|
case _:
|
||||||
|
for elem in self.map_gn_cad(obj):
|
||||||
|
self.add(elem)
|
||||||
|
|
||||||
|
|
||||||
|
def map_gn_cad(self, obj, locked=False, net_name=None):
|
||||||
|
match obj:
|
||||||
|
case cad_pr.Trace():
|
||||||
|
for elem in obj.to_graphic_objects():
|
||||||
|
elem.convert_to(MM)
|
||||||
|
match elem:
|
||||||
|
case go.Arc(x1, y1, x2, y2, xc, yc, cw, ap):
|
||||||
|
yield TrackArc(
|
||||||
|
start=XYCoord(x1, y1),
|
||||||
|
mid=XYCoord(x1+xc, y1+yc),
|
||||||
|
end=XYCoord(x2, y2),
|
||||||
|
width=ap.equivalent_width(MM),
|
||||||
|
layer=gn_side_to_kicad(obj.side),
|
||||||
|
locked=locked,
|
||||||
|
net=self.net_id(net_name))
|
||||||
|
|
||||||
|
case go.Line(x1, y1, x2, y2, ap):
|
||||||
|
yield TrackSegment(
|
||||||
|
start=XYCoord(x1, y1),
|
||||||
|
end=XYCoord(x2, y2),
|
||||||
|
width=ap.equivalent_width(MM),
|
||||||
|
layer=gn_side_to_kicad(obj.side),
|
||||||
|
locked=locked,
|
||||||
|
net=self.net_id(net_name))
|
||||||
|
|
||||||
|
case cad_pr.Via(pad_stack=cad_pr.ThroughViaStack(hole, dia, unit=st_unit)):
|
||||||
|
x, y, _a, _f = obj.abs_pos
|
||||||
|
x, y = MM(x, st_unit), MM(y, obj.unit)
|
||||||
|
yield Via(
|
||||||
|
locked=locked,
|
||||||
|
at=XYCoord(x, y),
|
||||||
|
size=MM(dia, st_unit),
|
||||||
|
drill=MM(hole, st_unit),
|
||||||
|
layers='*.Cu',
|
||||||
|
net=self.net_id(net_name))
|
||||||
|
|
||||||
|
case cad_pr.Text(_x, _y, text, font_size, stroke_width, h_align, v_align, layer, dark):
|
||||||
|
x, y, a, flip = obj.abs_pos
|
||||||
|
x, y = MM(x, st_unit), MM(y, st_unit)
|
||||||
|
size = MM(size, unit)
|
||||||
|
yield gr.Text(
|
||||||
|
text,
|
||||||
|
AtPos(x, y, -math.degrees(a)),
|
||||||
|
layer=gr.TextLayer(gn_layer_to_kicad(layer, flip), not dark),
|
||||||
|
effects=TextEffect(font=FontSpec(
|
||||||
|
size=XYCoord(size, size),
|
||||||
|
thickness=stroke_width),
|
||||||
|
justify=Justify(h=Atom(h_align) if h_align != 'center' else None,
|
||||||
|
v=Atom(v_align) if v_align != 'middle' else None,
|
||||||
|
mirror=flip)))
|
||||||
|
|
||||||
|
|
||||||
|
def unfill_zones(self):
|
||||||
|
for zone in self.zones:
|
||||||
|
zone.unfill()
|
||||||
|
|
||||||
|
|
||||||
|
def find_pads(self, net=None):
|
||||||
|
for fp in self.footprints:
|
||||||
|
for pad in fp.pads:
|
||||||
|
if net and not match_filter(net, pad.net.name):
|
||||||
|
continue
|
||||||
|
yield pad
|
||||||
|
|
||||||
|
|
||||||
|
def find_footprints(self, value=None, reference=None, name=None, net=None, sheetname=None, sheetfile=None):
|
||||||
|
for fp in self.footprints:
|
||||||
|
if name and not match_filter(name, fp.name):
|
||||||
|
continue
|
||||||
|
if value and not match_filter(value, fp.value):
|
||||||
|
continue
|
||||||
|
if reference and not match_filter(reference, fp.reference):
|
||||||
|
continue
|
||||||
|
if net and not any(pad.net and match_filter(net, pad.net.name) for pad in fp.pads):
|
||||||
|
continue
|
||||||
|
if sheetname and not match_filter(sheetname, fp.sheetname):
|
||||||
|
continue
|
||||||
|
if sheetfile and not match_filter(sheetfile, fp.sheetfile):
|
||||||
|
continue
|
||||||
|
yield fp
|
||||||
|
|
||||||
|
|
||||||
|
def find_traces(self, net=None, include_vias=True):
|
||||||
|
net_id = self.net_id(net, create=False)
|
||||||
|
match = lambda obj: obj.net == net_id
|
||||||
|
for obj in chain(self.track_segments, self.track_arcs, self.vias):
|
||||||
|
if obj.net == net_id:
|
||||||
|
yield obj
|
||||||
|
|
||||||
|
|
||||||
|
@property
|
||||||
|
def version(self):
|
||||||
|
return self._version
|
||||||
|
|
||||||
|
|
||||||
|
@version.setter
|
||||||
|
def version(self, value):
|
||||||
|
if value not in SUPPORTED_FILE_FORMAT_VERSIONS:
|
||||||
|
raise FormatError(f'File format version {value} is not supported. Supported versions are {", ".join(map(str, SUPPORTED_FILE_FORMAT_VERSIONS))}.')
|
||||||
|
|
||||||
|
|
||||||
|
def write(self, filename=None):
|
||||||
|
with open(filename or self.original_filename, 'w') as f:
|
||||||
|
f.write(self.serialize())
|
||||||
|
|
||||||
|
|
||||||
|
def serialize(self):
|
||||||
|
return build_sexp(sexp(type(self), self)[0])
|
||||||
|
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def open(kls, pcb_file, *args, **kwargs):
|
||||||
|
return kls.load(Path(pcb_file).read_text(), *args, **kwargs, original_filename=pcb_file)
|
||||||
|
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def load(kls, data, *args, **kwargs):
|
||||||
|
return kls.parse(data, *args, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
@property
|
||||||
|
def single_sided(self):
|
||||||
|
raise NotImplementedError()
|
||||||
|
|
||||||
|
|
||||||
|
def net_id(self, name, create=True):
|
||||||
|
if name is None:
|
||||||
|
return None
|
||||||
|
|
||||||
|
for i, n in self.nets.items():
|
||||||
|
if n == name:
|
||||||
|
return i
|
||||||
|
|
||||||
|
if create:
|
||||||
|
index = max(self.nets.keys()) + 1
|
||||||
|
self.nets[index] = name
|
||||||
|
return index
|
||||||
|
|
||||||
|
else:
|
||||||
|
raise IndexError(f'No such net: "{name}"')
|
||||||
|
|
||||||
|
|
||||||
|
# FIXME vvv
|
||||||
|
def graphic_objects(self, text=False, images=False):
|
||||||
|
return chain(
|
||||||
|
(self.texts if text else []),
|
||||||
|
(self.text_boxes if text else []),
|
||||||
|
self.lines,
|
||||||
|
self.rectangles,
|
||||||
|
self.circles,
|
||||||
|
self.arcs,
|
||||||
|
self.polygons,
|
||||||
|
self.curves,
|
||||||
|
(self.dimensions if text else []),
|
||||||
|
(self.images if images else []))
|
||||||
|
|
||||||
|
|
||||||
|
def tracks(self, vias=True):
|
||||||
|
return chain(self.track_segments, self.track_arcs, (self.vias if vias else []))
|
||||||
|
|
||||||
|
|
||||||
|
def objects(self, vias=True, text=False, images=False):
|
||||||
|
return chain(self.graphic_objects(text=text, images=images), self.tracks(vias=vias), self.footprints, self.zones, self.groups)
|
||||||
|
|
||||||
|
|
||||||
|
def render(self, layer_stack, layer_map, x=0, y=0, rotation=0, text=False, flip=False, variables={}, cache=None):
|
||||||
|
for obj in self.objects(images=False, vias=False, text=text):
|
||||||
|
if not (layer := layer_map.get(obj.layer)):
|
||||||
|
continue
|
||||||
|
|
||||||
|
for fe in obj.render(variables=variables):
|
||||||
|
fe.rotate(rotation)
|
||||||
|
fe.offset(x, -y, MM)
|
||||||
|
layer_stack[layer].objects.append(fe)
|
||||||
|
|
||||||
|
for obj in self.vias:
|
||||||
|
for glob in obj.layers or []:
|
||||||
|
for layer in fnmatch.filter(layer_map, glob):
|
||||||
|
for fe in obj.render(cache=cache):
|
||||||
|
fe.rotate(rotation)
|
||||||
|
fe.offset(x, -y, MM)
|
||||||
|
fe.aperture = fe.aperture.rotated(rotation)
|
||||||
|
layer_stack[layer_map[layer]].objects.append(fe)
|
||||||
|
|
||||||
|
for fe in obj.render_drill():
|
||||||
|
fe.rotate(rotation)
|
||||||
|
fe.offset(x, -y, MM)
|
||||||
|
layer_stack.drill_pth.append(fe)
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class BoardInstance(cad_pr.Positioned):
|
||||||
|
sexp: Board = None
|
||||||
|
variables: dict = field(default_factory=lambda: {})
|
||||||
|
|
||||||
|
def render(self, layer_stack, cache=None):
|
||||||
|
x, y, rotation, flip = self.abs_pos
|
||||||
|
x, y = MM(x, self.unit), MM(y, self.unit)
|
||||||
|
|
||||||
|
variables = dict(self.variables)
|
||||||
|
|
||||||
|
layer_map = {kc_id: gn_id for kc_id, gn_id in LAYER_MAP_K2G.items() if gn_id in layer_stack}
|
||||||
|
|
||||||
|
self.sexp.render(layer_stack, layer_map,
|
||||||
|
x=x, y=y, rotation=rotation,
|
||||||
|
flip=flip,
|
||||||
|
variables=variables, cache=cache)
|
||||||
|
|
||||||
|
def bounding_box(self, unit=MM):
|
||||||
|
return offset_bounds(self.sexp.bounding_box(unit), unit(self.x, self.unit), unit(self.y, self.unit))
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
import sys
|
||||||
|
from ...layers import LayerStack
|
||||||
|
fp = Board.open(sys.argv[1])
|
||||||
|
stack = LayerStack()
|
||||||
|
BoardInstance(0, 0, fp, unit=MM).render(stack)
|
||||||
|
print(stack.to_pretty_svg())
|
||||||
|
stack.save_to_directory('/tmp/testdir')
|
||||||
|
|
||||||
267
src/gerbonara/cad/kicad/primitives.py
Normal file
|
|
@ -0,0 +1,267 @@
|
||||||
|
|
||||||
|
import enum
|
||||||
|
import math
|
||||||
|
import re
|
||||||
|
|
||||||
|
from .sexp import *
|
||||||
|
from .base_types import *
|
||||||
|
|
||||||
|
|
||||||
|
def unfuck_layers(layers):
|
||||||
|
if layers and layers[0] == 'F&B.Cu':
|
||||||
|
return ['F.Cu', 'B.Cu', *layers[1:]]
|
||||||
|
else:
|
||||||
|
return layers
|
||||||
|
|
||||||
|
|
||||||
|
def fuck_layers(layers):
|
||||||
|
if layers and 'F.Cu' in layers and 'B.Cu' in layers and not any(re.match(r'^In[0-9]+\.Cu$', l) for l in layers):
|
||||||
|
return ['F&B.Cu', *(l for l in layers if l not in ('F.Cu', 'B.Cu'))]
|
||||||
|
else:
|
||||||
|
return layers
|
||||||
|
|
||||||
|
|
||||||
|
def layer_mask(layers):
|
||||||
|
if isinstance(layers, int):
|
||||||
|
return layers
|
||||||
|
|
||||||
|
if isinstance(layers, str):
|
||||||
|
layers = [l.strip() for l in layers.split(',')]
|
||||||
|
|
||||||
|
mask = 0
|
||||||
|
for layer in layers:
|
||||||
|
match layer:
|
||||||
|
case '*.Cu':
|
||||||
|
return 0xffffffff
|
||||||
|
case 'F.Cu':
|
||||||
|
mask |= 1<<0
|
||||||
|
case 'B.Cu':
|
||||||
|
mask |= 1<<31
|
||||||
|
case _:
|
||||||
|
if (m := re.match(fr'In([0-9]+)\.Cu', layer)):
|
||||||
|
mask |= 1<<int(m.group(1))
|
||||||
|
return mask
|
||||||
|
|
||||||
|
|
||||||
|
def center_arc_to_kicad_mid(center, start, end):
|
||||||
|
# Convert normal p1/p2/center notation to the insanity that is kicad's midpoint notation
|
||||||
|
cx, cy = center.x, center.y
|
||||||
|
x1, y1 = start.x - cx, start.y - cy
|
||||||
|
x2, y2 = end.x - cx, end.y - cy
|
||||||
|
# Get a vector pointing from the center to the "mid" point.
|
||||||
|
dx, dy = x1 - x2, y1 - y2 # Get a vector pointing from "end" to "start"
|
||||||
|
dx, dy = -dy, dx # rotate by 90 degrees counter-clockwise
|
||||||
|
# normalize vector, and multiply by radius to get final point
|
||||||
|
r = math.hypot(x1, y1)
|
||||||
|
l = math.hypot(dx, dy)
|
||||||
|
mx = cx + dx / l * r
|
||||||
|
my = cy + dy / l * r
|
||||||
|
return XYCoord(mx, my)
|
||||||
|
|
||||||
|
|
||||||
|
def kicad_mid_to_center_arc(mid, start, end):
|
||||||
|
""" Convert kicad's slightly insane midpoint notation to standrad center/p1/p2 notation.
|
||||||
|
|
||||||
|
returns a ((center_x, center_y), radius, clockwise) tuple in KiCad coordinates.
|
||||||
|
|
||||||
|
Returns the center and radius of the circle passing the given 3 points.
|
||||||
|
In case the 3 points form a line, raises a ValueError.
|
||||||
|
"""
|
||||||
|
# https://stackoverflow.com/questions/28910718/give-3-points-and-a-plot-circle
|
||||||
|
p1, p2, p3 = start, mid, end
|
||||||
|
|
||||||
|
temp = p2[0] * p2[0] + p2[1] * p2[1]
|
||||||
|
bc = (p1[0] * p1[0] + p1[1] * p1[1] - temp) / 2
|
||||||
|
cd = (temp - p3[0] * p3[0] - p3[1] * p3[1]) / 2
|
||||||
|
det = (p1[0] - p2[0]) * (p2[1] - p3[1]) - (p2[0] - p3[0]) * (p1[1] - p2[1])
|
||||||
|
|
||||||
|
if abs(det) < 1.0e-6:
|
||||||
|
raise ValueError()
|
||||||
|
|
||||||
|
# Center of circle
|
||||||
|
cx = (bc*(p2[1] - p3[1]) - cd*(p1[1] - p2[1])) / det
|
||||||
|
cy = ((p1[0] - p2[0]) * cd - (p2[0] - p3[0]) * bc) / det
|
||||||
|
|
||||||
|
radius = math.sqrt((cx - p1[0])**2 + (cy - p1[1])**2)
|
||||||
|
return (cx, cy), radius, det < 0
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('hatch')
|
||||||
|
class Hatch:
|
||||||
|
style: AtomChoice(Atom.none, Atom.edge, Atom.full) = Atom.edge
|
||||||
|
pitch: float = 0.5
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('connect_pads')
|
||||||
|
class PadConnection:
|
||||||
|
type: AtomChoice(Atom.yes, Atom.thru_hole_only, Atom.full, Atom.no) = None
|
||||||
|
clearance: Named(float) = 0
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('keepout')
|
||||||
|
class ZoneKeepout:
|
||||||
|
tracks_allowed: Named(YesNoAtom(yes=Atom.allowed, no=Atom.not_allowed), name='tracks') = True
|
||||||
|
vias_allowed: Named(YesNoAtom(yes=Atom.allowed, no=Atom.not_allowed), name='vias') = True
|
||||||
|
pads_allowed: Named(YesNoAtom(yes=Atom.allowed, no=Atom.not_allowed), name='pads') = True
|
||||||
|
copperpour_allowed: Named(YesNoAtom(yes=Atom.allowed, no=Atom.not_allowed), name='copperpour') = True
|
||||||
|
footprints_allowed: Named(YesNoAtom(yes=Atom.allowed, no=Atom.not_allowed), name='footprints') = True
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('smoothing')
|
||||||
|
class ZoneSmoothing:
|
||||||
|
style: AtomChoice(Atom.chamfer, Atom.fillet) = Atom.chamfer
|
||||||
|
radius: Named(float) = None
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('fill')
|
||||||
|
class ZoneFill:
|
||||||
|
yes: Flag() = False
|
||||||
|
mode: Named(Flag(atom=Atom.hatch)) = False
|
||||||
|
thermal_gap: Named(float) = 0.508
|
||||||
|
thermal_bridge_width: Named(float) = 0.508
|
||||||
|
smoothing: ZoneSmoothing = None
|
||||||
|
radius: Named(float) = 0.125
|
||||||
|
island_removal_mode: Named(int) = None
|
||||||
|
island_area_min: Named(float) = None
|
||||||
|
hatch_thickness: Named(float) = None
|
||||||
|
hatch_gap: Named(float) = None
|
||||||
|
hatch_orientation: Named(int) = None
|
||||||
|
hatch_smoothing_level: Named(int) = None
|
||||||
|
hatch_smoothing_value: Named(float) = None
|
||||||
|
hatch_border_algorithm: Named(AtomChoice(Atom.hatch_thickness, Atom.min_thickness)) = None
|
||||||
|
hatch_min_hole_area: Named(float) = None
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('filled_polygon')
|
||||||
|
class FillPolygon:
|
||||||
|
layer: Named(str) = ""
|
||||||
|
island: Wrap(Flag()) = False
|
||||||
|
pts: ArcPointList = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('fill_segments')
|
||||||
|
class FillSegment:
|
||||||
|
layer: Named(str) = ""
|
||||||
|
pts: ArcPointList = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('polygon')
|
||||||
|
class ZonePolygon:
|
||||||
|
pts: ArcPointList = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('placement')
|
||||||
|
class ZonePlacement:
|
||||||
|
enabled: Named(YesNoAtom()) = False
|
||||||
|
sheetname: Named(str) = ''
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('teardrop')
|
||||||
|
class ZoneTeardropSpec:
|
||||||
|
type: Named(AtomChoice(Atom.padvia, Atom.track_end)) = Atom.padvia
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('attr')
|
||||||
|
class ZoneAttr:
|
||||||
|
teardrop: ZoneTeardropSpec = None
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('zone')
|
||||||
|
class Zone:
|
||||||
|
locked: Flag() = False
|
||||||
|
net: Named(int) = 0
|
||||||
|
net_name: Named(str) = ""
|
||||||
|
layer: Named(str) = None
|
||||||
|
layers: Named(Array(str)) = None
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
tstamp: Timestamp = None
|
||||||
|
name: Named(str) = None
|
||||||
|
hatch: Hatch = None
|
||||||
|
priority: OmitDefault(Named(int)) = 0
|
||||||
|
attr: ZoneAttr = None
|
||||||
|
connect_pads: PadConnection = field(default_factory=PadConnection)
|
||||||
|
min_thickness: Named(float) = 0.254
|
||||||
|
filled_areas_thickness: Named(YesNoAtom()) = True
|
||||||
|
keepout: ZoneKeepout = None
|
||||||
|
placement: ZonePlacement = None
|
||||||
|
fill: ZoneFill = field(default_factory=ZoneFill)
|
||||||
|
polygon: ZonePolygon = field(default_factory=ZonePolygon)
|
||||||
|
fill_polygons: List(FillPolygon) = field(default_factory=list)
|
||||||
|
fill_segments: List(FillSegment) = field(default_factory=list)
|
||||||
|
|
||||||
|
def __after_parse__(self, parent=None):
|
||||||
|
self.layers = unfuck_layers(self.layers)
|
||||||
|
|
||||||
|
def __before_sexp__(self):
|
||||||
|
self.layers = fuck_layers(self.layers)
|
||||||
|
|
||||||
|
def unfill(self):
|
||||||
|
self.fill.yes = False
|
||||||
|
self.fill_polygons = []
|
||||||
|
self.fill_segments = []
|
||||||
|
|
||||||
|
def rotate(self, angle, cx=None, cy=None):
|
||||||
|
self.unfill()
|
||||||
|
self.polygon.pts = [pt.with_rotation(angle, cx, cy) for pt in self.polygon.pts]
|
||||||
|
|
||||||
|
def offset(self, x=0, y=0):
|
||||||
|
self.unfill()
|
||||||
|
self.polygon.pts = [pt.with_offset(x, y) for pt in self.polygon.pts]
|
||||||
|
|
||||||
|
|
||||||
|
def bounding_box(self):
|
||||||
|
min_x = min(pt.x for pt in self.polygon.pts)
|
||||||
|
min_y = min(pt.y for pt in self.polygon.pts)
|
||||||
|
max_x = max(pt.x for pt in self.polygon.pts)
|
||||||
|
max_y = max(pt.y for pt in self.polygon.pts)
|
||||||
|
return (min_x, min_y), (max_x, max_y)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('polygon')
|
||||||
|
class RenderCachePolygon:
|
||||||
|
pts: PointList = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('render_cache')
|
||||||
|
class RenderCache:
|
||||||
|
text: str = None
|
||||||
|
rotation: int = 0
|
||||||
|
polygons: List(RenderCachePolygon) = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('margins')
|
||||||
|
class Margins:
|
||||||
|
left: float = 0.0
|
||||||
|
top: float = 0.0
|
||||||
|
right: float = 0.0
|
||||||
|
bottom: float = 0.0
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('comment')
|
||||||
|
class TitleComment:
|
||||||
|
@classmethod
|
||||||
|
def __map__(kls, obj, parent=None, path=''):
|
||||||
|
lines = []
|
||||||
|
for lineno, content in zip(obj[1::2], obj[2::2]):
|
||||||
|
while lineno > len(lines):
|
||||||
|
lines.append('')
|
||||||
|
lines[lineno-1] = content
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def __sexp__(kls, value):
|
||||||
|
l = [Atom.comment]
|
||||||
|
for i, line in enumerate(value.splitlines(), start=1):
|
||||||
|
l.append(i)
|
||||||
|
l.append(line.rstrip('\n'))
|
||||||
|
return l
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('title_block')
|
||||||
|
class TitleBlock:
|
||||||
|
title: Named(str) = ''
|
||||||
|
date: Named(str) = ''
|
||||||
|
rev: Named(str) = ''
|
||||||
|
company: Named(str) = ''
|
||||||
|
comment: TitleComment = None
|
||||||
|
|
||||||
|
|
||||||
867
src/gerbonara/cad/kicad/schematic.py
Normal file
|
|
@ -0,0 +1,867 @@
|
||||||
|
"""
|
||||||
|
Library for handling KiCad's schematic files (`*.kicad_sch`).
|
||||||
|
"""
|
||||||
|
|
||||||
|
import math
|
||||||
|
import string
|
||||||
|
from pathlib import Path
|
||||||
|
from dataclasses import field, KW_ONLY
|
||||||
|
from itertools import chain
|
||||||
|
import re
|
||||||
|
import fnmatch
|
||||||
|
import os.path
|
||||||
|
import warnings
|
||||||
|
|
||||||
|
from .sexp import *
|
||||||
|
from .base_types import *
|
||||||
|
from .primitives import *
|
||||||
|
from .symbols import Symbol
|
||||||
|
from . import graphical_primitives as gr
|
||||||
|
|
||||||
|
from .. import primitives as cad_pr
|
||||||
|
from ... import __version__
|
||||||
|
|
||||||
|
from ... import graphic_primitives as gp
|
||||||
|
from ... import graphic_objects as go
|
||||||
|
from ... import apertures as ap
|
||||||
|
from ...layers import LayerStack
|
||||||
|
from ...newstroke import Newstroke
|
||||||
|
from ...utils import MM, rotate_point, Tag, setup_svg
|
||||||
|
from .schematic_colors import *
|
||||||
|
|
||||||
|
|
||||||
|
KICAD_PAPER_SIZES = {
|
||||||
|
'A5': (210, 148),
|
||||||
|
'A4': (297, 210),
|
||||||
|
'A3': (420, 297),
|
||||||
|
'A2': (594, 420),
|
||||||
|
'A1': (841, 594),
|
||||||
|
'A0': (1189, 841),
|
||||||
|
'A': (11*25.4, 8.5*25.4),
|
||||||
|
'B': (17*25.4, 11*15.4),
|
||||||
|
'C': (22*25.4, 17*25.4),
|
||||||
|
'D': (34*25.4, 22*25.4),
|
||||||
|
'E': (44*25.4, 34*25.4),
|
||||||
|
'USLetter': (11*25.4, 8.5*25.4),
|
||||||
|
'USLegal': (14*25.4, 8.5*25.4),
|
||||||
|
'USLedger': (17*25.4, 11*25.4),
|
||||||
|
}
|
||||||
|
|
||||||
|
@sexp_type('path')
|
||||||
|
class SheetPath:
|
||||||
|
path: str = '/'
|
||||||
|
page: Named(str) = '1'
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('junction')
|
||||||
|
class Junction:
|
||||||
|
at: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
diameter: Named(float) = 0
|
||||||
|
color: Color = field(default_factory=lambda: Color(0, 0, 0, 0))
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
|
||||||
|
def bounding_box(self, default=None):
|
||||||
|
r = (self.diameter/2 or 0.5)
|
||||||
|
return (self.at.x - r, self.at.y - r), (self.at.x + r, self.at.y + r)
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
yield Tag('circle', cx=f'{self.at.x:.3f}', cy=f'{self.at.y:.3f}', r=(self.diameter/2 or 0.5),
|
||||||
|
fill=self.color.svg(colorscheme.wire))
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('no_connect')
|
||||||
|
class NoConnect:
|
||||||
|
at: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
|
||||||
|
def bounding_box(self, default=None):
|
||||||
|
r = 0.635
|
||||||
|
return (self.at.x - r, self.at.y - r), (self.at.x + r, self.at.y + r)
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
r = 0.635
|
||||||
|
x, y = self.at.x, self.at.y
|
||||||
|
yield Tag('path', d=f'M {x-r:.3f} {y-r:.3f} L {x+r:.3f} {y+r:.3f} M {x-r:.3f} {y+r:.3f} L {x+r:.3f} {y-r:.3f}',
|
||||||
|
fill='none', stroke_width='0.254', stroke=colorscheme.no_connect)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('bus_alias')
|
||||||
|
class BusAlias:
|
||||||
|
name: str = ''
|
||||||
|
members: Named(Array(str)) = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('bus_entry')
|
||||||
|
class BusEntry:
|
||||||
|
at: AtPos = field(default_factory=AtPos)
|
||||||
|
size: Rename(XYCoord) = field(default_factory=lambda: XYCoord(2.54, 2.54))
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
|
||||||
|
def bounding_box(self, default=None):
|
||||||
|
r = math.hypot(self.size.x, self.size.y)
|
||||||
|
x1, y1 = self.at.x, self.at.y
|
||||||
|
x2, y2 = rotate_point(x1+r, y1+r, self.at.rotation or 0)
|
||||||
|
x1, x2 = min(x1, x2), max(x1, x2)
|
||||||
|
y1, y2 = min(y1, y2), max(y1, y2)
|
||||||
|
|
||||||
|
r = (self.stroke.width or 0.254) / 2
|
||||||
|
return (x1-r, y1-r), (x2+r, y2+r)
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
yield Tag('path', d='M {self.at.x} {self.at.y} l {self.size.x} {self.size.y}',
|
||||||
|
transform=f'rotate({self.at.rotation or 0})',
|
||||||
|
fill='none', stroke=self.stroke.svg_color(colorscheme.bus), width=self.stroke.width or '0.254')
|
||||||
|
|
||||||
|
|
||||||
|
def _polyline_svg(self, default_color):
|
||||||
|
da = Dasher(self)
|
||||||
|
if len(self.points.xy) < 2:
|
||||||
|
warnings.warn(f'Schematic {type(self)} with less than two points')
|
||||||
|
|
||||||
|
p0, *rest = self.points.xy
|
||||||
|
da.move(p0.x, p0.y)
|
||||||
|
for pn in rest:
|
||||||
|
da.line(pn.x, pn.y)
|
||||||
|
|
||||||
|
return da.svg(stroke=self.stroke.svg_color(default_color))
|
||||||
|
|
||||||
|
|
||||||
|
def _polyline_bounds(self):
|
||||||
|
x1 = min(pt.x for pt in self.points)
|
||||||
|
y1 = min(pt.y for pt in self.points)
|
||||||
|
x2 = max(pt.x for pt in self.points)
|
||||||
|
y2 = max(pt.y for pt in self.points)
|
||||||
|
|
||||||
|
r = (self.stroke.width or 0.254) / 2
|
||||||
|
return (x1-r, y1-r), (x2+r, y2+r)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('wire')
|
||||||
|
class Wire:
|
||||||
|
points: PointList = field(default_factory=list)
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
|
||||||
|
def bounding_box(self, default=None):
|
||||||
|
return _polyline_bounds(self)
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
yield _polyline_svg(self, colorscheme.wire)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('bus')
|
||||||
|
class Bus:
|
||||||
|
points: PointList = field(default_factory=list)
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
|
||||||
|
def bounding_box(self, default=None):
|
||||||
|
return _polyline_bounds(self)
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
yield _polyline_svg(self, colorscheme.bus)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('polyline')
|
||||||
|
class Polyline:
|
||||||
|
points: PointList = field(default_factory=list)
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
fill: OmitDefault(Fill) = None
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
|
||||||
|
def bounding_box(self, default=None):
|
||||||
|
return _polyline_bounds(self)
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
yield _polyline_svg(self, colorscheme.lines)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('circle')
|
||||||
|
class Circle:
|
||||||
|
center: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
radius: Named(float) = 0.0
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
fill: OmitDefault(Fill) = None
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('rectangle')
|
||||||
|
class Rectangle:
|
||||||
|
start: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
end: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
fill: OmitDefault(Fill) = None
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
|
||||||
|
|
||||||
|
def label_shape_path_d(shape, w, h):
|
||||||
|
l, r = {
|
||||||
|
Atom.input: '<]',
|
||||||
|
Atom.output: '[>',
|
||||||
|
Atom.bidirectional: '<>',
|
||||||
|
Atom.tri_state: '<>',
|
||||||
|
Atom.passive: '[]'}.get(shape, '<]')
|
||||||
|
r = h/2
|
||||||
|
|
||||||
|
if l == '[':
|
||||||
|
d = f'M {r:.3f} {r:.3f} L 0 {r:.3f} L 0 {-r:.3f} L {r:.3f} {-r:.3f}'
|
||||||
|
else:
|
||||||
|
d = f'M {r:.3f} {r:.3f} L 0 0 L {r:.3f} {-r:.3f}'
|
||||||
|
|
||||||
|
e = w+r
|
||||||
|
d += f' L {e:.3f} {-r:.3f}'
|
||||||
|
|
||||||
|
if l == '[':
|
||||||
|
return d + f'L {e+r:.3f} {-r:.3f} L {e+r:.3f} {r:.3f} L {e:.3f} {r:.3f} Z'
|
||||||
|
else:
|
||||||
|
return d + f'L {e+r:.3f} {0:.3f} L {e:.3f} {r:.3f} Z'
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class TextLabel(TextMixin):
|
||||||
|
text: str = ''
|
||||||
|
shape: Named(AtomChoice(Atom.input, Atom.output, Atom.bidirectional, Atom.tri_state, Atom.passive, Atom.dot, Atom.round, Atom.diamond, Atom.rectangle)) = Atom.passive
|
||||||
|
exclude_from_sim: Named(YesNoAtom()) = False
|
||||||
|
at: AtPos = field(default_factory=AtPos)
|
||||||
|
fields_autoplaced: Named(YesNoAtom()) = False
|
||||||
|
effects: TextEffect = field(default_factory=TextEffect)
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
properties: List(DrawnProperty) = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('text')
|
||||||
|
class Text(TextLabel):
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
yield from TextMixin.to_svg(self, colorscheme.text)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('label')
|
||||||
|
class LocalLabel(TextLabel):
|
||||||
|
@property
|
||||||
|
def _text_offset(self):
|
||||||
|
return (0, -2*self.line_width)
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
yield from TextMixin.to_svg(self, colorscheme.labels)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('global_label')
|
||||||
|
class GlobalLabel(TextLabel):
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
text = super(TextMixin, self).to_svg(colorscheme.labels),
|
||||||
|
text.attrs['transform'] = f'translate({self.size*0.6:.3f} 0)'
|
||||||
|
(x1, y1), (x2, y2) = self.bounding_box()
|
||||||
|
frame = Tag('path', fill='none', stroke_width=0.254, stroke=colorscheme.lines,
|
||||||
|
d=label_shape_path_d(self.shape, self.size*0.2 + y2-y1, self.size*1.2 + 0.254))
|
||||||
|
yield Tag('g', children=[frame, text])
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('hierarchical_label')
|
||||||
|
class HierarchicalLabel(TextLabel):
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
text, = TextMixin.to_svg(self, colorscheme.labels),
|
||||||
|
text.attrs['transform'] = f'translate({self.size*1.2:.3f} 0)'
|
||||||
|
frame = Tag('path', fill='none', stroke_width=0.254, stroke=colorscheme.lines,
|
||||||
|
d=label_shape_path_d(self.shape, self.size, self.size))
|
||||||
|
yield Tag('g', children=[frame, text])
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('netclass_flag')
|
||||||
|
class NetclassFlag(TextLabel):
|
||||||
|
length: Named(float) = 2.54
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
# FIXME
|
||||||
|
yield from TextMixin.to_svg(self, colorscheme.text)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('pin')
|
||||||
|
class Pin:
|
||||||
|
name: str = '1'
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
alternate: Named(str) = None
|
||||||
|
|
||||||
|
|
||||||
|
# Suddenly, we're doing syntax like this is yaml or something.
|
||||||
|
@sexp_type('path')
|
||||||
|
class SymbolCrosslinkSheet:
|
||||||
|
path: str = ''
|
||||||
|
reference: Named(str) = ''
|
||||||
|
unit: Named(int) = 1
|
||||||
|
value: OmitDefault(Named(str)) = None
|
||||||
|
footprint: OmitDefault(Named(str)) = None
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('project')
|
||||||
|
class SymbolCrosslinkProject:
|
||||||
|
project_name: str = ''
|
||||||
|
instances: List(SymbolCrosslinkSheet) = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('mirror')
|
||||||
|
class MirrorFlags:
|
||||||
|
x: Flag() = False
|
||||||
|
y: Flag() = False
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('property')
|
||||||
|
class DrawnProperty(TextMixin):
|
||||||
|
key: str = None
|
||||||
|
value: str = None
|
||||||
|
id: Named(int) = None
|
||||||
|
at: AtPos = field(default_factory=AtPos)
|
||||||
|
hide: Flag() = False
|
||||||
|
effects: TextEffect = field(default_factory=TextEffect)
|
||||||
|
_: SEXP_END = None
|
||||||
|
parent: object = None
|
||||||
|
|
||||||
|
def __after_parse__(self, parent=None):
|
||||||
|
self.parent = parent
|
||||||
|
|
||||||
|
# Alias value for text mixin
|
||||||
|
@property
|
||||||
|
def text(self):
|
||||||
|
if self.key == 'Reference' and self.parent.unit > 0:
|
||||||
|
return f'{self.value}{string.ascii_uppercase[self.parent.unit-1]}'
|
||||||
|
else:
|
||||||
|
return self.value
|
||||||
|
|
||||||
|
@text.setter
|
||||||
|
def text(self, value):
|
||||||
|
self.value = value
|
||||||
|
|
||||||
|
@property
|
||||||
|
def default_v_align(self):
|
||||||
|
return 'middle'
|
||||||
|
|
||||||
|
@property
|
||||||
|
def h_align(self):
|
||||||
|
align = self.effects.justify.h_str
|
||||||
|
if self.rotation in (90, 270):
|
||||||
|
align = {'left': 'right', 'right': 'left'}.get(align, align)
|
||||||
|
return align
|
||||||
|
|
||||||
|
@property
|
||||||
|
def rotation(self):
|
||||||
|
rot = self.at.rotation
|
||||||
|
rot += getattr(self.parent.at, 'rotation', 0)
|
||||||
|
return rot%360
|
||||||
|
|
||||||
|
@property
|
||||||
|
def mirrored(self):
|
||||||
|
if hasattr(self.parent, 'mirror'):
|
||||||
|
return self.parent.mirror.x, self.parent.mirror.y
|
||||||
|
return False, False
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
if not self.hide:
|
||||||
|
yield from TextMixin.to_svg(self, colorscheme.values)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('default_instance')
|
||||||
|
class DefaultSymbolInstance:
|
||||||
|
reference: Named(str) = ''
|
||||||
|
unit: Named(int) = 1
|
||||||
|
value: Named(str) = ''
|
||||||
|
footprint: Named(str) = ''
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('symbol')
|
||||||
|
class SymbolInstance:
|
||||||
|
name: str = None
|
||||||
|
lib_name: Named(str) = ''
|
||||||
|
lib_id: Named(str) = ''
|
||||||
|
at: AtPos = field(default_factory=AtPos)
|
||||||
|
mirror: OmitDefault(MirrorFlags) = field(default_factory=MirrorFlags)
|
||||||
|
unit: Named(int) = 1
|
||||||
|
exclude_from_sim: Named(YesNoAtom()) = False
|
||||||
|
in_bom: Named(YesNoAtom()) = True
|
||||||
|
on_board: Named(YesNoAtom()) = True
|
||||||
|
dnp: Named(YesNoAtom()) = True
|
||||||
|
fields_autoplaced: Named(YesNoAtom()) = True
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
default_instance: DefaultSymbolInstance = None
|
||||||
|
properties: List(DrawnProperty) = field(default_factory=list)
|
||||||
|
# AFAICT this property is completely redundant.
|
||||||
|
pins: List(Pin) = field(default_factory=list)
|
||||||
|
# AFAICT this property, too, is completely redundant. It ultimately just lists paths and references of at most
|
||||||
|
# three other uses of the same symbol in this schematic.
|
||||||
|
instances: Named(Array(SymbolCrosslinkProject)) = field(default_factory=list)
|
||||||
|
_ : SEXP_END = None
|
||||||
|
schematic: object = field(repr=False, default=None)
|
||||||
|
|
||||||
|
def __after_parse__(self, parent):
|
||||||
|
self.schematic = parent
|
||||||
|
|
||||||
|
@property
|
||||||
|
def reference(self):
|
||||||
|
return self['Reference'].value
|
||||||
|
|
||||||
|
@reference.setter
|
||||||
|
def reference(self, value):
|
||||||
|
self['Reference'].value = value
|
||||||
|
|
||||||
|
@property
|
||||||
|
def value(self):
|
||||||
|
return self['Value'].value
|
||||||
|
|
||||||
|
@value.setter
|
||||||
|
def value(self, value):
|
||||||
|
self['Value'].value = value
|
||||||
|
|
||||||
|
@property
|
||||||
|
def footprint(self):
|
||||||
|
return self['Footprint'].value
|
||||||
|
|
||||||
|
@footprint.setter
|
||||||
|
def footprint(self, value):
|
||||||
|
self['Footprint'].value = value
|
||||||
|
|
||||||
|
def __getitem__(self, key):
|
||||||
|
for prop in self.properties:
|
||||||
|
if prop.key == key:
|
||||||
|
return prop
|
||||||
|
|
||||||
|
@property
|
||||||
|
def rotation(self):
|
||||||
|
return self.at.rotation
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
children = []
|
||||||
|
rot = self.at.rotation
|
||||||
|
|
||||||
|
sym = self.schematic.lookup_symbol(self.lib_name, self.lib_id)
|
||||||
|
|
||||||
|
units = [unit for unit in sym.units if unit.unit_global or unit.unit_index == self.unit]
|
||||||
|
|
||||||
|
at_xform = xform = f'translate({self.at.x:.3f} {self.at.y:.3f})'
|
||||||
|
if self.mirror.y:
|
||||||
|
xform += f'scale(-1 -1)'
|
||||||
|
elif self.mirror.x:
|
||||||
|
xform += f'scale(1 1)'
|
||||||
|
else:
|
||||||
|
xform += f'scale(1 -1)'
|
||||||
|
if rot:
|
||||||
|
xform += f'rotate({rot})'
|
||||||
|
|
||||||
|
children = [foo for unit in units for elem in unit.graphical_elements for foo in elem.to_svg(colorscheme)]
|
||||||
|
yield Tag('g', children=children, transform=xform, fill=colorscheme.fill, stroke=colorscheme.lines)
|
||||||
|
|
||||||
|
children = [foo for unit in units for pin in unit.pins for foo in pin.to_svg(colorscheme, self.mirror, rot)]
|
||||||
|
yield Tag('g', children=children, transform=at_xform, fill=colorscheme.fill, stroke=colorscheme.lines)
|
||||||
|
|
||||||
|
for prop in self.properties:
|
||||||
|
yield from prop.to_svg(colorscheme)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('path')
|
||||||
|
class SubsheetCrosslinkSheet:
|
||||||
|
path: str = ''
|
||||||
|
page: Named(str) = ''
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('project')
|
||||||
|
class SubsheetCrosslinkProject:
|
||||||
|
project_name: str = ''
|
||||||
|
instances: List(SymbolCrosslinkSheet) = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('pin')
|
||||||
|
class SubsheetPin:
|
||||||
|
name: str = '1'
|
||||||
|
shape: AtomChoice(Atom.input, Atom.output, Atom.bidirectional, Atom.tri_state, Atom.passive) = Atom.input
|
||||||
|
at: AtPos = field(default_factory=AtPos)
|
||||||
|
effects: TextEffect = field(default_factory=TextEffect)
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
_ : SEXP_END = None
|
||||||
|
subsheet: object = None
|
||||||
|
|
||||||
|
def __after_parse__(self, parent):
|
||||||
|
self.subsheet = parent
|
||||||
|
|
||||||
|
def to_svg(self):
|
||||||
|
size = self.effects.font.size.y or 1.27
|
||||||
|
yield Tag('path', fill='none', d=label_shape_path_d(self.shape, 0, size+0.5),
|
||||||
|
transform=f'translate({self.at.x:.3f} {self.at.y:.3f}) rotate({180-self.at.rotation})')
|
||||||
|
|
||||||
|
lx, ly = self.at.x, self.at.y
|
||||||
|
dx, dy = rotate_point(-(size+1), 0, math.radians(self.at.rotation))
|
||||||
|
lx += dx
|
||||||
|
ly += dy
|
||||||
|
frot = self.at.rotation
|
||||||
|
h_align = 'right'
|
||||||
|
if frot == 180:
|
||||||
|
frot = 0
|
||||||
|
h_align = 'left'
|
||||||
|
|
||||||
|
font = Newstroke.load()
|
||||||
|
yield font.render_svg(self.name,
|
||||||
|
size=size,
|
||||||
|
x0=0,
|
||||||
|
y0=0,
|
||||||
|
h_align=h_align,
|
||||||
|
v_align='middle',
|
||||||
|
rotation=-frot,
|
||||||
|
transform=f'translate({lx:.3f} {ly:.3f})',
|
||||||
|
scale=(1, 1),
|
||||||
|
mirror=(False, False),
|
||||||
|
)
|
||||||
|
|
||||||
|
@sexp_type('fill')
|
||||||
|
class SubsheetFill:
|
||||||
|
color: Color = field(default_factory=lambda: Color(0, 0, 0, 0))
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('sheet')
|
||||||
|
class Subsheet:
|
||||||
|
at: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
size: Rename(XYCoord) = field(default_factory=lambda: XYCoord(2.54, 2.54))
|
||||||
|
exclude_from_sim: Named(YesNoAtom()) = False
|
||||||
|
in_bom: Named(YesNoAtom()) = False
|
||||||
|
on_board: Named(YesNoAtom()) = False
|
||||||
|
dnp: Named(YesNoAtom()) = False
|
||||||
|
fields_autoplaced: Named(YesNoAtom()) = True
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
fill: SubsheetFill = field(default_factory=SubsheetFill)
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
_properties: List(DrawnProperty) = field(default_factory=list)
|
||||||
|
pins: List(SubsheetPin) = field(default_factory=list)
|
||||||
|
# AFAICT this is completely redundant, just like the one in SymbolInstance
|
||||||
|
instances: Named(List(SubsheetCrosslinkProject)) = field(default_factory=list)
|
||||||
|
_ : SEXP_END = None
|
||||||
|
sheet_name: object = field(default_factory=lambda: DrawnProperty('Sheetname', ''))
|
||||||
|
file_name: object = field(default_factory=lambda: DrawnProperty('Sheetfile', ''))
|
||||||
|
schematic: object = field(repr=False, default=None)
|
||||||
|
|
||||||
|
def __after_parse__(self, parent):
|
||||||
|
self.sheet_name, self.file_name, *_extra_params = self._properties
|
||||||
|
self.schematic = parent
|
||||||
|
|
||||||
|
def __before_sexp__(self):
|
||||||
|
self._properties = [self.sheet_name, self.file_name]
|
||||||
|
|
||||||
|
@property
|
||||||
|
def rotation(self):
|
||||||
|
return 0
|
||||||
|
|
||||||
|
def open(self, search_dir=None, safe=True):
|
||||||
|
if search_dir is None:
|
||||||
|
if not self.schematic.original_filename:
|
||||||
|
raise FileNotFoundError('No search path given and path of parent schematic unknown')
|
||||||
|
else:
|
||||||
|
search_dir = Path(self.schematic.original_filename).parent
|
||||||
|
else:
|
||||||
|
search_dir = Path(search_dir)
|
||||||
|
|
||||||
|
resolved = search_dir / self.file_name.value
|
||||||
|
if safe and os.path.commonprefix((search_dir.parts, resolved.parts)) != search_dir.parts:
|
||||||
|
raise ValueError('Subsheet path traversal to parent directory attempted in Subsheet.open(..., safe=True)')
|
||||||
|
|
||||||
|
return Schematic.open(resolved)
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
children = []
|
||||||
|
|
||||||
|
for prop in self._properties:
|
||||||
|
yield from prop.to_svg(colorscheme)
|
||||||
|
|
||||||
|
yield Tag('rect', x=f'{self.at.x:.3f}', y=f'{self.at.y:.3f}',
|
||||||
|
width=f'{self.size.x:.3f}', height=f'{self.size.y:.3f}',
|
||||||
|
**self.stroke.svg_attrs(colorscheme.lines), fill=self.fill.color.svg(colorscheme.fill))
|
||||||
|
|
||||||
|
children = []
|
||||||
|
for pin in self.pins:
|
||||||
|
children += pin.to_svg()
|
||||||
|
|
||||||
|
#xform = f'translate({self.at.x:.3f} {self.at.y:.3f})'
|
||||||
|
yield Tag('g', children=children, #transform=xform,
|
||||||
|
fill=self.fill.color.svg(colorscheme.fill),
|
||||||
|
**self.stroke.svg_attrs(colorscheme.lines))
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('rule_area')
|
||||||
|
class RuleArea:
|
||||||
|
polyline: Polyline = None
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('text_box')
|
||||||
|
class TextBox(TextMixin):
|
||||||
|
text: str = None
|
||||||
|
exclude_from_sim: Named(YesNoAtom()) = False
|
||||||
|
at: AtPos = field(default_factory=AtPos)
|
||||||
|
size: Rename(XYCoord) = None
|
||||||
|
margins: Rename(gr.Margins) = None
|
||||||
|
effects: TextEffect = field(default_factory=TextEffect)
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
fill: OmitDefault(Fill) = None
|
||||||
|
effects: TextEffect = field(default_factory=TextEffect)
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
|
||||||
|
def render(self, variables={}, cache=None):
|
||||||
|
yield from gr.TextBox.render(self, variables=variables)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('lib_symbols')
|
||||||
|
class LocalLibrary:
|
||||||
|
symbols: List(Symbol) = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
SUPPORTED_FILE_FORMAT_VERSIONS = [20230620]
|
||||||
|
@sexp_type('kicad_sch')
|
||||||
|
class Schematic:
|
||||||
|
_version: Named(int, name='version') = 20230620
|
||||||
|
generator: Named(str) = 'gerbonara'
|
||||||
|
generator_version: Named(str) = __version__
|
||||||
|
legacy_generator: Named(Array(str), name='host') = None
|
||||||
|
uuid: UUID = field(default_factory=UUID)
|
||||||
|
page_settings: PageSettings = field(default_factory=PageSettings)
|
||||||
|
legacy_page: Named(Array(int), name='page') = None
|
||||||
|
legacy_paper: Named(str, name='paper') = None
|
||||||
|
title_block: TitleBlock = None
|
||||||
|
# The doc says this is expected, but eeschema barfs when it's there.
|
||||||
|
# path: SheetPath = field(default_factory=SheetPath)
|
||||||
|
lib_symbols: LocalLibrary = field(default_factory=list)
|
||||||
|
junctions: List(Junction) = field(default_factory=list)
|
||||||
|
no_connects: List(NoConnect) = field(default_factory=list)
|
||||||
|
rule_areas: List(RuleArea) = field(default_factory=list)
|
||||||
|
netclass_flags: List(NetclassFlag) = field(default_factory=list)
|
||||||
|
bus_aliases: List(BusAlias) = field(default_factory=list)
|
||||||
|
bus_entries: List(BusEntry) = field(default_factory=list)
|
||||||
|
wires: List(Wire) = field(default_factory=list)
|
||||||
|
buses: List(Bus) = field(default_factory=list)
|
||||||
|
images: List(gr.Image) = field(default_factory=list)
|
||||||
|
polylines: List(Polyline) = field(default_factory=list)
|
||||||
|
circles: List(Circle) = field(default_factory=list)
|
||||||
|
rectangles: List(Rectangle) = field(default_factory=list)
|
||||||
|
texts: List(Text) = field(default_factory=list)
|
||||||
|
text_boxes: List(TextBox) = field(default_factory=list)
|
||||||
|
local_labels: List(LocalLabel) = field(default_factory=list)
|
||||||
|
global_labels: List(GlobalLabel) = field(default_factory=list)
|
||||||
|
hierarchical_labels: List(HierarchicalLabel) = field(default_factory=list)
|
||||||
|
symbols: List(SymbolInstance) = field(default_factory=list)
|
||||||
|
subsheets: List(Subsheet) = field(default_factory=list)
|
||||||
|
sheet_instances: Named(Array(SubsheetCrosslinkSheet)) = field(default_factory=list)
|
||||||
|
symbol_instances: Named(Array(SymbolCrosslinkProject)) = field(default_factory=list)
|
||||||
|
embedded_fonts: Named(YesNoAtom()) = False
|
||||||
|
_ : SEXP_END = None
|
||||||
|
original_filename: str = None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def version(self):
|
||||||
|
return self._version
|
||||||
|
|
||||||
|
@version.setter
|
||||||
|
def version(self, value):
|
||||||
|
if value not in SUPPORTED_FILE_FORMAT_VERSIONS:
|
||||||
|
raise FormatError(f'File format version {value} is not supported. Supported versions are {", ".join(map(str, SUPPORTED_FILE_FORMAT_VERSIONS))}.')
|
||||||
|
|
||||||
|
|
||||||
|
def lookup_symbol(self, lib_name, lib_id):
|
||||||
|
key = lib_name or lib_id
|
||||||
|
for sym in self.lib_symbols.symbols:
|
||||||
|
if sym.name == key or sym.raw_name == key:
|
||||||
|
return sym
|
||||||
|
raise KeyError(f'Symbol with {lib_name=} {lib_id=} not found')
|
||||||
|
|
||||||
|
def write(self, filename=None):
|
||||||
|
with open(filename or self.original_filename, 'w') as f:
|
||||||
|
f.write(self.serialize())
|
||||||
|
|
||||||
|
def serialize(self):
|
||||||
|
return build_sexp(sexp(type(self), self)[0])
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def open(kls, pcb_file, *args, **kwargs):
|
||||||
|
return kls.load(Path(pcb_file).read_text(), *args, **kwargs, original_filename=pcb_file)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def load(kls, data, *args, **kwargs):
|
||||||
|
return kls.parse(data, *args, **kwargs)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def elements(self):
|
||||||
|
yield from self.subsheets
|
||||||
|
yield from self.images
|
||||||
|
yield from self.polylines
|
||||||
|
yield from self.symbols
|
||||||
|
yield from self.junctions
|
||||||
|
yield from self.no_connects
|
||||||
|
yield from self.bus_entries
|
||||||
|
yield from self.wires
|
||||||
|
yield from self.buses
|
||||||
|
yield from self.texts
|
||||||
|
yield from self.local_labels
|
||||||
|
yield from self.global_labels
|
||||||
|
yield from self.hierarchical_labels
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
children = []
|
||||||
|
for elem in self.elements:
|
||||||
|
children += elem.to_svg(colorscheme)
|
||||||
|
w, h = KICAD_PAPER_SIZES[self.page_settings.page_format]
|
||||||
|
return setup_svg(children, ((0, 0), (w, h)), pagecolor=colorscheme.background)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
# From: https://jakevdp.github.io/blog/2012/10/07/xkcd-style-plots-in-matplotlib/
|
||||||
|
#def xkcd_line(x, y, xlim=None, ylim=None, mag=1.0, f1=30, f2=0.05, f3=15):
|
||||||
|
def xkcd_line(x, y, xlim=None, ylim=None, mag=1.0, f1=10, f2=0.05, f3=5):
|
||||||
|
"""
|
||||||
|
Mimic a hand-drawn line from (x, y) data
|
||||||
|
|
||||||
|
Parameters
|
||||||
|
----------
|
||||||
|
x, y : array_like
|
||||||
|
arrays to be modified
|
||||||
|
xlim, ylim : data range
|
||||||
|
the assumed plot range for the modification. If not specified,
|
||||||
|
they will be guessed from the data
|
||||||
|
mag : float
|
||||||
|
magnitude of distortions
|
||||||
|
f1, f2, f3 : int, float, int
|
||||||
|
filtering parameters. f1 gives the size of the window, f2 gives
|
||||||
|
the high-frequency cutoff, f3 gives the size of the filter
|
||||||
|
|
||||||
|
Returns
|
||||||
|
-------
|
||||||
|
x, y : ndarrays
|
||||||
|
The modified lines
|
||||||
|
"""
|
||||||
|
import numpy as np
|
||||||
|
from scipy import interpolate, signal
|
||||||
|
|
||||||
|
x = np.asarray(x)
|
||||||
|
y = np.asarray(y)
|
||||||
|
|
||||||
|
# get limits for rescaling
|
||||||
|
if xlim is None:
|
||||||
|
xlim = (x.min(), x.max())
|
||||||
|
if ylim is None:
|
||||||
|
ylim = (y.min(), y.max())
|
||||||
|
|
||||||
|
if xlim[1] == xlim[0]:
|
||||||
|
xlim = ylim
|
||||||
|
|
||||||
|
if ylim[1] == ylim[0]:
|
||||||
|
ylim = xlim
|
||||||
|
|
||||||
|
# scale the data
|
||||||
|
x_scaled = (x - xlim[0]) * 1. / (xlim[1] - xlim[0])
|
||||||
|
y_scaled = (y - ylim[0]) * 1. / (ylim[1] - ylim[0])
|
||||||
|
|
||||||
|
# compute the total distance along the path
|
||||||
|
dx = x_scaled[1:] - x_scaled[:-1]
|
||||||
|
dy = y_scaled[1:] - y_scaled[:-1]
|
||||||
|
dist_tot = np.sum(np.sqrt(dx * dx + dy * dy))
|
||||||
|
|
||||||
|
# number of interpolated points is proportional to the distance
|
||||||
|
Nu = int(50 * dist_tot)
|
||||||
|
u = np.arange(-1, Nu + 1) * 1. / (Nu - 1)
|
||||||
|
|
||||||
|
# interpolate curve at sampled points
|
||||||
|
k = min(3, len(x) - 1)
|
||||||
|
res = interpolate.splprep([x_scaled, y_scaled], s=0, k=k)
|
||||||
|
x_int, y_int = interpolate.splev(u, res[0])
|
||||||
|
|
||||||
|
# we'll perturb perpendicular to the drawn line
|
||||||
|
dx = x_int[2:] - x_int[:-2]
|
||||||
|
dy = y_int[2:] - y_int[:-2]
|
||||||
|
dist = np.sqrt(dx * dx + dy * dy)
|
||||||
|
|
||||||
|
# create a filtered perturbation
|
||||||
|
coeffs = mag * np.random.normal(0, 0.01, len(x_int) - 2)
|
||||||
|
b = signal.firwin(f1, f2 * dist_tot, window=('kaiser', f3))
|
||||||
|
response = signal.lfilter(b, 1, coeffs)
|
||||||
|
|
||||||
|
x_int[1:-1] += response * dy / dist
|
||||||
|
y_int[1:-1] += response * dx / dist
|
||||||
|
|
||||||
|
# un-scale data
|
||||||
|
x_int = x_int[1:-1] * (xlim[1] - xlim[0]) + xlim[0]
|
||||||
|
y_int = y_int[1:-1] * (ylim[1] - ylim[0]) + ylim[0]
|
||||||
|
|
||||||
|
return x_int, y_int
|
||||||
|
|
||||||
|
def wonkify(path):
|
||||||
|
out = []
|
||||||
|
for segment in path.attrs['d'].split('M')[1:]:
|
||||||
|
if 'A' in segment:
|
||||||
|
out.append(segment)
|
||||||
|
continue
|
||||||
|
|
||||||
|
points = segment.split('L')
|
||||||
|
if points[-1].rstrip().endswith('Z'):
|
||||||
|
closed = True
|
||||||
|
points[-1] = points[-1].rstrip()[:-1].rstrip()
|
||||||
|
points.append(points[0])
|
||||||
|
else:
|
||||||
|
closed = False
|
||||||
|
|
||||||
|
pts = []
|
||||||
|
lx, ly = None, None
|
||||||
|
for pt in points:
|
||||||
|
x, y = pt.strip().split()
|
||||||
|
x, y = float(x), float(y)
|
||||||
|
if (x, y) == (lx, ly):
|
||||||
|
continue
|
||||||
|
|
||||||
|
lx, ly = x, y
|
||||||
|
pts.append((x, y))
|
||||||
|
|
||||||
|
if len(pts) == 2:
|
||||||
|
segs = [pts]
|
||||||
|
|
||||||
|
else:
|
||||||
|
seg = [pts[0]]
|
||||||
|
segs = [seg]
|
||||||
|
for p0, p1, p2 in zip(pts[0::], pts[1::], pts[2::]):
|
||||||
|
dx1, dy1 = p1[0] - p0[0], p1[1] - p0[1]
|
||||||
|
dx2, dy2 = p2[0] - p1[0], p2[1] - p1[1]
|
||||||
|
l1, l2 = math.hypot(dx1, dy1), math.hypot(dx2, dy2)
|
||||||
|
a1, a2 = math.atan2(dy1, dx1), math.atan2(dy2, dx2)
|
||||||
|
da = (a2 - a1 + math.pi) % (2*math.pi) - math.pi
|
||||||
|
if abs(da) > math.pi/4 and l1+l2 > 3:
|
||||||
|
seg.append(p1)
|
||||||
|
seg = [p1, p2]
|
||||||
|
segs.append(seg)
|
||||||
|
seg.append(p1)
|
||||||
|
seg.append(p2)
|
||||||
|
|
||||||
|
for seg in segs:
|
||||||
|
xs, ys = [x for x, y in seg], [y for x, y in seg]
|
||||||
|
xs, ys = xkcd_line(xs, ys)
|
||||||
|
d = ' L '.join(f'{x:.3f} {y:.3f}' for x, y in zip(xs, ys))
|
||||||
|
if closed:
|
||||||
|
d += ' Z'
|
||||||
|
out.append(d)
|
||||||
|
|
||||||
|
path.attrs['d'] = ' '.join(f'M {seg}' for seg in out)
|
||||||
|
|
||||||
|
|
||||||
|
def postprocess(tag):
|
||||||
|
if tag.name == 'path':
|
||||||
|
wonkify(tag)
|
||||||
|
else:
|
||||||
|
for child in tag.children:
|
||||||
|
postprocess(child)
|
||||||
|
return tag
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
import sys
|
||||||
|
from ...layers import LayerStack
|
||||||
|
from .tmtheme import *
|
||||||
|
sch = Schematic.open(sys.argv[1])
|
||||||
|
print('Loaded schematic with', len(sch.wires), 'wires and', len(sch.symbols), 'symbols.')
|
||||||
|
for subsh in sch.subsheets:
|
||||||
|
subsh = subsh.open()
|
||||||
|
print('Loaded sub-sheet with', len(subsh.wires), 'wires and', len(subsh.symbols), 'symbols.')
|
||||||
|
|
||||||
|
sch.write('/tmp/test.kicad_sch')
|
||||||
|
for p in Path('/tmp').glob('*.tmTheme'):
|
||||||
|
cs = TmThemeSchematic(p.read_text())
|
||||||
|
Path(f'/tmp/test-{p.stem}.svg').write_text(str(postprocess(sch.to_svg(cs))))
|
||||||
|
for p in Path('/tmp').glob('*.sublime-color-scheme'):
|
||||||
|
cs = SublimeSchematic(p.read_text())
|
||||||
|
Path(f'/tmp/test-{p.stem}.svg').write_text(str(postprocess(sch.to_svg(cs))))
|
||||||
|
|
||||||
13
src/gerbonara/cad/kicad/schematic_colors.py
Normal file
|
|
@ -0,0 +1,13 @@
|
||||||
|
|
||||||
|
class Colorscheme:
|
||||||
|
class KiCad:
|
||||||
|
wire = 'black'
|
||||||
|
bus = 'black'
|
||||||
|
lines = 'black'
|
||||||
|
no_connect = 'black'
|
||||||
|
text = 'black'
|
||||||
|
values = 'black'
|
||||||
|
labels = 'black'
|
||||||
|
fill = '#cccccc'
|
||||||
|
background = 'white'
|
||||||
|
|
||||||
|
|
@ -64,7 +64,7 @@ term_regex = r"""(?mx)
|
||||||
(\))|
|
(\))|
|
||||||
([+-]?\d+\.\d+(?=[\s\)]))|
|
([+-]?\d+\.\d+(?=[\s\)]))|
|
||||||
(\-?\d+(?=[\s\)]))|
|
(\-?\d+(?=[\s\)]))|
|
||||||
([^0-9"\s()][^"\s)]*)
|
([^"\s()][^"\s)]*)
|
||||||
)"""
|
)"""
|
||||||
|
|
||||||
|
|
||||||
404
src/gerbonara/cad/kicad/sexp_mapper.py
Normal file
|
|
@ -0,0 +1,404 @@
|
||||||
|
|
||||||
|
import textwrap
|
||||||
|
|
||||||
|
import copy
|
||||||
|
from dataclasses import MISSING, replace, fields
|
||||||
|
from .sexp import *
|
||||||
|
|
||||||
|
|
||||||
|
SEXP_END = type('SEXP_END', (), {})
|
||||||
|
|
||||||
|
|
||||||
|
class AtomChoice:
|
||||||
|
def __init__(self, *choices):
|
||||||
|
self.choices = choices
|
||||||
|
|
||||||
|
def __contains__(self, value):
|
||||||
|
return value in self.choices
|
||||||
|
|
||||||
|
def __atoms__(self):
|
||||||
|
return self.choices
|
||||||
|
|
||||||
|
def __map__(self, obj, parent=None):
|
||||||
|
obj, = obj
|
||||||
|
if obj not in self:
|
||||||
|
raise TypeError(f'Invalid atom {obj} for {type(self)}, valid choices are: {", ".join(map(str, self.choices))}')
|
||||||
|
return obj
|
||||||
|
|
||||||
|
def __sexp__(self, value):
|
||||||
|
yield value
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
choices = '|'.join(map(str, self.choices))
|
||||||
|
return f'AtomChoice({choices})'
|
||||||
|
|
||||||
|
|
||||||
|
class Flag:
|
||||||
|
def __init__(self, atom=None, invert=None):
|
||||||
|
self.atom, self.invert = atom, invert
|
||||||
|
|
||||||
|
def __bind_field__(self, field):
|
||||||
|
if self.atom is None:
|
||||||
|
self.atom = Atom(field.name)
|
||||||
|
if self.invert is None:
|
||||||
|
self.invert = bool(field.default)
|
||||||
|
|
||||||
|
def __atoms__(self):
|
||||||
|
return [self.atom]
|
||||||
|
|
||||||
|
def __map__(self, obj, parent=None):
|
||||||
|
return not self.invert
|
||||||
|
|
||||||
|
def __sexp__(self, value):
|
||||||
|
if bool(value) == (not self.invert):
|
||||||
|
yield self.atom
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
if self.invert is not None:
|
||||||
|
return f'Flag({self.atom}/{self.invert})'
|
||||||
|
return f'Flag({self.atom})'
|
||||||
|
|
||||||
|
|
||||||
|
def sexp(t, v):
|
||||||
|
try:
|
||||||
|
if v is None:
|
||||||
|
return []
|
||||||
|
elif t in (int, float, str, Atom):
|
||||||
|
return [t(v)]
|
||||||
|
elif hasattr(t, '__sexp__'):
|
||||||
|
return list(t.__sexp__(v))
|
||||||
|
elif isinstance(t, list):
|
||||||
|
t, = t
|
||||||
|
return [sexp(t, elem) for elem in v]
|
||||||
|
else:
|
||||||
|
raise TypeError(f'Python type {t} of value {v!r} has no defined s-expression serialization')
|
||||||
|
|
||||||
|
except MappingError as e:
|
||||||
|
raise e
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
raise MappingError(f'Error trying to serialize {textwrap.shorten(str(v), width=120)} into type {t}', t, v) from e
|
||||||
|
|
||||||
|
|
||||||
|
class MappingError(TypeError):
|
||||||
|
def __init__(self, msg, t, sexp):
|
||||||
|
super().__init__(msg)
|
||||||
|
self.t, self.sexp = t, sexp
|
||||||
|
|
||||||
|
def map_sexp(t, v, parent=None, path=''):
|
||||||
|
try:
|
||||||
|
if t is not Atom and hasattr(t, '__map__'):
|
||||||
|
return t.__map__(v, parent=parent)
|
||||||
|
|
||||||
|
elif t in (int, float, str, Atom):
|
||||||
|
v, = v
|
||||||
|
if not isinstance(v, t):
|
||||||
|
types = set({type(v), t})
|
||||||
|
if types == {int, float} or types == {str, Atom}:
|
||||||
|
v = t(v)
|
||||||
|
else:
|
||||||
|
raise TypeError(f'Cannot map s-expression value {v} of type {type(v)} to Python type {t}')
|
||||||
|
return v
|
||||||
|
|
||||||
|
elif isinstance(t, list):
|
||||||
|
t, = t
|
||||||
|
return [map_sexp(t, elem, parent=parent, path=f'{path}/{t}') for elem in v]
|
||||||
|
|
||||||
|
else:
|
||||||
|
raise TypeError(f'Python type {t} has no defined s-expression deserialization')
|
||||||
|
|
||||||
|
except MappingError as e:
|
||||||
|
raise e
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
raise MappingError(f'Error at {path} trying to map {textwrap.shorten(str(v), width=60)} into type {t}', t, v) from e
|
||||||
|
|
||||||
|
|
||||||
|
class WrapperType:
|
||||||
|
def __init__(self, next_type):
|
||||||
|
self.next_type = next_type
|
||||||
|
|
||||||
|
def __bind_field__(self, field):
|
||||||
|
self.field = field
|
||||||
|
if self.next_type is not Atom:
|
||||||
|
getattr(self.next_type, '__bind_field__', lambda x: None)(field)
|
||||||
|
|
||||||
|
def __atoms__(self):
|
||||||
|
if hasattr(self, 'name_atom'):
|
||||||
|
return [self.name_atom]
|
||||||
|
elif self.next_type is Atom:
|
||||||
|
return []
|
||||||
|
else:
|
||||||
|
return getattr(self.next_type, '__atoms__', lambda: [])()
|
||||||
|
|
||||||
|
class Named(WrapperType):
|
||||||
|
def __init__(self, next_type, name=None, omit_empty=True):
|
||||||
|
super().__init__(next_type)
|
||||||
|
self.name_atom = Atom(name) if name else None
|
||||||
|
self.omit_empty = omit_empty
|
||||||
|
|
||||||
|
def __bind_field__(self, field):
|
||||||
|
if self.next_type is not Atom:
|
||||||
|
getattr(self.next_type, '__bind_field__', lambda x: None)(field)
|
||||||
|
if self.name_atom is None:
|
||||||
|
self.name_atom = Atom(field.name)
|
||||||
|
|
||||||
|
def __map__(self, obj, parent=None, path=''):
|
||||||
|
k, *obj = obj
|
||||||
|
if self.next_type in (int, float, str, Atom) or isinstance(self.next_type, AtomChoice):
|
||||||
|
return map_sexp(self.next_type, [*obj], parent=parent, path=f'{path}/{self.name_atom}')
|
||||||
|
else:
|
||||||
|
return map_sexp(self.next_type, obj, parent=parent, path=f'{path}/{self.name_atom}')
|
||||||
|
|
||||||
|
def __sexp__(self, value):
|
||||||
|
value = sexp(self.next_type, value)
|
||||||
|
if value is None:
|
||||||
|
return
|
||||||
|
|
||||||
|
if self.omit_empty and not value:
|
||||||
|
return
|
||||||
|
|
||||||
|
yield [self.name_atom, *value]
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return f'Named={self.name_atom}({self.next_type})'
|
||||||
|
|
||||||
|
|
||||||
|
class Rename(WrapperType):
|
||||||
|
def __init__(self, next_type, name=None):
|
||||||
|
super().__init__(next_type)
|
||||||
|
self.name_atom = Atom(name) if name else None
|
||||||
|
|
||||||
|
def __bind_field__(self, field):
|
||||||
|
if self.name_atom is None:
|
||||||
|
self.name_atom = Atom(field.name)
|
||||||
|
if hasattr(self.next_type, '__bind_field__'):
|
||||||
|
self.next_type.__bind_field__(field)
|
||||||
|
|
||||||
|
def __map__(self, obj, parent=None, path=''):
|
||||||
|
return map_sexp(self.next_type, obj, parent=parent, path=f'{path}/{self.name_atom}')
|
||||||
|
|
||||||
|
def __sexp__(self, value):
|
||||||
|
value, = sexp(self.next_type, value)
|
||||||
|
if self.next_type in (str, float, int, Atom):
|
||||||
|
yield [self.name_atom, *value]
|
||||||
|
else:
|
||||||
|
key, *rest = value
|
||||||
|
yield [self.name_atom, *rest]
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return f'Rename={self.name_atom}({self.next_type})'
|
||||||
|
|
||||||
|
|
||||||
|
class OmitDefault(WrapperType):
|
||||||
|
def __bind_field__(self, field):
|
||||||
|
getattr(self.next_type, '__bind_field__', lambda x: None)(field)
|
||||||
|
if field.default_factory != MISSING:
|
||||||
|
self.default = field.default_factory()
|
||||||
|
else:
|
||||||
|
self.default = field.default
|
||||||
|
|
||||||
|
def __map__(self, obj, parent=None, path=''):
|
||||||
|
return map_sexp(self.next_type, obj, parent=parent, path=path)
|
||||||
|
|
||||||
|
def __sexp__(self, value):
|
||||||
|
if value != self.default:
|
||||||
|
yield from sexp(self.next_type, value)
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return f'OmitDefault({self.field})'
|
||||||
|
|
||||||
|
|
||||||
|
class YesNoAtom:
|
||||||
|
def __init__(self, yes=Atom.yes, no=Atom.no):
|
||||||
|
self.yes, self.no = yes, no
|
||||||
|
|
||||||
|
def __map__(self, value, parent=None):
|
||||||
|
if not value: # compatibility with legacy flag style
|
||||||
|
return False
|
||||||
|
value, = value
|
||||||
|
return value == self.yes
|
||||||
|
|
||||||
|
def __sexp__(self, value):
|
||||||
|
yield self.yes if value else self.no
|
||||||
|
|
||||||
|
|
||||||
|
class LegacyCompatibleFlag:
|
||||||
|
'''Variant of YesNoAtom that accepts both the `(flag <yes/no>)` variant and the bare `flag` variant for compatibility.'''
|
||||||
|
|
||||||
|
def __init__(self, yes=Atom.yes, no=Atom.no, value_when_empty=True):
|
||||||
|
self.yes, self.no = yes, no
|
||||||
|
self.value_when_empty = value_when_empty
|
||||||
|
|
||||||
|
def __map__(self, value, parent=None):
|
||||||
|
if value == []:
|
||||||
|
return self.value_when_empty
|
||||||
|
|
||||||
|
value, = value
|
||||||
|
return value == self.yes
|
||||||
|
|
||||||
|
def __sexp__(self, value):
|
||||||
|
yield self.yes if value else self.no
|
||||||
|
|
||||||
|
|
||||||
|
class Wrap(WrapperType):
|
||||||
|
def __map__(self, value, parent=None, path=''):
|
||||||
|
value, = value
|
||||||
|
return map_sexp(self.next_type, value, parent=parent, path=path)
|
||||||
|
|
||||||
|
def __sexp__(self, value):
|
||||||
|
for inner in sexp(self.next_type, value):
|
||||||
|
yield [inner]
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return f'Wrap({self.next_type})'
|
||||||
|
|
||||||
|
|
||||||
|
class Array(WrapperType):
|
||||||
|
def __map__(self, value, parent=None, path=''):
|
||||||
|
return [map_sexp(self.next_type, [elem], parent=parent, path=path) for elem in value]
|
||||||
|
|
||||||
|
def __sexp__(self, value):
|
||||||
|
for e in value:
|
||||||
|
yield from sexp(self.next_type, e)
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return f'Array({self.next_type})'
|
||||||
|
|
||||||
|
|
||||||
|
class Untagged(WrapperType):
|
||||||
|
def __map__(self, value, parent=None, path=''):
|
||||||
|
value, = value
|
||||||
|
return self.next_type.__map__([self.next_type.name_atom, *value], parent=parent, path=path)
|
||||||
|
|
||||||
|
def __sexp__(self, value):
|
||||||
|
for inner in sexp(self.next_type, value):
|
||||||
|
_tag, *rest = inner
|
||||||
|
yield rest
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return f'Untagged({self.next_type})'
|
||||||
|
|
||||||
|
class List(WrapperType):
|
||||||
|
def __bind_field__(self, field):
|
||||||
|
self.attr = field.name
|
||||||
|
|
||||||
|
def __map__(self, value, parent, path=''):
|
||||||
|
l = getattr(parent, self.attr, [])
|
||||||
|
mapped = map_sexp(self.next_type, value, parent=parent, path=f'{path}/{self.attr}')
|
||||||
|
l.append(mapped)
|
||||||
|
setattr(parent, self.attr, l)
|
||||||
|
|
||||||
|
def __sexp__(self, value):
|
||||||
|
for elem in value:
|
||||||
|
yield from sexp(self.next_type, elem)
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return f'List@{self.attr}({self.next_type})'
|
||||||
|
|
||||||
|
|
||||||
|
class _SexpTemplate:
|
||||||
|
@staticmethod
|
||||||
|
def __atoms__(kls):
|
||||||
|
return [kls.name_atom]
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def __map__(kls, value, *args, parent=None, path='', **kwargs):
|
||||||
|
positional = iter(kls.positional)
|
||||||
|
inst = kls(*args, **kwargs)
|
||||||
|
|
||||||
|
for v in value[1:]: # skip key
|
||||||
|
if isinstance(v, Atom) and v in kls.keys:
|
||||||
|
name, etype = kls.keys[v]
|
||||||
|
mapped = map_sexp(etype, [v], parent=inst, path=f'{path}/{kls.name_atom}')
|
||||||
|
if mapped is not None:
|
||||||
|
setattr(inst, name, mapped)
|
||||||
|
|
||||||
|
elif isinstance(v, list):
|
||||||
|
key = v[0]
|
||||||
|
if key in kls.keys:
|
||||||
|
name, etype = kls.keys[key]
|
||||||
|
mapped = map_sexp(etype, v, parent=inst, path=f'{path}/{kls.name_atom}')
|
||||||
|
if mapped is not None:
|
||||||
|
setattr(inst, name, mapped)
|
||||||
|
|
||||||
|
elif hasattr(inst, '__catchall__'):
|
||||||
|
inst.__catchall__(v, path=f'{path}/{kls.name_atom}')
|
||||||
|
|
||||||
|
else:
|
||||||
|
#print('class has keys:')
|
||||||
|
#print('\n'.join(map(str, kls.keys)))
|
||||||
|
raise TypeError(f'Unhandled keyed argument {v!r} while parsing {kls}')
|
||||||
|
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
pos_key = next(positional)
|
||||||
|
setattr(inst, pos_key.name, v)
|
||||||
|
except StopIteration:
|
||||||
|
raise TypeError(f'Unhandled positional argument {v!r} while parsing {kls}')
|
||||||
|
|
||||||
|
getattr(inst, '__after_parse__', lambda x: None)(parent)
|
||||||
|
return inst
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def __sexp__(kls, value):
|
||||||
|
getattr(value, '__before_sexp__', lambda: None)()
|
||||||
|
|
||||||
|
out = [kls.name_atom]
|
||||||
|
for f in fields(kls):
|
||||||
|
if f.type is SEXP_END:
|
||||||
|
break
|
||||||
|
out += sexp(f.type, getattr(value, f.name))
|
||||||
|
yield out
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def parse(kls, data, *args, **kwargs):
|
||||||
|
return kls.__map__(parse_sexp(data), *args, **kwargs)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def sexp(self):
|
||||||
|
return next(self.__sexp__(self))
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def __deepcopy__(self, memo):
|
||||||
|
return replace(self, **{f.name: copy.deepcopy(getattr(self, f.name), memo) for f in fields(self) if not f.kw_only})
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def __copy__(self):
|
||||||
|
# Even during a shallow copy, we need to deep copy any fields whose types have a __before_sexp__ method to avoid
|
||||||
|
# those from being called more than once on the same object.
|
||||||
|
return replace(self, **{f.name: copy.copy(getattr(self, f.name)) for f in fields(self) if not f.kw_only and hasattr(f.type, '__before_sexp__')})
|
||||||
|
|
||||||
|
def sexp_type(name=None):
|
||||||
|
def register(cls):
|
||||||
|
cls = dataclass(cls)
|
||||||
|
cls.name_atom = Atom(name) if name is not None else None
|
||||||
|
for key in '__sexp__', '__map__', '__atoms__', 'parse':
|
||||||
|
if not hasattr(cls, key):
|
||||||
|
setattr(cls, key, classmethod(getattr(_SexpTemplate, key)))
|
||||||
|
|
||||||
|
for key in 'sexp', '__deepcopy__', '__copy__':
|
||||||
|
if not hasattr(cls, key):
|
||||||
|
setattr(cls, key, getattr(_SexpTemplate, key))
|
||||||
|
|
||||||
|
cls.positional = []
|
||||||
|
cls.keys = {}
|
||||||
|
for f in fields(cls):
|
||||||
|
f_type = f.type
|
||||||
|
if f_type is SEXP_END:
|
||||||
|
break
|
||||||
|
|
||||||
|
if hasattr(f_type, '__bind_field__'):
|
||||||
|
f_type.__bind_field__(f)
|
||||||
|
|
||||||
|
atoms = getattr(f_type, '__atoms__', lambda: [])
|
||||||
|
atoms = list(atoms())
|
||||||
|
for atom in atoms:
|
||||||
|
cls.keys[atom] = (f.name, f_type)
|
||||||
|
if not atoms:
|
||||||
|
cls.positional.append(f)
|
||||||
|
|
||||||
|
return cls
|
||||||
|
return register
|
||||||
|
|
||||||
|
|
||||||
631
src/gerbonara/cad/kicad/symbols.py
Normal file
|
|
@ -0,0 +1,631 @@
|
||||||
|
"""
|
||||||
|
Library for processing KiCad's symbol files.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import json
|
||||||
|
import string
|
||||||
|
import math
|
||||||
|
import re
|
||||||
|
import sys
|
||||||
|
import itertools
|
||||||
|
from fnmatch import fnmatch
|
||||||
|
from collections import defaultdict
|
||||||
|
from dataclasses import field
|
||||||
|
from pathlib import Path
|
||||||
|
from typing import Any, Dict, List, Optional, Tuple
|
||||||
|
|
||||||
|
from .sexp import *
|
||||||
|
from .sexp_mapper import *
|
||||||
|
from .base_types import *
|
||||||
|
from ...utils import rotate_point, Tag, arc_bounds
|
||||||
|
from ... import __version__
|
||||||
|
from ...newstroke import Newstroke
|
||||||
|
from .schematic_colors import *
|
||||||
|
from .primitives import kicad_mid_to_center_arc, Margins
|
||||||
|
|
||||||
|
|
||||||
|
PIN_ETYPE = AtomChoice(Atom.input, Atom.output, Atom.bidirectional, Atom.tri_state, Atom.passive, Atom.free,
|
||||||
|
Atom.unspecified, Atom.power_in, Atom.power_out, Atom.open_collector, Atom.open_emitter,
|
||||||
|
Atom.no_connect, Atom.unconnected)
|
||||||
|
|
||||||
|
|
||||||
|
PIN_STYLE = AtomChoice(Atom.line, Atom.inverted, Atom.clock, Atom.inverted_clock, Atom.input_low, Atom.clock_low,
|
||||||
|
Atom.output_low, Atom.edge_clock_high, Atom.non_logic)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('alternate')
|
||||||
|
class AltFunction:
|
||||||
|
name: str = None
|
||||||
|
etype: PIN_ETYPE = Atom.unspecified
|
||||||
|
shape: PIN_STYLE = Atom.line
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('__styled_text')
|
||||||
|
class StyledText:
|
||||||
|
value: str = None
|
||||||
|
effects: TextEffect = field(default_factory=TextEffect)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('pin')
|
||||||
|
class Pin:
|
||||||
|
etype: PIN_ETYPE = Atom.unspecified
|
||||||
|
style: PIN_STYLE = Atom.line
|
||||||
|
at: AtPos = field(default_factory=AtPos)
|
||||||
|
length: Named(float) = 2.54
|
||||||
|
hide: OmitDefault(Named(YesNoAtom())) = False
|
||||||
|
name: Rename(StyledText) = field(default_factory=StyledText)
|
||||||
|
number: Rename(StyledText) = field(default_factory=StyledText)
|
||||||
|
alternates: List(AltFunction) = field(default_factory=list)
|
||||||
|
_: SEXP_END = None
|
||||||
|
unit: object = None
|
||||||
|
|
||||||
|
def __after_parse__(self, parent=None):
|
||||||
|
self.unit = parent
|
||||||
|
|
||||||
|
@property
|
||||||
|
def direction(self):
|
||||||
|
return {0: 'R', 90: 'U', 180: 'L', 270: 'D'}.get(self.at.rotation, 'R')
|
||||||
|
|
||||||
|
@direction.setter
|
||||||
|
def direction(self, value):
|
||||||
|
self.at.rotation = {0: 'R', 90: 'U', 180: 'L', 270: 'D'}[value[0].upper()]
|
||||||
|
|
||||||
|
def bounding_box(self, default=None):
|
||||||
|
font = Newstroke.load()
|
||||||
|
strokes = list(font.render(self.name, size=2.54))
|
||||||
|
min_x = min(x for st in strokes for x, y in st)
|
||||||
|
min_y = min(y for st in strokes for x, y in st)
|
||||||
|
max_x = max(x for st in strokes for x, y in st)
|
||||||
|
max_y = max(y for st in strokes for x, y in st)
|
||||||
|
w, h = max_x - min_x, max_y - min_y
|
||||||
|
l = self.length + 0.2 + w
|
||||||
|
|
||||||
|
x1, y1 = x2, y2 = self.at.x, self.at.y
|
||||||
|
if self.at.rotation == 0:
|
||||||
|
x2 += w
|
||||||
|
y1 -= h/2
|
||||||
|
y2 += h/2
|
||||||
|
if self.at.rotation == 90:
|
||||||
|
y2 += w
|
||||||
|
x1 -= h/2
|
||||||
|
x2 += h/2
|
||||||
|
if self.at.rotation == 180:
|
||||||
|
x1 -= w
|
||||||
|
y1 -= h/2
|
||||||
|
y2 += h/2
|
||||||
|
if self.at.rotation == 270:
|
||||||
|
y1 -= w
|
||||||
|
x1 -= h/2
|
||||||
|
x2 += h/2
|
||||||
|
else:
|
||||||
|
raise ValueError(f'Invalid pin rotation {self.at.rotation}')
|
||||||
|
|
||||||
|
return (x1, y1), (x2, y2)
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme, p_mirror, p_rotation):
|
||||||
|
if self.hide:
|
||||||
|
return
|
||||||
|
|
||||||
|
psx, psy = (-1 if p_mirror.x else 1), (-1 if p_mirror.y else 1)
|
||||||
|
x1, y1 = self.at.x, self.at.y
|
||||||
|
x2, y2 = self.at.x+self.length, self.at.y
|
||||||
|
if p_mirror.y:
|
||||||
|
p_xf = f'scale(-1 -1)'
|
||||||
|
elif p_mirror.x:
|
||||||
|
p_xf = f'scale(1 1)'
|
||||||
|
else:
|
||||||
|
p_xf = f'scale(1 -1)'
|
||||||
|
p_xf += f'rotate({p_rotation})'
|
||||||
|
xform = {'transform': f'{p_xf} translate({self.at.x:.3f} {self.at.y:.3f}) rotate({self.at.rotation})'}
|
||||||
|
style = {'stroke_width': 0.254, 'stroke': colorscheme.lines, 'stroke_linecap': 'round'}
|
||||||
|
|
||||||
|
yield Tag('path', **xform, **style, d=f'M 0 0 L {self.length:.3f} 0')
|
||||||
|
|
||||||
|
eps = 1
|
||||||
|
for tag in {
|
||||||
|
'line': [],
|
||||||
|
'inverted': [
|
||||||
|
Tag('circle', **xform, **style, cx=x2-eps/3-0.2, cy=y2, r=eps/3)],
|
||||||
|
'clock': [
|
||||||
|
Tag('path', **xform, **style, d=f'M {x2} {y2-eps/2} L {x2+eps/2} {y2} L {x2} {y2+eps/2}')],
|
||||||
|
'inverted_clock': [
|
||||||
|
Tag('circle', **xform, **style, cx=x2-eps/3-0.2, cy=y2, r=eps/3),
|
||||||
|
Tag('path', **xform, **style, d=f'M {x2} {y2-eps/2} L {x2+eps/2} {y2} L {x2} {y2+eps/2}')],
|
||||||
|
'input_low': [
|
||||||
|
Tag('path', **xform, **style, d=f'M {x2} {y2} L {x2-eps} {y2-eps} L {x2-eps} {y2}')],
|
||||||
|
'clock_low': [
|
||||||
|
Tag('path', **xform, **style, d=f'M {x2} {y2} L {x2-eps} {y2-eps} L {x2-eps} {y2}'),
|
||||||
|
Tag('path', **xform, **style, d=f'M {x2} {y2-eps/2} L {x2+eps/2} {y2} L {x2} {y2+eps/2}')],
|
||||||
|
'output_low': [
|
||||||
|
Tag('path', **xform, **style, d=f'M {x2} {y2-eps} L {x2-eps} {y2}')],
|
||||||
|
'edge_clock_high': [
|
||||||
|
Tag('path', **xform, **style, d=f'M {x2} {y2} L {x2-eps} {y2-eps} L {x2-eps} {y2}'),
|
||||||
|
Tag('path', **xform, **style, d=f'M {x2} {y2-eps/2} L {x2+eps/2} {y2} L {x2} {y2+eps/2}')],
|
||||||
|
'non_logic': [
|
||||||
|
Tag('path', **xform, **style, d=f'M {x2-eps/2} {y2-eps/2} L {x2+eps/2} {y2+eps/2}'),
|
||||||
|
Tag('path', **xform, **style, d=f'M {x2-eps/2} {y2+eps/2} L {x2+eps/2} {y2-eps/2}')],
|
||||||
|
# FIXME...
|
||||||
|
}.get(self.style, []):
|
||||||
|
yield tag
|
||||||
|
|
||||||
|
rot = self.at.rotation + p_rotation
|
||||||
|
trot = self.at.rotation
|
||||||
|
ax, ay = self.length+0.2, 0
|
||||||
|
ax, ay = rotate_point(ax, ay, math.radians(-self.at.rotation))
|
||||||
|
|
||||||
|
#lx, ly = self.at.x, -self.at.y
|
||||||
|
#lx, ly = rotate_point(lx, ly, math.radians(p_rotation))
|
||||||
|
#if p_mirror.y:
|
||||||
|
# lx, ly = -lx, ly
|
||||||
|
#elif p_mirror.x:
|
||||||
|
# lx, ly = lx, -ly
|
||||||
|
#yield Tag('circle', cx=lx, cy=ly, r='0.5', stroke='blue', stroke_width='0.1', fill='none', z_index='100')
|
||||||
|
|
||||||
|
lx, ly = self.at.x + ax, -self.at.y - ay
|
||||||
|
lx, ly = rotate_point(lx, ly, math.radians(p_rotation))
|
||||||
|
if p_mirror.y:
|
||||||
|
lx, ly = -lx, ly
|
||||||
|
elif p_mirror.x:
|
||||||
|
lx, ly = lx, -ly
|
||||||
|
#yield Tag('circle', cx=lx, cy=ly, r='0.5', stroke='red', stroke_width='0.1', fill='none', z_index='100')
|
||||||
|
|
||||||
|
h_align = 'left'
|
||||||
|
if p_mirror.y:
|
||||||
|
if trot in (0, 180):
|
||||||
|
trot = 180 - trot
|
||||||
|
elif p_mirror.x:
|
||||||
|
if p_rotation == 0:
|
||||||
|
if trot in (90, 270):
|
||||||
|
trot = 360-trot
|
||||||
|
else:
|
||||||
|
if trot in (0, 180):
|
||||||
|
trot = 180 - trot
|
||||||
|
frot = (trot + p_rotation)%360
|
||||||
|
sx, sy = 1, 1
|
||||||
|
|
||||||
|
if frot == 180:
|
||||||
|
frot = 0
|
||||||
|
h_align = 'right'
|
||||||
|
elif frot == 270:
|
||||||
|
frot = 90
|
||||||
|
h_align = 'right'
|
||||||
|
|
||||||
|
font = Newstroke.load()
|
||||||
|
if self.name.value != '~' and not self.unit.symbol.pin_names.hide:
|
||||||
|
yield font.render_svg(self.name.value,
|
||||||
|
size=self.name.effects.font.size.y or 1.27,
|
||||||
|
x0=0,
|
||||||
|
y0=0,
|
||||||
|
h_align=h_align,
|
||||||
|
v_align='middle',
|
||||||
|
rotation=-frot,
|
||||||
|
stroke=colorscheme.pin_names,
|
||||||
|
transform=f'translate({lx:.3f} {ly:.3f})',
|
||||||
|
scale=(sx, sy),
|
||||||
|
mirror=(False, False),
|
||||||
|
)
|
||||||
|
|
||||||
|
if self.number.value != '~' and not self.unit.symbol.pin_numbers.hide:
|
||||||
|
yield font.render_svg(self.number.value,
|
||||||
|
size=self.number.effects.font.size.y or 1.27,
|
||||||
|
x0=-0.4 if h_align == 'left' else 0.4,
|
||||||
|
y0=-0.4,
|
||||||
|
h_align={'left': 'right', 'right': 'left'}[h_align],
|
||||||
|
v_align='bottom',
|
||||||
|
rotation=-frot,
|
||||||
|
stroke=colorscheme.pin_numbers,
|
||||||
|
scale=(sx, sy),
|
||||||
|
transform=f'translate({lx:.3f} {ly:.3f})',
|
||||||
|
mirror=(False, False),
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('fill')
|
||||||
|
class Fill:
|
||||||
|
type: Named(AtomChoice(Atom.none, Atom.outline, Atom.background)) = Atom.none
|
||||||
|
|
||||||
|
def svg(self, fg, bg):
|
||||||
|
if self.type == 'outline':
|
||||||
|
return fg
|
||||||
|
elif self.type == 'background':
|
||||||
|
return bg
|
||||||
|
else:
|
||||||
|
return 'none'
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('circle')
|
||||||
|
class Circle:
|
||||||
|
center: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
radius: Named(float) = 0.0
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
fill: Fill = field(default_factory=Fill)
|
||||||
|
|
||||||
|
def bounding_box(self, default=None):
|
||||||
|
x, y, r = self.center.x, self.center.y, self.radius
|
||||||
|
return (x-r, y-r), (x+r, y+r)
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
yield Tag('circle', cx=f'{self.center.x:.3f}', cy=f'{self.center.y:.3f}', r=f'{self.radius:.3f}',
|
||||||
|
fill=self.fill.svg(colorscheme.lines, colorscheme.fill),
|
||||||
|
**self.stroke.svg_attrs(colorscheme.lines))
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('radius')
|
||||||
|
class ArcRadius:
|
||||||
|
at: AtPos = field(default_factory=AtPos)
|
||||||
|
length: Named(float) = 0.0
|
||||||
|
angles: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('arc')
|
||||||
|
class Arc:
|
||||||
|
start: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
mid: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
end: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
radius: ArcRadius = None
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
fill: Fill = field(default_factory=Fill)
|
||||||
|
|
||||||
|
def bounding_box(self, default=None):
|
||||||
|
(cx, cy), r = kicad_mid_to_center_arc(self.mid, self.start, self.end)
|
||||||
|
x1, y1 = self.start.x, self.start.y
|
||||||
|
x2, y2 = self.mid.x-x1, self.mid.y-x2
|
||||||
|
x3, y3 = (self.end.x - x1)/2, (self.end.y - y1)/2
|
||||||
|
clockwise = math.atan2(x2*y3-x3*y2, x2*x3+y2*y3) > 0
|
||||||
|
return arc_bounds(x1, y1, self.end.x, self.end.y, cx, cy, clockwise)
|
||||||
|
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
(cx, cy), r = kicad_mid_to_center_arc(self.mid, self.start, self.end)
|
||||||
|
|
||||||
|
x1r = self.start.x - cx
|
||||||
|
y1r = self.start.y - cy
|
||||||
|
x2r = self.end.x - cx
|
||||||
|
y2r = self.end.y - cy
|
||||||
|
a1 = math.atan2(x1r, y1r)
|
||||||
|
a2 = math.atan2(x2r, y2r)
|
||||||
|
da = (a2 - a1 + math.pi) % (2*math.pi) - math.pi
|
||||||
|
|
||||||
|
large_arc = int(da > math.pi)
|
||||||
|
d = f'M {self.start.x:.3f} {self.start.y:.3f} A {r:.3f} {r:.3f} 0 {large_arc} 0 {self.end.x:.3f} {self.end.y:.3f}'
|
||||||
|
yield Tag('path', d=d, fill=self.fill.svg(colorscheme.lines, colorscheme.fill),
|
||||||
|
**self.stroke.svg_attrs(colorscheme.lines))
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('polyline')
|
||||||
|
class Polyline:
|
||||||
|
pts: PointList = field(default_factory=PointList)
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
fill: Fill = field(default_factory=Fill)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def points(self):
|
||||||
|
return self.pts.xy
|
||||||
|
|
||||||
|
@points.setter
|
||||||
|
def points(self, value):
|
||||||
|
self.pts.xy = value
|
||||||
|
|
||||||
|
@property
|
||||||
|
def closed(self):
|
||||||
|
# if the last and first point are the same, we consider the polyline closed
|
||||||
|
# a closed triangle will have 4 points (A-B-C-A) stored in the list of points
|
||||||
|
return len(self.points) > 3 and self.points[0].isclose(self.points[-1])
|
||||||
|
|
||||||
|
def bounding_box(self, default=None):
|
||||||
|
if not self.points:
|
||||||
|
return default
|
||||||
|
|
||||||
|
return (min(p.x for p in self.points), min(p.y for p in self.points)), \
|
||||||
|
(max(p.x for p in self.points), max(p.y for p in self.points))
|
||||||
|
|
||||||
|
def as_rectangle(self):
|
||||||
|
(maxx, maxy, minx, miny) = self.bbox()
|
||||||
|
return Rectangle(minx, maxy, maxx, miny, self.stroke, self.fill)
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
p0, *rest = self.points
|
||||||
|
if not rest:
|
||||||
|
return
|
||||||
|
|
||||||
|
d = ' '.join([f'M {p0.x:.3f} {p0.y:.3f}', *(f'L {pn.x:.3f} {pn.y:.3f}' for pn in rest)])
|
||||||
|
yield Tag('path', d=d, fill=self.fill.svg(colorscheme.lines, colorscheme.fill), **self.stroke.svg_attrs(colorscheme.lines))
|
||||||
|
|
||||||
|
def is_rectangle(self):
|
||||||
|
# A rectangle has 5 points and is closed
|
||||||
|
if len(self.points) != 5 or not self.is_closed():
|
||||||
|
return False
|
||||||
|
|
||||||
|
# Check that we have all four corners present
|
||||||
|
(x1, y1), (x2, y2) = self.bbox()
|
||||||
|
if not all(any(cand.isclose(pt) for cand in self.points[:-1]) for pt in
|
||||||
|
[(x1, y1), (x1, y2), (x2, y2), (x2, y1)]):
|
||||||
|
return False
|
||||||
|
|
||||||
|
# Check that we only have horizontal or vertical lines
|
||||||
|
if any(x2-x1 and y2-y1 for (x1, y1), (x2, y2) in zip(self.points[:-1], self.points[1:])):
|
||||||
|
return False
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('at')
|
||||||
|
class TextPos(XYCoord):
|
||||||
|
x: float = 0 # in millimeter
|
||||||
|
y: float = 0 # in millimeter
|
||||||
|
rotation: int = 0 # in degrees
|
||||||
|
|
||||||
|
def __after_parse__(self, parent):
|
||||||
|
self.rotation = self.rotation / 10
|
||||||
|
|
||||||
|
def __before_sexp__(self):
|
||||||
|
self.rotation = round((self.rotation % 360) * 10)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def rotation_rad(self):
|
||||||
|
return math.radians(self.rotation)
|
||||||
|
|
||||||
|
@rotation_rad.setter
|
||||||
|
def rotation_rad(self, value):
|
||||||
|
self.rotation = math.degrees(value)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('text')
|
||||||
|
class Text(TextMixin):
|
||||||
|
text: str = None
|
||||||
|
at: TextPos = field(default_factory=TextPos)
|
||||||
|
rotation: float = None
|
||||||
|
effects: TextEffect = field(default_factory=TextEffect)
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
yield from TextMixin.to_svg(self, colorscheme.text)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('rectangle')
|
||||||
|
class Rectangle:
|
||||||
|
# Some v6 symbols use rectangles, newer ones encode them as polylines.
|
||||||
|
# At some point in time we can most likely remove this class since its not used anymore
|
||||||
|
|
||||||
|
start: Rename(XYCoord) = None
|
||||||
|
end: Rename(XYCoord) = None
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
fill: Fill = field(default_factory=Fill)
|
||||||
|
|
||||||
|
def to_polyline(self):
|
||||||
|
x1, y1 = self.start.x, self.start.y
|
||||||
|
x2, y2 = self.end.x, self.end.y
|
||||||
|
return Polyline(PointList([XYCoord(x1, y1), XYCoord(x2, y1), XYCoord(x2, y2), XYCoord(x1, y2), XYCoord(x1, y1)]),
|
||||||
|
self.stroke, self.fill)
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
return self.to_polyline().to_svg(colorscheme)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('property')
|
||||||
|
class Property(TextMixin):
|
||||||
|
private: Flag() = False
|
||||||
|
name: str = None
|
||||||
|
value: str = None
|
||||||
|
id: Named(int) = None
|
||||||
|
at: AtPos = field(default_factory=AtPos)
|
||||||
|
show_name: Flag() = False
|
||||||
|
effects: TextEffect = field(default_factory=TextEffect)
|
||||||
|
|
||||||
|
# Alias value for text mixin
|
||||||
|
@property
|
||||||
|
def text(self):
|
||||||
|
return self.value
|
||||||
|
|
||||||
|
@text.setter
|
||||||
|
def text(self, value):
|
||||||
|
self.value = value
|
||||||
|
|
||||||
|
def to_svg(self, colorscheme=Colorscheme.KiCad):
|
||||||
|
yield from TextMixin.to_svg(self, colorscheme.text)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('pin_numbers')
|
||||||
|
class PinNumberSpec:
|
||||||
|
hide: Named(YesNoAtom()) = False
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('pin_names')
|
||||||
|
class PinNameSpec:
|
||||||
|
offset: OmitDefault(Named(float)) = 0.508
|
||||||
|
hide: OmitDefault(Named(YesNoAtom())) = False
|
||||||
|
|
||||||
|
@sexp_type('text_box')
|
||||||
|
class TextBox:
|
||||||
|
text: str = ''
|
||||||
|
exclude_from_sim: OmitDefault(Named(YesNoAtom())) = False
|
||||||
|
at: AtPos = field(default_factory=AtPos)
|
||||||
|
size: Rename(XYCoord) = field(default_factory=XYCoord)
|
||||||
|
margins: Margins = None
|
||||||
|
stroke: Stroke = field(default_factory=Stroke)
|
||||||
|
fill: Fill = field(default_factory=Fill)
|
||||||
|
effects: TextEffect = field(default_factory=TextEffect)
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('symbol')
|
||||||
|
class Unit:
|
||||||
|
name: str = None
|
||||||
|
circles: List(Circle) = field(default_factory=list)
|
||||||
|
arcs: List(Arc) = field(default_factory=list)
|
||||||
|
polylines: List(Polyline) = field(default_factory=list)
|
||||||
|
rectangles: List(Rectangle) = field(default_factory=list)
|
||||||
|
texts: List(Text) = field(default_factory=list)
|
||||||
|
text_boxes: List(TextBox) = field(default_factory=list)
|
||||||
|
pins: List(Pin) = field(default_factory=list)
|
||||||
|
unit_name: Named(str) = None
|
||||||
|
_ : SEXP_END = None
|
||||||
|
unit_global: Flag() = False
|
||||||
|
style_global: Flag() = False
|
||||||
|
demorgan_style: int = 1
|
||||||
|
unit_index: int = 1
|
||||||
|
symbol = None
|
||||||
|
|
||||||
|
def __after_parse__(self, parent):
|
||||||
|
self.symbol = parent
|
||||||
|
|
||||||
|
if not (m := re.fullmatch(r'(.*)_([0-9]+)_([0-9]+)', self.name)):
|
||||||
|
raise FormatError(f'Invalid unit name "{self.name}"')
|
||||||
|
sym_name, unit_index, demorgan_style = m.groups()
|
||||||
|
if sym_name != self.symbol.raw_name.rpartition(':')[2]:
|
||||||
|
raise FormatError(f'Unit name "{self.name}" does not match symbol name "{self.symbol.name}"')
|
||||||
|
self.demorgan_style = int(demorgan_style)
|
||||||
|
self.unit_index = int(unit_index)
|
||||||
|
self.style_global = self.demorgan_style == 0
|
||||||
|
self.unit_global = self.unit_index == 0
|
||||||
|
|
||||||
|
@property
|
||||||
|
def graphical_elements(self):
|
||||||
|
yield from self.rectangles
|
||||||
|
yield from self.polylines
|
||||||
|
yield from self.circles
|
||||||
|
yield from self.arcs
|
||||||
|
yield from self.texts
|
||||||
|
|
||||||
|
def __before_sexp__(self):
|
||||||
|
self.name = f'{self.symbol.name}_{self.unit_index}_{self.demorgan_style}'
|
||||||
|
|
||||||
|
def pin_stacks(self):
|
||||||
|
stacks = defaultdict(lambda: set())
|
||||||
|
for pin in self.all_pins():
|
||||||
|
stacks[(pin.at.x, pin.at.y)].add(pin)
|
||||||
|
return stacks
|
||||||
|
|
||||||
|
|
||||||
|
@sexp_type('symbol')
|
||||||
|
class Symbol:
|
||||||
|
raw_name: str = None
|
||||||
|
extends: Named(str) = None
|
||||||
|
power: Wrap(Flag()) = False
|
||||||
|
pin_numbers: OmitDefault(PinNumberSpec) = field(default_factory=PinNumberSpec)
|
||||||
|
pin_names: OmitDefault(PinNameSpec) = field(default_factory=PinNameSpec)
|
||||||
|
exclude_from_sim: OmitDefault(Named(YesNoAtom())) = False
|
||||||
|
exclude_from_sim: Named(YesNoAtom()) = False
|
||||||
|
in_bom: Named(YesNoAtom()) = True
|
||||||
|
on_board: Named(YesNoAtom()) = True
|
||||||
|
properties: List(Property) = field(default_factory=list)
|
||||||
|
units: List(Unit) = field(default_factory=list)
|
||||||
|
embedded_fonts: Named(YesNoAtom()) = False
|
||||||
|
_ : SEXP_END = None
|
||||||
|
library = None
|
||||||
|
name: str = None
|
||||||
|
library_name: str = None
|
||||||
|
|
||||||
|
def __after_parse__(self, parent):
|
||||||
|
self.library = parent
|
||||||
|
|
||||||
|
self.library_name, _, self.name = self.raw_name.rpartition(':')
|
||||||
|
|
||||||
|
if self.extends:
|
||||||
|
self.in_bom = None
|
||||||
|
self.on_board = None
|
||||||
|
|
||||||
|
self.properties = {prop.name: prop for prop in self.properties}
|
||||||
|
if (prop := self.properties.get('ki_fp_filters')):
|
||||||
|
prop.value = prop.value.split() if prop.value else []
|
||||||
|
|
||||||
|
def __before_sexp__(self):
|
||||||
|
if (prop := self.properties.get('ki_fp_filters')):
|
||||||
|
if not isinstance(prop.value, str):
|
||||||
|
prop.value = ' '.join(prop.value)
|
||||||
|
self.properties = list(self.properties.values())
|
||||||
|
|
||||||
|
def default_properties(self):
|
||||||
|
for i, (name, value, hide) in enumerate([
|
||||||
|
('Reference', 'U', False),
|
||||||
|
('Value', None, False),
|
||||||
|
('Footprint', None, True),
|
||||||
|
('Datasheet', None, True),
|
||||||
|
('ki_locked', None, True),
|
||||||
|
('ki_keywords', None, True),
|
||||||
|
('ki_description', None, True),
|
||||||
|
('ki_fp_filters', None, False),
|
||||||
|
]):
|
||||||
|
self.properties[name] = Property(name=name, value=value, id=i, effects=TextEffect(hide=hide))
|
||||||
|
|
||||||
|
def resolve(self):
|
||||||
|
if self.extends:
|
||||||
|
return self.library[self.extends]
|
||||||
|
else:
|
||||||
|
return self
|
||||||
|
|
||||||
|
def is_graphic_symbol(self):
|
||||||
|
return self.extends is None and (
|
||||||
|
not self.pins or self.get_property("Reference").value == "#SYM"
|
||||||
|
)
|
||||||
|
|
||||||
|
def pins_by_name(self, demorgan_style=None):
|
||||||
|
pins = defaultdict(lambda: set())
|
||||||
|
for unit in self.units(demorgan_style):
|
||||||
|
for pin in unit.all_pins:
|
||||||
|
pins[pin.name].add(pin)
|
||||||
|
return pins
|
||||||
|
|
||||||
|
def pins_by_number(self, demorgan_style=None):
|
||||||
|
pins = defaultdict(lambda: set())
|
||||||
|
for unit in self.units(demorgan_style):
|
||||||
|
for pin in unit.all_pins:
|
||||||
|
pins[pin.number].add(pin)
|
||||||
|
return pins
|
||||||
|
|
||||||
|
def filter_pins(self, name=None, direction=None, electrical_type=None):
|
||||||
|
for pin in self.all_pins:
|
||||||
|
if name and not fnmatch(pin.name, name):
|
||||||
|
continue
|
||||||
|
if direction and not pin.direction in direction:
|
||||||
|
continue
|
||||||
|
if electrical_type and not pin.etype in electical_type:
|
||||||
|
continue
|
||||||
|
yield pin
|
||||||
|
|
||||||
|
def heuristically_small(self):
|
||||||
|
""" Heuristically try to determine whether this is a "small" component like a resistor, capacitor, LED, diode,
|
||||||
|
or transistor etc. When we have at most two pins, or there is no filled rectangle as symbol outline and we have
|
||||||
|
3 or 4 pins, we assume this is a small symbol.
|
||||||
|
"""
|
||||||
|
if len(self.all_pins) <= 2:
|
||||||
|
return True
|
||||||
|
if len(self.all_pins) > 4:
|
||||||
|
return False
|
||||||
|
return bool(self.get_center_rectangle(range(self.unit_count)))
|
||||||
|
|
||||||
|
|
||||||
|
SUPPORTED_FILE_FORMAT_VERSIONS = [20211014, 20220914]
|
||||||
|
@sexp_type('kicad_symbol_lib')
|
||||||
|
class Library:
|
||||||
|
_version: Named(int, name='version') = 20211014
|
||||||
|
generator: Named(str) = Atom.gerbonara
|
||||||
|
generator_version: Named(str) = __version__
|
||||||
|
symbols: List(Symbol) = field(default_factory=list)
|
||||||
|
_ : SEXP_END = None
|
||||||
|
original_filename: str = None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def version(self):
|
||||||
|
return self._version
|
||||||
|
|
||||||
|
@version.setter
|
||||||
|
def version(self, value):
|
||||||
|
if value not in SUPPORTED_FILE_FORMAT_VERSIONS:
|
||||||
|
raise FormatError(f'File format version {value} is not supported. Supported versions are {", ".join(map(str, SUPPORTED_FILE_FORMAT_VERSIONS))}.')
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def open(cls, filename: str):
|
||||||
|
with open(filename) as f:
|
||||||
|
return cls.parse(f.read())
|
||||||
|
|
||||||
|
def write(self, filename=None):
|
||||||
|
with open(filename or self.original_filename, 'w') as f:
|
||||||
|
f.write(build_sexp(sexp(self)))
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
if len(sys.argv) >= 2:
|
||||||
|
a = Library.open(sys.argv[1])
|
||||||
|
print(build_sexp(sexp(a)))
|
||||||
|
else:
|
||||||
|
print("pass a .kicad_sym file please")
|
||||||
88
src/gerbonara/cad/kicad/tmtheme.py
Normal file
|
|
@ -0,0 +1,88 @@
|
||||||
|
|
||||||
|
from xml.etree import ElementTree
|
||||||
|
import base64
|
||||||
|
import json
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
def _map_primitive(element):
|
||||||
|
match element.tag:
|
||||||
|
case 'data':
|
||||||
|
return base64.b64decode(element.text)
|
||||||
|
case 'date':
|
||||||
|
return element.text
|
||||||
|
case 'true':
|
||||||
|
return True
|
||||||
|
case 'false':
|
||||||
|
return False
|
||||||
|
case 'real':
|
||||||
|
return float(element.text)
|
||||||
|
case 'integer':
|
||||||
|
return int(element.text)
|
||||||
|
case 'string':
|
||||||
|
return element.text
|
||||||
|
case 'array':
|
||||||
|
return [_map_primitive(child) for child in element]
|
||||||
|
case 'dict':
|
||||||
|
children = list(element)
|
||||||
|
return {k.text: _map_primitive(v) for k, v in zip(children[0::2], children[1::2])}
|
||||||
|
|
||||||
|
|
||||||
|
def parse_shitty_json(data):
|
||||||
|
# Parse apple plist XML
|
||||||
|
root = ElementTree.fromstring(data)
|
||||||
|
return _map_primitive(root[0])
|
||||||
|
|
||||||
|
|
||||||
|
class _SublimeColorschemeSuper:
|
||||||
|
def __init__(self, s, by_scope):
|
||||||
|
def lookup(default, *scopes):
|
||||||
|
for scope in scopes:
|
||||||
|
if not (elem := by_scope.get(scope)):
|
||||||
|
continue
|
||||||
|
|
||||||
|
if 'foreground' not in elem:
|
||||||
|
continue
|
||||||
|
|
||||||
|
return elem['foreground']
|
||||||
|
return default
|
||||||
|
|
||||||
|
self.background = s.get('background', 'white')
|
||||||
|
fg = s.get('foreground', 'black')
|
||||||
|
self.bus = lookup(fg, 'constant.other', 'storage.type')
|
||||||
|
self.wire = self.lines = lookup(fg, 'constant.other')
|
||||||
|
self.no_connect = lookup(fg, 'constant.language', 'variable')
|
||||||
|
self.text = lookup(fg, 'constant.numeric', 'constant.numeric.hex', 'storage.type.number')
|
||||||
|
self.pin_names = lookup(fg, 'constant.character', 'constant.other')
|
||||||
|
self.pin_numbers = fg
|
||||||
|
self.values = lookup(fg, 'constant.character.format.placeholder', 'constant.other.placeholder', 'entity.name.tag', 'support.type', 'support.class', 'entity.other.inherited-class')
|
||||||
|
self.labels = lookup(fg, 'constant.numeric', 'constant.numeric.hex', 'storage.type.number')
|
||||||
|
self.fill = s.get('background')
|
||||||
|
|
||||||
|
|
||||||
|
class TmThemeSchematic(_SublimeColorschemeSuper):
|
||||||
|
def __init__(self, data):
|
||||||
|
self.theme = parse_shitty_json(data)
|
||||||
|
s = self.theme['settings'][0]['settings']
|
||||||
|
by_scope = {}
|
||||||
|
for elem in self.theme['settings']:
|
||||||
|
if 'scope' not in elem:
|
||||||
|
continue
|
||||||
|
for scope in elem['scope'].split(','):
|
||||||
|
by_scope[scope.strip()] = elem.get('settings', {})
|
||||||
|
super().__init__(s, by_scope)
|
||||||
|
|
||||||
|
|
||||||
|
class SublimeSchematic(_SublimeColorschemeSuper):
|
||||||
|
def __init__(self, data):
|
||||||
|
self.theme = json.loads(data)
|
||||||
|
s = self.theme['globals']
|
||||||
|
by_scope = {}
|
||||||
|
for elem in self.theme['rules']:
|
||||||
|
for scope in elem['scope'].split(','):
|
||||||
|
by_scope[scope.strip()] = elem
|
||||||
|
super().__init__(s, by_scope)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
print(parse_shitty_json(Path('/tmp/witchhazelhypercolor.tmTheme').read_text()))
|
||||||
|
|
||||||
|
|
@ -4,7 +4,7 @@ import math
|
||||||
import warnings
|
import warnings
|
||||||
from copy import copy
|
from copy import copy
|
||||||
from itertools import zip_longest, chain
|
from itertools import zip_longest, chain
|
||||||
from dataclasses import dataclass, field, KW_ONLY
|
from dataclasses import dataclass, field, replace, KW_ONLY
|
||||||
from collections import defaultdict
|
from collections import defaultdict
|
||||||
|
|
||||||
from ..utils import LengthUnit, MM, rotate_point, svg_arc, sum_bounds, bbox_intersect, Tag, offset_bounds
|
from ..utils import LengthUnit, MM, rotate_point, svg_arc, sum_bounds, bbox_intersect, Tag, offset_bounds
|
||||||
|
|
@ -14,6 +14,9 @@ from ..apertures import Aperture, CircleAperture, ObroundAperture, RectangleAper
|
||||||
from ..newstroke import Newstroke
|
from ..newstroke import Newstroke
|
||||||
|
|
||||||
|
|
||||||
|
class UNDEFINED:
|
||||||
|
pass
|
||||||
|
|
||||||
def sgn(x):
|
def sgn(x):
|
||||||
return -1 if x < 0 else 1
|
return -1 if x < 0 else 1
|
||||||
|
|
||||||
|
|
@ -51,7 +54,7 @@ class Board:
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def abs_pos(self):
|
def abs_pos(self):
|
||||||
return self.x, self.y, self.rotation
|
return self.x, self.y, self.rotation, False
|
||||||
|
|
||||||
def add_silk(self, side, obj):
|
def add_silk(self, side, obj):
|
||||||
if side not in ('top', 'bottom'):
|
if side not in ('top', 'bottom'):
|
||||||
|
|
@ -115,7 +118,7 @@ class Board:
|
||||||
|
|
||||||
def layer_stack(self, layer_stack=None):
|
def layer_stack(self, layer_stack=None):
|
||||||
if layer_stack is None:
|
if layer_stack is None:
|
||||||
layer_stack = LayerStack()
|
layer_stack = LayerStack(board_name='proto')
|
||||||
|
|
||||||
cache = {}
|
cache = {}
|
||||||
for obj in chain(self.objects):
|
for obj in chain(self.objects):
|
||||||
|
|
@ -142,21 +145,56 @@ class Positioned:
|
||||||
y: float
|
y: float
|
||||||
_: KW_ONLY
|
_: KW_ONLY
|
||||||
rotation: float = 0.0
|
rotation: float = 0.0
|
||||||
side: str = 'top'
|
flip: bool = False
|
||||||
unit: LengthUnit = MM
|
unit: LengthUnit = MM
|
||||||
parent: object = None
|
parent: object = None
|
||||||
|
|
||||||
def flip(self):
|
|
||||||
self.side = 'top' if self.side == 'bottom' else 'bottom'
|
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def abs_pos(self):
|
def abs_pos(self):
|
||||||
if self.parent is None:
|
if self.parent is None:
|
||||||
px, py, pa = 0, 0, 0
|
px, py, pa, pf = 0, 0, 0, False
|
||||||
else:
|
else:
|
||||||
px, py, pa = self.parent.abs_pos
|
px, py, pa, pf = self.parent.abs_pos
|
||||||
|
|
||||||
return self.x+px, self.y+py, self.rotation+pa
|
return self.x+px, self.y+py, self.rotation+pa, (bool(self.flip) != bool(pf))
|
||||||
|
|
||||||
|
def bounding_box(self, unit=MM):
|
||||||
|
stack = LayerStack()
|
||||||
|
self.render(stack)
|
||||||
|
objects = chain(*(l.objects for l in stack.graphic_layers.values()),
|
||||||
|
stack.drill_pth.objects, stack.drill_npth.objects)
|
||||||
|
objects = list(objects)
|
||||||
|
#print('foo', type(self).__name__,
|
||||||
|
# [(type(obj).__name__, [prim.bounding_box() for prim in obj.to_primitives(unit)]) for obj in objects], file=sys.stderr)
|
||||||
|
return sum_bounds(prim.bounding_box() for obj in objects for prim in obj.to_primitives(unit))
|
||||||
|
|
||||||
|
def overlaps(self, bbox, unit=MM):
|
||||||
|
return bbox_intersect(self.bounding_box(unit), bbox)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def single_sided(self):
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
# The dataclass API is slightly idiotic here, so we have to duplicate the entire thing.
|
||||||
|
@dataclass(frozen=True)
|
||||||
|
class FrozenPositioned:
|
||||||
|
x: float
|
||||||
|
y: float
|
||||||
|
_: KW_ONLY
|
||||||
|
rotation: float = 0.0
|
||||||
|
flip: bool = False
|
||||||
|
unit: LengthUnit = MM
|
||||||
|
parent: object = None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def abs_pos(self):
|
||||||
|
if self.parent is None:
|
||||||
|
px, py, pa, pf = 0, 0, 0, False
|
||||||
|
else:
|
||||||
|
px, py, pa, pf = self.parent.abs_pos
|
||||||
|
|
||||||
|
return self.x+px, self.y+py, self.rotation+pa, (bool(self.flip) != bool(pf))
|
||||||
|
|
||||||
def bounding_box(self, unit=MM):
|
def bounding_box(self, unit=MM):
|
||||||
stack = LayerStack()
|
stack = LayerStack()
|
||||||
|
|
@ -177,7 +215,7 @@ class Positioned:
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class ObjectGroup(Positioned):
|
class Graphics(Positioned):
|
||||||
top_copper: list = field(default_factory=list)
|
top_copper: list = field(default_factory=list)
|
||||||
top_mask: list = field(default_factory=list)
|
top_mask: list = field(default_factory=list)
|
||||||
top_silk: list = field(default_factory=list)
|
top_silk: list = field(default_factory=list)
|
||||||
|
|
@ -188,15 +226,10 @@ class ObjectGroup(Positioned):
|
||||||
bottom_paste: list = field(default_factory=list)
|
bottom_paste: list = field(default_factory=list)
|
||||||
drill_npth: list = field(default_factory=list)
|
drill_npth: list = field(default_factory=list)
|
||||||
drill_pth: list = field(default_factory=list)
|
drill_pth: list = field(default_factory=list)
|
||||||
objects: list = field(default_factory=list)
|
|
||||||
|
|
||||||
def render(self, layer_stack, cache=None):
|
def render(self, layer_stack, cache=None):
|
||||||
x, y, rotation = self.abs_pos
|
x, y, rotation, flip = self.abs_pos
|
||||||
top, bottom = ('bottom', 'top') if self.side == 'bottom' else ('top', 'bottom')
|
top, bottom = ('bottom', 'top') if flip else ('top', 'bottom')
|
||||||
|
|
||||||
for obj in self.objects:
|
|
||||||
obj.parent = self
|
|
||||||
obj.render(layer_stack, cache=cache)
|
|
||||||
|
|
||||||
for target, source in [
|
for target, source in [
|
||||||
(layer_stack[top, 'copper'], self.top_copper),
|
(layer_stack[top, 'copper'], self.top_copper),
|
||||||
|
|
@ -229,7 +262,6 @@ class ObjectGroup(Positioned):
|
||||||
self.bottom_paste,
|
self.bottom_paste,
|
||||||
self.drill_npth,
|
self.drill_npth,
|
||||||
self.drill_pth,
|
self.drill_pth,
|
||||||
self.objects,
|
|
||||||
))), unit(self.x, self.unit), unit(self.y, self.unit))
|
))), unit(self.x, self.unit), unit(self.y, self.unit))
|
||||||
else:
|
else:
|
||||||
return super().bounding_box(unit)
|
return super().bounding_box(unit)
|
||||||
|
|
@ -242,6 +274,30 @@ class ObjectGroup(Positioned):
|
||||||
return not (any_drill or (any_top and any_bottom))
|
return not (any_drill or (any_top and any_bottom))
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class ObjectGroup(Positioned):
|
||||||
|
objects: list = field(default_factory=list)
|
||||||
|
|
||||||
|
def render(self, layer_stack, cache=None):
|
||||||
|
for obj in self.objects:
|
||||||
|
if not isinstance(obj, Positioned):
|
||||||
|
raise ValueError(f'ObjectGroup members must be children of Positioned, not {type(obj)}')
|
||||||
|
|
||||||
|
obj.parent = self
|
||||||
|
obj.render(layer_stack, cache=cache)
|
||||||
|
|
||||||
|
def bounding_box(self, unit=MM):
|
||||||
|
if math.isclose(self.rotation, 0, abs_tol=1e-3):
|
||||||
|
return offset_bounds(sum_bounds((obj.bounding_box(unit=unit) for obj in self.objects)),
|
||||||
|
unit(self.x, self.unit), unit(self.y, self.unit))
|
||||||
|
else:
|
||||||
|
return super().bounding_box(unit)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def single_sided(self):
|
||||||
|
return all(obj.single_sided for obj in self.objects)
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class Text(Positioned):
|
class Text(Positioned):
|
||||||
text: str
|
text: str
|
||||||
|
|
@ -253,7 +309,7 @@ class Text(Positioned):
|
||||||
polarity_dark: bool = True
|
polarity_dark: bool = True
|
||||||
|
|
||||||
def render(self, layer_stack, cache=None):
|
def render(self, layer_stack, cache=None):
|
||||||
obj_x, obj_y, rotation = self.abs_pos
|
obj_x, obj_y, rotation, flip = self.abs_pos
|
||||||
global newstroke_font
|
global newstroke_font
|
||||||
|
|
||||||
if newstroke_font is None:
|
if newstroke_font is None:
|
||||||
|
|
@ -266,6 +322,7 @@ class Text(Positioned):
|
||||||
xs = [x for points in strokes for x, _y in points]
|
xs = [x for points in strokes for x, _y in points]
|
||||||
ys = [y for points in strokes for _x, y in points]
|
ys = [y for points in strokes for _x, y in points]
|
||||||
min_x, min_y, max_x, max_y = min(xs), min(ys), max(xs), max(ys)
|
min_x, min_y, max_x, max_y = min(xs), min(ys), max(xs), max(ys)
|
||||||
|
h = self.font_size + self.stroke_width # (max_y - min_y)
|
||||||
|
|
||||||
if self.h_align == 'left':
|
if self.h_align == 'left':
|
||||||
x0 = 0
|
x0 = 0
|
||||||
|
|
@ -276,16 +333,16 @@ class Text(Positioned):
|
||||||
else:
|
else:
|
||||||
raise ValueError('h_align must be one of "left", "center", or "right".')
|
raise ValueError('h_align must be one of "left", "center", or "right".')
|
||||||
|
|
||||||
if self.v_align == 'top':
|
if self.v_align == 'bottom':
|
||||||
y0 = -(max_y - min_y)
|
y0 = h
|
||||||
elif self.v_align == 'middle':
|
elif self.v_align == 'middle':
|
||||||
y0 = -(max_y - min_y)/2
|
y0 = h/2
|
||||||
elif self.v_align == 'bottom':
|
elif self.v_align == 'top':
|
||||||
y0 = 0
|
y0 = 0
|
||||||
else:
|
else:
|
||||||
raise ValueError('v_align must be one of "top", "middle", or "bottom".')
|
raise ValueError('v_align must be one of "top", "middle", or "bottom".')
|
||||||
|
|
||||||
if self.side == 'bottom':
|
if self.flip:
|
||||||
x0 += min_x + max_x
|
x0 += min_x + max_x
|
||||||
x_sign = -1
|
x_sign = -1
|
||||||
else:
|
else:
|
||||||
|
|
@ -295,10 +352,10 @@ class Text(Positioned):
|
||||||
|
|
||||||
for stroke in strokes:
|
for stroke in strokes:
|
||||||
for (x1, y1), (x2, y2) in zip(stroke[:-1], stroke[1:]):
|
for (x1, y1), (x2, y2) in zip(stroke[:-1], stroke[1:]):
|
||||||
obj = Line(x0+x_sign*x1, y0-y1, x0+x_sign*x2, y0-y2, aperture=ap, unit=self.unit, polarity_dark=self.polarity_dark)
|
obj = Line(x0+x_sign*x1, y0+y1, x0+x_sign*x2, y0+y2, aperture=ap, unit=self.unit, polarity_dark=self.polarity_dark)
|
||||||
obj.rotate(rotation)
|
obj.rotate(rotation)
|
||||||
obj.offset(obj_x, obj_y)
|
obj.offset(obj_x, obj_y)
|
||||||
layer_stack[self.side, self.layer].objects.append(obj)
|
layer_stack['bottom' if flip else 'top', self.layer].objects.append(obj)
|
||||||
|
|
||||||
def bounding_box(self, unit=MM):
|
def bounding_box(self, unit=MM):
|
||||||
approx_w = len(self.text)*self.font_size*0.75 + self.stroke_width
|
approx_w = len(self.text)*self.font_size*0.75 + self.stroke_width
|
||||||
|
|
@ -312,166 +369,209 @@ class Text(Positioned):
|
||||||
x0 = -approx_w
|
x0 = -approx_w
|
||||||
|
|
||||||
if self.v_align == 'top':
|
if self.v_align == 'top':
|
||||||
y0 = -approx_h
|
y0 = 0
|
||||||
elif self.v_align == 'middle':
|
elif self.v_align == 'middle':
|
||||||
y0 = -approx_h/2
|
y0 = -approx_h/2
|
||||||
elif self.v_align == 'bottom':
|
elif self.v_align == 'bottom':
|
||||||
y0 = 0
|
y0 = -approx_h
|
||||||
|
|
||||||
return (self.x+x0, self.y+y0), (self.x+x0+approx_w, self.y+y0+approx_h)
|
return (self.x+x0, self.y+y0), (self.x+x0+approx_w, self.y+y0+approx_h)
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass(frozen=True, slots=True)
|
||||||
class Pad(Positioned):
|
class PadStackAperture:
|
||||||
pass
|
aperture: Aperture
|
||||||
|
side: str
|
||||||
|
layer: str
|
||||||
|
offset_x: float = 0 # in PadStack units
|
||||||
|
offset_y: float = 0
|
||||||
|
rotation: float = 0
|
||||||
|
invert: bool = False
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass(frozen=True, slots=True)
|
||||||
class SMDPad(Pad):
|
class PadStack:
|
||||||
copper_aperture: Aperture
|
_: KW_ONLY
|
||||||
mask_aperture: Aperture
|
unit: LengthUnit = MM
|
||||||
paste_aperture: Aperture
|
|
||||||
silk_features: list = field(default_factory=list)
|
|
||||||
|
|
||||||
def render(self, layer_stack, cache=None):
|
@property
|
||||||
x, y, rotation = self.abs_pos
|
def apertures(self):
|
||||||
layer_stack[self.side, 'copper'].objects.append(Flash(x, y, self.copper_aperture.rotated(rotation), unit=self.unit))
|
raise NotImplementedError()
|
||||||
layer_stack[self.side, 'mask' ].objects.append(Flash(x, y, self.mask_aperture.rotated(rotation), unit=self.unit))
|
|
||||||
if self.paste_aperture:
|
def flashes(self, x, y, rotation: float = 0, flip: bool = False):
|
||||||
layer_stack[self.side, 'paste' ].objects.append(Flash(x, y, self.paste_aperture.rotated(rotation), unit=self.unit))
|
for ap in self.apertures:
|
||||||
layer_stack[self.side, 'silk' ].objects.extend([copy(feature).rotate(rotation).offset(x, y, self.unit)
|
aperture = ap.aperture.rotated(ap.rotation + rotation)
|
||||||
for feature in self.silk_features])
|
fl = Flash(ap.offset_x, ap.offset_y, aperture, polarity_dark=not ap.invert, unit=self.unit)
|
||||||
|
fl.rotate(rotation)
|
||||||
|
fl.offset(x, y)
|
||||||
|
side = ap.side
|
||||||
|
if flip:
|
||||||
|
side = {'top': 'bottom', 'bottom': 'top'}.get(side, side)
|
||||||
|
yield side, ap.layer, fl
|
||||||
|
|
||||||
|
def render(self, layer_stack, x, y, rotation: float = 0, flip: bool = False):
|
||||||
|
for side, layer, flash in self.flashes(x, y, rotation, flip):
|
||||||
|
if side == 'drill' and layer == 'plated':
|
||||||
|
layer_stack.drill_pth.objects.append(flash)
|
||||||
|
|
||||||
|
elif side == 'drill' and layer == 'nonplated':
|
||||||
|
layer_stack.drill_npth.objects.append(flash)
|
||||||
|
|
||||||
|
elif (side, layer) in layer_stack:
|
||||||
|
layer_stack[side, layer].objects.append(flash)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def single_sided(self):
|
||||||
|
return len({ap.side for ap in self.apertures}) <= 1
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass(frozen=True, slots=True)
|
||||||
|
class SMDStack(PadStack):
|
||||||
|
aperture: Aperture
|
||||||
|
mask_expansion: float = 0.0
|
||||||
|
paste_expansion: float = 0.0
|
||||||
|
paste: bool = True
|
||||||
|
flip: bool = False
|
||||||
|
|
||||||
|
@property
|
||||||
|
def side(self):
|
||||||
|
return 'bottom' if self.flip else 'top'
|
||||||
|
|
||||||
|
@property
|
||||||
|
def apertures(self):
|
||||||
|
yield PadStackAperture(self.aperture, self.side, 'copper')
|
||||||
|
yield PadStackAperture(self.aperture.dilated(self.mask_expansion, self.unit), self.side, 'mask')
|
||||||
|
if self.paste:
|
||||||
|
yield PadStackAperture(self.aperture.dilated(self.paste_expansion, self.unit), self.side, 'paste')
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def rect(kls, x, y, w, h, rotation=0, side='top', mask_expansion=0.0, paste_expansion=0.0, paste=True, unit=MM):
|
def rect(kls, w, h, rotation=0, mask_expansion=0.0, paste_expansion=0.0, paste=True, flip=False, unit=MM):
|
||||||
ap_c = RectangleAperture(w, h, unit=unit)
|
ap = RectangleAperture(w, h, unit=unit).rotated(rotation)
|
||||||
ap_m = RectangleAperture(w+2*mask_expansion, h+2*mask_expansion, unit=unit)
|
return kls(ap, mask_expansion, paste_expansion, paste, flip, unit=unit)
|
||||||
ap_p = RectangleAperture(w+2*paste_expansion, h+2*paste_expansion, unit=unit) if paste else None
|
|
||||||
return kls(x, y, side=side, copper_aperture=ap_c, mask_aperture=ap_m, paste_aperture=ap_p, rotation=rotation,
|
|
||||||
unit=unit)
|
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def circle(kls, x, y, dia, side='top', mask_expansion=0.0, paste_expansion=0.0, paste=True, unit=MM):
|
def circle(kls, dia, mask_expansion=0.0, paste_expansion=0.0, paste=True, flip=False, unit=MM):
|
||||||
ap_c = CircleAperture(dia, unit=unit)
|
return kls(CircleAperture(dia, unit=unit), mask_expansion, paste_expansion, paste, flip, unit=unit)
|
||||||
ap_m = CircleAperture(dia+2*mask_expansion, unit=unit)
|
|
||||||
ap_p = CircleAperture(dia+2*paste_expansion, unit=unit) if paste else None
|
|
||||||
return kls(x, y, side=side, copper_aperture=ap_c, mask_aperture=ap_m, paste_aperture=ap_p, unit=unit)
|
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass(frozen=True, slots=True)
|
||||||
class THTPad(Pad):
|
class MechanicalHoleStack(PadStack):
|
||||||
drill_dia: float
|
drill_dia: float
|
||||||
pad_top: SMDPad
|
mask_expansion: float = 0.0
|
||||||
pad_bottom: SMDPad = None
|
mask_aperture = None
|
||||||
aperture_inner: Aperture = None
|
|
||||||
|
@property
|
||||||
|
def apertures(self):
|
||||||
|
mask_aperture = self.mask_aperture or CircleAperture(self.drill_dia + self.mask_expansion, unit=self.unit)
|
||||||
|
yield PadStackAperture(mask_aperture, 'top', 'mask')
|
||||||
|
yield PadStackAperture(mask_aperture, 'bottom', 'mask')
|
||||||
|
|
||||||
|
@property
|
||||||
|
def single_sided(self):
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass(frozen=True, slots=True)
|
||||||
|
class THTPad(PadStack):
|
||||||
|
drill_dia: float
|
||||||
|
pad_top: SMDStack
|
||||||
|
pad_bottom: SMDStack = None
|
||||||
|
aperture_inner: Aperture = UNDEFINED
|
||||||
plated: bool = True
|
plated: bool = True
|
||||||
|
|
||||||
def __post_init__(self):
|
def __post_init__(self):
|
||||||
if self.pad_bottom is None:
|
if self.pad_bottom is None:
|
||||||
import sys
|
object.__setattr__(self, 'pad_bottom', replace(self.pad_top, flip=True))
|
||||||
self.pad_bottom = copy(self.pad_top)
|
|
||||||
self.pad_bottom.flip()
|
|
||||||
|
|
||||||
self.pad_top.parent = self.pad_bottom.parent = self
|
|
||||||
|
|
||||||
if (self.pad_top.side, self.pad_bottom.side) != ('top', 'bottom'):
|
|
||||||
raise ValueError(f'The top and bottom pads must have side set to top and bottom, respectively. Currently, the top pad side is set to "{self.pad_top.side}" and the bottom pad side to "{self.pad_bottom.side}".')
|
|
||||||
|
|
||||||
def render(self, layer_stack, cache=None):
|
|
||||||
x, y, rotation = self.abs_pos
|
|
||||||
self.pad_top.parent = self
|
|
||||||
self.pad_top.render(layer_stack)
|
|
||||||
if self.pad_bottom:
|
|
||||||
self.pad_bottom.parent = self
|
|
||||||
self.pad_bottom.render(layer_stack)
|
|
||||||
|
|
||||||
if self.aperture_inner is None:
|
|
||||||
(x_min, y_min), (x_max, y_max) = self.pad_top.bounding_box(MM)
|
|
||||||
w_top = x_max - x_min
|
|
||||||
h_top = y_max - y_min
|
|
||||||
if self.pad_bottom:
|
|
||||||
(x_min, y_min), (x_max, y_max) = self.pad_bottom.bounding_box(MM)
|
|
||||||
w_bottom = x_max - x_min
|
|
||||||
h_bottom = y_max - y_min
|
|
||||||
w_top = min(w_top, w_bottom)
|
|
||||||
h_top = min(h_top, h_bottom)
|
|
||||||
self.aperture_inner = CircleAperture(min(w_top, h_top), unit=MM)
|
|
||||||
|
|
||||||
for (side, use), layer in layer_stack.inner_layers:
|
|
||||||
layer.objects.append(Flash(x, y, self.aperture_inner.rotated(rotation), unit=self.unit))
|
|
||||||
|
|
||||||
hole = Flash(x, y, ExcellonTool(self.drill_dia, plated=self.plated, unit=self.unit), unit=self.unit)
|
|
||||||
if self.plated:
|
|
||||||
layer_stack.drill_pth.objects.append(hole)
|
|
||||||
else:
|
|
||||||
layer_stack.drill_npth.objects.append(hole)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def single_sided(self):
|
|
||||||
return False
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def rect(kls, x, y, hole_dia, w, h=None, rotation=0, mask_expansion=0.0, paste_expansion=0.0, paste=True, plated=True, unit=MM):
|
|
||||||
if h is None:
|
|
||||||
h = w
|
|
||||||
pad = SMDPad.rect(0, 0, w, h, mask_expansion=mask_expansion, paste_expansion=paste_expansion, paste=paste, unit=unit)
|
|
||||||
return kls(x, y, hole_dia, pad, rotation=rotation, plated=plated, unit=unit)
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def circle(kls, x, y, hole_dia, dia, mask_expansion=0.0, paste_expansion=0.0, paste=True, plated=True, unit=MM):
|
|
||||||
pad = SMDPad.circle(0, 0, dia, mask_expansion=mask_expansion, paste_expansion=paste_expansion, paste=paste, unit=unit)
|
|
||||||
return kls(x, y, hole_dia, pad, plated=plated, unit=unit)
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def obround(kls, x, y, hole_dia, w, h, rotation=0, mask_expansion=0.0, paste_expanson=0.0, paste=True, plated=True, unit=MM):
|
|
||||||
ap_c = ObroundAperture(w, h, unit=unit)
|
|
||||||
ap_m = ObroundAperture(w+2*mask_expansion, h+2*mask_expansion, unit=unit)
|
|
||||||
ap_p = ObroundAperture(w, h, unit=unit) if paste else None
|
|
||||||
pad = SMDPad(0, 0, side='top', copper_aperture=ap_c, mask_aperture=ap_m, paste_aperture=ap_p, unit=unit)
|
|
||||||
return kls(x, y, hole_dia, pad, rotation=rotation, plated=plated, unit=unit)
|
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
|
||||||
class Hole(Positioned):
|
|
||||||
diameter: float
|
|
||||||
mask_copper_margin: float = 0.2
|
|
||||||
|
|
||||||
def render(self, layer_stack, cache=None):
|
|
||||||
x, y, rotation = self.abs_pos
|
|
||||||
|
|
||||||
hole = Flash(x, y, ExcellonTool(self.diameter, plated=False, unit=self.unit), unit=self.unit)
|
|
||||||
layer_stack.drill_npth.objects.append(hole)
|
|
||||||
|
|
||||||
if self.mask_copper_margin > 0:
|
|
||||||
mask = Flash(x, y, CircleAperture(self.mask_copper_margin, unit=self.unit), polarity_dark=False, unit=self.unit)
|
|
||||||
layer_stack['top', 'copper'].objects.append(mask)
|
|
||||||
layer_stack['bottom', 'copper'].objects.append(mask)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def single_sided(self):
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
|
||||||
class Via(Positioned):
|
|
||||||
diameter: float
|
|
||||||
hole: float
|
|
||||||
|
|
||||||
def render(self, layer_stack, cache=None):
|
|
||||||
x, y, rotation = self.abs_pos
|
|
||||||
|
|
||||||
aperture = CircleAperture(diameter=self.diameter, unit=self.unit)
|
|
||||||
tool = ExcellonTool(diameter=self.hole, unit=self.unit)
|
|
||||||
|
|
||||||
for (side, use), layer in layer_stack.copper_layers:
|
if self.aperture_inner is UNDEFINED:
|
||||||
layer.objects.append(Flash(x, y, aperture, unit=self.unit))
|
object.__setattr__(self, 'aperture_inner', self.pad_top.aperture)
|
||||||
|
|
||||||
layer_stack.drill_pth.objects.append(Flash(x, y, tool, unit=self.unit))
|
if self.pad_top.flip:
|
||||||
|
raise ValueError('top pad cannot be flipped')
|
||||||
|
|
||||||
|
@property
|
||||||
|
def plating(self):
|
||||||
|
return 'plated' if self.plated else 'nonplated'
|
||||||
|
|
||||||
|
@property
|
||||||
|
def apertures(self):
|
||||||
|
yield from self.pad_top.apertures
|
||||||
|
yield from self.pad_bottom.apertures
|
||||||
|
if self.aperture_inner is not None:
|
||||||
|
yield PadStackAperture(self.aperture_inner, 'inner', 'copper')
|
||||||
|
yield PadStackAperture(ExcellonTool(self.drill_dia, plated=self.plated, unit=self.unit), 'drill', self.plating)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def single_sided(self):
|
def single_sided(self):
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def rect(kls, drill_dia, w, h, rotation=0, mask_expansion=0.0, paste_expansion=0.0, paste=True, plated=True, unit=MM):
|
||||||
|
pad = SMDStack.rect(w, h, rotation, mask_expansion, paste_expansion, paste, unit=unit)
|
||||||
|
return kls(drill_dia, pad, plated=plated)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def circle(kls, drill_dia, dia, rotation=0, mask_expansion=0.0, paste_expansion=0.0, paste=True, plated=True, unit=MM):
|
||||||
|
pad = SMDStack.circle(dia, mask_expansion, paste_expansion, paste, unit=unit)
|
||||||
|
return kls(drill_dia, pad, plated=plated)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def obround(kls, drill_dia, w, h, rotation=0, mask_expansion=0.0, paste_expansion=0.0, paste=True, plated=True, unit=MM):
|
||||||
|
ap = ObroundAperture(w, h, unit=unit).rotated(rotation)
|
||||||
|
pad = SMDStack(ap, mask_expansion, paste_expansion, paste, unit=unit)
|
||||||
|
return kls(drill_dia, pad, plated=plated)
|
||||||
|
|
||||||
|
@dataclass(frozen=True, slots=True)
|
||||||
|
class ThroughViaStack(PadStack):
|
||||||
|
hole: float
|
||||||
|
dia: float = None
|
||||||
|
tented: bool = True
|
||||||
|
|
||||||
|
def __post_init__(self):
|
||||||
|
if self.dia == None:
|
||||||
|
object.__setattr__(self, 'dia', self.hole*2)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def single_sided(self):
|
||||||
|
return False
|
||||||
|
|
||||||
|
@property
|
||||||
|
def apertures(self):
|
||||||
|
copper_aperture = CircleAperture(self.dia, unit=self.unit)
|
||||||
|
yield PadStackAperture(copper_aperture, 'top', 'copper')
|
||||||
|
yield PadStackAperture(copper_aperture, 'bottom', 'copper')
|
||||||
|
yield PadStackAperture(copper_aperture, 'inner', 'copper')
|
||||||
|
if self.tented:
|
||||||
|
yield PadStackAperture(copper_aperture, 'top', 'mask')
|
||||||
|
yield PadStackAperture(copper_aperture, 'bottom', 'mask')
|
||||||
|
yield PadStackAperture(ExcellonTool(self.hole, plated=True, unit=self.unit), 'drill', 'plated')
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass(frozen=True, slots=True)
|
||||||
|
class Via(FrozenPositioned):
|
||||||
|
pad_stack: PadStack
|
||||||
|
|
||||||
|
def render(self, layer_stack, cache=None):
|
||||||
|
x, y, rotation, flip = self.abs_pos
|
||||||
|
self.pad_stack.render(layer_stack, x, y, rotation, flip)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def at(kls, x, y, hole, dia=None, tented=True, unit=MM):
|
||||||
|
return kls(x, y, ThroughViaStack(hole, dia, tented, unit=unit), unit=unit)
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class Pad(Positioned):
|
||||||
|
pad_stack: PadStack
|
||||||
|
|
||||||
|
def render(self, layer_stack, cache=None):
|
||||||
|
x, y, rotation, flip = self.abs_pos
|
||||||
|
self.pad_stack.render(layer_stack, x, y, rotation, flip)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def single_sided(self):
|
||||||
|
return self.pad_stack.single_sided
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class Trace:
|
class Trace:
|
||||||
|
|
@ -480,8 +580,9 @@ class Trace:
|
||||||
end: object = None
|
end: object = None
|
||||||
waypoints: [(float, float)] = field(default_factory=list)
|
waypoints: [(float, float)] = field(default_factory=list)
|
||||||
style: str = 'oblique'
|
style: str = 'oblique'
|
||||||
orientation: [str] = tuple() # 'top' or 'bottom'
|
orientation: [str] = tuple() # 'cw' or 'ccw'
|
||||||
roundover: float = 0
|
roundover: float = 0
|
||||||
|
side: str = 'top'
|
||||||
unit: LengthUnit = MM
|
unit: LengthUnit = MM
|
||||||
parent: object = None
|
parent: object = None
|
||||||
|
|
||||||
|
|
@ -629,13 +730,13 @@ class Trace:
|
||||||
|
|
||||||
yield Line(line_b.x1, line_b.y1, x3, y3, aperture=aperture, unit=self.unit)
|
yield Line(line_b.x1, line_b.y1, x3, y3, aperture=aperture, unit=self.unit)
|
||||||
|
|
||||||
def _to_graphic_objects(self):
|
def to_graphic_objects(self):
|
||||||
start, end = self.start, self.end
|
start, end = self.start, self.end
|
||||||
|
|
||||||
if not isinstance(start, tuple):
|
if not isinstance(start, tuple):
|
||||||
*start, _rotation = start.abs_pos
|
*start, _rotation, _flip = start.abs_pos
|
||||||
if not isinstance(end, tuple):
|
if not isinstance(end, tuple):
|
||||||
*end, _rotation = end.abs_pos
|
*end, _rotation, _flip = end.abs_pos
|
||||||
|
|
||||||
aperture = CircleAperture(diameter=self.width, unit=self.unit)
|
aperture = CircleAperture(diameter=self.width, unit=self.unit)
|
||||||
|
|
||||||
|
|
@ -649,7 +750,7 @@ class Trace:
|
||||||
return self._round_over(points, aperture)
|
return self._round_over(points, aperture)
|
||||||
|
|
||||||
def render(self, layer_stack, cache=None):
|
def render(self, layer_stack, cache=None):
|
||||||
layer_stack[self.side, 'copper'].objects.extend(self._to_graphic_objects())
|
layer_stack[self.side, 'copper'].objects.extend(self.to_graphic_objects())
|
||||||
|
|
||||||
def _route_demo():
|
def _route_demo():
|
||||||
from ..utils import setup_svg, Tag
|
from ..utils import setup_svg, Tag
|
||||||
1218
src/gerbonara/cad/protoboard.py
Normal file
|
|
@ -8,6 +8,7 @@ from quart import Quart, request, Response, send_file, abort
|
||||||
|
|
||||||
from . import protoboard as pb
|
from . import protoboard as pb
|
||||||
from . import protoserve_data
|
from . import protoserve_data
|
||||||
|
from .primitives import SMDStack
|
||||||
from ..utils import MM, Inch
|
from ..utils import MM, Inch
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -25,7 +26,7 @@ def extract_importlib(package):
|
||||||
else:
|
else:
|
||||||
assert item.is_dir()
|
assert item.is_dir()
|
||||||
item_out.mkdir()
|
item_out.mkdir()
|
||||||
stack.push((item, item_out))
|
stack.append((item, item_out))
|
||||||
|
|
||||||
return root
|
return root
|
||||||
|
|
||||||
|
|
@ -62,10 +63,10 @@ def deserialize(obj, unit):
|
||||||
case 'smd':
|
case 'smd':
|
||||||
match obj['pad_shape']:
|
match obj['pad_shape']:
|
||||||
case 'rect':
|
case 'rect':
|
||||||
pad = pb.SMDPad.rect(0, 0, pitch_x-clearance, pitch_y-clearance, paste=False, unit=unit)
|
stack = SMDStack.rect(pitch_x-clearance, pitch_y-clearance, paste=False, unit=unit)
|
||||||
case 'circle':
|
case 'circle':
|
||||||
pad = pb.SMDPad.circle(0, 0, min(pitch_x, pitch_y)-clearance, paste=False, unit=unit)
|
stack = SMDStack.circle(min(pitch_x, pitch_y)-clearance, paste=False, unit=unit)
|
||||||
return pb.PatternProtoArea(pitch_x, pitch_y, obj=pad, unit=unit)
|
return pb.PatternProtoArea(pitch_x, pitch_y, obj=stack, unit=unit)
|
||||||
|
|
||||||
case 'tht':
|
case 'tht':
|
||||||
hole_dia = mil(float(obj['hole_dia']))
|
hole_dia = mil(float(obj['hole_dia']))
|
||||||
|
|
@ -79,11 +80,11 @@ def deserialize(obj, unit):
|
||||||
|
|
||||||
match obj['pad_shape']:
|
match obj['pad_shape']:
|
||||||
case 'rect':
|
case 'rect':
|
||||||
pad = pb.THTPad.rect(0, 0, hole_dia, pitch_x-clearance, pitch_y-clearance, paste=False, plated=plated, unit=unit)
|
pad = pb.THTPad.rect(hole_dia, pitch_x-clearance, pitch_y-clearance, paste=False, plated=plated, unit=unit)
|
||||||
case 'circle':
|
case 'circle':
|
||||||
pad = pb.THTPad.circle(0, 0, hole_dia, min(pitch_x, pitch_y)-clearance, paste=False, plated=plated, unit=unit)
|
pad = pb.THTPad.circle(hole_dia, min(pitch_x, pitch_y)-clearance, paste=False, plated=plated, unit=unit)
|
||||||
case 'obround':
|
case 'obround':
|
||||||
pad = pb.THTPad.obround(0, 0, hole_dia, pitch_x-clearance, pitch_y-clearance, paste=False, plated=plated, unit=unit)
|
pad = pb.THTPad.obround(hole_dia, pitch_x-clearance, pitch_y-clearance, paste=False, plated=plated, unit=unit)
|
||||||
|
|
||||||
if oneside:
|
if oneside:
|
||||||
pad.pad_bottom = None
|
pad.pad_bottom = None
|
||||||
|
|
@ -106,7 +107,8 @@ def deserialize(obj, unit):
|
||||||
pitch = mil(float(obj.get('pitch', 2.54)))
|
pitch = mil(float(obj.get('pitch', 2.54)))
|
||||||
hole_dia = mil(float(obj['hole_dia']))
|
hole_dia = mil(float(obj['hole_dia']))
|
||||||
pattern_dia = mil(float(obj['pattern_dia']))
|
pattern_dia = mil(float(obj['pattern_dia']))
|
||||||
return pb.PatternProtoArea(2*pitch, 2*pitch, pb.THTFlowerProto(pitch, hole_dia, pattern_dia, unit=unit), unit=unit)
|
clearance = mil(float(obj['clearance']))
|
||||||
|
return pb.PatternProtoArea(pitch, pitch, pb.THTFlowerProto(pitch, hole_dia, pattern_dia, clearance, unit=unit), unit=unit)
|
||||||
|
|
||||||
case 'spiky':
|
case 'spiky':
|
||||||
return pb.PatternProtoArea(2.54, 2.54, pb.SpikyProto(), unit=unit)
|
return pb.PatternProtoArea(2.54, 2.54, pb.SpikyProto(), unit=unit)
|
||||||
|
|
@ -127,6 +129,20 @@ def deserialize(obj, unit):
|
||||||
via_size=via_size
|
via_size=via_size
|
||||||
), margin=unit(1.5, MM), unit=unit)
|
), margin=unit(1.5, MM), unit=unit)
|
||||||
|
|
||||||
|
case 'breadboard':
|
||||||
|
horizontal = obj.get('direction', 'v') == 'h'
|
||||||
|
drill = float(obj.get('hole_dia', 0.9))
|
||||||
|
return pb.BreadboardArea(clearance=clearance, drill=drill, horizontal=horizontal, unit=unit)
|
||||||
|
|
||||||
|
case 'starburst':
|
||||||
|
trace_width_x = float(obj.get('trace_width_x', 1.8))
|
||||||
|
trace_width_y = float(obj.get('trace_width_y', 1.8))
|
||||||
|
drill = float(obj.get('hole_dia', 0.9))
|
||||||
|
annular_ring = float(obj.get('annular', 1.2))
|
||||||
|
clearance = float(obj.get('clearance', 0.4))
|
||||||
|
mask_width = float(obj.get('mask_width', 0.5))
|
||||||
|
return pb.PatternProtoArea(pitch_x, pitch_y, pb.StarburstPad(pitch_x, pitch_y, trace_width_x, trace_width_y, clearance, mask_width, drill, annular_ring, unit=unit), unit=unit)
|
||||||
|
|
||||||
case 'rf':
|
case 'rf':
|
||||||
pitch = float(obj.get('pitch', 2.54))
|
pitch = float(obj.get('pitch', 2.54))
|
||||||
hole_dia = float(obj['hole_dia'])
|
hole_dia = float(obj['hole_dia'])
|
||||||
|
|
@ -139,6 +155,7 @@ def to_board(obj):
|
||||||
w = float(obj.get('width', unit(100, MM)))
|
w = float(obj.get('width', unit(100, MM)))
|
||||||
h = float(obj.get('height', unit(80, MM)))
|
h = float(obj.get('height', unit(80, MM)))
|
||||||
corner_radius = float(obj.get('round_corners', {}).get('radius', unit(1.5, MM)))
|
corner_radius = float(obj.get('round_corners', {}).get('radius', unit(1.5, MM)))
|
||||||
|
margin = float(obj.get('margin', unit(2.0, MM)))
|
||||||
holes = obj.get('mounting_holes', {})
|
holes = obj.get('mounting_holes', {})
|
||||||
mounting_hole_dia = float(holes.get('diameter', unit(3.2, MM)))
|
mounting_hole_dia = float(holes.get('diameter', unit(3.2, MM)))
|
||||||
mounting_hole_offset = float(holes.get('offset', unit(5, MM)))
|
mounting_hole_offset = float(holes.get('offset', unit(5, MM)))
|
||||||
|
|
@ -155,13 +172,14 @@ def to_board(obj):
|
||||||
corner_radius=corner_radius,
|
corner_radius=corner_radius,
|
||||||
mounting_hole_dia=mounting_hole_dia,
|
mounting_hole_dia=mounting_hole_dia,
|
||||||
mounting_hole_offset=mounting_hole_offset,
|
mounting_hole_offset=mounting_hole_offset,
|
||||||
|
margin=margin,
|
||||||
unit=unit)
|
unit=unit)
|
||||||
|
|
||||||
@app.route('/preview.svg', methods=['POST'])
|
@app.route('/preview_<side>.svg', methods=['POST'])
|
||||||
async def preview():
|
async def preview(side):
|
||||||
obj = await request.get_json()
|
obj = await request.get_json()
|
||||||
board = to_board(obj)
|
board = to_board(obj)
|
||||||
return Response(str(board.pretty_svg()), mimetype='image/svg+xml')
|
return Response(str(board.pretty_svg(side=side)), mimetype='image/svg+xml')
|
||||||
|
|
||||||
@app.route('/gerbers.zip', methods=['POST'])
|
@app.route('/gerbers.zip', methods=['POST'])
|
||||||
async def gerbers():
|
async def gerbers():
|
||||||
|
|
@ -172,7 +190,10 @@ async def gerbers():
|
||||||
board.layer_stack().save_to_zipfile(f)
|
board.layer_stack().save_to_zipfile(f)
|
||||||
return Response(f.read_bytes(), mimetype='image/svg+xml')
|
return Response(f.read_bytes(), mimetype='image/svg+xml')
|
||||||
|
|
||||||
|
def main():
|
||||||
if __name__ == '__main__':
|
|
||||||
app.run()
|
app.run()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|
||||||
|
|
||||||
|
|
@ -177,11 +177,14 @@ input[type="text"]:focus:valid {
|
||||||
position: relative;
|
position: relative;
|
||||||
grid-area: main;
|
grid-area: main;
|
||||||
padding: 20px;
|
padding: 20px;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
flex-wrap: wrap;
|
||||||
|
align-items: stretch;
|
||||||
}
|
}
|
||||||
|
|
||||||
#preview-image {
|
#preview > img {
|
||||||
width: 100%;
|
flex-grow: 1;
|
||||||
height: 100%;
|
|
||||||
object-fit: contain;
|
object-fit: contain;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -280,6 +283,12 @@ input[type="text"]:focus:valid {
|
||||||
<span class="unit us">inch</span>
|
<span class="unit us">inch</span>
|
||||||
</label>
|
</label>
|
||||||
|
|
||||||
|
<label>Margin
|
||||||
|
<input type="text" placeholder="margin" name="margin" value="2.0" pattern="[0-9]+\.?[0-9]*"/>
|
||||||
|
<span class="unit metric">mm</span>
|
||||||
|
<span class="unit us">inch</span>
|
||||||
|
</label>
|
||||||
|
|
||||||
<div class="group expand" data-group="round_corners">
|
<div class="group expand" data-group="round_corners">
|
||||||
<label>Round corners
|
<label>Round corners
|
||||||
<input name="enabled" type="checkbox" checked/>
|
<input name="enabled" type="checkbox" checked/>
|
||||||
|
|
@ -316,7 +325,8 @@ input[type="text"]:focus:valid {
|
||||||
</form>
|
</form>
|
||||||
</div>
|
</div>
|
||||||
<div id="preview">
|
<div id="preview">
|
||||||
<img id="preview-image" alt="Automatically generated preview image"/>
|
<img id="preview-image-top" alt="Automatically generated top side preview image"/>
|
||||||
|
<img id="preview-image-bottom" alt="Automatically generated bottom side preview image"/>
|
||||||
<div id="preview-message"></div>
|
<div id="preview-message"></div>
|
||||||
</div>
|
</div>
|
||||||
<div id="links">
|
<div id="links">
|
||||||
|
|
@ -401,6 +411,8 @@ input[type="text"]:focus:valid {
|
||||||
<a href="#" data-placeholder="rf" class="double-sided-only">RF THT area</a>
|
<a href="#" data-placeholder="rf" class="double-sided-only">RF THT area</a>
|
||||||
<a href="#" data-placeholder="spiky" class="double-sided-only">Spiky hybrid area</a>
|
<a href="#" data-placeholder="spiky" class="double-sided-only">Spiky hybrid area</a>
|
||||||
<a href="#" data-placeholder="alio" class="double-sided-only">ALio hybrid area</a>
|
<a href="#" data-placeholder="alio" class="double-sided-only">ALio hybrid area</a>
|
||||||
|
<a href="#" data-placeholder="starburst" class="double-sided-only">THT starburst area</a>
|
||||||
|
<a href="#" data-placeholder="breadboard" class="double-sided-only">Permanent breadboard area</a>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</template>
|
</template>
|
||||||
|
|
@ -468,7 +480,7 @@ input[type="text"]:focus:valid {
|
||||||
<span class="unit us">mil</span>
|
<span class="unit us">mil</span>
|
||||||
</label>
|
</label>
|
||||||
<label>Plating
|
<label>Plating
|
||||||
<select name="plating" value="through">
|
<select name="plating" value="plated">
|
||||||
<option value="plated">Double-sided, through-plated</option>
|
<option value="plated">Double-sided, through-plated</option>
|
||||||
<option value="nonplated">Double-sided, non-plated</option>
|
<option value="nonplated">Double-sided, non-plated</option>
|
||||||
<option value="singleside">Single-sided, non-plated</option>
|
<option value="singleside">Single-sided, non-plated</option>
|
||||||
|
|
@ -494,6 +506,34 @@ input[type="text"]:focus:valid {
|
||||||
</div>
|
</div>
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
|
<template id="tpl-g-breadboard">
|
||||||
|
<div data-type="breadboard" class="group breadboard">
|
||||||
|
<h4>Permanent breadboard area</h4>
|
||||||
|
<span class="content area-controls">(<a href="#" class="area-remove">Remove</a><a href="#" class="area-move">Move</a>)</span>
|
||||||
|
<label class="proportion">Proportion
|
||||||
|
<input type="text" name="layout_prop" value="1" pattern="[0-9]+\.?[0-9]*"/>
|
||||||
|
</label>
|
||||||
|
|
||||||
|
<h5>Area Settings</h5>
|
||||||
|
<label>Direction
|
||||||
|
<select name="direction" value="v">
|
||||||
|
<option value="v">Vertical</option>
|
||||||
|
<option value="h">Horizontal</option>
|
||||||
|
</select>
|
||||||
|
</label>
|
||||||
|
<label>Clearance
|
||||||
|
<input type="text" name="clearance" placeholder="length" value="0.5" pattern="[0-9]+\.?[0-9]*"/>
|
||||||
|
<span class="unit metric">mm</span>
|
||||||
|
<span class="unit us">mil</span>
|
||||||
|
</label>
|
||||||
|
<label>Hole diameter
|
||||||
|
<input type="text" name="hole_dia" placeholder="length" value="0.9" pattern="[0-9]+\.?[0-9]*"/>
|
||||||
|
<span class="unit metric">mm</span>
|
||||||
|
<span class="unit us">mil</span>
|
||||||
|
</label>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
<template id="tpl-g-manhattan">
|
<template id="tpl-g-manhattan">
|
||||||
<div data-type="manhattan" class="group manhattan">
|
<div data-type="manhattan" class="group manhattan">
|
||||||
<h4>Manhattan area</h4>
|
<h4>Manhattan area</h4>
|
||||||
|
|
@ -699,6 +739,58 @@ input[type="text"]:focus:valid {
|
||||||
</div>
|
</div>
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
|
<template id="tpl-g-starburst">
|
||||||
|
<div data-type="starburst" class="group starburst">
|
||||||
|
<h4>Starburst area</h4>
|
||||||
|
<span class="content area-controls">(<a href="#" class="area-remove">Remove</a><a href="#" class="area-move">Move</a>)</span>
|
||||||
|
<label class="proportion">Proportion
|
||||||
|
<input type="text" name="layout_prop" value="1" pattern="[0-9]+\.?[0-9]*"/>
|
||||||
|
</label>
|
||||||
|
|
||||||
|
<h5>Area Settings</h5>
|
||||||
|
<label>Pitch X
|
||||||
|
<input type="text" name="pitch_x" placeholder="length" value="2.54" pattern="[0-9]+\.?[0-9]*"/>
|
||||||
|
<span class="unit metric">mm</span>
|
||||||
|
<span class="unit us">mil</span>
|
||||||
|
</label>
|
||||||
|
<label>Pitch Y
|
||||||
|
<input type="text" name="pitch_y" placeholder="length" value="2.54" pattern="[0-9]+\.?[0-9]*"/>
|
||||||
|
<span class="unit metric">mm</span>
|
||||||
|
<span class="unit us">mil</span>
|
||||||
|
</label>
|
||||||
|
<label>Drill diameter
|
||||||
|
<input type="text" name="drill" placeholder="length" value="0.9" pattern="[0-9]+\.?[0-9]*"/>
|
||||||
|
<span class="unit metric">mm</span>
|
||||||
|
<span class="unit us">mil</span>
|
||||||
|
</label>
|
||||||
|
<label>Annular ring
|
||||||
|
<input type="text" name="annular" placeholder="length" value="1.2" pattern="[0-9]+\.?[0-9]*"/>
|
||||||
|
<span class="unit metric">mm</span>
|
||||||
|
<span class="unit us">mil</span>
|
||||||
|
</label>
|
||||||
|
<label>Pad clearance
|
||||||
|
<input type="text" name="clearance" placeholder="length" value="0.4" pattern="[0-9]+\.?[0-9]*"/>
|
||||||
|
<span class="unit metric">mm</span>
|
||||||
|
<span class="unit us">mil</span>
|
||||||
|
</label>
|
||||||
|
<label>Soldermask wall
|
||||||
|
<input type="text" name="mask_width" placeholder="length" value="0.5" pattern="[0-9]+\.?[0-9]*"/>
|
||||||
|
<span class="unit metric">mm</span>
|
||||||
|
<span class="unit us">mil</span>
|
||||||
|
</label>
|
||||||
|
<label>Trace width X
|
||||||
|
<input type="text" name="trace_width_x" placeholder="length" value="1.40" pattern="[0-9]+\.?[0-9]*"/>
|
||||||
|
<span class="unit metric">mm</span>
|
||||||
|
<span class="unit us">mil</span>
|
||||||
|
</label>
|
||||||
|
<label>Trace width Y
|
||||||
|
<input type="text" name="trace_width_y" placeholder="length" value="1.40" pattern="[0-9]+\.?[0-9]*"/>
|
||||||
|
<span class="unit metric">mm</span>
|
||||||
|
<span class="unit us">mil</span>
|
||||||
|
</label>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
<script>
|
<script>
|
||||||
document.querySelectorAll('.expand').forEach((elem) => {
|
document.querySelectorAll('.expand').forEach((elem) => {
|
||||||
const checkbox = elem.querySelector(':first-child > input');
|
const checkbox = elem.querySelector(':first-child > input');
|
||||||
|
|
@ -985,26 +1077,43 @@ input[type="text"]:focus:valid {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let previewBlobURL = null;
|
let previewTopBlobURL = null;
|
||||||
|
let previewBotBlobURL = null;
|
||||||
previewReloader = new RateLimiter(async () => {
|
previewReloader = new RateLimiter(async () => {
|
||||||
if (document.querySelector('form').checkValidity()) {
|
if (document.querySelector('form').checkValidity()) {
|
||||||
document.querySelector('#preview-message').textContent = 'Reloading...';
|
document.querySelector('#preview-message').textContent = 'Reloading...';
|
||||||
document.querySelector('#preview-message').classList.add('loading');
|
document.querySelector('#preview-message').classList.add('loading');
|
||||||
const response = await fetch('preview.svg', {
|
|
||||||
|
const response_top = await fetch('preview_top.svg', {
|
||||||
method: 'POST',
|
method: 'POST',
|
||||||
mode: 'same-origin',
|
mode: 'same-origin',
|
||||||
cache: 'no-cache',
|
cache: 'no-cache',
|
||||||
headers: {'Content-Type': 'application/json'},
|
headers: {'Content-Type': 'application/json'},
|
||||||
body: serialize(),
|
body: serialize(),
|
||||||
});
|
});
|
||||||
const data = await response.blob();
|
const data_top = await response_top.blob();
|
||||||
if (previewBlobURL) {
|
if (previewTopBlobURL) {
|
||||||
URL.revokeObjectURL(previewBlobURL);
|
URL.revokeObjectURL(previewTopBlobURL);
|
||||||
}
|
}
|
||||||
previewBlobURL = URL.createObjectURL(data);
|
previewTopBlobURL = URL.createObjectURL(data_top);
|
||||||
document.querySelector('#preview-image').src = previewBlobURL;
|
document.querySelector('#preview-image-top').src = previewTopBlobURL;
|
||||||
|
|
||||||
document.querySelector('#preview-message').textContent = '';
|
document.querySelector('#preview-message').textContent = '';
|
||||||
document.querySelector('#preview-message').classList.remove('loading');
|
document.querySelector('#preview-message').classList.remove('loading');
|
||||||
|
|
||||||
|
const response_bot = await fetch('preview_bottom.svg', {
|
||||||
|
method: 'POST',
|
||||||
|
mode: 'same-origin',
|
||||||
|
cache: 'no-cache',
|
||||||
|
headers: {'Content-Type': 'application/json'},
|
||||||
|
body: serialize(),
|
||||||
|
});
|
||||||
|
const data_bot = await response_bot.blob();
|
||||||
|
if (previewBotBlobURL) {
|
||||||
|
URL.revokeObjectURL(previewBotBlobURL);
|
||||||
|
}
|
||||||
|
previewBotBlobURL = URL.createObjectURL(data_bot);
|
||||||
|
document.querySelector('#preview-image-bottom').src = previewBotBlobURL;
|
||||||
} else {
|
} else {
|
||||||
document.querySelector('#preview-message').classList.add('loading');
|
document.querySelector('#preview-message').classList.add('loading');
|
||||||
document.querySelector('#preview-message').textContent = 'Please correct any invalid fields.';
|
document.querySelector('#preview-message').textContent = 'Please correct any invalid fields.';
|
||||||
|
|
@ -26,7 +26,7 @@ import shutil
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from functools import cached_property
|
from functools import cached_property
|
||||||
|
|
||||||
from .utils import LengthUnit, MM, Inch, Tag, sum_bounds, setup_svg
|
from .utils import LengthUnit, MM, Inch, Tag, sum_bounds, setup_svg, convex_hull
|
||||||
from . import graphic_primitives as gp
|
from . import graphic_primitives as gp
|
||||||
from . import graphic_objects as go
|
from . import graphic_objects as go
|
||||||
|
|
||||||
|
|
@ -54,6 +54,15 @@ class FileSettings:
|
||||||
zeros : bool = None
|
zeros : bool = None
|
||||||
#: Number format. ``(integer, decimal)`` tuple of number of integer and decimal digits. At most ``(6,7)`` by spec.
|
#: Number format. ``(integer, decimal)`` tuple of number of integer and decimal digits. At most ``(6,7)`` by spec.
|
||||||
number_format : tuple = (None, None)
|
number_format : tuple = (None, None)
|
||||||
|
#: At least the aperture macro implementations of gerbv and whatever JLCPCB uses are severely broken and simply
|
||||||
|
#: ignore parentheses in numeric expressions without throwing an error or a warning, leading to broken rendering.
|
||||||
|
#: To avoid trouble with severely broken software like this, we just calculate out all macros by default.
|
||||||
|
#: If you want to export the macros with their original formulaic expressions (which is completely fine by the
|
||||||
|
#: Gerber standard, btw), set this parameter to ``False`` before exporting.
|
||||||
|
calculate_out_all_aperture_macros: bool = True
|
||||||
|
#: Internal field used to communicate if only decimal coordinates were found inside an Excellon file, or if it
|
||||||
|
#: contained at least some coordinates in fixed-width notation.
|
||||||
|
_file_has_fixed_width_coordinates: bool = False
|
||||||
|
|
||||||
# input validation
|
# input validation
|
||||||
def __setattr__(self, name, value):
|
def __setattr__(self, name, value):
|
||||||
|
|
@ -342,6 +351,24 @@ class CamFile:
|
||||||
|
|
||||||
return sum_bounds(( p.bounding_box(unit) for p in self.objects ), default=default)
|
return sum_bounds(( p.bounding_box(unit) for p in self.objects ), default=default)
|
||||||
|
|
||||||
|
def convex_hull(self, tol=0.01, unit=None):
|
||||||
|
unit = unit or self.unit
|
||||||
|
points = []
|
||||||
|
|
||||||
|
for obj in self.objects:
|
||||||
|
if isinstance(obj, go.Line):
|
||||||
|
line = obj.as_primitive(unit)
|
||||||
|
points.append((line.x1, line.y1))
|
||||||
|
points.append((line.x2, line.y2))
|
||||||
|
|
||||||
|
elif isinstance(obj, go.Arc):
|
||||||
|
for obj in obj.approximate(tol, unit):
|
||||||
|
line = obj.as_primitive(unit)
|
||||||
|
points.append((line.x1, line.y1))
|
||||||
|
points.append((line.x2, line.y2))
|
||||||
|
|
||||||
|
return convex_hull(points)
|
||||||
|
|
||||||
def to_excellon(self):
|
def to_excellon(self):
|
||||||
""" Convert to a :py:class:`.ExcellonFile`. Returns ``self`` if it already is one. """
|
""" Convert to a :py:class:`.ExcellonFile`. Returns ``self`` if it already is one. """
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
|
@ -23,7 +23,10 @@ import dataclasses
|
||||||
import re
|
import re
|
||||||
import warnings
|
import warnings
|
||||||
import json
|
import json
|
||||||
|
import sys
|
||||||
import itertools
|
import itertools
|
||||||
|
import webbrowser
|
||||||
|
import warnings
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
from .utils import MM, Inch
|
from .utils import MM, Inch
|
||||||
|
|
@ -31,8 +34,23 @@ from .cam import FileSettings
|
||||||
from .rs274x import GerberFile
|
from .rs274x import GerberFile
|
||||||
from . import layers as lyr
|
from . import layers as lyr
|
||||||
from . import __version__
|
from . import __version__
|
||||||
|
from .cad.kicad import schematic as kc_schematic
|
||||||
|
from .cad.kicad import tmtheme
|
||||||
|
from .cad import protoserve
|
||||||
|
|
||||||
|
|
||||||
|
def _showwarning(message, category, filename, lineno, file=None, line=None):
|
||||||
|
if file is None:
|
||||||
|
file = sys.stderr
|
||||||
|
|
||||||
|
filename = Path(filename)
|
||||||
|
gerbonara_module_install_location = Path(__file__).parent.parent
|
||||||
|
if filename.is_relative_to(gerbonara_module_install_location):
|
||||||
|
filename = filename.relative_to(gerbonara_module_install_location)
|
||||||
|
|
||||||
|
print(f'{filename}:{lineno}: {message}', file=file)
|
||||||
|
warnings.showwarning = _showwarning
|
||||||
|
|
||||||
def _print_version(ctx, param, value):
|
def _print_version(ctx, param, value):
|
||||||
if value and not ctx.resilient_parsing:
|
if value and not ctx.resilient_parsing:
|
||||||
click.echo(f'Version {__version__}')
|
click.echo(f'Version {__version__}')
|
||||||
|
|
@ -128,6 +146,44 @@ def cli():
|
||||||
well as sets of those files """
|
well as sets of those files """
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
@cli.group('protoboard')
|
||||||
|
def protoboard_group():
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@protoboard_group.command()
|
||||||
|
@click.option('-h', '--host', default=None, help='Hostname to listen on. Defaults to localhost.')
|
||||||
|
@click.option('-p', '--port', type=int, default=1337, help='Port to listen on. Defaults to 1337')
|
||||||
|
def interactive(host, port):
|
||||||
|
''' Launch gerbonar's interactive protoboard designer in your browser '''
|
||||||
|
|
||||||
|
if host is None:
|
||||||
|
@protoserve.app.before_serving
|
||||||
|
async def open_browser():
|
||||||
|
webbrowser.open_new(f'http://localhost:{port}/')
|
||||||
|
protoserve.app.run(host=host, port=port, use_reloader=False, debug=False)
|
||||||
|
|
||||||
|
|
||||||
|
@cli.group('kicad')
|
||||||
|
def kicad_group():
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@kicad_group.group('schematic')
|
||||||
|
def schematic_group():
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@schematic_group.command()
|
||||||
|
@click.argument('inpath', type=click.Path(exists=True))
|
||||||
|
@click.argument('theme', type=click.Path(exists=True))
|
||||||
|
@click.argument('outfile', type=click.File('w'), default='-')
|
||||||
|
def render(inpath, theme, outfile):
|
||||||
|
sch = kc_schematic.Schematic.open(inpath)
|
||||||
|
cs = tmtheme.TmThemeSchematic(Path(theme).read_text())
|
||||||
|
with outfile as f:
|
||||||
|
f.write(str(sch.to_svg(cs)))
|
||||||
|
|
||||||
|
|
||||||
@cli.command()
|
@cli.command()
|
||||||
@click.option('--warnings', 'format_warnings', type=click.Choice(['default', 'ignore', 'once']), default='default',
|
@click.option('--warnings', 'format_warnings', type=click.Choice(['default', 'ignore', 'once']), default='default',
|
||||||
|
|
@ -142,12 +198,17 @@ def cli():
|
||||||
rules and use only rules given by --input-map''')
|
rules and use only rules given by --input-map''')
|
||||||
@click.option('--force-zip', is_flag=True, help='''Force treating input path as a zip file (default: guess file type
|
@click.option('--force-zip', is_flag=True, help='''Force treating input path as a zip file (default: guess file type
|
||||||
from extension and contents)''')
|
from extension and contents)''')
|
||||||
@click.option('--top/--bottom', default=True, help='Which side of the board to render')
|
@click.option('--top', 'side', flag_value='top', help='Render top side')
|
||||||
|
@click.option('--bottom', 'side', flag_value='bottom', help='Render top side')
|
||||||
@click.option('--command-line-units', type=Unit(), help='''Units for values given in other options. Default:
|
@click.option('--command-line-units', type=Unit(), help='''Units for values given in other options. Default:
|
||||||
millimeter''')
|
millimeter''')
|
||||||
@click.option('--margin', type=float, default=0.0, help='Add space around the board inside the viewport')
|
@click.option('--margin', type=float, default=0.0, help='Add space around the board inside the viewport')
|
||||||
@click.option('--force-bounds', help='Force SVG bounding box to value given as "min_x,min_y,max_x,max_y"')
|
@click.option('--force-bounds', help='Force SVG bounding box to value given as "min_x,min_y,max_x,max_y"')
|
||||||
@click.option('--inkscape/--standard-svg', default=True, help='Export in Inkscape SVG format with layers and stuff.')
|
@click.option('--inkscape/--standard-svg', default=True, help='Export in Inkscape SVG format with layers and stuff.')
|
||||||
|
@click.option('--pretty/--no-filters', default=True, help='''Export pseudo-realistic render using filters (default) or
|
||||||
|
just stack up layers using given colorscheme. In "--no-filters" mode, by default all layers are exported
|
||||||
|
unless either "--top" or "--bottom" is given.''')
|
||||||
|
@click.option('--drills/--no-drills', default=True, help='''Include (default) or exclude drills ("--no-filters" only!)''')
|
||||||
@click.option('--colorscheme', type=click.Path(exists=True, path_type=Path), help='''Load colorscheme from given JSON
|
@click.option('--colorscheme', type=click.Path(exists=True, path_type=Path), help='''Load colorscheme from given JSON
|
||||||
file. The JSON file must contain a single dict with keys copper, silk, mask, paste, drill and outline.
|
file. The JSON file must contain a single dict with keys copper, silk, mask, paste, drill and outline.
|
||||||
Each key must map to a string containing either a normal 6-digit hex color with leading hash sign, or an
|
Each key must map to a string containing either a normal 6-digit hex color with leading hash sign, or an
|
||||||
|
|
@ -155,8 +216,8 @@ def cli():
|
||||||
with FF being completely opaque, and 00 being invisibly transparent.''')
|
with FF being completely opaque, and 00 being invisibly transparent.''')
|
||||||
@click.argument('inpath', type=click.Path(exists=True))
|
@click.argument('inpath', type=click.Path(exists=True))
|
||||||
@click.argument('outfile', type=click.File('w'), default='-')
|
@click.argument('outfile', type=click.File('w'), default='-')
|
||||||
def render(inpath, outfile, format_warnings, input_map, use_builtin_name_rules, force_zip, top, command_line_units,
|
def render(inpath, outfile, format_warnings, input_map, use_builtin_name_rules, force_zip, side, drills,
|
||||||
margin, force_bounds, inkscape, colorscheme):
|
command_line_units, margin, force_bounds, inkscape, pretty, colorscheme):
|
||||||
""" Render a gerber file, or a directory or zip of gerber files into an SVG file. """
|
""" Render a gerber file, or a directory or zip of gerber files into an SVG file. """
|
||||||
|
|
||||||
overrides = json.loads(input_map.read_bytes()) if input_map else None
|
overrides = json.loads(input_map.read_bytes()) if input_map else None
|
||||||
|
|
@ -174,9 +235,14 @@ def render(inpath, outfile, format_warnings, input_map, use_builtin_name_rules,
|
||||||
if colorscheme:
|
if colorscheme:
|
||||||
colorscheme = json.loads(colorscheme.read_text())
|
colorscheme = json.loads(colorscheme.read_text())
|
||||||
|
|
||||||
outfile.write(str(stack.to_pretty_svg(side='top' if top else 'bottom', margin=margin,
|
if pretty:
|
||||||
arg_unit=(command_line_units or MM),
|
svg = stack.to_pretty_svg(side='bottom' if side == 'bottom' else 'top', margin=margin,
|
||||||
svg_unit=MM, force_bounds=force_bounds, inkscape=inkscape, colors=colorscheme)))
|
arg_unit=(command_line_units or MM),
|
||||||
|
svg_unit=MM, force_bounds=force_bounds, inkscape=inkscape, colors=colorscheme)
|
||||||
|
else:
|
||||||
|
svg = stack.to_svg(side_re=side or '.*', margin=margin, drills=drills, arg_unit=(command_line_units or MM),
|
||||||
|
svg_unit=MM, force_bounds=force_bounds, colors=colorscheme)
|
||||||
|
outfile.write(str(svg))
|
||||||
|
|
||||||
|
|
||||||
@cli.command()
|
@cli.command()
|
||||||
|
|
@ -272,9 +338,9 @@ def rewrite(transform, command_line_units, number_format, units, zero_suppressio
|
||||||
scheme instead of keeping the old file names.''')
|
scheme instead of keeping the old file names.''')
|
||||||
@click.argument('transform')
|
@click.argument('transform')
|
||||||
@click.argument('inpath')
|
@click.argument('inpath')
|
||||||
@click.argument('outpath')
|
@click.argument('outpath', type=click.Path(path_type=Path))
|
||||||
def transform(transform, units, output_format, inpath, outpath,
|
def transform(transform, units, output_format, inpath, outpath, format_warnings, input_map, use_builtin_name_rules,
|
||||||
format_warnings, input_map, use_builtin_name_rules, output_naming_scheme):
|
output_naming_scheme, number_format, force_zip):
|
||||||
""" Transform all gerber files in a given directory or zip file using the given python transformation script.
|
""" Transform all gerber files in a given directory or zip file using the given python transformation script.
|
||||||
|
|
||||||
In the python transformation script you have access to the functions translate(x, y), scale(factor) and
|
In the python transformation script you have access to the functions translate(x, y), scale(factor) and
|
||||||
|
|
@ -289,16 +355,26 @@ def transform(transform, units, output_format, inpath, outpath,
|
||||||
with warnings.catch_warnings():
|
with warnings.catch_warnings():
|
||||||
warnings.simplefilter(format_warnings)
|
warnings.simplefilter(format_warnings)
|
||||||
if force_zip:
|
if force_zip:
|
||||||
stack = lyr.LayerStack.open_zip(path, overrides=overrides, autoguess=use_builtin_name_rules)
|
stack = lyr.LayerStack.open_zip(inpath, overrides=overrides, autoguess=use_builtin_name_rules)
|
||||||
else:
|
else:
|
||||||
stack = lyr.LayerStack.open(path, overrides=overrides, autoguess=use_builtin_name_rules)
|
stack = lyr.LayerStack.open(inpath, overrides=overrides, autoguess=use_builtin_name_rules)
|
||||||
|
|
||||||
_apply_transform(transform, units, stack)
|
_apply_transform(transform, units, stack)
|
||||||
|
|
||||||
output_format = None if output_format == 'reuse' else FileSettings.defaults()
|
output_format = None if output_format == 'reuse' else FileSettings.defaults()
|
||||||
stack.save_to_directory(outpath, naming_scheme=output_naming_scheme or {},
|
if number_format:
|
||||||
gerber_settings=output_format,
|
if output_format is None:
|
||||||
excellon_settings=dataclasses.replace(output_format, zeros=None))
|
output_format = FileSettings.defaults()
|
||||||
|
a, _, b = number_format.partition('.')
|
||||||
|
output_format.number_format = (int(a), int(b))
|
||||||
|
if outpath.is_file() or outpath.suffix.lower() == '.zip':
|
||||||
|
stack.save_to_zipfile(outpath, naming_scheme=output_naming_scheme or {},
|
||||||
|
gerber_settings=output_format,
|
||||||
|
excellon_settings=dataclasses.replace(output_format, zeros=None))
|
||||||
|
else:
|
||||||
|
stack.save_to_directory(outpath, naming_scheme=output_naming_scheme or {},
|
||||||
|
gerber_settings=output_format,
|
||||||
|
excellon_settings=dataclasses.replace(output_format, zeros=None))
|
||||||
|
|
||||||
|
|
||||||
@cli.command()
|
@cli.command()
|
||||||
|
|
@ -379,7 +455,7 @@ def merge(inpath, outpath, offset, rotation, input_map, command_line_units, outp
|
||||||
@click.option('--version', is_flag=True, callback=_print_version, expose_value=False, is_eager=True)
|
@click.option('--version', is_flag=True, callback=_print_version, expose_value=False, is_eager=True)
|
||||||
@click.option('--warnings', 'format_warnings', type=click.Choice(['default', 'ignore', 'once']), default='default',
|
@click.option('--warnings', 'format_warnings', type=click.Choice(['default', 'ignore', 'once']), default='default',
|
||||||
help='''Enable or disable file format warnings during parsing (default: on)''')
|
help='''Enable or disable file format warnings during parsing (default: on)''')
|
||||||
@click.option('--units', type=Unit(), help='Output bounding box in this unit (default: millimeter)')
|
@click.option('--units', type=Unit(), default='metric', help='Output bounding box in this unit (default: millimeter)')
|
||||||
@click.option('--input-number-format', help='Override number format of input file (mostly useful for Excellon files)')
|
@click.option('--input-number-format', help='Override number format of input file (mostly useful for Excellon files)')
|
||||||
@click.option('--input-units', type=Unit(), help='Override units of input file')
|
@click.option('--input-units', type=Unit(), help='Override units of input file')
|
||||||
@click.option('--input-zero-suppression', type=click.Choice(['off', 'leading', 'trailing']), help='Override zero suppression setting of input file')
|
@click.option('--input-zero-suppression', type=click.Choice(['off', 'leading', 'trailing']), help='Override zero suppression setting of input file')
|
||||||
|
|
@ -30,9 +30,10 @@ from pathlib import Path
|
||||||
|
|
||||||
from .cam import CamFile, FileSettings
|
from .cam import CamFile, FileSettings
|
||||||
from .graphic_objects import Flash, Line, Arc
|
from .graphic_objects import Flash, Line, Arc
|
||||||
from .apertures import ExcellonTool
|
from .apertures import ExcellonTool, CircleAperture
|
||||||
from .utils import Inch, MM, to_unit, InterpMode, RegexMatcher
|
from .utils import Inch, MM, to_unit, InterpMode, RegexMatcher
|
||||||
|
|
||||||
|
|
||||||
class ExcellonContext:
|
class ExcellonContext:
|
||||||
""" Internal helper class used for tracking graphics state when writing Excellon. """
|
""" Internal helper class used for tracking graphics state when writing Excellon. """
|
||||||
|
|
||||||
|
|
@ -268,17 +269,19 @@ class ExcellonFile(CamFile):
|
||||||
""" Counterpart to :py:meth:`~.rs274x.GerberFile.to_excellon`. Does nothing and returns :py:obj:`self`. """
|
""" Counterpart to :py:meth:`~.rs274x.GerberFile.to_excellon`. Does nothing and returns :py:obj:`self`. """
|
||||||
return self
|
return self
|
||||||
|
|
||||||
def to_gerber(self, errros='raise'):
|
def to_gerber(self, errors='raise'):
|
||||||
""" Convert this excellon file into a :py:class:`~.rs274x.GerberFile`. """
|
""" Convert this excellon file into a :py:class:`~.rs274x.GerberFile`. """
|
||||||
|
from .rs274x import GerberFile
|
||||||
out = GerberFile()
|
out = GerberFile()
|
||||||
out.comments = self.comments
|
out.comments = self.comments
|
||||||
|
|
||||||
apertures = {}
|
apertures = {}
|
||||||
for obj in self.objects:
|
for obj in self.objects:
|
||||||
if not (ap := apertures[obj.tool]):
|
if not (ap := apertures.get(obj.tool)):
|
||||||
ap = apertures[obj.tool] = CircleAperture(obj.tool.diameter)
|
ap = apertures[obj.tool] = CircleAperture(obj.tool.diameter, unit=obj.aperture.unit)
|
||||||
|
|
||||||
out.objects.append(dataclasses.replace(obj, aperture=ap))
|
out.objects.append(dataclasses.replace(obj, aperture=ap))
|
||||||
|
return out
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def generator(self):
|
def generator(self):
|
||||||
|
|
@ -325,7 +328,7 @@ class ExcellonFile(CamFile):
|
||||||
for fn in 'nc_param.txt', 'ncdrill.log':
|
for fn in 'nc_param.txt', 'ncdrill.log':
|
||||||
if (param_file := filename.parent / fn).is_file():
|
if (param_file := filename.parent / fn).is_file():
|
||||||
settings = parse_allegro_ncparam(param_file.read_text())
|
settings = parse_allegro_ncparam(param_file.read_text())
|
||||||
warnings.warn(f'Loaded allegro-style excellon settings file {param_file}')
|
warnings.warn(f'Loaded allegro-style excellon settings file {param_file}', SyntaxWarning)
|
||||||
break
|
break
|
||||||
|
|
||||||
# Parse Zuken log file for settings
|
# Parse Zuken log file for settings
|
||||||
|
|
@ -333,7 +336,7 @@ class ExcellonFile(CamFile):
|
||||||
logfile = filename.with_suffix('.fdl')
|
logfile = filename.with_suffix('.fdl')
|
||||||
if logfile.is_file():
|
if logfile.is_file():
|
||||||
settings = parse_zuken_logfile(logfile.read_text())
|
settings = parse_zuken_logfile(logfile.read_text())
|
||||||
warnings.warn(f'Loaded zuken-style excellon log file {logfile}: {settings}')
|
warnings.warn(f'Loaded zuken-style excellon log file {logfile}: {settings}', SyntaxWarning)
|
||||||
|
|
||||||
if external_tools is None:
|
if external_tools is None:
|
||||||
# Parse allegro log files for tools.
|
# Parse allegro log files for tools.
|
||||||
|
|
@ -376,7 +379,7 @@ class ExcellonFile(CamFile):
|
||||||
|
|
||||||
mixed_plating = (len({ tool.plated for tool in tool_map.values() }) > 1)
|
mixed_plating = (len({ tool.plated for tool in tool_map.values() }) > 1)
|
||||||
if mixed_plating:
|
if mixed_plating:
|
||||||
warnings.warn('Multiple plating values in same file. Will use non-standard Altium comment syntax to indicate hole plating.')
|
warnings.warn('Multiple plating values in same file. Will use non-standard Altium comment syntax to indicate hole plating.', SyntaxWarning)
|
||||||
|
|
||||||
defined_tools = {}
|
defined_tools = {}
|
||||||
tool_indices = {}
|
tool_indices = {}
|
||||||
|
|
@ -566,6 +569,8 @@ class ExcellonParser(object):
|
||||||
self.filename = None
|
self.filename = None
|
||||||
self.external_tools = external_tools or {}
|
self.external_tools = external_tools or {}
|
||||||
self.found_kicad_format_comment = False
|
self.found_kicad_format_comment = False
|
||||||
|
self.allegro_eof_toolchange_hack = False
|
||||||
|
self.allegro_eof_toolchange_hack_index = 1
|
||||||
|
|
||||||
def warn(self, msg):
|
def warn(self, msg):
|
||||||
warnings.warn(f'{self.filename}:{self.lineno} "{self.line}": {msg}', SyntaxWarning)
|
warnings.warn(f'{self.filename}:{self.lineno} "{self.line}": {msg}', SyntaxWarning)
|
||||||
|
|
@ -606,18 +611,25 @@ class ExcellonParser(object):
|
||||||
exprs = RegexMatcher()
|
exprs = RegexMatcher()
|
||||||
|
|
||||||
# NOTE: These must be kept before the generic comment handler at the end of this class so they match first.
|
# NOTE: These must be kept before the generic comment handler at the end of this class so they match first.
|
||||||
@exprs.match(r';T(?P<index1>[0-9]+) Holesize (?P<index2>[0-9]+)\. = (?P<diameter>[0-9/.]+) Tolerance = \+[0-9/.]+/-[0-9/.]+ (?P<plated>PLATED|NON_PLATED|OPTIONAL) (?P<unit>MILS|MM) Quantity = [0-9]+')
|
@exprs.match(r';(?P<index1_prefix>T(?P<index1>[0-9]+))?\s+Holesize (?P<index2>[0-9]+)\. = (?P<diameter>[0-9/.]+) Tolerance = \+[0-9/.]+/-[0-9/.]+ (?P<plated>PLATED|NON_PLATED|OPTIONAL) (?P<unit>MILS|MM) Quantity = [0-9]+')
|
||||||
def parse_allegro_tooldef(self, match):
|
def parse_allegro_tooldef(self, match):
|
||||||
# NOTE: We ignore the given tolerances here since they are non-standard.
|
# NOTE: We ignore the given tolerances here since they are non-standard.
|
||||||
self.program_state = ProgramState.HEADER # TODO is this needed? we need a test file.
|
self.program_state = ProgramState.HEADER # TODO is this needed? we need a test file.
|
||||||
self.generator_hints.append('allegro')
|
self.generator_hints.append('allegro')
|
||||||
|
|
||||||
if (index := int(match['index1'])) != int(match['index2']): # index1 has leading zeros, index2 not.
|
index = int(match['index2'])
|
||||||
|
|
||||||
|
if match['index1'] and index != int(match['index1']): # index1 has leading zeros, index2 not.
|
||||||
raise SyntaxError('BUG: Allegro excellon tool def has mismatching tool indices. Please file a bug report on our issue tracker and provide this file!')
|
raise SyntaxError('BUG: Allegro excellon tool def has mismatching tool indices. Please file a bug report on our issue tracker and provide this file!')
|
||||||
|
|
||||||
if index in self.tools:
|
if index in self.tools:
|
||||||
self.warn('Re-definition of tool index {index}, overwriting old definition.')
|
self.warn('Re-definition of tool index {index}, overwriting old definition.')
|
||||||
|
|
||||||
|
if not match['index1_prefix']:
|
||||||
|
# This is a really nasty orcad file without tool change commands, that instead just puts all holes in order
|
||||||
|
# of the hole size definitions with M00's in between.
|
||||||
|
self.allegro_eof_toolchange_hack = True
|
||||||
|
|
||||||
# NOTE: We map "optionally" plated holes to plated holes for API simplicity. If you hit a case where that's a
|
# NOTE: We map "optionally" plated holes to plated holes for API simplicity. If you hit a case where that's a
|
||||||
# problem, please raise an issue on our issue tracker, explain why you need this and provide an example file.
|
# problem, please raise an issue on our issue tracker, explain why you need this and provide an example file.
|
||||||
is_plated = None if match['plated'] is None else (match['plated'] in ('PLATED', 'OPTIONAL'))
|
is_plated = None if match['plated'] is None else (match['plated'] in ('PLATED', 'OPTIONAL'))
|
||||||
|
|
@ -630,13 +642,19 @@ class ExcellonParser(object):
|
||||||
else:
|
else:
|
||||||
unit = MM
|
unit = MM
|
||||||
|
|
||||||
if unit != self.settings.unit:
|
if self.settings.unit is None:
|
||||||
|
self.settings.unit = unit
|
||||||
|
|
||||||
|
elif unit != self.settings.unit:
|
||||||
self.warn('Allegro Excellon drill file tool definitions in {unit.name}, but file parameters say the '
|
self.warn('Allegro Excellon drill file tool definitions in {unit.name}, but file parameters say the '
|
||||||
'file should be in {settings.unit.name}. Please double-check that this is correct, and if it is, '
|
'file should be in {settings.unit.name}. Please double-check that this is correct, and if it is, '
|
||||||
'please raise an issue on our issue tracker.')
|
'please raise an issue on our issue tracker.')
|
||||||
|
|
||||||
self.tools[index] = ExcellonTool(diameter=diameter, plated=is_plated, unit=unit)
|
self.tools[index] = ExcellonTool(diameter=diameter, plated=is_plated, unit=unit)
|
||||||
|
|
||||||
|
if self.allegro_eof_toolchange_hack and self.active_tool is None:
|
||||||
|
self.active_tool = self.tools[index]
|
||||||
|
|
||||||
# Searching Github I found that EasyEDA has two different variants of the unit specification here.
|
# Searching Github I found that EasyEDA has two different variants of the unit specification here.
|
||||||
@exprs.match(';Holesize (?P<index>[0-9]+) = (?P<diameter>[.0-9]+) (?P<unit>INCH|inch|METRIC|mm)')
|
@exprs.match(';Holesize (?P<index>[0-9]+) = (?P<diameter>[.0-9]+) (?P<unit>INCH|inch|METRIC|mm)')
|
||||||
def parse_easyeda_tooldef(self, match):
|
def parse_easyeda_tooldef(self, match):
|
||||||
|
|
@ -753,6 +771,12 @@ class ExcellonParser(object):
|
||||||
def handle_end_of_program(self, match):
|
def handle_end_of_program(self, match):
|
||||||
if self.program_state in (None, ProgramState.HEADER):
|
if self.program_state in (None, ProgramState.HEADER):
|
||||||
self.warn('M30 statement found before end of header.')
|
self.warn('M30 statement found before end of header.')
|
||||||
|
|
||||||
|
if self.allegro_eof_toolchange_hack:
|
||||||
|
self.allegro_eof_toolchange_hack_index = min(max(self.tools), self.allegro_eof_toolchange_hack_index + 1)
|
||||||
|
self.active_tool = self.tools[self.allegro_eof_toolchange_hack_index]
|
||||||
|
return
|
||||||
|
|
||||||
self.program_state = ProgramState.FINISHED
|
self.program_state = ProgramState.FINISHED
|
||||||
# TODO: maybe add warning if this is followed by other commands.
|
# TODO: maybe add warning if this is followed by other commands.
|
||||||
|
|
||||||
|
|
@ -762,14 +786,17 @@ class ExcellonParser(object):
|
||||||
def do_move(self, coord_groups):
|
def do_move(self, coord_groups):
|
||||||
x_s, x, y_s, y = coord_groups
|
x_s, x, y_s, y = coord_groups
|
||||||
|
|
||||||
if self.settings.number_format == (None, None) and '.' not in x:
|
if (x is not None and '.' not in x) or (y is not None and '.' not in y):
|
||||||
# TARGET3001! exports zeros as "00" even when it uses an explicit decimal point everywhere else.
|
self.settings._file_has_fixed_width_coordinates = True
|
||||||
if x != '00':
|
|
||||||
raise SyntaxError('No number format set and value does not contain a decimal point. If this is an Allegro '
|
if self.settings.number_format == (None, None):
|
||||||
'Excellon drill file make sure either nc_param.txt or ncdrill.log ends up in the same folder as '
|
# TARGET3001! exports zeros as "00" even when it uses an explicit decimal point everywhere else.
|
||||||
'it, because Allegro does not include this critical information in their Excellon output. If you '
|
if x != '00':
|
||||||
'call this through ExcellonFile.from_string, you must manually supply from_string with a '
|
raise SyntaxError('No number format set and value does not contain a decimal point. If this is an Allegro '
|
||||||
'FileSettings object from excellon.parse_allegro_ncparam.')
|
'Excellon drill file make sure either nc_param.txt or ncdrill.log ends up in the same folder as '
|
||||||
|
'it, because Allegro does not include this critical information in their Excellon output. If you '
|
||||||
|
'call this through ExcellonFile.from_string, you must manually supply from_string with a '
|
||||||
|
'FileSettings object from excellon.parse_allegro_ncparam.')
|
||||||
|
|
||||||
x = self.settings.parse_gerber_value(x)
|
x = self.settings.parse_gerber_value(x)
|
||||||
if x_s:
|
if x_s:
|
||||||
|
|
@ -863,12 +890,17 @@ class ExcellonParser(object):
|
||||||
# from https://math.stackexchange.com/a/1781546
|
# from https://math.stackexchange.com/a/1781546
|
||||||
if a_s:
|
if a_s:
|
||||||
raise ValueError('Negative arc radius given')
|
raise ValueError('Negative arc radius given')
|
||||||
r = settings.parse_gerber_value(a)
|
r = self.settings.parse_gerber_value(a)
|
||||||
x1, y1 = start
|
x1, y1 = start
|
||||||
x2, y2 = end
|
x2, y2 = end
|
||||||
dx, dy = (x2-x1)/2, (y2-y1)/2
|
dx, dy = (x2-x1)/2, (y2-y1)/2
|
||||||
x0, y0 = x1+dx, y1+dy
|
x0, y0 = x1+dx, y1+dy
|
||||||
f = math.hypot(dx, dy) / math.sqrt(r**2 - a**2)
|
d = math.hypot(dx, dy)
|
||||||
|
if d == 0:
|
||||||
|
raise ValueError('Arc radius notation requires distinct start and end points')
|
||||||
|
if r < d:
|
||||||
|
raise ValueError('Arc radius too small for endpoint distance')
|
||||||
|
f = math.sqrt(r**2 - d**2) / d
|
||||||
if clockwise:
|
if clockwise:
|
||||||
cx = x0 + f*dy
|
cx = x0 + f*dy
|
||||||
cy = y0 - f*dx
|
cy = y0 - f*dx
|
||||||
|
|
@ -878,16 +910,16 @@ class ExcellonParser(object):
|
||||||
i, j = cx-start[0], cy-start[1]
|
i, j = cx-start[0], cy-start[1]
|
||||||
|
|
||||||
else: # explicit center given
|
else: # explicit center given
|
||||||
i = settings.parse_gerber_value(i)
|
i = self.settings.parse_gerber_value(i) or 0
|
||||||
if i_s:
|
if i_s:
|
||||||
i = -i
|
i = -i
|
||||||
j = settings.parse_gerber_value(j)
|
j = self.settings.parse_gerber_value(j) or 0
|
||||||
if j_s:
|
if j_s:
|
||||||
j = -i
|
j = -j
|
||||||
|
|
||||||
self.objects.append(Arc(*start, *end, i, j, True, self.active_tool, unit=self.settings.unit))
|
self.objects.append(Arc(*start, *end, i, j, clockwise, self.active_tool, unit=self.settings.unit))
|
||||||
|
|
||||||
@exprs.match(r'(M71|METRIC|M72|INCH)(,LZ|,TZ)?(,0*\.0*)?')
|
@exprs.match(r'(M71|METRIC|M72|INCH)(,LZ|,TZ)?(,([0-9]+\.[0-9]+|0*\.0*))?')
|
||||||
def parse_easyeda_format(self, match):
|
def parse_easyeda_format(self, match):
|
||||||
metric = match[1] in ('METRIC', 'M71')
|
metric = match[1] in ('METRIC', 'M71')
|
||||||
|
|
||||||
|
|
@ -900,7 +932,10 @@ class ExcellonParser(object):
|
||||||
# This is used by newer autodesk eagles, fritzing and diptrace
|
# This is used by newer autodesk eagles, fritzing and diptrace
|
||||||
if match[3]:
|
if match[3]:
|
||||||
integer, _, fractional = match[3][1:].partition('.')
|
integer, _, fractional = match[3][1:].partition('.')
|
||||||
self.settings.number_format = len(integer), len(fractional)
|
if integer.strip('0') or fractional.strip('0'):
|
||||||
|
self.settings.number_format = int(integer), int(fractional)
|
||||||
|
else:
|
||||||
|
self.settings.number_format = len(integer), len(fractional)
|
||||||
|
|
||||||
elif self.settings.number_format == (None, None) and not metric and not self.found_kicad_format_comment:
|
elif self.settings.number_format == (None, None) and not metric and not self.found_kicad_format_comment:
|
||||||
self.warn('Using implicit number format from bare "INCH" statement. This is normal for Fritzing, Diptrace, Geda and pcb-rnd.')
|
self.warn('Using implicit number format from bare "INCH" statement. This is normal for Fritzing, Diptrace, Geda and pcb-rnd.')
|
||||||
|
|
@ -926,10 +961,10 @@ class ExcellonParser(object):
|
||||||
@exprs.match('(FMAT|VER),?([0-9]*)')
|
@exprs.match('(FMAT|VER),?([0-9]*)')
|
||||||
def handle_command_format(self, match):
|
def handle_command_format(self, match):
|
||||||
if match[1] == 'FMAT':
|
if match[1] == 'FMAT':
|
||||||
# We do not support integer/fractional decimals specification via FMAT because that's stupid. If you need this,
|
# We only use FMAT as a compatibility marker. Version 1 drill files encountered in the wild still use the
|
||||||
# please raise an issue on our issue tracker, provide a sample file and tell us where on earth you found that
|
# same coordinate and routing statements that we already support, so rejecting the header unconditionally
|
||||||
# file.
|
# needlessly breaks otherwise parseable files.
|
||||||
if match[2] not in ('', '2'):
|
if match[2] not in ('', '1', '2'):
|
||||||
raise SyntaxError(f'Unsupported FMAT format version {match[2]}')
|
raise SyntaxError(f'Unsupported FMAT format version {match[2]}')
|
||||||
|
|
||||||
else: # VER
|
else: # VER
|
||||||
|
|
@ -958,6 +993,19 @@ class ExcellonParser(object):
|
||||||
else:
|
else:
|
||||||
self.warn('Bare coordinate after end of file')
|
self.warn('Bare coordinate after end of file')
|
||||||
|
|
||||||
|
@exprs.match(xy_coord + 'G85' + xy_coord)
|
||||||
|
def handle_g85_slot(self, match):
|
||||||
|
if self.program_state == ProgramState.HEADER:
|
||||||
|
return
|
||||||
|
|
||||||
|
self.do_move(match.groups()[:4])
|
||||||
|
start, end = self.do_move(match.groups()[4:])
|
||||||
|
|
||||||
|
if not self.ensure_active_tool():
|
||||||
|
return
|
||||||
|
|
||||||
|
self.objects.append(Line(*start, *end, self.active_tool, unit=self.settings.unit))
|
||||||
|
|
||||||
@exprs.match(r'DETECT,ON|ATC,ON|M06')
|
@exprs.match(r'DETECT,ON|ATC,ON|M06')
|
||||||
def parse_zuken_legacy_statements(self, match):
|
def parse_zuken_legacy_statements(self, match):
|
||||||
self.generator_hints.append('zuken')
|
self.generator_hints.append('zuken')
|
||||||
|
|
@ -19,9 +19,9 @@
|
||||||
import math
|
import math
|
||||||
import copy
|
import copy
|
||||||
from dataclasses import dataclass, astuple, field, fields
|
from dataclasses import dataclass, astuple, field, fields
|
||||||
from itertools import zip_longest
|
from itertools import zip_longest, pairwise, islice, cycle
|
||||||
|
|
||||||
from .utils import MM, InterpMode, to_unit, rotate_point, sum_bounds
|
from .utils import MM, InterpMode, to_unit, rotate_point, sum_bounds, approximate_arc, sweep_angle
|
||||||
from . import graphic_primitives as gp
|
from . import graphic_primitives as gp
|
||||||
from .aperture_macros import primitive as amp
|
from .aperture_macros import primitive as amp
|
||||||
|
|
||||||
|
|
@ -278,9 +278,15 @@ class Region(GraphicObject):
|
||||||
* A region is always exactly one connected component.
|
* A region is always exactly one connected component.
|
||||||
* A region must not overlap itself anywhere.
|
* A region must not overlap itself anywhere.
|
||||||
* A region cannot have holes.
|
* A region cannot have holes.
|
||||||
|
* The last outline point of the region must be equal to the first.
|
||||||
|
|
||||||
There is one exception from the last two rules: To emulate a region with a hole in it, *cut-ins* are allowed. At a
|
There is one exception from the last two rules: To emulate a region with a hole in it, *cut-ins* are allowed. At a
|
||||||
cut-in, the region is allowed to touch (but never overlap!) itself.
|
cut-in, the region is allowed to touch (but never overlap!) itself.
|
||||||
|
|
||||||
|
When ``arc_centers`` is empty, this region has only straight outline segments. When ``arc_centers`` is not empty,
|
||||||
|
the i-th entry defines the i-th outline segment, with a ``None`` entry designating a straight line segment.
|
||||||
|
An arc is defined by a ``(clockwise, (cx, cy))`` tuple, where ``clockwise`` can be ``True`` for a clockwise arc, or
|
||||||
|
``False`` for a counter-clockwise arc. ``cx`` and ``cy`` are the absolute coordinates of the arc's center.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self, outline=None, arc_centers=None, *, unit=MM, polarity_dark=True):
|
def __init__(self, outline=None, arc_centers=None, *, unit=MM, polarity_dark=True):
|
||||||
|
|
@ -288,6 +294,7 @@ class Region(GraphicObject):
|
||||||
self.polarity_dark = polarity_dark
|
self.polarity_dark = polarity_dark
|
||||||
self.outline = [] if outline is None else outline
|
self.outline = [] if outline is None else outline
|
||||||
self.arc_centers = [] if arc_centers is None else arc_centers
|
self.arc_centers = [] if arc_centers is None else arc_centers
|
||||||
|
self.close()
|
||||||
|
|
||||||
def __len__(self):
|
def __len__(self):
|
||||||
return len(self.outline)
|
return len(self.outline)
|
||||||
|
|
@ -300,12 +307,13 @@ class Region(GraphicObject):
|
||||||
|
|
||||||
def _offset(self, dx, dy):
|
def _offset(self, dx, dy):
|
||||||
self.outline = [ (x+dx, y+dy) for x, y in self.outline ]
|
self.outline = [ (x+dx, y+dy) for x, y in self.outline ]
|
||||||
|
self.arc_centers = [ (c[0], (c[1][0]+dx, c[1][1]+dy)) if c else None for c in self.arc_centers ]
|
||||||
|
|
||||||
def _rotate(self, angle, cx=0, cy=0):
|
def _rotate(self, angle, cx=0, cy=0):
|
||||||
self.outline = [ gp.rotate_point(x, y, angle, cx, cy) for x, y in self.outline ]
|
self.outline = [ gp.rotate_point(x, y, angle, cx, cy) for x, y in self.outline ]
|
||||||
self.arc_centers = [
|
self.arc_centers = [
|
||||||
(arc[0], gp.rotate_point(*arc[1], angle, cx-p[0], cy-p[1])) if arc else None
|
(arc[0], gp.rotate_point(*arc[1], angle, cx, cy)) if arc else None
|
||||||
for p, arc in zip_longest(self.outline, self.arc_centers) ]
|
for arc in self.arc_centers ]
|
||||||
|
|
||||||
def _scale(self, factor):
|
def _scale(self, factor):
|
||||||
self.outline = [ (x*factor, y*factor) for x, y in self.outline ]
|
self.outline = [ (x*factor, y*factor) for x, y in self.outline ]
|
||||||
|
|
@ -313,6 +321,12 @@ class Region(GraphicObject):
|
||||||
(arc[0], (arc[1][0]*factor, arc[1][1]*factor)) if arc else None
|
(arc[0], (arc[1][0]*factor, arc[1][1]*factor)) if arc else None
|
||||||
for p, arc in zip_longest(self.outline, self.arc_centers) ]
|
for p, arc in zip_longest(self.outline, self.arc_centers) ]
|
||||||
|
|
||||||
|
def close(self):
|
||||||
|
if self.outline and self.outline[-1] != self.outline[0]:
|
||||||
|
self.outline.append(self.outline[0])
|
||||||
|
if self.arc_centers:
|
||||||
|
self.arc_centers.append((None, (None, None)))
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_rectangle(kls, x, y, w, h, unit=MM):
|
def from_rectangle(kls, x, y, w, h, unit=MM):
|
||||||
return kls([
|
return kls([
|
||||||
|
|
@ -322,6 +336,11 @@ class Region(GraphicObject):
|
||||||
(x, y+h),
|
(x, y+h),
|
||||||
], unit=unit)
|
], unit=unit)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def from_arc_poly(kls, arc_poly, polarity_dark=None, unit=MM):
|
||||||
|
polarity = arc_poly.polarity_dark if polarity_dark is None else polarity_dark
|
||||||
|
return kls(arc_poly.outline, arc_poly.arc_centers, polarity_dark=polarity, unit=unit)
|
||||||
|
|
||||||
def append(self, obj):
|
def append(self, obj):
|
||||||
if obj.unit != self.unit:
|
if obj.unit != self.unit:
|
||||||
obj = obj.converted(self.unit)
|
obj = obj.converted(self.unit)
|
||||||
|
|
@ -331,47 +350,52 @@ class Region(GraphicObject):
|
||||||
self.outline.append(obj.p2)
|
self.outline.append(obj.p2)
|
||||||
|
|
||||||
if isinstance(obj, Arc):
|
if isinstance(obj, Arc):
|
||||||
self.arc_centers.append((obj.clockwise, obj.center_relative))
|
self.arc_centers.append((obj.clockwise, obj.center))
|
||||||
else:
|
else:
|
||||||
self.arc_centers.append(None)
|
self.arc_centers.append(None)
|
||||||
|
|
||||||
def close(self):
|
def iter_segments(self, tolerance=1e-6):
|
||||||
if not self.outline:
|
for points, arc in zip_longest(pairwise(self.outline), self.arc_centers):
|
||||||
return
|
if arc:
|
||||||
|
if points:
|
||||||
|
yield *points, arc
|
||||||
|
else:
|
||||||
|
yield self.outline[-1], self.outline[0], arc
|
||||||
|
return
|
||||||
|
else:
|
||||||
|
if not points:
|
||||||
|
break
|
||||||
|
yield *points, (None, (None, None))
|
||||||
|
|
||||||
if self.outline[-1] != self.outline[0]:
|
# Close outline if necessary.
|
||||||
self.outline.append(self.outline[0])
|
if math.dist(self.outline[0], self.outline[-1]) > tolerance:
|
||||||
|
yield self.outline[-1], self.outline[0], (None, (None, None))
|
||||||
|
|
||||||
def outline_objects(self, aperture=None):
|
def outline_objects(self, aperture=None):
|
||||||
for p1, p2, arc in zip_longest(self.outline, self.outline[1:] + self.outline[:1], self.arc_centers):
|
for p1, p2, (clockwise, center) in self.iter_segments():
|
||||||
if arc:
|
if clockwise is not None:
|
||||||
clockwise, pc = arc
|
yield Arc(*p1, *p2, *center, clockwise, aperture=aperture, unit=self.unit, polarity_dark=self.polarity_dark)
|
||||||
yield Arc(*p1, *p2, *pc, clockwise, aperture=aperture, unit=self.unit, polarity_dark=self.polarity_dark)
|
|
||||||
else:
|
else:
|
||||||
yield Line(*p1, *p2, aperture=aperture, unit=self.unit, polarity_dark=self.polarity_dark)
|
yield Line(*p1, *p2, aperture=aperture, unit=self.unit, polarity_dark=self.polarity_dark)
|
||||||
|
|
||||||
def _aperture_macro_primitives(self, max_error=1e-2, unit=MM):
|
def _aperture_macro_primitives(self, max_error=1e-2, clip_max_error=True, unit=MM):
|
||||||
# unit is only for max_error, the resulting primitives will always be in MM
|
# unit is only for max_error, the resulting primitives will always be in MM
|
||||||
|
|
||||||
if len(self.outline) < 2:
|
if len(self.outline) < 2:
|
||||||
return
|
return
|
||||||
|
|
||||||
points = [self.outline[0]]
|
points = []
|
||||||
for p1, p2, arc in zip_longest(self.outline[:-1], self.outline[1:], self.arc_centers):
|
for p1, p2, (clockwise, center) in self.iter_segments():
|
||||||
if arc:
|
if clockwise is not None:
|
||||||
clockwise, pc = arc
|
for p in approximate_arc(*center, *p1, *p2, clockwise,
|
||||||
#r = math.hypot(*pc) # arc center is relative to p1.
|
max_error=max_error, clip_max_error=clip_max_error):
|
||||||
#d = math.dist(p1, p2)
|
points.append(p)
|
||||||
#err = r - math.sqrt(r**2 - (d/(2*n))**2)
|
points.pop()
|
||||||
#n = math.ceil(1/(2*math.sqrt(r**2 - (r - max_err)**2)/d))
|
|
||||||
arc = Arc(*p1, *p2, *pc, clockwise, unit=self.unit, polarity_dark=self.polarity_dark, aperture=None)
|
|
||||||
for line in arc.approximate(max_error=max_error, unit=unit):
|
|
||||||
points.append(line.p2)
|
|
||||||
|
|
||||||
else:
|
else:
|
||||||
points.append(p2)
|
points.append(p1)
|
||||||
|
points.append(p2)
|
||||||
|
|
||||||
if points[-1] != points[0]:
|
if points[0] != points[-1]:
|
||||||
points.append(points[0])
|
points.append(points[0])
|
||||||
|
|
||||||
yield amp.Outline(self.unit, int(self.polarity_dark), len(points)-1, tuple(coord for p in points for coord in p))
|
yield amp.Outline(self.unit, int(self.polarity_dark), len(points)-1, tuple(coord for p in points for coord in p))
|
||||||
|
|
@ -389,6 +413,9 @@ class Region(GraphicObject):
|
||||||
yield gp.ArcPoly(conv_outline, conv_arc, polarity_dark=self.polarity_dark)
|
yield gp.ArcPoly(conv_outline, conv_arc, polarity_dark=self.polarity_dark)
|
||||||
|
|
||||||
def to_statements(self, gs):
|
def to_statements(self, gs):
|
||||||
|
if len(self.outline) < 3:
|
||||||
|
return
|
||||||
|
|
||||||
yield from gs.set_polarity(self.polarity_dark)
|
yield from gs.set_polarity(self.polarity_dark)
|
||||||
yield 'G36*'
|
yield 'G36*'
|
||||||
# Repeat interpolation mode at start of region statement to work around gerbv bug. Without this, gerbv will
|
# Repeat interpolation mode at start of region statement to work around gerbv bug. Without this, gerbv will
|
||||||
|
|
@ -398,32 +425,24 @@ class Region(GraphicObject):
|
||||||
|
|
||||||
yield from gs.set_current_point(self.outline[0], unit=self.unit)
|
yield from gs.set_current_point(self.outline[0], unit=self.unit)
|
||||||
|
|
||||||
for point, arc_center in zip_longest(self.outline[1:], self.arc_centers):
|
for previous_point, point, (clockwise, center) in self.iter_segments():
|
||||||
if point is None and arc_center is None:
|
if point is None and center is None:
|
||||||
break
|
break
|
||||||
|
|
||||||
if arc_center is None:
|
x = gs.file_settings.write_gerber_value(point[0], self.unit)
|
||||||
yield from gs.set_interpolation_mode(InterpMode.LINEAR)
|
y = gs.file_settings.write_gerber_value(point[1], self.unit)
|
||||||
|
|
||||||
x = gs.file_settings.write_gerber_value(point[0], self.unit)
|
if clockwise is None:
|
||||||
y = gs.file_settings.write_gerber_value(point[1], self.unit)
|
yield from gs.set_interpolation_mode(InterpMode.LINEAR)
|
||||||
yield f'X{x}Y{y}D01*'
|
yield f'X{x}Y{y}D01*'
|
||||||
|
|
||||||
gs.update_point(*point, unit=self.unit)
|
|
||||||
|
|
||||||
else:
|
else:
|
||||||
clockwise, (cx, cy) = arc_center
|
|
||||||
x2, y2 = point
|
|
||||||
yield from gs.set_interpolation_mode(InterpMode.CIRCULAR_CW if clockwise else InterpMode.CIRCULAR_CCW)
|
yield from gs.set_interpolation_mode(InterpMode.CIRCULAR_CW if clockwise else InterpMode.CIRCULAR_CCW)
|
||||||
|
i = gs.file_settings.write_gerber_value(center[0]-previous_point[0], self.unit)
|
||||||
x = gs.file_settings.write_gerber_value(x2, self.unit)
|
j = gs.file_settings.write_gerber_value(center[1]-previous_point[1], self.unit)
|
||||||
y = gs.file_settings.write_gerber_value(y2, self.unit)
|
|
||||||
# TODO are these coordinates absolute or relative now?!
|
|
||||||
i = gs.file_settings.write_gerber_value(cx, self.unit)
|
|
||||||
j = gs.file_settings.write_gerber_value(cy, self.unit)
|
|
||||||
yield f'X{x}Y{y}I{i}J{j}D01*'
|
yield f'X{x}Y{y}I{i}J{j}D01*'
|
||||||
|
|
||||||
gs.update_point(x2, y2, unit=self.unit)
|
gs.update_point(*point, unit=self.unit)
|
||||||
|
|
||||||
yield 'G37*'
|
yield 'G37*'
|
||||||
|
|
||||||
|
|
@ -605,22 +624,8 @@ class Arc(GraphicObject):
|
||||||
:returns: Angle in clockwise radian between ``0`` and ``2*math.pi``
|
:returns: Angle in clockwise radian between ``0`` and ``2*math.pi``
|
||||||
:rtype: float
|
:rtype: float
|
||||||
"""
|
"""
|
||||||
cx, cy = self.cx + self.x1, self.cy + self.y1
|
|
||||||
x1, y1 = self.x1 - cx, self.y1 - cy
|
|
||||||
x2, y2 = self.x2 - cx, self.y2 - cy
|
|
||||||
|
|
||||||
a1, a2 = math.atan2(y1, x1), math.atan2(y2, x2)
|
return sweep_angle(self.cx+self.x1, self.cy+self.y1, self.x1, self.y1, self.x2, self.y2, self.clockwise)
|
||||||
f = abs(a2 - a1)
|
|
||||||
if not self.clockwise:
|
|
||||||
if a2 > a1:
|
|
||||||
return a2 - a1
|
|
||||||
else:
|
|
||||||
return 2*math.pi - abs(a2 - a1)
|
|
||||||
else:
|
|
||||||
if a1 > a2:
|
|
||||||
return a1 - a2
|
|
||||||
else:
|
|
||||||
return 2*math.pi - abs(a1 - a2)
|
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def p1(self):
|
def p1(self):
|
||||||
|
|
@ -677,34 +682,16 @@ class Arc(GraphicObject):
|
||||||
:returns: list of :py:class:`~.graphic_objects.Line` instances.
|
:returns: list of :py:class:`~.graphic_objects.Line` instances.
|
||||||
:rtype: list
|
:rtype: list
|
||||||
"""
|
"""
|
||||||
# TODO the max_angle calculation below is a bit off -- we over-estimate the error, and thus produce finer
|
|
||||||
# results than necessary. Fix this.
|
|
||||||
|
|
||||||
r = math.hypot(self.cx, self.cy)
|
|
||||||
|
|
||||||
max_error = self.unit(max_error, unit)
|
max_error = self.unit(max_error, unit)
|
||||||
if clip_max_error:
|
return [Line(*p1, *p2, aperture=self.aperture, polarity_dark=self.polarity_dark, unit=self.unit)
|
||||||
# 1 - math.sqrt(1 - 0.5*math.sqrt(2))
|
for p1, p2 in pairwise(approximate_arc(
|
||||||
max_error = min(max_error, r*0.4588038998538031)
|
self.cx+self.x1, self.cy+self.y1,
|
||||||
|
self.x1, self.y1,
|
||||||
elif max_error >= r:
|
self.x2, self.y2,
|
||||||
return [Line(*self.p1, *self.p2, aperture=self.aperture, polarity_dark=self.polarity_dark, unit=self.unit)]
|
self.clockwise,
|
||||||
|
max_error=max_error,
|
||||||
# see https://www.mathopenref.com/sagitta.html
|
clip_max_error=clip_max_error))]
|
||||||
l = math.sqrt(r**2 - (r - max_error)**2)
|
|
||||||
|
|
||||||
angle_max = math.asin(l/r)
|
|
||||||
sweep_angle = self.sweep_angle()
|
|
||||||
num_segments = math.ceil(sweep_angle / angle_max)
|
|
||||||
angle = sweep_angle / num_segments
|
|
||||||
|
|
||||||
if not self.clockwise:
|
|
||||||
angle = -angle
|
|
||||||
|
|
||||||
cx, cy = self.center
|
|
||||||
points = [ rotate_point(self.x1, self.y1, i*angle, cx, cy) for i in range(num_segments + 1) ]
|
|
||||||
return [ Line(*p1, *p2, aperture=self.aperture, polarity_dark=self.polarity_dark, unit=self.unit)
|
|
||||||
for p1, p2 in zip(points[0::], points[1::]) ]
|
|
||||||
|
|
||||||
def _rotate(self, rotation, cx=0, cy=0):
|
def _rotate(self, rotation, cx=0, cy=0):
|
||||||
# rotate center first since we need old x1, y1 here
|
# rotate center first since we need old x1, y1 here
|
||||||
|
|
@ -726,7 +713,7 @@ class Arc(GraphicObject):
|
||||||
w = self.aperture.equivalent_width(unit) if self.aperture else 0
|
w = self.aperture.equivalent_width(unit) if self.aperture else 0
|
||||||
return gp.Arc(x1=conv.x1, y1=conv.y1,
|
return gp.Arc(x1=conv.x1, y1=conv.y1,
|
||||||
x2=conv.x2, y2=conv.y2,
|
x2=conv.x2, y2=conv.y2,
|
||||||
cx=conv.cx, cy=conv.cy,
|
cx=conv.cx+conv.x1, cy=conv.cy+conv.y1,
|
||||||
clockwise=self.clockwise,
|
clockwise=self.clockwise,
|
||||||
width=w,
|
width=w,
|
||||||
polarity_dark=self.polarity_dark)
|
polarity_dark=self.polarity_dark)
|
||||||
|
|
@ -19,7 +19,7 @@
|
||||||
import math
|
import math
|
||||||
import itertools
|
import itertools
|
||||||
|
|
||||||
from dataclasses import dataclass, replace
|
from dataclasses import dataclass, replace, field
|
||||||
|
|
||||||
from .utils import *
|
from .utils import *
|
||||||
|
|
||||||
|
|
@ -62,6 +62,12 @@ class GraphicPrimitive:
|
||||||
|
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
|
||||||
|
def is_zero_size(self):
|
||||||
|
""" Return whether this primitive is zero size
|
||||||
|
|
||||||
|
:rtype: bool
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True)
|
@dataclass(frozen=True)
|
||||||
class Circle(GraphicPrimitive):
|
class Circle(GraphicPrimitive):
|
||||||
|
|
@ -79,6 +85,14 @@ class Circle(GraphicPrimitive):
|
||||||
color = fg if self.polarity_dark else bg
|
color = fg if self.polarity_dark else bg
|
||||||
return tag('circle', cx=prec(self.x), cy=prec(self.y), r=prec(self.r), fill=color)
|
return tag('circle', cx=prec(self.x), cy=prec(self.y), r=prec(self.r), fill=color)
|
||||||
|
|
||||||
|
def to_arc_poly(self):
|
||||||
|
return ArcPoly([(self.x-self.r, self.y), (self.x+self.r, self.y)],
|
||||||
|
[(True, (self.x, self.y)), (True, (self.x, self.y))],
|
||||||
|
polarity_dark=self.polarity_dark)
|
||||||
|
|
||||||
|
def is_zero_size(self):
|
||||||
|
return math.isclose(self.r, 0)
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True)
|
@dataclass(frozen=True)
|
||||||
class ArcPoly(GraphicPrimitive):
|
class ArcPoly(GraphicPrimitive):
|
||||||
|
|
@ -88,28 +102,51 @@ class ArcPoly(GraphicPrimitive):
|
||||||
#: connected.
|
#: connected.
|
||||||
outline : list
|
outline : list
|
||||||
#: Must be either None (all segments are straight lines) or same length as outline.
|
#: Must be either None (all segments are straight lines) or same length as outline.
|
||||||
#: Straight line segments have None entry.
|
#: Straight line segments have None entry. Arc segments have (clockwise, (cx, cy)) tuple with cx, cy being absolute
|
||||||
arc_centers : list = None
|
#: coords.
|
||||||
|
arc_centers : list = field(default_factory=list)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def segments(self):
|
def segments(self):
|
||||||
""" Return an iterator through all *segments* of this polygon. For each outline segment (line or arc), this
|
""" Return an iterator through all *segments* of this polygon. For each outline segment (line or arc), this
|
||||||
iterator will yield a ``(p1, p2, center)`` tuple. If the segment is a straight line, ``center`` will be
|
iterator will yield a ``(p1, p2, (clockwise, center))`` tuple. If the segment is a straight line, ``clockwise``
|
||||||
``None``.
|
will be ``None``.
|
||||||
"""
|
"""
|
||||||
ol = self.outline
|
for points, arc in itertools.zip_longest(itertools.pairwise(self.outline), self.arc_centers):
|
||||||
return itertools.zip_longest(ol, ol[1:] + [ol[0]], self.arc_centers or [])
|
if arc:
|
||||||
|
if points:
|
||||||
|
yield *points, arc
|
||||||
|
else:
|
||||||
|
yield self.outline[-1], self.outline[0], arc
|
||||||
|
return
|
||||||
|
else:
|
||||||
|
if not points:
|
||||||
|
break
|
||||||
|
yield *points, (None, (None, None))
|
||||||
|
|
||||||
|
# Close outline if necessary.
|
||||||
|
if math.dist(self.outline[0], self.outline[-1]) > 1e-6:
|
||||||
|
yield self.outline[-1], self.outline[0], (None, (None, None))
|
||||||
|
|
||||||
|
def approximate_arcs(self, max_error=1e-2, clip_max_error=True):
|
||||||
|
outline = []
|
||||||
|
for (x1, y1), (x2, y2), (clockwise, (cx, cy)) in self.segments:
|
||||||
|
if clockwise is None:
|
||||||
|
outline.append((x1, y1))
|
||||||
|
else:
|
||||||
|
outline.extend(approximate_arc(cx, cy, x1, y1, x2, y2, clockwise,
|
||||||
|
max_error=max_error, clip_max_error=clip_max_error))
|
||||||
|
outline.pop() # remove arc end point
|
||||||
|
return type(self)(outline, polarity_dark=self.polarity_dark)
|
||||||
|
|
||||||
def bounding_box(self):
|
def bounding_box(self):
|
||||||
bbox = (None, None), (None, None)
|
bbox = (None, None), (None, None)
|
||||||
for (x1, y1), (x2, y2), arc in self.segments:
|
for (x1, y1), (x2, y2), (clockwise, (cx, cy)) in self.segments:
|
||||||
if arc:
|
if clockwise is None:
|
||||||
clockwise, (cx, cy) = arc
|
|
||||||
bbox = add_bounds(bbox, arc_bounds(x1, y1, x2, y2, cx, cy, clockwise))
|
|
||||||
|
|
||||||
else:
|
|
||||||
line_bounds = (min(x1, x2), min(y1, y2)), (max(x1, x2), max(y1, y2))
|
line_bounds = (min(x1, x2), min(y1, y2)), (max(x1, x2), max(y1, y2))
|
||||||
bbox = add_bounds(bbox, line_bounds)
|
bbox = add_bounds(bbox, line_bounds)
|
||||||
|
else:
|
||||||
|
bbox = add_bounds(bbox, arc_bounds(x1, y1, x2, y2, cx, cy, clockwise))
|
||||||
return bbox
|
return bbox
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
|
|
@ -138,17 +175,33 @@ class ArcPoly(GraphicPrimitive):
|
||||||
|
|
||||||
yield f'M {float(self.outline[0][0]):.6} {float(self.outline[0][1]):.6}'
|
yield f'M {float(self.outline[0][0]):.6} {float(self.outline[0][1]):.6}'
|
||||||
|
|
||||||
for old, new, arc in self.segments:
|
for old, new, (clockwise, center) in self.segments:
|
||||||
if not arc:
|
if clockwise is None:
|
||||||
yield f'L {float(new[0]):.6} {float(new[1]):.6}'
|
yield f'L {float(new[0]):.6} {float(new[1]):.6}'
|
||||||
else:
|
else:
|
||||||
clockwise, center = arc
|
|
||||||
yield svg_arc(old, new, center, clockwise)
|
yield svg_arc(old, new, center, clockwise)
|
||||||
|
|
||||||
def to_svg(self, fg='black', bg='white', tag=Tag):
|
def to_svg(self, fg='black', bg='white', tag=Tag):
|
||||||
color = fg if self.polarity_dark else bg
|
color = fg if self.polarity_dark else bg
|
||||||
return tag('path', d=' '.join(self.path_d()), fill=color)
|
return tag('path', d=' '.join(self.path_d()), fill=color)
|
||||||
|
|
||||||
|
def to_arc_poly(self):
|
||||||
|
return self
|
||||||
|
|
||||||
|
|
||||||
|
def is_zero_size(self):
|
||||||
|
for (x1, y1), (x2, y2), (clockwise, (cx, cy)) in self.segments:
|
||||||
|
if clockwise is not None: # arc
|
||||||
|
if math.isclose(cx, x1) and math.isclose(cy, y1):
|
||||||
|
continue
|
||||||
|
|
||||||
|
if math.isclose(x1, x2) and math.isclose(y1, y2):
|
||||||
|
return False
|
||||||
|
|
||||||
|
if math.isclose(polygon_area(self.outline), 0):
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True)
|
@dataclass(frozen=True)
|
||||||
class Line(GraphicPrimitive):
|
class Line(GraphicPrimitive):
|
||||||
|
|
@ -189,7 +242,34 @@ class Line(GraphicPrimitive):
|
||||||
color = fg if self.polarity_dark else bg
|
color = fg if self.polarity_dark else bg
|
||||||
width = f'{self.width:.6}' if not math.isclose(self.width, 0) else '0.01mm'
|
width = f'{self.width:.6}' if not math.isclose(self.width, 0) else '0.01mm'
|
||||||
return tag('path', d=f'M {float(self.x1):.6} {float(self.y1):.6} L {float(self.x2):.6} {float(self.y2):.6}',
|
return tag('path', d=f'M {float(self.x1):.6} {float(self.y1):.6} L {float(self.x2):.6} {float(self.y2):.6}',
|
||||||
fill='none', stroke=color, stroke_width=str(width))
|
fill='none', stroke=color, stroke_width=str(width), stroke_linecap='round')
|
||||||
|
|
||||||
|
def to_arc_poly(self):
|
||||||
|
l = math.dist((self.x1, self.y1), (self.x2, self.y2))
|
||||||
|
if math.isclose(l, 0):
|
||||||
|
# degenerate case: a zero-length line becomes a circle.
|
||||||
|
return ArcPoly([(self.x1-self.width/2, self.y1), (self.x1+self.width/2, self.y1)],
|
||||||
|
[(True, (self.x1, self.y1)), (True, (self.x1, self.y1))],
|
||||||
|
polarity_dark=self.polarity_dark)
|
||||||
|
|
||||||
|
dx, dy = self.x2-self.x1, self.y2-self.y1
|
||||||
|
nx, ny = -dy/l, dx/l
|
||||||
|
rx, ry = nx*self.width/2, ny*self.width/2
|
||||||
|
return ArcPoly([
|
||||||
|
(self.x2+rx, self.y2+ry),
|
||||||
|
(self.x2-rx, self.y2-ry),
|
||||||
|
(self.x1-rx, self.y1-ry),
|
||||||
|
(self.x1+rx, self.y1+ry),
|
||||||
|
], [
|
||||||
|
(True, (self.x2, self.y2)),
|
||||||
|
None,
|
||||||
|
(True, (self.x1, self.y1)),
|
||||||
|
None,
|
||||||
|
], polarity_dark=self.polarity_dark)
|
||||||
|
|
||||||
|
def is_zero_size(self):
|
||||||
|
return math.isclose(self.x1, self.x2) and math.isclose(self.y1, self.y2)
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True)
|
@dataclass(frozen=True)
|
||||||
class Arc(GraphicPrimitive):
|
class Arc(GraphicPrimitive):
|
||||||
|
|
@ -202,9 +282,9 @@ class Arc(GraphicPrimitive):
|
||||||
x2 : float
|
x2 : float
|
||||||
#: End Y coodinate
|
#: End Y coodinate
|
||||||
y2 : float
|
y2 : float
|
||||||
#: Center X coordinate relative to ``x1``
|
#: Center X coordinate (absolute)
|
||||||
cx : float
|
cx : float
|
||||||
#: Center Y coordinate relative to ``y1``
|
#: Center Y coordinate (absolute)
|
||||||
cy : float
|
cy : float
|
||||||
#: ``True`` if this arc is clockwise from start to end. Selects between the large arc and the small arc given this
|
#: ``True`` if this arc is clockwise from start to end. Selects between the large arc and the small arc given this
|
||||||
#: start, end and center
|
#: start, end and center
|
||||||
|
|
@ -212,9 +292,12 @@ class Arc(GraphicPrimitive):
|
||||||
#: Line width of this arc.
|
#: Line width of this arc.
|
||||||
width : float
|
width : float
|
||||||
|
|
||||||
|
@property
|
||||||
|
def is_circle(self):
|
||||||
|
return math.isclose(self.x1, self.x2, abs_tol=1e-6) and math.isclose(self.y1, self.y2, abs_tol=1e-6)
|
||||||
|
|
||||||
def flip(self):
|
def flip(self):
|
||||||
return replace(self, x1=self.x2, y1=self.y2, x2=self.x1, y2=self.y1,
|
return replace(self, x1=self.x2, y1=self.y2, x2=self.x1, y2=self.y1, clockwise=not self.clockwise)
|
||||||
cx=(self.x1 + self.cx) - self.x2, cy=(self.y1 + self.cy) - self.y2, clockwise=not self.clockwise)
|
|
||||||
|
|
||||||
def bounding_box(self):
|
def bounding_box(self):
|
||||||
r = self.width/2
|
r = self.width/2
|
||||||
|
|
@ -226,7 +309,36 @@ class Arc(GraphicPrimitive):
|
||||||
arc = svg_arc((self.x1, self.y1), (self.x2, self.y2), (self.cx, self.cy), self.clockwise)
|
arc = svg_arc((self.x1, self.y1), (self.x2, self.y2), (self.cx, self.cy), self.clockwise)
|
||||||
width = f'{self.width:.6}' if not math.isclose(self.width, 0) else '0.01mm'
|
width = f'{self.width:.6}' if not math.isclose(self.width, 0) else '0.01mm'
|
||||||
return tag('path', d=f'M {float(self.x1):.6} {float(self.y1):.6} {arc}',
|
return tag('path', d=f'M {float(self.x1):.6} {float(self.y1):.6} {arc}',
|
||||||
fill='none', stroke=color, stroke_width=width)
|
fill='none', stroke=color, stroke_width=width, stroke_linecap='round')
|
||||||
|
|
||||||
|
def to_arc_poly(self):
|
||||||
|
r = math.dist((self.x1, self.y1), (self.cx, self.cy))
|
||||||
|
|
||||||
|
if math.isclose(r, 0):
|
||||||
|
# degenerate case: a zero-radius arc becomes a circle.
|
||||||
|
return ArcPoly([(self.x1-self.width/2, self.y1), (self.x1+self.width/2, self.y1)],
|
||||||
|
[(True, (self.x1, self.y1)), (True, (self.x1, self.y1))],
|
||||||
|
polarity_dark=self.polarity_dark)
|
||||||
|
|
||||||
|
dx1, dy1 = self.x1-self.cx, self.y1-self.cy
|
||||||
|
nx1, ny1 = dx1/r * self.width/2, dy1/r * self.width/2
|
||||||
|
dx2, dy2 = self.x2-self.cx, self.y2-self.cy
|
||||||
|
nx2, ny2 = dx2/r * self.width/2, dy2/r * self.width/2
|
||||||
|
return ArcPoly([ # vertices
|
||||||
|
(self.x1+nx1, self.y1+ny1),
|
||||||
|
(self.x1-nx1, self.y1-ny1),
|
||||||
|
(self.x2-nx2, self.y2-ny2),
|
||||||
|
(self.x2+nx2, self.y2+ny2),
|
||||||
|
], [ # arc segments (direction, center)
|
||||||
|
(not self.clockwise, (self.x1, self.y1)),
|
||||||
|
(self.clockwise, (self.cx, self.cy)),
|
||||||
|
(self.clockwise, (self.x2, self.y2)),
|
||||||
|
(not self.clockwise, (self.cx, self.cy)),
|
||||||
|
], polarity_dark=self.polarity_dark)
|
||||||
|
|
||||||
|
def is_zero_size(self):
|
||||||
|
return False # an arc with identical start and end points is defined as a circle
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True)
|
@dataclass(frozen=True)
|
||||||
class Rectangle(GraphicPrimitive):
|
class Rectangle(GraphicPrimitive):
|
||||||
|
|
@ -254,7 +366,7 @@ class Rectangle(GraphicPrimitive):
|
||||||
(x - (cw+sh), y + (ch+sw)),
|
(x - (cw+sh), y + (ch+sw)),
|
||||||
(x + (cw+sh), y + (ch+sw)),
|
(x + (cw+sh), y + (ch+sw)),
|
||||||
(x + (cw+sh), y - (ch+sw)),
|
(x + (cw+sh), y - (ch+sw)),
|
||||||
])
|
], polarity_dark=self.polarity_dark)
|
||||||
|
|
||||||
def to_svg(self, fg='black', bg='white', tag=Tag):
|
def to_svg(self, fg='black', bg='white', tag=Tag):
|
||||||
color = fg if self.polarity_dark else bg
|
color = fg if self.polarity_dark else bg
|
||||||
|
|
@ -262,3 +374,6 @@ class Rectangle(GraphicPrimitive):
|
||||||
return tag('rect', x=prec(x), y=prec(y), width=prec(self.w), height=prec(self.h),
|
return tag('rect', x=prec(x), y=prec(y), width=prec(self.w), height=prec(self.h),
|
||||||
**svg_rotation(self.rotation, self.x, self.y), fill=color)
|
**svg_rotation(self.rotation, self.x, self.y), fill=color)
|
||||||
|
|
||||||
|
def is_zero_size(self):
|
||||||
|
return math.isclose(self.w, 0) or math.isclose(self.h, 0)
|
||||||
|
|
||||||
|
|
@ -82,6 +82,7 @@ MATCH_RULES = {
|
||||||
'bottom paste': r'.*_boardoutline\.\w+', # FIXME verify this
|
'bottom paste': r'.*_boardoutline\.\w+', # FIXME verify this
|
||||||
'drill plated': r'.*\.(drl)', # diptrace has unplated drills on the outline layer
|
'drill plated': r'.*\.(drl)', # diptrace has unplated drills on the outline layer
|
||||||
'other netlist': r'.*\.ipc', # default rule due to lack of tool-specific examples
|
'other netlist': r'.*\.ipc', # default rule due to lack of tool-specific examples
|
||||||
|
'header regex': [['sufficient', r'top .*|bottom .*', r'G04 DipTrace [.-0-9a-z]*\*']],
|
||||||
},
|
},
|
||||||
|
|
||||||
'target': {
|
'target': {
|
||||||
|
|
@ -151,22 +152,25 @@ MATCH_RULES = {
|
||||||
|
|
||||||
'allegro': {
|
'allegro': {
|
||||||
# Allegro doesn't have any widespread convention, so we rely heavily on the layer name auto-guesser here.
|
# Allegro doesn't have any widespread convention, so we rely heavily on the layer name auto-guesser here.
|
||||||
'drill mech': r'.*\.(drl|rou)',
|
'drill plated': r'.*\.(drl)',
|
||||||
'generic gerber': r'.*\.art',
|
'drill nonplated': r'.*\.(rou)',
|
||||||
|
'other unknown': r'.*(place|assembly|keep.?in|keep.?out).*\.art',
|
||||||
|
'autoguess': r'.*\.art',
|
||||||
'excellon params': r'nc_param\.txt|ncdrill\.log|ncroute\.log',
|
'excellon params': r'nc_param\.txt|ncdrill\.log|ncroute\.log',
|
||||||
'other netlist': r'.*\.ipc', # default rule due to lack of tool-specific examples
|
'other netlist': r'.*\.ipc', # default rule due to lack of tool-specific examples
|
||||||
|
'header regex': [['required,sufficient', r'.*\.art', r'G04 File Origin:\s+Cadence Allegro [0-9]+\.[0-9]+[-a-zA-Z0-9]*']],
|
||||||
},
|
},
|
||||||
|
|
||||||
'pads': {
|
'pads': {
|
||||||
# Pads also does not seem to have a factory-default naming schema. Or it has one but everyone ignores it.
|
# Pads also does not seem to have a factory-default naming schema. Or it has one but everyone ignores it.
|
||||||
'generic gerber': r'.*\.pho',
|
'autoguess': r'.*\.pho',
|
||||||
'drill mech': r'.*\.drl',
|
'drill plated': r'.*\.drl',
|
||||||
},
|
},
|
||||||
|
|
||||||
'zuken': {
|
'zuken': {
|
||||||
'generic gerber': r'.*\.fph',
|
'autoguess': r'.*\.fph',
|
||||||
'gerber params': r'.*\.fpl',
|
'gerber params': r'.*\.fpl',
|
||||||
'drill mech': r'.*\.fdr',
|
'drill unknown': r'.*\.fdr',
|
||||||
'excellon params': r'.*\.fdl',
|
'excellon params': r'.*\.fdl',
|
||||||
'other netlist': r'.*\.ipc',
|
'other netlist': r'.*\.ipc',
|
||||||
'ipc-2581': r'.*\.xml',
|
'ipc-2581': r'.*\.xml',
|
||||||
|
|
@ -39,6 +39,7 @@ from .cam import FileSettings, LazyCamFile
|
||||||
from .layer_rules import MATCH_RULES
|
from .layer_rules import MATCH_RULES
|
||||||
from .utils import sum_bounds, setup_svg, MM, Tag, convex_hull
|
from .utils import sum_bounds, setup_svg, MM, Tag, convex_hull
|
||||||
from . import graphic_objects as go
|
from . import graphic_objects as go
|
||||||
|
from . import apertures as ap
|
||||||
from . import graphic_primitives as gp
|
from . import graphic_primitives as gp
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -65,27 +66,27 @@ DEFAULT_COLORS = {
|
||||||
|
|
||||||
class NamingScheme:
|
class NamingScheme:
|
||||||
kicad = {
|
kicad = {
|
||||||
'top copper': '{board_name}-F.Cu.gbr',
|
'top copper': '{board_name}-F_Cu.gbr',
|
||||||
'top mask': '{board_name}-F.Mask.gbr',
|
'top mask': '{board_name}-F_Mask.gbr',
|
||||||
'top silk': '{board_name}-F.SilkS.gbr',
|
'top silk': '{board_name}-F_SilkS.gbr',
|
||||||
'top paste': '{board_name}-F.Paste.gbr',
|
'top paste': '{board_name}-F_Paste.gbr',
|
||||||
'bottom copper': '{board_name}-B.Cu.gbr',
|
'bottom copper': '{board_name}-B_Cu.gbr',
|
||||||
'bottom mask': '{board_name}-B.Mask.gbr',
|
'bottom mask': '{board_name}-B_Mask.gbr',
|
||||||
'bottom silk': '{board_name}-B.SilkS.gbr',
|
'bottom silk': '{board_name}-B_SilkS.gbr',
|
||||||
'bottom paste': '{board_name}-B.Paste.gbr',
|
'bottom paste': '{board_name}-B_Paste.gbr',
|
||||||
'inner copper': '{board_name}-In{layer_number}.Cu.gbr',
|
'inner copper': '{board_name}-In{layer_number}_Cu.gbr',
|
||||||
'mechanical outline': '{board_name}-Edge.Cuts.gbr',
|
'mechanical outline': '{board_name}-Edge_Cuts.gbr',
|
||||||
'drill unknown': '{board_name}.drl',
|
'drill unknown': '{board_name}.drl',
|
||||||
'drill plated': '{board_name}-PTH.drl',
|
'drill plated': '{board_name}-PTH.drl',
|
||||||
'drill nonplated': '{board_name}-NPTH.drl',
|
'drill nonplated': '{board_name}-NPTH.drl',
|
||||||
'other comments': '{board_name}-Cmts.User.gbr',
|
'other comments': '{board_name}-Cmts_User.gbr',
|
||||||
'other drawings': '{board_name}-Dwgs.User.gbr',
|
'other drawings': '{board_name}-Dwgs_User.gbr',
|
||||||
'top fabrication': '{board_name}-F.Fab.gbr',
|
'top fabrication': '{board_name}-F_Fab.gbr',
|
||||||
'bottom fabrication': '{board_name}-B.Fab.gbr',
|
'bottom fabrication': '{board_name}-B_Fab.gbr',
|
||||||
'top adhesive': '{board_name}-F.Adhes.gbr',
|
'top adhesive': '{board_name}-F_Adhes.gbr',
|
||||||
'bottom adhesive': '{board_name}-B.Adhes.gbr',
|
'bottom adhesive': '{board_name}-B_Adhes.gbr',
|
||||||
'top courtyard': '{board_name}-F.CrtYd.gbr',
|
'top courtyard': '{board_name}-F_CrtYd.gbr',
|
||||||
'bottom courtyard': '{board_name}-B.CrtYd.gbr',
|
'bottom courtyard': '{board_name}-B_CrtYd.gbr',
|
||||||
'other netlist': '{board_name}.d356',
|
'other netlist': '{board_name}.d356',
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -112,31 +113,61 @@ class NamingScheme:
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def apply_rules(filenames, rules):
|
||||||
|
certain = False
|
||||||
|
gen = {}
|
||||||
|
already_matched = set()
|
||||||
|
header_regex = rules.pop('header regex', [])
|
||||||
|
header_regex_matched = [False] * len(header_regex)
|
||||||
|
|
||||||
def _match_files(filenames):
|
file_headers = {}
|
||||||
matches = {}
|
def get_header(path):
|
||||||
for generator, rules in MATCH_RULES.items():
|
if path not in file_headers:
|
||||||
already_matched = set()
|
with open(path) as f:
|
||||||
gen = {}
|
file_headers[path] = f.read(16384)
|
||||||
matches[generator] = gen
|
return file_headers[path]
|
||||||
for layer, regex in rules.items():
|
|
||||||
for fn in filenames:
|
|
||||||
if fn in already_matched:
|
|
||||||
continue
|
|
||||||
|
|
||||||
if (m := re.fullmatch(regex, fn.name, re.IGNORECASE)):
|
for layer, regex in rules.items():
|
||||||
if layer == 'inner copper':
|
for fn in filenames:
|
||||||
target = 'inner_' + ''.join(e or '' for e in m.groups()) + ' copper'
|
if fn in already_matched:
|
||||||
else:
|
continue
|
||||||
target = layer
|
|
||||||
|
|
||||||
gen[target] = gen.get(target, []) + [fn]
|
target = None
|
||||||
already_matched.add(fn)
|
if (m := re.fullmatch(regex, fn.name, re.IGNORECASE)):
|
||||||
return matches
|
if layer == 'inner copper':
|
||||||
|
target = 'inner_' + ''.join(e or '' for e in m.groups()) + ' copper'
|
||||||
|
else:
|
||||||
|
target = layer
|
||||||
|
|
||||||
|
gen[target] = gen.get(target, []) + [fn]
|
||||||
|
already_matched.add(fn)
|
||||||
|
|
||||||
|
for i, (match_type, layer_match, header_match) in enumerate(header_regex):
|
||||||
|
if re.fullmatch(layer_match, fn.name, re.IGNORECASE) or (
|
||||||
|
target is not None and re.fullmatch(layer_match, target, re.IGNORECASE)):
|
||||||
|
if re.search(header_match, get_header(fn)):
|
||||||
|
|
||||||
|
if 'sufficient' in match_type:
|
||||||
|
certain = True
|
||||||
|
|
||||||
|
header_regex_matched[i] = True
|
||||||
|
|
||||||
|
if any('required' in match_type and not match
|
||||||
|
for match, (match_type, *_) in zip(header_regex_matched, header_regex)):
|
||||||
|
return False, {}
|
||||||
|
|
||||||
|
return certain, gen
|
||||||
|
|
||||||
def _best_match(filenames):
|
def _best_match(filenames):
|
||||||
matches = _match_files(filenames)
|
matches = {}
|
||||||
|
for generator, rules in MATCH_RULES.items():
|
||||||
|
certain, candidate = apply_rules(filenames, rules)
|
||||||
|
|
||||||
|
if certain:
|
||||||
|
return generator, candidate
|
||||||
|
|
||||||
|
matches[generator] = candidate
|
||||||
|
|
||||||
matches = sorted(matches.items(), key=lambda pair: len(pair[1]))
|
matches = sorted(matches.items(), key=lambda pair: len(pair[1]))
|
||||||
generator, files = matches[-1]
|
generator, files = matches[-1]
|
||||||
return generator, files
|
return generator, files
|
||||||
|
|
@ -243,7 +274,7 @@ def _layername_autoguesser(fn):
|
||||||
elif re.search('film', fn):
|
elif re.search('film', fn):
|
||||||
use = 'copper'
|
use = 'copper'
|
||||||
|
|
||||||
elif re.search('out(line)?', fn):
|
elif re.search('out(line)?|board.?geom(etry)?', fn):
|
||||||
use = 'outline'
|
use = 'outline'
|
||||||
side = 'mechanical'
|
side = 'mechanical'
|
||||||
|
|
||||||
|
|
@ -273,6 +304,9 @@ def _sort_layername(val):
|
||||||
assert side.startswith('inner_')
|
assert side.startswith('inner_')
|
||||||
return int(side[len('inner_'):])
|
return int(side[len('inner_'):])
|
||||||
|
|
||||||
|
def convex_hull_to_lines(points, unit=MM):
|
||||||
|
for (x1, y1), (x2, y2) in zip(points, points[1:] + points):
|
||||||
|
yield go.Line(x1, y1, x2, y2, aperture=ap.CircleAperture(unit(0.1, MM), unit=unit), unit=unit)
|
||||||
|
|
||||||
class LayerStack:
|
class LayerStack:
|
||||||
""" :py:class:`LayerStack` represents a set of Gerber files that describe different layers of the same board.
|
""" :py:class:`LayerStack` represents a set of Gerber files that describe different layers of the same board.
|
||||||
|
|
@ -385,7 +419,7 @@ class LayerStack:
|
||||||
with ZipFile(file) as f:
|
with ZipFile(file) as f:
|
||||||
f.extractall(path=tmp_indir)
|
f.extractall(path=tmp_indir)
|
||||||
|
|
||||||
inst = kls.open_dir(tmp_indir, board_name=board_name, lazy=lazy)
|
inst = kls.open_dir(tmp_indir, board_name=board_name, lazy=lazy, overrides=overrides, autoguess=autoguess)
|
||||||
inst.tmpdir = tmpdir
|
inst.tmpdir = tmpdir
|
||||||
inst.original_path = Path(original_path or file)
|
inst.original_path = Path(original_path or file)
|
||||||
inst.was_zipped = True
|
inst.was_zipped = True
|
||||||
|
|
@ -421,6 +455,7 @@ class LayerStack:
|
||||||
given value.
|
given value.
|
||||||
:rtype: :py:class:`LayerStack`
|
:rtype: :py:class:`LayerStack`
|
||||||
"""
|
"""
|
||||||
|
print_layermap = False
|
||||||
|
|
||||||
if autoguess:
|
if autoguess:
|
||||||
generator, filemap = _best_match(files)
|
generator, filemap = _best_match(files)
|
||||||
|
|
@ -445,14 +480,51 @@ class LayerStack:
|
||||||
filemap[layer].remove(fn)
|
filemap[layer].remove(fn)
|
||||||
filemap[layer] = filemap.get(layer, []) + [fn]
|
filemap[layer] = filemap.get(layer, []) + [fn]
|
||||||
|
|
||||||
|
if 'autoguess' in filemap:
|
||||||
|
warnings.warn(f'This generator ({generator}) often exports ambiguous filenames. Falling back to autoguesser for some files. Use at your own peril. Autoguessed files: {", ".join(f.name for f in filemap["autoguess"])}')
|
||||||
|
print_layermap = True
|
||||||
|
autoguess_filenames = filemap.pop('autoguess')
|
||||||
|
|
||||||
|
matched = set()
|
||||||
|
for key, values in _do_autoguess(autoguess_filenames).items():
|
||||||
|
filemap[key] = filemap.get(key, []) + values
|
||||||
|
matched |= set(values)
|
||||||
|
|
||||||
|
if generator == 'allegro':
|
||||||
|
# Allegro gerbers often contain the inner layers with completely random filenames and no indication of
|
||||||
|
# layer ordering except for drawings in the mechanical files. We fall back to alphabetic ordering.
|
||||||
|
for fn in autoguess_filenames:
|
||||||
|
if fn not in matched:
|
||||||
|
with open(fn) as f:
|
||||||
|
header = f.read(16384)
|
||||||
|
if re.search(r'G04 Layer:\s*ETCH/.*\*', header):
|
||||||
|
filemap['unknown copper'] = filemap.get('unknown copper', []) + [fn]
|
||||||
|
|
||||||
|
if (unk := filemap.pop('unknown copper', None)):
|
||||||
|
unk = sorted(unk, key=str)
|
||||||
|
if 'top copper' not in filemap:
|
||||||
|
filemap['top copper'], *unk = [unk]
|
||||||
|
if 'bottom copper' not in filemap:
|
||||||
|
*unk, filemap['bottom copper'] = [unk]
|
||||||
|
|
||||||
|
i = 1
|
||||||
|
while unk and i < 128:
|
||||||
|
key = f'inner_{i:02d} copper'
|
||||||
|
if key not in filemap:
|
||||||
|
filemap[key] = [unk.pop(0)]
|
||||||
|
i += 1
|
||||||
|
|
||||||
if sum(len(files) for files in filemap.values()) < 6 and autoguess:
|
if sum(len(files) for files in filemap.values()) < 6 and autoguess:
|
||||||
warnings.warn('Ambiguous gerber filenames. Trying last-resort autoguesser.')
|
warnings.warn('Ambiguous gerber filenames. Trying last-resort autoguesser.')
|
||||||
generator = None
|
generator = None
|
||||||
|
print_layermap = True
|
||||||
filemap = _do_autoguess(files)
|
filemap = _do_autoguess(files)
|
||||||
if len(filemap) < 6:
|
if len(filemap) < 6:
|
||||||
raise ValueError('Cannot figure out gerber file mapping. Partial map is: ', filemap)
|
raise ValueError('Cannot figure out gerber file mapping. Partial map is: ', filemap)
|
||||||
|
|
||||||
excellon_settings, external_tools = None, None
|
excellon_settings, external_tools = None, None
|
||||||
|
automatch_drill_scale = False
|
||||||
|
|
||||||
if generator == 'geda':
|
if generator == 'geda':
|
||||||
# geda is written by geniuses who waste no bytes of unnecessary output so it doesn't actually include the
|
# geda is written by geniuses who waste no bytes of unnecessary output so it doesn't actually include the
|
||||||
# number format in files that use imperial units. Unfortunately it also doesn't include any hints that the
|
# number format in files that use imperial units. Unfortunately it also doesn't include any hints that the
|
||||||
|
|
@ -470,16 +542,22 @@ class LayerStack:
|
||||||
if (external_tools := parse_allegro_logfile(file.read_text())):
|
if (external_tools := parse_allegro_logfile(file.read_text())):
|
||||||
break
|
break
|
||||||
del filemap['excellon params']
|
del filemap['excellon params']
|
||||||
# Ignore if we can't find the param file -- maybe the user has convinced Allegro to actually put this
|
else:
|
||||||
# information into a comment, or maybe they have made Allegro just use decimal points like XNC does.
|
# Ignore if we can't find the param file -- maybe the user has convinced Allegro to actually put this
|
||||||
|
# information into a comment, or maybe they have made Allegro just use decimal points like XNC does.
|
||||||
|
# We'll run an automatic scale matching later.
|
||||||
|
excellon_settings = FileSettings(number_format=(2, 4))
|
||||||
|
automatch_drill_scale = True
|
||||||
|
|
||||||
|
print('remaining filemap')
|
||||||
|
import pprint
|
||||||
|
pprint.pprint(filemap)
|
||||||
|
|
||||||
filemap = _do_autoguess([ f for files in filemap.values() for f in files ])
|
|
||||||
if len(filemap) < 6:
|
if len(filemap) < 6:
|
||||||
raise SystemError('Cannot figure out gerber file mapping')
|
raise SystemError('Cannot figure out gerber file mapping')
|
||||||
# FIXME use layer metadata from comments and ipc file if available
|
# FIXME use layer metadata from comments and ipc file if available
|
||||||
|
|
||||||
elif generator == 'zuken':
|
elif generator == 'zuken':
|
||||||
filemap = _do_autoguess([ f for files in filemap.values() for f in files ])
|
|
||||||
if len(filemap) < 6:
|
if len(filemap) < 6:
|
||||||
raise SystemError('Cannot figure out gerber file mapping')
|
raise SystemError('Cannot figure out gerber file mapping')
|
||||||
# FIXME use layer metadata from comments and ipc file if available
|
# FIXME use layer metadata from comments and ipc file if available
|
||||||
|
|
@ -503,7 +581,12 @@ class LayerStack:
|
||||||
else:
|
else:
|
||||||
excellon_settings = None
|
excellon_settings = None
|
||||||
|
|
||||||
ambiguous = [ f'{key} ({", ".join(x.name for x in value)})' for key, value in filemap.items() if len(value) > 1 and not 'drill' in key ]
|
ambiguous = [ f'{key} ({", ".join(x.name for x in value)})'
|
||||||
|
for key, value in filemap.items()
|
||||||
|
if len(value) > 1 and\
|
||||||
|
not 'drill' in key and\
|
||||||
|
not 'excellon' in key and\
|
||||||
|
not key == 'other unknown']
|
||||||
if ambiguous:
|
if ambiguous:
|
||||||
raise SystemError(f'Ambiguous layer names: {", ".join(ambiguous)}')
|
raise SystemError(f'Ambiguous layer names: {", ".join(ambiguous)}')
|
||||||
|
|
||||||
|
|
@ -512,8 +595,11 @@ class LayerStack:
|
||||||
netlist = None
|
netlist = None
|
||||||
layers = {} # { tuple(key.split()): None for key in STANDARD_LAYERS }
|
layers = {} # { tuple(key.split()): None for key in STANDARD_LAYERS }
|
||||||
for key, paths in filemap.items():
|
for key, paths in filemap.items():
|
||||||
if len(paths) > 1 and not 'drill' in key:
|
if len(paths) > 1 and\
|
||||||
raise ValueError(f'Multiple matching files found for {key} layer: {", ".join(value)}')
|
not 'drill' in key and\
|
||||||
|
not 'excellon' in key and\
|
||||||
|
not key == 'other unknown':
|
||||||
|
raise ValueError(f'Multiple matching files found for {key} layer: {", ".join(map(str, value))}')
|
||||||
|
|
||||||
for path in paths:
|
for path in paths:
|
||||||
id_result = identify_file(path.read_text())
|
id_result = identify_file(path.read_text())
|
||||||
|
|
@ -574,9 +660,72 @@ class LayerStack:
|
||||||
board_name = re.sub(r'^\W+', '', board_name)
|
board_name = re.sub(r'^\W+', '', board_name)
|
||||||
board_name = re.sub(r'\W+$', '', board_name)
|
board_name = re.sub(r'\W+$', '', board_name)
|
||||||
|
|
||||||
return kls(layers, drill_pth, drill_npth, drill_layers, board_name=board_name,
|
if automatch_drill_scale:
|
||||||
|
top_copper = layers[('top', 'copper')].to_excellon(errors='ignore', holes_only=True)
|
||||||
|
|
||||||
|
# precision is matching precision in mm
|
||||||
|
def map_coords(obj, precision=0.01, scale=1):
|
||||||
|
obj = obj.converted(MM)
|
||||||
|
return round(obj.x*scale/precision), round(obj.y*scale/precision)
|
||||||
|
|
||||||
|
aper_coords = {map_coords(obj) for obj in top_copper.drills()}
|
||||||
|
|
||||||
|
for drill_file in [drill_pth, drill_npth, *drill_layers]:
|
||||||
|
if not drill_file or not drill_pth.import_settings._file_has_fixed_width_coordinates:
|
||||||
|
continue
|
||||||
|
|
||||||
|
scale_matches = {}
|
||||||
|
for exp in range(-6, 6):
|
||||||
|
scale = 10**exp
|
||||||
|
hole_coords = {map_coords(obj, scale=scale) for obj in drill_file.drills()}
|
||||||
|
|
||||||
|
scale_matches[scale] = len(aper_coords - hole_coords), len(hole_coords - aper_coords)
|
||||||
|
scales_out = [(max(a, b), scale) for scale, (a, b) in scale_matches.items()]
|
||||||
|
_matches, scale = sorted(scales_out)[0]
|
||||||
|
warnings.warn(f'Performing automatic alignment of poorly exported drill layer. Scale matching results: {scale_matches}. Chosen scale: {scale}')
|
||||||
|
|
||||||
|
# Note: This is only used with allegro files, which use decimal points and explicit units in their tool
|
||||||
|
# definitions. Thus, we only scale object coordinates, and not apertures.
|
||||||
|
for obj in drill_file.objects:
|
||||||
|
obj.scale(scale)
|
||||||
|
|
||||||
|
stack = kls(layers, drill_pth, drill_npth, drill_layers, board_name=board_name,
|
||||||
original_path=original_path, was_zipped=was_zipped, generator=[*all_generator_hints, None][0])
|
original_path=original_path, was_zipped=was_zipped, generator=[*all_generator_hints, None][0])
|
||||||
|
|
||||||
|
if print_layermap:
|
||||||
|
warnings.warn('Auto-guessed layer map:\n' + stack.format_layer_map())
|
||||||
|
return stack
|
||||||
|
|
||||||
|
def format_layer_map(self):
|
||||||
|
lines = []
|
||||||
|
def print_layer(prefix, file):
|
||||||
|
nonlocal lines
|
||||||
|
if file is None:
|
||||||
|
lines.append(f'{prefix} <not found>')
|
||||||
|
else:
|
||||||
|
lines.append(f'{prefix} {file.original_path.name} {file}')
|
||||||
|
|
||||||
|
lines.append(' Drill files:')
|
||||||
|
print_layer(' Plated holes:', self.drill_pth)
|
||||||
|
print_layer(' Nonplated holes:', self.drill_npth)
|
||||||
|
for i, l in enumerate(self._drill_layers):
|
||||||
|
print_layer(f' Additional drill layer {i}:', l)
|
||||||
|
|
||||||
|
print_layer(' Board outline:', self.get('mechanical outline'))
|
||||||
|
|
||||||
|
lines.append(' Soldermask:')
|
||||||
|
print_layer(' Top:', self.get('top mask'))
|
||||||
|
print_layer(' Bottom:', self.get('bottom mask'))
|
||||||
|
|
||||||
|
lines.append(' Silkscreen:')
|
||||||
|
print_layer(' Top:', self.get('top silk'))
|
||||||
|
print_layer(' Bottom:', self.get('bottom silk'))
|
||||||
|
|
||||||
|
lines.append(' Copper:')
|
||||||
|
for (side, _use), layer in self.copper_layers:
|
||||||
|
print_layer(f' {side}:', layer)
|
||||||
|
return '\n'.join(lines)
|
||||||
|
|
||||||
def save_to_zipfile(self, path, prefix='', overwrite_existing=True, board_name=None, naming_scheme={},
|
def save_to_zipfile(self, path, prefix='', overwrite_existing=True, board_name=None, naming_scheme={},
|
||||||
gerber_settings=None, excellon_settings=None):
|
gerber_settings=None, excellon_settings=None):
|
||||||
""" Save this board into a zip file at the given path. For other options, see
|
""" Save this board into a zip file at the given path. For other options, see
|
||||||
|
|
@ -589,10 +738,7 @@ class LayerStack:
|
||||||
|
|
||||||
:param prefix: Store output files under the given prefix inside the zip file
|
:param prefix: Store output files under the given prefix inside the zip file
|
||||||
"""
|
"""
|
||||||
if path.is_file():
|
if path.is_file() and not overwrite_existing:
|
||||||
if overwrite_existing:
|
|
||||||
path.unlink()
|
|
||||||
else:
|
|
||||||
raise ValueError('output zip file already exists and overwrite_existing is False')
|
raise ValueError('output zip file already exists and overwrite_existing is False')
|
||||||
|
|
||||||
if gerber_settings and not excellon_settings:
|
if gerber_settings and not excellon_settings:
|
||||||
|
|
@ -699,7 +845,7 @@ class LayerStack:
|
||||||
def __repr__(self):
|
def __repr__(self):
|
||||||
return str(self)
|
return str(self)
|
||||||
|
|
||||||
def to_svg(self, margin=0, arg_unit=MM, svg_unit=MM, force_bounds=None, color_map=None, tag=Tag):
|
def to_svg(self, margin=0, side_re='.*', drills=True, arg_unit=MM, svg_unit=MM, force_bounds=None, colors=None, tag=Tag):
|
||||||
""" Convert this layer stack to a plain SVG string. This is intended for use cases where the resulting SVG will
|
""" Convert this layer stack to a plain SVG string. This is intended for use cases where the resulting SVG will
|
||||||
be processed by other tools, and thus styling with colors or extra markup like Inkscape layer information are
|
be processed by other tools, and thus styling with colors or extra markup like Inkscape layer information are
|
||||||
unwanted. If you want to instead generate a nice-looking preview image for display or graphical editing in tools
|
unwanted. If you want to instead generate a nice-looking preview image for display or graphical editing in tools
|
||||||
|
|
@ -709,6 +855,9 @@ class LayerStack:
|
||||||
mirrored vertically.
|
mirrored vertically.
|
||||||
|
|
||||||
:param margin: Export SVG file with given margin around the board's bounding box.
|
:param margin: Export SVG file with given margin around the board's bounding box.
|
||||||
|
:param side_re: A regex, such as ``'top'``, ``'bottom'``, or ``'.*'`` (default). Selects which layers to export.
|
||||||
|
The default includes inner layers.
|
||||||
|
:param drills: :py:obj:`bool` setting if drills are included (default) or not.
|
||||||
:param arg_unit: :py:class:`.LengthUnit` or str (``'mm'`` or ``'inch'``). Which unit ``margin`` and
|
:param arg_unit: :py:class:`.LengthUnit` or str (``'mm'`` or ``'inch'``). Which unit ``margin`` and
|
||||||
``force_bounds`` are specified in. Default: mm
|
``force_bounds`` are specified in. Default: mm
|
||||||
:param svg_unit: :py:class:`.LengthUnit` or str (``'mm'`` or ``'inch'``). Which unit to use inside the SVG file.
|
:param svg_unit: :py:class:`.LengthUnit` or str (``'mm'`` or ``'inch'``). Which unit to use inside the SVG file.
|
||||||
|
|
@ -716,6 +865,7 @@ class LayerStack:
|
||||||
:param force_bounds: Use bounds given as :py:obj:`((min_x, min_y), (max_x, max_y))` tuple for the output SVG
|
:param force_bounds: Use bounds given as :py:obj:`((min_x, min_y), (max_x, max_y))` tuple for the output SVG
|
||||||
file instead of deriving them from this board's bounding box and ``margin``. Note that this
|
file instead of deriving them from this board's bounding box and ``margin``. Note that this
|
||||||
will not scale or move the board, but instead will only crop the viewport.
|
will not scale or move the board, but instead will only crop the viewport.
|
||||||
|
:param colors: Dict mapping ``f'{side} {use}'`` strings to SVG colors.
|
||||||
:param tag: Extension point to support alternative XML serializers in addition to the built-in one.
|
:param tag: Extension point to support alternative XML serializers in addition to the built-in one.
|
||||||
:rtype: :py:obj:`str`
|
:rtype: :py:obj:`str`
|
||||||
"""
|
"""
|
||||||
|
|
@ -726,29 +876,29 @@ class LayerStack:
|
||||||
|
|
||||||
stroke_attrs = {'stroke_linejoin': 'round', 'stroke_linecap': 'round'}
|
stroke_attrs = {'stroke_linejoin': 'round', 'stroke_linecap': 'round'}
|
||||||
|
|
||||||
if color_map is None:
|
if colors is None:
|
||||||
color_map = default_dict(lambda: 'black')
|
colors = defaultdict(lambda: 'black')
|
||||||
|
|
||||||
tags = []
|
tags = []
|
||||||
|
layer_transform = f'translate(0 {bounds[0][1] + bounds[1][1]}) scale(1 -1)'
|
||||||
for (side, use), layer in reversed(self.graphic_layers.items()):
|
for (side, use), layer in reversed(self.graphic_layers.items()):
|
||||||
fg = color_map[(side, use)]
|
if re.fullmatch(side_re, side) and (fg := colors.get(f'{side} {use}')):
|
||||||
tags.append(tag('g', list(layer.svg_objects(svg_unit=svg_unit, fg=fg, bg="white", tag=Tag)),
|
tags.append(tag('g', list(layer.svg_objects(svg_unit=svg_unit, fg=fg, bg="white", tag=Tag)),
|
||||||
**stroke_attrs, id=f'l-{side}-{use}'))
|
**stroke_attrs, id=f'l-{side}-{use}', transform=layer_transform))
|
||||||
|
|
||||||
if self.drill_pth:
|
if drills:
|
||||||
fg = color_map[('drill', 'pth')]
|
if self.drill_pth and (fg := colors.get('drill pth')):
|
||||||
tags.append(tag('g', list(self.drill_pth.svg_objects(svg_unit=svg_unit, fg=fg, bg="white", tag=Tag)),
|
tags.append(tag('g', list(self.drill_pth.svg_objects(svg_unit=svg_unit, fg=fg, bg="white", tag=Tag)),
|
||||||
**stroke_attrs, id=f'l-drill-pth'))
|
**stroke_attrs, id=f'l-drill-pth', transform=layer_transform))
|
||||||
|
|
||||||
if self.drill_npth:
|
if self.drill_npth and (fg := colors.get('drill npth')):
|
||||||
fg = color_map[('drill', 'npth')]
|
tags.append(tag('g', list(self.drill_npth.svg_objects(svg_unit=svg_unit, fg=fg, bg="white", tag=Tag)),
|
||||||
tags.append(tag('g', list(self.drill_npth.svg_objects(svg_unit=svg_unit, fg=fg, bg="white", tag=Tag)),
|
**stroke_attrs, id=f'l-drill-npth', transform=layer_transform))
|
||||||
**stroke_attrs, id=f'l-drill-npth'))
|
|
||||||
|
|
||||||
for i, layer in enumerate(self._drill_layers):
|
if (fg := colors.get('drill unknown')):
|
||||||
fg = color_map[('drill', 'unknown')]
|
for i, layer in enumerate(self._drill_layers):
|
||||||
tags.append(tag('g', list(layer.svg_objects(svg_unit=svg_unit, fg=fg, bg="white", tag=Tag)),
|
tags.append(tag('g', list(layer.svg_objects(svg_unit=svg_unit, fg=fg, bg="white", tag=Tag)),
|
||||||
**stroke_attrs, id=f'l-drill-{i}'))
|
**stroke_attrs, id=f'l-drill-{i}', transform=layer_transform))
|
||||||
|
|
||||||
return setup_svg(tags, bounds, margin=margin, arg_unit=arg_unit, svg_unit=svg_unit, tag=tag)
|
return setup_svg(tags, bounds, margin=margin, arg_unit=arg_unit, svg_unit=svg_unit, tag=tag)
|
||||||
|
|
||||||
|
|
@ -819,6 +969,7 @@ class LayerStack:
|
||||||
|
|
||||||
inkscape_attrs = lambda label: dict(inkscape__groupmode='layer', inkscape__label=label) if inkscape else {}
|
inkscape_attrs = lambda label: dict(inkscape__groupmode='layer', inkscape__label=label) if inkscape else {}
|
||||||
stroke_attrs = {'stroke_linejoin': 'round', 'stroke_linecap': 'round'}
|
stroke_attrs = {'stroke_linejoin': 'round', 'stroke_linecap': 'round'}
|
||||||
|
layer_transform=f'translate(0 {bounds[0][1] + bounds[1][1]}) scale(1 -1)'
|
||||||
|
|
||||||
use_defs = []
|
use_defs = []
|
||||||
|
|
||||||
|
|
@ -828,7 +979,7 @@ class LayerStack:
|
||||||
warnings.warn(f'Layer "{side} {use}" not found. Found layers: {", ".join(side + " " + use for side, use in self.graphic_layers)}')
|
warnings.warn(f'Layer "{side} {use}" not found. Found layers: {", ".join(side + " " + use for side, use in self.graphic_layers)}')
|
||||||
continue
|
continue
|
||||||
|
|
||||||
layer = self[(side, use)]
|
layer = self[(side, use)].instance
|
||||||
|
|
||||||
fg, bg = ('white', 'black') if use != 'mask' else ('black', 'white')
|
fg, bg = ('white', 'black') if use != 'mask' else ('black', 'white')
|
||||||
default_fill = {'copper': fg, 'mask': fg, 'silk': 'none', 'paste': fg}[use]
|
default_fill = {'copper': fg, 'mask': fg, 'silk': 'none', 'paste': fg}[use]
|
||||||
|
|
@ -861,18 +1012,25 @@ class LayerStack:
|
||||||
if use == 'mask':
|
if use == 'mask':
|
||||||
objects.insert(0, tag('path', id='outline-path', d=self.outline_svg_d(unit=svg_unit), fill='white'))
|
objects.insert(0, tag('path', id='outline-path', d=self.outline_svg_d(unit=svg_unit), fill='white'))
|
||||||
layers.append(tag('g', objects, id=f'l-{side}-{use}', filter=f'url(#f-{use})',
|
layers.append(tag('g', objects, id=f'l-{side}-{use}', filter=f'url(#f-{use})',
|
||||||
fill=default_fill, stroke=default_stroke, **stroke_attrs,
|
fill=default_fill, stroke=default_stroke, **stroke_attrs, fill_rule='evenodd',
|
||||||
**inkscape_attrs(f'{side} {use}')))
|
**inkscape_attrs(f'{side} {use}'), transform=layer_transform))
|
||||||
|
|
||||||
for i, layer in enumerate(self.drill_layers):
|
for i, layer in enumerate(self.drill_layers):
|
||||||
layers.append(tag('g', list(layer.instance.svg_objects(svg_unit=svg_unit, fg='white', bg='black', tag=Tag)),
|
layers.append(tag('g', list(layer.instance.svg_objects(svg_unit=svg_unit, fg='white', bg='black', tag=Tag)),
|
||||||
id=f'l-drill-{i}', filter=f'url(#f-drill)', **stroke_attrs, **inkscape_attrs(f'drill-{i}')))
|
id=f'l-drill-{i}', filter=f'url(#f-drill)', **stroke_attrs, **inkscape_attrs(f'drill-{i}'),
|
||||||
|
transform=layer_transform))
|
||||||
|
|
||||||
if self.outline:
|
if self.outline:
|
||||||
layers.append(tag('g', list(self.outline.instance.svg_objects(svg_unit=svg_unit, fg='white', bg='black', tag=Tag)),
|
layers.append(tag('g', list(self.outline.instance.svg_objects(svg_unit=svg_unit, fg='white', bg='black', tag=Tag)),
|
||||||
id=f'l-mechanical-outline', **stroke_attrs, **inkscape_attrs(f'outline')))
|
id=f'l-mechanical-outline', **stroke_attrs, **inkscape_attrs(f'outline'),
|
||||||
|
transform=layer_transform))
|
||||||
|
|
||||||
layer_group = tag('g', layers, transform=f'translate(0 {bounds[0][1] + bounds[1][1]}) scale(1 -1)')
|
sc_y, tl_y = 1, 0
|
||||||
|
if side == 'bottom':
|
||||||
|
sc_x, tl_x = -1, (bounds[0][0] + bounds[1][0])
|
||||||
|
else:
|
||||||
|
sc_x, tl_x = 1, 0
|
||||||
|
layer_group = tag('g', layers, transform=f'translate({tl_x} {tl_y}) scale({sc_x} {sc_y})')
|
||||||
tags = [tag('defs', filter_defs + use_defs), layer_group]
|
tags = [tag('defs', filter_defs + use_defs), layer_group]
|
||||||
return setup_svg(tags, bounds, margin=margin, arg_unit=arg_unit, svg_unit=svg_unit, pagecolor="white", tag=tag, inkscape=inkscape)
|
return setup_svg(tags, bounds, margin=margin, arg_unit=arg_unit, svg_unit=svg_unit, pagecolor="white", tag=tag, inkscape=inkscape)
|
||||||
|
|
||||||
|
|
@ -1102,22 +1260,6 @@ class LayerStack:
|
||||||
polys.append(' '.join(poly.path_d()) + ' Z')
|
polys.append(' '.join(poly.path_d()) + ' Z')
|
||||||
return ' '.join(polys)
|
return ' '.join(polys)
|
||||||
|
|
||||||
def outline_convex_hull(self, tol=0.01, unit=MM):
|
|
||||||
points = []
|
|
||||||
for obj in self.outline.instance.objects:
|
|
||||||
if isinstance(obj, go.Line):
|
|
||||||
line = obj.as_primitive(unit)
|
|
||||||
points.append((line.x1, line.y1))
|
|
||||||
points.append((line.x2, line.y2))
|
|
||||||
|
|
||||||
elif isinstance(obj, go.Arc):
|
|
||||||
for obj in obj.approximate(tol, unit):
|
|
||||||
line = obj.as_primitive(unit)
|
|
||||||
points.append((line.x1, line.y1))
|
|
||||||
points.append((line.x2, line.y2))
|
|
||||||
|
|
||||||
return convex_hull(points)
|
|
||||||
|
|
||||||
def outline_polygons(self, tol=0.01, unit=MM):
|
def outline_polygons(self, tol=0.01, unit=MM):
|
||||||
""" Iterator yielding this boards outline as a list of ordered :py:class:`~.graphic_objects.Arc` and
|
""" Iterator yielding this boards outline as a list of ordered :py:class:`~.graphic_objects.Arc` and
|
||||||
:py:class:`~.graphic_objects.Line` objects. This method first sorts all lines and arcs on the outline layer into
|
:py:class:`~.graphic_objects.Line` objects. This method first sorts all lines and arcs on the outline layer into
|
||||||
|
|
@ -1132,14 +1274,29 @@ class LayerStack:
|
||||||
:param tol: :py:obj:`float` setting the tolerance below which two points are considered equal
|
:param tol: :py:obj:`float` setting the tolerance below which two points are considered equal
|
||||||
:param unit: :py:class:`.LengthUnit` or str (``'mm'`` or ``'inch'``). SVG document unit. Default: mm
|
:param unit: :py:class:`.LengthUnit` or str (``'mm'`` or ``'inch'``). SVG document unit. Default: mm
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
if not self.outline:
|
||||||
|
warnings.warn("Board has no outline layer, or the outline layer could not be identified by file name. Using the copper layers' convex hull instead.")
|
||||||
|
points = sum((layer.instance.convex_hull(tol, unit) for (_side, _use), layer in self.copper_layers), start=[])
|
||||||
|
yield list(convex_hull_to_lines(convex_hull(points), unit))
|
||||||
|
return
|
||||||
|
|
||||||
|
maybe_allegro_hint = '' if self.generator != 'allegro' else ' This file looks like it was generated by Allegro/OrCAD. These tools produce quite mal-formed gerbers, and often export text on the outline layer. If you generated this file yourself, maybe try twiddling with the export settings.'
|
||||||
polygons = []
|
polygons = []
|
||||||
lines = [ obj.as_primitive(unit) for obj in self.outline.instance.objects if isinstance(obj, (go.Line, go.Arc)) ]
|
lines = [ obj.as_primitive(unit) for obj in self.outline.instance.objects if isinstance(obj, (go.Line, go.Arc)) ]
|
||||||
|
lines = [ prim for prim in lines if not prim.is_zero_size() ]
|
||||||
|
|
||||||
by_x = sorted([ (obj.x1, obj) for obj in lines ] + [ (obj.x2, obj) for obj in lines ], key=lambda x: x[0])
|
by_x = sorted([ (obj.x1, obj) for obj in lines ] + [ (obj.x2, obj) for obj in lines ], key=lambda x: x[0])
|
||||||
dist_sq = lambda x1, y1, x2, y2: (x2-x1)**2 + (y2-y1)**2
|
dist_sq = lambda x1, y1, x2, y2: (x2-x1)**2 + (y2-y1)**2
|
||||||
|
|
||||||
joins = {}
|
joins = {}
|
||||||
for cur in lines:
|
for cur in lines:
|
||||||
|
# Special case: An arc may describe a complete circle, in which case we have to return it as-is since it
|
||||||
|
# is the only primitive that can join itself.
|
||||||
|
if isinstance(cur, gp.Arc) and cur.is_circle:
|
||||||
|
yield [cur]
|
||||||
|
continue
|
||||||
|
|
||||||
for (i, x, y) in [(0, cur.x1, cur.y1), (1, cur.x2, cur.y2)]:
|
for (i, x, y) in [(0, cur.x1, cur.y1), (1, cur.x2, cur.y2)]:
|
||||||
x_left = bisect.bisect_left (by_x, x, key=lambda elem: elem[0] + tol)
|
x_left = bisect.bisect_left (by_x, x, key=lambda elem: elem[0] + tol)
|
||||||
x_right = bisect.bisect_right(by_x, x, key=lambda elem: elem[0] - tol)
|
x_right = bisect.bisect_right(by_x, x, key=lambda elem: elem[0] - tol)
|
||||||
|
|
@ -1154,13 +1311,14 @@ class LayerStack:
|
||||||
j = 0 if d1 < d2 else 1
|
j = 0 if d1 < d2 else 1
|
||||||
|
|
||||||
if (nearest, j) in joins and joins[(nearest, j)] != (cur, i):
|
if (nearest, j) in joins and joins[(nearest, j)] != (cur, i):
|
||||||
warnings.warn(f'Three-way intersection on outline layer at: {(nearest, j)}; {(cur, i)}; and {joins[(nearest, j)]}. Falling back to returning the convex hull of the outline layer.')
|
warnings.warn(f'Three-way intersection on outline layer at: {(nearest, j)}; {(cur, i)}; and {joins[(nearest, j)]}. Falling back to returning the convex hull of the outline layer.{maybe_allegro_hint}')
|
||||||
return self.outline_convex_hull(tol, unit)
|
yield list(convex_hull_to_lines(self.outline.instance.convex_hull(tol, unit), unit))
|
||||||
|
return
|
||||||
|
|
||||||
if (cur, i) in joins and joins[(cur, i)] != (nearest, j):
|
if (cur, i) in joins and joins[(cur, i)] != (nearest, j):
|
||||||
warnings.warn(f'three-way intersection on outline layer at: {(nearest, j)}; {(cur, i)}; and {joins[(nearest, j)]}. Falling back to returning the convex hull of the outline layer.')
|
warnings.warn(f'Three-way intersection on outline layer at: {(nearest, j)}; {(cur, i)}; and {joins[(cur, i)]}. Falling back to returning the convex hull of the outline layer.{maybe_allegro_hint}')
|
||||||
return self.outline_convex_hull(tol, unit)
|
yield list(convex_hull_to_lines(self.outline.instance.convex_hull(tol, unit), unit))
|
||||||
|
return
|
||||||
|
|
||||||
joins[(cur, i)] = (nearest, j)
|
joins[(cur, i)] = (nearest, j)
|
||||||
joins[(nearest, j)] = (cur, i)
|
joins[(nearest, j)] = (cur, i)
|
||||||
167
src/gerbonara/newstroke.py
Normal file
|
|
@ -0,0 +1,167 @@
|
||||||
|
#!/usr/bin/env python
|
||||||
|
|
||||||
|
from pathlib import Path
|
||||||
|
import unicodedata
|
||||||
|
import re
|
||||||
|
import ast
|
||||||
|
from functools import lru_cache
|
||||||
|
import math
|
||||||
|
from importlib.resources import files
|
||||||
|
|
||||||
|
from . import data
|
||||||
|
from .utils import rotate_point, Tag
|
||||||
|
|
||||||
|
|
||||||
|
STROKE_FONT_SCALE = 1/21
|
||||||
|
FONT_OFFSET = -10
|
||||||
|
DEFAULT_SPACE_WIDTH = 0.6
|
||||||
|
DEFAULT_CHAR_GAP = 0.2
|
||||||
|
|
||||||
|
_dec = lambda c: ord(c)-ord('R')
|
||||||
|
|
||||||
|
|
||||||
|
class Newstroke:
|
||||||
|
def __init__(self, newstroke_cpp=None):
|
||||||
|
if newstroke_cpp is None:
|
||||||
|
newstroke_cpp = files(data).joinpath('newstroke_font.cpp').read_bytes()
|
||||||
|
self.glyphs = dict(self.load_font(newstroke_cpp))
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
@lru_cache
|
||||||
|
def load(kls):
|
||||||
|
return kls()
|
||||||
|
|
||||||
|
def render(self, text, size=1.0, x0=0, y0=0, rotation=0, h_align='left', v_align='bottom', space_width=DEFAULT_SPACE_WIDTH, char_gap=DEFAULT_CHAR_GAP, scale=(1, 1), mirror=(False, False)):
|
||||||
|
text = unicodedata.normalize('NFC', text)
|
||||||
|
missing_glyph = self.glyphs['?']
|
||||||
|
sx, sy = scale
|
||||||
|
mx, my = mirror
|
||||||
|
x = 0
|
||||||
|
|
||||||
|
if rotation >= 180:
|
||||||
|
rotation -= 180
|
||||||
|
h_align = {'left': 'right', 'right': 'left'}.get(h_align, h_align)
|
||||||
|
x0, y0 = -x0, y0
|
||||||
|
|
||||||
|
# if mx:
|
||||||
|
# y0 = -y0
|
||||||
|
# if rotation == 0:
|
||||||
|
# v_align = {'top': 'bottom', 'bottom': 'top'}.get(v_align, v_align)
|
||||||
|
# else:
|
||||||
|
# h_align = {'left': 'right', 'right': 'left'}.get(h_align, h_align)
|
||||||
|
|
||||||
|
x0, y0 = rotate_point(x0, y0, math.radians(-rotation))
|
||||||
|
|
||||||
|
alx, aly = 0, 0
|
||||||
|
(minx, miny), (maxx, maxy) = bbox = self.bounding_box(text, size, space_width, char_gap)
|
||||||
|
w = maxx - minx
|
||||||
|
|
||||||
|
if my:
|
||||||
|
if rotation == 0:
|
||||||
|
sx = -1
|
||||||
|
h_align = {'left': 'right', 'right': 'left'}.get(h_align, h_align)
|
||||||
|
else:
|
||||||
|
sy = -sy
|
||||||
|
|
||||||
|
if h_align != 'left':
|
||||||
|
if h_align == 'right':
|
||||||
|
alx = -w
|
||||||
|
elif h_align == 'center':
|
||||||
|
alx = -w/2
|
||||||
|
else:
|
||||||
|
raise ValueError(f'Invalid h_align value "{h_align}"')
|
||||||
|
|
||||||
|
if v_align == 'top':
|
||||||
|
aly = sy*1.2*size
|
||||||
|
elif v_align == 'middle':
|
||||||
|
aly = sy*1.2*size/2
|
||||||
|
elif v_align != 'bottom':
|
||||||
|
raise ValueError(f'Invalid v_align value "{v_align}"')
|
||||||
|
|
||||||
|
for c in text:
|
||||||
|
if c == ' ':
|
||||||
|
x += space_width
|
||||||
|
continue
|
||||||
|
|
||||||
|
width, strokes = self.glyphs.get(c, missing_glyph)
|
||||||
|
glyph_w = max(width, max(x for st in strokes for x, _y in st))
|
||||||
|
|
||||||
|
for st in strokes:
|
||||||
|
yield [rotate_point((px+x)*sx*size+alx+x0, py*sy*size+aly+y0, math.radians(-rotation), x0, y0) for px, py in st]
|
||||||
|
|
||||||
|
x += glyph_w
|
||||||
|
|
||||||
|
def render_svg(self, text, size=1.0, x0=0, y0=0, rotation=0, h_align='left', v_align='bottom', space_width=DEFAULT_SPACE_WIDTH, char_gap=DEFAULT_CHAR_GAP, scale=(1, -1), mirror=(False, False), **svg_attrs):
|
||||||
|
if 'stroke_linecap' not in svg_attrs:
|
||||||
|
svg_attrs['stroke_linecap'] = 'round'
|
||||||
|
if 'stroke_linejoin' not in svg_attrs:
|
||||||
|
svg_attrs['stroke_linejoin'] = 'round'
|
||||||
|
if 'stroke_width' not in svg_attrs:
|
||||||
|
svg_attrs['stroke_width'] = f'{0.2*size:.3f}'
|
||||||
|
svg_attrs['fill'] = 'none'
|
||||||
|
|
||||||
|
strokes = ['M ' + ' L '.join(f'{x:.3f} {y:.3f}' for x, y in stroke)
|
||||||
|
for stroke in self.render(text, size=size, x0=x0, y0=y0, rotation=rotation, h_align=h_align,
|
||||||
|
v_align=v_align, mirror=mirror, space_width=space_width, char_gap=char_gap,
|
||||||
|
scale=scale)]
|
||||||
|
return Tag('path', d=' '.join(strokes), **svg_attrs)
|
||||||
|
|
||||||
|
def bounding_box(self, text, size=1.0, space_width=DEFAULT_SPACE_WIDTH, char_gap=DEFAULT_CHAR_GAP):
|
||||||
|
text = unicodedata.normalize('NFC', text)
|
||||||
|
missing_glyph = self.glyphs['?']
|
||||||
|
x = 0
|
||||||
|
for c in text:
|
||||||
|
if c == ' ':
|
||||||
|
x += space_width*size
|
||||||
|
continue
|
||||||
|
|
||||||
|
width, strokes = self.glyphs.get(c, missing_glyph)
|
||||||
|
glyph_w = max(width, max(x for st in strokes for x, _y in st))
|
||||||
|
x += glyph_w*size
|
||||||
|
|
||||||
|
return (0, -0.2*size), (x, 1.2*size)
|
||||||
|
|
||||||
|
def load_font(self, newstroke_cpp):
|
||||||
|
e = []
|
||||||
|
for char, (width, strokes) in self.load_glyphs(newstroke_cpp):
|
||||||
|
yield char, (width, strokes)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def decode_stroke(kls, stroke, start_x):
|
||||||
|
for i in range(0, len(stroke), 2):
|
||||||
|
x = (stroke[i]-0x52-start_x)*STROKE_FONT_SCALE
|
||||||
|
y = (stroke[i+1]-0x52+FONT_OFFSET)*STROKE_FONT_SCALE
|
||||||
|
yield (x, y)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def decode_glyph(kls, data):
|
||||||
|
start_x, end_x = data[0]-0x52, data[1]-0x52
|
||||||
|
width = end_x - start_x
|
||||||
|
|
||||||
|
strokes = tuple(tuple(kls.decode_stroke(st, start_x)) for st in data[2:].split(b' R'))
|
||||||
|
return width*STROKE_FONT_SCALE, strokes
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def load_glyphs(kls, newstroke_cpp):
|
||||||
|
it = iter(newstroke_cpp.splitlines())
|
||||||
|
|
||||||
|
for line in it:
|
||||||
|
if re.search(rb'char.*\*', line):
|
||||||
|
break
|
||||||
|
|
||||||
|
charcode = 0x20
|
||||||
|
for line in it:
|
||||||
|
if (match := re.search(rb'".*"', line)):
|
||||||
|
yield chr(charcode), kls.decode_glyph(match.group(0)[1:-1].replace(b'\\\\', b'\\'))
|
||||||
|
charcode += 1
|
||||||
|
else:
|
||||||
|
if b'}' in line:
|
||||||
|
break
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
import time
|
||||||
|
t1 = time.time()
|
||||||
|
Newstroke()
|
||||||
|
t2 = time.time()
|
||||||
|
print((t2-t1)*1000)
|
||||||
|
|
@ -21,6 +21,7 @@
|
||||||
|
|
||||||
import re
|
import re
|
||||||
import math
|
import math
|
||||||
|
import copy
|
||||||
import warnings
|
import warnings
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
import dataclasses
|
import dataclasses
|
||||||
|
|
@ -131,7 +132,7 @@ class GerberFile(CamFile):
|
||||||
nonlocal cache, settings
|
nonlocal cache, settings
|
||||||
if isinstance(aperture, apertures.ApertureMacroInstance):
|
if isinstance(aperture, apertures.ApertureMacroInstance):
|
||||||
macro = aperture.macro
|
macro = aperture.macro
|
||||||
macro_def = macro.to_gerber(unit=settings.unit)
|
macro_def = macro.to_gerber(settings)
|
||||||
if macro_def not in cache:
|
if macro_def not in cache:
|
||||||
cache[macro_def] = macro
|
cache[macro_def] = macro
|
||||||
|
|
||||||
|
|
@ -151,7 +152,7 @@ class GerberFile(CamFile):
|
||||||
|
|
||||||
self.map_apertures(lookup)
|
self.map_apertures(lookup)
|
||||||
|
|
||||||
def to_excellon(self, plated=None, errors='raise'):
|
def to_excellon(self, plated=None, errors='raise', holes_only=False):
|
||||||
""" Convert this excellon file into a :py:class:`~.excellon.ExcellonFile`. This will convert interpolated lines
|
""" Convert this excellon file into a :py:class:`~.excellon.ExcellonFile`. This will convert interpolated lines
|
||||||
into slots, and circular aperture flashes into holes. Other features such as ``G36`` polygons or flashes with
|
into slots, and circular aperture flashes into holes. Other features such as ``G36`` polygons or flashes with
|
||||||
non-circular apertures will result in a :py:obj:`ValueError`. You can, of course, programmatically remove such
|
non-circular apertures will result in a :py:obj:`ValueError`. You can, of course, programmatically remove such
|
||||||
|
|
@ -159,7 +160,10 @@ class GerberFile(CamFile):
|
||||||
new_objs = []
|
new_objs = []
|
||||||
new_tools = {}
|
new_tools = {}
|
||||||
for obj in self.objects:
|
for obj in self.objects:
|
||||||
if not (isinstance(obj, go.Line) or isinstance(obj, go.Arc) or isinstance(obj, go.Flash)) or \
|
if holes_only and not isinstance(obj, go.Flash):
|
||||||
|
continue
|
||||||
|
|
||||||
|
if not isinstance(obj, (go.Line, go.Arc, go.Flash)) or \
|
||||||
not isinstance(getattr(obj, 'aperture', None), apertures.CircleAperture):
|
not isinstance(getattr(obj, 'aperture', None), apertures.CircleAperture):
|
||||||
if errors == 'raise':
|
if errors == 'raise':
|
||||||
raise ValueError(f'Cannot convert {obj} to excellon.')
|
raise ValueError(f'Cannot convert {obj} to excellon.')
|
||||||
|
|
@ -283,13 +287,24 @@ class GerberFile(CamFile):
|
||||||
|
|
||||||
self.dedup_apertures()
|
self.dedup_apertures()
|
||||||
|
|
||||||
am_stmt = lambda macro: f'%AM{macro.name}*\n{macro.to_gerber(unit=settings.unit)}*\n%'
|
am_stmt = lambda macro: f'%AM{macro.name}*\n{macro.to_gerber(settings)}*\n%'
|
||||||
for macro in self.aperture_macros():
|
|
||||||
yield am_stmt(macro)
|
|
||||||
|
|
||||||
aperture_map = {ap: num for num, ap in enumerate(self.apertures(), start=10)}
|
aperture_map = {ap: num for num, ap in enumerate(self.apertures(), start=10)}
|
||||||
for aperture, number in aperture_map.items():
|
|
||||||
yield f'%ADD{number}{aperture.to_gerber(settings)}*%'
|
if settings.calculate_out_all_aperture_macros:
|
||||||
|
adds = []
|
||||||
|
for aperture, number in aperture_map.items():
|
||||||
|
if isinstance(aperture, apertures.ApertureMacroInstance):
|
||||||
|
aperture = aperture.calculate_out(settings.unit, macro_name=f'CALCM{number}')
|
||||||
|
yield am_stmt(aperture.macro)
|
||||||
|
adds.append(f'%ADD{number}{aperture.to_gerber(settings)}*%')
|
||||||
|
yield from adds
|
||||||
|
|
||||||
|
else:
|
||||||
|
for macro in self.aperture_macros():
|
||||||
|
yield am_stmt(macro)
|
||||||
|
|
||||||
|
for aperture, number in aperture_map.items():
|
||||||
|
yield f'%ADD{number}{aperture.to_gerber(settings)}*%'
|
||||||
|
|
||||||
def warn(msg, kls=SyntaxWarning):
|
def warn(msg, kls=SyntaxWarning):
|
||||||
warnings.warn(msg, kls)
|
warnings.warn(msg, kls)
|
||||||
|
|
@ -360,8 +375,8 @@ class GerberFile(CamFile):
|
||||||
def invert_polarity(self):
|
def invert_polarity(self):
|
||||||
""" Invert the polarity (color) of each object in this file. """
|
""" Invert the polarity (color) of each object in this file. """
|
||||||
for obj in self.objects:
|
for obj in self.objects:
|
||||||
obj.polarity_dark = not p.polarity_dark
|
obj.polarity_dark = not obj.polarity_dark
|
||||||
|
|
||||||
|
|
||||||
class GraphicsState:
|
class GraphicsState:
|
||||||
""" Internal class used to track Gerber processing state during import and export.
|
""" Internal class used to track Gerber processing state during import and export.
|
||||||
|
|
@ -448,7 +463,7 @@ class GraphicsState:
|
||||||
obj = go.Flash(*self.map_coord(*self.point), self.aperture,
|
obj = go.Flash(*self.map_coord(*self.point), self.aperture,
|
||||||
polarity_dark=self._polarity_dark,
|
polarity_dark=self._polarity_dark,
|
||||||
unit=self.unit,
|
unit=self.unit,
|
||||||
attrs=self.object_attrs)
|
attrs=copy.copy(self.object_attrs))
|
||||||
return obj
|
return obj
|
||||||
|
|
||||||
def interpolate(self, x, y, i=None, j=None, aperture=True, multi_quadrant=False):
|
def interpolate(self, x, y, i=None, j=None, aperture=True, multi_quadrant=False):
|
||||||
|
|
@ -474,13 +489,13 @@ class GraphicsState:
|
||||||
raise SyntaxError("i/j coordinates given for linear D01 operation (which doesn't take i/j)")
|
raise SyntaxError("i/j coordinates given for linear D01 operation (which doesn't take i/j)")
|
||||||
|
|
||||||
return go.Line(*old_point, *self.map_coord(*self.point), aperture,
|
return go.Line(*old_point, *self.map_coord(*self.point), aperture,
|
||||||
polarity_dark=self._polarity_dark, unit=unit, attrs=self.object_attrs)
|
polarity_dark=self._polarity_dark, unit=unit, attrs=copy.copy(self.object_attrs))
|
||||||
|
|
||||||
else:
|
else:
|
||||||
if i is None and j is None:
|
if i is None and j is None:
|
||||||
self.warn('Linear segment implied during arc interpolation mode through D01 w/o I, J values')
|
self.warn('Linear segment implied during arc interpolation mode through D01 w/o I, J values')
|
||||||
return go.Line(*old_point, *self.map_coord(*self.point), aperture,
|
return go.Line(*old_point, *self.map_coord(*self.point), aperture,
|
||||||
polarity_dark=self._polarity_dark, unit=unit, attrs=self.object_attrs)
|
polarity_dark=self._polarity_dark, unit=unit, attrs=copy.copy(self.object_attrs))
|
||||||
|
|
||||||
else:
|
else:
|
||||||
if i is None:
|
if i is None:
|
||||||
|
|
@ -497,7 +512,7 @@ class GraphicsState:
|
||||||
if not multi_quadrant:
|
if not multi_quadrant:
|
||||||
return go.Arc(*old_point, *new_point, *self.map_coord(i, j, relative=True),
|
return go.Arc(*old_point, *new_point, *self.map_coord(i, j, relative=True),
|
||||||
clockwise=clockwise, aperture=(self.aperture if aperture else None),
|
clockwise=clockwise, aperture=(self.aperture if aperture else None),
|
||||||
polarity_dark=self._polarity_dark, unit=unit, attrs=self.object_attrs)
|
polarity_dark=self._polarity_dark, unit=unit, attrs=copy.copy(self.object_attrs))
|
||||||
|
|
||||||
else:
|
else:
|
||||||
if math.isclose(old_point[0], new_point[0]) and math.isclose(old_point[1], new_point[1]):
|
if math.isclose(old_point[0], new_point[0]) and math.isclose(old_point[1], new_point[1]):
|
||||||
|
|
@ -510,7 +525,7 @@ class GraphicsState:
|
||||||
|
|
||||||
arc = lambda cx, cy: go.Arc(*old_point, *new_point, cx, cy,
|
arc = lambda cx, cy: go.Arc(*old_point, *new_point, cx, cy,
|
||||||
clockwise=clockwise, aperture=aperture,
|
clockwise=clockwise, aperture=aperture,
|
||||||
polarity_dark=self._polarity_dark, unit=unit, attrs=self.object_attrs)
|
polarity_dark=self._polarity_dark, unit=unit, attrs=copy.copy(self.object_attrs))
|
||||||
arcs = [ arc(cx, cy), arc(-cx, cy), arc(cx, -cy), arc(-cx, -cy) ]
|
arcs = [ arc(cx, cy), arc(-cx, cy), arc(cx, -cy), arc(-cx, -cy) ]
|
||||||
arcs = sorted(arcs, key=lambda a: a.numeric_error())
|
arcs = sorted(arcs, key=lambda a: a.numeric_error())
|
||||||
|
|
||||||
|
|
@ -572,9 +587,9 @@ class GraphicsState:
|
||||||
|
|
||||||
def interpolation_mode_statement(self):
|
def interpolation_mode_statement(self):
|
||||||
return {
|
return {
|
||||||
InterpMode.LINEAR: 'G01',
|
InterpMode.LINEAR: 'G01*',
|
||||||
InterpMode.CIRCULAR_CW: 'G02',
|
InterpMode.CIRCULAR_CW: 'G02*',
|
||||||
InterpMode.CIRCULAR_CCW: 'G03'}[self.interpolation_mode]
|
InterpMode.CIRCULAR_CCW: 'G03*'}[self.interpolation_mode]
|
||||||
|
|
||||||
|
|
||||||
class GerberParser:
|
class GerberParser:
|
||||||
|
|
@ -584,6 +599,8 @@ class GerberParser:
|
||||||
NUMBER = r"[\+-]?\d+"
|
NUMBER = r"[\+-]?\d+"
|
||||||
DECIMAL = r"[\+-]?\d+([.]?\d+)?"
|
DECIMAL = r"[\+-]?\d+([.]?\d+)?"
|
||||||
NAME = r"[a-zA-Z_$\.][a-zA-Z_$\.0-9+\-]+"
|
NAME = r"[a-zA-Z_$\.][a-zA-Z_$\.0-9+\-]+"
|
||||||
|
MAX_STEP_REPEAT_INSTANCES = 100000
|
||||||
|
MAX_STEP_REPEAT_RESULT_OBJECTS = 100000
|
||||||
|
|
||||||
STATEMENT_REGEXES = {
|
STATEMENT_REGEXES = {
|
||||||
'coord': fr"(G0?[123]|G74|G75|G54|G55)?\s*(?:X\+?(-?)({NUMBER}))?(?:Y\+?(-?)({NUMBER}))?" \
|
'coord': fr"(G0?[123]|G74|G75|G54|G55)?\s*(?:X\+?(-?)({NUMBER}))?(?:Y\+?(-?)({NUMBER}))?" \
|
||||||
|
|
@ -591,6 +608,7 @@ class GerberParser:
|
||||||
fr"(?:D0?([123]))?$",
|
fr"(?:D0?([123]))?$",
|
||||||
'region_start': r'G36$',
|
'region_start': r'G36$',
|
||||||
'region_end': r'G37$',
|
'region_end': r'G37$',
|
||||||
|
'eof': r"(D02)?M0?[02]", # P-CAD 2006 files have a spurious D02 before M02 as in "D02M02"
|
||||||
'aperture': r"(G54|G55)?\s*D(?P<number>\d+)",
|
'aperture': r"(G54|G55)?\s*D(?P<number>\d+)",
|
||||||
# Allegro combines format spec and unit into one long illegal extended command.
|
# Allegro combines format spec and unit into one long illegal extended command.
|
||||||
'allegro_format_spec': r"FS(?P<zero>(L|T|D))?(?P<notation>(A|I))[NG0-9]*X(?P<x>[0-7][0-7])Y(?P<y>[0-7][0-7])[DM0-9]*\*MO(?P<unit>IN|MM)",
|
'allegro_format_spec': r"FS(?P<zero>(L|T|D))?(?P<notation>(A|I))[NG0-9]*X(?P<x>[0-7][0-7])Y(?P<y>[0-7][0-7])[DM0-9]*\*MO(?P<unit>IN|MM)",
|
||||||
|
|
@ -611,9 +629,9 @@ class GerberParser:
|
||||||
'aperture_definition': fr"ADD(?P<number>\d+)(?P<shape>C|R|O|P|{NAME})(,(?P<modifiers>[^,%]*))?$",
|
'aperture_definition': fr"ADD(?P<number>\d+)(?P<shape>C|R|O|P|{NAME})(,(?P<modifiers>[^,%]*))?$",
|
||||||
'aperture_macro': fr"AM(?P<name>{NAME})\*(?P<macro>[^%]*)",
|
'aperture_macro': fr"AM(?P<name>{NAME})\*(?P<macro>[^%]*)",
|
||||||
'siemens_garbage': r'^ICAS$',
|
'siemens_garbage': r'^ICAS$',
|
||||||
|
'step_repeat': fr'^SR(?P<coords>X(?P<X>[0-9]+)Y(?P<Y>[0-9]+)I(?P<I>{DECIMAL})J(?P<J>{DECIMAL}))?$',
|
||||||
'old_unit':r'(?P<mode>G7[01])',
|
'old_unit':r'(?P<mode>G7[01])',
|
||||||
'old_notation': r'(?P<mode>G9[01])',
|
'old_notation': r'(?P<mode>G9[01])',
|
||||||
'eof': r"M0?[02]",
|
|
||||||
'ignored': r"(?P<stmt>M01)",
|
'ignored': r"(?P<stmt>M01)",
|
||||||
# NOTE: The official spec says names can be empty or contain commas. I think that doesn't make sense.
|
# NOTE: The official spec says names can be empty or contain commas. I think that doesn't make sense.
|
||||||
'attribute': r"(?P<eagle_garbage>G04 #@! %)?(?P<type>TF|TA|TO|TD)(?P<name>[._$a-zA-Z][._$a-zA-Z0-9]*)?(,(?P<value>.*))?",
|
'attribute': r"(?P<eagle_garbage>G04 #@! %)?(?P<type>TF|TA|TO|TD)(?P<name>[._$a-zA-Z][._$a-zA-Z0-9]*)?(,(?P<value>.*))?",
|
||||||
|
|
@ -631,6 +649,8 @@ class GerberParser:
|
||||||
self.aperture_map = {}
|
self.aperture_map = {}
|
||||||
self.aperture_macros = {}
|
self.aperture_macros = {}
|
||||||
self.current_region = None
|
self.current_region = None
|
||||||
|
self.step_repeat_coords = None
|
||||||
|
self.step_repeat_objects = None
|
||||||
self.eof_found = False
|
self.eof_found = False
|
||||||
self.multi_quadrant_mode = None # used only for syntax checking
|
self.multi_quadrant_mode = None # used only for syntax checking
|
||||||
self.macros = {}
|
self.macros = {}
|
||||||
|
|
@ -773,7 +793,10 @@ class GerberParser:
|
||||||
# in multi-quadrant mode this may return None if start and end point of the arc are the same.
|
# in multi-quadrant mode this may return None if start and end point of the arc are the same.
|
||||||
obj = self.graphics_state.interpolate(x, y, i, j, multi_quadrant=self.multi_quadrant_mode)
|
obj = self.graphics_state.interpolate(x, y, i, j, multi_quadrant=self.multi_quadrant_mode)
|
||||||
if obj is not None:
|
if obj is not None:
|
||||||
self.target.objects.append(obj)
|
if self.step_repeat_objects:
|
||||||
|
self.step_repeat_objects.append(obj)
|
||||||
|
else:
|
||||||
|
self.target.objects.append(obj)
|
||||||
else:
|
else:
|
||||||
obj = self.graphics_state.interpolate(x, y, i, j, aperture=False, multi_quadrant=self.multi_quadrant_mode)
|
obj = self.graphics_state.interpolate(x, y, i, j, aperture=False, multi_quadrant=self.multi_quadrant_mode)
|
||||||
if obj is not None:
|
if obj is not None:
|
||||||
|
|
@ -784,14 +807,21 @@ class GerberParser:
|
||||||
if self.current_region:
|
if self.current_region:
|
||||||
# Start a new region for every outline. As gerber has no concept of fill rules or winding numbers,
|
# Start a new region for every outline. As gerber has no concept of fill rules or winding numbers,
|
||||||
# it does not make a graphical difference, and it makes the implementation slightly easier.
|
# it does not make a graphical difference, and it makes the implementation slightly easier.
|
||||||
self.target.objects.append(self.current_region)
|
if self.step_repeat_objects:
|
||||||
|
self.step_repeat_objects.append(self.current_region)
|
||||||
|
else:
|
||||||
|
self.target.objects.append(self.current_region)
|
||||||
self.current_region = go.Region(
|
self.current_region = go.Region(
|
||||||
polarity_dark=self.graphics_state.polarity_dark,
|
polarity_dark=self.graphics_state.polarity_dark,
|
||||||
unit=self.file_settings.unit)
|
unit=self.file_settings.unit)
|
||||||
|
|
||||||
elif op == '3':
|
elif op == '3':
|
||||||
if self.current_region is None:
|
if self.current_region is None:
|
||||||
self.target.objects.append(self.graphics_state.flash(x, y))
|
obj = self.graphics_state.flash(x, y)
|
||||||
|
if self.step_repeat_objects:
|
||||||
|
self.step_repeat_objects.append(obj)
|
||||||
|
else:
|
||||||
|
self.target.objects.append(obj)
|
||||||
else:
|
else:
|
||||||
raise SyntaxError('DO3 flash statement inside region')
|
raise SyntaxError('DO3 flash statement inside region')
|
||||||
|
|
||||||
|
|
@ -832,6 +862,11 @@ class GerberParser:
|
||||||
if match['shape'] in 'RO' and (math.isclose(modifiers[0], 0) or math.isclose(modifiers[1], 0)):
|
if match['shape'] in 'RO' and (math.isclose(modifiers[0], 0) or math.isclose(modifiers[1], 0)):
|
||||||
self.warn('Definition of zero-width and/or zero-height rectangle or obround aperture. This is invalid according to spec.' )
|
self.warn('Definition of zero-width and/or zero-height rectangle or obround aperture. This is invalid according to spec.' )
|
||||||
|
|
||||||
|
# Polygon aperture rotation is specified in degrees, but radians are easier to work with
|
||||||
|
if match['shape'] == 'P':
|
||||||
|
if len(modifiers) > 2:
|
||||||
|
modifiers[2] = math.radians(modifiers[2])
|
||||||
|
|
||||||
new_aperture = kls(*modifiers, unit=self.file_settings.unit, attrs=tuple(self.aperture_attrs.items()),
|
new_aperture = kls(*modifiers, unit=self.file_settings.unit, attrs=tuple(self.aperture_attrs.items()),
|
||||||
original_number=number)
|
original_number=number)
|
||||||
|
|
||||||
|
|
@ -1048,11 +1083,40 @@ class GerberParser:
|
||||||
|
|
||||||
else:
|
else:
|
||||||
target = {'TF': self.file_attrs, 'TO': self.graphics_state.object_attrs, 'TA': self.aperture_attrs}[match['type']]
|
target = {'TF': self.file_attrs, 'TO': self.graphics_state.object_attrs, 'TA': self.aperture_attrs}[match['type']]
|
||||||
target[match['name']] = match['value'].split(',')
|
target[match['name']] = tuple(match['value'].split(',')) if match['value'] else ()
|
||||||
|
|
||||||
if 'EAGLE' in self.file_attrs.get('.GenerationSoftware', []) or match['eagle_garbage']:
|
if 'EAGLE' in self.file_attrs.get('.GenerationSoftware', []) or match['eagle_garbage']:
|
||||||
self.generator_hints.append('eagle')
|
self.generator_hints.append('eagle')
|
||||||
|
|
||||||
|
def _parse_step_repeat(self, match):
|
||||||
|
if match['coords']:
|
||||||
|
if self.step_repeat_coords:
|
||||||
|
raise SyntaxError('SR step-repeat called inside ongoing SR step-repeat')
|
||||||
|
|
||||||
|
x, y = int(match['X']), int(match['Y'])
|
||||||
|
i, j = float(match['I']), float(match['J'])
|
||||||
|
if x < 1 or y < 1:
|
||||||
|
raise SyntaxError('SR step-repeat X and Y values must be at least 1')
|
||||||
|
if x * y > self.MAX_STEP_REPEAT_INSTANCES:
|
||||||
|
raise SyntaxError('SR step-repeat expands to too many instances')
|
||||||
|
|
||||||
|
self.step_repeat_coords = (x, y, i, j)
|
||||||
|
self.step_repeat_objects = []
|
||||||
|
|
||||||
|
else:
|
||||||
|
x, y, i, j = self.step_repeat_coords
|
||||||
|
if len(self.step_repeat_objects) * x * y > self.MAX_STEP_REPEAT_RESULT_OBJECTS:
|
||||||
|
raise SyntaxError('SR step-repeat expands to too many objects')
|
||||||
|
|
||||||
|
for obj in self.step_repeat_objects:
|
||||||
|
for nx in range(x):
|
||||||
|
for ny in range(y):
|
||||||
|
new_obj = copy.copy(obj)
|
||||||
|
new_obj.offset(i * nx, j * ny)
|
||||||
|
self.target.objects.append(new_obj)
|
||||||
|
self.step_repeat_coords = None
|
||||||
|
self.step_repeat_objects = None
|
||||||
|
|
||||||
def _parse_eof(self, match):
|
def _parse_eof(self, match):
|
||||||
self.eof_found = True
|
self.eof_found = True
|
||||||
|
|
||||||
|
|
@ -244,6 +244,59 @@ def rotate_point(x, y, angle, cx=0, cy=0):
|
||||||
cy + (x - cx) * math.sin(-angle) + (y - cy) * math.cos(-angle))
|
cy + (x - cx) * math.sin(-angle) + (y - cy) * math.cos(-angle))
|
||||||
|
|
||||||
|
|
||||||
|
def sweep_angle(cx, cy, x1, y1, x2, y2, clockwise):
|
||||||
|
""" Calculate absolute sweep angle of arc. This is always a positive number.
|
||||||
|
|
||||||
|
:returns: Angle in clockwise radian between ``0`` and ``2*math.pi``
|
||||||
|
:rtype: float
|
||||||
|
"""
|
||||||
|
x1, y1 = x1-cx, y1-cy
|
||||||
|
x2, y2 = x2-cx, y2-cy
|
||||||
|
|
||||||
|
a1, a2 = math.atan2(y1, x1), math.atan2(y2, x2)
|
||||||
|
f = abs(a2 - a1)
|
||||||
|
if not clockwise:
|
||||||
|
if a2 > a1:
|
||||||
|
return a2 - a1
|
||||||
|
else:
|
||||||
|
return 2*math.pi - abs(a2 - a1)
|
||||||
|
else:
|
||||||
|
if a1 > a2:
|
||||||
|
return a1 - a2
|
||||||
|
else:
|
||||||
|
return 2*math.pi - abs(a1 - a2)
|
||||||
|
|
||||||
|
|
||||||
|
def approximate_arc(cx, cy, x1, y1, x2, y2, clockwise, max_error=1e-2, clip_max_error=True):
|
||||||
|
# TODO the max_angle calculation below is a bit off -- we over-estimate the error, and thus produce finer
|
||||||
|
# results than necessary. Fix this.
|
||||||
|
|
||||||
|
r = math.dist((x1, y1), (cx, cy))
|
||||||
|
|
||||||
|
if clip_max_error:
|
||||||
|
# 1 - math.sqrt(1 - 0.5*math.sqrt(2))
|
||||||
|
max_error = min(max_error, r*0.4588038998538031)
|
||||||
|
|
||||||
|
elif max_error >= r:
|
||||||
|
yield (x1, y1)
|
||||||
|
yield (x2, y2)
|
||||||
|
return
|
||||||
|
|
||||||
|
# see https://www.mathopenref.com/sagitta.html
|
||||||
|
l = math.sqrt(r**2 - (r - max_error)**2)
|
||||||
|
|
||||||
|
angle_max = math.asin(l/r)
|
||||||
|
alpha = sweep_angle(cx, cy, x1, y1, x2, y2, clockwise)
|
||||||
|
num_segments = math.ceil(alpha / angle_max)
|
||||||
|
angle = alpha / num_segments
|
||||||
|
|
||||||
|
if not clockwise:
|
||||||
|
angle = -angle
|
||||||
|
|
||||||
|
for i in range(num_segments + 1):
|
||||||
|
yield rotate_point(x1, y1, i*angle, cx, cy)
|
||||||
|
|
||||||
|
|
||||||
def min_none(a, b):
|
def min_none(a, b):
|
||||||
""" Like the ``min(..)`` builtin, but if either value is ``None``, returns the other. """
|
""" Like the ``min(..)`` builtin, but if either value is ``None``, returns the other. """
|
||||||
if a is None:
|
if a is None:
|
||||||
|
|
@ -340,11 +393,9 @@ def arc_bounds(x1, y1, x2, y2, cx, cy, clockwise):
|
||||||
# This solution manages to handle circular arcs given in gerber format (with explicit center and endpoints, plus
|
# This solution manages to handle circular arcs given in gerber format (with explicit center and endpoints, plus
|
||||||
# sweep direction instead of a format with e.g. angles and radius) without any trigonometric functions (e.g. atan2).
|
# sweep direction instead of a format with e.g. angles and radius) without any trigonometric functions (e.g. atan2).
|
||||||
#
|
#
|
||||||
# cx, cy are relative to p1.
|
# cx, cy are in absolute coordinates.
|
||||||
|
|
||||||
# Center arc on cx, cy
|
# Center arc on cx, cy
|
||||||
cx += x1
|
|
||||||
cy += y1
|
|
||||||
x1 -= cx
|
x1 -= cx
|
||||||
x2 -= cx
|
x2 -= cx
|
||||||
y1 -= cy
|
y1 -= cy
|
||||||
|
|
@ -354,6 +405,10 @@ def arc_bounds(x1, y1, x2, y2, cx, cy, clockwise):
|
||||||
# Calculate radius
|
# Calculate radius
|
||||||
r = math.sqrt(x1**2 + y1**2)
|
r = math.sqrt(x1**2 + y1**2)
|
||||||
|
|
||||||
|
# Special case: Gerber defines an arc with p1 == p2 as a full circle.
|
||||||
|
if math.isclose(x1, x2) and math.isclose(y1, y2):
|
||||||
|
return (cx-r, cy-r), (cx+r, cy+r)
|
||||||
|
|
||||||
# Calculate in which half-planes (north/south, west/east) P1 and P2 lie.
|
# Calculate in which half-planes (north/south, west/east) P1 and P2 lie.
|
||||||
# Note that we assume the y axis points upwards, as in Gerber and maths.
|
# Note that we assume the y axis points upwards, as in Gerber and maths.
|
||||||
# SVG has its y axis pointing downwards.
|
# SVG has its y axis pointing downwards.
|
||||||
|
|
@ -457,25 +512,25 @@ def point_line_distance(l1, l2, p):
|
||||||
|
|
||||||
|
|
||||||
def svg_arc(old, new, center, clockwise):
|
def svg_arc(old, new, center, clockwise):
|
||||||
""" Format an SVG circular arc "A" path data entry given an arc in Gerber notation (i.e. with center relative to
|
""" Format an SVG circular arc "A" path data entry given an arc in Gerber notation (but with center in absolute
|
||||||
first point).
|
coordinates).
|
||||||
|
|
||||||
:rtype: str
|
:rtype: str
|
||||||
"""
|
"""
|
||||||
r = float(math.hypot(*center))
|
r = float(math.dist(old, center))
|
||||||
# invert sweep flag since the svg y axis is mirrored
|
# invert sweep flag since the svg y axis is mirrored
|
||||||
sweep_flag = int(not clockwise)
|
sweep_flag = int(not clockwise)
|
||||||
# In the degenerate case where old == new, we always take the long way around. To represent this "full-circle arc"
|
# In the degenerate case where old == new, we always take the long way around. To represent this "full-circle arc"
|
||||||
# in SVG, we have to split it into two.
|
# in SVG, we have to split it into two.
|
||||||
if math.isclose(math.dist(old, new), 0):
|
if math.isclose(math.dist(old, new), 0):
|
||||||
intermediate = old[0] + 2*center[0], old[1] + 2*center[1]
|
intermediate = old[0] + 2*(center[0]-old[0]), old[1] + 2*(center[1]-old[1])
|
||||||
# Note that we have to preserve the sweep flag to avoid causing self-intersections by flipping the direction of
|
# Note that we have to preserve the sweep flag to avoid causing self-intersections by flipping the direction of
|
||||||
# a circular cutin
|
# a circular cutin
|
||||||
return f'A {r:.6} {r:.6} 0 1 {sweep_flag} {float(intermediate[0]):.6} {float(intermediate[1]):.6} ' +\
|
return f'A {r:.6} {r:.6} 0 1 {sweep_flag} {float(intermediate[0]):.6} {float(intermediate[1]):.6} ' +\
|
||||||
f'A {r:.6} {r:.6} 0 1 {sweep_flag} {float(new[0]):.6} {float(new[1]):.6}'
|
f'A {r:.6} {r:.6} 0 1 {sweep_flag} {float(new[0]):.6} {float(new[1]):.6}'
|
||||||
|
|
||||||
else: # normal case
|
else: # normal case
|
||||||
d = point_line_distance(old, new, (old[0]+center[0], old[1]+center[1]))
|
d = point_line_distance(old, new, (center[0], center[1]))
|
||||||
large_arc = int((d < 0) == clockwise)
|
large_arc = int((d < 0) == clockwise)
|
||||||
return f'A {r:.6} {r:.6} 0 {large_arc} {sweep_flag} {float(new[0]):.6} {float(new[1]):.6}'
|
return f'A {r:.6} {r:.6} 0 {large_arc} {sweep_flag} {float(new[0]):.6} {float(new[1]):.6}'
|
||||||
|
|
||||||
|
|
@ -519,6 +574,7 @@ def setup_svg(tags, bounds, margin=0, arg_unit=MM, svg_unit=MM, pagecolor='white
|
||||||
return tag('svg', tags,
|
return tag('svg', tags,
|
||||||
width=f'{w}{svg_unit}', height=f'{h}{svg_unit}',
|
width=f'{w}{svg_unit}', height=f'{h}{svg_unit}',
|
||||||
viewBox=f'{min_x} {min_y} {w} {h}',
|
viewBox=f'{min_x} {min_y} {w} {h}',
|
||||||
|
style=f'background-color:{pagecolor}',
|
||||||
**namespaces,
|
**namespaces,
|
||||||
root=True)
|
root=True)
|
||||||
|
|
||||||
|
|
@ -549,6 +605,18 @@ def point_in_polygon(point, poly):
|
||||||
return res
|
return res
|
||||||
|
|
||||||
|
|
||||||
|
def polygon_area(poly):
|
||||||
|
# https://en.wikipedia.org/wiki/Shoelace_formula
|
||||||
|
|
||||||
|
if not poly or len(poly) < 3:
|
||||||
|
return 0
|
||||||
|
|
||||||
|
acc = 0
|
||||||
|
for (x1, y1), (x2, y2) in zip(poly, poly[-1:] + poly):
|
||||||
|
acc += (y1 + y2) * (x1 - x2)
|
||||||
|
return acc/2
|
||||||
|
|
||||||
|
|
||||||
def bbox_intersect(a, b):
|
def bbox_intersect(a, b):
|
||||||
if a is None or b is None:
|
if a is None or b is None:
|
||||||
return False
|
return False
|
||||||
|
|
@ -561,3 +629,16 @@ def bbox_intersect(a, b):
|
||||||
|
|
||||||
return x_overlap and y_overlap
|
return x_overlap and y_overlap
|
||||||
|
|
||||||
|
|
||||||
|
def bbox_contains(outer, inner):
|
||||||
|
if outer is None or inner is None:
|
||||||
|
return False
|
||||||
|
|
||||||
|
(xa_min, ya_min), (xa_max, ya_max) = outer
|
||||||
|
(xb_min, yb_min), (xb_max, yb_max) = inner
|
||||||
|
|
||||||
|
contained_x = xa_min < xb_min and xb_max < xa_max
|
||||||
|
contained_y = ya_min < yb_min and yb_max < ya_max
|
||||||
|
|
||||||
|
return contained_x and contained_y
|
||||||
|
|
||||||
53
test2.py
Normal file
|
|
@ -0,0 +1,53 @@
|
||||||
|
from gerbonara import *
|
||||||
|
from shapely import *
|
||||||
|
|
||||||
|
stack = layers.LayerStack.open('/home/jaseg/proj/ihsm-strain-gage-controller-hw/pcb/gerber')
|
||||||
|
# Let's work in mm here. Gerbonara will take care to convert units when the file is in US customary units.
|
||||||
|
(x1, y1), (x2, y2) = stack.bounding_box(unit=utils.MM)
|
||||||
|
|
||||||
|
for l in [stack['bottom mask'], stack['top mask']]:
|
||||||
|
|
||||||
|
# The solder mask gerber layer by convention is "negative". That is, a "dark" polarity (drawn) Gerber primitive
|
||||||
|
# will result in an opening in the solder mask. Conversely, an empty gerber file would lead to the entire board
|
||||||
|
# being covered in solder mask.
|
||||||
|
#
|
||||||
|
# Here, we add a rectangle covering the entire board so the entire board is *free* of solder mask.
|
||||||
|
|
||||||
|
new = [graphic_objects.Region(
|
||||||
|
[(x1, y1), (x1, y2), (x2, y2), (x2, y1), (x1, y1)],
|
||||||
|
unit=utils.MM,
|
||||||
|
polarity_dark=True)]
|
||||||
|
|
||||||
|
# Iterate through all objects on the solder mask layer. In later KiCad versions, everything on the solder mask
|
||||||
|
# layer is exported as a Gerber region, which is a really bad idea, but makes things easy for us here.
|
||||||
|
for obj in l.objects:
|
||||||
|
if not obj.polarity_dark:
|
||||||
|
continue
|
||||||
|
|
||||||
|
if isinstance(obj, graphic_objects.Region):
|
||||||
|
regions = []
|
||||||
|
else:
|
||||||
|
regions = [graphic_objects.Region.from_arc_poly(prim.to_arc_poly())
|
||||||
|
for prim in obj.to_primitives(unit=utils.MM)]
|
||||||
|
|
||||||
|
for obj in regions:
|
||||||
|
# Convert the region to a shapely line string
|
||||||
|
ls = LineString(obj.outline).normalize()
|
||||||
|
|
||||||
|
# Ask shapely to offset the line string by 1 mm
|
||||||
|
out = ls.offset_curve(obj.unit(1, 'mm'))
|
||||||
|
|
||||||
|
# For negative offsets, this operation can result in an object being split up into multiple parts, so we
|
||||||
|
# might get back a MultiLineString instead of a LineString.
|
||||||
|
for ls in (out.geoms if hasattr(out, 'geoms') else [out]):
|
||||||
|
|
||||||
|
# Convert the resulting shapely object back to a Gerber region.
|
||||||
|
new.append(graphic_objects.Region(
|
||||||
|
unit=obj.unit,
|
||||||
|
polarity_dark=not obj.polarity_dark,
|
||||||
|
outline=list(ls.coords)))
|
||||||
|
|
||||||
|
# Append the new objects to the original layer data
|
||||||
|
l.objects = new + l.objects
|
||||||
|
# Write the modified layer stack to a new Gerber directory
|
||||||
|
stack.save_to_directory('/tmp/out')
|
||||||
0
tests/__init__.py
Normal file
172
tests/conftest.py
Normal file
|
|
@ -0,0 +1,172 @@
|
||||||
|
|
||||||
|
import os
|
||||||
|
from pathlib import Path
|
||||||
|
import tqdm
|
||||||
|
import multiprocessing.pool
|
||||||
|
import subprocess
|
||||||
|
from itertools import chain
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
from .image_support import ImageDifference, run_cargo_cmd, ImageSupport
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def kicad_container(request):
|
||||||
|
return request.config.kicad_container
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def kicad_footprints_libdir(request):
|
||||||
|
return request.config.kicad_footprints_libdir
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def kicad_symbols_libdir(request):
|
||||||
|
return request.config.kicad_symbols_libdir
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def img_support(request):
|
||||||
|
return request.config.image_support
|
||||||
|
|
||||||
|
|
||||||
|
def pytest_assertrepr_compare(op, left, right):
|
||||||
|
if isinstance(left, ImageDifference) or isinstance(right, ImageDifference):
|
||||||
|
diff = left if isinstance(left, ImageDifference) else right
|
||||||
|
return [
|
||||||
|
f'Image difference assertion failed.',
|
||||||
|
f' Calculated difference: {diff}',
|
||||||
|
f' Histogram: {diff.histogram}', ]
|
||||||
|
|
||||||
|
|
||||||
|
# store report in node object so tmp_gbr can determine if the test failed.
|
||||||
|
@pytest.hookimpl(tryfirst=True, hookwrapper=True)
|
||||||
|
def pytest_runtest_makereport(item, call):
|
||||||
|
outcome = yield
|
||||||
|
rep = outcome.get_result()
|
||||||
|
setattr(item, f'rep_{rep.when}', rep)
|
||||||
|
|
||||||
|
|
||||||
|
fail_dir = Path('gerbonara_test_failures')
|
||||||
|
def pytest_sessionstart(session):
|
||||||
|
if 'PYTEST_XDIST_WORKER' in os.environ: # only run this on the controller
|
||||||
|
return
|
||||||
|
|
||||||
|
for f in chain(fail_dir.glob('*.gbr'), fail_dir.glob('*.png')):
|
||||||
|
f.unlink()
|
||||||
|
|
||||||
|
try:
|
||||||
|
run_cargo_cmd('resvg', '--help', stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
||||||
|
except FileNotFoundError:
|
||||||
|
pytest.exit('resvg binary not found, aborting test.', 2)
|
||||||
|
|
||||||
|
|
||||||
|
def _update_repo_cache(lib_dir, repo_url, tag):
|
||||||
|
if not lib_dir.is_dir():
|
||||||
|
print(f'Checking out KiCad footprint repo tag {tag}')
|
||||||
|
subprocess.run(['git', '-c', 'advice.detachedHead=false', 'clone', '--branch', tag, '--depth', '1', repo_url, str(lib_dir)], check=True)
|
||||||
|
return True
|
||||||
|
|
||||||
|
else:
|
||||||
|
print(f'Found cached KiCad footprint checkout, updating to {tag}')
|
||||||
|
res = subprocess.run(['git', '-C', str(lib_dir), 'rev-parse', 'HEAD', f'{tag}^{{commit}}'], check=True, capture_output=True, text=True)
|
||||||
|
head_commit, tag_commit = res.stdout.strip().splitlines()
|
||||||
|
print('got commits', head_commit, tag_commit)
|
||||||
|
if head_commit != tag_commit:
|
||||||
|
subprocess.run(['git', '-C', str(lib_dir), 'fetch', '--depth', '1', 'origin', tag], check=True)
|
||||||
|
subprocess.run(['git', '-c', 'advice.detachedHead=false', '-C', str(lib_dir), 'reset', '--hard', tag], check=True)
|
||||||
|
subprocess.run(['git', '-C', str(lib_dir), 'clean', '--force', '-d', '-x'], check=True)
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
print('Up to date, only cleaning.')
|
||||||
|
subprocess.run(['git', '-C', str(lib_dir), 'clean', '--force', '-d', '-x'], check=True)
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
def pytest_addoption(parser):
|
||||||
|
parser.addini('kicad_footprints_tag', 'git tag or branch for KiCad footprint library repo used as testdata', default='main')
|
||||||
|
parser.addini('kicad_symbols_tag', 'git tag or branch for KiCad symbol library repo used as testdata', default='main')
|
||||||
|
parser.addini('kicad_container_tag', 'docker hub tag for the KiCad container to use for exporting footprint images', default='main')
|
||||||
|
parser.addini('kicad_source_tag', 'git tag for the KiCad source repo whose demos directory is used as testdata', default='main')
|
||||||
|
parser.addoption("--use-cached-data", action="store_true", help="Do not re-check git repo caches and podman image")
|
||||||
|
|
||||||
|
|
||||||
|
def pytest_configure(config):
|
||||||
|
os.nice(20)
|
||||||
|
# Resvg can sometimes consume a lot of memory. Make sure we don't kill the user's session.
|
||||||
|
if (oom_adj := Path('/proc/self/oom_adj')).is_file():
|
||||||
|
oom_adj.write_text('15\n')
|
||||||
|
|
||||||
|
if (lib_dir := os.environ.get('KICAD_FOOTPRINTS')):
|
||||||
|
config.kicad_footprints_libdir = Path(lib_dir).expanduser()
|
||||||
|
else:
|
||||||
|
config.kicad_footprints_libdir = config.cache.mkdir('kicad-footprints') / 'repo'
|
||||||
|
|
||||||
|
if (lib_dir := os.environ.get('KICAD_SYMBOLS')):
|
||||||
|
config.kicad_symbols_libdir = Path(lib_dir).expanduser()
|
||||||
|
else:
|
||||||
|
config.kicad_symbols_libdir = config.cache.mkdir('kicad-symbols') / 'repo'
|
||||||
|
|
||||||
|
if (lib_dir := os.environ.get('KICAD_SOURCE')):
|
||||||
|
config.kicad_source_dir = Path(lib_dir).expanduser()
|
||||||
|
else:
|
||||||
|
config.kicad_source_dir = config.cache.mkdir('kicad-source') / 'repo'
|
||||||
|
|
||||||
|
did_updates = False
|
||||||
|
is_pytest_controller = 'PYTEST_XDIST_WORKER' not in os.environ
|
||||||
|
if is_pytest_controller and not config.getoption("--use-cached-data"):
|
||||||
|
# Update cached library repos unless they are overridden from outside.
|
||||||
|
if not os.environ.get('KICAD_FOOTPRINTS'):
|
||||||
|
tag = config.getini('kicad_footprints_tag')
|
||||||
|
did_updates |= _update_repo_cache(config.kicad_footprints_libdir, 'https://gitlab.com/kicad/libraries/kicad-footprints', tag)
|
||||||
|
|
||||||
|
if not os.environ.get('KICAD_SYMBOLS'):
|
||||||
|
tag = config.getini('kicad_symbols_tag')
|
||||||
|
did_updates |= _update_repo_cache(config.kicad_symbols_libdir, 'https://gitlab.com/kicad/libraries/kicad-symbols', tag)
|
||||||
|
|
||||||
|
if not os.environ.get('KICAD_SOURCE'):
|
||||||
|
tag = config.getini('kicad_source_tag')
|
||||||
|
did_updates |= _update_repo_cache(config.kicad_source_dir, 'https://gitlab.com/kicad/code/kicad', tag)
|
||||||
|
|
||||||
|
tag = config.getini("kicad_container_tag")
|
||||||
|
config.kicad_container = os.environ.get('KICAD_CONTAINER', f'registry.hub.docker.com/kicad/kicad:{tag}')
|
||||||
|
|
||||||
|
if is_pytest_controller and not config.getoption("--use-cached-data"):
|
||||||
|
print('Checking podman image')
|
||||||
|
res = subprocess.run(['podman', 'image', 'exists', config.kicad_container])
|
||||||
|
if res.returncode:
|
||||||
|
print('Updating podman image')
|
||||||
|
subprocess.run(['podman', 'pull', config.kicad_container], check=True)
|
||||||
|
did_updates = True
|
||||||
|
else:
|
||||||
|
print('Up to date.')
|
||||||
|
|
||||||
|
config.image_support = ImageSupport(config.cache.mkdir('image_cache'), config.kicad_container)
|
||||||
|
|
||||||
|
if is_pytest_controller and did_updates and not config.getoption("--use-cached-data"):
|
||||||
|
print('Checking KiCad footprint library render cache')
|
||||||
|
with multiprocessing.pool.ThreadPool() as pool: # use thread pool here since we're only monitoring podman processes
|
||||||
|
lib_dirs = list(config.kicad_footprints_libdir.glob('*.pretty'))
|
||||||
|
res = list(tqdm.tqdm(pool.imap(lambda path: config.image_support.bulk_populate_kicad_fp_export_cache(path), lib_dirs), total=len(lib_dirs)))
|
||||||
|
|
||||||
|
|
||||||
|
def pytest_generate_tests(metafunc):
|
||||||
|
if 'kicad_library_file' in metafunc.fixturenames:
|
||||||
|
library_files = list(metafunc.config.kicad_symbols_libdir.glob('*.kicad_sym'))
|
||||||
|
metafunc.parametrize('kicad_library_file', library_files, ids=list(map(str, library_files)))
|
||||||
|
|
||||||
|
if 'kicad_mod_file' in metafunc.fixturenames:
|
||||||
|
mod_files = list(metafunc.config.kicad_footprints_libdir.glob('*.pretty/*.kicad_mod'))
|
||||||
|
metafunc.parametrize('kicad_mod_file', mod_files, ids=list(map(str, mod_files)))
|
||||||
|
|
||||||
|
if 'kicad_sch_file' in metafunc.fixturenames:
|
||||||
|
files = list(metafunc.config.kicad_source_dir.glob('demos/*.kicad_sch'))
|
||||||
|
files += list(metafunc.config.kicad_source_dir.glob('qa/data/**/*.kicad_sch'))
|
||||||
|
metafunc.parametrize('kicad_sch_file', files, ids=list(map(str, files)))
|
||||||
|
|
||||||
|
if 'kicad_pcb_file' in metafunc.fixturenames:
|
||||||
|
files = list(metafunc.config.kicad_source_dir.glob('demos/*.kicad_pcb'))
|
||||||
|
files += list(metafunc.config.kicad_source_dir.glob('qa/data/**/*.kicad_pcb'))
|
||||||
|
metafunc.parametrize('kicad_pcb_file', files, ids=list(map(str, files)))
|
||||||
|
|
||||||
|
Before Width: | Height: | Size: 6.1 KiB After Width: | Height: | Size: 6.1 KiB |
|
Before Width: | Height: | Size: 46 KiB After Width: | Height: | Size: 46 KiB |
|
Before Width: | Height: | Size: 6.3 KiB After Width: | Height: | Size: 6.3 KiB |
|
Before Width: | Height: | Size: 1.3 KiB After Width: | Height: | Size: 1.3 KiB |
|
Before Width: | Height: | Size: 5.8 KiB After Width: | Height: | Size: 5.8 KiB |
|
Before Width: | Height: | Size: 3.4 KiB After Width: | Height: | Size: 3.4 KiB |
|
Before Width: | Height: | Size: 4 KiB After Width: | Height: | Size: 4 KiB |
|
Before Width: | Height: | Size: 1.7 KiB After Width: | Height: | Size: 1.7 KiB |
|
Before Width: | Height: | Size: 70 KiB After Width: | Height: | Size: 70 KiB |
|
Before Width: | Height: | Size: 11 KiB After Width: | Height: | Size: 11 KiB |
|
Before Width: | Height: | Size: 9.8 KiB After Width: | Height: | Size: 9.8 KiB |
|
Before Width: | Height: | Size: 70 KiB After Width: | Height: | Size: 70 KiB |
|
Before Width: | Height: | Size: 94 KiB After Width: | Height: | Size: 94 KiB |
|
Before Width: | Height: | Size: 32 KiB After Width: | Height: | Size: 32 KiB |
|
Before Width: | Height: | Size: 32 KiB After Width: | Height: | Size: 32 KiB |
|
Before Width: | Height: | Size: 31 KiB After Width: | Height: | Size: 31 KiB |
|
Before Width: | Height: | Size: 556 B After Width: | Height: | Size: 556 B |
|
Before Width: | Height: | Size: 2.2 KiB After Width: | Height: | Size: 2.2 KiB |
|
Before Width: | Height: | Size: 18 KiB After Width: | Height: | Size: 18 KiB |