]> git.tdb.fi Git - libs/gl.git/blobdiff - blender/io_mspgl/export_mesh.py
Deal with some irregularities in the Blender exporter
[libs/gl.git] / blender / io_mspgl / export_mesh.py
index 3f0771fe426b60ad1a59d4fbeb47f85016c5a686..3da09f9ab92d99cc918cf7375ed692e3f12aeb63 100644 (file)
@@ -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()
 
@@ -271,21 +286,24 @@ class MeshExporter:
                if self.object:
                        out_file.begin("mesh")
 
-               fmt = "NORMAL3"
+               fmt = ["NORMAL3"]
                if texunits:
                        for i, u in texunits:
                                if u.unit==0:
-                                       fmt += "_TEXCOORD2"
+                                       fmt.append("TEXCOORD2")
                                else:
-                                       fmt += "_TEXCOORD2%d"%u.unit
+                                       fmt.append("TEXCOORD2_%d"%u.unit)
                        if self.tbn_vecs:
-                               fmt += "_ATTRIB33_ATTRIB34"
-               fmt += "_VERTEX3"
-               out_file.begin("vertices", fmt)
+                               fmt += ["TANGENT3", "BINORMAL3"]
+               if self.export_groups:
+                       fmt.append("ATTRIB%d_5"%(self.max_groups*2))
+               fmt.append("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)
@@ -299,11 +317,19 @@ class MeshExporter:
                                        uvs[i] = v.uvs[i]
                        if self.tbn_vecs:
                                if v.tan!=tan:
-                                       out_file.write("attrib3", 3, *v.tan)
+                                       out_file.write("tangent3", *v.tan)
                                        tan = v.tan
                                if v.bino!=bino:
-                                       out_file.write("attrib3", 4, *v.bino)
+                                       out_file.write("binormal3", *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: