Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-09-19 14:41:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Properly detect all possible representation contexts for an object. See #756.

Browse Source
This commit is contained in:
Dion Moult
2020-01-18 12:47:55 +11:00
parent 368f76b713
commit 3fb268f687
1 changed files with 34 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcblenderexport/blenderbim/export_ifc.py
+34 -24
View File
@@ -694,19 +694,8 @@ class IfcParser():
def load_product_representations(self, product): def load_product_representations(self, product):
obj = product['raw'] obj = product['raw']
if obj.data and obj.data.name in self.representations: if obj.data and obj.data.name in self.representations:
return return
self.append_default_representation(obj)
if self.is_point_cloud(obj):
self.append_point_cloud_representation(obj)
elif self.is_structural(obj):
self.append_structural_reference_representation(obj)
elif obj.type == 'CURVE':
self.append_curve_axis_representation(obj)
self.append_representation_per_context(obj) self.append_representation_per_context(obj)
def is_point_cloud(self, obj): def is_point_cloud(self, obj):
@@ -717,9 +706,8 @@ class IfcParser():
return 'IfcStructural' in obj.name return 'IfcStructural' in obj.name
def append_default_representation(self, obj): def append_default_representation(self, obj):
if obj.data and not self.is_mesh_context_sensitive(obj.data.name): self.representations['Model/Body/MODEL_VIEW/{}'.format(obj.data.name)] = self.get_representation(
self.representations['Model/Body/MODEL_VIEW/{}'.format(obj.data.name)] = self.get_representation( obj.data, obj, 'Model', 'Body', 'MODEL_VIEW')
obj.data, obj, 'Model', 'Body', 'MODEL_VIEW')
def append_point_cloud_representation(self, obj): def append_point_cloud_representation(self, obj):
self.representations['Model/Body/MODEL_VIEW/{}'.format(obj.name)] = self.get_representation( self.representations['Model/Body/MODEL_VIEW/{}'.format(obj.name)] = self.get_representation(
@@ -738,19 +726,41 @@ class IfcParser():
obj.data, obj, 'Model', 'Reference', 'GRAPH_VIEW') obj.data, obj, 'Model', 'Reference', 'GRAPH_VIEW')
def append_representation_per_context(self, obj): def append_representation_per_context(self, obj):
if not obj.data: if obj.data:
return name = self.get_ifc_representation_name(obj.data.name)
name = self.get_ifc_representation_name(obj.data.name) else:
name = obj.name
for context in self.ifc_export_settings.context_tree: for context in self.ifc_export_settings.context_tree:
for subcontext in context['subcontexts']: for subcontext in context['subcontexts']:
for target_view in subcontext['target_views']: for target_view in subcontext['target_views']:
mesh_name = '/'.join([context['name'], subcontext['name'], target_view, name]) self.append_representation_in_context(obj, context['name'], subcontext['name'], target_view, name)
mesh = bpy.data.meshes.get(mesh_name)
if not mesh: def append_representation_in_context(self, obj, context, subcontext, target_view, name):
mesh = bpy.data.curves.get(mesh_name) context_prefix = '/'.join([context, subcontext, target_view])
if mesh: mesh_name = '/'.join([context_prefix, name])
self.representations[mesh_name] = self.get_representation( mesh = self.search_for_mesh_or_curve_data(mesh_name)
mesh, obj, context['name'], subcontext['name'], target_view) if mesh:
self.representations[mesh_name] = self.get_representation(
mesh, obj, context, subcontext, target_view)
elif context_prefix == 'Model/Body/MODEL_VIEW' \
and obj.data \
and not self.is_mesh_context_sensitive(obj.data.name):
self.append_default_representation(obj)
elif context_prefix == 'Model/Body/MODEL_VIEW' \
and self.is_point_cloud(obj):
self.append_point_cloud_representation(obj)
elif context_prefix == 'Model/Reference/GRAPH_VIEW' \
and self.is_structural(obj):
self.append_structural_reference_representation(obj)
elif context_prefix == 'Model/Axis/GRAPH_VIEW' \
and obj.type == 'CURVE':
self.append_curve_axis_representation(obj)
def search_for_mesh_or_curve_data(self, name):
data = bpy.data.meshes.get(name)
if not data:
data = bpy.data.curves.get(name)
return data
def get_representation(self, mesh, obj, context, subcontext, target_view): def get_representation(self, mesh, obj, context, subcontext, target_view):
return { return {