# 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()