mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-09-12 22:43:41 +00:00
admin
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
from behave import step
|
|
|
|
import geometric_detail_methods as gdm
|
|
from utils import assert_elements
|
|
from utils import IfcFile
|
|
from utils import switch_locale
|
|
|
|
|
|
the_lang = "en"
|
|
|
|
|
|
@step("All elements must be under {number} polygons")
|
|
def step_impl(context, number):
|
|
number = int(number)
|
|
errors = []
|
|
for element in IfcFile.get().by_type("IfcElement"):
|
|
if not element.Representation:
|
|
continue
|
|
total_polygons = 0
|
|
tree = IfcFile.get().traverse(element.Representation)
|
|
for e in tree:
|
|
if e.is_a("IfcFace"):
|
|
total_polygons += 1
|
|
elif e.is_a("IfcPolygonalFaceSet"):
|
|
total_polygons += len(e.Faces)
|
|
elif e.is_a("IfcTriangulatedFaceSet"):
|
|
total_polygons += len(e.CoordIndex)
|
|
if total_polygons > number:
|
|
errors.append((total_polygons, element))
|
|
if errors:
|
|
message = (
|
|
"The following {} elements are over 500 polygons:\n"
|
|
.format(len(errors))
|
|
)
|
|
for error in errors:
|
|
message += "Polygons: {} - {}\n".format(error[0], error[1])
|
|
assert False, message
|
|
|
|
|
|
@step("all {ifc_class} elements have an {representation_class} representation")
|
|
def step_impl(context, ifc_class, representation_class):
|
|
switch_locale(context.localedir, the_lang)
|
|
gdm.eleclass_has_geometric_representation_of_specific_class(
|
|
context,
|
|
ifc_class,
|
|
representation_class
|
|
)
|