Method patch

src/ifcpatch/ifcpatch/recipes/FixRevitTINs.py:80–146

(self)

        self.is_solid = is_solid

    def patch(self) -> None:
        from math import degrees

        import bmesh  # pyright: ignore[reportMissingImports]  # ty:ignore[unresolved-import]
        import bonsai.tool as tool
        import bpy  # pyright: ignore[reportMissingImports]  # ty:ignore[unresolved-import]

        props = tool.Project.get_project_props()
        props.should_use_native_meshes = True
        bpy.ops.bim.load_project(filepath=self.filepath)

        old_history_size = tool.Ifc.get().history_size
        old_undo_steps = bpy.context.preferences.edit.undo_steps
        tool.Ifc.get().history_size = 0
        bpy.context.preferences.edit.undo_steps = 0

        angle_threshold = 0.3

        for obj in bpy.data.objects:
            ifc_id = tool.Blender.get_ifc_definition_id(obj)
            if not ifc_id or not obj.data:
                continue
            if not obj.data.polygons:
                continue
            data = obj.data
            bm = bmesh.new()
            bm.from_mesh(data)

            bm.faces.ensure_lookup_table()
            faces_to_delete = []

            if self.is_solid:
                for face in bm.faces:
                    if face.normal.z < 0.5:
                        faces_to_delete.append(face)

            bmesh.ops.delete(bm, geom=faces_to_delete, context="FACES_ONLY")

            bmesh.ops.remove_doubles(bm, verts=bm.verts, dist=0.01)
            bmesh.ops.triangulate(bm, faces=bm.faces[:], quad_method="BEAUTY", ngon_method="BEAUTY")
            bm.faces.ensure_lookup_table()
            for polygon in bm.faces:
                try:
                    v1, v2, v3 = [v.co.to_2d() for v in polygon.verts]
                    d1 = degrees((v2 - v1).angle(v3 - v1))
                    d2 = degrees((v3 - v2).angle(v1 - v2))
                    d3 = degrees((v1 - v3).angle(v2 - v3))
                    if d1 < angle_threshold or d2 < angle_threshold or d3 < angle_threshold:
                        bm.faces.remove(polygon)
                except:
                    bm.faces.remove(polygon)
            if bm.faces:
                if not self.is_solid:
                    bmesh.ops.recalc_face_normals(bm, faces=bm.faces)
                    for face in bm.faces:
                        global_normal = obj.matrix_world.to_3x3() @ face.normal
                        if global_normal.z < 0:
                            face.normal_flip()