149 lines
6.7 KiB
Python
149 lines
6.7 KiB
Python
import torch
|
|
import random
|
|
from PIL import Image, ImageFont, ImageDraw
|
|
from .imagefunc import AnyType, log, tensor2pil, pil2tensor, image2mask, get_resource_dir, RGB2RGBA, random_numbers
|
|
|
|
|
|
any = AnyType("*")
|
|
|
|
class TextImage_v2:
|
|
|
|
def __init__(self):
|
|
self.NODE_NAME = 'TextImageV2'
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(self):
|
|
(_, FONT_DICT) = get_resource_dir()
|
|
FONT_LIST = list(FONT_DICT.keys())
|
|
|
|
layout_list = ['horizontal', 'vertical']
|
|
return {
|
|
"required": {
|
|
"text": ("STRING", {"multiline": True, "default": "Text"}),
|
|
"font_file": (FONT_LIST,),
|
|
"spacing": ("INT", {"default": 0, "min": -9999, "max": 9999, "step": 1}),
|
|
"leading": ("INT", {"default": 0, "min": -9999, "max": 9999, "step": 1}),
|
|
"horizontal_border": ("FLOAT", {"default": 5, "min": -100, "max": 100, "step": 0.1}), # 左右距离百分比，横排为距左侧，竖排为距右侧
|
|
"vertical_border": ("FLOAT", {"default": 5, "min": -100, "max": 100, "step": 0.1}), # 上距离百分比
|
|
"scale": ("FLOAT", {"default": 80, "min": 0.1, "max": 999, "step": 0.01}), # 整体大小与画面长宽比，横排与宽比，竖排与高比
|
|
"variation_range": ("INT", {"default": 0, "min": 0, "max": 100, "step": 1}), # 随机大小和位置范围
|
|
"variation_seed": ("INT", {"default": 123, "min": 0, "max": 999999999999, "step": 1}), # 随机种子
|
|
"layout": (layout_list,), # 横排or竖排
|
|
"width": ("INT", {"default": 512, "min": 4, "max": 999999, "step": 1}),
|
|
"height": ("INT", {"default": 512, "min": 4, "max": 999999, "step": 1}),
|
|
"text_color": ("STRING", {"default": "#FFA000"}), # 文字颜色
|
|
"background_color": ("STRING", {"default": "#FFFFFF"}), # 背景颜色
|
|
},
|
|
"optional": {
|
|
"size_as": (any, {}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("IMAGE", "MASK",)
|
|
RETURN_NAMES = ("image", "mask",)
|
|
FUNCTION = 'text_image'
|
|
CATEGORY = '😺dzNodes/LayerUtility'
|
|
|
|
def text_image(self, text, font_file, spacing, leading, horizontal_border, vertical_border, scale,
|
|
variation_range, variation_seed, layout, width, height, text_color, background_color,
|
|
size_as=None
|
|
):
|
|
|
|
|
|
(_, FONT_DICT) = get_resource_dir()
|
|
FONT_LIST = list(FONT_DICT.keys())
|
|
|
|
if size_as is not None:
|
|
width, height = tensor2pil(size_as).size
|
|
text_table = []
|
|
max_char_in_line = 0
|
|
total_char = 0
|
|
lines = []
|
|
text_lines = text.split("\n")
|
|
for l in text_lines:
|
|
if len(l) > 0:
|
|
lines.append(l)
|
|
total_char += len(l)
|
|
if len(l) > max_char_in_line:
|
|
max_char_in_line = len(l)
|
|
else:
|
|
lines.append(" ")
|
|
|
|
if layout == 'vertical':
|
|
char_horizontal_size = width // len(lines)
|
|
char_vertical_size = height // max_char_in_line
|
|
char_size = min(char_horizontal_size, char_vertical_size)
|
|
else:
|
|
char_horizontal_size = width // max_char_in_line
|
|
char_vertical_size = height // len(lines)
|
|
char_size = min(char_horizontal_size, char_vertical_size)
|
|
|
|
# Adjust char_size based on scale
|
|
char_size = int(char_size * scale / 100)
|
|
char_size_h = int(char_size/width*height)
|
|
spacing = int(spacing * scale / 100)
|
|
leading = int(leading * scale / 100)
|
|
|
|
# Calculate the starting position based on the center
|
|
text_width = char_size * max_char_in_line + spacing * (max_char_in_line - 1)
|
|
text_height = char_size * len(lines) + leading * (len(lines) - 1)
|
|
|
|
# Adjust start_y to control the vertical position of the text center
|
|
start_x = int((width - text_width) * horizontal_border / 100)
|
|
|
|
# Ensure text does not go out of bounds and smoothly transitions
|
|
start_y = int((height - text_height) * vertical_border / 100-(char_size_h-char_size))
|
|
|
|
# calculate every char position and size to a table list
|
|
for i in range(len(lines)):
|
|
_x = start_x
|
|
_y = start_y
|
|
line_table = []
|
|
line_random = random_numbers(total=len(lines[i]),
|
|
random_range=int(char_size * variation_range / 25),
|
|
seed=variation_seed, sum_of_numbers=0)
|
|
for j in range(0, len(lines[i])):
|
|
offset = int((char_size + line_random[j]) * variation_range / 250)
|
|
font_size = char_size + line_random[j]
|
|
if font_size < 4:
|
|
font_size = 4
|
|
axis_x = _x + offset // 3 if random.random() > 0.5 else _x - offset // 3
|
|
axis_y = _y + offset // 3 if random.random() > 0.5 else _y - offset // 3
|
|
char_dict = {'char': lines[i][j],
|
|
'axis': (axis_x, axis_y),
|
|
'size': font_size}
|
|
line_table.append(char_dict)
|
|
if layout == 'vertical':
|
|
_y += char_size + line_random[j] + spacing
|
|
else:
|
|
_x += char_size + line_random[j] + spacing
|
|
if layout == 'vertical':
|
|
start_x -= leading * (i + 1) + char_size
|
|
else:
|
|
start_y += leading * (i + 1) + char_size
|
|
text_table.append(line_table)
|
|
|
|
# draw char
|
|
_mask = Image.new('RGB', size=(width, height), color='black')
|
|
draw = ImageDraw.Draw(_mask)
|
|
for l in range(len(lines)):
|
|
for c in range(len(lines[l])):
|
|
font_path = FONT_DICT.get(font_file)
|
|
font_size = text_table[l][c].get('size')
|
|
font = ImageFont.truetype(font_path, font_size)
|
|
draw.text(text_table[l][c].get('axis'), text_table[l][c].get('char'), font=font, fill='white')
|
|
_canvas = Image.new('RGB', size=(width, height), color=background_color)
|
|
_color = Image.new('RGB', size=(width, height), color=text_color)
|
|
_canvas.paste(_color, mask=_mask.convert('L'))
|
|
_canvas = RGB2RGBA(_canvas, _mask)
|
|
log(f"{self.NODE_NAME} Processed.", message_type='finish')
|
|
return (pil2tensor(_canvas), image2mask(_mask),)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"LayerUtility: TextImage V2": TextImage_v2
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"LayerUtility: TextImage V2": "LayerUtility: TextImage V2"
|
|
}
|