xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-24 04:04:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Multiline text mask

Browse Source
This commit is contained in:
kijai 2024-03-31 14:05:47 +03:00
parent d7f04582f7
commit 776bf61380
1 changed files with 41 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
nodes.py
View File

@ -679,7 +679,7 @@ class CreateTextMask:
"text_y": ("INT", {"default": 0,"min": 0, "max": 4096, "step": 1}), "text_y": ("INT", {"default": 0,"min": 0, "max": 4096, "step": 1}),
"font_size": ("INT", {"default": 32,"min": 8, "max": 4096, "step": 1}), "font_size": ("INT", {"default": 32,"min": 8, "max": 4096, "step": 1}),
"font_color": ("STRING", {"default": "white"}), "font_color": ("STRING", {"default": "white"}),
"text": ("STRING", {"default": "HELLO!"}), "text": ("STRING", {"default": "HELLO!", "multiline": True}),
"font": (folder_paths.get_filename_list("kjnodes_fonts"), ), "font": (folder_paths.get_filename_list("kjnodes_fonts"), ),
"width": ("INT", {"default": 512,"min": 16, "max": 4096, "step": 1}), "width": ("INT", {"default": 512,"min": 16, "max": 4096, "step": 1}),
"height": ("INT", {"default": 512,"min": 16, "max": 4096, "step": 1}), "height": ("INT", {"default": 512,"min": 16, "max": 4096, "step": 1}),
@ -689,7 +689,7 @@ class CreateTextMask:
} }
def createtextmask(self, frames, width, height, invert, text_x, text_y, text, font_size, font_color, font, start_rotation, end_rotation): def createtextmask(self, frames, width, height, invert, text_x, text_y, text, font_size, font_color, font, start_rotation, end_rotation):
# Define the number of images in the batch # Define the number of images in the batch
batch_size = frames batch_size = frames
out = [] out = []
masks = [] masks = []
@ -703,17 +703,47 @@ class CreateTextMask:
image = Image.new("RGB", (width, height), "black") image = Image.new("RGB", (width, height), "black")
draw = ImageDraw.Draw(image) draw = ImageDraw.Draw(image)
font = ImageFont.truetype(font_path, font_size) font = ImageFont.truetype(font_path, font_size)
text_width = font.getlength(text)
text_height = font_size # Split the text into words
text_center_x = text_x + text_width / 2 words = text.split()
text_center_y = text_y + text_height / 2
try: # Initialize variables for line creation
draw.text((text_x, text_y), text, font=font, fill=font_color, features=['-liga']) lines = []
except: current_line = []
draw.text((text_x, text_y), text, font=font, fill=font_color) current_line_width = 0
# Iterate through words to create lines
for word in words:
word_width = font.getsize(word)[0]
if current_line_width + word_width <= width - 2 * text_x:
current_line.append(word)
current_line_width += word_width + font.getsize(" ")[0] # Add space width
else:
lines.append(" ".join(current_line))
current_line = [word]
current_line_width = word_width
# Add the last line if it's not empty
if current_line:
lines.append(" ".join(current_line))
# Draw each line of text separately
y_offset = text_y
for line in lines:
text_width = font.getlength(line)
text_height = font_size
text_center_x = text_x + text_width / 2
text_center_y = y_offset + text_height / 2
try:
draw.text((text_x, y_offset), line, font=font, fill=font_color, features=['-liga'])
except:
draw.text((text_x, y_offset), line, font=font, fill=font_color)
y_offset += text_height # Move to the next line
if start_rotation != end_rotation: if start_rotation != end_rotation:
image = image.rotate(rotation, center=(text_center_x, text_center_y)) image = image.rotate(rotation, center=(text_center_x, text_center_y))
rotation += rotation_increment rotation += rotation_increment
image = np.array(image).astype(np.float32) / 255.0 image = np.array(image).astype(np.float32) / 255.0
image = torch.from_numpy(image)[None,] image = torch.from_numpy(image)[None,]
mask = image[:, :, :, 0] mask = image[:, :, :, 0]