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

AddRepresentation executes on all selected objects

Browse Source
This commit is contained in:
Gorgious
2021-08-24 13:40:18 +02:00
parent b149e8c218
commit 5b88740ad6
1 changed files with 68 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/blenderbim/blenderbim/bim/module/geometry/operator.py
+68 -69
View File
@@ -88,87 +88,86 @@ class AddRepresentation(bpy.types.Operator):
return IfcStore.execute_ifc_operator(self, context) return IfcStore.execute_ifc_operator(self, context)
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()
bpy.ops.bim.edit_object_placement(obj=obj.name)
if not obj.data:
return {"FINISHED"}
product = self.file.by_id(obj.BIMObjectProperties.ifc_definition_id)
context_id = self.context_id or int(context.scene.BIMProperties.contexts)
context_of_items = self.file.by_id(context_id)
gprop = context.scene.BIMGeoreferenceProperties gprop = context.scene.BIMGeoreferenceProperties
coordinate_offset = None objs = [bpy.data.objects.get(self.obj)] if self.obj else context.selected_objects
if gprop.has_blender_offset and obj.BIMObjectProperties.blender_offset_type == "CARTESIAN_POINT": for obj in objs:
coordinate_offset = Vector( if not obj.BIMObjectProperties.ifc_definition_id:
( continue
float(gprop.blender_eastings), bpy.ops.bim.edit_object_placement(obj=obj.name)
float(gprop.blender_northings), if not obj.data:
float(gprop.blender_orthogonal_height), continue
product = self.file.by_id(obj.BIMObjectProperties.ifc_definition_id)
context_id = self.context_id or int(context.scene.BIMProperties.contexts)
context_of_items = self.file.by_id(context_id)
coordinate_offset = None
if gprop.has_blender_offset and obj.BIMObjectProperties.blender_offset_type == "CARTESIAN_POINT":
coordinate_offset = Vector(
(
float(gprop.blender_eastings),
float(gprop.blender_northings),
float(gprop.blender_orthogonal_height),
)
) )
)
representation_data = { representation_data = {
"context": context_of_items, "context": context_of_items,
"blender_object": obj, "blender_object": obj,
"geometry": obj.data, "geometry": obj.data,
"coordinate_offset": coordinate_offset, "coordinate_offset": coordinate_offset,
"total_items": max(1, len(obj.material_slots)), "total_items": max(1, len(obj.material_slots)),
"should_force_faceted_brep": context.scene.BIMGeometryProperties.should_force_faceted_brep, "should_force_faceted_brep": context.scene.BIMGeometryProperties.should_force_faceted_brep,
"should_force_triangulation": context.scene.BIMGeometryProperties.should_force_triangulation, "should_force_triangulation": context.scene.BIMGeometryProperties.should_force_triangulation,
"ifc_representation_class": self.ifc_representation_class, "ifc_representation_class": self.ifc_representation_class,
"profile_set_usage": self.file.by_id(self.profile_set_usage) if self.profile_set_usage else None, "profile_set_usage": self.file.by_id(self.profile_set_usage) if self.profile_set_usage else None,
} }
result = ifcopenshell.api.run("geometry.add_representation", self.file, **representation_data) result = ifcopenshell.api.run("geometry.add_representation", self.file, **representation_data)
if not result: if not result:
print("Failed to write shape representation") print("Failed to write shape representation")
return {"FINISHED"} continue
[ [
bpy.ops.bim.add_style(material=s.material.name) bpy.ops.bim.add_style(material=s.material.name)
for s in obj.material_slots for s in obj.material_slots
if s.material and not s.material.BIMMaterialProperties.ifc_style_id if s.material and not s.material.BIMMaterialProperties.ifc_style_id
] ]
if isinstance(obj.data, bpy.types.Mesh) and len(obj.data.polygons): if obj.type == "MESH" and len(obj.data.polygons):
ifcopenshell.api.run(
"style.assign_representation_styles",
self.file,
**{
"shape_representation": result,
"styles": [
self.file.by_id(s.material.BIMMaterialProperties.ifc_style_id)
for s in obj.material_slots
if s.material
],
"should_use_presentation_style_assignment": context.scene.BIMGeometryProperties.should_use_presentation_style_assignment,
},
)
ifcopenshell.api.run( ifcopenshell.api.run(
"style.assign_representation_styles", "geometry.assign_representation", self.file, **{"product": product, "representation": result}
self.file,
**{
"shape_representation": result,
"styles": [
self.file.by_id(s.material.BIMMaterialProperties.ifc_style_id)
for s in obj.material_slots
if s.material
],
"should_use_presentation_style_assignment": context.scene.BIMGeometryProperties.should_use_presentation_style_assignment,
},
) )
ifcopenshell.api.run(
"geometry.assign_representation", self.file, **{"product": product, "representation": result}
)
existing_mesh = obj.data mesh = obj.data.copy()
mesh = obj.data.copy() mesh.name = "{}/{}".format(context_id, result.id())
mesh.name = "{}/{}".format(context_id, result.id()) mesh.BIMMeshProperties.ifc_definition_id = int(result.id())
mesh.BIMMeshProperties.ifc_definition_id = int(result.id()) obj.data = mesh
obj.data = mesh Data.load(IfcStore.get_file(), obj.BIMObjectProperties.ifc_definition_id)
Data.load(IfcStore.get_file(), obj.BIMObjectProperties.ifc_definition_id)
if product.is_a("IfcTypeProduct"): if product.is_a("IfcTypeProduct"):
if self.file.schema == "IFC2X3": if self.file.schema == "IFC2X3":
types = product.ObjectTypeOf types = product.ObjectTypeOf
else: else:
types = product.Types types = product.Types
if types: if types:
for element in types[0].RelatedObjects: for element in types[0].RelatedObjects:
Data.load(IfcStore.get_file(), element.id()) Data.load(IfcStore.get_file(), element.id())
return {"FINISHED"} return {"FINISHED"}