| from PIL import Image, ImageDraw, ImageFont |
|
|
| import folder_paths |
|
|
|
|
| class ImageText: |
| def __init__(self): |
| pass |
|
|
| @classmethod |
| def INPUT_TYPES(cls): |
| return { |
| "required": { |
| "text": ("STRING", {"multiline": False}), |
| "font": (folder_paths.get_filename_list("fonts"),), |
| "size": ("INT", { |
| "default": 28, |
| "min": 1, |
| "step": 1 |
| }), |
| "red": ("INT", { |
| "default": 255, |
| "max": 255, |
| "step": 1 |
| }), |
| "green": ("INT", { |
| "default": 255, |
| "max": 255, |
| "step": 1 |
| }), |
| "blue": ("INT", { |
| "default": 255, |
| "max": 255, |
| "step": 1 |
| }), |
| "alpha": ("FLOAT", { |
| "default": 1.0, |
| "max": 1.0, |
| "step": 0.01 |
| }), |
| "margin_x": ("INT", { |
| "default": 0, |
| "step": 1 |
| }), |
| "margin_y": ("INT", { |
| "default": 0, |
| "step": 1 |
| }), |
| }, |
| } |
|
|
| RETURN_TYPES = ("IMAGE",) |
| FUNCTION = "node" |
| CATEGORY = "image/draw" |
|
|
| def node(self, text, font, size, red, green, blue, alpha, margin_x, margin_y): |
| outline_size = 0 |
| outline_red = 255 |
| outline_green = 255 |
| outline_blue = 255 |
|
|
| return ImageTextOutlined().node( |
| text, |
| font, |
| size, |
| red, |
| green, |
| blue, |
| outline_size, |
| outline_red, |
| outline_green, |
| outline_blue, |
| alpha, |
| margin_x, |
| margin_y |
| ) |
|
|
|
|
| class ImageTextOutlined: |
| def __init__(self): |
| pass |
|
|
| @classmethod |
| def INPUT_TYPES(cls): |
| return { |
| "required": { |
| "text": ("STRING", {"multiline": False}), |
| "font": (folder_paths.get_filename_list("fonts"),), |
| "size": ("INT", { |
| "default": 28, |
| "min": 1, |
| "step": 1 |
| }), |
| "red": ("INT", { |
| "default": 255, |
| "max": 255, |
| "step": 1 |
| }), |
| "green": ("INT", { |
| "default": 255, |
| "max": 255, |
| "step": 1 |
| }), |
| "blue": ("INT", { |
| "default": 255, |
| "max": 255, |
| "step": 1 |
| }), |
| "outline_size": ("INT", { |
| "default": 1, |
| "step": 1 |
| }), |
| "outline_red": ("INT", { |
| "default": 0, |
| "max": 255, |
| "step": 1 |
| }), |
| "outline_green": ("INT", { |
| "default": 0, |
| "max": 255, |
| "step": 1 |
| }), |
| "outline_blue": ("INT", { |
| "default": 0, |
| "max": 255, |
| "step": 1 |
| }), |
| "alpha": ("FLOAT", { |
| "default": 1.0, |
| "max": 1.0, |
| "step": 0.01 |
| }), |
| "margin_x": ("INT", { |
| "default": 0, |
| "step": 1 |
| }), |
| "margin_y": ("INT", { |
| "default": 0, |
| "step": 1 |
| }), |
| }, |
| } |
|
|
| RETURN_TYPES = ("IMAGE",) |
| FUNCTION = "node" |
| CATEGORY = "image/draw" |
|
|
| def node( |
| self, text, font, size, red, green, blue, outline_size, outline_red, outline_green, outline_blue, alpha, |
| margin_x, margin_y |
| ): |
| font_path = folder_paths.get_full_path("fonts", font) |
| font = ImageFont.truetype(font_path, size, encoding="unic") |
|
|
| (left, top, right, bottom) = font.getbbox(text) |
|
|
| canvas = Image.new("RGBA", ( |
| right + (margin_x + outline_size) * 2, |
| bottom - top + (margin_y + outline_size) * 2 |
| ), (0, 0, 0, 0)) |
|
|
| draw = ImageDraw.Draw(canvas) |
|
|
| draw.text( |
| (margin_x + outline_size, margin_y + outline_size - top), |
| text=text, fill=(red, green, blue, int(alpha * 255)), |
| stroke_fill=(outline_red, outline_green, outline_blue, int(alpha * 255)), |
| stroke_width=outline_size, font=font |
| ) |
|
|
| return (canvas.image_to_tensor().unsqueeze(0),) |
|
|
|
|
| class ImageTextMultiline: |
| def __init__(self): |
| pass |
|
|
| @classmethod |
| def INPUT_TYPES(cls): |
| return { |
| "required": { |
| "text": ("STRING", {"multiline": True}), |
| "font": (folder_paths.get_filename_list("fonts"),), |
| "align": (["left", "center", "right"],), |
| "size": ("INT", { |
| "default": 28, |
| "min": 1, |
| "step": 1 |
| }), |
| "red": ("INT", { |
| "default": 255, |
| "max": 255, |
| "step": 1 |
| }), |
| "green": ("INT", { |
| "default": 255, |
| "max": 255, |
| "step": 1 |
| }), |
| "blue": ("INT", { |
| "default": 255, |
| "max": 255, |
| "step": 1 |
| }), |
| "alpha": ("FLOAT", { |
| "default": 1.0, |
| "max": 1.0, |
| "step": 0.01 |
| }), |
| "margin_x": ("INT", { |
| "default": 0, |
| "step": 1 |
| }), |
| "margin_y": ("INT", { |
| "default": 0, |
| "step": 1 |
| }), |
| }, |
| } |
|
|
| RETURN_TYPES = ("IMAGE",) |
| FUNCTION = "node" |
| CATEGORY = "image/draw" |
|
|
| def node(self, text, font, align, size, red, green, blue, alpha, margin_x, margin_y): |
| outline_size = 0 |
| outline_red = 255 |
| outline_green = 255 |
| outline_blue = 255 |
|
|
| return ImageTextMultilineOutlined().node( |
| text, |
| font, |
| align, |
| size, |
| red, |
| green, |
| blue, |
| outline_size, |
| outline_red, |
| outline_green, |
| outline_blue, |
| alpha, |
| margin_x, |
| margin_y |
| ) |
|
|
|
|
| class ImageTextMultilineOutlined: |
| def __init__(self): |
| pass |
|
|
| @classmethod |
| def INPUT_TYPES(cls): |
| return { |
| "required": { |
| "text": ("STRING", {"multiline": True}), |
| "font": (folder_paths.get_filename_list("fonts"),), |
| "align": (["left", "center", "right"],), |
| "size": ("INT", { |
| "default": 28, |
| "min": 1, |
| "step": 1 |
| }), |
| "red": ("INT", { |
| "default": 255, |
| "max": 255, |
| "step": 1 |
| }), |
| "green": ("INT", { |
| "default": 255, |
| "max": 255, |
| "step": 1 |
| }), |
| "blue": ("INT", { |
| "default": 255, |
| "max": 255, |
| "step": 1 |
| }), |
| "outline_size": ("INT", { |
| "default": 1, |
| "step": 1 |
| }), |
| "outline_red": ("INT", { |
| "default": 0, |
| "max": 255, |
| "step": 1 |
| }), |
| "outline_green": ("INT", { |
| "default": 0, |
| "max": 255, |
| "step": 1 |
| }), |
| "outline_blue": ("INT", { |
| "default": 0, |
| "max": 255, |
| "step": 1 |
| }), |
| "alpha": ("FLOAT", { |
| "default": 1.0, |
| "max": 1.0, |
| "step": 0.01 |
| }), |
| "margin_x": ("INT", { |
| "default": 0, |
| "step": 1 |
| }), |
| "margin_y": ("INT", { |
| "default": 0, |
| "step": 1 |
| }), |
| }, |
| } |
|
|
| RETURN_TYPES = ("IMAGE",) |
| FUNCTION = "node" |
| CATEGORY = "image/draw" |
|
|
| def node( |
| self, text, font, align, size, red, green, blue, outline_size, outline_red, outline_green, outline_blue, |
| alpha, margin_x, margin_y |
| ): |
| font_path = folder_paths.get_full_path("fonts", font) |
| font = ImageFont.truetype(font_path, size, encoding="unic") |
|
|
| lines = text.split('\n').__len__() |
| (_, top, _, _) = font.getbbox(text) |
|
|
| canvas = Image.new("RGBA", (0, 0)) |
| draw = ImageDraw.Draw(canvas) |
| text_size = draw.multiline_textbbox((0, 0), text, font) |
|
|
| canvas = Image.new("RGBA", ( |
| text_size[2] + (margin_x + outline_size) * 2, |
| text_size[3] - top + (margin_y + (outline_size * lines)) * 2 |
| ), (0, 0, 0, 0)) |
|
|
| draw = ImageDraw.Draw(canvas) |
|
|
| draw.text( |
| (margin_x + outline_size, margin_y + outline_size - top), |
| text=text, fill=(red, green, blue, int(alpha * 255)), |
| stroke_fill=(outline_red, outline_green, outline_blue, int(alpha * 255)), |
| stroke_width=outline_size, font=font, align=align |
| ) |
|
|
| return (canvas.image_to_tensor().unsqueeze(0),) |
|
|
|
|
| NODE_CLASS_MAPPINGS = { |
| "ImageText": ImageText, |
| "ImageTextOutlined": ImageTextOutlined, |
| "ImageTextMultiline": ImageTextMultiline, |
| "ImageTextMultilineOutlined": ImageTextMultilineOutlined |
| } |
|
|
