inkstitch/lib/extensions/jump_to_trim.py

65 wiersze
2.4 KiB
Python

# Authors: see git history
#
# Copyright (c) 2024 Authors
# Licensed under the GNU GPL version 3.0 or later. See the file LICENSE for details.
from inkex import Boolean, DirectedLineSegment
from ..commands import add_commands
from ..svg import PIXELS_PER_MM
from .base import InkstitchExtension
class JumpToTrim(InkstitchExtension):
"""Adds a trim command to avoid jump stitches."""
def __init__(self, *args, **kwargs):
InkstitchExtension.__init__(self, *args, **kwargs)
self.arg_parser.add_argument("--tab")
self.arg_parser.add_argument("-i", "--minimum-jump-length", type=float, default=3.0, dest="min_jump")
self.arg_parser.add_argument("-a", "--maximum-jump-length", type=float, default=0, dest="max_jump")
self.arg_parser.add_argument("-t", "--use-command-symbols", type=Boolean, default=False, dest="use_command_symbols")
def effect(self):
self._set_selection()
self.get_elements()
last_element = None
last_stitch_group = None
for element in self.elements:
stitch_groups = element.to_stitch_groups(last_stitch_group)
for stitch_group in stitch_groups:
if last_stitch_group is None or stitch_group.color != last_stitch_group.color:
last_stitch_group = stitch_group
continue
start = last_stitch_group.stitches[-1]
end = stitch_group.stitches[0]
last_stitch_group = stitch_group
line = DirectedLineSegment((start.x, start.y), (end.x, end.y))
# do not add a running stitch if the distance is smaller than min_jump setting
if line.length < self.options.min_jump * PIXELS_PER_MM:
continue
# do not add a running stitch if the distance is longer than max_jump setting
if self.options.max_jump > 0 and line.length > self.options.max_jump * PIXELS_PER_MM:
continue
self._add_trim(last_element)
last_element = element
def _add_trim(self, element):
if self.options.use_command_symbols:
add_commands(element, ["trim"])
else:
element.node.set('inkstitch:trim_after', True)
def _set_selection(self):
if not self.svg.selection:
self.svg.selection.clear()
if __name__ == '__main__':
JumpToTrim().run()