diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/extensions/__init__.py | 3 | ||||
| -rw-r--r-- | lib/extensions/stitch_plan_preview.py | 24 |
2 files changed, 26 insertions, 1 deletions
diff --git a/lib/extensions/__init__.py b/lib/extensions/__init__.py index 38276a64..cd8d79fb 100644 --- a/lib/extensions/__init__.py +++ b/lib/extensions/__init__.py @@ -15,10 +15,11 @@ from params import Params from print_pdf import Print from remove_embroidery_settings import RemoveEmbroiderySettings from simulate import Simulate +from stitch_plan_preview import StitchPlanPreview from zip import Zip - __all__ = extensions = [Embroider, + StitchPlanPreview, Install, Params, Print, 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]) |