147 lines
4.0 KiB
Python
147 lines
4.0 KiB
Python
|
|
from .logic_gates import CLASS_MAPPINGS as LogicMapping, CLASS_NAMES as LogicNames
|
|
from .randomness import CLASS_MAPPINGS as RandomMapping, CLASS_NAMES as RandomNames
|
|
from .conversion import CLASS_MAPPINGS as ConversionMapping, CLASS_NAMES as ConversionNames
|
|
from .math_nodes import CLASS_MAPPINGS as MathMapping, CLASS_NAMES as MathNames
|
|
from .exif.exif import read_info_from_image_stealth
|
|
from .autonode import node_wrapper, get_node_names_mappings, validate, anytype
|
|
import time
|
|
|
|
fundamental_classes = []
|
|
fundamental_node = node_wrapper(fundamental_classes)
|
|
|
|
@fundamental_node
|
|
class SleepNodeFloat:
|
|
FUNCTION = "sleep"
|
|
RETURN_TYPES = ("FLOAT",)
|
|
CATEGORY = "Misc"
|
|
custom_name = "Sleep (Float tunnel)"
|
|
@staticmethod
|
|
def sleep(interval):
|
|
time.sleep(interval)
|
|
return (interval,)
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"interval": ("FLOAT", {"default": 0.0}),
|
|
}
|
|
}
|
|
@fundamental_node
|
|
class SleepNodeImage:
|
|
FUNCTION = "sleep"
|
|
RETURN_TYPES = (anytype,)
|
|
CATEGORY = "Misc"
|
|
custom_name = "Sleep (Image tunnel)"
|
|
@staticmethod
|
|
def sleep(interval, image):
|
|
time.sleep(interval)
|
|
return (image,)
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"interval": ("FLOAT", {"default": 0.0}),
|
|
"image": (anytype,),
|
|
}
|
|
}
|
|
|
|
@fundamental_node
|
|
class ErrorNode:
|
|
FUNCTION = "raise_error"
|
|
RETURN_TYPES = ("STRING",)
|
|
CATEGORY = "Misc"
|
|
custom_name = "ErrorNode"
|
|
@staticmethod
|
|
def raise_error(error_msg = "Error"):
|
|
raise Exception("Error: {}".format(error_msg))
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"error_msg": ("STRING", {"default": "Error"}),
|
|
}
|
|
}
|
|
|
|
@fundamental_node
|
|
class DebugComboInputNode:
|
|
FUNCTION = "debug_combo_input"
|
|
RETURN_TYPES = ("STRING",)
|
|
CATEGORY = "Misc"
|
|
custom_name = "Debug Combo Input"
|
|
@staticmethod
|
|
def debug_combo_input(input1):
|
|
print(input1)
|
|
return (input1,)
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"input1": (["0", "1", "2"], { "default": "0" }),
|
|
}
|
|
}
|
|
|
|
# https://github.com/comfyanonymous/ComfyUI/blob/340177e6e85d076ab9e222e4f3c6a22f1fb4031f/custom_nodes/example_node.py.example#L18
|
|
@fundamental_node
|
|
class TextPreviewNode:
|
|
"""
|
|
Displays text in the UI
|
|
"""
|
|
FUNCTION = "text_preview"
|
|
RETURN_TYPES = ()
|
|
CATEGORY = "Misc"
|
|
custom_name = "Text Preview"
|
|
RESULT_NODE = True
|
|
|
|
def text_preview(self, text):
|
|
print(text)
|
|
# below does not work, why?
|
|
return {"ui": {"text": str(text)}}
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"text": (anytype,{"default": "text"}),
|
|
}
|
|
}
|
|
|
|
@fundamental_node
|
|
class ParseExifNode:
|
|
"""
|
|
Parses exif data from image
|
|
"""
|
|
FUNCTION = "parse_exif"
|
|
RETURN_TYPES = ("STRING",)
|
|
CATEGORY = "Misc"
|
|
custom_name = "Parse Exif"
|
|
@staticmethod
|
|
def parse_exif(image):
|
|
return (read_info_from_image_stealth(image),)
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"image": ("IMAGE",),
|
|
}
|
|
}
|
|
|
|
|
|
classes_parsed, node_names_parsed = get_node_names_mappings(fundamental_classes)
|
|
validate(fundamental_classes)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
}
|
|
NODE_CLASS_MAPPINGS.update(classes_parsed)
|
|
NODE_CLASS_MAPPINGS.update(LogicMapping)
|
|
NODE_CLASS_MAPPINGS.update(RandomMapping)
|
|
NODE_CLASS_MAPPINGS.update(ConversionMapping)
|
|
NODE_CLASS_MAPPINGS.update(MathMapping)
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
|
|
}
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(node_names_parsed)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(LogicNames)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(RandomNames)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(ConversionNames)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(MathNames)
|