]> git.tdb.fi Git - libs/gl.git/blobdiff - blender/io_mspgl/export_object.py
Only use texunit zero in exporter-generated techniques
[libs/gl.git] / blender / io_mspgl / export_object.py
index b910add12fad3ed86ef397861de92fc8bb6a50b9..abe33630ce29b76b6233dd2173d42acfe95adb2a 100644 (file)
@@ -1,4 +1,5 @@
 import os
+import mathutils
 
 def linear_to_srgb(l):
        if l<0.0031308:
@@ -6,6 +7,12 @@ def linear_to_srgb(l):
        else:
                return 1.055*(l**(1/2.4))-0.055
 
+def get_colormap(srgb):
+       if srgb:
+               return linear_to_srgb
+       else:
+               return lambda x: x
+
 def image_name(i):
        fp = i.filepath
        if fp:
@@ -23,20 +30,16 @@ def external_name(out_file, ext, index):
 
 class ObjectExporter:
        def __init__(self):
-               self.material_tex = False
-               self.srgb_colors = True
                self.textures = "REF"
                self.separate_mesh = False
+               self.shared_mesh = True
                self.separate_tech = False
-               self.external_tech = True
                self.shared_tech = True
                self.export_lods = True
 
-       def export(self, context, out_file, objs=None, progress=None):
-               if objs is None:
+       def export(self, context, out_file, obj=None, progress=None):
+               if obj is None:
                        obj = context.active_object
-               else:
-                       obj = objs[0]
 
                lods = [obj]
                for c in obj.children:
@@ -48,16 +51,26 @@ class ObjectExporter:
                from .outfile import open_output
                out_file = open_output(out_file)
 
+               p1 = max(((v.co, v.co.length) for v in obj.data.vertices), key=lambda x:x[1])[0]
+               p2 = max(((v.co, (v.co-p1).length) for v in obj.data.vertices), key=lambda x:x[1])[0]
+               center = (p1+p2)/2
+               radius = (p1-p2).length/2
+               for v in obj.data.vertices:
+                       d = v.co-center
+                       if d.length>radius:
+                               center += d*(1-radius/d.length)/2
+                               radius = (radius+d.length)/2
+
+               out_file.write("bounding_sphere_hint", center[0], center[1], center[2], radius)
+
                prev_mesh = None
                prev_tech = (None, None)
                for i, l in enumerate(lods):
                        if i>0:
                                out_file.begin("level_of_detail", i)
-                               objs = [l]
 
-                       same_mesh = (l.data.name==prev_mesh)
-                       if i==0 or not same_mesh:
-                               mesh = self.export_object_mesh(context, out_file, i, objs, progress)
+                       if i==0 or l.data.name!=prev_mesh:
+                               mesh = self.export_object_mesh(context, out_file, l, progress)
                                prev_mesh = l.data.name
 
                        same_tech = True
@@ -66,7 +79,7 @@ class ObjectExporter:
                                mat = l.material_slots[0].material.name
                                if mat!=prev_tech[1]:
                                        same_tech = False
-                       if self.external_tech and l.technique!=prev_tech[0]:
+                       if l.technique!=prev_tech[0]:
                                same_tech = False
                        if i==0 or not same_tech:
                                self.export_object_technique(l, mesh, out_file, i)
@@ -75,31 +88,32 @@ class ObjectExporter:
                        if i>0:
                                out_file.end()
 
-       def export_object_mesh(self, context, out_file, lod_index, objs, progress):
+       def export_object_mesh(self, context, out_file, obj, progress):
                from .export_mesh import MeshExporter
                mesh_export = MeshExporter()
                for k, v in self.__dict__.items():
                        setattr(mesh_export, k, v)
 
+               lod_index = 0
+               if obj.lod_for_parent:
+                       lod_index = obj.lod_index
+
                if self.separate_mesh:
                        from .outfile import open_output
                        path, name = external_name(out_file, ".mesh", lod_index)
+                       if self.shared_mesh:
+                               name = obj.data.name+".mesh"
                        mesh_out = open_output(os.path.join(path, name))
