inkstitch/lib/gui/tartan/customize_panel.py

301 wiersze
14 KiB
Python

# Authors: see git history
#
# Copyright (c) 2023 Authors
# Licensed under the GNU GPL version 3.0 or later. See the file LICENSE for details.
from math import floor
import wx
from wx.lib.scrolledpanel import ScrolledPanel
from ...i18n import _
class CustomizePanel(ScrolledPanel):
def __init__(self, parent, panel):
self.panel = panel
self.mouse_position = None
ScrolledPanel.__init__(self, parent)
self.customize_sizer = wx.BoxSizer(wx.VERTICAL)
general_settings_sizer = wx.BoxSizer(wx.HORIZONTAL)
positional_settings_sizer = wx.FlexGridSizer(2, 4, 5, 5)
stripe_header_sizer = wx.BoxSizer(wx.HORIZONTAL)
self.stripe_sizer = wx.BoxSizer(wx.HORIZONTAL)
self.warp_outer_sizer = wx.BoxSizer(wx.VERTICAL)
self.weft_outer_sizer = wx.BoxSizer(wx.VERTICAL)
self.warp_sizer = wx.BoxSizer(wx.VERTICAL)
self.weft_sizer = wx.BoxSizer(wx.VERTICAL)
general_settings_headline = wx.StaticText(self, label=_("Pattern Settings"))
general_settings_headline.SetFont(wx.Font().Bold())
self.symmetry_checkbox = wx.CheckBox(self, label=_("Symmetrical / reflective sett"))
self.symmetry_checkbox.SetToolTip(_("Disabled: asymmetrical / repeating sett"))
self.symmetry_checkbox.Bind(wx.EVT_CHECKBOX, self.update_symmetry)
self.warp_weft_checkbox = wx.CheckBox(self, label=_("Equal threadcount for warp and weft"))
self.warp_weft_checkbox.Bind(wx.EVT_CHECKBOX, self._update_warp_weft_event)
positional_settings_headline = wx.StaticText(self, label=_("Position"))
positional_settings_headline.SetFont(wx.Font().Bold())
self.rotate = wx.SpinCtrlDouble(self, min=-180, max=180, initial=0, inc=0.1, style=wx.SP_WRAP)
self.rotate.SetDigits(2)
self.rotate.Bind(wx.EVT_SPINCTRLDOUBLE, lambda event: self.on_change("rotate", event))
rotate_label = wx.StaticText(self, label=_("Rotate"))
self.scale = wx.SpinCtrl(self, min=0, max=1000, initial=100, style=wx.SP_WRAP)
self.scale.Bind(wx.EVT_SPINCTRL, self.update_scale)
scale_label = wx.StaticText(self, label=_("Scale (%)"))
self.offset_x = wx.SpinCtrlDouble(self, min=0, max=500, initial=0, style=wx.SP_WRAP)
self.offset_x.Bind(wx.EVT_SPINCTRLDOUBLE, lambda event: self.on_change("offset_x", event))
self.offset_x.SetDigits(2)
offset_x_label = wx.StaticText(self, label=_("Offset X (mm)"))
self.offset_y = wx.SpinCtrlDouble(self, min=0, max=500, initial=0, style=wx.SP_WRAP)
self.offset_y.Bind(wx.EVT_SPINCTRLDOUBLE, lambda event: self.on_change("offset_y", event))
self.offset_y.SetDigits(2)
offset_y_label = wx.StaticText(self, label=_("Offset Y (mm)"))
stripe_settings_headline = wx.StaticText(self, label=_("Stripes"))
stripe_settings_headline.SetFont(wx.Font().Bold())
self.link_colors_checkbox = wx.CheckBox(self, label=_("Link colors"))
self.link_colors_checkbox.SetToolTip(_("When enabled update all equal colors simultaneously."))
self.warp_headline = wx.StaticText(self, label=_("Warp"))
self.warp_headline.SetFont(wx.Font().Bold())
self.weft_headline = wx.StaticText(self, label=_("Weft"))
self.weft_headline.SetFont(wx.Font().Bold())
self.add_warp_button = wx.Button(self, label=_("Add"))
self.add_warp_button.Bind(wx.EVT_BUTTON, self._add_warp_event)
self.add_weft_button = wx.Button(self, label=_("Add"))
self.add_weft_button.Bind(wx.EVT_BUTTON, self._add_weft_event)
# Add to sizers
general_settings_sizer.Add(self.symmetry_checkbox, 0, wx.CENTER | wx.ALL, 10)
general_settings_sizer.Add(self.warp_weft_checkbox, 0, wx.CENTER | wx.ALL, 10)
positional_settings_sizer.Add(rotate_label, 0, wx.ALIGN_CENTRE, 0)
positional_settings_sizer.Add(self.rotate, 0, wx.EXPAND | wx.RIGHT, 30)
positional_settings_sizer.Add(offset_x_label, 0, wx.ALIGN_CENTRE, 0)
positional_settings_sizer.Add(self.offset_x, 0, wx.EXPAND, 0)
positional_settings_sizer.Add(scale_label, 0, wx.ALIGN_CENTRE, 0)
positional_settings_sizer.Add(self.scale, 0, wx.EXPAND | wx.RIGHT, 30)
positional_settings_sizer.Add(offset_y_label, 0, wx.ALIGN_CENTRE, 0)
positional_settings_sizer.Add(self.offset_y, 0, wx.EXPAND, 0)
positional_settings_sizer.AddGrowableCol(1)
positional_settings_sizer.AddGrowableCol(3)
self.warp_outer_sizer.Add(self.warp_headline, 0, wx.EXPAND, 0)
self.weft_outer_sizer.Add(self.weft_headline, 0, wx.EXPAND, 0)
self.warp_outer_sizer.Add(self.warp_sizer, 0, wx.EXPAND, 0)
self.weft_outer_sizer.Add(self.weft_sizer, 0, wx.EXPAND, 0)
self.warp_outer_sizer.Add(self.add_warp_button, 0, wx.ALIGN_RIGHT | wx.ALL, 10)
self.weft_outer_sizer.Add(self.add_weft_button, 0, wx.ALIGN_RIGHT | wx.ALL, 10)
self.stripe_sizer.Add(self.warp_outer_sizer, 1, wx.EXPAND, 0)
self.stripe_sizer.Add(self.weft_outer_sizer, 1, wx.EXPAND, 0)
stripe_header_sizer.Add(stripe_settings_headline, 0, wx.ALL, 10)
stripe_header_sizer.Add((0, 0), 1, wx.ALL | wx.EXPAND, 10)
stripe_header_sizer.Add(self.link_colors_checkbox, 0, wx.ALL, 10)
self.customize_sizer.Add(positional_settings_headline, 0, wx.ALL, 10)
self.customize_sizer.Add(positional_settings_sizer, 0, wx.ALL | wx.EXPAND, 10)
self.customize_sizer.Add(wx.StaticLine(self), 0, wx.ALL | wx.EXPAND, 10)
self.customize_sizer.Add(general_settings_headline, 0, wx.ALL, 10)
self.customize_sizer.Add(general_settings_sizer, 0, wx.ALL | wx.EXPAND, 10)
self.customize_sizer.Add(wx.StaticLine(self), 0, wx.ALL | wx.EXPAND, 10)
self.customize_sizer.Add(stripe_header_sizer, 0, wx.EXPAND | wx.ALL, 10)
self.customize_sizer.Add(self.stripe_sizer, 0, wx.EXPAND | wx.ALL, 10)
self.SetSizer(self.customize_sizer)
def _add_warp_event(self, event):
self.add_stripe()
def _add_weft_event(self, event):
self.add_stripe(False)
def add_stripe(self, warp=True, stripe=None, update=True):
stripesizer = wx.BoxSizer(wx.HORIZONTAL)
position = wx.Button(self, label='', style=wx.BU_EXACTFIT)
position.SetToolTip(_("Drag and drop to adjust position."))
position.Bind(wx.EVT_LEFT_DOWN, self._move_stripe_start)
position.Bind(wx.EVT_LEFT_UP, self._move_stripe_end)
visibility = wx.CheckBox(self)
visibility.SetToolTip(_("Stitch this stripe"))
visibility.SetValue(True)
visibility.Bind(wx.EVT_CHECKBOX, self._update_stripes_event)
# hidden label used for linked colors
# there seems to be no native way to catch the old color setting
colorinfo = wx.StaticText(self, label='black')
colorinfo.Hide()
colorpicker = wx.ColourPickerCtrl(self, colour=wx.Colour('black'))
colorpicker.SetToolTip(_("Select stripe color"))
colorpicker.Bind(wx.EVT_COLOURPICKER_CHANGED, self._update_color)
stripe_width = wx.SpinCtrlDouble(self, min=0.01, max=500, initial=5, style=wx.SP_WRAP)
stripe_width.SetDigits(2)
stripe_width.SetToolTip(_("Set stripe width (mm)"))
stripe_width.Bind(wx.EVT_SPINCTRLDOUBLE, self._update_stripes_event)
remove_button = wx.Button(self, label='X')
remove_button.SetToolTip(_("Remove stripe"))
remove_button.Bind(wx.EVT_BUTTON, self._remove_stripe)
stripesizer.Add(position, 0, wx.CENTER | wx.RIGHT | wx.TOP, 5)
stripesizer.Add(visibility, 0, wx.CENTER | wx.RIGHT | wx.TOP, 5)
stripesizer.Add(colorinfo, 0, wx.RIGHT | wx.TOP, 5)
stripesizer.Add(colorpicker, 0, wx.RIGHT | wx.TOP, 5)
stripesizer.Add(stripe_width, 1, wx.RIGHT | wx.TOP, 5)
stripesizer.Add(remove_button, 0, wx.CENTER | wx.TOP, 5)
if stripe is not None:
visibility.SetValue(stripe['render'])
colorinfo.SetLabel(wx.Colour(stripe['color']).GetAsString(wx.C2S_HTML_SYNTAX))
colorpicker.SetColour(wx.Colour(stripe['color']))
stripe_width.SetValue(stripe['width'])
if warp:
self.warp_sizer.Add(stripesizer, 0, wx.EXPAND | wx.ALL, 5)
else:
self.weft_sizer.Add(stripesizer, 0, wx.EXPAND | wx.ALL, 5)
if update:
self.panel.update_from_stripes()
self.set_stripe_width_color(stripe_width)
self.FitInside()
def _move_stripe_start(self, event):
self.mouse_position = wx.GetMousePosition()
def _move_stripe_end(self, event):
stripe = event.GetEventObject()
sizer = stripe.GetContainingSizer()
if self.warp_sizer.GetItem(sizer):
main_sizer = self.warp_sizer
else:
main_sizer = self.weft_sizer
for i, item in enumerate(main_sizer.GetChildren()):
if item.GetSizer() == sizer:
index = i
break
position = wx.GetMousePosition()
sizer_height = sizer.GetSize()[1] + 10
move = floor((position[1] - self.mouse_position[1]) / sizer_height)
index = min(len(main_sizer.Children) - 1, max(0, (index + move)))
main_sizer.Detach(sizer)
main_sizer.Insert(index, sizer, 0, wx.EXPAND | wx.ALL, 5)
self.panel.update_from_stripes()
self.FitInside()
def _remove_stripe(self, event):
sizer = event.GetEventObject().GetContainingSizer()
sizer.Clear(True)
self.warp_sizer.Remove(sizer)
try:
self.weft_sizer.Remove(sizer)
except RuntimeError:
# we may have removed it already
pass
self.panel.update_from_stripes()
self.FitInside()
def on_change(self, attribute, event):
self.panel.settings[attribute] = event.EventObject.GetValue()
self.panel.update_preview()
def update_scale(self, event):
self.panel.settings['scale'] = event.EventObject.GetValue()
# self.update_stripes(self.panel.palette.palette_stripes)
self.update_stripe_width_colors()
self.panel.update_preview()
def _update_stripes_event(self, event):
self.set_stripe_width_color(event.EventObject)
self.panel.update_from_stripes()
def update_stripe_width_colors(self):
for sizer in [self.warp_sizer, self.weft_sizer]:
for stripe_sizer in sizer.GetChildren():
inner_sizer = stripe_sizer.GetSizer()
for stripe_widget in inner_sizer:
widget = stripe_widget.GetWindow()
if isinstance(widget, wx.SpinCtrlDouble):
self.set_stripe_width_color(widget)
def set_stripe_width_color(self, stripe_width_ctrl):
scale = self.scale.GetValue()
min_stripe_width = self.panel.embroidery_panel.min_stripe_width.GetValue()
stripe_width = stripe_width_ctrl.GetValue() * scale / 100
if stripe_width <= min_stripe_width:
stripe_width_ctrl.SetBackgroundColour(wx.Colour('#efefef'))
stripe_width_ctrl.SetForegroundColour('black')
else:
stripe_width_ctrl.SetBackgroundColour(wx.NullColour)
stripe_width_ctrl.SetForegroundColour(wx.NullColour)
def update_stripes(self, stripes):
self.warp_sizer.Clear(True)
self.weft_sizer.Clear(True)
warp = True
for direction in stripes:
for stripe in direction:
self.add_stripe(warp, stripe, False)
warp = False
self.panel.update_from_stripes()
def _update_color(self, event):
linked = self.link_colors_checkbox.GetValue()
widget = event.GetEventObject()
colorinfo = widget.GetPrevSibling()
old_color = wx.Colour(colorinfo.GetLabel())
new_color = event.Colour
if linked:
self._update_color_picker(old_color, new_color, self.warp_sizer)
self._update_color_picker(old_color, new_color, self.weft_sizer)
colorinfo.SetLabel(new_color.GetAsString(wx.C2S_HTML_SYNTAX))
self.panel.update_from_stripes()
def _update_color_picker(self, old_color, new_color, sizer):
for stripe_sizer in sizer.Children:
stripe_info = stripe_sizer.GetSizer()
for widget in stripe_info.GetChildren():
widget = widget.GetWindow()
if isinstance(widget, wx.ColourPickerCtrl):
color = widget.GetColour()
if color == old_color:
widget.SetColour(new_color)
widget.GetPrevSibling().SetLabel(new_color.GetAsString(wx.C2S_HTML_SYNTAX))
def update_symmetry(self, event=None):
symmetry = self.symmetry_checkbox.GetValue()
self.panel.settings['symmetry'] = symmetry
self.panel.palette.update_symmetry(symmetry)
self.panel.update_from_stripes()
self.FitInside()
def update_warp_weft(self):
equal_warp_weft = self.warp_weft_checkbox.GetValue()
if equal_warp_weft:
self.stripe_sizer.Hide(self.warp_headline, recursive=True)
self.stripe_sizer.Hide(self.weft_outer_sizer, recursive=True)
else:
self.stripe_sizer.Show(self.warp_headline, recursive=True)
self.stripe_sizer.Show(self.weft_outer_sizer, recursive=True)
# We just made the weft colorinfo visible. Let's hide it again.
self._hide_colorinfo()
self.FitInside()
def _update_warp_weft_event(self, event):
self.panel.settings['equal_warp_weft'] = event.GetEventObject().GetValue()
self.update_warp_weft()
self.panel.update_from_stripes()
def _hide_colorinfo(self):
for stripe_sizer in self.weft_sizer.Children:
stripe_info = stripe_sizer.GetSizer()
for stripe in stripe_info.GetChildren():
widget = stripe.GetWindow()
if isinstance(widget, wx.StaticText):
widget.Hide()