From a736f7293ff1df73cf3ac653d966aa8c7390e2b8 Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Thu, 28 Dec 2017 15:39:56 -0500 Subject: rename draw_one_stitch to be more accurate --- embroider_simulate.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/embroider_simulate.py b/embroider_simulate.py index 2629c812..13a84e8a 100644 --- a/embroider_simulate.py +++ b/embroider_simulate.py @@ -118,7 +118,7 @@ class EmbroiderySimulator(wx.Frame): def go(self): self.current_stitch = 0 - self.timer = wx.PyTimer(self.draw_one_stitch) + self.timer = wx.PyTimer(self.draw_one_frame) self.timer.Start(self.frame_period) def clear(self): @@ -149,7 +149,7 @@ class EmbroiderySimulator(wx.Frame): dc = wx.ClientDC(self) dc.DrawBitmap(self.buffer, 0, 0) - def draw_one_stitch(self): + def draw_one_frame(self): for i in xrange(self.stitches_per_frame): try: ((x1, y1), (x2, y2)), color = self.stitches[self.current_stitch] -- cgit v1.2.3 From 69590df70cdacb35eae7bdd6ca647b904b52d777 Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Thu, 28 Dec 2017 15:55:43 -0500 Subject: generalize EmbroiderySimulator * add ability to use patches instead of a stitch file * add stop() and load() methods to use a new stitch file / patches list --- embroider_simulate.py | 55 +++++++++++++++++++++++++++++++++++++++++---------- 1 file changed, 45 insertions(+), 10 deletions(-) diff --git a/embroider_simulate.py b/embroider_simulate.py index 13a84e8a..7b096a2d 100644 --- a/embroider_simulate.py +++ b/embroider_simulate.py @@ -4,9 +4,12 @@ import numpy import wx import inkex +from embroider import patches_to_stitches, stitches_to_polylines + class EmbroiderySimulator(wx.Frame): def __init__(self, *args, **kwargs): - stitch_file = kwargs.pop('stitch_file') + stitch_file = kwargs.pop('stitch_file', None) + patches = kwargs.pop('patches', None) self.frame_period = kwargs.pop('frame_period', 80) self.stitches_per_frame = kwargs.pop('stitches_per_frame', 1) @@ -14,10 +17,9 @@ class EmbroiderySimulator(wx.Frame): 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.load(stitch_file, patches) - self.buffer = wx.EmptyBitmap(self.width, self.height) + self.buffer = wx.Bitmap(self.width, self.height) self.dc = wx.MemoryDC() self.dc.SelectObject(self.buffer) self.canvas = wx.GraphicsContext.Create(self.dc) @@ -30,6 +32,16 @@ class EmbroiderySimulator(wx.Frame): self.last_pos = None + def load(self, stitch_file=None, patches=None): + if stitch_file: + self.segments = self._parse_stitch_file(stitch_file) + elif patches: + self.segments = self._patches_to_stitches(patches) + else: + raise TypeError("EmbroiderySimulator requires a stitch_file or list of patches") + + self.width, self.height = self.get_dimensions() + def on_key_down(self, event): keycode = event.GetKeyCode() @@ -64,12 +76,30 @@ class EmbroiderySimulator(wx.Frame): return string - def _parse_stitches(self, stitch_file_path): + def _patches_to_segments(patches): + stitches = patches_to_stitches(patches) + + segments = [] + + last_pos = None + last_color = None + + for stitch in stitches: + if stitch.color == last_color: + pos = (stitch.x, stitch.y) + segments.append(((last_pos, pos), stitch.color)) + + last_pos = pos + last_color = stitch.color + + return segments + + def _parse_stitch_file(self, stitch_file_path): # "$","1","229","229","229","(null)","(null)" # "*","JUMP","1.595898","48.731899" # "*","STITCH","1.595898","48.731899" - stitches = [] + segments = [] pos = (0, 0) color = wx.Brush('black') @@ -97,18 +127,18 @@ class EmbroiderySimulator(wx.Frame): new_pos = (int(float(x) * 10), int(float(y) * 10)) if not cut: - stitches.append(((pos, new_pos), color)) + segments.append(((pos, new_pos), color)) cut = False pos = new_pos - return stitches + return segments def get_dimensions(self): width = 0 height = 0 - for stitch in self.stitches: + for stitch in self.segments: (start_x, start_y), (end_x, end_y) = stitch[0] width = max(width, start_x, end_x) @@ -117,10 +147,15 @@ class EmbroiderySimulator(wx.Frame): return width, height def go(self): + self.clear() + self.current_stitch = 0 self.timer = wx.PyTimer(self.draw_one_frame) self.timer.Start(self.frame_period) + def stop(self): + self.timer.Stop() + def clear(self): self.dc.SetBackground(wx.Brush('white')) self.dc.Clear() @@ -152,7 +187,7 @@ class EmbroiderySimulator(wx.Frame): def draw_one_frame(self): for i in xrange(self.stitches_per_frame): try: - ((x1, y1), (x2, y2)), color = self.stitches[self.current_stitch] + ((x1, y1), (x2, y2)), color = self.segments[self.current_stitch] y1 = self.height - y1 y2 = self.height - y2 -- cgit v1.2.3 From 42b83d4ec47669f8d88180ebb750dd20fe644079 Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Thu, 28 Dec 2017 19:55:04 -0500 Subject: fix trailing whitespace --- embroider_params.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/embroider_params.py b/embroider_params.py index 4e69ac33..8754f273 100644 --- a/embroider_params.py +++ b/embroider_params.py @@ -305,7 +305,7 @@ class SettingsFrame(wx.Frame): def __init__(self, *args, **kwargs): # begin wxGlade: MyFrame.__init__ self.tabs_factory = kwargs.pop('tabs_factory', []) - wx.Frame.__init__(self, None, wx.ID_ANY, + wx.Frame.__init__(self, None, wx.ID_ANY, "Embroidery Params" ) self.notebook = wx.Notebook(self, wx.ID_ANY) @@ -397,10 +397,10 @@ class SettingsFrame(wx.Frame): def _load_preset(self, preset_name): - preset = self.check_and_load_preset(preset_name) + preset = self.check_and_load_preset(preset_name) if not preset: return - + for tab in self.tabs: tab.load_preset(preset) @@ -420,11 +420,11 @@ class SettingsFrame(wx.Frame): if not preset_name: return - preset = self.check_and_load_preset(preset_name) + preset = self.check_and_load_preset(preset_name) if not preset: return - delete_preset(preset_name) + delete_preset(preset_name) self.update_preset_list() self.preset_chooser.SetValue("") @@ -521,7 +521,7 @@ class EmbroiderParams(inkex.Effect): else: getter = 'get_param' - values = filter(lambda item: item is not None, + values = filter(lambda item: item is not None, (getattr(node, getter)(param.name, param.default) for node in nodes)) return values -- cgit v1.2.3 From 61ed1da1cfb39f6350156bf56ce969efa8899891 Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Sat, 30 Dec 2017 16:05:21 -0500 Subject: params simulate window when you change params, a simulate window opens to preview the results right away --- embroider.py | 14 ++++----- embroider_params.py | 86 +++++++++++++++++++++++++++++++++++++++++++++++---- embroider_simulate.py | 36 ++++++++++++++++----- 3 files changed, 114 insertions(+), 22 deletions(-) diff --git a/embroider.py b/embroider.py index 7ad9ee22..392182e4 100644 --- a/embroider.py +++ b/embroider.py @@ -121,7 +121,7 @@ class EmbroideryElement(object): if param.endswith('_mm'): # print >> dbg, "get_float_param", param, value, "*", self.options.pixels_per_mm - value = value * self.options.pixels_per_mm + value = value * getattr(self.options, "pixels_per_mm", 10) return value @@ -133,7 +133,7 @@ class EmbroideryElement(object): return default if param.endswith('_mm'): - value = int(value * self.options.pixels_per_mm) + value = int(value * getattr(self.options, "pixels_per_mm", 10)) return value @@ -207,7 +207,7 @@ class EmbroideryElement(object): path = deepcopy(path) - cspsubdiv(path, self.options.flat) + cspsubdiv(path, getattr(self.options, "flat", 0.1)) flattened = [] @@ -500,7 +500,7 @@ class Fill(EmbroideryElement): # only stitch the first point if it's a reasonable distance away from the # last stitch - if not patch.stitches or (beg - patch.stitches[-1]).length() > 0.5 * self.options.pixels_per_mm: + if not patch.stitches or (beg - patch.stitches[-1]).length() > 0.5 * getattr(self.options, "pixels_per_mm", 10): patch.add_stitch(beg) first_stitch = self.adjust_stagger(beg, angle, row_spacing, max_stitch_length) @@ -515,7 +515,7 @@ class Fill(EmbroideryElement): patch.add_stitch(beg + offset * row_direction) offset += max_stitch_length - if (end - patch.stitches[-1]).length() > 0.1 * self.options.pixels_per_mm: + if (end - patch.stitches[-1]).length() > 0.1 * getattr(self.options, "pixels_per_mm", 10): patch.add_stitch(end) @@ -1000,7 +1000,7 @@ class AutoFill(Fill): patch.add_stitch(PyEmb.Point(*outline.interpolate(pos).coords[0])) end = PyEmb.Point(*end) - if (end - patch.stitches[-1]).length() > 0.1 * self.options.pixels_per_mm: + if (end - patch.stitches[-1]).length() > 0.1 * getattr(self.options, "pixels_per_mm", 10): patch.add_stitch(end) print >> dbg, "end connect_points" @@ -1941,5 +1941,3 @@ if __name__ == '__main__': print >> dbg, traceback.format_exc() dbg.flush() - -dbg.close() diff --git a/embroider_params.py b/embroider_params.py index 8754f273..7d76c9d9 100644 --- a/embroider_params.py +++ b/embroider_params.py @@ -5,6 +5,7 @@ import os import sys import json import traceback +from copy import copy from cStringIO import StringIO import wx from wx.lib.scrolledpanel import ScrolledPanel @@ -13,7 +14,7 @@ import inkex from embroider import Param, EmbroideryElement, Fill, AutoFill, Stroke, SatinColumn, descendants from functools import partial from itertools import groupby - +from embroider_simulate import EmbroiderySimulator def presets_path(): try: @@ -128,8 +129,14 @@ class ParamsTab(ScrolledPanel): def set_parent_tab(self, tab): self.parent_tab = tab + def is_dependent_tab(self): + return self.parent_tab is not None + + def enabled(self): + return self.toggle_checkbox.IsChecked() + def update_toggle_state(self, event=None, notify_pair=True): - enable = self.toggle_checkbox.IsChecked() + enable = self.enabled() # print self.name, "update_toggle_state", enable for child in self.settings_grid.GetChildren(): widget = child.GetWindow() @@ -137,7 +144,7 @@ class ParamsTab(ScrolledPanel): child.GetWindow().Enable(enable) if notify_pair and self.paired_tab: - self.paired_tab.pair_changed(self.toggle_checkbox.IsChecked()) + self.paired_tab.pair_changed(enable) for tab in self.dependent_tabs: tab.dependent_enable(enable) @@ -149,7 +156,7 @@ class ParamsTab(ScrolledPanel): # print self.name, "pair_changed", value new_value = not value - if self.toggle_checkbox.IsChecked() != new_value: + if self.enabled() != new_value: self.set_toggle_state(not value) self.toggle_checkbox.changed = True self.update_toggle_state(notify_pair=False) @@ -170,7 +177,7 @@ class ParamsTab(ScrolledPanel): values = {} if self.toggle: - checked = self.toggle_checkbox.IsChecked() + checked = self.enabled() if self.toggle_checkbox in self.changed_inputs and not self.toggle.inverse: values[self.toggle.name] = checked @@ -192,10 +199,16 @@ class ParamsTab(ScrolledPanel): for name, value in values.iteritems(): node.set_param(name, value) + def on_change(self, callable): + self.on_change_hook = callable + def changed(self, event): self.changed_inputs.add(event.GetEventObject()) event.Skip() + if self.on_change_hook: + self.on_change_hook(self) + def load_preset(self, preset): preset_data = preset.get(self.name, {}) @@ -311,6 +324,11 @@ class SettingsFrame(wx.Frame): self.notebook = wx.Notebook(self, wx.ID_ANY) self.tabs = self.tabs_factory(self.notebook) + for tab in self.tabs: + tab.on_change(self.params_changed) + + self.simulate_window = None + self.presets_box = wx.StaticBox(self, wx.ID_ANY, label="Presets") self.preset_chooser = wx.ComboBox(self, wx.ID_ANY) @@ -341,6 +359,56 @@ class SettingsFrame(wx.Frame): self.__do_layout() # end wxGlade + def params_changed(self, tab): + patches = self.generate_patches() + + if not patches: + return + + if self.simulate_window: + self.simulate_window.stop() + self.simulate_window.load(patches=patches) + else: + my_rect = self.GetRect() + simulator_pos = my_rect.GetTopRight() + simulator_pos.x += 5 + + try: + self.simulate_window = EmbroiderySimulator(None, -1, "Embroidery Simulator", simulator_pos, size=(300, 300), patches=patches, on_close=self.simulate_window_closed) + except: + with open('/tmp/params_debug.log', 'a') as log: + print >> log, traceback.format_exc() + log.flush() + + self.simulate_window.Show() + wx.CallLater(10, self.Raise) + + wx.CallAfter(self.simulate_window.go) + + def simulate_window_closed(self): + self.simulate_window = None + + def generate_patches(self): + patches = [] + + for tab in self.tabs: + tab.apply() + + try: + if tab.enabled() and not tab.is_dependent_tab(): + for node in tab.nodes: + # Making a copy of the embroidery element is an easy + # way to drop the cache in the @cache decorators used + # for many params in embroider.py. + + patches.extend(copy(node).to_patches(None)) + except: + # Ignore errors. This can be things like incorrect paths for + # satins or division by zero caused by incorrect param values. + pass + + return patches + def update_preset_list(self): preset_names = load_presets().keys() preset_names = [preset for preset in preset_names if preset != "__LAST__"] @@ -430,10 +498,12 @@ class SettingsFrame(wx.Frame): event.Skip() - def apply(self, event): + def _apply(self): for tab in self.tabs: tab.apply() + def apply(self, event): + self._apply() save_preset("__LAST__", self.get_preset_data()) self.Close() @@ -442,6 +512,10 @@ class SettingsFrame(wx.Frame): self.apply(event) def close(self, event): + if self.simulate_window: + self.simulate_window.stop() + self.simulate_window.Close() + self.Close() def __set_properties(self): diff --git a/embroider_simulate.py b/embroider_simulate.py index 7b096a2d..d80dbe6e 100644 --- a/embroider_simulate.py +++ b/embroider_simulate.py @@ -3,6 +3,7 @@ import os import numpy import wx import inkex +import simplestyle from embroider import patches_to_stitches, stitches_to_polylines @@ -10,10 +11,12 @@ class EmbroiderySimulator(wx.Frame): def __init__(self, *args, **kwargs): stitch_file = kwargs.pop('stitch_file', None) patches = kwargs.pop('patches', None) + self.on_close_hook = kwargs.pop('on_close', None) self.frame_period = kwargs.pop('frame_period', 80) 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() @@ -30,15 +33,19 @@ class EmbroiderySimulator(wx.Frame): self.panel.Bind(wx.EVT_PAINT, self.on_paint) self.panel.Bind(wx.EVT_KEY_DOWN, self.on_key_down) + self.timer = None + self.last_pos = None + self.Bind(wx.EVT_CLOSE, self.on_close) + def load(self, stitch_file=None, patches=None): if stitch_file: self.segments = self._parse_stitch_file(stitch_file) elif patches: - self.segments = self._patches_to_stitches(patches) + self.segments = self._patches_to_segments(patches) else: - raise TypeError("EmbroiderySimulator requires a stitch_file or list of patches") + return self.width, self.height = self.get_dimensions() @@ -76,18 +83,22 @@ class EmbroiderySimulator(wx.Frame): return string - def _patches_to_segments(patches): + def _patches_to_segments(self, patches): stitches = patches_to_stitches(patches) segments = [] last_pos = None last_color = None + pen = None for stitch in stitches: + pos = (stitch.x, stitch.y) + if stitch.color == last_color: - pos = (stitch.x, stitch.y) - segments.append(((last_pos, pos), stitch.color)) + segments.append(((last_pos, pos), pen)) + else: + pen = wx.Pen(simplestyle.parseColor(stitch.color)) last_pos = pos last_color = stitch.color @@ -138,8 +149,8 @@ class EmbroiderySimulator(wx.Frame): width = 0 height = 0 - for stitch in self.segments: - (start_x, start_y), (end_x, end_y) = stitch[0] + for segment in self.segments: + (start_x, start_y), (end_x, end_y) = segment[0] width = max(width, start_x, end_x) height = max(height, start_y, end_y) @@ -153,8 +164,17 @@ class EmbroiderySimulator(wx.Frame): self.timer = wx.PyTimer(self.draw_one_frame) self.timer.Start(self.frame_period) + def on_close(self, event): + self.stop() + + if self.on_close_hook: + self.on_close_hook() + + self.Destroy() + def stop(self): - self.timer.Stop() + if self.timer: + self.timer.Stop() def clear(self): self.dc.SetBackground(wx.Brush('white')) -- cgit v1.2.3 From 7152caa14d9c0ec912b5326fde58c0c410e45391 Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Sat, 30 Dec 2017 20:54:35 -0500 Subject: render in a background thread The problem with this approach is that sometimes the SystemExit happens down inside shapely and it complains bitterly (on stderr). May have to rethink. --- embroider_params.py | 57 +++++++++++++++++++++++++++++++++++++++++++++++++-- embroider_simulate.py | 2 ++ 2 files changed, 57 insertions(+), 2 deletions(-) diff --git a/embroider_params.py b/embroider_params.py index 7d76c9d9..58feafcb 100644 --- a/embroider_params.py +++ b/embroider_params.py @@ -5,6 +5,8 @@ import os import sys import json import traceback +import time +from threading import Thread from copy import copy from cStringIO import StringIO import wx @@ -16,6 +18,40 @@ from functools import partial from itertools import groupby from embroider_simulate import EmbroiderySimulator +class KillableThread(Thread): + """A subclass of threading.Thread, with a kill() method.""" + def __init__(self, *args, **keywords): + Thread.__init__(self, *args, **keywords) + self.killed = False + + def start(self): + """Start the thread.""" + self.__run_backup = self.run + self.run = self.__run # Force the Thread to install our trace. + Thread.start(self) + + def __run(self): + """Hacked run function, which installs the trace.""" + sys.settrace(self.globaltrace) + self.__run_backup() + self.run = self.__run_backup + + def globaltrace(self, frame, why, arg): + if why == 'call': + return self.localtrace + else: + return None + + def localtrace(self, frame, why, arg): + if self.killed: + if why == 'line': + raise SystemExit() + return self.localtrace + + def kill(self): + self.killed = True + + def presets_path(): try: import appdirs @@ -328,6 +364,8 @@ class SettingsFrame(wx.Frame): tab.on_change(self.params_changed) self.simulate_window = None + self.simulate_thread = None + self.thread_num = 0 self.presets_box = wx.StaticBox(self, wx.ID_ANY, label="Presets") @@ -360,11 +398,24 @@ class SettingsFrame(wx.Frame): # end wxGlade def params_changed(self, tab): + if self.simulate_window: + self.simulate_window.stop() + self.simulate_window.clear() + + if self.simulate_thread and self.simulate_thread.is_alive(): + self.simulate_thread.kill() + + self.simulate_thread = KillableThread(target=self.update_patches, name="Simulate%d" % self.thread_num) + self.simulate_thread.start() + self.thread_num += 1 + + def update_patches(self): patches = self.generate_patches() - if not patches: - return + if patches: + wx.CallAfter(self.update_simulator, patches) + def update_simulator(self, patches): if self.simulate_window: self.simulate_window.stop() self.simulate_window.load(patches=patches) @@ -402,6 +453,8 @@ class SettingsFrame(wx.Frame): # for many params in embroider.py. patches.extend(copy(node).to_patches(None)) + except SystemExit: + raise except: # Ignore errors. This can be things like incorrect paths for # satins or division by zero caused by incorrect param values. diff --git a/embroider_simulate.py b/embroider_simulate.py index d80dbe6e..6db203ad 100644 --- a/embroider_simulate.py +++ b/embroider_simulate.py @@ -179,6 +179,8 @@ class EmbroiderySimulator(wx.Frame): def clear(self): self.dc.SetBackground(wx.Brush('white')) self.dc.Clear() + self.last_pos = None + self.Refresh() def on_size(self, e): # ensure that the whole canvas is visible -- cgit v1.2.3 From cb6df397738c3a926a55c6d2e96c32787c46f4c4 Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Sat, 30 Dec 2017 21:15:34 -0500 Subject: get rid of thread killing Switching to a cooperative model. This has the downside that a long patch computation will cause the simulate window to take longer to update, but in practice the longest this will be is a couple of seconds (for a very complicated autofill region). --- embroider_params.py | 58 ++++++++++++++++------------------------------------- 1 file changed, 17 insertions(+), 41 deletions(-) diff --git a/embroider_params.py b/embroider_params.py index 58feafcb..fc472d36 100644 --- a/embroider_params.py +++ b/embroider_params.py @@ -6,7 +6,7 @@ import sys import json import traceback import time -from threading import Thread +from threading import Thread, Event from copy import copy from cStringIO import StringIO import wx @@ -18,40 +18,6 @@ from functools import partial from itertools import groupby from embroider_simulate import EmbroiderySimulator -class KillableThread(Thread): - """A subclass of threading.Thread, with a kill() method.""" - def __init__(self, *args, **keywords): - Thread.__init__(self, *args, **keywords) - self.killed = False - - def start(self): - """Start the thread.""" - self.__run_backup = self.run - self.run = self.__run # Force the Thread to install our trace. - Thread.start(self) - - def __run(self): - """Hacked run function, which installs the trace.""" - sys.settrace(self.globaltrace) - self.__run_backup() - self.run = self.__run_backup - - def globaltrace(self, frame, why, arg): - if why == 'call': - return self.localtrace - else: - return None - - def localtrace(self, frame, why, arg): - if self.killed: - if why == 'line': - raise SystemExit() - return self.localtrace - - def kill(self): - self.killed = True - - def presets_path(): try: import appdirs @@ -365,7 +331,7 @@ class SettingsFrame(wx.Frame): self.simulate_window = None self.simulate_thread = None - self.thread_num = 0 + self.simulate_refresh_needed = Event() self.presets_box = wx.StaticBox(self, wx.ID_ANY, label="Presets") @@ -402,12 +368,18 @@ class SettingsFrame(wx.Frame): self.simulate_window.stop() self.simulate_window.clear() - if self.simulate_thread and self.simulate_thread.is_alive(): - self.simulate_thread.kill() + if not self.simulate_thread or not self.simulate_thread.is_alive(): + self.simulate_thread = Thread(target=self.simulate_worker) + self.simulate_thread.daemon = True + self.simulate_thread.start() - self.simulate_thread = KillableThread(target=self.update_patches, name="Simulate%d" % self.thread_num) - self.simulate_thread.start() - self.thread_num += 1 + self.simulate_refresh_needed.set() + + def simulate_worker(self): + while True: + self.simulate_refresh_needed.wait() + self.simulate_refresh_needed.clear() + self.update_patches() def update_patches(self): patches = self.generate_patches() @@ -448,6 +420,10 @@ class SettingsFrame(wx.Frame): try: if tab.enabled() and not tab.is_dependent_tab(): for node in tab.nodes: + if self.simulate_refresh_needed.is_set(): + # cancel, we need to start over + return [] + # Making a copy of the embroidery element is an easy # way to drop the cache in the @cache decorators used # for many params in embroider.py. -- cgit v1.2.3 From 85aafccf25621224426c9893610d581d7a6a24bb Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Sat, 30 Dec 2017 21:30:53 -0500 Subject: cancel should not output the SVG also sped up simulation --- embroider_params.py | 34 ++++++++++++++++++++++++++++------ 1 file changed, 28 insertions(+), 6 deletions(-) diff --git a/embroider_params.py b/embroider_params.py index fc472d36..f8ebdfcb 100644 --- a/embroider_params.py +++ b/embroider_params.py @@ -320,6 +320,7 @@ class SettingsFrame(wx.Frame): def __init__(self, *args, **kwargs): # begin wxGlade: MyFrame.__init__ self.tabs_factory = kwargs.pop('tabs_factory', []) + self.cancel_hook = kwargs.pop('on_cancel', None) wx.Frame.__init__(self, None, wx.ID_ANY, "Embroidery Params" ) @@ -351,7 +352,8 @@ class SettingsFrame(wx.Frame): self.delete_preset_button.Bind(wx.EVT_BUTTON, self.delete_preset) self.cancel_button = wx.Button(self, wx.ID_ANY, "Cancel") - self.cancel_button.Bind(wx.EVT_BUTTON, self.close) + self.cancel_button.Bind(wx.EVT_BUTTON, self.cancel) + self.Bind(wx.EVT_CLOSE, self.cancel) self.use_last_button = wx.Button(self, wx.ID_ANY, "Use Last Settings") self.use_last_button.Bind(wx.EVT_BUTTON, self.use_last) @@ -397,7 +399,7 @@ class SettingsFrame(wx.Frame): simulator_pos.x += 5 try: - self.simulate_window = EmbroiderySimulator(None, -1, "Embroidery Simulator", simulator_pos, size=(300, 300), patches=patches, on_close=self.simulate_window_closed) + self.simulate_window = EmbroiderySimulator(None, -1, "Embroidery Simulator", simulator_pos, size=(300, 300), patches=patches, on_close=self.simulate_window_closed, stitches_per_frame=10) except: with open('/tmp/params_debug.log', 'a') as log: print >> log, traceback.format_exc() @@ -534,18 +536,24 @@ class SettingsFrame(wx.Frame): def apply(self, event): self._apply() save_preset("__LAST__", self.get_preset_data()) - self.Close() + self.close() def use_last(self, event): self._load_preset("__LAST__") self.apply(event) - def close(self, event): + def close(self): if self.simulate_window: self.simulate_window.stop() self.simulate_window.Close() - self.Close() + self.Destroy() + + def cancel(self, event): + if self.cancel_hook: + self.cancel_hook() + + self.close() def __set_properties(self): # begin wxGlade: MyFrame.__set_properties @@ -579,6 +587,10 @@ class SettingsFrame(wx.Frame): # end wxGlade class EmbroiderParams(inkex.Effect): + def __init__(self, *args, **kwargs): + self.cancelled = False + inkex.Effect.__init__(self, *args, **kwargs) + def get_nodes(self): if self.selected: nodes = [] @@ -690,12 +702,20 @@ class EmbroiderParams(inkex.Effect): return tabs + def cancel(self): + self.cancelled = True + def effect(self): app = wx.App() - frame = SettingsFrame(tabs_factory=self.create_tabs) + frame = SettingsFrame(tabs_factory=self.create_tabs, on_cancel=self.cancel) frame.Show() app.MainLoop() + if self.cancelled: + # This prevents the superclass from outputting the SVG, because we + # may have modified the DOM. + sys.exit(0) + def save_stderr(): # GTK likes to spam stderr, which inkscape will show in a dialog. @@ -719,6 +739,8 @@ if __name__ == "__main__": try: e = EmbroiderParams() e.affect() + except SystemExit: + raise except: traceback.print_exc() -- cgit v1.2.3 From 824dd3c4e6a42dbdb032f93ccfc050e5edb9b242 Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Sun, 31 Dec 2017 22:05:05 -0500 Subject: don't mirror Y axis for patches --- embroider_simulate.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/embroider_simulate.py b/embroider_simulate.py index 6db203ad..2d913bbb 100644 --- a/embroider_simulate.py +++ b/embroider_simulate.py @@ -41,8 +41,10 @@ class EmbroiderySimulator(wx.Frame): def load(self, stitch_file=None, patches=None): if stitch_file: + self.mirror = True self.segments = self._parse_stitch_file(stitch_file) elif patches: + self.mirror = False self.segments = self._patches_to_segments(patches) else: return @@ -210,8 +212,10 @@ class EmbroiderySimulator(wx.Frame): for i in xrange(self.stitches_per_frame): try: ((x1, y1), (x2, y2)), color = self.segments[self.current_stitch] - y1 = self.height - y1 - y2 = self.height - y2 + + if self.mirror: + y1 = self.height - y1 + y2 = self.height - y2 self.canvas.SetPen(color) self.canvas.DrawLines(((x1, y1), (x2, y2))) -- cgit v1.2.3 From 82fe2b7144a9b2a845a542304be11a407f18478d Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Sun, 31 Dec 2017 22:10:10 -0500 Subject: show simulate window at start --- embroider_params.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/embroider_params.py b/embroider_params.py index f8ebdfcb..1be10881 100644 --- a/embroider_params.py +++ b/embroider_params.py @@ -328,12 +328,14 @@ class SettingsFrame(wx.Frame): self.tabs = self.tabs_factory(self.notebook) for tab in self.tabs: - tab.on_change(self.params_changed) + tab.on_change(self.update_simulator) self.simulate_window = None self.simulate_thread = None self.simulate_refresh_needed = Event() + wx.CallLater(1000, self.update_simulator) + self.presets_box = wx.StaticBox(self, wx.ID_ANY, label="Presets") self.preset_chooser = wx.ComboBox(self, wx.ID_ANY) @@ -365,7 +367,7 @@ class SettingsFrame(wx.Frame): self.__do_layout() # end wxGlade - def params_changed(self, tab): + def update_simulator(self, tab=None): if self.simulate_window: self.simulate_window.stop() self.simulate_window.clear() @@ -387,9 +389,9 @@ class SettingsFrame(wx.Frame): patches = self.generate_patches() if patches: - wx.CallAfter(self.update_simulator, patches) + wx.CallAfter(self.refresh_simulator, patches) - def update_simulator(self, patches): + def refresh_simulator(self, patches): if self.simulate_window: self.simulate_window.stop() self.simulate_window.load(patches=patches) -- cgit v1.2.3 From c2388a53a66be19199f84d332cfb9f613512db4a Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Sun, 31 Dec 2017 22:24:18 -0500 Subject: adjust default simulation speed to always take ~5 seconds --- embroider_params.py | 2 +- embroider_simulate.py | 12 ++++++++++++ 2 files changed, 13 insertions(+), 1 deletion(-) diff --git a/embroider_params.py b/embroider_params.py index 1be10881..8300cbc2 100644 --- a/embroider_params.py +++ b/embroider_params.py @@ -401,7 +401,7 @@ class SettingsFrame(wx.Frame): simulator_pos.x += 5 try: - self.simulate_window = EmbroiderySimulator(None, -1, "Embroidery Simulator", simulator_pos, size=(300, 300), patches=patches, on_close=self.simulate_window_closed, stitches_per_frame=10) + self.simulate_window = EmbroiderySimulator(None, -1, "Embroidery Simulator", simulator_pos, size=(300, 300), patches=patches, on_close=self.simulate_window_closed, target_duration=5) except: with open('/tmp/params_debug.log', 'a') as log: print >> log, traceback.format_exc() diff --git a/embroider_simulate.py b/embroider_simulate.py index 2d913bbb..6fdff57c 100644 --- a/embroider_simulate.py +++ b/embroider_simulate.py @@ -14,6 +14,7 @@ class EmbroiderySimulator(wx.Frame): self.on_close_hook = kwargs.pop('on_close', None) self.frame_period = kwargs.pop('frame_period', 80) self.stitches_per_frame = kwargs.pop('stitches_per_frame', 1) + self.target_duration = kwargs.pop('target_duration', None) wx.Frame.__init__(self, *args, **kwargs) @@ -22,6 +23,9 @@ class EmbroiderySimulator(wx.Frame): self.load(stitch_file, patches) + if self.target_duration: + self.adjust_speed(self.target_duration) + self.buffer = wx.Bitmap(self.width, self.height) self.dc = wx.MemoryDC() self.dc.SelectObject(self.buffer) @@ -51,6 +55,14 @@ class EmbroiderySimulator(wx.Frame): self.width, self.height = self.get_dimensions() + def adjust_speed(self, duration): + self.frame_period = 1000 * float(duration) / len(self.segments) + self.stitches_per_frame = 1 + + while self.frame_period < 1.0: + self.frame_period *= 2 + self.stitches_per_frame *= 2 + def on_key_down(self, event): keycode = event.GetKeyCode() -- cgit v1.2.3 From fd4d19bf8dd04a978a6cad4d89949f3c08a1cd78 Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Sun, 31 Dec 2017 22:28:43 -0500 Subject: don't start outdated simulation --- embroider_params.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/embroider_params.py b/embroider_params.py index 8300cbc2..ffd9d53a 100644 --- a/embroider_params.py +++ b/embroider_params.py @@ -388,7 +388,7 @@ class SettingsFrame(wx.Frame): def update_patches(self): patches = self.generate_patches() - if patches: + if patches and not self.simulate_refresh_needed.is_set(): wx.CallAfter(self.refresh_simulator, patches) def refresh_simulator(self, patches): -- cgit v1.2.3 From 75241f5f9ddb1e48744878e987136224f9526845 Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Sun, 31 Dec 2017 22:38:15 -0500 Subject: trim unnecessary whitespace --- embroider_simulate.py | 44 +++++++++++++++++++++++++++++++++++++++----- 1 file changed, 39 insertions(+), 5 deletions(-) diff --git a/embroider_simulate.py b/embroider_simulate.py index 6fdff57c..0bee6327 100644 --- a/embroider_simulate.py +++ b/embroider_simulate.py @@ -53,6 +53,7 @@ class EmbroiderySimulator(wx.Frame): else: return + self.trim_margins() self.width, self.height = self.get_dimensions() def adjust_speed(self, duration): @@ -159,15 +160,48 @@ class EmbroiderySimulator(wx.Frame): return segments + def all_coordinates(self): + for segment in self.segments: + start, end = segment[0] + + yield start + yield end + + def trim_margins(self): + """remove any unnecessary whitespace around the design""" + + min_x = sys.maxint + min_y = sys.maxint + + for x, y in self.all_coordinates(): + min_x = min(min_x, x) + min_y = min(min_y, y) + + + new_segments = [] + + for segment in self.segments: + (start, end), color = segment + + new_segment = ( + ( + (start[0] - min_x, start[1] - min_y), + (end[0] - min_x, end[1] - min_y), + ), + color + ) + + new_segments.append(new_segment) + + self.segments = new_segments + def get_dimensions(self): width = 0 height = 0 - for segment in self.segments: - (start_x, start_y), (end_x, end_y) = segment[0] - - width = max(width, start_x, end_x) - height = max(height, start_y, end_y) + for x, y in self.all_coordinates(): + width = max(width, x) + height = max(height, y) return width, height -- cgit v1.2.3 From 3cb4427fe6a5bdc83e08bc4e1c970598609a5f49 Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Sun, 31 Dec 2017 22:53:29 -0500 Subject: make debugging slightly easier by popping up traceback --- embroider_params.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/embroider_params.py b/embroider_params.py index ffd9d53a..289f0d8e 100644 --- a/embroider_params.py +++ b/embroider_params.py @@ -403,9 +403,16 @@ class SettingsFrame(wx.Frame): try: self.simulate_window = EmbroiderySimulator(None, -1, "Embroidery Simulator", simulator_pos, size=(300, 300), patches=patches, on_close=self.simulate_window_closed, target_duration=5) except: - with open('/tmp/params_debug.log', 'a') as log: - print >> log, traceback.format_exc() - log.flush() + error = traceback.format_exc() + + try: + # a window may have been created, so we need to destroy it + # or the app will never exit + wx.Window.FindWindowByName("Embroidery Simulator").Destroy() + except: + pass + + info_dialog(self, error, "Internal Error") self.simulate_window.Show() wx.CallLater(10, self.Raise) @@ -731,7 +738,7 @@ def save_stderr(): def restore_stderr(): os.dup2(sys.stderr_dup, 2) sys.stderr_backup.write(sys.stderr.getvalue()) - sys.sys.stderr = stderr_backup + sys.stderr = stderr_backup # end of class MyFrame -- cgit v1.2.3 From e9bddedf36e91f33b13a22997872812e849b2157 Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Mon, 1 Jan 2018 14:56:27 -0500 Subject: render objects in the correct order --- embroider_params.py | 49 ++++++++++++++++++++++++++++--------------------- embroider_simulate.py | 1 - 2 files changed, 28 insertions(+), 22 deletions(-) diff --git a/embroider_params.py b/embroider_params.py index 289f0d8e..88d5772c 100644 --- a/embroider_params.py +++ b/embroider_params.py @@ -424,28 +424,34 @@ class SettingsFrame(wx.Frame): def generate_patches(self): patches = [] + nodes = [] for tab in self.tabs: tab.apply() - try: - if tab.enabled() and not tab.is_dependent_tab(): - for node in tab.nodes: - if self.simulate_refresh_needed.is_set(): - # cancel, we need to start over - return [] - - # Making a copy of the embroidery element is an easy - # way to drop the cache in the @cache decorators used - # for many params in embroider.py. - - patches.extend(copy(node).to_patches(None)) - except SystemExit: - raise - except: - # Ignore errors. This can be things like incorrect paths for - # satins or division by zero caused by incorrect param values. - pass + if tab.enabled() and not tab.is_dependent_tab(): + nodes.extend(tab.nodes) + + # sort nodes into the proper stacking order + nodes.sort(key=lambda node: node.order) + + try: + for node in nodes: + if self.simulate_refresh_needed.is_set(): + # cancel; params were updated and we need to start over + return [] + + # Making a copy of the embroidery element is an easy + # way to drop the cache in the @cache decorators used + # for many params in embroider.py. + + patches.extend(copy(node).to_patches(None)) + except SystemExit: + raise + except: + # Ignore errors. This can be things like incorrect paths for + # satins or division by zero caused by incorrect param values. + pass return patches @@ -631,9 +637,11 @@ class EmbroiderParams(inkex.Effect): nodes = self.get_nodes() nodes_by_class = defaultdict(list) - for node in self.get_nodes(): + for z, node in enumerate(self.get_nodes()): for cls in self.embroidery_classes(node): - nodes_by_class[cls].append(node) + element = cls(node) + element.order = z + nodes_by_class[cls].append(element) return sorted(nodes_by_class.items(), key=lambda (cls, nodes): cls.__name__) @@ -659,7 +667,6 @@ class EmbroiderParams(inkex.Effect): def create_tabs(self, parent): tabs = [] for cls, nodes in self.get_nodes_by_class(): - nodes = [cls(node) for node in nodes] params = cls.get_params() for param in params: diff --git a/embroider_simulate.py b/embroider_simulate.py index 0bee6327..5654b9c3 100644 --- a/embroider_simulate.py +++ b/embroider_simulate.py @@ -177,7 +177,6 @@ class EmbroiderySimulator(wx.Frame): min_x = min(min_x, x) min_y = min(min_y, y) - new_segments = [] for segment in self.segments: -- cgit v1.2.3 From 084dc783d2f2a94e77dfb6afe9b5a2ca14d7fc64 Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Mon, 1 Jan 2018 15:00:58 -0500 Subject: add restart option for simulator --- embroider_simulate.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/embroider_simulate.py b/embroider_simulate.py index 5654b9c3..31d3cd75 100644 --- a/embroider_simulate.py +++ b/embroider_simulate.py @@ -84,6 +84,10 @@ class EmbroiderySimulator(wx.Frame): self.timer.Stop() else: self.timer.Start(self.frame_period) + elif keycode == ord("R"): + self.stop() + self.clear() + self.go() self.frame_period = max(1, self.frame_period) self.stitches_per_frame = max(self.stitches_per_frame, 1) @@ -208,7 +212,10 @@ class EmbroiderySimulator(wx.Frame): self.clear() self.current_stitch = 0 - self.timer = wx.PyTimer(self.draw_one_frame) + + if not self.timer: + self.timer = wx.PyTimer(self.draw_one_frame) + self.timer.Start(self.frame_period) def on_close(self, event): -- cgit v1.2.3 From 09b9dd94e6591e0e74b8896c8cb8fdf173e32f5b Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Mon, 1 Jan 2018 15:26:18 -0500 Subject: adjust colors in the simulator to make them visible Colors too close to white are darkened just a bit to make them stand out against the white background. --- embroider_simulate.py | 27 +++++++++++++++++++++++---- 1 file changed, 23 insertions(+), 4 deletions(-) diff --git a/embroider_simulate.py b/embroider_simulate.py index 31d3cd75..f3f9e5e2 100644 --- a/embroider_simulate.py +++ b/embroider_simulate.py @@ -4,6 +4,7 @@ import numpy import wx import inkex import simplestyle +import colorsys from embroider import patches_to_stitches, stitches_to_polylines @@ -102,6 +103,24 @@ class EmbroiderySimulator(wx.Frame): return string + def color_to_pen(self, color): + # python colorsys module uses floats from 0 to 1.0 + color = [value / 255.0 for value in color] + + hls = list(colorsys.rgb_to_hls(*color)) + + # Our background is white. If the color is too close to white, then + # it won't be visible. Capping lightness should make colors visible + # without changing them too much. + hls[1] = min(hls[1], 0.85) + + color = colorsys.hls_to_rgb(*hls) + + # convert back to values in the range of 0-255 + color = [value * 255 for value in color] + + return wx.Pen(color) + def _patches_to_segments(self, patches): stitches = patches_to_stitches(patches) @@ -117,7 +136,7 @@ class EmbroiderySimulator(wx.Frame): if stitch.color == last_color: segments.append(((last_pos, pos), pen)) else: - pen = wx.Pen(simplestyle.parseColor(stitch.color)) + pen = self.color_to_pen(simplestyle.parseColor(stitch.color)) last_pos = pos last_color = stitch.color @@ -132,7 +151,7 @@ class EmbroiderySimulator(wx.Frame): segments = [] pos = (0, 0) - color = wx.Brush('black') + pen = wx.Brush('black') cut = True with open(stitch_file_path) as stitch_file: @@ -144,7 +163,7 @@ class EmbroiderySimulator(wx.Frame): if symbol == "$": red, green, blue = fields[2:5] - color = wx.Pen((int(red), int(green), int(blue))) + color = color_to_pen((int(red), int(green), int(blue))) elif symbol == "*": if command == "COLOR": # change color @@ -157,7 +176,7 @@ class EmbroiderySimulator(wx.Frame): new_pos = (int(float(x) * 10), int(float(y) * 10)) if not cut: - segments.append(((pos, new_pos), color)) + segments.append(((pos, new_pos), pen)) cut = False pos = new_pos -- cgit v1.2.3