From 6ec368901bf573f3641331c2b5403c45e73b314b Mon Sep 17 00:00:00 2001 From: Dion Moult Date: Sat, 22 Aug 2020 19:34:52 +1000 Subject: [PATCH] Types are now categorised in the project upon assignment --- src/ifcblenderexport/blenderbim/bim/operator.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/src/ifcblenderexport/blenderbim/bim/operator.py b/src/ifcblenderexport/blenderbim/bim/operator.py index 54bb5c95f5..c14873ae6c 100644 --- a/src/ifcblenderexport/blenderbim/bim/operator.py +++ b/src/ifcblenderexport/blenderbim/bim/operator.py @@ -207,6 +207,17 @@ class AssignClass(bpy.types.Operator): object_type = obj.BIMObjectProperties.attributes.add() object_type.name = 'ObjectType' object_type.string_value = bpy.context.scene.BIMProperties.ifc_userdefined_type + if bpy.context.scene.BIMProperties.ifc_product == 'IfcElementType': + for project in [c for c in bpy.context.view_layer.layer_collection.children if 'IfcProject' in c.name]: + if not [c for c in project.children if 'Types' in c.name]: + types = bpy.data.collections.new('Types') + project.collection.children.link(types) + for collection in [c for c in project.children if 'Types' in c.name]: + for user_collection in obj.users_collection: + user_collection.objects.unlink(obj) + collection.collection.objects.link(obj) + break + break return {'FINISHED'}