Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-09-26 10:11:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add support for plan representations and all target views

Browse Source
This commit is contained in:
Dion Moult
2019-11-07 21:49:14 +11:00
parent 609ef5f3a7
commit ee7956d921
1 changed files with 74 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcblenderexport/io_export_ifc/export_ifc.py
+74 -57
View File
@@ -539,40 +539,34 @@ class IfcParser():
self.append_default_representation(object, results) self.append_default_representation(object, results)
self.append_representation_per_context(object, results) self.append_representation_per_context(object, results)
self.append_generated_representation_per_context(object, results)
return results return results
def append_default_representation(self, object, results): def append_default_representation(self, object, results):
if not self.is_mesh_context_sensitive(object.data.name): if not self.is_mesh_context_sensitive(object.data.name):
results['Model/Body/{}'.format(object.data.name)] = self.get_representation( results['Model/Body/MODEL_VIEW/{}'.format(object.data.name)] = self.get_representation(
object.data, object, 'Model', 'Body') object.data, object, 'Model', 'Body', 'MODEL_VIEW')
def append_representation_per_context(self, object, results): def append_representation_per_context(self, object, results):
context = self.get_ifc_context(object.data.name)
name = self.get_ifc_representation_name(object.data.name) name = self.get_ifc_representation_name(object.data.name)
for subcontext in self.ifc_export_settings.subcontexts: for context in self.ifc_export_settings.context_tree:
try: for subcontext in context['subcontexts']:
mesh = bpy.data.meshes['/'.join([context, subcontext, name])] for target_view in subcontext['target_views']:
except: mesh_name = '/'.join([context['name'], subcontext['name'], target_view, name])
continue try:
results[mesh.name] = self.get_representation( mesh = bpy.data.meshes[mesh_name]
mesh, object, context, subcontext) except:
continue
results[mesh_name] = self.get_representation(
mesh, object, context['name'], subcontext['name'], target_view)
def append_generated_representation_per_context(self, object, results): def get_representation(self, mesh, object, context, subcontext, target_view, is_generated=False):
context = self.get_ifc_context(object.data.name)
name = self.get_ifc_representation_name(object.data.name)
for subcontext in self.ifc_export_settings.generated_subcontexts:
representation = self.get_representation(
object.data, object, context, subcontext, is_generated=True)
results['/'.join([context, subcontext, name])] = representation
def get_representation(self, mesh, object, context, subcontext, is_generated=False):
return { return {
'ifc': None, 'ifc': None,
'raw': mesh, 'raw': mesh,
'raw_object': object, 'raw_object': object,
'context': context, 'context': context,
'subcontext': subcontext, 'subcontext': subcontext,
'target_view': target_view,
'is_curve': isinstance(mesh, bpy.types.Curve), 'is_curve': isinstance(mesh, bpy.types.Curve),
'is_wireframe': mesh.BIMMeshProperties.is_wireframe if hasattr(mesh, 'BIMMeshProperties') else False, 'is_wireframe': mesh.BIMMeshProperties.is_wireframe if hasattr(mesh, 'BIMMeshProperties') else False,
'is_swept_solid': mesh.BIMMeshProperties.is_swept_solid if hasattr(mesh, 'BIMMeshProperties') else False, 'is_swept_solid': mesh.BIMMeshProperties.is_swept_solid if hasattr(mesh, 'BIMMeshProperties') else False,
@@ -583,14 +577,9 @@ class IfcParser():
def is_mesh_context_sensitive(self, name): def is_mesh_context_sensitive(self, name):
return '/' in name return '/' in name
def get_ifc_context(self, name):
if self.is_mesh_context_sensitive(name):
return name.split('/')[0]
return 'Model'
def get_ifc_representation_name(self, name): def get_ifc_representation_name(self, name):
if self.is_mesh_context_sensitive(name): if self.is_mesh_context_sensitive(name):
return name.split('/')[2] return name.split('/')[3]
return name return name
def get_materials(self): def get_materials(self):
@@ -706,18 +695,17 @@ class IfcParser():
if not object.data: if not object.data:
return names return names
if not self.is_mesh_context_sensitive(object.data.name): if not self.is_mesh_context_sensitive(object.data.name):
names.append('Model/Body/{}'.format(object.data.name)) names.append('Model/Body/MODEL_VIEW/{}'.format(object.data.name))
for subcontext in self.ifc_export_settings.subcontexts: name = self.get_ifc_representation_name(object.data.name)
try: for context in self.ifc_export_settings.context_tree:
mesh = bpy.data.meshes['/'.join([ for subcontext in context['subcontexts']:
self.get_ifc_context(object.data.name), for target_view in subcontext['target_views']:
subcontext, mesh_name = '/'.join([context['name'], subcontext['name'], target_view, name])
self.get_ifc_representation_name(object.data.name)])] try:
except: mesh = bpy.data.meshes[mesh_name]
continue except:
names.append(mesh.name) continue
for subcontext in self.ifc_export_settings.generated_subcontexts: names.append(mesh_name)
names.append('Model/{}/{}'.format(subcontext, self.get_ifc_representation_name(object.data.name)))
return names return names
def get_spatial_structure_elements_tree(self, collections, name_filter): def get_spatial_structure_elements_tree(self, collections, name_filter):
@@ -970,17 +958,18 @@ class IfcExporter():
self.ifc_rep_context['Model'] = { self.ifc_rep_context['Model'] = {
'ifc': self.file.createIfcGeometricRepresentationContext( 'ifc': self.file.createIfcGeometricRepresentationContext(
None, 'Model', 3, 1.0E-05, self.origin)} None, 'Model', 3, 1.0E-05, self.origin)}
# TODO Make optional if 'Plan' in self.ifc_export_settings.contexts:
self.ifc_rep_context['Plan'] = { self.ifc_rep_context['Plan'] = {
'ifc': self.file.createIfcGeometricRepresentationContext( 'ifc': self.file.createIfcGeometricRepresentationContext(
None, 'Plan', 2, 1.0E-05, self.origin)} None, 'Plan', 2, 1.0E-05, self.origin)}
for subcontext in (self.ifc_export_settings.subcontexts + for context in self.ifc_export_settings.context_tree:
self.ifc_export_settings.generated_subcontexts): for subcontext in context['subcontexts']:
self.ifc_rep_context['Model'][subcontext] = { self.ifc_rep_context[context['name']][subcontext['name']] = {}
'ifc': self.file.createIfcGeometricRepresentationSubContext( for target_view in subcontext['target_views']:
subcontext, 'Model', self.ifc_rep_context[context['name']][subcontext['name']][target_view] = {
None, None, None, None, 'ifc': self.file.createIfcGeometricRepresentationSubContext(
self.ifc_rep_context['Model']['ifc'], None, 'MODEL_VIEW', None)} subcontext['name'], context['name'], None, None, None, None,
self.ifc_rep_context[context['name']]['ifc'], None, target_view, None)}
def create_project(self): def create_project(self):
self.ifc_parser.project['attributes'].update({ self.ifc_parser.project['attributes'].update({
@@ -1119,7 +1108,7 @@ class IfcExporter():
for material in self.ifc_parser.materials.values(): for material in self.ifc_parser.materials.values():
styled_item = self.create_styled_item(material) styled_item = self.create_styled_item(material)
styled_representation = self.file.createIfcStyledRepresentation( styled_representation = self.file.createIfcStyledRepresentation(
self.ifc_rep_context['Model']['Body']['ifc'], None, None, [styled_item]) self.ifc_rep_context['Model']['Body']['MODEL_VIEW']['ifc'], None, None, [styled_item])
material['ifc'] = self.file.createIfcMaterial(material['raw'].name, None, None) material['ifc'] = self.file.createIfcMaterial(material['raw'].name, None, None)
self.file.createIfcMaterialDefinitionRepresentation( self.file.createIfcMaterialDefinitionRepresentation(
material['raw'].name, None, [styled_representation], material['ifc']) material['raw'].name, None, [styled_representation], material['ifc'])
@@ -1283,7 +1272,7 @@ class IfcExporter():
object.bound_box[0][0], object.bound_box[0][1], object.bound_box[0][2]), object.bound_box[0][0], object.bound_box[0][1], object.bound_box[0][2]),
object.dimensions[0], object.dimensions[1], object.dimensions[2]) object.dimensions[0], object.dimensions[1], object.dimensions[2])
return self.file.createIfcShapeRepresentation( return self.file.createIfcShapeRepresentation(
self.ifc_rep_context[representation['context']][representation['subcontext']]['ifc'], self.ifc_rep_context[representation['context']][representation['subcontext']][representation['target_view']]['ifc'],
representation['subcontext'], 'BoundingBox', [bounding_box]) representation['subcontext'], 'BoundingBox', [bounding_box])
def create_cog_representation(self, representation): def create_cog_representation(self, representation):
@@ -1291,7 +1280,7 @@ class IfcExporter():
cog = self.create_cartesian_point( cog = self.create_cartesian_point(
mesh.vertices[0].co.x, mesh.vertices[0].co.y, mesh.vertices[0].co.z) mesh.vertices[0].co.x, mesh.vertices[0].co.y, mesh.vertices[0].co.z)
return self.file.createIfcShapeRepresentation( return self.file.createIfcShapeRepresentation(
self.ifc_rep_context[representation['context']][representation['subcontext']]['ifc'], self.ifc_rep_context[representation['context']][representation['subcontext']][representation['target_view']]['ifc'],
representation['subcontext'], 'BoundingBox', [cog]) representation['subcontext'], 'BoundingBox', [cog])
def create_wireframe_representation(self, representation): def create_wireframe_representation(self, representation):
@@ -1301,7 +1290,7 @@ class IfcExporter():
self.ifc_edges.append(self.file.createIfcPolyline([ self.ifc_edges.append(self.file.createIfcPolyline([
self.ifc_vertices[v] for v in edge.vertices])) self.ifc_vertices[v] for v in edge.vertices]))
return self.file.createIfcShapeRepresentation( return self.file.createIfcShapeRepresentation(
self.ifc_rep_context[representation['context']][representation['subcontext']]['ifc'], self.ifc_rep_context[representation['context']][representation['subcontext']][representation['target_view']]['ifc'],
representation['subcontext'], 'Curve', representation['subcontext'], 'Curve',
self.ifc_edges) self.ifc_edges)
@@ -1323,7 +1312,7 @@ class IfcExporter():
self.ifc_vertices[i] for i in loop_vertex_indices])) self.ifc_vertices[i] for i in loop_vertex_indices]))
geometric_curve_set = self.file.createIfcGeometricCurveSet(loops) geometric_curve_set = self.file.createIfcGeometricCurveSet(loops)
return self.file.createIfcShapeRepresentation( return self.file.createIfcShapeRepresentation(
self.ifc_rep_context[representation['context']][representation['subcontext']]['ifc'], self.ifc_rep_context[representation['context']][representation['subcontext']][representation['target_view']]['ifc'],
representation['subcontext'], 'GeometricCurveSet', [geometric_curve_set]) representation['subcontext'], 'GeometricCurveSet', [geometric_curve_set])
# https://medium.com/@behreajj/scripting-curves-in-blender-with-python-c487097efd13 # https://medium.com/@behreajj/scripting-curves-in-blender-with-python-c487097efd13
@@ -1377,7 +1366,7 @@ class IfcExporter():
# swept_area, self.origin, self.create_curve(representation['raw']), # swept_area, self.origin, self.create_curve(representation['raw']),
# 0., 1., self.file.createIfcDirection((0.0, -1.0, 0.0))) # 0., 1., self.file.createIfcDirection((0.0, -1.0, 0.0)))
return self.file.createIfcShapeRepresentation( return self.file.createIfcShapeRepresentation(
self.ifc_rep_context[representation['context']][representation['subcontext']]['ifc'], self.ifc_rep_context[representation['context']][representation['subcontext']][representation['target_view']]['ifc'],
representation['subcontext'], 'AdvancedSweptSolid', representation['subcontext'], 'AdvancedSweptSolid',
swept_area_solids) swept_area_solids)
@@ -1420,7 +1409,7 @@ class IfcExporter():
self.file.createIfcDirection((unit_direction.x, unit_direction.y, unit_direction.z)), self.file.createIfcDirection((unit_direction.x, unit_direction.y, unit_direction.z)),
self.convert_si_to_unit(direction.length)) self.convert_si_to_unit(direction.length))
return self.file.createIfcShapeRepresentation( return self.file.createIfcShapeRepresentation(
self.ifc_rep_context[representation['context']][representation['subcontext']]['ifc'], self.ifc_rep_context[representation['context']][representation['subcontext']][representation['target_view']]['ifc'],
representation['subcontext'], 'SweptSolid', representation['subcontext'], 'SweptSolid',
[extruded_area_solid]) [extruded_area_solid])
@@ -1482,7 +1471,7 @@ class IfcExporter():
self.file.createIfcPolyLoop([self.ifc_vertices[vertice] for vertice in polygon.vertices]), self.file.createIfcPolyLoop([self.ifc_vertices[vertice] for vertice in polygon.vertices]),
True)])) True)]))
return self.file.createIfcShapeRepresentation( return self.file.createIfcShapeRepresentation(
self.ifc_rep_context[representation['context']][representation['subcontext']]['ifc'], self.ifc_rep_context[representation['context']][representation['subcontext']][representation['target_view']]['ifc'],
representation['subcontext'], 'Brep', representation['subcontext'], 'Brep',
[self.file.createIfcFacetedBrep(self.file.createIfcClosedShell(self.ifc_faces))]) [self.file.createIfcFacetedBrep(self.file.createIfcClosedShell(self.ifc_faces))])
@@ -1633,5 +1622,33 @@ class IfcExportSettings:
self.output_file = None self.output_file = None
self.has_representations = True self.has_representations = True
self.has_quantities = True self.has_quantities = True
self.contexts = ['Model', 'Plan']
self.subcontexts = ['Axis', 'FootPrint', 'Reference', 'Body', 'Clearance', 'CoG', 'SurveyPoints'] self.subcontexts = ['Axis', 'FootPrint', 'Reference', 'Body', 'Clearance', 'CoG', 'SurveyPoints']
self.generated_subcontexts = ['Box'] self.generated_subcontexts = ['Box']
self.target_views = ['GRAPH_VIEW', 'SKETCH_VIEW', 'MODEL_VIEW', 'PLAN_VIEW', 'REFLECTED_PLAN_VIEW', 'SECTION_VIEW', 'ELEVATION_VIEW', 'USERDEFINED', 'NOTDEFINED']
# TODO make this configurable via UI
self.context_tree = self.build_context_tree()
def build_context_tree(self):
tree = []
for context in self.contexts:
subcontexts = []
for subcontext in self.subcontexts + self.generated_subcontexts:
target_views = []
for target_view in self.target_views:
if context == 'Model' \
and target_view != 'MODEL_VIEW':
continue
elif context == 'Plan' \
and target_view == 'MODEL_VIEW':
continue
target_views.append(target_view)
subcontexts.append({
'name': subcontext,
'target_views': target_views
})
tree.append({
'name': context,
'subcontexts': subcontexts
})
return tree