mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-19 03:33:48 +00:00
major update, refactoring to consider generalization of connections
This commit is contained in:
+180
-47
@@ -7,20 +7,49 @@ class IFC2CA:
|
||||
self.filename = filename
|
||||
self.file = None
|
||||
self.result = {}
|
||||
self.warnings = []
|
||||
|
||||
def convert(self):
|
||||
self.file = ifcopenshell.open(self.filename)
|
||||
for model in self.file.by_type('IfcStructuralAnalysisModel'):
|
||||
elements = self.get_structural_items(model, item_type='IfcStructuralMember')
|
||||
connections = self.get_structural_items(model, item_type='IfcStructuralConnection')
|
||||
|
||||
materialdb = []
|
||||
materials = list(dict.fromkeys([e['material'] for e in elements]))
|
||||
for mat in materials:
|
||||
id = int(mat.split('|')[1])
|
||||
material = self.get_material_properties(self.file.by_id(id))
|
||||
material['relatedElements'] = [e['ifcName'] for e in elements if 'material' in e and e['material'] == mat]
|
||||
materialdb.append(material)
|
||||
|
||||
profiledb = []
|
||||
profiles = list(dict.fromkeys([e['profile'] for e in elements if 'profile' in e]))
|
||||
for prof in profiles:
|
||||
id = int(prof.split('|')[1])
|
||||
profile = self.get_profile_properties(self.file.by_id(id))
|
||||
profile['relatedElements'] = [e['ifcName'] for e in elements if 'profile' in e and e['profile'] == prof]
|
||||
profiledb.append(profile)
|
||||
|
||||
self.result = {
|
||||
'ifcName': model.is_a() + '|' + str(model.id()),
|
||||
'name': model.Name,
|
||||
'id': model.GlobalId,
|
||||
'elements': self.get_structural_items(model, item_type='IfcStructuralMember'),
|
||||
'connections': self.get_structural_items(model, item_type='IfcStructuralConnection')
|
||||
'elements': elements,
|
||||
'connections': connections,
|
||||
'db': {
|
||||
'materials': materialdb,
|
||||
'profiles': profiledb
|
||||
},
|
||||
'warnings': self.warnings
|
||||
}
|
||||
|
||||
print('Number of elements: ', len(self.result['elements']))
|
||||
print('Number of connections: ', len(self.result['connections']))
|
||||
print('Model %s converted' % model.Name)
|
||||
print('Number of elements: ', len(elements))
|
||||
print('Number of connections: ', len(connections))
|
||||
print('Number of materials: ', len(materialdb))
|
||||
print('Number of profiles: ', len(profiledb))
|
||||
print('')
|
||||
|
||||
break
|
||||
|
||||
@@ -36,6 +65,8 @@ class IFC2CA:
|
||||
return items
|
||||
|
||||
def get_item_data(self, item):
|
||||
transformation = self.get_transformation(item.ObjectPlacement)
|
||||
|
||||
if item.is_a('IfcStructuralCurveMember'):
|
||||
representation = self.get_representation(item, 'Edge')
|
||||
material_profile = self.get_material_profile(item)
|
||||
@@ -43,7 +74,30 @@ class IFC2CA:
|
||||
print(representation, material_profile)
|
||||
return
|
||||
|
||||
material = material_profile.Material
|
||||
profile = material_profile.Profile
|
||||
geometry = self.get_geometry(representation)
|
||||
orientation = self.get_1D_orientation(geometry, item.Axis)
|
||||
connections = self.get_connection_data(item.ConnectedBy)
|
||||
for conn in connections:
|
||||
if not conn['orientation']:
|
||||
conn['orientation'] = orientation
|
||||
# --> Correct pointOnElement for eccentricity connection for ETABS files
|
||||
length = np.linalg.norm(np.array(geometry[1]) - np.array(geometry[0]))
|
||||
for c in connections:
|
||||
if c['eccentricity']:
|
||||
if np.linalg.norm(np.array(c['eccentricity']['pointOnElement'])) > length:
|
||||
# print('Eccentricity in %s corrected' % item.is_a() + '|' + str(item.id()))
|
||||
self.warnings.append('Eccentricity in %s corrected' % (item.is_a() + '|' + str(item.id())))
|
||||
c['eccentricity']['pointOnElement'][0] = length
|
||||
# End <--
|
||||
if transformation:
|
||||
geometry = self.transform_vectors(geometry, transformation)
|
||||
orientation = self.transform_vectors(orientation, transformation, include_translation=False)
|
||||
for c in connections:
|
||||
c['orientation'] = self.transform_vectors(c['orientation'], transformation, include_translation=False)
|
||||
if c['eccentricity']:
|
||||
c['eccentricity']['vector'] = self.transform_vectors(c['eccentricity']['vector'], transformation, include_translation=False)
|
||||
|
||||
return {
|
||||
'ifcName': item.is_a() + '|' + str(item.id()),
|
||||
@@ -52,10 +106,10 @@ class IFC2CA:
|
||||
'geometryType': 'line',
|
||||
'predefinedType': item.PredefinedType,
|
||||
'geometry': geometry,
|
||||
'orientation': self.get_1D_orientation(geometry, item.Axis),
|
||||
'material': self.get_material_properties(material_profile.Material),
|
||||
'profile': self.get_profile_properties(material_profile.Profile),
|
||||
'connections': self.get_connection_data(item.ConnectedBy)
|
||||
'orientation': orientation,
|
||||
'material': material.is_a() + '|' + str(material.id()),
|
||||
'profile': profile.is_a() + '|' + str(profile.id()),
|
||||
'connections': connections
|
||||
}
|
||||
|
||||
elif item.is_a('IfcStructuralSurfaceMember'):
|
||||
@@ -65,6 +119,18 @@ class IFC2CA:
|
||||
print(representation)
|
||||
return
|
||||
|
||||
geometry = self.get_geometry(representation)
|
||||
orientation = self.get_2D_orientation(representation)
|
||||
connections = self.get_connection_data(item.ConnectedBy)
|
||||
for conn in connections:
|
||||
if not conn['orientation']:
|
||||
conn['orientation'] = orientation
|
||||
if transformation:
|
||||
geometry = self.transform_vectors(geometry, transformation)
|
||||
orientation = self.transform_vectors(orientation, transformation, include_translation=False)
|
||||
for c in connections:
|
||||
c['orientation'] = self.transform_vectors(c['orientation'], transformation, include_translation=False)
|
||||
|
||||
return {
|
||||
'ifcName': item.is_a() + '|' + str(item.id()),
|
||||
'name': item.Name,
|
||||
@@ -72,9 +138,10 @@ class IFC2CA:
|
||||
'geometryType': 'surface',
|
||||
'predefinedType': item.PredefinedType,
|
||||
'thickness': item.Thickness,
|
||||
'geometry': self.get_geometry(representation),
|
||||
'material': self.get_material_properties(material),
|
||||
'connections': self.get_connection_data(item.ConnectedBy)
|
||||
'geometry': geometry,
|
||||
'orientation': orientation,
|
||||
'material': material.is_a() + '|' + str(material.id()),
|
||||
'connections': connections
|
||||
}
|
||||
|
||||
elif item.is_a('IfcStructuralPointConnection'):
|
||||
@@ -83,17 +150,60 @@ class IFC2CA:
|
||||
print(representation)
|
||||
return
|
||||
|
||||
geometry = self.get_geometry(representation)
|
||||
orientation = self.get_0D_orientation(item.ConditionCoordinateSystem)
|
||||
if not orientation:
|
||||
orientation = np.eye(3).tolist()
|
||||
if transformation:
|
||||
geometry = self.transform_vectors(geometry, transformation)
|
||||
orientation = self.transform_vectors(orientation, transformation, include_translation=False)
|
||||
|
||||
return {
|
||||
'ifcName': item.is_a() + '|' + str(item.id()),
|
||||
'name': item.Name,
|
||||
'id': item.GlobalId,
|
||||
'geometryType': 'point',
|
||||
'geometry': self.get_geometry(representation),
|
||||
'orientation': self.get_0D_orientation(item.ConditionCoordinateSystem),
|
||||
'geometry': geometry,
|
||||
'orientation': orientation,
|
||||
'appliedCondition': self.get_connection_input(item),
|
||||
'relatedElements': self.get_connection_data(item.ConnectsStructuralMembers)
|
||||
'relatedElements': [con.is_a() + '|' + str(con.id()) for con in item.ConnectsStructuralMembers]
|
||||
}
|
||||
|
||||
def get_transformation(self, placement):
|
||||
if not placement:
|
||||
return None
|
||||
if placement.is_a('IfcLocalPlacement'):
|
||||
if placement.PlacementRelTo:
|
||||
print('Warning! Object Placement with PlacementRelTo attribute is not supported and will be neglected')
|
||||
axes = placement.RelativePlacement
|
||||
location = np.array(self.get_coordinate(axes.Location))
|
||||
if axes.Axis and axes.RefDirection:
|
||||
xAxis = np.array(axes.RefDirection.DirectionRatios) # this can be not accurate (in the xz plane)
|
||||
zAxis = np.array(axes.Axis.DirectionRatios)
|
||||
zAxis /= np.linalg.norm(zAxis)
|
||||
yAxis = np.cross(zAxis, xAxis)
|
||||
yAxis /= np.linalg.norm(yAxis)
|
||||
xAxis = np.cross(yAxis, zAxis)
|
||||
xAxis /= np.linalg.norm(xAxis)
|
||||
else:
|
||||
if np.allclose(location, np.array([0., 0., 0.])):
|
||||
return None
|
||||
xAxis = np.array([1., 0., 0.])
|
||||
yAxis = np.array([0., 1., 0.])
|
||||
zAxis = np.array([0., 0., 1.])
|
||||
if (np.allclose(location, np.array([0., 0., 0.])) and
|
||||
np.allclose(xAxis, np.array([1., 0., 0.])) and
|
||||
np.allclose(yAxis, np.array([0., 1., 0.])) and
|
||||
np.allclose(zAxis, np.array([0., 0., 1.]))):
|
||||
return None
|
||||
return {
|
||||
'location': location,
|
||||
'rotationMatrix': np.array([xAxis, yAxis, zAxis]).transpose()
|
||||
}
|
||||
else:
|
||||
print('Warning! Object Placement is of type %s, which is not supported. Default considered' % placement.is_a())
|
||||
return None
|
||||
|
||||
def get_representation(self, element, rep_type):
|
||||
if not element.Representation:
|
||||
return None
|
||||
@@ -140,41 +250,58 @@ class IFC2CA:
|
||||
|
||||
def get_coordinate(self, point):
|
||||
if point.is_a('IfcCartesianPoint'):
|
||||
return point.Coordinates
|
||||
return list(point.Coordinates)
|
||||
|
||||
def get_0D_orientation(self, axes):
|
||||
if axes and axes.Axis and axes.RefDirection:
|
||||
xAxis = np.array(axes.RefDirection.DirectionRatios) # this can be not accurate (in the xz plane)
|
||||
xAxis = np.array(axes.RefDirection.DirectionRatios) # this can be not strictly perpendicular (in the xz plane)
|
||||
zAxis = np.array(axes.Axis.DirectionRatios)
|
||||
zAxis /= np.linalg.norm(zAxis)
|
||||
yAxis = np.cross(zAxis, xAxis)
|
||||
yAxis /= np.linalg.norm(yAxis)
|
||||
xAxis = np.cross(yAxis, zAxis)
|
||||
xAxis /= np.linalg.norm(xAxis)
|
||||
# print('0D:', xAxis, yAxis, zAxis)
|
||||
value = xAxis.tolist()
|
||||
value.extend(yAxis.tolist())
|
||||
return {
|
||||
'type': 'xyPlane',
|
||||
'value': value
|
||||
}
|
||||
|
||||
return [xAxis.tolist(), yAxis.tolist(), zAxis.tolist()]
|
||||
else: # return None and copy the elements orientation
|
||||
return None
|
||||
|
||||
def get_1D_orientation(self, geometry, zAxis):
|
||||
if zAxis:
|
||||
xAxis = np.array(geometry[1]) - np.array(geometry[0])
|
||||
zAxis = np.array(zAxis.DirectionRatios)
|
||||
yAxis = np.cross(zAxis, xAxis)
|
||||
yAxis /= np.linalg.norm(yAxis)
|
||||
# print('1D:', xAxis, yAxis, zAxis)
|
||||
return {
|
||||
'type': 'yAxis',
|
||||
'value': yAxis.tolist()
|
||||
}
|
||||
else:
|
||||
print('Warning! Orientation for curve member missing. Default considered')
|
||||
return {
|
||||
'type': 'rotationAngle',
|
||||
'value': 0
|
||||
}
|
||||
xAxis = np.array(geometry[1]) - np.array(geometry[0])
|
||||
xAxis /= np.linalg.norm(xAxis)
|
||||
zAxis = np.array(zAxis.DirectionRatios) # this can be not strictly perpendicular (in the xz plane)
|
||||
yAxis = np.cross(zAxis, xAxis)
|
||||
yAxis /= np.linalg.norm(yAxis)
|
||||
zAxis = np.cross(xAxis, yAxis)
|
||||
zAxis /= np.linalg.norm(zAxis)
|
||||
|
||||
return [xAxis.tolist(), yAxis.tolist(), zAxis.tolist()]
|
||||
|
||||
def get_2D_orientation(self, representation):
|
||||
item = representation.Items[0]
|
||||
if item.is_a('IfcFaceSurface'):
|
||||
item.SameSense
|
||||
axes = item.FaceSurface.Position
|
||||
orientation = self.get_0D_orientation(axes)
|
||||
if not item.SameSense:
|
||||
orientation = [[-v for v in vec] for vec in orientation]
|
||||
return orientation
|
||||
|
||||
def transform_vectors(self, geometry, trsf, include_translation=True):
|
||||
if not any(isinstance(el, list) for el in geometry): # single point which contains no list
|
||||
geometry = [geometry]
|
||||
globalGeometry = []
|
||||
|
||||
for p in geometry:
|
||||
gp = trsf['rotationMatrix'].dot(np.array(p))
|
||||
if include_translation:
|
||||
gp += trsf['location']
|
||||
globalGeometry.append(gp.tolist())
|
||||
|
||||
if len(globalGeometry) == 1: # single point
|
||||
globalGeometry = globalGeometry[0]
|
||||
|
||||
return globalGeometry
|
||||
|
||||
def get_material_profile(self, element):
|
||||
if not element.HasAssociations:
|
||||
@@ -270,12 +397,13 @@ class IFC2CA:
|
||||
'orientation': self.get_0D_orientation(rel.ConditionCoordinateSystem),
|
||||
'appliedCondition': self.get_connection_input(rel),
|
||||
'eccentricity': None if not rel.is_a('IfcRelConnectsWithEccentricity') else {
|
||||
'inX': 0.0 if not rel.ConnectionConstraint.EccentricityInX else rel.ConnectionConstraint.EccentricityInX,
|
||||
'inY': 0.0 if not rel.ConnectionConstraint.EccentricityInY else rel.ConnectionConstraint.EccentricityInY,
|
||||
'inZ': 0.0 if not rel.ConnectionConstraint.EccentricityInZ else rel.ConnectionConstraint.EccentricityInZ,
|
||||
'vector': [
|
||||
0.0 if not rel.ConnectionConstraint.EccentricityInX else rel.ConnectionConstraint.EccentricityInX,
|
||||
0.0 if not rel.ConnectionConstraint.EccentricityInY else rel.ConnectionConstraint.EccentricityInY,
|
||||
0.0 if not rel.ConnectionConstraint.EccentricityInZ else rel.ConnectionConstraint.EccentricityInZ
|
||||
],
|
||||
'pointOnElement': self.get_coordinate(rel.ConnectionConstraint.PointOnRelatingElement)
|
||||
}
|
||||
# 'geometryPointIndex': None
|
||||
} for rel in itemList]
|
||||
|
||||
def get_connection_input(self, connection):
|
||||
@@ -310,7 +438,12 @@ class IFC2CA:
|
||||
}
|
||||
|
||||
if __name__ == '__main__':
|
||||
IFC_FILENAME = ''
|
||||
ifc2ca = IFC2CA(IFC_FILENAME)
|
||||
ifc2ca.convert()
|
||||
print(json.dumps(ifc2ca.result, indent=4))
|
||||
fileNames = ['cantilever_01', 'portal_01'];
|
||||
files = fileNames
|
||||
|
||||
for fileName in files:
|
||||
BASE_PATH = '/home/jesusbill/Dev-Projects/github.com/IfcOpenShell/analysis-models/ifcFiles/'
|
||||
ifc2ca = IFC2CA(BASE_PATH + fileName + '.ifc')
|
||||
ifc2ca.convert()
|
||||
with open(BASE_PATH + fileName + '.json', 'w') as f:
|
||||
f.write(json.dumps(ifc2ca.result, indent = 4))
|
||||
|
||||
Reference in New Issue
Block a user