# IfcOpenShell - IFC toolkit and geometry engine # Copyright (C) 2021 Thomas Krijnen # # 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 . import operator import itertools import codegen import templates import documentation from collections import defaultdict USE_VIRTUAL_INHERITANCE = True class Header(codegen.Base): def __init__(self, mapping): declarations = [] case_lookup = lambda nm: [k for k in mapping.schema.keys if k.lower() == nm.lower()][0] case_normalize = lambda nm: nm if nm.startswith("IfcUtil::") else case_lookup(nm) create_supertype_statement = lambda nms: ", ".join( "public %s %s" % ("" if c.startswith("IfcUtil::") else "", c) for c in nms ) write = lambda str, **kwargs: declarations.append( str % dict({"documentation": templates.multi_line_comment(documentation.description(kwargs["name"]))}, **kwargs) ) forward_names = list(mapping.schema.entities.keys()) + list(mapping.schema.simpletypes.keys()) forward_definitions = "".join(["class %s; " % n for n in forward_names]) select_super_types = defaultdict(list) for name, type in mapping.schema.selects.items(): for nm in type.values: select_super_types[str(nm).lower()].append(name) write(templates.select_virtual if USE_VIRTUAL_INHERITANCE else templates.select_plain, name=name) def get_select_super_types(nm, bases=[]): x = list(select_super_types[nm.lower()]) for y in x: x.extend(get_select_super_types(y)) return sorted(set(x) - set(itertools.chain.from_iterable(map(get_select_super_types, bases)))) for name, type in mapping.schema.enumerations.items(): short_name = name[:-4] if name.endswith("Enum") else name write(templates.enumeration, name=name, values=", ".join(["%s_%s" % (short_name, v) for v in type.values])) emitted_simpletypes = set() while len(emitted_simpletypes) < len(mapping.schema.simpletypes): for name, type in mapping.schema.simpletypes.items(): if name.lower() in emitted_simpletypes: continue type_str = mapping.make_type_string(mapping.flatten_type_string(type)) attr_type = mapping.make_argument_type(type) superclass = mapping.simple_type_parent(name) superclasses = [] all_superclasses = [] if superclass is not None: superclasses.append(superclass) while superclass: all_superclasses.append(superclass) superclass = mapping.simple_type_parent(superclass) else: superclasses.append("IfcUtil::IfcBaseType") if USE_VIRTUAL_INHERITANCE: superclasses.extend(get_select_super_types(name, bases=all_superclasses)) is_emitted = ( lambda nm: nm == "IfcUtil::IfcBaseType" or nm in mapping.schema.selects or nm.lower() in emitted_simpletypes ) if not all(map(is_emitted, superclasses)): continue superclasses = list(map(case_normalize, superclasses)) emitted_simpletypes.add(name.lower()) superclass_statement = create_supertype_statement(superclasses) write( templates.simpletype, name=name, type=type_str, attr_type=attr_type, superclass=superclass_statement ) class_definitions = [] write = lambda str, **kwargs: class_definitions.append( str % dict({"documentation": templates.multi_line_comment(documentation.description(kwargs["name"]))}, **kwargs) ) emitted_entities = set() while len(emitted_entities) < len(mapping.schema.entities): for name, type in mapping.schema.entities.items(): if name.lower() in emitted_entities: continue if len(type.supertypes) == 0 or set(map(str.lower, type.supertypes)) <= emitted_entities: attr_lines = [] def write_method(attr): attr_lines.extend( ["/// %s" % d for d in documentation.description(".".join((name, attr.name)))] ) type_str = mapping.get_parameter_type(attr) if mapping.make_argument_type(attr) != "IfcUtil::Argument_UNKNOWN": attr_lines.append("%s %s() const;" % (type_str, attr.name)) attr_lines.append("void set%s(%s v);" % (attr.name, type_str)) [write_method(attr) for attr in type.attributes] inv_lines = [] def write_inverse(attr): inv_lines.append( templates.inverse_attr % {"name": attr.name, "entity": attr.entity, "attribute": attr.attribute} ) if type.inverse: [write_inverse(attr) for attr in type.inverse] attributes = "\n".join(["%s%s" % (" " * 4, a) for a in attr_lines]) if len(attributes): attributes += "\n" inverse = "\n".join(["%s%s" % (" " * 4, a) for a in inv_lines]) if len(inverse): inverse += "\n" all_supertypes = [] tt = type while len(tt.supertypes): all_supertypes.append(tt.supertypes[0]) tt = mapping.schema.entities[tt.supertypes[0]] supertypes = list(type.supertypes) if len(type.supertypes) else ["IfcUtil::IfcBaseEntity"] if USE_VIRTUAL_INHERITANCE: supertypes.extend(get_select_super_types(name, bases=all_supertypes)) supertypes = list(map(case_normalize, supertypes)) superclass = create_supertype_statement(supertypes) argument_count = mapping.argument_count(type) argument_start = argument_count - len(type.attributes) argument_name_function_body_switch_stmt = ( " switch (i) {%s}" % ( "".join( [ 'case %d: return "%s"; ' % (i + argument_start, attr.name) for i, attr in enumerate(type.attributes) ] ) ) if len(type.attributes) else "" ) argument_name_function_body_tail = ( (" return %s::getArgumentName(i); " % type.supertypes[0]) if len(type.supertypes) == 1 else ' (void)i; throw IfcParse::IfcAttributeOutOfRangeException("Argument index out of range"); ' ) argument_name_function_body = ( argument_name_function_body_switch_stmt + argument_name_function_body_tail ) derived = mapping.derived_in_supertype(type) attribute_names = list(map(operator.attrgetter("name"), mapping.arguments(type))) derived_in_supertype = set(derived) & set(attribute_names) derived_in_supertype_indices = sorted(attribute_names.index(nm) for nm in derived_in_supertype) attribute_type_cases = [ "case %d: return IfcUtil::Argument_DERIVED; " % idx for idx in derived_in_supertype_indices ] attribute_type_cases += [ "case %d: return %s; " % (i + argument_start, mapping.make_argument_type(attr)) for i, attr in enumerate(type.attributes) ] argument_type_function_body_switch_stmt = ( " switch (i) {%s}" % ("".join(attribute_type_cases)) if len(type.attributes) else "" ) argument_type_function_body_tail = ( (" return %s::getArgumentType(i); " % type.supertypes[0]) if len(type.supertypes) == 1 else ' (void)i; throw IfcParse::IfcAttributeOutOfRangeException("Argument index out of range"); ' ) argument_type_function_body = ( argument_type_function_body_switch_stmt + argument_type_function_body_tail ) argument_entity_function_body_switch_stmt = ( " switch (i) {%s}" % ( "".join( [ "case %d: return %s; " % (i + argument_start, mapping.make_argument_entity(attr)) for i, attr in enumerate(type.attributes) ] ) ) if len(type.attributes) else "" ) argument_entity_function_body_tail = ( (" return %s::getArgumentEntity(i); " % type.supertypes[0]) if len(type.supertypes) == 1 else ' (void)i; throw IfcParse::IfcAttributeOutOfRangeException("Argument index out of range"); ' ) argument_entity_function_body = ( argument_entity_function_body_switch_stmt + argument_entity_function_body_tail ) constructor_arguments = ", ".join( "%(full_type)s v%(index)d_%(name)s" % a for a in mapping.get_assignable_arguments(type) ) write(templates.entity, **locals()) emitted_entities.add(name) self.str = templates.header % { "schema_name_upper": mapping.schema.name.upper(), "schema_name": mapping.schema.name.capitalize(), "declarations": "".join(declarations), "forward_definitions": forward_definitions, "class_definitions": "".join(class_definitions), } self.schema_name = mapping.schema.name.capitalize() self.file_name = "%s.h" % self.schema_name def __repr__(self): return self.str Generator = Header