mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-12 10:33:20 +00:00
IfcBlender: Ensure correct matrices on nested parenting
This commit is contained in:
Thomas Krijnen
@@ -59,6 +59,122 @@ bpy.types.Object.ifc_type = StringProperty(name="IFC Entity Type",
|
||||
description="The STEP Datatype keyword")
|
||||
|
||||
|
||||
def import_ifc(filename, use_names, process_relations):
|
||||
from . import IfcImport
|
||||
if not IfcImport.Init(filename):
|
||||
return
|
||||
id_to_object = {}
|
||||
id_to_parent = {}
|
||||
id_to_matrix = {}
|
||||
old_progress = -1
|
||||
print("Creating geometry...")
|
||||
while True:
|
||||
ob = IfcImport.Get()
|
||||
if ob.type != 'IfcOpeningElement':
|
||||
f = ob.mesh.faces
|
||||
v = ob.mesh.verts
|
||||
m = ob.matrix
|
||||
t = ob.type[0:21]
|
||||
nm = ob.name if len(ob.name) and use_names else ob.guid
|
||||
|
||||
verts = [[v[i], v[i + 1], v[i + 2]] \
|
||||
for i in range(0, len(v), 3)]
|
||||
faces = [[f[i], f[i + 1], f[i + 2]] \
|
||||
for i in range(0, len(f), 3)]
|
||||
|
||||
me = bpy.data.meshes.new('mesh%d' % ob.mesh.id)
|
||||
me.from_pydata(verts, [], faces)
|
||||
if t in bpy.data.materials:
|
||||
mat = bpy.data.materials[t]
|
||||
mat.use_fake_user = True
|
||||
else:
|
||||
mat = bpy.data.materials.new(t)
|
||||
me.materials.append(mat)
|
||||
|
||||
bob = bpy.data.objects.new(nm, me)
|
||||
mat = mathutils.Matrix(([m[0], m[1], m[2], 0],
|
||||
[m[3], m[4], m[5], 0],
|
||||
[m[6], m[7], m[8], 0],
|
||||
[m[9], m[10], m[11], 1]))
|
||||
if process_relations:
|
||||
id_to_matrix[ob.id] = mat
|
||||
else:
|
||||
bob.matrix_world = mat
|
||||
bpy.context.scene.objects.link(bob)
|
||||
|
||||
bpy.ops.object.select_all(action='DESELECT')
|
||||
bpy.ops.object.select_name(name=bob.name)
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
bpy.ops.mesh.normals_make_consistent()
|
||||
bpy.ops.object.mode_set(mode='OBJECT')
|
||||
|
||||
bob.ifc_id, bob.ifc_guid, bob.ifc_name, bob.ifc_type = \
|
||||
ob.id, ob.guid, ob.name, ob.type
|
||||
|
||||
bob.hide = ob.type == 'IfcSpace'
|
||||
id_to_object[ob.id] = bob
|
||||
|
||||
if ob.parent_id > 0:
|
||||
id_to_parent[ob.id] = ob.parent_id
|
||||
progress = IfcImport.Progress() // 2
|
||||
if progress > old_progress:
|
||||
print("\r[" + "#" * progress + " " * (50 - progress) + "]", end="")
|
||||
old_progress = progress
|
||||
if not IfcImport.Next():
|
||||
break
|
||||
|
||||
print("\rDone creating geometry" + " " * 30)
|
||||
|
||||
id_to_parent_temp = dict(id_to_parent)
|
||||
|
||||
while len(id_to_parent_temp) and process_relations:
|
||||
id, parent_id = id_to_parent_temp.popitem()
|
||||
|
||||
if parent_id in id_to_object:
|
||||
bob = id_to_object[parent_id]
|
||||
else:
|
||||
parent_ob = IfcImport.GetObject(parent_id)
|
||||
if parent_ob.id == -1:
|
||||
bob = None
|
||||
else:
|
||||
m = parent_ob.matrix
|
||||
nm = parent_ob.name if len(parent_ob.name) and use_names \
|
||||
else parent_ob.guid
|
||||
bob = bpy.data.objects.new(nm, None)
|
||||
id_to_matrix[parent_ob.id] = mathutils.Matrix((
|
||||
[m[0], m[1], m[2], 0],
|
||||
[m[3], m[4], m[5], 0],
|
||||
[m[6], m[7], m[8], 0],
|
||||
[m[9], m[10], m[11], 1]))
|
||||
bpy.context.scene.objects.link(bob)
|
||||
|
||||
bob.ifc_id = parent_ob.id
|
||||
bob.ifc_guid = parent_ob.guid
|
||||
bob.ifc_name = parent_ob.name
|
||||
bob.ifc_type = parent_ob.type
|
||||
|
||||
if parent_ob.parent_id > 0:
|
||||
id_to_parent[parent_id] = parent_ob.parent_id
|
||||
id_to_parent_temp[parent_id] = parent_ob.parent_id
|
||||
id_to_object[parent_id] = bob
|
||||
if bob:
|
||||
ob = id_to_object[id]
|
||||
ob.parent = bob
|
||||
|
||||
id_to_matrix_temp = dict(id_to_matrix)
|
||||
|
||||
while len(id_to_matrix_temp):
|
||||
id, matrix = id_to_matrix_temp.popitem()
|
||||
parent_id = id_to_parent.get(id, None)
|
||||
parent_matrix = id_to_matrix.get(parent_id, None)
|
||||
if parent_matrix:
|
||||
id_to_object[id].matrix_local = parent_matrix.inverted() * matrix
|
||||
else:
|
||||
id_to_object[id].matrix_world = matrix
|
||||
|
||||
IfcImport.CleanUp()
|
||||
|
||||
|
||||
class ImportIFC(bpy.types.Operator, ImportHelper):
|
||||
bl_idname = "import_scene.ifc"
|
||||
bl_label = "Import .ifc file"
|
||||
@@ -70,104 +186,12 @@ class ImportIFC(bpy.types.Operator, ImportHelper):
|
||||
description="Use entity names rather than GlobalIds for objects",
|
||||
default=True)
|
||||
process_relations = BoolProperty(name="Process relations",
|
||||
description="Convert containment and aggregation relations to parenting",
|
||||
description="Convert containment and aggregation" \
|
||||
" relations to parenting",
|
||||
default=True)
|
||||
|
||||
def execute(self, context):
|
||||
from . import IfcImport
|
||||
if not IfcImport.Init(self.filepath):
|
||||
return {'FINISHED'}
|
||||
id_to_object = {}
|
||||
id_to_parent = {}
|
||||
old_progress = -1
|
||||
print ("Creating geometry...")
|
||||
while True:
|
||||
ob = IfcImport.Get()
|
||||
if ob.type != 'IfcOpeningElement':
|
||||
f = ob.mesh.faces
|
||||
v = ob.mesh.verts
|
||||
m = ob.matrix
|
||||
t = ob.type[0:21]
|
||||
nm = ob.name if len(ob.name) and self.use_names else ob.guid
|
||||
|
||||
verts = [[v[i], v[i + 1], v[i + 2]] \
|
||||
for i in range(0, len(v), 3)]
|
||||
faces = [[f[i], f[i + 1], f[i + 2]] \
|
||||
for i in range(0, len(f), 3)]
|
||||
|
||||
me = bpy.data.meshes.new('mesh%d' % ob.mesh.id)
|
||||
me.from_pydata(verts, [], faces)
|
||||
if t in bpy.data.materials:
|
||||
mat = bpy.data.materials[t]
|
||||
mat.use_fake_user = True
|
||||
else:
|
||||
mat = bpy.data.materials.new(t)
|
||||
me.materials.append(mat)
|
||||
|
||||
bob = bpy.data.objects.new(nm, me)
|
||||
bob.matrix_world = mathutils.Matrix(([m[0], m[1], m[2], 0],
|
||||
[m[3], m[4], m[5], 0],
|
||||
[m[6], m[7], m[8], 0],
|
||||
[m[9], m[10], m[11], 1]))
|
||||
bpy.context.scene.objects.link(bob)
|
||||
|
||||
bpy.ops.object.select_all(action='DESELECT')
|
||||
bpy.ops.object.select_name(name=bob.name)
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
bpy.ops.mesh.normals_make_consistent()
|
||||
bpy.ops.object.mode_set(mode='OBJECT')
|
||||
|
||||
bob.ifc_id, bob.ifc_guid, bob.ifc_name, bob.ifc_type = \
|
||||
ob.id, ob.guid, ob.name, ob.type
|
||||
|
||||
bob.hide = ob.type == 'IfcSpace'
|
||||
id_to_object[ob.id] = bob
|
||||
|
||||
if ob.parent_id > 0:
|
||||
id_to_parent[ob.id] = ob.parent_id
|
||||
progress = IfcImport.Progress() // 2
|
||||
if progress > old_progress:
|
||||
print ("\r[" + "=" * progress + " " * (50 - progress) + "]",end="")
|
||||
old_progress = progress
|
||||
if not IfcImport.Next():
|
||||
break
|
||||
|
||||
print ("\rDone creating geometry"+" "*30)
|
||||
|
||||
while len(id_to_parent) and self.process_relations:
|
||||
id, parent_id = id_to_parent.popitem()
|
||||
|
||||
if parent_id in id_to_object:
|
||||
bob = id_to_object[parent_id]
|
||||
else:
|
||||
parent_ob = IfcImport.GetObject(parent_id)
|
||||
if parent_ob.id == -1:
|
||||
bob = None
|
||||
else:
|
||||
m = parent_ob.matrix
|
||||
nm = parent_ob.name if len(parent_ob.name) and self.use_names \
|
||||
else parent_ob.guid
|
||||
bob = bpy.data.objects.new(nm, None)
|
||||
bob.matrix_world = mathutils.Matrix(([m[0], m[1], m[2], 0],
|
||||
[m[3], m[4], m[5], 0],
|
||||
[m[6], m[7], m[8], 0],
|
||||
[m[9], m[10], m[11], 1]))
|
||||
bpy.context.scene.objects.link(bob)
|
||||
|
||||
bob.ifc_id = parent_ob.id
|
||||
bob.ifc_guid = parent_ob.guid
|
||||
bob.ifc_name = parent_ob.name
|
||||
bob.ifc_type = parent_ob.type
|
||||
|
||||
if parent_ob.parent_id > 0:
|
||||
id_to_parent[parent_id] = parent_ob.parent_id
|
||||
id_to_object[parent_id] = bob
|
||||
if bob:
|
||||
ob = id_to_object[id]
|
||||
ob.matrix_world = bob.matrix_world.inverted() * ob.matrix_world
|
||||
ob.parent = bob
|
||||
|
||||
IfcImport.CleanUp()
|
||||
import_ifc(self.filepath, self.use_names, self.process_relations)
|
||||
return {'FINISHED'}
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user