mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-09-26 10:11:46 +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")
|
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):
|
class ImportIFC(bpy.types.Operator, ImportHelper):
|
||||||
bl_idname = "import_scene.ifc"
|
bl_idname = "import_scene.ifc"
|
||||||
bl_label = "Import .ifc file"
|
bl_label = "Import .ifc file"
|
||||||
@@ -70,104 +186,12 @@ class ImportIFC(bpy.types.Operator, ImportHelper):
|
|||||||
description="Use entity names rather than GlobalIds for objects",
|
description="Use entity names rather than GlobalIds for objects",
|
||||||
default=True)
|
default=True)
|
||||||
process_relations = BoolProperty(name="Process relations",
|
process_relations = BoolProperty(name="Process relations",
|
||||||
description="Convert containment and aggregation relations to parenting",
|
description="Convert containment and aggregation" \
|
||||||
|
" relations to parenting",
|
||||||
default=True)
|
default=True)
|
||||||
|
|
||||||
def execute(self, context):
|
def execute(self, context):
|
||||||
from . import IfcImport
|
import_ifc(self.filepath, self.use_names, self.process_relations)
|
||||||
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()
|
|
||||||
return {'FINISHED'}
|
return {'FINISHED'}
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user