kopia lustrzana https://github.com/inkstitch/inkstitch
855 wiersze
31 KiB
Python
Executable File
855 wiersze
31 KiB
Python
Executable File
# Authors: see git history
|
|
#
|
|
# Copyright (c) 2010 Authors
|
|
# Licensed under the GNU GPL version 3.0 or later. See the file LICENSE for details.
|
|
|
|
# -*- coding: UTF-8 -*-
|
|
|
|
import os
|
|
import sys
|
|
from collections import defaultdict
|
|
from copy import copy
|
|
from itertools import groupby, zip_longest
|
|
from secrets import randbelow
|
|
|
|
import wx
|
|
from wx.lib.scrolledpanel import ScrolledPanel
|
|
|
|
from ..commands import is_command, is_command_symbol
|
|
from ..elements import (Clone, EmbroideryElement, FillStitch, SatinColumn,
|
|
Stroke)
|
|
from ..elements.clone import is_clone
|
|
from ..exceptions import InkstitchException, format_uncaught_exception
|
|
from ..gui import PresetsPanel, PreviewRenderer, WarningPanel
|
|
from ..gui.simulator import SplitSimulatorWindow
|
|
from ..i18n import _
|
|
from ..stitch_plan import stitch_groups_to_stitch_plan
|
|
from ..svg.tags import EMBROIDERABLE_TAGS
|
|
from ..utils import get_resource_dir
|
|
from ..utils.param import ParamOption
|
|
from ..utils.svg_data import get_pagecolor
|
|
from ..utils.threading import ExitThread, check_stop_flag
|
|
from .base import InkstitchExtension
|
|
|
|
|
|
def grouper(iterable_obj, count, fillvalue=None):
|
|
args = [iter(iterable_obj)] * count
|
|
return zip_longest(*args, fillvalue=fillvalue)
|
|
|
|
|
|
class ParamsTab(ScrolledPanel):
|
|
def __init__(self, *args, **kwargs):
|
|
self.params = kwargs.pop('params', [])
|
|
self.name = kwargs.pop('name', None)
|
|
|
|
# TODO: this is actually a list of embroidery elements, not DOM nodes, and needs to be renamed
|
|
self.nodes = kwargs.pop('nodes')
|
|
kwargs["style"] = wx.TAB_TRAVERSAL
|
|
ScrolledPanel.__init__(self, *args, **kwargs)
|
|
self.SetupScrolling()
|
|
|
|
self.changed_inputs = set()
|
|
self.dependent_tabs = []
|
|
self.parent_tab = None
|
|
self.param_inputs = {}
|
|
self.choice_widgets = defaultdict(list)
|
|
self.dict_of_choices = {}
|
|
self.paired_tab = None
|
|
self.disable_notify_pair = False
|
|
|
|
toggles = [param for param in self.params if param.type == 'toggle']
|
|
|
|
if toggles:
|
|
self.toggle = toggles[0]
|
|
self.params.remove(self.toggle)
|
|
self.toggle_checkbox = wx.CheckBox(
|
|
self, label=self.toggle.description)
|
|
|
|
value = any(self.toggle.values)
|
|
if self.toggle.inverse:
|
|
value = not value
|
|
self.toggle_checkbox.SetValue(value)
|
|
|
|
self.toggle_checkbox.Bind(
|
|
wx.EVT_CHECKBOX, self.update_toggle_state)
|
|
self.toggle_checkbox.Bind(wx.EVT_CHECKBOX, self.changed)
|
|
|
|
self.param_inputs[self.toggle.name] = self.toggle_checkbox
|
|
else:
|
|
self.toggle = None
|
|
|
|
self.param_change_indicators = {}
|
|
|
|
self.settings_grid = wx.FlexGridSizer(rows=0, cols=4, hgap=10, vgap=15)
|
|
self.settings_grid.AddGrowableCol(2, 1)
|
|
|
|
self.pencil_icon = wx.Image(os.path.join(get_resource_dir(
|
|
"icons"), "pencil_20x20.png")).ConvertToBitmap()
|
|
|
|
randomize_icon = 'randomize_20x20.png'
|
|
if wx.SystemSettings().GetAppearance().IsDark():
|
|
randomize_icon = 'randomize_20x20_dark.png'
|
|
self.randomize_icon = wx.Image(
|
|
os.path.join(get_resource_dir("icons"), randomize_icon)).ConvertToBitmap()
|
|
|
|
self.__set_properties()
|
|
self.__do_layout()
|
|
|
|
if self.toggle:
|
|
self.update_toggle_state()
|
|
|
|
self.update_enable_widgets()
|
|
|
|
def pair(self, tab):
|
|
self.paired_tab = tab
|
|
self.update_description()
|
|
|
|
def add_dependent_tab(self, tab):
|
|
self.dependent_tabs.append(tab)
|
|
self.update_description()
|
|
|
|
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):
|
|
if self.toggle_checkbox:
|
|
return self.toggle_checkbox.IsChecked()
|
|
else:
|
|
return True
|
|
|
|
def update_toggle_state(self, event=None, notify_pair=True):
|
|
enable = self.enabled()
|
|
for child in self.settings_grid.GetChildren():
|
|
widget = child.GetWindow()
|
|
if widget:
|
|
child.GetWindow().Enable(enable)
|
|
|
|
if notify_pair and self.paired_tab:
|
|
self.paired_tab.pair_changed(enable)
|
|
|
|
for tab in self.dependent_tabs:
|
|
tab.dependent_enable(enable)
|
|
|
|
if event:
|
|
event.Skip()
|
|
|
|
def update_choice_state(self, event=None, combo=False):
|
|
input = event.GetEventObject()
|
|
if combo:
|
|
selection = input.GetClientData(input.GetSelection()).id
|
|
else:
|
|
selection = input.GetSelection()
|
|
|
|
param = self.inputs_to_params[input]
|
|
|
|
self.update_choice_widgets((param, selection))
|
|
self.settings_grid.Layout()
|
|
self.Fit()
|
|
self.Layout()
|
|
|
|
if event:
|
|
event.Skip()
|
|
|
|
def update_combo_state(self, event=None):
|
|
self.update_choice_state(event, True)
|
|
|
|
def get_combo_value_index(self, param, options, default):
|
|
for option in options:
|
|
if option.id == param:
|
|
return options.index(option)
|
|
return default
|
|
|
|
def pair_changed(self, value):
|
|
new_value = not value
|
|
|
|
if self.enabled() != new_value:
|
|
self.set_toggle_state(not value)
|
|
self.update_toggle_state(notify_pair=False)
|
|
|
|
if self.on_change_hook:
|
|
self.on_change_hook(self)
|
|
|
|
def dependent_enable(self, enable):
|
|
if enable:
|
|
self.toggle_checkbox.Enable()
|
|
else:
|
|
self.set_toggle_state(False)
|
|
self.toggle_checkbox.Disable()
|
|
self.update_toggle_state()
|
|
|
|
if self.on_change_hook:
|
|
self.on_change_hook(self)
|
|
|
|
def set_toggle_state(self, value):
|
|
if self.toggle_checkbox:
|
|
self.toggle_checkbox.SetValue(value)
|
|
self.changed_inputs.add(self.toggle_checkbox)
|
|
|
|
def get_values(self):
|
|
values = {}
|
|
|
|
if self.toggle:
|
|
checked = self.enabled()
|
|
if self.toggle_checkbox in self.changed_inputs:
|
|
if self.toggle.inverse:
|
|
values[self.toggle.name] = not checked
|
|
else:
|
|
values[self.toggle.name] = checked
|
|
|
|
if not checked:
|
|
# Ignore params on this tab if the toggle is unchecked,
|
|
# because they're grayed out anyway.
|
|
return values
|
|
|
|
for name, input in self.param_inputs.items():
|
|
if input in self.changed_inputs and input != self.toggle_checkbox:
|
|
# there are two types of combo boxes:
|
|
# 1. multiple values for the same param on selected elements - 2. param type
|
|
# multiple values will be handled with the GetValue() method
|
|
if name in self.dict_of_choices and self.dict_of_choices[name]['param'].type == 'combo':
|
|
values[name] = input.GetClientData(input.GetSelection()).id
|
|
elif isinstance(input, wx.Choice):
|
|
values[name] = input.GetSelection()
|
|
else:
|
|
values[name] = input.GetValue()
|
|
|
|
return values
|
|
|
|
def on_reroll(self, event):
|
|
if len(self.nodes) == 1:
|
|
new_seed = str(randbelow(int(1e8)))
|
|
input = self.param_inputs['random_seed']
|
|
input.SetValue(new_seed)
|
|
self.changed_inputs.add(input)
|
|
else:
|
|
for node in self.nodes:
|
|
new_seed = str(randbelow(int(1e8)))
|
|
node.set_param('random_seed', new_seed)
|
|
|
|
self.enable_change_indicator('random_seed')
|
|
event.Skip()
|
|
if self.on_change_hook:
|
|
self.on_change_hook(self)
|
|
|
|
def apply(self):
|
|
values = self.get_values()
|
|
for node in self.nodes:
|
|
for name, value in values.items():
|
|
node.set_param(name, value)
|
|
|
|
def on_change(self, callable):
|
|
self.on_change_hook = callable
|
|
|
|
def changed(self, event):
|
|
input = event.GetEventObject()
|
|
self.changed_inputs.add(input)
|
|
|
|
param = self.inputs_to_params[input]
|
|
self.enable_change_indicator(param)
|
|
event.Skip()
|
|
|
|
if self.on_change_hook:
|
|
self.on_change_hook(self)
|
|
|
|
def load_preset(self, preset):
|
|
preset_data = preset.get(self.name, {})
|
|
|
|
# print(self.param_inputs, '\n\n', preset_data.items(), file=sys.stderr)
|
|
|
|
for name, value in preset_data.items():
|
|
if name in self.param_inputs:
|
|
if name in self.dict_of_choices and self.dict_of_choices[name]['param'].type == 'combo':
|
|
param = self.dict_of_choices[name]["param"]
|
|
self.param_inputs[name].SetSelection(self.get_combo_value_index(value, param.options, param.default))
|
|
elif isinstance(self.param_inputs[name], wx.Choice):
|
|
self.param_inputs[name].SetSelection(int(value))
|
|
else:
|
|
input = self.param_inputs[name]
|
|
if name == "satin_column" and input.Label == _('Running stitch along paths'):
|
|
value = not value
|
|
input.SetValue(value)
|
|
self.changed_inputs.add(self.param_inputs[name])
|
|
|
|
self.update_toggle_state()
|
|
|
|
def save_preset(self, storage):
|
|
storage[self.name] = self.get_values()
|
|
|
|
def update_description(self):
|
|
if len(self.nodes) == 1:
|
|
description = _("These settings will be applied to 1 object.")
|
|
else:
|
|
description = _(
|
|
"These settings will be applied to %d objects.") % len(self.nodes)
|
|
|
|
if any(len(param.values) > 1 for param in self.params):
|
|
description += "\n • " + \
|
|
_("Some settings had different values across objects. Select a value from the dropdown or enter a new one.")
|
|
|
|
if self.dependent_tabs:
|
|
if len(self.dependent_tabs) == 1:
|
|
description += "\n • " + \
|
|
_("Disabling this tab will disable the following %d tabs.") % len(
|
|
self.dependent_tabs)
|
|
else:
|
|
description += "\n • " + \
|
|
_("Disabling this tab will disable the following tab.")
|
|
|
|
if self.paired_tab:
|
|
description += "\n • " + \
|
|
_("Enabling this tab will disable %s and vice-versa.") % self.paired_tab.name
|
|
|
|
self.description_text = description
|
|
|
|
def resized(self, event):
|
|
if not hasattr(self, 'rewrap_timer'):
|
|
self.rewrap_timer = wx.Timer()
|
|
self.rewrap_timer.Bind(wx.EVT_TIMER, self.rewrap)
|
|
|
|
# If we try to rewrap every time we get EVT_SIZE then a resize is
|
|
# extremely slow.
|
|
self.rewrap_timer.Start(50, oneShot=True)
|
|
event.Skip()
|
|
|
|
def rewrap(self, event=None):
|
|
self.description.SetLabel(self.description_text)
|
|
self.description.Wrap(self.GetSize().x - 20)
|
|
self.description_container.Layout()
|
|
if event:
|
|
event.Skip()
|
|
|
|
def __set_properties(self):
|
|
# begin wxGlade: SatinPane.__set_properties
|
|
# end wxGlade
|
|
pass
|
|
|
|
# choice tuple is None or contains ("choice widget param name", "actual selection")
|
|
def update_choice_widgets(self, choice_tuple=None):
|
|
if choice_tuple is None: # update all choices
|
|
for choice in self.dict_of_choices.values():
|
|
if choice["param"].type == "combo":
|
|
self.update_choice_widgets((choice["param"].name, choice["widget"].GetClientData(choice["widget"].GetSelection()).id))
|
|
else:
|
|
self.update_choice_widgets((choice["param"].name, choice["widget"].GetSelection()))
|
|
else:
|
|
choice = self.dict_of_choices[choice_tuple[0]]
|
|
last_selection = choice["last_initialized_choice"]
|
|
if choice["param"].type == "combo":
|
|
current_selection = choice["widget"].GetClientData(choice["widget"].GetSelection()).id
|
|
else:
|
|
current_selection = choice["widget"].GetSelection()
|
|
|
|
if last_selection != -1 and last_selection != current_selection: # Hide the old widgets
|
|
for widget in self.choice_widgets[(choice["param"].name, last_selection)]:
|
|
widget.Hide()
|
|
|
|
for widgets in grouper(self.choice_widgets[choice_tuple], 4):
|
|
widgets[0].Show(True)
|
|
widgets[1].Show(True)
|
|
widgets[2].Show(True)
|
|
widgets[3].Show(True)
|
|
choice["last_initialized_choice"] = current_selection
|
|
|
|
def update_enable_widgets(self, event=None):
|
|
if event is None:
|
|
for param in self.params:
|
|
if param.enables is not None:
|
|
value = self.param_inputs[param.name].GetValue()
|
|
for item in param.enables:
|
|
enable_input = self.param_inputs[item]
|
|
enable_input.Enable(value)
|
|
enable_input.GetPrevSibling().Enable(value)
|
|
else:
|
|
input = event.GetEventObject()
|
|
param = [param for param in self.params if param.name == input.Name]
|
|
if not param:
|
|
return
|
|
param = param[0]
|
|
if param.enables is not None:
|
|
value = input.GetValue()
|
|
for item in param.enables:
|
|
enable_input = self.param_inputs[item]
|
|
enable_input.Enable(value)
|
|
enable_input.GetPrevSibling().Enable(value)
|
|
|
|
def __do_layout(self, only_settings_grid=False): # noqa: C901
|
|
|
|
# just to add space around the settings
|
|
box = wx.BoxSizer(wx.VERTICAL)
|
|
|
|
summary_box = wx.StaticBox(self, wx.ID_ANY, label=_("Inkscape objects"))
|
|
sizer = wx.StaticBoxSizer(summary_box, wx.HORIZONTAL)
|
|
self.description = wx.StaticText(self)
|
|
self.update_description()
|
|
self.description.SetLabel(self.description_text)
|
|
self.description_container = box
|
|
self.Bind(wx.EVT_SIZE, self.resized)
|
|
sizer.Add(self.description, proportion=0, flag=wx.EXPAND | wx.ALL, border=5)
|
|
box.Add(sizer, proportion=0, flag=wx.ALL, border=5)
|
|
|
|
if self.toggle:
|
|
toggle_sizer = wx.BoxSizer(wx.HORIZONTAL)
|
|
toggle_sizer.Add(self.create_change_indicator(
|
|
self.toggle.name), proportion=0, flag=wx.ALIGN_CENTER_VERTICAL | wx.RIGHT, border=5)
|
|
toggle_sizer.Add(self.toggle_checkbox, proportion=0, flag=wx.ALIGN_CENTER_VERTICAL)
|
|
box.Add(toggle_sizer, proportion=0, flag=wx.BOTTOM, border=10)
|
|
|
|
for param in self.params:
|
|
col1 = self.create_change_indicator(param.name)
|
|
description = wx.StaticText(self, label=param.description)
|
|
description.SetToolTip(param.tooltip)
|
|
|
|
if param.select_items is not None:
|
|
col1.Hide()
|
|
description.Hide()
|
|
for item in param.select_items:
|
|
self.choice_widgets[item].extend([col1, description])
|
|
|
|
self.settings_grid.Add(col1, flag=wx.ALIGN_CENTER_VERTICAL)
|
|
self.settings_grid.Add(description, flag=wx.RIGHT | wx.ALIGN_CENTER_VERTICAL | wx.TOP, border=5)
|
|
|
|
if param.type == 'boolean':
|
|
if len(param.values) > 1:
|
|
input = wx.CheckBox(self, style=wx.CHK_3STATE)
|
|
input.Set3StateValue(wx.CHK_UNDETERMINED)
|
|
else:
|
|
input = wx.CheckBox(self)
|
|
if param.values:
|
|
input.SetValue(param.values[0])
|
|
input.SetName(param.name)
|
|
input.Bind(wx.EVT_CHECKBOX, self.update_enable_widgets)
|
|
input.Bind(wx.EVT_CHECKBOX, self.changed)
|
|
elif param.type == 'dropdown':
|
|
input = wx.Choice(self, wx.ID_ANY, choices=param.options)
|
|
input.SetSelection(int(param.values[0]))
|
|
input.Bind(wx.EVT_CHOICE, self.changed)
|
|
input.Bind(wx.EVT_CHOICE, self.update_choice_state)
|
|
self.dict_of_choices[param.name] = {
|
|
"param": param, "widget": input, "last_initialized_choice": 1}
|
|
elif param.type == 'combo':
|
|
input = wx.adv.BitmapComboBox(self, wx.ID_ANY, choices=[], style=wx.CB_READONLY)
|
|
for option in param.options:
|
|
if option.preview_image:
|
|
image = wx.Image(option.preview_image)
|
|
image.Rescale(60, 60, quality=wx.IMAGE_QUALITY_HIGH)
|
|
image = wx.Bitmap(image)
|
|
else:
|
|
image = wx.NullBitmap
|
|
input.Append(option.name, image, option)
|
|
if not param.options:
|
|
input.Append(_('No options available'), ParamOption('not_available'))
|
|
value = self.get_combo_value_index(param.values[0], param.options, param.default or 0)
|
|
input.SetSelection(value)
|
|
input.Bind(wx.EVT_COMBOBOX, self.changed)
|
|
input.Bind(wx.EVT_COMBOBOX, self.update_combo_state)
|
|
self.dict_of_choices[param.name] = {
|
|
"param": param, "widget": input, "last_initialized_choice": 1}
|
|
elif len(param.values) > 1:
|
|
input = wx.ComboBox(self, wx.ID_ANY, choices=sorted(
|
|
str(value) for value in param.values), style=wx.CB_DROPDOWN)
|
|
input.Bind(wx.EVT_COMBOBOX, self.changed)
|
|
input.Bind(wx.EVT_TEXT, self.changed)
|
|
else:
|
|
value = param.values[0] if param.values else ""
|
|
input = wx.TextCtrl(self, wx.ID_ANY, value=str(value))
|
|
input.Bind(wx.EVT_TEXT, self.changed)
|
|
|
|
self.param_inputs[param.name] = input
|
|
|
|
if param.type == 'random_seed':
|
|
col4 = wx.Button(self, wx.ID_ANY, _("Re-roll"))
|
|
col4.Bind(wx.EVT_BUTTON, self.on_reroll)
|
|
col4.SetBitmap(self.randomize_icon)
|
|
else:
|
|
col4 = wx.StaticText(self, label=param.unit or "")
|
|
|
|
if param.select_items is not None:
|
|
input.Hide()
|
|
col4.Hide()
|
|
for item in param.select_items:
|
|
self.choice_widgets[item].extend([input, col4])
|
|
|
|
self.settings_grid.Add(input, flag=wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.EXPAND, border=10)
|
|
self.settings_grid.Add(col4, flag=wx.ALIGN_CENTER_VERTICAL)
|
|
|
|
self.inputs_to_params = {v: k for k, v in self.param_inputs.items()}
|
|
|
|
box.Add(self.settings_grid, proportion=1, flag=wx.ALL | wx.EXPAND, border=10)
|
|
|
|
self.update_choice_widgets()
|
|
self.SetSizer(box)
|
|
self.SetMinSize((box.CalcMin()[0], 200))
|
|
|
|
self.Layout()
|
|
|
|
def create_change_indicator(self, param):
|
|
indicator = wx.Button(self, style=wx.BORDER_NONE |
|
|
wx.BU_NOTEXT, size=(28, 28))
|
|
indicator.SetToolTip(
|
|
_('Click to force this parameter to be saved when you click "Apply and Quit"'))
|
|
indicator.Bind(
|
|
wx.EVT_BUTTON, lambda event: self.enable_change_indicator(param))
|
|
|
|
self.param_change_indicators[param] = indicator
|
|
return indicator
|
|
|
|
def enable_change_indicator(self, param):
|
|
self.param_change_indicators[param].SetBitmapLabel(self.pencil_icon)
|
|
self.param_change_indicators[param].SetToolTip(
|
|
_('This parameter will be saved when you click "Apply and Quit"'))
|
|
|
|
# end of class SatinPane
|
|
|
|
|
|
class SettingsPanel(wx.Panel):
|
|
def __init__(self, parent, tabs_factory=None, metadata=None, background_color='white', simulator=None):
|
|
self.tabs_factory = tabs_factory
|
|
self.metadata = metadata
|
|
self.background_color = background_color
|
|
self.simulator = simulator
|
|
self.parent = parent
|
|
|
|
super().__init__(self.parent, wx.ID_ANY)
|
|
|
|
self.preview_renderer = PreviewRenderer(self.render_stitch_plan, self.on_stitch_plan_rendered)
|
|
|
|
self.notebook = wx.Notebook(self, wx.ID_ANY)
|
|
self.tabs = self.tabs_factory(self.notebook)
|
|
|
|
for tab in self.tabs:
|
|
tab.on_change(self.update_preview)
|
|
|
|
self.presets_panel = PresetsPanel(self)
|
|
self.warning_panel = WarningPanel(self)
|
|
self.warning_panel.Hide()
|
|
|
|
self.cancel_button = wx.Button(self, wx.ID_ANY, _("Cancel"))
|
|
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)
|
|
|
|
self.apply_button = wx.Button(self, wx.ID_ANY, _("Apply and Quit"))
|
|
self.apply_button.Bind(wx.EVT_BUTTON, self.apply)
|
|
|
|
self.__do_layout()
|
|
self.update_preview()
|
|
|
|
def update_preview(self, tab=None):
|
|
self.simulator.stop()
|
|
self.simulator.clear()
|
|
self.preview_renderer.update()
|
|
|
|
def render_stitch_plan(self):
|
|
stitch_groups = []
|
|
nodes = []
|
|
# move the stroke tab to the end of the list
|
|
tabs = self.get_stroke_last_tabs()
|
|
for tab in tabs:
|
|
tab.apply()
|
|
if tab.enabled() and not tab.is_dependent_tab():
|
|
nodes.extend(tab.nodes)
|
|
|
|
check_stop_flag()
|
|
|
|
# sort nodes into the proper stacking order
|
|
nodes.sort(key=lambda node: node.order)
|
|
|
|
try:
|
|
wx.CallAfter(self._hide_warning)
|
|
last_stitch_group = None
|
|
for node, next_node in zip_longest(nodes, self._get_next_nodes(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.
|
|
|
|
stitch_groups.extend(copy(node).embroider(last_stitch_group, next_node))
|
|
if stitch_groups:
|
|
last_stitch_group = stitch_groups[-1]
|
|
|
|
check_stop_flag()
|
|
|
|
if stitch_groups:
|
|
return stitch_groups_to_stitch_plan(
|
|
stitch_groups,
|
|
collapse_len=self.metadata['collapse_len_mm'],
|
|
min_stitch_len=self.metadata['min_stitch_len_mm']
|
|
)
|
|
except (SystemExit, ExitThread):
|
|
raise
|
|
except InkstitchException as exc:
|
|
wx.CallAfter(self._show_warning, str(exc))
|
|
except Exception:
|
|
wx.CallAfter(self._show_warning, format_uncaught_exception())
|
|
|
|
def _get_next_nodes(self, nodes):
|
|
if len(nodes) > 1:
|
|
return nodes[1:]
|
|
return []
|
|
|
|
def get_stroke_last_tabs(self):
|
|
tabs = self.tabs
|
|
stroke_index = [tabs.index(tab) for tab in tabs if tab.name == _("Stroke")]
|
|
if stroke_index:
|
|
tabs.append(tabs.pop(stroke_index[0]))
|
|
return tabs
|
|
|
|
def on_stitch_plan_rendered(self, stitch_plan):
|
|
try:
|
|
self.simulator.stop()
|
|
self.simulator.load(stitch_plan)
|
|
self.simulator.go()
|
|
except RuntimeError:
|
|
# this can happen when they close the window at a bad time
|
|
pass
|
|
|
|
def _hide_warning(self):
|
|
self.warning_panel.clear()
|
|
self.warning_panel.Hide()
|
|
self.Layout()
|
|
|
|
def _show_warning(self, warning_text):
|
|
self.warning_panel.set_warning_text(warning_text)
|
|
self.warning_panel.Show()
|
|
self.Layout()
|
|
|
|
def get_preset_data(self):
|
|
# called by self.presets_panel
|
|
|
|
preset = {}
|
|
|
|
current_tab = self.tabs[self.notebook.GetSelection()]
|
|
while current_tab.parent_tab:
|
|
current_tab = current_tab.parent_tab
|
|
|
|
tabs = [current_tab]
|
|
if current_tab.paired_tab:
|
|
tabs.append(current_tab.paired_tab)
|
|
tabs.extend(current_tab.paired_tab.dependent_tabs)
|
|
tabs.extend(current_tab.dependent_tabs)
|
|
|
|
for tab in tabs:
|
|
tab.save_preset(preset)
|
|
|
|
return preset
|
|
|
|
def apply_preset_data(self, preset_data):
|
|
# called by self.presets_panel
|
|
|
|
for tab in self.tabs:
|
|
tab.load_preset(preset_data)
|
|
|
|
self.preview_renderer.update()
|
|
|
|
def _apply(self):
|
|
for tab in self.tabs:
|
|
tab.apply()
|
|
|
|
def apply(self, event):
|
|
self._apply()
|
|
self.presets_panel.store_preset("__LAST__", self.get_preset_data())
|
|
self.close()
|
|
|
|
def use_last(self, event):
|
|
self.presets_panel.load_preset("__LAST__")
|
|
self.apply(event)
|
|
|
|
def close(self):
|
|
self.simulator.stop()
|
|
wx.CallAfter(self.GetTopLevelParent().close)
|
|
|
|
def cancel(self, event):
|
|
self.simulator.stop()
|
|
wx.CallAfter(self.GetTopLevelParent().cancel)
|
|
|
|
def __do_layout(self):
|
|
# begin wxGlade: MyFrame.__do_layout
|
|
sizer_1 = wx.BoxSizer(wx.VERTICAL)
|
|
# self.sizer_3_staticbox.Lower()
|
|
sizer_3 = wx.BoxSizer(wx.HORIZONTAL)
|
|
for tab in self.tabs:
|
|
self.notebook.AddPage(tab, tab.name)
|
|
sizer_1.Add(self.warning_panel, 0, flag=wx.EXPAND | wx.ALL, border=10)
|
|
sizer_1.Add(self.notebook, 1, wx.EXPAND |
|
|
wx.LEFT | wx.TOP | wx.RIGHT, 10)
|
|
sizer_1.Add(self.presets_panel, 0, flag=wx.EXPAND | wx.ALL, border=10)
|
|
sizer_3.Add(self.cancel_button, 0, wx.RIGHT, 5)
|
|
sizer_3.Add(self.use_last_button, 0, wx.RIGHT | wx.BOTTOM, 5)
|
|
sizer_3.Add(self.apply_button, 0, wx.RIGHT | wx.BOTTOM, 5)
|
|
sizer_1.Add(sizer_3, 0, wx.ALIGN_RIGHT, 0)
|
|
self.SetSizer(sizer_1)
|
|
sizer_1.Fit(self)
|
|
|
|
# prevent the param dialog to become smaller than 500*400
|
|
# otherwise the scrollbar jumps to the side and produces a
|
|
# deprecation warning in wxpythons scrolledpanel.py line 225 -
|
|
# which is expecting an integer, but uses previously a division
|
|
# self.SetSizeHints(600, 400)
|
|
|
|
self.Layout()
|
|
# end wxGlade
|
|
|
|
|
|
class NoValidObjects(Exception):
|
|
pass
|
|
|
|
|
|
class Params(InkstitchExtension):
|
|
def __init__(self, *args, **kwargs):
|
|
self.cancelled = False
|
|
InkstitchExtension.__init__(self, *args, **kwargs)
|
|
|
|
def embroidery_classes(self, node):
|
|
element = EmbroideryElement(node)
|
|
classes = []
|
|
|
|
if not is_command(node) and not is_command_symbol(node):
|
|
if is_clone(node):
|
|
classes.append(Clone)
|
|
elif node.tag in EMBROIDERABLE_TAGS and not node.get_path():
|
|
pass
|
|
else:
|
|
if element.get_style("fill", 'black') and not element.get_style("fill-opacity", 1) == "0":
|
|
classes.append(FillStitch)
|
|
if element.get_style("stroke") is not None:
|
|
classes.append(Stroke)
|
|
if len(element.path) > 1:
|
|
classes.append(SatinColumn)
|
|
return classes
|
|
|
|
def get_nodes_by_class(self):
|
|
# returns embroidery elements (not nodes) by class
|
|
# TODO: rename this
|
|
nodes = self.get_nodes()
|
|
nodes_by_class = defaultdict(list)
|
|
|
|
for z, node in enumerate(nodes):
|
|
for cls in self.embroidery_classes(node):
|
|
element = cls(node)
|
|
element.order = z
|
|
nodes_by_class[cls].append(element)
|
|
|
|
return sorted(list(nodes_by_class.items()), key=lambda cls_nodes: cls_nodes[0].__name__)
|
|
|
|
def get_values(self, param, nodes):
|
|
if param.type in ('toggle', 'boolean'):
|
|
getter = 'get_boolean_param'
|
|
values = [item for item in (getattr(node, getter)(
|
|
param.name, param.default) for node in nodes) if item is not None]
|
|
else:
|
|
getter = 'get_param'
|
|
values = [item if item is not None else "" for item in (getattr(node, getter)(
|
|
param.name, param.default) for node in nodes)]
|
|
|
|
return values
|
|
|
|
def group_params(self, params):
|
|
def by_group_and_sort_index(param):
|
|
return param.group or "", param.sort_index
|
|
|
|
def by_group(param):
|
|
return param.group or ""
|
|
|
|
return groupby(sorted(params, key=by_group_and_sort_index), by_group)
|
|
|
|
def sort_tabs(self, tabs):
|
|
def tab_sort_key(tab):
|
|
parent = tab.parent_tab or tab
|
|
|
|
sort_key = (
|
|
# For Stroke and SatinColumn, place the one that's
|
|
# enabled first. Place dependent tabs first too.
|
|
parent.toggle and parent.toggle_checkbox.IsChecked(),
|
|
|
|
# If multiple tabs are enabled, make sure dependent
|
|
# tabs are grouped with the parent.
|
|
parent and parent.name,
|
|
|
|
# Within parent/dependents, put the parent first.
|
|
tab == parent
|
|
)
|
|
|
|
return sort_key
|
|
|
|
tabs.sort(key=tab_sort_key, reverse=True)
|
|
|
|
def pair_tabs(self, tabs):
|
|
for tab in tabs:
|
|
if tab.toggle and tab.toggle.inverse:
|
|
for other_tab in tabs:
|
|
if other_tab != tab and other_tab.toggle.name == tab.toggle.name:
|
|
tab.pair(other_tab)
|
|
other_tab.pair(tab)
|
|
|
|
def assign_parents(self, tabs, parent_tab):
|
|
for tab in tabs:
|
|
if tab != parent_tab:
|
|
parent_tab.add_dependent_tab(tab)
|
|
tab.set_parent_tab(parent_tab)
|
|
|
|
def create_tabs(self, parent):
|
|
tabs = []
|
|
nodes_by_class = self.get_nodes_by_class()
|
|
|
|
if not nodes_by_class:
|
|
raise NoValidObjects()
|
|
|
|
for cls, nodes in self.get_nodes_by_class():
|
|
params = cls.get_params()
|
|
|
|
for param in params:
|
|
param.values = list(set(self.get_values(param, nodes)))
|
|
|
|
parent_tab = None
|
|
new_tabs = []
|
|
|
|
for group, params in self.group_params(params):
|
|
tab_name = group or cls.element_name
|
|
tab = ParamsTab(parent, id=wx.ID_ANY, name=tab_name,
|
|
params=list(params), nodes=nodes)
|
|
new_tabs.append(tab)
|
|
|
|
if group == "":
|
|
parent_tab = tab
|
|
|
|
self.assign_parents(new_tabs, parent_tab)
|
|
tabs.extend(new_tabs)
|
|
|
|
self.pair_tabs(tabs)
|
|
self.sort_tabs(tabs)
|
|
|
|
return tabs
|
|
|
|
def cancel(self):
|
|
self.cancelled = True
|
|
|
|
def effect(self):
|
|
try:
|
|
app = wx.App()
|
|
metadata = self.get_inkstitch_metadata()
|
|
background_color = get_pagecolor(self.svg.namedview)
|
|
frame = SplitSimulatorWindow(
|
|
title=_("Embroidery Params"),
|
|
panel_class=SettingsPanel,
|
|
tabs_factory=self.create_tabs,
|
|
on_cancel=self.cancel,
|
|
metadata=metadata,
|
|
background_color=background_color,
|
|
target_duration=5
|
|
)
|
|
|
|
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)
|
|
except NoValidObjects:
|
|
self.no_elements_error()
|