feat: improving the user experience

bioxelnodes/bioxelutils/node.py

@@ -0,0 +1,68 @@
+from pathlib import Path
+import shutil
+import bpy
+
+from .common import get_file_prop, set_file_prop
+from ..exceptions import Incompatible, NoFound
+
+from ..constants import NODE_LIB_FILEPATH, VERSION
+
+
+def get_node_group(node_type: str, use_link=True):
+    # unannotate below for local debug in node lib file.
+    # node_group = bpy.data.node_groups[node_type]
+    # return node_group
+
+    if get_file_prop("addon_version") is None:
+        set_file_prop("addon_version", VERSION)
+
+    local_lib_file = None
+    addon_lib_file = NODE_LIB_FILEPATH.as_posix()
+
+    for node_group in bpy.data.node_groups:
+        if node_group.name.startswith("BioxelNodes"):
+            node_group_lib = node_group.library
+            if node_group_lib:
+                abs_filepath = bpy.path.abspath(node_group_lib.filepath)
+                _local_lib_file = Path(abs_filepath).resolve().as_posix()
+                if _local_lib_file != addon_lib_file:
+                    local_lib_file = _local_lib_file
+                    break
+
+    # local lib first
+    lib_file = local_lib_file or addon_lib_file
+    bpy.ops.wm.append('EXEC_DEFAULT',
+                      directory=f"{lib_file}/NodeTree",
+                      filename=node_type,
+                      link=use_link,
+                      use_recursive=True,
+                      do_reuse_local_id=True)
+    node_group = bpy.data.node_groups.get(node_type)
+
+    if node_group is None:
+        raise NoFound('No custom node found')
+
+    return node_group
+
+
+def assign_node_group(node, node_type: str):
+    node.node_tree = bpy.data.node_groups[node_type]
+    node.width = 200.0
+    node.name = node_type
+    return node
+
+
+def add_node_to_graph(node_name: str, node_group, use_link=True):
+    node_type = f"BioxelNodes_{node_name}"
+
+    # Deselect all nodes first
+    for node in node_group.nodes:
+        if node.select:
+            node.select = False
+
+    get_node_group(node_type, use_link)
+    node = node_group.nodes.new("GeometryNodeGroup")
+    assign_node_group(node, node_type)
+
+    node.show_options = False
+    return node