major update, refactoring to consider generalization of connections

This commit is contained in:
Jesusbill
2020-05-27 05:33:36 +02:00
parent 8bef6b9347
commit e071c7812f
5 changed files with 882 additions and 558 deletions
+180 -47
View File
@@ -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))