stbkit 0.1.0a9.dev1__py3-none-any.whl

stbkit/tools/_internal/utils/atomic_writer.py

@@ -0,0 +1,57 @@
+# Copyright 2026 TAISEI CORPORATION
+# SPDX-License-Identifier: MPL-2.0
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+import os
+import tempfile
+from collections.abc import Callable
+from pathlib import Path
+from typing import TextIO
+
+
+def write_with_temp_file(output_path: Path, writer: Callable[[Path], None]) -> None:
+    output_path.parent.mkdir(parents=True, exist_ok=True)
+    with tempfile.NamedTemporaryFile(
+        mode="wb",
+        dir=output_path.parent,
+        prefix=f".{output_path.name}.",
+        suffix=".tmp",
+        delete=False,
+    ) as stream:
+        temporary_path: Path = Path(stream.name)
+    try:
+        writer(temporary_path)
+        os.replace(temporary_path, output_path)
+    finally:
+        temporary_path.unlink(missing_ok=True)
+
+
+def write_text_with_temp_file(
+    output_path: Path, text: str, *, encoding: str = "utf-8", newline: str = "\n"
+) -> None:
+    def _write(temporary_path: Path) -> None:
+        with open(temporary_path, "w", encoding=encoding, newline=newline) as stream:
+            stream.write(text)
+
+    write_with_temp_file(output_path, _write)
+
+
+def write_text_stream_with_temp_file(
+    output_path: Path,
+    writer: Callable[[TextIO], None],
+    *,
+    encoding: str = "utf-8",
+    newline: str = "\n",
+) -> None:
+    def _write(temporary_path: Path) -> None:
+        with open(
+            temporary_path,
+            "w",
+            encoding=encoding,
+            newline=newline,
+        ) as stream:
+            writer(stream)
+
+    write_with_temp_file(output_path, _write)

stbkit/tools/_internal/utils/float_rounding.py

@@ -0,0 +1,49 @@
+# Copyright 2026 TAISEI CORPORATION
+# SPDX-License-Identifier: MPL-2.0
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+from ..vectors import Vector2d, Vector3d
+
+
+def validated_round_digits(round_digits: int | None) -> int | None:
+    if round_digits is None:
+        return None
+    if round_digits < 0:
+        raise AssertionError("round_digitsは0以上である必要があります")
+    return round_digits
+
+
+def round_float(value: float, *, round_digits: int | None) -> float:
+    round_digits = validated_round_digits(round_digits)
+    if round_digits is None:
+        return value
+    return round(value, ndigits=round_digits)
+
+
+def round_vector2d(
+    value: Vector2d,
+    *,
+    round_digits: int | None,
+) -> Vector2d:
+    if round_digits is None:
+        return value
+    return Vector2d(
+        round_float(value.x, round_digits=round_digits),
+        round_float(value.y, round_digits=round_digits),
+    )
+
+
+def round_vector3d(
+    value: Vector3d,
+    *,
+    round_digits: int | None,
+) -> Vector3d:
+    if round_digits is None:
+        return value
+    return Vector3d(
+        round_float(value.x, round_digits=round_digits),
+        round_float(value.y, round_digits=round_digits),
+        round_float(value.z, round_digits=round_digits),
+    )

stbkit/tools/_internal/utils/girder_utils.py

@@ -0,0 +1,91 @@
+# Copyright 2025-2026 TAISEI CORPORATION
+# SPDX-License-Identifier: MPL-2.0
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+from dataclasses import dataclass
+from typing import Protocol
+
+from stbkit.core.data_model.common import StBridgeRoot
+from stbkit.core.stb_exceptions import SchemaError
+
+from ..vectors import Vector3d
+from .node_utils import get_node_coordinate
+
+
+class GirderLike(Protocol):
+    @property
+    def id_node_start(self) -> int: ...
+    @property
+    def id_node_start_or_none(self) -> int | None: ...
+    @property
+    def id_node_end(self) -> int: ...
+    @property
+    def id_node_end_or_none(self) -> int | None: ...
+    @property
+    def offset_start_x_or_none(self) -> float | None: ...
+    @property
+    def offset_start_y_or_none(self) -> float | None: ...
+    @property
+    def offset_start_z_or_none(self) -> float | None: ...
+    @property
+    def offset_end_x_or_none(self) -> float | None: ...
+    @property
+    def offset_end_y_or_none(self) -> float | None: ...
+    @property
+    def offset_end_z_or_none(self) -> float | None: ...
+    def _name_for_log(self) -> str: ...
+
+
+def girder_offset_start(girder: GirderLike) -> Vector3d:
+    return Vector3d(
+        girder.offset_start_x_or_none or 0.0,
+        girder.offset_start_y_or_none or 0.0,
+        girder.offset_start_z_or_none or 0.0,
+    )
+
+
+def girder_offset_end(girder: GirderLike) -> Vector3d:
+    return Vector3d(
+        girder.offset_end_x_or_none or 0.0,
+        girder.offset_end_y_or_none or 0.0,
+        girder.offset_end_z_or_none or 0.0,
+    )
+
+
+@dataclass(slots=True)
+class GirderLengthValue:
+    length: float
+    length_with_offset: float
+    pos_control_point_start: float
+    pos_control_point_end: float
+
+
+def girder_length(
+    stb_girder: GirderLike,
+    stb: StBridgeRoot,
+) -> GirderLengthValue:
+    if (
+        stb_girder.id_node_start_or_none is None
+        or stb_girder.id_node_end_or_none is None
+    ):
+        raise SchemaError(f"{stb_girder._name_for_log()} 節点idがNoneです")
+    coord_start: Vector3d = get_node_coordinate(stb, stb_girder.id_node_start)
+    coord_end: Vector3d = get_node_coordinate(stb, stb_girder.id_node_end)
+    offset_start: Vector3d = girder_offset_start(stb_girder)
+    offset_end: Vector3d = girder_offset_end(stb_girder)
+    vx: Vector3d = (coord_end - coord_start).normalize()
+    length: float = coord_start.distance_to(coord_end)
+    length_with_offset: float = (coord_start + offset_start).distance_to(
+        coord_end + offset_end
+    )
+    pos_control_point_start: float = vx.dot(offset_start)
+    pos_control_point_end: float = length - vx.dot(offset_end)
+
+    return GirderLengthValue(
+        length=length,
+        length_with_offset=length_with_offset,
+        pos_control_point_start=pos_control_point_start,
+        pos_control_point_end=pos_control_point_end,
+    )

stbkit/tools/_internal/utils/guid_utils.py

@@ -0,0 +1,30 @@
+# Copyright 2025-2026 TAISEI CORPORATION
+# SPDX-License-Identifier: MPL-2.0
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+from uuid import UUID
+
+from ..optional_dependencies.ifc_common import ensure_ifcopenshell
+
+
+def to_ifc_guid(guid: UUID) -> str:
+    """UUIDをIFCのGlobalId形式に変換する
+
+    TODO:ifcopenshell無しで変換できるようにする"""
+    ensure_ifcopenshell()
+    import ifcopenshell
+
+    return ifcopenshell.guid.compress(str(guid))
+
+
+def from_ifc_guid(ifc_guid: str) -> UUID:
+    """IFCのGlobalId形式をUUIDに変換する
+
+    TODO:ifcopenshell無しで変換できるようにする"""
+    ensure_ifcopenshell()
+    import ifcopenshell
+
+    decompressed = ifcopenshell.guid.expand(ifc_guid)
+    return UUID(decompressed)