From 88ba830a991ae68ee1a11c989961da78fbb3ef5c Mon Sep 17 00:00:00 2001
From: Kaalleen <36401965+kaalleen@users.noreply.github.com>
Date: Sun, 22 Mar 2020 09:16:28 +0100
Subject: add stitch plan extension (#640)

---
 lib/extensions/stitch_plan_preview.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 lib/extensions/stitch_plan_preview.py

(limited to 'lib/extensions/stitch_plan_preview.py')

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])
-- 
cgit v1.2.3