add stitch plan extension (#640)

1 files changed, 24 insertions, 0 deletions

diff --git a/lib/extensions/stitch_plan_preview.py b/lib/extensions/stitch_plan_preview.py
new file mode 100644
index 00000000..b89e24a7
--- /dev/null
+++ b/lib/extensions/stitch_plan_preview.py
@@ -0,0 +1,24 @@
+from ..stitch_plan import patches_to_stitch_plan
+from ..svg import render_stitch_plan
+from .base import InkstitchExtension
+
+
+class StitchPlanPreview(InkstitchExtension):
+
+    def effect(self):
+        # delete old stitch plan
+        svg = self.document.getroot()
+        layer = svg.find(".//*[@id='__inkstitch_stitch_plan__']")
+        if layer is not None:
+            del layer[:]
+
+        # create new stitch plan
+        if not self.get_elements():
+            return
+        patches = self.elements_to_patches(self.elements)
+        stitch_plan = patches_to_stitch_plan(patches)
+        render_stitch_plan(svg, stitch_plan)
+
+        # translate stitch plan to the right side of the canvas
+        layer = svg.find(".//*[@id='__inkstitch_stitch_plan__']")
+        layer.set('transform', 'translate(%s)' % svg.get('viewBox', '0 0 800 0').split(' ')[2])