summaryrefslogtreecommitdiff
path: root/lib/extensions/jump_to_stroke.py
blob: 7cf6ff17e353ab4906e3af66ab90bfe0c6131ccd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
# Authors: see git history
#
# Copyright (c) 2010 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 .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 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[-1]
            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))
        parent.insert(index, line)


if __name__ == '__main__':
    JumpToStroke().run()