First Commit

2026-05-31 10:17:09 +07:00
commit 17a9c69379
4547 changed files with 1170384 additions and 0 deletions
@@ -0,0 +1,92 @@
+# IfcOpenShell - IFC toolkit and geometry engine
+# Copyright (C) 2021 Dion Moult <dion@thinkmoult.com>
+#
+# 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
+
+import numpy as np
+
+import ifcopenshell
+import ifcopenshell.util.element
+import ifcopenshell.util.placement
+import ifcopenshell.util.unit
+from ifcopenshell.util.shape_builder import (
+    V,
+    VectorType,
+    ifc_safe_vector_type,
+    np_apply_matrix,
+)
+
+
+def create_axis_curve(
+    file: ifcopenshell.file,
+    *,
+    p1: VectorType,
+    p2: VectorType,
+    grid_axis: ifcopenshell.entity_instance,
+    is_si: bool = True,
+) -> None:
+    """Adds curve geometry to a grid axis to represent the axis extents
+
+    An IFC grid will have a minimum of two axes (typically perpendicular). Each
+    axis will then have a line which represents the extents of the axis.
+
+    Points are provided as 3D coordinates in world space.
+    During axis creation, the coordinates will be localized relative to IfcGrid
+    and saved as 2D.
+
+    :param p1: The first point of the grid axis
+    :param p2: The second point of the grid axis
+    :param grid_axis: The IfcGridAxis element to add geometry to.
+    :param is_si: If true, the points are in meters, not project units
+
+    Example:
+
+    .. code:: python
+
+        # A pretty standard rectangular grid, with only two axes.
+        grid = ifcopenshell.api.root.create_entity(model, ifc_class="IfcGrid")
+        axis_a = ifcopenshell.api.grid.create_grid_axis(model,
+            axis_tag="A", uvw_axes="UAxes", grid=grid)
+        axis_1 = ifcopenshell.api.grid.create_grid_axis(model,
+            axis_tag="1", uvw_axes="VAxes", grid=grid)
+
+        # By convention, alphabetic grids are horizontal, and numeric are vertical
+        ifcopenshell.api.grid.create_axis_curve(
+            model, p1=np.array((0., 0., 0.)), p2=np.array((10., 0., 0.)), grid_axis=axis_a)
+        ifcopenshell.api.grid.create_axis_curve(
+            model, p1=np.array((0., 0., 0.)), p2=np.array((0., 10., 0.)), grid_axis=axis_1)
+    """
+    existing_curve = grid_axis.AxisCurve
+    points = V([p1, p2])
+    if is_si:
+        unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file)
+        points /= unit_scale
+
+    grid = next(i for i in file.get_inverse(grid_axis) if i.is_a("IfcGrid"))
+    grid_matrix_i = np.linalg.inv(ifcopenshell.util.placement.get_local_placement(grid.ObjectPlacement))
+    p1, p2 = ifc_safe_vector_type(np_apply_matrix(points, grid_matrix_i))
+    grid_axis.AxisCurve = file.create_entity(
+        "IfcPolyline",
+        (
+            file.create_entity("IfcCartesianPoint", p1[:2]),
+            file.create_entity("IfcCartesianPoint", p2[:2]),
+        ),
+    )
+
+    if existing_curve:
+        ifcopenshell.util.element.remove_deep2(file, existing_curve)