add Simulate effect to visualize stitching order

embroider_simulate.inx

@@ -0,0 +1,17 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<inkscape-extension xmlns="http://www.inkscape.org/namespace/inkscape/extension">
+    <_name>Simulate</_name>
+    <id>jonh.embroider.simulate</id>
+    <dependency type="executable" location="extensions">embroider_simulate.py</dependency>
+    <dependency type="executable" location="extensions">inkex.py</dependency>
+    <param name="path" type="string" _gui-text="Directory where embroidery output is stored"></param>
+    <effect>
+        <object-type>all</object-type>
+        <effects-menu>
+            <submenu _name="Embroidery"/>
+        </effects-menu>
+    </effect>
+    <script>
+        <command reldir="extensions" interpreter="python">embroider_simulate.py</command>
+    </script>
+</inkscape-extension>

embroider_simulate.py

@@ -0,0 +1,196 @@
+import sys
+import os
+import numpy
+import wx
+from wx.lib.floatcanvas.FloatCanvas import FloatCanvas
+import inkex
+
+class EmbroiderySimulator(wx.Frame):
+    def __init__(self, *args, **kwargs):
+        stitch_file = kwargs.pop('stitch_file')
+        self.frame_period = kwargs.pop('frame_period', 10)
+        self.stitches_per_frame = kwargs.pop('stitches_per_frame', 1)
+
+        wx.Frame.__init__(self, *args, **kwargs)
+        self.panel = wx.Panel(self, wx.ID_ANY)
+        self.panel.SetFocus()
+
+        self.stitches = self._parse_stitches(stitch_file)
+        self.width, self.height = self.get_dimensions()
+
+        self.buffer = wx.EmptyBitmap(self.width, self.height)
+        self.dc = wx.MemoryDC()
+        self.dc.SelectObject(self.buffer)
+        self.canvas = wx.GraphicsContext.Create(self.dc)
+
+        self.clear()
+
+        self.Bind(wx.EVT_SIZE, self.on_size)
+        self.panel.Bind(wx.EVT_PAINT, self.on_paint)
+        self.panel.Bind(wx.EVT_KEY_DOWN, self.on_key_down)
+
+        self.last_pos = None
+
+    def on_key_down(self, event):
+        keycode = event.GetKeyCode()
+
+        if keycode == ord("+") or keycode == ord("=") or keycode == wx.WXK_UP:
+            if self.frame_period == 1:
+                self.stitches_per_frame *= 2
+            else:
+                self.frame_period = self.frame_period / 2
+        elif keycode == ord("-") or keycode == ord("_") or keycode == wx.WXK_DOWN:
+            if self.stitches_per_frame == 1:
+                self.frame_period *= 2
+            else:
+                self.stitches_per_frame /= 2
+        elif keycode == ord("Q"):
+            self.Close()
+        elif keycode == ord('P'):
+            if self.timer.IsRunning():
+                self.timer.Stop()
+            else:
+                self.timer.Start(self.frame_period)
+
+        self.frame_period = max(1, self.frame_period)
+        self.stitches_per_frame = max(self.stitches_per_frame, 1)
+
+        if self.timer.IsRunning():
+            self.timer.Stop()
+            self.timer.Start(self.frame_period)
+
+    def _strip_quotes(self, string):
+        if string.startswith('"') and string.endswith('"'):
+            string = string[1:-1]
+
+        return string
+
+    def _parse_stitches(self, stitch_file_path):
+        # "$","1","229","229","229","(null)","(null)"
+        # "*","JUMP","1.595898","48.731899"
+        # "*","STITCH","1.595898","48.731899"
+
+        stitches = []
+
+        pos = (0, 0)
+        color = wx.Brush('black')
+        cut = True
+
+        with open(stitch_file_path) as stitch_file:
+            for line in stitch_file:
+                fields = line.strip().split(",")
+                fields = [self._strip_quotes(field) for field in fields]
+
+                symbol, command = fields[:2]
+
+                if symbol == "$":
+                    red, green, blue = fields[2:5]
+                    color = wx.Pen((int(red), int(green), int(blue)))
+                elif symbol == "*":
+                    if command == "COLOR":
+                        # change color
+                        # The next command should be a JUMP, and we'll need to skip stitching.
+                        cut = True
+                    elif command == "JUMP" or command == "STITCH":
+                        # JUMP just means a long stitch, really.
+
+                        x, y = fields[2:]
+                        new_pos = (int(float(x) * 10), int(float(y) * 10))
+
+                        if not cut:
+                            stitches.append(((pos, new_pos), color))
+
+                        cut = False
+                        pos = new_pos
+
+        return stitches
+
+    def get_dimensions(self):
+        width = 0
+        height = 0
+
+        for stitch in self.stitches:
+            (start_x, start_y), (end_x, end_y) = stitch[0]
+
+            width = max(width, start_x, end_x)
+            height = max(height, start_y, end_y)
+
+        return width, height
+
+    def go(self):
+        self.current_stitch = 0
+        self.timer = wx.PyTimer(self.draw_one_stitch)
+        self.timer.Start(self.frame_period)
+
+    def clear(self):
+        self.dc.SetBackground(wx.Brush('white'))
+        self.dc.Clear()
+
+    def on_size(self, e):
+        # ensure that the whole canvas is visible
+        window_width, window_height = self.GetSize()
+        client_width, client_height = self.GetClientSize()
+
+        decorations_width = window_width - client_width
+        decorations_height = window_height - client_height
+
+        self.SetSize((self.width + decorations_width, self.height + decorations_height))
+
+        e.Skip()
+
+    def on_paint(self, e):
+        dc = wx.PaintDC(self.panel)
+        dc.DrawBitmap(self.buffer, 0, 0)
+
+        if self.last_pos:
+            dc.DrawLine(self.last_pos[0] - 10, self.last_pos[1],      self.last_pos[0] + 10, self.last_pos[1])
+            dc.DrawLine(self.last_pos[0],      self.last_pos[1] - 10, self.last_pos[0],      self.last_pos[1] + 10)
+
+    def redraw(self):
+        dc = wx.ClientDC(self)
+        dc.DrawBitmap(self.buffer, 0, 0)
+
+    def draw_one_stitch(self):
+        for i in xrange(self.stitches_per_frame):
+            try:
+                ((x1, y1), (x2, y2)), color = self.stitches[self.current_stitch]
+                y1 = self.height - y1
+                y2 = self.height - y2
+
+                self.canvas.SetPen(color)
+                self.canvas.DrawLines(((x1, y1), (x2, y2)))
+                self.Refresh()
+
+                self.current_stitch += 1
+                self.last_pos = (x2, y2)
+            except IndexError:
+                self.timer.Stop()
+
+class SimulateEffect(inkex.Effect):
+    def __init__(self):
+        inkex.Effect.__init__(self)
+        self.OptionParser.add_option("-P", "--path",
+                                     action="store", type="string",
+                                     dest="path", default=".",
+                                     help="Directory in which to store output file")
+
+    def effect(self):
+        app = wx.App()
+        frame = EmbroiderySimulator(None, -1, "Embroidery Simulation", wx.DefaultPosition, size=(1000, 1000), stitch_file=self.get_stitch_file())
+        app.SetTopWindow(frame)
+        frame.Show()
+        wx.CallAfter(frame.go)
+        app.MainLoop()
+
+    def get_stitch_file(self):
+        svg_filename = self.document.getroot().get(inkex.addNS('docname', 'sodipodi'))
+        csv_filename = svg_filename.replace('.svg', '.csv')
+        stitch_file = os.path.join(self.options.path, csv_filename)
+
+        return stitch_file
+
+
+if __name__ == "__main__":
+    effect = SimulateEffect()
+    effect.affect()
+    sys.exit(0)