Add Slice node

2023-09-10 20:35:49 +01:00
parent 25c5f19d58
commit db319d7e86
1 changed files with 26 additions and 3 deletions
@@ -14,9 +14,9 @@ class Range:
    def INPUT_TYPES(s):
        return {
            "required": {
-                "start": ("INT", {"default": 0, "min": 0, "max": 9999999, "step": 1}),
-                "stop": ("INT", {"default": 0, "min": 0, "max": 9999999, "step": 1}),
-                "step": ("INT", {"default": 0, "min": 0, "max": 9999999, "step": 1}),
+                "start": ("INT", {"default": 0, "min": -9999999, "max": 9999999, "step": 1}),
+                "stop": ("INT", {"default": 10, "min": -9999999, "max": 9999999, "step": 1}),
+                "step": ("INT", {"default": 1, "min": -9999999, "max": 9999999, "step": 1}),
            },
        }

@@ -116,6 +116,29 @@ class Permutations:
        return ([x for x in itertools.permutations(sequence, count)], )


+@register_node
+class Slice:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "sequence": ("SEQUENCE",),
+                "start": ("INT", {"default": 0, "min": -9999999, "max": 9999999, "step": 1}),
+                "stop": ("INT", {"default": 9999999, "min": -9999999, "max": 9999999, "step": 1}),
+                "step": ("INT", {"default": 1, "min": -9999999, "max": 9999999, "step": 1}),
+            },
+        }
+
+    RETURN_TYPES = ("SEQUENCE", )
+    RETURN_NAMES = ("sequence", )
+    FUNCTION = "go"
+    CATEGORY = "ali1234/sequence"
+
+    def go(self, sequence, start, stop, step):
+        return (sequence[start:stop:step], )
+
+
+
@register_node
 class Join:
    @classmethod