mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-13 02:47:48 +00:00
Dion Moult
233 lines
8.3 KiB
Python
Executable File
233 lines
8.3 KiB
Python
Executable File
#!python
|
|
|
|
import collision
|
|
import ifcopenshell
|
|
import ifcopenshell.geom
|
|
import ifcopenshell.util.selector
|
|
import multiprocessing
|
|
import numpy as np
|
|
import json
|
|
import sys
|
|
import argparse
|
|
import logging
|
|
|
|
|
|
class Mesh:
|
|
faces: []
|
|
vertices: []
|
|
|
|
|
|
class IfcClasher:
|
|
def __init__(self, settings):
|
|
self.settings = settings
|
|
self.geom_settings = ifcopenshell.geom.settings()
|
|
self.clash_sets = []
|
|
#self.tolerance = 0.01
|
|
|
|
def clash(self):
|
|
for clash_set in self.clash_sets:
|
|
self.process_clash_set(clash_set)
|
|
|
|
def process_clash_set(self, clash_set):
|
|
for ab in ['a', 'b']:
|
|
self.settings.logger.info(f'Creating collision manager {ab} ...')
|
|
clash_set[f'{ab}_cm'] = collision.CollisionManager()
|
|
self.settings.logger.info(f'Loading files {ab} ...')
|
|
for data in clash_set[ab]:
|
|
data['ifc'] = ifcopenshell.open(data['file'])
|
|
self.patch_ifc(data['ifc'])
|
|
self.settings.logger.info(f'Purging unnecessary elements {ab} ...')
|
|
self.purge_elements(data)
|
|
self.settings.logger.info(f'Creating collision data for {ab} ...')
|
|
if len(data['ifc'].by_type('IfcElement')) > 0:
|
|
self.add_collision_objects(data, clash_set[f'{ab}_cm'])
|
|
|
|
if 'b' in clash_set:
|
|
results = clash_set['a_cm'].in_collision_other(clash_set['b_cm'], return_data=True)
|
|
else:
|
|
results = clash_set['a_cm'].in_collision_internal(return_data=True)
|
|
|
|
if not results[0]:
|
|
return
|
|
|
|
tolerance = clash_set['tolerance'] if 'tolerance' in clash_set else 0.01
|
|
clash_set['clashes'] = {}
|
|
|
|
for contact in results[1]:
|
|
a_global_id, b_global_id = contact.names
|
|
a = self.get_element(clash_set['a'], a_global_id)
|
|
if 'b' in clash_set:
|
|
b = self.get_element(clash_set['b'], b_global_id)
|
|
else:
|
|
b = self.get_element(clash_set['a'], b_global_id)
|
|
if contact.raw.penetration_depth < tolerance:
|
|
continue
|
|
key = f'{a_global_id}-{b_global_id}'
|
|
if key in clash_set['clashes'] \
|
|
and clash_set['clashes'][key]['penetration_depth'] > contact.raw.penetration_depth:
|
|
continue
|
|
clash_set['clashes'][key] = {
|
|
'a_global_id': a_global_id,
|
|
'b_global_id': b_global_id,
|
|
'a_ifc_class': a.is_a(),
|
|
'b_ifc_class': b.is_a(),
|
|
'a_name': a.Name,
|
|
'b_name': b.Name,
|
|
'normal': list(contact.raw.normal),
|
|
'position': list(contact.raw.pos),
|
|
'penetration_depth': contact.raw.penetration_depth
|
|
}
|
|
|
|
def export(self):
|
|
results = self.clash_sets.copy()
|
|
for result in results:
|
|
del result['a_cm']
|
|
del result['b_cm']
|
|
for ab in ['a', 'b']:
|
|
for data in result[ab]:
|
|
for key in ['ifc', 'meshes']:
|
|
if key in data:
|
|
del data[key]
|
|
with open(self.settings.output, 'w', encoding='utf-8') as clashes_file:
|
|
json.dump(results, clashes_file, indent=4)
|
|
|
|
def get_element(self, clash_group, global_id):
|
|
for data in clash_group:
|
|
try:
|
|
element = data['ifc'].by_guid(global_id)
|
|
if element:
|
|
return element
|
|
except:
|
|
pass
|
|
|
|
def purge_elements(self, data):
|
|
if 'selector' not in data:
|
|
for element in data['ifc'].by_type('IfcSpace'):
|
|
ifc_file.remove(element)
|
|
return
|
|
|
|
selector = ifcopenshell.util.selector.Selector()
|
|
elements = selector.parse(data['ifc'], data['selector'])
|
|
|
|
if data['mode'] == 'e':
|
|
for element in data['ifc'].by_type('IfcElement'):
|
|
if element in elements:
|
|
data['ifc'].remove(element)
|
|
elif data['mode'] == 'i':
|
|
for element in data['ifc'].by_type('IfcElement'):
|
|
if element not in elements:
|
|
data['ifc'].remove(element)
|
|
|
|
def add_collision_objects(self, data, cm):
|
|
data['meshes'] = {}
|
|
iterator = ifcopenshell.geom.iterator(self.geom_settings, data['ifc'], multiprocessing.cpu_count())
|
|
valid_file = iterator.initialize()
|
|
if not valid_file:
|
|
return False
|
|
old_progress = -1
|
|
while True:
|
|
progress = iterator.progress() // 2
|
|
if progress > old_progress:
|
|
print("\r[" + "#" * progress + " " * (50 - progress) + "]", end="")
|
|
old_progress = progress
|
|
self.add_collision_object(data, cm, iterator.get())
|
|
if not iterator.next():
|
|
break
|
|
|
|
def add_collision_object(self, data, cm, shape):
|
|
if shape is None:
|
|
return
|
|
element = data['ifc'].by_id(shape.guid)
|
|
self.settings.logger.info('Creating object {}'.format(element))
|
|
mesh_name = f'mesh-{shape.geometry.id}'
|
|
if mesh_name in data['meshes']:
|
|
mesh = data['meshes'][mesh_name]
|
|
else:
|
|
mesh = self.create_mesh(shape)
|
|
data['meshes'][mesh_name] = mesh
|
|
|
|
m = shape.transformation.matrix.data
|
|
mat = np.array(
|
|
[
|
|
[m[0], m[3], m[6], m[9]],
|
|
[m[1], m[4], m[7], m[10]],
|
|
[m[2], m[5], m[8], m[11]],
|
|
[0, 0, 0, 1]
|
|
]
|
|
)
|
|
mat.transpose()
|
|
cm.add_object(shape.guid, mesh, mat)
|
|
|
|
def create_mesh(self, shape):
|
|
f = shape.geometry.faces
|
|
v = shape.geometry.verts
|
|
mesh = Mesh()
|
|
mesh.vertices = np.array([[v[i], v[i + 1], v[i + 2]]
|
|
for i in range(0, len(v), 3)])
|
|
mesh.faces = np.array([[f[i], f[i + 1], f[i + 2]]
|
|
for i in range(0, len(f), 3)])
|
|
return mesh
|
|
|
|
def patch_ifc(self, ifc_file):
|
|
project = ifc_file.by_type('IfcProject')[0]
|
|
sites = self.find_decomposed_ifc_class(project, 'IfcSite')
|
|
for site in sites:
|
|
self.patch_placement_to_origin(site)
|
|
buildings = self.find_decomposed_ifc_class(project, 'IfcBuilding')
|
|
for building in buildings:
|
|
self.patch_placement_to_origin(building)
|
|
|
|
def find_decomposed_ifc_class(self, element, ifc_class):
|
|
results = []
|
|
rel_aggregates = element.IsDecomposedBy
|
|
if not rel_aggregates:
|
|
return results
|
|
for rel_aggregate in rel_aggregates:
|
|
for part in rel_aggregate.RelatedObjects:
|
|
if part.is_a(ifc_class):
|
|
results.append(part)
|
|
results.extend(self.find_decomposed_ifc_class(part, ifc_class))
|
|
return results
|
|
|
|
def patch_placement_to_origin(self, element):
|
|
element.ObjectPlacement.RelativePlacement.Location.Coordinates = (0., 0., 0.)
|
|
if element.ObjectPlacement.RelativePlacement.Axis:
|
|
element.ObjectPlacement.RelativePlacement.Axis.DirectionRatios = (0., 0., 1.)
|
|
if element.ObjectPlacement.RelativePlacement.RefDirection:
|
|
element.ObjectPlacement.RelativePlacement.RefDirection.DirectionRatios = (1., 0., 0.)
|
|
|
|
|
|
class IfcClashSettings:
|
|
def __init__(self):
|
|
self.logger = None
|
|
self.output = 'clashes.json'
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(
|
|
description='Clashes geometry between two IFC files')
|
|
parser.add_argument(
|
|
'input',
|
|
type=str,
|
|
help='A JSON dataset describing a series of clashsets')
|
|
parser.add_argument(
|
|
'-o',
|
|
'--output',
|
|
type=str,
|
|
help='The JSON diff file to output. Defaults to output.json',
|
|
default='output.json')
|
|
args = parser.parse_args()
|
|
|
|
settings = IfcClashSettings()
|
|
settings.output = args.output
|
|
settings.logger = logging.getLogger('Clash')
|
|
settings.logger.setLevel(logging.DEBUG)
|
|
handler = logging.StreamHandler(sys.stdout)
|
|
handler.setLevel(logging.DEBUG)
|
|
settings.logger.addHandler(handler)
|
|
ifc_clasher = IfcClasher(settings)
|
|
with open(args.input, 'r') as clash_sets_file:
|
|
ifc_clasher.clash_sets = json.loads(clash_sets_file.read())
|
|
ifc_clasher.clash()
|
|
ifc_clasher.export()
|