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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
import os
import sys
import tempfile
from zipfile import ZipFile
import inkex
import pyembroidery
from ..i18n import _
from ..output import write_embroidery_file
from ..stitch_plan import patches_to_stitch_plan
from ..svg import PIXELS_PER_MM
from .base import InkstitchExtension
from ..threads import ThreadCatalog
class Zip(InkstitchExtension):
def __init__(self, *args, **kwargs):
InkstitchExtension.__init__(self)
self.OptionParser.add_option("-c", "--collapse_len_mm",
action="store", type="float",
dest="collapse_length_mm", default=3.0,
help="max collapse length (mm)")
# it's kind of obnoxious that I have to do this...
self.formats = []
for format in pyembroidery.supported_formats():
if 'writer' in format and format['category'] == 'embroidery':
extension = format['extension']
self.OptionParser.add_option('--format-%s' % extension, type="inkbool", dest=extension)
self.formats.append(extension)
self.OptionParser.add_option('--format-svg', type="inkbool", dest='svg')
self.OptionParser.add_option('--format-threadlist', type="inkbool", dest='threadlist')
self.formats.append('svg')
self.formats.append('threadlist')
def effect(self):
if not self.get_elements():
return
patches = self.elements_to_patches(self.elements)
stitch_plan = patches_to_stitch_plan(patches, self.options.collapse_length_mm * PIXELS_PER_MM)
base_file_name = self.get_base_file_name()
path = tempfile.mkdtemp()
files = []
for format in self.formats:
if getattr(self.options, format):
output_file = os.path.join(path, "%s.%s" % (base_file_name, format))
if format == 'svg':
output = open(output_file, 'w')
output.write(inkex.etree.tostring(self.document.getroot()))
output.close()
if format == 'threadlist':
output_file = os.path.join(path, "%s_%s.txt" % (base_file_name, _("threadlist")))
output = open(output_file, 'w')
output.write(self.get_threadlist(stitch_plan, base_file_name))
output.close()
else:
write_embroidery_file(output_file, stitch_plan, self.document.getroot())
files.append(output_file)
if not files:
self.errormsg(_("No embroidery file formats selected."))
temp_file = tempfile.NamedTemporaryFile(suffix=".zip", delete=False)
# in windows, failure to close here will keep the file locked
temp_file.close()
with ZipFile(temp_file.name, "w") as zip_file:
for file in files:
zip_file.write(file, os.path.basename(file))
# inkscape will read the file contents from stdout and copy
# to the destination file that the user chose
with open(temp_file.name) as output_file:
sys.stdout.write(output_file.read())
os.remove(temp_file.name)
for file in files:
os.remove(file)
os.rmdir(path)
# don't let inkex output the SVG!
sys.exit(0)
def get_threadlist(self, stitch_plan, design_name):
ThreadCatalog().match_and_apply_palette(stitch_plan, self.get_inkstitch_metadata()['thread-palette'])
thread_used = []
thread_output = "%s\n" % _("Design Details")
thread_output += "==============\n\n"
thread_output += "%s: %s\n" % (_("Title"), design_name)
thread_output += "%s (mm): %.2f x %.2f\n" % (_("Size"), stitch_plan.dimensions_mm[0], stitch_plan.dimensions_mm[1])
thread_output += "%s: %s\n" % (_("Stitches"), stitch_plan.num_stitches)
thread_output += "%s: %s\n\n" % (_("Colors"), stitch_plan.num_colors)
thread_output += "%s\n" % _("Thread Order")
thread_output += "============\n\n"
for i, color_block in enumerate(stitch_plan):
thread = color_block.color
thread_output += str(i + 1) + " "
string = "%s #%s - %s (#%s)" % (thread.name, thread.number, thread.manufacturer, thread.hex_digits.lower())
thread_output += string + "\n"
thread_used.append(string)
thread_output += "\n"
thread_output += _("Thread Used") + "\n"
thread_output += "============" + "\n\n"
for thread in set(thread_used):
thread_output += thread + "\n"
return "%s" % thread_output
|
