Initial commit - Text On Image node with fonts

__init__.py

@@ -0,0 +1,179 @@
+import os
+from PIL import Image, ImageDraw, ImageFont
+import numpy as np
+import torch
+import logging
+
+class TextOnImage:
+    def __init__(self):
+        pass
+
+    @classmethod
+    def INPUT_TYPES(s):
+        font_dir = os.path.join(os.path.dirname(__file__), "font")
+        font_files = [f for f in os.listdir(font_dir) if f.endswith(('.ttc', '.ttf'))]
+        if not font_files:
+            font_files = ["default"]
+
+        return {
+            "required": {
+                "text": ("STRING", {"default": "", "multiline": True}),
+                "image": ("IMAGE",),
+                "x": ("INT", {"default": 0, "min": 0, "max": 4096, "step": 1}),
+                "y": ("INT", {"default": 0, "min": 0, "max": 4096, "step": 1}),
+                "font_size": ("INT", {"default": 12, "min": 0, "max": 320, "step": 1}),
+                "text_color": ("STRING", {"default": "#ffffff"}),
+                "text_opacity": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01}),
+                "stroke_width": ("INT", {"default": 0, "min": 0, "max": 100, "step": 1}),
+                "stroke_color": ("STRING", {"default": "#000000"}),
+                "stroke_opacity": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01}),
+                "shadow_x": ("INT", {"default": 0, "min": -100, "max": 100, "step": 1}),
+                "shadow_y": ("INT", {"default": 0, "min": -100, "max": 100, "step": 1}),
+                "shadow_color": ("STRING", {"default": "#000000"}),
+                "shadow_opacity": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01}),
+                "font_file": (font_files, {"default": font_files[0]}),
+            },
+        }
+
+    RETURN_TYPES = ("IMAGE",)
+    FUNCTION = "apply_text"
+    CATEGORY = "💀Text On Image"
+
+    def hex_to_rgb(self, hex_color):
+        """将 HEX 色值转换为 RGB 元组"""
+        hex_color = hex_color.lstrip('#')
+        return tuple(int(hex_color[i:i+2], 16) for i in (0, 2, 4))
+
+    def apply_text(self, text, image, x, y, font_size, text_color, text_opacity,
+                  stroke_width, stroke_color, stroke_opacity,
+                  shadow_x, shadow_y, shadow_color, shadow_opacity,
+                  font_file):
+        if text == "":
+            logging.info("Text is empty, returning original image.")
+            return (image,)
+
+        # 将输入的 torch.Tensor 转换为 PIL 图像
+        img = image[0].cpu().numpy()
+        img = (img * 255).astype(np.uint8)
+        base_img = Image.fromarray(img).convert('RGBA')
+        img_width, img_height = base_img.size
+        logging.info(f"Image size: {img_width}x{img_height}")
+
+        # 构建字体文件路径
+        font_dir = os.path.join(os.path.dirname(__file__), "font")
+        font_path = os.path.join(font_dir, font_file)
+        logging.info(f"Font path: {font_path}")
+
+        # 加载字体
+        try:
+            font = ImageFont.truetype(font_path, font_size)
+            logging.info(f"Font loaded successfully: {font_path}")
+        except IOError as e:
+            logging.warning(f"Failed to load font {font_path}: {e}, using default font.")
+            font = ImageFont.load_default()
+
+        # 转换颜色
+        text_color_rgb = self.hex_to_rgb(text_color)
+        stroke_color_rgb = self.hex_to_rgb(stroke_color)
+        shadow_color_rgb = None if shadow_color == "none" else self.hex_to_rgb(shadow_color)
+
+        # 使用 ImageDraw.textbbox 精确计算文字区域
+        temp_img = Image.new('RGBA', (1, 1), (0, 0, 0, 0))
+        temp_draw = ImageDraw.Draw(temp_img)
+        text_bbox = temp_draw.textbbox((0, 0), text, font=font)
+        text_width = text_bbox[2] - text_bbox[0]
+        text_height = text_bbox[3] - text_bbox[1]
+
+        # 增加安全高度
+        safety_margin = font_size // 2
+        text_height += safety_margin
+
+        logging.info(f"Text dimensions: width={text_width}, height={text_height}")
+
+        # 计算总padding（包括描边和投影）
+        padding = max(stroke_width * 2, abs(shadow_x), abs(shadow_y))
+
+        # 调整文字位置
+        text_left = max(0, x - padding)
+        text_top = max(0, y - padding)
+        text_right = min(img_width, x + text_width + padding)
+        text_bottom = min(img_height, y + text_height + padding)
+
+        # 如果文字超出图像底部，向上调整
+        if text_bottom > img_height:
+            offset = text_bottom - img_height
+            text_top = max(0, text_top - offset)
+            text_bottom = img_height
+
+        logging.info(f"Canvas bounds: left={text_left}, top={text_top}, right={text_right}, bottom={text_bottom}")
+
+        # 创建文字层
+        canvas_width = text_right - text_left
+        canvas_height = text_bottom - text_top
+        if canvas_width <= 0 or canvas_height <= 0:
+            logging.warning("Canvas dimensions are invalid, returning original image.")
+            return (image,)
+        canvas = Image.new('RGBA', (canvas_width, canvas_height), (0, 0, 0, 0))
+        canvas_draw = ImageDraw.Draw(canvas)
+        logging.info(f"Canvas created: {canvas_width}x{canvas_height}")
+
+        # 计算文字在画布中的相对位置（y 作为顶部位置）
+        relative_x = x - text_left
+        relative_y = y - text_top
+        logging.info(f"Relative text position: x={relative_x}, y={relative_y}")
+
+        # 绘制投影
+        if shadow_color_rgb and (shadow_x != 0 or shadow_y != 0):
+            try:
+                shadow_pos = (relative_x + shadow_x, relative_y + shadow_y)
+                canvas_draw.text(shadow_pos, text, font=font, 
+                               fill=shadow_color_rgb + (int(255 * shadow_opacity),))
+                logging.info("Shadow drawn successfully.")
+            except Exception as e:
+                logging.error(f"Failed to draw shadow: {e}")
+
+        # 绘制描边
+        if stroke_width > 0:
+            try:
+                for dx in range(-stroke_width, stroke_width + 1):
+                    for dy in range(-stroke_width, stroke_width + 1):
+                        if dx != 0 or dy != 0:
+                            stroke_pos = (relative_x + dx, relative_y + dy)
+                            canvas_draw.text(stroke_pos, text, font=font, 
+                                           fill=stroke_color_rgb + (int(255 * stroke_opacity),))
+                logging.info(f"Stroke drawn with width {stroke_width}.")
+            except Exception as e:
+                logging.error(f"Failed to draw stroke: {e}")
+
+        # 绘制主文字
+        try:
+            canvas_draw.text((relative_x, relative_y), text, font=font, 
+                           fill=text_color_rgb + (int(255 * text_opacity),))
+            logging.info("Main text drawn successfully.")
+        except Exception as e:
+            logging.error(f"Failed to draw main text: {e}")
+            return (image,)
+
+        # 将文字画布粘贴到主图像上
+        try:
+            base_img.paste(canvas, (text_left, text_top), canvas)
+            logging.info("Canvas pasted to base image successfully.")
+        except Exception as e:
+            logging.error(f"Failed to paste canvas to base image: {e}")
+            return (image,)
+
+        # 将 PIL.Image 转换回 torch.Tensor
+        img = np.array(base_img).astype(np.float32) / 255.0
+        img = torch.from_numpy(img)[None,]
+        logging.info("Image converted back to tensor.")
+
+        return (img,)
+
+# 注册节点
+NODE_CLASS_MAPPINGS = {
+    "TextOnImage": TextOnImage
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "TextOnImage": "💀Text On Image"
+}