Added new nodes: api_WIP, bmesh_to_ifc, create_project and quick_project_setup

Modified nodes: api, create_entity
This commit is contained in:
martinaCodes
2022-08-25 22:01:35 +02:00
parent 0f85890c72
commit acee90fa89
9 changed files with 445 additions and 15 deletions
+21 -10
View File
@@ -26,10 +26,7 @@ from sverchok.node_tree import SverchCustomTreeNode
from sverchok.data_structure import updateNode
def update_usecase(self, context):
module_usecase = self.get_module_usecase()
if module_usecase:
self.generate_node(*module_usecase)
class SvIfcTooltip(bpy.types.Operator):
@@ -46,13 +43,25 @@ class SvIfcTooltip(bpy.types.Operator):
class SvIfcApi(bpy.types.Node, SverchCustomTreeNode, ifcsverchok.helper.SvIfcCore):
def update_usecase(self, context):
try:
module_usecase = self.get_module_usecase()
if module_usecase:
self.generate_node(*module_usecase)
except:
raise Exception(
f"Couldn't run generate_node(). Module usecase: {module_usecase}"
)
bl_idname = "SvIfcApi"
bl_label = "IFC API"
tooltip: StringProperty(name="Tooltip")
usecase: StringProperty(name="Usecase", update=update_usecase)
usecase: StringProperty(update=updateNode)
def sv_init(self, context):
self.inputs.new("SvStringsSocket", "usecase").prop_name = "usecase"
input_socket = self.inputs.new("SvStringsSocket", "usecase").use_prop = True
input_socket.tooltip = "ifcopenshell.api usecase, written like 'module.usecase' \n E.g.: 'project.create_file'"
self.outputs.new("SvVerticesSocket", "file")
def draw_buttons(self, context, layout):
@@ -63,8 +72,9 @@ class SvIfcApi(bpy.types.Node, SverchCustomTreeNode, ifcsverchok.helper.SvIfcCor
print("process")
module_usecase = self.get_module_usecase()
if module_usecase:
self.generate_node(*module_usecase)
self.sv_input_names = [i.name for i in self.inputs]
super().process()
super().process() # super().process() uses zip_long_repeat() which doesn't take objects like bmesh
def get_module_usecase(self):
usecase = self.inputs["usecase"].sv_get()[0][0]
@@ -75,7 +85,7 @@ class SvIfcApi(bpy.types.Node, SverchCustomTreeNode, ifcsverchok.helper.SvIfcCor
try:
node_data = ifcopenshell.api.extract_docs(module, usecase)
except:
print("Node not yet implemented:", module, usecase)
raise Exception("Node not yet implemented:", module, usecase)
return
while len(self.inputs) > 1:
self.inputs.remove(self.inputs[-1])
@@ -83,7 +93,7 @@ class SvIfcApi(bpy.types.Node, SverchCustomTreeNode, ifcsverchok.helper.SvIfcCor
self.tooltip = ""
for name, data in node_data["inputs"].items():
setattr(SvIfcApi, name, StringProperty(name=name, update=updateNode))
self.inputs.new("SvStringsSocket", name).prop_name = name
self.inputs.new("SvStringsSocket", name).use_prop = True
if "default" in data:
self.tooltip = f"{name} ({data['default']}): {data['description']}\n"
else:
@@ -97,6 +107,7 @@ class SvIfcApi(bpy.types.Node, SverchCustomTreeNode, ifcsverchok.helper.SvIfcCor
self.outputs["file"].sv_set([ifcopenshell.api.run(usecase, **settings)])
def register():
bpy.utils.register_class(SvIfcTooltip)
bpy.utils.register_class(SvIfcApi)
@@ -104,4 +115,4 @@ def register():
def unregister():
bpy.utils.unregister_class(SvIfcApi)
bpy.utils.unregister_class(SvIfcTooltip)
bpy.utils.unregister_class(SvIfcTooltip)