New IFC create shape node for Sverchok

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"),

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)

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)