IFC2JSON_python/ifc2json.py at master · BIM-Tools/IFC2JSON_python · GitHub

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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
import ifcopenshell
import json

ifc_file = ifcopenshell.open('./samples/7m900_tue_hello_wall_with_door.ifc')
schema = ifcopenshell.ifcopenshell_wrapper.schema_by_name(ifc_file.schema)
id_objects = {}

def entityToDict(entity):
    ref = {
        "Type": entity.is_a()
    }
    attr_dict = entity.__dict__

    # check for globalid
    if "GlobalId" in attr_dict:
        ref["ref"] = attr_dict["GlobalId"]
        if not attr_dict["GlobalId"] in id_objects:
            d = {
                "Type": entity.is_a()
            }

            for i in range(0,len(entity)):
                attr = entity.attribute_name(i)
                if attr in attr_dict:
                    if not attr == "OwnerHistory":
                        jsonValue = getEntityValue(attr_dict[attr])
                        if jsonValue:
                            d[attr] = jsonValue
                        if attr_dict[attr] == None:
                            continue
                        elif isinstance(attr_dict[attr], ifcopenshell.entity_instance):
                            d[attr] = entityToDict(attr_dict[attr])
                        elif isinstance(attr_dict[attr], tuple):
                            subEnts = []
                            for subEntity in attr_dict[attr]:
                                if isinstance(subEntity, ifcopenshell.entity_instance):
                                    subEntJson = entityToDict(subEntity)
                                    if subEntJson:
                                        subEnts.append(subEntJson)
                                else:
                                    subEnts.append(subEntity)
                            if len(subEnts) > 0:
                                d[attr] = subEnts
                        else:
                            d[attr] = attr_dict[attr]
            id_objects[attr_dict["GlobalId"]] = d
        return ref
    else:
        d = {
            "Type": entity.is_a()
        }

        for i in range(0,len(entity)):
            attr = entity.attribute_name(i)
            if attr in attr_dict:
                if not attr == "OwnerHistory":
                    jsonValue = getEntityValue(attr_dict[attr])
                    if jsonValue:
                        d[attr] = jsonValue
                    if attr_dict[attr] == None:
                        continue
                    elif isinstance(attr_dict[attr], ifcopenshell.entity_instance):
                        d[attr] = entityToDict(attr_dict[attr])
                    elif isinstance(attr_dict[attr], tuple):
                        subEnts = []
                        for subEntity in attr_dict[attr]:
                            if isinstance(subEntity, ifcopenshell.entity_instance):
                                # subEnts.append(None)
                                subEntJson = entityToDict(subEntity)
                                if subEntJson:
                                    subEnts.append(subEntJson)
                            else:
                                subEnts.append(subEntity)
                        if len(subEnts) > 0:
                            d[attr] = subEnts
                    else:
                        d[attr] = attr_dict[attr]
        return d

def getEntityValue(value):
    if value == None:
        jsonValue = None
    elif isinstance(value, ifcopenshell.entity_instance):
        jsonValue = entityToDict(value)
    elif isinstance(value, tuple):
        jsonValue = None
        subEnts = []
        for subEntity in value:
            subEnts.append(getEntityValue(subEntity))
        jsonValue = subEnts
    else:
        jsonValue = value
    return jsonValue


jsonObjects= []
entityIter = iter(ifc_file)
for entity in entityIter:
    entityToDict(entity)
for key in id_objects:
    jsonObjects.append(id_objects[key])
with open('7m900_tue_hello_wall_with_door.json', 'w') as outfile:
    json.dump(jsonObjects, outfile, indent=4)