/
element.py
53 lines (47 loc) · 1.88 KB
/
element.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
def get_psets(element):
psets = {}
try:
if element.is_a('IfcTypeObject'):
if element.HasPropertySets:
for definition in element.HasPropertySets:
psets[definition.Name] = get_property_definition(definition)
else:
for relationship in element.IsDefinedBy:
if relationship.is_a('IfcRelDefinesByProperties'):
definition = relationship.RelatingPropertyDefinition
psets[definition.Name] = get_property_definition(definition)
except Exception as e:
import traceback
print('failed to load properties: {}'.format(e))
traceback.print_exc()
return psets
def get_property_definition(definition):
if definition is not None:
props = {}
if definition.is_a('IfcElementQuantity'):
props.update(get_quantities(definition.Quantities))
elif definition.is_a('IfcPropertySet'):
props.update(get_properties(definition.HasProperties))
else:
# Entity introduced in IFC4
# definition.is_a('IfcPreDefinedPropertySet'):
for prop in range(4, len(definition)):
props[definition.attribute_name(prop)] = definition[prop]
return props
def get_quantities(quantities):
results = {}
for quantity in quantities:
if quantity.is_a('IfcPhysicalSimpleQuantity'):
results[quantity.Name] = quantity[3]
return results
def get_properties(properties):
results = {}
for prop in properties:
if prop.is_a('IfcPropertySingleValue'):
results[prop.Name] = prop.NominalValue
elif prop.is_a('IfcComplexProperty'):
data = prop.get_info()
data['properties'] = get_properties(prop.HasProperties)
del(data['HasProperties'])
results[prop.Name] = data
return results