diff options
Diffstat (limited to 'lib/extensions/element_info.py')
| -rw-r--r-- | lib/extensions/element_info.py | 164 |
1 files changed, 164 insertions, 0 deletions
diff --git a/lib/extensions/element_info.py b/lib/extensions/element_info.py new file mode 100644 index 00000000..a086108b --- /dev/null +++ b/lib/extensions/element_info.py @@ -0,0 +1,164 @@ +# 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 ..elements import FillStitch, SatinColumn, Stroke +from ..gui.element_info import ElementInfoApp +from ..i18n import _ +from ..stitch_plan import stitch_groups_to_stitch_plan +from ..svg import PIXELS_PER_MM +from .base import InkstitchExtension + + +class ElementInfo(InkstitchExtension): + + def effect(self): + if not self.svg.selection or not self.get_elements(): + return + + self.metadata = self.get_inkstitch_metadata() + self.list_items = [] + + self._general_info() + + previous_stitch_group = None + for element in self.elements: + previous_stitch_group = self._element_info(element, previous_stitch_group) + + app = ElementInfoApp(self.list_items) + app.MainLoop() + + def _element_info(self, element, previous_stitch_group): + stitch_groups = element.to_stitch_groups(previous_stitch_group) + stitch_plan = stitch_groups_to_stitch_plan( + stitch_groups, + collapse_len=self.metadata['collapse_len'], + min_stitch_len=self.metadata['min_stitch_len'] + ) + + self.list_items.append(ListItem( + name=f"{ element.node.label } ({ element.node.get_id() })", + headline=True + )) + self.list_items.append(ListItem( + name=_("Type"), + value=element.element_name + )) + if isinstance(element, FillStitch): + fill_method = next((method.name for method in element._fill_methods if method.id == element.fill_method), "") + self.list_items.append(ListItem( + name=_("Fill Method"), + value=fill_method + )) + + if isinstance(element, SatinColumn): + satin_method = next((method.name for method in element._satin_methods if method.id == element.satin_method), "") + self.list_items.append(ListItem( + name=_("Satin Method"), + value=satin_method + )) + + if isinstance(element, Stroke): + stroke_method = next((method.name for method in element._stroke_methods if method.id == element.stroke_method), "") + self.list_items.append(ListItem( + name=_("Satin Method"), + value=stroke_method + )) + + self.list_items.append(ListItem( + name=_("Dimensions (mm)"), + value="{:.2f} x {:.2f}".format(stitch_plan.dimensions_mm[0], stitch_plan.dimensions_mm[1]) + )) + + stitch_lengths = [] + for group in stitch_groups: + stitches = group.stitches + + previous_stitch = stitches[0] + for stitch in stitches[1:]: + st = stitch - previous_stitch + length = st.length() / PIXELS_PER_MM + if length <= self.metadata['min_stitch_len_mm']: + continue + stitch_lengths.append(length) + previous_stitch = stitch + + stitches_per_group = "" + if len(stitch_groups) > 1: + stitches_per_group = f" ({', '.join([str(len(group.stitches)) for group in stitch_groups]) })" + + self.list_items.append(ListItem( + name=_("Stitches"), + value=str(stitch_plan.num_stitches - stitch_plan.num_jumps) + stitches_per_group + )) + self.list_items.append(ListItem( + name=_("Jumps"), + value=str(stitch_plan.num_jumps - 1) + )) + self.list_items.append(ListItem( + name=_("Max stitch length"), + value="{:.2f}".format(max(stitch_lengths)) + )) + self.list_items.append(ListItem( + name=_("Min stitch length"), + value="{:.2f}".format(min(stitch_lengths)) + )) + self.list_items.append(ListItem()) + return stitch_groups[0] + + def _general_info(self): + stitch_groups = self.elements_to_stitch_groups(self.elements) + stitch_plan = stitch_groups_to_stitch_plan( + stitch_groups, + collapse_len=self.metadata['collapse_len'], + min_stitch_len=self.metadata['min_stitch_len'] + ) + + self.list_items.append(ListItem( + name=_("All Selected Elements"), + headline=True + )) + self.list_items.append(ListItem( + name=_("Dimensions (mm)"), + value="{:.2f} x {:.2f}".format(stitch_plan.dimensions_mm[0], stitch_plan.dimensions_mm[1]) + )) + self.list_items.append(ListItem( + name=_("Colors"), + value=str(stitch_plan.num_colors) + )) + self.list_items.append(ListItem( + name=_("Color Changes"), + value=str(stitch_plan.num_color_blocks - 1) + )) + self.list_items.append(ListItem( + name=_("Jumps"), + value=str(stitch_plan.num_jumps) + )) + self.list_items.append(ListItem( + name=_("Trims"), + value=str(stitch_plan.num_trims) + )) + self.list_items.append(ListItem( + name=_("Stops"), + value=str(stitch_plan.num_stops) + )) + self.list_items.append(ListItem( + name=_("Stitches"), + value=str(stitch_plan.num_stitches - stitch_plan.num_jumps) + )) + self.list_items.append(ListItem( + name=_("Filter stitches smaller than (mm)"), + value=str(self.metadata['min_stitch_len_mm']) + )) + self.list_items.append(ListItem()) + + +class ListItem: + def __init__(self, name="", value="", headline=False): + self.name: str = name + self.value: str = value + self.headline: bool = headline + + def __repr__(self): + return "ListItem(%s, %s, %s)" % (self.name, self.value, self.headline) |