From 0506475f265d11d24dfe5e8169cff6b8eb1cb921 Mon Sep 17 00:00:00 2001 From: bosonprojets Date: Tue, 20 Apr 2021 23:51:55 +0000 Subject: [PATCH] Feature to add building elements to construction scheduling tasks in blenderBIM and assign product to process with the ifcopenshell.api --- .../bim/module/sequence/__init__.py | 1 + .../bim/module/sequence/operator.py | 21 +++++++ .../blenderbim/bim/module/sequence/prop.py | 1 + .../blenderbim/bim/module/sequence/ui.py | 5 ++ .../api/sequence/assign_product.py | 56 +++++++++++++++++++ 5 files changed, 84 insertions(+) create mode 100644 src/ifcopenshell-python/ifcopenshell/api/sequence/assign_product.py diff --git a/src/blenderbim/blenderbim/bim/module/sequence/__init__.py b/src/blenderbim/blenderbim/bim/module/sequence/__init__.py index 6ba11ff1e0..6e3658388f 100644 --- a/src/blenderbim/blenderbim/bim/module/sequence/__init__.py +++ b/src/blenderbim/blenderbim/bim/module/sequence/__init__.py @@ -39,6 +39,7 @@ classes = ( operator.EnableEditingTaskTime, operator.DisableEditingTaskTime, operator.EditTaskTime, + operator.AssignProduct, prop.WorkPlan, prop.BIMWorkPlanProperties, prop.Task, diff --git a/src/blenderbim/blenderbim/bim/module/sequence/operator.py b/src/blenderbim/blenderbim/bim/module/sequence/operator.py index 3b80089d32..0f7f714e8a 100644 --- a/src/blenderbim/blenderbim/bim/module/sequence/operator.py +++ b/src/blenderbim/blenderbim/bim/module/sequence/operator.py @@ -628,6 +628,7 @@ class EnableEditingTask(bpy.types.Operator): if data[attribute.name()]: new.enum_value = data[attribute.name()] props.active_task_id = self.task + props.should_show_times = True return {"FINISHED"} @@ -741,3 +742,23 @@ class UnassignSuccessor(bpy.types.Operator): ) Data.load(self.file) return {"FINISHED"} + + +class AssignProduct(bpy.types.Operator): + bl_idname = "bim.assign_product" + bl_label = "Assign Product" + task: bpy.props.IntProperty() + + def execute(self, context): + obj = bpy.context.active_object.BIMObjectProperties.ifc_definition_id + props = context.scene.BIMWorkScheduleProperties + self.file = IfcStore.get_file() + ifcopenshell.api.run( + "sequence.assign_product", + self.file, + relating_product = self.file.by_id(obj), + related_process = self.file.by_id(self.task), + ) + props.has_assignment = True + Data.load(self.file) + return {"FINISHED"} diff --git a/src/blenderbim/blenderbim/bim/module/sequence/prop.py b/src/blenderbim/blenderbim/bim/module/sequence/prop.py index 39075fe847..e59a744cd8 100644 --- a/src/blenderbim/blenderbim/bim/module/sequence/prop.py +++ b/src/blenderbim/blenderbim/bim/module/sequence/prop.py @@ -107,6 +107,7 @@ class Task(PropertyGroup): identification: StringProperty(name="Identification", update=updateTaskIdentification) ifc_definition_id: IntProperty(name="IFC Definition ID") has_children: BoolProperty(name="Has Children") + has_assignment: BoolProperty(name="Has Assignement") is_expanded: BoolProperty(name="Is Expanded") level_index: IntProperty(name="Level Index") duration: StringProperty(name="Duration") diff --git a/src/blenderbim/blenderbim/bim/module/sequence/ui.py b/src/blenderbim/blenderbim/bim/module/sequence/ui.py index 8e5930c40c..917076dc3b 100644 --- a/src/blenderbim/blenderbim/bim/module/sequence/ui.py +++ b/src/blenderbim/blenderbim/bim/module/sequence/ui.py @@ -293,3 +293,8 @@ class BIM_UL_tasks(UIList): row.operator("bim.enable_editing_task", text="", icon="GREASEPENCIL").task = item.ifc_definition_id row.operator("bim.add_task", text="", icon="ADD").task = item.ifc_definition_id row.operator("bim.remove_task", text="", icon="X").task = item.ifc_definition_id + if context.selected_objects: + obj = context.selected_objects[0] + pass + row = layout.row(align=True) + row.operator("bim.assign_product", text="ADD", icon="OUTLINER_COLLECTION").task = item.ifc_definition_id diff --git a/src/ifcopenshell-python/ifcopenshell/api/sequence/assign_product.py b/src/ifcopenshell-python/ifcopenshell/api/sequence/assign_product.py new file mode 100644 index 0000000000..cd4f39176c --- /dev/null +++ b/src/ifcopenshell-python/ifcopenshell/api/sequence/assign_product.py @@ -0,0 +1,56 @@ +import ifcopenshell +import ifcopenshell.api + + +class Usecase: + def __init__(self, file, **settings): + self.file = file + self.settings = { + "relating_product": None, + "related_process": None, + } + for key, value in settings.items(): + self.settings[key] = value + + def execute(self): + referenced_by = None + + if self.settings["relating_product"].ReferencedBy: + for rel in self.settings["relating_product"].ReferencedBy: + if rel.is_a("IfcRelAssignsToProduct"): + referenced_by = rel + + assignment = None + for rel in self.settings["related_process"].HasAssignments: + if rel.is_a('IfcRelAssignsToProduct'): + assignment = rel + break + if referenced_by and referenced_by == assignment: + return + + if assignment: + related_objects = set(assignment.RelatedObjects) + related_objects.add(self.settings["relating_product"]) + assignment.RelatedObjects = list(related_objects) + ifcopenshell.api.run("owner.update_owner_history", self.file, **{"element": assignment}) + else: + rel = self.file.create_entity( + "IfcRelAssignsToProduct", + **{ + "GlobalId": ifcopenshell.guid.new(), + "OwnerHistory": ifcopenshell.api.run("owner.create_owner_history", self.file), + "RelatingProduct": self.settings["relating_product"], + "RelatedObjects": [self.settings["related_process"]], + }) + return rel + + if referenced_by: + related_objects = list(referenced_by.RelatedObjects) + related_objects.remove(self.settings["relating_product"]) + if related_objects: + referenced_by.RelatedObjects = related_objects + ifcopenshell.api.run("owner.update_owner_history", self.file, **{"element": referenced_by}) + else: + self.file.remove(referenced_by) + + return rel