Files
Comfyui-VACE-Tools/mode_select_node.py
2026-02-22 20:09:02 +01:00

40 lines
1.1 KiB
Python

from .nodes import VACE_MODES
class VACEModeSelect:
"""Select a VACE mode by integer index (0-9)."""
@classmethod
def INPUT_TYPES(cls):
mode_list = " | ".join(f"{i}={m}" for i, m in enumerate(VACE_MODES))
return {
"required": {
"index": ("INT", {
"default": 0,
"min": 0,
"max": len(VACE_MODES) - 1,
"step": 1,
"tooltip": f"Mode index: {mode_list}",
}),
},
}
RETURN_TYPES = (VACE_MODES,)
RETURN_NAMES = ("mode",)
FUNCTION = "select"
CATEGORY = "VACE Tools"
DESCRIPTION = "Select a VACE mode by integer index (0-9). Useful when driving mode selection from another node's integer output instead of a dropdown."
def select(self, index):
index = max(0, min(index, len(VACE_MODES) - 1))
return (VACE_MODES[index],)
NODE_CLASS_MAPPINGS = {
"VACEModeSelect": VACEModeSelect,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"VACEModeSelect": "VACE Mode Select",
}