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 4cb6be5baf63a45d521511a42d28d86de0c86c43..cb62a1b0d40401996427a15b1da11e230783c017 100644 (file)
--- a/blender/io_mspgl/scene.py
+++ b/blender/io_mspgl/scene.py
@@ -10,18 +10,24 @@ def is_same_object(obj1, obj2):
 
        return True
 
-class Instance:
+class ObjectPrototype:
+       def __init__(self, obj):
+               self.name = obj.name
+               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.rotation_mode
-                       self.prototype = bpy.data.objects[prototype.name]
+                       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
+               self.prototype = prototype
 
 class Scene:
        def __init__(self, scene, obj_filter=None):
@@ -64,9 +70,10 @@ class Scene:
                if obj_filter:
                        objects = list(filter(obj_filter, objects))
 
+               proto_map = {}
                for o in objects:
                        if o.type=='MESH':
-                               self.add_instance(Instance(o, o))
+                               self.add_instance(o, o, proto_map)
                        elif o.type=='LIGHT':
                                self.lights.append(o)
                                if o.data.use_shadow:
@@ -74,36 +81,33 @@ class Scene:
 
                for i in scene.view_layers[0].depsgraph.object_instances:
                        if i.is_instance and i.object.type=='MESH':
-                               self.add_instance(Instance(i, i.object))
+                               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
 
-               proto_map = {}
-               for i in itertools.chain(self.instances, self.blended_instances):
-                       p = proto_map.get(i.prototype)
-                       if p:
-                               i.prototype = p
-                       else:
-                               found = False
-                               for p in proto_map.values():
-                                       if is_same_object(i.prototype, p):
-                                               proto_map[i.prototype] = p
-                                               i.prototype = p
-                                               found = True
-                                               break
-
-                               if not found:
-                                       proto_map[i.prototype] = i.prototype
-                                       self.prototypes.append(i.prototype)
-
-       def add_instance(self, instance):
-               obj = instance.prototype
                instance_list = self.instances
-               if obj.material_slots and obj.material_slots[0].material:
-                       mat = obj.material_slots[0].material
+               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 = []