diff --git a/src/blenderbim/blenderbim/bim/helper.py b/src/blenderbim/blenderbim/bim/helper.py index f0dab98427..6b11944f87 100644 --- a/src/blenderbim/blenderbim/bim/helper.py +++ b/src/blenderbim/blenderbim/bim/helper.py @@ -38,11 +38,14 @@ def draw_attribute(attribute, layout, copy_operator=None): if not value_name: layout.label(text=attribute.name) return - layout.prop( - attribute, - value_name, - text=attribute.name, - ) + if value_name == "enum_value": + prop_with_search(layout, attribute, "enum_value", text=attribute.name) + else: + layout.prop( + attribute, + value_name, + text=attribute.name, + ) if attribute.is_optional: layout.prop(attribute, "is_null", icon="RADIOBUT_OFF" if attribute.is_null else "RADIOBUT_ON", text="") if copy_operator: @@ -105,6 +108,16 @@ def export_attributes(props, callback=None): return attributes +def prop_with_search(layout, data, prop_name, **kwargs): + # kwargs are layout.prop arguments (text, icon, etc.) + row = layout.row(align=True) + # Magick courtesy of https://blender.stackexchange.com/a/203443/86891 + row.context_pointer_set(name="data", data=data) + row.prop(data, prop_name, **kwargs) + op = row.operator("bim.enum_property_search", text="", icon="VIEWZOOM") + op.prop_name = prop_name + + class IfcHeaderExtractor: def __init__(self, filepath: str): self.filepath = filepath diff --git a/src/blenderbim/blenderbim/bim/module/brick/ui.py b/src/blenderbim/blenderbim/bim/module/brick/ui.py index f0b1cedce7..bb8898b2c0 100644 --- a/src/blenderbim/blenderbim/bim/module/brick/ui.py +++ b/src/blenderbim/blenderbim/bim/module/brick/ui.py @@ -18,7 +18,7 @@ import blenderbim.tool as tool from bpy.types import Panel, UIList -from blenderbim.bim.ui import prop_with_search +from blenderbim.bim.helper import prop_with_search from blenderbim.bim.module.brick.data import BrickschemaData, BrickschemaReferencesData diff --git a/src/blenderbim/blenderbim/bim/module/geometry/ui.py b/src/blenderbim/blenderbim/bim/module/geometry/ui.py index 4ce050ffb9..5f7c918de0 100644 --- a/src/blenderbim/blenderbim/bim/module/geometry/ui.py +++ b/src/blenderbim/blenderbim/bim/module/geometry/ui.py @@ -19,7 +19,7 @@ import bpy from bpy.types import Panel from blenderbim.bim.ifc import IfcStore -from blenderbim.bim.ui import prop_with_search +from blenderbim.bim.helper import prop_with_search from blenderbim.bim.module.geometry.data import RepresentationsData, DerivedPlacementsData diff --git a/src/blenderbim/blenderbim/bim/module/material/ui.py b/src/blenderbim/blenderbim/bim/module/material/ui.py index f657ab0a40..cddd181189 100644 --- a/src/blenderbim/blenderbim/bim/module/material/ui.py +++ b/src/blenderbim/blenderbim/bim/module/material/ui.py @@ -22,7 +22,7 @@ from ifcopenshell.api.material.data import Data from ifcopenshell.api.profile.data import Data as ProfileData from blenderbim.bim.ifc import IfcStore from blenderbim.bim.helper import draw_attributes -from blenderbim.bim.ui import prop_with_search +from blenderbim.bim.helper import prop_with_search from blenderbim.bim.module.material.data import MaterialsData, ObjectMaterialData diff --git a/src/blenderbim/blenderbim/bim/module/patch/ui.py b/src/blenderbim/blenderbim/bim/module/patch/ui.py index 9d58971d7d..9e7a241e51 100644 --- a/src/blenderbim/blenderbim/bim/module/patch/ui.py +++ b/src/blenderbim/blenderbim/bim/module/patch/ui.py @@ -16,7 +16,7 @@ # You should have received a copy of the GNU General Public License # along with BlenderBIM Add-on. If not, see . -from blenderbim.bim.ui import prop_with_search +from blenderbim.bim.helper import prop_with_search import bpy from blenderbim.bim.helper import draw_attributes diff --git a/src/blenderbim/blenderbim/bim/module/project/ui.py b/src/blenderbim/blenderbim/bim/module/project/ui.py index 8a0a21f109..4aff3bb1d3 100644 --- a/src/blenderbim/blenderbim/bim/module/project/ui.py +++ b/src/blenderbim/blenderbim/bim/module/project/ui.py @@ -17,7 +17,7 @@ # along with BlenderBIM Add-on. If not, see . import os -from blenderbim.bim.ui import prop_with_search +from blenderbim.bim.helper import prop_with_search from bpy.types import Panel, UIList from blenderbim.bim.ifc import IfcStore from blenderbim.bim.module.project.data import ProjectData diff --git a/src/blenderbim/blenderbim/bim/module/pset/ui.py b/src/blenderbim/blenderbim/bim/module/pset/ui.py index 7f6f4d7cf0..b6c67b1611 100644 --- a/src/blenderbim/blenderbim/bim/module/pset/ui.py +++ b/src/blenderbim/blenderbim/bim/module/pset/ui.py @@ -18,7 +18,7 @@ from bpy.types import Panel from blenderbim.bim.ifc import IfcStore -from blenderbim.bim.ui import prop_with_search +from blenderbim.bim.helper import prop_with_search from blenderbim.bim.module.pset.data import ( ObjectPsetsData, ObjectQtosData, diff --git a/src/blenderbim/blenderbim/bim/module/pset_template/ui.py b/src/blenderbim/blenderbim/bim/module/pset_template/ui.py index 4fa0b67644..b15ca05bd4 100644 --- a/src/blenderbim/blenderbim/bim/module/pset_template/ui.py +++ b/src/blenderbim/blenderbim/bim/module/pset_template/ui.py @@ -19,7 +19,7 @@ import bpy from bpy.types import Panel from blenderbim.bim.ifc import IfcStore -from blenderbim.bim.ui import prop_with_search +from blenderbim.bim.helper import prop_with_search from blenderbim.bim.module.pset_template.prop import ( getPsetTemplates, ) diff --git a/src/blenderbim/blenderbim/bim/module/root/ui.py b/src/blenderbim/blenderbim/bim/module/root/ui.py index b6a5f99268..18ca900cea 100644 --- a/src/blenderbim/blenderbim/bim/module/root/ui.py +++ b/src/blenderbim/blenderbim/bim/module/root/ui.py @@ -20,7 +20,7 @@ import bpy import blenderbim.bim.module.root.prop as root_prop from bpy.types import Panel from blenderbim.bim.ifc import IfcStore -from blenderbim.bim.ui import prop_with_search +from blenderbim.bim.helper import prop_with_search from blenderbim.bim.module.root.data import IfcClassData diff --git a/src/blenderbim/blenderbim/bim/module/unit/ui.py b/src/blenderbim/blenderbim/bim/module/unit/ui.py index 183ed5cbe7..10a3682296 100644 --- a/src/blenderbim/blenderbim/bim/module/unit/ui.py +++ b/src/blenderbim/blenderbim/bim/module/unit/ui.py @@ -19,7 +19,7 @@ import blenderbim.bim.helper from bpy.types import Panel, UIList from blenderbim.bim.ifc import IfcStore -from blenderbim.bim.ui import prop_with_search +from blenderbim.bim.helper import prop_with_search from blenderbim.bim.module.unit.data import UnitsData diff --git a/src/blenderbim/blenderbim/bim/prop.py b/src/blenderbim/blenderbim/bim/prop.py index ffdfc8bb27..e86896fd39 100644 --- a/src/blenderbim/blenderbim/bim/prop.py +++ b/src/blenderbim/blenderbim/bim/prop.py @@ -240,6 +240,10 @@ class Attribute(PropertyGroup): value = str(value) setattr(self, self.get_value_name(), value) + getter_enum = { + "enum_value": getAttributeEnumValues, + } + class ModuleVisibility(PropertyGroup): name: StringProperty(name="Name") diff --git a/src/blenderbim/blenderbim/bim/ui.py b/src/blenderbim/blenderbim/bim/ui.py index e04e05d5be..2457cd99bf 100644 --- a/src/blenderbim/blenderbim/bim/ui.py +++ b/src/blenderbim/blenderbim/bim/ui.py @@ -170,15 +170,6 @@ class BIM_ADDON_preferences(bpy.types.AddonPreferences): row.operator("bim.configure_visibility") -def prop_with_search(layout, data, prop_name, **kwargs): # Kwargs are layout.prop arguments (text, icon, etc.) - row = layout.row(align=True) - # Magick courtesy of https://blender.stackexchange.com/a/203443/86891 - row.context_pointer_set(name="data", data=data) - row.prop(data, prop_name, **kwargs) - op = row.operator("bim.enum_property_search", text="", icon="VIEWZOOM") - op.prop_name = prop_name - - def ifc_units(self, context): scene = context.scene props = scene.BIMProperties