Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-09-20 15:08:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

added the option to refresh different groups of linked aggregate

Browse Source
This commit is contained in:
Bruno Perdigão
2023-11-23 10:27:54 -03:00
parent f859c06858
commit 2fb8fafbb4
2 changed files with 77 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/blenderbim/blenderbim/bim/module/geometry/operator.py
+77 -60
View File
@@ -927,6 +927,7 @@ class DuplicateMoveLinkedAggregate(bpy.types.Operator):
def _execute(self, context): def _execute(self, context):
return DuplicateMoveLinkedAggregate.execute_ifc_duplicate_linked_aggregate_operator(self, context) return DuplicateMoveLinkedAggregate.execute_ifc_duplicate_linked_aggregate_operator(self, context)
@staticmethod
def execute_ifc_duplicate_linked_aggregate_operator(self, context): def execute_ifc_duplicate_linked_aggregate_operator(self, context):
self.new_active_obj = None self.new_active_obj = None
self.group_name = "BBIM_Linked_Aggregate" self.group_name = "BBIM_Linked_Aggregate"
@@ -1063,80 +1064,96 @@ class RefreshLinkedAggregate(bpy.types.Operator):
return element return element
else: else:
return None return None
def handle_selection(selected_objs):
selected_elements = [tool.Ifc.get_entity(selected_obj) for selected_obj in selected_objs]
selected_parents = []
for selected_element in selected_elements:
selected_element = get_element_assembly(selected_element)
if not selected_element:
self.report({"INFO"}, "Object is not part of a IfcElementAssembly.")
return {'FINISHED'}
selected_parents.append(selected_element)
selected_parents = list(set(selected_parents))
linked_aggregate_groups = []
for selected_element in selected_parents:
product_linked_agg_group = [
r.RelatingGroup
for r in getattr(selected_element, "HasAssignments", []) or []
if r.is_a("IfcRelAssignsToGroup")
if self.group_name in r.RelatingGroup.Name
]
linked_aggregate_groups.append(product_linked_agg_group[0].id())
return list(set(linked_aggregate_groups)), selected_parents
selected_objs = context.selected_objects selected_objs = context.selected_objects
selected_elements = [tool.Ifc.get_entity(selected_obj) for selected_obj in selected_objs] selected_elements = [tool.Ifc.get_entity(selected_obj) for selected_obj in selected_objs]
active_element = tool.Ifc.get_entity(context.active_object) active_element = tool.Ifc.get_entity(context.active_object)
active_element = get_element_assembly(active_element) active_element = get_element_assembly(active_element)
selected_objs = context.selected_objects
linked_aggregate_groups, selected_parents = handle_selection(selected_objs)
working_selection = [] if len(linked_aggregate_groups) > 1:
for selected_element in selected_elements: if len(selected_parents) != len(linked_aggregate_groups):
selected_element = get_element_assembly(selected_element) self.report({"INFO"}, "Select only one object from each Linked Aggregate or multiple objects from the same Linked Aggregate.")
if not selected_element: return {"FINISHED"}
self.report({"INFO"}, "Object is not part of a IfcElementAssembly.")
return {'FINISHED'}
working_selection.append(selected_element)
working_selection = list(set(working_selection)) for group in linked_aggregate_groups:
working_ids = [e.id() for e in working_selection] elements = tool.Drawing.get_group_elements(tool.Ifc.get().by_id(group))
selection_group = [] if len(linked_aggregate_groups) > 1:
for selected_element in working_selection: base_instance = [e for e in elements if e in selected_parents][0]
product_linked_agg_group = [ instances_to_refresh = elements
r.RelatingGroup
for r in getattr(selected_element, "HasAssignments", []) or [] elif (len(linked_aggregate_groups) == 1) and (len(selected_parents) > 1):
if r.is_a("IfcRelAssignsToGroup") base_instance = active_element
if self.group_name in r.RelatingGroup.Name instances_to_refresh = [element for element in elements if element in selected_parents]
]
selection_group.append(product_linked_agg_group[0].id()) else:
base_instance = active_element
if len(set(selection_group))> 1: instances_to_refresh = elements
self.report({"INFO"}, "Objects are not part of the same Linked Aggregate")
return {"FINISHED"}
elements = tool.Drawing.get_group_elements(tool.Ifc.get().by_id(selection_group[0])) for element in instances_to_refresh:
if element.GlobalId == base_instance.GlobalId:
for element in elements: continue
if element.GlobalId == active_element.GlobalId:
continue
if (len(working_selection) > 1) and not (element.id() in working_ids):
continue
element_aggregate = ifcopenshell.util.element.get_aggregate(element) element_aggregate = ifcopenshell.util.element.get_aggregate(element)
selected_obj = tool.Ifc.get_object(active_element) selected_obj = tool.Ifc.get_object(base_instance)
selected_matrix = selected_obj.matrix_world selected_matrix = selected_obj.matrix_world
object_duplicate = tool.Ifc.get_object(element) object_duplicate = tool.Ifc.get_object(element)
duplicate_matrix = object_duplicate.matrix_world.decompose() duplicate_matrix = object_duplicate.matrix_world.decompose()
delete_objects(element)
for obj in context.selected_objects:
obj.select_set(False)
tool.Ifc.get_object(active_element).select_set(True) delete_objects(element)
old_to_new = DuplicateMoveLinkedAggregate.execute_ifc_duplicate_linked_aggregate_operator(self, context) for obj in context.selected_objects:
for old, new in old_to_new.items(): obj.select_set(False)
new_obj = tool.Ifc.get_object(new[0]) tool.Ifc.get_object(base_instance).select_set(True)
new_base_matrix = Matrix.LocRotScale(*duplicate_matrix)
matrix_diff = Matrix.inverted(selected_matrix) @ new_obj.matrix_world old_to_new = DuplicateMoveLinkedAggregate.execute_ifc_duplicate_linked_aggregate_operator(self, context)
new_obj_matrix = new_base_matrix @ matrix_diff for old, new in old_to_new.items():
new_obj.matrix_world = new_obj_matrix new_obj = tool.Ifc.get_object(new[0])
new_base_matrix = Matrix.LocRotScale(*duplicate_matrix)
matrix_diff = Matrix.inverted(selected_matrix) @ new_obj.matrix_world
if element_aggregate and new[0].is_a("IfcElementAssembly"): new_obj_matrix = new_base_matrix @ matrix_diff
new_aggregate = ifcopenshell.util.element.get_aggregate(new[0]) new_obj.matrix_world = new_obj_matrix
if not new_aggregate:
blenderbim.core.aggregate.assign_object( if element_aggregate and new[0].is_a("IfcElementAssembly"):
tool.Ifc, new_aggregate = ifcopenshell.util.element.get_aggregate(new[0])
tool.Aggregate,
tool.Collector, if not new_aggregate:
relating_obj=tool.Ifc.get_object(element_aggregate), blenderbim.core.aggregate.assign_object(
related_obj=tool.Ifc.get_object(new[0]), tool.Ifc,
) tool.Aggregate,
tool.Collector,
relating_obj=tool.Ifc.get_object(element_aggregate),
related_obj=tool.Ifc.get_object(new[0]),
)
blenderbim.bim.handler.refresh_ui_data() blenderbim.bim.handler.refresh_ui_data()
src/blenderbim/blenderbim/tool/root.py
-2
View File
@@ -234,10 +234,8 @@ class Root(blenderbim.core.tool.Root):
@classmethod @classmethod
def recreate_aggregate(cls, old_to_new): def recreate_aggregate(cls, old_to_new):
print("O_T_N", old_to_new)
for old, new in old_to_new.items(): for old, new in old_to_new.items():
old_aggregate = ifcopenshell.util.element.get_aggregate(old) old_aggregate = ifcopenshell.util.element.get_aggregate(old)
print("Old", old_aggregate)
if old_aggregate: if old_aggregate:
try: try:
new_aggregate = old_to_new[old_aggregate] new_aggregate = old_to_new[old_aggregate]