diff --git a/src/ifcblenderexport/blenderbim/bim/__init__.py b/src/ifcblenderexport/blenderbim/bim/__init__.py index 3d3bb92abd..429a4de4a6 100644 --- a/src/ifcblenderexport/blenderbim/bim/__init__.py +++ b/src/ifcblenderexport/blenderbim/bim/__init__.py @@ -14,6 +14,7 @@ if bpy is not None: from .module.model import wall as model_wall from .module.model import stair as model_stair from .module.model import door as model_door + from .module.model import window as model_window from .module.model import slab as model_slab classes = ( @@ -274,6 +275,7 @@ if bpy is not None: model_wall.BIM_OT_add_object, model_stair.BIM_OT_add_object, model_door.BIM_OT_add_object, + model_window.BIM_OT_add_object, model_slab.BIM_OT_add_object, ) @@ -313,6 +315,7 @@ if bpy is not None: bpy.types.VIEW3D_MT_mesh_add.append(model_wall.add_object_button) bpy.types.VIEW3D_MT_mesh_add.append(model_stair.add_object_button) bpy.types.VIEW3D_MT_mesh_add.append(model_door.add_object_button) + bpy.types.VIEW3D_MT_mesh_add.append(model_window.add_object_button) bpy.types.VIEW3D_MT_mesh_add.append(model_slab.add_object_button) def unregister(): @@ -336,4 +339,5 @@ if bpy is not None: bpy.types.VIEW3D_MT_mesh_add.remove(model_wall.add_object_button) bpy.types.VIEW3D_MT_mesh_add.remove(model_stair.add_object_button) bpy.types.VIEW3D_MT_mesh_add.remove(model_door.add_object_button) + bpy.types.VIEW3D_MT_mesh_add.remove(model_window.add_object_button) bpy.types.VIEW3D_MT_mesh_add.remove(model_slab.add_object_button) diff --git a/src/ifcblenderexport/blenderbim/bim/module/model/window.py b/src/ifcblenderexport/blenderbim/bim/module/model/window.py new file mode 100644 index 0000000000..3e81e94c99 --- /dev/null +++ b/src/ifcblenderexport/blenderbim/bim/module/model/window.py @@ -0,0 +1,114 @@ +import bpy +import math +from bpy.types import Operator +from bpy.props import FloatVectorProperty, FloatProperty, IntProperty +from bpy_extras.object_utils import AddObjectHelper, object_data_add +from mathutils import Vector + +def add_object(self, context): + # Window lining profile + verts = [ + Vector((0, 0, 0)), + Vector((0, -self.depth, 0)), + Vector((.04, -self.depth, 0)), + Vector((.04, 0, 0)), + ] + edges = [] + faces = [[0, 1, 2, 3]] + mesh = bpy.data.meshes.new(name="Dumb Window Profile") + mesh.from_pydata(verts, edges, faces) + obj = object_data_add(context, mesh, operator=self) + bpy.ops.object.convert(target='CURVE') + + # Window lining sweep + verts = [ + Vector((0, 0, 0)), + Vector((0, self.overall_height, 0)), + Vector((self.overall_width, self.overall_height, 0)), + Vector((self.overall_width, 0, 0)), + ] + edges = [[0, 1], [1, 2], [2, 3]] + faces = [] + mesh = bpy.data.meshes.new(name="IfcWindow/Dumb Window") + mesh.from_pydata(verts, edges, faces) + obj2 = object_data_add(context, mesh, operator=self) + bpy.ops.object.convert(target='CURVE') + obj2.data.splines[0].use_cyclic_u = True + + obj2.data.dimensions = '2D' + obj2.data.bevel_object = obj + + obj2.rotation_euler[0] = math.pi/2 + bpy.ops.object.convert(target='MESH') + bpy.ops.object.transform_apply(location=False) + bpy.data.objects.remove(obj, do_unlink=True) + + # Window panel + verts = [ + Vector((.04, (self.depth/2)+.005, .04)), + Vector((.04, (self.depth/2)-.005, .04)), + Vector((self.overall_width-.04, (self.depth/2)-.005, .04)), + Vector((self.overall_width-.04, (self.depth/2)+.005, .04)), + ] + edges = [] + faces = [[0, 1, 2, 3]] + mesh = bpy.data.meshes.new(name="Dumb Window Panel") + mesh.from_pydata(verts, edges, faces) + obj3 = object_data_add(context, mesh, operator=self) + modifier = obj3.modifiers.new('Panel Height', 'SOLIDIFY') + modifier.offset = 1 + modifier.thickness = self.overall_height-.08 + bpy.ops.object.convert(target='MESH') + + ctx = bpy.context.copy() + ctx['active_object'] = obj2 + ctx['selected_editable_objects'] = [obj2, obj3] + bpy.ops.object.join(ctx) + + # Window Opening + verts = [ + Vector((0, -.1, 0)), + Vector((0, self.depth+.1, 0)), + Vector((self.overall_width, self.depth+.1, 0)), + Vector((self.overall_width, -.1, 0)), + ] + edges = [] + faces = [[0, 1, 2, 3]] + mesh = bpy.data.meshes.new(name="Dumb Window Opening") + mesh.from_pydata(verts, edges, faces) + obj4 = object_data_add(context, mesh, operator=self) + modifier = obj4.modifiers.new('Panel Height', 'SOLIDIFY') + modifier.offset = -1 + modifier.thickness = self.overall_height + bpy.ops.object.convert(target='MESH') + obj4.display_type = 'WIRE' + obj4.parent = obj2 + obj4.matrix_parent_inverse = obj2.matrix_world.inverted() + obj4.name = 'IfcOpeningElement/Dumb Window Opening' + attribute = obj4.BIMObjectProperties.attributes.add() + attribute.name = 'PredefinedType' + attribute.string_value = 'OPENING' + + attribute = obj2.BIMObjectProperties.attributes.add() + attribute.name = 'PredefinedType' + attribute.string_value = 'WINDOW' + + +class BIM_OT_add_object(Operator, AddObjectHelper): + bl_idname = "mesh.add_window" + bl_label = "Dumb Window" + bl_options = {'REGISTER', 'UNDO'} + + overall_width: FloatProperty(name='Overall Width', default=.7) + overall_height: FloatProperty(name='Overall Height', default=1) + depth: FloatProperty(name='Depth', default=.1) + + 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')