Compare commits
16 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
 Jonathon Broughton
|
fac5601b35 | ||
|
NLSA
|
cde3a58b08 | ||
|
NLSA
|
682a21130f | ||
|
NLSA
|
50e62020ef | ||
|
NLSA
|
1681d756e8 | ||
|
NLSA
|
0fb8ceb73a | ||
|
NLSA
|
09dc8eb8c8 | ||
|
NLSA
|
38e05ccc28 | ||
|
NLSA
|
f9193ee3a0 | ||
|
NLSA
|
112f031608 | ||
|
NLSA
|
673c024ac5 | ||
|
NLSA
|
f8c9d4237d | ||
|
NLSA
|
11acb02fd1 | ||
|
NLSA
|
f7aa6c29da | ||
|
NLSA
|
63082a881c | ||
|
NLSA
|
bdd030ba86 |
@@ -30,3 +30,4 @@ jobs:
|
||||
speckle_function_id: ${{ secrets.SPECKLE_FUNCTION_ID }}
|
||||
speckle_function_input_schema_file_path: ${{ env.FUNCTION_SCHEMA_FILE_NAME }}
|
||||
speckle_function_command: "python -u main.py run"
|
||||
speckle_function_recommended_memory_mi: 8000
|
||||
|
||||
@@ -1,174 +1,240 @@
|
||||
# Speckle Automate function template - Python
|
||||
# Speckle-Revit to IFC 4.3 Exporter
|
||||
|
||||
This template repository is for a Speckle Automate function written in Python
|
||||
using the [specklepy](https://pypi.org/project/specklepy/) SDK to interact with Speckle data.
|
||||
## 🚧 Project Status: WIP
|
||||
Hey there! This project is still under active development, so expect changes, bugs, and incomplete features.
|
||||
If you have any questions or suggestions, don’t hesitate to reach out at: **your@email.com**
|
||||
|
||||
This template contains the full scaffolding required to publish a function to the Automate environment.
|
||||
It also has some sane defaults for development environment setups.
|
||||
A [Speckle Automate](https://automate.speckle.dev/) function that converts Speckle Revit models into IFC 4.3 files using [ifcopenshell](https://ifcopenshell.org/). This exporter is specifically designed for models sent to Speckle from Autodesk Revit and relies on Revit-specific object structures, categories, and parameters.
|
||||
|
||||
## Getting started
|
||||
## What It Does
|
||||
|
||||
1. Use this template repository to create a new repository in your own / organization's profile.
|
||||
1. Register the function
|
||||
The exporter receives a Speckle model version, walks its object tree, and produces a standards-compliant IFC 4.3 file. Each Speckle object becomes an IFC element with:
|
||||
|
||||
### Add new dependencies
|
||||
- Correct IFC entity classification (IfcWall, IfcSlab, IfcColumn, etc.)
|
||||
- Tessellated geometry (IfcPolygonalFaceSet)
|
||||
- Curve geometry for Lines and Arcs (IfcGeometricCurveSet with IfcPolyline)
|
||||
- Material colours from Speckle render materials
|
||||
- Revit property sets (Common psets, instance/type parameters, material quantities)
|
||||
- IFC type objects (IfcWallType, IfcSlabType, etc.) shared across instances
|
||||
- Spatial structure (IfcProject > IfcSite > IfcBuilding > IfcBuildingStorey)
|
||||
- IfcSpace elements aggregated under storeys with Room properties
|
||||
- Automatic skipping of analytical/energy categories (e.g. Energy Analysis, MEP Analytical, Solar Shading)
|
||||
|
||||
To add new Python package dependencies to the project, edit the `pyproject.toml` file:
|
||||
## Pipeline Overview
|
||||
|
||||
**For packages your function needs to run** (like pandas, requests, etc.):
|
||||
```toml
|
||||
dependencies = [
|
||||
"specklepy==3.0.0",
|
||||
"pandas==2.1.0", # Add production dependencies here
|
||||
]
|
||||
```
|
||||
Speckle Model
|
||||
│
|
||||
▼
|
||||
1. Receive version (specklepy)
|
||||
│
|
||||
▼
|
||||
2. Build definition map (for instance geometry reuse)
|
||||
│
|
||||
▼
|
||||
3. Create IFC scaffold (Project → Site → Building)
|
||||
│
|
||||
▼
|
||||
4. Traverse object tree
|
||||
│ For each leaf element:
|
||||
│ ├── Classify → IFC entity class (skip analytical categories)
|
||||
│ ├── Convert geometry → IfcPolygonalFaceSet or IfcGeometricCurveSet
|
||||
│ ├── Create IFC element + placement
|
||||
│ ├── Write property sets & quantities
|
||||
│ └── Assign IFC type object
|
||||
│
|
||||
▼
|
||||
5. Flush spatial containment & type relationships
|
||||
│
|
||||
▼
|
||||
6. Write .ifc file
|
||||
```
|
||||
|
||||
**For development tools** (like testing or formatting tools):
|
||||
```toml
|
||||
[project.optional-dependencies]
|
||||
dev = [
|
||||
"black==23.12.1",
|
||||
"pytest-mock==3.11.1", # Add development dependencies here
|
||||
# ... other dev tools
|
||||
]
|
||||
```
|
||||
## Module Structure
|
||||
|
||||
**How to decide which section?**
|
||||
- If your `main.py` (or other function logic) imports it → `dependencies`
|
||||
- If it's just a tool to help you code → `[project.optional-dependencies].dev`
|
||||
| File | Purpose |
|
||||
|------|---------|
|
||||
| `main.py` | Entry point, orchestrates the full pipeline |
|
||||
| `utils/traversal.py` | Walks the Speckle Collection tree (Project > Level > Category > Element) |
|
||||
| `utils/mapper.py` | Classifies Speckle objects into IFC entity types |
|
||||
| `utils/geometry.py` | Converts Speckle meshes to IfcPolygonalFaceSet and Lines/Arcs to IfcGeometricCurveSet geometry |
|
||||
| `utils/instances.py` | Handles InstanceProxy objects with shared geometry (IfcMappedItem) |
|
||||
| `utils/properties.py` | Writes IFC property sets and quantities from Revit parameters |
|
||||
| `utils/type_manager.py` | Creates and caches IfcTypeObjects (IfcWallType, etc.) |
|
||||
| `utils/materials.py` | Maps Speckle render materials to IfcSurfaceStyle colours |
|
||||
| `utils/writer.py` | Creates the IFC file scaffold and manages storey creation |
|
||||
| `utils/receiver.py` | Connects to Speckle server and receives model data (uses `.env`) |
|
||||
|
||||
Example:
|
||||
```python
|
||||
# In your main.py
|
||||
import pandas as pd # ← This goes in dependencies
|
||||
import specklepy # ← This goes in dependencies
|
||||
## Mapping Logic
|
||||
|
||||
# You won't import these in main.py:
|
||||
# pytest, black, mypy ← These go in [project.optional-dependencies].dev
|
||||
```
|
||||
Classification of Speckle objects to IFC entity types follows a priority chain. The first match wins.
|
||||
|
||||
### Change launch variables
|
||||
### Priority 1: `builtInCategory` (OST_ enum)
|
||||
|
||||
Describe how the launch.json should be edited.
|
||||
The most reliable source. Read from `obj.properties.builtInCategory`, which contains the Revit `BuiltInCategory` enum value.
|
||||
|
||||
### GitHub Codespaces
|
||||
Examples:
|
||||
| builtInCategory | IFC Class |
|
||||
|---|---|
|
||||
| `OST_Walls` | `IfcWall` |
|
||||
| `OST_Floors` | `IfcSlab` |
|
||||
| `OST_StructuralColumns` | `IfcColumn` |
|
||||
| `OST_StructuralFraming` | `IfcBeam` |
|
||||
| `OST_Doors` | `IfcDoor` |
|
||||
| `OST_Windows` | `IfcWindow` |
|
||||
| `OST_Roofs` | `IfcRoof` |
|
||||
| `OST_CurtainWallPanels` | `IfcCurtainWall` |
|
||||
| `OST_DuctCurves` | `IfcDuctSegment` |
|
||||
| `OST_PipeCurves` | `IfcPipeSegment` |
|
||||
| `OST_PipeFitting` | `IfcPipeFitting` |
|
||||
| `OST_PlumbingEquipment` | `IfcSanitaryTerminal` |
|
||||
| `OST_Rebar` | `IfcReinforcingBar` |
|
||||
| `OST_StructConnections` | `IfcMechanicalFastener` |
|
||||
| `OST_LightingFixtures` | `IfcLightFixture` |
|
||||
| `OST_Furniture` | `IfcFurnishingElement` |
|
||||
| `OST_Rooms` | `IfcSpace` |
|
||||
|
||||
Create a new repo from this template, and use the create new code.
|
||||
The full table covers ~70 Revit categories across Architectural, Structural, MEP (HVAC, Plumbing, Electrical), and Site/Civil disciplines.
|
||||
|
||||
### Using this Speckle Function
|
||||
### Skipped Categories
|
||||
|
||||
1. [Create](https://automate.speckle.dev/) a new Speckle Automation.
|
||||
1. Select your Speckle Project and Speckle Model.
|
||||
1. Select the deployed Speckle Function.
|
||||
1. Enter a phrase to use in the comment.
|
||||
1. Click `Create Automation`.
|
||||
The following analytical/energy OST categories are automatically skipped (not exported to IFC):
|
||||
|
||||
## Getting Started with Creating Your Own Speckle Function
|
||||
`OST_MEPLoadAreaSeparationLines`, `OST_EnergyAnalysisZones`, `OST_EnergyAnalysisSurface`, `OST_SolarShading`, `OST_MEPAnalyticalPipeSegments`, `OST_MEPAnalyticalDuctSegments`, `OST_MEPAnalyticalSpaces`, `OST_ElectricalConduitAnalyticalLines`, `OST_MEPLoadBoundaryLines`, `OST_FlowTerminalSeparationLines`
|
||||
|
||||
1. [Register](https://automate.speckle.dev/) your Function with [Speckle Automate](https://automate.speckle.dev/) and select the Python template.
|
||||
1. A new repository will be created in your GitHub account.
|
||||
1. Make changes to your Function in `main.py`. See below for the Developer Requirements and instructions on how to test.
|
||||
1. To create a new version of your Function, create a new [GitHub release](https://docs.github.com/en/repositories/releasing-projects-on-github/managing-releases-in-a-repository) in your repository.
|
||||
### Priority 2: Category name (display string)
|
||||
|
||||
## Developer Requirements
|
||||
The category name from the traversal context (the name of the parent Collection in the Speckle tree). Exact match first, then case-insensitive substring match.
|
||||
|
||||
1. Install the following:
|
||||
- [Python 3.11+](https://www.python.org/downloads/)
|
||||
1. Run the following to set up your development environment:
|
||||
```bash
|
||||
python -m venv .venv
|
||||
# On Windows
|
||||
.venv\Scripts\activate
|
||||
# On macOS/Linux
|
||||
source .venv/bin/activate
|
||||
Examples:
|
||||
| Category Name | IFC Class |
|
||||
|---|---|
|
||||
| `Walls` | `IfcWall` |
|
||||
| `Structural Columns` | `IfcColumn` |
|
||||
| `Plumbing Fixtures` | `IfcSanitaryTerminal` |
|
||||
| `Structural Rebar` | `IfcReinforcingBar` |
|
||||
| `Structural Connections` | `IfcMechanicalFastener` |
|
||||
| `Lighting Fixtures` | `IfcLightFixture` |
|
||||
|
||||
pip install --upgrade pip
|
||||
pip install .[dev]
|
||||
```
|
||||
### Priority 3: `obj.category` field
|
||||
|
||||
**What this installs:**
|
||||
- All the packages your function needs to run (`dependencies`)
|
||||
- Plus development tools like testing and code formatting (`[project.optional-dependencies].dev`)
|
||||
Same lookup as Priority 2, but using the object's own `category` attribute.
|
||||
|
||||
**Why separate sections?**
|
||||
- `dependencies`: Only what gets deployed with your function (lightweight)
|
||||
- `dev` dependencies: Extra tools to help you write better code locally
|
||||
### Fallback
|
||||
|
||||
## Building and Testing
|
||||
If none of the above match, the object is classified as `IfcBuildingElementProxy`.
|
||||
|
||||
The code can be tested locally by running `pytest`.
|
||||
## Geometry Handling
|
||||
|
||||
### Alternative dependency managers
|
||||
### Direct Meshes (Path B1)
|
||||
|
||||
This template uses the modern **PEP 621** standard in `pyproject.toml`, which works with all modern Python dependency managers:
|
||||
Objects with `displayValue` containing Mesh objects are converted directly:
|
||||
|
||||
#### Using Poetry
|
||||
```bash
|
||||
poetry install # Automatically reads pyproject.toml
|
||||
```
|
||||
1. Extract vertices and faces from each mesh in `displayValue`
|
||||
2. Scale vertices to millimetres based on the mesh's unit declaration
|
||||
3. Deduplicate vertices via snap grid (0.01mm tolerance) to avoid IFC GEM111 errors
|
||||
4. Build `IfcPolygonalFaceSet` with `IfcCartesianPointList3D` + `IfcIndexedPolygonalFace`
|
||||
5. Compute bounding box origin for `IfcLocalPlacement`, offset vertices relative to it
|
||||
|
||||
#### Using uv
|
||||
```bash
|
||||
uv sync # Automatically reads pyproject.toml
|
||||
```
|
||||
### Instance Objects (Path A / B2)
|
||||
|
||||
#### Using pip-tools
|
||||
```bash
|
||||
pip-compile pyproject.toml # Generate requirements.txt from pyproject.toml
|
||||
pip install -r requirements.txt
|
||||
```
|
||||
Speckle `InstanceProxy` objects reference shared definition geometry via `definitionId`. The exporter supports two formats:
|
||||
|
||||
#### Using pdm
|
||||
```bash
|
||||
pdm install # Automatically reads pyproject.toml
|
||||
```
|
||||
- **Revit format**: `definitionId` is a 64-char hex hash; geometry is found by walking the object tree
|
||||
- **IFC format**: `definitionId` starts with `DEFINITION:`; geometry is in `definitionGeometry` collection
|
||||
|
||||
**Advantage**: All tools read the same `pyproject.toml` file, so there's no need to keep multiple files in sync!
|
||||
Performance optimisation: geometry is built once as an `IfcRepresentationMap`, then each instance references it via `IfcMappedItem` + `IfcCartesianTransformationOperator3DnonUniform`. This avoids duplicating vertex data across hundreds of identical elements.
|
||||
|
||||
### Building and running the Docker Container Image
|
||||
### Curve Geometry (Path B3)
|
||||
|
||||
Running and testing your code on your machine is a great way to develop your Function; the following instructions are a bit more in-depth and only required if you are having issues with your Function in GitHub Actions or on Speckle Automate.
|
||||
Objects whose `displayValue` contains `Objects.Geometry.Line` or `Objects.Geometry.Arc` items (and no meshes or instances) are exported as curve geometry:
|
||||
|
||||
#### Building the Docker Container Image
|
||||
- **Lines** → `IfcPolyline` with start and end points
|
||||
- **Arcs** → `IfcPolyline` approximated with 8 segments, sampled parametrically from the arc's plane origin, radius, and domain angles. Falls back to start/mid/end points if plane data is unavailable.
|
||||
|
||||
The GitHub Action packages your code into the format required by Speckle Automate. This is done by building a Docker Image, which Speckle Automate runs. You can attempt to build the Docker Image locally to test the building process.
|
||||
All curves are wrapped in an `IfcGeometricCurveSet` inside an `IfcShapeRepresentation` with `RepresentationType="GeometricCurveSet"`.
|
||||
|
||||
To build the Docker Container Image, you must have [Docker](https://docs.docker.com/get-docker/) installed.
|
||||
### Composite Objects (Path B2 — merged instances)
|
||||
|
||||
Once you have Docker running on your local machine:
|
||||
Objects like Windows and Doors may have multiple `InstanceProxy` items in their `displayValue` (e.g. frame, glass, sill). These are **not** separate IFC elements — all instance geometries are merged into a single `IfcShapeRepresentation` with combined `IfcMappedItem` entries, producing one IFC element per Speckle object.
|
||||
|
||||
1. Open a terminal
|
||||
1. Navigate to the directory in which you cloned this repository
|
||||
1. Run the following command:
|
||||
## Property Sets
|
||||
|
||||
```bash
|
||||
docker build -f ./Dockerfile -t speckle_automate_python_example .
|
||||
```
|
||||
The exporter writes property sets matching Revit's native IFC export structure:
|
||||
|
||||
#### Running the Docker Container Image
|
||||
| Property Set | Content |
|
||||
|---|---|
|
||||
| `Pset_<Entity>Common` | Standard IFC properties: Reference, IsExternal, LoadBearing, ThermalTransmittance |
|
||||
| `Pset_SpaceCommon` | Room-specific: Reference, RoomNumber, RoomName, Category (Occupant) |
|
||||
| `RVT_InstanceParameters` | All Revit instance parameters |
|
||||
| `RVT_Identity` | Family, Type, ElementId, BuiltInCategory |
|
||||
|
||||
Once the GitHub Action has built the image, it is sent to Speckle Automate. When Speckle Automate runs your Function as part of an Automation, it will run the Docker Container Image. You can test that your Docker Container Image runs correctly locally.
|
||||
## Quantities
|
||||
|
||||
1. To then run the Docker Container Image, run the following command:
|
||||
Quantities follow the IFC standard naming convention: `Qto_<EntityType>BaseQuantities` and `Qto_<MaterialName>BaseQuantities`.
|
||||
|
||||
```bash
|
||||
docker run --rm speckle_automate_python_example \
|
||||
python -u main.py run \
|
||||
'{"projectId": "1234", "modelId": "1234", "branchName": "myBranch", "versionId": "1234", "speckleServerUrl": "https://speckle.xyz", "automationId": "1234", "automationRevisionId": "1234", "automationRunId": "1234", "functionId": "1234", "functionName": "my function", "functionLogo": "base64EncodedPng"}' \
|
||||
'{}' \
|
||||
yourSpeckleServerAuthenticationToken
|
||||
```
|
||||
| Quantity Set | Content |
|
||||
|---|---|
|
||||
| `Qto_<EntityType>BaseQuantities` | Element-level quantities from Revit computed parameters (area, volume, length, width, height, perimeter) |
|
||||
| `Qto_SpaceBaseQuantities` | Room quantities: NetFloorArea, NetVolume |
|
||||
| `Qto_<MaterialName>BaseQuantities` | Per-material quantities: GrossArea, GrossVolume, Density |
|
||||
|
||||
Let's explain this in more detail:
|
||||
### Element Quantity Mapping
|
||||
|
||||
`docker run—-rm speckle_automate_python_example` tells Docker to run the Docker Container Image we built earlier. `speckle_automate_python_example` is the name of the Docker Container Image. The `--rm` flag tells Docker to remove the container after it has finished running, freeing up space on your machine.
|
||||
| IFC Quantity | Revit Parameter(s) |
|
||||
|---|---|
|
||||
| GrossArea | `HOST_AREA_COMPUTED` |
|
||||
| GrossVolume | `HOST_VOLUME_COMPUTED` |
|
||||
| Length | `CURVE_ELEM_LENGTH`, `INSTANCE_LENGTH_PARAM` |
|
||||
| Height | `WALL_USER_HEIGHT_PARAM`, `FAMILY_HEIGHT_PARAM`, `INSTANCE_HEAD_HEIGHT_PARAM` |
|
||||
| Width | `INSTANCE_WIDTH_PARAM`, `FURNITURE_WIDTH`, `FLOOR_ATTR_THICKNESS_PARAM` |
|
||||
| Perimeter | `HOST_PERIMETER_COMPUTED` |
|
||||
|
||||
The line `python -u main.py run` is the command run inside the Docker Container Image. The rest of the command is the arguments passed to the command. The arguments are:
|
||||
### Supported Entity Qto Sets
|
||||
|
||||
- `'{"projectId": "1234", "modelId": "1234", "branchName": "myBranch", "versionId": "1234", "speckleServerUrl": "https://speckle.xyz", "automationId": "1234", "automationRevisionId": "1234", "automationRunId": "1234", "functionId": "1234", "functionName": "my function", "functionLogo": "base64EncodedPng"}'` - the metadata that describes the automation and the function.
|
||||
- `{}` - the input parameters for the function the Automation creator can set. Here, they are blank, but you can add your parameters to test your function.
|
||||
- `yourSpeckleServerAuthenticationToken`—the authentication token for the Speckle Server that the Automation can connect to. This is required to interact with the Speckle Server, for example, to get data from the Model.
|
||||
`Qto_WallBaseQuantities`, `Qto_SlabBaseQuantities`, `Qto_ColumnBaseQuantities`, `Qto_BeamBaseQuantities`, `Qto_DoorBaseQuantities`, `Qto_WindowBaseQuantities`, `Qto_RoofBaseQuantities`, `Qto_CoveringBaseQuantities`, `Qto_RailingBaseQuantities`, `Qto_StairBaseQuantities`, `Qto_RampBaseQuantities`, `Qto_MemberBaseQuantities`, `Qto_FootingBaseQuantities`, `Qto_CurtainWallBaseQuantities`, `Qto_BuildingElementProxyBaseQuantities`, `Qto_PipeFittingBaseQuantities`, `Qto_SanitaryTerminalBaseQuantities`, `Qto_ReinforcingElementBaseQuantities`, `Qto_MechanicalFastenerBaseQuantities`
|
||||
|
||||
## IfcSpace (Rooms)
|
||||
|
||||
Revit Rooms (`OST_Rooms`) are exported as `IfcSpace` elements with special handling:
|
||||
|
||||
- **Spatial relationship**: Aggregated under `IfcBuildingStorey` via `IfcRelAggregates` (not contained)
|
||||
- **Naming**: Uses the Speckle object `name` attribute (not Family:Type which is "none:none" for rooms)
|
||||
- **IfcSpace.Name**: Set to `ROOM_NUMBER`
|
||||
- **IfcSpace.LongName**: Set to `ROOM_NAME`
|
||||
- **Geometry**: Converted from `displayValue` meshes like any other element
|
||||
|
||||
## Function Inputs
|
||||
|
||||
| Input | Description |
|
||||
|---|---|
|
||||
| `file_name` | Output IFC filename (timestamp is appended automatically) |
|
||||
| `IFC_PROJECT_NAME` | Name for the IfcProject entity |
|
||||
| `IFC_SITE_NAME` | Name for the IfcSite entity |
|
||||
| `IFC_BUILDING_NAME` | Name for the IfcBuilding entity |
|
||||
|
||||
## Environment Variables
|
||||
|
||||
For local testing via `receiver.py`, configure a `.env` file:
|
||||
|
||||
| Variable | Description |
|
||||
|---|---|
|
||||
| `SPECKLE_SERVER_URL` | Speckle server URL (default: `https://app.speckle.systems`) |
|
||||
| `SPECKLE_TOKEN` | Personal access token for authentication |
|
||||
| `SPECKLE_PROJECT_ID` | Project (stream) ID |
|
||||
|
||||
## Testing
|
||||
|
||||
| Model Name | Revit Size | IFC Size | Conversion Time |
|
||||
|----------------------------------|------------|----------|-----------------|
|
||||
| Huge confidential model | 450 MB | 391 MB | 2h 30m |
|
||||
| Snowdon Towers (Architecture) | 93.2 MB | 118 MB | 8m 37s |
|
||||
| Speckle Tower | 51 MB | 45 MB | 3m |
|
||||
| Rac Basic Sample Model | 18.8 MB | 12 MB | 12s |
|
||||
|
||||
## Resources
|
||||
|
||||
- [Learn](https://speckle.guide/dev/python.html) more about SpecklePy and interacting with Speckle from Python.
|
||||
- [Speckle Developer Docs](https://speckle.guide/dev/python.html)
|
||||
- [ifcopenshell Documentation](https://ifcopenshell.org/)
|
||||
- [IFC 4.3 Schema](https://standards.buildingsmart.org/IFC/RELEASE/IFC4x3/HTML/)
|
||||
- [Revit BuiltInCategory Reference](https://www.revitapidocs.com/2019/ba1c5b30-242f-5fdc-8ea9-ec3b61e6e722.htm)
|
||||
|
||||
-27
@@ -1,27 +0,0 @@
|
||||
"""Helper module for a simple speckle object tree flattening."""
|
||||
|
||||
from collections.abc import Iterable
|
||||
|
||||
from specklepy.objects import Base
|
||||
|
||||
|
||||
def flatten_base(base: Base) -> Iterable[Base]:
|
||||
"""Flatten a base object into an iterable of bases.
|
||||
|
||||
This function recursively traverses the `elements` or `@elements` attribute of the
|
||||
base object, yielding each nested base object.
|
||||
|
||||
Args:
|
||||
base (Base): The base object to flatten.
|
||||
|
||||
Yields:
|
||||
Base: Each nested base object in the hierarchy.
|
||||
"""
|
||||
# Attempt to get the elements attribute, fallback to @elements if necessary
|
||||
elements = getattr(base, "elements", getattr(base, "@elements", None))
|
||||
|
||||
if elements is not None:
|
||||
for element in elements:
|
||||
yield from flatten_base(element)
|
||||
|
||||
yield base
|
||||
@@ -2,20 +2,19 @@ from datetime import datetime
|
||||
|
||||
import ifcopenshell.api
|
||||
|
||||
import utils.config as config
|
||||
|
||||
from utils.materials import MaterialManager
|
||||
from utils.traversal import traverse, print_tree
|
||||
from utils.mapper import classify
|
||||
from utils.geometry import mesh_to_ifc, get_display_instances, _make_placement
|
||||
from utils.instances import is_instance, instance_to_ifc, build_definition_map, print_instance_stats
|
||||
from utils.properties import write_properties, write_common_properties
|
||||
from utils.mapper import classify, reset_caches as reset_mapper_caches
|
||||
from utils.geometry import mesh_to_ifc, get_display_instances, curves_to_ifc, _make_placement
|
||||
from utils.instances import is_instance, instance_to_ifc, build_definition_map, print_instance_stats, get_definition_object
|
||||
from utils.properties import write_properties, write_common_properties, build_element_name, get_element_tag, get_ifc_guid, reset_caches as reset_props_caches
|
||||
from utils.writer import create_ifc_scaffold, StoreyManager
|
||||
from utils.type_manager import TypeManager
|
||||
|
||||
|
||||
SPATIAL_STRUCTURE_TYPES = {
|
||||
"IfcSite", "IfcBuilding", "IfcBuildingStorey",
|
||||
"IfcSpace", "IfcExternalSpatialElement", "IfcSpatialZone",
|
||||
"IfcBuilding", "IfcBuildingStorey",
|
||||
"IfcExternalSpatialElement", "IfcSpatialZone",
|
||||
"IfcGrid", "IfcAnnotation",
|
||||
}
|
||||
|
||||
@@ -59,7 +58,9 @@ def automate_function(
|
||||
print(" Speckle -> IFC4.3 Exporter")
|
||||
print("=" * 60)
|
||||
|
||||
#version_root_object = automate_context.receive_version()
|
||||
# Reset caches from any previous run
|
||||
reset_props_caches()
|
||||
reset_mapper_caches()
|
||||
|
||||
# ------------------------------------------------------------------ #
|
||||
# 1. Receive
|
||||
@@ -72,21 +73,24 @@ def automate_function(
|
||||
|
||||
# ------------------------------------------------------------------ #
|
||||
# 2. Build definition map (for instance resolution)
|
||||
# ------------------------------------------------------------------ #
|
||||
print("\n🔍 Building definition map...")
|
||||
# ----------------------------------------------
|
||||
definition_map = build_definition_map(base)
|
||||
|
||||
# ------------------------------------------------------------------ #
|
||||
# 3. Set up IFC
|
||||
# ------------------------------------------------------------------ #
|
||||
ifc, building, body_context = create_ifc_scaffold()
|
||||
ifc, _site, building, body_context = create_ifc_scaffold(
|
||||
project_name=function_inputs.IFC_PROJECT_NAME,
|
||||
site_name=function_inputs.IFC_SITE_NAME,
|
||||
building_name=function_inputs.IFC_BUILDING_NAME,
|
||||
)
|
||||
storey_manager = StoreyManager(ifc, building)
|
||||
|
||||
# ------------------------------------------------------------------ #
|
||||
# 3b. Build material map from renderMaterialProxies
|
||||
# ------------------------------------------------------------------ #
|
||||
print("\n🎨 Building material map...")
|
||||
material_manager = MaterialManager(ifc, base)
|
||||
type_manager = TypeManager(ifc)
|
||||
|
||||
# ------------------------------------------------------------------ #
|
||||
# 4. Traverse & export
|
||||
@@ -102,25 +106,44 @@ def automate_function(
|
||||
|
||||
ifc_class = classify(obj, category_name)
|
||||
|
||||
if ifc_class is None:
|
||||
continue
|
||||
|
||||
if ifc_class in SPATIAL_STRUCTURE_TYPES:
|
||||
skipped_spatial += 1
|
||||
continue
|
||||
|
||||
name = getattr(obj, "name", None) or getattr(obj, "applicationId", None) or getattr(obj, "id", "unnamed")
|
||||
# IfcSpace uses the Speckle object name (e.g. "Rooms - Live/Work Unit 507")
|
||||
# instead of Family:Type (which is "none:none" for Revit rooms)
|
||||
if ifc_class == "IfcSpace":
|
||||
name = getattr(obj, "name", None) or build_element_name(obj)
|
||||
else:
|
||||
name = build_element_name(obj)
|
||||
storey = storey_manager.get_or_create(level_name)
|
||||
|
||||
# ------------------------------------------------------------------ #
|
||||
# Path A: Instance object (has transform + definitionId, no displayValue)
|
||||
# ------------------------------------------------------------------ #
|
||||
if is_instance(obj):
|
||||
# Instances may lack category info — inherit from definition object
|
||||
if ifc_class == "IfcBuildingElementProxy":
|
||||
def_obj = get_definition_object(obj, definition_map)
|
||||
if def_obj:
|
||||
ifc_class = classify(def_obj, category_name)
|
||||
|
||||
rep, placement = instance_to_ifc(ifc, body_context, obj, definition_map, scale=scale, material_manager=material_manager)
|
||||
element = _create_element(ifc, ifc_class, name, rep, placement, storey)
|
||||
write_common_properties(ifc, element, obj, category_name)
|
||||
write_properties(ifc, element, obj)
|
||||
instance_count += 1
|
||||
total += 1
|
||||
if not rep:
|
||||
no_geometry += 1
|
||||
continue
|
||||
element = _create_element(ifc, ifc_class, name, rep, placement, storey,
|
||||
storey_manager=storey_manager,
|
||||
tag=get_element_tag(obj), guid=get_ifc_guid(obj),
|
||||
object_type=getattr(obj, "type", None),
|
||||
)
|
||||
write_properties(ifc, element, obj, ifc_class=ifc_class, category_name=category_name)
|
||||
type_manager.assign(element, obj, ifc_class)
|
||||
instance_count += 1
|
||||
total += 1
|
||||
|
||||
else:
|
||||
# ------------------------------------------------------------------ #
|
||||
@@ -131,29 +154,62 @@ def automate_function(
|
||||
|
||||
# B1: Mesh geometry on the parent object
|
||||
rep, placement = mesh_to_ifc(ifc, body_context, obj, scale=scale, material_manager=material_manager)
|
||||
element = _create_element(ifc, ifc_class, name, rep, placement, storey)
|
||||
write_common_properties(ifc, element, obj, category_name)
|
||||
write_properties(ifc, element, obj)
|
||||
total += 1
|
||||
if not rep:
|
||||
no_geometry += 1
|
||||
if rep:
|
||||
element = _create_element(ifc, ifc_class, name, rep, placement, storey,
|
||||
storey_manager=storey_manager,
|
||||
tag=get_element_tag(obj), guid=get_ifc_guid(obj),
|
||||
object_type=getattr(obj, "type", None),
|
||||
)
|
||||
write_properties(ifc, element, obj, ifc_class=ifc_class, category_name=category_name)
|
||||
type_manager.assign(element, obj, ifc_class)
|
||||
total += 1
|
||||
|
||||
# B2: Instance objects nested inside displayValue
|
||||
# Each becomes its own IFC element (same class as parent)
|
||||
# Use the parent object's name — the InstanceProxy has no meaningful name
|
||||
# All instances are parts of the SAME element (e.g. window frame + glass + sill)
|
||||
# Merge all into a single IFC element with combined geometry
|
||||
nested_instances = get_display_instances(obj)
|
||||
for inst in nested_instances:
|
||||
inst_rep, inst_placement = instance_to_ifc(
|
||||
ifc, body_context, inst, definition_map, scale=scale, material_manager=material_manager
|
||||
)
|
||||
inst_element = _create_element(
|
||||
ifc, ifc_class, name, inst_rep, inst_placement, storey
|
||||
)
|
||||
write_common_properties(ifc, inst_element, obj, category_name)
|
||||
write_properties(ifc, inst_element, obj)
|
||||
instance_count += 1
|
||||
total += 1
|
||||
if not inst_rep:
|
||||
if nested_instances:
|
||||
mapped_items = []
|
||||
inst_placement = None
|
||||
for inst in nested_instances:
|
||||
inst_rep, inst_pl = instance_to_ifc(
|
||||
ifc, body_context, inst, definition_map, scale=scale, material_manager=material_manager
|
||||
)
|
||||
if inst_rep:
|
||||
mapped_items.extend(inst_rep.Items)
|
||||
if inst_placement is None:
|
||||
inst_placement = inst_pl
|
||||
if mapped_items:
|
||||
combined_rep = ifc.createIfcShapeRepresentation(
|
||||
ContextOfItems=body_context,
|
||||
RepresentationIdentifier="Body",
|
||||
RepresentationType="MappedRepresentation",
|
||||
Items=mapped_items,
|
||||
)
|
||||
element = _create_element(
|
||||
ifc, ifc_class, name, combined_rep, inst_placement, storey,
|
||||
storey_manager=storey_manager,
|
||||
tag=get_element_tag(obj), guid=get_ifc_guid(obj),
|
||||
object_type=getattr(obj, "type", None),
|
||||
)
|
||||
write_properties(ifc, element, obj, ifc_class=ifc_class, category_name=category_name)
|
||||
type_manager.assign(element, obj, ifc_class)
|
||||
instance_count += 1
|
||||
total += 1
|
||||
|
||||
# B3: Curve geometry (Lines, Arcs in displayValue)
|
||||
if not rep and not nested_instances:
|
||||
curve_rep, curve_placement = curves_to_ifc(ifc, body_context, obj, scale=scale, material_manager=material_manager)
|
||||
if curve_rep:
|
||||
element = _create_element(ifc, ifc_class, name, curve_rep, curve_placement, storey,
|
||||
storey_manager=storey_manager,
|
||||
tag=get_element_tag(obj), guid=get_ifc_guid(obj),
|
||||
object_type=getattr(obj, "type", None),
|
||||
)
|
||||
write_properties(ifc, element, obj, ifc_class=ifc_class, category_name=category_name)
|
||||
type_manager.assign(element, obj, ifc_class)
|
||||
total += 1
|
||||
else:
|
||||
no_geometry += 1
|
||||
|
||||
if total % 100 == 0:
|
||||
@@ -162,13 +218,24 @@ def automate_function(
|
||||
# ------------------------------------------------------------------ #
|
||||
# 5. Write output
|
||||
# ------------------------------------------------------------------ #
|
||||
print("\n🔗 Flushing spatial containment...")
|
||||
storey_manager.flush()
|
||||
print("🔗 Flushing type relationships...")
|
||||
type_manager.flush()
|
||||
|
||||
file_name = function_inputs.file_name
|
||||
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||||
|
||||
ifc_filename = f"{file_name}_{timestamp}.ifc"
|
||||
|
||||
ifc.write(ifc_filename)
|
||||
automate_context.store_file_result(f"./{ifc_filename}")
|
||||
print(f"\n💾 IFC file written: {ifc_filename}")
|
||||
try:
|
||||
automate_context.mark_run_success("Success! You can download the IF file below.")
|
||||
automate_context.store_file_result(f"./{ifc_filename}")
|
||||
except Exception as e:
|
||||
print(f" ⚠️ Could not upload file result (network issue?): {e}")
|
||||
automate_context.mark_run_failed(f"Something went wrong when storing file result. Exception detail: {e}")
|
||||
|
||||
print(f"\n{'=' * 60}")
|
||||
print(f" Export complete!")
|
||||
@@ -181,13 +248,27 @@ def automate_function(
|
||||
print_instance_stats()
|
||||
print(f"{'=' * 60}\n")
|
||||
|
||||
def _create_element(ifc, ifc_class, name, rep, placement, storey):
|
||||
"""Helper: create an IFC element, assign geometry + placement + container."""
|
||||
element = ifcopenshell.api.run(
|
||||
"root.create_entity", ifc,
|
||||
ifc_class=ifc_class,
|
||||
name=str(name),
|
||||
)
|
||||
def _create_element(ifc, ifc_class, name, rep, placement, storey,
|
||||
storey_manager=None,
|
||||
tag=None, guid=None, object_type=None):
|
||||
"""Helper: create an IFC element, assign geometry + placement, queue containment."""
|
||||
element = ifcopenshell.api.run("root.create_entity", ifc,
|
||||
ifc_class=ifc_class, name=str(name))
|
||||
if tag:
|
||||
try:
|
||||
element.Tag = str(tag)
|
||||
except AttributeError:
|
||||
pass
|
||||
if object_type:
|
||||
try:
|
||||
element.ObjectType = str(object_type)
|
||||
except AttributeError:
|
||||
pass
|
||||
if guid:
|
||||
try:
|
||||
element.GlobalId = guid
|
||||
except Exception:
|
||||
pass
|
||||
if rep and placement:
|
||||
element.Representation = ifc.createIfcProductDefinitionShape(
|
||||
Representations=(rep,)
|
||||
@@ -198,11 +279,14 @@ def _create_element(ifc, ifc_class, name, rep, placement, storey):
|
||||
else:
|
||||
element.ObjectPlacement = _make_placement(ifc, 0.0, 0.0, 0.0)
|
||||
|
||||
ifcopenshell.api.run(
|
||||
"spatial.assign_container", ifc,
|
||||
relating_structure=storey,
|
||||
products=[element],
|
||||
)
|
||||
# Queue spatial assignment (batched flush at end for performance)
|
||||
# IfcSpace is a spatial structure element — must be decomposed (aggregated)
|
||||
# under its IfcBuildingStorey, not spatially contained.
|
||||
if storey_manager:
|
||||
if ifc_class in ("IfcSite", "IfcSpace"):
|
||||
storey_manager.queue_aggregate(storey, element)
|
||||
else:
|
||||
storey_manager.queue_contain(storey, element)
|
||||
return element
|
||||
|
||||
# make sure to call the function with the executor
|
||||
|
||||
+3
-2
@@ -7,10 +7,11 @@ maintainers = [{ name = "Speckle Systems", email = "hello@speckle.systems" }]
|
||||
description = "A Speckle Automate function template using specklepy"
|
||||
readme = "README.md"
|
||||
license = "Apache-2.0"
|
||||
keywords = ["speckle", "automate", "bim", "aec"]
|
||||
keywords = ["speckle", "automate", "bim", "aec", "ifc", "export", "revit"]
|
||||
|
||||
dependencies = ["specklepy==3.1.0",
|
||||
"ifcopenshell==0.8.4.post1",]
|
||||
"ifcopenshell==0.8.4.post1",
|
||||
"python-dotenv>=1.0.0",]
|
||||
|
||||
[project.optional-dependencies]
|
||||
dev = [
|
||||
|
||||
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
@@ -1,34 +0,0 @@
|
||||
# =============================================================================
|
||||
# config.py
|
||||
# All user-facing settings. Edit this file before running main.py.
|
||||
# =============================================================================
|
||||
|
||||
# --- Speckle Connection ---
|
||||
SPECKLE_HOST = "app.speckle.systems" # or your self-hosted server URL
|
||||
SPECKLE_TOKEN = "****" # from app.speckle.systems/profile
|
||||
|
||||
# --- Speckle Project ---
|
||||
PROJECT_ID = "d7d987146d" # the stream/project ID from the URL
|
||||
VERSION_ID = "d59178f01e" # the specific version/commit to export
|
||||
|
||||
# --- IFC Output ---
|
||||
OUTPUT_PATH = "output3.ifc" # where to write the IFC file
|
||||
IFC_SCHEMA = "IFC4X3" # IFC4X3 = IFC4.3
|
||||
|
||||
# --- Project Metadata (written into the IFC file) ---
|
||||
IFC_PROJECT_NAME = "Speckle Export"
|
||||
IFC_SITE_NAME = "Site"
|
||||
IFC_BUILDING_NAME = "Building"
|
||||
|
||||
# --- Units ---
|
||||
# Speckle unit → metres scale factor
|
||||
# The exporter reads units from the root object automatically,
|
||||
# but this is the fallback if units are not set on the stream.
|
||||
DEFAULT_UNITS = "mm"
|
||||
UNIT_SCALE = {
|
||||
"mm": 0.001,
|
||||
"cm": 0.01,
|
||||
"m": 1.0,
|
||||
"ft": 0.3048,
|
||||
"in": 0.0254,
|
||||
}
|
||||
+336
-171
@@ -1,14 +1,18 @@
|
||||
# =============================================================================
|
||||
# geometry.py
|
||||
# Converts Speckle DataObject geometry → IFC IfcFacetedBrep + IfcLocalPlacement
|
||||
# Converts Speckle DataObject geometry → IFC IfcPolygonalFaceSet + IfcLocalPlacement
|
||||
#
|
||||
# Key facts:
|
||||
# - After specklepy receive(), vertices and faces are FLAT Python lists
|
||||
# - displayValue is an array of Mesh objects
|
||||
# - Units are in mm (for Revit), scale to metres for IFC
|
||||
# - Vertices are in absolute world coordinates
|
||||
# - Uses IfcPolygonalFaceSet (indexed vertices) instead of IfcFacetedBrep
|
||||
# for compact output — each vertex stored once, not once per face.
|
||||
# =============================================================================
|
||||
|
||||
import math
|
||||
|
||||
import ifcopenshell
|
||||
from specklepy.objects.base import Base
|
||||
|
||||
@@ -24,139 +28,88 @@ _UNIT_SCALES = {
|
||||
|
||||
|
||||
# --------------------------------------------------------------------------- #
|
||||
# Geometry validation helpers (GEM111 + BRP002 fixes)
|
||||
# Geometry validation helpers (GEM111 fix)
|
||||
# --------------------------------------------------------------------------- #
|
||||
|
||||
# Minimum distance in mm below which two vertices are considered identical (GEM111).
|
||||
_VERTEX_MERGE_TOL = 0.01 # 0.01 mm
|
||||
_INV_TOL = 1.0 / _VERTEX_MERGE_TOL # pre-computed: multiply instead of divide
|
||||
|
||||
|
||||
def snap_coord(v: float) -> int:
|
||||
"""Snap a coordinate to integer grid at _VERTEX_MERGE_TOL resolution."""
|
||||
return round(v / _VERTEX_MERGE_TOL)
|
||||
|
||||
|
||||
def _find_connected_components(snapped_faces: list) -> list:
|
||||
def build_ifc_facesets(ifc, verts_scaled: list, face_groups: list) -> list:
|
||||
"""
|
||||
Union-Find: group face indices into connected components.
|
||||
Two faces are connected if they share an edge (pair of snapped vertex keys).
|
||||
Returns list of components, each a list of face indices.
|
||||
Build a list of IfcPolygonalFaceSet from scaled (x,y,z) vertices and face index groups.
|
||||
|
||||
BRP002 requires all faces in an IfcClosedShell to form ONE component.
|
||||
If multiple components exist, each must become a separate IfcClosedShell.
|
||||
"""
|
||||
n = len(snapped_faces)
|
||||
if n == 0:
|
||||
return []
|
||||
|
||||
parent = list(range(n))
|
||||
|
||||
def find(x):
|
||||
while parent[x] != x:
|
||||
parent[x] = parent[parent[x]]
|
||||
x = parent[x]
|
||||
return x
|
||||
|
||||
def union(a, b):
|
||||
parent[find(a)] = find(b)
|
||||
|
||||
# Map each edge to the first face that used it, then union subsequent faces
|
||||
edge_to_face = {}
|
||||
for fi, keys in enumerate(snapped_faces):
|
||||
for i in range(len(keys)):
|
||||
edge = frozenset([keys[i], keys[(i + 1) % len(keys)]])
|
||||
if edge in edge_to_face:
|
||||
union(fi, edge_to_face[edge])
|
||||
else:
|
||||
edge_to_face[edge] = fi
|
||||
|
||||
from collections import defaultdict
|
||||
groups: dict = defaultdict(list)
|
||||
for fi in range(n):
|
||||
groups[find(fi)].append(fi)
|
||||
return list(groups.values())
|
||||
|
||||
|
||||
def build_ifc_breps(ifc, verts_scaled: list, face_groups: list) -> list:
|
||||
"""
|
||||
Build a list of IfcFacetedBrep from scaled (x,y,z) vertices and face index groups.
|
||||
Uses IfcCartesianPointList3D + IfcIndexedPolygonalFace for compact output.
|
||||
Vertices are deduplicated via snap grid so each unique position is stored once.
|
||||
|
||||
GEM111 fix: skip faces with near-duplicate vertices (snapped to same grid cell).
|
||||
BRP002 fix: split faces into connected components; each component → its own
|
||||
IfcClosedShell → IfcFacetedBrep so every shell is arc-wise connected.
|
||||
|
||||
verts_scaled: flat list of already-scaled floats [x0,y0,z0, x1,y1,z1, ...]
|
||||
face_groups: list of index lists [[i,j,k], [i,j,k,l], ...]
|
||||
Returns: list of IfcFacetedBrep (one per connected component, never empty).
|
||||
Returns: list of IfcPolygonalFaceSet (typically one, empty on failure).
|
||||
"""
|
||||
# Pass 1: validate faces and build snapped key lists for connectivity analysis
|
||||
valid_faces = [] # list of (pts_raw, snapped_keys)
|
||||
snap_to_idx = {} # snap_key → 0-based index in deduped_verts
|
||||
deduped_verts = [] # [[x, y, z], ...] — lists for direct IFC use
|
||||
inv_tol = _INV_TOL
|
||||
|
||||
# Validate faces and remap indices to deduplicated vertex list
|
||||
valid_faces = [] # list of (idx0+1, idx1+1, ...) tuples (1-based for IFC)
|
||||
vert_len = len(verts_scaled)
|
||||
for indices in face_groups:
|
||||
try:
|
||||
pts_raw = []
|
||||
snapped = []
|
||||
degenerate = False
|
||||
seen = set()
|
||||
|
||||
for i in indices:
|
||||
x = float(verts_scaled[i * 3])
|
||||
y = float(verts_scaled[i * 3 + 1])
|
||||
z = float(verts_scaled[i * 3 + 2])
|
||||
key = (snap_coord(x), snap_coord(y), snap_coord(z))
|
||||
if key in seen:
|
||||
degenerate = True
|
||||
break
|
||||
seen.add(key)
|
||||
pts_raw.append((x, y, z))
|
||||
snapped.append(key)
|
||||
|
||||
if degenerate or len(pts_raw) < 3:
|
||||
continue
|
||||
|
||||
valid_faces.append((pts_raw, snapped))
|
||||
except Exception:
|
||||
if indices is None:
|
||||
continue
|
||||
if not isinstance(indices, (list, tuple)):
|
||||
continue
|
||||
|
||||
if not valid_faces:
|
||||
remapped = []
|
||||
seen_snaps = set()
|
||||
degenerate = False
|
||||
invalid = False
|
||||
|
||||
for i in indices:
|
||||
if not isinstance(i, int):
|
||||
invalid = True
|
||||
break
|
||||
|
||||
i3 = i * 3
|
||||
if i3 < 0 or i3 + 2 >= vert_len:
|
||||
invalid = True
|
||||
break
|
||||
|
||||
x = verts_scaled[i3]
|
||||
y = verts_scaled[i3 + 1]
|
||||
z = verts_scaled[i3 + 2]
|
||||
key = (round(x * inv_tol), round(y * inv_tol), round(z * inv_tol))
|
||||
if key in seen_snaps:
|
||||
degenerate = True
|
||||
break
|
||||
seen_snaps.add(key)
|
||||
|
||||
idx = snap_to_idx.get(key)
|
||||
if idx is None:
|
||||
idx = len(deduped_verts)
|
||||
snap_to_idx[key] = idx
|
||||
deduped_verts.append([x, y, z])
|
||||
remapped.append(idx + 1) # 1-based for IFC
|
||||
|
||||
if invalid or degenerate or len(remapped) < 3:
|
||||
continue
|
||||
valid_faces.append(remapped)
|
||||
|
||||
if not valid_faces or not deduped_verts:
|
||||
return []
|
||||
|
||||
# Pass 2: split into connected components (BRP002)
|
||||
snapped_only = [f[1] for f in valid_faces]
|
||||
components = _find_connected_components(snapped_only)
|
||||
|
||||
# Pass 3: build one IfcFacetedBrep per component
|
||||
breps = []
|
||||
for component_indices in components:
|
||||
ifc_faces = []
|
||||
for fi in component_indices:
|
||||
pts_raw, _ = valid_faces[fi]
|
||||
try:
|
||||
pts = [ifc.createIfcCartesianPoint([x, y, z]) for x, y, z in pts_raw]
|
||||
poly = ifc.createIfcPolyLoop(pts)
|
||||
bound = ifc.createIfcFaceOuterBound(poly, True)
|
||||
ifc_faces.append(ifc.createIfcFace([bound]))
|
||||
except Exception:
|
||||
continue
|
||||
|
||||
if not ifc_faces:
|
||||
continue
|
||||
|
||||
shell = ifc.createIfcClosedShell(ifc_faces)
|
||||
breps.append(ifc.createIfcFacetedBrep(shell))
|
||||
|
||||
return breps
|
||||
|
||||
|
||||
# Keep old name as alias so instances.py import works unchanged
|
||||
def build_ifc_faces(ifc, verts_scaled: list, face_groups: list) -> list:
|
||||
"""Legacy wrapper — returns flat list of IfcFace (no connectivity splitting)."""
|
||||
# Used only as a fallback; callers should prefer build_ifc_breps directly.
|
||||
breps = build_ifc_breps(ifc, verts_scaled, face_groups)
|
||||
# Return the faces from all shells combined (for callers that need face lists)
|
||||
faces = []
|
||||
for brep in breps:
|
||||
faces.extend(brep.Outer.CfsFaces)
|
||||
return faces
|
||||
# Build IFC entities
|
||||
try:
|
||||
point_list = ifc.createIfcCartesianPointList3D(deduped_verts)
|
||||
ifc_faces = [
|
||||
ifc.createIfcIndexedPolygonalFace(fi) for fi in valid_faces
|
||||
]
|
||||
faceset = ifc.createIfcPolygonalFaceSet(point_list, None, ifc_faces, None)
|
||||
return [faceset]
|
||||
except Exception:
|
||||
return []
|
||||
|
||||
|
||||
# --------------------------------------------------------------------------- #
|
||||
@@ -189,31 +142,32 @@ def unwrap_chunks(raw) -> list:
|
||||
|
||||
Handles two cases:
|
||||
1. Already flat list of numbers (after specklepy receive deserializes)
|
||||
→ [3, 0, 1, 2, 3, ...] returned as-is
|
||||
→ returned as-is (fast path)
|
||||
2. List of DataChunk objects (raw from server before deserialization)
|
||||
→ each chunk's .data list is concatenated
|
||||
|
||||
Both cases are handled so this function is always safe to call.
|
||||
"""
|
||||
if not raw:
|
||||
return []
|
||||
|
||||
# Fast path: if first item is a number, assume all items are numbers
|
||||
first = raw[0]
|
||||
if isinstance(first, (int, float)):
|
||||
return raw
|
||||
|
||||
# Slow path: DataChunk objects or mixed content
|
||||
result = []
|
||||
for item in raw:
|
||||
if item is None:
|
||||
continue
|
||||
# Plain number — already flat
|
||||
if isinstance(item, (int, float)):
|
||||
result.append(item)
|
||||
continue
|
||||
# DataChunk — unwrap .data
|
||||
speckle_type = getattr(item, "speckle_type", "") or ""
|
||||
if "DataChunk" in speckle_type:
|
||||
chunk_data = _get(item, "data") or _get(item, "@data")
|
||||
if chunk_data:
|
||||
result.extend(list(chunk_data))
|
||||
else:
|
||||
# Unknown — try iterating (handles nested lists)
|
||||
try:
|
||||
result.extend(list(item))
|
||||
except Exception:
|
||||
@@ -306,6 +260,181 @@ def get_display_instances(obj: Base) -> list:
|
||||
return instances
|
||||
|
||||
|
||||
# --------------------------------------------------------------------------- #
|
||||
# Curve detection & extraction (Lines, Arcs)
|
||||
# --------------------------------------------------------------------------- #
|
||||
|
||||
def _is_line(item) -> bool:
|
||||
"""Detect Objects.Geometry.Line (but not Polyline)."""
|
||||
if item is None:
|
||||
return False
|
||||
st = _get(item, "speckle_type") or ""
|
||||
return "Line" in st and "Polyline" not in st
|
||||
|
||||
|
||||
def _is_arc(item) -> bool:
|
||||
"""Detect Objects.Geometry.Arc."""
|
||||
if item is None:
|
||||
return False
|
||||
st = _get(item, "speckle_type") or ""
|
||||
return "Arc" in st
|
||||
|
||||
|
||||
def get_display_curves(obj: Base) -> list:
|
||||
"""Extract Line and Arc objects from a DataObject's displayValue."""
|
||||
curves = []
|
||||
for key in ["displayValue", "@displayValue"]:
|
||||
display = _get(obj, key)
|
||||
if display is None:
|
||||
continue
|
||||
items = display if isinstance(display, list) else [display]
|
||||
for item in items:
|
||||
if _is_line(item) or _is_arc(item):
|
||||
curves.append(item)
|
||||
if curves:
|
||||
break
|
||||
return curves
|
||||
|
||||
|
||||
def _point_coords(pt, fallback_scale: float) -> tuple:
|
||||
"""Extract (x, y, z) from a Speckle Point, scaled to mm."""
|
||||
scale = _resolve_scale(pt, fallback_scale)
|
||||
x = float(_get(pt, "x") or 0.0) * scale
|
||||
y = float(_get(pt, "y") or 0.0) * scale
|
||||
z = float(_get(pt, "z") or 0.0) * scale
|
||||
return x, y, z
|
||||
|
||||
|
||||
def _arc_to_points(arc, scale: float, num_segments: int = 8) -> list:
|
||||
"""
|
||||
Approximate a Speckle Arc as a list of (x, y, z) points in mm.
|
||||
Uses plane origin (center), radius, and domain angles for parametric sampling.
|
||||
Falls back to start/mid/end points if plane data is missing.
|
||||
"""
|
||||
plane = _get(arc, "plane")
|
||||
radius = _get(arc, "radius")
|
||||
domain = _get(arc, "domain")
|
||||
|
||||
if not plane or not radius or not domain:
|
||||
points = []
|
||||
for key in ["startPoint", "midPoint", "endPoint"]:
|
||||
pt = _get(arc, key)
|
||||
if pt:
|
||||
points.append(_point_coords(pt, scale))
|
||||
return points if len(points) >= 2 else []
|
||||
|
||||
origin = _get(plane, "origin")
|
||||
xdir = _get(plane, "xdir")
|
||||
ydir = _get(plane, "ydir")
|
||||
|
||||
if not origin or not xdir or not ydir:
|
||||
points = []
|
||||
for key in ["startPoint", "midPoint", "endPoint"]:
|
||||
pt = _get(arc, key)
|
||||
if pt:
|
||||
points.append(_point_coords(pt, scale))
|
||||
return points if len(points) >= 2 else []
|
||||
|
||||
cx, cy, cz = _point_coords(origin, scale)
|
||||
# Direction vectors are unitless — do not scale
|
||||
dxx = float(_get(xdir, "x") or 0.0)
|
||||
dxy = float(_get(xdir, "y") or 0.0)
|
||||
dxz = float(_get(xdir, "z") or 0.0)
|
||||
dyx = float(_get(ydir, "x") or 0.0)
|
||||
dyy = float(_get(ydir, "y") or 0.0)
|
||||
dyz = float(_get(ydir, "z") or 0.0)
|
||||
|
||||
r = float(radius) * scale
|
||||
t_start = float(_get(domain, "start") or 0.0)
|
||||
t_end = float(_get(domain, "end") or 0.0)
|
||||
|
||||
points = []
|
||||
for i in range(num_segments + 1):
|
||||
t = t_start + (t_end - t_start) * i / num_segments
|
||||
cos_t = math.cos(t)
|
||||
sin_t = math.sin(t)
|
||||
x = cx + r * (cos_t * dxx + sin_t * dyx)
|
||||
y = cy + r * (cos_t * dxy + sin_t * dyy)
|
||||
z = cz + r * (cos_t * dxz + sin_t * dyz)
|
||||
points.append((x, y, z))
|
||||
return points
|
||||
|
||||
|
||||
def curves_to_ifc(
|
||||
ifc: ifcopenshell.file,
|
||||
body_context,
|
||||
obj: Base,
|
||||
scale: float = 0.001,
|
||||
material_manager=None,
|
||||
) -> tuple:
|
||||
"""
|
||||
Convert Speckle Line/Arc objects in displayValue to IFC curve geometry.
|
||||
Lines → IfcPolyline (2 points), Arcs → IfcPolyline (sampled points).
|
||||
Wrapped in IfcGeometricCurveSet.
|
||||
Returns (IfcShapeRepresentation, IfcLocalPlacement) or (None, None).
|
||||
"""
|
||||
curves = get_display_curves(obj)
|
||||
if not curves:
|
||||
return None, None
|
||||
|
||||
obj_scale = _resolve_scale(obj, scale)
|
||||
polylines = []
|
||||
all_points = []
|
||||
|
||||
for curve in curves:
|
||||
cs = _resolve_scale(curve, obj_scale)
|
||||
|
||||
if _is_line(curve):
|
||||
start = _get(curve, "start")
|
||||
end = _get(curve, "end")
|
||||
if not start or not end:
|
||||
continue
|
||||
p1 = _point_coords(start, cs)
|
||||
p2 = _point_coords(end, cs)
|
||||
all_points.extend([p1, p2])
|
||||
polylines.append([p1, p2])
|
||||
|
||||
elif _is_arc(curve):
|
||||
pts = _arc_to_points(curve, cs)
|
||||
if len(pts) >= 2:
|
||||
all_points.extend(pts)
|
||||
polylines.append(pts)
|
||||
|
||||
if not polylines or not all_points:
|
||||
return None, None
|
||||
|
||||
# Compute origin from all curve points
|
||||
xs = [p[0] for p in all_points]
|
||||
ys = [p[1] for p in all_points]
|
||||
zs = [p[2] for p in all_points]
|
||||
ox = (min(xs) + max(xs)) / 2.0
|
||||
oy = (min(ys) + max(ys)) / 2.0
|
||||
oz = min(zs)
|
||||
|
||||
# Build IfcPolylines offset from origin
|
||||
ifc_polylines = []
|
||||
for pts in polylines:
|
||||
ifc_points = [
|
||||
ifc.createIfcCartesianPoint([p[0] - ox, p[1] - oy, p[2] - oz])
|
||||
for p in pts
|
||||
]
|
||||
ifc_polylines.append(ifc.createIfcPolyline(ifc_points))
|
||||
|
||||
if not ifc_polylines:
|
||||
return None, None
|
||||
|
||||
curve_set = ifc.createIfcGeometricCurveSet(ifc_polylines)
|
||||
|
||||
rep = ifc.createIfcShapeRepresentation(
|
||||
ContextOfItems=body_context,
|
||||
RepresentationIdentifier="Body",
|
||||
RepresentationType="GeometricCurveSet",
|
||||
Items=[curve_set],
|
||||
)
|
||||
placement = _make_placement(ifc, ox, oy, oz)
|
||||
return rep, placement
|
||||
|
||||
|
||||
# --------------------------------------------------------------------------- #
|
||||
# Face decoding
|
||||
# --------------------------------------------------------------------------- #
|
||||
@@ -318,17 +447,22 @@ def decode_faces(faces_raw: list) -> list:
|
||||
"""
|
||||
decoded = []
|
||||
i = 0
|
||||
while i < len(faces_raw):
|
||||
n = int(faces_raw[i])
|
||||
total = len(faces_raw)
|
||||
# Check if values are already ints (common after unwrap_chunks)
|
||||
already_int = total > 0 and isinstance(faces_raw[0], int)
|
||||
while i < total:
|
||||
n = faces_raw[i] if already_int else int(faces_raw[i])
|
||||
if n == 0:
|
||||
n = 3
|
||||
elif n == 1:
|
||||
n = 4
|
||||
end = i + 1 + n
|
||||
if end > len(faces_raw):
|
||||
if end > total:
|
||||
break
|
||||
indices = [int(faces_raw[i + 1 + j]) for j in range(n)]
|
||||
decoded.append(indices)
|
||||
if already_int:
|
||||
decoded.append(faces_raw[i + 1:end])
|
||||
else:
|
||||
decoded.append([int(v) for v in faces_raw[i + 1:end]])
|
||||
i = end
|
||||
return decoded
|
||||
|
||||
@@ -339,25 +473,55 @@ def decode_faces(faces_raw: list) -> list:
|
||||
|
||||
def compute_origin(flat_verts: list) -> tuple:
|
||||
"""
|
||||
Compute placement origin from scaled vertex list (metres).
|
||||
Compute placement origin from scaled vertex list (mm).
|
||||
X, Y = bounding box centroid
|
||||
Z = minimum Z (bottom face of element — more natural for IFC)
|
||||
Single-pass to avoid creating 3 sliced copies of a large list.
|
||||
"""
|
||||
xs = flat_verts[0::3]
|
||||
ys = flat_verts[1::3]
|
||||
zs = flat_verts[2::3]
|
||||
cx = (min(xs) + max(xs)) / 2.0
|
||||
cy = (min(ys) + max(ys)) / 2.0
|
||||
cz = min(zs)
|
||||
return cx, cy, cz
|
||||
x0 = flat_verts[0]
|
||||
y0 = flat_verts[1]
|
||||
z0 = flat_verts[2]
|
||||
xmin = xmax = x0
|
||||
ymin = ymax = y0
|
||||
zmin = z0
|
||||
for i in range(3, len(flat_verts) - 2, 3):
|
||||
x = flat_verts[i]
|
||||
y = flat_verts[i + 1]
|
||||
z = flat_verts[i + 2]
|
||||
if x < xmin:
|
||||
xmin = x
|
||||
elif x > xmax:
|
||||
xmax = x
|
||||
if y < ymin:
|
||||
ymin = y
|
||||
elif y > ymax:
|
||||
ymax = y
|
||||
if z < zmin:
|
||||
zmin = z
|
||||
return (xmin + xmax) / 2.0, (ymin + ymax) / 2.0, zmin
|
||||
|
||||
|
||||
# Cache for shared IFC direction/point entities (keyed by ifc file id)
|
||||
_shared_entities: dict[int, dict] = {}
|
||||
|
||||
|
||||
def _get_shared(ifc):
|
||||
"""Return (or create) shared IfcDirection and IfcCartesianPoint entities for this file."""
|
||||
fid = id(ifc)
|
||||
if fid not in _shared_entities:
|
||||
_shared_entities[fid] = {
|
||||
"z_axis": ifc.createIfcDirection([0.0, 0.0, 1.0]),
|
||||
"x_axis": ifc.createIfcDirection([1.0, 0.0, 0.0]),
|
||||
"origin_0": ifc.createIfcCartesianPoint([0.0, 0.0, 0.0]),
|
||||
}
|
||||
return _shared_entities[fid]
|
||||
|
||||
|
||||
def _make_placement(ifc, x: float, y: float, z: float):
|
||||
"""Create an IfcLocalPlacement at absolute world coordinates (metres)."""
|
||||
shared = _get_shared(ifc)
|
||||
origin = ifc.createIfcCartesianPoint([x, y, z])
|
||||
z_axis = ifc.createIfcDirection([0.0, 0.0, 1.0])
|
||||
x_axis = ifc.createIfcDirection([1.0, 0.0, 0.0])
|
||||
a2p = ifc.createIfcAxis2Placement3D(origin, z_axis, x_axis)
|
||||
a2p = ifc.createIfcAxis2Placement3D(origin, shared["z_axis"], shared["x_axis"])
|
||||
return ifc.createIfcLocalPlacement(PlacementRelTo=None, RelativePlacement=a2p)
|
||||
|
||||
|
||||
@@ -374,7 +538,7 @@ def mesh_to_ifc(
|
||||
) -> tuple:
|
||||
"""
|
||||
Convert a Speckle DataObject → (IfcShapeRepresentation, IfcLocalPlacement).
|
||||
Creates one IfcFacetedBrep per mesh so each can carry its own material style.
|
||||
Creates one IfcPolygonalFaceSet per mesh so each can carry its own material style.
|
||||
Returns (None, None) if no usable geometry is found.
|
||||
"""
|
||||
meshes = get_display_meshes(obj)
|
||||
@@ -384,21 +548,22 @@ def mesh_to_ifc(
|
||||
obj_scale = _resolve_scale(obj, scale)
|
||||
|
||||
# ------------------------------------------------------------------ #
|
||||
# Pass 1: collect all scaled vertices to compute world origin
|
||||
# Pass 1: unpack vertices once per mesh, collect all scaled coords
|
||||
# to compute world origin. Cache (verts, ms) for Pass 2.
|
||||
# ------------------------------------------------------------------ #
|
||||
mesh_cache = [] # [(verts_list, ms, scaled)] or None per mesh
|
||||
all_scaled = []
|
||||
for mesh in meshes:
|
||||
raw_verts = _get(mesh, "vertices") or []
|
||||
verts = unwrap_chunks(list(raw_verts))
|
||||
verts = unwrap_chunks(raw_verts if isinstance(raw_verts, list) else list(raw_verts))
|
||||
if not verts:
|
||||
mesh_cache.append(None)
|
||||
continue
|
||||
ms = _resolve_scale(mesh, obj_scale)
|
||||
for i in range(0, len(verts) - 2, 3):
|
||||
all_scaled.extend([
|
||||
float(verts[i]) * ms,
|
||||
float(verts[i+1]) * ms,
|
||||
float(verts[i+2]) * ms,
|
||||
])
|
||||
# Pre-scale vertices once, reuse in Pass 2
|
||||
scaled = [float(v) * ms for v in verts]
|
||||
mesh_cache.append((verts, ms, scaled))
|
||||
all_scaled.extend(scaled)
|
||||
|
||||
if not all_scaled:
|
||||
return None, None
|
||||
@@ -406,20 +571,19 @@ def mesh_to_ifc(
|
||||
ox, oy, oz = compute_origin(all_scaled)
|
||||
|
||||
# ------------------------------------------------------------------ #
|
||||
# Pass 2: one brep per mesh (so each can have its own material style)
|
||||
# Pass 2: one faceset per mesh — reuse cached verts, only unpack faces
|
||||
# ------------------------------------------------------------------ #
|
||||
brep_items = []
|
||||
geom_items = []
|
||||
|
||||
for mesh in meshes:
|
||||
raw_verts = _get(mesh, "vertices") or []
|
||||
raw_faces = _get(mesh, "faces") or []
|
||||
verts = unwrap_chunks(list(raw_verts))
|
||||
faces_raw = unwrap_chunks(list(raw_faces))
|
||||
|
||||
if not verts or not faces_raw:
|
||||
for mesh, cached in zip(meshes, mesh_cache):
|
||||
if cached is None:
|
||||
continue
|
||||
verts, ms, scaled = cached
|
||||
raw_faces = _get(mesh, "faces") or []
|
||||
faces_raw = unwrap_chunks(raw_faces if isinstance(raw_faces, list) else list(raw_faces))
|
||||
|
||||
ms = _resolve_scale(mesh, obj_scale)
|
||||
if not faces_raw:
|
||||
continue
|
||||
|
||||
try:
|
||||
face_groups = decode_faces(faces_raw)
|
||||
@@ -427,28 +591,29 @@ def mesh_to_ifc(
|
||||
print(f" ⚠️ Face decode error: {e}")
|
||||
continue
|
||||
|
||||
# Build pre-scaled vertex list (relative to origin) for this mesh
|
||||
verts_scaled = []
|
||||
for vi in range(0, len(verts) - 2, 3):
|
||||
verts_scaled.append(float(verts[vi]) * ms - ox)
|
||||
verts_scaled.append(float(verts[vi+1]) * ms - oy)
|
||||
verts_scaled.append(float(verts[vi+2]) * ms - oz)
|
||||
# Offset pre-scaled vertices relative to origin (flat list, no tuples)
|
||||
n = len(scaled)
|
||||
verts_scaled = [0.0] * n
|
||||
for vi in range(0, n, 3):
|
||||
verts_scaled[vi] = scaled[vi] - ox
|
||||
verts_scaled[vi + 1] = scaled[vi + 1] - oy
|
||||
verts_scaled[vi + 2] = scaled[vi + 2] - oz
|
||||
|
||||
mesh_breps = build_ifc_breps(ifc, verts_scaled, face_groups)
|
||||
mesh_facesets = build_ifc_facesets(ifc, verts_scaled, face_groups)
|
||||
|
||||
if not mesh_breps:
|
||||
if not mesh_facesets:
|
||||
continue
|
||||
|
||||
# Apply material style to every component brep of this mesh
|
||||
# Apply material style to every faceset of this mesh
|
||||
if material_manager:
|
||||
mesh_app_id = _get(mesh, "applicationId")
|
||||
if mesh_app_id:
|
||||
for brep in mesh_breps:
|
||||
material_manager.apply_to_item(brep, str(mesh_app_id))
|
||||
for fs in mesh_facesets:
|
||||
material_manager.apply_to_item(fs, str(mesh_app_id))
|
||||
|
||||
brep_items.extend(mesh_breps)
|
||||
geom_items.extend(mesh_facesets)
|
||||
|
||||
if not brep_items:
|
||||
if not geom_items:
|
||||
return None, None
|
||||
|
||||
# ------------------------------------------------------------------ #
|
||||
@@ -457,8 +622,8 @@ def mesh_to_ifc(
|
||||
rep = ifc.createIfcShapeRepresentation(
|
||||
ContextOfItems=body_context,
|
||||
RepresentationIdentifier="Body",
|
||||
RepresentationType="Brep",
|
||||
Items=brep_items,
|
||||
RepresentationType="Tessellation",
|
||||
Items=geom_items,
|
||||
)
|
||||
placement = _make_placement(ifc, ox, oy, oz)
|
||||
|
||||
|
||||
+229
-133
@@ -15,10 +15,14 @@
|
||||
# Definition geometry lives in root → Collection("definitionGeometry")
|
||||
#
|
||||
# We detect the format by the definitionId prefix.
|
||||
#
|
||||
# Performance: uses IfcRepresentationMap + IfcMappedItem so that all instances
|
||||
# sharing the same definition reference a single copy of the geometry.
|
||||
# =============================================================================
|
||||
|
||||
import math
|
||||
from specklepy.objects.base import Base
|
||||
from utils.geometry import _get, unwrap_chunks, decode_faces, _UNIT_SCALES, build_ifc_breps
|
||||
from utils.geometry import _get, unwrap_chunks, decode_faces, _UNIT_SCALES, build_ifc_facesets, _get_shared
|
||||
|
||||
|
||||
def is_instance(obj) -> bool:
|
||||
@@ -72,33 +76,6 @@ def build_definition_map(root: Base) -> dict:
|
||||
print(f" IFC definition proxies: {len(ifc_proxies)}")
|
||||
print(f" IFC definition meshes: {len(ifc_meshes)}")
|
||||
|
||||
# Diagnostic: dump first 3 instanceDefinitionProxies to understand structure
|
||||
print("\n [PROXY DIAG] First 3 instanceDefinitionProxies from root:")
|
||||
proxies_raw2 = _get(root, "instanceDefinitionProxies")
|
||||
if proxies_raw2:
|
||||
sample = proxies_raw2 if isinstance(proxies_raw2, list) else [proxies_raw2]
|
||||
for i, proxy in enumerate(sample[:3]):
|
||||
app_id = _get(proxy, "applicationId") or "?"
|
||||
name = _get(proxy, "name") or "?"
|
||||
objects = _get(proxy, "objects") or []
|
||||
obj_ids = list(objects)[:3] if objects else []
|
||||
print(f" [{i}] appId={app_id}")
|
||||
print(f" name={name}")
|
||||
print(f" objects={obj_ids} (len={len(list(objects)) if objects else 0})")
|
||||
# Check if first object is found in our maps
|
||||
if obj_ids:
|
||||
oid = str(obj_ids[0])
|
||||
in_by_id = oid.lower()[:32] in by_id
|
||||
in_by_app_id = oid.lower() in by_app_id
|
||||
print(f" objects[0]='{oid}' → in by_id: {in_by_id}, in by_app_id: {in_by_app_id}")
|
||||
else:
|
||||
print(" [PROXY DIAG] No instanceDefinitionProxies found on root!")
|
||||
# Check where they might be
|
||||
for key in ["@instanceDefinitionProxies", "instancedefinitionproxies"]:
|
||||
val = _get(root, key)
|
||||
if val:
|
||||
print(f" Found under key '{key}': {type(val)}")
|
||||
|
||||
return {
|
||||
"by_id": by_id,
|
||||
"by_app_id": by_app_id,
|
||||
@@ -213,31 +190,19 @@ def _resolve_instance_scale(obj, stream_scale: float) -> float:
|
||||
return stream_scale
|
||||
|
||||
|
||||
def _parse_transform(t: list, scale: float) -> tuple:
|
||||
"""
|
||||
Row-major 4x4 matrix.
|
||||
Translation at t[3], t[7], t[11] — scaled to metres.
|
||||
Local X axis = row 0, Local Z axis = row 2.
|
||||
"""
|
||||
tx = float(t[3]) * scale
|
||||
ty = float(t[7]) * scale
|
||||
tz = float(t[11]) * scale
|
||||
x_axis = (float(t[0]), float(t[1]), float(t[2]))
|
||||
z_axis = (float(t[8]), float(t[9]), float(t[10]))
|
||||
return (tx, ty, tz), x_axis, z_axis
|
||||
|
||||
|
||||
def _make_ifc_placement(ifc, tx, ty, tz, x_axis, z_axis):
|
||||
origin = ifc.createIfcCartesianPoint([tx, ty, tz])
|
||||
x_dir = ifc.createIfcDirection(list(x_axis))
|
||||
z_dir = ifc.createIfcDirection(list(z_axis))
|
||||
a2p = ifc.createIfcAxis2Placement3D(origin, z_dir, x_dir)
|
||||
return ifc.createIfcLocalPlacement(PlacementRelTo=None, RelativePlacement=a2p)
|
||||
|
||||
|
||||
# Stats
|
||||
_stats = {"found": 0, "not_found": 0}
|
||||
_dbg_cnt = [0]
|
||||
_stats = {"found": 0, "not_found": 0}
|
||||
|
||||
# Cache: mesh id → (verts_scaled, face_groups) to avoid re-unpacking
|
||||
# AND re-scaling the same definition mesh across many instances that share it.
|
||||
_mesh_data_cache: dict = {}
|
||||
|
||||
# Cache: definition_id → IfcRepresentationMap (or None if no geometry)
|
||||
# All instances sharing the same definition reuse one geometry copy.
|
||||
_rep_map_cache: dict = {}
|
||||
|
||||
# Shared identity placement for all instances (keyed by ifc file id)
|
||||
_identity_placement_cache: dict[int, object] = {}
|
||||
|
||||
|
||||
_MM_SCALES = {
|
||||
@@ -248,28 +213,153 @@ _MM_SCALES = {
|
||||
}
|
||||
|
||||
|
||||
def _apply_transform(t: list, vx: float, vy: float, vz: float, ts: float) -> tuple:
|
||||
"""
|
||||
Apply a row-major 4x4 transform to a single vertex.
|
||||
ts = scale factor applied to the translation components only (not rotation).
|
||||
For Revit mm data with IFC in mm: ts=1.0 (no conversion).
|
||||
For IFC-format transforms (metres): ts=1000.0 (m→mm).
|
||||
Rotation components are dimensionless and never scaled.
|
||||
"""
|
||||
x = t[0]*vx + t[1]*vy + t[2]*vz + t[3] * ts
|
||||
y = t[4]*vx + t[5]*vy + t[6]*vz + t[7] * ts
|
||||
z = t[8]*vx + t[9]*vy + t[10]*vz + t[11] * ts
|
||||
return x, y, z
|
||||
# --------------------------------------------------------------------------- #
|
||||
# IfcRepresentationMap builder — geometry created once per definition
|
||||
# --------------------------------------------------------------------------- #
|
||||
|
||||
def _build_rep_map(ifc, body_context, meshes: list, ifc_format: bool,
|
||||
material_manager=None):
|
||||
"""
|
||||
Build an IfcRepresentationMap from definition meshes.
|
||||
Geometry is in local coordinates (mm, no instance transform applied).
|
||||
Returns IfcRepresentationMap or None if no valid geometry.
|
||||
"""
|
||||
geom_items = []
|
||||
|
||||
for mesh in meshes:
|
||||
mesh_id = _get(mesh, "id") or _get(mesh, "applicationId")
|
||||
if mesh_id and mesh_id in _mesh_data_cache:
|
||||
verts_local, face_groups = _mesh_data_cache[mesh_id]
|
||||
else:
|
||||
raw_verts = _get(mesh, "vertices") or []
|
||||
raw_faces = _get(mesh, "faces") or []
|
||||
verts = unwrap_chunks(list(raw_verts))
|
||||
faces_raw = unwrap_chunks(list(raw_faces))
|
||||
if not verts or not faces_raw:
|
||||
continue
|
||||
|
||||
mesh_units = _get(mesh, "units") or _get(mesh, "_units") or ("m" if ifc_format else "mm")
|
||||
ms = _MM_SCALES.get(mesh_units.lower().strip(), 1.0)
|
||||
|
||||
try:
|
||||
face_groups = decode_faces(faces_raw)
|
||||
except Exception as e:
|
||||
print(f" ⚠️ Instance face decode: {e}")
|
||||
continue
|
||||
|
||||
# Scale vertices once and cache the result
|
||||
verts_local = [float(v) * ms for v in verts]
|
||||
|
||||
if mesh_id:
|
||||
_mesh_data_cache[mesh_id] = (verts_local, face_groups)
|
||||
|
||||
mesh_facesets = build_ifc_facesets(ifc, verts_local, face_groups)
|
||||
|
||||
if not mesh_facesets:
|
||||
continue
|
||||
|
||||
# Apply material style to each faceset
|
||||
if material_manager:
|
||||
mesh_app_id = _get(mesh, "applicationId")
|
||||
if mesh_app_id:
|
||||
for fs in mesh_facesets:
|
||||
material_manager.apply_to_item(fs, str(mesh_app_id))
|
||||
|
||||
geom_items.extend(mesh_facesets)
|
||||
|
||||
if not geom_items:
|
||||
return None
|
||||
|
||||
# Mapping origin = identity (local coords origin) — reuse shared origin
|
||||
shared = _get_shared(ifc)
|
||||
a2p = ifc.createIfcAxis2Placement3D(shared["origin_0"], None, None)
|
||||
|
||||
# The mapped representation holds the actual geometry
|
||||
mapped_rep = ifc.createIfcShapeRepresentation(
|
||||
ContextOfItems=body_context,
|
||||
RepresentationIdentifier="Body",
|
||||
RepresentationType="Tessellation",
|
||||
Items=geom_items,
|
||||
)
|
||||
|
||||
return ifc.createIfcRepresentationMap(a2p, mapped_rep)
|
||||
|
||||
|
||||
# --------------------------------------------------------------------------- #
|
||||
# Transform → IfcCartesianTransformationOperator3D
|
||||
# --------------------------------------------------------------------------- #
|
||||
|
||||
def _vec_magnitude(x, y, z):
|
||||
return math.sqrt(x*x + y*y + z*z)
|
||||
|
||||
|
||||
def _make_transform_operator(ifc, t: list, ts: float):
|
||||
"""
|
||||
Convert a row-major 4x4 matrix + translation scale into an
|
||||
IfcCartesianTransformationOperator3DnonUniform.
|
||||
|
||||
t: 16 floats, row-major [r00,r01,r02,tx, r10,r11,r12,ty, r20,r21,r22,tz, 0,0,0,1]
|
||||
ts: scale factor for translation components (e.g. 1000.0 for m→mm)
|
||||
|
||||
The matrix acts as: p' = M * p + translation, where M rows are:
|
||||
row0 = (t[0], t[1], t[2])
|
||||
row1 = (t[4], t[5], t[6])
|
||||
row2 = (t[8], t[9], t[10])
|
||||
|
||||
IfcCartesianTransformationOperator axes represent the COLUMNS of M:
|
||||
Axis1 = column 0 = where local X maps → (t[0], t[4], t[8])
|
||||
Axis2 = column 1 = where local Y maps → (t[1], t[5], t[9])
|
||||
Axis3 = column 2 = where local Z maps → (t[2], t[6], t[10])
|
||||
|
||||
Returns the IFC entity, or None if the transform is degenerate.
|
||||
"""
|
||||
# Extract COLUMNS of the 3x3 rotation/scale sub-matrix
|
||||
ax1 = (float(t[0]), float(t[4]), float(t[8])) # column 0: X-axis direction
|
||||
ax2 = (float(t[1]), float(t[5]), float(t[9])) # column 1: Y-axis direction
|
||||
ax3 = (float(t[2]), float(t[6]), float(t[10])) # column 2: Z-axis direction
|
||||
|
||||
s1 = _vec_magnitude(*ax1)
|
||||
s2 = _vec_magnitude(*ax2)
|
||||
s3 = _vec_magnitude(*ax3)
|
||||
|
||||
if s1 < 1e-10 or s2 < 1e-10 or s3 < 1e-10:
|
||||
return None # degenerate transform
|
||||
|
||||
# Normalized direction vectors
|
||||
d1 = ifc.createIfcDirection([ax1[0]/s1, ax1[1]/s1, ax1[2]/s1])
|
||||
d2 = ifc.createIfcDirection([ax2[0]/s2, ax2[1]/s2, ax2[2]/s2])
|
||||
d3 = ifc.createIfcDirection([ax3[0]/s3, ax3[1]/s3, ax3[2]/s3])
|
||||
|
||||
# Translation, scaled to mm
|
||||
tx = float(t[3]) * ts
|
||||
ty = float(t[7]) * ts
|
||||
tz = float(t[11]) * ts
|
||||
origin = ifc.createIfcCartesianPoint([tx, ty, tz])
|
||||
|
||||
# Use non-uniform variant to handle mirrors and non-uniform scale
|
||||
return ifc.createIfcCartesianTransformationOperator3DnonUniform(
|
||||
d1, # Axis1
|
||||
d2, # Axis2
|
||||
origin, # LocalOrigin
|
||||
s1, # Scale
|
||||
d3, # Axis3
|
||||
s2, # Scale2
|
||||
s3, # Scale3
|
||||
)
|
||||
|
||||
|
||||
# --------------------------------------------------------------------------- #
|
||||
# Main conversion — IfcMappedItem approach
|
||||
# --------------------------------------------------------------------------- #
|
||||
|
||||
def instance_to_ifc(ifc, body_context, obj: Base, definition_map: dict,
|
||||
scale: float = 1.0, material_manager=None):
|
||||
"""
|
||||
Convert a Speckle InstanceProxy → (IfcShapeRepresentation, IfcLocalPlacement).
|
||||
|
||||
Strategy: BAKE the full 4x4 transform into every vertex (world coordinates).
|
||||
Creates one IfcFacetedBrep per definition mesh so each can carry its own
|
||||
material style via renderMaterialProxies.
|
||||
Strategy: create geometry once per definition as an IfcRepresentationMap,
|
||||
then reference it via IfcMappedItem + IfcCartesianTransformationOperator3D
|
||||
for each instance. This avoids duplicating geometry across instances.
|
||||
"""
|
||||
transform_raw = _get(obj, "transform")
|
||||
if not transform_raw:
|
||||
@@ -285,89 +375,95 @@ def instance_to_ifc(ifc, body_context, obj: Base, definition_map: dict,
|
||||
# Revit format transform is already in mm (same as IFC file units)
|
||||
ts = 1000.0 if ifc_format else _resolve_instance_scale(obj, scale)
|
||||
|
||||
if _dbg_cnt[0] < 6:
|
||||
_dbg_cnt[0] += 1
|
||||
fmt = "IFC" if ifc_format else "Revit"
|
||||
x_axis = (round(t[0],2), round(t[1],2), round(t[2],2))
|
||||
z_axis = (round(t[8],2), round(t[9],2), round(t[10],2))
|
||||
print(f" [INST {_dbg_cnt[0]} {fmt}] {definition_id[:40]}")
|
||||
print(f" t[3]={t[3]:.1f} t[7]={t[7]:.1f} t[11]={t[11]:.1f} x={x_axis} z={z_axis}")
|
||||
# Identity placement (transform is encoded in the MappedItem) — shared across all instances
|
||||
fid = id(ifc)
|
||||
if fid not in _identity_placement_cache:
|
||||
shared = _get_shared(ifc)
|
||||
a2p = ifc.createIfcAxis2Placement3D(shared["origin_0"], None, None)
|
||||
_identity_placement_cache[fid] = ifc.createIfcLocalPlacement(PlacementRelTo=None, RelativePlacement=a2p)
|
||||
placement = _identity_placement_cache[fid]
|
||||
|
||||
# World-origin placement (geometry is baked to world coords)
|
||||
origin = ifc.createIfcCartesianPoint([0.0, 0.0, 0.0])
|
||||
a2p = ifc.createIfcAxis2Placement3D(origin, None, None)
|
||||
placement = ifc.createIfcLocalPlacement(PlacementRelTo=None, RelativePlacement=a2p)
|
||||
# --- Get or build IfcRepresentationMap (cached per definition_id) ---
|
||||
if definition_id not in _rep_map_cache:
|
||||
if ifc_format:
|
||||
meshes = _get_ifc_meshes(definition_id, definition_map)
|
||||
else:
|
||||
meshes = _get_revit_meshes(definition_id, definition_map)
|
||||
|
||||
# Get definition meshes
|
||||
if ifc_format:
|
||||
meshes = _get_ifc_meshes(definition_id, definition_map)
|
||||
if not meshes:
|
||||
_stats["not_found"] += 1
|
||||
_rep_map_cache[definition_id] = None
|
||||
return None, placement
|
||||
|
||||
_stats["found"] += 1
|
||||
_rep_map_cache[definition_id] = _build_rep_map(
|
||||
ifc, body_context, meshes, ifc_format, material_manager
|
||||
)
|
||||
else:
|
||||
meshes = _get_revit_meshes(definition_id, definition_map)
|
||||
# Track stats even for cached definitions
|
||||
if _rep_map_cache[definition_id] is not None:
|
||||
_stats["found"] += 1
|
||||
else:
|
||||
_stats["not_found"] += 1
|
||||
|
||||
if not meshes:
|
||||
_stats["not_found"] += 1
|
||||
rep_map = _rep_map_cache[definition_id]
|
||||
if rep_map is None:
|
||||
return None, placement
|
||||
|
||||
_stats["found"] += 1
|
||||
|
||||
# One brep per mesh so each can have its own material style
|
||||
brep_items = []
|
||||
for mesh in meshes:
|
||||
raw_verts = _get(mesh, "vertices") or []
|
||||
raw_faces = _get(mesh, "faces") or []
|
||||
verts = unwrap_chunks(list(raw_verts))
|
||||
faces_raw = unwrap_chunks(list(raw_faces))
|
||||
if not verts or not faces_raw:
|
||||
continue
|
||||
|
||||
mesh_units = _get(mesh, "units") or _get(mesh, "_units") or ("m" if ifc_format else "mm")
|
||||
ms = _MM_SCALES.get(mesh_units.lower().strip(), 1.0)
|
||||
|
||||
try:
|
||||
face_groups = decode_faces(faces_raw)
|
||||
except Exception as e:
|
||||
print(f" ⚠️ Instance face decode: {e}")
|
||||
continue
|
||||
|
||||
# Pre-compute world coords for all vertices in this mesh
|
||||
verts_world = []
|
||||
for vi in range(0, len(verts) - 2, 3):
|
||||
lx = float(verts[vi]) * ms
|
||||
ly = float(verts[vi+1]) * ms
|
||||
lz = float(verts[vi+2]) * ms
|
||||
wx, wy, wz = _apply_transform(t, lx, ly, lz, ts)
|
||||
verts_world.append(wx)
|
||||
verts_world.append(wy)
|
||||
verts_world.append(wz)
|
||||
|
||||
mesh_breps = build_ifc_breps(ifc, verts_world, face_groups)
|
||||
|
||||
if not mesh_breps:
|
||||
continue
|
||||
|
||||
# Apply material style to every component brep of this mesh
|
||||
if material_manager:
|
||||
mesh_app_id = _get(mesh, "applicationId")
|
||||
if mesh_app_id:
|
||||
for brep in mesh_breps:
|
||||
material_manager.apply_to_item(brep, str(mesh_app_id))
|
||||
|
||||
brep_items.extend(mesh_breps)
|
||||
|
||||
if not brep_items:
|
||||
# --- Build transform operator from instance's 4x4 matrix ---
|
||||
transform_op = _make_transform_operator(ifc, t, ts)
|
||||
if transform_op is None:
|
||||
return None, placement
|
||||
|
||||
# --- Create IfcMappedItem referencing the shared geometry ---
|
||||
mapped_item = ifc.createIfcMappedItem(rep_map, transform_op)
|
||||
|
||||
rep = ifc.createIfcShapeRepresentation(
|
||||
ContextOfItems=body_context,
|
||||
RepresentationIdentifier="Body",
|
||||
RepresentationType="Brep",
|
||||
Items=brep_items,
|
||||
RepresentationType="MappedRepresentation",
|
||||
Items=[mapped_item],
|
||||
)
|
||||
return rep, placement
|
||||
|
||||
|
||||
def get_definition_object(obj: Base, definition_map: dict):
|
||||
"""
|
||||
Resolve the definition's source object for an InstanceProxy.
|
||||
Returns the first object referenced by the definition proxy, which
|
||||
carries the proper category/type info. Returns None if not found.
|
||||
"""
|
||||
definition_id = _get(obj, "definitionId") or ""
|
||||
if not definition_id:
|
||||
return None
|
||||
|
||||
ifc_proxies = definition_map.get("ifc_proxies", {})
|
||||
proxy = ifc_proxies.get(definition_id) or ifc_proxies.get(definition_id.lower())
|
||||
if proxy is None:
|
||||
return None
|
||||
|
||||
object_ids = _get(proxy, "objects") or []
|
||||
if not isinstance(object_ids, list):
|
||||
object_ids = list(object_ids)
|
||||
if not object_ids:
|
||||
return None
|
||||
|
||||
by_app_id = definition_map.get("by_app_id", {})
|
||||
source = by_app_id.get(str(object_ids[0]).lower())
|
||||
return source
|
||||
|
||||
|
||||
def print_instance_stats():
|
||||
total = _stats["found"] + _stats["not_found"]
|
||||
print(f" Instance resolution: {_stats['found']}/{total} definitions found")
|
||||
if _stats["not_found"] > 0:
|
||||
print(f" ⚠️ {_stats['not_found']} instances had no definition geometry")
|
||||
|
||||
|
||||
def reset_caches():
|
||||
"""Reset module-level caches (call at start of each export run)."""
|
||||
_mesh_data_cache.clear()
|
||||
_rep_map_cache.clear()
|
||||
_identity_placement_cache.clear()
|
||||
_stats["found"] = 0
|
||||
_stats["not_found"] = 0
|
||||
|
||||
+218
-72
@@ -1,53 +1,141 @@
|
||||
# =============================================================================
|
||||
# mapper.py
|
||||
# Maps Speckle speckle_type strings and Revit category names → IFC entity classes.
|
||||
# Maps Speckle objects → IFC entity classes.
|
||||
#
|
||||
# Strategy:
|
||||
# 1. Try to match speckle_type exactly or by prefix
|
||||
# 2. Fall back to Revit category name (e.g. "Floors" → IfcSlab)
|
||||
# 3. Fall back to IfcBuildingElementProxy if nothing matches
|
||||
# Strategy (priority order):
|
||||
# 1. builtInCategory (OST_ enum from properties.builtInCategory) — most reliable
|
||||
# 2. category_name string (traversal context) — display name fallback
|
||||
# 3. IfcBuildingElementProxy — last resort
|
||||
#
|
||||
# builtInCategory values: https://www.revitapidocs.com/2019/ba1c5b30-242f-5fdc-8ea9-ec3b61e6e722.htm
|
||||
# =============================================================================
|
||||
|
||||
|
||||
# --- speckle_type → IFC class ---
|
||||
# Covers Objects.BuiltElements.* from the Speckle Objects kit
|
||||
SPECKLE_TYPE_MAP: dict[str, str] = {
|
||||
"Objects.BuiltElements.Wall": "IfcWall",
|
||||
"Objects.BuiltElements.Floor": "IfcSlab",
|
||||
"Objects.BuiltElements.Roof": "IfcRoof",
|
||||
"Objects.BuiltElements.Column": "IfcColumn",
|
||||
"Objects.BuiltElements.Beam": "IfcBeam",
|
||||
"Objects.BuiltElements.Brace": "IfcMember",
|
||||
"Objects.BuiltElements.Duct": "IfcDuctSegment",
|
||||
"Objects.BuiltElements.Pipe": "IfcPipeSegment",
|
||||
"Objects.BuiltElements.Wire": "IfcCableCarrierSegment",
|
||||
"Objects.BuiltElements.Opening": "IfcOpeningElement",
|
||||
"Objects.BuiltElements.Room": "IfcSpace",
|
||||
"Objects.BuiltElements.Ceiling": "IfcCovering",
|
||||
"Objects.BuiltElements.Stair": "IfcStair",
|
||||
"Objects.BuiltElements.Ramp": "IfcRamp",
|
||||
"Objects.BuiltElements.Foundation": "IfcFooting",
|
||||
"Objects.BuiltElements.Grid": "IfcGrid",
|
||||
"Objects.BuiltElements.Level": "IfcBuildingStorey",
|
||||
"Objects.BuiltElements.Revit.RevitWall": "IfcWall",
|
||||
"Objects.BuiltElements.Revit.RevitFloor": "IfcSlab",
|
||||
"Objects.BuiltElements.Revit.RevitRoof": "IfcRoof",
|
||||
"Objects.BuiltElements.Revit.RevitColumn": "IfcColumn",
|
||||
"Objects.BuiltElements.Revit.RevitBeam": "IfcBeam",
|
||||
"Objects.BuiltElements.Revit.RevitBrace": "IfcMember",
|
||||
"Objects.BuiltElements.Revit.RevitDuct": "IfcDuctSegment",
|
||||
"Objects.BuiltElements.Revit.RevitPipe": "IfcPipeSegment",
|
||||
"Objects.BuiltElements.Revit.RevitRoom": "IfcSpace",
|
||||
"Objects.BuiltElements.Revit.RevitStair": "IfcStair",
|
||||
"Objects.BuiltElements.Revit.RevitRailing": "IfcRailing",
|
||||
"Objects.BuiltElements.Revit.RevitCeiling": "IfcCovering",
|
||||
"Objects.BuiltElements.Revit.RevitTopography": "IfcGeographicElement",
|
||||
"Objects.BuiltElements.Revit.RevitElementType": "IfcBuildingElementProxy",
|
||||
"Objects.Geometry.Mesh": "IfcBuildingElementProxy",
|
||||
"Objects.Geometry.Brep": "IfcBuildingElementProxy",
|
||||
# --- OST_ BuiltInCategory → IFC class (primary lookup) ---
|
||||
BUILTIN_CATEGORY_MAP: dict[str, str] = {
|
||||
# Architectural - Walls
|
||||
"OST_Walls": "IfcWall",
|
||||
"OST_CurtainWallPanels": "IfcCurtainWall",
|
||||
"OST_CurtainWallMullions": "IfcMember",
|
||||
"OST_Fascia": "IfcCovering",
|
||||
"OST_Gutters": "IfcPipeSegment",
|
||||
|
||||
# Architectural - Floors / Roofs / Ceilings
|
||||
"OST_Floors": "IfcSlab",
|
||||
"OST_Roofs": "IfcRoof",
|
||||
"OST_Ceilings": "IfcCovering",
|
||||
"OST_RoofSoffit": "IfcCovering",
|
||||
|
||||
# Architectural - Doors / Windows / Openings
|
||||
"OST_Doors": "IfcDoor",
|
||||
"OST_Windows": "IfcWindow",
|
||||
"OST_CurtainWallFamilies": "IfcCurtainWall",
|
||||
"OST_Skylights": "IfcWindow",
|
||||
|
||||
# Architectural - Stairs / Ramps / Railings
|
||||
"OST_Stairs": "IfcStair",
|
||||
"OST_StairsRailing": "IfcRailing",
|
||||
"OST_RailingTopRail": "IfcRailing",
|
||||
"OST_Ramps": "IfcRamp",
|
||||
"OST_StairsLandings": "IfcStairFlight",
|
||||
"OST_StairsRuns": "IfcStairFlight",
|
||||
"OST_StairsSupports": "IfcMember",
|
||||
|
||||
# Architectural - Rooms / Spaces
|
||||
"OST_Rooms": "IfcSpace",
|
||||
"OST_Parking": "IfcSpace",
|
||||
"OST_Areas": "IfcSpace",
|
||||
|
||||
# Architectural - Furniture / Casework
|
||||
"OST_Furniture": "IfcFurnishingElement",
|
||||
"OST_FurnitureSystems": "IfcFurnishingElement",
|
||||
"OST_Casework": "IfcFurnishingElement",
|
||||
"OST_SpecialtyEquipment": "IfcFurnishingElement",
|
||||
"OST_Entourage": "IfcFurnishingElement",
|
||||
|
||||
# Structural
|
||||
"OST_StructuralColumns": "IfcColumn",
|
||||
"OST_Columns": "IfcColumn",
|
||||
"OST_StructuralFraming": "IfcBeam",
|
||||
"OST_StructuralFoundation": "IfcFooting",
|
||||
"OST_FoundationSlab": "IfcSlab",
|
||||
"OST_StructuralStiffener": "IfcMember",
|
||||
"OST_StructuralTruss": "IfcMember",
|
||||
"OST_StructuralConnectionModel": "IfcMechanicalFastener",
|
||||
"OST_StructConnections": "IfcMechanicalFastener",
|
||||
"OST_Rebar": "IfcReinforcingBar",
|
||||
"OST_FabricAreas": "IfcReinforcingMesh",
|
||||
"OST_FabricReinforcement": "IfcReinforcingMesh",
|
||||
|
||||
# MEP - HVAC
|
||||
"OST_DuctCurves": "IfcDuctSegment",
|
||||
"OST_DuctFitting": "IfcDuctFitting",
|
||||
"OST_DuctAccessory": "IfcDuctSegment",
|
||||
"OST_DuctTerminal": "IfcAirTerminal",
|
||||
"OST_FlexDuctCurves": "IfcDuctSegment",
|
||||
"OST_MechanicalEquipment": "IfcUnitaryEquipment",
|
||||
"OST_AirTerminal": "IfcAirTerminal",
|
||||
|
||||
# MEP - Plumbing
|
||||
"OST_PipeCurves": "IfcPipeSegment",
|
||||
"OST_PipeFitting": "IfcPipeFitting",
|
||||
"OST_PipeAccessory": "IfcPipeSegment",
|
||||
"OST_FlexPipeCurves": "IfcPipeSegment",
|
||||
"OST_PlumbingFixtures": "IfcSanitaryTerminal",
|
||||
"OST_PlumbingEquipment": "IfcSanitaryTerminal",
|
||||
"OST_Sprinklers": "IfcFireSuppressionTerminal",
|
||||
|
||||
# MEP - Electrical
|
||||
"OST_ElectricalEquipment": "IfcElectricDistributionBoard",
|
||||
"OST_ElectricalFixtures": "IfcElectricAppliance",
|
||||
"OST_LightingFixtures": "IfcLightFixture",
|
||||
"OST_LightingDevices": "IfcLightFixture",
|
||||
"OST_CableTray": "IfcCableCarrierSegment",
|
||||
"OST_CableTrayFitting": "IfcCableCarrierFitting",
|
||||
"OST_Conduit": "IfcCableCarrierSegment",
|
||||
"OST_ConduitFitting": "IfcCableCarrierFitting",
|
||||
"OST_CommunicationDevices": "IfcElectricAppliance",
|
||||
"OST_DataDevices": "IfcElectricAppliance",
|
||||
"OST_FireAlarmDevices": "IfcAlarm",
|
||||
"OST_SecurityDevices": "IfcAlarm",
|
||||
"OST_NurseCallDevices": "IfcElectricAppliance",
|
||||
|
||||
# Site / Civil
|
||||
"OST_Site": "IfcSite",
|
||||
"OST_Topography": "IfcGeographicElement",
|
||||
"OST_Toposolid": "IfcGeographicElement",
|
||||
"OST_Roads": "IfcRoad",
|
||||
"OST_Hardscape": "IfcPavement",
|
||||
"OST_Planting": "IfcGeographicElement",
|
||||
"OST_SiteSurface": "IfcGeographicElement",
|
||||
|
||||
# Generic / Annotation (skip or proxy)
|
||||
"OST_GenericModel": "IfcBuildingElementProxy",
|
||||
"OST_Mass": "IfcBuildingElementProxy",
|
||||
"OST_DetailComponents": "IfcAnnotation",
|
||||
"OST_Lines": "IfcAnnotation",
|
||||
"OST_Grids": "IfcGrid",
|
||||
"OST_Levels": "IfcBuildingStorey",
|
||||
"OST_Views": "IfcAnnotation",
|
||||
}
|
||||
|
||||
# --- Revit category name → IFC class (fallback) ---
|
||||
|
||||
# --- OST categories to skip entirely (analytical / energy / separation lines) ---
|
||||
SKIP_CATEGORIES: set[str] = {
|
||||
"OST_MEPLoadAreaSeparationLines",
|
||||
"OST_EnergyAnalysisZones",
|
||||
"OST_EnergyAnalysisSurface",
|
||||
"OST_SolarShading",
|
||||
"OST_MEPAnalyticalPipeSegments",
|
||||
"OST_MEPAnalyticalDuctSegments",
|
||||
"OST_MEPAnalyticalSpaces",
|
||||
"OST_ElectricalConduitAnalyticalLines",
|
||||
"OST_MEPLoadBoundaryLines",
|
||||
"OST_FlowTerminalSeparationLines",
|
||||
}
|
||||
|
||||
|
||||
# --- Display category name → IFC class (secondary fallback) ---
|
||||
CATEGORY_MAP: dict[str, str] = {
|
||||
"Walls": "IfcWall",
|
||||
"Floors": "IfcSlab",
|
||||
@@ -66,6 +154,7 @@ CATEGORY_MAP: dict[str, str] = {
|
||||
"Stairs": "IfcStair",
|
||||
"Ramps": "IfcRamp",
|
||||
"Railings": "IfcRailing",
|
||||
"Top Rails": "IfcRailing",
|
||||
"Curtain Panels": "IfcCurtainWall",
|
||||
"Curtain Wall Mullions": "IfcMember",
|
||||
"Doors": "IfcDoor",
|
||||
@@ -74,64 +163,121 @@ CATEGORY_MAP: dict[str, str] = {
|
||||
"Furniture Systems": "IfcFurnishingElement",
|
||||
"Casework": "IfcFurnishingElement",
|
||||
"Plumbing Fixtures": "IfcSanitaryTerminal",
|
||||
"Plumbing Equipment": "IfcSanitaryTerminal",
|
||||
"Electrical Fixtures": "IfcElectricAppliance",
|
||||
"Lighting Fixtures": "IfcLightFixture",
|
||||
"Mechanical Equipment": "IfcUnitaryEquipment",
|
||||
"Electrical Equipment": "IfcElectricDistributionBoard",
|
||||
"Structural Rebar": "IfcReinforcingBar",
|
||||
"Structural Connections": "IfcMechanicalFastener",
|
||||
"Structural Foundations": "IfcFooting",
|
||||
"Foundation Slabs": "IfcSlab",
|
||||
"Topography": "IfcGeographicElement",
|
||||
"Toposolid": "IfcGeographicElement",
|
||||
"Planting": "IfcGeographicElement",
|
||||
"Site": "IfcSite",
|
||||
"Parking": "IfcSpace",
|
||||
"Generic Models": "IfcBuildingElementProxy",
|
||||
"Mass": "IfcBuildingElementProxy",
|
||||
"Specialty Equipment": "IfcBuildingElementProxy",
|
||||
"Specialty Equipment": "IfcFurnishingElement",
|
||||
}
|
||||
|
||||
|
||||
def classify(obj, category_name: str = "") -> str:
|
||||
_bic_cache: dict[int, str | None] = {} # id(obj) → builtInCategory
|
||||
|
||||
|
||||
def _get_builtin_category(obj) -> str | None:
|
||||
"""
|
||||
Read builtInCategory from obj.properties.builtInCategory.
|
||||
Returns the OST_ string or None. Cached per object.
|
||||
"""
|
||||
oid = id(obj)
|
||||
if oid in _bic_cache:
|
||||
return _bic_cache[oid]
|
||||
result = None
|
||||
try:
|
||||
props = getattr(obj, "properties", None)
|
||||
if props is None:
|
||||
try:
|
||||
props = obj["properties"]
|
||||
except Exception:
|
||||
pass
|
||||
if props is not None:
|
||||
val = getattr(props, "builtInCategory", None)
|
||||
if val is None:
|
||||
try:
|
||||
val = props["builtInCategory"]
|
||||
except Exception:
|
||||
pass
|
||||
if val and isinstance(val, str):
|
||||
result = val.strip()
|
||||
except Exception:
|
||||
pass
|
||||
_bic_cache[oid] = result
|
||||
return result
|
||||
|
||||
|
||||
# Pre-computed lowercase category map for substring matching
|
||||
_CATEGORY_MAP_LOWER: list[tuple[str, str]] = [
|
||||
(k.lower(), v) for k, v in CATEGORY_MAP.items()
|
||||
]
|
||||
|
||||
# Classification cache: (obj_id, category_name) → ifc_class
|
||||
_classify_cache: dict[tuple, str] = {}
|
||||
|
||||
|
||||
def classify(obj, category_name: str = "") -> str | None:
|
||||
"""
|
||||
Determine the IFC class for a Speckle object.
|
||||
|
||||
With the new Objects.Data.DataObject:Objects.Data.RevitObject speckle_type,
|
||||
category name is now the primary classification signal.
|
||||
|
||||
Args:
|
||||
obj: A specklepy Base object (leaf element).
|
||||
category_name: The Revit category string from the traversal context
|
||||
e.g. "Floors", "Walls", "Structural Columns"
|
||||
|
||||
Returns:
|
||||
An IFC class name string e.g. "IfcWall"
|
||||
Priority:
|
||||
1. properties.builtInCategory (OST_ enum) — definitive Revit classification
|
||||
2. category_name from traversal context (display string)
|
||||
3. obj.category field
|
||||
4. IfcBuildingElementProxy fallback
|
||||
"""
|
||||
speckle_type = getattr(obj, "speckle_type", "") or ""
|
||||
cache_key = (id(obj), category_name)
|
||||
if cache_key in _classify_cache:
|
||||
return _classify_cache[cache_key]
|
||||
|
||||
# 1. Category name — PRIMARY lookup for RevitObject types
|
||||
result = _classify_impl(obj, category_name)
|
||||
_classify_cache[cache_key] = result
|
||||
return result
|
||||
|
||||
|
||||
def _classify_impl(obj, category_name: str) -> str | None:
|
||||
# 0. Skip analytical / energy / separation-line categories
|
||||
bic = _get_builtin_category(obj)
|
||||
if bic and bic in SKIP_CATEGORIES:
|
||||
return None
|
||||
|
||||
# 1. builtInCategory — most reliable, direct Revit enum
|
||||
if bic and bic in BUILTIN_CATEGORY_MAP:
|
||||
return BUILTIN_CATEGORY_MAP[bic]
|
||||
|
||||
# 2. category_name from traversal context — exact match first
|
||||
if category_name:
|
||||
# Exact match
|
||||
if category_name in CATEGORY_MAP:
|
||||
return CATEGORY_MAP[category_name]
|
||||
# Partial match handles Revit appending IDs e.g. "Structural Framing [12345]"
|
||||
for key, ifc_class in CATEGORY_MAP.items():
|
||||
if key.lower() in category_name.lower():
|
||||
cat_lower = category_name.lower()
|
||||
for key_lower, ifc_class in _CATEGORY_MAP_LOWER:
|
||||
if key_lower in cat_lower:
|
||||
return ifc_class
|
||||
|
||||
# 2. Read 'category' directly off the object itself
|
||||
# Per docs: category is a TOP-LEVEL field on RevitObject, not inside properties
|
||||
# 3. obj.category field
|
||||
obj_category = getattr(obj, "category", None)
|
||||
if obj_category and isinstance(obj_category, str):
|
||||
if obj_category in CATEGORY_MAP:
|
||||
return CATEGORY_MAP[obj_category]
|
||||
for key, ifc_class in CATEGORY_MAP.items():
|
||||
if key.lower() in obj_category.lower():
|
||||
obj_cat_lower = obj_category.lower()
|
||||
for key_lower, ifc_class in _CATEGORY_MAP_LOWER:
|
||||
if key_lower in obj_cat_lower:
|
||||
return ifc_class
|
||||
|
||||
# 3. speckle_type — fallback for non-RevitObject types (geometry, structural, etc.)
|
||||
if speckle_type in SPECKLE_TYPE_MAP:
|
||||
return SPECKLE_TYPE_MAP[speckle_type]
|
||||
for key, ifc_class in SPECKLE_TYPE_MAP.items():
|
||||
if speckle_type.startswith(key):
|
||||
return ifc_class
|
||||
return "IfcBuildingElementProxy"
|
||||
|
||||
# 4. Last resort
|
||||
return "IfcBuildingElementProxy"
|
||||
|
||||
def reset_caches():
|
||||
"""Clear module-level caches (call at start of each export run)."""
|
||||
_bic_cache.clear()
|
||||
_classify_cache.clear()
|
||||
|
||||
+1
-1
@@ -136,7 +136,7 @@ class MaterialManager:
|
||||
return style
|
||||
|
||||
def apply_to_item(self, item, mesh_app_id: str):
|
||||
"""Assign the material style to a single IFC geometry item (e.g. IfcFacetedBrep)."""
|
||||
"""Assign the material style to a single IFC geometry item (e.g. IfcPolygonalFaceSet)."""
|
||||
style = self.get_style(mesh_app_id)
|
||||
if style is None:
|
||||
return
|
||||
|
||||
+817
-138
@@ -1,177 +1,856 @@
|
||||
# =============================================================================
|
||||
# properties.py
|
||||
# Extracts Revit data from Speckle DataObjects and writes IFC property sets.
|
||||
# Writes IFC property sets matching the structure of Revit's native IFC export.
|
||||
#
|
||||
# Revit parameter structure from the Speckle connector:
|
||||
# obj.properties = {
|
||||
# "elementId": "704282",
|
||||
# "Parameters": {
|
||||
# "Type Parameters": {
|
||||
# "Dimensions": {
|
||||
# "Thickness": {"name": "Thickness", "value": 25.4, "units": "Millimeters", ...}
|
||||
# },
|
||||
# ...
|
||||
# },
|
||||
# "Instance Parameters": {
|
||||
# "Constraints": {
|
||||
# "Level": {"name": "Level", "value": "Level 1", ...}
|
||||
# },
|
||||
# ...
|
||||
# }
|
||||
# }
|
||||
# }
|
||||
# Revit native IFC export produces:
|
||||
# - Element Name: "Family:TypeName:ElementId" e.g. "Basic Roof:SG Metal Panels roof:243274"
|
||||
# - Element Tag: ElementId string e.g. "243274"
|
||||
# - Element GlobalId: from IFC Parameters.IfcGUID
|
||||
# - Pset_<EntityType>Common with typed properties (IfcBoolean, IfcIdentifier, etc.)
|
||||
# - Pset_EnvironmentalImpactIndicators with Reference = TypeName
|
||||
#
|
||||
# We flatten this into two IFC property sets:
|
||||
# Pset_RevitTypeParameters — from "Type Parameters"
|
||||
# Pset_RevitInstanceParameters — from "Instance Parameters"
|
||||
# Our Speckle source fields:
|
||||
# obj.family → Family name
|
||||
# obj.type → Type name (= Reference in all Common psets)
|
||||
# properties.elementId → Revit ElementId → Tag
|
||||
# properties.Parameters.Instance Parameters.IFC Parameters.IfcGUID.value → GlobalId
|
||||
# properties.Parameters.Type Parameters.* → typed IFC properties
|
||||
# properties.Parameters.Instance Parameters.* → typed IFC properties
|
||||
# =============================================================================
|
||||
|
||||
import ifcopenshell.api
|
||||
from specklepy.objects.base import Base
|
||||
|
||||
|
||||
def _safe_val(value) -> str | None:
|
||||
"""Convert a value to a clean IFC-safe string."""
|
||||
# ---------------------------------------------------------------------------
|
||||
# IFC entity → standard Common pset name
|
||||
# ---------------------------------------------------------------------------
|
||||
COMMON_PSET: dict[str, str] = {
|
||||
"IfcWall": "Pset_WallCommon",
|
||||
"IfcWallStandardCase": "Pset_WallCommon",
|
||||
"IfcSlab": "Pset_SlabCommon",
|
||||
"IfcRoof": "Pset_RoofCommon",
|
||||
"IfcColumn": "Pset_ColumnCommon",
|
||||
"IfcBeam": "Pset_BeamCommon",
|
||||
"IfcMember": "Pset_MemberCommon",
|
||||
"IfcDoor": "Pset_DoorCommon",
|
||||
"IfcWindow": "Pset_WindowCommon",
|
||||
"IfcStair": "Pset_StairCommon",
|
||||
"IfcStairFlight": "Pset_StairFlightCommon",
|
||||
"IfcRamp": "Pset_RampCommon",
|
||||
"IfcRailing": "Pset_RailingCommon",
|
||||
"IfcCovering": "Pset_CoveringCommon",
|
||||
"IfcCurtainWall": "Pset_CurtainWallCommon",
|
||||
"IfcFooting": "Pset_FootingCommon",
|
||||
"IfcPile": "Pset_PileCommon",
|
||||
"IfcSpace": "Pset_SpaceCommon",
|
||||
"IfcSite": "Pset_SiteCommon",
|
||||
"IfcBuildingStorey": "Pset_BuildingStoreyCommon",
|
||||
"IfcBuilding": "Pset_BuildingCommon",
|
||||
"IfcBuildingElementProxy": "Pset_BuildingElementProxyCommon",
|
||||
"IfcFurnishingElement": "Pset_FurnitureTypeCommon",
|
||||
"IfcLightFixture": "Pset_LightFixtureTypeCommon",
|
||||
"IfcOpeningElement": "Pset_OpeningElementCommon",
|
||||
"IfcPlate": "Pset_PlateCommon",
|
||||
"IfcGeographicElement": "Pset_SiteCommon",
|
||||
"IfcPipeFitting": "Pset_PipeFittingTypeCommon",
|
||||
"IfcSanitaryTerminal": "Pset_SanitaryTerminalTypeCommon",
|
||||
"IfcReinforcingBar": "Pset_ReinforcingBarBendingsBECCommon",
|
||||
"IfcMechanicalFastener": "Pset_MechanicalFastenerTypeCommon",
|
||||
}
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Revit parameter internal names → (IFC pset property name, IFC value factory)
|
||||
# These are harvested from the Common psets Revit native export produces.
|
||||
# ---------------------------------------------------------------------------
|
||||
def _bool(v):
|
||||
return ("IfcBoolean", bool(v))
|
||||
|
||||
def _identifier(v):
|
||||
return ("IfcIdentifier", str(v))
|
||||
|
||||
def _label(v):
|
||||
return ("IfcLabel", str(v))
|
||||
|
||||
def _real(v):
|
||||
return ("IfcReal", float(v))
|
||||
|
||||
def _thermal(v):
|
||||
return ("IfcThermalTransmittanceMeasure", float(v))
|
||||
|
||||
def _length(v):
|
||||
return ("IfcPositiveLengthMeasure", float(v))
|
||||
|
||||
def _count(v):
|
||||
return ("IfcCountMeasure", int(v))
|
||||
|
||||
def _angle(v):
|
||||
return ("IfcPlaneAngleMeasure", float(v))
|
||||
|
||||
|
||||
# Map: Revit internalDefinitionName → (IFC property name, value factory fn)
|
||||
REVIT_PARAM_TO_IFC: dict[str, tuple] = {
|
||||
# Wall
|
||||
"WALL_ATTR_ROOM_BOUNDING": ("IsExternal", _bool),
|
||||
"WALL_STRUCTURAL_SIGNIFICANT": ("LoadBearing", _bool),
|
||||
"WALL_STRUCTURAL_USAGE_PARAM": ("LoadBearing", _bool),
|
||||
"ANALYTICAL_THERMAL_RESISTANCE": ("ThermalTransmittance", _thermal),
|
||||
"ANALYTICAL_HEAT_TRANSFER_COEFFICIENT": ("ThermalTransmittance", _thermal),
|
||||
|
||||
# Slab / Roof / Floor
|
||||
"HOST_AREA_COMPUTED": ("NetArea", _real),
|
||||
"HOST_VOLUME_COMPUTED": ("NetVolume", _real),
|
||||
"ROOF_SLOPE": ("PitchAngle", _angle),
|
||||
|
||||
# Stair
|
||||
"STAIR_RISER_HEIGHT": ("RiserHeight", _length),
|
||||
"STAIR_TREAD_DEPTH": ("TreadLength", _length),
|
||||
"STAIR_NUMBER_OF_RISERS": ("NumberOfRiser", _count),
|
||||
"STAIR_NUMBER_OF_TREADS": ("NumberOfTreads", _count),
|
||||
"STAIR_NOSING_LENGTH": ("NosingLength", _length),
|
||||
|
||||
# Railing
|
||||
"RAILING_HEIGHT": ("Height", _length),
|
||||
|
||||
# Door / Window
|
||||
"DOOR_FIRE_RATING": ("FireExit", _bool),
|
||||
|
||||
# General identity
|
||||
"ALL_MODEL_FAMILY_NAME": ("Reference", _identifier),
|
||||
"ALL_MODEL_TYPE_NAME": ("Reference", _identifier),
|
||||
"ASSEMBLY_CODE": ("Reference", _identifier),
|
||||
}
|
||||
|
||||
# External category OST_ codes (used to infer IsExternal)
|
||||
EXTERNAL_CATEGORIES = {
|
||||
"OST_Walls", "OST_Roofs", "OST_Windows", "OST_Doors",
|
||||
"OST_CurtainWallPanels", "OST_CurtainWallMullions",
|
||||
"OST_StructuralColumns", "OST_StructuralFraming",
|
||||
"OST_Stairs", "OST_StairsRailing", "OST_Ramps",
|
||||
}
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Helpers
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
_props_cache: dict[int, dict] = {} # id(obj) → props dict
|
||||
|
||||
|
||||
def _get_props_dict(obj: Base) -> dict:
|
||||
"""Get properties as a plain dict. Cached per object to avoid repeated conversion."""
|
||||
oid = id(obj)
|
||||
if oid in _props_cache:
|
||||
return _props_cache[oid]
|
||||
# Try getattr first — matches the pattern that works in other Speckle scripts
|
||||
p = getattr(obj, "properties", None)
|
||||
if p is None:
|
||||
for key in ["properties", "@properties"]:
|
||||
try:
|
||||
p = obj[key]
|
||||
if p is not None:
|
||||
break
|
||||
except Exception:
|
||||
continue
|
||||
if p is None:
|
||||
_props_cache[oid] = {}
|
||||
return {}
|
||||
result = _to_dict(p)
|
||||
_props_cache[oid] = result
|
||||
return result
|
||||
|
||||
|
||||
def _get_nested(d, *keys):
|
||||
"""Safely walk nested dicts/objects."""
|
||||
cur = d
|
||||
for k in keys:
|
||||
if cur is None:
|
||||
return None
|
||||
cur = _safe_get(cur, k)
|
||||
return cur
|
||||
|
||||
|
||||
_to_dict_cache: dict[int, dict] = {} # id(obj) → converted dict
|
||||
|
||||
|
||||
def _to_dict(obj) -> dict:
|
||||
"""Convert a Speckle Base object or dict to a plain dict. Returns {} on failure.
|
||||
Cached per object identity to avoid repeated conversion."""
|
||||
if obj is None:
|
||||
return {}
|
||||
if isinstance(obj, dict):
|
||||
return obj
|
||||
oid = id(obj)
|
||||
if oid in _to_dict_cache:
|
||||
return _to_dict_cache[oid]
|
||||
# Try .get_dynamic_member_names() for Speckle Base objects
|
||||
if hasattr(obj, "get_dynamic_member_names"):
|
||||
result = {}
|
||||
try:
|
||||
names = obj.get_dynamic_member_names()
|
||||
except Exception:
|
||||
_to_dict_cache[oid] = {}
|
||||
return {}
|
||||
for n in names:
|
||||
try:
|
||||
result[n] = obj[n]
|
||||
except Exception:
|
||||
pass
|
||||
_to_dict_cache[oid] = result
|
||||
return result
|
||||
# Last resort: try common dict-like patterns
|
||||
if hasattr(obj, "items"):
|
||||
try:
|
||||
result = dict(obj.items())
|
||||
_to_dict_cache[oid] = result
|
||||
return result
|
||||
except Exception:
|
||||
pass
|
||||
_to_dict_cache[oid] = {}
|
||||
return {}
|
||||
|
||||
|
||||
def _safe_get(obj, key, default=None):
|
||||
"""Safe key access for both dicts and Speckle Base objects."""
|
||||
if obj is None:
|
||||
return default
|
||||
if isinstance(obj, dict):
|
||||
return obj.get(key, default)
|
||||
# Try getattr first (works reliably for Speckle Base)
|
||||
try:
|
||||
val = getattr(obj, key, None)
|
||||
if val is not None:
|
||||
return val
|
||||
except Exception:
|
||||
pass
|
||||
# Fallback to bracket access
|
||||
try:
|
||||
val = obj[key]
|
||||
if val is not None:
|
||||
return val
|
||||
except Exception:
|
||||
pass
|
||||
return default
|
||||
|
||||
|
||||
def _param_value(params_block, internal_name: str):
|
||||
"""
|
||||
Search all groups in a parameter block for a param with the given
|
||||
internalDefinitionName. Returns the raw value or None.
|
||||
Handles both plain dicts and Speckle Base objects.
|
||||
"""
|
||||
block = _to_dict(params_block)
|
||||
if not block:
|
||||
return None
|
||||
for group in block.values():
|
||||
group_d = _to_dict(group)
|
||||
if not group_d:
|
||||
continue
|
||||
for entry in group_d.values():
|
||||
entry_d = _to_dict(entry)
|
||||
if not entry_d:
|
||||
continue
|
||||
if entry_d.get("internalDefinitionName") == internal_name:
|
||||
return entry_d.get("value")
|
||||
return None
|
||||
|
||||
|
||||
def _make_prop(ifc, name: str, ifc_type: str, value) -> object | None:
|
||||
"""Create an IfcPropertySingleValue with the correct IFC measure type."""
|
||||
try:
|
||||
nominal = ifc.create_entity(ifc_type, wrappedValue=value)
|
||||
return ifc.create_entity(
|
||||
"IfcPropertySingleValue",
|
||||
Name=name,
|
||||
NominalValue=nominal,
|
||||
)
|
||||
except Exception as e:
|
||||
return None
|
||||
|
||||
|
||||
def _write_pset(ifc, element, pset_name: str, props: list):
|
||||
"""Write an IfcPropertySet with the given list of IfcProperty objects."""
|
||||
if not props:
|
||||
return
|
||||
try:
|
||||
pset = ifcopenshell.api.run("pset.add_pset", ifc, product=element, name=pset_name)
|
||||
# Directly attach the pre-built property objects
|
||||
pset.HasProperties = props
|
||||
except Exception as e:
|
||||
print(f" ⚠️ {pset_name}: {e}")
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Element name + tag (matching Revit native IFC format)
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def build_element_name(obj: Base) -> str:
|
||||
"""
|
||||
Build element name in Revit native IFC format: "Family:TypeName:ElementId"
|
||||
Falls back gracefully if any part is missing.
|
||||
"""
|
||||
family = getattr(obj, "family", None) or ""
|
||||
typ = getattr(obj, "type", None) or ""
|
||||
|
||||
# Treat literal "none" (case-insensitive) the same as empty — Revit exports
|
||||
# placeholder objects with family/type set to the string "none".
|
||||
if family.strip().lower() == "none":
|
||||
family = ""
|
||||
if typ.strip().lower() == "none":
|
||||
typ = ""
|
||||
|
||||
parts = [p for p in [family, typ] if p]
|
||||
return ":".join(parts) if parts else (getattr(obj, "id", None) or "unnamed")
|
||||
|
||||
|
||||
def get_element_tag(obj: Base) -> str | None:
|
||||
"""Return Revit ElementId as the IFC Tag."""
|
||||
props = _get_props_dict(obj)
|
||||
elem_id = _safe_get(props, "elementId")
|
||||
return str(elem_id) if elem_id else None
|
||||
|
||||
|
||||
def get_ifc_guid(obj: Base) -> str | None:
|
||||
"""
|
||||
Read IfcGUID from the Revit IFC Parameters.
|
||||
Falls back to None (ifcopenshell will auto-generate a GUID).
|
||||
"""
|
||||
props = _get_props_dict(obj)
|
||||
params = _safe_get(props, "Parameters", {})
|
||||
inst = _safe_get(params, "Instance Parameters", {})
|
||||
ifc_p = _safe_get(inst, "IFC Parameters", {})
|
||||
entry = _safe_get(ifc_p, "IfcGUID", {})
|
||||
entry_d = _to_dict(entry) if not isinstance(entry, dict) else entry
|
||||
val = entry_d.get("value") if entry_d else None
|
||||
return str(val) if val else None
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Standard Common pset (Pset_WallCommon etc.)
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def write_common_pset(ifc, element, obj: Base, ifc_class: str, category_name: str = ""):
|
||||
"""
|
||||
Write the standard Pset_<Entity>Common property set, matching Revit native export.
|
||||
Properties: Reference (TypeName), IsExternal, LoadBearing, ThermalTransmittance, etc.
|
||||
"""
|
||||
pset_name = COMMON_PSET.get(ifc_class)
|
||||
if not pset_name:
|
||||
return
|
||||
|
||||
props = _get_props_dict(obj)
|
||||
params = _safe_get(props, "Parameters", {})
|
||||
type_params = _safe_get(params, "Type Parameters", {})
|
||||
inst_params = _safe_get(params, "Instance Parameters", {})
|
||||
|
||||
ifc_props = []
|
||||
|
||||
# Reference = TypeName (always present in Revit IFC)
|
||||
type_name = getattr(obj, "type", None) or ""
|
||||
if type_name:
|
||||
p = _make_prop(ifc, "Reference", "IfcIdentifier", type_name)
|
||||
if p:
|
||||
ifc_props.append(p)
|
||||
|
||||
# IsExternal — derive from builtInCategory or "Constraints" parameters
|
||||
bic = _safe_get(props, "builtInCategory", "")
|
||||
is_external = bic in EXTERNAL_CATEGORIES
|
||||
if not is_external:
|
||||
# Some elements expose it directly as a parameter
|
||||
ext_val = _param_value(inst_params, "WALL_ATTR_ROOM_BOUNDING")
|
||||
if ext_val is not None:
|
||||
is_external = bool(ext_val)
|
||||
if ifc_class not in {"IfcSpace", "IfcSite", "IfcBuildingStorey", "IfcBuilding",
|
||||
"IfcFurnishingElement", "IfcOpeningElement"}:
|
||||
p = _make_prop(ifc, "IsExternal", "IfcBoolean", is_external)
|
||||
if p:
|
||||
ifc_props.append(p)
|
||||
|
||||
# LoadBearing — walls, columns, beams, slabs
|
||||
if ifc_class in {"IfcWall", "IfcWallStandardCase", "IfcSlab", "IfcColumn", "IfcBeam"}:
|
||||
lb_val = (_param_value(inst_params, "WALL_STRUCTURAL_SIGNIFICANT") or
|
||||
_param_value(inst_params, "WALL_STRUCTURAL_USAGE_PARAM") or
|
||||
_param_value(type_params, "WALL_STRUCTURAL_SIGNIFICANT"))
|
||||
lb = bool(lb_val) if lb_val is not None else False
|
||||
p = _make_prop(ifc, "LoadBearing", "IfcBoolean", lb)
|
||||
if p:
|
||||
ifc_props.append(p)
|
||||
|
||||
# ThermalTransmittance — walls, roofs, slabs, doors, windows
|
||||
if ifc_class in {"IfcWall", "IfcWallStandardCase", "IfcRoof", "IfcSlab",
|
||||
"IfcDoor", "IfcWindow"}:
|
||||
u_val = (_param_value(type_params, "ANALYTICAL_HEAT_TRANSFER_COEFFICIENT") or
|
||||
_param_value(inst_params, "ANALYTICAL_HEAT_TRANSFER_COEFFICIENT"))
|
||||
if u_val is not None:
|
||||
try:
|
||||
p = _make_prop(ifc, "ThermalTransmittance", "IfcThermalTransmittanceMeasure", float(u_val))
|
||||
if p:
|
||||
ifc_props.append(p)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# PitchAngle — roofs/slabs
|
||||
if ifc_class in {"IfcRoof", "IfcSlab"}:
|
||||
slope = _param_value(inst_params, "ROOF_SLOPE")
|
||||
if slope is not None:
|
||||
try:
|
||||
p = _make_prop(ifc, "PitchAngle", "IfcPlaneAngleMeasure", float(slope))
|
||||
if p:
|
||||
ifc_props.append(p)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# Stair-specific
|
||||
if ifc_class in {"IfcStair", "IfcStairFlight"}:
|
||||
for internal, prop_name, factory in [
|
||||
("STAIR_RISER_HEIGHT", "RiserHeight", "IfcPositiveLengthMeasure"),
|
||||
("STAIR_TREAD_DEPTH", "TreadLength", "IfcPositiveLengthMeasure"),
|
||||
("STAIR_NUMBER_OF_RISERS","NumberOfRiser", "IfcCountMeasure"),
|
||||
("STAIR_NUMBER_OF_TREADS","NumberOfTreads", "IfcCountMeasure"),
|
||||
]:
|
||||
v = _param_value(inst_params, internal) or _param_value(type_params, internal)
|
||||
if v is not None:
|
||||
try:
|
||||
p = _make_prop(ifc, prop_name, factory, float(v) if "Measure" in factory else int(v))
|
||||
if p:
|
||||
ifc_props.append(p)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# Railing height
|
||||
if ifc_class == "IfcRailing":
|
||||
h = _param_value(inst_params, "RAILING_HEIGHT") or _param_value(type_params, "RAILING_HEIGHT")
|
||||
if h is not None:
|
||||
try:
|
||||
p = _make_prop(ifc, "Height", "IfcPositiveLengthMeasure", float(h))
|
||||
if p:
|
||||
ifc_props.append(p)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# IfcSpace-specific: set Name, LongName, Category, and BaseQuantities
|
||||
if ifc_class == "IfcSpace":
|
||||
_write_space_properties(ifc, element, obj, ifc_props)
|
||||
|
||||
_write_pset(ifc, element, pset_name, ifc_props)
|
||||
|
||||
|
||||
def _write_space_properties(ifc, element, obj: Base, ifc_props: list):
|
||||
"""
|
||||
Set IfcSpace attributes and BaseQuantities from Revit Room parameters.
|
||||
|
||||
Uses internalDefinitionName to find values:
|
||||
ROOM_NUMBER → IfcSpace.Name + Pset_SpaceCommon.Reference
|
||||
ROOM_NAME → IfcSpace.LongName
|
||||
Occupant → Pset_SpaceCommon.Category
|
||||
ROOM_AREA → Qto_SpaceBaseQuantities.NetFloorArea
|
||||
ROOM_VOLUME → Qto_SpaceBaseQuantities.NetVolume
|
||||
"""
|
||||
props = _get_props_dict(obj)
|
||||
params = _safe_get(props, "Parameters", {})
|
||||
inst_params = _safe_get(params, "Instance Parameters", {})
|
||||
|
||||
# --- Room Number → IfcSpace.Name + Pset_SpaceCommon.Reference ---
|
||||
room_number = _param_value(inst_params, "ROOM_NUMBER")
|
||||
if room_number:
|
||||
room_number = str(room_number).strip()
|
||||
element.Name = room_number
|
||||
# Replace any existing Reference in ifc_props
|
||||
ifc_props[:] = [p for p in ifc_props if p.Name != "Reference"]
|
||||
p = _make_prop(ifc, "Reference", "IfcIdentifier", room_number)
|
||||
if p:
|
||||
ifc_props.append(p)
|
||||
# Also add as explicit RoomNumber in the pset
|
||||
p = _make_prop(ifc, "RoomNumber", "IfcLabel", room_number)
|
||||
if p:
|
||||
ifc_props.append(p)
|
||||
|
||||
# --- Room Name → IfcSpace.LongName + Pset_SpaceCommon.RoomName ---
|
||||
room_name = _param_value(inst_params, "ROOM_NAME")
|
||||
if not room_name:
|
||||
# Fallback to the Speckle object's own name
|
||||
room_name = getattr(obj, "name", None)
|
||||
if room_name:
|
||||
room_name = str(room_name).strip()
|
||||
try:
|
||||
element.LongName = room_name
|
||||
except AttributeError:
|
||||
pass
|
||||
p = _make_prop(ifc, "RoomName", "IfcLabel", room_name)
|
||||
if p:
|
||||
ifc_props.append(p)
|
||||
|
||||
# --- Occupant → Pset_SpaceCommon.Category ---
|
||||
occupant = _param_value(inst_params, "Occupant")
|
||||
if occupant:
|
||||
p = _make_prop(ifc, "Category", "IfcLabel", str(occupant).strip())
|
||||
if p:
|
||||
ifc_props.append(p)
|
||||
|
||||
# --- Area & Volume → Qto_SpaceBaseQuantities ---
|
||||
quantities = []
|
||||
|
||||
area_val = _param_value(inst_params, "ROOM_AREA")
|
||||
if area_val is not None:
|
||||
try:
|
||||
q = ifc.create_entity(
|
||||
"IfcQuantityArea",
|
||||
Name="NetFloorArea",
|
||||
AreaValue=float(area_val),
|
||||
)
|
||||
quantities.append(q)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
volume_val = _param_value(inst_params, "ROOM_VOLUME")
|
||||
if volume_val is not None:
|
||||
try:
|
||||
q = ifc.create_entity(
|
||||
"IfcQuantityVolume",
|
||||
Name="NetVolume",
|
||||
VolumeValue=float(volume_val),
|
||||
)
|
||||
quantities.append(q)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
if quantities:
|
||||
try:
|
||||
qto = ifcopenshell.api.run(
|
||||
"pset.add_qto", ifc,
|
||||
product=element,
|
||||
name="Qto_SpaceBaseQuantities",
|
||||
)
|
||||
qto.Quantities = quantities
|
||||
except Exception as e:
|
||||
print(f" ⚠️ Qto_SpaceBaseQuantities: {e}")
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Pset_EnvironmentalImpactIndicators (always written, Reference = TypeName)
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def write_environmental_pset(ifc, element, obj: Base):
|
||||
"""Write Pset_EnvironmentalImpactIndicators with Reference = TypeName."""
|
||||
type_name = getattr(obj, "type", None) or ""
|
||||
if not type_name:
|
||||
return
|
||||
p = _make_prop(ifc, "Reference", "IfcIdentifier", type_name)
|
||||
if p:
|
||||
_write_pset(ifc, element, "Pset_EnvironmentalImpactIndicators", [p])
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Custom Revit parameters pset (all remaining instance + type params)
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def _safe_str(value) -> str | None:
|
||||
if value is None:
|
||||
return None
|
||||
if isinstance(value, bool):
|
||||
return "Yes" if value else "No"
|
||||
if isinstance(value, float):
|
||||
# Trim excessive decimals
|
||||
return f"{value:.6g}"
|
||||
if isinstance(value, (int, str)):
|
||||
s = str(value).strip()
|
||||
return s if s else None
|
||||
return str(value).strip() or None
|
||||
s = str(value).strip()
|
||||
return s or None
|
||||
|
||||
|
||||
def _extract_param(entry) -> tuple[str, str] | None:
|
||||
"""
|
||||
Given a Revit parameter entry dict like:
|
||||
{"name": "Thickness", "value": 25.4, "units": "Millimeters", ...}
|
||||
Returns (display_name, display_value) or None if unusable.
|
||||
"""
|
||||
if not isinstance(entry, dict):
|
||||
return None
|
||||
name = entry.get("name")
|
||||
value = entry.get("value")
|
||||
if not name or value is None:
|
||||
return None
|
||||
units = entry.get("units", "")
|
||||
# Skip non-informative unit labels
|
||||
def _flatten_params(params_block) -> dict:
|
||||
"""Flatten Type or Instance parameter block into {name: display_value}.
|
||||
Handles both plain dicts and Speckle Base objects at every nesting level."""
|
||||
result = {}
|
||||
skip_units = {"", "None", "General", "Currency", "Integer"}
|
||||
val_str = _safe_val(value)
|
||||
if val_str is None:
|
||||
return None
|
||||
if units and units not in skip_units:
|
||||
display = f"{val_str} {units}"
|
||||
else:
|
||||
display = val_str
|
||||
return str(name), display
|
||||
|
||||
|
||||
def _flatten_param_group(group: dict) -> dict:
|
||||
"""
|
||||
Flatten one parameter group (e.g. "Dimensions", "Constraints") dict.
|
||||
Each value is a Revit parameter entry {"name":..., "value":..., "units":...}.
|
||||
Returns {display_name: display_value}.
|
||||
"""
|
||||
result = {}
|
||||
if not isinstance(group, dict):
|
||||
return result
|
||||
for _internal_key, entry in group.items():
|
||||
pair = _extract_param(entry)
|
||||
if pair:
|
||||
name, val = pair
|
||||
result[name] = val
|
||||
return result
|
||||
|
||||
|
||||
def _extract_parameter_block(params_block: dict) -> dict:
|
||||
"""
|
||||
Flatten all groups in a parameter block (Type Parameters or Instance Parameters).
|
||||
Returns a merged {display_name: display_value} dict.
|
||||
"""
|
||||
result = {}
|
||||
if not isinstance(params_block, dict):
|
||||
return result
|
||||
for _group_name, group in params_block.items():
|
||||
result.update(_flatten_param_group(group))
|
||||
return result
|
||||
|
||||
|
||||
def _get_properties_dict(obj: Base) -> dict:
|
||||
"""Extract the raw properties dict from a DataObject."""
|
||||
for key in ["properties", "@properties", "_properties"]:
|
||||
try:
|
||||
props = obj[key]
|
||||
if props is None:
|
||||
continue
|
||||
if hasattr(props, "get_dynamic_member_names"):
|
||||
names = props.get_dynamic_member_names()
|
||||
return {n: props[n] for n in names}
|
||||
if isinstance(props, dict):
|
||||
return props
|
||||
except Exception:
|
||||
block = _to_dict(params_block)
|
||||
for group in block.values():
|
||||
group_d = _to_dict(group)
|
||||
if not group_d:
|
||||
continue
|
||||
return {}
|
||||
for entry in group_d.values():
|
||||
entry_d = _to_dict(entry)
|
||||
if not entry_d:
|
||||
continue
|
||||
name = entry_d.get("name")
|
||||
value = entry_d.get("value")
|
||||
units = entry_d.get("units", "") or ""
|
||||
if not name or value is None:
|
||||
continue
|
||||
val_str = _safe_str(value)
|
||||
if val_str is None:
|
||||
continue
|
||||
display = f"{val_str} {units}".strip() if units not in skip_units else val_str
|
||||
result[name] = display
|
||||
return result
|
||||
|
||||
|
||||
def _write_pset(ifc, element, pset_name: str, props: dict):
|
||||
"""Write a property set if there are any properties."""
|
||||
if not props:
|
||||
return
|
||||
try:
|
||||
pset = ifcopenshell.api.run("pset.add_pset", ifc, product=element, name=pset_name)
|
||||
ifcopenshell.api.run("pset.edit_pset", ifc, pset=pset, properties=props)
|
||||
except Exception as e:
|
||||
print(f" ⚠️ {pset_name}: {e}")
|
||||
|
||||
|
||||
def write_properties(ifc, element, obj: Base):
|
||||
def write_revit_params(ifc, element, obj: Base):
|
||||
"""
|
||||
Write Revit parameters as IFC property sets.
|
||||
Creates separate psets for Type and Instance parameters.
|
||||
Write remaining Revit instance parameters as a custom property set
|
||||
using the vendor prefix 'RVT_' (not 'Pset_' which is reserved):
|
||||
RVT_InstanceParameters — from Instance Parameters
|
||||
|
||||
Note: RVT_TypeParameters are written on the IfcTypeObject (via TypeManager),
|
||||
not on individual elements, to avoid duplication.
|
||||
"""
|
||||
props_dict = _get_properties_dict(obj)
|
||||
parameters = props_dict.get("Parameters") or {}
|
||||
props = _get_props_dict(obj)
|
||||
params = _safe_get(props, "Parameters", {})
|
||||
|
||||
# Type Parameters → Pset_RevitTypeParameters
|
||||
type_params = parameters.get("Type Parameters") or {}
|
||||
type_flat = _extract_parameter_block(type_params)
|
||||
_write_pset(ifc, element, "RVT_TypeParameters", type_flat)
|
||||
inst_flat = _flatten_params(_safe_get(params, "Instance Parameters", {}))
|
||||
|
||||
# Instance Parameters → Pset_RevitInstanceParameters
|
||||
inst_params = parameters.get("Instance Parameters") or {}
|
||||
inst_flat = _extract_parameter_block(inst_params)
|
||||
_write_pset(ifc, element, "RVT_InstanceParameters", inst_flat)
|
||||
def build_str_props(flat: dict) -> list:
|
||||
out = []
|
||||
for name, val in flat.items():
|
||||
try:
|
||||
nominal = ifc.create_entity("IfcLabel", wrappedValue=val)
|
||||
p = ifc.create_entity("IfcPropertySingleValue", Name=name, NominalValue=nominal)
|
||||
out.append(p)
|
||||
except Exception:
|
||||
pass
|
||||
return out
|
||||
|
||||
# Top-level semantic fields → Pset_RevitIdentity
|
||||
inst_props = build_str_props(inst_flat)
|
||||
|
||||
if inst_props:
|
||||
_write_pset(ifc, element, "RVT_InstanceParameters", inst_props)
|
||||
|
||||
# Identity: family, type, elementId, builtInCategory
|
||||
identity = {}
|
||||
for field in ["type", "family", "category", "level"]:
|
||||
for field in ["family", "type", "category"]:
|
||||
val = getattr(obj, field, None)
|
||||
if val and isinstance(val, str) and val.strip():
|
||||
identity[field.capitalize()] = val.strip()
|
||||
# Also include elementId if present
|
||||
elem_id = props_dict.get("elementId")
|
||||
elem_id = _safe_get(props, "elementId")
|
||||
if elem_id:
|
||||
identity["ElementId"] = str(elem_id)
|
||||
bic = _safe_get(props, "builtInCategory")
|
||||
if bic:
|
||||
identity["BuiltInCategory"] = str(bic)
|
||||
|
||||
_write_pset(ifc, element, "RVT_Identity", identity)
|
||||
id_props = []
|
||||
for name, val in identity.items():
|
||||
try:
|
||||
nominal = ifc.create_entity("IfcLabel", wrappedValue=val)
|
||||
p = ifc.create_entity("IfcPropertySingleValue", Name=name, NominalValue=nominal)
|
||||
id_props.append(p)
|
||||
except Exception:
|
||||
pass
|
||||
if id_props:
|
||||
_write_pset(ifc, element, "RVT_Identity", id_props)
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Public API — called from main.py
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def write_material_quantities(ifc, element, obj: Base):
|
||||
"""
|
||||
Write Material Quantities from Revit as IfcElementQuantity sets.
|
||||
|
||||
Source: properties."Material Quantities".<MaterialName>.{area, volume, density,
|
||||
materialName, materialClass, materialCategory}
|
||||
|
||||
Each material produces one IfcElementQuantity named "Qto_<MaterialName>BaseQuantities" with:
|
||||
- GrossArea (IfcQuantityArea)
|
||||
- GrossVolume (IfcQuantityVolume)
|
||||
- Density (IfcPropertySingleValue — no standard IFC quantity type)
|
||||
- MaterialClass (IfcPropertySingleValue)
|
||||
- MaterialCategory (IfcPropertySingleValue)
|
||||
"""
|
||||
props = _get_props_dict(obj)
|
||||
mat_quantities = _safe_get(props, "Material Quantities")
|
||||
if mat_quantities is None:
|
||||
return
|
||||
|
||||
mat_dict = _to_dict(mat_quantities)
|
||||
if not mat_dict:
|
||||
return
|
||||
|
||||
for mat_key, mat_data in mat_dict.items():
|
||||
mat_d = _to_dict(mat_data)
|
||||
if not mat_d:
|
||||
continue
|
||||
|
||||
mat_name = mat_d.get("materialName") or mat_key
|
||||
quantities = []
|
||||
|
||||
# Area → IfcQuantityArea
|
||||
area_entry = _to_dict(mat_d.get("area"))
|
||||
if area_entry and area_entry.get("value") is not None:
|
||||
try:
|
||||
q = ifc.create_entity(
|
||||
"IfcQuantityArea",
|
||||
Name="GrossArea",
|
||||
AreaValue=float(area_entry["value"]),
|
||||
)
|
||||
quantities.append(q)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# Volume → IfcQuantityVolume
|
||||
vol_entry = _to_dict(mat_d.get("volume"))
|
||||
if vol_entry and vol_entry.get("value") is not None:
|
||||
try:
|
||||
q = ifc.create_entity(
|
||||
"IfcQuantityVolume",
|
||||
Name="GrossVolume",
|
||||
VolumeValue=float(vol_entry["value"]),
|
||||
)
|
||||
quantities.append(q)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# Density → IfcQuantityWeight (mass per volume, stored as weight)
|
||||
density_entry = _to_dict(mat_d.get("density"))
|
||||
if density_entry and density_entry.get("value") is not None:
|
||||
try:
|
||||
q = ifc.create_entity(
|
||||
"IfcQuantityWeight",
|
||||
Name="Density",
|
||||
WeightValue=float(density_entry["value"]),
|
||||
)
|
||||
quantities.append(q)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
if not quantities:
|
||||
continue
|
||||
|
||||
# Create IfcElementQuantity and link via IfcRelDefinesByProperties
|
||||
qto_name = f"Qto_{mat_name}BaseQuantities"
|
||||
try:
|
||||
qto = ifcopenshell.api.run(
|
||||
"pset.add_qto", ifc,
|
||||
product=element,
|
||||
name=qto_name,
|
||||
)
|
||||
qto.Quantities = quantities
|
||||
except Exception as e:
|
||||
print(f" ⚠️ {qto_name}: {e}")
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Qto_<EntityType>BaseQuantities — standard element-level quantities
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
# IFC entity → Qto name (only entities with standard Qto sets)
|
||||
_ENTITY_QTO_NAME: dict[str, str] = {
|
||||
"IfcWall": "Qto_WallBaseQuantities",
|
||||
"IfcWallStandardCase": "Qto_WallBaseQuantities",
|
||||
"IfcSlab": "Qto_SlabBaseQuantities",
|
||||
"IfcColumn": "Qto_ColumnBaseQuantities",
|
||||
"IfcBeam": "Qto_BeamBaseQuantities",
|
||||
"IfcDoor": "Qto_DoorBaseQuantities",
|
||||
"IfcWindow": "Qto_WindowBaseQuantities",
|
||||
"IfcRoof": "Qto_RoofBaseQuantities",
|
||||
"IfcCovering": "Qto_CoveringBaseQuantities",
|
||||
"IfcRailing": "Qto_RailingBaseQuantities",
|
||||
"IfcStair": "Qto_StairBaseQuantities",
|
||||
"IfcRamp": "Qto_RampBaseQuantities",
|
||||
"IfcMember": "Qto_MemberBaseQuantities",
|
||||
"IfcFooting": "Qto_FootingBaseQuantities",
|
||||
"IfcCurtainWall": "Qto_CurtainWallBaseQuantities",
|
||||
"IfcBuildingElementProxy": "Qto_BuildingElementProxyBaseQuantities",
|
||||
"IfcPipeFitting": "Qto_PipeFittingBaseQuantities",
|
||||
"IfcSanitaryTerminal": "Qto_SanitaryTerminalBaseQuantities",
|
||||
"IfcReinforcingBar": "Qto_ReinforcingElementBaseQuantities",
|
||||
"IfcMechanicalFastener": "Qto_MechanicalFastenerBaseQuantities",
|
||||
}
|
||||
|
||||
# IFC quantity name → (IFC entity type, value attribute, [Revit param fallbacks])
|
||||
# First matching Revit param wins for each quantity name.
|
||||
_ELEMENT_QUANTITY_DEFS: list[tuple[str, str, str, list[str]]] = [
|
||||
("GrossArea", "IfcQuantityArea", "AreaValue", ["HOST_AREA_COMPUTED"]),
|
||||
("GrossVolume", "IfcQuantityVolume", "VolumeValue", ["HOST_VOLUME_COMPUTED"]),
|
||||
("Length", "IfcQuantityLength", "LengthValue", [
|
||||
"CURVE_ELEM_LENGTH", "INSTANCE_LENGTH_PARAM",
|
||||
]),
|
||||
("Height", "IfcQuantityLength", "LengthValue", [
|
||||
"WALL_USER_HEIGHT_PARAM", "FAMILY_HEIGHT_PARAM",
|
||||
"INSTANCE_HEAD_HEIGHT_PARAM",
|
||||
]),
|
||||
("Width", "IfcQuantityLength", "LengthValue", [
|
||||
"INSTANCE_WIDTH_PARAM", "FURNITURE_WIDTH",
|
||||
"FLOOR_ATTR_THICKNESS_PARAM",
|
||||
]),
|
||||
("Perimeter", "IfcQuantityLength", "LengthValue", [
|
||||
"HOST_PERIMETER_COMPUTED",
|
||||
]),
|
||||
]
|
||||
|
||||
|
||||
def write_element_quantities(ifc, element, obj: Base, ifc_class: str = ""):
|
||||
"""
|
||||
Write Qto_<EntityType>BaseQuantities from Revit computed instance parameters.
|
||||
|
||||
Reads HOST_AREA_COMPUTED, HOST_VOLUME_COMPUTED, CURVE_ELEM_LENGTH,
|
||||
FURNITURE_WIDTH, FAMILY_HEIGHT_PARAM, etc.
|
||||
IfcSpace is handled separately in _write_space_properties.
|
||||
"""
|
||||
if ifc_class == "IfcSpace":
|
||||
return # Already handled by Qto_SpaceBaseQuantities
|
||||
|
||||
qto_name = _ENTITY_QTO_NAME.get(ifc_class)
|
||||
if not qto_name:
|
||||
return
|
||||
|
||||
props = _get_props_dict(obj)
|
||||
params = _safe_get(props, "Parameters", {})
|
||||
inst_params = _safe_get(params, "Instance Parameters", {})
|
||||
if not inst_params:
|
||||
return
|
||||
|
||||
quantities = []
|
||||
|
||||
for qty_name, ifc_entity, value_attr, revit_params in _ELEMENT_QUANTITY_DEFS:
|
||||
val = None
|
||||
for internal_name in revit_params:
|
||||
val = _param_value(inst_params, internal_name)
|
||||
if val is not None:
|
||||
break
|
||||
if val is None:
|
||||
continue
|
||||
try:
|
||||
q = ifc.create_entity(ifc_entity, Name=qty_name, **{value_attr: float(val)})
|
||||
quantities.append(q)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
if not quantities:
|
||||
return
|
||||
|
||||
try:
|
||||
qto = ifcopenshell.api.run(
|
||||
"pset.add_qto", ifc,
|
||||
product=element,
|
||||
name=qto_name,
|
||||
)
|
||||
qto.Quantities = quantities
|
||||
except Exception as e:
|
||||
print(f" ⚠️ {qto_name}: {e}")
|
||||
|
||||
|
||||
def write_properties(ifc, element, obj: Base, ifc_class: str = "", category_name: str = ""):
|
||||
"""
|
||||
Write all property sets for an IFC element, matching Revit native IFC export structure:
|
||||
1. Pset_<Entity>Common — standard typed properties (Reference, IsExternal, etc.)
|
||||
2. Pset_EnvironmentalImpactIndicators — Reference = TypeName
|
||||
3. RVT_TypeParameters — all remaining Revit type parameters
|
||||
4. RVT_InstanceParameters — all remaining Revit instance parameters
|
||||
5. RVT_Identity — family, type, elementId, builtInCategory
|
||||
6. Qto_<EntityType>BaseQuantities — element-level quantities (area, volume, length)
|
||||
7. Qto_<MaterialName>BaseQuantities — material quantities (area, volume, density)
|
||||
"""
|
||||
write_common_pset(ifc, element, obj, ifc_class, category_name)
|
||||
write_revit_params(ifc, element, obj)
|
||||
write_element_quantities(ifc, element, obj, ifc_class)
|
||||
write_material_quantities(ifc, element, obj)
|
||||
|
||||
|
||||
def write_common_properties(ifc, element, obj: Base, category_name: str = ""):
|
||||
"""
|
||||
Write Pset_SpeckleData for traceability back to the Speckle source object.
|
||||
"""
|
||||
props = {}
|
||||
speckle_id = getattr(obj, "id", None)
|
||||
app_id = getattr(obj, "applicationId", None)
|
||||
speckle_type = getattr(obj, "speckle_type", None)
|
||||
"""Legacy shim — kept for compatibility with main.py call sites."""
|
||||
pass # All handled by write_properties now
|
||||
|
||||
if speckle_id: props["SpeckleId"] = str(speckle_id)
|
||||
if app_id: props["ApplicationId"] = str(app_id)
|
||||
if speckle_type: props["SpeckleType"] = str(speckle_type)
|
||||
if category_name: props["RevitCategory"] = str(category_name)
|
||||
|
||||
_write_pset(ifc, element, "RVT_SpeckleData", props)
|
||||
def reset_caches():
|
||||
"""Clear module-level caches (call at start of each export run)."""
|
||||
_props_cache.clear()
|
||||
_to_dict_cache.clear()
|
||||
+16
-9
@@ -3,29 +3,36 @@
|
||||
# Connects to Speckle and receives the root Base object for a given version.
|
||||
# =============================================================================
|
||||
|
||||
import os
|
||||
from dotenv import load_dotenv
|
||||
from specklepy.api.client import SpeckleClient
|
||||
from specklepy.api.credentials import get_default_account
|
||||
from specklepy.api import operations
|
||||
from specklepy.transports.server import ServerTransport
|
||||
import utils.config as config
|
||||
|
||||
load_dotenv()
|
||||
|
||||
SPECKLE_HOST = os.getenv("SPECKLE_SERVER_URL", "https://app.speckle.systems")
|
||||
SPECKLE_TOKEN = os.getenv("SPECKLE_TOKEN", "")
|
||||
DEFAULT_UNITS = "mm"
|
||||
|
||||
|
||||
def get_client() -> SpeckleClient:
|
||||
"""
|
||||
Create and authenticate a SpeckleClient.
|
||||
Uses a personal access token from config.py.
|
||||
Uses a personal access token from the .env file.
|
||||
To use your local Speckle Manager account instead, swap to get_default_account().
|
||||
"""
|
||||
client = SpeckleClient(host=config.SPECKLE_HOST)
|
||||
client = SpeckleClient(host=SPECKLE_HOST)
|
||||
|
||||
if config.SPECKLE_TOKEN and config.SPECKLE_TOKEN != "YOUR_PERSONAL_ACCESS_TOKEN":
|
||||
client.authenticate_with_token(config.SPECKLE_TOKEN)
|
||||
if SPECKLE_TOKEN and SPECKLE_TOKEN != "YOUR_PERSONAL_ACCESS_TOKEN":
|
||||
client.authenticate_with_token(SPECKLE_TOKEN)
|
||||
else:
|
||||
# Fallback: use account from Speckle Manager desktop app
|
||||
account = get_default_account()
|
||||
if account is None:
|
||||
raise RuntimeError(
|
||||
"No Speckle account found. Either set SPECKLE_TOKEN in config.py "
|
||||
"No Speckle account found. Either set SPECKLE_TOKEN in .env "
|
||||
"or log in via Speckle Manager."
|
||||
)
|
||||
client.authenticate_with_account(account)
|
||||
@@ -46,11 +53,11 @@ def receive_version(project_id: str, version_id: str):
|
||||
"""
|
||||
client = get_client()
|
||||
|
||||
print(f"🔗 Connecting to {config.SPECKLE_HOST}...")
|
||||
print(f"🔗 Connecting to {SPECKLE_HOST}...")
|
||||
print(f"📦 Receiving project={project_id} version={version_id}")
|
||||
|
||||
# Get version metadata to find the referenced object ID
|
||||
version = client.version.get(version_id,project_id)
|
||||
version = client.version.get(version_id, project_id)
|
||||
referenced_object_id = version.referenced_object
|
||||
|
||||
# Download the full object graph
|
||||
@@ -58,7 +65,7 @@ def receive_version(project_id: str, version_id: str):
|
||||
base = operations.receive(referenced_object_id, transport)
|
||||
|
||||
# Read units from the root object
|
||||
units = getattr(base, "units", config.DEFAULT_UNITS) or config.DEFAULT_UNITS
|
||||
units = getattr(base, "units", DEFAULT_UNITS) or DEFAULT_UNITS
|
||||
|
||||
# IFC file is declared in MILLIMETRES — no conversion needed.
|
||||
# All geometry stays in source units (mm). scale=1.0 means "keep as-is".
|
||||
|
||||
+6
-8
@@ -55,10 +55,10 @@ def get_prop(obj, key: str, default=None):
|
||||
|
||||
|
||||
# speckle_type fragments that mark a non-exportable / spatial-structure object
|
||||
_SKIP_TYPE_FRAGMENTS = {
|
||||
"Collection", "Level", "Grid", "View", "RenderMaterial",
|
||||
"Site", "Building", "Storey",
|
||||
}
|
||||
import re
|
||||
_SKIP_TYPE_RE = re.compile(
|
||||
r"Collection|Level|Grid|View|RenderMaterial|Site|Building|Storey"
|
||||
)
|
||||
|
||||
|
||||
def _is_valid_element(obj) -> bool:
|
||||
@@ -70,10 +70,8 @@ def _is_valid_element(obj) -> bool:
|
||||
return False
|
||||
|
||||
speckle_type = getattr(obj, "speckle_type", "") or ""
|
||||
|
||||
for fragment in _SKIP_TYPE_FRAGMENTS:
|
||||
if fragment in speckle_type:
|
||||
return False
|
||||
if _SKIP_TYPE_RE.search(speckle_type):
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
|
||||
@@ -0,0 +1,214 @@
|
||||
# =============================================================================
|
||||
# type_manager.py
|
||||
# Creates and caches IfcTypeObjects (IfcWallType, IfcRoofType, etc.) and
|
||||
# links element instances to them via IfcRelDefinesByType.
|
||||
#
|
||||
# Revit native IFC export pattern:
|
||||
# IfcWallType
|
||||
# Name = "Family:TypeName" (no ElementId)
|
||||
# Tag = Type's Revit ElementId (from Instance Parameters > Other > Type Id)
|
||||
# GlobalId = from Type IfcGUID param (from Type Parameters > IFC Parameters > Type IfcGUID)
|
||||
# HasPropertySets:
|
||||
# Pset_WallCommon: IsExternal, ThermalTransmittance (type-level)
|
||||
# Pset_EnvironmentalImpactIndicators: Reference = TypeName
|
||||
# RVT_TypeParameters: all remaining type params
|
||||
#
|
||||
# Type objects are SHARED — multiple instances of the same Revit type
|
||||
# map to one IfcTypeObject, keyed by (ifc_class, family, type_name).
|
||||
# =============================================================================
|
||||
|
||||
import ifcopenshell
|
||||
import ifcopenshell.api
|
||||
from specklepy.objects.base import Base
|
||||
from utils.properties import (
|
||||
_get_props_dict, _get_nested, _param_value, _make_prop, _write_pset,
|
||||
_safe_get, _to_dict,
|
||||
COMMON_PSET, EXTERNAL_CATEGORIES, _flatten_params
|
||||
)
|
||||
|
||||
|
||||
# IFC element class → IFC type class
|
||||
TYPE_CLASS_MAP: dict[str, str] = {
|
||||
"IfcWall": "IfcWallType",
|
||||
"IfcWallStandardCase": "IfcWallType",
|
||||
"IfcSlab": "IfcSlabType",
|
||||
"IfcRoof": "IfcRoofType",
|
||||
"IfcColumn": "IfcColumnType",
|
||||
"IfcBeam": "IfcBeamType",
|
||||
"IfcMember": "IfcMemberType",
|
||||
"IfcDoor": "IfcDoorType",
|
||||
"IfcWindow": "IfcWindowType",
|
||||
"IfcStair": "IfcStairType",
|
||||
"IfcStairFlight": "IfcStairFlightType",
|
||||
"IfcRamp": "IfcRampType",
|
||||
"IfcRailing": "IfcRailingType",
|
||||
"IfcCovering": "IfcCoveringType",
|
||||
"IfcCurtainWall": "IfcCurtainWallType",
|
||||
"IfcFooting": "IfcFootingType",
|
||||
"IfcBuildingElementProxy": "IfcBuildingElementProxyType",
|
||||
"IfcFurnishingElement": "IfcFurnitureType",
|
||||
"IfcLightFixture": "IfcLightFixtureType",
|
||||
"IfcElectricAppliance": "IfcElectricApplianceType",
|
||||
"IfcElectricDistributionBoard": "IfcElectricDistributionBoardType",
|
||||
"IfcSanitaryTerminal": "IfcSanitaryTerminalType",
|
||||
"IfcUnitaryEquipment": "IfcUnitaryEquipmentType",
|
||||
"IfcDuctSegment": "IfcDuctSegmentType",
|
||||
"IfcPipeSegment": "IfcPipeSegmentType",
|
||||
"IfcCableCarrierSegment": "IfcCableCarrierSegmentType",
|
||||
"IfcPlate": "IfcPlateType",
|
||||
}
|
||||
|
||||
|
||||
class TypeManager:
|
||||
"""
|
||||
Creates IfcTypeObjects on demand and caches them by (ifc_class, family, type_name).
|
||||
Call assign(element, obj, ifc_class) for each exported element.
|
||||
"""
|
||||
|
||||
def __init__(self, ifc: ifcopenshell.file):
|
||||
self._ifc = ifc
|
||||
# key: (ifc_class, family, type_name) → IfcTypeObject
|
||||
self._cache: dict[tuple, object] = {}
|
||||
# type_object → [element, ...] (for batched IfcRelDefinesByType)
|
||||
self._pending: dict[int, list] = {}
|
||||
|
||||
def assign(self, element, obj: Base, ifc_class: str):
|
||||
"""Create (or retrieve cached) type object and queue the assignment."""
|
||||
type_class = TYPE_CLASS_MAP.get(ifc_class)
|
||||
if not type_class:
|
||||
return
|
||||
|
||||
family = getattr(obj, "family", None) or ""
|
||||
type_name = getattr(obj, "type", None) or ""
|
||||
if not type_name:
|
||||
return
|
||||
|
||||
cache_key = (ifc_class, family, type_name)
|
||||
|
||||
if cache_key not in self._cache:
|
||||
type_obj = self._create_type(type_class, family, type_name, obj, ifc_class)
|
||||
self._cache[cache_key] = type_obj
|
||||
|
||||
type_obj = self._cache[cache_key]
|
||||
type_id = type_obj.id()
|
||||
|
||||
if type_id not in self._pending:
|
||||
self._pending[type_id] = []
|
||||
self._pending[type_id].append(element)
|
||||
|
||||
def flush(self):
|
||||
"""Write all IfcRelDefinesByType relationships."""
|
||||
for type_id, elements in self._pending.items():
|
||||
type_obj = self._ifc.by_id(type_id)
|
||||
ifcopenshell.api.run(
|
||||
"type.assign_type", self._ifc,
|
||||
related_objects=elements,
|
||||
relating_type=type_obj,
|
||||
)
|
||||
self._pending.clear()
|
||||
print(f" Type objects created: {len(self._cache)}")
|
||||
|
||||
# -----------------------------------------------------------------------
|
||||
def _create_type(self, type_class: str, family: str, type_name: str,
|
||||
obj: Base, ifc_class: str):
|
||||
"""Instantiate the IfcTypeObject with name, tag, GlobalId, and psets."""
|
||||
props = _get_props_dict(obj)
|
||||
params = _safe_get(props, "Parameters", {})
|
||||
type_params = _safe_get(params, "Type Parameters", {})
|
||||
inst_params = _safe_get(params, "Instance Parameters", {})
|
||||
|
||||
# Name: "Family:TypeName" (no ElementId)
|
||||
name_parts = [p for p in [family, type_name] if p]
|
||||
name = ":".join(name_parts)
|
||||
|
||||
# Tag: Type's Revit ElementId
|
||||
type_id_entry = _get_nested(inst_params, "Other", "Type Id")
|
||||
type_id_d = _to_dict(type_id_entry)
|
||||
tag = str(type_id_d.get("value")) if type_id_d.get("value") else None
|
||||
|
||||
# GlobalId: from Type IfcGUID parameter
|
||||
type_guid_entry = _get_nested(type_params, "IFC Parameters", "Type IfcGUID")
|
||||
type_guid_d = _to_dict(type_guid_entry)
|
||||
guid = type_guid_d.get("value") if type_guid_d else None
|
||||
|
||||
# Create type entity
|
||||
type_obj = ifcopenshell.api.run(
|
||||
"root.create_entity", self._ifc,
|
||||
ifc_class=type_class,
|
||||
name=name,
|
||||
)
|
||||
if tag:
|
||||
try:
|
||||
type_obj.Tag = str(tag)
|
||||
except Exception:
|
||||
pass
|
||||
if guid:
|
||||
try:
|
||||
type_obj.GlobalId = str(guid)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# Write type-level property sets
|
||||
self._write_type_psets(type_obj, obj, ifc_class, type_name, props,
|
||||
type_params, inst_params)
|
||||
return type_obj
|
||||
|
||||
def _write_type_psets(self, type_obj, obj, ifc_class, type_name,
|
||||
props, type_params, inst_params):
|
||||
"""Write psets on the type object (type-level parameters only)."""
|
||||
ifc = self._ifc
|
||||
pset_name = COMMON_PSET.get(ifc_class)
|
||||
|
||||
# ── Standard Common pset on the type ──────────────────────────────
|
||||
if pset_name:
|
||||
type_ifc_props = []
|
||||
|
||||
# IsExternal (type-level)
|
||||
bic = _safe_get(props, "builtInCategory", "")
|
||||
is_external = bic in EXTERNAL_CATEGORIES
|
||||
if ifc_class not in {"IfcSpace", "IfcSite", "IfcBuildingStorey",
|
||||
"IfcBuilding", "IfcFurnishingElement", "IfcOpeningElement"}:
|
||||
p = _make_prop(ifc, "IsExternal", "IfcBoolean", is_external)
|
||||
if p:
|
||||
type_ifc_props.append(p)
|
||||
|
||||
# ThermalTransmittance (from type parameters)
|
||||
if ifc_class in {"IfcWall", "IfcWallStandardCase", "IfcRoof",
|
||||
"IfcSlab", "IfcDoor", "IfcWindow"}:
|
||||
u_val = _param_value(type_params,
|
||||
"ANALYTICAL_HEAT_TRANSFER_COEFFICIENT")
|
||||
if u_val is not None:
|
||||
try:
|
||||
p = _make_prop(ifc, "ThermalTransmittance",
|
||||
"IfcThermalTransmittanceMeasure", float(u_val))
|
||||
if p:
|
||||
type_ifc_props.append(p)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# LoadBearing (from type parameters)
|
||||
if ifc_class in {"IfcWall", "IfcWallStandardCase", "IfcColumn",
|
||||
"IfcBeam", "IfcSlab"}:
|
||||
lb_val = _param_value(type_params, "WALL_STRUCTURAL_SIGNIFICANT")
|
||||
if lb_val is not None:
|
||||
p = _make_prop(ifc, "LoadBearing", "IfcBoolean", bool(lb_val))
|
||||
if p:
|
||||
type_ifc_props.append(p)
|
||||
|
||||
if type_ifc_props:
|
||||
_write_pset(ifc, type_obj, pset_name, type_ifc_props)
|
||||
|
||||
# ── RVT_TypeParameters — all type-level Revit params ──────────────
|
||||
type_flat = _flatten_params(type_params)
|
||||
if type_flat:
|
||||
type_str_props = []
|
||||
for name_p, val in type_flat.items():
|
||||
try:
|
||||
nominal = ifc.create_entity("IfcLabel", wrappedValue=val)
|
||||
prop = ifc.create_entity("IfcPropertySingleValue",
|
||||
Name=name_p, NominalValue=nominal)
|
||||
type_str_props.append(prop)
|
||||
except Exception:
|
||||
pass
|
||||
if type_str_props:
|
||||
_write_pset(ifc, type_obj, "RVT_TypeParameters", type_str_props)
|
||||
+53
-8
@@ -9,16 +9,20 @@
|
||||
|
||||
import ifcopenshell
|
||||
import ifcopenshell.api
|
||||
import utils.config as config
|
||||
|
||||
|
||||
def create_ifc_scaffold() -> tuple:
|
||||
def create_ifc_scaffold(
|
||||
project_name: str = "Default Project",
|
||||
site_name: str = "Default Site",
|
||||
building_name: str = "Default Building",
|
||||
) -> tuple:
|
||||
"""
|
||||
Create the IFC file with the required project/site/building hierarchy.
|
||||
|
||||
Returns:
|
||||
(ifc_file, building, body_context)
|
||||
(ifc_file, site, building, body_context)
|
||||
- ifc_file: The ifcopenshell file object
|
||||
- site: The IfcSite entity
|
||||
- building: The IfcBuilding entity (storeys are assigned under this)
|
||||
- body_context: The Body geometry subcontext for shape representations
|
||||
"""
|
||||
@@ -28,7 +32,7 @@ def create_ifc_scaffold() -> tuple:
|
||||
project = ifcopenshell.api.run(
|
||||
"root.create_entity", ifc,
|
||||
ifc_class="IfcProject",
|
||||
name=config.IFC_PROJECT_NAME,
|
||||
name=project_name,
|
||||
)
|
||||
|
||||
# Units — millimetres (matching Revit/Speckle source data)
|
||||
@@ -55,12 +59,12 @@ def create_ifc_scaffold() -> tuple:
|
||||
site = ifcopenshell.api.run(
|
||||
"root.create_entity", ifc,
|
||||
ifc_class="IfcSite",
|
||||
name=config.IFC_SITE_NAME,
|
||||
name=site_name,
|
||||
)
|
||||
building = ifcopenshell.api.run(
|
||||
"root.create_entity", ifc,
|
||||
ifc_class="IfcBuilding",
|
||||
name=config.IFC_BUILDING_NAME,
|
||||
name=building_name,
|
||||
)
|
||||
|
||||
ifcopenshell.api.run(
|
||||
@@ -74,19 +78,26 @@ def create_ifc_scaffold() -> tuple:
|
||||
products=[building],
|
||||
)
|
||||
|
||||
return ifc, building, body_ctx
|
||||
return ifc, site, building, body_ctx
|
||||
|
||||
|
||||
class StoreyManager:
|
||||
"""
|
||||
Lazily creates IfcBuildingStorey entities as new level names are encountered.
|
||||
Keeps storeys in insertion order so the IFC file is logically ordered.
|
||||
|
||||
Spatial containment is batched — call flush() after all elements are created
|
||||
to write all IfcRelContainedInSpatialStructure / aggregate relationships at once.
|
||||
"""
|
||||
|
||||
def __init__(self, ifc: ifcopenshell.file, building):
|
||||
self.ifc = ifc
|
||||
self.building = building
|
||||
self._storeys: dict[str, object] = {} # level_name → IfcBuildingStorey
|
||||
# Batched containment: storey_id → [element, ...]
|
||||
self._contained: dict[int, list] = {}
|
||||
# Batched aggregation (IfcSite etc.): storey_id → [element, ...]
|
||||
self._aggregated: dict[int, list] = {}
|
||||
|
||||
def get_or_create(self, level_name: str):
|
||||
"""Return existing storey or create a new one for this level name."""
|
||||
@@ -106,10 +117,44 @@ class StoreyManager:
|
||||
|
||||
return self._storeys[level_name]
|
||||
|
||||
def queue_contain(self, storey, element):
|
||||
"""Queue an element for spatial containment (batched flush)."""
|
||||
sid = storey.id()
|
||||
if sid not in self._contained:
|
||||
self._contained[sid] = []
|
||||
self._contained[sid].append(element)
|
||||
|
||||
def queue_aggregate(self, storey, element):
|
||||
"""Queue an element for aggregation under storey (e.g. IfcSite)."""
|
||||
sid = storey.id()
|
||||
if sid not in self._aggregated:
|
||||
self._aggregated[sid] = []
|
||||
self._aggregated[sid].append(element)
|
||||
|
||||
def flush(self):
|
||||
"""Write all batched spatial containment and aggregation relationships."""
|
||||
ifc = self.ifc
|
||||
for sid, elements in self._contained.items():
|
||||
storey = ifc.by_id(sid)
|
||||
ifcopenshell.api.run(
|
||||
"spatial.assign_container", ifc,
|
||||
relating_structure=storey,
|
||||
products=elements,
|
||||
)
|
||||
for sid, elements in self._aggregated.items():
|
||||
storey = ifc.by_id(sid)
|
||||
ifcopenshell.api.run(
|
||||
"aggregate.assign_object", ifc,
|
||||
relating_object=storey,
|
||||
products=elements,
|
||||
)
|
||||
self._contained.clear()
|
||||
self._aggregated.clear()
|
||||
|
||||
@property
|
||||
def count(self) -> int:
|
||||
return len(self._storeys)
|
||||
|
||||
@property
|
||||
def names(self) -> list[str]:
|
||||
return list(self._storeys.keys())
|
||||
return list(self._storeys.keys())
|
||||
Reference in New Issue
Block a user