Compare commits

..

210 commits
dev ... main

Author SHA1 Message Date
Flavien Solt
736107f7a4 Reject oversized step-repeat expansions 2026-04-25 12:04:51 +02:00
Flavien Solt
e3674de08d Accept valueless Gerber attributes 2026-04-25 12:04:44 +02:00
Flavien Solt
516a9d337f Improve Excellon compatibility parsing 2026-04-21 11:13:15 +08:00
jaseg
2451b517e8 Add aperture macro expression tests 2026-03-22 13:22:11 +01:00
jaseg
bdd4008ab9 Fix macro polygon offset handling and add regression test
Applies github PR #1. Thanks to @SaumyaShah08 on github!
2026-03-22 11:46:47 +01:00
jaseg
6f006e2782 Start work on proper aperture macro tests 2026-03-21 13:24:50 +01:00
jaseg
8df709f55f Bump version to v1.6.2 2026-03-09 17:22:02 +01:00
jaseg
985e263cfe Add missing test data 2026-03-09 17:11:34 +01:00
jaseg
5ccfd7a259 Fix handling of zero-length line segments during pretty SVG export
Fixess gitlab issue #8
2026-03-09 17:10:15 +01:00
jaseg
bcc4aeefa7 doc: State more explicitly the nature of gerbonara's file name matching rules 2026-03-09 16:34:09 +01:00
jaseg
0c15111463 Add pretty_svg test, fix some bugs. 2026-03-09 16:31:08 +01:00
jaseg
1a0f519720 Add invert polarity test
Fixes gitlab issue #6
2026-03-09 16:07:37 +01:00
jaseg
3d0ddc3dc8 Update layer naming to track modern kicad 2026-03-08 16:49:26 +01:00
jaseg
575046a60c Fix ExcellonFile.to_gerber and add a unit test 2026-03-08 15:12:58 +01:00
jaseg
8de776616c Fix aperture macro thermal primitive
Fixes gitlab issues #10 and #11
2026-03-08 15:03:11 +01:00
jaseg
e19dec20b6 Fix ArcPoly primitive approximate_arcs 2026-03-08 14:38:13 +01:00
jaseg
97513df5d9 Improve regression test for gitlab issue #17
Fixes gitlab issue #17
2026-03-08 14:29:56 +01:00
jaseg
2ce55ebdca Fix arcpoly conversion bugs 2026-03-08 14:19:41 +01:00
jaseg
f3c95a42d4 Fix bug in rs274x Region close
Fixes gitlab issue #18
2026-03-08 13:23:29 +01:00
jaseg
8a2599f5f4 Add missing degree/radian conversion for polygon aperture rotation 2026-03-08 13:17:16 +01:00
jaseg
2647709215 Fix arc gerber export bug 2026-03-08 11:24:46 +01:00
jaseg
87413855bf tests WIP 2026-02-22 14:50:03 +01:00
Nitsan-Suron
af4fb2668f fixed invert_polarity 2026-02-22 12:42:22 +01:00
Matthew Mets
a47a694ba0 Convert polygonal aperture rotations to radians
Polygonal aperture rotations are specified in degrees, however
gerbonara wants radians. This is a fix for issue #12 :

