93 lines
3.4 KiB
Python
93 lines
3.4 KiB
Python
# 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)
|