Export the entire scene by default

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

diff --git a/blender/io_mspgl/export_scene.py b/blender/io_mspgl/export_scene.py
index c66feed7aadf9c1218c379a51b94d92fb74e119e..b28c8e041db4c66d2d43b1dc59d95c1c77e76797 100644 (file)
--- a/blender/io_mspgl/export_scene.py
+++ b/blender/io_mspgl/export_scene.py
@@ -3,11 +3,20 @@ import os
 
 class SceneExporter:
        def __init__(self):
+               self.selected_only = False
+               self.active_layers = True
                self.resource_collection = True
                self.show_progress = True
 
        def export_to_file(self, context, out_fn):
-               objs = [o for o in context.selected_objects if o.type=="MESH" and not o.lod_for_parent]
+               if self.selected_only:
+                       objs = context.selected_objects
+               else:
+                       objs = context.scene.objects
+               if self.active_layers:
+                       layers = context.scene.layers
+                       objs = [o for o in objs if any(a and b for a, b in zip(layers, o.layers))]
+               objs = [o for o in objs if o.type=="MESH" and not o.lod_for_parent]
                objs = [o for o in objs if (not o.compound or o.parent not in objs)]
 
                path, base = os.path.split(out_fn)
@@ -30,8 +39,6 @@ class SceneExporter:
                                                continue
                                        if u.data.name!=o.data.name:
                                                continue
-                                       if u.technique!=o.technique:
-                                               continue
                                        if any(s.link=="OBJECT" for s in u.material_slots):
                                                continue
 
@@ -64,8 +71,8 @@ class SceneExporter:
                scene_res = self.export_scene(context, objs, progress, prototypes=object_prototypes, resources=resources)
                refs = scene_res.collect_references()
 
-               from .datafile import Statement
                if self.resource_collection:
+                       from .datafile import Statement
                        keywords = { ".mat": "material",
                                ".mesh": "mesh",
                                ".object": "object",
@@ -110,21 +117,22 @@ class SceneExporter:
                for o in objs:
                        obj_res = resources[prototypes[o.name].name+".object"]
                        st = scene_res.create_reference_statement("object", obj_res, o.name)
-                       # XXX Parent relationships screw up the location and rotation
+
                        ss = Statement("transform")
-                       ss.sub.append(Statement("position", o.location[0], o.location[1], o.location[2]))
-                       if o.rotation_mode=="AXIS_ANGLE":
-                               angle = o.rotation_axis_angle[0]
-                               axis = o.rotation_axis_angle[1:]
+
+                       loc = o.matrix_world.to_translation()
+                       ss.sub.append(Statement("position", *tuple(loc)))
+
+                       quat = o.matrix_world.to_quaternion()
+                       if o.rotation_mode in ('XYZ', 'XZY', 'YXZ', 'YZX', 'ZXY', 'ZYX'):
+                               angles = [a*180/math.pi for a in quat.to_euler()]
+                               ss.sub.append(Statement("euler", *angles));
                        else:
-                               if o.rotation_mode=="QUATERNION":
-                                       q = o.rotation_quaternion
-                               else:
-                                       q = o.rotation_euler.to_quaternion()
-                               angle = q.angle
-                               axis = q.axis
-                       ss.sub.append(Statement("rotation", angle*180/math.pi, axis[0], axis[1], axis[2]))
-                       ss.sub.append(Statement("scale", o.scale[0], o.scale[1], o.scale[2]))
+                               ss.sub.append(Statement("rotation", quat.angle*180/math.pi, *tuple(quat.axis)))
+
+                       scale = o.matrix_world.to_scale()
+                       ss.sub.append(Statement("scale", *tuple(scale)))
+
                        st.sub.append(ss)
                        scene_res.statements.append(st)