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

Update nodes.py

Browse Source
This commit is contained in:
kijai 2023-10-11 01:31:23 +03:00
parent 4e9cc0b8a2
commit 7595958f5a
1 changed files with 30 additions and 112 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

134
nodes.py
View File

@ -475,6 +475,14 @@ class ConditioningMultiCombine:
cond = cond_combine_node.combine(new_cond, cond)[0]
return (cond, inputcount,)
def append_helper(t, mask, c, set_area_to_bounds, strength):
n = [t[0], t[1].copy()]
_, h, w = mask.shape
n[1]['mask'] = mask
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c.append(n)
class ConditioningSetMaskAndCombine:
@classmethod
def INPUT_TYPES(cls):
@ -507,33 +515,13 @@ class ConditioningSetMaskAndCombine:
if len(mask_2.shape) < 3:
mask_2 = mask_2.unsqueeze(0)
for t in positive_1:
n = [t[0], t[1].copy()]
_, h, w = mask_1.shape
n[1]['mask'] = mask_1
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c.append(n)
append_helper(t, mask_1, c, set_area_to_bounds, strength)
for t in positive_2:
n = [t[0], t[1].copy()]
_, h, w = mask_2.shape
n[1]['mask'] = mask_2
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c.append(n)
append_helper(t, mask_2, c, set_area_to_bounds, strength)
for t in negative_1:
n = [t[0], t[1].copy()]
_, h, w = mask_1.shape
n[1]['mask'] = mask_1
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c2.append(n)
append_helper(t, mask_1, c2, set_area_to_bounds, strength)
for t in negative_2:
n = [t[0], t[1].copy()]
_, h, w = mask_2.shape
n[1]['mask'] = mask_2
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c2.append(n)
append_helper(t, mask_2, c2, set_area_to_bounds, strength)
return (c, c2)
class ConditioningSetMaskAndCombine3:
@ -573,47 +561,17 @@ class ConditioningSetMaskAndCombine3:
if len(mask_3.shape) < 3:
mask_3 = mask_3.unsqueeze(0)
for t in positive_1:
n = [t[0], t[1].copy()]
_, h, w = mask_1.shape
n[1]['mask'] = mask_1
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c.append(n)
append_helper(t, mask_1, c, set_area_to_bounds, strength)
for t in positive_2:
n = [t[0], t[1].copy()]
_, h, w = mask_2.shape
n[1]['mask'] = mask_2
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c.append(n)
append_helper(t, mask_2, c, set_area_to_bounds, strength)
for t in positive_3:
n = [t[0], t[1].copy()]
_, h, w = mask_3.shape
n[1]['mask'] = mask_3
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c.append(n)
append_helper(t, mask_3, c, set_area_to_bounds, strength)
for t in negative_1:
n = [t[0], t[1].copy()]
_, h, w = mask_1.shape
n[1]['mask'] = mask_1
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c2.append(n)
append_helper(t, mask_1, c2, set_area_to_bounds, strength)
for t in negative_2:
n = [t[0], t[1].copy()]
_, h, w = mask_2.shape
n[1]['mask'] = mask_2
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c2.append(n)
append_helper(t, mask_2, c2, set_area_to_bounds, strength)
for t in negative_3:
n = [t[0], t[1].copy()]
_, h, w = mask_3.shape
n[1]['mask'] = mask_3
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c2.append(n)
append_helper(t, mask_3, c2, set_area_to_bounds, strength)
return (c, c2)
class ConditioningSetMaskAndCombine4:
@ -658,61 +616,21 @@ class ConditioningSetMaskAndCombine4:
if len(mask_4.shape) < 3:
mask_4 = mask_4.unsqueeze(0)
for t in positive_1:
n = [t[0], t[1].copy()]
_, h, w = mask_1.shape
n[1]['mask'] = mask_1
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c.append(n)
append_helper(t, mask_1, c, set_area_to_bounds, strength)
for t in positive_2:
n = [t[0], t[1].copy()]
_, h, w = mask_2.shape
n[1]['mask'] = mask_2
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c.append(n)
append_helper(t, mask_2, c, set_area_to_bounds, strength)
for t in positive_3:
n = [t[0], t[1].copy()]
_, h, w = mask_3.shape
n[1]['mask'] = mask_3
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c.append(n)
append_helper(t, mask_3, c, set_area_to_bounds, strength)
for t in positive_4:
n = [t[0], t[1].copy()]
_, h, w = mask_3.shape
n[1]['mask'] = mask_3
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c.append(n)
append_helper(t, mask_4, c, set_area_to_bounds, strength)
for t in negative_1:
n = [t[0], t[1].copy()]
_, h, w = mask_1.shape
n[1]['mask'] = mask_1
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c2.append(n)
append_helper(t, mask_1, c2, set_area_to_bounds, strength)
for t in negative_2:
n = [t[0], t[1].copy()]
_, h, w = mask_2.shape
n[1]['mask'] = mask_2
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c2.append(n)
append_helper(t, mask_2, c2, set_area_to_bounds, strength)
for t in negative_3:
n = [t[0], t[1].copy()]
_, h, w = mask_3.shape
n[1]['mask'] = mask_3
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c2.append(n)
append_helper(t, mask_3, c2, set_area_to_bounds, strength)
for t in negative_4:
n = [t[0], t[1].copy()]
_, h, w = mask_3.shape
n[1]['mask'] = mask_3
n[1]['set_area_to_bounds'] = set_area_to_bounds
n[1]['mask_strength'] = strength
c2.append(n)
append_helper(t, mask_4, c2, set_area_to_bounds, strength)
return (c, c2)
NODE_CLASS_MAPPINGS = {