X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=blender%2Fio_mspgl%2Fexport_mesh.py;h=6d90c8faf6d5f435ac4daec803e8a02cb0200b86;hb=306be12bfad9de3c0fee353feaaa9904bc925fb6;hp=541874b2bd2018c052d286962956a0a0a45f77b6;hpb=fc7566b2d83eb5b73aa516b045116e02150df83c;p=libs%2Fgl.git diff --git a/blender/io_mspgl/export_mesh.py b/blender/io_mspgl/export_mesh.py index 541874b2..6d90c8fa 100644 --- a/blender/io_mspgl/export_mesh.py +++ b/blender/io_mspgl/export_mesh.py @@ -41,15 +41,7 @@ class MeshExporter: self.max_strip_len = 1024 self.optimize_cache = True self.cache_size = 64 - self.export_lines = False - self.export_uv = "UNIT0" - self.tbn_vecs = False - self.tbn_uvtex = "" - self.compound = False self.material_tex = False - self.smoothing = "MSPGL" - self.export_groups = False - self.max_groups = 2 def stripify(self, mesh, progress=None): for f in mesh.faces: @@ -196,32 +188,11 @@ class MeshExporter: return strips, loose - def export(self, context, out_file, objs=None, progress=None): - if objs: - objs = [(o, mathutils.Matrix()) for o in objs] - - if self.compound: - if objs is None: - objs = [(o, mathutils.Matrix()) for o in context.selected_objects] - check = objs - while check: - children = [] - for o, m in check: - for c in o.children: - if c.compound: - children.append((c, m*c.matrix_local)) - objs += children - check = children - elif objs is None: - objs = [(context.active_object, mathutils.Matrix())] - - if not objs: - raise Exception("Nothing to export") - for o, m in objs: - if o.type!="MESH": - raise Exception("Can only export Mesh data") - - from .mesh import Mesh + def export(self, context, out_file, obj=None, progress=None): + if obj is None: + obj = context.active_object + + from .mesh import create_mesh_from_object from .util import Progress if self.show_progress: @@ -231,82 +202,7 @@ class MeshExporter: else: progress = None - mesh = None - bmeshes = [] - winding_test = False - for o, m in objs: - if o.data.winding_test: - winding_test = True - bmesh = o.to_mesh(context.scene, True, "PREVIEW") - bmeshes.append(bmesh) - me = Mesh(bmesh) - me.transform(m) - if not mesh: - mesh = me - else: - mesh.splice(me) - - if progress: - progress.set_task("Smoothing", 0.05, 0.35) - if self.smoothing=="NONE": - mesh.flatten_faces() - mesh.split_smooth(progress) - - 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 - first_obj = objs[0][0] - group_index_map = dict((i, i) for i in range(len(first_obj.vertex_groups))) - if first_obj.parent and first_obj.parent.type=="ARMATURE": - armature = first_obj.parent.data - bone_indices = dict((armature.bones[i].name, i) for i in range(len(armature.bones))) - for g in first_obj.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() - - texunits = [] - force_unit0 = False - if mesh.uv_layers and (self.export_uv!="NONE" or self.material_tex): - # Figure out which UV layers to export - if self.export_uv=="ALL": - texunits = range(len(mesh.uv_layers)) - elif self.material_tex: - # The material UV layer is always the last one - texunits = [len(mesh.uv_layers)-1] - force_unit0 = True - else: - for i, u in enumerate(mesh.uv_layers): - if u.unit==0: - texunits = [i] - break - texunits = [(i, mesh.uv_layers[i]) for i in texunits] - texunits = [u for u in texunits if not u[1].hidden] - - if self.tbn_vecs: - # TBN coordinates must be generated before vertices are split by any other layer - uv_names = [u.name for i, u in texunits] - if self.tbn_uvtex in uv_names: - tbn_index = uv_names.index(self.tbn_uvtex) - unit = texunits[tbn_index] - del texunits[tbn_index] - texunits.insert(0, unit) - - for i, u in texunits: - if progress: - progress.set_task("Splitting UVs", 0.35+0.3*i/len(texunits), 0.35+0.3*(i+1)/len(texunits)) - mesh.split_uv(i, progress) - if self.tbn_vecs and u.name==self.tbn_uvtex: - mesh.compute_uv() - mesh.compute_tbn(i) - - mesh.compute_uv() + mesh = create_mesh_from_object(context, obj, progress) strips = [] loose = mesh.faces @@ -322,17 +218,17 @@ class MeshExporter: out_file = open_output(out_file) fmt = ["NORMAL3"] - if texunits: - for i, u in texunits: - size = str(len(mesh.vertices[0].uvs[i])) - if u.unit==0 or force_unit0: + if mesh.uv_layers: + for u in mesh.uv_layers: + size = str(len(u.uvs[0])) + if u.unit==0: fmt.append("TEXCOORD"+size) else: fmt.append("TEXCOORD%s_%d"%(size, u.unit)) - if self.tbn_vecs: + if mesh.tbn_vecs: fmt += ["TANGENT3", "BINORMAL3"] - if self.export_groups: - fmt.append("ATTRIB%d_5"%(self.max_groups*2)) + if mesh.vertex_groups: + fmt.append("ATTRIB%d_5"%(mesh.max_groups_per_vertex*2)) fmt.append("VERTEX3") out_file.begin("vertices", *fmt) normal = None @@ -344,24 +240,24 @@ class MeshExporter: if v.normal!=normal: out_file.write("normal3", *v.normal) normal = v.normal - for i, u in texunits: + for i, u in enumerate(mesh.uv_layers): if v.uvs[i]!=uvs.get(i): size = str(len(v.uvs[i])) - if u.unit==0 or force_unit0: + if u.unit==0: out_file.write("texcoord"+size, *v.uvs[i]) else: out_file.write("multitexcoord"+size, u.unit, *v.uvs[i]) uvs[i] = v.uvs[i] - if self.tbn_vecs: + if mesh.tbn_vecs: if v.tan!=tan: out_file.write("tangent3", *v.tan) tan = v.tan if v.bino!=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)