X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=blender%2Fio_mspgl%2Fscene.py;h=4cb6be5baf63a45d521511a42d28d86de0c86c43;hb=ccc48ed8c2ab0c934e820131746442462ddd8f93;hp=28c30ad78b58a452309f9f24552edb6b56647ac2;hpb=c17add060e6db20bb4de3965801565f2f5f0c275;p=libs%2Fgl.git

diff --git a/blender/io_mspgl/scene.py b/blender/io_mspgl/scene.py
index 28c30ad7..4cb6be5b 100644
--- 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):
@@ -10,10 +12,16 @@ def is_same_object(obj1, obj2):
 
 class Instance:
 	def __init__(self, obj, prototype):
-		self.name = obj.name
-		self.matrix_world = obj.matrix_world
-		self.rotation_mode = obj.rotation_mode
-		self.prototype = prototype.name
+		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]
+		else:
+			self.name = obj.name
+			self.matrix_world = obj.matrix_world
+			self.rotation_mode = obj.rotation_mode
+			self.prototype = prototype
 
 class Scene:
 	def __init__(self, scene, obj_filter=None):
@@ -23,7 +31,10 @@ class Scene:
 		self.camera = scene.camera
 		self.prototypes = []
 		self.instances = []
+		self.blended_instances = []
 		self.lights = []
+		self.realtime_sky = False
+		self.sun_light = None
 		self.ambient_light = mathutils.Color((0.0, 0.0, 0.0))
 		self.exposure = scene.view_settings.exposure
 
@@ -42,24 +53,57 @@ class Scene:
 					s = surface_node.inputs["Strength"].default_value
 					self.ambient_light = mathutils.Color(c[:3])*s
 
+			self.use_sky = scene.world.use_sky and scene.world.sun_light
+			self.sun_light = scene.world.sun_light
+
+		self.use_shadow = False
+		self.use_ibl = False
+
 		objects = scene.objects[:]
 		objects.sort(key=lambda o:o.name)
 		if obj_filter:
 			objects = list(filter(obj_filter, objects))
 
-		processed = set()
 		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)
-				for c in clones:
-					self.instances.append(Instance(c, o))
-					processed.add(c.name)
+				self.add_instance(Instance(o, o))
 			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(Instance(i, i.object))
+
+		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 mat.blend_method=='BLEND':
+				instance_list = self.blended_instances
+			if mat.image_based_lighting:
+				self.use_ibl = True
+		instance_list.append(instance)
 
 	def get_chain(self):
 		result = []
@@ -74,7 +118,7 @@ def get_all_collections(collection):
 		result += get_all_collections(c)
 	return result
 
-def create_scene_from_current(context, *, selected_only=False, visible_only=True):
+def create_scene_from_current(ctx, *, selected_only=False, visible_only=True):
 	obj_filters = []
 
 	if selected_only:
@@ -82,7 +126,7 @@ def create_scene_from_current(context, *, selected_only=False, visible_only=True
 
 	if visible_only:
 		visible_names = set()
-		for c in get_all_collections(context.view_layer.layer_collection):
+		for c in get_all_collections(ctx.context.view_layer.layer_collection):
 			if not c.hide_viewport and not c.collection.hide_viewport:
 				visible_names.update(o.name for o in c.collection.objects)
 		obj_filters.append(lambda o: o.name in visible_names)
@@ -93,7 +137,7 @@ def create_scene_from_current(context, *, selected_only=False, visible_only=True
 	if obj_filters:
 		obj_filter = lambda o: all(f(o) for f in obj_filters)
 
-	return Scene(context.scene, obj_filter)
+	return Scene(ctx.context.scene, obj_filter)
 
 def create_scene(scene, *, visible_only=True):
 	obj_filter = None