1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
# Authors: see git history
#
# Copyright (c) 2010 Authors
# Licensed under the GNU GPL version 3.0 or later. See the file LICENSE for details.
import os
import sys
from html import escape
import inkex
import pyembroidery
from ..i18n import _
from ..svg import PIXELS_PER_MM, render_stitch_plan
from ..svg.tags import INKSCAPE_LABEL
from .stitch import Stitch
from .stitch_plan import StitchPlan
def generate_stitch_plan(embroidery_file, import_commands="symbols"): # noqa: C901
validate_file_path(embroidery_file)
pattern = pyembroidery.read(embroidery_file)
stitch_plan = StitchPlan()
color_block = None
for raw_stitches, thread in pattern.get_as_colorblocks():
color_block = stitch_plan.new_color_block(thread)
for x, y, command in raw_stitches:
if command == pyembroidery.STITCH:
color_block.add_stitch(Stitch(x * PIXELS_PER_MM / 10.0, y * PIXELS_PER_MM / 10.0))
if len(color_block) > 0:
if import_commands == "none" and command in [pyembroidery.TRIM, pyembroidery.STOP]:
# Importing commands is not wanted:
# start a new color block without inserting the command
color_block = stitch_plan.new_color_block(thread)
elif command == pyembroidery.TRIM:
color_block.add_stitch(trim=True)
elif command == pyembroidery.STOP:
color_block.add_stitch(stop=True)
color_block = stitch_plan.new_color_block(thread)
stitch_plan.delete_empty_color_blocks()
if stitch_plan.last_color_block:
if stitch_plan.last_color_block.last_stitch:
if stitch_plan.last_color_block.last_stitch.stop:
# ending with a STOP command is redundant, so remove it
del stitch_plan.last_color_block[-1]
extents = stitch_plan.extents
svg = inkex.SvgDocumentElement("svg", nsmap=inkex.NSS, attrib={
"width": str(extents[0] * 2),
"height": str(extents[1] * 2),
"viewBox": "0 0 %s %s" % (extents[0] * 2, extents[1] * 2),
})
visual_commands = True if import_commands == "symbols" else False
render_stitch_plan(svg, stitch_plan, visual_commands=visual_commands)
# rename the Stitch Plan layer so that it doesn't get overwritten by Embroider
layer = svg.find(".//*[@id='__inkstitch_stitch_plan__']")
layer.set(INKSCAPE_LABEL, escape(os.path.basename(embroidery_file)))
layer.attrib.pop('id')
# Shift the design so that its origin is at the center of the canvas
# Note: this is NOT the same as centering the design in the canvas!
layer.set('transform', 'translate(%s,%s)' % (extents[0], extents[1]))
return svg
def validate_file_path(path):
# Check if the file exists
if not os.path.isfile(path):
inkex.errormsg(_('File does not exist and cannot be opened. Please correct the file path and try again.\r%s') % path)
sys.exit(1)
|
