summaryrefslogtreecommitdiff
path: root/gerberex
diff options
context:
space:
mode:
Diffstat (limited to 'gerberex')
-rw-r--r--gerberex/dxf.py198
-rw-r--r--gerberex/dxf_path.py197
2 files changed, 342 insertions, 53 deletions
diff --git a/gerberex/dxf.py b/gerberex/dxf.py
index a2c26d0..e6eb971 100644
--- a/gerberex/dxf.py
+++ b/gerberex/dxf.py
@@ -5,16 +5,29 @@
import io
from math import pi, cos, sin, tan, atan, atan2, acos, asin, sqrt
+import dxfgrabber
from gerber.cam import CamFile, FileSettings
from gerber.utils import inch, metric, write_gerber_value
from gerber.gerber_statements import ADParamStmt
from gerber.excellon_statements import ExcellonTool
from gerber.excellon_statements import CoordinateStmt
-import dxfgrabber
+from gerberex.dxf_path import generate_closed_paths
+
+ACCEPTABLE_ERROR = 0.001
+
+def is_equal_value(a, b, error_range=0):
+ return a - b <= error_range and a - b >= -error_range
+
+def is_equal_point(a, b, error_range=0):
+ return is_equal_value(a[0], b[0], error_range) and \
+ is_equal_value(a[1], b[1], error_range)
class DxfStatement(object):
def __init__(self, entity):
self.entity = entity
+ self.start = None
+ self.end = None
+ self.is_closed = False
def to_gerber(self, settings=None, pitch=0, width=0):
pass
@@ -28,16 +41,22 @@ class DxfStatement(object):
def to_metric(self):
pass
+ def is_equal_to(self, target, error_range=0):
+ return False
+
+ def reverse(self):
+ raise Exception('Not implemented')
+
class DxfLineStatement(DxfStatement):
def __init__(self, entity):
super(DxfLineStatement, self).__init__(entity)
+ self.start = (self.entity.start[0], self.entity.start[1])
+ self.end = (self.entity.end[0], self.entity.end[1])
def to_gerber(self, settings=FileSettings(), pitch=0, width=0):
if pitch == 0:
- x0 = self.entity.start[0]
- y0 = self.entity.start[1]
- x1 = self.entity.end[0]
- y1 = self.entity.end[1]
+ x0, y0 = self.start
+ x1, y1 = self.end
return 'G01*\nX{0}Y{1}D02*\nX{2}Y{3}D01*'.format(
write_gerber_value(x0, settings.format,
settings.zero_suppression),
@@ -67,22 +86,34 @@ class DxfLineStatement(DxfStatement):
return gstr
def to_inch(self):
- self.entity.start = (
- inch(self.entity.start[0]), inch(self.entity.start[1]))
- self.entity.end = (
- inch(self.entity.end[0]), inch(self.entity.end[1]))
+ self.start = (
+ inch(self.start[0]), inch(self.start[1]))
+ self.end = (
+ inch(self.end[0]), inch(self.end[1]))
def to_metric(self):
- self.entity.start = (
- metric(self.entity.start[0]), inch(self.entity.start[1]))
- self.entity.end = (
- metric(self.entity.end[0]), inch(self.entity.end[1]))
+ self.start = (
+ metric(self.start[0]), inch(self.start[1]))
+ self.end = (
+ metric(self.end[0]), inch(self.end[1]))
+
+ def is_equal_to(self, target, error_range=0):
+ if not isinstance(target, DxfLineStatement):
+ return False
+ return (is_equal_point(self.start, target.start, error_range) and \
+ is_equal_point(self.end, target.end, error_range)) or \
+ (is_equal_point(self.start, target.end, error_range) and \
+ is_equal_point(self.end, target.start, error_range))
+
+ def reverse(self):
+ pt = self.start
+ self.start = self.end
+ self.end = pt
def _dots(self, pitch, width):
- x0 = self.entity.start[0]
- y0 = self.entity.start[1]
- x1 = self.entity.end[0]
- y1 = self.entity.end[1]
+ x0, y0 = self.start
+ x1, y1 = self.end
+ y1 = self.end[1]
xp = x1 - x0
yp = y1 - y0
l = sqrt(xp * xp + yp * yp)
@@ -99,13 +130,17 @@ class DxfLineStatement(DxfStatement):
class DxfCircleStatement(DxfStatement):
def __init__(self, entity):
super(DxfCircleStatement, self).__init__(entity)
+ self.radius = self.entity.radius
+ self.center = (self.entity.center[0], self.entity.center[1])
+ self.start = (self.center[0] + self.radius, self.center[1])
+ self.end = self.start
+ self.is_closed = True
def to_gerber(self, settings=FileSettings(), pitch=0, width=0):
if pitch:
return
- r = self.entity.radius
- x0 = self.entity.center[0]
- y0 = self.entity.center[1]
+ r = self.radius
+ x0, y0 = self.center
return 'G01*\nX{0}Y{1}D02*\n' \
'G75*\nG03*\nX{2}Y{3}I{4}J{5}D01*'.format(
write_gerber_value(x0 + r, settings.format,
@@ -124,66 +159,107 @@ class DxfCircleStatement(DxfStatement):
)
def to_inch(self):
- self.entity.radius = inch(self.entity.radius)
- self.entity.center = (
- inch(self.entity.center[0]), inch(self.entity.center[1]))
+ self.radius = inch(self.radius)
+ self.center = (
+ inch(self.center[0]), inch(self.center[1]))
def to_metric(self):
- self.entity.radius = metric(self.entity.radius)
- self.entity.center = (
- metric(self.entity.center[0]), metric(self.entity.center[1]))
+ self.radius = metric(self.radius)
+ self.center = (
+ metric(self.center[0]), metric(self.center[1]))
+
+ def is_equal_to(self, target, error_range=0):
+ if not isinstance(target, DxfCircleStatement):
+ return False
+ return is_equal_point(self.center, target.enter, error_range) and \
+ is_equal_value(self.radius, target.radius)
+
+ def reverse(self):
+ pass
class DxfArcStatement(DxfStatement):
def __init__(self, entity):
super(DxfArcStatement, self).__init__(entity)
+ self.start_angle = self.entity.start_angle
+ self.end_angle = self.entity.end_angle
+ self.radius = self.entity.radius
+ self.center = (self.entity.center[0], self.entity.center[1])
+ self.start = (
+ self.center[0] + self.radius * cos(self.start_angle / 180. * pi),
+ self.center[1] + self.radius * sin(self.start_angle / 180. * pi),
+ )
+ self.end = (
+ self.center[0] + self.radius * cos(self.end_angle / 180. * pi),
+ self.center[1] + self.radius * sin(self.end_angle / 180. * pi),
+ )
+ angle = self.end_angle - self.start_angle
+ self.is_closed = angle >= 360 or angle <= -360
def to_gerber(self, settings=FileSettings(), pitch=0, width=0):
if pitch:
return
- deg0 = self.entity.start_angle
- deg1 = self.entity.end_angle
- r = self.entity.radius
- x0 = self.entity.center[0]
- y0 = self.entity.center[1]
- begin_x = x0 + r * cos(deg0 / 180. * pi)
- begin_y = y0 + r * sin(deg0 / 180. * pi)
- end_x = x0 + r * cos(deg1 / 180. * pi)
- end_y = y0 + r * sin(deg1 / 180. * pi)
+ x0 = self.center[0]
+ y0 = self.center[1]
+ start_x, start_y = self.start
+ end_x, end_y = self.end
return 'G01*\nX{0}Y{1}D02*\n' \
'G75*\nG{2}*\nX{3}Y{4}I{5}J{6}D01*'.format(
- write_gerber_value(begin_x, settings.format,
+ write_gerber_value(start_x, settings.format,
settings.zero_suppression),
- write_gerber_value(begin_y, settings.format,
+ write_gerber_value(start_y, settings.format,
settings.zero_suppression),
- '03',
+ '02' if self.start_angle > self.end_angle else '03',
write_gerber_value(end_x, settings.format,
settings.zero_suppression),
write_gerber_value(end_y, settings.format,
settings.zero_suppression),
- write_gerber_value(x0 - begin_x, settings.format,
+ write_gerber_value(x0 - start_x, settings.format,
settings.zero_suppression),
- write_gerber_value(y0 - begin_y, settings.format,
+ write_gerber_value(y0 - start_y, settings.format,
settings.zero_suppression)
)
def to_inch(self):
- self.entity.start_angle = inch(self.entity.start_angle)
- self.entity.end_angle = inch(self.entity.end_angle)
- self.entity.radius = inch(self.entity.radius)
- self.entity.center = (
- inch(self.entity.center[0]), inch(self.entity.center[1]))
+ self.radius = inch(self.radius)
+ self.center = (inch(self.center[0]), inch(self.center[1]))
+ self.start = (inch(self.start[0]), inch(self.start[1]))
+ self.end = (inch(self.end[0]), inch(self.end[1]))
def to_metric(self):
- self.entity.start_angle = metric(self.entity.start_angle)
- self.entity.end_angle = metric(self.entity.end_angle)
- self.entity.radius = metric(self.entity.radius)
- self.entity.center = (
- metric(self.entity.center[0]), metric(self.entity.center[1]))
+ self.radius = metric(self.radius)
+ self.center = (metric(self.center[0]), metric(self.center[1]))
+ self.start = (metric(self.start[0]), metric(self.start[1]))
+ self.end = (metric(self.end[0]), metric(self.end[1]))
+
+ def is_equal_to(self, target, error_range=0):
+ if not isinstance(target, DxfArcStatement):
+ return False
+ aerror_range = error_range / pi * self.radius * 180
+ return is_equal_point(self.center, target.center, error_range) and \
+ is_equal_value(self.radius, target.radius, error_range) and \
+ ((is_equal_value(self.start_angle, target.start_angle, aerror_range) and
+ is_equal_value(self.end_angle, target.end_angle, aerror_range)) or
+ (is_equal_value(self.start_angle, target.end_angle, aerror_range) and
+ is_equal_value(self.end_angle, target.end_angle, aerror_range)))
+
+ def reverse(self):
+ tmp = self.start_angle
+ self.start_angle = self.end_angle
+ self.end_angle = tmp
+ tmp = self.start
+ self.start = self.end
+ self.end = tmp
class DxfPolylineStatement(DxfStatement):
def __init__(self, entity):
super(DxfPolylineStatement, self).__init__(entity)
+ self.start = (self.entity.points[0][0], self.entity.points[0][1])
+ self.is_closed = self.entity.is_closed
+ if self.is_closed:
+ self.end = self.start
+ else:
+ self.start = (self.entity.points[-1][0], self.entity.points[-1][1])
def to_gerber(self, settings=FileSettings(), pitch=0, width=0):
if pitch:
@@ -243,12 +319,16 @@ class DxfPolylineStatement(DxfStatement):
return gerber
def to_inch(self):
+ self.start = (inch(self.start[0]), inch(self.start[1]))
+ self.end = (inch(self.end[0]), inch(self.end[1]))
for idx in range(0, len(self.entity.points)):
self.entity.points[idx] = (
inch(self.entity.points[idx][0]), inch(self.entity.points[idx][1]))
self.entity.bulge[idx] = inch(self.entity.bulge[idx])
def to_metric(self):
+ self.start = (metric(self.start[0]), metric(self.start[1]))
+ self.end = (metric(self.end[0]), metric(self.end[1]))
for idx in range(0, len(self.entity.points)):
self.entity.points[idx] = (
metric(self.entity.points[idx][0]), metric(self.entity.points[idx][1]))
@@ -261,8 +341,9 @@ class DxfStatements(object):
self._units = units
self.dcode = dcode
self.draw_mode = draw_mode
- self.pitch = inch(1) if self._units == 'unit' else 1
+ self.pitch = inch(1) if self._units == 'inch' else 1
self.width = 0
+ self.error_range = inch(ACCEPTABLE_ERROR) if self._units == 'inch' else ACCEPTABLE_ERROR
self.statements = []
for entity in entities:
if entity.dxftype == 'LWPOLYLINE':
@@ -273,6 +354,7 @@ class DxfStatements(object):
self.statements.append(DxfCircleStatement(entity))
elif entity.dxftype == 'ARC':
self.statements.append(DxfArcStatement(entity))
+ self.paths = generate_closed_paths(self.statements, self.error_range)
@property
def units(self):
@@ -282,12 +364,16 @@ class DxfStatements(object):
def gerbers():
yield 'D{0}*'.format(self.dcode)
if self.draw_mode == DxfFile.DM_FILL:
- yield 'G36*'
for statement in self.statements:
+ yield 'G36*'
if isinstance(statement, DxfCircleStatement) or \
(isinstance(statement, DxfPolylineStatement) and statement.entity.is_closed):
yield statement.to_gerber(settings)
- yield 'G37*'
+ yield 'G37*'
+ for path in self.paths:
+ yield 'G36*'
+ yield path.to_gerber(settings)
+ yield 'G37*'
else:
for statement in self.statements:
yield statement.to_gerber(
@@ -310,15 +396,21 @@ class DxfStatements(object):
if self._units == 'metric':
self._units = 'inch'
self.pitch = inch(self.pitch)
+ self.error_range = inch(self.error_range)
for statement in self.statements:
statement.to_inch()
+ for path in self.paths:
+ path.to_inch()
def to_metric(self):
if self._units == 'inch':
self._units = 'metric'
self.pitch = metric(self.pitch)
+ self.error_range = metric(self.error_range)
for statement in self.statements:
statement.to_metric()
+ for path in self.paths:
+ path.to_metric()
class DxfHeaderStatement(object):
def to_gerber(self, settings):
diff --git a/gerberex/dxf_path.py b/gerberex/dxf_path.py
new file mode 100644
index 0000000..b066b08
--- /dev/null
+++ b/gerberex/dxf_path.py
@@ -0,0 +1,197 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+# Copyright 2019 Hiroshi Murayama <opiopan@gmail.com>
+
+from gerber.utils import inch, metric, write_gerber_value
+from gerber.cam import FileSettings
+
+class DxfPath(object):
+ def __init__(self, statement, error_range=0):
+ self.statements = [statement]
+ self.error_range = error_range
+
+ @property
+ def start(self):
+ return self.statements[0].start
+
+ @property
+ def end(self):
+ return self.statements[-1].end
+
+ @property
+ def is_closed(self):
+ from gerberex.dxf import is_equal_point
+ return len(self.statements) > 1 and \
+ is_equal_point(self.start, self.end, self.error_range)
+
+ def is_equal_to(self, target, error_range=0):
+ from gerberex.dxf import is_equal_point
+ if not isinstance(target, DxfPath):
+ return False
+ if len(self.statements) != len(target.statements):
+ return False
+ if is_equal_point(self.start, target.start, error_range) and \
+ is_equal_point(self.end, target.end, error_range):
+ for i in range(0, len(self.statements)):
+ if not self.statements[i].is_equal_to(target.statements[i], error_range):
+ return False
+ return True
+ elif is_equal_point(self.start, target.end, error_range) and \
+ is_equal_point(self.end, target.start, error_range):
+ for i in range(0, len(self.statements)):
+ if not self.statements[i].is_equal_to(target.statements[-1 - i], error_range):
+ return False
+ return True
+ return False
+
+ def to_inch(self):
+ self.error_range = inch(self.error_range)
+
+ def to_metric(self):
+ self.error_range = metric(self.error_range)
+
+ def reverse(self):
+ rlist = []
+ for statement in reversed(self.statements):
+ statement.reverse()
+ rlist.append(statement)
+ self.statements = rlist
+
+ def merge(self, element, error_range=0):
+ from gerberex.dxf import is_equal_point
+ if self.is_closed or element.is_closed:
+ return False
+ if not error_range:
+ error_range = self.error_range
+ if is_equal_point(self.end, element.start, error_range):
+ return self._append_at_end(element, error_range)
+ elif is_equal_point(self.end, element.end, error_range):
+ element.reverse()
+ return self._append_at_end(element, error_range)
+ elif is_equal_point(self.start, element.end, error_range):
+ return self._insert_on_top(element, error_range)
+ elif is_equal_point(self.start, element.start, error_range):
+ element.reverse()
+ return self._insert_on_top(element, error_range)
+ else:
+ return False
+
+ def _append_at_end(self, element, error_range=0):
+ if isinstance(element, DxfPath):
+ if self.is_equal_to(element, error_range):
+ return False
+ for i in range(0, min(len(self.statements), len(element.statements))):
+ if not self.statements[-1 - i].is_equal_to(element.statements[i]):
+ break
+ for j in range(0, min(len(self.statements), len(element.statements))):
+ if not self.statements[j].is_equal_to(element.statements[-1 - j]):
+ break
+ if i + j >= len(element.statements):
+ return False
+ mergee = list(element.statements)
+ if i > 0:
+ del mergee[0:i]
+ del self.statements[-i]
+ if j > 0:
+ del mergee[-j]
+ del self.statements[0:j]
+ self.statements.extend(mergee)
+ return True
+ else:
+ if self.statements[-1].is_equal_to(element, error_range) or \
+ self.statements[0].is_equal_to(element, error_range):
+ return False
+ self.statements.appen(element)
+ return True
+
+ def _insert_on_top(self, element, error_range=0):
+ if isinstance(element, DxfPath):
+ if self.is_equal_to(element, error_range):
+ return False
+ for i in range(0, min(len(self.statements), len(element.statements))):
+ if not self.statements[-1 - i].is_equal_to(element.statements[i]):
+ break
+ for j in range(0, min(len(self.statements), len(element.statements))):
+ if not self.statements[j].is_equal_to(element.statements[-1 - j]):
+ break
+ if i + j >= len(element.statements):
+ return False
+ mergee = list(element.statements)
+ if i > 0:
+ del mergee[0:i]
+ del self.statements[-i]
+ if j > 0:
+ del mergee[-j]
+ del self.statements[0:j]
+ self.statements[0:0] = mergee
+ return True
+ else:
+ if self.statements[-1].is_equal_to(element, error_range) or \
+ self.statements[0].is_equal_to(element, error_range):
+ return False
+ self.statements.insert(0, element)
+ return True
+
+ def to_gerber(self, settings=FileSettings(), pitch=0, width=0):
+ from gerberex.dxf import DxfArcStatement
+ if pitch:
+ return
+
+ x0, y0 = self.statements[0].start
+ gerber = 'G01*\nX{0}Y{1}D02*\nG75*'.format(
+ write_gerber_value(x0, settings.format,
+ settings.zero_suppression),
+ write_gerber_value(y0, settings.format,
+ settings.zero_suppression),
+ )
+
+ for statement in self.statements:
+ x0, y0 = statement.start
+ x1, y1 = statement.end
+ if isinstance(statement, DxfArcStatement):
+ xc, yc = statement.center
+ gerber += '\nG{0}*\nX{1}Y{2}I{3}J{4}D01*'.format(
+ '03' if statement.end_angle > statement.start_angle else '02',
+ write_gerber_value(x1, settings.format,
+ settings.zero_suppression),
+ write_gerber_value(y1, settings.format,
+ settings.zero_suppression),
+ write_gerber_value(xc - x0, settings.format,
+ settings.zero_suppression),
+ write_gerber_value(yc - y0, settings.format,
+ settings.zero_suppression)
+ )
+ else:
+ gerber += '\nG01*\nX{0}Y{1}D01*'.format(
+ write_gerber_value(x1, settings.format,
+ settings.zero_suppression),
+ write_gerber_value(y1, settings.format,
+ settings.zero_suppression),
+ )
+
+ return gerber
+
+def generate_closed_paths(statements, error_range=0):
+ from gerberex.dxf import DxfLineStatement, DxfArcStatement
+
+ paths = [DxfPath(s, error_range) \
+ for s in filter(lambda s: isinstance(s, DxfLineStatement) or \
+ isinstance(s, DxfArcStatement), statements)]
+
+ prev_paths_num = 0
+ while prev_paths_num != len(paths):
+ working = []
+ for i in range(len(paths)):
+ mergee = paths[i]
+ for j in range(i + 1, len(paths) - 1):
+ target = paths[j]
+ if target.merge(mergee, error_range):
+ break
+ else:
+ working.append(mergee)
+ prev_paths_num = len(paths)
+ paths = working
+
+ return list(filter(lambda p: p.is_closed, paths))
+