https://gitlab.com/gerbolyze/gerbonara/-/issues/12
2025-12-10 13:32:10 +00:00
jaseg
dd8ad98f13 Bump version to v1.6.1 2025-12-10 14:23:14 +01:00
jaseg
a1ea416269 Fix uv build for newer container versions 2025-12-10 14:23:14 +01:00
jaseg
a7f0324506 Add protoserve entrypoint 2025-12-10 14:13:00 +01:00
jaseg
6ab4640099 Bump version to v1.6.0 2025-12-08 12:13:14 +01:00
jaseg
04a088c3e8 Update gitlab CI for uv build 2025-12-08 12:13:11 +01:00
jaseg
df4895abfa kicad: make footprint render layer_map argument optional 2025-12-08 11:52:42 +01:00
jaseg
42dfd1be7f Bring kicad PCB file format up to 9.0.5 2025-11-19 12:26:03 +01:00
jaseg
fd6880640d Add missing test file 2025-11-19 11:00:38 +01:00
jaseg
6a8bd8dc3f Kicad 9.0.5 PCB compatibility WIP 2025-11-18 23:43:29 +01:00
jaseg
46df184358 Fix remaining failing tests
This brings schematic i/o compatiblity up to 9.0.5.
2025-11-18 23:11:22 +01:00
jaseg
c2679260a0 Make test logic resource updating smarter
Only update repos/podman images if they're not as expected. Also only
run footprint export on changes.
2025-11-18 22:13:45 +01:00
jaseg
9d2d635eee kiacd 9.0 compatibility WIP 2025-11-17 17:38:59 +01:00
jaseg
75905f7d0c Fix failing tests
This adds basic support for kicad 9.0 library i/o
2025-11-17 12:07:15 +01:00
jaseg
7073b6e33f Finish transition to uv 2025-11-16 21:19:13 +01:00
jaseg
718de2b338 WIP move project to uv 2025-11-16 19:37:48 +01:00
jaseg
379cf273cb Fix out-of-date documentation 2025-08-22 13:24:34 +02:00
jaseg
da6d8349fa Bump version to v1.5.0 2025-07-09 18:46:10 +02:00
jaseg
1f5a9261e1 cli: Fix transform cli 2025-07-09 18:45:16 +02:00
jaseg
3ae6b9be61 kicad: fix for newer format versions 2025-04-02 17:49:13 +02:00
jaseg
a6434f16cd Fix clobbered object attributes 2025-04-01 15:52:53 +02:00
jaseg
63faf4280d Fix failing tests 2024-12-21 18:54:00 +01:00
jaseg
7fb8215a1e Add recent fusion-360 files as test resource 2024-12-21 17:32:16 +01:00
jaseg
81ae51d4be Improve allegro inner layer matching 2024-11-08 12:26:02 +01:00
jaseg
8ffa7c1b76 Fix svg orientation and improve OrCAD rendering 2024-11-08 11:41:11 +01:00
jaseg
be8371c7bc Improve allegro/orcad import 2024-11-06 14:49:50 +01:00
jaseg
1a854b1812 Tests: Make kicad docker image configurable 2024-11-06 14:49:36 +01:00
jaseg
091ee84910 kicad: add rotation method to circles and polygons 2024-07-19 19:28:15 +02:00
jaseg
fd63c44314 Kicad: add missing zone connection Atom 2024-07-19 19:23:20 +02:00
jaseg
cbe8dfa252 kicad: small syntax fix 2024-07-19 19:23:11 +02:00
jaseg
67ce4af957 kicad: Fix bug in footprint search 2024-07-19 19:22:56 +02:00
jaseg
eadd250ee3 kicad: Fix footprint rotation
Previously, footprint rotation was mirrored compared with everything
else in the kicad API
2024-07-19 19:22:26 +02:00
jaseg
5f5bbccd05 kicad: Add copy_placement method to footprints
Moving footprints around is ugly because of kicad's really weird way of
specifying sub-object coordinates and rotations. This commit adds a
helper method to deal with that.
2024-07-19 19:21:24 +02:00
jaseg
ea2664219d kicad: Make reprs more reasonable
This excludes parent back-links from reprs, which would previously blow
up the repr of many objects.
2024-07-19 19:20:51 +02:00
jaseg
be25b860a9 kicad: Improve net access
This adds net_name and net_index properties to a bunch of objects that
automatically look into the (net s-expr of the object
2024-07-19 19:20:09 +02:00
jaseg
e42b7462c9 kicad: Add nicer width access
This commit straightens up the mess a bit with kicad's two conflicting
"width" fields in different file format versions. The new variant is
preferred when saving, but the old variant is accepted as API
2024-07-19 19:18:55 +02:00
jaseg
754c9557e5 kicad: Make point lists more ergonomic
This commit removes the unnecessary "PointList" wrapper class, and just
puts regular python lists in its place.
2024-07-19 19:17:46 +02:00
jaseg
d7efa57732 kicad: Add bounding box support to lots of s-expr objects 2024-07-19 19:15:52 +02:00
jaseg
689ce748db kicad: Update empty PCB template for 8.99 2024-07-19 14:09:40 +02:00
jaseg
8e337c0506 kicad: Small fixes to 8.99 compatibility 2024-07-19 14:07:29 +02:00
jaseg
b23cafbfd4 tests: add status messages 2024-07-19 14:04:24 +02:00
jaseg
78ffb61aee kicad: Fixes for latest git version 2024-07-19 12:25:13 +02:00
jaseg
58d5784903 rs274x: Add support for SR step-repeat command 2024-07-18 16:39:58 +02:00
jaseg
1ecb7be6f9 Improve aperture macros doc 2024-07-18 16:39:38 +02:00
jaseg
ef9d61ffd5 Fix single contour example syntax errors 2024-07-18 14:19:31 +02:00
jaseg
79f555a465 macro parser: improve parameter number warning message 2024-07-18 14:19:17 +02:00
jaseg
501882ea3d Fix ast deprecation warnings 2024-07-18 14:19:04 +02:00
jaseg
e3a6716187 Fix symbol round trip tests 2024-07-18 13:50:01 +02:00
jaseg
8950a593f8 Fix bug in SVG bounding box calculation 2024-07-18 13:49:38 +02:00
jaseg
344825c5da kicad: Fix bug when searching pads with empty nets 2024-07-18 11:36:16 +02:00
jaseg
f1b3ab5e72 kicad: Improve compatibility with old symbol files 2024-07-18 11:36:13 +02:00
jaseg
2092b86431 tests: Always update kicad nightly container before run 2024-07-18 11:35:03 +02:00
jaseg
885ce36fd3 tests: Adjust nice and oom_adj values on linux
The testsuite calls resvg, and that sometimes can use up *a lot* of
memory. We adjust the test process tree's nice values and oom_adj values
to make sure the test processes get killed during an OOM condition
before anything else on the system.
2024-07-18 11:33:46 +02:00
jaseg
edaf246b9d Update KiCad PCB format support to 8.99 2024-07-17 10:20:35 +02:00
jaseg
b1d0260c70 Bump version to v1.4.0 2024-07-08 18:11:18 +02:00
jaseg
a2c6e2d64a Add missing package data 2024-07-08 18:10:57 +02:00
jaseg
1d5f4c8193 protoboard: Fix bug with empty points in breadboard rendering 2024-07-08 18:10:57 +02:00
jaseg
30878adfb1 Bump version to v1.3.0 2024-07-08 17:47:02 +02:00
jaseg
66811af966 kicad: Bring file format up to current 8.99 nightly 2024-07-08 16:29:57 +02:00
jaseg
ec85d6c169 tests: Speed up tests by a lot by bulk-caching kicad footprint renders 2024-07-08 16:29:35 +02:00
jaseg
f447b12571 tests: Fix compatibility with recent kicad nightly containers 2024-07-08 16:29:10 +02:00
jaseg
93fd764482 protoboard: Add CLI 2024-07-07 22:48:19 +02:00
jaseg
26e85279dc protoboard: improve permanent breadboard rail rendering 2024-07-07 21:57:29 +02:00
jaseg
1ed127e3b3 protoboard: improve permanent breadboard rendering 2024-07-07 21:55:17 +02:00
jaseg
1ee6b6587a protoboard: Add permanent breadboard rendering 2024-07-07 21:42:10 +02:00
jaseg
e98f3f3ace Add starburst pattern 2024-07-07 19:07:58 +02:00
jaseg
6f8d4bb999 protoboard: Finish up sides of Alio better 2024-07-07 17:08:06 +02:00
jaseg
4aab344a18 protoboard: add split front/back view in webthing 2024-07-07 16:52:24 +02:00
jaseg
21218239e4 protoboard: Fix alio and two-sided SMD rendering 2024-07-07 16:37:34 +02:00
jaseg
a1d6ebf79f protoboard: Improve layout distribution and index rendering 2024-07-07 16:25:01 +02:00
jaseg
b0274a93c0 protoboard: Improve layout packing 2024-07-07 16:08:32 +02:00
jaseg
0150c318bb protoboard: Improve row/column numbering 2024-07-07 15:48:36 +02:00
jaseg
6de138bf7c protoboard: reduce hole keepout margins 2024-07-07 15:16:01 +02:00
jaseg
224a666219 protoboard: improve border handling 2024-07-07 15:14:29 +02:00
jaseg
cee355ff57 protoboard: fix column label alignment 2024-07-07 14:40:27 +02:00
jaseg
4c3815b25a Fix THT flower proto area 2024-07-07 14:36:41 +02:00
jaseg
ef3b5d5e1c Spiky proto also works now 2024-07-07 00:52:38 +02:00
jaseg
552f30c15d Protoboard: All layouts except for spiky proto work 2024-07-07 00:32:17 +02:00
jaseg
f721692bf3 Protoboard generator WIP 2024-07-06 15:51:08 +02:00
jaseg
04c4b3ff0c kicad_sch render: Fix nightly import and wire rendering 2024-05-28 18:07:09 +02:00
jaseg
227d4ed1cd Bump version to v1.2.0 2023-11-14 21:56:03 +01:00
jaseg
ea4c28e307 Make new test files pass 2023-11-14 21:54:04 +01:00
jaseg
51ef4882a1 Fix failing tests 2023-11-14 21:54:04 +01:00
jaseg
df75a2fddb Small bugfix 2023-11-14 21:52:12 +01:00
jaseg
11325b213b Calculate out all aperture macros by default.
There are just too many severely buggy implementations around. Today I
ran into problems with both gerbv and with whatever JLC uses. You can
still export macros with raw expressions by setting a flag in the export
FileSettings.
2023-11-14 21:52:12 +01:00
jaseg
74fb384c4c aperture macros: work around gerbv/jlc wonkiness 2023-11-14 21:52:12 +01:00
jaseg
9af0713445 Remove debug print 2023-11-14 21:52:12 +01:00
jaseg
09e3731b74 aperture macros: Add expression simplification 2023-11-14 21:52:12 +01:00
jaseg
37b6b8f8d2 Aperture macro expression simplification WIP 2023-11-14 21:52:12 +01:00
jaseg
07362c592f Make sure we asterisk-terminate all G0x commands.
While this is common in the wild, not terminating them violates the
spec. It also breaks JLCPCB pretty badly. It seems their human review
process uses a Gerber viewer that like most can handle this, and won't
notice anything out of the ordinary, but then their photoplotter chokes
on this and literally stops plotting the file, discarding anything that
is after that line. This error is then apparently ignored and the
resulting broken boards shipped to the customer.
2023-11-14 21:52:12 +01:00
jaseg
2f5f7719c6 Split CLI into pretty svg and layer export sub-commands 2023-11-14 21:52:12 +01:00
jaseg
cb1d3eb3fb pretty svg export: Mirror board bottom side 2023-11-14 21:52:12 +01:00
jaseg
4ee5c51f22 Add JLCPCB KiCad Gerber X2/aperture macro test files 2023-11-14 21:52:12 +01:00
jaseg
53788354e8 Add JLCPCB/FAB-3000 example gerbers 2023-11-14 21:52:12 +01:00
jaseg
10962ae2f4 Add P-CAD 2006 example gerbers 2023-11-14 21:52:12 +01:00
jaseg
a19d307a7d Bump version to v1.1.0 2023-10-27 14:47:42 +02:00
jaseg
165e101dda ci: Disable tests for now, since upstream kicad-cli is broken. 2023-10-27 14:40:50 +02:00
jaseg
187c44555c Work around pip now requiring a new random feature switch to work
...for no good reason except to annoy anyone using it in a container.
2023-10-27 12:55:38 +02:00
jaseg
36da1fd68b Fix failing test cases 2023-10-26 23:53:23 +02:00
jaseg
9624e46147 Move coil stuff to separate repo 2023-10-26 00:48:52 +02:00
jaseg
a35125b123 Fix all failing tests that don't involve kicad-cli 2023-10-26 00:36:24 +02:00
jaseg
31af2b260c WIP 2023-10-20 18:24:45 +02:00
jaseg
dd49698df9 Update coil generator for the board house's latest fit 2023-10-20 12:44:48 +02:00
jaseg
ef2b53325c Test board fixes 2023-10-20 11:44:10 +02:00
jaseg
5d3cd4694d Improve mouse bite spacing for JLC 2023-10-18 17:19:05 +02:00
jaseg
3631871a40 Optimize coil model mesh precision 2023-10-18 17:15:20 +02:00
jaseg
b710462419 WIP 2023-10-18 10:59:28 +02:00
jaseg
313aa7dd26 Make coil test board more amenable to our favorite fab 2023-10-16 14:34:06 +02:00
jaseg
1ce02a9d25 Update sims & gen 2023-10-13 18:22:00 +02:00
jaseg
0ae13de322 Fix trace orientation 2023-10-12 20:50:40 +02:00
jaseg
c212663bb2 Update gen script 2023-10-12 20:47:55 +02:00
jaseg
2a9c91b025 Add coil test board gen 2023-10-12 20:44:52 +02:00
jaseg
2d4c40c0f7 Get self capacitance simulation to work 2023-10-11 17:06:42 +02:00
jaseg
a11f144c67 Give notebook a sensible name 2023-10-10 17:52:31 +02:00
jaseg
e78e939a13 Run more sims 2023-10-10 17:52:01 +02:00
jaseg
4ea1b26293 Coupling sims work 2023-10-09 19:56:42 +02:00
jaseg
cc59a6567e magnetic sim agreees with calculations now 2023-10-09 14:46:27 +02:00
jaseg
a2d1429036 Try to fix mesh export 2023-10-06 18:49:08 +02:00
jaseg
f793f12edb Try to fix mesh fragmentation 2023-10-06 16:45:21 +02:00
jaseg
78f5bf965f Mesh WIP 2023-10-06 16:17:25 +02:00
jaseg
84f7e5d25b Add missing simulation yamls 2023-10-06 14:42:35 +02:00
jaseg
ba689e632e EM solver WIP 2023-10-04 16:02:46 +02:00
jaseg
4400dc361e mag mesh gen looks good 2023-09-29 18:30:02 +02:00
jaseg
fdaf3c7e93 Add sim files 2023-09-29 10:37:40 +02:00
jaseg
6e12adb07e mesh gen works 2023-09-27 15:50:40 +02:00
jaseg
cff22b9e08 WIP 2023-09-26 22:42:57 +02:00
jaseg
f711c1d91c cli: Add kicad schematic svg rendering 2023-09-26 16:44:40 +02:00
jaseg
61e591b5b8 WiP 2023-09-22 18:54:11 +02:00
jaseg
95da482033 WIP 2023-09-22 18:54:11 +02:00
jaseg
d2143bdf4d Trace connectivity WIP 2023-09-22 13:30:11 +02:00
jaseg
5f1350d4f4 coil gen: add kicad pcb export 2023-09-20 14:24:15 +02:00
jaseg
5ff40e0ad1 twisted coil gen: add trace length calculation 2023-09-19 19:24:41 +02:00
jaseg
7d21dafd6a Two layer coil gen works with arcs! 2023-09-19 18:58:18 +02:00
jaseg
301601e81d Multilayer coil WIP 2023-09-19 12:44:22 +02:00
jaseg
3e47e7c2da Add line wonkifier 2023-07-22 18:29:04 +02:00
jaseg
ba4cafa3a4 Add tmtheme support 2023-07-22 14:25:18 +02:00
jaseg
2f0a21abf9 Made junctions smaller 2023-07-22 13:20:48 +02:00
jaseg
eb5c01ddd0 Subsheet rendering works too 2023-07-22 13:19:26 +02:00
jaseg
6e7337cca5 Rendering looks pretty good 2023-07-22 12:40:34 +02:00
jaseg
eb20595e00 WIP 2023-07-21 20:55:35 +02:00
jaseg
e4dcbe177f symbol pin rendering works 2023-07-21 20:27:14 +02:00
jaseg
8cb91dabdb WIP 2023-07-21 19:15:21 +02:00
jaseg
91b99a0452 WIP 2023-07-21 17:56:24 +02:00
jaseg
09c9d26728 WIP 2023-07-21 14:46:35 +02:00
jaseg
21ca5f7f5c WIP 2023-07-21 14:38:09 +02:00
jaseg
a39af853c8 Schematics WIP 2023-07-21 13:27:02 +02:00
jaseg
b69e9fded4 Kicad schematic rendering WIP 2023-07-21 01:46:11 +02:00
jaseg
60674ab5b3 Fix line rendering 2023-07-20 16:56:20 +02:00
jaseg
bdbdf7f586 Schematic rendering WIP 2023-07-20 16:42:19 +02:00
jaseg
a1b8cbf861 Make kicad eat schematics written by gerbonara 2023-07-18 21:54:26 +02:00
jaseg
58142cb0c7 kicad: Add schematic file format support 2023-07-18 21:15:08 +02:00
jaseg
08c4091e57 kicad: Improve API and fix kicad-nightly compat 2023-07-17 23:23:19 +02:00
jaseg
860fa4c53b LayerStack: Fix issue SVG rendering lazy-loaded stacks 2023-07-17 23:23:00 +02:00
jaseg
b2729a46ac Improve auto layout API 2023-07-07 20:19:36 +02:00
jaseg
572486aa25 kicad: Fix layers attribute handling and improve rotation API 2023-07-06 22:42:39 +02:00
jaseg
0799cc55ad kicad: Fix Footprint.property_value 2023-07-05 17:33:29 +02:00
jaseg
9f2f1f97f9 Improve coil gen, and fix some kicad s-expr issues 2023-07-05 17:25:28 +02:00
jaseg
cb188ac593 kicad: various pcb re-serialization fixes 2023-07-05 13:42:37 +02:00
jaseg
a5087636ab kicad: Fix additional dimension flags 2023-07-05 13:31:51 +02:00
jaseg
fb1d392831 kicad: Fix dimension.locked attr 2023-07-05 13:30:27 +02:00
jaseg
8f7b2893dc Fix failing symbol tests 2023-07-04 16:20:11 +02:00
jaseg
e696c09eac kicad: Small file format fixes 2023-07-04 12:55:34 +02:00
jaseg
3bc92876b4 Fix arc bounding box calculation for full circles 2023-07-03 23:00:00 +02:00
jaseg
34fae0a7c2 Fix handling of circles on outline layer 2023-07-03 22:59:47 +02:00
jaseg
2eb88e8127 Fix variable expression rendering 2023-07-03 17:50:15 +02:00
jaseg
95728fb33c Fix GerberX2 attribtue handling 2023-07-03 17:49:52 +02:00
jaseg
0920af4149 Fix test when ran with pytest-xdist 2023-07-01 10:36:57 +02:00
jaseg
7a95a0dde7 Remove obsolete workaround for breakage in kicad-cli 2023-06-29 22:55:01 +02:00
jaseg
88642a1803 Fix some failing unit tests 2023-06-29 22:45:47 +02:00
jaseg
56d55fda5d kicad: Extend query API 2023-06-29 19:47:31 +02:00
jaseg
467e482bf4 Fix pcb stackup handling 2023-06-21 13:37:38 +02:00
jaseg
dd8507d202 Polish twisted coil generator 2023-06-20 14:40:55 +02:00
jaseg
d43308c5cc Add twisted coil gen 2023-06-20 14:07:03 +02:00
jaseg
c29802c2b9 Add missing file for initial PCB support 2023-06-14 17:29:43 +02:00
jaseg
a92177904e Coil gen done 2023-06-14 17:29:26 +02:00
jaseg
0148db6249 Beautiful spirals! 2023-06-14 12:05:33 +02:00
jaseg
5178eba26f WIP 2023-06-14 12:01:51 +02:00
jaseg
ddd0641649 basic coils working 2023-06-14 11:39:44 +02:00
jaseg
e349c2c08b WIP 2023-06-14 10:47:49 +02:00
jaseg
96fde32c0b Coils WIP 2023-06-13 19:26:11 +02:00
jaseg
86276490eb coil gen WIP 2023-06-13 18:53:03 +02:00
jaseg
070494a1c3 stroke WIP 2023-06-12 19:43:13 +02:00
jaseg
1d0ba4da70 coil gen WIP 2023-06-12 19:43:00 +02:00
jaseg
3561817903 Add basic KiCad PCB file format support 2023-06-12 18:39:33 +02:00
jaseg
03f2ec0a30 rework WIP 2023-05-08 23:22:55 +02:00
jaseg
732c58f70b Extend breakout generator 2023-05-07 12:59:10 +02:00
jaseg
4bac77d0b4 Add pad ring generator 2023-05-07 12:59:10 +02:00
585 changed files with 248518 additions and 3883 deletions

2
.gitignore vendored
View file

@ -3,3 +3,5 @@ gerbonara_test_failures
__pycache__ __pycache__
.tox .tox
docs/_build/ docs/_build/
build
dist

View file

@ -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

View file

@ -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

Binary file not shown.

After

Width:  |  Height:  |  Size: 243 KiB

64
docs/examples.rst Normal file
View 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')

View file

@ -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
====================== ======================

View file

@ -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

View file

@ -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))

