diff --git a/__init__.py b/__init__.py index 7cc8f6f..1133b95 100644 --- a/__init__.py +++ b/__init__.py @@ -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'] diff --git a/relay_node.py b/relay_node.py new file mode 100644 index 0000000..16a49c2 --- /dev/null +++ b/relay_node.py @@ -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", + +} \ No newline at end of file