Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-08-13 10:57:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement support for door attributes

Browse Source
This commit is contained in:
Dion Moult
2019-09-29 22:12:51 +10:00
parent 22d35cf89a
commit a406f3d78a
1 changed files with 33 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcblenderexport/io_export_ifc/export.py
+33 -5
View File
@@ -120,6 +120,7 @@ class IfcParser():
self.rel_aggregates = {}
self.representations = {}
self.type_products = []
self.door_attributes = {}
self.project = {}
self.libraries = []
self.products = []
@@ -141,6 +142,7 @@ class IfcParser():
self.project = self.get_project()
self.libraries = self.get_libraries()
self.door_attributes = self.get_door_attributes()
self.type_products = self.get_type_products()
self.get_products()
self.map_conversion = self.get_map_conversion()
@@ -219,7 +221,11 @@ class IfcParser():
product['attributes'].update(attribute_override)
product.update(metadata_override)
if product['raw'].parent:
if object.parent \
and self.is_a_type(self.get_ifc_class(object.parent.name)):
reference = self.get_type_product_reference(object.parent.name)
self.rel_defines_by_type.setdefault(reference, []).append(self.product_index)
elif product['raw'].parent:
parent_product_index = self.get_product_index_from_raw_name(product['raw'].parent.name)
self.rel_nests.setdefault(parent_product_index, []).append(product)
product['relating_host'] = parent_product_index
@@ -265,10 +271,6 @@ class IfcParser():
for slot in object.material_slots:
self.rel_associates_material.setdefault( slot.material.name, []).append(product)
if object.parent \
and self.is_a_type(self.get_ifc_class(object.parent.name)):
reference = self.get_type_product_reference(object.parent.name)
self.rel_defines_by_type.setdefault(reference, []).append(self.product_index)
return product
def parse_product_collection(self, product, collection):
@@ -329,6 +331,19 @@ class IfcParser():
}
return psets
def get_door_attributes(self):
results = {}
for filename in Path(self.data_dir + 'door/').glob('**/*.csv'):
with open(filename, 'r') as f:
type_name = filename.parts[-2]
pset_name = filename.stem
results.setdefault(type_name, []).append({
'ifc': None,
'raw': { x[0]: x[1] for x in list(csv.reader(f)) },
'pset_name': pset_name.split('.')[0]
})
return results
def get_classifications(self):
results = []
class_path = self.data_dir + 'class/'
@@ -908,11 +923,24 @@ class IfcExporter():
[self.ifc_parser.psets[pset]['ifc'] for pset in
product['psets']] })
if product['class'] == 'IfcDoorType' \
and product['attributes']['Name'] in self.ifc_parser.door_attributes:
attributes = self.ifc_parser.door_attributes[product['attributes']['Name']]
self.create_door_attributes(attributes)
product['attributes'].setdefault('HasPropertySets', [])
for attribute in attributes:
product['attributes']['HasPropertySets'].append(attribute['ifc'])
try:
product['ifc'] = self.file.create_entity(product['class'], **product['attributes'])
except RuntimeError as e:
print('The type product "{}/{}" could not be created: {}'.format(product['class'], product['attributes']['Name'], e.args))
def create_door_attributes(self, attributes):
for attribute in attributes:
attribute['ifc'] = self.file.create_entity(attribute['pset_name'],
**{k: float(v) if v.replace('.', '', 1).isdigit() else v for k, v in attribute['raw'].items()})
def relate_definitions_to_contexts(self):
for library in self.ifc_parser.libraries:
self.file.createIfcRelDeclares(