mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-09-24 20:00:02 +00:00
Dion Moult
97 lines
3.5 KiB
Python
97 lines
3.5 KiB
Python
import bpy
|
|
import blenderbim.bim.module.grid.create_grid_axis as create_grid_axis
|
|
from bpy.types import Operator
|
|
from bpy.props import FloatProperty, IntProperty
|
|
from mathutils import Vector
|
|
from blenderbim.bim.ifc import IfcStore
|
|
|
|
|
|
def add_object(self, context):
|
|
obj = bpy.data.objects.new("Grid", None)
|
|
obj.name = "Grid"
|
|
|
|
collection = bpy.data.collections.new(obj.name)
|
|
has_site_collection = False
|
|
for child in bpy.context.view_layer.layer_collection.children:
|
|
if "IfcProject/" not in child.name:
|
|
continue
|
|
for grandchild in child.children:
|
|
if "IfcSite/" not in grandchild.name:
|
|
continue
|
|
has_site_collection = True
|
|
grandchild.collection.children.link(collection)
|
|
break
|
|
if not has_site_collection:
|
|
bpy.context.view_layer.active_layer_collection.collection.children.link(collection)
|
|
collection.objects.link(obj)
|
|
|
|
self.file = IfcStore.get_file()
|
|
if self.file:
|
|
bpy.ops.bim.assign_class(obj=obj.name, ifc_class="IfcGrid")
|
|
grid = self.file.by_id(obj.BIMObjectProperties.ifc_definition_id)
|
|
if has_site_collection:
|
|
bpy.ops.bim.assign_container(relating_structure=grandchild.name, related_element=obj.name)
|
|
|
|
axes_collection = bpy.data.collections.new("UAxes")
|
|
collection.children.link(axes_collection)
|
|
for i in range(0, self.total_u):
|
|
verts = [
|
|
Vector((-2, i * self.u_spacing, 0)),
|
|
Vector((((self.total_v - 1) * self.v_spacing) + 2, i * self.u_spacing, 0)),
|
|
]
|
|
edges = [[0, 1]]
|
|
faces = []
|
|
mesh = bpy.data.meshes.new(name="Grid Axis")
|
|
mesh.from_pydata(verts, edges, faces)
|
|
tag = chr(ord("A") + i)
|
|
obj = bpy.data.objects.new(f"IfcGridAxis/{tag}", mesh)
|
|
|
|
axes_collection.objects.link(obj)
|
|
|
|
if self.file:
|
|
result = create_grid_axis.Usecase(
|
|
self.file, {"AxisTag": tag, "AxisCurve": obj, "UVWAxes": "UAxes", "Grid": grid}
|
|
).execute()
|
|
obj.BIMObjectProperties.ifc_definition_id = result.id()
|
|
|
|
axes_collection = bpy.data.collections.new("VAxes")
|
|
collection.children.link(axes_collection)
|
|
for i in range(0, self.total_v):
|
|
verts = [
|
|
Vector((i * self.v_spacing, -2, 0)),
|
|
Vector((i * self.v_spacing, ((self.total_u - 1) * self.u_spacing) + 2, 0)),
|
|
]
|
|
edges = [[0, 1]]
|
|
faces = []
|
|
mesh = bpy.data.meshes.new(name="Grid Axis")
|
|
mesh.from_pydata(verts, edges, faces)
|
|
tag = str(i + 1).zfill(2)
|
|
obj = bpy.data.objects.new(f"IfcGridAxis/{tag}", mesh)
|
|
|
|
axes_collection.objects.link(obj)
|
|
|
|
if IfcStore.get_file():
|
|
result = create_grid_axis.Usecase(
|
|
self.file, {"AxisTag": tag, "AxisCurve": obj, "UVWAxes": "VAxes", "Grid": grid}
|
|
).execute()
|
|
obj.BIMObjectProperties.ifc_definition_id = result.id()
|
|
|
|
|
|
class BIM_OT_add_object(Operator):
|
|
bl_idname = "mesh.add_grid"
|
|
bl_label = "Grid"
|
|
bl_options = {"REGISTER", "UNDO"}
|
|
|
|
u_spacing: FloatProperty(name="U Spacing", default=10)
|
|
total_u: IntProperty(name="Number of U Grids", default=3)
|
|
v_spacing: FloatProperty(name="V Spacing", default=10)
|
|
total_v: IntProperty(name="Number of V Grids", default=3)
|
|
|
|
def execute(self, context):
|
|
add_object(self, context)
|
|
return {"FINISHED"}
|
|
|
|
|
|
def add_object_button(self, context):
|
|
self.layout.operator(BIM_OT_add_object.bl_idname, icon="PLUGIN")
|