X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=blender%2Fio_mspgl%2Fscene.py;h=7532f48ccdd82d88de86cafcef2b4fc9237d40d8;hb=7d8c187f036a7c91f9073ca0071a20807c0f1aa5;hp=079647ce00985a41252084cdfd6a3ca435b06626;hpb=5590886411b5779a7e6aa6a5ffeda8db6c515a55;p=libs%2Fgl.git

diff --git a/blender/io_mspgl/scene.py b/blender/io_mspgl/scene.py
index 079647ce..7532f48c 100644
--- a/blender/io_mspgl/scene.py
+++ b/blender/io_mspgl/scene.py
@@ -1,3 +1,5 @@
+import mathutils
+
 def is_same_object(obj1, obj2):
 	if obj1.data.name!=obj2.data.name:
 		return False
@@ -17,10 +19,30 @@ class Scene:
 	def __init__(self, scene, obj_filter=None):
 		self.name = scene.name
 		self.scene_type = scene.scene_type
+		self.export_disposition = scene.export_disposition
+		self.background_set = None
+		self.camera = scene.camera
 		self.prototypes = []
 		self.instances = []
+		self.lights = []
+		self.ambient_light = mathutils.Color((0.0, 0.0, 0.0))
+		self.exposure = scene.view_settings.exposure
+
+		self.use_hdr = False
+		if scene.world:
+			self.use_hdr = scene.world.use_hdr
+
+			out_node = next((n for n in scene.world.node_tree.nodes if n.type=='OUTPUT_WORLD'), None)
+			if out_node:
+				from .util import get_linked_node_and_socket
+
+				surface_node, _ = get_linked_node_and_socket(scene.world.node_tree, out_node.inputs["Surface"])
+				if surface_node and surface_node.type=='BACKGROUND':
+					c = surface_node.inputs["Color"].default_value
+					s = surface_node.inputs["Strength"].default_value
+					self.ambient_light = mathutils.Color(c[:3])*s
 
-		objects = [o for o in scene.objects if o.type=='MESH']
+		objects = scene.objects[:]
 		objects.sort(key=lambda o:o.name)
 		if obj_filter:
 			objects = list(filter(obj_filter, objects))
@@ -30,11 +52,21 @@ class Scene:
 			if o.name in processed:
 				continue
 
-			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)
+			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)
+			elif o.type=='LIGHT':
+				self.lights.append(o)
+
+	def get_chain(self):
+		result = []
+		if self.background_set:
+			result = self.background_set.get_chain()
+		result.append(self)
+		return result
 
 def get_all_collections(collection):
 	result = [collection]
@@ -62,3 +94,39 @@ def create_scene_from_current(context, *, selected_only=False, visible_only=True
 		obj_filter = lambda o: all(f(o) for f in obj_filters)
 
 	return Scene(context.scene, obj_filter)
+
+def create_scene(scene, *, visible_only=True):
+	obj_filter = None
+
+	if visible_only:
+		visible_names = set()
+		for c in get_all_collections(scene.collection):
+			if not c.hide_viewport:
+				visible_names.update(o.name for o in c.objects)
+		obj_filter = lambda o: o.name in visible_names
+
+	return Scene(scene, obj_filter)
+
+def create_scene_chain(scene, cache, *, visible_only=True):
+	if cache is None:
+		cache = {}
+
+	top = None
+	prev = None
+	while scene:
+		converted = None
+		if scene.name in cache:
+			converted = cache[scene.name]
+		else:
+			converted = create_scene(scene, visible_only=visible_only)
+			cache[scene.name] = converted
+
+		if not top:
+			top = converted
+		if prev:
+			prev.background_set = converted
+
+		prev = converted
+		scene = scene.background_set
+
+	return top