inkstitch/lib/extensions/input.py

import os
from os.path import realpath, dirname, join as path_join
import sys
from inkex import etree
import inkex

# help python find libembroidery when running in a local repo clone
if getattr(sys, 'frozen', None) is None:
    sys.path.append(realpath(path_join(dirname(__file__), '..', '..')))

from libembroidery import *

from ..svg import PIXELS_PER_MM, render_stitch_plan
from ..svg.tags import INKSCAPE_LABEL
from ..i18n import _
from ..stitch_plan import StitchPlan
from ..utils.io import save_stdout


class Input(object):
    def pattern_stitches(self, pattern):
        stitch_pointer = pattern.stitchList
        while stitch_pointer:
            yield stitch_pointer.stitch
            stitch_pointer = stitch_pointer.next


    def affect(self, args):
        # libembroidery likes to dump a bunch of debugging stuff to stdout
        save_stdout()

        embroidery_file = args[0]
        pattern = embPattern_create()
        embPattern_read(pattern, embroidery_file)
        embPattern_flipVertical(pattern)

        stitch_plan = StitchPlan()
        color_block = None
        current_color = None

        for stitch in self.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,
                                       color_change=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 >> sys.real_stdout, etree.tostring(svg)