simulator standalone mode

2 files changed, 23 insertions, 1 deletions

diff --git a/lib/stitch_plan/__init__.py b/lib/stitch_plan/__init__.py
index 791a5f20..2aaa0ab9 100644
--- a/lib/stitch_plan/__init__.py
+++ b/lib/stitch_plan/__init__.py
@@ -1,2 +1,3 @@
-from stitch_plan import patches_to_stitch_plan, StitchPlan, ColorBlock
+from .stitch_plan import patches_to_stitch_plan, StitchPlan, ColorBlock
 from .stitch import Stitch
+from .read_file import stitch_plan_from_file
diff --git a/lib/stitch_plan/read_file.py b/lib/stitch_plan/read_file.py
new file mode 100644
index 00000000..ff5a68ac
--- /dev/null
+++ b/lib/stitch_plan/read_file.py
@@ -0,0 +1,21 @@
+import pyembroidery
+from .stitch_plan import StitchPlan
+
+from ..svg import PIXELS_PER_MM
+
+
+def stitch_plan_from_file(embroidery_file):
+    """Read a machine embroidery file in any supported format and return a stitch plan."""
+    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:
+            color_block.add_stitch(x * PIXELS_PER_MM / 10.0, y * PIXELS_PER_MM / 10.0,
+                                   jump=(command == pyembroidery.JUMP),
+                                   trim=(command == pyembroidery.TRIM))
+
+    return stitch_plan