View file

@ -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=' ')

View file

@ -1,9 +0,0 @@
#!/usr/bin/env python3
import click
from .cli import cli
if __name__ == '__main__':
cli()

View file

@ -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)

View file

@ -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>')

View file

@ -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 []

View file

@ -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)

View file

@ -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

View file

@ -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")

View file

@ -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)

View file

@ -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)

View file

@ -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)))

View file

@ -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))

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

View file

@ -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*

View file

@ -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
View 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"

View file

@ -1,3 +0,0 @@
[pytest]
testpaths = gerbonara/tests
norecursedirs=*

View file

@ -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',
)

View file

@ -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
View 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()

View 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)

View file

@ -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
if isinstance(l, ConstantExpression) and isinstance(r, ConstantExpression):
return ConstantExpression(self.op(float(l), float(r))) return ConstantExpression(self.op(float(l), float(r)))
# Minimize operations with neutral elements and zeros
# 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
case _: # default
return OperatorExpression(self.op, l, r) return OperatorExpression(self.op, l, r)
def to_gerber(self, unit=None): return expr(rv).optimized(variable_binding)
lval = self.l.to_gerber(unit)
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()

View 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)

View file

@ -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

View file

@ -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)

View 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()

View file

@ -0,0 +1 @@

View 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>')

