FabLab_Chemnitz/mightyscape-1.1-deprecated
FabLab_Chemnitz/mightyscape-1.1-deprecated
Archived
1
0
Fork 0
Code Issues Pull Requests Releases Activity

no message

Browse Source
This commit is contained in:
Mario Voigt 2020-08-18 14:29:20 +02:00
parent 32ab4b21c2
commit 807fa69a16
2 changed files with 53 additions and 59 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

102
extensions/fablabchemnitz_triangulation.py
View File

@ -22,31 +22,21 @@ Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
------------------------------------------------------------------------
'''
import base64
from io import BytesIO
import inkex
import urllib.parse
import urllib.request
import os
import random
from PIL import Image
from lxml import etree
from inkex import Transform
import numpy as np
from scipy.spatial import Delaunay
from scipy.cluster.vq import kmeans2
import cv2
import urllib.request as urllib
class Triangulation(inkex.Effect):
def __init__(self):
# Call base class construtor.
inkex.Effect.__init__(self)
# Option parser:
# -n, --num_points
# -m, --edge_thresh_min
# -M, --edge_thresh_max
# -c, --add_corners
# -g, --gradient_fill
# -b, --tab
self.arg_parser.add_argument("-n", "--num_points", type=int, default=100, help="Number of points to be sampled")
self.arg_parser.add_argument("-m", "--edge_thresh_min", type=int, default=200, help="Minimum threshold for edge detection")
self.arg_parser.add_argument("-M", "--edge_thresh_max", type=int, default=255, help="Maximum threshold for edge detection")
@ -63,6 +53,30 @@ class Triangulation(inkex.Effect):
path = etree.SubElement(parent, inkex.addNS('path','svg'), {'style' : str(inkex.Style(style)), 'd' : pathdesc})
return path
def checkImagePath(self, node):
"""Embed the data of the selected Image Tag element"""
xlink = node.get('xlink:href')
if xlink and xlink[:5] == 'data:':
# No need, data alread embedded
return
url = urllib.urlparse(xlink)
href = urllib.url2pathname(url.path)
# Primary location always the filename itself.
path = self.absolute_href(href or '')
# Backup directory where we can find the image
if not os.path.isfile(path):
path = node.get('sodipodi:absref', path)
if not os.path.isfile(path):
inkex.errormsg(_('File not found "{}". Unable to embed image.').format(path))
return
if (os.path.isfile(path)):
return path
def effect(self):
# Check we have something selected
@ -76,13 +90,7 @@ class Triangulation(inkex.Effect):
inkex.errormsg("The selected object (" + id + ") is not an image, skipping.")
continue
else:
(self.path, errcode) = self.checkImagePath(obj) # This also ensures the file exists
if errcode==1:
inkex.errormsg("Embedded images are not (yet?) supported, please use a linked image. Skipping.")
continue
elif errcode==2:
inkex.errormsg("The image points to a file, which seems to be missing: "+self.path+". Skipping.")
continue
self.path = self.checkImagePath(obj) # This also ensures the file exists
grpname = 'img_triangles'
# Make sure that the id/name is unique
@ -95,7 +103,6 @@ class Triangulation(inkex.Effect):
grp_attribs = {inkex.addNS('label','inkscape'):grp_name}
# The group to put everything in
grp = etree.SubElement(self.svg.get_current_layer(), 'g', grp_attribs)
# Find image size and position in Inkscape
try:
self.img_x_pos = float(obj.get("x"))
@ -105,7 +112,19 @@ class Triangulation(inkex.Effect):
self.img_y_pos = 0
self.img_width = float(obj.get("width"))
self.img_height = float(obj.get("height"))
if self.path is None: #check if image is embedded or linked
image_string = obj.get('{http://www.w3.org/1999/xlink}href')
# find comma position
i = 0
while i < 40:
if image_string[i] == ',':
break
i = i + 1
im = Image.open(BytesIO(base64.b64decode(image_string[i + 1:len(image_string)])))
else:
im = Image.open(self.path)
# IMPORTANT!
# The numpy array is accessed as im.data[row,column], that is data[y_coord, x_coord]
# Be careful not to pass coordinates as (x,y): rather use (y,x)!
@ -116,34 +135,7 @@ class Triangulation(inkex.Effect):
# Find real image size
(self.img_real_width, self.img_real_height) = im.size
self.doTriangulation(grp)
# Check file exists and returns its path
def checkImagePath(self, obj):
xlink = obj.get(inkex.addNS('href','xlink'))
if xlink[:5] == 'data:': # Embedded image
return (None, 1)
# Code shamelessly copied from the Embed image extension :)
if xlink is None or xlink[:5] != 'data:':
absref = obj.get(inkex.addNS('absref','sodipodi'))
url = urllib.parse.urlparse(xlink)
href = urllib.request.url2pathname(url.path)
path=''
#path selection strategy:
# 1. href if absolute
# 2. realpath-ified href
# 3. absref, only if the above does not point to a file
if (href != None):
path = os.path.realpath(href)
if (not os.path.isfile(path)):
if (absref != None):
path=absref
if (not os.path.isfile(path)):
return (path, 2)
return (path, 0)
self.doTriangulation(im, grp)
# Converts image coordinates to screen coordinates
def imgToScreen(self, x, y):
@ -177,10 +169,10 @@ class Triangulation(inkex.Effect):
stop2 = etree.SubElement(gradient, inkex.addNS('stop','svg'), attribs)
return gradient
def doTriangulation (self, grp):
#inkex.utils.debug(self.path)
def doTriangulation (self, im, grp):
# Read image with OpenCV
imcv = cv2.imread(self.path)
imcv = np.array(im)
#imcv = cv2.imread(self.path)
# Convert to grayscale
gray = cv2.cvtColor(imcv,cv2.COLOR_RGB2GRAY)
gray = np.float32(gray)
@ -188,9 +180,11 @@ class Triangulation(inkex.Effect):
edges = cv2.Canny(imcv, self.options.edge_thresh_min, self.options.edge_thresh_max, 100)
# Find coordinates of the edges
coords = [(float(x),float(y)) for y, row in enumerate(edges) for x, col in enumerate(row) if col>0]
#pt = random.sample(coords, self.options.num_points)
try:
pt, idx = kmeans2(np.array(coords), self.options.num_points, minit="points")
except ValueError:
inkex.utils.debug("Too much points. Reduce sampled points and try again!")
exit(1)
if self.options.add_corners:
# Add the four corners
corners = [(0, 0),