455 lines
17 KiB
Python
Executable file
455 lines
17 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# coding=utf-8
|
|
#
|
|
# Copyright (C) 2005,2007,2008 Aaron Spike, aaron@ekips.org
|
|
# Copyright (C) 2008,2010 Alvin Penner, penner@vaxxine.com
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
#
|
|
"""
|
|
This file output script for Inkscape creates a AutoCAD R14 DXF file.
|
|
The spec can be found here: http://www.autodesk.com/techpubs/autocad/acadr14/dxf/index.htm.
|
|
|
|
File history:
|
|
- template dxf_outlines.dxf added Feb 2008 by Alvin Penner
|
|
- ROBO-Master output option added Aug 2008
|
|
- ROBO-Master multispline output added Sept 2008
|
|
- LWPOLYLINE output modification added Dec 2008
|
|
- toggle between LINE/LWPOLYLINE added Jan 2010
|
|
- support for transform elements added July 2010
|
|
- support for layers added July 2010
|
|
- support for rectangle added Dec 2010
|
|
"""
|
|
|
|
from __future__ import print_function
|
|
|
|
import inkex
|
|
from inkex import (
|
|
colors,
|
|
bezier,
|
|
Transform,
|
|
Group,
|
|
Layer,
|
|
Use,
|
|
PathElement,
|
|
Rectangle,
|
|
Line,
|
|
Circle,
|
|
Ellipse,
|
|
)
|
|
from inkex.localization import inkex_gettext as _
|
|
|
|
|
|
def get_matrix(u, i, j):
|
|
if j == i + 2:
|
|
return (
|
|
(u[i] - u[i - 1])
|
|
* (u[i] - u[i - 1])
|
|
/ (u[i + 2] - u[i - 1])
|
|
/ (u[i + 1] - u[i - 1])
|
|
)
|
|
elif j == i + 1:
|
|
return (
|
|
(u[i] - u[i - 1]) * (u[i + 2] - u[i]) / (u[i + 2] - u[i - 1])
|
|
+ (u[i + 1] - u[i]) * (u[i] - u[i - 2]) / (u[i + 1] - u[i - 2])
|
|
) / (u[i + 1] - u[i - 1])
|
|
elif j == i:
|
|
return (
|
|
(u[i + 1] - u[i])
|
|
* (u[i + 1] - u[i])
|
|
/ (u[i + 1] - u[i - 2])
|
|
/ (u[i + 1] - u[i - 1])
|
|
)
|
|
else:
|
|
return 0
|
|
|
|
|
|
def get_fit(u, csp, col):
|
|
return (
|
|
(1 - u) ** 3 * csp[0][col]
|
|
+ 3 * (1 - u) ** 2 * u * csp[1][col]
|
|
+ 3 * (1 - u) * u**2 * csp[2][col]
|
|
+ u**3 * csp[3][col]
|
|
)
|
|
|
|
|
|
class DxfOutlines(inkex.OutputExtension):
|
|
def add_arguments(self, pars):
|
|
pars.add_argument("--tab")
|
|
pars.add_argument("-R", "--ROBO", type=inkex.Boolean, default=False)
|
|
pars.add_argument("-P", "--POLY", type=inkex.Boolean, default=False)
|
|
pars.add_argument("-F", "--FLATTENBEZ", type=inkex.Boolean, default=False)
|
|
pars.add_argument(
|
|
"--unit_from_document", type=inkex.Boolean, default=True
|
|
) # px
|
|
pars.add_argument("--units", default="px") # px
|
|
pars.add_argument("--encoding", dest="char_encode", default="latin_1")
|
|
pars.add_argument("--layer_option", default="all")
|
|
pars.add_argument("--layer_name")
|
|
|
|
self.dxf = []
|
|
self.handle = 255 # handle for DXF ENTITY
|
|
self.layers = ["0"]
|
|
self.layer = "0" # mandatory layer
|
|
self.layernames = []
|
|
self.csp_old = [[0.0, 0.0]] * 4 # previous spline
|
|
self.d = [0.0] # knot vector
|
|
self.poly = [[0.0, 0.0]] # LWPOLYLINE data
|
|
|
|
def dxf_add(self, str):
|
|
self.dxf.append(str.encode(self.options.char_encode))
|
|
|
|
def dxf_line(self, csp):
|
|
"""Draw a line in the DXF format"""
|
|
self.handle += 1
|
|
self.dxf_add(
|
|
" 0\nLINE\n 5\n%x\n100\nAcDbEntity\n 8\n%s\n 62\n%d\n100\nAcDbLine\n"
|
|
% (self.handle, self.layer, self.color)
|
|
)
|
|
self.dxf_add(
|
|
" 10\n%f\n 20\n%f\n 30\n0.0\n 11\n%f\n 21\n%f\n 31\n0.0\n"
|
|
% (csp[0][0], csp[0][1], csp[1][0], csp[1][1])
|
|
)
|
|
|
|
def LWPOLY_line(self, csp):
|
|
if (
|
|
abs(csp[0][0] - self.poly[-1][0]) > 0.0001
|
|
or abs(csp[0][1] - self.poly[-1][1]) > 0.0001
|
|
or self.color_LWPOLY != self.color
|
|
): # THIS LINE IS NEW
|
|
self.LWPOLY_output() # terminate current polyline
|
|
self.poly = [csp[0]] # initiallize new polyline
|
|
self.color_LWPOLY = self.color
|
|
self.layer_LWPOLY = self.layer
|
|
self.poly.append(csp[1])
|
|
|
|
def LWPOLY_output(self):
|
|
if len(self.poly) == 1:
|
|
return
|
|
self.handle += 1
|
|
closed = 1
|
|
if (
|
|
abs(self.poly[0][0] - self.poly[-1][0]) > 0.0001
|
|
or abs(self.poly[0][1] - self.poly[-1][1]) > 0.0001
|
|
):
|
|
closed = 0
|
|
self.dxf_add(
|
|
" 0\nLWPOLYLINE\n 5\n%x\n100\nAcDbEntity\n 8\n%s\n 62\n%d\n100\nAcDbPolyline\n 90\n%d\n 70\n%d\n"
|
|
% (
|
|
self.handle,
|
|
self.layer_LWPOLY,
|
|
self.color_LWPOLY,
|
|
len(self.poly) - closed,
|
|
closed,
|
|
)
|
|
)
|
|
for i in range(len(self.poly) - closed):
|
|
self.dxf_add(
|
|
" 10\n%f\n 20\n%f\n 30\n0.0\n" % (self.poly[i][0], self.poly[i][1])
|
|
)
|
|
|
|
def dxf_spline(self, csp):
|
|
knots = 8
|
|
ctrls = 4
|
|
self.handle += 1
|
|
self.dxf_add(
|
|
" 0\nSPLINE\n 5\n%x\n100\nAcDbEntity\n 8\n%s\n 62\n%d\n100\nAcDbSpline\n"
|
|
% (self.handle, self.layer, self.color)
|
|
)
|
|
self.dxf_add(" 70\n8\n 71\n3\n 72\n%d\n 73\n%d\n 74\n0\n" % (knots, ctrls))
|
|
for i in range(2):
|
|
for j in range(4):
|
|
self.dxf_add(" 40\n%d\n" % i)
|
|
for i in csp:
|
|
self.dxf_add(" 10\n%f\n 20\n%f\n 30\n0.0\n" % (i[0], i[1]))
|
|
|
|
def ROBO_spline(self, csp):
|
|
"""this spline has zero curvature at the endpoints, as in ROBO-Master"""
|
|
if (
|
|
abs(csp[0][0] - self.csp_old[3][0]) > 0.0001
|
|
or abs(csp[0][1] - self.csp_old[3][1]) > 0.0001
|
|
or abs(
|
|
(csp[1][1] - csp[0][1]) * (self.csp_old[3][0] - self.csp_old[2][0])
|
|
- (csp[1][0] - csp[0][0]) * (self.csp_old[3][1] - self.csp_old[2][1])
|
|
)
|
|
> 0.001
|
|
):
|
|
self.ROBO_output() # terminate current spline
|
|
self.xfit = [csp[0][0]] # initiallize new spline
|
|
self.yfit = [csp[0][1]]
|
|
self.d = [0.0]
|
|
self.color_ROBO = self.color
|
|
self.layer_ROBO = self.layer
|
|
self.xfit += 3 * [0.0]
|
|
self.yfit += 3 * [0.0]
|
|
self.d += 3 * [0.0]
|
|
for i in range(1, 4):
|
|
j = len(self.d) + i - 4
|
|
self.xfit[j] = get_fit(i / 3.0, csp, 0)
|
|
self.yfit[j] = get_fit(i / 3.0, csp, 1)
|
|
self.d[j] = self.d[j - 1] + bezier.pointdistance(
|
|
(self.xfit[j - 1], self.yfit[j - 1]), (self.xfit[j], self.yfit[j])
|
|
)
|
|
self.csp_old = csp
|
|
|
|
def ROBO_output(self):
|
|
try:
|
|
import numpy
|
|
from numpy.linalg import solve
|
|
except ImportError:
|
|
inkex.errormsg(
|
|
_(
|
|
"Failed to import the numpy or numpy.linalg modules. "
|
|
"These modules are required by the ROBO option."
|
|
"Please install them and try again."
|
|
)
|
|
)
|
|
return
|
|
|
|
if len(self.d) == 1:
|
|
return
|
|
fits = len(self.d)
|
|
ctrls = fits + 2
|
|
knots = ctrls + 4
|
|
self.xfit += 2 * [0.0] # pad with 2 endpoint constraints
|
|
self.yfit += 2 * [0.0]
|
|
self.d += 6 * [0.0] # pad with 3 duplicates at each end
|
|
self.d[fits + 2] = self.d[fits + 1] = self.d[fits] = self.d[fits - 1]
|
|
|
|
solmatrix = numpy.zeros((ctrls, ctrls), dtype=float)
|
|
for i in range(fits):
|
|
solmatrix[i, i] = get_matrix(self.d, i, i)
|
|
solmatrix[i, i + 1] = get_matrix(self.d, i, i + 1)
|
|
solmatrix[i, i + 2] = get_matrix(self.d, i, i + 2)
|
|
solmatrix[fits, 0] = self.d[2] / self.d[fits - 1] # curvature at start = 0
|
|
solmatrix[fits, 1] = -(self.d[1] + self.d[2]) / self.d[fits - 1]
|
|
solmatrix[fits, 2] = self.d[1] / self.d[fits - 1]
|
|
solmatrix[fits + 1, fits - 1] = (self.d[fits - 1] - self.d[fits - 2]) / self.d[
|
|
fits - 1
|
|
] # curvature at end = 0
|
|
solmatrix[fits + 1, fits] = (
|
|
self.d[fits - 3] + self.d[fits - 2] - 2 * self.d[fits - 1]
|
|
) / self.d[fits - 1]
|
|
solmatrix[fits + 1, fits + 1] = (self.d[fits - 1] - self.d[fits - 3]) / self.d[
|
|
fits - 1
|
|
]
|
|
xctrl = solve(solmatrix, self.xfit)
|
|
yctrl = solve(solmatrix, self.yfit)
|
|
self.handle += 1
|
|
self.dxf_add(
|
|
" 0\nSPLINE\n 5\n%x\n100\nAcDbEntity\n 8\n%s\n 62\n%d\n100\nAcDbSpline\n"
|
|
% (self.handle, self.layer_ROBO, self.color_ROBO)
|
|
)
|
|
self.dxf_add(
|
|
" 70\n0\n 71\n3\n 72\n%d\n 73\n%d\n 74\n%d\n" % (knots, ctrls, fits)
|
|
)
|
|
for i in range(knots):
|
|
self.dxf_add(" 40\n%f\n" % self.d[i - 3])
|
|
for i in range(ctrls):
|
|
self.dxf_add(" 10\n%f\n 20\n%f\n 30\n0.0\n" % (xctrl[i], yctrl[i]))
|
|
for i in range(fits):
|
|
self.dxf_add(" 11\n%f\n 21\n%f\n 31\n0.0\n" % (self.xfit[i], self.yfit[i]))
|
|
|
|
def process_shape(self, node, mat):
|
|
if not isinstance(node, (PathElement, Rectangle, Line, Circle, Ellipse)):
|
|
return
|
|
|
|
rgb = (0, 0, 0)
|
|
style = node.style("stroke")
|
|
if style is not None and isinstance(style, inkex.Color):
|
|
rgb = style.to_rgb()
|
|
hsl = colors.rgb_to_hsl(rgb[0] / 255.0, rgb[1] / 255.0, rgb[2] / 255.0)
|
|
self.color = 7 # default is blac
|
|
if hsl[2]:
|
|
self.color = 1 + (int(6 * hsl[0] + 0.5) % 6) # use 6 hues
|
|
|
|
# Transforming /after/ superpath is more reliable than before
|
|
# because of some issues with arcs in transformations
|
|
path = node.path.to_superpath().transform(Transform(mat) @ node.transform)
|
|
|
|
# If Flatten Beziers is enabled, subdivide our beziers and
|
|
# we'll later just ignore the curve and output flat lines
|
|
if self.options.FLATTENBEZ:
|
|
bezier.cspsubdiv(path, 0.1) # default to most detailed (0.1)
|
|
|
|
# Now output the path.
|
|
for sub in path:
|
|
for i in range(len(sub) - 1):
|
|
s = sub[i]
|
|
e = sub[i + 1]
|
|
# If flattening beziers, ignore curves and output flat lines
|
|
if (s[1] == s[2] and e[0] == e[1]) or self.options.FLATTENBEZ:
|
|
if self.options.POLY:
|
|
self.LWPOLY_line([s[1], e[1]])
|
|
else:
|
|
self.dxf_line([s[1], e[1]])
|
|
elif self.options.ROBO:
|
|
self.ROBO_spline([s[1], s[2], e[0], e[1]])
|
|
else:
|
|
self.dxf_spline([s[1], s[2], e[0], e[1]])
|
|
|
|
def process_clone(self, node):
|
|
"""Process a clone node, looking for internal paths"""
|
|
trans = node.get("transform")
|
|
x = node.get("x")
|
|
y = node.get("y")
|
|
mat = Transform([[1.0, 0.0, 0.0], [0.0, 1.0, 0.0]])
|
|
if trans:
|
|
mat @= Transform(trans)
|
|
if x:
|
|
mat @= Transform([[1.0, 0.0, float(x)], [0.0, 1.0, 0.0]])
|
|
if y:
|
|
mat @= Transform([[1.0, 0.0, 0.0], [0.0, 1.0, float(y)]])
|
|
# push transform
|
|
if trans or x or y:
|
|
self.groupmat.append(Transform(self.groupmat[-1]) @ mat)
|
|
# get referenced node
|
|
refid = node.get("xlink:href")
|
|
refnode = self.svg.getElementById(refid[1:])
|
|
if refnode is not None:
|
|
if isinstance(refnode, Group):
|
|
self.process_group(refnode)
|
|
elif isinstance(refnode, Use):
|
|
self.process_clone(refnode)
|
|
else:
|
|
self.process_shape(refnode, self.groupmat[-1])
|
|
# pop transform
|
|
if trans or x or y:
|
|
self.groupmat.pop()
|
|
|
|
def process_group(self, group):
|
|
"""Process group elements"""
|
|
if isinstance(group, Layer):
|
|
style = group.style
|
|
if (
|
|
style("display") == "none"
|
|
and self.options.layer_option
|
|
and self.options.layer_option == "visible"
|
|
):
|
|
return
|
|
layer = group.label
|
|
if self.options.layer_name and self.options.layer_option == "name":
|
|
if not layer.lower() in self.options.layer_name:
|
|
return
|
|
|
|
layer = layer.replace(" ", "_")
|
|
if layer in self.layers:
|
|
self.layer = layer
|
|
trans = group.get("transform")
|
|
if trans:
|
|
self.groupmat.append(Transform(self.groupmat[-1]) @ Transform(trans))
|
|
for node in group:
|
|
try:
|
|
if isinstance(node, Group):
|
|
self.process_group(node)
|
|
elif isinstance(node, Use):
|
|
self.process_clone(node)
|
|
else:
|
|
self.process_shape(node, self.groupmat[-1])
|
|
except RecursionError as e:
|
|
raise inkex.AbortExtension(
|
|
_(
|
|
'Too many nested groups. Please use the "Deep Ungroup" extension first.'
|
|
)
|
|
) from e # pylint: disable=line-too-long
|
|
if trans:
|
|
self.groupmat.pop()
|
|
|
|
def save(self, stream):
|
|
# Warn user if name match field is empty
|
|
if (
|
|
self.options.layer_option
|
|
and self.options.layer_option == "name"
|
|
and not self.options.layer_name
|
|
):
|
|
return inkex.errormsg(
|
|
_(
|
|
"Error: Field 'Layer match name' must be filled when using "
|
|
"'By name match' option"
|
|
)
|
|
)
|
|
|
|
if len(self.svg.xpath("//svg:use|//svg:flowRoot|//svg:text")) > 0:
|
|
self.preprocess(["flowRoot", "text"])
|
|
|
|
# Split user layer data into a list: "layerA,layerb,LAYERC" becomes ["layera", "layerb", "layerc"]
|
|
if self.options.layer_name:
|
|
self.options.layer_name = self.options.layer_name.lower().split(",")
|
|
|
|
# References: Minimum Requirements for Creating a DXF File of a 3D Model By Paul Bourke
|
|
# NURB Curves: A Guide for the Uninitiated By Philip J. Schneider
|
|
# The NURBS Book By Les Piegl and Wayne Tiller (Springer, 1995)
|
|
# self.dxf_add("999\nDXF created by Inkscape\n") # Some programs do not take comments in DXF files (KLayout 0.21.12 for example)
|
|
if self.options.unit_from_document:
|
|
unit = self.svg.document_unit
|
|
else:
|
|
unit = self.options.units
|
|
with open(self.get_resource("dxf14_header.txt"), "r") as fhl:
|
|
header = fhl.read()
|
|
unit_map = {"px": 0, "in": 1, "ft": 2, "mm": 4, "cm": 5, "m": 6}
|
|
header = header.replace("<unit specifier>", str(unit_map.get(unit, 0)))
|
|
self.dxf_add(header)
|
|
for node in self.svg.xpath("//svg:g"):
|
|
if isinstance(node, Layer):
|
|
layer = node.label
|
|
self.layernames.append(layer.lower())
|
|
if (
|
|
self.options.layer_name
|
|
and self.options.layer_option
|
|
and self.options.layer_option == "name"
|
|
and not layer.lower() in self.options.layer_name
|
|
):
|
|
continue
|
|
layer = layer.replace(" ", "_")
|
|
if layer and layer not in self.layers:
|
|
self.layers.append(layer)
|
|
self.dxf_add(
|
|
" 2\nLAYER\n 5\n2\n100\nAcDbSymbolTable\n 70\n%s\n" % len(self.layers)
|
|
)
|
|
for i in range(len(self.layers)):
|
|
self.dxf_add(
|
|
" 0\nLAYER\n 5\n%x\n100\nAcDbSymbolTableRecord\n100\nAcDbLayerTableRecord\n 2\n%s\n 70\n0\n 6\nCONTINUOUS\n"
|
|
% (i + 80, self.layers[i])
|
|
)
|
|
with open(self.get_resource("dxf14_style.txt"), "r") as fhl:
|
|
self.dxf_add(fhl.read())
|
|
|
|
# Set toplevel transform
|
|
scale = self.svg.inkscape_scale
|
|
self.groupmat = [
|
|
[[scale, 0.0, 0.0], [0.0, -scale, self.svg.viewbox_height * scale]]
|
|
]
|
|
self.process_group(self.svg)
|
|
if self.options.ROBO:
|
|
self.ROBO_output()
|
|
if self.options.POLY:
|
|
self.LWPOLY_output()
|
|
with open(self.get_resource("dxf14_footer.txt"), "r") as fhl:
|
|
self.dxf_add(fhl.read())
|
|
# Warn user if layer data seems wrong
|
|
if (
|
|
self.options.layer_name
|
|
and self.options.layer_option
|
|
and self.options.layer_option == "name"
|
|
):
|
|
for layer in self.options.layer_name:
|
|
if layer not in self.layernames:
|
|
inkex.errormsg(_("Warning: Layer '{}' not found!").format(layer))
|
|
|
|
stream.write(b"".join(self.dxf))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
DxfOutlines().run()
|