fix conversion

This commit is contained in:
aria1th
2023-12-11 02:47:42 +09:00
parent b77910af45
commit 60319133e6
2 changed files with 91 additions and 1 deletions
+1 -1
View File
@@ -17,7 +17,7 @@ def create_class(type_from, type_to):
return None
class_name = "{}2{}".format(type_from, type_to)
class CustomClass:
FUNCTION = class_name.lower()
FUNCTION = "convert"
RETURN_TYPES = (type_to.upper(),)
CATEGORY = "Conversion"
custom_name = "Convert {} to {}".format(type_from, type_to)
+90
View File
@@ -2,14 +2,104 @@
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 .autonode import node_wrapper, get_node_names_mappings, validate
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 = ("IMAGE",)
CATEGORY = "Misc"
custom_name = "Sleep (Image tunnel)"
@staticmethod
def sleep(interval, image):
time.sleep(interval)
return (None,)
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"interval": ("FLOAT", {"default": 0.0}),
"image": ("IMAGE",),
}
}
@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"}),
}
}
# 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"
OUTPUT_NODE = True
custom_name = "Text Preview"
def __init__(self) -> None:
self.type = "output"
def text_preview(self, text):
print(text)
# below does not work, why?
return {"ui": {"text": text}}
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"text": ("STRING",{"default": "text"}),
}
}
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_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)