UV Layout export, writes an SVG, uses material colors.
[blender.git] / release / scripts / op / uv.py
1 # ##### BEGIN GPL LICENSE BLOCK #####
2 #
3 #  This program is free software; you can redistribute it and/or
4 #  modify it under the terms of the GNU General Public License
5 #  as published by the Free Software Foundation; either version 2
6 #  of the License, or (at your option) any later version.
7 #
8 #  This program is distributed in the hope that it will be useful,
9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11 #  GNU General Public License for more details.
12 #
13 #  You should have received a copy of the GNU General Public License
14 #  along with this program; if not, write to the Free Software Foundation,
15 #  Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
16 #
17 # ##### END GPL LICENSE BLOCK #####
18
19 # <pep8-80 compliant>
20
21 import bpy
22 from bpy.props import *
23
24 class ExportUVLayout(bpy.types.Operator):
25     '''Export the Mesh as SVG.'''
26
27     bl_idname = "uv.export_layout"
28     bl_label = "Export UV Layout"
29     bl_register = True
30     bl_undo = True
31     
32     path = StringProperty(name="File Path", description="File path used for exporting the SVG file", maxlen=1024, default="")
33     
34     def poll(self, context):
35         ob = context.active_object
36         return (ob and ob.type == 'MESH')
37         
38     def execute(self, context):
39         ob = context.active_object
40         is_editmode = (ob.mode == 'EDIT')
41         if is_editmode:
42             bpy.ops.object.mode_set(mode='OBJECT', toggle=False)
43
44         mesh = ob.data
45
46         active_uv_layer = None
47         for lay in mesh.uv_textures:
48             if lay.active:
49                 active_uv_layer = lay.data
50                 break
51
52         fuvs = [(uv.uv1, uv.uv2, uv.uv3, uv.uv4) for uv in active_uv_layer]
53         fuvs_cpy = [(uv[0].copy(), uv[1].copy(), uv[2].copy(), uv[3].copy()) for uv in fuvs]
54         
55         # as a list
56         faces = mesh.faces[:]
57         
58         fuvsel = [(False not in uv.uv_selected) for uv in active_uv_layer]
59         
60         file = open(self.properties.path, "w")
61         fw = file.write
62         
63         fw('<?xml version="1.0" standalone="no"?>\n')
64         fw('<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" \n')
65         fw('  "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">\n')
66         fw('<svg width="10cm" height="10cm" viewBox="0 0 1000 1000"\n')
67         fw('     xmlns="http://www.w3.org/2000/svg" version="1.1">\n')
68         
69         fw('<desc>%s, %s, %s</desc>\n' % (bpy.data.filename, ob.name, mesh.name))
70         
71         # svg colors
72         fill_settings = []
73         for mat in mesh.materials:
74             if mat:
75                 fill_settings.append('fill="rgb(%d, %d, %d)"' % tuple(int(c*255) for c in mat.diffuse_color))
76             else:
77                 fill_settings.append('fill="grey"')
78         
79         for i, uv in enumerate(active_uv_layer):
80             if len(faces[i].verts) == 3:
81                 uvs = uv.uv1, uv.uv2, uv.uv3
82             else:
83                 uvs = uv.uv1, uv.uv2, uv.uv3, uv.uv4
84
85             fw('<polygon %s fill-opacity="0.5" stroke="black" stroke-width="1" \n' % fill_settings[faces[i].material_index])
86             fw('  points="')
87             
88             for j, uv in enumerate(uvs):
89                 x, y = uv.x, 1.0 - uv.y
90                 fw('%f.3f,%f.3f ' % (x * 1000.0, y * 1000.0))
91             fw('" />\n')
92         fw('\n')
93         fw('</svg>\n')
94         
95         if is_editmode:
96             bpy.ops.object.mode_set(mode='EDIT', toggle=False)
97
98         return {'FINISHED'}
99
100     def invoke(self, context, event):
101         wm = context.manager
102         wm.add_fileselect(self)
103         return {'RUNNING_MODAL'}
104
105 # Register the operator
106 bpy.types.register(ExportUVLayout)
107
108 def menu_func(self, context):
109     default_path = bpy.data.filename.replace(".blend", ".svg")
110     self.layout.operator(ExportUVLayout.bl_idname).path = default_path
111
112 bpy.types.IMAGE_MT_uvs.append(menu_func)
113
114 #if __name__ == "__main__":
115 #    bpy.ops.uv.export_layout(path="/home/ideasman42/foo.svg")