mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-17 10:59:17 +00:00
You can now add/edit/remove/assign/unassign structural analysis models
This commit is contained in:
Dion Moult
@@ -26,6 +26,7 @@ if bpy is not None:
|
||||
"cobie": None,
|
||||
"sequence": None,
|
||||
"group": None,
|
||||
"structural": None,
|
||||
"material": None,
|
||||
"style": None,
|
||||
"layer": None,
|
||||
|
||||
@@ -0,0 +1,26 @@
|
||||
import bpy
|
||||
from . import ui, prop, operator
|
||||
|
||||
classes = (
|
||||
operator.LoadStructuralAnalysisModels,
|
||||
operator.DisableStructuralAnalysisModelEditingUI,
|
||||
operator.AddStructuralAnalysisModel,
|
||||
operator.EditStructuralAnalysisModel,
|
||||
operator.RemoveStructuralAnalysisModel,
|
||||
operator.AssignStructuralAnalysisModel,
|
||||
operator.UnassignStructuralAnalysisModel,
|
||||
operator.EnableEditingStructuralAnalysisModel,
|
||||
operator.DisableEditingStructuralAnalysisModel,
|
||||
prop.StructuralAnalysisModel,
|
||||
prop.BIMStructuralProperties,
|
||||
ui.BIM_PT_structural,
|
||||
ui.BIM_UL_structural_analysis_models,
|
||||
)
|
||||
|
||||
|
||||
def register():
|
||||
bpy.types.Scene.BIMStructuralProperties = bpy.props.PointerProperty(type=prop.BIMStructuralProperties)
|
||||
|
||||
|
||||
def unregister():
|
||||
del bpy.types.Scene.BIMStructuralProperties
|
||||
+18
@@ -0,0 +1,18 @@
|
||||
import ifcopenshell
|
||||
from blenderbim.bim.module.owner.api import create_owner_history
|
||||
|
||||
|
||||
class Usecase:
|
||||
def __init__(self, file, settings={}):
|
||||
self.file = file
|
||||
self.settings = {}
|
||||
for key, value in settings.items():
|
||||
self.settings[key] = value
|
||||
|
||||
def execute(self):
|
||||
return self.file.create_entity("IfcStructuralAnalysisModel", **{
|
||||
"GlobalId": ifcopenshell.guid.new(),
|
||||
"OwnerHistory": create_owner_history(),
|
||||
"Name": "Unnamed",
|
||||
"PredefinedType": "LOADING_3D"
|
||||
})
|
||||
+28
@@ -0,0 +1,28 @@
|
||||
import ifcopenshell
|
||||
from blenderbim.bim.module.owner.api import create_owner_history
|
||||
from blenderbim.bim.module.owner.api import update_owner_history
|
||||
|
||||
|
||||
class Usecase:
|
||||
def __init__(self, file, settings=None):
|
||||
self.file = file
|
||||
self.settings = {
|
||||
"product": None,
|
||||
"structural_analysis_model": None,
|
||||
}
|
||||
for key, value in settings.items():
|
||||
self.settings[key] = value
|
||||
|
||||
def execute(self):
|
||||
if not self.settings["structural_analysis_model"].IsGroupedBy:
|
||||
return self.file.create_entity("IfcRelAssignsToGroup", **{
|
||||
"GlobalId": ifcopenshell.guid.new(),
|
||||
"OwnerHistory": create_owner_history(),
|
||||
"RelatedObjects": [self.settings["product"]],
|
||||
"RelatingGroup": self.settings["structural_analysis_model"]
|
||||
})
|
||||
rel = self.settings["structural_analysis_model"].IsGroupedBy[0]
|
||||
related_objects = set(rel.RelatedObjects) or set()
|
||||
related_objects.add(self.settings["product"])
|
||||
rel.RelatedObjects = list(related_objects)
|
||||
update_owner_history(rel)
|
||||
@@ -0,0 +1,39 @@
|
||||
from blenderbim.bim.ifc import IfcStore
|
||||
|
||||
|
||||
class Data:
|
||||
is_loaded = False
|
||||
products = {}
|
||||
structural_analysis_models = {}
|
||||
|
||||
@classmethod
|
||||
def purge(cls):
|
||||
cls.is_loaded = False
|
||||
cls.products = {}
|
||||
cls.structural_analysis_models = {}
|
||||
|
||||
@classmethod
|
||||
def load(cls):
|
||||
cls.products = {}
|
||||
cls.structural_analysis_models = {}
|
||||
for model in IfcStore.get_file().by_type("IfcStructuralAnalysisModel"):
|
||||
if model.IsGroupedBy:
|
||||
for rel in model.IsGroupedBy:
|
||||
for product in rel.RelatedObjects:
|
||||
cls.products.setdefault(product.id(), []).append(model.id())
|
||||
data = model.get_info()
|
||||
del data["OwnerHistory"]
|
||||
del data["OrientationOf2DPlane"]
|
||||
|
||||
loaded_by = []
|
||||
for load_group in model.LoadedBy or []:
|
||||
loaded_by.append(load_group.id())
|
||||
data["LoadedBy"] = loaded_by
|
||||
|
||||
has_results = []
|
||||
for result_group in model.HasResults or []:
|
||||
has_results.append(result_group.id())
|
||||
data["HasResults"] = has_results
|
||||
|
||||
cls.structural_analysis_models[model.id()] = data
|
||||
cls.is_loaded=True
|
||||
+13
@@ -0,0 +1,13 @@
|
||||
class Usecase:
|
||||
def __init__(self, file, settings=None):
|
||||
self.file = file
|
||||
self.settings = {
|
||||
"structural_analysis_model": None,
|
||||
"attributes": {}
|
||||
}
|
||||
for key, value in settings.items():
|
||||
self.settings[key] = value
|
||||
|
||||
def execute(self):
|
||||
for name, value in self.settings["attributes"].items():
|
||||
setattr(self.settings["structural_analysis_model"], name, value)
|
||||
@@ -0,0 +1,160 @@
|
||||
import bpy
|
||||
import json
|
||||
import blenderbim.bim.module.structural.add_structural_analysis_model as add_structural_analysis_model
|
||||
import blenderbim.bim.module.structural.edit_structural_analysis_model as edit_structural_analysis_model
|
||||
import blenderbim.bim.module.structural.remove_structural_analysis_model as remove_structural_analysis_model
|
||||
import blenderbim.bim.module.structural.assign_structural_analysis_model as assign_structural_analysis_model
|
||||
import blenderbim.bim.module.structural.unassign_structural_analysis_model as unassign_structural_analysis_model
|
||||
from blenderbim.bim.ifc import IfcStore
|
||||
from blenderbim.bim.module.structural.data import Data
|
||||
|
||||
|
||||
class LoadStructuralAnalysisModels(bpy.types.Operator):
|
||||
bl_idname = "bim.load_structural_analysis_models"
|
||||
bl_label = "Load Structural Analysis Models"
|
||||
|
||||
def execute(self, context):
|
||||
props = context.scene.BIMStructuralProperties
|
||||
while len(props.structural_analysis_models) > 0:
|
||||
props.structural_analysis_models.remove(0)
|
||||
for ifc_definition_id, structural_analysis_model in Data.structural_analysis_models.items():
|
||||
new = props.structural_analysis_models.add()
|
||||
new.ifc_definition_id = ifc_definition_id
|
||||
new.name = structural_analysis_model["Name"]
|
||||
props.is_editing = True
|
||||
bpy.ops.bim.disable_editing_structural_analysis_model()
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class DisableStructuralAnalysisModelEditingUI(bpy.types.Operator):
|
||||
bl_idname = "bim.disable_structural_analysis_model_editing_ui"
|
||||
bl_label = "Disable Structural Analysis Model Editing UI"
|
||||
|
||||
def execute(self, context):
|
||||
context.scene.BIMStructuralProperties.is_editing = False
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class AddStructuralAnalysisModel(bpy.types.Operator):
|
||||
bl_idname = "bim.add_structural_analysis_model"
|
||||
bl_label = "Add Structural Analysis Model"
|
||||
|
||||
def execute(self, context):
|
||||
result = add_structural_analysis_model.Usecase(IfcStore.get_file()).execute()
|
||||
Data.load()
|
||||
bpy.ops.bim.load_structural_analysis_models()
|
||||
bpy.ops.bim.enable_editing_structural_analysis_model(structural_analysis_model=result.id())
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class EditStructuralAnalysisModel(bpy.types.Operator):
|
||||
bl_idname = "bim.edit_structural_analysis_model"
|
||||
bl_label = "Edit Structural Analysis Model"
|
||||
|
||||
def execute(self, context):
|
||||
props = context.scene.BIMStructuralProperties
|
||||
attributes = {}
|
||||
for attribute in props.structural_analysis_model_attributes:
|
||||
if attribute.is_null:
|
||||
attributes[attribute.name] = None
|
||||
else:
|
||||
if attribute.data_type == "string":
|
||||
attributes[attribute.name] = attribute.string_value
|
||||
elif attribute.data_type == "enum":
|
||||
attributes[attribute.name] = attribute.enum_value
|
||||
self.file = IfcStore.get_file()
|
||||
edit_structural_analysis_model.Usecase(
|
||||
self.file, {"structural_analysis_model": self.file.by_id(props.active_structural_analysis_model_id), "attributes": attributes}
|
||||
).execute()
|
||||
Data.load()
|
||||
bpy.ops.bim.load_structural_analysis_models()
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class RemoveStructuralAnalysisModel(bpy.types.Operator):
|
||||
bl_idname = "bim.remove_structural_analysis_model"
|
||||
bl_label = "Remove Structural Analysis Model"
|
||||
structural_analysis_model: bpy.props.IntProperty()
|
||||
|
||||
def execute(self, context):
|
||||
props = context.scene.BIMStructuralProperties
|
||||
self.file = IfcStore.get_file()
|
||||
remove_structural_analysis_model.Usecase(self.file, {"structural_analysis_model": self.file.by_id(self.structural_analysis_model)}).execute()
|
||||
Data.load()
|
||||
bpy.ops.bim.load_structural_analysis_models()
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class EnableEditingStructuralAnalysisModel(bpy.types.Operator):
|
||||
bl_idname = "bim.enable_editing_structural_analysis_model"
|
||||
bl_label = "Enable Editing Structural Analysis Model"
|
||||
structural_analysis_model: bpy.props.IntProperty()
|
||||
|
||||
def execute(self, context):
|
||||
props = context.scene.BIMStructuralProperties
|
||||
while len(props.structural_analysis_model_attributes) > 0:
|
||||
props.structural_analysis_model_attributes.remove(0)
|
||||
|
||||
data = Data.structural_analysis_models[self.structural_analysis_model]
|
||||
|
||||
for attribute in IfcStore.get_schema().declaration_by_name("IfcStructuralAnalysisModel").all_attributes():
|
||||
data_type = str(attribute.type_of_attribute)
|
||||
if "<entity" in data_type:
|
||||
continue
|
||||
new = props.structural_analysis_model_attributes.add()
|
||||
new.name = attribute.name()
|
||||
new.is_null = data[attribute.name()] is None
|
||||
new.is_optional = attribute.optional()
|
||||
if attribute.name() == "PredefinedType":
|
||||
new.enum_items = json.dumps(attribute.type_of_attribute().declared_type().enumeration_items())
|
||||
new.data_type = "enum"
|
||||
if data[attribute.name()]:
|
||||
new.enum_value = data[attribute.name()]
|
||||
else:
|
||||
new.string_value = "" if new.is_null else data[attribute.name()]
|
||||
new.data_type = "string"
|
||||
props.active_structural_analysis_model_id = self.structural_analysis_model
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class DisableEditingStructuralAnalysisModel(bpy.types.Operator):
|
||||
bl_idname = "bim.disable_editing_structural_analysis_model"
|
||||
bl_label = "Disable Editing Structural Analysis Model"
|
||||
|
||||
def execute(self, context):
|
||||
context.scene.BIMStructuralProperties.active_structural_analysis_model_id = 0
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class AssignStructuralAnalysisModel(bpy.types.Operator):
|
||||
bl_idname = "bim.assign_structural_analysis_model"
|
||||
bl_label = "Assign Structural Analysis Model"
|
||||
product: bpy.props.StringProperty()
|
||||
structural_analysis_model: bpy.props.IntProperty()
|
||||
|
||||
def execute(self, context):
|
||||
product = bpy.data.objects.get(self.product) if self.product else context.active_object
|
||||
self.file = IfcStore.get_file()
|
||||
assign_structural_analysis_model.Usecase(self.file, {
|
||||
"product": self.file.by_id(product.BIMObjectProperties.ifc_definition_id),
|
||||
"structural_analysis_model": self.file.by_id(self.structural_analysis_model)
|
||||
}).execute()
|
||||
Data.load()
|
||||
return {"FINISHED"}
|
||||
|
||||
|
||||
class UnassignStructuralAnalysisModel(bpy.types.Operator):
|
||||
bl_idname = "bim.unassign_structural_analysis_model"
|
||||
bl_label = "Unassign Structural Analysis Model"
|
||||
product: bpy.props.StringProperty()
|
||||
structural_analysis_model: bpy.props.IntProperty()
|
||||
|
||||
def execute(self, context):
|
||||
product = bpy.data.objects.get(self.product) if self.product else context.active_object
|
||||
self.file = IfcStore.get_file()
|
||||
unassign_structural_analysis_model.Usecase(self.file, {
|
||||
"product": self.file.by_id(product.BIMObjectProperties.ifc_definition_id),
|
||||
"structural_analysis_model": self.file.by_id(self.structural_analysis_model)
|
||||
}).execute()
|
||||
Data.load()
|
||||
return {"FINISHED"}
|
||||
@@ -0,0 +1,26 @@
|
||||
import bpy
|
||||
from blenderbim.bim.prop import StrProperty, Attribute
|
||||
from bpy.types import PropertyGroup
|
||||
from bpy.props import (
|
||||
PointerProperty,
|
||||
StringProperty,
|
||||
EnumProperty,
|
||||
BoolProperty,
|
||||
IntProperty,
|
||||
FloatProperty,
|
||||
FloatVectorProperty,
|
||||
CollectionProperty,
|
||||
)
|
||||
|
||||
|
||||
class StructuralAnalysisModel(PropertyGroup):
|
||||
name: StringProperty(name="Name")
|
||||
ifc_definition_id: IntProperty(name="IFC Definition ID")
|
||||
|
||||
|
||||
class BIMStructuralProperties(PropertyGroup):
|
||||
structural_analysis_model_attributes: CollectionProperty(name="Structural Analysis Model Attributes", type=Attribute)
|
||||
is_editing: BoolProperty(name="Is Editing", default=False)
|
||||
structural_analysis_models: CollectionProperty(name="structural_analysis_models", type=StructuralAnalysisModel)
|
||||
active_structural_analysis_model_index: IntProperty(name="Active Structural Analysis Model Index")
|
||||
active_structural_analysis_model_id: IntProperty(name="Active Structural Analysis Model Id")
|
||||
+11
@@ -0,0 +1,11 @@
|
||||
class Usecase:
|
||||
def __init__(self, file, settings=None):
|
||||
self.file = file
|
||||
self.settings = {"structural_analysis_model": None}
|
||||
for key, value in settings.items():
|
||||
self.settings[key] = value
|
||||
|
||||
def execute(self):
|
||||
for rel in self.settings["structural_analysis_model"].IsGroupedBy or []:
|
||||
self.file.remove(rel)
|
||||
self.file.remove(self.settings["structural_analysis_model"])
|
||||
@@ -0,0 +1,87 @@
|
||||
from bpy.types import Panel, UIList
|
||||
from blenderbim.bim.ifc import IfcStore
|
||||
from blenderbim.bim.module.structural.data import Data
|
||||
|
||||
|
||||
class BIM_PT_structural(Panel):
|
||||
bl_label = "IFC Structural Analysis Models"
|
||||
bl_idname = "BIM_PT_structural"
|
||||
bl_options = {"DEFAULT_CLOSED"}
|
||||
bl_space_type = "PROPERTIES"
|
||||
bl_region_type = "WINDOW"
|
||||
bl_context = "scene"
|
||||
|
||||
@classmethod
|
||||
def poll(cls, context):
|
||||
return IfcStore.get_file()
|
||||
|
||||
def draw(self, context):
|
||||
if not Data.is_loaded:
|
||||
Data.load()
|
||||
self.props = context.scene.BIMStructuralProperties
|
||||
|
||||
row = self.layout.row(align=True)
|
||||
row.label(
|
||||
text="{} Structural Analysis Models Found".format(len(Data.structural_analysis_models)), icon="MOD_SIMPLIFY"
|
||||
)
|
||||
if self.props.is_editing:
|
||||
row.operator("bim.add_structural_analysis_model", text="", icon="ADD")
|
||||
row.operator("bim.disable_structural_analysis_model_editing_ui", text="", icon="CHECKMARK")
|
||||
else:
|
||||
row.operator("bim.load_structural_analysis_models", text="", icon="GREASEPENCIL")
|
||||
|
||||
if self.props.is_editing:
|
||||
self.layout.template_list(
|
||||
"BIM_UL_structural_analysis_models",
|
||||
"",
|
||||
self.props,
|
||||
"structural_analysis_models",
|
||||
self.props,
|
||||
"active_structural_analysis_model_index",
|
||||
)
|
||||
|
||||
if self.props.active_structural_analysis_model_id:
|
||||
self.draw_editable_ui(context)
|
||||
|
||||
def draw_editable_ui(self, context):
|
||||
for attribute in self.props.structural_analysis_model_attributes:
|
||||
row = self.layout.row(align=True)
|
||||
if attribute.data_type == "string":
|
||||
row.prop(attribute, "string_value", text=attribute.name)
|
||||
elif attribute.data_type == "enum":
|
||||
row.prop(attribute, "enum_value", text=attribute.name)
|
||||
if attribute.is_optional:
|
||||
row.prop(attribute, "is_null", icon="RADIOBUT_OFF" if attribute.is_null else "RADIOBUT_ON", text="")
|
||||
|
||||
|
||||
class BIM_UL_structural_analysis_models(UIList):
|
||||
def draw_item(self, context, layout, data, item, icon, active_data, active_propname):
|
||||
if item:
|
||||
row = layout.row(align=True)
|
||||
row.label(text=item.name)
|
||||
|
||||
if context.active_object:
|
||||
oprops = context.active_object.BIMObjectProperties
|
||||
if (
|
||||
oprops.ifc_definition_id in Data.products
|
||||
and item.ifc_definition_id in Data.products[oprops.ifc_definition_id]
|
||||
):
|
||||
op = row.operator(
|
||||
"bim.unassign_structural_analysis_model", text="", icon="KEYFRAME_HLT", emboss=False
|
||||
)
|
||||
op.structural_analysis_model = item.ifc_definition_id
|
||||
else:
|
||||
op = row.operator("bim.assign_structural_analysis_model", text="", icon="KEYFRAME", emboss=False)
|
||||
op.structural_analysis_model = item.ifc_definition_id
|
||||
|
||||
if context.scene.BIMStructuralProperties.active_structural_analysis_model_id == item.ifc_definition_id:
|
||||
row.operator("bim.edit_structural_analysis_model", text="", icon="CHECKMARK")
|
||||
row.operator("bim.disable_editing_structural_analysis_model", text="", icon="X")
|
||||
elif context.scene.BIMStructuralProperties.active_structural_analysis_model_id:
|
||||
op = row.operator("bim.remove_structural_analysis_model", text="", icon="X")
|
||||
op.structural_analysis_model = item.ifc_definition_id
|
||||
else:
|
||||
op = row.operator("bim.enable_editing_structural_analysis_model", text="", icon="GREASEPENCIL")
|
||||
op.structural_analysis_model = item.ifc_definition_id
|
||||
op = row.operator("bim.remove_structural_analysis_model", text="", icon="X")
|
||||
op.structural_analysis_model = item.ifc_definition_id
|
||||
+25
@@ -0,0 +1,25 @@
|
||||
import ifcopenshell
|
||||
from blenderbim.bim.module.owner.api import update_owner_history
|
||||
|
||||
|
||||
class Usecase:
|
||||
def __init__(self, file, settings=None):
|
||||
self.file = file
|
||||
self.settings = {
|
||||
"product": None,
|
||||
"structural_analysis_model": None,
|
||||
}
|
||||
for key, value in settings.items():
|
||||
self.settings[key] = value
|
||||
|
||||
def execute(self):
|
||||
if not self.settings["structural_analysis_model"].IsGroupedBy:
|
||||
return
|
||||
rel = self.settings["structural_analysis_model"].IsGroupedBy[0]
|
||||
related_objects = set(rel.RelatedObjects) or set()
|
||||
related_objects.remove(self.settings["product"])
|
||||
if len(related_objects):
|
||||
rel.RelatedObjects = list(related_objects)
|
||||
update_owner_history(rel)
|
||||
else:
|
||||
self.file.remove(rel)
|
||||
Reference in New Issue
Block a user