mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-09-20 23:36:20 +00:00
Prevent adding a crazy number of material slots, especially in messy IFCs coming from Revit
This commit is contained in:
@@ -31,10 +31,12 @@ class MaterialCreator():
|
|||||||
def __init__(self, ifc_import_settings):
|
def __init__(self, ifc_import_settings):
|
||||||
self.mesh = None
|
self.mesh = None
|
||||||
self.materials = {}
|
self.materials = {}
|
||||||
|
self.current_object_materials = []
|
||||||
self.parsed_meshes = []
|
self.parsed_meshes = []
|
||||||
self.ifc_import_settings = ifc_import_settings
|
self.ifc_import_settings = ifc_import_settings
|
||||||
|
|
||||||
def create(self, element, obj, mesh):
|
def create(self, element, obj, mesh):
|
||||||
|
self.current_object_materials = []
|
||||||
self.obj = obj
|
self.obj = obj
|
||||||
self.mesh = mesh
|
self.mesh = mesh
|
||||||
if (hasattr(element, 'Representation') and not element.Representation) \
|
if (hasattr(element, 'Representation') and not element.Representation) \
|
||||||
@@ -76,7 +78,10 @@ class MaterialCreator():
|
|||||||
|
|
||||||
material_name = self.get_material_name(styled_item)
|
material_name = self.get_material_name(styled_item)
|
||||||
|
|
||||||
if material_name not in self.materials:
|
if material_name in self.current_object_materials:
|
||||||
|
return True
|
||||||
|
|
||||||
|
if material_name not in self.materials.keys():
|
||||||
material = bpy.data.materials.get(material_name)
|
material = bpy.data.materials.get(material_name)
|
||||||
if material:
|
if material:
|
||||||
self.materials[material_name] = material
|
self.materials[material_name] = material
|
||||||
@@ -214,6 +219,7 @@ class MaterialCreator():
|
|||||||
|
|
||||||
def assign_material_to_mesh(self, material, is_styled_item=False):
|
def assign_material_to_mesh(self, material, is_styled_item=False):
|
||||||
self.mesh.materials.append(material)
|
self.mesh.materials.append(material)
|
||||||
|
self.current_object_materials.append(material.name)
|
||||||
if is_styled_item:
|
if is_styled_item:
|
||||||
index = len(self.obj.material_slots) - 1
|
index = len(self.obj.material_slots) - 1
|
||||||
self.obj.material_slots[index].link = 'OBJECT'
|
self.obj.material_slots[index].link = 'OBJECT'
|
||||||
|
|||||||
Reference in New Issue
Block a user