Added Delaunay Triangulation
This commit is contained in:
leyghisbb
parent
5b785c06a7
commit
b10fc90a4c
64
extensions/fablabchemnitz/delaunay.inx
Normal file
64
extensions/fablabchemnitz/delaunay.inx
Normal file
@ -0,0 +1,64 @@
|
|||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<inkscape-extension
|
||||||
|
xmlns="http://www.inkscape.org/namespace/inkscape/extension">
|
||||||
|
<name>Delaunay Triangulation</name>
|
||||||
|
<id>fablabchemnitz.de.delaunay_triangulation</id>
|
||||||
|
<param name="tab" type="notebook">
|
||||||
|
<page name="Options" gui-text="Options">
|
||||||
|
<param name="joggling" type="bool" gui-text="Support concavity and holes">false</param>
|
||||||
|
<param name="furthest" type="bool" gui-text="Use furthest-site triangulation">false</param>
|
||||||
|
<param name="elt_type" type="enum" gui-text="Object type to generate">
|
||||||
|
<item value="poly">Triangles</item>
|
||||||
|
<item value="line">Individual lines</item>
|
||||||
|
</param>
|
||||||
|
<spacer />
|
||||||
|
<separator />
|
||||||
|
<spacer />
|
||||||
|
<hbox>
|
||||||
|
<param name="fill_type" type="enum" gui-text="Fill color source">
|
||||||
|
<item value="first_sel">Same as first object selected</item>
|
||||||
|
<item value="last_sel">Same as last object selected</item>
|
||||||
|
<item value="random">Random</item>
|
||||||
|
<item value="specified">Explicitly specified</item>
|
||||||
|
</param>
|
||||||
|
<param name="fill_color" type="color" appearance="colorbutton" gui-text=" " gui-description="Specific fill color">-1</param>
|
||||||
|
</hbox>
|
||||||
|
<hbox>
|
||||||
|
<param name="stroke_type" type="enum" gui-text="Stroke color source">
|
||||||
|
<item value="first_sel">Same as first object selected</item>
|
||||||
|
<item value="last_sel">Same as last object selected</item>
|
||||||
|
<item value="random">Random</item>
|
||||||
|
<item value="specified">Explicitly specified</item>
|
||||||
|
</param>
|
||||||
|
<param name="stroke_color" type="color" appearance="colorbutton" gui-text=" " gui-description="Specific stroke color">255</param>
|
||||||
|
</hbox>
|
||||||
|
</page>
|
||||||
|
<page name="Advanced" gui-text="Advanced">
|
||||||
|
<param name="qhull" type="string" gui-text="qhull options">Qbb Qc Qz Q12</param>
|
||||||
|
<param name="name" type="description">
|
||||||
|
If "Support concavity" is enabled on the Options tab, "QJ" will be
|
||||||
|
prepended to the qhull options listed above. The default options
|
||||||
|
are "Qbb Qc Qz Q12". The following website describes the available
|
||||||
|
options.
|
||||||
|
</param>
|
||||||
|
<label appearance="url">http://www.qhull.org/html/qhull.htm#options</label>
|
||||||
|
</page>
|
||||||
|
<page name="Help" gui-text="Help">
|
||||||
|
<label>
|
||||||
|
This effect uses the Delaunay triangulation algorithm to create
|
||||||
|
triangles from all of the points found in the selected objects.
|
||||||
|
</label>
|
||||||
|
</page>
|
||||||
|
</param>
|
||||||
|
<effect>
|
||||||
|
<object-type>all</object-type>
|
||||||
|
<effects-menu>
|
||||||
|
<submenu name="FabLab Chemnitz">
|
||||||
|
<submenu name="Shape/Pattern from existing Path(s)" />
|
||||||
|
</submenu>
|
||||||
|
</effects-menu>
|
||||||
|
</effect>
|
||||||
|
<script>
|
||||||
|
<command location="inx" interpreter="python">delaunay.py</command>
|
||||||
|
</script>
|
||||||
|
</inkscape-extension>
|
||||||
198
extensions/fablabchemnitz/delaunay.py
Normal file
198
extensions/fablabchemnitz/delaunay.py
Normal file
@ -0,0 +1,198 @@
|
|||||||
|
#! /usr/bin/python
|
||||||
|
|
||||||
|
'''
|
||||||
|
Copyright (C) 2020 Scott Pakin, scott-ink@pakin.org
|
||||||
|
|
||||||
|
This program is free software; you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU General Public License as published by
|
||||||
|
the Free Software Foundation; either version 3 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
along with this program; if not, write to the Free Software
|
||||||
|
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
||||||
|
02110-1301, USA.
|
||||||
|
|
||||||
|
'''
|
||||||
|
|
||||||
|
import inkex
|
||||||
|
import numpy as np
|
||||||
|
import random
|
||||||
|
import sys
|
||||||
|
from inkex import Group, Line, Polygon, TextElement
|
||||||
|
from inkex.styles import Style
|
||||||
|
from inkex.transforms import Vector2d
|
||||||
|
from scipy.spatial import Delaunay
|
||||||
|
|
||||||
|
class DelaunayTriangulation(inkex.EffectExtension):
|
||||||
|
'Overlay selected objects with triangles.'
|
||||||
|
|
||||||
|
def add_arguments(self, pars):
|
||||||
|
'Parse the arguments passed to us from an Inkscape dialog box.'
|
||||||
|
pars.add_argument('--tab', help='The selected UI tab when OK was pressed')
|
||||||
|
pars.add_argument('--joggling', type=inkex.Boolean, default=False, help='Use joggled input instead of merged facets')
|
||||||
|
pars.add_argument('--furthest', type=inkex.Boolean, default=False, help='Furthest-site Delaunay triangulation')
|
||||||
|
pars.add_argument('--elt_type', default='poly', help='Element type to generate ("poly" or "line")')
|
||||||
|
pars.add_argument('--qhull', help='Triangulation options to pass to qhull')
|
||||||
|
pars.add_argument('--fill_type', help='How to fill generated polygons')
|
||||||
|
pars.add_argument('--fill_color', type=inkex.Color, help='Fill color to use with a fill type of "specified"')
|
||||||
|
pars.add_argument('--stroke_type', help='How to stroke generated polygons')
|
||||||
|
pars.add_argument('--stroke_color', type=inkex.Color, help='Stroke color to use with a stroke type of "specified"')
|
||||||
|
|
||||||
|
def _path_points(self, elt):
|
||||||
|
'Return a list of all points on a path (endpoints, not control points).'
|
||||||
|
pts = set()
|
||||||
|
first = None
|
||||||
|
prev = Vector2d()
|
||||||
|
for cmd in elt.path.to_absolute():
|
||||||
|
if first is None:
|
||||||
|
first = cmd.end_point(first, prev)
|
||||||
|
ep = cmd.end_point(first, prev)
|
||||||
|
pts.add((ep.x, ep.y))
|
||||||
|
prev = ep
|
||||||
|
return pts
|
||||||
|
|
||||||
|
def _create_styles(self, n):
|
||||||
|
'Return a style to use for the generated objects.'
|
||||||
|
# Use either the first or the last element's stroke for line caps,
|
||||||
|
# stroke widths, etc.
|
||||||
|
fstyle = self.svg.selection.first().style
|
||||||
|
lstyle = self.svg.selection[-1].style
|
||||||
|
if self.options.stroke_type == 'last_sel':
|
||||||
|
style = Style(lstyle)
|
||||||
|
else:
|
||||||
|
style = Style(fstyle)
|
||||||
|
|
||||||
|
# Apply the specified fill color.
|
||||||
|
if self.options.fill_type == 'first_sel':
|
||||||
|
fcolor = fstyle.get_color('fill')
|
||||||
|
style.set_color(fcolor, 'fill')
|
||||||
|
elif self.options.fill_type == 'last_sel':
|
||||||
|
fcolor = lstyle.get_color('fill')
|
||||||
|
style.set_color(fcolor, 'fill')
|
||||||
|
elif self.options.fill_type == 'specified':
|
||||||
|
style.set_color(self.options.fill_color, 'fill')
|
||||||
|
elif self.options.fill_type == 'random':
|
||||||
|
pass # Handled below
|
||||||
|
else:
|
||||||
|
sys.exit(inkex.utils.errormsg(_('Internal error: Unrecognized fill type "%s".')) % self.options.fill_type)
|
||||||
|
|
||||||
|
# Apply the specified stroke color.
|
||||||
|
if self.options.stroke_type == 'first_sel':
|
||||||
|
scolor = fstyle.get_color('stroke')
|
||||||
|
style.set_color(scolor, 'stroke')
|
||||||
|
elif self.options.stroke_type == 'last_sel':
|
||||||
|
scolor = lstyle.get_color('stroke')
|
||||||
|
style.set_color(scolor, 'stroke')
|
||||||
|
elif self.options.stroke_type == 'specified':
|
||||||
|
style.set_color(self.options.stroke_color, 'stroke')
|
||||||
|
elif self.options.stroke_type == 'random':
|
||||||
|
pass # Handled below
|
||||||
|
else:
|
||||||
|
sys.exit(inkex.utils.errormsg(_('Internal error: Unrecognized stroke type "%s".')) % self.options.stroke_type)
|
||||||
|
|
||||||
|
# Produce n copies of the style.
|
||||||
|
styles = [Style(style) for i in range(n)]
|
||||||
|
if self.options.fill_type == 'random':
|
||||||
|
for s in styles:
|
||||||
|
r = random.randint(0, 255)
|
||||||
|
g = random.randint(0, 255)
|
||||||
|
b = random.randint(0, 255)
|
||||||
|
s.set_color('#%02x%02x%02x' % (r, g, b), 'fill')
|
||||||
|
s['fill-opacity'] = 255
|
||||||
|
if self.options.stroke_type == 'random':
|
||||||
|
for s in styles:
|
||||||
|
r = random.randint(0, 255)
|
||||||
|
g = random.randint(0, 255)
|
||||||
|
b = random.randint(0, 255)
|
||||||
|
s.set_color('#%02x%02x%02x' % (r, g, b), 'stroke')
|
||||||
|
s['stroke-opacity'] = 255
|
||||||
|
|
||||||
|
# Return the list of styles.
|
||||||
|
return [str(s) for s in styles]
|
||||||
|
|
||||||
|
def _create_polygons(self, triangles):
|
||||||
|
'Render triangles as SVG polygons.'
|
||||||
|
styles = self._create_styles(len(triangles))
|
||||||
|
group = self.svg.get_current_layer().add(Group())
|
||||||
|
for tri, style in zip(triangles, styles):
|
||||||
|
tri_str = ' '.join(['%.10g %.10g' % (pt[0], pt[1]) for pt in tri])
|
||||||
|
poly = Polygon()
|
||||||
|
poly.set('points', tri_str)
|
||||||
|
poly.style = style
|
||||||
|
group.add(poly)
|
||||||
|
|
||||||
|
def _create_lines(self, triangles):
|
||||||
|
'Render triangles as individual SVG lines.'
|
||||||
|
# First, find all unique lines.
|
||||||
|
lines = set()
|
||||||
|
for tri in triangles:
|
||||||
|
if len(tri) != 3:
|
||||||
|
sys.exit(inkex.utils.errormsg(_('Internal error: Encountered a non-triangle.')))
|
||||||
|
for i, j in [(0, 1), (0, 2), (1, 2)]:
|
||||||
|
xy1 = tuple(tri[i])
|
||||||
|
xy2 = tuple(tri[j])
|
||||||
|
if xy1 < xy2:
|
||||||
|
lines.update([(xy1, xy2)])
|
||||||
|
else:
|
||||||
|
lines.update([(xy2, xy1)])
|
||||||
|
|
||||||
|
# Then, create SVG line elements.
|
||||||
|
styles = self._create_styles(len(lines))
|
||||||
|
group = self.svg.get_current_layer().add(Group())
|
||||||
|
for ([(x1, y1), (x2, y2)], style) in zip(lines, styles):
|
||||||
|
line = Line()
|
||||||
|
line.set('x1', x1)
|
||||||
|
line.set('y1', y1)
|
||||||
|
line.set('x2', x2)
|
||||||
|
line.set('y2', y2)
|
||||||
|
line.style = style
|
||||||
|
group.add(line)
|
||||||
|
|
||||||
|
def effect(self):
|
||||||
|
'Triangulate a set of objects.'
|
||||||
|
|
||||||
|
# Complain if the selection is empty.
|
||||||
|
if len(self.svg.selection) == 0:
|
||||||
|
return inkex.utils.errormsg(_('Please select at least one object.'))
|
||||||
|
|
||||||
|
# Acquire a set of all points from all selected objects.
|
||||||
|
all_points = set()
|
||||||
|
warned_text = False
|
||||||
|
for obj in self.svg.selection.values():
|
||||||
|
if isinstance(obj, TextElement) and not warned_text:
|
||||||
|
sys.stderr.write('Warning: Text elements are not currently supported. Ignoring all text in the selection.\n')
|
||||||
|
warned_text = True
|
||||||
|
all_points.update(self._path_points(obj.to_path_element()))
|
||||||
|
|
||||||
|
# Use SciPy to perform the Delaunay triangulation.
|
||||||
|
pts = np.array(list(all_points))
|
||||||
|
if len(pts) == 0:
|
||||||
|
return inkex.utils.errormsg(_('No points were found.'))
|
||||||
|
qopts = self.options.qhull
|
||||||
|
if self.options.joggling:
|
||||||
|
qopts = 'QJ ' + qopts
|
||||||
|
simplices = Delaunay(pts, furthest_site=self.options.furthest, qhull_options=qopts).simplices
|
||||||
|
|
||||||
|
# Create either triangles or lines, as request. Either option uses the
|
||||||
|
# style of the first object in the selection.
|
||||||
|
triangles = []
|
||||||
|
for s in simplices:
|
||||||
|
try:
|
||||||
|
triangles.append(pts[s])
|
||||||
|
except IndexError:
|
||||||
|
pass
|
||||||
|
if self.options.elt_type == 'poly':
|
||||||
|
self._create_polygons(triangles)
|
||||||
|
elif self.options.elt_type == 'line':
|
||||||
|
self._create_lines(triangles)
|
||||||
|
else:
|
||||||
|
return inkex.utils.errormsg(_('Internal error: unexpected element type "%s".') % self.options.elt_type)
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
DelaunayTriangulation().run()
|
||||||
Reference in New Issue
Block a user