stbkit 0.1.0a9.dev1__py3-none-any.whl

stbkit/tools/_internal/data_model/building_geometry_web.py

@@ -0,0 +1,106 @@
+# 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 stbkit.core.data_model.common import StBridgeRoot
+from stbkit.core.stb_io import loads
+from stbkit.core.stb_reporting import Reporter, get_reporter
+
+from stbkit.api import upgrade_to_latest
+from stbkit.api.stb_latest import StBridge
+from stbkit.tools.converters._internal.stb_to_building_geometry import (
+    stb_to_building_geometry,
+)
+
+from .building_geometry import BuildingGeometry, OptionalUuidColumn
+
+
+def _uuid_strings(column: OptionalUuidColumn) -> list[str | None]:
+    def _get_string(index: int) -> str | None:
+        value = column.get(index)
+        return None if value is None else str(value)
+
+    return [_get_string(index) for index in range(len(column))]
+
+
+def building_geometry_to_web_payload(
+    geometry: BuildingGeometry,
+) -> dict[str, object]:
+    geometry.freeze()
+
+    return {
+        "origin_xyz": list(geometry.origin_xyz),
+        "mesh": {
+            "positions": list(geometry.mesh.positions),
+            "face_vertex_indices": list(geometry.mesh.face_vertex_indices),
+            "face_offsets": list(geometry.mesh.face_offsets),
+        },
+        "elements": {
+            "kind": list(geometry.elements.kind),
+            "flags": list(geometry.elements.flags),
+            "vertex_start": list(geometry.elements.vertex_start),
+            "vertex_count": list(geometry.elements.vertex_count),
+            "face_start": list(geometry.elements.face_start),
+            "face_count": list(geometry.elements.face_count),
+            "line_shape_index": list(geometry.elements.line_shape_index),
+            "plane_shape_index": list(geometry.elements.plane_shape_index),
+            "node_start": list(geometry.elements.node_start),
+            "node_count": list(geometry.elements.node_count),
+            "story_start": list(geometry.elements.story_start),
+            "story_count": list(geometry.elements.story_count),
+            "guid": _uuid_strings(geometry.elements.guid),
+            "xpath": list(geometry.elements.xpath),
+            "name": list(geometry.elements.name),
+            "source_type": list(geometry.elements.source_type),
+            "source_id": list(geometry.elements.source_id),
+        },
+        "line_shapes": {
+            "start_end_xyz": list(geometry.line_shapes.start_end_xyz),
+            "reference_y_xyz": list(geometry.line_shapes.reference_y_xyz),
+            "reference_y_valid": list(geometry.line_shapes.reference_y_valid),
+        },
+        "plane_shapes": {
+            "points_xyz": list(geometry.plane_shapes.points_xyz),
+            "loop_point_offsets": list(geometry.plane_shapes.loop_point_offsets),
+            "plane_loop_offsets": list(geometry.plane_shapes.plane_loop_offsets),
+            "loop_types": list(geometry.plane_shapes.loop_types),
+        },
+        "nodes": {
+            "positions_xyz": list(geometry.nodes.positions_xyz),
+            "source_id": list(geometry.nodes.source_id),
+            "guid": _uuid_strings(geometry.nodes.guid),
+            "story_index": list(geometry.nodes.story_index),
+            "xpath": list(geometry.nodes.xpath),
+        },
+        "stories": {
+            "source_id": list(geometry.stories.source_id),
+            "name": list(geometry.stories.name),
+            "elevations": list(geometry.stories.elevations),
+        },
+        "axes": {
+            "points_xyz": list(geometry.axes.points_xyz),
+            "point_start": list(geometry.axes.point_start),
+            "point_count": list(geometry.axes.point_count),
+            "source_id": list(geometry.axes.source_id),
+            "name": list(geometry.axes.name),
+        },
+        "element_node_indices": list(geometry.element_node_indices),
+        "element_story_indices": list(geometry.element_story_indices),
+    }
+
+
+def stb_xml_to_web_payload(
+    xml: str,
+    *,
+    reporter: Reporter | None = None,
+) -> dict[str, object]:
+    actual_reporter: Reporter = reporter or get_reporter()
+    root: StBridgeRoot = loads(xml, reporter=actual_reporter)
+    stb: StBridge = upgrade_to_latest(root, reporter=actual_reporter)
+    geometry: BuildingGeometry = stb_to_building_geometry(
+        stb,
+        reporter=actual_reporter,
+    )
+    return building_geometry_to_web_payload(geometry)

stbkit/tools/_internal/data_model/element_data.py

