diff options
Diffstat (limited to 'lib/simulator.py')
| -rw-r--r-- | lib/simulator.py | 151 |
1 files changed, 107 insertions, 44 deletions
diff --git a/lib/simulator.py b/lib/simulator.py index 2d2d3e08..5620f65b 100644 --- a/lib/simulator.py +++ b/lib/simulator.py @@ -10,38 +10,44 @@ from .svg import PIXELS_PER_MM, color_block_to_point_lists class EmbroiderySimulator(wx.Frame): def __init__(self, *args, **kwargs): stitch_plan = kwargs.pop('stitch_plan', None) + self.x_position = kwargs.pop('x_position', 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) self.target_duration = kwargs.pop('target_duration', None) - self.margin = 10 + self.margin = 30 - screen_rect = wx.Display(0).ClientArea - self.max_width = kwargs.pop('max_width', screen_rect.GetWidth()) - self.max_height = kwargs.pop('max_height', screen_rect.GetHeight()) + screen_rect = self.get_current_screen_rect() + self.max_width = kwargs.pop('max_width', screen_rect[2]) + self.max_height = kwargs.pop('max_height', screen_rect[3]) self.scale = 1 + self.min_width = 600 + if self.max_width < self.min_width: + self.max_width = self.min_width + wx.Frame.__init__(self, *args, **kwargs) self.panel = wx.Panel(self, wx.ID_ANY) - self.panel.SetFocus() + + self.panel.SetDoubleBuffered(True) self.sizer = wx.BoxSizer(wx.VERTICAL) self.button_sizer = wx.BoxSizer(wx.HORIZONTAL) self.button_label = ( - [_("Speed up"),_('Press + or arrow up to speed up')], - [_("Slow down"),_('Press - or arrow down to slow down')], - [_("Pause"),_("Press P to pause the animation")], - [_("Restart"),_("Press R to restart the animation")], - [_("Quit"),_("Press Q to close the simulation window")]) + [_("Speed up"), _('Press + or arrow up to speed up'), self.animation_speed_up], + [_("Slow down"), _('Press - or arrow down to slow down'), self.animation_slow_down], + [_("Pause"), _("Press P to pause the animation"), self.animation_pause], + [_("Restart"), _("Press R to restart the animation"), self.animation_restart], + [_("Quit"), _("Press Q to close the simulation window"), self.animation_quit]) self.buttons = [] for i in range(0, len(self.button_label)): self.buttons.append(wx.Button(self, -1, self.button_label[i][0])) self.button_sizer.Add(self.buttons[i], 1, wx.EXPAND) - self.buttons[i].Bind(wx.EVT_BUTTON, self.on_key_down) self.buttons[i].SetToolTip(self.button_label[i][1]) + self.buttons[i].Bind(wx.EVT_BUTTON, self.button_label[i][2]) self.sizer.Add(self.panel, 1, wx.EXPAND) self.sizer.Add(self.button_sizer, 0, wx.EXPAND) @@ -59,15 +65,51 @@ class EmbroiderySimulator(wx.Frame): self.clear() + self.set_stitch_counter(1) + + shortcut_keys = [ + (wx.ACCEL_NORMAL, ord('+'), self.animation_speed_up), + (wx.ACCEL_NORMAL, ord('='), self.animation_speed_up), + (wx.ACCEL_SHIFT, ord('='), self.animation_speed_up), + (wx.ACCEL_NORMAL, wx.WXK_ADD, self.animation_speed_up), + (wx.ACCEL_NORMAL, wx.WXK_NUMPAD_ADD, self.animation_speed_up), + (wx.ACCEL_NORMAL, wx.WXK_NUMPAD_UP, self.animation_speed_up), + (wx.ACCEL_NORMAL, wx.WXK_UP, self.animation_speed_up), + (wx.ACCEL_NORMAL, ord('-'), self.animation_slow_down), + (wx.ACCEL_NORMAL, ord('_'), self.animation_slow_down), + (wx.ACCEL_NORMAL, wx.WXK_SUBTRACT, self.animation_slow_down), + (wx.ACCEL_NORMAL, wx.WXK_NUMPAD_SUBTRACT, self.animation_slow_down), + (wx.ACCEL_NORMAL, wx.WXK_NUMPAD_DOWN, self.animation_slow_down), + (wx.ACCEL_NORMAL, wx.WXK_DOWN, self.animation_slow_down), + (wx.ACCEL_NORMAL, ord('r'), self.animation_restart), + (wx.ACCEL_NORMAL, ord('p'), self.animation_pause), + (wx.ACCEL_NORMAL, ord('q'), self.animation_quit)] + + accel_entries = [] + + for shortcut_key in shortcut_keys: + eventId = wx.NewId() + accel_entries.append((shortcut_key[0], shortcut_key[1], eventId)) + self.Bind(wx.EVT_MENU, shortcut_key[2], id=eventId) + + accel_table = wx.AcceleratorTable(accel_entries) + self.SetAcceleratorTable(accel_table) + self.Bind(wx.EVT_SIZE, self.on_size) + self.Bind(wx.EVT_CLOSE, self.on_close) self.panel.Bind(wx.EVT_PAINT, self.on_paint) - self.panel.Bind(wx.EVT_KEY_DOWN, self.on_key_down) + + self.panel.SetFocus() self.timer = None self.last_pos = None - self.Bind(wx.EVT_CLOSE, self.on_close) + def get_current_screen_rect(self): + current_screen = wx.Display.GetFromPoint(wx.GetMousePosition()) + display = wx.Display(current_screen) + screen_rect = display.GetClientArea() + return screen_rect def load(self, stitch_plan=None): if stitch_plan: @@ -87,42 +129,51 @@ class EmbroiderySimulator(wx.Frame): self.frame_period *= 2 self.stitches_per_frame *= 2 - def on_key_down(self, event): - if hasattr(event, 'GetKeyCode'): - keycode = event.GetKeyCode() + def animation_speed_up(self, event): + if self.frame_period == 1: + self.stitches_per_frame *= 2 else: - keycode = event.GetEventObject().GetLabelText() - self.panel.SetFocus() - - if keycode == ord("+") or keycode == ord("=") or keycode == wx.WXK_UP or keycode == "Speed 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 or keycode == "Slow down": - if self.stitches_per_frame == 1: - self.frame_period *= 2 - else: - self.stitches_per_frame /= 2 - elif keycode == ord("Q") or keycode == "Quit": - self.Close() - elif keycode == ord("P") or keycode == "Pause": - if self.timer.IsRunning(): - self.timer.Stop() - else: - self.timer.Start(self.frame_period) - elif keycode == ord("R") or keycode == "Restart": - self.stop() - self.clear() - self.go() + self.frame_period = self.frame_period / 2 + self.animation_update_timer() + def animation_slow_down(self, event): + if self.stitches_per_frame == 1: + self.frame_period *= 2 + else: + self.stitches_per_frame /= 2 + self.animation_update_timer() + + def animation_restart(self, event): + self.stop() + self.clear() + self.go() + + def animation_pause(self, event): + if self.timer.IsRunning(): + self.timer.Stop() + else: + self.timer.Start(self.frame_period) + + def animation_quit(self, event): + self.Close() + + def animation_update_timer(self): 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 set_stitch_counter(self, current_stitch): + if hasattr(self.panel, 'stitch_counter'): + self.panel.stitch_counter.SetLabel(_("Stitch # ") + str(current_stitch) + ' / ' + str(len(self.segments) + 1)) + else: + self.font = wx.Font(9, wx.FONTFAMILY_DEFAULT, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_NORMAL) + self.panel.stitch_counter = wx.StaticText(self, label=_("Stitch #") + '1 / ' + str(len(self.segments)), pos=(30, 10)) + self.panel.stitch_counter.SetFont(self.font) + self.panel.stitch_counter.SetForegroundColour('red') + self.panel.stitch_counter.SetBackgroundColour('white') + def _strip_quotes(self, string): if string.startswith('"') and string.endswith('"'): string = string[1:-1] @@ -194,7 +245,7 @@ class EmbroiderySimulator(wx.Frame): self.width = width self.height = height - self.scale = min(float(self.max_width) / width, float(self.max_height - 60) / height) + self.scale = min(float(self.max_width - self.margin * 2) / width, float(self.max_height - self.margin * 2 - 40) / height) # make room for decorations and the margin self.scale *= 0.95 @@ -241,11 +292,20 @@ class EmbroiderySimulator(wx.Frame): setsize_window_width = self.width * self.scale + decorations_width + self.margin * 2 setsize_window_height = (self.height) * self.scale + decorations_height + self.margin * 2 - if setsize_window_width < 600: - setsize_window_width = 600 + # set minimum width (force space for control buttons) + if setsize_window_width < self.min_width: + setsize_window_width = self.min_width self.SetSize(( setsize_window_width, setsize_window_height)) + # center the simulation on screen if not called by params + # else center vertically + if self.x_position == None: + self.Centre() + else: + display_rect = self.get_current_screen_rect() + self.SetPosition((self.x_position, display_rect[3] / 2 - setsize_window_height / 2)) + e.Skip() def on_paint(self, e): @@ -276,5 +336,8 @@ class EmbroiderySimulator(wx.Frame): self.current_stitch += 1 self.last_pos = (x2, y2) + + self.set_stitch_counter(self.current_stitch + 1) + except IndexError: self.timer.Stop() |