5 Name: 'MSP GL Mesh (.mesh)...'
15 def make_edge_key(i1, i2):
16 return (min(i1, i2), max(i1, i2))
20 def __init__(self, me):
21 if me.__class__==Edge:
31 def __getattr__(self, attr):
32 return getattr(self._medge, attr)
34 def __cmp__(self, other):
37 def check_smooth(self, limit):
38 if len(self.faces)!=2:
41 d=Blender.Mathutils.DotVecs(self.faces[0].no, self.faces[1].no)
42 if (d>limit and self.faces[0].smooth and self.faces[1].smooth) or d>0.999:
45 def other_face(self, f):
46 if f.index==self.faces[0].index:
47 if len(self.faces)>=2:
56 def __init__(self, mv):
57 if mv.__class__==Vertex:
64 self.orig_index=self._mvert.index
70 def __getattr__(self, attr):
71 return getattr(self._mvert, attr)
73 def __cmp__(self, other):
76 return cmp(self.index, other.index)
79 return "<Vert %d (%.4f, %.4f, %.4f) (%.4f, %.4f, %.4f)>"%(self.index, self.co[0], self.co[1], self.co[2], self.no[0], self.no[1], self.no[2])
85 def __init__(self, mf):
88 self.verts=[v for v in mf.verts]
91 def __getattr__(self, attr):
92 return getattr(self._mface, attr)
94 def __cmp__(self, other):
97 return cmp(self.index, other.index)
100 return "<Face %d (%s)>"%(self.index, " ".join([str(v.index) for v in self.verts]))
104 def pivot_vertices(self, reverse, *vt):
108 flags=[(v in vt) for v in verts]
111 if flags[i] and not flags[(i+l-1)%l]:
112 return verts[i:]+verts[:i]
114 def get_edge(self, v1, v2):
115 key=make_edge_key(v1.index, v2.index)
119 raise KeyError, "No edge %s"%(key,)
123 def __init__(self, e):
125 self.verts=[e.v1, e.v2]
129 return "<Line (%d %d)>"%(self.verts[0].index, self.verts[1].index)
135 def __init__(self, m):
137 self.verts=[Vertex(v) for v in m.verts]
138 self.faces=[Face(f) for f in m.faces]
141 for i in range(len(f.verts)):
142 f.verts[i]=self.verts[f.verts[i].index]
143 f.verts[i].faces.append(f)
145 self.edges=dict([(e.key, Edge(e)) for e in m.edges])
147 for k in f.edge_keys:
149 e.faces.append(self.faces[f.index])
152 self.lines=[Line(e) for e in self.edges.itervalues() if not e.faces]
154 if m.mode&Blender.Mesh.Modes.AUTOSMOOTH:
155 smooth_limit=math.cos(m.degr*math.pi/180)
158 for e in self.edges.itervalues():
159 e.v1=self.verts[e.v1.index]
160 e.v2=self.verts[e.v2.index]
161 e.check_smooth(smooth_limit)
163 def __getattr__(self, attr):
164 return getattr(self._mesh, attr)
166 def split_vertices(self, find_group_func, debug):
175 vg.append(find_group_func(v, f))
179 for i in range(len(self.verts)):
180 if len(groups[i])==1:
184 print "Vertex %s has %d groups"%(self.verts[i], len(groups[i]))
186 for g in groups[i][1:]:
187 v=Vertex(self.verts[i])
188 v.index=len(self.verts)
192 print " -> %d %s"%(v.index, [f.index for f in g])
195 for j in range(len(f.edges)):
198 if e.v1!=self.verts[i] and e.v2!=self.verts[i]:
202 print " Splitting edge %s with faces %s"%(e.key, e.faces)
204 if e.other_face(f) not in g and len(e.faces)>=2:
211 del self.edges[e.key]
213 if e.v1==self.verts[i]:
215 elif e.v2==self.verts[i]:
218 e.key=make_edge_key(e.v1.index, e.v2.index)
221 self.verts[i].faces.remove(f)
222 f.verts[f.verts.index(self.verts[i])]=v
225 def split_smooth(self, debug=False):
226 self.split_vertices(self.find_smooth_group, debug)
228 def split_uv(self, debug=False):
229 self.split_vertices(self.find_uv_group, debug)
231 def find_smooth_group(self, vert, face):
237 other=e.other_face(f)
238 #if not other or other.index not in face_indices:
239 if other not in vert.faces:
249 def find_uv_group(self, vert, face):
250 uv=face.uv[face.verts.index(vert)]
254 if not f.flag and f.uv[f.verts.index(vert)]==uv:
259 def compute_normals(self):
262 v.no=Blender.Mathutils.Vector()
267 # XXX Should use edges to compute normal
268 v.no=Blender.Mathutils.Vector(0, 0, 1)
270 def compute_uv(self):
273 v.uv=v.faces[0].uv[v.faces[0].verts.index(v)]
275 def compute_tbn(self):
277 v.tan=Blender.Mathutils.Vector()
278 v.bino=Blender.Mathutils.Vector()
280 fverts=f.pivot_vertices(False, v)
288 edge1=fverts[1].co-fverts[0].co
289 edge2=fverts[-1].co-fverts[0].co
290 v.tan+=(edge1*dv2-edge2*dv1)/div
291 v.bino+=(edge2*du1-edge1*du2)/div
295 def create_strip(self, face, reverse, debug):
298 other=e.other_face(face)
299 if other and not other.flag:
307 print "Starting strip from %s, edge %s, reverse=%s"%([v.index for v in face.verts], (edge.v1.index, edge.v2.index), reverse)
309 verts=face.pivot_vertices(reverse, edge.v1, edge.v2)
311 result=[verts[-1], verts[0]]
313 result=[verts[-2], verts[-1]]
316 verts=face.pivot_vertices(reverse, *result[-2:])
319 print " Adding %s"%face
322 if len(verts)==4 and not k:
323 result.append(verts[3])
324 result.append(verts[2])
325 if len(verts)==4 and k:
326 result.append(verts[3])
328 edge=face.get_edge(*result[-2:])
331 print " Next edge is %s"%(edge.key, )
333 next=edge.other_face(face)
334 if not next or next.flag:
339 print " %s"%[v.index for v in result]
345 def __init__(self, fn):
348 self.out_file=sys.stdout
350 self.out_file=file(fn, "w")
352 self.use_degen_tris=True
353 self.optimize_locality=True
354 self.export_lines=True
357 self.strip_debug=False
358 self.split_debug=False
360 def get_locality(self, strip):
362 for i in range(1, len(strip)):
363 if strip[i].index!=strip[i-1].index:
364 total+=1.0/(abs(strip[i].index-strip[i-1].index))
365 return total/len(strip)
367 def get_followers(self, strip):
369 for i in range(len(strip)-1):
373 if v.index not in result:
375 if n.index not in result[v.index]:
376 result[v.index][n.index]=1
378 result[v.index][n.index]+=1
382 scene=bpy.data.scenes.active
384 obj=scene.objects.active
385 if obj.getType()!="Mesh":
386 raise Exception, "Can only export Mesh data"
388 mesh=Blender.Mesh.New("export_tmp")
389 mesh.getFromObject(obj)
393 ntris=sum([len(f.verts)-2 for f in mesh.faces])
394 print "Starting with %d vertices, %d faces (%d triangles) and %d edges"%(len(mesh.verts), len(mesh.faces), ntris, len(mesh.edges))
396 mesh.split_smooth(self.split_debug)
399 print "After smooth splitting %d vertices and %d edges"%(len(mesh.verts), len(mesh.edges))
401 mesh.compute_normals()
404 mesh.split_uv(self.split_debug)
406 print "After UV splitting %d vertices and %d edges"%(len(mesh.verts), len(mesh.edges))
425 other=e.other_face(f)
426 if other and not other.flag:
428 if score>0 and score<best:
435 strip=mesh.create_strip(face, self.use_degen_tris and sum([len(s) for s in strips])%2, self.strip_debug)
440 print "%d strips:"%len(strips)
441 for i in range(len(strips)):
442 print " %d: %d indices"%(i, len(strips[i]))
443 print "%d loose faces"%len([f for f in mesh.faces if not f.flag])
444 nind=sum([len(s) for s in strips])+sum([len(f.verts) for f in mesh.faces if not f.flag])
445 print "%d indices total"%nind
447 if self.use_degen_tris and strips:
451 big_strip+=[big_strip[-1], s[0]]
461 big_strip+=[big_strip[-1], f.verts[order[0]]]
462 big_strip+=[f.verts[i] for i in order[:len(f.verts)]]
469 print "Big strip has %d indices"%len(big_strip)
472 print "%.2f indices per triangle"%(float(nind)/max(ntris, 1))
473 print "Locality before optimization: "+" ".join(["%.3f"%self.get_locality(s) for s in strips])
475 if self.optimize_locality and self.use_strips and strips:
478 followers.update(self.get_followers(s))
487 if vert.index in followers:
488 flw=followers[vert.index]
491 if flw[n]>best and not mesh.verts[n].flag:
493 best=flw[n]+0.9/abs(vert.index-n)
507 for i in range(len(mesh.verts)):
508 mesh.verts[i].index=i
511 print "Locality after optimization: "+" ".join(["%.3f"%self.get_locality(s) for s in strips])
513 self.out_file.write("vertices NORMAL3")
515 self.out_file.write("_TEXCOORD2")
517 self.out_file.write("_ATTRIB33_ATTRIB34")
518 self.out_file.write("_VERTEX3\n{\n")
525 self.out_file.write("\tnormal3 %f %f %f;\n"%tuple(v.no))
528 self.out_file.write("\ttexcoord2 %f %f;\n"%tuple(v.uv))
531 self.out_file.write("\tattrib3 3 %f %f %f;\n"%tuple(v.tan))
534 self.out_file.write("\tattrib3 4 %f %f %f;\n"%tuple(v.bino))
536 self.out_file.write("\tvertex3 %f %f %f;\n"%tuple(v.co))
537 self.out_file.write("};\n")
539 self.out_file.write("batch TRIANGLE_STRIP\n{\n\tindices")
542 self.out_file.write(" %u"%v.index)
545 self.out_file.write(";\n\tindices")
546 self.out_file.write(";\n};\n")
552 self.out_file.write("batch TRIANGLES\n{\n")
554 for i in range(2, len(f.verts)):
555 self.out_file.write("\tindices %u %u %u;\n"%(f.verts[0].index, f.verts[i-1].index, f.verts[i].index))
557 self.out_file.write("};\n")
559 if self.export_lines and mesh.lines:
560 self.out_file.write("batch LINES\n{\n")
562 self.out_file.write("\tindices %u %u;\n"%(l.verts[0].index, l.verts[1].index))
563 self.out_file.write("};\n")
568 self.config=Blender.Registry.GetKey('mspgl_export', True) or {}
569 self.temp_config=Blender.Registry.GetKey('mspgl_export_temp') or {}
572 self.use_strips=Blender.Draw.Create(self.config.get('use_strips', True))
573 self.use_degen_tris=Blender.Draw.Create(self.config.get('use_degen_tris', True))
574 self.optimize_locality=Blender.Draw.Create(self.config.get('optimize_locality', True))
575 self.export_lines=Blender.Draw.Create(self.config.get('export_lines', False))
576 self.tbn_vecs=Blender.Draw.Create(self.config.get('tbn_vecs', False))
577 self.debug=Blender.Draw.Create(self.config.get('debug', False))
578 self.strip_debug=Blender.Draw.Create(self.config.get('strip_debug', False))
579 self.split_debug=Blender.Draw.Create(self.config.get('split_debug', False))
580 ret=Blender.Draw.PupBlock("Export MSP GL mesh",
581 [("Use strips", self.use_strips, "Generage OpenGL triangle strips"),
582 ("Use degen tris", self.use_degen_tris, "Use degenerate triangles to combine triangle strips"),
583 ("Optimize locality", self.optimize_locality),
584 ("Export lines", self.export_lines, "Export lone edges as lines"),
585 ("Compute T/B vecs", self.tbn_vecs, "Compute tangent/binormal vectors for bumpmapping"),
586 ("Debugging options"),
587 ("Debug", self.debug),
588 ("Debug strips", self.strip_debug),
589 ("Debug splitting", self.split_debug)])
591 dirname=self.temp_config.get("dirname", Blender.sys.dirname(Blender.Get("filename")))
592 obj=bpy.data.scenes.active.objects.active
593 Blender.Window.FileSelector(self.export, "Export MSP GL mesh", "%s/%s.mesh"%(dirname, obj.name))
598 def export(self, fn):
599 self.config['use_strips']=self.use_strips.val
600 self.config['use_degen_tris']=self.use_degen_tris.val
601 self.config['optimize_locality']=self.optimize_locality.val
602 self.config['export_lines']=self.export_lines.val
603 self.config['tbn_vecs']=self.tbn_vecs.val
604 self.config['debug']=self.debug.val
605 self.config['strip_debug']=self.strip_debug.val
606 self.config['split_debug']=self.split_debug.val
607 Blender.Registry.SetKey('mspgl_export', self.config, True)
610 self.temp_config["dirname"]=os.path.dirname(fn)
611 Blender.Registry.SetKey('mspgl_export_temp', self.temp_config)
614 exp.use_strips=self.use_strips.val
615 exp.use_degen_tris=self.use_degen_tris.val
616 exp.optimize_locality=self.optimize_locality.val
617 exp.export_lines=self.export_lines.val
618 exp.tbn_vecs=self.tbn_vecs.val
619 exp.debug=self.debug.val
620 exp.strip_debug=self.strip_debug.val
621 exp.split_debug=self.split_debug.val
625 if __name__=="__main__":