xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-16 08:14:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2023-10-09 15:28:06 +03:00
parent e74c2b034a
commit 3f42ff5d05
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
nodes.py
View File

@ -137,11 +137,12 @@ class CreateFadeMask:
"height": ("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"],), "interpolation": (["linear", "ease_in", "ease_out", "ease_in_out"],),
"start_level": ("FLOAT", {"default": 1.0,"min": 0.0, "max": 1.0, "step": 0.01}), "start_level": ("FLOAT", {"default": 1.0,"min": 0.0, "max": 1.0, "step": 0.01}),
"midpoint_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}), "end_level": ("FLOAT", {"default": 0.0,"min": 0.0, "max": 1.0, "step": 0.01}),
}, },
} }
def createfademask(self, frames, width, height, invert, interpolation, start_level, end_level): def createfademask(self, frames, width, height, invert, interpolation, start_level, midpoint_level, end_level):
def ease_in(t): def ease_in(t):
return t * t return t * t
@ -165,7 +166,14 @@ class CreateFadeMask:
elif interpolation == "ease_in_out": elif interpolation == "ease_in_out":
t = ease_in_out(t) t = ease_in_out(t)
color = start_level - t * (start_level - end_level) if midpoint_level is not None:
if t < 0.5:
color = start_level - t * (start_level - midpoint_level) * 2
else:
color = midpoint_level - (t - 0.5) * (midpoint_level - end_level) * 2
else:
color = start_level - t * (start_level - end_level)
image = np.full((height, width), color, dtype=np.float32) image = np.full((height, width), color, dtype=np.float32)
image_batch[i] = image image_batch[i] = image