Move most properties from exporters to the relevant types

[libs/gl.git] / blender / io_mspgl / export_object.py

diff --git a/blender/io_mspgl/export_object.py b/blender/io_mspgl/export_object.py
index b910add12fad3ed86ef397861de92fc8bb6a50b9..9f1c1e581c286371c44b60918cb5f73d0e58c1c0 100644 (file)
--- a/blender/io_mspgl/export_object.py
+++ b/blender/io_mspgl/export_object.py
@@ -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,10 +30,9 @@ 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
@@ -48,6 +54,18 @@ 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):
@@ -55,9 +73,8 @@ class ObjectExporter:
                                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, objs, progress)
                                prev_mesh = l.data.name
 
                        same_tech = True
@@ -75,15 +92,21 @@ 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, lod, objs, 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 lod.lod_for_parent:
+                       lod_index = lod.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 = lod.data.name+".mesh"
                        mesh_out = open_output(os.path.join(path, name))
                        mesh = mesh_export.export(context, mesh_out, objs, progress)
                        out_file.write("mesh", '"{}"'.format(name))
@@ -95,11 +118,6 @@ class ObjectExporter:
                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
@@ -122,7 +140,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,15 +153,13 @@ 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()
@@ -158,7 +174,8 @@ class ObjectExporter:
                                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)
@@ -203,7 +220,7 @@ class ObjectExporter:
                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)