diff --git a/src/bonsai/bonsai/bim/module/geometry/ui.py b/src/bonsai/bonsai/bim/module/geometry/ui.py index 80f5751eeca..f76bbed8938 100644 --- a/src/bonsai/bonsai/bim/module/geometry/ui.py +++ b/src/bonsai/bonsai/bim/module/geometry/ui.py @@ -32,13 +32,30 @@ from bonsai.bim.module.layer.data import LayersData +class UIData: + data = {} + is_loaded = False + + @classmethod + def load(cls): + cls.data = {"menu_icon_color_mode": cls.icon_color_mode("user_interface.wcol_menu.text")} + cls.is_loaded = True + + @classmethod + def icon_color_mode(cls, color_path): + return tool.Blender.detect_icon_color_mode(color_path) + + def mode_menu(self, context): if not tool.Ifc.get(): return + if not UIData.is_loaded: + UIData.load() + ifc_icon = f"{UIData.data['menu_icon_color_mode']}_ifc" row = self.layout.row(align=True) if context.scene.BIMGeometryProperties.mode == "EDIT": row.operator("bim.override_mode_set_object", icon="CANCEL", text="Discard Changes").should_save = False - row.prop(context.scene.BIMGeometryProperties, "mode", text="", icon_value=bonsai.bim.icons["IFC"].icon_id) + row.prop(context.scene.BIMGeometryProperties, "mode", text="", icon_value=bonsai.bim.icons[ifc_icon].icon_id) def object_menu(self, context): diff --git a/src/bonsai/bonsai/bim/module/model/ops.authoring.geographicelement.dat b/src/bonsai/bonsai/bim/module/model/ops.authoring.geographicelement.dat new file mode 100644 index 00000000000..770d95e149d Binary files /dev/null and b/src/bonsai/bonsai/bim/module/model/ops.authoring.geographicelement.dat differ diff --git a/src/bonsai/bonsai/bim/module/project/ui.py b/src/bonsai/bonsai/bim/module/project/ui.py index 4b4b5dcad97..51505ed46e0 100644 --- a/src/bonsai/bonsai/bim/module/project/ui.py +++ b/src/bonsai/bonsai/bim/module/project/ui.py @@ -37,6 +37,24 @@ def file_import_menu(self, context): op.should_start_fresh_session = False +def refresh(): + UIData.is_loaded = False + + +class UIData: + data = {} + is_loaded = False + + @classmethod + def load(cls): + cls.data = {"menu_item_icon_color_mode": cls.icon_color_mode("user_interface.wcol_menu_item.text")} + cls.is_loaded = True + + @classmethod + def icon_color_mode(cls, color_path): + return tool.Blender.detect_icon_color_mode(color_path) + + class BIM_MT_project(Menu): bl_idname = "BIM_MT_project" bl_label = "New IFC Project" @@ -54,6 +72,9 @@ class BIM_MT_recent_projects(Menu): bl_label = "Open Recent IFC Project" def draw(self, context): + if not UIData.is_loaded: + UIData.load() + ifc_icon = f"{UIData.data['menu_item_icon_color_mode']}_ifc" layout = self.layout paths = tool.Project.get_recent_ifc_projects() if not paths: @@ -62,7 +83,7 @@ def draw(self, context): for path in paths: row = layout.row() - op = row.operator("bim.load_project", text=path.name, icon_value=bonsai.bim.icons["IFC"].icon_id) + op = row.operator("bim.load_project", text=path.name, icon_value=bonsai.bim.icons[ifc_icon].icon_id) op.filepath = str(path) op.should_start_fresh_session = True op.use_detailed_tooltip = True diff --git a/src/bonsai/bonsai/bim/prop.py b/src/bonsai/bonsai/bim/prop.py index caddd909755..8c790444f1b 100644 --- a/src/bonsai/bonsai/bim/prop.py +++ b/src/bonsai/bonsai/bim/prop.py @@ -406,11 +406,16 @@ def set_value(self, value: Union[Any, str, float, bool, int]) -> None: def get_tab( self: "Union[BIMAreaProperties, BIMTabProperties]", context: bpy.types.Context ) -> list[tuple[str, str, str, str, int]]: - # Items are not very dynamic, but can't make them completely static - # because icons are not yet available during prop registration. + + UIData = bonsai.bim.ui.UIData + if not UIData.is_loaded: + UIData.load() + + icon_key = f"{UIData.data['menu_icon_color_mode']}_ifc" + if not hasattr(get_tab, "enum_items"): get_tab.enum_items = [ - ("PROJECT", "Project Overview", "", bonsai.bim.icons["IFC"].icon_id, 0), + ("PROJECT", "Project Overview", "", bonsai.bim.icons[icon_key].icon_id, 0), ("OBJECT", "Object Information", "", "FILE_3D", 1), ("GEOMETRY", "Geometry and Materials", "", "MATERIAL", 2), ("DRAWINGS", "Drawings and Documents", "", "DOCUMENTS", 3), diff --git a/src/bonsai/bonsai/bim/ui.py b/src/bonsai/bonsai/bim/ui.py index 3579366a1e9..bc6a1a2b304 100644 --- a/src/bonsai/bonsai/bim/ui.py +++ b/src/bonsai/bonsai/bim/ui.py @@ -407,7 +407,7 @@ def draw(self, context): UIData.load() is_ifc_project = bool(tool.Ifc.get()) aprops = tool.Blender.get_area_props(context) - ifc_icon = f"{UIData.data['icon_color_mode']}_ifc" + ifc_icon = f"{UIData.data['tabs_icon_color_mode']}_ifc" row = self.layout.row() row.alignment = "CENTER" @@ -1118,7 +1118,11 @@ class UIData: @classmethod def load(cls): - cls.data = {"version": cls.version(), "icon_color_mode": cls.icon_color_mode()} + cls.data = { + "version": cls.version(), + "tabs_icon_color_mode": cls.icon_color_mode("user_interface.wcol_regular.text"), + "menu_icon_color_mode": cls.icon_color_mode("user_interface.wcol_menu.text"), + } cls.is_loaded = True @classmethod @@ -1126,8 +1130,8 @@ def version(cls): return tool.Blender.get_bonsai_version() @classmethod - def icon_color_mode(cls): - return tool.Blender.detect_icon_color_mode() + def icon_color_mode(cls, color_path): + return tool.Blender.detect_icon_color_mode(color_path) def draw_statusbar(self, context):