Further refactoring of instance handling in the Blender exporter

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

diff --git a/blender/io_mspgl/scene.py b/blender/io_mspgl/scene.py
index 70dbc8c9a7178d19470c5ba3d0ae5e1c2e142250..cb62a1b0d40401996427a15b1da11e230783c017 100644 (file)
--- a/blender/io_mspgl/scene.py
+++ b/blender/io_mspgl/scene.py
@@ -1,3 +1,5 @@
+import itertools
+import bpy
 import mathutils
 
 def is_same_object(obj1, obj2):
@@ -8,11 +10,23 @@ def is_same_object(obj1, obj2):
 
        return True
 
-class Instance:
-       def __init__(self, obj, prototype):
+class ObjectPrototype:
+       def __init__(self, obj):
                self.name = obj.name
-               self.matrix_world = obj.matrix_world
-               self.rotation_mode = obj.rotation_mode
+               self.object = obj
+               self.instances = []
+               self.use_array = False
+
+class ObjectInstance:
+       def __init__(self, obj, prototype):
+               if type(obj)==bpy.types.DepsgraphObjectInstance:
+                       self.name = None
+                       self.matrix_world = mathutils.Matrix(obj.matrix_world)
+                       self.rotation_mode = prototype.object.rotation_mode
+               else:
+                       self.name = obj.name
+                       self.matrix_world = obj.matrix_world
+                       self.rotation_mode = obj.rotation_mode
                self.prototype = prototype
 
 class Scene:
@@ -56,29 +70,45 @@ class Scene:
                if obj_filter:
                        objects = list(filter(obj_filter, objects))
 
-               processed = set()
+               proto_map = {}
                for o in objects:
-                       if o.name in processed:
-                               continue
-
                        if o.type=='MESH':
-                               clones = [c for c in objects if is_same_object(o, c)]
-                               self.prototypes.append(o)
-                               instance_list = self.instances
-                               if o.material_slots and o.material_slots[0].material:
-                                       mat = o.material_slots[0].material
-                                       if mat.blend_method=='BLEND':
-                                               instance_list = self.blended_instances
-                                       if mat.image_based_lighting:
-                                               self.use_ibl = True
-                               for c in clones:
-                                       instance_list.append(Instance(c, o))
-                                       processed.add(c.name)
+                               self.add_instance(o, o, proto_map)
                        elif o.type=='LIGHT':
                                self.lights.append(o)
                                if o.data.use_shadow:
                                        self.use_shadow = True
 
+               for i in scene.view_layers[0].depsgraph.object_instances:
+                       if i.is_instance and i.object.type=='MESH':
+                               self.add_instance(i, bpy.data.objects[i.object.name], proto_map)
+
+       def add_instance(self, obj, proto_obj, proto_map):
+               prototype = proto_map.get(proto_obj)
+               if not prototype:
+                       for p in proto_map.values():
+                               if is_same_object(proto_obj, p.object):
+                                       prototype = p
+                                       break
+
+                       if not prototype:
+                               prototype = ObjectPrototype(proto_obj)
+                               self.prototypes.append(prototype)
+
+                       proto_map[proto_obj] = prototype
+
+               instance_list = self.instances
+               if proto_obj.material_slots and proto_obj.material_slots[0].material:
+                       mat = proto_obj.material_slots[0].material
+                       if mat.blend_method=='BLEND':
+                               instance_list = self.blended_instances
+                       if mat.image_based_lighting:
+                               self.use_ibl = True
+
+               instance = ObjectInstance(obj, prototype)
+               instance_list.append(instance)
+               prototype.instances.append(instance)
+
        def get_chain(self):
                result = []
                if self.background_set: