1 files changed, 74 insertions, 0 deletions
diff --git a/lib/stitch_plan/generate_stitch_plan.py b/lib/stitch_plan/generate_stitch_plan.py
new file mode 100644
index 00000000..2d8ceeff
--- /dev/null
+++ b/lib/stitch_plan/generate_stitch_plan.py
@@ -0,0 +1,74 @@
+# Authors: see git history
+#
+# Copyright (c) 2010 Authors
+# Licensed under the GNU GPL version 3.0 or later.  See the file LICENSE for details.
+
+import os
+import sys
+
+import inkex
+
+import pyembroidery
+
+from ..i18n import _
+from ..svg import PIXELS_PER_MM, render_stitch_plan
+from ..svg.tags import INKSCAPE_LABEL
+from .stitch import Stitch
+from .stitch_plan import StitchPlan
+
+
+def generate_stitch_plan(embroidery_file, import_commands=True):  # noqa: C901
+    validate_file_path(embroidery_file)
+    pattern = pyembroidery.read(embroidery_file)
+    stitch_plan = StitchPlan()
+    color_block = None
+
+    for raw_stitches, thread in pattern.get_as_colorblocks():
+        color_block = stitch_plan.new_color_block(thread)
+        for x, y, command in raw_stitches:
+            if command == pyembroidery.STITCH:
+                color_block.add_stitch(Stitch(x * PIXELS_PER_MM / 10.0, y * PIXELS_PER_MM / 10.0))
+            if len(color_block) > 0:
+                if not import_commands and command in [pyembroidery.TRIM, pyembroidery.STOP]:
+                    # Importing commands is not wanted:
+                    # start a new color block without inserting the command
+                    color_block = stitch_plan.new_color_block(thread)
+                elif command == pyembroidery.TRIM:
+                    color_block.add_stitch(trim=True)
+                elif command == pyembroidery.STOP:
+                    color_block.add_stitch(stop=True)
+                    color_block = stitch_plan.new_color_block(thread)
+
+    stitch_plan.delete_empty_color_blocks()
+
+    if stitch_plan.last_color_block:
+        if stitch_plan.last_color_block.last_stitch:
+            if stitch_plan.last_color_block.last_stitch.stop:
+                # ending with a STOP command is redundant, so remove it
+                del stitch_plan.last_color_block[-1]
+
+    extents = stitch_plan.extents
+    svg = inkex.SvgDocumentElement("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]))
+
+    return svg
+
+
+def validate_file_path(path):
+    # Check if the file exists
+    if not os.path.isfile(path):
+        inkex.errormsg(_('File does not exist and cannot be opened. Please correct the file path and try again.\r%s') % path)
+        sys.exit(1)