Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-09-22 14:48:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

#2692 - bulk assign and unassign ifcmaterials

Browse Source
This commit is contained in:
Ryan Schultz
2023-01-28 15:13:27 -06:00
committed by Dion Moult
parent 70f35095ff
commit 2902d96690
1 changed files with 47 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/blenderbim/blenderbim/bim/module/material/operator.py
+47 -30
View File
@@ -146,23 +146,32 @@ class AssignMaterial(bpy.types.Operator, tool.Ifc.Operator):
bl_options = {"REGISTER", "UNDO"} bl_options = {"REGISTER", "UNDO"}
obj: bpy.props.StringProperty() obj: bpy.props.StringProperty()
material_type: bpy.props.StringProperty() material_type: bpy.props.StringProperty()
def _execute(self, context): def _execute(self, context):
obj = bpy.data.objects.get(self.obj) if self.obj else context.active_object
material_type = self.material_type or obj.BIMObjectMaterialProperties.material_type
self.file = IfcStore.get_file() self.file = IfcStore.get_file()
element = self.file.by_id(obj.BIMObjectProperties.ifc_definition_id)
ifcopenshell.api.run( if self.obj:
"material.assign_material", objects = [bpy.data.objects.get(self.obj)]
self.file, else:
**{ objects = context.selected_objects
"product": element, active_obj = context.active_object
"type": material_type, active_object_material_type = self.material_type or active_obj.BIMObjectMaterialProperties.material_type
"material": self.file.by_id(int(obj.BIMObjectMaterialProperties.material)), active_object_material = active_obj.BIMObjectMaterialProperties.material
}, for obj in objects:
) element = self.file.by_id(obj.BIMObjectProperties.ifc_definition_id)
Data.load(IfcStore.get_file()) ifcopenshell.api.run(
Data.load(IfcStore.get_file(), obj.BIMObjectProperties.ifc_definition_id) "material.assign_material",
self.file,
**{
"product": element,
"type": active_object_material_type,
"material": self.file.by_id(int(active_object_material)),
},
)
Data.load(IfcStore.get_file(), obj.BIMObjectProperties.ifc_definition_id)
class UnassignMaterial(bpy.types.Operator, tool.Ifc.Operator): class UnassignMaterial(bpy.types.Operator, tool.Ifc.Operator):
@@ -172,13 +181,18 @@ class UnassignMaterial(bpy.types.Operator, tool.Ifc.Operator):
obj: bpy.props.StringProperty() obj: bpy.props.StringProperty()
def _execute(self, context): def _execute(self, context):
obj = bpy.data.objects.get(self.obj) if self.obj else context.active_object
self.file = IfcStore.get_file() self.file = IfcStore.get_file()
ifcopenshell.api.run(
"material.unassign_material", if self.obj:
self.file, objects = [bpy.data.objects.get(self.obj)]
**{"product": self.file.by_id(obj.BIMObjectProperties.ifc_definition_id)}, else:
) objects = context.selected_objects
for obj in objects:
ifcopenshell.api.run(
"material.unassign_material",
self.file,
**{"product": self.file.by_id(obj.BIMObjectProperties.ifc_definition_id)},
)
Data.purge() Data.purge()
@@ -467,18 +481,21 @@ class EditAssignedMaterial(bpy.types.Operator, tool.Ifc.Operator):
def _execute(self, context): def _execute(self, context):
self.file = IfcStore.get_file() self.file = IfcStore.get_file()
obj = bpy.data.objects.get(self.obj) if self.obj else context.active_object active_obj = bpy.data.objects.get(self.obj) if self.obj else context.active_object
props = obj.BIMObjectMaterialProperties props = active_obj.BIMObjectMaterialProperties
product_data = Data.products[obj.BIMObjectProperties.ifc_definition_id] product_data = Data.products[active_obj.BIMObjectProperties.ifc_definition_id]
objects = context.selected_objects
if props.active_material_set_item_id != 0: # We were editing a material layer set item if props.active_material_set_item_id != 0: # We were editing a material layer set item
bpy.ops.bim.edit_material_set_item(material_set_item=props.active_material_set_item_id) bpy.ops.bim.edit_material_set_item(material_set_item=props.active_material_set_item_id)
if product_data["type"] == "IfcMaterial": if product_data["type"] == "IfcMaterial":
bpy.ops.bim.unassign_material(obj=obj.name) for obj in objects:
bpy.ops.bim.assign_material(obj=obj.name, material_type="IfcMaterial") bpy.ops.bim.unassign_material(obj=obj.name)
Data.load(IfcStore.get_file(), obj.BIMObjectProperties.ifc_definition_id) bpy.ops.bim.assign_material(obj=obj.name, material_type="IfcMaterial")
bpy.ops.bim.disable_editing_assigned_material(obj=obj.name) Data.load(IfcStore.get_file(), obj.BIMObjectProperties.ifc_definition_id)
bpy.ops.bim.disable_editing_assigned_material(obj=active_obj.name)
return {"FINISHED"} return {"FINISHED"}
material_set = self.file.by_id(self.material_set) material_set = self.file.by_id(self.material_set)
@@ -488,7 +505,7 @@ class EditAssignedMaterial(bpy.types.Operator, tool.Ifc.Operator):
self.file, self.file,
**{"element": material_set, "attributes": attributes}, **{"element": material_set, "attributes": attributes},
) )
Data.load(IfcStore.get_file(), obj.BIMObjectProperties.ifc_definition_id) Data.load(IfcStore.get_file(), active_obj.BIMObjectProperties.ifc_definition_id)
if self.material_set_usage: if self.material_set_usage:
material_set_usage = self.file.by_id(self.material_set_usage) material_set_usage = self.file.by_id(self.material_set_usage)
@@ -516,7 +533,7 @@ class EditAssignedMaterial(bpy.types.Operator, tool.Ifc.Operator):
Data.load_layers() Data.load_layers()
elif material_set.is_a("IfcMaterialProfileSet"): elif material_set.is_a("IfcMaterialProfileSet"):
Data.load_profiles() Data.load_profiles()
bpy.ops.bim.disable_editing_assigned_material(obj=obj.name) bpy.ops.bim.disable_editing_assigned_material(obj=active_obj.name)
class EnableEditingMaterialSetItemProfile(bpy.types.Operator): class EnableEditingMaterialSetItemProfile(bpy.types.Operator):