]> git.tdb.fi Git - libs/gl.git/blobdiff - blender/io_mesh_mspgl/export_mspgl.py
An extra sanity check for TBN export
[libs/gl.git] / blender / io_mesh_mspgl / export_mspgl.py
index 5a1bbe1130be97b39acaf49c067679da4798f070..c776c54fea3fddec0051941e7230884abc60d695 100644 (file)
@@ -75,6 +75,7 @@ class Exporter:
                self.compound = False
                self.object = False
                self.material_tex = False
+               self.textures = "REF"
                self.smoothing = "MSPGL"
 
        def stripify(self, mesh, progress = None):
@@ -93,6 +94,7 @@ class Exporter:
                island_strips = []
                while 1:
                        if not island:
+                               # No current island; find any unused face to start from
                                queue = []
                                for f in mesh.faces:
                                        if not f.flag:
@@ -103,38 +105,46 @@ class Exporter:
                                if not queue:
                                        break
 
+                               # Find all faces connected to the first one
                                while queue:
-                                       f = queue[0]
-                                       del queue[0]
-                                       island.append(f)
+                                       face = queue.pop(0)
+                                       island.append(face)
 
-                                       for e in f.edges:
-                                               other = e.other_face(f)
-                                               if other and not other.flag:
-                                                       other.flag = True
-                                                       queue.append(other)
+                                       for n in f.get_neighbors():
+                                               n.flag = True
+                                               queue.append(n)
 
+                               # Unflag the island for the next phase
                                for f in island:
                                        f.flag = False
 
+                       # Find an unused face with as few unused neighbors as possible, but
+                       # at least one.  This heuristic gives a preference to faces in corners
+                       # or along borders of a non-closed island.
                        best = 5
                        face = None
                        for f in island:
                                if f.flag:
                                        continue
-                               score = 0
-                               for e in f.edges:
-                                       other = e.other_face(f)
-                                       if other and not other.flag:
-                                               score += 1
+
+                               score = sum(n.flag for n in f.get_neighbors())
                                if score>0 and score<best:
                                        face = f
                                        best = score
 
-                       if not face:
+                       if face:
+                               # Create a strip starting from the face.  This will flag the faces.
+                               strip = mesh.create_strip(face, self.max_strip_len)
+                               if strip:
+                                       island_strips.append(strip)
+                       else:
+                               # Couldn't find a candidate face for starting a strip, so we're
+                               # done with this island
                                while island_strips:
                                        best = 0
                                        if cache:
+                                               # Find the strip that benefits the most from the current
+                                               # contents of the vertex cache
                                                best_hits = 0
                                                for i in range(len(island_strips)):
                                                        hits = cache.test_strip(island_strips[i])
@@ -142,28 +152,23 @@ class Exporter:
                                                                best = i
                                                                best_hits = hits
 
-                                       s = island_strips[best]
-                                       del island_strips[best]
-                                       strips.append(s)
+                                       strip = island_strips.pop(best)
+                                       strips.append(strip)
 
                                        if cache:
-                                               cache.fetch_strip(s)
+                                               cache.fetch_strip(strip)
 
                                faces_done += len(island)
                                if progress:
                                        progress.set_progress(float(faces_done)/len(mesh.faces))
 
+                               # Collect any faces that weren't used in strips
                                loose += [f for f in island if not f.flag]
                                for f in island:
                                        f.flag = True
 
                                island = []
                                island_strips = []
-                               continue
-
-                       strip = mesh.create_strip(face, self.max_strip_len)
-                       if strip:
-                               island_strips.append(strip)
 
                if cache:
                        cache = VertexCache(self.cache_size)
@@ -174,6 +179,8 @@ class Exporter:
 
                        for s in strips:
                                if big_strip:
+                                       # Generate glue elements, ensuring that the next strip begins at
+                                       # an even position
                                        glue = [big_strip[-1], s[0]]
                                        if len(big_strip)%2:
                                                glue += [s[0]]
@@ -187,6 +194,8 @@ class Exporter:
                                        total_hits += cache.fetch_strip(s)
 
                        for f in loose:
+                               # Add loose faces to the end.  This wastes space, using five
+                               # elements per triangle and six elements per quad.
                                if len(big_strip)%2:
                                        order = (-1, -2, 0, 1)
                                else:
