Merge pull request #163 from lexelby/lexelby-single-extension

single code entry point

1 files changed, 68 insertions, 0 deletions

diff --git a/lib/extensions/input.py b/lib/extensions/input.py
new file mode 100644
index 00000000..f8bf5a5d
--- /dev/null
+++ b/lib/extensions/input.py
@@ -0,0 +1,68 @@
+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
+
+
+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):
+        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,
+                                       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)