akko/diffvg
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #34 from robinsloan/master

Browse Source 
Fix bug in SVG export caused by missing string conversions
This commit is contained in:
Tzu-Mao Li
2021-10-17 13:49:17 -04:00
committed by GitHub
parent 2018a1bcec dac80fea96
commit e5955dbdcb
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
pydiffvg/save_svg.py
View File

@@ -60,7 +60,7 @@ def save_svg(filename, width, height, shapes, shape_groups, use_gamma = False):
stop_colors = lg.stop_colors.data.cpu().numpy() stop_colors = lg.stop_colors.data.cpu().numpy()
for j in range(offsets.shape[0]): for j in range(offsets.shape[0]):
stop = etree.SubElement(color, 'stop') stop = etree.SubElement(color, 'stop')
stop.set('offset', offsets[j]) stop.set('offset', str(offsets[j]))
c = lg.stop_colors[j, :] c = lg.stop_colors[j, :]
stop.set('stop-color', 'rgb({}, {}, {})'.format(\ stop.set('stop-color', 'rgb({}, {}, {})'.format(\
int(255 * c[0]), int(255 * c[1]), int(255 * c[2]))) int(255 * c[0]), int(255 * c[1]), int(255 * c[2])))
@@ -75,9 +75,9 @@ def save_svg(filename, width, height, shapes, shape_groups, use_gamma = False):
shape = shapes[shape_group.shape_ids[0]] shape = shapes[shape_group.shape_ids[0]]
if isinstance(shape, pydiffvg.Circle): if isinstance(shape, pydiffvg.Circle):
shape_node = etree.SubElement(g, 'circle') shape_node = etree.SubElement(g, 'circle')
shape_node.set('r', shape.radius.item()) shape_node.set('r', str(shape.radius.item()))
shape_node.set('cx', shape.center[0].item()) shape_node.set('cx', str(shape.center[0].item()))
shape_node.set('cy', shape.center[1].item()) shape_node.set('cy', str(shape.center[1].item()))
elif isinstance(shape, pydiffvg.Polygon): elif isinstance(shape, pydiffvg.Polygon):
shape_node = etree.SubElement(g, 'polygon') shape_node = etree.SubElement(g, 'polygon')
points = shape.points.data.cpu().numpy() points = shape.points.data.cpu().numpy()
@@ -117,10 +117,10 @@ def save_svg(filename, width, height, shapes, shape_groups, use_gamma = False):
shape_node.set('d', path_str) shape_node.set('d', path_str)
elif isinstance(shape, pydiffvg.Rect): elif isinstance(shape, pydiffvg.Rect):
shape_node = etree.SubElement(g, 'rect') shape_node = etree.SubElement(g, 'rect')
shape_node.set('x', shape.p_min[0].item()) shape_node.set('x', str(shape.p_min[0].item()))
shape_node.set('y', shape.p_min[1].item()) shape_node.set('y', str(shape.p_min[1].item()))
shape_node.set('width', shape.p_max[0].item() - shape.p_min[0].item()) shape_node.set('width', str(shape.p_max[0].item() - shape.p_min[0].item()))
shape_node.set('height', shape.p_max[1].item() - shape.p_min[1].item()) shape_node.set('height', str(shape.p_max[1].item() - shape.p_min[1].item()))
else: else:
assert(False) assert(False)