View file

@ -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)
if dasher.width > 0:
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)
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 objects(self, text=False, pads=True): def face(self, direction, pad=None, net=None):
if not net and not pad:
pad = '1'
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

View 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

View 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')

View 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

View 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))))

View 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'

View file

@ -64,7 +64,7 @@ term_regex = r"""(?mx)
(\))| (\))|
([+-]?\d+\.\d+(?=[\s\)]))| ([+-]?\d+\.\d+(?=[\s\)]))|
(\-?\d+(?=[\s\)]))| (\-?\d+(?=[\s\)]))|
([^0-9"\s()][^"\s)]*) ([^"\s()][^"\s)]*)
)""" )"""

View 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

View 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")

View 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()))

View file

@ -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.aperture_inner is UNDEFINED:
object.__setattr__(self, 'aperture_inner', self.pad_top.aperture)
if (self.pad_top.side, self.pad_bottom.side) != ('top', 'bottom'): if self.pad_top.flip:
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}".') raise ValueError('top pad cannot be flipped')
def render(self, layer_stack, cache=None): @property
x, y, rotation = self.abs_pos def plating(self):
self.pad_top.parent = self return 'plated' if self.plated else 'nonplated'
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: @property
(x_min, y_min), (x_max, y_max) = self.pad_top.bounding_box(MM) def apertures(self):
w_top = x_max - x_min yield from self.pad_top.apertures
h_top = y_max - y_min yield from self.pad_bottom.apertures
if self.pad_bottom: if self.aperture_inner is not None:
(x_min, y_min), (x_max, y_max) = self.pad_bottom.bounding_box(MM) yield PadStackAperture(self.aperture_inner, 'inner', 'copper')
w_bottom = x_max - x_min yield PadStackAperture(ExcellonTool(self.drill_dia, plated=self.plated, unit=self.unit), 'drill', self.plating)
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 @property
def single_sided(self): def single_sided(self):
return False return False
@classmethod @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): def rect(kls, drill_dia, w, h, rotation=0, mask_expansion=0.0, paste_expansion=0.0, paste=True, plated=True, unit=MM):
if h is None: pad = SMDStack.rect(w, h, rotation, mask_expansion, paste_expansion, paste, unit=unit)
h = w return kls(drill_dia, pad, plated=plated)
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 @classmethod
def circle(kls, x, y, hole_dia, dia, mask_expansion=0.0, paste_expansion=0.0, paste=True, plated=True, unit=MM): def circle(kls, drill_dia, dia, rotation=0, 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) pad = SMDStack.circle(dia, mask_expansion, paste_expansion, paste, unit=unit)
return kls(x, y, hole_dia, pad, plated=plated, unit=unit) return kls(drill_dia, pad, plated=plated)
@classmethod @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): def obround(kls, drill_dia, w, h, rotation=0, mask_expansion=0.0, paste_expansion=0.0, paste=True, plated=True, unit=MM):
ap_c = ObroundAperture(w, h, unit=unit) ap = ObroundAperture(w, h, unit=unit).rotated(rotation)
ap_m = ObroundAperture(w+2*mask_expansion, h+2*mask_expansion, unit=unit) pad = SMDStack(ap, mask_expansion, paste_expansion, paste, unit=unit)
ap_p = ObroundAperture(w, h, unit=unit) if paste else None return kls(drill_dia, pad, plated=plated)
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(frozen=True, slots=True)
@dataclass class ThroughViaStack(PadStack):
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 hole: float
dia: float = None
tented: bool = True
def render(self, layer_stack, cache=None): def __post_init__(self):
x, y, rotation = self.abs_pos if self.dia == None:
object.__setattr__(self, 'dia', self.hole*2)
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:
layer.objects.append(Flash(x, y, aperture, unit=self.unit))
layer_stack.drill_pth.objects.append(Flash(x, y, tool, unit=self.unit))
@property @property
def single_sided(self): def single_sided(self):
return False 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

File diff suppressed because it is too large Load diff

View 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()

View file

@ -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.';

View file

@ -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()

View file

@ -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:
svg = stack.to_pretty_svg(side='bottom' if side == 'bottom' else 'top', margin=margin,
arg_unit=(command_line_units or MM), arg_unit=(command_line_units or MM),
svg_unit=MM, force_bounds=force_bounds, inkscape=inkscape, colors=colorscheme))) 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,13 +355,23 @@ 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()
if number_format:
if output_format is 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 {}, stack.save_to_directory(outpath, naming_scheme=output_naming_scheme or {},
gerber_settings=output_format, gerber_settings=output_format,
excellon_settings=dataclasses.replace(output_format, zeros=None)) excellon_settings=dataclasses.replace(output_format, zeros=None))
@ -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')

View 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,7 +786,10 @@ 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):
self.settings._file_has_fixed_width_coordinates = True
if self.settings.number_format == (None, None):
# TARGET3001! exports zeros as "00" even when it uses an explicit decimal point everywhere else. # TARGET3001! exports zeros as "00" even when it uses an explicit decimal point everywhere else.
if x != '00': if x != '00':
raise SyntaxError('No number format set and value does not contain a decimal point. If this is an Allegro ' raise SyntaxError('No number format set and value does not contain a decimal point. If this is an Allegro '
@ -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,6 +932,9 @@ 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('.')
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) 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:
@ -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')

View file

@ -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):
if arc:
if points:
yield *points, arc
else:
yield self.outline[-1], self.outline[0], arc
return 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(p1)
points.append(p2) 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:
yield from gs.set_interpolation_mode(InterpMode.LINEAR)
x = gs.file_settings.write_gerber_value(point[0], self.unit) x = gs.file_settings.write_gerber_value(point[0], self.unit)
y = gs.file_settings.write_gerber_value(point[1], self.unit) y = gs.file_settings.write_gerber_value(point[1], self.unit)
if clockwise is None:
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:
# 1 - math.sqrt(1 - 0.5*math.sqrt(2))
max_error = min(max_error, r*0.4588038998538031)
elif max_error >= r:
return [Line(*self.p1, *self.p2, aperture=self.aperture, polarity_dark=self.polarity_dark, unit=self.unit)]
# see https://www.mathopenref.com/sagitta.html
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) return [Line(*p1, *p2, aperture=self.aperture, polarity_dark=self.polarity_dark, unit=self.unit)
for p1, p2 in zip(points[0::], points[1::]) ] for p1, p2 in pairwise(approximate_arc(
self.cx+self.x1, self.cy+self.y1,
self.x1, self.y1,
self.x2, self.y2,
self.clockwise,
max_error=max_error,
clip_max_error=clip_max_error))]
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)

View file

@ -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)

View file

@ -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',

View file

@ -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,18 +113,26 @@ class NamingScheme:
} }
def apply_rules(filenames, rules):
def _match_files(filenames): certain = False
matches = {}
for generator, rules in MATCH_RULES.items():
already_matched = set()
gen = {} gen = {}
matches[generator] = gen already_matched = set()
header_regex = rules.pop('header regex', [])
header_regex_matched = [False] * len(header_regex)
file_headers = {}
def get_header(path):
if path not in file_headers:
with open(path) as f:
file_headers[path] = f.read(16384)
return file_headers[path]
for layer, regex in rules.items(): for layer, regex in rules.items():
for fn in filenames: for fn in filenames:
if fn in already_matched: if fn in already_matched:
continue continue
target = None
if (m := re.fullmatch(regex, fn.name, re.IGNORECASE)): if (m := re.fullmatch(regex, fn.name, re.IGNORECASE)):
if layer == 'inner copper': if layer == 'inner copper':
target = 'inner_' + ''.join(e or '' for e in m.groups()) + ' copper' target = 'inner_' + ''.join(e or '' for e in m.groups()) + ' copper'
@ -132,11 +141,33 @@ def _match_files(filenames):
gen[target] = gen.get(target, []) + [fn] gen[target] = gen.get(target, []) + [fn]
already_matched.add(fn) already_matched.add(fn)
return matches
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']
else:
# Ignore if we can't find the param file -- maybe the user has convinced Allegro to actually put this # 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. # 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')) **stroke_attrs, id=f'l-drill-npth', transform=layer_transform))
if (fg := colors.get('drill unknown')):
for i, layer in enumerate(self._drill_layers): for i, layer in enumerate(self._drill_layers):
fg = color_map[('drill', 'unknown')]
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
View 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)

View file

@ -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,11 +287,22 @@ 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%'
aperture_map = {ap: num for num, ap in enumerate(self.apertures(), start=10)}
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(): for macro in self.aperture_macros():
yield am_stmt(macro) yield am_stmt(macro)
aperture_map = {ap: num for num, ap in enumerate(self.apertures(), start=10)}
for aperture, number in aperture_map.items(): for aperture, number in aperture_map.items():
yield f'%ADD{number}{aperture.to_gerber(settings)}*%' yield f'%ADD{number}{aperture.to_gerber(settings)}*%'
@ -360,7 +375,7 @@ 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:
@ -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,6 +793,9 @@ 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:
if self.step_repeat_objects:
self.step_repeat_objects.append(obj)
else:
self.target.objects.append(obj) 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)
@ -784,6 +807,9 @@ 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.
if self.step_repeat_objects:
self.step_repeat_objects.append(self.current_region)
else:
self.target.objects.append(self.current_region) 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,
@ -791,7 +817,11 @@ class GerberParser:
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

View file

@ -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
View 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
View file

172
tests/conftest.py Normal file
View 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)))

View file

Before

Width:  |  Height:  |  Size: 6.1 KiB

After

Width:  |  Height:  |  Size: 6.1 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 46 KiB

After

Width:  |  Height:  |  Size: 46 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 6.3 KiB

After

Width:  |  Height:  |  Size: 6.3 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 1.3 KiB

After

Width:  |  Height:  |  Size: 1.3 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 5.8 KiB

After

Width:  |  Height:  |  Size: 5.8 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 3.4 KiB

After

Width:  |  Height:  |  Size: 3.4 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 4 KiB

After

Width:  |  Height:  |  Size: 4 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 1.7 KiB

After

Width:  |  Height:  |  Size: 1.7 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 70 KiB

After

Width:  |  Height:  |  Size: 70 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 11 KiB

After

Width:  |  Height:  |  Size: 11 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 9.8 KiB

After

Width:  |  Height:  |  Size: 9.8 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 70 KiB

After

Width:  |  Height:  |  Size: 70 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 94 KiB

After

Width:  |  Height:  |  Size: 94 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 32 KiB

After

Width:  |  Height:  |  Size: 32 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 32 KiB

After

Width:  |  Height:  |  Size: 32 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 31 KiB

After

Width:  |  Height:  |  Size: 31 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 556 B

After

Width:  |  Height:  |  Size: 556 B

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 2.2 KiB

After

Width:  |  Height:  |  Size: 2.2 KiB

Before After
Before After

View file

Before

Width:  |  Height:  |  Size: 18 KiB

After

Width:  |  Height:  |  Size: 18 KiB

Before After
Before After

Some files were not shown because too many files have changed in this diff Show more