Ma Nan
ce397b35aa
fix: contrast formulation cause leak of high value fix: empty series_id cause fail to read chore: restructure folder chore!: change layer vars BREAKING CHANGE: not support 4.1 any more
148 lines
5.0 KiB
Python
148 lines
5.0 KiB
Python
import bpy
|
|
from pathlib import Path
|
|
from .nodes import AddCustomNode
|
|
|
|
|
|
class CustomNodes():
|
|
def __init__(
|
|
self,
|
|
menu_items,
|
|
nodes_file,
|
|
class_prefix="CUSTOMNODES_MT_NODES",
|
|
root_label='CustomNodes',
|
|
root_icon='NONE'
|
|
) -> None:
|
|
if not Path(nodes_file).is_file():
|
|
raise FileNotFoundError(str(Path(nodes_file).resolve()))
|
|
|
|
self.menu_items = menu_items
|
|
self.nodes_file = str(Path(nodes_file).resolve())
|
|
self.class_prefix = class_prefix
|
|
self.root_label = root_label
|
|
self.root_icon = root_icon
|
|
self.class_prefix = class_prefix
|
|
|
|
menu_classes = []
|
|
self._create_menu_class(
|
|
items=menu_items,
|
|
label=root_label,
|
|
menu_classes=menu_classes
|
|
)
|
|
self.menu_classes = menu_classes
|
|
|
|
idname = f"{class_prefix}_{root_label.replace(' ', '').upper()}"
|
|
|
|
def add_node_menu(self, context):
|
|
if ('GeometryNodeTree' == bpy.context.area.spaces[0].tree_type):
|
|
layout = self.layout
|
|
layout.separator()
|
|
layout.menu(idname, icon=root_icon)
|
|
|
|
self.add_node_menu = add_node_menu
|
|
|
|
def _create_menu_class(self, menu_classes, items, label='CustomNodes', icon=0, idname_namespace=None):
|
|
nodes_file = self.nodes_file
|
|
idname_namespace = idname_namespace or self.class_prefix
|
|
idname = f"{idname_namespace}_{label.replace(' ', '').upper()}"
|
|
|
|
# create submenu class if item is menu.
|
|
for item in items:
|
|
item_items = item.get('items') if item != 'separator' else None
|
|
if item_items:
|
|
menu_class = self._create_menu_class(
|
|
menu_classes=menu_classes,
|
|
items=item_items,
|
|
label=item.get('label') or 'CustomNodes',
|
|
icon=item.get('icon') or 0,
|
|
idname_namespace=idname
|
|
)
|
|
item['menu_class'] = menu_class
|
|
|
|
# create menu class
|
|
class Menu(bpy.types.Menu):
|
|
bl_idname = idname
|
|
bl_label = label
|
|
nodes_file = self.nodes_file
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
for item in items:
|
|
# print(item)
|
|
if item == "separator":
|
|
layout.separator()
|
|
elif item.get('menu_class'):
|
|
layout.menu(
|
|
item.get('menu_class').bl_idname,
|
|
icon=item.get('icon') or 'NONE'
|
|
)
|
|
else:
|
|
op = layout.operator(
|
|
'customnodes.add_custom_node',
|
|
text=item.get('label'),
|
|
icon=item.get('icon') or 'NONE'
|
|
)
|
|
op.nodes_file = nodes_file
|
|
op.node_type = item['node_type']
|
|
op.node_label = item.get('label') or ""
|
|
op.node_description = item.get(
|
|
'node_description') or "Add Custom Node."
|
|
op.node_link = item.get('link') or True
|
|
op.node_callback = item.get('node_callback') or ""
|
|
|
|
menu_classes.append(Menu)
|
|
return Menu
|
|
|
|
def _find_item(self, found_items, menu_items, node_type: str):
|
|
|
|
for item in menu_items:
|
|
if item == 'separator':
|
|
continue
|
|
|
|
if item.get("node_type") == node_type:
|
|
found_items.append(item)
|
|
|
|
item_items = item.get('items')
|
|
if item_items:
|
|
self._find_item(found_items, item_items, node_type)
|
|
|
|
def find_item(self, node_type: str):
|
|
found_items = []
|
|
self._find_item(found_items, self.menu_items, node_type)
|
|
return found_items[0] if len(found_items) > 0 else None
|
|
|
|
def add_node(self, node_group, node_type: str):
|
|
item = self.find_item(node_type)
|
|
op = AddCustomNode()
|
|
op.nodes_file = self.nodes_file
|
|
op.node_type = node_type
|
|
if item:
|
|
op.node_label = item.get('label') or ""
|
|
op.node_link = item.get('link') or True
|
|
op.node_callback = item.get('node_callback') or ""
|
|
else:
|
|
op.node_label = ""
|
|
op.node_link = True
|
|
op.node_callback = ""
|
|
return op.add_node(node_group)
|
|
|
|
def register(self):
|
|
for cls in self.menu_classes:
|
|
bpy.utils.register_class(cls)
|
|
|
|
bpy.types.NODE_MT_add.append(self.add_node_menu)
|
|
bpy.types.Scene.customnodes_node_dir = bpy.props.StringProperty(
|
|
name="Node Directory",
|
|
subtype='DIR_PATH',
|
|
default="//"
|
|
)
|
|
|
|
def unregister(self):
|
|
bpy.types.NODE_MT_add.remove(self.add_node_menu)
|
|
try:
|
|
for cls in reversed(self.menu_classes):
|
|
bpy.utils.unregister_class(cls)
|
|
|
|
except RuntimeError:
|
|
pass
|