@@ -244,28 +253,33 @@ class Exporter:
                if self.smoothing!="BLENDER":
                        mesh.compute_normals()
 
-               if self.material_tex:
+               if self.material_tex and mesh.materials:
                        mesh.generate_material_uv()
 
                texunits = []
                if mesh.uv_layers and self.export_uv!="NONE":
+                       # Figure out which UV layers to export
                        if self.export_uv=="UNIT0":
-                               texunits = [0]
+                               if mesh.uv_layers[0].unit==0:
+                                       texunits = [0]
                        else:
-                               texunits = list(range(len(mesh.uv_layers)))
+                               texunits = range(len(mesh.uv_layers))
+                       texunits = [(i, mesh.uv_layers[i]) for i in texunits]
+                       texunits = [u for u in texunits if not u[1].hidden]
 
-                       tbn_unit = 0
                        if self.tbn_vecs:
-                               uvtex_names = [u.name for u in mesh.uv_layers]
-                               if self.tbn_uvtex in uvtex_names:
-                                       tbn_unit = uvtex_names.index(uvtex)
-                                       del texunits[tbn_unit]
-                                       texunits.insert(0, tbn_unit)
-
-                       for i in texunits:
+                               # TBN coordinates must be generated before vertices are split by any other layer
+                               uv_names = [u.name for i, u in texunits]
+                               if self.tbn_uvtex in uv_names:
+                                       tbn_index = uv_names.index(self.tbn_uvtex)
+                                       unit = texunits[tbn_index]
+                                       del texunits[tbn_index]
+                                       texunits.insert(0, unit)
+
+                       for i, u in texunits:
                                progress.set_task("Splitting UVs", 0.35+0.3*i/len(texunits), 0.35+0.3*(i+1)/len(texunits))
                                mesh.split_uv(i, progress)
-                               if self.tbn_vecs and i==tbn_unit:
+                               if self.tbn_vecs and u.name==self.tbn_uvtex:
                                        mesh.compute_uv()
                                        mesh.compute_tbn(i)
 
@@ -285,34 +299,37 @@ class Exporter:
 
                fmt = "NORMAL3"
                if texunits:
-                       fmt += "_TEXCOORD2"
-                       for i in texunits[1:]:
-                               fmt += "_TEXCOORD2%d"%i
+                       for i, u in texunits:
+                               if u.unit==0:
+                                       fmt += "_TEXCOORD2"
+                               else:
+                                       fmt += "_TEXCOORD2%d"%u.unit
                        if self.tbn_vecs:
                                fmt += "_ATTRIB33_ATTRIB34"
                fmt += "_VERTEX3"
                out_file.begin("vertices", fmt)
                normal = None
-               uvs = [None]*len(mesh.uv_layers)
+               uvs = [None]*len(texunits)
                tan = None
                bino = None
                for v in mesh.vertices:
                        if v.normal!=normal:
                                out_file.write("normal3", *v.normal)
                                normal = v.normal
-                       for i in texunits:
+                       for i, u in texunits:
                                if v.uvs[i]!=uvs[i]:
-                                       if i==0:
+                                       if u.unit==0:
                                                out_file.write("texcoord2", *v.uvs[i])
                                        else:
-                                               out_file.write("multitexcoord2", i, *v.uvs[i])
+                                               out_file.write("multitexcoord2", u.unit, *v.uvs[i])
                                        uvs[i] = v.uvs[i]
-                       if v.tan!=tan:
-                               out_file.write("attrib3", 3, *v.tan)
-                               tan = v.tan
-                       if v.bino!=bino:
-                               out_file.write("attrib3", 4, *v.bino)
-                               bino = v.bino
+                       if self.tbn_vecs:
+                               if v.tan!=tan:
+                                       out_file.write("attrib3", 3, *v.tan)
+                                       tan = v.tan
+                               if v.bino!=bino:
+                                       out_file.write("attrib3", 4, *v.bino)
+                                       bino = v.bino
                        out_file.write("vertex3", *v.co)
                out_file.end()
                for s in strips:
@@ -345,34 +362,71 @@ class Exporter:
                        out_file.end()
                        out_file.begin("technique")
                        out_file.begin("pass", '""')
