From 340b4bbe22d1390e596bc2edaf6eefea27569efb Mon Sep 17 00:00:00 2001
From: longgui0318 <>
Date: Sun, 7 Jul 2024 14:59:03 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E4=B8=AD=E7=BB=A7=E8=8A=82?=
 =?UTF-8?q?=E7=82=B9=EF=BC=8C=E7=94=A8=E6=9D=A5=E8=BE=85=E5=8A=A9=E8=A7=A3?=
 =?UTF-8?q?=E5=86=B3=20UE=E8=8A=82=E7=82=B9=E5=AF=B9=E4=BA=8E=E6=B6=88?=
 =?UTF-8?q?=E6=81=AF=E7=9A=84=E6=B4=BE=E9=80=81=E5=86=B2=E7=AA=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 __init__.py   |   8 +++-
 relay_node.py | 102 ++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 108 insertions(+), 2 deletions(-)
 create mode 100644 relay_node.py

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