kopia lustrzana https://github.com/inkstitch/inkstitch
89 wiersze
3.2 KiB
Python
89 wiersze
3.2 KiB
Python
import sys
|
|
import traceback
|
|
import os
|
|
import inkex
|
|
import tempfile
|
|
from zipfile import ZipFile
|
|
from libembroidery import *
|
|
|
|
from .base import InkstitchExtension
|
|
from ..i18n import _
|
|
from ..output import write_embroidery_file
|
|
from ..stitch_plan import patches_to_stitch_plan
|
|
from ..svg import render_stitch_plan, PIXELS_PER_MM
|
|
from ..utils.io import save_stdout
|
|
|
|
|
|
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 = []
|
|
formatList = embFormatList_create()
|
|
curFormat = formatList
|
|
while(curFormat):
|
|
# extension includes the dot, so we'll remove it
|
|
extension = embFormat_extension(curFormat)[1:]
|
|
description = embFormat_description(curFormat)
|
|
writer_state = embFormat_writerState(curFormat)
|
|
|
|
if writer_state.strip() and embFormat_type(curFormat) != EMBFORMAT_OBJECTONLY:
|
|
self.OptionParser.add_option('--format-%s' % extension, type="inkbool", dest=extension)
|
|
self.formats.append(extension)
|
|
curFormat = curFormat.next
|
|
|
|
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 = []
|
|
|
|
# libembroidery likes to debug log things to stdout. No way to disable it.
|
|
save_stdout()
|
|
for format in self.formats:
|
|
if getattr(self.options, format):
|
|
output_file = os.path.join(path, "%s.%s" % (base_file_name, format))
|
|
write_embroidery_file(output_file, stitch_plan, self.document.getroot())
|
|
files.append(output_file)
|
|
|
|
# I'd love to do restore_stderr() here, but if I do, libembroidery's
|
|
# stuff still prints out and corrupts the zip! That's because it uses
|
|
# C's buffered stdout, so it hasn't actually written anything to the
|
|
# real standard output yet.
|
|
|
|
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.real_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)
|