-                       if self.material_tex:
-                               out_file.begin("material")
-                               out_file.write("diffuse", 1.0, 1.0, 1.0, 1.0)
-                               out_file.end()
-                               out_file.begin("texunit", 0)
-                               out_file.begin("texture2d")
-                               out_file.write("min_filter", "NEAREST")
-                               out_file.write("mag_filter", "NEAREST")
-                               out_file.write("storage", "RGB", len(mesh.materials), 1)
-                               texdata = '"'
-                               for m in mesh.materials:
-                                       color = [int(c*255) for c in m.diffuse_color]
-                                       texdata += "\\x%02X\\x%02X\\x%02X"%tuple(color)
-                               texdata += '"'
-                               out_file.write("raw_data", texdata)
-                               out_file.end()
-                               out_file.end()
-                       elif mesh.materials:
-                               mat = mesh.materials[0]
-                               out_file.begin("material")
-                               diff = mat.diffuse_color
-                               out_file.write("diffuse", diff.r, diff.g, diff.b, 1.0)
-                               amb = diff*mat.ambient
-                               out_file.write("ambient", amb.r, amb.g, amb.b, 1.0)
-                               spec = mat.specular_color*mat.specular_intensity
-                               out_file.write("specular", spec.r, spec.g, spec.b, 1.0)
-                               out_file.write("shininess", mat.specular_hardness);
-                               out_file.end()
+                       if mesh.materials:
+                               if self.material_tex:
+                                       out_file.begin("material")
+                                       out_file.write("diffuse", 1.0, 1.0, 1.0, 1.0)
+                                       out_file.end()
+                                       index = 0
+                                       for u in mesh.uv_layers:
+                                               if u.name=="material_tex":
+                                                       index = u.unit
+                                       out_file.begin("texunit", index)
+                                       out_file.begin("texture2d")
+                                       out_file.write("min_filter", "NEAREST")
+                                       out_file.write("mag_filter", "NEAREST")
+                                       out_file.write("storage", "RGB", len(mesh.materials), 1)
+                                       texdata = '"'
+                                       for m in mesh.materials:
+                                               color = [int(c*255) for c in m.diffuse_color]
+                                               texdata += "\\x%02X\\x%02X\\x%02X"%tuple(color)
+                                       texdata += '"'
+                                       out_file.write("raw_data", texdata)
+                                       out_file.end()
+                                       out_file.end()
+                               else:
+                                       mat = mesh.materials[0]
+                                       out_file.begin("material")
+                                       diff = mat.diffuse_color
+                                       out_file.write("diffuse", diff.r, diff.g, diff.b, 1.0)
+                                       amb = diff*mat.ambient
+                                       out_file.write("ambient", amb.r, amb.g, amb.b, 1.0)
+                                       spec = mat.specular_color*mat.specular_intensity
+                                       out_file.write("specular", spec.r, spec.g, spec.b, 1.0)
+                                       out_file.write("shininess", mat.specular_hardness);
+                                       out_file.end()
+
+                               if self.textures!="NONE":
+                                       for slot in mesh.materials[0].texture_slots:
+                                               if not slot:
+                                                       continue
+
+                                               tex = slot.texture
+                                               if tex.type!="IMAGE":
+                                                       continue
+
+                                               if slot.uv_layer:
+                                                       for u in mesh.uv_layers:
+                                                               if u.name==slot.uv_layer:
+                                                                       index = u.unit
+                                               else:
+                                                       index = mesh.uv_layers[0].unit
+
+                                               out_file.begin("texunit", index)
+                                               if self.textures=="INLINE":
+                                                       out_file.begin("texture2d")
+                                                       out_file.write("min_filter", "LINEAR")
+                                                       out_file.write("storage", "RGBA", tex.image.size[0], tex.image.size[1])
+                                                       texdata = '"'
+                                                       for p in tex.image.pixels:
+                                                               texdata += "\\x%02X"%int(p*255)
+                                                       texdata += '"'
+                                                       out_file.write("raw_data", texdata)
+                                                       out_file.end()
+                                               else:
+                                                       out_file.write("texture", '"%s"'%tex.image.name)
+                                               out_file.end()
+
                        out_file.end()
                        out_file.end()