inkstitch/embroider_input.py

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:]))