summaryrefslogtreecommitdiff
path: root/gerber/gerber.py
diff options
context:
space:
mode:
authorPaulo Henrique Silva <ph.silva@gmail.com>2013-12-20 02:56:42 -0200
committerPaulo Henrique Silva <ph.silva@gmail.com>2013-12-20 02:56:42 -0200
commitfbf5943792b19d4d051bac44c5f7953618ed1754 (patch)
tree142e691e6ba4f1eb117708e427e7bed4aa220df7 /gerber/gerber.py
parenta3f7c778c069a3cffaf86c7e6d78a4eda09bbae9 (diff)
downloadgerbonara-fbf5943792b19d4d051bac44c5f7953618ed1754.tar.gz
gerbonara-fbf5943792b19d4d051bac44c5f7953618ed1754.tar.bz2
gerbonara-fbf5943792b19d4d051bac44c5f7953618ed1754.zip
Moving SVG code to use svgwrite.
Still ugly output and not all apertures handled, specially rounded and macro based, but keeps improving.
Diffstat (limited to 'gerber/gerber.py')
-rw-r--r--gerber/gerber.py170
1 files changed, 113 insertions, 57 deletions
diff --git a/gerber/gerber.py b/gerber/gerber.py
index 11e9574..8480f24 100644
--- a/gerber/gerber.py
+++ b/gerber/gerber.py
@@ -3,6 +3,9 @@
import re
import json
+import copy
+
+import svgwrite
def red(s):
@@ -64,10 +67,10 @@ class LPParamStmt(ParamStmt):
class ADParamStmt(ParamStmt):
- def __init__(self, param, d, aperture, modifiers):
+ def __init__(self, param, d, shape, modifiers):
ParamStmt.__init__(self, param)
self.d = d
- self.aperture = aperture
+ self.shape = shape
self.modifiers = [[x for x in m.split("X")] for m in modifiers.split(",")]
@@ -136,9 +139,11 @@ NOTATION_INCREMENTAL = 2
UNIT_INCH = 1
UNIT_MM = 2
+INTERPOLATION_LINEAR = 1
+INTERPOLATION_ARC = 2
-class GerberCoordFormat(object):
+class GerberCoordFormat(object):
def __init__(self, zeroes, x, y):
self.omit_leading_zeroes = True if zeroes == "L" else False
self.omit_trailing_zeroes = True if zeroes == "T" else False
@@ -153,7 +158,7 @@ class GerberCoordFormat(object):
missing_zeroes = (self.x_int_digits + self.x_dec_digits) - len(new_x)
if missing_zeroes and self.omit_leading_zeroes:
- new_x = (missing_zeroes*"0") + new_x
+ new_x = (missing_zeroes * "0") + new_x
elif missing_zeroes and self.omit_trailing_zeroes:
new_x += missing_zeroes * "0"
@@ -163,7 +168,7 @@ class GerberCoordFormat(object):
missing_zeroes = (self.y_int_digits + self.y_dec_digits) - len(new_y)
if missing_zeroes and self.omit_leading_zeroes:
- new_y = (missing_zeroes*"0") + new_y
+ new_y = (missing_zeroes * "0") + new_y
elif missing_zeroes and self.omit_trailing_zeroes:
new_y += missing_zeroes * "0"
@@ -180,8 +185,8 @@ class GerberContext(object):
x = 0
y = 0
- current_aperture = 0
- interpolation = None
+ aperture = 0
+ interpolation = INTERPOLATION_LINEAR
image_polarity = IMAGE_POLARITY_POSITIVE
level_polarity = LEVEL_POLARITY_DARK
@@ -190,89 +195,136 @@ class GerberContext(object):
pass
def set_coord_format(self, zeroes, x, y):
- # print "<coord-format/>"
self.coord_format = GerberCoordFormat(zeroes, x, y)
def set_coord_notation(self, notation):
- # print "<coord-notation/>"
self.coord_notation = NOTATION_ABSOLUTE if notation == "A" else NOTATION_INCREMENTAL
def set_coord_unit(self, unit):
- # print "<coord-unit/>"
self.coord_unit = UNIT_INCH if unit == "IN" else UNIT_MM
def set_image_polarity(self, polarity):
- # print "<image-polarity/>"
self.image_polarity = IMAGE_POLARITY_POSITIVE if polarity == "POS" else IMAGE_POLARITY_NEGATIVE
def set_level_polarity(self, polarity):
- # print "<level-polarity/>"
self.level_polarity = LEVEL_POLARITY_DARK if polarity == "D" else LEVEL_POLARITY_CLEAR
+ def set_interpolation(self, interpolation):
+ self.interpolation = INTERPOLATION_LINEAR if interpolation in ("G01", "G1") else INTERPOLATION_ARC
+
def set_aperture(self, d):
- # print "<aperture %s/>" % d
- self.current_aperture = d
+ self.aperture = d
def resolve(self, x, y):
x, y = self.coord_format.resolve(x, y)
return x or self.x, y or self.y
- def move(self, x, y):
- # print "<x=%s y=%s/>" % (x, y)
- self.x, self.y = self.resolve(x, y)
+ def define_aperture(self, d, shape, modifiers):
+ pass
+
+ def move(self, x, y, resolve=True):
+ if resolve:
+ self.x, self.y = self.resolve(x, y)
+ else:
+ self.x, self.y = x, y
+
+ def stroke(self, x, y):
+ pass
- def line(self):
- # print "<line/>"
+ def line(self, x, y):
pass
- def arc(self):
- # print "<arc/>"
+ def arc(self, x, y):
pass
- def flash(self):
- # print "<flash/>"
+ def flash(self, x, y):
pass
+class Shape(object):
+ pass
+
+
+class Circle(Shape):
+ def __init__(self, diameter=0):
+ self.diameter = diameter
+
+ def draw(self, ctx, x, y):
+ return ctx.dwg.line(start=(ctx.x*300, ctx.y*300), end=(x*300, y*300), stroke="rgb(184, 115, 51)", stroke_width=2,
+ stroke_linecap="round")
+
+ def flash(self, ctx, x, y):
+ return ctx.dwg.circle(center=(x*300, y*300), r=300*(self.diameter/2.0), fill="rgb(184, 115, 51)")
+
+
+class Rect(Shape):
+ def __init__(self, size=0):
+ self.size = size
+
+ def draw(self, ctx, x, y):
+ return ctx.dwg.line(start=(ctx.x*300, ctx.y*300), end=(x*300, y*300), stroke="rgb(184, 115, 51)", stroke_width=2,
+ stroke_linecap="butt")
+
+ def flash(self, ctx, x, y):
+ return ctx.dwg.rect(insert=(300*x, 300*y), size=(300*float(self.size[0]), 300*float(self.size[1])), fill="rgb(184, 115, 51)")
+
+
class SvgContext(GerberContext):
def __init__(self):
GerberContext.__init__(self)
- self.header = "<svg width=\"1280\" height=\"720\">"
- self.footer = "</svg>"
+ self.apertures = {}
+ self.dwg = svgwrite.Drawing()
+ self.dwg.add(self.dwg.rect(insert=(0, 0), size=(2000, 2000), fill="black"))
+
+ def define_aperture(self, d, shape, modifiers):
+ aperture = None
+ if shape == "C":
+ aperture = Circle(diameter=float(modifiers[0][0]))
+ elif shape == "R":
+ aperture = Rect(size=modifiers[0][0:2])
+
+ self.apertures[d] = aperture
+
+ def stroke(self, x, y):
+ super(SvgContext, self).stroke(x, y)
+
+ if self.interpolation == INTERPOLATION_LINEAR:
+ self.line(x, y)
+ elif self.interpolation == INTERPOLATION_ARC:
+ self.arc(x, y)
+
+ def line(self, x, y):
+ super(SvgContext, self).line(x, y)
- self.lines = []
+ x, y = self.resolve(x, y)
- self.path = []
+ ap = self.apertures.get(str(self.aperture), None)
+ if ap is None:
+ return
+
+ self.dwg.add(ap.draw(self, x, y))
- def move(self, x, y):
- super(SvgContext, self).move(x, y)
- self.path.append((self.x, self.y))
+ self.move(x, y, resolve=False)
- def line(self):
- super(SvgContext, self).line()
+ def arc(self, x, y):
+ super(SvgContext, self).arc(x, y)
- for i, path in enumerate(self.path):
- if i > 0:
- x1 = self.path[i-1]
- x2 = self.path[i]
- self.lines.append('<line x1="{0}" y1="{1}" x2="{2}" y2="{3}" stroke="rgb(0,145,21)" stroke-width="2"/>'.format(x1[0]*300, x1[1]*300, x2[0]*300, x2[1]*300))
+ def flash(self, x, y):
+ super(SvgContext, self).flash(x, y)
- self.path = []
+ x, y = self.resolve(x, y)
+
+ ap = self.apertures.get(str(self.aperture), None)
+ if ap is None:
+ return
- def arc(self):
- super(SvgContext, self).arc()
- self.path = []
+ self.dwg.add(ap.flash(self, x, y))
- def flash(self):
- super(SvgContext, self).flash()
- self.path = []
+ self.move(x, y, resolve=False)
def dump(self):
- print self.header
- for line in self.lines:
- print line
- print self.footer
+ self.dwg.saveas("teste.svg")
class Gerber(object):
@@ -288,11 +340,11 @@ class Gerber(object):
MO = r"(?P<param>MO)(?P<mo>(MM|IN))"
IP = r"(?P<param>IP)(?P<ip>(POS|NEG))"
LP = r"(?P<param>LP)(?P<lp>(D|C))"
- AD_CIRCLE = r"(?P<param>AD)D(?P<d>\d+)(?P<aperture>C)[,](?P<modifiers>[^,]*)"
- AD_RECT = r"(?P<param>AD)D(?P<d>\d+)(?P<aperture>R)[,](?P<modifiers>[^,]*)"
- AD_OBROUND = r"(?P<param>AD)D(?P<d>\d+)(?P<aperture>O)[,](?P<modifiers>[^,]*)"
- AD_POLY = r"(?P<param>AD)D(?P<d>\d+)(?P<aperture>P)[,](?P<modifiers>[^,]*)"
- AD_MACRO = r"(?P<param>AD)D(?P<d>\d+)+(?P<aperture>{name})[,](?P<modifiers>[^,]*)".format(name=NAME)
+ AD_CIRCLE = r"(?P<param>AD)D(?P<d>\d+)(?P<shape>C)[,](?P<modifiers>[^,]*)"
+ AD_RECT = r"(?P<param>AD)D(?P<d>\d+)(?P<shape>R)[,](?P<modifiers>[^,]*)"
+ AD_OBROUND = r"(?P<param>AD)D(?P<d>\d+)(?P<shape>O)[,](?P<modifiers>[^,]*)"
+ AD_POLY = r"(?P<param>AD)D(?P<d>\d+)(?P<shape>P)[,](?P<modifiers>[^,]*)"
+ AD_MACRO = r"(?P<param>AD)D(?P<d>\d+)+(?P<shape>{name})[,](?P<modifiers>[^,]*)".format(name=NAME)
AM = r"(?P<param>AM)(?P<name>{name})\*(?P<macro>.*)".format(name=NAME)
# begin deprecated
@@ -477,16 +529,20 @@ class Gerber(object):
self.ctx.set_image_polarity(stmt.ip)
elif stmt.param == "LP:":
self.ctx.set_level_polarity(stmt.lp)
+ elif stmt.param == "AD":
+ self.ctx.define_aperture(stmt.d, stmt.shape, stmt.modifiers)
def _evaluate_coord(self, stmt):
+
+ if stmt.function in ("G01", "G1", "G02", "G2", "G03", "G3"):
+ self.ctx.set_interpolation(stmt.function)
+
if stmt.op == "D01":
- self.ctx.move(stmt.x, stmt.y)
- self.ctx.line()
+ self.ctx.stroke(stmt.x, stmt.y)
elif stmt.op == "D02":
self.ctx.move(stmt.x, stmt.y)
elif stmt.op == "D03":
- self.ctx.move(stmt.x, stmt.y)
- self.ctx.flash()
+ self.ctx.flash(stmt.x, stmt.y)
def _evaluate_aperture(self, stmt):
self.ctx.set_aperture(stmt.d)