inkstitch/lib/extensions/outline.py

84 wiersze
3.5 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.
from inkex import Boolean, Path, errormsg
from shapely import offset_curve
from shapely.geometry import LineString, MultiPolygon, Polygon
from ..i18n import _
from ..svg import PIXELS_PER_MM
from ..svg.tags import SVG_PATH_TAG
from ..utils.geometry import ensure_multi_line_string
from ..utils.smoothing import smooth_path
from .base import InkstitchExtension
class Outline(InkstitchExtension):
def __init__(self, *args, **kwargs):
InkstitchExtension.__init__(self, *args, **kwargs)
self.arg_parser.add_argument("--notebook")
self.arg_parser.add_argument("-k", "--keep-original", type=Boolean, default=False, dest="keep_original")
self.arg_parser.add_argument("-b", "--buffer", type=float, default=0.001, dest="buffer")
self.arg_parser.add_argument("-s", "--smoothness", type=float, default=0.3, dest="smoothness")
self.arg_parser.add_argument("-t", "--threshold", type=float, default=10.0, dest="threshold")
self.arg_parser.add_argument("-i", "--inset", type=float, default=0.001, dest="inset")
def effect(self):
if not self.svg.selection:
errormsg(_("Please select one or more shapes to convert to their outline."))
return
self.threshold = self.options.threshold * PIXELS_PER_MM
self.shape_buffer = max(self.options.buffer * PIXELS_PER_MM, 0.001)
self.smoothness = self.options.smoothness * PIXELS_PER_MM
self.inset = self.options.inset * PIXELS_PER_MM
for element in self.svg.selection:
self.element_to_outline(element)
def get_outline(self, element):
d = ''
transform = element.composed_transform()
path = Path(element.get_path()).transform(transform).break_apart()
for subpath in path:
points = subpath.end_points
shape = LineString(points).buffer(self.shape_buffer)
outline = ensure_multi_line_string(offset_curve(shape, -self.inset))
interiors = []
for interior in outline.geoms:
if Polygon(interior).area < self.threshold:
continue
interior_path = smooth_path(interior.coords, self.smoothness)
if len(interior_path) > 2:
interiors.append(Polygon(interior_path))
outline = MultiPolygon(interiors)
for geom in outline.geoms:
d += str(Path(geom.exterior.coords).transform(-transform))
return d
def element_to_outline(self, element):
element_id = element.label or element.get_id()
if element.tag_name == 'g':
for element in element.iterdescendants(SVG_PATH_TAG):
self.element_to_outline(element)
return
elif element.tag_name != 'path':
errormsg(_("{element_id} is not a path element. "
"This extension is designed to generate an outline of an embroidery pattern.").format(element_id=element_id))
return
d = self.get_outline(element)
if not d:
errormsg(_("Could not generate path from element {element_id} with the given settings.").format(element_id=element_id))
return
if self.options.keep_original:
new_element = element.duplicate()
new_element.set('d', d)
else:
element.set('d', d)