diff --git a/src/bonsai/bonsai/bim/module/unit/operator.py b/src/bonsai/bonsai/bim/module/unit/operator.py index f719eac74c..f7c129aba1 100644 --- a/src/bonsai/bonsai/bim/module/unit/operator.py +++ b/src/bonsai/bonsai/bim/module/unit/operator.py @@ -30,10 +30,12 @@ if TYPE_CHECKING: class AssignSceneUnits(bpy.types.Operator, tool.Ifc.Operator): bl_idname = "bim.assign_scene_units" bl_label = "Assign Scene Units" + bl_description = "Add new units based on the current Blender scene units and assign them as project default." bl_options = {"REGISTER", "UNDO"} def _execute(self, context): core.assign_scene_units(tool.Ifc, tool.Unit) + core.load_units(tool.Unit) class AssignUnit(bpy.types.Operator, tool.Ifc.Operator): diff --git a/src/bonsai/bonsai/bim/module/unit/ui.py b/src/bonsai/bonsai/bim/module/unit/ui.py index c3051b5cf9..9407ef01ac 100644 --- a/src/bonsai/bonsai/bim/module/unit/ui.py +++ b/src/bonsai/bonsai/bim/module/unit/ui.py @@ -53,6 +53,7 @@ class BIM_PT_units(Panel): row = self.layout.row(align=True) row.label(text="{} Units Found".format(UnitsData.data["total_units"]), icon="SNAP_GRID") if self.props.is_editing: + row.operator("bim.assign_scene_units", text="", icon="TOOL_SETTINGS") row.operator("bim.disable_unit_editing_ui", text="", icon="CANCEL") else: row.operator("bim.load_units", text="", icon="GREASEPENCIL")