Export UV Layout: Rewrite Export UV Layout addon

Differential Revision: https://developer.blender.org/D3715

Reviewer: brecht

1 files changed, 41 insertions, 60 deletions

diff --git a/io_mesh_uv_layout/export_uv_svg.py b/io_mesh_uv_layout/export_uv_svg.py
index fe727d09..d00f9402 100644
--- a/io_mesh_uv_layout/export_uv_svg.py
+++ b/io_mesh_uv_layout/export_uv_svg.py
@@ -19,65 +19,46 @@
 # <pep8-80 compliant>
 
 import bpy
-
-
-from xml.sax.saxutils import escape
 from os.path import basename
+from xml.sax.saxutils import escape
 
-
-class Export_UV_SVG:
-    def begin(self, fw, image_size, opacity):
-
-        self.fw = fw
-        self.image_width = image_size[0]
-        self.image_height = image_size[1]
-        self.opacity = opacity
-
-        fw('<?xml version="1.0" standalone="no"?>\n')
-        fw('<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" \n')
-        fw('  "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">\n')
-        fw('<svg width="%d" height="%d" viewBox="0 0 %d %d"\n' %
-           (self.image_width, self.image_height, self.image_width, self.image_height))
-        fw('     xmlns="http://www.w3.org/2000/svg" version="1.1">\n')
-        desc = ("%r, (Blender %s)" %
-                (basename(bpy.data.filepath), bpy.app.version_string))
-        fw('<desc>%s</desc>\n' % escape(desc))
-
-    def build(self, mesh, face_iter_func):
-        self.fw('<g>\n')
-        desc = ("Mesh: %s" % (mesh.name))
-        self.fw('<desc>%s</desc>\n' % escape(desc))
-
-        # svg colors
-        fill_settings = []
-        fill_default = 'fill="grey"'
-        for mat in mesh.materials if mesh.materials else [None]:
-            if mat:
-                fill_settings.append('fill="rgb(%d, %d, %d)"' %
-                                     tuple(int(c * 255) for c in mat.diffuse_color))
-            else:
-                fill_settings.append(fill_default)
-
-        polys = mesh.polygons
-        for i, uvs in face_iter_func():
-            try:  # rare cases material index is invalid.
-                fill = fill_settings[polys[i].material_index]
-            except IndexError:
-                fill = fill_default
-
-            self.fw('<polygon stroke="black" stroke-width="1"')
-            if self.opacity > 0.0:
-                self.fw(' %s fill-opacity="%.2g"' % (fill, self.opacity))
-
-            self.fw(' points="')
-
-            for j, uv in enumerate(uvs):
-                x, y = uv[0], 1.0 - uv[1]
-                self.fw('%.3f,%.3f ' % (x * self.image_width, y * self.image_height))
-            self.fw('" />\n')
-
-        self.fw('</g>\n')
-
-    def end(self):
-        self.fw('\n')
-        self.fw('</svg>\n')
+def export(filepath, face_data, colors, width, height, opacity):
+    with open(filepath, "w") as file:
+        for text in get_file_parts(face_data, colors, width, height, opacity):
+            file.write(text)
+
+def get_file_parts(face_data, colors, width, height, opacity):
+    yield from header(width, height)
+    yield from draw_polygons(face_data, width, height, opacity)
+    yield from footer()
+
+def header(width, height):
+    yield '<?xml version="1.0" standalone="no"?>\n'
+    yield '<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" \n'
+    yield '  "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">\n'
+    yield f'<svg width="{width}" height="{height}" viewBox="0 0 {width} {height}"\n'
+    yield '     xmlns="http://www.w3.org/2000/svg" version="1.1">\n'
+    desc = f"{basename(bpy.data.filepath)}, (Blender {bpy.app.version_string})"
+    yield f'<desc>{escape(desc)}</desc>\n'
+
+def draw_polygons(face_data, width, height, opacity):
+    for uvs, color in face_data:
+        fill = f'fill="{get_color_string(color)}"'
+
+        yield '<polygon stroke="black" stroke-width="1"'
+        yield f' {fill} fill-opacity="{opacity:.2g}"'
+
+        yield ' points="'
+
+        for uv in uvs:
+            x, y = uv[0], 1.0 - uv[1]
+            yield f'{x*width:.3f},{y*height:.3f} '
+        yield '" />\n'
+
+def get_color_string(color):
+    r, g, b = color
+    return f"rgb({round(r*255)}, {round(g*255)}, {round(b*255)})"
+
+def footer():
+    yield '\n'
+    yield '</svg>\n'
\ No newline at end of file