initial code
This commit is contained in:
Eric W. Burns
@@ -1,14 +1,113 @@
|
||||
"""Node definitions for ComfyUI-EBU-Workflow package."""
|
||||
import math
|
||||
|
||||
class EbuScalingResolution:
|
||||
aspect_ratios = {
|
||||
"16:9": [
|
||||
"640x360", "960x540", "1024x576", "1088x612", "1152x648", "1216x684", "1280x720", "1344x756", "1408x792", "1472x828", "1536x864", "1600x900", "1792x1024", "1856x1044", "1920x1080", "2048x1152", "2240x1260", "2304x1296", "2560x1440"
|
||||
],
|
||||
"16:10": [
|
||||
"640x400", "704x440", "768x480", "832x520", "896x560", "960x600", "1024x640", "1088x680", "1152x720", "1216x760", "1280x800", "1344x840", "1408x880", "1472x920", "1536x960", "1600x1000", "1792x1120", "1856x1160", "1920x1200", "2048x1280", "2240x1400", "2560x1600"
|
||||
],
|
||||
"3:2": [
|
||||
"768x512", "832x554", "896x597", "960x640", "1024x682", "1088x725", "1152x768", "1216x810", "1280x853", "1344x896", "1408x938", "1472x981", "1536x1024", "1600x1066"
|
||||
],
|
||||
"4:3": [
|
||||
"640x480", "704x528", "768x576", "832x624", "896x672", "960x720", "1024x768", "1088x816", "1152x864", "1216x912", "1280x960", "1344x1008", "1408x1056", "1472x1104", "1536x1152", "1600x1200"
|
||||
],
|
||||
"5:4": [
|
||||
"640x512", "704x564", "768x615", "832x667", "896x718", "960x770", "1024x821", "1088x872", "1152x924", "1216x975", "1280x1024", "1344x1076", "1408x1128", "1472x1179", "1536x1230", "1600x1280"
|
||||
],
|
||||
"6:5": [
|
||||
"768x640", "832x693", "896x746", "960x800", "1024x853", "1088x906", "1152x960", "1216x1013", "1280x1066", "1344x1120", "1408x1173", "1472x1226", "1536x1280", "1600x1333"
|
||||
],
|
||||
"1:1": [
|
||||
"512x512", "576x576", "640x640", "704x704", "768x768", "832x832", "896x896", "960x960", "1024x1024", "1088x1088", "1152x1152", "1216x1216", "1280x1280", "1344x1344", "1408x1408", "1472x1472", "1536x1536", "1600x1600"
|
||||
]
|
||||
}
|
||||
|
||||
@classmethod
|
||||
def INPUT_TYPES(cls):
|
||||
return {
|
||||
"required": {
|
||||
"active_aspect_ratio": (list(cls.aspect_ratios.keys()) + ["Other"],),
|
||||
**{key: (value,) for key, value in cls.aspect_ratios.items()},
|
||||
"other_width": ("INT", {"default": 1024, "min": 64, "max": 8192, "step": 8, "display": "number"}),
|
||||
"other_height": ("INT", {"default": 1024, "min": 64, "max": 8192, "step": 8, "display": "number"}),
|
||||
"mode": (["Landscape", "Profile"], {"default": "Landscape"}),
|
||||
"upscale_by": ("FLOAT", {"default": 1.5, "min": 1.0, "max": 10.0, "step": 0.5, "display": "number"})
|
||||
}
|
||||
}
|
||||
|
||||
RETURN_TYPES = ("INT", "INT", "INT", "INT", "FLOAT", "STRING",)
|
||||
RETURN_NAMES = ("width", "height", "upscaled_width", "upscaled_height", "upscale_by", "upscaled_resolution_string",)
|
||||
FUNCTION = "compute_resolution"
|
||||
CATEGORY = "Resolution"
|
||||
|
||||
def compute_resolution(self, active_aspect_ratio, **kwargs):
|
||||
if active_aspect_ratio == "Other":
|
||||
width = kwargs["other_width"]
|
||||
height = kwargs["other_height"]
|
||||
else:
|
||||
resolution = kwargs[active_aspect_ratio]
|
||||
width, height = [int(x) for x in resolution.split('x')]
|
||||
|
||||
def round_up_to_multiple_of_eight(value):
|
||||
return math.ceil(value / 8) * 8
|
||||
|
||||
upscale_by = kwargs["upscale_by"]
|
||||
scaled_width = round_up_to_multiple_of_eight(width * upscale_by)
|
||||
scaled_height = round_up_to_multiple_of_eight(height * upscale_by)
|
||||
|
||||
if kwargs["mode"] == "Profile":
|
||||
width, height = height, width
|
||||
scaled_width, scaled_height = scaled_height, scaled_width
|
||||
upscaled_resolution_string = f"{scaled_width}x{scaled_height}"
|
||||
|
||||
print(f"Original: {width}x{height}, Scaled: {upscaled_resolution_string}, Upscale factor: {upscale_by}")
|
||||
|
||||
return width, height, scaled_width, scaled_height, upscale_by, upscaled_resolution_string
|
||||
|
||||
|
||||
class EbuScalingTile:
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
@classmethod
|
||||
def INPUT_TYPES(cls):
|
||||
return {
|
||||
"required": {
|
||||
"upscaled_image_width": ("INT", {"default": 768, "step": 1, "display": "number"}),
|
||||
"upscaled_image_height": ("INT", {"default": 1024, "step": 1, "display": "number"}),
|
||||
"profile_width_div_by": ("FLOAT", {"default": 1, "min": 1.0, "max": 20.0, "step": 0.05, "display": "number"}),
|
||||
"profile_height_div_by": ("FLOAT", {"default": 3, "min": 1.0, "max": 20.0, "step": 0.05, "display": "number"}),
|
||||
"landscape_width_div_by": ("FLOAT", {"default": 2, "min": 1.0, "max": 20.0, "step": 0.05, "display": "number"}),
|
||||
"landscape_height_div_by": ("FLOAT", {"default": 2, "min": 1.0, "max": 20.0, "step": 0.05, "display": "number"}),
|
||||
"tile_width_padding": ("INT", {"default": 0, "step": 1, "display": "number"}),
|
||||
"tile_height_padding": ("INT", {"default": 0, "step": 1, "display": "number"}),
|
||||
}
|
||||
}
|
||||
|
||||
RETURN_TYPES = ("INT", "INT",)
|
||||
RETURN_NAMES = ("tile_width", "tile_height",)
|
||||
FUNCTION = "tile"
|
||||
CATEGORY = "Resolution"
|
||||
|
||||
def tile(self, upscaled_image_width, upscaled_image_height, profile_width_div_by, profile_height_div_by, landscape_width_div_by, landscape_height_div_by, tile_width_padding, tile_height_padding):
|
||||
if upscaled_image_height > upscaled_image_width:
|
||||
new_width = int(upscaled_image_width / profile_width_div_by) + tile_width_padding
|
||||
new_height = int(upscaled_image_height / profile_height_div_by) + tile_height_padding
|
||||
else:
|
||||
new_width = int(upscaled_image_width / landscape_width_div_by) + tile_width_padding
|
||||
new_height = int(upscaled_image_height / landscape_height_div_by) + tile_height_padding
|
||||
return new_width, new_height
|
||||
|
||||
class ExampleNode:
|
||||
"""A simple example node implementation."""
|
||||
def execute(self, *args, **kwargs):
|
||||
return "Example output"
|
||||
|
||||
NODE_CLASS_MAPPINGS = {
|
||||
"ExampleNode": ExampleNode,
|
||||
"EbuScalingResolution": EbuScalingResolution,
|
||||
"EbuScalingTile": EbuScalingTile
|
||||
}
|
||||
|
||||
NODE_DISPLAY_NAME_MAPPINGS = {
|
||||
"ExampleNode": "Example Node",
|
||||
"EbuScalingResolution": "EBU Scaling Resolution",
|
||||
"EbuScalingTile": "EBU Scaling Tile"
|
||||
}
|
||||
|
||||
@@ -0,0 +1,16 @@
|
||||
[project]
|
||||
name = "ComfyUI_EBU_PromptHelper"
|
||||
description = "This ComfyUI extension provides a suite of custom nodes for enhancing and manipulating prompts. Features include random color palette generation, prompt text replacement, prompt randomization using variable names and lists of options, list sampling, file loading, current datetime formatting, and season/weather/time-of-day generation."
|
||||
version = "1.0.0"
|
||||
authors = [
|
||||
{name = "burnsbert"}
|
||||
]
|
||||
keywords = keywords = ["comfyui", "prompt", "random", "color", "weather", "datetime", "utf8", "utility"]
|
||||
license = {file = "LICENSE"}
|
||||
|
||||
[project.urls]
|
||||
Repository = "https://github.com/burnsbert/ComfyUI-EBU-PromptHelper"
|
||||
|
||||
[tool.comfy]
|
||||
PublisherId = "burnsbert"
|
||||
DisplayName = "ComfyUI-EBU-PromptHelper"
|
||||
@@ -0,0 +1 @@
|
||||
# No additional requirements beyond Python standard library
|
||||
Reference in New Issue
Block a user