mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-10 09:48:32 +00:00
Thomas Krijnen
86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
import collections
|
|
|
|
import numpy as np
|
|
import ifcopenshell
|
|
import ifcopenshell.geom
|
|
|
|
import pytest
|
|
|
|
from test.bootstrap import file
|
|
|
|
|
|
@pytest.mark.parametrize("file", ["geom/polygonal-face-tessellation.ifc"], indirect=True)
|
|
def test_polyhedral_setting(file):
|
|
s = ifcopenshell.geom.settings()
|
|
# default is triangle mesh
|
|
assert s.get("triangulation-type") == ifcopenshell.ifcopenshell_wrapper.TRIANGLE_MESH
|
|
s = ifcopenshell.geom.settings(TRIANGULATION_TYPE=ifcopenshell.ifcopenshell_wrapper.POLYHEDRON_WITHOUT_HOLES)
|
|
obj = ifcopenshell.geom.create_shape(s, file[30])
|
|
# outer:
|
|
# - top
|
|
# - left
|
|
# - bottom
|
|
# - right
|
|
# - back;
|
|
# inner:
|
|
# - top
|
|
# - left
|
|
# - bottom
|
|
# - right
|
|
# - back;
|
|
# front:
|
|
# - 8 triangles
|
|
# --------------------- +
|
|
# 18 facets
|
|
assert len(obj.geometry.faces) == 18
|
|
s = ifcopenshell.geom.settings(TRIANGULATION_TYPE=ifcopenshell.ifcopenshell_wrapper.POLYHEDRON_WITH_HOLES)
|
|
obj = ifcopenshell.geom.create_shape(s, file[30])
|
|
# outer:
|
|
# - top
|
|
# - left
|
|
# - bottom
|
|
# - right
|
|
# - back;
|
|
# inner:
|
|
# - top
|
|
# - left
|
|
# - bottom
|
|
# - right
|
|
# - back;
|
|
# front:
|
|
# - single facet with 2 bounds
|
|
# --------------------- +
|
|
# 11 facets
|
|
assert len(obj.geometry.faces) == 11
|
|
# 10 facets with 1 bound, 1 facet with 2 bounds
|
|
assert sorted(collections.Counter(map(len, obj.geometry.faces)).items()) == [(1, 10), (2, 1)]
|
|
|
|
|
|
def get_edges(inst):
|
|
coords = inst.Coordinates.CoordList
|
|
for f in inst.Faces:
|
|
|
|
def emit(loop):
|
|
fcoords = list(map(lambda i: coords[i - 1], loop))
|
|
shifted = fcoords[1:] + [fcoords[0]]
|
|
return map(frozenset, zip(fcoords, shifted))
|
|
|
|
yield from emit(f.CoordIndex)
|
|
|
|
if f.is_a("IfcIndexedPolygonalFaceWithVoids"):
|
|
for inner in f.InnerCoordIndices:
|
|
yield from emit(inner)
|
|
|
|
|
|
@pytest.mark.parametrize("file", ["geom/polygonal-face-tessellation.ifc"], indirect=True)
|
|
@pytest.mark.parametrize("geometry_library", ["opencascade", "cgal", "cgal-simple"])
|
|
def test_correct_edges(file, geometry_library):
|
|
s = ifcopenshell.geom.settings()
|
|
obj = ifcopenshell.geom.create_shape(s, file[30], geometry_library=geometry_library)
|
|
vs = np.array(obj.geometry.verts).reshape((-1, 3))
|
|
eds = np.array(obj.geometry.edges).reshape((-1, 2))
|
|
edge_collection_1 = set(map(lambda t: frozenset(map(tuple, t)), (vs[eds] * 1000).tolist()))
|
|
rep = file[30].Representation.Representations[0].Items[0]
|
|
edge_collection_2 = set(get_edges(rep))
|
|
assert edge_collection_1 == edge_collection_2
|