ComfyUI/comfy_extras/nodes_constant_values,.py
2024-06-27 16:41:57 -04:00

80 lines
1.6 KiB
Python

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,
}