diff options
Diffstat (limited to 'gerber')
| -rw-r--r-- | gerber/render/cairo_backend.py | 156 |
1 files changed, 156 insertions, 0 deletions
diff --git a/gerber/render/cairo_backend.py b/gerber/render/cairo_backend.py new file mode 100644 index 0000000..d5efc2d --- /dev/null +++ b/gerber/render/cairo_backend.py @@ -0,0 +1,156 @@ +#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+
+# Copyright 2014 Hamilton Kibbe <ham@hamiltonkib.be>
+
+# 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.
+
+from .render import GerberContext
+from .apertures import Circle, Rect, Obround, Polygon
+import cairo
+import math
+
+SCALE = 200.
+
+class CairoCircle(Circle):
+ def line(self, ctx, x, y, color=(184/255., 115/255., 51/255.)):
+ ctx.set_source_rgb (*color)
+ ctx.set_line_width(self.diameter * SCALE)
+ ctx.set_line_cap(cairo.LINE_CAP_ROUND)
+ ctx.line_to(x * SCALE, y * SCALE)
+ ctx.stroke()
+
+ def arc(self, ctx, x, y, i, j, direction, color=(184/255., 115/255., 51/255.)):
+ ctx_x, ctx_y = ctx.get_current_point()
+
+ # Do the math
+ center = ((x + i) * SCALE, (y + j) * SCALE)
+ radius = math.sqrt(math.pow(ctx_x - center[0], 2) + math.pow(ctx_y - center[1], 2))
+ delta_x0 = (ctx_x - center[0])
+ delta_y0 = (ctx_y - center[1])
+ delta_x1 = (x * SCALE - center[0])
+ delta_y1 = (y * SCALE - center[1])
+ theta0 = math.atan2(delta_y0, delta_x0)
+ theta1 = math.atan2(delta_y1, delta_x1)
+ # Draw the arc
+ ctx.set_source_rgb (*color)
+ ctx.set_line_width(self.diameter * SCALE)
+ ctx.set_line_cap(cairo.LINE_CAP_ROUND)
+ if direction == 'clockwise':
+ ctx.arc_negative(center[0], center[1], radius, theta0, theta1)
+ else:
+ ctx.arc(center[0], center[1], radius, theta0, theta1)
+ ctx.stroke()
+
+ def flash(self, ctx, x, y, color=(184/255., 115/255., 51/255.)):
+ ctx.set_source_rgb (*color)
+ ctx.set_line_width(0)
+ ctx.arc(x * SCALE, y * SCALE, (self.diameter/2.) * SCALE, 0, 2 * math.pi)
+ ctx.fill()
+
+class CairoRect(Rect):
+ def line(self, ctx, x, y, color=(184/255., 115/255., 51/255.)):
+ ctx.set_source_rgb (*color)
+ ctx.set_line_width(self.diameter * SCALE)
+ ctx.set_line_cap(cairo.LINE_CAP_SQUARE)
+ ctx.line_to(x * SCALE, y * SCALE)
+ ctx.stroke()
+
+ def flash(self, ctx, x, y, color=(184/255., 115/255., 51/255.)):
+ xsize, ysize = self.size
+ ctx.set_source_rgb (*color)
+ ctx.set_line_width(0)
+ x0 = SCALE * (x - (xsize / 2.))
+ y0 = SCALE * (y - (ysize / 2.))
+
+ ctx.rectangle(x0,y0,SCALE * xsize, SCALE * ysize)
+ ctx.fill()
+
+
+
+class GerberCairoContext(GerberContext):
+ def __init__(self, surface=None, size=(1000, 1000),
+ color='rgb(184, 115, 51)', drill_color='gray'):
+ GerberContext.__init__(self)
+ if surface is None:
+ self.surface = cairo.ImageSurface(cairo.FORMAT_ARGB32,
+ size[0], size[1])
+ else:
+ self.surface = surface
+ self.ctx = cairo.Context(self.surface)
+ self.size = size
+ self.ctx.translate(0, self.size[1])
+ self.ctx.scale(1,-1)
+ self.apertures = {}
+ self.color = color
+ self.drill_color = drill_color
+ self.background = False
+
+ def set_bounds(self, bounds):
+ xbounds, ybounds = bounds
+ self.ctx.rectangle(SCALE * xbounds[0], SCALE * ybounds[0], SCALE * (xbounds[1]- xbounds[0]), SCALE * (ybounds[1] - ybounds[0]))
+ self.ctx.set_source_rgb(0,0,0)
+ self.ctx.fill()
+
+ def define_aperture(self, d, shape, modifiers):
+ aperture = None
+ if shape == 'C':
+ aperture = CairoCircle(diameter=float(modifiers[0][0]))
+ elif shape == 'R':
+ aperture = CairoRect(size=modifiers[0][0:2])
+ self.apertures[d] = aperture
+
+ def stroke(self, x, y, i, j):
+ super(GerberCairoContext, self).stroke(x, y, i, j)
+
+ if self.interpolation == 'linear':
+ self.line(x, y)
+ elif self.interpolation == 'arc':
+ self.arc(x, y, i, j)
+ self.move(x,y)
+
+ def line(self, x, y):
+ x, y = self.resolve(x, y)
+ ap = self.apertures.get(self.aperture, None)
+ if ap is None:
+ return
+ ap.line(self.ctx, x, y)
+
+ def arc(self, x, y, i, j):
+ super(GerberCairoContext, self).arc(x, y, i, j)
+ ap = self.apertures.get(self.aperture, None)
+ if ap is None:
+ return
+ ap.arc(self.ctx, x, y, i, j, self.direction)
+
+
+ def flash(self, x, y):
+ x, y = self.resolve(x, y)
+ ap = self.apertures.get(self.aperture, None)
+ if ap is None:
+ return
+ ap.flash(self.ctx, x, y)
+ self.move(x, y, resolve=False)
+
+ def move(self, x, y, resolve=True):
+ super(GerberCairoContext, self).move(x, y, resolve)
+ if x is None:
+ x = self.x
+ if y is None:
+ y = self.y
+ if self.x is not None and self.y is not None:
+ self.ctx.move_to(x * SCALE, y * SCALE)
+
+
+ def dump(self, filename):
+ self.surface.write_to_png(filename)
\ No newline at end of file |