inkstitch/lib/extensions/jump_to_stroke.py

65 wiersze
2.7 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 inkex import DirectedLineSegment, PathElement, errormsg
from ..i18n import _
from ..svg import PIXELS_PER_MM, get_correction_transform
from ..svg.tags import INKSTITCH_ATTRIBS
from .base import InkstitchExtension
class JumpToStroke(InkstitchExtension):
"""Adds a running stitch as a connection between two (or more) selected elements.
The elements must have the same color and a minimum distance (collapse_len)."""
def __init__(self, *args, **kwargs):
InkstitchExtension.__init__(self, *args, **kwargs)
self.arg_parser.add_argument("-l", "--stitch-length", type=float, default=2.5, dest="running_stitch_length_mm")
self.arg_parser.add_argument("-t", "--tolerance", type=float, default=2.0, dest="running_stitch_tolerance_mm")
def effect(self):
if not self.svg.selection or not self.get_elements() or len(self.elements) < 2:
errormsg(_("Please select at least two elements to convert the jump stitch to a running stitch."))
return
last_stitch_group = None
last_color = None
for element in self.elements:
stitch_group = element.to_stitch_groups(last_stitch_group)
end = stitch_group[-1].stitches[0]
if last_stitch_group is not None and element.color == last_color:
start = last_stitch_group.stitches[-1]
self.generate_stroke(element, start, end)
last_stitch_group = stitch_group[-1]
last_color = element.color
def generate_stroke(self, element, start, end):
node = element.node
parent = node.getparent()
index = parent.index(node)
# do not add a running stitch if the distance is smaller than the collapse setting
self.metadata = self.get_inkstitch_metadata()
collapse_len = self.metadata['collapse_len_mm'] or 3.0
collapse_len *= PIXELS_PER_MM
line = DirectedLineSegment((start.x, start.y), (end.x, end.y))
if collapse_len > line.length:
return
path = f'M {start.x}, {start.y} L {end.x}, {end.y}'
color = element.color
style = f'stroke:{color};stroke-width:1px;stroke-dasharray:3, 1;fill:none;'
line = PathElement(d=path, style=style, transform=get_correction_transform(node))
line.set(INKSTITCH_ATTRIBS['running_stitch_length_mm'], self.options.running_stitch_length_mm)
line.set(INKSTITCH_ATTRIBS['running_stitch_tolerance_mm'], self.options.running_stitch_tolerance_mm)
parent.insert(index, line)
if __name__ == '__main__':
JumpToStroke().run()