91 lines
2.6 KiB
Python
91 lines
2.6 KiB
Python
"""
|
|
@author: Avatech Limited
|
|
@title: Avatar Graph
|
|
@nickname: Avatar Graph
|
|
@description: Include nodes for sam + bpy operation, that allows workflow creations for generative 2d character rig.
|
|
"""
|
|
import routes
|
|
|
|
import inspect
|
|
import os
|
|
import sys
|
|
import importlib
|
|
import subprocess
|
|
from folder_paths import add_model_folder_path
|
|
|
|
ag_path = os.path.join(os.path.dirname(__file__))
|
|
|
|
# pip_install = [sys.executable, '-m', 'pip', 'install']
|
|
|
|
# def ensure_pip_packages():
|
|
# try:
|
|
# # import bpy
|
|
# import segment_anything
|
|
# except Exception:
|
|
# my_path = os.path.dirname(__file__)
|
|
# requirements_path = os.path.join(my_path, "requirements.txt")
|
|
# subprocess.check_call(pip_install + ['-r', requirements_path])
|
|
|
|
# ensure_pip_packages()
|
|
|
|
|
|
def get_python_files(path):
|
|
return [f[:-3] for f in os.listdir(path) if f.endswith('.py')]
|
|
|
|
|
|
def append_to_sys_path(path):
|
|
if path not in sys.path:
|
|
sys.path.append(path)
|
|
|
|
|
|
def create_sam_model_dir():
|
|
model_dir = os.path.join(ag_path, "../../models/sam")
|
|
if not os.path.isdir(model_dir):
|
|
os.makedirs(model_dir)
|
|
add_model_folder_path('sam', model_dir)
|
|
|
|
|
|
create_sam_model_dir()
|
|
|
|
paths = ['blender', 'sam', 'common']
|
|
files = []
|
|
|
|
for path in paths:
|
|
full_path = os.path.join(ag_path, path)
|
|
append_to_sys_path(full_path)
|
|
files.extend(get_python_files(full_path))
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
import blender_node
|
|
base_class = blender_node.ObjectOps
|
|
|
|
# Import all the modules and append their mappings
|
|
for file in files:
|
|
module = importlib.import_module(file)
|
|
|
|
for name, obj in inspect.getmembers(module):
|
|
if inspect.isclass(obj):
|
|
if issubclass(obj, base_class) and obj != base_class:
|
|
NODE_CLASS_MAPPINGS.update(obj.NODE_CLASS_MAPPINGS())
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(
|
|
obj.NODE_DISPLAY_NAME_MAPPINGS())
|
|
|
|
if (hasattr(module, 'BLENDER_NODES')):
|
|
for node in module.BLENDER_NODES:
|
|
# print(node)
|
|
# NODE_CLASS_MAPPINGS.update({node: module.BLENDER_NODES[node]})
|
|
# NODE_DISPLAY_NAME_MAPPINGS.update({node: node})
|
|
NODE_CLASS_MAPPINGS.update(node.NODE_CLASS_MAPPINGS())
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(
|
|
node.NODE_DISPLAY_NAME_MAPPINGS())
|
|
|
|
if hasattr(module, 'NODE_CLASS_MAPPINGS'):
|
|
NODE_CLASS_MAPPINGS.update(module.NODE_CLASS_MAPPINGS)
|
|
if hasattr(module, 'NODE_DISPLAY_NAME_MAPPINGS'):
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(module.NODE_DISPLAY_NAME_MAPPINGS)
|
|
|
|
WEB_DIRECTORY = "js"
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|