-                       mesh = mesh_export.export(context, mesh_out, objs, progress)
+                       mesh = mesh_export.export(context, mesh_out, obj, progress)
                        out_file.write("mesh", '"{}"'.format(name))
                else:
                        out_file.begin("mesh")
-                       mesh = mesh_export.export(context, out_file, objs, progress)
+                       mesh = mesh_export.export(context, out_file, obj, progress)
                        out_file.end()
 
                return mesh
 
        def export_object_technique(self, obj, mesh, out_file, lod_index):
-               if self.srgb_colors:
-                       self.colormap = linear_to_srgb
-               else:
-                       self.colormap = lambda x: x
-
                material = None
                if obj.material_slots:
                        material = obj.material_slots[0].material
@@ -107,7 +121,7 @@ class ObjectExporter:
                from .outfile import open_output
                path, name = external_name(out_file, ".tech", lod_index)
 
-               if self.external_tech and obj.technique:
+               if obj.technique:
                        if obj.inherit_tech and material and (obj.override_material or material.texture_slots):
                                out_file.begin("technique")
                                out_file.begin("inherit", '"{}"'.format(obj.technique))
@@ -122,7 +136,7 @@ class ObjectExporter:
                                        mat_name = material.name+".mat"
                                        mat_out = open_output(os.path.join(path, mat_name))
                                        self.export_material(material, mat_out)
-                                       out_file.write("material", '""', '"{}"'.format(mat_name))
+                                       out_file.write("material", '"surface"', '"{}"'.format(mat_name))
                                out_file.end()
                                out_file.end()
                        else:
@@ -135,30 +149,25 @@ class ObjectExporter:
                        out_file.write("technique", '"{}"'.format(name))
                else:
                        out_file.begin("technique")
-                       self.export_technique_definition(material, mesh, out_file)
+                       self.export_technique_definition(material, obj.material_tex, mesh, out_file)
                        out_file.end()
 
-       def export_technique_definition(self, material, mesh, out_file):
+       def export_technique_definition(self, material, material_tex, mesh, out_file):
                out_file.begin("pass", '""')
                if material:
-                       cm = self.colormap
-
-                       if self.material_tex:
+                       if 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("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(cm(c)*255) for c in m.diffuse_color*mat.diffuse_intensity]
+                                       cm = get_colormap(m.srgb_colors)
+                                       color = [int(cm(c)*255) for c in m.diffuse_color*m.diffuse_intensity]
                                        texdata += "\\x%02X\\x%02X\\x%02X"%tuple(color)
                                texdata += '"'
                                out_file.write("raw_data", texdata)
@@ -170,40 +179,32 @@ class ObjectExporter:
                                out_file.end()
 
                        if self.textures!="NONE":
+                               diffuse_tex = None
                                for slot in material.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
+                                       if slot and slot.texture.type=="IMAGE" and slot.use_map_color_diffuse:
+                                               diffuse_tex = slot.texture
+                                               break
 
-                                       out_file.begin("texunit", index)
+                               if diffuse_tex:
+                                       out_file.begin("texunit", 0)
                                        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])
+                                               out_file.write("storage", "RGBA", diffuse_tex.image.size[0], diffuse_tex.image.size[1])
                                                texdata = '"'
-                                               for p in tex.image.pixels:
+                                               for p in diffuse_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"'%image_name(tex.image))
+                                               out_file.write("texture", '"%s"'%image_name(diffuse_tex.image))
                                        out_file.end()
 
                out_file.end()
 
        def export_material(self, mat, out_file):
-               cm = self.colormap
+               cm = get_colormap(mat.srgb_colors)
                if any((s and s.use_map_color_diffuse) for s in mat.texture_slots):
                        out_file.write("diffuse", 1.0, 1.0, 1.0, 1.0)
                        amb = cm(mat.ambient)
@@ -215,4 +216,4 @@ class ObjectExporter:
                        out_file.write("ambient", cm(amb.r), cm(amb.g), cm(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.write("shininess", mat.specular_hardness)