122 lines
2.7 KiB
Python
122 lines
2.7 KiB
Python
import sys, os, xml.etree.ElementTree, shutil
|
|
from lib import util
|
|
from . import effect
|
|
|
|
|
|
def _export_dxf(in_path, out_path, layers):
|
|
dxf_effect = effect.ExportEffect(layers)
|
|
dxf_effect.affect(args = [in_path], output = False)
|
|
|
|
with open(out_path, 'w') as file:
|
|
dxf_effect.write_dxf(file)
|
|
|
|
|
|
def _get_inkscape_layers(svg_path):
|
|
document = xml.etree.ElementTree.parse(svg_path)
|
|
|
|
def iter_layers():
|
|
nodes = document.findall(
|
|
'{http://www.w3.org/2000/svg}g[@{http://www.inkscape.org/namespaces/inkscape}groupmode="layer"]')
|
|
|
|
for i in nodes:
|
|
inkscape_name = i.get('{http://www.inkscape.org/namespaces/inkscape}label').strip()
|
|
|
|
if inkscape_name.endswith(']'):
|
|
dxf_name, args = inkscape_name[:-1].rsplit('[', 1)
|
|
|
|
dxf_name = dxf_name.strip()
|
|
args = args.strip()
|
|
|
|
use_paths = 'p' in args
|
|
else:
|
|
use_paths = False
|
|
dxf_name = inkscape_name
|
|
|
|
yield effect.Layer(inkscape_name, dxf_name, use_paths = use_paths)
|
|
|
|
return list(iter_layers())
|
|
|
|
|
|
def _inkscape(svg_path, verbs):
|
|
def iter_args():
|
|
yield os.environ['INKSCAPE']
|
|
|
|
for i in verbs:
|
|
yield '--verb'
|
|
yield i
|
|
|
|
yield svg_path
|
|
|
|
util.command(list(iter_args()))
|
|
|
|
|
|
def _unfuck_svg_document(temp_svg_path, layers):
|
|
"""
|
|
Unfucks an SVG document so is can be processed by the better_dxf_export plugin (or what's left of it).
|
|
"""
|
|
|
|
def iter_inkscape_verbs():
|
|
yield 'LayerUnlockAll'
|
|
yield 'LayerShowAll'
|
|
|
|
# Go to the first layer.
|
|
for _ in layers:
|
|
yield 'LayerPrev'
|
|
|
|
# Copy each layer and flatten it to a single path object.
|
|
for i in layers:
|
|
yield 'LayerDuplicate'
|
|
yield 'EditSelectAll'
|
|
yield 'ObjectToPath'
|
|
yield 'EditSelectAll'
|
|
yield 'SelectionUnGroup'
|
|
|
|
if not i.use_paths:
|
|
yield 'EditSelectAll'
|
|
yield 'StrokeToPath'
|
|
yield 'EditSelectAll'
|
|
yield 'SelectionUnion'
|
|
|
|
yield 'LayerNext'
|
|
|
|
# Go to the first layer again.
|
|
for _ in range(2 * len(layers)):
|
|
yield 'LayerPrev'
|
|
|
|
# Move the flattened shapes to the original layers.
|
|
for _ in layers:
|
|
yield 'EditSelectAll'
|
|
yield 'EditDelete'
|
|
yield 'LayerNext'
|
|
|
|
yield 'EditSelectAll'
|
|
yield 'LayerMoveToPrev'
|
|
yield 'LayerNext'
|
|
yield 'LayerDelete'
|
|
|
|
yield 'FileSave'
|
|
yield 'FileClose'
|
|
yield 'FileQuit'
|
|
|
|
_inkscape(temp_svg_path, list(iter_inkscape_verbs()))
|
|
|
|
|
|
def main(in_path, out_path):
|
|
with util.TemporaryDirectory() as temp_dir:
|
|
temp_svg_path = os.path.join(temp_dir, 'temp.svg')
|
|
|
|
shutil.copyfile(in_path, temp_svg_path)
|
|
|
|
layers = _get_inkscape_layers(temp_svg_path)
|
|
_unfuck_svg_document(temp_svg_path, layers)
|
|
|
|
_export_dxf(temp_svg_path, out_path, layers)
|
|
|
|
|
|
try:
|
|
main(*sys.argv[1:])
|
|
except util.UserError as e:
|
|
print 'Error:', e
|
|
sys.exit(1)
|
|
except KeyboardInterrupt:
|
|
sys.exit(2)
|