增加中继节点,用来辅助解决 UE节点对于消息的派送冲突

This commit is contained in:
longgui0318
2024-07-07 14:59:03 +08:00
parent 53426f7ae6
commit 340b4bbe22
2 changed files with 108 additions and 2 deletions
+6 -2
View File
@@ -1,12 +1,16 @@
from .layer_nodes import NODE_CLASS_MAPPINGS as layer_nodes_class, NODE_DISPLAY_NAME_MAPPINGS as layer_nodes_display
from .light_nodes import NODE_CLASS_MAPPINGS as light_nodes_class, NODE_DISPLAY_NAME_MAPPINGS as light_nodes_display
from .relay_node import NODE_CLASS_MAPPINGS as relay_nodes_class, NODE_DISPLAY_NAME_MAPPINGS as relay_nodes_display
NODE_CLASS_MAPPINGS = {
**layer_nodes_class,
**light_nodes_class
**light_nodes_class,
**relay_nodes_class
}
NODE_DISPLAY_NAME_MAPPINGS = {
**layer_nodes_display,
**light_nodes_display
**light_nodes_display,
**relay_nodes_display
}
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
+102
View File
@@ -0,0 +1,102 @@
class ImageRelay:
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"image": ("IMAGE",),
}
}
RETURN_TYPES = ("IMAGE",)
RETURN_NAMES = ("IMAGE",)
FUNCTION = "relay"
CATEGORY = "utils"
def relay(self, image):
return (image,)
class MaskRelay:
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"mask": ("MASK",),
}
}
RETURN_TYPES = ("MASK",)
RETURN_NAMES = ("MASK",)
FUNCTION = "relay"
CATEGORY = "utils"
def relay(self, mask):
return (mask,)
class IntRelay:
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"int_": ("INT",{ "default": 0, "step": 1 }),
}
}
RETURN_TYPES = ("INT",)
RETURN_NAMES = ("INT",)
FUNCTION = "relay"
CATEGORY = "utils"
def relay(self, int_):
return (int_,)
class StringRelay:
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"string_": ("STRING",{ "default": "" }),
}
}
RETURN_TYPES = ("STRING",)
RETURN_NAMES = ("STRING",)
FUNCTION = "relay"
CATEGORY = "utils"
def relay(self, string_):
return (string_,)
class FloatRelay:
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"float_": ("FLOAT",{ "default": 1.0, "step": 0.01 }),
}
}
RETURN_TYPES = ("FLOAT",)
RETURN_NAMES = ("FLOAT",)
FUNCTION = "relay"
CATEGORY = "utils"
def relay(self, float_):
return (float_,)
NODE_CLASS_MAPPINGS = {
"Image Relay":ImageRelay,
"Mask Relay": MaskRelay,
"Int Relay": IntRelay,
"String Relay": StringRelay,
"Float Relay": FloatRelay,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"Image Relay":"Image Relay",
"Mask Relay": "Mask Relay",
"Int Relay": "Int Relay",
"String Relay": "String Relay",
"Float Relay": "Float Relay",
}