add missing file

2023-08-23 22:40:21 +09:00
parent 2255fba8a1
commit 8d6080e188
2 changed files with 67 additions and 1 deletions
@@ -2,7 +2,7 @@ import configparser
 import os


-version = "V3.24"
+version = "V3.24.1"

 dependency_version = 9

@@ -0,0 +1,66 @@
+from impact.utils import any_typ
+
+class GeneralSwitch:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {"required": {
+                    "select": ("INT", {"default": 1, "min": 1, "max": 999999, "step": 1}),
+                    "input1": (any_typ,),
+                    },
+                }
+
+    RETURN_TYPES = (any_typ, )
+
+    OUTPUT_NODE = True
+
+    FUNCTION = "doit"
+
+    CATEGORY = "ImpactPack/Util"
+
+    def doit(self, *args, **kwargs):
+        input_name = f"input{int(kwargs['select'])}"
+
+        if input_name in kwargs:
+            return (kwargs[input_name],)
+        else:
+            print(f"ImpactSwitch: invalid select index ('input1' is selected)")
+            return (kwargs['input1'],)
+
+
+class ImageMaskSwitch:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {"required": {
+            "select": ("INT", {"default": 1, "min": 1, "max": 4, "step": 1}),
+            "images1": ("IMAGE",),
+        },
+
+            "optional": {
+                "mask1_opt": ("MASK",),
+                "images2_opt": ("IMAGE",),
+                "mask2_opt": ("MASK",),
+                "images3_opt": ("IMAGE",),
+                "mask3_opt": ("MASK",),
+                "images4_opt": ("IMAGE",),
+                "mask4_opt": ("MASK",),
+            },
+        }
+
+    RETURN_TYPES = ("IMAGE", "MASK",)
+
+    OUTPUT_NODE = True
+
+    FUNCTION = "doit"
+
+    CATEGORY = "ImpactPack/Util"
+
+    def doit(self, select, images1, mask1_opt=None, images2_opt=None, mask2_opt=None, images3_opt=None, mask3_opt=None,
+             images4_opt=None, mask4_opt=None):
+        if select == 1:
+            return images1, mask1_opt,
+        elif select == 2:
+            return images2_opt, mask2_opt,
+        elif select == 3:
+            return images3_opt, mask3_opt,
+        else:
+            return images4_opt, mask4_opt,