@@ -0,0 +1,93 @@
+# 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, field
+from enum import StrEnum
+from uuid import UUID
+
+from stbkit.tools._internal.vectors import Vector2d, Vector3d
+
+from .shape_data import (
+    ShapePair,
+    ShapePosition,
+)
+
+
+class ElementType(StrEnum):
+    COLUMN = "column"
+    POST = "post"
+    GIRDER = "girder"
+    BEAM = "beam"
+    BRACE = "brace"
+    SLAB = "slab"
+    WALL = "wall"
+    ISOLATING_DEVICE = "isolating_device"
+    DAMPING_DEVICE = "damping_device"
+    FRAME_DAMPING_DEVICE = "frame_damping_device"
+    FOOTING = "footing"
+    STRIP_FOOTING = "strip_footing"
+    PILE = "pile"
+    FOUNDATION_COLUMN = "foundation_column"
+    PARAPET = "parapet"
+    OPEN = "open"
+    PENETRATION = "penetration"
+    JOINT = "joint"
+    PANEL_ZONE = "panel_zone"
+    CONNECTION = "connection"
+    OTHER = "other"
+
+
+@dataclass(frozen=True, kw_only=True, slots=True)
+class Node:
+    position: Vector3d
+    xpath: str | None = None
+    guid: UUID | None = None
+
+
+@dataclass(eq=False, order=False, kw_only=True, slots=True)
+class ElementPlane:
+    element_type: ElementType | None = None
+    name: str | None = None
+    guid: UUID | None = None
+    points: list[Vector3d] | None = None
+    thickness: float | None = None
+    opens: list[list[Vector2d]] = field(default_factory=list)
+    path: str | None = None
+    node_ids: list[int] = field(default_factory=list)
+    node_points: list[Vector3d] | None = None
+
+
+@dataclass(eq=False, order=False, kw_only=True, slots=True)
+class ElementLine:
+    element_type: ElementType | None = None
+    name: str | None = None
+    guid: UUID | None = None
+    lengths: list[float] = field(default_factory=list)
+    point_start: Vector3d | None = None
+    point_end: Vector3d | None = None
+    angle: float = 0.0
+    shapes: list[ShapePair] = field(default_factory=list)
+    axis_position: ShapePosition = ShapePosition.CENTER_CENTER
+    path: str | None = None
+    node_ids: list[int] = field(default_factory=list)
+
+
+@dataclass(eq=False, order=False, kw_only=True, slots=True)
+class ElementLineSrc:
+    element_type: ElementType | None = None
+    name: str | None = None
+    guid: UUID | None = None
+    steel: ElementLine | None = None
+    rc: ElementLine | None = None
+    path: str | None = None
+    node_ids: list[int] = field(default_factory=list)
+
+
+class ElementData(list[ElementLine | ElementPlane | ElementLineSrc]):
+    def __init__(
+        self, elements: list[ElementLine | ElementPlane | ElementLineSrc] | None = None
+    ):
+        super().__init__(elements or [])

stbkit/tools/_internal/data_model/ifc_data/entity_data.py

@@ -0,0 +1,67 @@
+# 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, field
+from typing import TYPE_CHECKING
+from uuid import UUID
+
+from stbkit.core.stb_reporting import Reporter
+
+from ..element_data import ElementData
+
+if TYPE_CHECKING:
+    import ifcopenshell.file
+
+
+@dataclass(eq=False, order=False, kw_only=True, slots=True)
+class IfcBuildingStorey:
+    name: str | None = None
+    guid: UUID | None = None
+    elements: ElementData = field(default_factory=ElementData)
+
+
+@dataclass(eq=False, order=False, kw_only=True, slots=True)
+class IfcBuilding:
+    name: str | None = None
+    guid: UUID | None = None
+    building_storeys: list[IfcBuildingStorey] = field(default_factory=list)
+
+
+@dataclass(eq=False, order=False, kw_only=True, slots=True)
+class IfcSite:
+    name: str | None = None
+    guid: UUID | None = None
+    building: IfcBuilding | None = None
+
+
+@dataclass(eq=False, order=False, kw_only=True, slots=True)
+class IfcProject:
+    name: str | None = None
+    guid: UUID | None = None
+    site: IfcSite | None = None
+
+
+@dataclass(eq=False, order=False, kw_only=True, slots=True)
+class IfcModel:
+    project: IfcProject | None = None
+
+    def to_ifc(
+        self,
+        *,
+        reporter: Reporter,
+        round_digits: int | None = None,
+        round_digits_mm: int | None = None,
+    ) -> "ifcopenshell.file":
+        from ...optional_dependencies.ifc_exporter.ifc_entity_exporter import (
+            IfcEntityExporter,
+        )
+
+        return IfcEntityExporter.ifc_model_to_file(
+            self,
+            reporter=reporter,
+            round_digits=round_digits,
+            round_digits_mm=round_digits_mm,
+        )

stbkit/tools/_internal/data_model/ifc_data/entity_names.py

@@ -0,0 +1,32 @@
+# 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 enum import StrEnum
+
+
+class EntityName(StrEnum):
+    IFC_BEAM = "IfcBeam"
+    IFC_BUILDING = "IfcBuilding"
+    IFC_BUILDING_STOREY = "IfcBuildingStorey"
+    IFC_C_SHAPE_PROFILE_DEF = "IfcCShapeProfileDef"
+    IFC_CIRCLE_PROFILE_DEF = "IfcCircleProfileDef"
+    IFC_CIRCLE_HOLLOW_PROFILE_DEF = "IfcCircleHollowProfileDef"
+    IFC_COLUMN = "IfcColumn"
+    IFC_ELEMENT_ASSEMBLY = "IfcElementAssembly"
+    IFC_FOOTING = "IfcFooting"
+    IFC_I_SHAPE_PROFILE_DEF = "IfcIShapeProfileDef"
+    IFC_MEMBER = "IfcMember"
+    IFC_PILE = "IfcPile"
+    IFC_PROJECT = "IfcProject"
+    IFC_REL_AGGREGATES = "IfcRelAggregates"
+    IFC_REL_CONTAINED_IN_SPATIAL_STRUCTURE = "IfcRelContainedInSpatialStructure"
+    IFC_RECTANGLE_HOLLOW_PROFILE_DEF = "IfcRectangleHollowProfileDef"
+    IFC_RECTANGLE_PROFILE_DEF = "IfcRectangleProfileDef"
+    IFC_SITE = "IfcSite"
+    IFC_SLAB = "IfcSlab"
+    IFC_T_SHAPE_PROFILE_DEF = "IfcTShapeProfileDef"
+    IFC_U_SHAPE_PROFILE_DEF = "IfcUShapeProfileDef"
+    IFC_WALL = "IfcWall"