88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
|
"""This submodule contains tools for creating path objects from SVG files.
|
||
|
|
The main tool being the svg2paths() function."""
|
||
|
|
|
||
|
|
# External dependencies
|
||
|
|
from __future__ import division, absolute_import, print_function
|
||
|
|
from xml.dom.minidom import parse
|
||
|
|
from os import path as os_path, getcwd
|
||
|
|
|
||
|
|
# Internal dependencies
|
||
|
|
from .parser import parse_path
|
||
|
|
|
||
|
|
|
||
|
|
def polyline2pathd(polyline_d):
|
||
|
|
"""converts the string from a polyline d-attribute to a string for a Path
|
||
|
|
object d-attribute"""
|
||
|
|
points = polyline_d.replace(', ', ',')
|
||
|
|
points = points.replace(' ,', ',')
|
||
|
|
points = points.split()
|
||
|
|
|
||
|
|
if points[0] == points[-1]:
|
||
|
|
closed = True
|
||
|
|
else:
|
||
|
|
closed = False
|
||
|
|
|
||
|
|
d = 'M' + points.pop(0).replace(',', ' ')
|
||
|
|
for p in points:
|
||
|
|
d += 'L' + p.replace(',', ' ')
|
||
|
|
if closed:
|
||
|
|
d += 'z'
|
||
|
|
return d
|
||
|
|
|
||
|
|
|
||
|
|
def svg2paths(svg_file_location,
|
||
|
|
convert_lines_to_paths=True,
|
||
|
|
convert_polylines_to_paths=True,
|
||
|
|
convert_polygons_to_paths=True):
|
||
|
|
"""
|
||
|
|
Converts an SVG file into a list of Path objects and a list of
|
||
|
|
dictionaries containing their attributes. This currently supports
|
||
|
|
SVG Path, Line, Polyline, and Polygon elements.
|
||
|
|
:param svg_file_location: the location of the svg file
|
||
|
|
:param convert_lines_to_paths: Set to False to disclude SVG-Line objects
|
||
|
|
(converted to Paths)
|
||
|
|
:param convert_polylines_to_paths: Set to False to disclude SVG-Polyline
|
||
|
|
objects (converted to Paths)
|
||
|
|
:param convert_polygons_to_paths: Set to False to disclude SVG-Polygon
|
||
|
|
objects (converted to Paths)
|
||
|
|
:return: list of Path objects
|
||
|
|
"""
|
||
|
|
if os_path.dirname(svg_file_location) == '':
|
||
|
|
svg_file_location = os_path.join(getcwd(), svg_file_location)
|
||
|
|
|
||
|
|
doc = parse(svg_file_location) # parseString also exists
|
||
|
|
|
||
|
|
def dom2dict(element):
|
||
|
|
"""Converts DOM elements to dictionaries of attributes."""
|
||
|
|
keys = element.attributes.keys()
|
||
|
|
values = [val.value for val in element.attributes.values()]
|
||
|
|
return dict(zip(keys, values))
|
||
|
|
|
||
|
|
# Use minidom to extract path strings from input SVG
|
||
|
|
paths = [dom2dict(el) for el in doc.getElementsByTagName('path')]
|
||
|
|
d_strings = [el['d'] for el in paths]
|
||
|
|
attribute_dictionary_list = paths
|
||
|
|
|
||
|
|
# Use minidom to extract polyline strings from input SVG, convert to
|
||
|
|
# path strings, add to list
|
||
|
|
if convert_polylines_to_paths:
|
||
|
|
plins = [dom2dict(el) for el in doc.getElementsByTagName('polyline')]
|
||
|
|
d_strings += [polyline2pathd(pl['points']) for pl in plins]
|
||
|
|
attribute_dictionary_list += plins
|
||
|
|
|
||
|
|
# Use minidom to extract polygon strings from input SVG, convert to
|
||
|
|
# path strings, add to list
|
||
|
|
if convert_polygons_to_paths:
|
||
|
|
pgons = [dom2dict(el) for el in doc.getElementsByTagName('polygon')]
|
||
|
|
d_strings += [polyline2pathd(pg['points']) + 'z' for pg in pgons]
|
||
|
|
attribute_dictionary_list += pgons
|
||
|
|
|
||
|
|
if convert_lines_to_paths:
|
||
|
|
lines = [dom2dict(el) for el in doc.getElementsByTagName('line')]
|
||
|
|
d_strings += [('M' + l['x1'] + ' ' + l['y1'] +
|
||
|
|
'L' + l['x2'] + ' ' + l['y2']) for l in lines]
|
||
|
|
attribute_dictionary_list += lines
|
||
|
|
doc.unlink()
|
||
|
|
path_list = [parse_path(d) for d in d_strings]
|
||
|
|
return path_list, attribute_dictionary_list
|