]> git.tdb.fi Git - libs/gl.git/blobdiff - blender/io_mspgl/export_mesh.py
Refactor face cull settings in Blender
[libs/gl.git] / blender / io_mspgl / export_mesh.py
index 68d93cbc525b9b3872e209c58a33bb7548ca0cba..012386185faa7755388f3aa108e5d02a093446ec 100644 (file)
@@ -1,91 +1,62 @@
-import itertools
-import os
-import bpy
-import mathutils
-
 class MeshExporter:
-       def __init__(self):
-               self.show_progress = True
-               self.use_strips = True
-               self.use_degen_tris = False
-               self.export_all = False
-
-       def join_strips(self, strips):
-               big_strip = []
-
-               for s in strips:
-                       if big_strip:
-                               # Generate glue elements, ensuring that the next strip begins at
-                               # an even position
-                               glue = [big_strip[-1], s[0]]
-                               if len(big_strip)%2:
-                                       glue += [s[0]]
-
-                               big_strip += glue
-
-                       big_strip += s
-
-               return big_strip
-
-       def export_to_file(self, context, out_fn):
-               if self.export_all:
-                       objs = [o for o in context.selected_objects if o.type=="MESH"]
-               else:
-                       objs = [context.active_object]
-
-               from .util import Progress
-
-               path, base = os.path.split(out_fn)
-               base, ext = os.path.splitext(base)
-
-               progress = Progress(self.show_progress and context)
-               for i, obj in enumerate(objs):
-                       if self.export_all:
-                               out_fn = os.path.join(path, obj.data.name+ext)
-
-                       progress.push_task_slice(obj.data.name, i, len(objs))
-                       resource = self.export_mesh(context, obj, progress)
-
-                       resource.write_to_file(out_fn)
-                       progress.pop_task()
-
-       def export_mesh(self, context, mesh_or_obj, progress):
+       def export_mesh(self, ctx, mesh_or_obj):
                from .mesh import Mesh, create_mesh_from_object
 
                if type(mesh_or_obj)==Mesh:
                        mesh = mesh_or_obj
                else:
-                       progress.push_task("", 0.0, 0.9)
-                       mesh = create_mesh_from_object(context, mesh_or_obj, progress)
-                       progress.pop_task()
+                       task = ctx.task("Preparing mesh", 0.9)
+                       mesh = create_mesh_from_object(task, mesh_or_obj)
 
                from .datafile import Resource, Statement, Token
                resource = Resource(mesh.name+".mesh", "mesh")
                statements = resource.statements
 
-               st = Statement("vertices", Token("NORMAL3"))
+               task = ctx.task("Creating statements", 1.0)
+
+               statements.append(Statement("winding", Token('COUNTERCLOCKWISE')))
+
+               st = Statement("vertices", Token("VERTEX3_FLOAT"))
+               stride = 12
                if mesh.vertices[0].color:
                        st.append(Token("COLOR4_UBYTE"))
+                       stride += 4
                if mesh.uv_layers:
                        for u in mesh.uv_layers:
-                               size = str(len(u.uvs[0]))
+                               size = len(u.uvs[0])
+                               min_val = min(*u.uvs[0])
+                               max_val = max(*u.uvs[1])
+                               for c in u.uvs:
+                                       min_val = min(min_val, *c)
+                                       max_val = max(max_val, *c)
+                               uv_type = "USHORT" if min_val>=0.0 and max_val<=1.0 else "FLOAT"
+                               if uv_type=="FLOAT" and stride%4:
+                                       pad = 4-stride%4
+                                       st.append(Token("PADDING{}".format(pad)))
+                                       stride += pad
                                if u.unit==0:
-                                       st.append(Token("TEXCOORD"+size))
+                                       st.append(Token("TEXCOORD{}_{}".format(size, uv_type)))
                                else:
-                                       st.append(Token("TEXCOORD{}_{}".format(size, u.unit)))
-                       if mesh.tbn_vecs:
-                               st.append(Token("TANGENT3"))
-                               st.append(Token("BINORMAL3"))
+                                       st.append(Token("TEXCOORD{}_{}_{}".format(size, u.unit, uv_type)))
+                               stride += (2 if uv_type=="USHORT" else 4)*size
                if mesh.vertex_groups:
-                       st.append(Token("GROUP{}".format(mesh.max_groups_per_vertex)))
-                       st.append(Token("WEIGHT{}".format(mesh.max_groups_per_vertex)))
-               st.append(Token("VERTEX3"))
+                       st.append(Token("WEIGHT{}_USHORT".format(mesh.max_groups_per_vertex)))
+                       st.append(Token("GROUP{}_UBYTE".format(mesh.max_groups_per_vertex)))
+                       stride += 3*mesh.max_groups_per_vertex
+               st.append(Token("NORMAL3_BYTE"))
+               stride += 3
+               if mesh.uv_layers and mesh.tangent_vecs:
+                       st.append(Token("TANGENT3_BYTE"))
+                       stride += 3
+               if stride%4:
+                       pad = 4-stride%4
+                       st.append(Token("PADDING{}_UBYTE".format(pad)))
+                       stride += pad
 
                normal = None
                color = None
                uvs = [None]*len(mesh.uv_layers)
                tan = None
-               bino = None
                group = None
                weight = None
                for v in mesh.vertices:
@@ -102,13 +73,10 @@ class MeshExporter:
                                        else:
                                                st.sub.append(Statement("multitexcoord", u.unit, *v.uvs[i]))
                                        uvs[i] = v.uvs[i]
-                       if mesh.tbn_vecs:
+                       if mesh.tangent_vecs:
                                if v.tan!=tan:
                                        st.sub.append(Statement("tangent", *v.tan))
                                        tan = v.tan
-                               if v.bino!=bino:
-                                       st.sub.append(Statement("binormal", *v.bino))
-                                       bino = v.bino
                        if mesh.vertex_groups:
                                v_group = [g.group for g in v.groups]
                                v_weight = [g.weight for g in v.groups]
@@ -122,12 +90,8 @@ class MeshExporter:
 
                statements.append(st)
 
-               if self.use_strips:
-                       strips = mesh.vertex_sequence
-                       if self.use_degen_tris:
-                               strips = [self.join_strips(strips)]
-
-                       for s in strips:
+               if mesh.use_strips:
+                       for s in mesh.vertex_sequence:
                                st = Statement("batch", Token("TRIANGLE_STRIP"))
                                for i in range(0, len(s), 32):
                                        st.sub.append(Statement("indices", *(v.index for v in s[i:i+32])))
@@ -144,9 +108,6 @@ class MeshExporter:
                                st.sub.append(Statement("indices", *(v.index for v in l.vertices)))
                        statements.append(st)
 
-               if mesh.winding_test:
-                       statements.append(Statement("winding", Token('COUNTERCLOCKWISE')))
-
-               progress.set_progress(1.0)
+               task.set_progress(1.0)
 
                return resource