import bpy
import Blender
+def make_edge_key(i1, i2):
+ return (min(i1, i2), max(i1, i2))
+
+
class Edge:
def __init__(self, me):
if me.__class__==Edge:
self._medge=me._medge
+ self.v1=me.v1
+ self.v2=me.v2
self.smooth=me.smooth
else:
self._medge=me
def __getattr__(self, attr):
return getattr(self._medge, attr)
+ def __cmp__(self, other):
+ return self is other
+
def check_smooth(self, limit):
if len(self.faces)!=2:
return
d=Blender.Mathutils.DotVecs(self.faces[0].no, self.faces[1].no)
if (d>limit and self.faces[0].smooth and self.faces[1].smooth) or d>0.999:
- self.smooth=1
+ self.smooth=True
def other_face(self, f):
if f.index==self.faces[0].index:
def __init__(self, mv):
if mv.__class__==Vertex:
self._mvert=mv._mvert
+ self.no=mv.no
self.uv=mv.uv
else:
self._mvert=mv
self.uv=None
self.orig_index=self._mvert.index
+ self.flag=False
+ self.faces=[]
+ self.tan=None
+ self.bino=None
def __getattr__(self, attr):
return getattr(self._mvert, attr)
+ def __cmp__(self, other):
+ if other is None:
+ return 1
+ return cmp(self.index, other.index)
+
+ def __str__(self):
+ 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])
+
+ __repr__=__str__
+
class Face:
def __init__(self, mf):
self._mface=mf
- 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)
+ def __cmp__(self, other):
+ if other is None:
+ return 1
+ return cmp(self.index, other.index)
+
+ def __str__(self):
+ return "<Face %d (%s)>"%(self.index, " ".join([str(v.index) for v in self.verts]))
+
+ __repr__=__str__
+
+ def pivot_vertices(self, reverse, *vt):
+ verts=self.verts[:]
+ if reverse:
+ verts.reverse()
+ flags=[(v in vt) for v in verts]
+ l=len(verts)
for i in range(l):
if flags[i] and not flags[(i+l-1)%l]:
- return self.verts[i:]+self.verts[:i]
+ return verts[i:]+verts[:i]
def get_edge(self, v1, v2):
- i1=v1.index
- i2=v2.index
+ key=make_edge_key(v1.index, v2.index)
for e in self.edges:
- if (e.v1.index==i1 or e.v1.index==i2) and (e.v2.index==i1 or e.v2.index==i2):
+ if e.key==key:
return e
+ raise KeyError, "No edge %s"%(key,)
-class SmoothGroup:
- def __init__(self, index):
- self.index=index
- self.faces=[]
- self.verts=[]
+class Line:
+ def __init__(self, e):
+ self.edge=e
+ self.verts=[e.v1, e.v2]
+ self.flag=False
+
+ def __str__(self):
+ return "<Line (%d %d)>"%(self.verts[0].index, self.verts[1].index)
+
+ __repr__=__str__
+
+
+class Mesh:
+ def __init__(self, m):
+ self._mesh=m
+ self.verts=[Vertex(v) for v in m.verts]
+ self.faces=[Face(f) for f in m.faces]
- def find_vertices(self):
- vert_map={}
for f in self.faces:
for i in range(len(f.verts)):
- v=f.verts[i]
- if v.index not in vert_map:
- vt=Vertex(v)
- vt.index=len(self.verts)
- self.verts.append(vt)
-
- vert_map[v.index]=vt
- vt.no=Blender.Mathutils.Vector(f.no)
- else:
- vt=vert_map[v.index]
- vt.no+=f.no
- f.verts[i]=vt
+ f.verts[i]=self.verts[f.verts[i].index]
+ f.verts[i].faces.append(f)
+
+ self.edges=dict([(e.key, Edge(e)) for e in m.edges])
+ for f in self.faces:
+ for k in f.edge_keys:
+ e=self.edges[k]
+ e.faces.append(self.faces[f.index])
+ f.edges.append(e)
+
+ self.lines=[Line(e) for e in self.edges.itervalues() if not e.faces]
+
+ if m.mode&Blender.Mesh.Modes.AUTOSMOOTH:
+ smooth_limit=math.cos(m.degr*math.pi/180)
+ else:
+ smooth_limit=-1
+ for e in self.edges.itervalues():
+ e.v1=self.verts[e.v1.index]
+ e.v2=self.verts[e.v2.index]
+ e.check_smooth(smooth_limit)
+
+ def __getattr__(self, attr):
+ return getattr(self._mesh, attr)
+ def split_vertices(self, find_group_func, debug):
+ groups=[]
for v in self.verts:
- v.no.normalize()
+ for f in v.faces:
+ f.flag=False
- def separate_uv(self):
- copies={}
- for f in self.faces:
- for i in range(len(f.verts)):
- v=f.verts[i]
- if not v.uv:
- v.uv=f.uv[i]
- elif f.uv[i]!=v.uv:
- if v.index not in copies:
- copies[v.index]=[]
-
- vt=None
- for w in copies[v.index]:
- if w.uv==f.uv[i]:
- vt=w
- break
+ vg=[]
+ for f in v.faces:
+ if not f.flag:
+ vg.append(find_group_func(v, f))
- if not vt:
- vt=Vertex(v)
- vt.index=len(self.verts)
- vt.uv=f.uv[i]
- self.verts.append(vt)
- copies[v.index].append(vt)
+ groups.append(vg)
- f.verts[i]=vt
+ for i in range(len(self.verts)):
+ if len(groups[i])==1:
+ continue
- def create_edges(self):
- edge_map={}
- for f in self.faces:
- vert_map=dict([(v.orig_index, v) for v in f.verts])
- for i in range(len(f.edges)):
- v1=vert_map[f.edges[i].v1.index]
- v2=vert_map[f.edges[i].v2.index]
- key=tuple(sorted((v1.index, v2.index)))
-
- if key in edge_map:
- e=edge_map[key]
- else:
- e=Edge(f.edges[i])
- edge_map[key]=e
- e.v1=v1
- e.v2=v2
- e.key=key
+ if debug:
+ print "Vertex %s has %d groups"%(self.verts[i], len(groups[i]))
- f.edges[i]=e
- e.faces.append(f)
+ 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])
-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.debug=False
- self.strip_debug=False
+ 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
- def find_smooth_group(self, face, sg):
- face.smooth_group=sg
- sg.faces.append(face)
+ 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]
- while queue:
- cur=queue.pop(0)
- for e in cur.edges:
+
+ 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:
- other=e.other_face(cur)
- if other and not other.smooth_group:
- other.smooth_group=sg
- sg.faces.append(other)
+ if not other.flag:
+ other.flag=True
queue.append(other)
- def create_strip(self, face):
+ 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, reverse, debug):
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:
+ if other and not other.flag:
edge=e
break
if not edge:
return None
- if self.strip_debug:
- print "Starting strip from %s, edge %s"%([v.index for v in face.verts], (edge.v1.index, edge.v2.index))
+ if debug:
+ print "Starting strip from %s, edge %s, reverse=%s"%([v.index for v in face.verts], (edge.v1.index, edge.v2.index), reverse)
- verts=face.get_vertices_from(edge.v1, edge.v2)
+ verts=face.pivot_vertices(reverse, edge.v1, edge.v2)
if len(verts)==3:
result=[verts[-1], verts[0]]
else:
result=[verts[-2], verts[-1]]
while 1:
- verts=face.get_vertices_from(*result[-2:])
+ verts=face.pivot_vertices(reverse, *result[-2:])
k=len(result)%2
- if self.strip_debug:
- print " Adding %s"%[v.index for v in verts]
+ if debug:
+ print " Adding %s"%face
face.flag=True
if len(verts)==4 and not k:
edge=face.get_edge(*result[-2:])
- if self.strip_debug:
- print " Next edge is %s"%((edge.v1.index, edge.v2.index), )
+ if debug:
+ print " Next edge is %s"%(edge.key, )
next=edge.other_face(face)
- if not next or next.smooth_group.index!=face.smooth_group.index or next.flag:
+ if not next or next.flag:
break
face=next
- if self.strip_debug:
+ if debug:
print " %s"%[v.index for v in result]
return result
+
+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.optimize_locality=True
+ self.export_lines=True
+ self.tbn_vecs=False
+ self.debug=False
+ self.strip_debug=False
+ self.split_debug=False
+
def get_locality(self, strip):
total=0
for i in range(1, len(strip)):
total+=1.0/(abs(strip[i].index-strip[i-1].index))
return total/len(strip)
+ def get_followers(self, strip):
+ result={}
+ for i in range(len(strip)-1):
+ v=strip[i]
+ n=strip[i+1]
+ if v.index!=n.index:
+ if v.index not in result:
+ result[v.index]={}
+ if n.index not in result[v.index]:
+ result[v.index][n.index]=1
+ else:
+ result[v.index][n.index]+=1
+ return result
+
def export(self):
scene=bpy.data.scenes.active
if obj.getType()!="Mesh":
raise Exception, "Can only export Mesh data"
- mesh=obj.getData(mesh=True)
-
- faces=[Face(f) for f in mesh.faces]
+ mesh=Blender.Mesh.New("export_tmp")
+ mesh.getFromObject(obj)
+ mesh=Mesh(mesh)
- edges=dict([(e.key, Edge(e)) for e in mesh.edges])
- for f in faces:
- for e in f.edge_keys:
- edges[e].faces.append(f)
- f.edges.append(edges[e])
+ 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))
- smooth_limit=math.cos(mesh.degr*math.pi/180)
- for e in edges.itervalues():
- e.check_smooth(smooth_limit)
+ mesh.split_smooth(self.split_debug)
if self.debug:
- ntris=sum([len(f.verts)-2 for f in faces])
- print "%d faces (%d triangles), %d edges"%(len(faces), ntris, len(edges))
-
- smooth_groups=[]
- for f in faces:
- if not f.smooth_group:
- sg=SmoothGroup(len(smooth_groups))
- smooth_groups.append(sg)
- self.find_smooth_group(f, sg)
-
- for sg in smooth_groups:
- sg.find_vertices()
- if mesh.faceUV:
- sg.separate_uv()
- sg.create_edges()
-
- verts=[]
- for sg in smooth_groups:
- for v in sg.verts:
- v.index=len(verts)
- verts.append(v)
+ print "After smooth splitting %d vertices and %d edges"%(len(mesh.verts), len(mesh.edges))
- if self.debug:
- print "%d smooth groups:"%len(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))
- print "%d vertices total"%len(verts)
+ 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=[]
if self.use_strips:
+ for f in mesh.faces:
+ f.flag=False
+
while 1:
best=5
face=None
- for f in faces:
+ for f in mesh.faces:
if f.flag:
continue
score=0
for e in f.edges:
other=e.other_face(f)
- if other and other.smooth_group.index==f.smooth_group.index and not other.flag:
+ if other and not other.flag:
score+=1
if score>0 and score<best:
face=f
best=score
+
if not face:
break
- strip=self.create_strip(face)
+
+ strip=mesh.create_strip(face, self.use_degen_tris and sum([len(s) for s in strips])%2, self.strip_debug)
if strip:
strips.append(strip)
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 faces if not f.flag])
- nind=sum([len(s) for s in strips])+sum([len(f.verts) for f in faces if not f.flag])
+ 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 mesh.faces if not f.flag])
print "%d indices total"%nind
- if self.use_degen_tris:
+ if self.use_degen_tris and strips:
big_strip=[]
- while strips:
- strip=None
- if big_strip:
- for i in range(len(strips)):
- if strips[i][0].index==big_strip[-1].index:
- strip=strips.pop(i)
- break
- if not strip:
- strip=strips.pop(0)
+ for s in strips:
if big_strip:
- if strip[0].index!=big_strip[-1].index:
- big_strip+=[big_strip[-1]]
- big_strip+=[strip[0]]
- big_strip+=strip
+ big_strip+=[big_strip[-1], s[0]]
+ big_strip+=s
- for f in faces:
+ for f in mesh.faces:
if not f.flag:
+ if len(big_strip)%2:
+ order=(-1, -2, 0, 1)
+ else:
+ order=(0, 1, -1, -2)
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])
+ big_strip+=[big_strip[-1], f.verts[order[0]]]
+ big_strip+=[f.verts[i] for i in order[:len(f.verts)]]
f.flag=True
strips=[big_strip]
print "Big strip has %d indices"%len(big_strip)
if self.debug:
- print "%.2f vertices per triangle"%(float(nind)/ntris)
+ print "%.2f indices per triangle"%(float(nind)/max(ntris, 1))
print "Locality before optimization: "+" ".join(["%.3f"%self.get_locality(s) for s in strips])
- used=[False]*len(verts)
- i=0
- for s in strips:
- for v in s:
- if not used[v.index]:
- used[v.index]=True
- v.index=i
- i+=1
+ if self.optimize_locality and self.use_strips and strips:
+ followers={}
+ for s in strips:
+ followers.update(self.get_followers(s))
+
+ verts2=[]
+ vert=strips[0][0]
+ while 1:
+ vert.flag=True
+ verts2.append(vert)
+
+ next=None
+ if vert.index in followers:
+ flw=followers[vert.index]
+ best=0
+ for n in flw:
+ if flw[n]>best and not mesh.verts[n].flag:
+ next=mesh.verts[n]
+ best=flw[n]+0.9/abs(vert.index-n)
+
+ if not next:
+ for v in mesh.verts:
+ if not v.flag:
+ next=v
+ break
+ if not next:
+ break
- verts.sort(cmp=lambda x,y: cmp(x.index, y.index))
+ vert=next
- if self.debug:
- print "Locality after optimization: "+" ".join(["%.3f"%self.get_locality(s) for s in strips])
+ mesh.verts=verts2
+
+ for i in range(len(mesh.verts)):
+ mesh.verts[i].index=i
+
+ if self.debug:
+ print "Locality after optimization: "+" ".join(["%.3f"%self.get_locality(s) for s in strips])
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
- for v in verts:
+ 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))
self.out_file.write("};\n")
for s in strips:
self.out_file.write(";\n};\n")
first=True
- for f in faces:
+ for f in mesh.faces:
if not f.flag:
if first:
self.out_file.write("batch TRIANGLES\n{\n")
if not first:
self.out_file.write("};\n")
+ if self.export_lines and mesh.lines:
+ self.out_file.write("batch LINES\n{\n")
+ for l in mesh.lines:
+ self.out_file.write("\tindices %u %u;\n"%(l.verts[0].index, l.verts[1].index))
+ self.out_file.write("};\n")
+
class FrontEnd:
+ def __init__(self):
+ self.config=Blender.Registry.GetKey('mspgl_export', True) or {}
+ self.temp_config=Blender.Registry.GetKey('mspgl_export_temp') or {}
+
def run(self):
- #self.export(None)
- Blender.Window.FileSelector(self.export, "Export MSP GL mesh", Blender.sys.makename(ext='.mesh'))
+ self.use_strips=Blender.Draw.Create(self.config.get('use_strips', True))
+ self.use_degen_tris=Blender.Draw.Create(self.config.get('use_degen_tris', True))
+ self.optimize_locality=Blender.Draw.Create(self.config.get('optimize_locality', True))
+ self.export_lines=Blender.Draw.Create(self.config.get('export_lines', False))
+ self.tbn_vecs=Blender.Draw.Create(self.config.get('tbn_vecs', False))
+ self.debug=Blender.Draw.Create(self.config.get('debug', False))
+ self.strip_debug=Blender.Draw.Create(self.config.get('strip_debug', False))
+ self.split_debug=Blender.Draw.Create(self.config.get('split_debug', False))
+ ret=Blender.Draw.PupBlock("Export MSP GL mesh",
+ [("Use strips", self.use_strips, "Generage OpenGL triangle strips"),
+ ("Use degen tris", self.use_degen_tris, "Use degenerate triangles to combine triangle strips"),
+ ("Optimize locality", self.optimize_locality),
+ ("Export lines", self.export_lines, "Export lone edges as lines"),
+ ("Compute T/B vecs", self.tbn_vecs, "Compute tangent/binormal vectors for bumpmapping"),
+ ("Debugging options"),
+ ("Debug", self.debug),
+ ("Debug strips", self.strip_debug),
+ ("Debug splitting", self.split_debug)])
+ if ret:
+ dirname=self.temp_config.get("dirname", Blender.sys.dirname(Blender.Get("filename")))
+ obj=bpy.data.scenes.active.objects.active
+ Blender.Window.FileSelector(self.export, "Export MSP GL mesh", "%s/%s.mesh"%(dirname, obj.name))
+
+ def draw(self):
+ pass
def export(self, fn):
+ self.config['use_strips']=self.use_strips.val
+ self.config['use_degen_tris']=self.use_degen_tris.val
+ self.config['optimize_locality']=self.optimize_locality.val
+ self.config['export_lines']=self.export_lines.val
+ self.config['tbn_vecs']=self.tbn_vecs.val
+ self.config['debug']=self.debug.val
+ self.config['strip_debug']=self.strip_debug.val
+ self.config['split_debug']=self.split_debug.val
+ Blender.Registry.SetKey('mspgl_export', self.config, True)
+
+ import os
+ self.temp_config["dirname"]=os.path.dirname(fn)
+ Blender.Registry.SetKey('mspgl_export_temp', self.temp_config)
+
exp=Exporter(fn)
- #exp.use_degen_tris=False
- #exp.debug=True
+ exp.use_strips=self.use_strips.val
+ exp.use_degen_tris=self.use_degen_tris.val
+ exp.optimize_locality=self.optimize_locality.val
+ exp.export_lines=self.export_lines.val
+ exp.tbn_vecs=self.tbn_vecs.val
+ exp.debug=self.debug.val
+ exp.strip_debug=self.strip_debug.val
+ exp.split_debug=self.split_debug.val
exp.export()