profile.add_parameterized_profile - add profile_type arg

To create more valid profiles by default. Noticed then Bonsai creates new profiles, it doesn't set ProfileType, leading to validation errors.
This commit is contained in:
Andrej730
2025-07-17 14:55:46 +05:00
parent 0b48120a66
commit 2edfe1e5d8
4 changed files with 51 additions and 3 deletions
+1
View File
@@ -223,6 +223,7 @@ class Profile(bonsai.core.tool.Profile):
"""Set default profile attributes to keep profile valid."""
class_match = False
si_conversion = ifcopenshell.util.unit.calculate_unit_scale(tool.Ifc.get())
# We're using for-loop as classes may not match exactly.
for ifc_class, params in cls.DEFAULT_PROFILE_ATTRS.items():
if profile.is_a(ifc_class):
class_match = True
@@ -16,9 +16,15 @@
# 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
from typing import Literal
def add_parameterized_profile(file: ifcopenshell.file, ifc_class: str) -> ifcopenshell.entity_instance:
ProfileType = Literal["AREA", "CURVE"]
def add_parameterized_profile(
file: ifcopenshell.file, ifc_class: str, profile_type: str = "AREA"
) -> ifcopenshell.entity_instance:
"""Adds a new parameterised profile
IFC offers parameterised profiles for common standardised hot roll
@@ -30,6 +36,7 @@ def add_parameterized_profile(file: ifcopenshell.file, ifc_class: str) -> ifcope
:param ifc_class: The subclass of IfcParameterizedProfileDef that you'd
like to create.
:param profile_type:
:return: The newly created element depending on the specified ifc_class.
Example:
@@ -40,4 +47,4 @@ def add_parameterized_profile(file: ifcopenshell.file, ifc_class: str) -> ifcope
ifc_class="IfcCircleProfileDef")
circle.Radius = 1.
"""
return file.create_entity(ifc_class)
return file.create_entity(ifc_class, ProfileType=profile_type)
@@ -118,7 +118,7 @@ def get_supertypes(
def get_subtypes(
declaration: ifcopenshell.ifcopenshell_wrapper.entity,
) -> list[ifcopenshell.ifcopenshell_wrapper.entity]:
"""Get a flat list of subtype declarations
"""Get a flat list of subtype declarations, recursively.
Abstract classes are skipped.
@@ -0,0 +1,40 @@
# 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 test.bootstrap
import ifcopenshell.api.profile
import ifcopenshell.util.schema
class TestAddParametrizedProfileIFC2X3(test.bootstrap.IFC2X3):
def test_run(self):
schema = ifcopenshell.schema_by_name(self.file.schema)
entity = schema.declaration_by_name("IfcParameterizedProfileDef").as_entity()
assert entity
for s in ifcopenshell.util.schema.get_subtypes(entity):
profile = ifcopenshell.api.profile.add_parameterized_profile(self.file, ifc_class=s.name())
assert profile.is_a() == s.name()
assert profile.ProfileType == "AREA"
class TestAddParametrizedProfileIFC4(test.bootstrap.IFC4, TestAddParametrizedProfileIFC2X3):
pass
class TestAddParametrizedProfileIFC4X3(test.bootstrap.IFC4, TestAddParametrizedProfileIFC2X3):
pass