kopia lustrzana https://github.com/inkstitch/inkstitch
65 wiersze
2.0 KiB
Python
65 wiersze
2.0 KiB
Python
import sys
|
|
|
|
import inkex
|
|
|
|
from ..elements import SatinColumn, Stroke
|
|
from ..i18n import _
|
|
from ..stitches.auto_satin import auto_satin
|
|
from .commands import CommandsExtension
|
|
|
|
|
|
class AutoSatin(CommandsExtension):
|
|
COMMANDS = ["trim"]
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
CommandsExtension.__init__(self, *args, **kwargs)
|
|
|
|
self.arg_parser.add_argument("-p", "--preserve_order", dest="preserve_order", type=inkex.Boolean, default=False)
|
|
|
|
def get_starting_point(self):
|
|
return self.get_point("satin_start")
|
|
|
|
def get_ending_point(self):
|
|
return self.get_point("satin_end")
|
|
|
|
def get_point(self, command_type):
|
|
command = None
|
|
for satin in self.elements:
|
|
this_command = satin.get_command(command_type)
|
|
if command is not None and this_command:
|
|
inkex.errormsg(_("Please ensure that at most one start and end command is attached to the selected satin columns."))
|
|
sys.exit(0)
|
|
elif this_command:
|
|
command = this_command
|
|
|
|
if command is not None:
|
|
return command.target_point
|
|
|
|
def check_selection(self):
|
|
if not self.get_elements():
|
|
return
|
|
|
|
if not self.svg.selected:
|
|
# L10N auto-route satin columns extension
|
|
inkex.errormsg(_("Please select one or more satin columns."))
|
|
return False
|
|
|
|
satincolumns = [element for element in self.elements if isinstance(element, SatinColumn)]
|
|
if len(satincolumns) == 0:
|
|
inkex.errormsg(_("Please select at least one satin column."))
|
|
return False
|
|
|
|
return True
|
|
|
|
def effect(self):
|
|
if not self.check_selection():
|
|
return
|
|
|
|
starting_point = self.get_starting_point()
|
|
ending_point = self.get_ending_point()
|
|
|
|
# Ignore fills
|
|
elements = [element for element in self.elements if isinstance(element, SatinColumn) or isinstance(element, Stroke)]
|
|
|
|
auto_satin(elements, self.options.preserve_order, starting_point, ending_point, self.options.trim)
|