summaryrefslogtreecommitdiff
path: root/gerber/render/rs274x_backend.py
diff options
context:
space:
mode:
Diffstat (limited to 'gerber/render/rs274x_backend.py')
-rw-r--r--gerber/render/rs274x_backend.py233
1 files changed, 116 insertions, 117 deletions
diff --git a/gerber/render/rs274x_backend.py b/gerber/render/rs274x_backend.py
index b4b4612..13e871c 100644
--- a/gerber/render/rs274x_backend.py
+++ b/gerber/render/rs274x_backend.py
@@ -6,7 +6,7 @@ try:
from cStringIO import StringIO
except(ImportError):
from io import StringIO
-
+
from .render import GerberContext
from ..am_statements import *
from ..gerber_statements import *
@@ -15,27 +15,27 @@ from ..primitives import AMGroup, Arc, Circle, Line, Obround, Outline, Polygon,
class AMGroupContext(object):
'''A special renderer to generate aperature macros from an AMGroup'''
-
+
def __init__(self):
self.statements = []
-
+
def render(self, amgroup, name):
-
+
if amgroup.stmt:
# We know the statement it was generated from, so use that to create the AMParamStmt
# It will give a much better result
-
+
stmt = deepcopy(amgroup.stmt)
stmt.name = name
-
+
return stmt
-
+
else:
# Clone ourselves, then offset by the psotion so that
# our render doesn't have to consider offset. Just makes things simpler
nooffset_group = deepcopy(amgroup)
nooffset_group.position = (0, 0)
-
+
# Now draw the shapes
for primitive in nooffset_group.primitives:
if isinstance(primitive, Outline):
@@ -50,46 +50,46 @@ class AMGroupContext(object):
self._render_polygon(primitive)
else:
raise ValueError('amgroup')
-
+
statement = AMParamStmt('AM', name, self._statements_to_string())
return statement
-
+
def _statements_to_string(self):
macro = ''
for statement in self.statements:
macro += statement.to_gerber()
-
+
return macro
-
+
def _render_circle(self, circle):
self.statements.append(AMCirclePrimitive.from_primitive(circle))
-
+
def _render_rectangle(self, rectangle):
self.statements.append(AMCenterLinePrimitive.from_primitive(rectangle))
-
+
def _render_line(self, line):
self.statements.append(AMVectorLinePrimitive.from_primitive(line))
-
+
def _render_outline(self, outline):
self.statements.append(AMOutlinePrimitive.from_primitive(outline))
-
+
def _render_polygon(self, polygon):
self.statements.append(AMPolygonPrimitive.from_primitive(polygon))
-
+
def _render_thermal(self, thermal):
pass
-
+
class Rs274xContext(GerberContext):
-
+
def __init__(self, settings):
GerberContext.__init__(self)
self.comments = []
self.header = []
self.body = []
self.end = [EofStmt()]
-
+
# Current values so we know if we have to execute
# moves, levey changes before anything else
self._level_polarity = None
@@ -97,65 +97,65 @@ class Rs274xContext(GerberContext):
self._func = None
self._quadrant_mode = None
self._dcode = None
-
+
# Primarily for testing and comarison to files, should we write
# flashes as a single statement or a move plus flash? Set to true
# to do in a single statement. Normally this can be false
self.condensed_flash = True
-
+
# When closing a region, force a D02 staement to close a region.
# This is normally not necessary because regions are closed with a G37
# staement, but this will add an extra statement for doubly close
# the region
self.explicit_region_move_end = False
-
+
self._next_dcode = 10
self._rects = {}
self._circles = {}
self._obrounds = {}
self._polygons = {}
self._macros = {}
-
+
self._i_none = 0
self._j_none = 0
-
+
self.settings = settings
self._start_header(settings)
-
+
def _start_header(self, settings):
self.header.append(FSParamStmt.from_settings(settings))
self.header.append(MOParamStmt.from_units(settings.units))
-
+
def _simplify_point(self, point):
return (point[0] if point[0] != self._pos[0] else None, point[1] if point[1] != self._pos[1] else None)
-
+
def _simplify_offset(self, point, offset):
-
+
if point[0] != offset[0]:
xoffset = point[0] - offset[0]
else:
xoffset = self._i_none
-
+
if point[1] != offset[1]:
yoffset = point[1] - offset[1]
else:
yoffset = self._j_none
-
+
return (xoffset, yoffset)
-
+
@property
def statements(self):
return self.comments + self.header + self.body + self.end
-
+
def set_bounds(self, bounds):
pass
-
+
def _paint_background(self):
pass
-
+
def _select_aperture(self, aperture):
-
+
# Select the right aperture if not already selected
if aperture:
if isinstance(aperture, Circle):
@@ -168,61 +168,61 @@ class Rs274xContext(GerberContext):
aper = self._get_amacro(aperture)
else:
raise NotImplementedError('Line with invalid aperture type')
-
+
if aper.d != self._dcode:
self.body.append(ApertureStmt(aper.d))
self._dcode = aper.d
-
+
def _pre_render_primitive(self, primitive):
-
+
if hasattr(primitive, 'comment'):
self.body.append(CommentStmt(primitive.comment))
-
+
def _render_line(self, line, color):
-
+
self._select_aperture(line.aperture)
-
+
self._render_level_polarity(line)
-
+
# Get the right function
if self._func != CoordStmt.FUNC_LINEAR:
func = CoordStmt.FUNC_LINEAR
else:
func = None
self._func = CoordStmt.FUNC_LINEAR
-
+
if self._pos != line.start:
self.body.append(CoordStmt.move(func, self._simplify_point(line.start)))
self._pos = line.start
# We already set the function, so the next command doesn't require that
func = None
-
+
point = self._simplify_point(line.end)
-
+
# In some files, we see a lot of duplicated ponts, so omit those
if point[0] != None or point[1] != None:
self.body.append(CoordStmt.line(func, self._simplify_point(line.end)))
self._pos = line.end
elif func:
self.body.append(CoordStmt.mode(func))
-
+
def _render_arc(self, arc, color):
-
+
# Optionally set the quadrant mode if it has changed:
if arc.quadrant_mode != self._quadrant_mode:
-
+
if arc.quadrant_mode != 'multi-quadrant':
self.body.append(QuadrantModeStmt.single())
else:
self.body.append(QuadrantModeStmt.multi())
-
+
self._quadrant_mode = arc.quadrant_mode
-
+
# Select the right aperture if not already selected
self._select_aperture(arc.aperture)
-
+
self._render_level_polarity(arc)
-
+
# Find the right movement mode. Always set to be sure it is really right
dir = arc.direction
if dir == 'clockwise':
@@ -233,59 +233,59 @@ class Rs274xContext(GerberContext):
self._func = CoordStmt.FUNC_ARC_CCW
else:
raise ValueError('Invalid circular interpolation mode')
-
+
if self._pos != arc.start:
# TODO I'm not sure if this is right
self.body.append(CoordStmt.move(CoordStmt.FUNC_LINEAR, self._simplify_point(arc.start)))
self._pos = arc.start
-
+
center = self._simplify_offset(arc.center, arc.start)
end = self._simplify_point(arc.end)
self.body.append(CoordStmt.arc(func, end, center))
self._pos = arc.end
def _render_region(self, region, color):
-
+
self._render_level_polarity(region)
-
+
self.body.append(RegionModeStmt.on())
-
+
for p in region.primitives:
-
+
if isinstance(p, Line):
self._render_line(p, color)
else:
self._render_arc(p, color)
-
+
if self.explicit_region_move_end:
self.body.append(CoordStmt.move(None, None))
self.body.append(RegionModeStmt.off())
-
+
def _render_level_polarity(self, region):
if region.level_polarity != self._level_polarity:
self._level_polarity = region.level_polarity
self.body.append(LPParamStmt.from_region(region))
-
+
def _render_flash(self, primitive, aperture):
-
+
self._render_level_polarity(primitive)
-
+
if aperture.d != self._dcode:
self.body.append(ApertureStmt(aperture.d))
self._dcode = aperture.d
-
+
if self.condensed_flash:
self.body.append(CoordStmt.flash(self._simplify_point(primitive.position)))
else:
self.body.append(CoordStmt.move(None, self._simplify_point(primitive.position)))
self.body.append(CoordStmt.flash(None))
-
+
self._pos = primitive.position
-
+
def _get_circle(self, diameter, hole_diameter, dcode = None):
'''Define a circlar aperture'''
-
+
aper = self._circles.get((diameter, hole_diameter), None)
if not aper:
@@ -294,13 +294,13 @@ class Rs274xContext(GerberContext):
self._next_dcode += 1
else:
self._next_dcode = max(dcode + 1, self._next_dcode)
-
+
aper = ADParamStmt.circle(dcode, diameter, hole_diameter)
self._circles[(diameter, hole_diameter)] = aper
self.header.append(aper)
return aper
-
+
def _render_circle(self, circle, color):
aper = self._get_circle(circle.diameter, circle.hole_diameter)
@@ -308,122 +308,122 @@ class Rs274xContext(GerberContext):
def _get_rectangle(self, width, height, dcode = None):
'''Get a rectanglar aperture. If it isn't defined, create it'''
-
+
key = (width, height)
aper = self._rects.get(key, None)
-
+
if not aper:
if not dcode:
dcode = self._next_dcode
self._next_dcode += 1
else:
self._next_dcode = max(dcode + 1, self._next_dcode)
-
+
aper = ADParamStmt.rect(dcode, width, height)
self._rects[(width, height)] = aper
self.header.append(aper)
-
+
return aper
def _render_rectangle(self, rectangle, color):
-
+
aper = self._get_rectangle(rectangle.width, rectangle.height)
self._render_flash(rectangle, aper)
-
+
def _get_obround(self, width, height, dcode = None):
-
+
key = (width, height)
aper = self._obrounds.get(key, None)
-
+
if not aper:
if not dcode:
dcode = self._next_dcode
self._next_dcode += 1
else:
self._next_dcode = max(dcode + 1, self._next_dcode)
-
+
aper = ADParamStmt.obround(dcode, width, height)
self._obrounds[key] = aper
self.header.append(aper)
-
+
return aper
-
+
def _render_obround(self, obround, color):
-
+
aper = self._get_obround(obround.width, obround.height)
self._render_flash(obround, aper)
-
+
def _render_polygon(self, polygon, color):
-
+
aper = self._get_polygon(polygon.radius, polygon.sides, polygon.rotation, polygon.hole_radius)
self._render_flash(polygon, aper)
-
+
def _get_polygon(self, radius, num_vertices, rotation, hole_radius, dcode = None):
-
+
key = (radius, num_vertices, rotation, hole_radius)
aper = self._polygons.get(key, None)
-
+
if not aper:
if not dcode:
dcode = self._next_dcode
self._next_dcode += 1
else:
self._next_dcode = max(dcode + 1, self._next_dcode)
-
+
aper = ADParamStmt.polygon(dcode, radius * 2, num_vertices, rotation, hole_radius * 2)
self._polygons[key] = aper
self.header.append(aper)
-
+
return aper
-
+
def _render_drill(self, drill, color):
raise ValueError('Drills are not valid in RS274X files')
-
+
def _hash_amacro(self, amgroup):
'''Calculate a very quick hash code for deciding if we should even check AM groups for comparision'''
-
+
# We always start with an X because this forms part of the name
# Basically, in some cases, the name might start with a C, R, etc. That can appear
# to conflict with normal aperture definitions. Technically, it shouldn't because normal
# aperture definitions should have a comma, but in some cases the commit is omitted
hash = 'X'
for primitive in amgroup.primitives:
-
+
hash += primitive.__class__.__name__[0]
-
+
bbox = primitive.bounding_box
hash += str((bbox[0][1] - bbox[0][0]) * 100000)[0:2]
hash += str((bbox[1][1] - bbox[1][0]) * 100000)[0:2]
-
+
if hasattr(primitive, 'primitives'):
hash += str(len(primitive.primitives))
-
+
if isinstance(primitive, Rectangle):
hash += str(primitive.width * 1000000)[0:2]
hash += str(primitive.height * 1000000)[0:2]
elif isinstance(primitive, Circle):
hash += str(primitive.diameter * 1000000)[0:2]
-
+
if len(hash) > 20:
# The hash might actually get quite complex, so stop before
# it gets too long
break
-
+
return hash
def _get_amacro(self, amgroup, dcode = None):
# Macros are a little special since we don't have a good way to compare them quickly
# but in most cases, this should work
-
+
hash = self._hash_amacro(amgroup)
macro = None
macroinfo = self._macros.get(hash, None)
-
+
if macroinfo:
-
+
# We have a definition, but check that the groups actually are the same
for macro in macroinfo:
-
+
# Macros should have positions, right? But if the macro is selected for non-flashes
# then it won't have a position. This is of course a bad gerber, but they do exist
if amgroup.position:
@@ -435,7 +435,7 @@ class Rs274xContext(GerberContext):
if amgroup.equivalent(macro[1], offset):
break
macro = None
-
+
# Did we find one in the group0
if not macro:
# This is a new macro, so define it
@@ -444,52 +444,51 @@ class Rs274xContext(GerberContext):
self._next_dcode += 1
else:
self._next_dcode = max(dcode + 1, self._next_dcode)
-
+
# Create the statements
# TODO
amrenderer = AMGroupContext()
statement = amrenderer.render(amgroup, hash)
-
+
self.header.append(statement)
-
+
aperdef = ADParamStmt.macro(dcode, hash)
self.header.append(aperdef)
-
+
# Store the dcode and the original so we can check if it really is the same
# If it didn't have a postition, set it to 0, 0
if amgroup.position == None:
amgroup.position = (0, 0)
macro = (aperdef, amgroup)
-
+
if macroinfo:
macroinfo.append(macro)
else:
- self._macros[hash] = [macro]
-
+ self._macros[hash] = [macro]
+
return macro[0]
-
+
def _render_amgroup(self, amgroup, color):
-
+
aper = self._get_amacro(amgroup)
self._render_flash(amgroup, aper)
def _render_inverted_layer(self):
pass
-
+
def _new_render_layer(self):
# TODO Might need to implement this
pass
-
+
def _flatten(self):
# TODO Might need to implement this
pass
-
+
def dump(self):
"""Write the rendered file to a StringIO steam"""
statements = map(lambda stmt: stmt.to_gerber(self.settings), self.statements)
stream = StringIO()
for statement in statements:
stream.write(statement + '\n')
-
+
return stream
- \ No newline at end of file