IfcOpenShell/src/ifcblenderexport/blenderbim/bim/module/model/door.py

import bpy
import math
import ifcopenshell
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):
    guid = ifcopenshell.guid.new()
    leaf_width = self.overall_width-0.045-0.045
    verts = [
        # Left lining
        Vector((0, 0, 0)),
        Vector((0, self.depth, 0)),
        Vector((.04, self.depth, 0)),
        Vector((.04, self.depth-.04, 0)),
        Vector((.065, self.depth-.04, 0)),
        Vector((.065, 0, 0)),
        # Right lining
        Vector((self.overall_width, 0, 0)),
        Vector((self.overall_width, self.depth, 0)),
        Vector((self.overall_width-.04, self.depth, 0)),
        Vector((self.overall_width-.04, self.depth-.04, 0)),
        Vector((self.overall_width-.065, self.depth-.04, 0)),
        Vector((self.overall_width-.065, 0, 0)),
        # Door panel
        Vector((.045, self.depth, 0)),
        Vector((.045, self.depth+leaf_width, 0)),
        Vector((.080, self.depth+leaf_width, 0)),
        Vector((.080, self.depth, 0)),
    ]
    edges = [
        [0, 1], [1, 2], [2, 3], [3, 4], [4, 5], # Left lining
        [6, 7], [7, 8], [8, 9], [9, 10], [10, 11], # Right lining
        [12, 13], [13, 14], [14, 15], [15, 12], # Door panel
    ]
    # Door swing
    for i in range(0, 9):
        verts.append(Vector((0.045+(leaf_width*math.cos((math.pi/2)/8*i)), self.depth+(leaf_width*math.sin((math.pi/2)/8*i)), 0)))
        edges.append([16+i, 17+i])
    edges.pop()
    faces = []
    mesh = bpy.data.meshes.new(name='Plan/Annotation/PLAN_VIEW/' + guid)
    mesh.use_fake_user = True
    mesh.from_pydata(verts, edges, faces)

    # Door lining profile
    verts = [
        Vector((0, 0, 0)),
        Vector((0, -self.depth, 0)),
        Vector((.04, -self.depth, 0)),
        Vector((.04, -self.depth+.04, 0)),
        Vector((.065, -self.depth+.04, 0)),
        Vector((.065, 0, 0)),
    ]
    edges = []
    faces = [[0, 1, 2, 3, 4, 5]]
    mesh = bpy.data.meshes.new(name="Dumb Door Profile")
    mesh.from_pydata(verts, edges, faces)
    obj = object_data_add(context, mesh, operator=self)
    bpy.ops.object.convert(target='CURVE')

    # Door 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='Dumb Door')
    mesh.from_pydata(verts, edges, faces)
    obj2 = object_data_add(context, mesh, operator=self)
    bpy.ops.object.convert(target='CURVE')

    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)

    # Door panel
    verts = [
        Vector((.045, self.depth, 0)),
        Vector((.045, self.depth-.035, 0)),
        Vector((self.overall_width-.045, self.depth-.035, 0)),
        Vector((self.overall_width-.045, self.depth, 0)),
    ]
    edges = []
    faces = [[0, 1, 2, 3]]
    mesh = bpy.data.meshes.new(name="Dumb Door 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-.045
    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)

    # Door Opening
    verts = [
        Vector((0, -.1, -.1)),
        Vector((0, self.depth+.1, -.1)),
        Vector((self.overall_width, self.depth+.1, -.1)),
        Vector((self.overall_width, -.1, -.1)),
    ]
    edges = []
    faces = [[0, 1, 2, 3]]
    mesh = bpy.data.meshes.new(name="Dumb Door 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+.1
    bpy.ops.object.convert(target='MESH')
    obj4.display_type = 'WIRE'
    obj4.parent = obj2
    obj4.matrix_parent_inverse = obj2.matrix_world.inverted()
    obj4.hide_render = True
    obj4.name = 'IfcOpeningElement/Dumb Door Opening'
    attribute = obj4.BIMObjectProperties.attributes.add()
    attribute.name = 'PredefinedType'
    attribute.string_value = 'OPENING'

    obj2.name = 'IfcDoor/Dumb Door'
    attribute = obj2.BIMObjectProperties.attributes.add()
    attribute.name = 'PredefinedType'
    attribute.string_value = 'DOOR'
    obj2.data.name = 'Model/Body/MODEL_VIEW/' + guid
    obj2.data.use_fake_user = True

    rep = obj2.BIMObjectProperties.representation_contexts.add()
    rep.context = 'Model'
    rep.name = 'Body'
    rep.target_view = 'MODEL_VIEW'

    rep = obj2.BIMObjectProperties.representation_contexts.add()
    rep.context = 'Plan'
    rep.name = 'Annotation'
    rep.target_view = 'PLAN_VIEW'


class BIM_OT_add_object(Operator, AddObjectHelper):
    bl_idname = "mesh.add_door"
    bl_label = "Dumb Door"
    bl_options = {'REGISTER', 'UNDO'}

    overall_width: FloatProperty(name='Overall Width', default=.85)
    overall_height: FloatProperty(name='Overall Height', default=2.1)
    depth: FloatProperty(name='Depth', default=.2)

    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')