# 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)