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

selector.set_element_value to support 'predefined_type' keyword

Browse Source 
See more details in https://community.osarch.org/discussion/comment/22134/#Comment_22134
This commit is contained in:
Andrej730
2024-09-06 17:35:31 +05:00
parent 39fd9ce462
commit 871cd45f53
2 changed files with 86 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/ifcopenshell/util/selector.py
+35 -3
View File
@@ -21,6 +21,7 @@ import lark
import numpy as np
import ifcopenshell.api.pset
import ifcopenshell.api.geometry
import ifcopenshell.ifcopenshell_wrapper as W
import ifcopenshell.util
import ifcopenshell.util.attribute
import ifcopenshell.util.fm
@@ -508,9 +509,40 @@ def set_element_value(
elif key == "id":
return
elif key == "predefined_type":
print(
f"WARNING. Assigning 'predefined_type' is not yet supported. Skipping value '{value}' for element: '{element}'."
)
current_value = ifcopenshell.util.element.get_predefined_type(element)
if current_value == value:
return
def set_predefined_type(
element: ifcopenshell.entity_instance, value: Union[str, None], *, is_type: bool
) -> None:
predefined_type = element.PredefinedType
declaration = element.wrapped_data.declaration()
entity = declaration.as_entity()
enum_attr = next(attr for attr in entity.attributes() if attr.name() == "PredefinedType")
enum_items = ifcopenshell.util.attribute.get_enum_items(enum_attr)
# USERDEFINED shouldn't occur here, if it does then it means
# then it was artificially added and PredefinedType is actually unset.
if value in (None, "NOTDEFINED", "USERDEFINED"):
element.PredefinedType = "NOTDEFINED"
setattr(element, "ElementType" if is_type else "ObjectType", None)
elif value in enum_items:
if predefined_type == value:
return
element.PredefinedType = value
return
# Value not in PredefinedType enum items.
if predefined_type != "USERDEFINED":
element.PredefinedType = "USERDEFINED"
setattr(element, "ElementType" if is_type else "ObjectType", value)
return
if element_type := ifcopenshell.util.element.get_type(element):
set_predefined_type(element_type, value, is_type=True)
return
set_predefined_type(element, value, is_type=False)
return
elif key == "classification":
element = ifcopenshell.util.classification.get_references(element)