mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-14 00:37:12 +08:00
Add constant primitive datatype nodes
This commit is contained in:
parent
97b409cd48
commit
6db26caba1
79
comfy_extras/nodes_constant_values,.py
Normal file
79
comfy_extras/nodes_constant_values,.py
Normal file
@ -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,
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user