embroider_input.py: input extension to read embroidery formats

1 files changed, 51 insertions, 0 deletions

diff --git a/embroider_input.py b/embroider_input.py
new file mode 100644
index 00000000..39a0e424
--- /dev/null
+++ b/embroider_input.py
@@ -0,0 +1,51 @@
+import sys
+from libembroidery import *
+from inkex import etree
+import inkex
+from inkstitch import PIXELS_PER_MM, _
+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)
+
+    dimensions = stitch_plan.dimensions
+    svg = etree.Element("svg", nsmap=inkex.NSS, attrib=
+                        {
+                            "width": "%s" % dimensions[0],
+                            "height": "%s" % dimensions[1],
+                            "viewBox": "0 0 %s %s" % dimensions,
+                        })
+    render_stitch_plan(svg, stitch_plan)
+
+    print etree.tostring(svg)
+
+if __name__ == '__main__':
+    sys.exit(main(*sys.argv[1:]))