kopia lustrzana https://github.com/inkstitch/inkstitch
62 wiersze
2.2 KiB
Python
62 wiersze
2.2 KiB
Python
import sys
|
|
import os
|
|
from libembroidery import *
|
|
from inkex import etree
|
|
import inkex
|
|
from inkstitch import PIXELS_PER_MM, INKSCAPE_LABEL, _
|
|
from inkstitch.stitch_plan import StitchPlan
|
|
from inkstitch.svg import render_stitch_plan
|
|
|
|
|
|
def pattern_stitches(pattern):
|
|
stitch_pointer = pattern.stitchList
|
|
while stitch_pointer:
|
|
yield stitch_pointer.stitch
|
|
stitch_pointer = stitch_pointer.next
|
|
|
|
|
|
def main(embroidery_file):
|
|
pattern = embPattern_create()
|
|
embPattern_read(pattern, embroidery_file)
|
|
embPattern_flipVertical(pattern)
|
|
|
|
stitch_plan = StitchPlan()
|
|
color_block = None
|
|
current_color = None
|
|
|
|
for stitch in pattern_stitches(pattern):
|
|
if stitch.color != current_color:
|
|
thread = embThreadList_getAt(pattern.threadList, stitch.color)
|
|
color = thread.color
|
|
color_block = stitch_plan.new_color_block((color.r, color.g, color.b))
|
|
current_color = stitch.color
|
|
|
|
if not stitch.flags & END:
|
|
color_block.add_stitch(stitch.xx * PIXELS_PER_MM, stitch.yy * PIXELS_PER_MM,
|
|
jump=stitch.flags & JUMP,
|
|
stop=stitch.flags & STOP,
|
|
trim=stitch.flags & TRIM)
|
|
|
|
extents = stitch_plan.extents
|
|
svg = etree.Element("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),
|
|
})
|
|
render_stitch_plan(svg, stitch_plan)
|
|
|
|
# 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, 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]))
|
|
|
|
print etree.tostring(svg)
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main(*sys.argv[1:]))
|