mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-10 09:48:32 +00:00
Andrej730
8b6235f42f
TessellateElements patch was failing because it was relying on optionality of `edges` argument. Though argument itself is currently useless and not used in the code.
108 lines
4.5 KiB
Python
108 lines
4.5 KiB
Python
# IfcOpenShell - IFC toolkit and geometry engine
|
|
# Copyright (C) 2022 Dion Moult <dion@thinkmoult.com>
|
|
#
|
|
# This file is part of IfcOpenShell.
|
|
#
|
|
# IfcOpenShell is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Lesser General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# IfcOpenShell is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public License
|
|
# along with IfcOpenShell. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import ifcopenshell.util.unit
|
|
from typing import Optional
|
|
|
|
COORD_3D = tuple[float, float, float]
|
|
|
|
|
|
def add_mesh_representation(
|
|
file: ifcopenshell.file,
|
|
# IfcGeometricRepresentationContext
|
|
context: ifcopenshell.entity_instance,
|
|
# Vertices, edges, and faces are given in the form of: [item1, item2, item3, ...]
|
|
# A list of coordinates
|
|
# ... where itemN = [(0., 0., 0.), (1., 1., 1.), (x, y, z), ...]
|
|
vertices: list[COORD_3D],
|
|
# A list of edges, represented by vertex index pairs
|
|
# ... where itemN = [(0, 1), (1, 2), (v1, v2), ...]
|
|
edges: list[tuple[int, int]] = None,
|
|
# A list of polygons, represented by vertex indices
|
|
# ... where itemN = [(0, 1, 2), (5, 4, 2, 3), (v1, v2, v3, ... vN), ...]
|
|
faces: list[list[int]] = None,
|
|
# Optionally apply a vector offset to all coordinates
|
|
cooridnate_offset: Optional[COORD_3D] = None,
|
|
# A scale factor to apply for all vectors in case the unit is different
|
|
unit_scale: Optional[float] = None,
|
|
# Force using IfcFacetedBreps instead of IfcPolygonalFaceSets
|
|
force_faceted_brep: bool = False,
|
|
) -> ifcopenshell.entity_instance:
|
|
# TODO: Support edges without faces.
|
|
assert faces is not None, f"Currently 'faces' argument is not optional."
|
|
usecase = Usecase()
|
|
usecase.file = file
|
|
usecase.settings = {
|
|
"context": context,
|
|
"vertices": vertices,
|
|
"edges": edges,
|
|
"faces": faces,
|
|
"coordinate_offset": cooridnate_offset,
|
|
"unit_scale": unit_scale,
|
|
"force_faceted_brep": force_faceted_brep,
|
|
}
|
|
return usecase.execute()
|
|
|
|
|
|
class Usecase:
|
|
def execute(self):
|
|
if self.settings["unit_scale"] is None:
|
|
self.settings["unit_scale"] = ifcopenshell.util.unit.calculate_unit_scale(self.file)
|
|
return self.create_mesh_representation()
|
|
|
|
def create_mesh_representation(self):
|
|
if self.settings["force_faceted_brep"] or self.file.schema == "IFC2X3":
|
|
return self.create_faceted_brep()
|
|
return self.create_polygonal_face_set()
|
|
|
|
def create_faceted_brep(self):
|
|
items = []
|
|
for i in range(0, len(self.settings["vertices"])):
|
|
vertices = [
|
|
self.file.createIfcCartesianPoint(self.convert_si_to_unit(v)) for v in self.settings["vertices"][i]
|
|
]
|
|
faces = [
|
|
self.file.createIfcFace(
|
|
[self.file.createIfcFaceOuterBound(self.file.createIfcPolyLoop([vertices[v] for v in f]), True)]
|
|
)
|
|
for f in self.settings["faces"][i]
|
|
]
|
|
items.append(self.file.createIfcFacetedBrep(self.file.createIfcClosedShell(faces)))
|
|
return self.file.createIfcShapeRepresentation(
|
|
self.settings["context"], self.settings["context"].ContextIdentifier, "Brep", items
|
|
)
|
|
|
|
def create_polygonal_face_set(self):
|
|
items = []
|
|
for i in range(0, len(self.settings["vertices"])):
|
|
coordinates = self.file.createIfcCartesianPointList3D(
|
|
[self.convert_si_to_unit(v) for v in self.settings["vertices"][i]]
|
|
)
|
|
faces = [self.file.createIfcIndexedPolygonalFace([v + 1 for v in f]) for f in self.settings["faces"][i]]
|
|
items.append(self.file.createIfcPolygonalFaceSet(coordinates, None, faces))
|
|
return self.file.createIfcShapeRepresentation(
|
|
self.settings["context"], self.settings["context"].ContextIdentifier, "Tessellation", items
|
|
)
|
|
|
|
def convert_si_to_unit(self, co):
|
|
if isinstance(co, (tuple, list)):
|
|
return [self.convert_si_to_unit(o) for o in co]
|
|
if self.settings["coordinate_offset"]:
|
|
return (co / self.settings["unit_scale"]) + self.settings["coordinate_offset"]
|
|
return co / self.settings["unit_scale"]
|