mesh_export.py

   1 #!BPY
   2 # $Id$
   3
   4 """
   5 Name: 'MSP GL Mesh (.mesh)...'
   6 Blender: 244
   7 Group: 'Export'
   8 """
   9
  10 import sys
  11 import math
  12 import bpy
  13 import Blender
  14
  15 def make_edge_key(i1, i2):
  16         return (min(i1, i2), max(i1, i2))
  17
  18
  19 class Edge:
  20         def __init__(self, me):
  21                 if me.__class__==Edge:
  22                         self._medge=me._medge
  23                         self.v1=me.v1
  24                         self.v2=me.v2
  25                         self.smooth=me.smooth
  26                 else:
  27                         self._medge=me
  28                         self.smooth=False
  29                 self.faces=[]
  30
  31         def __getattr__(self, attr):
  32                 return getattr(self._medge, attr)
  33
  34         def __cmp__(self, other):
  35                 return self is other
  36
  37         def check_smooth(self, limit):
  38                 if len(self.faces)!=2:
  39                         return
  40
  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:
  43                         self.smooth=True
  44
  45         def other_face(self, f):
  46                 if f.index==self.faces[0].index:
  47                         if len(self.faces)>=2:
  48                                 return self.faces[1]
  49                         else:
  50                                 return None
  51                 else:
  52                         return self.faces[0]
  53
  54
  55 class Vertex:
  56         def __init__(self, mv):
  57                 if mv.__class__==Vertex:
  58                         self._mvert=mv._mvert
  59                         self.no=mv.no
  60                         self.uv=mv.uv
  61                 else:
  62                         self._mvert=mv
  63                         self.uv=None
  64                 self.orig_index=self._mvert.index
  65                 self.flag=False
  66                 self.faces=[]
  67                 self.tan=None
  68                 self.bino=None
  69
  70         def __getattr__(self, attr):
  71                 return getattr(self._mvert, attr)
  72
  73         def __cmp__(self, other):
  74                 if other is None:
  75                         return 1
  76                 return cmp(self.index, other.index)
  77
  78         def __str__(self):
  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])
  80
  81         __repr__=__str__
  82
  83
  84 class Face:
  85         def __init__(self, mf):
  86                 self._mface=mf
  87                 self.edges=[]
  88                 self.verts=[v for v in mf.verts]
  89                 self.flag=False
  90
  91         def __getattr__(self, attr):
  92                 return getattr(self._mface, attr)
  93
  94         def __cmp__(self, other):
  95                 if other is None:
  96                         return 1
  97                 return cmp(self.index, other.index)
  98
  99         def __str__(self):
 100                 return "<Face %d (%s)>"%(self.index, " ".join([str(v.index) for v in self.verts]))
 101
 102         __repr__=__str__
 103
 104         def pivot_vertices(self, reverse, *vt):
 105                 verts=self.verts[:]
 106                 if reverse:
 107                         verts.reverse()
 108                 flags=[(v in vt) for v in verts]
 109                 l=len(verts)
 110                 for i in range(l):
 111                         if flags[i] and not flags[(i+l-1)%l]:
 112                                 return verts[i:]+verts[:i]
 113
 114         def get_edge(self, v1, v2):
 115                 key=make_edge_key(v1.index, v2.index)
 116                 for e in self.edges:
 117                         if e.key==key:
 118                                 return e
 119                 raise KeyError, "No edge %s"%(key,)
 120
 121
 122 class Line:
 123         def __init__(self, e):
 124                 self.edge=e
 125                 self.verts=[e.v1, e.v2]
 126                 self.flag=False
 127
 128         def __str__(self):
 129                 return "<Line (%d %d)>"%(self.verts[0].index, self.verts[1].index)
 130
 131         __repr__=__str__
 132
 133
 134 class Mesh:
 135         def __init__(self, m):
 136                 self._mesh=m
 137                 self.verts=[Vertex(v) for v in m.verts]
 138                 self.faces=[Face(f) for f in m.faces]
 139
 140                 for f in self.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)
 144
 145                 self.edges=dict([(e.key, Edge(e)) for e in m.edges])
 146                 for f in self.faces:
 147                         for k in f.edge_keys:
 148                                 e=self.edges[k]
 149                                 e.faces.append(self.faces[f.index])
 150                                 f.edges.append(e)
 151
 152                 self.lines=[Line(e) for e in self.edges.itervalues() if not e.faces]
 153
 154                 if m.mode&Blender.Mesh.Modes.AUTOSMOOTH:
 155                         smooth_limit=math.cos(m.degr*math.pi/180)
 156                 else:
 157                         smooth_limit=-1
 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)
 162
 163         def __getattr__(self, attr):
 164                 return getattr(self._mesh, attr)
 165
 166         def split_vertices(self, find_group_func, debug):
 167                 groups=[]
 168                 for v in self.verts:
 169                         for f in v.faces:
 170                                 f.flag=False
 171
 172                         vg=[]
 173                         for f in v.faces:
 174                                 if not f.flag:
 175                                         vg.append(find_group_func(v, f))
 176
 177                         groups.append(vg)
 178
 179                 for i in range(len(self.verts)):
 180                         if len(groups[i])==1:
 181                                 continue
 182
 183                         if debug:
 184                                 print "Vertex %s has %d groups"%(self.verts[i], len(groups[i]))
 185
 186                         for g in groups[i][1:]:
 187                                 v=Vertex(self.verts[i])
 188                                 v.index=len(self.verts)
 189                                 self.verts.append(v)
 190
 191                                 if debug:
 192                                         print "  -> %d %s"%(v.index, [f.index for f in g])
 193
 194                                 for f in g:
 195                                         for j in range(len(f.edges)):
 196                                                 e=f.edges[j]
 197
 198                                                 if e.v1!=self.verts[i] and e.v2!=self.verts[i]:
 199                                                         continue
 200
 201                                                 if debug:
 202                                                         print "  Splitting edge %s with faces %s"%(e.key, e.faces)
 203
 204                                                 if e.other_face(f) not in g and len(e.faces)>=2:
 205                                                         k=e.faces.index(f)
 206                                                         e.faces.remove(f)
 207                                                         e=Edge(e)
 208                                                         f.edges[j]=e
 209                                                         e.faces.append(f)
 210                                                 else:
 211                                                         del self.edges[e.key]
 212
 213                                                 if e.v1==self.verts[i]:
 214                                                         e.v1=v
 215                                                 elif e.v2==self.verts[i]:
 216                                                         e.v2=v
 217
 218                                                 e.key=make_edge_key(e.v1.index, e.v2.index)
 219                                                 self.edges[e.key]=e
 220
 221                                         self.verts[i].faces.remove(f)
 222                                         f.verts[f.verts.index(self.verts[i])]=v
 223                                         v.faces.append(f)
 224
 225         def split_smooth(self, debug=False):
 226                 self.split_vertices(self.find_smooth_group, debug)
 227
 228         def split_uv(self, debug=False):
 229                 self.split_vertices(self.find_uv_group, debug)
 230
 231         def find_smooth_group(self, vert, face):
 232                 face.flag=True
 233                 queue=[face]
 234
 235                 for f in queue:
 236                         for e in f.edges:
 237                                 other=e.other_face(f)
 238                                 #if not other or other.index not in face_indices:
 239                                 if other not in vert.faces:
 240                                         continue
 241
 242                                 if e.smooth:
 243                                         if not other.flag:
 244                                                 other.flag=True
 245                                                 queue.append(other)
 246
 247                 return queue
 248
 249         def find_uv_group(self, vert, face):
 250                 uv=face.uv[face.verts.index(vert)]
 251                 face.flag=True
 252                 group=[face]
 253                 for f in vert.faces:
 254                         if not f.flag and f.uv[f.verts.index(vert)]==uv:
 255                                 f.flag=True
 256                                 group.append(f)
 257                 return group
 258
 259         def compute_normals(self):
 260                 for v in self.verts:
 261                         if v.faces:
 262                                 v.no=Blender.Mathutils.Vector()
 263                                 for f in v.faces:
 264                                         v.no+=f.no
 265                                 v.no.normalize()
 266                         else:
 267                                 # XXX Should use edges to compute normal
 268                                 v.no=Blender.Mathutils.Vector(0, 0, 1)
 269
 270         def compute_uv(self):
 271                 for v in self.verts:
 272                         if v.faces:
 273                                 v.uv=v.faces[0].uv[v.faces[0].verts.index(v)]
 274
 275         def compute_tbn(self):
 276                 for v in self.verts:
 277                         v.tan=Blender.Mathutils.Vector()
 278                         v.bino=Blender.Mathutils.Vector()
 279                         for f in v.faces:
 280                                 fverts=f.pivot_vertices(False, v)
 281                                 v1=fverts[1]
 282                                 v2=fverts[-1]
 283                                 du1=v1.uv[0]-v.uv[0]
 284                                 du2=v2.uv[0]-v.uv[0]
 285                                 dv1=v1.uv[1]-v.uv[1]
 286                                 dv2=v2.uv[1]-v.uv[1]
 287                                 div=du1*dv2-du2*dv1
 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
 292                         v.tan.normalize()
 293                         v.bino.normalize()
 294
 295         def create_strip(self, face, reverse, debug):
 296                 edge=None
 297                 for e in face.edges:
 298                         other=e.other_face(face)
 299                         if other and not other.flag:
 300                                 edge=e
 301                                 break
 302
 303                 if not edge:
 304                         return None
 305
 306                 if debug:
 307                         print "Starting strip from %s, edge %s, reverse=%s"%([v.index for v in face.verts], (edge.v1.index, edge.v2.index), reverse)
 308
 309                 verts=face.pivot_vertices(reverse, edge.v1, edge.v2)
 310                 if len(verts)==3:
 311                         result=[verts[-1], verts[0]]
 312                 else:
 313                         result=[verts[-2], verts[-1]]
 314
 315                 while 1:
 316                         verts=face.pivot_vertices(reverse, *result[-2:])
 317                         k=len(result)%2
 318                         if debug:
 319                                 print "  Adding %s"%face
 320
 321                         face.flag=True
 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])
 327
 328                         edge=face.get_edge(*result[-2:])
 329
 330                         if debug:
 331                                 print "  Next edge is %s"%(edge.key, )
 332
 333                         next=edge.other_face(face)
 334                         if not next or next.flag:
 335                                 break
 336                         face=next
 337
 338                 if debug:
 339                         print "  %s"%[v.index for v in result]
 340
 341                 return result
 342
 343
 344 class Exporter:
 345         def __init__(self, fn):
 346                 self.filename=fn
 347                 if fn==None:
 348                         self.out_file=sys.stdout
 349                 else:
 350                         self.out_file=file(fn, "w")
 351                 self.use_strips=True
 352                 self.use_degen_tris=True
 353                 self.optimize_locality=True
 354                 self.export_lines=True
 355                 self.tbn_vecs=False
 356                 self.debug=False
 357                 self.strip_debug=False
 358                 self.split_debug=False
 359
 360         def get_locality(self, strip):
 361                 total=0
 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)
 366
 367         def get_followers(self, strip):
 368                 result={}
 369                 for i in range(len(strip)-1):
 370                         v=strip[i]
 371                         n=strip[i+1]
 372                         if v.index!=n.index:
 373                                 if v.index not in result:
 374                                         result[v.index]={}
 375                                 if n.index not in result[v.index]:
 376                                         result[v.index][n.index]=1
 377                                 else:
 378                                         result[v.index][n.index]+=1
 379                 return result
 380
 381         def export(self):
 382                 scene=bpy.data.scenes.active
 383
 384                 obj=scene.objects.active
 385                 if obj.getType()!="Mesh":
 386                         raise Exception, "Can only export Mesh data"
 387
 388                 mesh=Blender.Mesh.New("export_tmp")
 389                 mesh.getFromObject(obj)
 390                 mesh=Mesh(mesh)
 391
 392                 if self.debug:
 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))
 395
 396                 mesh.split_smooth(self.split_debug)
 397
 398                 if self.debug:
 399                         print "After smooth splitting %d vertices and %d edges"%(len(mesh.verts), len(mesh.edges))
 400
 401                 mesh.compute_normals()
 402
 403                 if mesh.faceUV:
 404                         mesh.split_uv(self.split_debug)
 405                         if self.debug:
 406                                 print "After UV splitting %d vertices and %d edges"%(len(mesh.verts), len(mesh.edges))
 407
 408                         mesh.compute_uv()
 409                         if self.tbn_vecs:
 410                                 mesh.compute_tbn()
 411
 412                 strips=[]
 413                 if self.use_strips:
 414                         for f in mesh.faces:
 415                                 f.flag=False
 416
 417                         while 1:
 418                                 best=5
 419                                 face=None
 420                                 for f in mesh.faces:
 421                                         if f.flag:
 422                                                 continue
 423                                         score=0
 424                                         for e in f.edges:
 425                                                 other=e.other_face(f)
 426                                                 if other and not other.flag:
 427                                                         score+=1
 428                                         if score>0 and score<best:
 429                                                 face=f
 430                                                 best=score
 431
 432                                 if not face:
 433                                         break
 434
 435                                 strip=mesh.create_strip(face, self.use_degen_tris and sum([len(s) for s in strips])%2, self.strip_debug)
 436                                 if strip:
 437                                         strips.append(strip)
 438
 439                         if self.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
 446
 447                         if self.use_degen_tris and strips:
 448                                 big_strip=[]
 449                                 for s in strips:
 450                                         if big_strip:
 451                                                 big_strip+=[big_strip[-1], s[0]]
 452                                         big_strip+=s
 453
 454                                 for f in mesh.faces:
 455                                         if not f.flag:
 456                                                 if len(big_strip)%2:
 457                                                         order=(-1, -2, 0, 1)
 458                                                 else:
 459                                                         order=(0, 1, -1, -2)
 460                                                 if big_strip:
 461                                                         big_strip+=[big_strip[-1], f.verts[order[0]]]
 462                                                 big_strip+=[f.verts[i] for i in order[:len(f.verts)]]
 463                                                 f.flag=True
 464
 465                                 strips=[big_strip]
 466
 467                                 if self.debug:
 468                                         nind=len(big_strip)
 469                                         print "Big strip has %d indices"%len(big_strip)
 470
 471                 if self.debug:
 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])
 474
 475                 if self.optimize_locality and self.use_strips and strips:
 476                         followers={}
 477                         for s in strips:
 478                                 followers.update(self.get_followers(s))
 479
 480                         verts2=[]
 481                         vert=strips[0][0]
 482                         while 1:
 483                                 vert.flag=True
 484                                 verts2.append(vert)
 485
 486                                 next=None
 487                                 if vert.index in followers:
 488                                         flw=followers[vert.index]
 489                                         best=0
 490                                         for n in flw:
 491                                                 if flw[n]>best and not mesh.verts[n].flag:
 492                                                         next=mesh.verts[n]
 493                                                         best=flw[n]+0.9/abs(vert.index-n)
 494
 495                                 if not next:
 496                                         for v in mesh.verts:
 497                                                 if not v.flag:
 498                                                         next=v
 499                                                         break
 500                                         if not next:
 501                                                 break
 502
 503                                 vert=next
 504
 505                         mesh.verts=verts2
 506
 507                         for i in range(len(mesh.verts)):
 508                                 mesh.verts[i].index=i
 509
 510                         if self.debug:
 511                                 print "Locality after optimization: "+" ".join(["%.3f"%self.get_locality(s) for s in strips])
 512
 513                 self.out_file.write("vertices NORMAL3")
 514                 if mesh.faceUV:
 515                         self.out_file.write("_TEXCOORD2")
 516                         if self.tbn_vecs:
 517                                 self.out_file.write("_ATTRIB33_ATTRIB34")
 518                 self.out_file.write("_VERTEX3\n{\n")
 519                 norm=None
 520                 uv=None
 521                 tan=None
 522                 bino=None
 523                 for v in mesh.verts:
 524                         if v.no!=norm:
 525                                 self.out_file.write("\tnormal3 %f %f %f;\n"%tuple(v.no))
 526                                 norm=v.no
 527                         if v.uv!=uv:
 528                                 self.out_file.write("\ttexcoord2 %f %f;\n"%tuple(v.uv))
 529                                 uv=v.uv
 530                         if v.tan!=tan:
 531                                 self.out_file.write("\tattrib3 3 %f %f %f;\n"%tuple(v.tan))
 532                                 tan=v.tan
 533                         if v.bino!=bino:
 534                                 self.out_file.write("\tattrib3 4 %f %f %f;\n"%tuple(v.bino))
 535                                 bino=v.bino
 536                         self.out_file.write("\tvertex3 %f %f %f;\n"%tuple(v.co))
 537                 self.out_file.write("};\n")
 538                 for s in strips:
 539                         self.out_file.write("batch TRIANGLE_STRIP\n{\n\tindices")
 540                         n=0
 541                         for v in s:
 542                                 self.out_file.write(" %u"%v.index)
 543                                 n+=1;
 544                                 if n%32==0:
 545                                         self.out_file.write(";\n\tindices")
 546                         self.out_file.write(";\n};\n")
 547
 548                 first=True
 549                 for f in mesh.faces:
 550                         if not f.flag:
 551                                 if first:
 552                                         self.out_file.write("batch TRIANGLES\n{\n")
 553                                         first=False
 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))
 556                 if not first:
 557                         self.out_file.write("};\n")
 558
 559                 if self.export_lines and mesh.lines:
 560                         self.out_file.write("batch LINES\n{\n")
 561                         for l in mesh.lines:
 562                                 self.out_file.write("\tindices %u %u;\n"%(l.verts[0].index, l.verts[1].index))
 563                         self.out_file.write("};\n")
 564
 565
 566 class FrontEnd:
 567         def __init__(self):
 568                 self.config=Blender.Registry.GetKey('mspgl_export', True) or {}
 569                 self.temp_config=Blender.Registry.GetKey('mspgl_export_temp') or {}
 570
 571         def run(self):
 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)])
 590                 if ret:
 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))
 594
 595         def draw(self):
 596                 pass
 597
 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)
 608
 609                 import os
 610                 self.temp_config["dirname"]=os.path.dirname(fn)
 611                 Blender.Registry.SetKey('mspgl_export_temp', self.temp_config)
 612
 613                 exp=Exporter(fn)
 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
 622                 exp.export()
 623
 624
 625 if __name__=="__main__":
 626         fe=FrontEnd()
 627         fe.run()