X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=blender%2Fio_mspgl%2Fmesh.py;h=ba9db8022c7e50652e5ed6258fab1821e1b7f229;hb=3d6f8d2781beb692e11488c34871ec62174e58df;hp=3b1c74dfbec9675403226a3435fed354ffed7ef8;hpb=74a5bc6159d2c753786a5ef6bf785263cd0538f1;p=libs%2Fgl.git diff --git a/blender/io_mspgl/mesh.py b/blender/io_mspgl/mesh.py index 3b1c74df..ba9db802 100644 --- a/blender/io_mspgl/mesh.py +++ b/blender/io_mspgl/mesh.py @@ -75,6 +75,12 @@ class VertexGroup: self.weight = 0.0 +class Batch: + def __init__(self, pt): + self.primitive_type = pt + self.vertices = [] + + class Face: def __init__(self, face): self.index = face.index @@ -158,6 +164,7 @@ class Mesh: self.use_uv = mesh.use_uv self.tangent_uvtex = mesh.tangent_uvtex self.use_strips = mesh.use_strips + self.use_lines = mesh.use_lines self.vertex_groups = mesh.vertex_groups # Clone basic data @@ -242,7 +249,7 @@ class Mesh: if normal_prop and normal_prop.texture: self.tangent_vecs = True - self.vertex_sequence = [] + self.batches = [] def transform(self, matrix): for v in self.vertices: @@ -257,12 +264,12 @@ class Mesh: # Merge materials and form a lookup from source material indices to the # merged material list - material_atlas = [] + material_lookup = [] for m in other.materials: if m in self.materials: - material_atlas.append(self.materials.index(m)) + material_lookup.append(self.materials.index(m)) else: - material_atlas.append(len(self.materials)) + material_lookup.append(len(self.materials)) self.materials.append(m) # Append data and adjust indices where necessary. Since the data is @@ -293,7 +300,7 @@ class Mesh: f.index += offset f.loop_indices = range(f.loop_indices.start+offset, f.loop_indices.stop+offset) if other.materials: - f.material_index = material_atlas[f.material_index] + f.material_index = material_lookup[f.material_index] offset = len(self.edges) self.edges += other.edges @@ -411,30 +418,6 @@ class Mesh: for g in v.groups: g.group = group_index_map[g.group] - def apply_material_atlas(self, material_atlas): - for m in self.materials: - if m.name not in material_atlas.material_names: - raise Exception("Material atlas {} is not compatible with Mesh {}".format(material_atlas.name, self.name)) - - if self.use_uv=='NONE': - return - - layer = UvLayer("material_atlas") - if self.use_uv=='UNIT0': - self.uv_layers = [layer] - layer.unit = 0 - else: - self.uv_layers.append(layer) - used_units = [u.unit for u in self.uv_layers] - layer.unit = next(i for i in itertools.count() if i not in used_units) - self.uv_layers.sort(key=lambda u: u.unit) - - layer.uvs = [(0.0, 0.0)]*len(self.loops) - for f in self.faces: - uv = material_atlas.get_material_uv(self.materials[f.material_index]) - for i in f.loop_indices: - layer.uvs[i] = uv - def prepare_uv(self, task): # Form a list of UV layers referenced by materials with the array atlas # property set @@ -658,6 +641,17 @@ class Mesh: self.reorder_faces(subtask) subtask = task.task("Building sequence", 1.0) + if self.use_strips: + self.build_tristrip_sequence(subtask) + else: + self.build_triangle_sequence(subtask) + + if self.use_lines: + self.build_line_sequence() + + self.reorder_vertices() + + def build_tristrip_sequence(self, task): sequence = None for i, f in enumerate(self.faces): if sequence: @@ -681,10 +675,23 @@ class Mesh: sequence += to_add if not sequence: - sequence = f.vertices[:] - self.vertex_sequence.append(sequence) + self.batches.append(Batch("TRIANGLE_STRIP")) + sequence = self.batches[-1].vertices + sequence += f.vertices - subtask.set_progress(i/len(self.faces)) + task.set_progress(i/len(self.faces)) + + def build_triangle_sequence(self, task): + batch = Batch("TRIANGLES") + for f in self.faces: + batch.vertices += f.vertices + self.batches.append(batch) + + def build_line_sequence(self): + batch = Batch("LINES") + for l in self.lines: + batch.vertices += l.vertices + self.batches.append(batch) self.reorder_vertices() @@ -776,8 +783,8 @@ class Mesh: v.index = -1 reordered_vertices = [] - for s in self.vertex_sequence: - for v in s: + for b in self.batches: + for v in b.vertices: if v.index<0: v.index = len(reordered_vertices) reordered_vertices.append(v) @@ -793,7 +800,7 @@ class Mesh: e.key = make_edge_key(e.vertices[0].index, e.vertices[1].index) -def create_mesh_from_object(ctx, obj, material_atlas): +def create_mesh_from_object(ctx, obj): if obj.type!="MESH": raise Exception("Object {} is not a mesh".format(obj.name)) @@ -838,9 +845,6 @@ def create_mesh_from_object(ctx, obj, material_atlas): mesh.name = obj.data.name - if material_atlas: - mesh.apply_material_atlas(material_atlas) - task = ctx.task("Triangulating", 0.3) mesh.prepare_triangles(task) task = ctx.task("Smoothing", 0.5)