Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-08-10 01:41:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bundle and implement basic UI with addon

Browse Source
This commit is contained in:
Dion Moult
2019-10-05 21:15:53 +10:00
parent 90fee4ef7f
commit 824b30b3da
6 changed files with 143 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcblenderexport/bimui.py
-78
View File
@@ -1,78 +0,0 @@
import bpy
import json
class IfcSchema():
def __init__(self):
with open('/home/dion/Projects/IfcOpenShell/src/ifcblenderexport/IfcElement.json') as f:
self.elements = json.load(f)
ifc_schema = IfcSchema()
class BIMProperties(bpy.types.PropertyGroup):
def getIfcClasses(self, context):
return [(e, e, '') for e in ifc_schema.elements]
def getIfcPredefinedTypes(self, context):
results = []
for name, data in ifc_schema.elements.items():
if name != bpy.context.scene.BIMProperties.ifc_class:
continue
for attribute in data['attributes']:
if attribute['name'] != 'PredefinedType':
continue
return [(e, e, '') for e in attribute['enum_values']]
data_dir: bpy.props.StringProperty(name="Data Directory")
ifc_class: bpy.props.EnumProperty(items = getIfcClasses, name="Class")
ifc_predefined_type: bpy.props.EnumProperty(
items = getIfcPredefinedTypes,
name="Predefined Type", default=None)
ifc_userdefined_type: bpy.props.StringProperty(name="Userdefined Type")
class BIMPanel(bpy.types.Panel):
"""Creates a Panel in the scene context of the properties editor"""
bl_label = "Building Information Modeling"
bl_idname = "SCENE_PT_layout"
bl_space_type = 'PROPERTIES'
bl_region_type = 'WINDOW'
bl_context = "scene"
def draw(self, context):
layout = self.layout
scene = context.scene
bim_properties = bpy.context.scene.BIMProperties
layout.label(text="System Setup:")
row = layout.row()
row.prop(bim_properties, "data_dir")
layout.label(text="IFC Categorisation:")
row = layout.row()
row.prop(bim_properties, "ifc_class")
row = layout.row()
row.prop(bim_properties, "ifc_predefined_type")
row = layout.row()
row.prop(bim_properties, "ifc_userdefined_type")
row = layout.row()
row.operator("render.render")
classes = (
BIMProperties,
BIMPanel
)
def register():
for cls in classes:
bpy.utils.register_class(cls)
bpy.types.Scene.BIMProperties = bpy.props.PointerProperty(type=BIMProperties)
def unregister():
for cls in classes:
bpy.utils.unregister_class(cls)
del(bpy.types.Scene.BIMProperties)
if __name__ == "__main__":
register()
src/ifcblenderexport/io_export_ifc/__init__.py
+14 -4
View File
@@ -17,8 +17,8 @@ if "bpy" in locals():
import bpy
import time
from . import export
from . import ui
import os
from bpy.props import (StringProperty,)
cwd = os.path.dirname(os.path.realpath(__file__)) + os.path.sep
@@ -26,7 +26,7 @@ class ExportIFC(bpy.types.Operator):
bl_idname = "export.ifc"
bl_label = "Export .ifc file"
filename_ext = ".ifc"
filepath: StringProperty(subtype='FILE_PATH')
filepath: bpy.props.StringProperty(subtype='FILE_PATH')
def invoke(self, context, event):
if not self.filepath:
@@ -39,7 +39,8 @@ class ExportIFC(bpy.types.Operator):
print('# Starting export')
start = time.time()
ifc_export_settings = export.IfcExportSettings()
ifc_export_settings.bim_path = cwd
ifc_export_settings.data_dir = bpy.context.scene.BIMProperties.data_dir
ifc_export_settings.schema_dir = bpy.context.scene.BIMProperties.schema_dir
ifc_export_settings.output_file = bpy.path.ensure_ext(self.filepath, '.ifc')
ifc_parser = export.IfcParser(ifc_export_settings)
ifc_schema = export.IfcSchema(ifc_export_settings)
@@ -53,17 +54,26 @@ def menu_func(self, context):
self.layout.operator(ExportIFC.bl_idname,
text="Industry Foundation Classes (.ifc)")
classes = (ExportIFC,)
classes = (
ui.BIMOpAssignClass,
ui.BIMOpSelectDataDir,
ui.BIMOpSelectSchemaDir,
ui.BIMProperties,
ui.BIMPanel,
ExportIFC,
)
def register():
for cls in classes:
bpy.utils.register_class(cls)
bpy.types.TOPBAR_MT_file_export.append(menu_func)
bpy.types.Scene.BIMProperties = bpy.props.PointerProperty(type=ui.BIMProperties)
def unregister():
for cls in reversed(classes):
bpy.utils.unregister_class(cls)
bpy.types.TOPBAR_MT_file_export.remove(menu_func)
del(bpy.types.Scene.BIMProperties)
if __name__ == "__main__":
register()
src/ifcblenderexport/io_export_ifc/export.py
+5 -4
View File
@@ -64,7 +64,7 @@ class QtoCalculator():
class IfcSchema():
def __init__(self, ifc_export_settings):
self.schema_dir = '{}schema/'.format(ifc_export_settings.bim_path)
self.schema_dir = ifc_export_settings.schema_dir
self.property_file = ifcopenshell.open(self.schema_dir + 'IFC4_ADD2.ifc')
self.psets = {}
self.qtos = {}
@@ -84,7 +84,7 @@ class IfcSchema():
class IfcParser():
def __init__(self, ifc_export_settings):
self.data_dir = '{}data/'.format(ifc_export_settings.bim_path)
self.data_dir = ifc_export_settings.data_dir
self.ifc_export_settings = ifc_export_settings
@@ -746,7 +746,7 @@ class SIUnitHelper:
class IfcExporter():
def __init__(self, ifc_export_settings, ifc_schema, ifc_parser, qto_calculator):
self.template_file = '{}template.ifc'.format(ifc_export_settings.bim_path)
self.template_file = '{}template.ifc'.format(ifc_export_settings.schema_dir)
self.output_file = ifc_export_settings.output_file
self.ifc_export_settings = ifc_export_settings
self.ifc_schema = ifc_schema
@@ -1414,7 +1414,8 @@ class IfcExporter():
class IfcExportSettings:
def __init__(self):
self.bim_path = None
self.schema_dir = None
self.data_dir = None
self.output_file = None
self.has_representations = True
self.has_quantities = True
src/ifcblenderexport/IfcElement.json → src/ifcblenderexport/io_export_ifc/schema/ifc_elements_IFC4.json
View File
src/ifcblenderexport/io_export_ifc/template.ifc → src/ifcblenderexport/io_export_ifc/schema/template.ifc
View File
src/ifcblenderexport/io_export_ifc/ui.py
+124
View File
@@ -0,0 +1,124 @@
import bpy
import json
import os
cwd = os.path.dirname(os.path.realpath(__file__)) + os.path.sep
class BIMOpAssignClass(bpy.types.Operator):
bl_idname = 'bim.assign_class'
bl_label = 'Assign IFC Class'
def execute(self, context):
for object in bpy.context.selected_objects:
existing_class = None
if '/' in object.name \
and object.name[0:3] == 'Ifc':
existing_class = object.name.split('/')[0]
if existing_class:
object.name = '{}/{}'.format(
bpy.context.scene.BIMProperties.ifc_class,
object.name.split('/')[1])
else:
object.name = '{}/{}'.format(
bpy.context.scene.BIMProperties.ifc_class,
object.name)
if existing_class != bpy.context.scene.BIMProperties.ifc_class \
and 'IfcPredefinedType' in object.keys():
del(object['IfcPredefinedType'])
object['IfcPredefinedType'] = bpy.context.scene.BIMProperties.ifc_predefined_type
if bpy.context.scene.BIMProperties.ifc_predefined_type == 'USERDEFINED':
object['IfcObjectType'] = bpy.context.scene.BIMProperties.ifc_userdefined_type
elif 'IfcObjectType' in object.keys():
del(object['IfcObjectType'])
return {'FINISHED'}
class BIMOpSelectDataDir(bpy.types.Operator):
bl_idname = "bim.select_data_dir"
bl_label = "Select Data Directory"
filepath: bpy.props.StringProperty(subtype="FILE_PATH")
def execute(self, context):
bpy.context.scene.BIMProperties.data_dir = self.filepath
return {'FINISHED'}
def invoke(self, context, event):
context.window_manager.fileselect_add(self)
return {'RUNNING_MODAL'}
class BIMOpSelectSchemaDir(bpy.types.Operator):
bl_idname = "bim.select_schema_dir"
bl_label = "Select Schema Directory"
filepath: bpy.props.StringProperty(subtype="FILE_PATH")
def execute(self, context):
bpy.context.scene.BIMProperties.schema_dir = self.filepath
return {'FINISHED'}
def invoke(self, context, event):
context.window_manager.fileselect_add(self)
return {'RUNNING_MODAL'}
class IfcSchema():
def __init__(self):
with open('{}ifc_elements_IFC4.json'.format(cwd + 'schema/')) as f:
self.elements = json.load(f)
ifc_schema = IfcSchema()
class BIMProperties(bpy.types.PropertyGroup):
def getIfcClasses(self, context):
return [(e, e, '') for e in ifc_schema.elements]
def getIfcPredefinedTypes(self, context):
results = []
for name, data in ifc_schema.elements.items():
if name != bpy.context.scene.BIMProperties.ifc_class:
continue
for attribute in data['attributes']:
if attribute['name'] != 'PredefinedType':
continue
return [(e, e, '') for e in attribute['enum_values']]
schema_dir: bpy.props.StringProperty(default=cwd + 'schema/', name="Schema Directory")
data_dir: bpy.props.StringProperty(default=cwd + 'data/', name="Data Directory")
ifc_class: bpy.props.EnumProperty(items = getIfcClasses, name="Class")
ifc_predefined_type: bpy.props.EnumProperty(
items = getIfcPredefinedTypes,
name="Predefined Type", default=None)
ifc_userdefined_type: bpy.props.StringProperty(name="Userdefined Type")
class BIMPanel(bpy.types.Panel):
bl_label = "Building Information Modeling"
bl_idname = "SCENE_PT_layout"
bl_space_type = 'PROPERTIES'
bl_region_type = 'WINDOW'
bl_context = "scene"
def draw(self, context):
layout = self.layout
scene = context.scene
bim_properties = bpy.context.scene.BIMProperties
layout.label(text="System Setup:")
col = layout.column()
row = col.row(align=True)
row.prop(bim_properties, "schema_dir")
row.operator("bim.select_schema_dir", icon="FILE_FOLDER", text="")
col = layout.column()
row = col.row(align=True)
row.prop(bim_properties, "data_dir")
row.operator("bim.select_data_dir", icon="FILE_FOLDER", text="")
layout.label(text="IFC Categorisation:")
row = layout.row()
row.prop(bim_properties, "ifc_class")
row = layout.row()
row.prop(bim_properties, "ifc_predefined_type")
row = layout.row()
row.prop(bim_properties, "ifc_userdefined_type")
row = layout.row()
row.operator("bim.assign_class")