#---------------------------------------------------------------------------------------------------------------------#
# Comfyroll Studio custom nodes by RockOfFire and Akatsuzi https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes
# for ComfyUI https://github.com/comfyanonymous/ComfyUI
#---------------------------------------------------------------------------------------------------------------------#
import os
import sys
import comfy.controlnet
import comfy.sd
import folder_paths
from nodes import ControlNetApplyAdvanced
from ..categories import icons
sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(__file__)), "comfy"))
#---------------------------------------------------------------------------------------------------------------------#
# This node will apply any type of ControlNet.
class CR_ApplyControlNet:
@classmethod
def INPUT_TYPES(s):
return {"required": {"conditioning": ("CONDITIONING", ),
"control_net": ("CONTROL_NET", ),
"image": ("IMAGE", ),
"switch": (["On","Off"],),
"strength": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 10.0, "step": 0.01})
}
}
RETURN_TYPES = ("CONDITIONING", "STRING", )
RETURN_NAMES = ("CONDITIONING", "show_help", )
FUNCTION = "apply_controlnet"
CATEGORY = icons.get("Comfyroll/ControlNet")
def apply_controlnet(self, conditioning, control_net, image, switch, strength):
show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/ControlNet-Nodes#cr-apply-controlnet"
if strength == 0 or switch == "Off":
return (conditioning, show_help, )
c = []
control_hint = image.movedim(-1,1)
for t in conditioning:
n = [t[0], t[1].copy()]
c_net = control_net.copy().set_cond_hint(control_hint, strength)
if 'control' in t[1]:
c_net.set_previous_controlnet(t[1]['control'])
n[1]['control'] = c_net
c.append(n)
return (c, show_help, )
#---------------------------------------------------------------------------------------------------------------------#
# This node is a stack of controlnets each with their own switch.
class CR_ControlNetStack:
controlnets = ["None"] + folder_paths.get_filename_list("controlnet")
@classmethod
def INPUT_TYPES(cls):
#controlnets = ["None"]
return {"required": {
},
"optional": {
"switch_1": (["Off","On"],),
"controlnet_1": (cls.controlnets,),
"controlnet_strength_1": ("FLOAT", {"default": 1.0, "min": -10.0, "max": 10.0, "step": 0.01}),
"start_percent_1": ("FLOAT", {"default": 0.0, "min": 0.0, "max": 1.0, "step": 0.001}),
"end_percent_1": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.001}),
#
"switch_2": (["Off","On"],),
"controlnet_2": (cls.controlnets,),
"controlnet_strength_2": ("FLOAT", {"default": 1.0, "min": -10.0, "max": 10.0, "step": 0.01}),
"start_percent_2": ("FLOAT", {"default": 0.0, "min": 0.0, "max": 1.0, "step": 0.001}),
"end_percent_2": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.001}),
#
"switch_3": (["Off","On"],),
"controlnet_3": (cls.controlnets,),
"controlnet_strength_3": ("FLOAT", {"default": 1.0, "min": -10.0, "max": 10.0, "step": 0.01}),
"start_percent_3": ("FLOAT", {"default": 0.0, "min": 0.0, "max": 1.0, "step": 0.001}),
"end_percent_3": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.001}),
#
"image_1": ("IMAGE",),
"image_2": ("IMAGE",),
"image_3": ("IMAGE",),
"controlnet_stack": ("CONTROL_NET_STACK",)
},
}
RETURN_TYPES = ("CONTROL_NET_STACK", "STRING", )
RETURN_NAMES = ("CONTROLNET_STACK", "show_help", )
FUNCTION = "controlnet_stacker"
CATEGORY = icons.get("Comfyroll/ControlNet")
def controlnet_stacker(self, switch_1, controlnet_1, controlnet_strength_1, start_percent_1, end_percent_1,
switch_2, controlnet_2, controlnet_strength_2, start_percent_2, end_percent_2,
switch_3, controlnet_3, controlnet_strength_3, start_percent_3, end_percent_3,
image_1=None, image_2=None, image_3=None, controlnet_stack=None):
# Initialise the list
controlnet_list= []
if controlnet_stack is not None:
controlnet_list.extend([l for l in controlnet_stack if l[0] != "None"])
if controlnet_1 != "None" and switch_1 == "On" and image_1 is not None:
controlnet_path = folder_paths.get_full_path("controlnet", controlnet_1)
controlnet_1 = comfy.controlnet.load_controlnet(controlnet_path)
controlnet_list.extend([(controlnet_1, image_1, controlnet_strength_1, start_percent_1, end_percent_1)]),
if controlnet_2 != "None" and switch_2 == "On" and image_2 is not None:
controlnet_path = folder_paths.get_full_path("controlnet", controlnet_2)
controlnet_2 = comfy.controlnet.load_controlnet(controlnet_path)
controlnet_list.extend([(controlnet_2, image_2, controlnet_strength_2, start_percent_2, end_percent_2)]),
if controlnet_3 != "None" and switch_3 == "On" and image_3 is not None:
controlnet_path = folder_paths.get_full_path("controlnet", controlnet_3)
controlnet_3 = comfy.controlnet.load_controlnet(controlnet_path)
controlnet_list.extend([(controlnet_3, image_3, controlnet_strength_3, start_percent_3, end_percent_3)]),
show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/ControlNet-Nodes#cr-multi-controlnet-stack"
return (controlnet_list, show_help, )
#---------------------------------------------------------------------------------------------------------------------#
# This applies the ControlNet stack.
class CR_ApplyControlNetStack:
@classmethod
def INPUT_TYPES(s):
return {"required": {"base_positive": ("CONDITIONING", ),
"base_negative": ("CONDITIONING",),
"switch": (["Off","On"],),
"controlnet_stack": ("CONTROL_NET_STACK", ),
}
}
RETURN_TYPES = ("CONDITIONING", "CONDITIONING", "STRING", )
RETURN_NAMES = ("base_pos", "base_neg", "show_help", )
FUNCTION = "apply_controlnet_stack"
CATEGORY = icons.get("Comfyroll/ControlNet")
def apply_controlnet_stack(self, base_positive, base_negative, switch, controlnet_stack=None,):
show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/ControlNet-Nodes#cr-apply-multi-controlnet-stack"
if switch == "Off":
return (base_positive, base_negative, show_help, )
if controlnet_stack is not None:
for controlnet_tuple in controlnet_stack:
controlnet_name, image, strength, start_percent, end_percent = controlnet_tuple
if type(controlnet_name) == str:
controlnet_path = folder_paths.get_full_path("controlnet", controlnet_name)
controlnet = comfy.sd.load_controlnet(controlnet_path)
else:
controlnet = controlnet_name
controlnet_conditioning = ControlNetApplyAdvanced().apply_controlnet(base_positive, base_negative,
controlnet, image, strength,
start_percent, end_percent)
base_positive, base_negative = controlnet_conditioning[0], controlnet_conditioning[1]
return (base_positive, base_negative, show_help, )
#---------------------------------------------------------------------------------------------------------------------#
# MAPPINGS
#---------------------------------------------------------------------------------------------------------------------#
# For reference only, actual mappings are in __init__.py
'''
NODE_CLASS_MAPPINGS = {
"CR Apply ControlNet": CR_ApplyControlNet,
"CR Multi-ControlNet Stack":CR_ControlNetStack,
"CR Apply Multi-ControlNet":CR_ApplyControlNetStack,
}
'''