Add support for exporting vertex colors

[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 974685b3d1fa8cf0ae579bea292dd7676a8e47a3..0d27a71cdf6c2b1e57121a2e064bec4d5067b9bb 100644 (file)
--- a/blender/io_mspgl/export_mesh.py
+++ b/blender/io_mspgl/export_mesh.py
@@ -64,6 +64,8 @@ class MeshExporter:
                statements = resource.statements
 
                st = Statement("vertices", Token("NORMAL3"))
+               if mesh.vertices[0].color:
+                       st.append(Token("COLOR4_UBYTE"))
                if mesh.uv_layers:
                        for u in mesh.uv_layers:
                                size = str(len(u.uvs[0]))
@@ -75,10 +77,11 @@ class MeshExporter:
                                st.append(Token("TANGENT3"))
                                st.append(Token("BINORMAL3"))
                if mesh.vertex_groups:
-                       st.append(Token("ATTRIB{}_5".format(mesh.max_groups_per_vertex*2)))
+                       st.append(Token("GENERIC{}_0".format(mesh.max_groups_per_vertex*2)))
                st.append(Token("VERTEX3"))
 
                normal = None
+               color = None
                uvs = [None]*len(mesh.uv_layers)
                tan = None
                bino = None
@@ -87,6 +90,9 @@ class MeshExporter:
                        if v.normal!=normal:
                                st.sub.append(Statement("normal", *v.normal))
                                normal = v.normal
+                       if v.color!=color:
+                               st.sub.append(Statement("color", *v.color))
+                               color = v.color
                        for i, u in enumerate(mesh.uv_layers):
                                if v.uvs[i]!=uvs[i]:
                                        if u.unit==0:
@@ -107,7 +113,7 @@ class MeshExporter:
                                        group_attr.append((0, 0.0))
                                group_attr = list(itertools.chain(*group_attr))
                                if group_attr!=group:
-                                       st.sub.append(Statement("attrib", 5, *group_attr))
+                                       st.sub.append(Statement("generic", 0, *group_attr))
                                        group = group_attr
                        st.sub.append(Statement("vertex", *v.co))