X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=mesh_export.py;h=0158bdbf9189d3cff6e2bedae25b8e040601d02b;hb=926d77eb54f915a6620eb879a91c1c4bd0f06bbc;hp=932433360f93649db45d61b6c63b290990a37f0b;hpb=b31a5873556fe046488587568fe35236227e3606;p=libs%2Fgl.git diff --git a/mesh_export.py b/mesh_export.py index 93243336..0158bdbf 100644 --- a/mesh_export.py +++ b/mesh_export.py @@ -9,6 +9,7 @@ Group: 'Export' import sys import math +import bpy import Blender class Edge: @@ -55,13 +56,23 @@ class Face: self.smooth_group=None self.edges=[] self.verts=[v for v in mf.verts] - + self.flag=False + def __getattr__(self, attr): return getattr(self._mface, attr) + def get_vertices_from(self, *vt): + indices=[u.index for u in vt] + flags=[(v.index in indices) for v in self.verts] + l=len(self.verts) + for i in range(l): + if flags[i] and not flags[(i+l-1)%l]: + return self.verts[i:]+self.verts[:i] + class SmoothGroup: - def __init__(self): + def __init__(self, index): + self.index=index self.faces=[] self.verts=[] @@ -72,13 +83,16 @@ class SmoothGroup: v=f.verts[i] if v.index not in vert_map: vt=Vertex(v) - if not f.smooth: - vt.no=f.no + vt.no=Blender.Mathutils.Vector(f.no) self.verts.append(vt) vert_map[v.index]=vt f.verts[i]=vt else: f.verts[i]=vert_map[v.index] + vert_map[v.index].no+=f.no + + for v in self.verts: + v.no.normalize() class Exporter: @@ -88,25 +102,76 @@ class Exporter: self.out_file=sys.stdout else: self.out_file=file(fn, "w") + self.use_strips=True + self.use_degen_tris=True + self.debug=False def find_smooth_group(self, face, sg): face.smooth_group=sg sg.faces.append(face) queue=[face] while queue: - f=queue.pop(0) - for e in f.edges: + cur=queue.pop(0) + for e in cur.edges: if e.smooth: - f2=e.other_face(f) - if f2 and not f2.smooth_group: - f2.smooth_group=sg - sg.faces.append(f2) - queue.append(f2) + other=e.other_face(cur) + if other and not other.smooth_group: + other.smooth_group=sg + sg.faces.append(other) + queue.append(other) + + def create_strip(self, face): + edge=None + for e in face.edges: + other=e.other_face(face) + if other and other.smooth_group.index==face.smooth_group.index and not other.flag: + edge=e + break + + if not edge: + return None + + if self.debug: + print "Starting strip from %s"%[v.index for v in face.verts] + + verts=face.get_vertices_from(edge.v1, edge.v2) + result=[verts[-2], verts[-1]] + + while 1: + verts=face.get_vertices_from(*result[-2:]) + k=len(result)%2 + if self.debug: + print " %d %s"%(len(result), [v.index for v in verts]) + + face.flag=True + if len(verts)==4 and not k: + result.append(verts[3]) + result.append(verts[2]) + if len(verts)==4 and k: + result.append(verts[3]) + + i1=result[-2].index + i2=result[-1].index + ekey=(min(i1, i2), max(i1, i2)) + for e in face.edges: + if e.key==ekey: + edge=e + break + + next=edge.other_face(face) + if not next or next.smooth_group.index!=face.smooth_group.index or next.flag: + break + face=next + + if self.debug: + print " %s"%[v.index for v in result] + + return result def export(self): - scene=Blender.Scene.GetCurrent() + scene=bpy.data.scenes.active - obj=scene.getActiveObject() + obj=scene.objects.active if obj.getType()!="Mesh": raise Exception, "Can only export Mesh data" @@ -124,30 +189,86 @@ class Exporter: for e in edges.itervalues(): e.check_smooth(smooth_limit) + if self.debug: + print "%d faces, %d edges"%(len(faces), len(edges)) + smooth_groups=[] for f in faces: if not f.smooth_group: - sg=SmoothGroup() + sg=SmoothGroup(len(smooth_groups)) smooth_groups.append(sg) self.find_smooth_group(f, sg) - verts=[] for sg in smooth_groups: sg.find_vertices() + + if self.debug: + print "%d smooth groups:" + for i in range(len(smooth_groups)): + sg=smooth_groups[i] + print " %d: %d faces, %d vertices"%(i, len(sg.faces), len(sg.verts)) + + strips=[] + if self.use_strips: + for sg in smooth_groups: + for f in sg.faces: + if not f.flag: + strip=self.create_strip(f) + if strip: + strips.append(strip) + + if self.use_degen_tris: + big_strip=[] + for s in strips: + if big_strip: + big_strip+=[big_strip[-1], s[0]] + big_strip+=s + + for f in faces: + if not f.flag: + if big_strip: + big_strip+=[big_strip[-1], f.verts[0]] + big_strip+=[f.verts[i] for i in (0, 1, -1)] + if len(f.verts)==4: + big_strip.append(f.verts[-2]) + f.flag=True + + strips=[big_strip] + + verts=[] + for sg in smooth_groups: for v in sg.verts: v.index=len(verts) verts.append(v) self.out_file.write("vertices NORMAL3_VERTEX3\n{\n") + norm=None for v in verts: - self.out_file.write("\tnormal3 %g %g %g;\n"%tuple(v.no)) - self.out_file.write("\tvertex3 %g %g %g;\n"%tuple(v.co)) + if v.no!=norm: + self.out_file.write("\tnormal3 %f %f %f;\n"%tuple(v.no)) + norm=v.no + self.out_file.write("\tvertex3 %f %f %f;\n"%tuple(v.co)) self.out_file.write("};\n") - self.out_file.write("batch TRIANGLES\n{\n") + for s in strips: + self.out_file.write("batch TRIANGLE_STRIP\n{\n\tindices") + n=0 + for v in s: + self.out_file.write(" %u"%v.index) + n+=1; + if n%32==0: + self.out_file.write(";\n\tindices") + self.out_file.write(";\n};\n") + + first=True for f in faces: - for i in range(2, len(f.verts)): - self.out_file.write("\tindices %u %u %u;\n"%(f.verts[0].index, f.verts[i-1].index, f.verts[i].index)) - self.out_file.write("};\n") + if not f.flag: + if first: + self.out_file.write("batch TRIANGLES\n{\n") + first=False + for i in range(2, len(f.verts)): + self.out_file.write("\tindices %u %u %u;\n"%(f.verts[0].index, f.verts[i-1].index, f.verts[i].index)) + if not first: + self.out_file.write("};\n") class FrontEnd: @@ -157,6 +278,7 @@ class FrontEnd: def export(self, fn): exp=Exporter(fn) + #exp.use_degen_tris=False exp.export()