Add Slice node

This commit is contained in:
Alistair Buxton
2023-09-10 20:35:49 +01:00
parent 25c5f19d58
commit db319d7e86
+26 -3
View File
@@ -14,9 +14,9 @@ class Range:
def INPUT_TYPES(s): def INPUT_TYPES(s):
return { return {
"required": { "required": {
"start": ("INT", {"default": 0, "min": 0, "max": 9999999, "step": 1}), "start": ("INT", {"default": 0, "min": -9999999, "max": 9999999, "step": 1}),
"stop": ("INT", {"default": 0, "min": 0, "max": 9999999, "step": 1}), "stop": ("INT", {"default": 10, "min": -9999999, "max": 9999999, "step": 1}),
"step": ("INT", {"default": 0, "min": 0, "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)], ) 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 @register_node
class Join: class Join:
@classmethod @classmethod