112 lines
3.9 KiB
Python
112 lines
3.9 KiB
Python
import sys
|
|
from . import hooks
|
|
from . import defs
|
|
import comfy
|
|
|
|
|
|
class SEGSOrderedFilterDetailerHookProvider:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {"required": {
|
|
"target": (["area(=w*h)", "width", "height", "x1", "y1", "x2", "y2"],),
|
|
"order": ("BOOLEAN", {"default": True, "label_on": "descending", "label_off": "ascending"}),
|
|
"take_start": ("INT", {"default": 0, "min": 0, "max": sys.maxsize, "step": 1}),
|
|
"take_count": ("INT", {"default": 1, "min": 0, "max": sys.maxsize, "step": 1}),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("DETAILER_HOOK", )
|
|
FUNCTION = "doit"
|
|
|
|
CATEGORY = "ImpactPack/Util"
|
|
|
|
def doit(self, target, order, take_start, take_count):
|
|
hook = hooks.SEGSOrderedFilterDetailerHook(target, order, take_start, take_count)
|
|
return (hook, )
|
|
|
|
|
|
class SEGSRangeFilterDetailerHookProvider:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {"required": {
|
|
"target": (["area(=w*h)", "width", "height", "x1", "y1", "x2", "y2", "length_percent"],),
|
|
"mode": ("BOOLEAN", {"default": True, "label_on": "inside", "label_off": "outside"}),
|
|
"min_value": ("INT", {"default": 0, "min": 0, "max": sys.maxsize, "step": 1}),
|
|
"max_value": ("INT", {"default": 67108864, "min": 0, "max": sys.maxsize, "step": 1}),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("DETAILER_HOOK", )
|
|
FUNCTION = "doit"
|
|
|
|
CATEGORY = "ImpactPack/Util"
|
|
|
|
def doit(self, target, mode, min_value, max_value):
|
|
hook = hooks.SEGSRangeFilterDetailerHook(target, mode, min_value, max_value)
|
|
return (hook, )
|
|
|
|
|
|
class SEGSLabelFilterDetailerHookProvider:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {"required": {
|
|
"segs": ("SEGS", ),
|
|
"preset": (['all'] + defs.detection_labels,),
|
|
"labels": ("STRING", {"multiline": True, "placeholder": "List the types of segments to be allowed, separated by commas"}),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("DETAILER_HOOK", )
|
|
FUNCTION = "doit"
|
|
|
|
CATEGORY = "ImpactPack/Util"
|
|
|
|
def doit(self, preset, labels):
|
|
hook = hooks.SEGSLabelFilterDetailerHook(labels)
|
|
return (hook, )
|
|
|
|
|
|
class PreviewDetailerHookProvider:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {"quality": ("INT", {"default": 95, "min": 20, "max": 100})},
|
|
"hidden": {"unique_id": "UNIQUE_ID"},
|
|
}
|
|
|
|
RETURN_TYPES = ("DETAILER_HOOK", )
|
|
FUNCTION = "doit"
|
|
|
|
CATEGORY = "ImpactPack/Util"
|
|
|
|
def doit(self, quality, unique_id):
|
|
hook = hooks.PreviewDetailerHook(unique_id, quality)
|
|
return (hook, )
|
|
|
|
|
|
class StableCascade_DetailerHookProvider:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"b_model": ("MODEL",),
|
|
"b_vae": ("VAE",),
|
|
"b_seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}),
|
|
"b_steps": ("INT", {"default": 5, "min": 1, "max": 10000}),
|
|
"b_cfg": ("FLOAT", {"default": 1.1, "min": 0.0, "max": 100.0, "step": 0.1, "round": 0.01}),
|
|
"b_sampler_name": (comfy.samplers.KSampler.SAMPLERS,),
|
|
"b_scheduler": (comfy.samplers.KSampler.SCHEDULERS,),
|
|
"c_compression": ("INT", {"default": 42, "min": 4, "max": 128, "step": 1}),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("DETAILER_HOOK", )
|
|
FUNCTION = "doit"
|
|
|
|
CATEGORY = "ImpactPack/Util"
|
|
|
|
def doit(self, b_model, b_vae, b_seed, b_steps, b_cfg, b_sampler_name, b_scheduler, c_compression):
|
|
hook = hooks.StableCascade_DetailerHook(b_model, b_vae, b_seed, b_steps, b_cfg, b_sampler_name, b_scheduler, c_compression)
|
|
return (hook, )
|
|
|