From e74c2b034a4c91ed41bb24ba4f681fe27ffa095c Mon Sep 17 00:00:00 2001 From: kijai Date: Mon, 9 Oct 2023 14:31:32 +0300 Subject: [PATCH] Add start/end levels --- nodes.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/nodes.py b/nodes.py index 51bbb6b..f467cf4 100644 --- a/nodes.py +++ b/nodes.py @@ -136,13 +136,12 @@ class CreateFadeMask: "width": ("INT", {"default": 256,"min": 16, "max": 4096, "step": 1}), "height": ("INT", {"default": 256,"min": 16, "max": 4096, "step": 1}), "interpolation": (["linear", "ease_in", "ease_out", "ease_in_out"],), + "start_level": ("FLOAT", {"default": 1.0,"min": 0.0, "max": 1.0, "step": 0.01}), + "end_level": ("FLOAT", {"default": 0.0,"min": 0.0, "max": 1.0, "step": 0.01}), }, } - - - - def createfademask(self, frames, width, height, invert, interpolation): + def createfademask(self, frames, width, height, invert, interpolation, start_level, end_level): def ease_in(t): return t * t @@ -166,7 +165,7 @@ class CreateFadeMask: elif interpolation == "ease_in_out": t = ease_in_out(t) - color = 1.0 - t + color = start_level - t * (start_level - end_level) image = np.full((height, width), color, dtype=np.float32) image_batch[i] = image