From 8b982df1c0875a3f25a7e6a584d89352d609f1b4 Mon Sep 17 00:00:00 2001
From: Dion Moult <dion@thinkmoult.com>
Date: Sat, 10 Oct 2020 10:21:17 +1100
Subject: [PATCH] New IFC create shape node for Sverchok

---
 src/ifcsverchok/__init__.py                   |  1 +
 src/ifcsverchok/nodes/ifc/create_shape.py     | 65 +++++++++++++++++++
 .../nodes/ifc/select_blender_objects.py       | 19 ++++++
 3 files changed, 85 insertions(+)
 create mode 100644 src/ifcsverchok/nodes/ifc/create_shape.py

diff --git a/src/ifcsverchok/__init__.py b/src/ifcsverchok/__init__.py
index 4ac86a8f24..22514f885c 100644
--- a/src/ifcsverchok/__init__.py
+++ b/src/ifcsverchok/__init__.py
@@ -26,6 +26,7 @@ def nodes_index():
         ("ifc.read_file", "SvIfcReadFile"),
         ("ifc.write_file", "SvIfcWriteFile"),
         ("ifc.create_entity", "SvIfcCreateEntity"),
+        ("ifc.create_shape", "SvIfcCreateShape"),
         ("ifc.by_id", "SvIfcById"),
         ("ifc.by_guid", "SvIfcByGuid"),
         ("ifc.by_type", "SvIfcByType"),
diff --git a/src/ifcsverchok/nodes/ifc/create_shape.py b/src/ifcsverchok/nodes/ifc/create_shape.py
new file mode 100644
index 0000000000..874635a552
--- /dev/null
+++ b/src/ifcsverchok/nodes/ifc/create_shape.py
@@ -0,0 +1,65 @@
+import bpy
+import logging
+import ifcopenshell
+import ifcsverchok.helper
+import blenderbim.bim.import_ifc
+from bpy.props import StringProperty
+from sverchok.node_tree import SverchCustomTreeNode
+from sverchok.data_structure import updateNode
+
+
+class SvIfcCreateShapeRefresh(bpy.types.Operator):
+    bl_idname = "node.sv_ifc_create_shape_refresh"
+    bl_label = "IFC Create Shape Refresh"
+    bl_options = {'UNDO'}
+
+    idtree: StringProperty(default='')
+    idname: StringProperty(default='')
+
+    def execute(self, context):
+        node = bpy.data.node_groups[self.idtree].nodes[self.idname]
+        node.process()
+        return {'FINISHED'}
+
+
+class SvIfcCreateShape(bpy.types.Node, SverchCustomTreeNode, ifcsverchok.helper.SvIfcCore):
+    bl_idname = 'SvIfcCreateShape'
+    bl_label = 'IFC Create Shape'
+    file: StringProperty(name='file', update=updateNode)
+    entity: StringProperty(name='entity', update=updateNode)
+
+    def sv_init(self, context):
+        self.inputs.new('SvStringsSocket', 'file').prop_name = 'file'
+        self.inputs.new('SvStringsSocket', 'entity').prop_name = 'entity'
+
+    def draw_buttons(self, context, layout):
+        self.wrapper_tracked_ui_draw_op(layout, 'node.sv_ifc_create_shape_refresh', icon='FILE_REFRESH', text='Refresh')
+
+    def process(self):
+        self.sv_input_names = ['file', 'entity']
+        super().process()
+
+    def process_ifc(self, file, entity):
+        try:
+            if not entity.is_a('IfcProduct'):
+                return
+            logger = logging.getLogger('ImportIFC')
+            self.ifc_import_settings = blenderbim.bim.import_ifc.IfcImportSettings.factory(bpy.context, '', logger)
+            settings = ifcopenshell.geom.settings()
+            shape = ifcopenshell.geom.create_shape(settings, entity)
+            ifc_importer = blenderbim.bim.import_ifc.IfcImporter(self.ifc_import_settings)
+            ifc_importer.file = file
+            mesh = ifc_importer.create_mesh(entity, shape)
+            obj = bpy.data.objects.new('IFC Element', mesh)
+            bpy.context.scene.collection.objects.link(obj)
+        except:
+            print('Entity could not be converted into a shape', entity)
+
+
+def register():
+    bpy.utils.register_class(SvIfcCreateShapeRefresh)
+    bpy.utils.register_class(SvIfcCreateShape)
+
+def unregister():
+    bpy.utils.unregister_class(SvIfcCreateShape)
+    bpy.utils.unregister_class(SvIfcCreateShapeRefresh)
diff --git a/src/ifcsverchok/nodes/ifc/select_blender_objects.py b/src/ifcsverchok/nodes/ifc/select_blender_objects.py
index abac004cbf..86acf67e99 100644
--- a/src/ifcsverchok/nodes/ifc/select_blender_objects.py
+++ b/src/ifcsverchok/nodes/ifc/select_blender_objects.py
@@ -7,6 +7,20 @@ from sverchok.node_tree import SverchCustomTreeNode
 from sverchok.data_structure import updateNode
 
 
+class SvIfcSelectBlenderObjectsRefresh(bpy.types.Operator):
+    bl_idname = "node.sv_ifc_select_blender_objects_refresh"
+    bl_label = "IFC Select Blender Objects Refresh"
+    bl_options = {'UNDO'}
+
+    idtree: StringProperty(default='')
+    idname: StringProperty(default='')
+
+    def execute(self, context):
+        node = bpy.data.node_groups[self.idtree].nodes[self.idname]
+        node.process()
+        return {'FINISHED'}
+
+
 class SvIfcSelectBlenderObjects(bpy.types.Node, SverchCustomTreeNode, ifcsverchok.helper.SvIfcCore):
     bl_idname = 'SvIfcSelectBlenderObjects'
     bl_label = 'IFC Select Blender Objects'
@@ -16,6 +30,9 @@ class SvIfcSelectBlenderObjects(bpy.types.Node, SverchCustomTreeNode, ifcsvercho
     def sv_init(self, context):
         self.inputs.new('SvStringsSocket', 'entities').prop_name = 'entities'
 
+    def draw_buttons(self, context, layout):
+        self.wrapper_tracked_ui_draw_op(layout, 'node.sv_ifc_select_blender_objects_refresh', icon='FILE_REFRESH', text='Refresh')
+
     def process(self):
         self.sv_input_names = ['entities']
         self.guids = []
@@ -31,7 +48,9 @@ class SvIfcSelectBlenderObjects(bpy.types.Node, SverchCustomTreeNode, ifcsvercho
 
 
 def register():
+    bpy.utils.register_class(SvIfcSelectBlenderObjectsRefresh)
     bpy.utils.register_class(SvIfcSelectBlenderObjects)
 
 def unregister():
     bpy.utils.unregister_class(SvIfcSelectBlenderObjects)
+    bpy.utils.unregister_class(SvIfcSelectBlenderObjectsRefresh)