summaryrefslogtreecommitdiff
path: root/gerbonara/gerber/rs274x.py
diff options
context:
space:
mode:
Diffstat (limited to 'gerbonara/gerber/rs274x.py')
-rw-r--r--gerbonara/gerber/rs274x.py52
1 files changed, 50 insertions, 2 deletions
diff --git a/gerbonara/gerber/rs274x.py b/gerbonara/gerber/rs274x.py
index f2473b9..1b62cc4 100644
--- a/gerbonara/gerber/rs274x.py
+++ b/gerbonara/gerber/rs274x.py
@@ -51,6 +51,49 @@ class GerberFile(CamFile):
self.comments = []
self.objects = []
+ def merge(self, other):
+ """ Merge other GerberFile into this one """
+ # FIXME unit handling
+
+ self.comments += other.comments
+
+ # dedup apertures
+ new_apertures = {}
+ replace_apertures = {}
+ for ap in self.apertures + other.apertures:
+ gbr = ap.to_gerber()
+ if gbr not in new_apertures:
+ new_apertures[gbr] = ap
+ else:
+ replace_apertures[id(ap)] = new_apertures[gbr]
+ self.apertures = new_apertures
+
+ self.objects += other.objects
+ for obj in self.objects:
+ if (ap := replace_apertures.get(id(getattr(obj, 'aperture')))):
+ obj.aperture = ap
+
+ # dedup aperture macros
+ macros = { m.to_gerber(): m
+ for m in [ GenericMacros.circle, GenericMacros.rect, GenericMacros.oblong, GenericMacros.polygon] }
+ for ap in new_apertures:
+ if isinstance(aperture, ApertureMacroInstance):
+ macro_grb = ap.macro.to_gerber() # use native units to compare macros
+ if macro_grb in macros:
+ ap.macro = macros[macro_grb]
+ else:
+ macros[macro_grb] = ap.macro
+
+ # make macro names unique
+ seen_macro_names = set()
+ for macro in macros.values():
+ i = 2
+ while (new_name := f'{macro.name}{i}') in seen_macro_names:
+ i += 1
+ macro.name = new_name
+ seen_macro_names.add(new_name)
+
+
@classmethod
def open(kls, filename, enable_includes=False, enable_include_dir=None):
with open(filename, "r") as f:
@@ -91,7 +134,9 @@ class GerberFile(CamFile):
for cmt in self.comments:
yield CommentStmt(cmt)
- # Emit gerbonara's generic, rotation-capable aperture macro replacements for the standard C/R/O/P shapes.
+ # Always emit gerbonara's generic, rotation-capable aperture macro replacements for the standard C/R/O/P shapes.
+ # Unconditionally emitting these here is easier than first trying to figure out if we need them later,
+ # and they are only a few bytes anyway.
yield ApertureMacroStmt(GenericMacros.circle)
yield ApertureMacroStmt(GenericMacros.rect)
yield ApertureMacroStmt(GenericMacros.oblong)
@@ -117,9 +162,12 @@ class GerberFile(CamFile):
yield EofStmt()
- def __str__(self):
+ def to_gerber(self):
return '\n'.join(self.generate_statements())
+ def __str__(self):
+ return f'<GerberFile with {len(self.apertures)} apertures, {len(self.objects)} objects>'
+
def save(self, filename):
with open(filename, 'w', encoding='utf-8') as f: # Encoding is specified as UTF-8 by spec.
for stmt in self.generate_statements():