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

diff --git a/blender/io_mspgl/scene.py b/blender/io_mspgl/scene.py
new file mode 100644
index 00000000..079647ce
--- /dev/null
+++ b/blender/io_mspgl/scene.py
@@ -0,0 +1,64 @@
+def is_same_object(obj1, obj2):
+	if obj1.data.name!=obj2.data.name:
+		return False
+	if any(m1.name!=m2.name for m1, m2 in zip(obj1.material_slots, obj2.material_slots)):
+		return False
+
+	return True
+
+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
+
+class Scene:
+	def __init__(self, scene, obj_filter=None):
+		self.name = scene.name
+		self.scene_type = scene.scene_type
+		self.prototypes = []
+		self.instances = []
+
+		objects = [o for o in scene.objects if o.type=='MESH']
+		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
+
+			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)
+
+def get_all_collections(collection):
+	result = [collection]
+	for c in collection.children:
+		result += get_all_collections(c)
+	return result
+
+def create_scene_from_current(context, *, selected_only=False, visible_only=True):
+	obj_filters = []
+
+	if selected_only:
+		obj_filters.append(lambda o: o.select_get())
+
+	if visible_only:
+		visible_names = set()
+		for c in get_all_collections(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)
+
+	obj_filter = None
+	if len(obj_filters)==1:
+		obj_filter = obj_filters[0]
+	if obj_filters:
+		obj_filter = lambda o: all(f(o) for f in obj_filters)
+
+	return Scene(context.scene, obj_filter)