kopia lustrzana https://github.com/inkstitch/inkstitch
686 wiersze
28 KiB
Python
686 wiersze
28 KiB
Python
# Authors: see git history
|
||
#
|
||
# Copyright (c) 2010 Authors
|
||
# Licensed under the GNU GPL version 3.0 or later. See the file LICENSE for details.
|
||
|
||
import json
|
||
import os
|
||
from collections import defaultdict
|
||
from copy import deepcopy
|
||
from random import randint
|
||
|
||
import inkex
|
||
|
||
from ..commands import add_commands, ensure_symbol
|
||
from ..elements import SatinColumn, Stroke, nodes_to_elements
|
||
from ..exceptions import InkstitchException
|
||
from ..extensions.lettering_custom_font_dir import get_custom_font_dir
|
||
from ..i18n import _, get_languages
|
||
from ..marker import MARKER, ensure_marker, has_marker, is_grouped_with_marker
|
||
from ..stitches.auto_satin import auto_satin
|
||
from ..svg.clip import get_clips
|
||
from ..svg.tags import (CONNECTION_END, CONNECTION_START, EMBROIDERABLE_TAGS,
|
||
INKSCAPE_LABEL, INKSTITCH_ATTRIBS, SVG_GROUP_TAG,
|
||
SVG_PATH_TAG, SVG_USE_TAG, XLINK_HREF)
|
||
from ..utils import Point
|
||
from .font_variant import FontVariant
|
||
|
||
|
||
class FontError(InkstitchException):
|
||
pass
|
||
|
||
|
||
def font_metadata(name, default=None, multiplier=None):
|
||
def getter(self):
|
||
value = self.metadata.get(name, default)
|
||
|
||
if multiplier is not None:
|
||
value *= multiplier
|
||
|
||
return value
|
||
|
||
return property(getter)
|
||
|
||
|
||
def localized_font_metadata(name, default=None):
|
||
def getter(self):
|
||
# If the font contains a localized version of the attribute, use it.
|
||
for language in get_languages():
|
||
attr = "%s_%s" % (name, language)
|
||
if attr in self.metadata:
|
||
return self.metadata.get(attr)
|
||
|
||
if name in self.metadata:
|
||
# This may be a font packaged with Ink/Stitch, in which case the
|
||
# text will have been sent to CrowdIn for community translation.
|
||
# Try to fetch the translated version.
|
||
original_metadata = self.metadata.get(name)
|
||
localized_metadata = ""
|
||
if original_metadata != "":
|
||
localized_metadata = _(original_metadata)
|
||
return localized_metadata
|
||
else:
|
||
return default
|
||
|
||
return property(getter)
|
||
|
||
|
||
class Font(object):
|
||
"""Represents a font with multiple variants.
|
||
|
||
Each font may have multiple FontVariants for left-to-right, right-to-left,
|
||
etc. Each variant has a set of Glyphs, one per character.
|
||
|
||
Properties:
|
||
path -- the path to the directory containing this font
|
||
metadata -- A dict of information about the font.
|
||
name -- Shortcut property for metadata["name"]
|
||
license -- contents of the font's LICENSE file, or None if no LICENSE file exists.
|
||
variants -- A dict of FontVariants, with keys in FontVariant.VARIANT_TYPES.
|
||
"""
|
||
|
||
def __init__(self, font_path):
|
||
self.path = font_path
|
||
self.metadata = {}
|
||
self.license = None
|
||
self.variants = {}
|
||
|
||
self._load_metadata()
|
||
self._load_license()
|
||
|
||
def _load_metadata(self):
|
||
try:
|
||
with open(os.path.join(self.path, "font.json"), encoding="utf-8-sig") as metadata_file:
|
||
self.metadata = json.load(metadata_file)
|
||
except IOError:
|
||
path = os.path.join(self.path, "font.json")
|
||
msg = _("JSON file missing. Expected a JSON file at the following location:")
|
||
msg += f"\n{path}\n\n"
|
||
msg += _("Generate the JSON file through:\nExtensions > Ink/Stitch > Font Management > Generate JSON...")
|
||
msg += '\n\n'
|
||
inkex.errormsg(msg)
|
||
except json.decoder.JSONDecodeError as exception:
|
||
path = os.path.join(self.path, "font.json")
|
||
msg = _("Corrupt JSON file")
|
||
msg += f" ({exception}):\n{path}\n\n"
|
||
msg += _("Regenerate the JSON file through:\nExtensions > Ink/Stitch > Font Management > Generate JSON...")
|
||
msg += '\n\n'
|
||
inkex.errormsg(msg)
|
||
|
||
def _load_license(self):
|
||
try:
|
||
with open(os.path.join(self.path, "LICENSE"), encoding="utf-8-sig") as license_file:
|
||
self.license = license_file.read()
|
||
except IOError:
|
||
pass
|
||
|
||
def _load_variants(self):
|
||
if not self.variants:
|
||
for variant in FontVariant.VARIANT_TYPES:
|
||
try:
|
||
self.variants[variant] = FontVariant(self.path, variant, self.default_glyph)
|
||
except IOError:
|
||
# we'll deal with missing variants when we apply lettering
|
||
pass
|
||
|
||
name = font_metadata('name', '')
|
||
description = localized_font_metadata('description', '')
|
||
keywords = font_metadata('keywords', '')
|
||
json_default_variant = font_metadata('default_variant', FontVariant.LEFT_TO_RIGHT)
|
||
text_direction = font_metadata('text_direction', 'ltr')
|
||
letter_case = font_metadata('letter_case', '')
|
||
default_glyph = font_metadata('default_glyph', "<EFBFBD>")
|
||
leading = font_metadata('leading', 100)
|
||
kerning_pairs = font_metadata('kerning_pairs', {})
|
||
auto_satin = font_metadata('auto_satin', True)
|
||
min_scale = font_metadata('min_scale', 1.0)
|
||
max_scale = font_metadata('max_scale', 1.0)
|
||
size = font_metadata('size', 0)
|
||
available_glyphs = font_metadata('glyphs', [])
|
||
|
||
# use values from SVG Font, example:
|
||
# <font horiz-adv-x="45" ... <glyph .... horiz-adv-x="49" glyph-name="A" /> ... <hkern ... k="3"g1="A" g2="B" /> .... />
|
||
|
||
# Example font.json : "horiz_adv_x": {"A":49},
|
||
horiz_adv_x = font_metadata('horiz_adv_x', {})
|
||
|
||
# Example font.json : "horiz_adv_x_default" : 45,
|
||
horiz_adv_x_default = font_metadata('horiz_adv_x_default')
|
||
|
||
# Define by <glyph glyph-name="space" unicode=" " horiz-adv-x="22" />, Example font.json : "horiz_adv_x_space":22,
|
||
word_spacing = font_metadata('horiz_adv_x_space', 20)
|
||
|
||
reversible = font_metadata('reversible', True)
|
||
sortable = font_metadata('sortable', False)
|
||
combine_at_sort_indices = font_metadata('combine_at_sort_indices', [])
|
||
|
||
@property
|
||
def id(self):
|
||
return os.path.basename(self.path)
|
||
|
||
@property
|
||
def default_variant(self):
|
||
# Set default variant to any existing variant if default font file is missing
|
||
default_variant = self.json_default_variant
|
||
font_variants = self.has_variants()
|
||
if default_variant not in font_variants and len(font_variants) > 0:
|
||
default_variant = font_variants[0]
|
||
return default_variant
|
||
|
||
@property
|
||
def preview_image(self):
|
||
preview_image_path = os.path.join(self.path, "preview.png")
|
||
if os.path.isfile(preview_image_path):
|
||
return preview_image_path
|
||
return None
|
||
|
||
def has_variants(self):
|
||
# returns available variants
|
||
font_variants = []
|
||
for variant in FontVariant.VARIANT_TYPES:
|
||
if os.path.isfile(os.path.join(self.path, "%s.svg" % variant)):
|
||
font_variants.append(variant)
|
||
elif (os.path.isdir(os.path.join(self.path, "%s" % variant)) and
|
||
[svg for svg in os.listdir(os.path.join(self.path, "%s" % variant)) if svg.endswith('.svg')]):
|
||
font_variants.append(variant)
|
||
if not font_variants:
|
||
raise FontError(_("The font '%s' has no variants.") % self.name)
|
||
return font_variants
|
||
|
||
@property
|
||
def marked_custom_font_id(self):
|
||
if not self.is_custom_font():
|
||
return self.id
|
||
else:
|
||
return self.id + '*'
|
||
|
||
@property
|
||
def marked_custom_font_name(self):
|
||
if not self.is_custom_font():
|
||
return self.name
|
||
else:
|
||
return self.name + '*'
|
||
|
||
def is_custom_font(self):
|
||
custom_dir = get_custom_font_dir()
|
||
if not custom_dir:
|
||
return False
|
||
return custom_dir in self.path
|
||
|
||
def render_text(self, text, destination_group, variant=None, back_and_forth=True, # noqa: C901
|
||
trim_option=0, use_trim_symbols=False, color_sort=0, text_align=0):
|
||
|
||
"""Render text into an SVG group element."""
|
||
self._load_variants()
|
||
|
||
if variant is None:
|
||
variant = self.default_variant
|
||
|
||
if back_and_forth and self.reversible:
|
||
glyph_sets = [self.get_variant(variant), self.get_variant(FontVariant.reversed_variant(variant))]
|
||
else:
|
||
glyph_sets = [self.get_variant(variant)] * 2
|
||
|
||
max_line_width = 0
|
||
position = Point(0, 0)
|
||
for i, line in enumerate(text.splitlines()):
|
||
glyph_set = glyph_sets[i % 2]
|
||
line = line.strip()
|
||
|
||
if self.text_direction == "rtl":
|
||
line = line[::-1]
|
||
|
||
letter_group = self._render_line(destination_group, line, position, glyph_set)
|
||
if ((variant == '→' and back_and_forth and self.reversible and i % 2 == 1) or
|
||
(variant == '←' and not (back_and_forth and self.reversible and i % 2 == 1))):
|
||
letter_group[:] = reversed(letter_group)
|
||
for group in letter_group:
|
||
group[:] = reversed(group)
|
||
|
||
position.x = 0
|
||
position.y += self.leading
|
||
|
||
# We need to insert the destination_group now, even though it is possibly empty
|
||
# otherwise we could run into a FragmentError in case a glyph contains commands
|
||
destination_group.append(letter_group)
|
||
bounding_box = None
|
||
try:
|
||
bounding_box = letter_group.bounding_box()
|
||
except AttributeError:
|
||
# letter group is None
|
||
continue
|
||
# remove destination_group if it is empty
|
||
if not bounding_box:
|
||
letter_group.delete()
|
||
continue
|
||
|
||
line_width = bounding_box.width
|
||
max_line_width = max(max_line_width, line_width)
|
||
if text_align == 1:
|
||
# align center
|
||
letter_group.transform = f'translate({-line_width/2}, 0)'
|
||
if text_align == 2:
|
||
letter_group.transform = f'translate({-line_width}, 0)'
|
||
|
||
if text_align in [3, 4]:
|
||
for line_group in destination_group.iterchildren():
|
||
if text_align == 4 and len(line_group) == 1:
|
||
line_group = line_group[0]
|
||
if len(line_group) > 1:
|
||
distance = max_line_width - line_group.bounding_box().width
|
||
distance_per_space = distance / (len(line_group) - 1)
|
||
for i, word in enumerate(line_group.getchildren()[1:]):
|
||
transform = word.transform
|
||
translate = distance_per_space * (i + 1)
|
||
transform.add_translate(translate, 0)
|
||
|
||
if self.auto_satin and len(destination_group) > 0:
|
||
self._apply_auto_satin(destination_group)
|
||
|
||
self._set_style(destination_group)
|
||
|
||
# add trims
|
||
self._add_trims(destination_group, text, trim_option, use_trim_symbols, back_and_forth, color_sort)
|
||
# make sure necessary marker and command symbols are in the defs section
|
||
self._ensure_command_symbols(destination_group)
|
||
self._ensure_marker_symbols(destination_group)
|
||
|
||
if color_sort != 0 and self.sortable:
|
||
self.do_color_sort(destination_group, color_sort)
|
||
|
||
return destination_group
|
||
|
||
def _set_style(self, destination_group):
|
||
# make sure font stroke styles have always a similar look
|
||
for element in destination_group.iterdescendants(SVG_PATH_TAG):
|
||
style = inkex.Style(element.get('style'))
|
||
if style.get('fill') == 'none':
|
||
style += inkex.Style("stroke-width:1px")
|
||
if style.get('stroke-dasharray') and style.get('stroke-dasharray') != 'none':
|
||
style += inkex.Style("stroke-dasharray:3, 1")
|
||
# Set a smaller width to auto-route running stitches
|
||
if self.auto_satin or element.get_id().startswith("autosatinrun"):
|
||
style += inkex.Style("stroke-width:0.5px")
|
||
element.set('style', '%s' % style.to_str())
|
||
|
||
def get_variant(self, variant):
|
||
return self.variants.get(variant, self.variants[self.default_variant])
|
||
|
||
def _render_line(self, destination_group, line, position, glyph_set):
|
||
"""Render a line of text.
|
||
|
||
An SVG XML node tree will be returned, with an svg:g at its root. If
|
||
the font metadata requests it, Auto-Satin will be applied.
|
||
|
||
Parameters:
|
||
line -- the line of text to render.
|
||
position -- Current position. Will be updated to point to the spot
|
||
immediately after the last character.
|
||
glyph_set -- a FontVariant instance.
|
||
|
||
Returns:
|
||
An svg:g element containing the rendered text.
|
||
"""
|
||
|
||
group = inkex.Group()
|
||
group.label = line
|
||
if self.text_direction == 'rtl':
|
||
group.label = line[::-1]
|
||
group.set("inkstitch:letter-group", "line")
|
||
last_character = None
|
||
|
||
words = line.split(" ")
|
||
for word in words:
|
||
|
||
word_group = inkex.Group()
|
||
label = word
|
||
if self.text_direction == 'rtl':
|
||
label = word[::-1]
|
||
word_group.label = label
|
||
word_group.set("inkstitch:letter-group", "word")
|
||
|
||
if self.text_direction == 'rtl':
|
||
glyphs = self._get_word_glyphs(glyph_set, word[::-1])
|
||
glyphs = glyphs[::-1]
|
||
else:
|
||
glyphs = self._get_word_glyphs(glyph_set, word)
|
||
|
||
last_character = None
|
||
for glyph in glyphs:
|
||
if glyph is None:
|
||
position.x += self.word_spacing
|
||
last_character = None
|
||
continue
|
||
node = self._render_glyph(destination_group, glyph, position, glyph.name, last_character)
|
||
word_group.append(node)
|
||
last_character = glyph.name
|
||
group.append(word_group)
|
||
position.x += self.word_spacing
|
||
return group
|
||
|
||
def _get_word_glyphs(self, glyph_set, word):
|
||
glyphs = []
|
||
skip = []
|
||
previous_is_binding = True
|
||
|
||
# forced letter case
|
||
if self.letter_case == "upper":
|
||
word = word.upper()
|
||
elif self.letter_case == "lower":
|
||
word = word.lower()
|
||
|
||
for i, character in enumerate(word):
|
||
if i in skip:
|
||
continue
|
||
|
||
glyph, glyph_len, binding = glyph_set.get_next_glyph(word, i, previous_is_binding)
|
||
previous_is_binding = binding
|
||
|
||
skip = list(range(i, i+glyph_len))
|
||
|
||
if glyph is None and self.default_glyph == " ":
|
||
glyphs.append(None)
|
||
else:
|
||
if glyph is None:
|
||
glyphs.append(glyph_set[self.default_glyph])
|
||
if glyph is not None:
|
||
glyphs.append(glyph)
|
||
|
||
return glyphs
|
||
|
||
def _render_glyph(self, destination_group, glyph, position, character, last_character):
|
||
"""Render a single glyph.
|
||
|
||
An SVG XML node tree will be returned, with an svg:g at its root.
|
||
|
||
Parameters:
|
||
glyph -- a Glyph instance
|
||
position -- Current position. Will be updated based on the width
|
||
of this character and the letter spacing.
|
||
character -- the current Unicode character.
|
||
last_character -- the previous character in the line, or None if
|
||
we're at the start of the line or a word.
|
||
"""
|
||
|
||
# Concerning min_x: I add it before moving the letter because it is to
|
||
# take into account the margin in the drawing of the letter. With respect
|
||
# to point 0 the letter can start at 5 or -5. The letters have a defined
|
||
# place in the drawing that's important.
|
||
# Then to calculate the position of x for the next letter I have to remove
|
||
# the min_x margin because the horizontal adv is calculated from point 0 of the drawing.
|
||
|
||
node = deepcopy(glyph.node)
|
||
if last_character is not None:
|
||
if self.text_direction != "rtl":
|
||
kerning = self.kerning_pairs.get(f'{last_character} {character}', None)
|
||
if kerning is None:
|
||
# legacy kerning without space
|
||
kerning = self.kerning_pairs.get(last_character + character, 0)
|
||
position.x += glyph.min_x - kerning
|
||
else:
|
||
kerning = self.kerning_pairs.get(f'{character} {last_character}', None)
|
||
if kerning is None:
|
||
# legacy kerning without space
|
||
kerning = self.kerning_pairs.get(character + last_character, 0)
|
||
position.x += glyph.min_x - kerning
|
||
|
||
transform = "translate(%s, %s)" % position.as_tuple()
|
||
node.set('transform', transform)
|
||
|
||
horiz_adv_x_default = self.horiz_adv_x_default
|
||
if horiz_adv_x_default is None:
|
||
horiz_adv_x_default = glyph.width + glyph.min_x
|
||
|
||
position.x += self.horiz_adv_x.get(character, horiz_adv_x_default) - glyph.min_x
|
||
|
||
self._update_commands(node, glyph)
|
||
self._update_clips(destination_group, node, glyph)
|
||
|
||
# this is used to recognize a glyph layer later in the process
|
||
# because this is not unique it will be overwritten by inkscape when inserted into the document
|
||
node.set("id", "glyph")
|
||
node.set("inkstitch:letter-group", "glyph")
|
||
|
||
return node
|
||
|
||
def _update_commands(self, node, glyph):
|
||
for element, connectors in glyph.commands.items():
|
||
# update element
|
||
el = node.find(".//*[@id='%s']" % element)
|
||
# we cannot get a unique id from the document at this point
|
||
# so let's create a random id which will most probably work as well
|
||
new_element_id = "%s_%s" % (element, randint(0, 9999))
|
||
el.set_id(new_element_id)
|
||
for connector, symbol in connectors:
|
||
# update symbol
|
||
new_symbol_id = "%s_%s" % (symbol, randint(0, 9999))
|
||
s = node.find(".//*[@id='%s']" % symbol)
|
||
s.set_id(new_symbol_id)
|
||
# update connector
|
||
c = node.find(".//*[@id='%s']" % connector)
|
||
c.set(CONNECTION_END, "#%s" % new_element_id)
|
||
c.set(CONNECTION_START, "#%s" % new_symbol_id)
|
||
|
||
def _update_clips(self, destination_group, node, glyph):
|
||
svg = destination_group.getroottree().getroot()
|
||
for node_id, clip in glyph.clips.items():
|
||
if clip not in svg.defs:
|
||
svg.defs.append(clip)
|
||
el = node.find(f".//*[@id='{node_id}']")
|
||
el.clip = clip
|
||
|
||
def _add_trims(self, destination_group, text, trim_option, use_trim_symbols, back_and_forth, color_sort):
|
||
"""
|
||
trim_option == 0 --> no trims
|
||
trim_option == 1 --> trim at the end of each line
|
||
trim_option == 2 --> trim after each word
|
||
trim_option == 3 --> trim after each letter
|
||
"""
|
||
if trim_option == 0:
|
||
return
|
||
|
||
# reverse every second line of text if back and forth is true and strip spaces
|
||
text = text.splitlines()
|
||
text = [t[::-1].strip() if i % 2 != 0 and back_and_forth else t.strip() for i, t in enumerate(text)]
|
||
text = "\n".join(text)
|
||
|
||
i = -1
|
||
space_indices = [i for i, t in enumerate(text) if t == " "]
|
||
line_break_indices = [i for i, t in enumerate(text) if t == "\n"]
|
||
for group in destination_group.iterdescendants(SVG_GROUP_TAG):
|
||
# make sure we are only looking at glyph groups
|
||
if not group.get("id", "").startswith("glyph"):
|
||
continue
|
||
|
||
i += 1
|
||
while i in space_indices + line_break_indices:
|
||
i += 1
|
||
|
||
# letter
|
||
if trim_option == 3:
|
||
self._process_trim(group, use_trim_symbols, color_sort)
|
||
# word
|
||
elif trim_option == 2 and i+1 in space_indices + line_break_indices:
|
||
self._process_trim(group, use_trim_symbols, color_sort)
|
||
# line
|
||
elif trim_option == 1 and i+1 in line_break_indices:
|
||
self._process_trim(group, use_trim_symbols, color_sort)
|
||
|
||
def _process_trim(self, group, use_trim_symbols, color_sort):
|
||
if color_sort != 0 and self.sortable:
|
||
elements = defaultdict(list)
|
||
for path_child in group.iterdescendants(EMBROIDERABLE_TAGS):
|
||
if not has_marker(path_child):
|
||
sort_index = path_child.get('inkstitch:color_sort_index', None)
|
||
if sort_index is not None:
|
||
elements[sort_index] = path_child
|
||
else:
|
||
elements[404] = path_child
|
||
for value in elements.values():
|
||
self._add_trim_to_element(Stroke(value), use_trim_symbols)
|
||
else:
|
||
# find the last path that does not carry a marker or belongs to a visual command and add a trim there
|
||
for path_child in group.iterdescendants(EMBROIDERABLE_TAGS):
|
||
if not has_marker(path_child) and not path_child.get_id().startswith('command_connector'):
|
||
path = path_child
|
||
element = Stroke(path)
|
||
self._add_trim_to_element(element, use_trim_symbols)
|
||
|
||
def _add_trim_to_element(self, element, use_trim_symbols):
|
||
if element.shape:
|
||
element_id = "%s_%s" % (element.node.get('id'), randint(0, 9999))
|
||
element.node.set("id", element_id)
|
||
if use_trim_symbols is False:
|
||
element.node.set(INKSTITCH_ATTRIBS['trim_after'], 'true')
|
||
else:
|
||
add_commands(element, ['trim'])
|
||
|
||
def _ensure_command_symbols(self, group):
|
||
# collect commands
|
||
commands = set()
|
||
for element in group.iterdescendants(SVG_USE_TAG):
|
||
xlink = element.get(XLINK_HREF, ' ')
|
||
if xlink.startswith('#inkstitch_'):
|
||
commands.add(xlink[11:])
|
||
# make sure all necessary command symbols are in the document
|
||
for command in commands:
|
||
ensure_symbol(group.getroottree().getroot(), command)
|
||
|
||
def _ensure_marker_symbols(self, group):
|
||
for marker in MARKER:
|
||
xpath = ".//*[contains(@style, 'marker-start:url(#inkstitch-%s-marker')]" % marker
|
||
marked_elements = group.xpath(xpath, namespaces=inkex.NSS)
|
||
if marked_elements:
|
||
ensure_marker(group.getroottree().getroot(), marker)
|
||
for element in marked_elements:
|
||
element.style['marker-start'] = "url(#inkstitch-%s-marker)" % marker
|
||
|
||
def _apply_auto_satin(self, group):
|
||
"""Apply Auto-Satin to an SVG XML node tree with an svg:g at its root.
|
||
|
||
The group's contents will be replaced with the results of the auto-
|
||
satin operation. Any nested svg:g elements will be removed.
|
||
"""
|
||
|
||
elements = nodes_to_elements(group.iterdescendants(EMBROIDERABLE_TAGS))
|
||
elements = [element for element in elements if isinstance(element, SatinColumn) or isinstance(element, Stroke)]
|
||
|
||
if elements and any(isinstance(element, SatinColumn) for element in elements):
|
||
auto_satin(elements, preserve_order=True, trim=False)
|
||
|
||
def do_color_sort(self, group, color_sort):
|
||
"""Sort elements by their color sort index as defined by font author"""
|
||
|
||
if color_sort == 1:
|
||
# Whole text
|
||
self._color_sort_group(group, 'line')
|
||
elif color_sort == 2:
|
||
# per line
|
||
groups = group.getchildren()
|
||
for group in groups:
|
||
self._color_sort_group(group, 'word')
|
||
elif color_sort == 3:
|
||
# per word
|
||
line_groups = group.getchildren()
|
||
for line_group in line_groups:
|
||
for group in line_group.iterchildren():
|
||
self._color_sort_group(group, 'glyph')
|
||
|
||
def _color_sort_group(self, group, transform_key):
|
||
elements_by_color = self._get_color_sorted_elements(group, transform_key)
|
||
|
||
# there are no sort indexes defined, abort color sorting and return to normal
|
||
if not elements_by_color:
|
||
return
|
||
|
||
group.remove_all()
|
||
for index, grouped_elements in sorted(elements_by_color.items()):
|
||
color_group = inkex.Group(attrib={
|
||
INKSCAPE_LABEL: _("Color Group") + f' {index}'
|
||
})
|
||
|
||
# combined indices
|
||
if index in self.combine_at_sort_indices:
|
||
path = ""
|
||
for element_list in grouped_elements:
|
||
for element in element_list:
|
||
path += element.get("d", "")
|
||
grouped_elements[0][0].set("d", path)
|
||
if grouped_elements[0][0].get("inkstitch:fill_method", False) in ['tartan_fill', 'linear_gradient_fill']:
|
||
grouped_elements[0][0].set('inkstitch:stop_at_ending_point', True)
|
||
color_group.append(grouped_elements[0][0])
|
||
group.append(color_group)
|
||
continue
|
||
|
||
# everything else, create marker groups if applicable
|
||
for element_list in grouped_elements:
|
||
if len(element_list) == 1:
|
||
color_group.append(element_list[0])
|
||
continue
|
||
elements_group = inkex.Group()
|
||
for element in element_list:
|
||
elements_group.append(element)
|
||
color_group.append(elements_group)
|
||
|
||
group.append(color_group)
|
||
|
||
def _get_color_sorted_elements(self, group, transform_key): # noqa: C901
|
||
elements_by_color = defaultdict(list)
|
||
last_parent = None
|
||
|
||
for element in group.iterdescendants(EMBROIDERABLE_TAGS, SVG_GROUP_TAG):
|
||
sort_index = element.get('inkstitch:color_sort_index', None)
|
||
|
||
# Skip command connectors, we only aim for command groups
|
||
# Skip command connectors as well, they will be included with the command group
|
||
if (element.TAG == 'g' and not element.get_id().startswith('command_group')
|
||
or element.get_id().startswith('command_connector')):
|
||
continue
|
||
|
||
clips = get_clips(element)
|
||
if len(clips) > 1:
|
||
# multiple clips: wrap the element into clipped groups
|
||
parent = element.getparent()
|
||
index = parent.index(element)
|
||
for clip in clips:
|
||
new_group = inkex.Group()
|
||
new_group.clip = clip
|
||
parent.insert(index, new_group)
|
||
new_group.append(element)
|
||
element = new_group
|
||
elif len(clips) == 1:
|
||
# only one clip: we can apply the clip directly to the element
|
||
element.clip = clips[0]
|
||
|
||
# get glyph group to calculate transform
|
||
glyph_group = None
|
||
for ancestor in element.ancestors(group):
|
||
if ancestor.get("inkstitch:letter-group", '') == transform_key:
|
||
glyph_group = ancestor
|
||
break
|
||
if glyph_group is None:
|
||
# this should never happen
|
||
continue
|
||
|
||
element.transform = element.composed_transform(glyph_group.getparent())
|
||
if sort_index is not None and int(sort_index) in self.combine_at_sort_indices:
|
||
element.apply_transform()
|
||
|
||
if not sort_index:
|
||
elements_by_color[404].append([element])
|
||
continue
|
||
|
||
if element.get_id().startswith('command_group'):
|
||
elements_by_color[int(sort_index)].append([element])
|
||
continue
|
||
|
||
parent = element.getparent()
|
||
if element.clip is None and parent.clip is not None:
|
||
element.clip = parent.clip
|
||
if last_parent != parent or int(sort_index) not in elements_by_color or not is_grouped_with_marker(element):
|
||
elements_by_color[int(sort_index)].append([element])
|
||
else:
|
||
elements_by_color[int(sort_index)][-1].append(element)
|
||
last_parent = parent
|
||
return elements_by_color
|