mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-09-17 14:02:27 +00:00
Added new nodes: api_WIP, bmesh_to_ifc, create_project and quick_project_setup
Modified nodes: api, create_entity
This commit is contained in:
@@ -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)
|
||||
Reference in New Issue
Block a user