Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-09-23 21:38:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

pset.get_template use fallback schema for odd schemas #6672

Browse Source
This commit is contained in:
Andrej730
2025-08-11 18:06:29 +05:00
parent e54dce5dcf
commit 4423e0bb10
1 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/ifcopenshell/util/pset.py
+14 -9
View File
@@ -26,14 +26,15 @@ from ifcopenshell.entity_instance import entity_instance
from functools import lru_cache from functools import lru_cache
from typing import Optional, Literal, NamedTuple, Union from typing import Optional, Literal, NamedTuple, Union
templates: dict[str, "PsetQto"] = {} templates: dict[ifcopenshell.util.schema.IFC_SCHEMA, "PsetQto"] = {}
def get_template(schema: str) -> "PsetQto": def get_template(schema_identiier: str) -> "PsetQto":
""" """
:param schema: As in ``file.schema_identifier``, not ``file.schema``. :param schema_identiier: As in ``file.schema_identifier``, not ``file.schema``.
""" """
global templates global templates
schema = ifcopenshell.util.schema.get_fallback_schema(schema_identiier)
if schema not in templates: if schema not in templates:
templates[schema] = PsetQto(schema) templates[schema] = PsetQto(schema)
return templates[schema] return templates[schema]
@@ -41,24 +42,28 @@ def get_template(schema: str) -> "PsetQto":
class PsetQto: class PsetQto:
# fmt: off # fmt: off
templates_path = { templates_path: dict[ifcopenshell.util.schema.IFC_SCHEMA, str] = {
"IFC2X3": "Pset_IFC2X3.ifc", "IFC2X3": "Pset_IFC2X3.ifc",
"IFC4": "Pset_IFC4_ADD2.ifc", "IFC4": "Pset_IFC4_ADD2.ifc",
"IFC4X3_ADD2": "Pset_IFC4X3.ifc" "IFC4X3": "Pset_IFC4X3.ifc"
} }
# fmt: on # fmt: on
templates: list[ifcopenshell.file] templates: list[ifcopenshell.file]
def __init__(self, schema_identifier: str, templates: Optional[list[ifcopenshell.file]] = None) -> None: def __init__(
self.schema = ifcopenshell.schema_by_name(schema_identifier) self,
schema: ifcopenshell.util.schema.IFC_SCHEMA,
templates: Optional[list[ifcopenshell.file]] = None,
) -> None:
self.schema = ifcopenshell.schema_by_name(schema)
if not templates: if not templates:
folder_path = pathlib.Path(__file__).parent.absolute() folder_path = pathlib.Path(__file__).parent.absolute()
path = str(folder_path.joinpath("schema", self.templates_path[schema_identifier])) path = str(folder_path.joinpath("schema", self.templates_path[schema]))
ifc_file: ifcopenshell.file = ifcopenshell.open(path) ifc_file: ifcopenshell.file = ifcopenshell.open(path)
templates = [ifc_file] templates = [ifc_file]
# See bug 3583. We backport this change from IFC4X3 because it just makes sense. # See bug 3583. We backport this change from IFC4X3 because it just makes sense.
# Users aren't forced to use it. # Users aren't forced to use it.
if schema_identifier == "IFC4": if schema == "IFC4":
for element in templates[0].by_type("IfcPropertySetTemplate"): for element in templates[0].by_type("IfcPropertySetTemplate"):
if element.TemplateType == "QTO_OCCURRENCEDRIVEN": if element.TemplateType == "QTO_OCCURRENCEDRIVEN":
element.TemplateType = "QTO_TYPEDRIVENOVERRIDE" element.TemplateType = "QTO_TYPEDRIVENOVERRIDE"