Cycles: svn merge -r40358:40411 ^/trunk/blender
[blender.git] / intern / cycles / blender / addon / xml.py
1 #
2 # Copyright 2011, Blender Foundation.
3 #
4 # This program is free software; you can redistribute it and/or
5 # modify it under the terms of the GNU General Public License
6 # as published by the Free Software Foundation; either version 2
7 # of the License, or (at your option) any later version.
8 #
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 # GNU General Public License for more details.
13 #
14 # You should have received a copy of the GNU General Public License
15 # along with this program; if not, write to the Free Software Foundation,
16 # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17 #
18
19 # XML exporter for generating test files, not intended for end users
20
21 import os
22 import bpy
23 from bpy_extras.io_utils import ExportHelper
24 import xml.etree.ElementTree as etree
25 import xml.dom.minidom as dom
26
27 def strip(root):
28     root.text = None
29     root.tail = None
30
31     for elem in root:
32         strip(elem)
33
34 def write(node, fname):
35     strip(node)
36
37     s = etree.tostring(node)
38     s = dom.parseString(s).toprettyxml()
39
40     f = open(fname, "w")
41     f.write(s)
42
43 class ExportCyclesXML(bpy.types.Operator, ExportHelper):
44     ''''''
45     bl_idname = "export_mesh.cycles_xml"
46     bl_label = "Export Cycles XML"
47
48     filename_ext = ".xml"
49
50     @classmethod
51     def poll(cls, context):
52         return context.active_object != None
53
54     def execute(self, context):
55         filepath = bpy.path.ensure_ext(self.filepath, ".xml")
56
57         # get mesh
58         scene = context.scene
59         object = context.object
60
61         if not object:
62             raise Exception("No active object")
63
64         mesh = object.to_mesh(scene, True, 'PREVIEW')
65
66         if not mesh:
67             raise Exception("No mesh data in active object")
68
69         # generate mesh node
70         nverts = ""
71         verts = ""
72         P = ""
73
74         for v in mesh.vertices:
75             P += "%f %f %f  " % (v.co[0], v.co[1], v.co[2])
76
77         for i, f in enumerate(mesh.faces):
78             nverts += str(len(f.vertices)) + " "
79
80             for v in f.vertices:
81                 verts += str(v) + " "
82             verts += " "
83
84         node = etree.Element('mesh', attrib={'nverts': nverts, 'verts': verts, 'P': P})
85         
86         # write to file
87         write(node, filepath)
88
89         return {'FINISHED'}
90
91 def register():
92     pass
93
94 def unregister():
95     pass
96
97 if __name__ == "__main__":
98     register()
99