32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
# Import all node classes
|
|
from .scripts.geekyremb_core import GeekyRemB
|
|
from .scripts.geekyremb_animator import GeekyRemB_Animator
|
|
from .scripts.geekyremb_lightshadow import GeekyRemB_LightShadow
|
|
from .scripts.geekyremb_keyframe_position import GeekyRemB_KeyframePosition
|
|
|
|
# Create node mappings
|
|
NODE_CLASS_MAPPINGS = {
|
|
"GeekyRemB": GeekyRemB,
|
|
"GeekyRemB_Animator": GeekyRemB_Animator,
|
|
"GeekyRemB_LightShadow": GeekyRemB_LightShadow,
|
|
"GeekyRemB_KeyframePosition": GeekyRemB_KeyframePosition
|
|
}
|
|
|
|
# Create display name mappings
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"GeekyRemB": "Geeky RemB",
|
|
"GeekyRemB_Animator": "Geeky RemB Animator",
|
|
"GeekyRemB_LightShadow": "Geeky RemB Light & Shadow",
|
|
"GeekyRemB_KeyframePosition": "Geeky RemB Keyframe Position"
|
|
}
|
|
|
|
# Register custom types for ComfyUI
|
|
custom_output_types = {
|
|
"ANIMATOR": ("ANIMATOR", ),
|
|
"LIGHTSHADOW": ("LIGHTSHADOW", ),
|
|
"KEYFRAME": ("KEYFRAME", )
|
|
}
|
|
|
|
# Export the mappings
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'custom_output_types']
|