#---------------------------------------------------------------------------------------------------------------------# # Comfyroll Studio custom nodes by RockOfFire and Akatsuzi https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes # for ComfyUI https://github.com/comfyanonymous/ComfyUI #---------------------------------------------------------------------------------------------------------------------# import numpy as np import torch import os from PIL import Image, ImageDraw, ImageOps, ImageFont, ImageFilter from ..categories import icons from ..config import color_mapping, COLORS from ..config import iso_sizes from .functions_graphics import * #---------------------------------------------------------------------------------------------------------------------# ALIGN_OPTIONS = ["top", "center", "bottom"] ROTATE_OPTIONS = ["text center", "image center"] JUSTIFY_OPTIONS = ["left", "center", "right"] PERSPECTIVE_OPTIONS = ["top", "bottom", "left", "right"] #---------------------------------------------------------------------------------------------------------------------# class CR_PageLayout: @classmethod def INPUT_TYPES(s): font_dir = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), "fonts") file_list = [f for f in os.listdir(font_dir) if os.path.isfile(os.path.join(font_dir, f)) and f.lower().endswith(".ttf")] layout_options = ["header", "footer", "header and footer", "no header or footer"] return {"required": { "layout_options": (layout_options,), "image_panel": ("IMAGE",), "header_height": ("INT", {"default": 0, "min": 0, "max": 1024}), "header_text": ("STRING", {"multiline": True, "default": "text"}), "header_align": (JUSTIFY_OPTIONS, ), "footer_height": ("INT", {"default": 0, "min": 0, "max": 1024}), "footer_text": ("STRING", {"multiline": True, "default": "text"}), "footer_align": (JUSTIFY_OPTIONS, ), "font_name": (file_list,), "font_color": (COLORS,), "header_font_size": ("INT", {"default": 150, "min": 0, "max": 1024}), "footer_font_size": ("INT", {"default": 50, "min": 0, "max": 1024}), "border_thickness": ("INT", {"default": 0, "min": 0, "max": 1024}), "border_color": (COLORS,), "background_color": (COLORS,), }, "optional": { "font_color_hex": ("STRING", {"multiline": False, "default": "#000000"}), "border_color_hex": ("STRING", {"multiline": False, "default": "#000000"}), "bg_color_hex": ("STRING", {"multiline": False, "default": "#000000"}), } } RETURN_TYPES = ("IMAGE", "STRING", ) RETURN_NAMES = ("image", "show_help", ) FUNCTION = "layout" CATEGORY = icons.get("Comfyroll/Graphics/Layout") def layout(self, layout_options, image_panel, border_thickness, border_color, background_color, header_height, header_text, header_align, footer_height, footer_text, footer_align, font_name, font_color, header_font_size, footer_font_size, font_color_hex='#000000', border_color_hex='#000000', bg_color_hex='#000000'): # Get RGB values for the text and background colors font_color = get_color_values(font_color, font_color_hex, color_mapping) border_color = get_color_values(border_color, border_color_hex, color_mapping) bg_color = get_color_values(background_color, bg_color_hex, color_mapping) main_panel = tensor2pil(image_panel) # Get image width and height image_width = main_panel.width image_height = main_panel.height # Set defaults margins = 50 line_spacing = 0 position_x = 0 position_y = 0 align = "center" rotation_angle = 0 rotation_options = "image center" font_outline_thickness = 0 font_outline_color = "black" images = [] ### Create text panels and add to images array if layout_options == "header" or layout_options == "header and footer": header_panel = text_panel(image_width, header_height, header_text, font_name, header_font_size, font_color, font_outline_thickness, font_outline_color, bg_color, margins, line_spacing, position_x, position_y, align, header_align, rotation_angle, rotation_options) images.append(header_panel) images.append(main_panel) if layout_options == "footer" or layout_options == "header and footer": footer_panel = text_panel(image_width, footer_height, footer_text, font_name, footer_font_size, font_color, font_outline_thickness, font_outline_color, bg_color, margins, line_spacing, position_x, position_y, align, footer_align, rotation_angle, rotation_options) images.append(footer_panel) combined_image = combine_images(images, 'vertical') # Add a border to the combined image if border_thickness > 0: combined_image = ImageOps.expand(combined_image, border_thickness, border_color) show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/Layout-Nodes#cr-page-layout" return (pil2tensor(combined_image), show_help, ) #---------------------------------------------------------------------------------------------------------------------# class CR_SimpleTitles: @classmethod def INPUT_TYPES(s): font_dir = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), "fonts") file_list = [f for f in os.listdir(font_dir) if os.path.isfile(os.path.join(font_dir, f)) and f.lower().endswith(".ttf")] layout_options = ["header", "footer", "header and footer", "no header or footer"] return {"required": { "image": ("IMAGE",), "header_text": ("STRING", {"multiline": True, "default": "text"}), "header_height": ("INT", {"default": 0, "min": 0, "max": 1024}), "header_font_size": ("INT", {"default": 150, "min": 0, "max": 1024}), "header_align": (JUSTIFY_OPTIONS, ), "footer_text": ("STRING", {"multiline": True, "default": "text"}), "footer_height": ("INT", {"default": 0, "min": 0, "max": 1024}), "footer_font_size": ("INT", {"default": 50, "min": 0, "max": 1024}), "footer_align": (JUSTIFY_OPTIONS, ), "font_name": (file_list,), "font_color": (COLORS,), "background_color": (COLORS,), }, "optional": { "font_color_hex": ("STRING", {"multiline": False, "default": "#000000"}), "bg_color_hex": ("STRING", {"multiline": False, "default": "#000000"}), } } RETURN_TYPES = ("IMAGE", "STRING", ) RETURN_NAMES = ("image", "show_help", ) FUNCTION = "layout" CATEGORY = icons.get("Comfyroll/Graphics/Layout") def layout(self, image, header_height, header_text, header_align, header_font_size, footer_height, footer_text, footer_align, footer_font_size, font_name, font_color, background_color, font_color_hex='#000000', bg_color_hex='#000000',): # Get RGB values for the text and background colors font_color = get_color_values(font_color, font_color_hex, color_mapping) bg_color = get_color_values(background_color, bg_color_hex, color_mapping) main_panel = tensor2pil(image) # Get image width and height image_width = main_panel.width image_height = main_panel.height # Set defaults margins = 50 line_spacing = 0 position_x = 0 position_y = 0 align = "center" rotation_angle = 0 rotation_options = "image center" font_outline_thickness = 0 font_outline_color = "black" images = [] ### Create text panels and add to images array if header_height >0: header_panel = text_panel(image_width, header_height, header_text, font_name, header_font_size, font_color, font_outline_thickness, font_outline_color, bg_color, margins, line_spacing, position_x, position_y, align, header_align, rotation_angle, rotation_options) images.append(header_panel) images.append(main_panel) if footer_height >0: footer_panel = text_panel(image_width, footer_height, footer_text, font_name, footer_font_size, font_color, font_outline_thickness, font_outline_color, bg_color, margins, line_spacing, position_x, position_y, align, footer_align, rotation_angle, rotation_options) images.append(footer_panel) combined_image = combine_images(images, 'vertical') show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/Layout-Nodes#cr-simple_titles" return (pil2tensor(combined_image), show_help, ) #---------------------------------------------------------------------------------------------------------------------# class CR_ImagePanel: @classmethod def INPUT_TYPES(s): directions = ["horizontal", "vertical"] return {"required": { "image_1": ("IMAGE",), "border_thickness": ("INT", {"default": 0, "min": 0, "max": 1024}), "border_color": (COLORS,), "outline_thickness": ("INT", {"default": 0, "min": 0, "max": 1024}), "outline_color": (COLORS[1:],), "layout_direction": (directions,), }, "optional": { "image_2": ("IMAGE",), "image_3": ("IMAGE",), "image_4": ("IMAGE",), "border_color_hex": ("STRING", {"multiline": False, "default": "#000000"}) } } RETURN_TYPES = ("IMAGE", "STRING", ) RETURN_NAMES = ("image", "show_help", ) FUNCTION = "make_panel" CATEGORY = icons.get("Comfyroll/Graphics/Layout") def make_panel(self, image_1, border_thickness, border_color, outline_thickness, outline_color, layout_direction, image_2=None, image_3=None, image_4=None, border_color_hex='#000000'): border_color = get_color_values(border_color, border_color_hex, color_mapping) # Convert PIL images to NumPy arrays images = [] #image_1 = image_1[0, :, :, :] images.append(tensor2pil(image_1)) if image_2 is not None: #image_2 = image_2[0, :, :, :] images.append(tensor2pil(image_2)) if image_3 is not None: #image_3 = image_3[0, :, :, :] images.append(tensor2pil(image_3)) if image_4 is not None: #image_4 = image_4[0, :, :, :] images.append(tensor2pil(image_4)) # Apply borders and outlines to each image images = apply_outline_and_border(images, outline_thickness, outline_color, border_thickness, border_color) combined_image = combine_images(images, layout_direction) show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/Layout-Nodes#cr-image-panel" return (pil2tensor(combined_image), show_help, ) #---------------------------------------------------------------------------------------------------------------------# class CR_ImageGridPanel: @classmethod def INPUT_TYPES(s): return {"required": { "images": ("IMAGE",), "border_thickness": ("INT", {"default": 0, "min": 0, "max": 1024}), "border_color": (COLORS,), "outline_thickness": ("INT", {"default": 0, "min": 0, "max": 1024}), "outline_color": (COLORS[1:],), "max_columns": ("INT", {"default": 5, "min": 0, "max": 256}), }, "optional": { "border_color_hex": ("STRING", {"multiline": False, "default": "#000000"}) } } RETURN_TYPES = ("IMAGE", "STRING", ) RETURN_NAMES = ("image", "show_help", ) FUNCTION = "make_panel" CATEGORY = icons.get("Comfyroll/Graphics/Layout") def make_panel(self, images, border_thickness, border_color, outline_thickness, outline_color, max_columns, border_color_hex='#000000'): show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/Layout-Nodes#cr-image-grid-panel" border_color = get_color_values(border_color, border_color_hex, color_mapping) # Convert PIL images images = [tensor2pil(image) for image in images] # Apply borders and outlines to each image images = apply_outline_and_border(images, outline_thickness, outline_color, border_thickness, border_color) combined_image = make_grid_panel(images, max_columns) image_out = pil2tensor(combined_image) return (image_out, show_help, ) #---------------------------------------------------------------------------------------------------------------------# class CR_ImageBorder: @classmethod def INPUT_TYPES(s): return {"required": { "image": ("IMAGE",), "top_thickness": ("INT", {"default": 0, "min": 0, "max": 4096}), "bottom_thickness": ("INT", {"default": 0, "min": 0, "max": 4096}), "left_thickness": ("INT", {"default": 0, "min": 0, "max": 4096}), "right_thickness": ("INT", {"default": 0, "min": 0, "max": 4096}), "border_color": (COLORS,), "outline_thickness": ("INT", {"default": 0, "min": 0, "max": 1024}), "outline_color": (COLORS[1:],), }, "optional": { "border_color_hex": ("STRING", {"multiline": False, "default": "#000000"}) } } RETURN_TYPES = ("IMAGE", "STRING", ) RETURN_NAMES = ("image", "show_help", ) FUNCTION = "make_panel" CATEGORY = icons.get("Comfyroll/Graphics/Layout") def make_panel(self, image, top_thickness, bottom_thickness, left_thickness, right_thickness, border_color, outline_thickness, outline_color, border_color_hex='#000000'): images = [] border_color = get_color_values(border_color, border_color_hex, color_mapping) for img in image: img = tensor2pil(img) # Apply the outline if outline_thickness > 0: img = ImageOps.expand(img, outline_thickness, fill=outline_color) # Apply the borders if left_thickness > 0 or right_thickness > 0 or top_thickness > 0 or bottom_thickness > 0: img = ImageOps.expand(img, (left_thickness, top_thickness, right_thickness, bottom_thickness), fill=border_color) images.append(pil2tensor(img)) images = torch.cat(images, dim=0) show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/Layout-Nodes#cr-image-border" return (images, show_help, ) #---------------------------------------------------------------------------------------------------------------------# class CR_ColorPanel: @classmethod def INPUT_TYPES(s): return {"required": { "panel_width": ("INT", {"default": 512, "min": 8, "max": 4096}), "panel_height": ("INT", {"default": 512, "min": 8, "max": 4096}), "fill_color": (COLORS,), }, "optional": { "fill_color_hex": ("STRING", {"multiline": False, "default": "#000000"}) } } RETURN_TYPES = ("IMAGE", "STRING", ) RETURN_NAMES = ("image", "show_help", ) FUNCTION = "make_panel" CATEGORY = icons.get("Comfyroll/Graphics/Layout") def make_panel(self, panel_width, panel_height, fill_color, fill_color_hex='#000000'): fill_color = get_color_values(fill_color, fill_color_hex, color_mapping) size = (panel_width, panel_height) panel = Image.new('RGB', size, fill_color) show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/Layout-Nodes#cr-color-panel" return (pil2tensor(panel), show_help, ) #---------------------------------------------------------------------------------------------------------------------# class CR_SimpleTextPanel: @classmethod def INPUT_TYPES(s): font_dir = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), "fonts") file_list = [f for f in os.listdir(font_dir) if os.path.isfile(os.path.join(font_dir, f)) and f.lower().endswith(".ttf")] return {"required": { "panel_width": ("INT", {"default": 512, "min": 8, "max": 4096}), "panel_height": ("INT", {"default": 512, "min": 8, "max": 4096}), "text": ("STRING", {"multiline": True, "default": "text"}), "font_name": (file_list,), "font_color": (COLORS,), "font_size": ("INT", {"default": 100, "min": 0, "max": 1024}), "font_outline_thickness": ("INT", {"default": 0, "min": 0, "max": 50}), "font_outline_color": (COLORS,), "background_color": (COLORS,), "align": (ALIGN_OPTIONS, ), "justify": (JUSTIFY_OPTIONS, ), }, "optional": { "font_color_hex": ("STRING", {"multiline": False, "default": "#000000"}), "bg_color_hex": ("STRING", {"multiline": False, "default": "#000000"}), } } RETURN_TYPES = ("IMAGE", "STRING", ) RETURN_NAMES = ("image", "show_help", ) FUNCTION = "layout" CATEGORY = icons.get("Comfyroll/Graphics/Layout") def layout(self, panel_width, panel_height, text, align, justify, font_name, font_color, font_size, font_outline_thickness, font_outline_color, background_color, font_color_hex='#000000', font_outline_color_hex='#000000', bg_color_hex='#000000'): # Get RGB values for the text and background colors font_color = get_color_values(font_color, font_color_hex, color_mapping) outline_color = get_color_values(font_outline_color, font_outline_color_hex, color_mapping) bg_color = get_color_values(background_color, bg_color_hex, color_mapping) # Set defaults margins = 50 line_spacing = 0 position_x = 0 position_y = 0 rotation_angle = 0 rotation_options = "image center" ### Create text panels panel = text_panel(panel_width, panel_height, text, font_name, font_size, font_color, font_outline_thickness, outline_color, bg_color, margins, line_spacing, position_x, position_y, align, justify, rotation_angle, rotation_options) show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/Layout-Nodes#cr-simple-text-panel" return (pil2tensor(panel), show_help, ) #---------------------------------------------------------------------------------------------------------------------# class CR_OverlayTransparentImage: @classmethod def INPUT_TYPES(s): return {"required": { "back_image": ("IMAGE",), "overlay_image": ("IMAGE",), "transparency": ("FLOAT", {"default": 0.0, "min": 0.0, "max": 1.0, "step": 0.1}), "offset_x": ("INT", {"default": 0, "min": -4096, "max": 4096}), "offset_y": ("INT", {"default": 0, "min": -4096, "max": 4096}), "rotation_angle": ("FLOAT", {"default": 0.0, "min": -360.0, "max": 360.0, "step": 0.1}), "overlay_scale_factor": ("FLOAT", {"default": 1.000, "min": 0.000, "max": 100.000, "step": 0.001}), } } RETURN_TYPES = ("IMAGE", ) FUNCTION = "overlay_image" CATEGORY = icons.get("Comfyroll/Graphics/Layout") def overlay_image(self, back_image, overlay_image, transparency, offset_x, offset_y, rotation_angle, overlay_scale_factor=1.0): show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/Layout-Nodes#cr-overlay-transparent-image" # Create PIL images for the text and background layers and text mask back_image = tensor2pil(back_image) overlay_image = tensor2pil(overlay_image) # Apply transparency to overlay image overlay_image.putalpha(int(255 * (1 - transparency))) # Rotate overlay image overlay_image = overlay_image.rotate(rotation_angle, expand=True) # Scale overlay image overlay_width, overlay_height = overlay_image.size new_size = (int(overlay_width * overlay_scale_factor), int(overlay_height * overlay_scale_factor)) overlay_image = overlay_image.resize(new_size, Image.Resampling.LANCZOS) # Calculate centered position relative to the center of the background image center_x = back_image.width // 2 center_y = back_image.height // 2 position_x = center_x - overlay_image.width // 2 + offset_x position_y = center_y - overlay_image.height // 2 + offset_y # Paste the rotated overlay image onto the new back image at the specified position back_image.paste(overlay_image, (position_x, position_y), overlay_image) # Convert the PIL image back to a torch tensor return pil2tensor(back_image), #---------------------------------------------------------------------------------------------------------------------# class CR_FeatheredBorder: @classmethod def INPUT_TYPES(s): return {"required": { "image": ("IMAGE",), "top_thickness": ("INT", {"default": 0, "min": 0, "max": 4096}), "bottom_thickness": ("INT", {"default": 0, "min": 0, "max": 4096}), "left_thickness": ("INT", {"default": 0, "min": 0, "max": 4096}), "right_thickness": ("INT", {"default": 0, "min": 0, "max": 4096}), "border_color": (COLORS,), "feather_amount": ("INT", {"default": 0, "min": 0, "max": 1024}), }, "optional": { "border_color_hex": ("STRING", {"multiline": False, "default": "#000000"}) } } RETURN_TYPES = ("IMAGE", "STRING", ) RETURN_NAMES = ("image", "show_help", ) FUNCTION = "make_border" CATEGORY = icons.get("Comfyroll/Graphics/Layout") def make_border(self, image, top_thickness, bottom_thickness, left_thickness, right_thickness, border_color, feather_amount, border_color_hex='#000000'): show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/Layout-Nodes#cr-feathered-border" images = [] border_color = get_color_values(border_color, border_color_hex, color_mapping) for img in image: im = tensor2pil(img) RADIUS = feather_amount # Paste image on white background diam = 2*RADIUS back = Image.new('RGB', (im.size[0]+diam, im.size[1]+diam), border_color) back.paste(im, (RADIUS, RADIUS)) # Create paste mask mask = Image.new('L', back.size, 0) draw = ImageDraw.Draw(mask) x0, y0 = 0, 0 x1, y1 = back.size for d in range(diam+RADIUS): x1, y1 = x1-1, y1-1 alpha = 255 if d 0 or right_thickness > 0 or top_thickness > 0 or bottom_thickness > 0: img = ImageOps.expand(back, (left_thickness, top_thickness, right_thickness, bottom_thickness), fill=border_color) else: img = back images.append(pil2tensor(img)) images = torch.cat(images, dim=0) return (images, show_help, ) #---------------------------------------------------------------------------------------------------------------------# class CR_HalfDropPanel: @classmethod def INPUT_TYPES(s): patterns = ["none", "half drop", "quarter drop", "custom drop %"] return {"required": { "image": ("IMAGE",), "pattern": (patterns,), }, "optional": { "drop_percentage": ("FLOAT", {"default": 0.50, "min": 0.00, "max": 1.00, "step": 0.01}), } } RETURN_TYPES = ("IMAGE", "STRING", ) RETURN_NAMES = ("image", "show_help", ) FUNCTION = "make_panel" CATEGORY = icons.get("Comfyroll/Graphics/Layout") def make_panel(self, image, pattern, drop_percentage=0.5): show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/Layout-Nodes#cr-half-drop-panel" if pattern == "none": return (image, show_help, ) # Convert to PIL image pil_img = tensor2pil(image) pil_img = pil_img.convert('RGBA') x, y = pil_img.size aspect_ratio = x / y d = int(drop_percentage * 100) panel_image = Image.new('RGBA', (x*2, y*2)) if pattern == "half drop": panel_image.paste(pil_img, (0, 0)) panel_image.paste(pil_img, (0, y)) panel_image.paste(pil_img, (x, -y//2)) panel_image.paste(pil_img, (x, y//2)) panel_image.paste(pil_img, (x, 3*y//2)) elif pattern == "quarter drop": panel_image.paste(pil_img, (0, 0)) panel_image.paste(pil_img, (0, y)) panel_image.paste(pil_img, (x, -3*y//4)) panel_image.paste(pil_img, (x, y//4)) panel_image.paste(pil_img, (x, 5*y//4)) elif pattern == "custom drop %": panel_image.paste(pil_img, (0, 0)) panel_image.paste(pil_img, (0, y)) panel_image.paste(pil_img, (x, (d-100)*y//100)) panel_image.paste(pil_img, (x, d*y//100)) panel_image.paste(pil_img, (x, y + d*y//100)) image_out = pil2tensor(panel_image.convert('RGB')) return (image_out, show_help, ) #---------------------------------------------------------------------------------------------------------------------# class CR_DiamondPanel: @classmethod def INPUT_TYPES(s): patterns = ["none", "diamond"] return {"required": { "image": ("IMAGE",), "pattern": (patterns,), } } RETURN_TYPES = ("IMAGE", "STRING", ) RETURN_NAMES = ("image", "show_help", ) FUNCTION = "make_panel" CATEGORY = icons.get("Comfyroll/Graphics/Layout") def make_panel(self, image, pattern, drop_percentage=0.5): show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/Layout-Nodes#cr-diamond-panel" if pattern == "none": return (image, show_help, ) # Convert to PIL image pil_img = tensor2pil(image) pil_img = pil_img.convert('RGBA') x, y = pil_img.size aspect_ratio = x / y d = int(drop_percentage * 100) panel_image = Image.new('RGBA', (x*2, y*2)) if pattern == "diamond": diamond_size = min(x, y) diamond_width = min(x, y * aspect_ratio) diamond_height = min(y, x / aspect_ratio) diamond_mask = Image.new('L', (x, y), 0) draw = ImageDraw.Draw(diamond_mask) # Make sure the polygon points form a diamond shape draw.polygon([(x // 2, 0), (x, y // 2), (x // 2, y), (0, y // 2)], fill=255) # Create a copy of the original image diamond_image = pil_img.copy() # Set alpha channel using the diamond-shaped mask diamond_image.putalpha(diamond_mask) # Paste the diamond-shaped image onto the panel_image at position (0, 0) panel_image.paste(diamond_image, (-x//2, (d-100)*y//100), diamond_image) panel_image.paste(diamond_image, (-x//2, d*y//100), diamond_image) panel_image.paste(diamond_image, (-x//2, y + d*y//100), diamond_image) panel_image.paste(diamond_image, (0, 0), diamond_image) panel_image.paste(diamond_image, (0, y), diamond_image) panel_image.paste(diamond_image, (x//2, (d-100)*y//100), diamond_image) panel_image.paste(diamond_image, (x//2, d*y//100), diamond_image) panel_image.paste(diamond_image, (x//2, y + d*y//100), diamond_image) panel_image.paste(diamond_image, (x, 0), diamond_image) panel_image.paste(diamond_image, (x, y), diamond_image) panel_image.paste(diamond_image, (3*x//2, (d-100)*y//100), diamond_image) panel_image.paste(diamond_image, (3*x//2, d*y//100), diamond_image) panel_image.paste(diamond_image, (3*x//2, y + d*y//100), diamond_image) image_out = pil2tensor(panel_image.convert('RGB')) return (image_out, show_help, ) #---------------------------------------------------------------------------------------------------------------------# class CR_SelectISOSize: @classmethod def INPUT_TYPES(cls): sizes = list(iso_sizes.keys()) return { "required": { "iso_size": (sizes, ), } } RETURN_TYPES =("INT", "INT","STRING", ) RETURN_NAMES =("width", "height","show_help", ) FUNCTION = "get_size" CATEGORY = icons.get("Comfyroll/Utils/Other") def get_size(self, iso_size): if iso_size in iso_sizes: width, height = iso_sizes[iso_size] else: print("Size not found.") show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/Other-Nodes#cr-select-iso-size" return (width, height, show_help, ) #---------------------------------------------------------------------------------------------------------------------# # MAPPINGS #---------------------------------------------------------------------------------------------------------------------# # For reference only, actual mappings are in __init__.py ''' NODE_CLASS_MAPPINGS = { "CR Page Layout": CR_PageLayout, "CR Image Grid Panel": CR_ImageGridPanel, "CR Half Drop Panel": CR_HalfDropPanel, "CR Diamond Panel": CR_DiamondPanel, "CR Image Border": CR_ImageBorder, "CR Feathered Border": CR_FeatheredBorder, "CR Color Panel": CR_ColorPanel, "CR Simple Text Panel": CR_SimpleTextPanel, "CR Overlay Transparent Image": CR_OverlayTransparentImage, #"CR Simple Titles": CR_SimpleTitles, "CR Select ISO Size": CR_SelectISOSize, } '''