Support for exporting armatures and vertex groups

[libs/gl.git] / blender / io_mspgl / export_mesh.py

diff --git a/blender/io_mspgl/export_mesh.py b/blender/io_mspgl/export_mesh.py
index 3f0771fe426b60ad1a59d4fbeb47f85016c5a686..0d2e66ab442206b9f2454baa0ffe2e37f7557cb3 100644 (file)
--- a/blender/io_mspgl/export_mesh.py
+++ b/blender/io_mspgl/export_mesh.py
@@ -1,3 +1,4 @@
+import itertools
 import bpy
 from .outfile import OutFile
 
@@ -48,6 +49,8 @@ class MeshExporter:
                self.material_tex = False
                self.textures = "REF"
                self.smoothing = "MSPGL"
+               self.export_groups = False
+               self.max_groups = 2
 
        def stripify(self, mesh, progress = None):
                for f in mesh.faces:
@@ -227,6 +230,18 @@ class MeshExporter:
                if self.smoothing!="BLENDER":
                        mesh.compute_normals()
 
+               if self.export_groups:
+                       mesh.sort_vertex_groups(self.max_groups)
+
+                       # Create a mapping from vertex group indices to bone indices
+                       group_index_map = dict((i, i) for i in range(len(objs[0].vertex_groups)))
+                       if objs[0].parent and objs[0].parent.type=="ARMATURE":
+                               armature = objs[0].parent.data
+                               bone_indices = dict((armature.bones[i].name, i) for i in range(len(armature.bones)))
+                               for g in objs[0].vertex_groups:
+                                       if g.name in bone_indices:
+                                               group_index_map[g.index] = bone_indices[g.name]
+
                if self.material_tex and mesh.materials:
                        mesh.generate_material_uv()
 
@@ -280,12 +295,15 @@ class MeshExporter:
                                        fmt += "_TEXCOORD2%d"%u.unit
                        if self.tbn_vecs:
                                fmt += "_ATTRIB33_ATTRIB34"
+               if self.export_groups:
+                       fmt += "_ATTRIB%d5"%(self.max_groups*2)
                fmt += "_VERTEX3"
                out_file.begin("vertices", fmt)
                normal = None
                uvs = [None]*len(texunits)
                tan = None
                bino = None
+               group = None
                for v in mesh.vertices:
                        if v.normal!=normal:
                                out_file.write("normal3", *v.normal)
@@ -304,6 +322,14 @@ class MeshExporter:
                                if v.bino!=bino:
                                        out_file.write("attrib3", 4, *v.bino)
                                        bino = v.bino
+                       if self.export_groups:
+                               group_attr = [(group_index_map[g.group], g.weight*v.group_weight_scale) for g in v.groups[:self.max_groups]]
+                               while len(group_attr)<self.max_groups:
+                                       group_attr.append((0, 0.0))
+                               group_attr = list(itertools.chain(*group_attr))
+                               if group_attr!=group:
+                                       out_file.write("attrib%d"%len(group_attr), 5, *group_attr)
+                                       group = group_attr
                        out_file.write("vertex3", *v.co)
                out_file.end()
                for s in strips: