+ for i in range(len(self.verts)):
+ if len(groups[i])==1:
+ continue
+
+ if debug:
+ print "Vertex %s has %d groups"%(self.verts[i], len(groups[i]))
+
+ for g in groups[i][1:]:
+ v = Vertex(self.verts[i])
+ v.index = len(self.verts)
+ self.verts.append(v)
+
+ if debug:
+ print " -> %d %s"%(v.index, [f.index for f in g])
+
+ for f in g:
+ for j in range(len(f.edges)):
+ e = f.edges[j]
+
+ if e.v1!=self.verts[i] and e.v2!=self.verts[i]:
+ continue
+
+ if debug:
+ print " Splitting edge %s with faces %s"%(e.key, e.faces)
+
+ if e.other_face(f) not in g and len(e.faces)>=2:
+ k = e.faces.index(f)
+ e.faces.remove(f)
+ e = Edge(e)
+ f.edges[j] = e
+ e.faces.append(f)
+ else:
+ del self.edges[e.key]
+
+ if e.v1==self.verts[i]:
+ e.v1 = v
+ elif e.v2==self.verts[i]:
+ e.v2 = v
+
+ e.key = make_edge_key(e.v1.index, e.v2.index)
+ self.edges[e.key] = e
+
+ self.verts[i].faces.remove(f)
+ f.verts[f.verts.index(self.verts[i])] = v
+ v.faces.append(f)
+
+ def split_smooth(self, debug = False):
+ self.split_vertices(self.find_smooth_group, debug)
+
+ def split_uv(self, debug = False):
+ self.split_vertices(self.find_uv_group, debug)
+
+ def find_smooth_group(self, vert, face):
+ face.flag = True
+ queue = [face]
+
+ for f in queue:
+ for e in f.edges:
+ other = e.other_face(f)
+ #if not other or other.index not in face_indices:
+ if other not in vert.faces:
+ continue
+
+ if e.smooth:
+ if not other.flag:
+ other.flag = True
+ queue.append(other)
+
+ return queue
+
+ def find_uv_group(self, vert, face):
+ uv = face.uv[face.verts.index(vert)]
+ face.flag = True
+ group = [face]
+ for f in vert.faces:
+ if not f.flag and f.uv[f.verts.index(vert)]==uv:
+ f.flag = True
+ group.append(f)
+ return group
+
+ def compute_normals(self):
+ for v in self.verts:
+ if v.faces:
+ v.no = Blender.Mathutils.Vector()
+ for f in v.faces:
+ v.no += f.no
+ v.no.normalize()
+ else:
+ # XXX Should use edges to compute normal
+ v.no = Blender.Mathutils.Vector(0, 0, 1)
+
+ def compute_uv(self):
+ for v in self.verts:
+ if v.faces:
+ v.uv = v.faces[0].uv[v.faces[0].verts.index(v)]
+
+ def compute_tbn(self):
+ for v in self.verts:
+ v.tan = Blender.Mathutils.Vector()
+ v.bino = Blender.Mathutils.Vector()
+ for f in v.faces:
+ fverts = f.pivot_vertices(False, v)
+ v1 = fverts[1]
+ v2 = fverts[-1]
+ du1 = v1.uv[0]-v.uv[0]
+ du2 = v2.uv[0]-v.uv[0]
+ dv1 = v1.uv[1]-v.uv[1]
+ dv2 = v2.uv[1]-v.uv[1]
+ div = du1*dv2-du2*dv1
+ edge1 = fverts[1].co-fverts[0].co
+ edge2 = fverts[-1].co-fverts[0].co
+ v.tan += (edge1*dv2-edge2*dv1)/div
+ v.bino += (edge2*du1-edge1*du2)/div
+ v.tan.normalize()
+ v.bino.normalize()
+
+ def create_strip(self, face, max_len, debug):
+ edge = None
+ for e in face.edges:
+ other = e.other_face(face)
+ if other and not other.flag:
+ edge = e
+ break
+
+ if not edge:
+ return None
+
+ if debug:
+ print "Starting strip from %s, edge %s"%([v.index for v in face.verts], (edge.v1.index, edge.v2.index))
+
+ verts = face.pivot_vertices(edge.v1, edge.v2)
+ if len(verts)==3:
+ result = [verts[-1], verts[0]]
+ else:
+ result = [verts[-2], verts[-1]]
+
+ while 1:
+ if debug:
+ print " Adding %s"%face
+
+ verts = face.pivot_vertices(*result[-2:])
+ k = len(result)%2
+
+ 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])
+
+ if len(result)>=max_len:
+ if debug:
+ print " Max length exceeded"
+ break
+
+ edge = face.get_edge(*result[-2:])
+
+ if debug:
+ print " Next edge is %s"%(edge.key, )
+
+ next = edge.other_face(face)
+ if not next or next.flag:
+ break
+ face = next
+
+ if debug:
+ print " %s"%[v.index for v in result]
+
+ return result
+
+
+class VertexCache:
+ def __init__(self, size):
+ self.size = size
+ self.slots = [-1]*self.size
+
+ def fetch(self, v):
+ hit = v.index in self.slots
+ if hit:
+ self.slots.remove(v.index)
+ self.slots.append(v.index)
+ if not hit:
+ del self.slots[0]
+ return hit
+
+ def fetch_strip(self, strip):
+ hits = 0
+ for v in strip:
+ if self.fetch(v):
+ hits += 1
+ return hits
+
+ def test_strip(self, strip):
+ hits = 0
+ for i in range(len(strip)):
+ if i>=self.size:
+ break
+ if strip[i].index in self.slots[i:]:
+ hits += 1
+ return hits
+
+
+class Exporter:
+ def __init__(self, fn):
+ self.filename = fn
+ if fn==None:
+ self.out_file = sys.stdout
+ else:
+ self.out_file = file(fn, "w")
+ self.use_strips = True
+ self.use_degen_tris = True
+ self.max_strip_len = 1024
+ self.optimize_cache = False
+ self.cache_size = 64
+ self.export_lines = True
+ self.tbn_vecs = False
+ self.compound = False
+ self.debug = False
+ self.strip_debug = False
+ self.split_debug = False
+
+ def stripify(self, mesh):
+ for f in mesh.faces:
+ f.flag = False
+
+ strips = []
+
+ while 1:
+ best = 5
+ face = None
+ for f in mesh.faces:
+ if f.flag:
+ continue
+ score = 0
+ for e in f.edges:
+ other = e.other_face(f)
+ if other and not other.flag:
+ score += 1
+ if score>0 and score<best:
+ face = f
+ best = score
+
+ if not face:
+ break
+
+ strip = mesh.create_strip(face, self.max_strip_len, self.strip_debug)
+ if strip:
+ strips.append(strip)
+
+ loose = [f for f in mesh.faces if not f.flag]
+
+ if self.debug:
+ print "%d strips:"%len(strips)
+ for i in range(len(strips)):
+ print " %d: %d indices"%(i, len(strips[i]))
+ print "%d loose faces"%len([f for f in mesh.faces if not f.flag])
+ nind = sum([len(s) for s in strips])+sum([len(f.verts) for f in loose])
+ print "%d indices total"%nind
+
+ if self.use_degen_tris and strips:
+ big_strip = []
+
+ cache = None
+ total_hits = 0
+ if self.optimize_cache:
+ cache = VertexCache(self.cache_size)
+
+ while strips:
+ best = 0
+ if cache:
+ best_hits = 0
+ for i in range(len(strips)):
+ hits = cache.test_strip(strips[i])
+ if hits>best_hits:
+ best = i
+ best_hits = hits
+
+ s = strips[best]
+
+ if big_strip:
+ glue = [big_strip[-1], s[0]]
+ if len(big_strip)%2:
+ glue += [s[0]]
+
+ big_strip += glue
+ if cache:
+ total_hits += cache.fetch_strip(glue)
+
+ big_strip += s
+ if cache:
+ total_hits += cache.fetch_strip(s)
+
+ del strips[best]
+
+ for f in loose:
+ if len(big_strip)%2:
+ order = (-1, -2, 0, 1)
+ else:
+ order = (0, 1, -1, -2)
+ verts = [f.verts[i] for i in order[:len(f.verts)]]
+ if big_strip:
+ glue = [big_strip[-1], verts[0]]
+ big_strip += glue
+ if cache:
+ total_hits += cache.fetch_strip(glue)
+ big_strip += verts
+ if cache:
+ total_hits += cache.fetch_strip(verts)
+
+ strips = [big_strip]
+ loose = []
+
+ if self.debug:
+ nind = len(big_strip)
+ print "Big strip has %d indices"%nind
+ if self.optimize_cache:
+ print "%d cache hits"%total_hits
+
+ if self.debug:
+ ntris = sum([len(f.verts)-2 for f in mesh.faces])
+ print "%.2f indices per triangle"%(float(nind)/max(ntris, 1))
+
+ return strips, loose
+
+ def export(self):
+ scene = bpy.data.scenes.active
+
+ objs = Blender.Object.GetSelected()
+ if not objs:
+ raise Exception, "Nothing to export"
+ for o in objs:
+ if o.getType()!="Mesh":
+ raise Exception, "Can only export Mesh data"
+
+ mesh = Blender.Mesh.New("export_tmp")
+ mesh.getFromObject(objs[0])
+ mesh = Mesh(mesh)
+ if self.compound:
+ bmeshes = []
+ for o in objs[1:]:
+ bmesh = Blender.Mesh.New("export_tmp")
+ bmesh.getFromObject(o)
+ bmeshes.append(bmesh)
+ mesh.splice(Mesh(bmesh))
+
+ if self.debug:
+ ntris = sum([len(f.verts)-2 for f in mesh.faces])
+ print "Starting with %d vertices, %d faces (%d triangles) and %d edges"%(len(mesh.verts), len(mesh.faces), ntris, len(mesh.edges))
+
+ mesh.split_smooth(self.split_debug)
+
+ if self.debug:
+ print "After smooth splitting %d vertices and %d edges"%(len(mesh.verts), len(mesh.edges))
+
+ mesh.compute_normals()
+
+ if mesh.faceUV:
+ mesh.split_uv(self.split_debug)
+ if self.debug:
+ print "After UV splitting %d vertices and %d edges"%(len(mesh.verts), len(mesh.edges))
+
+ mesh.compute_uv()
+ if self.tbn_vecs:
+ mesh.compute_tbn()
+
+ strips = []
+ loose = mesh.faces
+ if self.use_strips:
+ strips, loose = self.stripify(mesh)
+
+ self.out_file.write("vertices NORMAL3")
+ if mesh.faceUV:
+ self.out_file.write("_TEXCOORD2")
+ if self.tbn_vecs:
+ self.out_file.write("_ATTRIB33_ATTRIB34")
+ self.out_file.write("_VERTEX3\n{\n")
+ norm = None
+ uv = None
+ tan = None
+ bino = None
+ for v in mesh.verts:
+ if v.no!=norm:
+ self.out_file.write("\tnormal3 %f %f %f;\n"%tuple(v.no))
+ norm = v.no
+ if v.uv!=uv:
+ self.out_file.write("\ttexcoord2 %f %f;\n"%tuple(v.uv))
+ uv = v.uv
+ if v.tan!=tan:
+ self.out_file.write("\tattrib3 3 %f %f %f;\n"%tuple(v.tan))
+ tan = v.tan
+ if v.bino!=bino:
+ self.out_file.write("\tattrib3 4 %f %f %f;\n"%tuple(v.bino))
+ bino = v.bino
+ self.out_file.write("\tvertex3 %f %f %f;\n"%tuple(v.co))