92 lines
3.1 KiB
Python
92 lines
3.1 KiB
Python
# Copyright (C) 2024 Reece H. Dunn. SPDX-License-Identifier: GPL-3
|
|
|
|
|
|
MAX_RESOLUTION=16384
|
|
|
|
|
|
# noinspection PyPep8Naming,PyMethodMayBeStatic,PyRedundantParentheses,PyMethodParameters
|
|
class ImageSizeBusNode:
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {"required": {"width": ("INT", {"default": 1024, "min": 256, "max": MAX_RESOLUTION, "step": 8, "defaultInput": True}),
|
|
"height": ("INT", {"default": 1024, "min": 256, "max": MAX_RESOLUTION, "step": 8, "defaultInput": True})}
|
|
}
|
|
|
|
RETURN_TYPES = ("INT", "INT", )
|
|
RETURN_NAMES = ("width", "height", )
|
|
|
|
FUNCTION = "bus"
|
|
CATEGORY = "ComfyBus"
|
|
|
|
def bus(self, width, height):
|
|
return (width, height)
|
|
|
|
|
|
# noinspection PyPep8Naming,PyMethodMayBeStatic,PyRedundantParentheses,PyMethodParameters
|
|
class ImageParameterBusNode:
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {"required": {"seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff, "defaultInput": True}),
|
|
"width": ("INT", {"default": 1024, "min": 256, "max": MAX_RESOLUTION, "step": 8, "defaultInput": True}),
|
|
"height": ("INT", {"default": 1024, "min": 256, "max": MAX_RESOLUTION, "step": 8, "defaultInput": True})}
|
|
}
|
|
|
|
RETURN_TYPES = ("INT", "INT", "INT", )
|
|
RETURN_NAMES = ("seed", "width", "height", )
|
|
|
|
FUNCTION = "bus"
|
|
CATEGORY = "ComfyBus"
|
|
|
|
def bus(self, seed, width, height):
|
|
return (seed, width, height)
|
|
|
|
|
|
# noinspection PyPep8Naming,PyMethodMayBeStatic,PyRedundantParentheses,PyMethodParameters
|
|
class LatentImageParameterBusNode:
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {"required": {"latent": ("LATENT", ),
|
|
"seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff, "defaultInput": True}),
|
|
"denoise": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01, "defaultInput": True})}
|
|
}
|
|
|
|
RETURN_TYPES = ("LATENT", "INT", "FLOAT", )
|
|
RETURN_NAMES = ("latent", "seed", "denoise", )
|
|
|
|
FUNCTION = "bus"
|
|
CATEGORY = "ComfyBus"
|
|
|
|
def bus(self, latent, seed, denoise):
|
|
return (latent, seed, denoise)
|
|
|
|
|
|
# noinspection PyPep8Naming,PyMethodMayBeStatic,PyRedundantParentheses,PyMethodParameters
|
|
class ImageScaleToSideParameterBusNode:
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {"required": {"side_length": ("INT", {"default": 1024, "min": 256, "max": MAX_RESOLUTION, "step": 8, "defaultInput": True}),
|
|
"seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff, "defaultInput": True}),
|
|
"denoise": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01, "defaultInput": True})}
|
|
}
|
|
|
|
RETURN_TYPES = ("INT", "INT", "FLOAT", )
|
|
RETURN_NAMES = ("side_length", "seed", "denoise", )
|
|
|
|
FUNCTION = "bus"
|
|
CATEGORY = "ComfyBus"
|
|
|
|
def bus(self, side_length, seed, denoise):
|
|
return (side_length, seed, denoise)
|