diff --git a/comfy_extras/nodes_constant_values,.py b/comfy_extras/nodes_constant_values,.py new file mode 100644 index 000000000..f5dc30413 --- /dev/null +++ b/comfy_extras/nodes_constant_values,.py @@ -0,0 +1,79 @@ +class _CONSTANT_BASE: + def __init__(self): + pass + + @classmethod + def INPUT_TYPES(s) -> dict: + raise NotImplementedError + + RETURN_TYPES: tuple = () + RETURN_NAMES: tuple = () + + CATEGORY: str = "constants" + + FUNCTION: str = "process" + + OUTPUT_NODE: bool = False + + def process(self, value) -> tuple: + return (value,) + + +class ConstantFloat(_CONSTANT_BASE): + @classmethod + def INPUT_TYPES(s) -> dict: + return { + "required": { + "value": ("FLOAT", {"default": 0.0}) + } + } + + RETURN_TYPES = ("FLOAT",) + RETURN_NAMES = ("float",) + + +class ConstantInteger(_CONSTANT_BASE): + @classmethod + def INPUT_TYPES(s) -> dict: + return { + "required": { + "value": ("INT", {"default": 0}) + } + } + + RETURN_TYPES = ("INT",) + RETURN_NAMES = ("int",) + + +class ConstantString(_CONSTANT_BASE): + @classmethod + def INPUT_TYPES(s) -> dict: + return { + "required" : { + "value": ("STRING", {"multiline": False}) + } + } + + RETURN_TYPES = ("STRING",) + RETURN_NAMES = ("string",) + + +class ConstantStringMultiline(_CONSTANT_BASE): + @classmethod + def INPUT_TYPES(s) -> dict: + return { + "required": { + "value": ("STRING", {"multiline": True}) + } + } + + RETURN_TYPES = ("STRING",) + RETURN_NAMES = ("string",) + + +NODE_CLASS_MAPPINGS = { + "ConstantFloat": ConstantFloat, + "ConstantInteger": ConstantInteger, + "ConstantString": ConstantString, + "ConstantStringMultiline": ConstantStringMultiline, +}