Files
IfcOpenShell/src/ifcopenshell-python/ifcopenshell/util/shape_builder.py
T

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

2048 lines
85 KiB
Python
Raw Normal View History

2023-01-08 13:32:13 +05:00
# IfcOpenShell - IFC toolkit and geometry engine
2023-08-28 15:25:10 +05:00
# Copyright (C) 2022, 2023 @Andrej730
2023-01-08 13:32:13 +05:00
#
# This file is part of IfcOpenShell.
#
# IfcOpenShell is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# IfcOpenShell is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with IfcOpenShell. If not, see <http://www.gnu.org/licenses/>.
from __future__ import annotations
2025-12-19 18:53:04 +05:00
2024-04-25 11:24:42 +05:00
import collections.abc
2025-12-19 18:53:04 +05:00
from collections.abc import Sequence
from itertools import chain
from math import atan, cos, degrees, pi, radians, sin, sqrt, tan
from typing import TYPE_CHECKING, Any, Literal, Optional, Union
import numpy as np
import numpy.typing as npt
2023-01-08 13:32:13 +05:00
import ifcopenshell
2024-04-25 11:24:42 +05:00
import ifcopenshell.util.element
2024-12-16 17:42:35 +05:00
import ifcopenshell.util.placement
2024-04-25 11:24:42 +05:00
import ifcopenshell.util.representation
import ifcopenshell.util.unit
2023-01-08 13:32:13 +05:00
2023-08-21 17:22:20 +05:00
PRECISION = 1.0e-5
2023-08-28 15:25:10 +05:00
if TYPE_CHECKING:
# NOTE: mathutils is never used at runtime in ifcopenshell,
# only for type checking to ensure methods are compatible with
# Blender vectors.
from mathutils import Vector
# Support both numpy arrays and python sequences as inputs.
VectorType = Union[Sequence[float], Vector, np.ndarray]
else:
# Ensure it's exportable, so other modules can reuse it for typing.
2025-02-10 17:26:03 +11:00
VectorType = Any
SequenceOfVectors = Union[Sequence[VectorType], np.ndarray]
2023-08-28 15:25:10 +05:00
def V(*args: Union[float, int, VectorType, SequenceOfVectors]) -> npt.NDArray[np.float64]:
"""Convert floats / vector / sequence of vectors to numpy array.
Note that `float` argument type also allows passing ints,
which will be converted to floats (a double type) as IfcOpenShell is strict
about setting int/float attributes.
"""
if isinstance(args[0], (float, int)):
return np.array(args, dtype="d")
assert len(args) == 1, "Only single argument is supported if providing a vector or a sequence of them."
return np.array(args[0], dtype="d")
def ifc_safe_vector_type(v: Union[VectorType, SequenceOfVectors]) -> Any:
"""Convert vector / sequence of vectors to a list of floats
that's safe to save IFC attribute.
Basically converting all numbers in sequences to Python floats.
"""
return np.array(v, dtype="d").tolist()
def is_x(value: float, x: float, si_conversion: Optional[float] = None) -> bool:
if si_conversion is not None:
2023-08-28 15:25:10 +05:00
value = value * si_conversion
return (x + PRECISION) > value > (x - PRECISION)
def round_to_precision(x: float, si_conversion: float) -> float:
return round(x * si_conversion, 5) / si_conversion
def np_round_to_precision(v: np.ndarray, si_conversion: float) -> np.ndarray:
return np.round(v * si_conversion, 5) / si_conversion
2023-08-21 17:22:20 +05:00
2023-01-08 13:32:13 +05:00
def np_normalized(v: VectorType) -> np.ndarray:
return np.divide(v, np.linalg.norm(v))
2025-01-13 13:22:04 +05:00
def np_matrix_normalized(matrix: np.ndarray) -> np.ndarray:
# Ensure translation is not affected.
scale_factors = np.linalg.norm(matrix[:3, :3], axis=0)
rotation_matrix = matrix.copy()
rotation_matrix[:3, :3] /= scale_factors
return rotation_matrix
2024-12-18 12:40:11 +05:00
def np_lerp(a: VectorType, b: VectorType, t: float) -> np.ndarray:
return a + np.subtract(b, a) * t
2024-12-16 17:42:35 +05:00
def np_to_3d(v: VectorType, z: float = 0.0) -> np.ndarray:
"""Convert 2D/4D vector to 3D."""
l = len(v)
if l == 2:
return np.append(v, z)
elif l == 4:
return v[:3]
assert False, f"Unexpected vector length: {l} ({v})."
def np_to_4d(v: VectorType, z: float = 0.0, w: float = 1.0) -> np.ndarray:
"""Convert 2D/3D vector to 4D (e.g. for multiplying with 4x4 matrix)."""
l = len(v)
if l == 2:
return np.append(v, (z, w))
elif l == 3:
return np.append(v, w)
assert False, f"Unexpected vector length: {l} ({v})."
2024-12-16 17:28:54 +05:00
def np_to_4x4(matrix_3x3: np.ndarray) -> np.ndarray:
"""Convert 3x3 matrix to 4x4."""
matrix_4x4 = np.pad(matrix_3x3, ((0, 1), (0, 1)))
matrix_4x4[3, 3] = 1
return matrix_4x4
def np_apply_matrix(vectors: SequenceOfVectors, matrix: npt.NDArray) -> npt.NDArray:
"""
:param vectors: Nx3 array of vectors.
:param matrix: 4x4 transformation matrix.
"""
m3x3 = matrix[:3, :3]
translation = matrix[:3, 3]
return vectors @ m3x3.T + translation
2024-12-16 17:42:35 +05:00
def np_angle(a: VectorType, b: VectorType) -> float:
"""Get angle between vectors in radians.
Designed to work similar to `Vector.angle`.
"""
return np.arccos(np.dot(a, b) / (np.linalg.norm(a) * np.linalg.norm(b)))
2024-12-18 12:40:11 +05:00
def np_angle_signed(a: VectorType, b: VectorType) -> float:
"""Get signed angle between 2D vectors in radians (clockwise is positive).
Designed to work similar to `Vector.angle_signed`.
"""
assert len(a) == 2 and len(b) == 2, "Only 2D vectors are supported."
det = a[1] * b[0] - a[0] * b[1]
dot = np.dot(a, b)
return np.arctan2(det, dot)
def np_translation_matrix(vector: VectorType) -> npt.NDArray[np.float64]:
"""Get translation matrix.
Designed to be similar to mathutils Matrix.Rotation but to use numpy.
:param vector: 3D translation vector.
:return: An 4x4 identity matrix with a translation
"""
eye = np.eye(4, dtype=np.float64)
M_TRANSLATION = (slice(0, 3), 3)
eye[M_TRANSLATION] = vector
return eye
2024-12-16 17:28:54 +05:00
def np_rotation_matrix(
angle: float, size: int, axis: Optional[Union[Literal["X", "Y", "Z"], VectorType]] = None
) -> np.ndarray:
"""Get rotation matrix. Designed to be similar to mathutils Matrix.Rotation but to use numpy.
:param float: Rotation angle, in radians.
:param size: Matrix size ([2;4]).
:param axis: Rotation axis.
For 2x2 matrices Z assumed by default and argument can be omitted,
for 3x3/4x4 matrices could be either axis literal
or a rotation axis presented as a vector.
:return: Rotation matrix.
"""
if not (2 <= size <= 4):
raise ValueError(f"Size must be [2;4], got {size}.")
cos_theta: float = np.cos(angle)
sin_theta: float = np.sin(angle)
if size == 2:
return np.array([[cos_theta, -sin_theta], [sin_theta, cos_theta]])
assert axis, "For non-2D matrices 'axis' argument is not optional."
if isinstance(axis, str):
if axis == "X":
matrix = np.array([[1, 0, 0], [0, cos_theta, -sin_theta], [0, sin_theta, cos_theta]])
elif axis == "Y":
matrix = np.array([[cos_theta, 0, sin_theta], [0, 1, 0], [-sin_theta, 0, cos_theta]])
elif axis == "Z":
matrix = np.array([[cos_theta, -sin_theta, 0], [sin_theta, cos_theta, 0], [0, 0, 1]])
else:
# Assume axis is a vector.
axis = axis / np.linalg.norm(axis)
# Rodrigues' rotation formula.
K = np.array([[0, -axis[2], axis[1]], [axis[2], 0, -axis[0]], [-axis[1], axis[0], 0]])
matrix = cos_theta * np.eye(3) + (1 - cos_theta) * np.outer(axis, axis) + sin_theta * K
if size == 4:
return np_to_4x4(matrix)
return matrix
2025-01-13 13:22:04 +05:00
def np_matrix_to_euler(matrix: np.ndarray) -> tuple[float, float, float]:
"""Convert a rotation matrix to Euler angles.
Designed to work similar to `mathutils.Matrix.to_euler`.
Currently only XYZ rotation is supported.
"""
if matrix.shape not in ((3, 3), (4, 4)):
raise ValueError(f"Matrix must be 3x3 or 4x4, got {matrix.shape}.")
matrix = np_matrix_normalized(matrix)
y = -np.arcsin(matrix[2, 0])
cos_y = np.cos(y)
x = np.arctan2(matrix[2, 1] / cos_y, matrix[2, 2] / cos_y)
z = np.arctan2(matrix[1, 0] / cos_y, matrix[0, 0] / cos_y)
return (x, y, z)
2024-12-18 12:40:11 +05:00
def np_normal(vectors: SequenceOfVectors) -> np.ndarray:
"""Normal of 3D Polygon.
Designed to work similar to `mathutils.geometry.normal`.
"""
assert len(vectors) == 3, "3 vectors required"
# TODO: can be optimized?
verts_np = np.array(vectors[:3])
v0, v1, v2 = verts_np[:3]
edge1 = v1 - v0
edge2 = v2 - v0
normal = np.cross(edge1, edge2)
norm = np.linalg.norm(normal)
return normal / norm
def np_intersect_line_line(
v1: VectorType, v2: VectorType, v3: VectorType, v4: VectorType
) -> tuple[np.ndarray, np.ndarray]:
"""Get 2 closest points on each line.
First line - (v1, v2). Second line - (v3, v4).
Designed to work similar to `mathutils.geometry.intersect_line_line`.
"""
# TODO: could be optimized?
d1 = np.subtract(v2, v1)
d2 = np.subtract(v4, v3)
# Cross product of the directions
cross_d1_d2 = np.cross(d1, d2)
cross_d1_d2_norm: float = np.linalg.norm(cross_d1_d2)
# Check if the lines are parallel.
if is_x(cross_d1_d2_norm, 0):
raise ValueError("Lines are parallel and do not intersect uniquely.")
r = np.subtract(v3, v1)
t = np.dot(np.cross(r, d2), cross_d1_d2) / (cross_d1_d2_norm**2)
u = np.dot(np.cross(r, d1), cross_d1_d2) / (cross_d1_d2_norm**2)
# Closest points on each line
point_on_line1 = v1 + t * d1
point_on_line2 = v3 + u * d2
return point_on_line1, point_on_line2
def intersect_x_axis_2d(p1: VectorType, p2: VectorType, y=0) -> Optional[float]:
"""Intersect a line defined by 2 points to a horizontal line defined by y
Useful for axis-aligned intersection checks.
:param p1: First 2D point of the line, order doesn't matter
:param p2: Second 2D point of the line, order doesn't matter
:param y: Intersect at this y value (i.e. defaults to y=0)
"""
x1, y1 = p1
x2, y2 = p2
if is_x(y1, y2): # Parallel
return
t = (y - y1) / (y2 - y1)
return x1 + t * (x2 - x1)
2023-01-08 13:32:13 +05:00
# Note: using ShapeBuilder try not to reuse IFC elements in the process
# otherwise you might run into situation where builder.mirror or other operation
# is applied twice during one run to the same element
# which might produce undesirable results
2023-01-25 10:49:03 +05:00
2023-01-08 13:32:13 +05:00
class ShapeBuilder:
2024-01-24 11:47:19 +05:00
def __init__(self, ifc_file: ifcopenshell.file):
self.file = ifc_file
2023-01-08 13:32:13 +05:00
def polyline(
2024-05-15 10:55:29 +05:00
self,
points: SequenceOfVectors,
2024-05-15 10:55:29 +05:00
closed: bool = False,
position_offset: Optional[VectorType] = None,
arc_points: Sequence[int] = (),
2024-01-24 16:14:26 +05:00
) -> ifcopenshell.entity_instance:
"""
Generate an IfcIndexedPolyCurve based on the provided points.
2024-01-24 16:14:26 +05:00
:param points: List of 2d or 3d points
:param closed: Whether polyline should be closed.
2024-01-24 16:14:26 +05:00
:param position_offset: offset to be applied to all points
:param arc_points: Indices of the middle points for arcs. For creating an arc segment,
2024-01-25 16:24:43 +05:00
provide 3 points: `arc_start`, `arc_middle` and `arc_end` to `points` and add the `arc_middle`
point's index to `arc_points`
:return: IfcIndexedPolyCurve
2024-01-25 16:24:43 +05:00
Example:
.. code:: python
# rectangle
points = Vector((0, 0)), Vector((1, 0)), Vector((1, 1)), Vector((0, 1))
position = Vector((2, 0))
# #2=IfcIndexedPolyCurve(#1,(IfcLineIndex((1,2,3,4,1))),$)
polyline = builder.polyline(points, closed=True, position_offset=position)
# arc between points (1,0) and (0,1). Second point in the arc should be it's middle
points = Vector((1, 0)), Vector((0.707, 0.707)), Vector((0, 1)), Vector((0,2))
arc_points = (1,) # point with index 1 is a middle of the arc
# 4=IfcIndexedPolyCurve(#3,(IfcArcIndex((1,2,3)),IfcLineIndex((3,4,1))),$)
curved_polyline = builder.polyline(points, closed=False, position_offset=position, arc_points=arc_points)
"""
if arc_points and self.file.schema == "IFC2X3":
raise Exception("Arcs are not supported for IFC2X3.")
points: np.ndarray
points = np.array(points)
if position_offset is not None:
points = points + position_offset
if self.file.schema == "IFC2X3":
ifc_points = [self.file.create_entity("IfcCartesianPoint", p) for p in points.tolist()]
if closed:
ifc_points.append(ifc_points[0])
ifc_curve = self.file.createIfcPolyline(Points=ifc_points)
return ifc_curve
dimensions = len(points[0])
if dimensions == 2:
ifc_points = self.file.create_entity("IfcCartesianPointList2D", points.tolist())
elif dimensions == 3:
ifc_points = self.file.create_entity("IfcCartesianPointList3D", points.tolist())
else:
raise Exception(f"Point has unexpected number of dimensions - {dimensions}.")
if not closed and not arc_points:
ifc_curve = self.file.createIfcIndexedPolyCurve(Points=ifc_points)
return ifc_curve
# if curve is closed or we have arc points
# then we do need to create segments
segments = []
cur_i = 0
2024-07-11 18:52:18 +05:00
closed_by_arc = False
while cur_i < len(points) - 1:
cur_i_ifc = cur_i + 1
if cur_i + 1 in arc_points:
2024-07-11 18:52:18 +05:00
if cur_i_ifc + 1 < len(points):
segments.append((cur_i_ifc, cur_i_ifc + 1, cur_i_ifc + 2))
else:
segments.append((cur_i_ifc, cur_i_ifc + 1, 1))
closed_by_arc = True
cur_i += 2
else:
segments.append((cur_i_ifc, cur_i_ifc + 1))
cur_i += 1
2024-07-11 18:52:18 +05:00
if closed and not closed_by_arc:
segments.append((len(points), 1))
ifc_segments = []
# because IfcLineIndex support 2+ points
# we merge neighbor line segments into one
current_line_segment = []
last_segment = len(segments) - 1
for seg_i, segment in enumerate(segments):
if len(segment) == 2:
# check if `current_line_segment` is empty to avoid duplicated indices like `IfcLineIndex((1,2,2,3,3,4,4,1))`
current_line_segment += segment if not current_line_segment else segment[1:]
if current_line_segment and (len(segment) == 3 or seg_i == last_segment):
ifc_segments.append(self.file.createIfcLineIndex(current_line_segment))
current_line_segment = []
if len(segment) == 3:
ifc_segments.append(self.file.createIfcArcIndex(segment))
2023-08-28 15:25:10 +05:00
# NOTE: IfcIndexPolyCurve support only consecutive segments
ifc_curve = self.file.createIfcIndexedPolyCurve(Points=ifc_points, Segments=ifc_segments)
2023-01-08 13:32:13 +05:00
return ifc_curve
2025-11-06 17:49:12 +05:00
@staticmethod
def get_rectangle_coords(size: VectorType = (1.0, 1.0), position: Optional[VectorType] = None) -> np.ndarray:
2024-01-25 16:35:28 +05:00
"""
Get rectangle coords arranged as below:
::
3 2
0 1
:param size: rectangle size, could be either 2d or 3d.
:param position: rectangle position.
2024-01-25 16:35:28 +05:00
if `position` not specified zero-vector will be used
:return: list of rectangle coords
"""
size_np = np.array(size)
if position is None:
dimensions = len(size_np)
points = np.full((4, dimensions), 0.0)
else:
points = np.tile(position, (4, 1))
# Support both 2d and 3d sizes defined in different dimensions.
non_empty_coords = np.nonzero(size_np)[0]
points[1, non_empty_coords[0]] += size_np[non_empty_coords[0]]
points[2] += size_np
points[3, non_empty_coords[1]] += size_np[non_empty_coords[1]]
2023-01-08 13:32:13 +05:00
return points
2024-01-24 16:14:26 +05:00
def rectangle(
self, size: VectorType = (1.0, 1.0), position: Optional[VectorType] = None
2024-01-24 16:14:26 +05:00
) -> ifcopenshell.entity_instance:
"""
2024-01-24 16:14:26 +05:00
Generate a rectangle polyline.
:param size: rectangle size, could be either 2d or 3d.
:param position: rectangle position.
2024-01-24 16:14:26 +05:00
if `position` not specified zero-vector will be used
2023-08-28 16:13:37 +05:00
:return: IfcIndexedPolyCurve
"""
2023-01-08 13:32:13 +05:00
return self.polyline(self.get_rectangle_coords(size, position), closed=True)
def circle(self, center: VectorType = (0.0, 0.0), radius: float = 1.0) -> ifcopenshell.entity_instance:
2024-01-24 11:47:19 +05:00
"""
:param center: circle 2D position
:param radius: radius of the circle
2024-01-24 11:47:19 +05:00
:return: IfcCircle
"""
2024-09-11 15:05:18 +05:00
ifc_center = self.create_axis2_placement_2d(center)
ifc_curve = self.file.create_entity("IfcCircle", ifc_center, radius)
2023-01-08 13:32:13 +05:00
return ifc_curve
def plane(
self, location: VectorType = (0.0, 0.0, 0.0), normal: VectorType = (0.0, 0.0, 1.0)
) -> ifcopenshell.entity_instance:
"""
Create IfcPlane.
:param location: plane position.
:param normal: plane normal direction.
:return: IfcPlane
"""
if np.allclose(np.round(normal, 2), (0.0, 0.0, 1.0)):
arbitrary_vector = (0.0, 1.0, 0.0)
else:
arbitrary_vector = (0.0, 0.0, 1.0)
x_axis = np_normalized(np.cross(normal, arbitrary_vector))
axis_placement = self.create_axis2_placement_3d(location, normal, x_axis)
return self.file.createIfcPlane(axis_placement)
2023-01-08 13:32:13 +05:00
# TODO: explain points order for the curve_between_two_points
# because the order is important and defines the center of the curve
# currently it seems like the first point shifted by x-axis defines the center
def curve_between_two_points(self, points: tuple[VectorType, VectorType]) -> ifcopenshell.entity_instance:
2023-01-08 13:32:13 +05:00
"""Simple circle based curve between two points
Good for creating curves and fillets, won't work for continuous ellipse shapes.
:param points: tuple of 2 points.
:return: IfcIndexePolyCurve
2023-01-08 13:32:13 +05:00
"""
diff = np.subtract(points[1], points[0])
max_diff_i = np.argmax(np.abs(diff))
diff_sign = np.zeros_like(diff)
diff_sign[max_diff_i] = np.sign(diff[max_diff_i])
2023-01-08 13:32:13 +05:00
# diff should be applied only to one axis
# if it's applied to two (like in a case of circle) it will create
# a straight line instead of a curve
diff = (0.01, 0.01) * diff_sign
2023-01-08 13:32:13 +05:00
middle_point = points[0] + diff
points: list[VectorType]
2023-01-08 13:32:13 +05:00
points = [points[0], middle_point, points[1]]
points = [ifc_safe_vector_type(p) for p in points]
seg = self.file.createIfcArcIndex((1, 2, 3))
ifc_points = self.file.createIfcCartesianPointList2D(points)
curve = self.file.createIfcIndexedPolyCurve(Points=ifc_points, Segments=[seg])
2023-01-08 13:32:13 +05:00
return curve
2024-05-15 10:55:29 +05:00
def get_trim_points_from_mask(
self,
x_axis_radius: float,
y_axis_radius: float,
trim_points_mask: Sequence[int],
position_offset: Optional[VectorType] = None,
) -> np.ndarray:
2023-01-08 13:32:13 +05:00
"""Handy way to get edge points of the ellipse like shape of a given radiuses.
Mask points are numerated from 0 to 3 ccw starting from (x_axis_radius/2; 0).
Example: mask (0, 1, 2, 3) will return points (x, 0), (0, y), (-x, 0), (0, -y)
"""
points = np.array(
(
(x_axis_radius, 0),
(0, y_axis_radius),
(-x_axis_radius, 0),
(0, -y_axis_radius),
)
2023-01-08 13:32:13 +05:00
)
# list type is important for selecting items by the indices.
trim_points = points[list(trim_points_mask)]
if position_offset is None:
return trim_points
return trim_points + position_offset
2023-01-08 13:32:13 +05:00
def create_ellipse_curve(
self,
2024-05-15 10:55:29 +05:00
x_axis_radius: float,
y_axis_radius: float,
position: VectorType = (0.0, 0.0),
trim_points: SequenceOfVectors = (),
ref_x_direction: VectorType = (1.0, 0.0),
2024-06-04 13:50:00 +05:00
trim_points_mask: Sequence[int] = (),
2024-05-15 10:55:29 +05:00
) -> ifcopenshell.entity_instance:
2023-01-08 13:32:13 +05:00
"""
Ellipse trimming points should be specified in counter clockwise order.
2023-01-25 10:49:03 +05:00
2023-01-08 13:32:13 +05:00
For example, if you need to get the part of the ellipse ABOVE y-axis, you need to use mask (0,2). Below y-axis - (2,0)
For more information about trim_points_mask check builder.get_trim_points_from_mask
Notion: trimmed ellipse also contains polyline between trim points, meaning IfcTrimmedCurve could be used
for further extrusion.
"""
2024-09-18 22:58:13 +10:00
ifc_position = self.create_axis2_placement_2d(position, ref_x_direction)
ifc_ellipse = self.file.createIfcEllipse(
Position=ifc_position, SemiAxis1=x_axis_radius, SemiAxis2=y_axis_radius
2023-01-25 10:49:03 +05:00
)
2023-01-08 13:32:13 +05:00
if not trim_points:
if not trim_points_mask:
return ifc_ellipse
trim_points = self.get_trim_points_from_mask(
x_axis_radius, y_axis_radius, trim_points_mask, position_offset=position
2023-01-08 13:32:13 +05:00
)
trim1 = [self.file.create_entity("IfcCartesianPoint", ifc_safe_vector_type(trim_points[0]))]
trim2 = [self.file.create_entity("IfcCartesianPoint", ifc_safe_vector_type(trim_points[1]))]
2023-01-08 13:32:13 +05:00
trim_ellipse = self.file.createIfcTrimmedCurve(
2023-01-08 13:32:13 +05:00
BasisCurve=ifc_ellipse, Trim1=trim1, Trim2=trim2, SenseAgreement=True, MasterRepresentation="CARTESIAN"
)
return trim_ellipse
2024-05-15 10:55:29 +05:00
def profile(
self,
outer_curve: ifcopenshell.entity_instance,
name: Optional[str] = None,
2024-06-04 13:50:00 +05:00
inner_curves: Sequence[ifcopenshell.entity_instance] = (),
2024-05-15 10:55:29 +05:00
profile_type: str = "AREA",
) -> ifcopenshell.entity_instance:
"""Create a profile.
:param outer_curve: Profile IfcCurve.
:param inner_curves: a sequence of IfcCurves.
:return: IfcArbitraryClosedProfileDef or IfcArbitraryProfileDefWithVoids.
"""
2023-01-08 13:32:13 +05:00
# inner_curves could be used as a tool for boolean operation
# but if any point of inner curve will go outside the outer curve
# it will just add shape on top instead of "boolean" it
# because of that you can't create bool edges of outer_curve this way
if outer_curve.Dim != 2:
raise Exception(
f"Outer curve for IfcArbitraryClosedProfileDef/IfcIfcArbitraryProfileDefWithVoid should be 2D to be valid, currently it has {outer_curve.Dim} dimensions.\n"
"Ref: https://ifc43-docs.standards.buildingsmart.org/IFC/RELEASE/IFC4x3/HTML/lexical/IfcArbitraryClosedProfileDef.htm#8.15.3.1.4-Formal-propositions"
)
kwargs = {
"ProfileName": name,
"ProfileType": profile_type,
"OuterCurve": outer_curve,
}
2023-01-08 13:32:13 +05:00
if inner_curves:
if not isinstance(inner_curves, collections.abc.Iterable):
inner_curves = [inner_curves]
if any(curve.Dim != 2 for curve in inner_curves):
raise Exception(
"WARNING. InnerCurve for IfcIfcArbitraryProfileDefWithVoid sould be 2D to be valid, "
"currently on one of the inner curves is using different amount of dimensions.\n"
"Ref: https://ifc43-docs.standards.buildingsmart.org/IFC/RELEASE/IFC4x3/HTML/lexical/IfcArbitraryClosedProfileDef.htm#8.15.3.1.4-Formal-propositions"
)
2023-01-08 13:32:13 +05:00
profile = self.file.create_entity("IfcArbitraryProfileDefWithVoids", InnerCurves=inner_curves, **kwargs)
2023-01-08 13:32:13 +05:00
else:
profile = self.file.create_entity("IfcArbitraryClosedProfileDef", **kwargs)
2023-01-08 13:32:13 +05:00
return profile
2024-05-15 10:55:29 +05:00
def translate(
self,
2024-12-16 16:53:00 +05:00
curve_or_item: Union[ifcopenshell.entity_instance, Sequence[ifcopenshell.entity_instance]],
translation: VectorType,
2024-05-15 10:55:29 +05:00
create_copy: bool = False,
) -> Union[ifcopenshell.entity_instance, list[ifcopenshell.entity_instance]]:
2024-12-16 16:53:00 +05:00
"""
Translate curve/representaiton item/representation.
:param curve_or_item: A single item to translate or a sequence of them.
:param translation: Translation vector.
:param create_copy: Whether to translate the provided item or it's copy.
:return: Translated curve/item/representation or a sequence of them.
"""
2023-01-08 13:32:13 +05:00
multiple_objects = isinstance(curve_or_item, collections.abc.Iterable)
if not multiple_objects:
curve_or_item = [curve_or_item]
2024-12-16 16:53:00 +05:00
processed_objects: list[ifcopenshell.entity_instance] = []
2023-01-08 13:32:13 +05:00
for c in curve_or_item:
if create_copy:
c = ifcopenshell.util.element.copy_deep(self.file, c)
2023-01-08 13:32:13 +05:00
if c.is_a() in ("IfcIndexedPolyCurve", "IfcPolyline"):
coords = self.get_polyline_coords(c)
2024-12-16 16:53:00 +05:00
coords += translation
self.set_polyline_coords(c, coords)
2023-01-08 13:32:13 +05:00
elif c.is_a("IfcCircle") or c.is_a("IfcExtrudedAreaSolid") or c.is_a("IfcEllipse"):
2024-12-16 16:53:00 +05:00
base_position = np.array(c.Position.Location.Coordinates)
c.Position.Location.Coordinates = ifc_safe_vector_type(base_position + translation)
2023-01-08 13:32:13 +05:00
elif c.is_a("IfcTessellatedFaceSet"):
c.Coordinates.CoordList = ifc_safe_vector_type(np.array(c.Coordinates.CoordList) + translation)
2023-01-08 13:32:13 +05:00
elif c.is_a("IfcShapeRepresentation"):
for item in c.Items:
self.translate(item, translation)
elif c.is_a("IfcTrimmedCurve"):
2024-12-16 16:53:00 +05:00
base_position = np.array(c.Trim1[0].Coordinates)
c.Trim1[0].Coordinates = ifc_safe_vector_type(base_position + translation)
2023-01-08 13:32:13 +05:00
2024-12-16 16:53:00 +05:00
base_position = np.array(c.Trim2[0].Coordinates)
c.Trim2[0].Coordinates = ifc_safe_vector_type(base_position + translation)
2023-01-08 13:32:13 +05:00
self.translate(c.BasisCurve, translation)
else:
raise Exception(f"{c} is not supported for translate() method.")
processed_objects.append(c)
return processed_objects if multiple_objects else processed_objects[0]
2023-01-25 10:49:03 +05:00
def rotate_2d_point(
2024-12-16 17:28:54 +05:00
self,
point_2d: VectorType,
angle: float = 90.0,
pivot_point: VectorType = (0.0, 0.0),
counter_clockwise: bool = False,
) -> np.ndarray:
angle_rad = radians(angle) * (1 if counter_clockwise else -1)
relative_point = np.array(point_2d) - pivot_point
relative_point = np_rotation_matrix(angle_rad, 2) @ relative_point
final_point = relative_point + pivot_point
return final_point
2023-01-08 13:32:13 +05:00
def rotate(
self,
2024-12-16 17:28:54 +05:00
curve_or_item: Union[ifcopenshell.entity_instance, Sequence[ifcopenshell.entity_instance]],
angle: float = 90.0,
pivot_point: VectorType = (0.0, 0.0),
2024-05-15 10:55:29 +05:00
counter_clockwise: bool = False,
create_copy: bool = False,
) -> Union[ifcopenshell.entity_instance, list[ifcopenshell.entity_instance]]:
2024-12-16 17:28:54 +05:00
"""Rotate curve/representaiton item/representation.
:param curve_or_item: A single item to rotate or a sequence of them.
:param angle: Rotation angle, in degrees.
:param pivot_point: Rotation pivot point.
:param counter_clockwise: Whether rotation is counter-clockwise.
:param create_copy: Whether to rotate the provided item or it's copy.
:return: Rotated curve/representaiton item/representation or a sequence of them.
"""
2023-01-08 13:32:13 +05:00
multiple_objects = isinstance(curve_or_item, collections.abc.Iterable)
if not multiple_objects:
curve_or_item = [curve_or_item]
2024-12-16 17:28:54 +05:00
processed_objects: list[ifcopenshell.entity_instance] = []
2023-01-08 13:32:13 +05:00
for c in curve_or_item:
if create_copy:
c = ifcopenshell.util.element.copy_deep(self.file, c)
2023-01-08 13:32:13 +05:00
if c.is_a() in ("IfcIndexedPolyCurve", "IfcPolyline"):
original_coords = self.get_polyline_coords(c)
2024-12-16 17:28:54 +05:00
coords = [self.rotate_2d_point(co, angle, pivot_point, counter_clockwise) for co in original_coords]
self.set_polyline_coords(c, coords)
2023-01-08 13:32:13 +05:00
elif c.is_a("IfcCircle"):
2024-12-16 17:28:54 +05:00
base_position = c.Position.Location.Coordinates
2023-01-08 13:32:13 +05:00
new_position = self.rotate_2d_point(base_position, angle, pivot_point, counter_clockwise)
2024-12-16 17:28:54 +05:00
c.Position.Location.Coordinates = ifc_safe_vector_type(new_position)
2023-01-08 13:32:13 +05:00
elif c.is_a("IfcExtrudedAreaSolid"):
# TODO: add support for Z-axis too
2024-12-16 17:28:54 +05:00
base_position = c.Position.Location.Coordinates
new_position = self.rotate_2d_point(base_position[:2], angle, pivot_point, counter_clockwise)
new_position = np_to_3d(new_position)
new_position[2] = base_position[2]
c.Position.Location.Coordinates = ifc_safe_vector_type(new_position)
2023-01-08 13:32:13 +05:00
# TODO: add inner axis too and test it
self.rotate(c.SweptArea.OuterCurve, angle, pivot_point, counter_clockwise)
else:
raise Exception(f"{c} is not supported for rotate() method.")
processed_objects.append(c)
return processed_objects if multiple_objects else processed_objects[0]
def mirror_2d_point(
self,
2024-12-16 17:42:35 +05:00
point_2d: VectorType,
mirror_axes: VectorType = (1.0, 1.0),
mirror_point: VectorType = (0.0, 0.0),
) -> np.ndarray:
"""mirror_axes - along which axes mirror will be applied"""
2024-12-16 17:42:35 +05:00
mirror_axes: np.ndarray = np.where(np.array(mirror_axes) > 0, -1, 1)
mirror_point: np.ndarray = np.array(mirror_point)
relative_point = point_2d - mirror_point
2023-01-08 13:32:13 +05:00
relative_point = relative_point * mirror_axes
2024-12-16 17:42:35 +05:00
point_2d_res = relative_point + mirror_point
return point_2d_res
2023-01-08 13:32:13 +05:00
def create_axis2_placement_3d(
self,
position: VectorType = (0.0, 0.0, 0.0),
z_axis: VectorType = (0.0, 0.0, 1.0),
x_axis: VectorType = (1.0, 0.0, 0.0),
) -> ifcopenshell.entity_instance:
"""
Create IfcAxis2Placement3D.
:param position: placement position (Axis).
:param z_axis: local Z axis direction.
:param x_axis: local X axis direction (RefDirection).
:return: IfcAxis2Placement3D
"""
return self.file.create_entity(
"IfcAxis2Placement3D",
self.file.create_entity("IfcCartesianPoint", ifc_safe_vector_type(position)),
Axis=self.file.create_entity("IfcDirection", ifc_safe_vector_type(z_axis)),
RefDirection=self.file.create_entity("IfcDirection", ifc_safe_vector_type(x_axis)),
)
def create_axis2_placement_3d_from_matrix(
self,
2024-04-25 15:55:32 +05:00
matrix: Union[npt.NDArray[np.float64], None] = None,
) -> ifcopenshell.entity_instance:
"""
Create IfcAxis2Placement3D from numpy matrix.
:param matrix: 4x4 transformation matrix, defaults to ``np.eye(4)``
:return: IfcAxis2Placement3D
"""
if matrix is None:
matrix = np.eye(4, dtype=float)
return self.create_axis2_placement_3d(position=matrix[:3, 3], z_axis=matrix[:3, 2], x_axis=matrix[:3, 0])
2024-09-11 15:05:18 +05:00
def create_axis2_placement_2d(
self, position: VectorType = (0.0, 0.0), x_direction: Optional[VectorType] = None
2024-09-11 15:05:18 +05:00
) -> ifcopenshell.entity_instance:
"""Create IfcAxis2Placement2D."""
ref_direction = (
self.file.create_entity("IfcDirection", ifc_safe_vector_type(x_direction)) if x_direction else None
)
2024-09-11 15:05:18 +05:00
return self.file.create_entity(
"IfcAxis2Placement2D",
Location=self.file.create_entity("IfcCartesianPoint", ifc_safe_vector_type(position)),
2024-09-11 15:05:18 +05:00
RefDirection=ref_direction,
)
def vertex(self, position: VectorType = (0.0, 0.0, 0.0)) -> ifcopenshell.entity_instance:
"""Create a topological vertex
Commonly used in structural point elements.
:param position: The 3D coordinate of the vertex
:return: IfcVertexPoint
"""
return self.file.create_entity(
"IfcVertexPoint", self.file.create_entity("IfcCartesianPoint", ifc_safe_vector_type(position))
)
def edge(
self, start: VectorType = (0.0, 0.0, 0.0), end: VectorType = (1.0, 0.0, 0.0)
) -> ifcopenshell.entity_instance:
"""Create a topological edge
:param start: The start coordinates of the vertex.
:param end: The end coordinates of the vertex.
:return: IfcEdge
"""
return self.file.create_entity("IfcEdge", self.vertex(start), self.vertex(end))
def face(self, points: SequenceOfVectors) -> ifcopenshell.entity_instance:
"""Create a single topological face
There are many types of faces, but for now we only support planar
polyloop defined faces with an outer boundary.
:param points: ordered list of 3d coordinates representing the outer boundary
:return: IfcFace
"""
verts = [self.file.createIfcCartesianPoint(p) for p in ifc_safe_vector_type(points)]
return self.file.createIfcFace([self.file.createIfcFaceOuterBound(self.file.createIfcPolyLoop(verts), True)])
2023-01-08 13:32:13 +05:00
def mirror(
self,
2024-05-15 10:55:29 +05:00
curve_or_item: Union[ifcopenshell.entity_instance, list[ifcopenshell.entity_instance]],
2024-12-16 17:42:35 +05:00
mirror_axes: Union[VectorType, SequenceOfVectors] = (1.0, 1.0),
mirror_point: VectorType = (0.0, 0.0),
2024-05-15 10:55:29 +05:00
create_copy: bool = False,
2024-12-16 17:42:35 +05:00
placement_matrix: Optional[np.ndarray] = None,
2024-05-15 10:55:29 +05:00
) -> Union[ifcopenshell.entity_instance, list[ifcopenshell.entity_instance]]:
2024-12-16 17:42:35 +05:00
"""Mirror curve/representaiton item/representation.
:param curve_or_item: A single item to mirror or a sequence of them.
:param mirror_axes: A vector of values, should have value > 0 for axes where mirror should be applied.
Example: mirroring `A(1,0)` by axis `(1,0)` will result in `A'(-1,0)`
Also could be a list of mirrors to apply to `curve_or_item`
multiple mirror_axes will result in multiple resulting curves
Example: curve_or_item = [a, b], mirror_axes=[v1, v2], result = [av1, av2, bv1, bv2]
:param mirror_point: Point relative to which mirror should be applied.
:param create_copy: Whether to mirror the provided item or it's copy.
:param placement_matrix: Optional placement matrix to use for polylines.
:return: Mirrored curve/item/representation or a sequence of them.
"""
2023-01-08 13:32:13 +05:00
# TODO: need to add placement_matrix for other types besides polycurve?
2024-12-16 17:42:35 +05:00
np_XY = slice(2)
np_X, np_Y, np_Z = 0, 1, 2
2023-01-08 13:32:13 +05:00
multiple_objects = isinstance(curve_or_item, collections.abc.Iterable)
curve_or_item = [curve_or_item] if not multiple_objects else curve_or_item
2024-12-16 17:42:35 +05:00
multiple_transformations = not isinstance(mirror_axes[0], (float, int))
2023-01-08 13:32:13 +05:00
mirror_axes_data = [mirror_axes] if not multiple_transformations else mirror_axes
2024-12-16 17:42:35 +05:00
processed_objects: list[ifcopenshell.entity_instance] = []
2023-01-08 13:32:13 +05:00
for curve_or_item_el in curve_or_item:
for mirror_axes in mirror_axes_data:
c = (
ifcopenshell.util.element.copy_deep(self.file, curve_or_item_el)
if create_copy
else curve_or_item_el
)
2023-01-08 13:32:13 +05:00
if c.is_a() in ("IfcIndexedPolyCurve", "IfcPolyline"):
original_coords = self.get_polyline_coords(c)
2024-12-16 17:42:35 +05:00
inverted_placement_matrix = (
np.linalg.inv(placement_matrix) if placement_matrix is not None else None
)
2023-01-08 13:32:13 +05:00
coords = []
for co in original_coords:
2024-12-16 17:42:35 +05:00
co_base = co.copy()
if placement_matrix is not None:
2023-01-08 13:32:13 +05:00
# TODO: add support for Z-axis too
2024-12-16 17:42:35 +05:00
co_base = placement_matrix @ np_to_3d(co_base)
co = self.mirror_2d_point(co_base[np_XY], mirror_axes, mirror_point)
co = np_to_3d(co, z=co_base[2])
co = (inverted_placement_matrix @ co)[np_XY]
2023-01-08 13:32:13 +05:00
else:
co = self.mirror_2d_point(co_base, mirror_axes, mirror_point)
coords.append(co)
self.set_polyline_coords(c, coords)
2023-01-08 13:32:13 +05:00
elif c.is_a("IfcCircle") or c.is_a("IfcEllipse"):
2024-12-16 17:42:35 +05:00
base_position = c.Position.Location.Coordinates
2023-01-08 13:32:13 +05:00
new_position = self.mirror_2d_point(base_position, mirror_axes, mirror_point)
2024-12-16 17:42:35 +05:00
c.Position.Location.Coordinates = ifc_safe_vector_type(new_position)
2023-01-08 13:32:13 +05:00
elif c.is_a("IfcExtrudedAreaSolid"):
2024-12-16 17:42:35 +05:00
placement_matrix_ = ifcopenshell.util.placement.get_axis2placement(c.Position)[:3, :3]
base_position = c.Position.Location.Coordinates
2023-01-08 13:32:13 +05:00
# TODO: add support for Z-axis too
2024-12-16 17:42:35 +05:00
new_position = self.mirror_2d_point(base_position[np_XY], mirror_axes, mirror_point)
new_position = np_to_3d(new_position, base_position[np_Z])
c.Position.Location.Coordinates = ifc_safe_vector_type(new_position)
2023-01-08 13:32:13 +05:00
# TODO: add support for Z-axis too
2024-12-16 17:42:35 +05:00
self.translate(c.SweptArea.OuterCurve, base_position[np_XY])
self.mirror(c.SweptArea.OuterCurve, mirror_axes, mirror_point, placement_matrix=placement_matrix_)
self.translate(c.SweptArea.OuterCurve, -new_position[np_XY])
2023-01-08 13:32:13 +05:00
if hasattr(c.SweptArea, "InnerCurves"):
for inner_curve in c.SweptArea.InnerCurves:
2024-12-16 17:42:35 +05:00
self.translate(inner_curve, base_position[np_XY])
self.mirror(inner_curve, mirror_axes, mirror_point, placement_matrix=placement_matrix_)
self.translate(inner_curve, -new_position[np_XY])
2023-01-08 13:32:13 +05:00
# extrusion converted to world space
2024-12-16 17:42:35 +05:00
base_extruded_direction = c.ExtrudedDirection.DirectionRatios
extruded_direction = placement_matrix_ @ base_extruded_direction
2023-01-08 13:32:13 +05:00
# TODO: add support for Z-axis too
# mirror point is ignored for extrusion direction
2024-12-16 17:42:35 +05:00
new_direction = self.mirror_2d_point(
extruded_direction[np_XY], mirror_axes, mirror_point=(0.0, 0.0)
)
new_direction = np_to_3d(new_direction, extruded_direction[np_Z])
2023-01-08 13:32:13 +05:00
# extrusion direction converted back to placement space
2024-12-16 18:22:00 +05:00
new_direction = np.linalg.inv(placement_matrix_) @ (new_direction)
c.ExtrudedDirection.DirectionRatios = ifc_safe_vector_type(new_direction)
2023-01-08 13:32:13 +05:00
elif c.is_a("IfcTrimmedCurve"):
trim_coords = [c.Trim1[0].Coordinates, c.Trim2[0].Coordinates]
trim_coords = [
self.mirror_2d_point(base_position, mirror_axes, mirror_point) for base_position in trim_coords
]
2023-01-08 13:32:13 +05:00
# if mirror only by 1 axis we need to preserve the counter-clockwise order
# for the trim points
if 0 in mirror_axes:
trim_coords = [trim_coords[1], trim_coords[0]]
2024-12-16 17:42:35 +05:00
trim_coords = ifc_safe_vector_type(np.array(trim_coords))
2023-01-08 13:32:13 +05:00
c.Trim1[0].Coordinates, c.Trim2[0].Coordinates = trim_coords
self.mirror(c.BasisCurve, mirror_axes, mirror_point)
else:
raise Exception(f"{c} is not supported for mirror() method.")
processed_objects.append(c)
return processed_objects if (multiple_objects or multiple_transformations) else processed_objects[0]
2024-12-16 18:22:00 +05:00
def sphere(self, radius: float = 1.0, center: VectorType = (0.0, 0.0, 0.0)) -> ifcopenshell.entity_instance:
2024-01-24 11:47:19 +05:00
"""
:param radius: radius of the sphere.
:param center: sphere position.
2024-01-24 11:47:19 +05:00
:return: IfcSphere
"""
ifc_position = self.create_axis2_placement_3d(position=center)
2024-01-24 11:47:19 +05:00
return self.file.createIfcSphere(Radius=radius, Position=ifc_position)
2025-01-18 18:34:40 +11:00
def block(
self,
position: VectorType = (0.0, 0.0, 0.0),
x_length: float = 1.0,
y_length: float = 1.0,
z_length: float = 1.0,
) -> ifcopenshell.entity_instance:
"""
:param position: the bottom left (min X/Y of the cube).
:param x_length: the X length, in the +X direction
:param y_length: the Y length, in the +Y direction
:param z_length: the Z length, in the +Z direction
:return: IfcBlock
"""
return self.file.createIfcBlock(self.create_axis2_placement_3d(position), x_length, y_length, z_length)
def half_space_solid(
self, plane: ifcopenshell.entity_instance, agreement_flag: bool = False
) -> ifcopenshell.entity_instance:
"""
:param plane: The IfcPlane representing the half space.
:param agreement_flag: False if +Z represents the void
:return: IfcHalfSpaceSolid
"""
return self.file.createIfcHalfSpaceSolid(plane, AgreementFlag=agreement_flag)
2023-01-08 13:32:13 +05:00
def extrude(
self,
2024-05-15 10:55:29 +05:00
profile_or_curve: ifcopenshell.entity_instance,
magnitude: float = 1.0,
2024-12-16 18:22:00 +05:00
position: VectorType = (0.0, 0.0, 0.0),
extrusion_vector: VectorType = (0.0, 0.0, 1.0),
position_z_axis: VectorType = (0.0, 0.0, 1.0),
position_x_axis: VectorType = (1.0, 0.0, 0.0),
position_y_axis: Optional[VectorType] = None,
2024-05-15 10:55:29 +05:00
) -> ifcopenshell.entity_instance:
2023-01-08 13:32:13 +05:00
"""Extrude profile or curve to get IfcExtrudedAreaSolid.
REMEMBER when handling custom axes - IFC is using RIGHT handed coordinate system.
Position and position axes are in world space, extrusion vector in placement space defined by
position_x_axis/position_y_axis/position_z_axis
2023-08-28 15:25:10 +05:00
NOTE: changing position also changes the resulting geometry origin.
2024-12-16 18:22:00 +05:00
:param profile_or_curve: Profile or a curve to extrude (curve will automatically converted to a profile).
:param extrusion_vector: as defined in coordinate system position_x_axis+position_z_axis
:param position: as defined in default IFC coordinate system, not in position_x_axis+position_z_axis
:param position_y_axis: optional, could be used to calculate Z-axis based on Y-axis
:return: IfcExtrudedAreaSolid
2023-01-08 13:32:13 +05:00
"""
if not magnitude:
raise Exception(
"Extrusion magnitude must be greater than 0 to be valid.\n"
"Ref: https://ifc43-docs.standards.buildingsmart.org/IFC/RELEASE/IFC4x3/HTML/lexical/IfcPositiveLengthMeasure.htm#8.11.2.71.3-Formal-representation"
)
2023-08-04 14:46:48 +05:00
if not profile_or_curve.is_a("IfcProfileDef"):
2023-01-08 13:32:13 +05:00
profile_or_curve = self.profile(profile_or_curve)
if position_y_axis:
2024-12-16 18:22:00 +05:00
position_z_axis = np.cross(position_x_axis, position_y_axis)
2023-01-08 13:32:13 +05:00
ifc_position = self.create_axis2_placement_3d(position, position_z_axis, position_x_axis)
2024-12-16 18:22:00 +05:00
ifc_direction = self.file.create_entity("IfcDirection", ifc_safe_vector_type(extrusion_vector))
extruded_area = self.file.createIfcExtrudedAreaSolid(
2023-01-08 13:32:13 +05:00
SweptArea=profile_or_curve, Position=ifc_position, ExtrudedDirection=ifc_direction, Depth=magnitude
)
return extruded_area
2024-05-15 10:55:29 +05:00
def create_swept_disk_solid(
self, path_curve: ifcopenshell.entity_instance, radius: float
) -> ifcopenshell.entity_instance:
"""Create IfcSweptDiskSolid from `path_curve` (must be 3D) and `radius`"""
if path_curve.Dim != 3:
raise Exception(
f"Path curve for IfcSweptDiskSolid should be 3D to be valid, currently it has {path_curve.Dim} dimensions.\n"
"Ref: https://ifc43-docs.standards.buildingsmart.org/IFC/RELEASE/IFC4x3/HTML/lexical/IfcSweptDiskSolid.htm#8.8.3.42.4-Formal-propositions"
)
disk_solid = self.file.createIfcSweptDiskSolid(Directrix=path_curve, Radius=radius)
return disk_solid
2024-05-15 10:55:29 +05:00
def get_representation(
self,
context: ifcopenshell.entity_instance,
2024-12-16 18:22:00 +05:00
items: Union[ifcopenshell.entity_instance, Sequence[ifcopenshell.entity_instance]],
2024-05-15 10:55:29 +05:00
representation_type: Optional[str] = None,
) -> ifcopenshell.entity_instance:
2023-08-28 16:13:37 +05:00
"""Create IFC representation for the specified context and items.
2023-08-28 15:25:10 +05:00
2023-08-28 16:13:37 +05:00
:param context: IfcGeometricRepresentationSubContext
:param items: could be a list or single curve/IfcExtrudedAreaSolid
:param representation_type: Explicitly specified RepresentationType.
2024-01-24 16:14:26 +05:00
If not provided it will be guessed from the items types
2024-05-15 10:55:29 +05:00
:return: IfcShapeRepresentation
2023-08-28 16:13:37 +05:00
"""
2023-01-08 13:32:13 +05:00
if not isinstance(items, collections.abc.Iterable):
items = [items]
if not representation_type:
representation_type = ifcopenshell.util.representation.guess_type(items)
return self.file.create_entity(
(
"IfcTopologyRepresentation"
if representation_type in ("Vertex", "Edge", "Path", "Face", "Shell")
else "IfcShapeRepresentation"
),
2023-01-08 13:32:13 +05:00
ContextOfItems=context,
RepresentationIdentifier=context.ContextIdentifier,
RepresentationType=representation_type,
2023-01-08 13:32:13 +05:00
Items=items,
)
2024-05-15 10:55:29 +05:00
def deep_copy(self, element: ifcopenshell.entity_instance) -> ifcopenshell.entity_instance:
return ifcopenshell.util.element.copy_deep(self.file, element)
# UTILITIES
2024-12-16 18:22:00 +05:00
def extrude_kwargs(self, axis: Literal["Y", "X", "Z"]) -> dict[str, tuple[float, float, float]]:
"""Shortcut to get kwargs for `ShapeBuilder.extrude` to extrude by some axis.
It assumes you have 2D profile in:
XZ plane for Y axis extrusion, \n
YZ plane for X axis extrusion, \n
XY plane for Z axis extrusion, \n
Extruding by X/Y using other kwargs might break ValidExtrusionDirection."""
if axis == "Y":
return {
2024-12-16 18:22:00 +05:00
"position_x_axis": (1, 0, 0),
"position_z_axis": (0, -1, 0),
"extrusion_vector": (0, 0, -1),
}
elif axis == "X":
return {
2024-12-16 18:22:00 +05:00
"position_x_axis": (0, 1, 0),
"position_z_axis": (1, 0, 0),
"extrusion_vector": (0, 0, 1),
}
elif axis == "Z":
return {
2024-12-16 18:22:00 +05:00
"position_x_axis": (1, 0, 0),
"position_z_axis": (0, 0, 1),
"extrusion_vector": (0, 0, 1),
}
2024-05-15 10:55:29 +05:00
def rotate_extrusion_kwargs_by_z(
self, kwargs: dict[str, Any], angle: float, counter_clockwise: bool = False
2024-12-16 18:22:00 +05:00
) -> dict[str, VectorType]:
"""shortcut to rotate extrusion kwargs by z axis
`kwargs` expected to have `position_x_axis` and `position_z_axis` keys
`angle` is a rotation value in radians
by default rotation is clockwise, to make it counter clockwise use `counter_clockwise` flag
"""
2024-12-16 18:22:00 +05:00
rot = np_rotation_matrix(-angle, 3, "Z")
kwargs = kwargs.copy() # prevent mutation of original kwargs
2024-12-16 18:22:00 +05:00
kwargs["position_x_axis"] = rot @ kwargs["position_x_axis"]
kwargs["position_z_axis"] = rot @ kwargs["position_z_axis"]
return kwargs
2024-12-16 16:53:00 +05:00
def get_polyline_coords(self, polyline: ifcopenshell.entity_instance) -> np.ndarray:
"""polyline should be either `IfcIndexedPolyCurve` or `IfcPolyline`"""
coords = None
if polyline.is_a("IfcIndexedPolyCurve"):
2024-12-16 16:53:00 +05:00
coords = np.array(polyline.Points.CoordList)
elif polyline.is_a("IfcPolyline"):
coords = np.array(tuple(p.Coordinates for p in polyline.Points))
2024-12-16 16:53:00 +05:00
else:
raise Exception(f"Unsupported polyline type: {polyline.is_a()}")
return coords
2024-12-16 16:53:00 +05:00
def set_polyline_coords(self, polyline: ifcopenshell.entity_instance, coords: SequenceOfVectors) -> None:
"""polyline should be either `IfcIndexedPolyCurve` or `IfcPolyline`"""
if polyline.is_a("IfcIndexedPolyCurve"):
2024-12-16 16:53:00 +05:00
polyline.Points.CoordList = ifc_safe_vector_type(coords)
elif polyline.is_a("IfcPolyline"):
2024-12-16 16:53:00 +05:00
ifc_points: list[ifcopenshell.entity_instance] = polyline.Points
assert len(ifc_points) == len(coords)
for point, co in zip(ifc_points, ifc_safe_vector_type(coords)):
point.Coordinates = co
else:
raise Exception(f"Unsupported polyline type: {polyline.is_a()}")
2023-07-21 11:00:12 +05:00
2024-05-15 10:55:29 +05:00
def get_simple_2dcurve_data(
self,
coords: SequenceOfVectors,
2024-06-04 13:50:00 +05:00
fillets: Sequence[int] = (),
fillet_radius: Union[float, Sequence[float]] = (),
2024-05-15 10:55:29 +05:00
closed: bool = True,
create_ifc_curve: bool = False,
) -> tuple[list[VectorType], list[list[int]], Union[ifcopenshell.entity_instance, None]]:
2023-07-21 11:00:12 +05:00
"""
Creates simple 2D curve from set of 2d coords and list of points with fillets.
Simple curve means that all fillets are based on 90 degree angle.
:param coords: list of 2d coords. Example: ((x0,y0), (x1,y1), (x2, y2))
:param fillets: list of points from `coords` to base fillet on. Example: (1,)
:param fillet_radius: list of fillet radius for each of corresponding point form `fillets`.
Example: (5.,) Note: `fillet_radius` could be just 1 float value if it's the same for all fillets.
:param closed: boolean whether curve should be closed (whether last point connected to first one).
:param create_ifc_curve: create IfcIndexedPolyCurve or just return the data.
2023-07-21 11:00:12 +05:00
:return: (points, segments, ifc_curve) for the created simple curve
if both points in e are equally far from pt, then v1 is returned.
"""
2023-07-21 11:00:12 +05:00
def remove_redundant_points(
points: list[VectorType], segments: list[list[int]]
) -> tuple[list[VectorType], list[list[int]]]:
2023-07-21 11:00:12 +05:00
# prevent mutating
points = [tuple(p) for p in points]
segments = segments.copy()
# find duplicate points, reindex them in segments
# and mark them to delete later
points_to_remove: list[int] = []
2023-07-21 11:00:12 +05:00
prev_point = 0
for i, p in enumerate(points[1:], 1):
if p != points[prev_point]:
prev_point = i
continue
valid_segments: list[list[int]] = []
2023-07-21 11:00:12 +05:00
for s in segments:
s = [ps if ps != i else prev_point for ps in s]
valid_segments.append(s)
segments = valid_segments
points_to_remove.append(i)
# remove duplicate segments
valid_segments = [segment for segment in segments if len(set(segment)) != 1]
points = [point for i, point in enumerate(points) if i not in points_to_remove]
# correct the order in segments
unique_points = sorted(set(chain(*valid_segments)))
unique_points_translation = {prev: i for i, prev in enumerate(unique_points)}
valid_segments = [[unique_points_translation[p] for p in s] for s in valid_segments]
return points, valid_segments
# option to use same fillet radius for all fillets
if isinstance(fillet_radius, (float, int)):
2023-07-21 11:00:12 +05:00
fillet_radius = [fillet_radius] * len(fillets)
fillets: dict[int, float] = dict(zip(fillets, fillet_radius))
segments: list[list[int]] = []
points: list[VectorType] = []
2023-07-21 11:00:12 +05:00
for co_i, co in enumerate(coords, 0):
current_point = len(points)
if co_i in fillets:
r = fillets[co_i]
rsb = r * cos(pi / 4) # radius shift big
rss = r - rsb # radius shift small
next_co = coords[(co_i + 1) % len(coords)]
previous_co = coords[co_i - 1]
# identify fillet type (1 of 4 possible types)
x_direction = 1 if coords[co_i][0] < previous_co[0] or coords[co_i][0] < next_co[0] else -1
y_direction = 1 if coords[co_i][1] < previous_co[1] or coords[co_i][1] < next_co[1] else -1
xshift_point = (co[0] + r * x_direction, co[1])
middle_point = (co[0] + rss * x_direction, co[1] + rss * y_direction)
yshift_point = (co[0], co[1] + r * y_direction)
# identify fillet direction
if co[1] == previous_co[1]:
points.extend((xshift_point, middle_point, yshift_point))
else:
points.extend((yshift_point, middle_point, xshift_point))
segments.append([current_point - 1, current_point])
segments.append([current_point, current_point + 1, current_point + 2])
else:
points.append(co)
if co_i != 0:
segments.append([current_point - 1, current_point])
if closed:
segments.append([len(points) - 1, 0])
# replace negative index
if segments[0][0] == -1:
segments[0][0] = len(points) - 1
# sometime fillet points could match previous or next points in line
# I remove them at the end to avoid making fillet algorithm even less readable
points, segments = remove_redundant_points(points, segments)
ifc_curve = None
if create_ifc_curve:
ifc_points = self.file.createIfcCartesianPointList2D(ifc_safe_vector_type(points))
2023-07-21 11:00:12 +05:00
ifc_segments = []
for segment in segments:
segment = [i + 1 for i in segment]
if len(segment) == 2:
ifc_segments.append(self.file.createIfcLineIndex(segment))
elif len(segment) == 3:
ifc_segments.append(self.file.createIfcArcIndex(segment))
ifc_curve = self.file.createIfcIndexedPolyCurve(Points=ifc_points, Segments=ifc_segments)
return (points, segments, ifc_curve)
2023-08-04 14:46:48 +05:00
def create_z_profile_lips_curve(
2024-05-15 10:55:29 +05:00
self,
FirstFlangeWidth: float,
SecondFlangeWidth: float,
Depth: float,
Girth: float,
WallThickness: float,
FilletRadius: float,
) -> ifcopenshell.entity_instance:
2023-07-21 11:00:12 +05:00
x1 = FirstFlangeWidth
x2 = SecondFlangeWidth
y = Depth / 2
g = Girth
t = WallThickness
r = FilletRadius
# fmt: off
coords = (
(-t/2, y),
(x2, y),
(x2, y-g),
(x2-t, y-g),
(x2-t, y-t),
(t/2, y-t),
(t/2, -y),
(-x1, -y),
(-x1, -y+g),
(-x1+t, -y+g),
(-x1+t, -y+t),
(-t/2, -y+t)
)
# option for no additional thickness in outer radius:
# points, segments, ifc_curve = create_curve_from_coords(
# coords, fillets = (0, 1, 4, 5, 6, 7, 10, 11), fillet_radius=r, closed=True, ifc_file=ifc_file
# )
2023-08-04 14:46:48 +05:00
points, segments, ifc_curve = self.get_simple_2dcurve_data(
coords,
fillets = (0, 1, 4, 5, 6, 7, 10, 11),
fillet_radius=(r+t, r+t, r, r, r+t, r+t, r, r),
2023-07-21 11:00:12 +05:00
closed=True, create_ifc_curve=True)
2023-08-04 14:46:48 +05:00
# fmt: on
assert ifc_curve
2023-07-21 11:00:12 +05:00
return ifc_curve
2023-08-04 14:46:48 +05:00
2024-05-15 10:55:29 +05:00
def create_transition_arc_ifc(
2024-12-16 18:37:07 +05:00
self, width: float, height: float, create_ifc_curve: bool = False
) -> tuple[SequenceOfVectors, list[list[int]], Union[ifcopenshell.entity_instance, None]]:
"""Create an arc in the rectangle with specified width and height.
If it's not possible to make a complete arc, create an arc with longest radius possible
and straight segment in the middle.
"""
2023-07-21 11:00:12 +05:00
fillet_size = (width / 2) / height
if fillet_size <= 1:
fillet_radius = height * fillet_size
curve_coords = [
(0.0, 0.0),
(0.0, height),
(width * 0.5, height),
(width, height),
(width, 0.0),
]
fillets = (1, 3)
else:
fillet_radius = height
curve_coords = [
(0.0, 0.0),
(0.0, height),
(fillet_radius, height),
(width - fillet_radius, height),
(width, height),
(width, 0.0),
]
fillets = (1, 4)
points, segments, transition_arc = self.get_simple_2dcurve_data(
curve_coords, fillets, fillet_radius, closed=False, create_ifc_curve=create_ifc_curve
)
2023-08-04 14:46:48 +05:00
return points, segments, transition_arc
2025-02-12 11:43:48 +05:00
def mesh(self, points: SequenceOfVectors, faces: Sequence[Sequence[int]]) -> ifcopenshell.entity_instance:
if self.file.schema == "IFC2X3":
return self.faceted_brep(points, faces)
return self.polygonal_face_set(points, faces)
2024-12-16 18:37:07 +05:00
def faceted_brep(self, points: SequenceOfVectors, faces: Sequence[Sequence[int]]) -> ifcopenshell.entity_instance:
"""Generate an IfcFacetedBrep with a closed shell
Note that :func:`polygonal_face_set` is recommended in IFC4.
:param points: list of 3d coordinates
:param faces: list of faces consisted of point indices (points indices starting from 0)
:return: IfcFacetedBrep
"""
2024-12-16 18:37:07 +05:00
verts = [self.file.createIfcCartesianPoint(p) for p in ifc_safe_vector_type(points)]
faces: list[ifcopenshell.entity_instance] = [
self.file.createIfcFace(
[self.file.createIfcFaceOuterBound(self.file.createIfcPolyLoop([verts[v] for v in f]), True)]
)
for f in faces
]
return self.file.createIfcFacetedBrep(self.file.createIfcClosedShell(faces))
def triangulated_face_set(
self, points: SequenceOfVectors, faces: Sequence[Sequence[int]]
) -> ifcopenshell.entity_instance:
"""
Generate an IfcTriangulatedFaceSet
Note that this is not available in IFC2X3.
:param points: list of 3d coordinates
:param faces: list of triangles consisted of point indices (points indices starting from 0)
:return: IfcTriangulatedFaceSet
"""
ifc_points = self.file.createIfcCartesianPointList3D(ifc_safe_vector_type(points))
ifc_faces = [[i + 1 for i in face][:3] for face in faces]
return self.file.createIfcTriangulatedFaceSet(Coordinates=ifc_points, CoordIndex=ifc_faces)
2024-12-16 18:37:07 +05:00
def polygonal_face_set(
self, points: SequenceOfVectors, faces: Sequence[Union[Sequence[int], Sequence[Sequence[int]]]]
2024-12-16 18:37:07 +05:00
) -> ifcopenshell.entity_instance:
2023-08-04 14:46:48 +05:00
"""
Generate an IfcPolygonalFaceSet
Note that this is not available in IFC2X3.
2023-08-04 14:46:48 +05:00
:param points: list of 3d coordinates
2024-06-17 11:30:38 +05:00
:param faces: list of faces consisted of point indices (points indices starting from 0)
in case of multiple sequences per face, the subsequent ones are inner voids
2024-06-17 11:30:38 +05:00
:return: IfcPolygonalFaceSet
2023-08-04 14:46:48 +05:00
"""
def is_sequence_of_ints(x):
return isinstance(x, Sequence) and not isinstance(x, (str, bytes)) and all(isinstance(el, int) for el in x)
def is_sequence_of_sequence_of_ints(x):
return (
isinstance(x, Sequence) and not isinstance(x, (str, bytes)) and all(is_sequence_of_ints(el) for el in x)
)
def incr(face):
return [i + 1 for i in face]
if not all(is_sequence_of_ints(f) or is_sequence_of_sequence_of_ints(f) for f in faces):
raise ValueError("Expected a sequence of int or sequence of sequence of int for each face")
2024-12-16 18:37:07 +05:00
ifc_points = self.file.createIfcCartesianPointList3D(ifc_safe_vector_type(points))
ifc_faces = [
(
self.file.createIfcIndexedPolygonalFace(incr(face))
if is_sequence_of_ints(face)
else self.file.createIfcIndexedPolygonalFaceWithVoids(incr(face[0]), list(map(incr, face[1:])))
)
for face in faces
]
return self.file.createIfcPolygonalFaceSet(Coordinates=ifc_points, Faces=ifc_faces)
2023-08-04 14:46:48 +05:00
2023-08-28 15:25:10 +05:00
def extrude_face_set(
2024-05-15 10:55:29 +05:00
self,
2024-12-16 18:37:07 +05:00
points: SequenceOfVectors,
2024-05-15 10:55:29 +05:00
magnitude: float,
2024-12-16 18:37:07 +05:00
extrusion_vector: VectorType = (0, 0, 1),
offset: Optional[VectorType] = None,
2024-05-15 10:55:29 +05:00
start_cap: bool = True,
end_cap: bool = True,
) -> ifcopenshell.entity_instance:
2023-08-28 15:25:10 +05:00
"""
Method to extrude by creating face sets rather than creating IfcExtrudedAreaSolid.
Useful if your representation is already using face sets and you need to avoid using SweptSolid
to assure CorrectItemsForType.
:param points: list of points, assuming they form consecutive closed polyline.
:param magnitude: extrusion magnitude
:param extrusion_vector: extrusion direction.
2023-08-28 15:25:10 +05:00
:param offset: offset from the points
:param start_cap: if True, create start cap.
:param end_cap: if True, create end cap.
2023-08-28 15:25:10 +05:00
:return: IfcPolygonalFaceSet
"""
# prevent mutating arguments, deepcopy doesn't work
2024-12-16 18:37:07 +05:00
start_points = np.array(points)
if offset is not None and offset.any():
2024-12-16 18:37:07 +05:00
start_points += offset
extrusion_offset = np.multiply(extrusion_vector, magnitude)
end_points = start_points + extrusion_offset
2023-08-28 15:25:10 +05:00
2024-12-16 18:37:07 +05:00
all_points = np.vstack((start_points, end_points))
2023-08-28 15:25:10 +05:00
faces = []
n_verts = len(start_points)
last_vert_i = n_verts - 1
for i in range(last_vert_i):
face = (i, i + 1, n_verts + i + 1, n_verts + i)
faces.append(face)
faces.append((last_vert_i, 0, n_verts + 0, n_verts + last_vert_i)) # close the loop
if end_cap:
faces.append(tuple(range(n_verts, n_verts * 2)))
if start_cap:
faces.append(tuple(reversed(range(n_verts))))
2024-12-16 18:37:07 +05:00
face_set = self.polygonal_face_set(all_points, faces)
2023-08-28 15:25:10 +05:00
return face_set
# TODO: move MEP to separate shape builder sub module
2023-08-21 17:22:20 +05:00
def mep_transition_shape(
self,
start_segment: ifcopenshell.entity_instance,
end_segment: ifcopenshell.entity_instance,
start_length: float,
end_length: float,
angle: float = 30.0,
profile_offset: VectorType = (0.0, 0.0),
) -> Union[tuple[ifcopenshell.entity_instance, dict[str, Any]], tuple[None, None]]:
"""Generate a MEP transition shape for the provided segments.
:param start_segment: Starting segment.
:param end_segment: Ending segment.
:param start_length: Start transition length.
:param end_length: End transition length.
:param angle: Transition angle, in degrees.
Good default values from angle = 30/60 deg
30 degree angle will result in 75 degrees on the transition (= 90 - α/2) - https://i.imgur.com/tcoYDWu.png
:param profile_offset: 2D vector for profile offset.
:return: A tuple of Model/Body/MODEL_VIEW IfcRepresentation and dictionary of transition shape data.
Or (None, None) if there was an error in the process.
2023-08-04 14:46:48 +05:00
"""
# TODO: get rid of reliance on profiles
def get_profile(element: ifcopenshell.entity_instance) -> Union[ifcopenshell.entity_instance, None]:
2023-08-04 14:46:48 +05:00
material = ifcopenshell.util.element.get_material(element, should_skip_usage=True)
if material and material.is_a("IfcMaterialProfileSet") and len(material.MaterialProfiles) == 1:
return material.MaterialProfiles[0].Profile
def get_circle_points(radius: float, segments: int = 16) -> np.ndarray:
2023-08-21 17:22:20 +05:00
"""starting from (R,0), going counter-clockwise"""
angles = np.linspace(0, 2 * np.pi, segments, endpoint=False)
verts = np.column_stack((np.cos(angles), np.sin(angles), np.zeros(segments))) * radius
2023-08-21 17:22:20 +05:00
return verts
def get_rectangle_points(dim: np.ndarray) -> np.ndarray:
2023-08-21 17:22:20 +05:00
"""Starting from (+X/2, +Y/2) going counter-clockwise"""
dim = dim / 2
offsets = np.array([[1, 1, 0], [-1, 1, 0], [-1, -1, 0], [1, -1, 0]])
return dim * offsets
2023-08-21 17:22:20 +05:00
# TODO: support more profiles
def get_dim(profile: ifcopenshell.entity_instance, depth: float) -> Union[np.ndarray, None]:
2023-08-21 17:22:20 +05:00
if profile.is_a("IfcRectangleProfileDef"):
return np.array([profile.XDim / 2, profile.YDim / 2, depth])
2023-08-21 17:22:20 +05:00
elif profile.is_a("IfcCircleProfileDef"):
return np.array([profile.Radius, profile.Radius, depth])
2023-08-21 17:22:20 +05:00
return None
2023-08-04 14:46:48 +05:00
start_profile = get_profile(start_segment)
end_profile = get_profile(end_segment)
if start_profile is None or end_profile is None:
return None, None
2023-08-04 14:46:48 +05:00
2023-08-21 17:22:20 +05:00
start_half_dim = get_dim(start_profile, start_length)
end_half_dim = get_dim(end_profile, end_length)
2023-08-04 14:46:48 +05:00
2023-08-21 17:22:20 +05:00
# if profile types are not supported
if start_half_dim is None or end_half_dim is None:
2023-08-21 17:22:20 +05:00
return None, None
2023-08-04 14:46:48 +05:00
transition_items = []
start_offset = np.array([0, 0, start_length])
2023-08-21 17:22:20 +05:00
end_extrusion_offset = start_offset.copy()
transition_length = self.mep_transition_length(start_half_dim, end_half_dim, angle, profile_offset)
2023-08-21 17:22:20 +05:00
if transition_length is None:
return None, None
faces: list[Sequence[int]] = []
end_extrusion_offset[2] += transition_length
end_extrusion_offset[:2] += profile_offset
2023-08-21 17:22:20 +05:00
if start_profile.is_a("IfcRectangleProfileDef") and end_profile.is_a("IfcRectangleProfileDef"):
# no transitions for exactly the same profiles
if transition_length == 0:
return None, None
2023-08-04 14:46:48 +05:00
faces += [(3, 4, 7, 0), (11, 8, 15, 12), (3, 11, 12, 4), (7, 15, 8, 0)]
2023-08-21 17:22:20 +05:00
# NOTE: clockwise order for correct face orientation
faces += [
# start extrusion
(0, 1, 2, 3),
(8, 11, 10, 9),
(0, 8, 9, 1),
(1, 9, 10, 2),
(2, 10, 11, 3),
# end extrusion
(4, 5, 6, 7),
(12, 15, 14, 13),
(4, 12, 13, 5),
(5, 13, 14, 6),
(6, 14, 15, 7),
]
points = [
start_half_dim * (-1, -1, 1),
start_half_dim * (-1, -1, 0),
start_half_dim * (1, -1, 0),
start_half_dim * (1, -1, 1),
end_half_dim * (1, -1, 0) + end_extrusion_offset,
end_half_dim * (1, -1, 1) + end_extrusion_offset,
end_half_dim * (-1, -1, 1) + end_extrusion_offset,
end_half_dim * (-1, -1, 0) + end_extrusion_offset,
start_half_dim * (-1, 1, 1),
start_half_dim * (-1, 1, 0),
start_half_dim * (1, 1, 0),
start_half_dim * (1, 1, 1),
end_half_dim * (1, 1, 0) + end_extrusion_offset,
end_half_dim * (1, 1, 1) + end_extrusion_offset,
end_half_dim * (-1, 1, 1) + end_extrusion_offset,
end_half_dim * (-1, 1, 0) + end_extrusion_offset,
2023-08-21 17:22:20 +05:00
]
elif start_profile.is_a("IfcCircleProfileDef") and end_profile.is_a("IfcCircleProfileDef"):
# no transitions for exactly the same profiles
if transition_length == 0:
return None, None
n_segments = 16
first_profile_points = get_circle_points(start_profile.Radius, n_segments)
second_profile_points = get_circle_points(end_profile.Radius, n_segments)
faces = []
for i in range(n_segments):
# For wrapping around the circle
next_i = (i + 1) % n_segments
face = [i, next_i, next_i + n_segments, i + n_segments]
faces.append(face)
2023-08-28 15:25:10 +05:00
transition_items.append(self.extrude_face_set(first_profile_points, start_length, end_cap=False))
transition_items.append(
2023-09-18 11:30:36 +05:00
self.extrude_face_set(second_profile_points, end_length, offset=end_extrusion_offset, start_cap=False)
2023-08-28 15:25:10 +05:00
)
2023-08-21 17:22:20 +05:00
first_profile_points += start_offset
second_profile_points += end_extrusion_offset
points = np.vstack((first_profile_points, second_profile_points))
2023-08-21 17:22:20 +05:00
else: # one is circular, another one is rectangular
# support transition from rectangle to circle of the same dimensions
if transition_length == 0:
transition_length = (start_length + end_length) / 2
end_extrusion_offset[2] += transition_length
2023-08-21 17:22:20 +05:00
starting_with_circle = start_profile.is_a("IfcCircleProfileDef")
if starting_with_circle:
circle_profile, rect_profile = start_profile, end_profile
else:
circle_profile, rect_profile = end_profile, start_profile
circle_points = get_circle_points(circle_profile.Radius)
rect_points = get_rectangle_points(np.array([rect_profile.XDim, rect_profile.YDim, 0]))
2023-08-21 17:22:20 +05:00
if starting_with_circle:
start_points, end_points = circle_points, rect_points
else:
start_points, end_points = rect_points, circle_points
2023-08-28 15:25:10 +05:00
transition_items.append(self.extrude_face_set(start_points, start_length, end_cap=False))
transition_items.append(
2023-09-18 11:30:36 +05:00
self.extrude_face_set(end_points, end_length, offset=end_extrusion_offset, start_cap=False)
2023-08-28 15:25:10 +05:00
)
2023-08-21 17:22:20 +05:00
# offset verts
if starting_with_circle:
circle_points += start_offset
rect_points += end_extrusion_offset
2023-08-21 17:22:20 +05:00
else:
rect_points += start_offset
circle_points += end_extrusion_offset
2023-08-21 17:22:20 +05:00
# circle verts are 0-15, rect verts are 16-19
points = np.concatenate((circle_points, rect_points))
2023-08-21 17:22:20 +05:00
transition_faces = [
(0, 19, 16), # base
(0, 16, 1),
(1, 16, 2),
(2, 16, 3),
(3, 16, 4),
(4, 16, 17), # base
(4, 17, 5),
(5, 17, 6),
(6, 17, 7),
(7, 17, 8),
(8, 17, 18), # base
(8, 18, 9),
(9, 18, 10),
(10, 18, 11),
(11, 18, 12),
(12, 18, 19), # base
(12, 19, 13),
(13, 19, 14),
(14, 19, 15),
(15, 19, 0),
]
# revert them in case it's starting with circle profile to keep the face orientation
if starting_with_circle:
transition_faces = [f[::-1] for f in transition_faces]
faces += transition_faces
2023-08-04 14:46:48 +05:00
face_set = self.polygonal_face_set(points, faces)
transition_items.append(face_set)
body = ifcopenshell.util.representation.get_context(self.file, "Model", "Body", "MODEL_VIEW")
assert body
2023-08-04 14:46:48 +05:00
representation = self.get_representation(body, transition_items, "Tesselation")
2023-08-04 14:46:48 +05:00
transition_data = {
"start_length": start_length,
"end_length": end_length,
"angle": angle,
"profile_offset": profile_offset,
2023-08-04 14:46:48 +05:00
"transition_length": transition_length,
"full_transition_length": start_length + transition_length + end_length,
}
return representation, transition_data
# TODO: move to separate shape_builder method
# so we could check transition length without creating representation
def mep_transition_length(
self,
start_half_dim: np.ndarray,
end_half_dim: np.ndarray,
angle: float,
profile_offset: VectorType = (0.0, 0.0),
verbose: bool = True,
):
"""get the final transition length for two profiles dimensions, angle and XY offset between them,
the difference from `calculate_transition` - `get_transition_length` is making sure
that length will fit both sides of the transition
"""
print = lambda *args, **kwargs: __builtins__["print"](*args, **kwargs) if verbose else None
np_X, np_Y = 0, 1
np_XY = slice(2)
# vectors tend to have bunch of float point garbage
# that can result in errors when we're calculating value for square root below
offset = np_round_to_precision(np.array(profile_offset), 1)
diff = start_half_dim[np_XY] - end_half_dim[np_XY]
diff = np.abs(diff)
print(f"offset = {profile_offset} / {offset}")
print(f"diff = {diff}")
calculation_arguments = {
"start_half_dim": start_half_dim,
"end_half_dim": end_half_dim,
"diff": diff,
"offset": offset,
"verbose": verbose,
}
def check_transition(end_profile: bool = False) -> Union[float, None]:
length = self.mep_transition_calculate(**calculation_arguments, angle=angle, end_profile=end_profile)
2023-08-23 15:06:05 +05:00
if length is None:
return
other_side_angle = self.mep_transition_calculate(
**calculation_arguments, length=length, end_profile=not end_profile
)
if other_side_angle is None:
return None
# NOTE: for now we just hardcode the good value for that case
same_dimension = is_x(diff[np_Y] if not end_profile else diff[np_X], 0)
if same_dimension and is_x(offset[np_Y] if not end_profile else offset[np_X], 0):
requested_angle = 90.0
else:
requested_angle = angle
print(f"other_side_angle = {other_side_angle}, requested_angle = {requested_angle}")
# need to make sure that the worst angle (maximum angle)
# for this transition angle is `requested_angle`
if other_side_angle < requested_angle or is_x(other_side_angle, requested_angle):
print(f"final length = {length}, angle = {requested_angle}, other side angle = {other_side_angle}")
return length
return check_transition() or check_transition(True)
def mep_transition_calculate(
self,
start_half_dim: np.ndarray,
end_half_dim: np.ndarray,
offset: np.ndarray,
diff: Optional[np.ndarray] = None,
end_profile: bool = False,
length: Optional[float] = None,
angle: Optional[float] = None,
verbose: bool = True,
) -> Union[float, None]:
"""will return transition length based on the profile dimension differences and offset.
If `length` is provided will return transition angle"""
print = lambda *args, **kwargs: __builtins__["print"](*args, **kwargs) if verbose else None
if diff is None:
diff = start_half_dim[:2] - end_half_dim[:2]
diff = np.abs(diff)
np_X, np_Y = 0, 1
np_YX = [1, 0]
if end_profile:
diff, offset = diff[np_YX], offset[np_YX]
same_dimension = is_x(diff[0], 0)
a = diff[np_X] + offset[np_X]
b = diff[np_X] - offset[np_X]
if length is None:
2023-08-23 15:06:05 +05:00
if not same_dimension:
assert angle is not None
t = tan(radians(angle))
2023-08-23 15:06:05 +05:00
h0 = a**2 + 4 * a * b * t**2 + 2 * a * b + b**2
# TODO: we might need to specify the exact failing cases in the future
if h0 < 0:
print(
f"B. Coulndn't calculate transition length for angle = {angle}, offset = {offset}, diff = {diff}"
)
return None
h = (a + b + sqrt(h0)) / (2 * t)
length_squared = h**2 - offset[np_Y] ** 2
if length_squared <= 0:
print(f"B. angle = {angle} requires h = {h} which is not possible with y offset = {offset[np_Y]}")
2023-08-23 15:06:05 +05:00
return None
length = sqrt(length_squared)
if verbose:
A = (end_half_dim if end_profile else start_half_dim) * (1, 0, 0)
end_profile_offset = np_to_3d(offset, length)
D = (start_half_dim if end_profile else end_half_dim) * (1, 0, 0)
B, C = -A, -D
C += end_profile_offset
D += end_profile_offset
tested_angle = degrees(np_angle(A - D, B - C))
print(f"A. length = {length}, requested angle = {angle}, tested angle = {tested_angle}")
else:
if is_x(offset[np_X], 0):
angle = 90 # NOTE: for now we just hardcode the good value for that case
h = start_half_dim[np_X] / tan(radians(angle / 2))
length_squared = h**2 - offset[np_Y] ** 2
if length_squared <= 0:
print(
f"B. angle = {angle} requires h = {h} which is not possible with y offset = {offset[np_Y]}"
)
2023-08-23 15:06:05 +05:00
return None
length = sqrt(length_squared)
if verbose:
O = np.zeros(3)
A = (-start_half_dim[np_X], 0, length) + np_to_3d(offset)
B = A * (-1, 1, 1)
tested_angle = degrees(np_angle(A - O, B - O))
print(f"B. length = {length}, requested angle = {angle}, tested angle = {tested_angle}")
else:
assert angle is not None
h = offset[np_X] / tan(radians(angle))
length_squared = h**2 - offset[np_Y] ** 2
if length_squared <= 0:
print(
f"C. angle = {angle} requires h = {h} which is not possible with y offset = {offset[np_Y]}"
)
2023-08-23 15:06:05 +05:00
return None
length = sqrt(length_squared)
if verbose:
A = np.array((-start_half_dim[np_X], 0, 0))
H = A + (0, 0, length)
H[np_Y] += offset[np_Y]
D = H.copy()
D[np_X] += offset[np_X]
tested_angle = degrees(np_angle(H - A, D - A))
print(f"C. length = {length}, requested angle = {angle}, tested angle = {tested_angle}")
return length
elif angle is None:
2023-08-23 15:06:05 +05:00
if not same_dimension:
if length == 0:
return 0
h = sqrt(length**2 + offset[np_Y] ** 2)
t = -h * (a + b) / (a * b - h**2)
angle = degrees(atan(t))
2023-08-23 15:06:05 +05:00
else:
h = sqrt(length**2 + offset[np_Y] ** 2)
if is_x(offset[np_X], 0):
angle = degrees(2 * atan(start_half_dim[np_X] / h))
else:
angle = degrees(atan(offset[np_X] / h))
return angle
2023-08-28 15:25:10 +05:00
def mep_bend_shape(
self,
segment: ifcopenshell.entity_instance,
start_length: float,
end_length: float,
angle: float,
radius: float,
bend_vector: VectorType,
flip_z_axis: bool,
) -> tuple[ifcopenshell.entity_instance, dict[str, Any]]:
2023-08-28 15:25:10 +05:00
"""
Generate a MEP bend shape for the provided segments.
2023-08-28 15:25:10 +05:00
:param segment: IfcFlowSegment for a bend.
2024-01-24 16:14:26 +05:00
Note that for a bend start and end segments types should match.
2023-08-28 15:25:10 +05:00
:param angle: bend angle, in radians
:param radius: bend radius
:param bend_vector: offset between start and end segments in local space of start segment
used mainly to determine the second bend axis and it's direction (positive or negative),
the actual magnitude of the vector is not important (though near zero values will be ignored).
:param flip_z_axis: since we cannot determine z axis direction from the profile offset,
2024-01-24 16:14:26 +05:00
there is an option to flip it if bend is going by start segment Z- axis.
:return: tuple of Model/Body/MODEL_VIEW IfcRepresentation and dictionary of transition shape data
2023-08-28 15:25:10 +05:00
"""
def get_profile(element: ifcopenshell.entity_instance) -> Union[ifcopenshell.entity_instance, None]:
2023-08-28 15:25:10 +05:00
material = ifcopenshell.util.element.get_material(element, should_skip_usage=True)
if material and material.is_a("IfcMaterialProfileSet") and len(material.MaterialProfiles) == 1:
return material.MaterialProfiles[0].Profile
def get_dim(profile: ifcopenshell.entity_instance, depth: float) -> Union[np.ndarray, None]:
2023-08-28 15:25:10 +05:00
if profile.is_a("IfcRectangleProfileDef"):
return np.array([profile.XDim / 2, profile.YDim / 2, depth])
2023-08-28 15:25:10 +05:00
elif profile.is_a("IfcCircleProfileDef"):
return np.array([profile.Radius, profile.Radius, depth])
2023-08-28 15:25:10 +05:00
return None
np_Z = 2
2023-08-28 15:25:10 +05:00
si_conversion = ifcopenshell.util.unit.calculate_unit_scale(self.file)
profile = get_profile(segment)
assert profile
2023-09-12 09:40:14 +05:00
is_circular_profile = profile.is_a("IfcCircleProfileDef")
2023-08-28 15:25:10 +05:00
profile_dim = get_dim(profile, start_length)
assert profile_dim is not None
2023-08-28 15:25:10 +05:00
rounded_bend_vector = np_round_to_precision(bend_vector, si_conversion)
lateral_axis = next(i for i in range(2) if not is_x(rounded_bend_vector[i], 0))
2023-08-28 15:25:10 +05:00
non_lateral_axis = 1 if lateral_axis == 0 else 0
lateral_sign = np.sign(bend_vector[lateral_axis])
z_sign = -1 if flip_z_axis else 1
2023-08-28 15:25:10 +05:00
rep_items: list[ifcopenshell.entity_instance] = []
2023-08-28 15:25:10 +05:00
# bend circle center
O = np.zeros(3)
2023-08-28 15:25:10 +05:00
O[lateral_axis] = (radius + profile_dim[lateral_axis]) * lateral_sign
theta = angle
def get_circle_points(angles: np.ndarray, radius: float) -> np.ndarray:
"""
:param angles: Angles, in radians.
"""
angles = angles - pi / 2
points = np.zeros((len(angles), 3))
2023-09-12 15:25:54 +05:00
# fmt: off
points[:, np_Z] = z_sign * np.cos(angles) * radius
points[:, lateral_axis] = lateral_sign * np.sin(angles) * radius
2023-09-12 15:25:54 +05:00
# fmt: on
return points
def get_circle_tangent(angle: float) -> np.ndarray:
"""
:param angle: Angle, in radians.
:return: Tangent vector.
"""
tangent = np.zeros(3)
tangent[np_Z] = cos(angle) * z_sign
2023-09-12 15:25:54 +05:00
tangent[lateral_axis] = sin(angle) * lateral_sign
return tangent
def get_bend_representation_item() -> ifcopenshell.entity_instance:
2023-08-28 15:25:10 +05:00
r = radius
theta_segments = np.array([0.0, theta / 2, theta])
points: np.ndarray
2023-09-12 09:40:14 +05:00
if is_circular_profile:
r += profile_dim[lateral_axis]
points = get_circle_points(theta_segments, r)
arc_points = (1,)
2023-09-12 15:25:54 +05:00
else:
outer_r = r + 2 * profile_dim[lateral_axis]
outer_points = get_circle_points(theta_segments[::-1], outer_r)
2023-09-14 13:15:43 +05:00
if is_x(r, 0):
points = get_circle_points(np.full(1, theta), r)
points = np.vstack((points, outer_points))
2023-09-14 13:15:43 +05:00
arc_points = (2,)
else:
inner_points = get_circle_points(theta_segments, r)
points = np.vstack((inner_points, outer_points))
2023-09-14 13:15:43 +05:00
arc_points = (1, 4)
2023-08-28 15:25:10 +05:00
points += O
offset = np.zeros(3)
offset[np_Z] = z_sign * start_length
2023-08-28 15:25:10 +05:00
2023-09-12 09:40:14 +05:00
if is_circular_profile:
bend_path = self.polyline(points, closed=False, arc_points=arc_points, position_offset=offset)
2023-09-12 09:40:14 +05:00
bend = self.create_swept_disk_solid(bend_path, profile_dim[lateral_axis])
else:
offset[non_lateral_axis] = -profile_dim[non_lateral_axis]
extrusion_kwargs = self.extrude_kwargs("XY"[non_lateral_axis])
polyline_points = points[:, [lateral_axis, np_Z]]
profile_curve = self.polyline(polyline_points, arc_points=arc_points, closed=True)
bend = self.extrude(
self.profile(profile_curve), profile_dim[non_lateral_axis] * 2, position=offset, **extrusion_kwargs
2023-09-12 09:40:14 +05:00
)
return bend
rep_items.append(get_bend_representation_item())
2023-09-12 15:25:54 +05:00
if start_length:
rep_items.append(self.extrude(profile, start_length, extrusion_vector=(0, 0, z_sign)))
2023-09-12 15:25:54 +05:00
if end_length:
end_position = O + get_circle_points(np.full(1, theta), radius + profile_dim[lateral_axis])[0]
end_position[np_Z] += start_length * z_sign
2023-09-12 15:25:54 +05:00
# define extrusion space for the segment after the bend
z_axis = get_circle_tangent(theta)
extrude_kwargs = {
"position_z_axis": z_axis,
"extrusion_vector": (0, 0, 1),
2023-09-12 15:25:54 +05:00
}
# since we are sure that tangent involves only two axis
# it's safe to assume that non lateral axis is untouched
if lateral_axis == 0:
x_axis = np.cross(z_axis, (0, 1, 0))
2023-09-12 15:25:54 +05:00
else:
x_axis = (1, 0, 0)
2023-09-12 15:25:54 +05:00
extrude_kwargs["position_x_axis"] = x_axis
rep_items.append(self.extrude(profile, end_length, end_position, **extrude_kwargs))
2023-08-28 15:25:10 +05:00
body = ifcopenshell.util.representation.get_context(self.file, "Model", "Body", "MODEL_VIEW")
assert body
2023-08-28 15:25:10 +05:00
rep = self.get_representation(body, rep_items)
bend_data = {
"start_length": start_length,
"end_length": end_length,
"radius": radius,
"angle": degrees(theta),
"lateral_axis": lateral_axis,
"lateral_sign": lateral_sign,
"z_axis_sign": -1 if flip_z_axis else 1,
"main_profile_dimension": profile_dim[lateral_axis],
}
2023-08-28 15:25:10 +05:00
return rep, bend_data