every_holiday/VRCT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

👍️[Update] Model : Overlay LargeLogのUIを修正

Browse Source
This commit is contained in:
misyaguziya
2024-11-14 14:07:51 +09:00
parent 547a410ecc
commit d7231e17b8
1 changed files with 81 additions and 111 deletions
Download Patch File Download Diff File Expand all files Collapse all files

192
src-python/models/overlay/overlay_image.py
View File

@@ -15,60 +15,53 @@ class OverlayImage:
self.message_log = [] self.message_log = []
@staticmethod @staticmethod
def concatenateImagesVertically(img1: Image, img2: Image, margin:int=0) -> Image: def concatenateImagesVertically(img1: Image, img2: Image, margin: int = 0) -> Image:
total_height = img1.height + img2.height + margin total_height = img1.height + img2.height + margin
dst = Image.new("RGBA", (img1.width, total_height)) dst = Image.new("RGBA", (img1.width, total_height))
dst.paste(img1, (0, 0)) dst.paste(img1, (0, 0))
dst.paste(img2, (0, img1.height)) dst.paste(img2, (0, img1.height + margin))
return dst return dst
@staticmethod @staticmethod
def addImageMargin(image: Image, top: int, right: int, bottom: int, left: int, color: Tuple[int, int, int, int]) -> Image: def addImageMargin(image: Image, top: int, right: int, bottom: int, left: int, color: Tuple[int, int, int, int]) -> Image:
width, height = image.size new_width = image.width + right + left
new_width = width + right + left new_height = image.height + top + bottom
new_height = height + top + bottom
result = Image.new(image.mode, (new_width, new_height), color) result = Image.new(image.mode, (new_width, new_height), color)
result.paste(image, (left, top)) result.paste(image, (left, top))
return result return result
@staticmethod @staticmethod
def getUiSizeSmallLog(): def getUiSizeSmallLog() -> dict:
return { return {
"width": int(960*4), "width": 3840,
"height": int(23*4), "height": 92,
"font_size": int(23*4), "font_size": 92,
} }
@staticmethod @staticmethod
def getUiColorSmallLog(ui_type): def getUiColorSmallLog() -> dict:
background_color = (41, 42, 45) colors = {
background_outline_color = (41, 42, 45) "background_color": (41, 42, 45),
text_color = (223, 223, 223) "background_outline_color": (41, 42, 45),
match ui_type: "text_color": (223, 223, 223)
case "default":
pass
case _:
pass
return {
"background_color": background_color,
"background_outline_color": background_outline_color,
"text_color": text_color
} }
return colors
def createTextboxSmallLog(self, text, language, text_color, base_width, base_height, font_size): def createTextboxSmallLog(self, text:str, language:str, text_color:tuple, base_width:int, base_height:int, font_size:int) -> Image:
font_family = self.LANGUAGES.get(language, "NotoSansJP-Regular") font_family = self.LANGUAGES.get(language, "NotoSansJP-Regular")
img = Image.new("RGBA", (base_width, base_height), (0, 0, 0, 0)) img = Image.new("RGBA", (base_width, base_height), (0, 0, 0, 0))
draw = ImageDraw.Draw(img) draw = ImageDraw.Draw(img)
font_path = os_path.join(os_path.dirname(__file__), "..", "..", "..", "fonts", f"{font_family}.ttf")
try: try:
font = ImageFont.truetype(os_path.join(os_path.dirname(os_path.dirname(os_path.dirname(__file__))), "fonts", f"{font_family}.ttf"), font_size) font = ImageFont.truetype(font_path, font_size)
except Exception: except Exception:
font = ImageFont.truetype(os_path.join(os_path.dirname(__file__), "..", "..", "..", "fonts", f"{font_family}.ttf"), font_size) font = ImageFont.truetype(font_path, font_size)
text_width = draw.textlength(text, font) text_width = draw.textlength(text, font)
character_width = text_width // len(text) character_width = text_width // len(text)
character_line_num = int((base_width) // character_width) - 12 character_line_num = (base_width // character_width) - 12
if len(text) > character_line_num: if len(text) > character_line_num:
text = "\n".join([text[i:i+character_line_num] for i in range(0, len(text), character_line_num)]) text = "\n".join([text[i:i + character_line_num] for i in range(0, len(text), character_line_num)])
text_height = font_size * (len(text.split("\n")) + 1) + 20 text_height = font_size * (len(text.split("\n")) + 1) + 20
img = Image.new("RGBA", (base_width, text_height), (0, 0, 0, 0)) img = Image.new("RGBA", (base_width, text_height), (0, 0, 0, 0))
draw = ImageDraw.Draw(img) draw = ImageDraw.Draw(img)
@@ -78,19 +71,17 @@ class OverlayImage:
draw.text((text_x, text_y), text, text_color, anchor="mm", stroke_width=0, font=font, align="center") draw.text((text_x, text_y), text, text_color, anchor="mm", stroke_width=0, font=font, align="center")
return img return img
def createOverlayImageSmallLog(self, message, your_language, translation="", target_language=None, ui_type="default"): def createOverlayImageSmallLog(self, message:str, your_language:str, translation:str="", target_language:str=None) -> Image:
ui_size = self.getUiSizeSmallLog() ui_size = self.getUiSizeSmallLog()
height = ui_size["height"] width, height, font_size = ui_size["width"], ui_size["height"], ui_size["font_size"]
width = ui_size["width"]
font_size = ui_size["font_size"]
ui_colors = self.getUiColorSmallLog(ui_type) ui_colors = self.getUiColorSmallLog()
text_color = ui_colors["text_color"] text_color = ui_colors["text_color"]
background_color = ui_colors["background_color"] background_color = ui_colors["background_color"]
background_outline_color = ui_colors["background_outline_color"] background_outline_color = ui_colors["background_outline_color"]
img = self.createTextboxSmallLog(message, your_language, text_color, width, height, font_size) img = self.createTextboxSmallLog(message, your_language, text_color, width, height, font_size)
if len(translation) > 0 and target_language is not None: if translation and target_language:
translation_img = self.createTextboxSmallLog(translation, target_language, text_color, width, height, font_size) translation_img = self.createTextboxSmallLog(translation, target_language, text_color, width, height, font_size)
img = self.concatenateImagesVertically(img, translation_img) img = self.concatenateImagesVertically(img, translation_img)
@@ -98,15 +89,14 @@ class OverlayImage:
draw = ImageDraw.Draw(background) draw = ImageDraw.Draw(background)
draw.rounded_rectangle([(0, 0), img.size], radius=50, fill=background_color, outline=background_outline_color, width=5) draw.rounded_rectangle([(0, 0), img.size], radius=50, fill=background_color, outline=background_outline_color, width=5)
img = Image.alpha_composite(background, img) return Image.alpha_composite(background, img)
return img
@staticmethod @staticmethod
def getUiSizeLargeLog(): def getUiSizeLargeLog() -> dict:
return { return {
"width": int(960), "width": 960,
"font_size_large": int(15*2), "font_size_large": 30,
"font_size_small": int(15*2*2/3), "font_size_small": 20,
"margin": 25, "margin": 25,
"radius": 25, "radius": 25,
"padding": 10, "padding": 10,
@@ -114,105 +104,89 @@ class OverlayImage:
} }
@staticmethod @staticmethod
def getUiColorLargeLog(): def getUiColorLargeLog() -> dict:
background_color = (41, 42, 45)
background_outline_color = (41, 42, 45)
text_color_large = (223, 223, 223)
text_color_small = (190, 190, 190)
text_color_send = (97, 151, 180)
text_color_receive = (168, 97, 180)
text_color_time = (120, 120, 120)
return { return {
"background_color": background_color, "background_color": (41, 42, 45),
"background_outline_color": background_outline_color, "background_outline_color": (41, 42, 45),
"text_color_large": text_color_large, "text_color_large": (223, 223, 223),
"text_color_small": text_color_small, "text_color_small": (190, 190, 190),
"text_color_send": text_color_send, "text_color_send": (97, 151, 180),
"text_color_receive": text_color_receive, "text_color_receive": (168, 97, 180),
"text_color_time": text_color_time "text_color_time": (120, 120, 120)
} }
def createTextImageLargeLog(self, message_type, size, text, language): def createTextImageLargeLog(self, message_type:str, size:str, text:str, language:str) -> Image:
ui_size = self.getUiSizeLargeLog() ui_size = self.getUiSizeLargeLog()
font_size_large = ui_size["font_size_large"] font_size = ui_size["font_size_large"] if size == "large" else ui_size["font_size_small"]
font_size_small = ui_size["font_size_small"] text_color = self.getUiColorLargeLog()[f"text_color_{size}"]
ui_width = ui_size["width"]
ui_padding = ui_size["padding"]
ui_color = self.getUiColorLargeLog()
text_color_large = ui_color["text_color_large"]
text_color_small = ui_color["text_color_small"]
font_size = font_size_large if size == "large" else font_size_small
text_color = text_color_large if size == "large" else text_color_small
anchor = "lm" if message_type == "receive" else "rm" anchor = "lm" if message_type == "receive" else "rm"
text_x = 0 if message_type == "receive" else ui_width text_x = 0 if message_type == "receive" else ui_size["width"]
align = "left" if message_type == "receive" else "right" align = "left" if message_type == "receive" else "right"
font_family = self.LANGUAGES.get(language, "NotoSansJP-Regular") font_family = self.LANGUAGES.get(language, "NotoSansJP-Regular")
img = Image.new("RGBA", (0, 0), (0, 0, 0, 0)) img = Image.new("RGBA", (0, 0), (0, 0, 0, 0))
draw = ImageDraw.Draw(img) draw = ImageDraw.Draw(img)
font_path = os_path.join(os_path.dirname(__file__), "..", "..", "..", "fonts", f"{font_family}.ttf")
try: try:
font = ImageFont.truetype(os_path.join(os_path.dirname(os_path.dirname(os_path.dirname(__file__))), "fonts", f"{font_family}.ttf"), font_size) font = ImageFont.truetype(font_path, font_size)
except Exception: except Exception:
font = ImageFont.truetype(os_path.join(os_path.dirname(__file__), "..", "..", "..", "fonts", f"{font_family}.ttf"), font_size) font = ImageFont.truetype(font_path, font_size)
text_width = draw.textlength(text, font) text_width = draw.textlength(text, font)
character_width = text_width // len(text) character_width = text_width // len(text)
character_line_num = int(ui_width // character_width) - 1 # 1 is for margin character_line_num = int((ui_size["width"] // character_width) - 1)
if len(text) > character_line_num: if len(text) > character_line_num:
text = "\n".join([text[i:i+character_line_num] for i in range(0, len(text), character_line_num)]) text = "\n".join([text[i:i + character_line_num] for i in range(0, len(text), character_line_num)])
n_num = len(text.split("\n")) text_height = font_size * len(text.split("\n")) + ui_size["padding"]
text_height = int(font_size*n_num) + ui_padding img = Image.new("RGBA", (ui_size["width"], text_height), (0, 0, 0, 0))
img = Image.new("RGBA", (ui_width, text_height), (0, 0, 0, 0))
draw = ImageDraw.Draw(img) draw = ImageDraw.Draw(img)
text_y = text_height // 2 text_y = text_height // 2
draw.multiline_text((text_x, text_y), text, text_color, anchor=anchor, stroke_width=0, font=font, align=align) draw.multiline_text((text_x, text_y), text, text_color, anchor=anchor, stroke_width=0, font=font, align=align)
return img return img
def createTextImageMessageType(self, message_type, date_time): def createTextImageMessageType(self, message_type:str, date_time:str) -> Image:
ui_size = self.getUiSizeLargeLog() ui_size = self.getUiSizeLargeLog()
ui_width = ui_size["width"]
font_size = ui_size["font_size_small"] font_size = ui_size["font_size_small"]
ui_padding = ui_size["padding"] ui_padding = ui_size["padding"]
ui_color = self.getUiColorLargeLog() ui_color = self.getUiColorLargeLog()
text_color_send = ui_color["text_color_send"] text_color = ui_color[f"text_color_{message_type}"]
text_color_receive = ui_color["text_color_receive"]
text_color_time = ui_color["text_color_time"] text_color_time = ui_color["text_color_time"]
anchor = "lm" if message_type == "receive" else "rm" anchor = "lm" if message_type == "receive" else "rm"
text = "Receive" if message_type == "receive" else "Send" text = "Receive" if message_type == "receive" else "Send"
text_color = text_color_receive if message_type == "receive" else text_color_send
img = Image.new("RGBA", (0, 0), (0, 0, 0, 0)) img = Image.new("RGBA", (0, 0), (0, 0, 0, 0))
draw = ImageDraw.Draw(img) draw = ImageDraw.Draw(img)
font_path = os_path.join(os_path.dirname(__file__), "..", "..", "..", "fonts", "NotoSansJP-Regular.ttf")
try: try:
font = ImageFont.truetype(os_path.join(os_path.dirname(os_path.dirname(os_path.dirname(__file__))), "fonts", "NotoSansJP-Regular.ttf"), font_size) font = ImageFont.truetype(font_path, font_size)
except Exception: except Exception:
font = ImageFont.truetype(os_path.join(os_path.dirname(__file__), "..", "..", "..", "fonts", "NotoSansJP-Regular.ttf"), font_size) font = ImageFont.truetype(font_path, font_size)
text_height = int(font_size) + ui_padding
text_height = font_size + ui_padding
text_width = draw.textlength(date_time, font) text_width = draw.textlength(date_time, font)
character_width = text_width // len(date_time) character_width = text_width // len(date_time)
img = Image.new("RGBA", (ui_width, text_height), (0, 0, 0, 0)) img = Image.new("RGBA", (ui_size["width"], text_height), (0, 0, 0, 0))
draw = ImageDraw.Draw(img) draw = ImageDraw.Draw(img)
text_y = text_height // 2 text_y = text_height // 2
text_time_x = 0 if message_type == "receive" else ui_width - (text_width + character_width) text_time_x = 0 if message_type == "receive" else ui_size["width"] - (text_width + character_width)
text_x = (text_width + character_width) if message_type == "receive" else ui_width text_x = (text_width + character_width) if message_type == "receive" else ui_size["width"]
draw.text((text_time_x, text_y), date_time, text_color_time, anchor=anchor, stroke_width=0, font=font) draw.text((text_time_x, text_y), date_time, text_color_time, anchor=anchor, stroke_width=0, font=font)
draw.text((text_x, text_y), text, text_color, anchor=anchor, stroke_width=0, font=font) draw.text((text_x, text_y), text, text_color, anchor=anchor, stroke_width=0, font=font)
return img return img
def createTextboxLargeLog(self, message_type, message, your_language, translation, target_language, date_time): def createTextboxLargeLog(self, message_type:str, message:str, your_language:str, translation:str, target_language:str, date_time:str) -> Image:
message_type_img = self.createTextImageMessageType(message_type, date_time) message_type_img = self.createTextImageMessageType(message_type, date_time)
if len(translation) > 0 and target_language is not None: if translation and target_language:
img = self.createTextImageLargeLog(message_type, "small", message, your_language) img = self.createTextImageLargeLog(message_type, "small", message, your_language)
translation_img = self.createTextImageLargeLog(message_type, "large",translation, target_language) translation_img = self.createTextImageLargeLog(message_type, "large", translation, target_language)
img = self.concatenateImagesVertically(img, translation_img) img = self.concatenateImagesVertically(img, translation_img)
else: else:
img = self.createTextImageLargeLog(message_type, "large", message, your_language) img = self.createTextImageLargeLog(message_type, "large", message, your_language)
return self.concatenateImagesVertically(message_type_img, img) return self.concatenateImagesVertically(message_type_img, img)
def createOverlayImageLargeLog(self, message_type, message, your_language, translation="", target_language=None): def createOverlayImageLargeLog(self, message_type:str, message:str, your_language:str, translation:str="", target_language:str=None) -> Image:
ui_color = self.getUiColorLargeLog() ui_color = self.getUiColorLargeLog()
background_color = ui_color["background_color"] background_color = ui_color["background_color"]
background_outline_color = ui_color["background_outline_color"] background_outline_color = ui_color["background_outline_color"]
@@ -222,30 +196,27 @@ class OverlayImage:
ui_radius = ui_size["radius"] ui_radius = ui_size["radius"]
ui_clause_margin = ui_size["clause_margin"] ui_clause_margin = ui_size["clause_margin"]
self.message_log.append( self.message_log.append({
{ "message_type": message_type,
"message_type":message_type, "message": message,
"message":message, "your_language": your_language,
"your_language":your_language, "translation": translation,
"translation":translation, "target_language": target_language,
"target_language":target_language, "datetime": datetime.now().strftime("%H:%M")
"datetime":datetime.now().strftime("%H:%M") })
}
)
if len(self.message_log) > 10: if len(self.message_log) > 10:
self.message_log = self.message_log[-10:] self.message_log = self.message_log[-10:]
imgs = [] imgs = [
for log in self.message_log: self.createTextboxLargeLog(
message_type = log["message_type"] log["message_type"],
message = log["message"] log["message"],
your_language = log["your_language"] log["your_language"],
translation = log["translation"] log["translation"],
target_language = log["target_language"] log["target_language"],
date_time = log["datetime"] log["datetime"]) for log in self.message_log
img = self.createTextboxLargeLog(message_type, message, your_language, translation, target_language, date_time) ]
imgs.append(img)
img = imgs[0] img = imgs[0]
for i in imgs[1:]: for i in imgs[1:]:
@@ -256,8 +227,7 @@ class OverlayImage:
background = Image.new("RGBA", (width, height), (0, 0, 0, 0)) background = Image.new("RGBA", (width, height), (0, 0, 0, 0))
draw = ImageDraw.Draw(background) draw = ImageDraw.Draw(background)
draw.rounded_rectangle([(0, 0), (width, height)], radius=ui_radius, fill=background_color, outline=background_outline_color, width=5) draw.rounded_rectangle([(0, 0), (width, height)], radius=ui_radius, fill=background_color, outline=background_outline_color, width=5)
img = Image.alpha_composite(background, img) return Image.alpha_composite(background, img)
return img
if __name__ == "__main__": if __name__ == "__main__":
overlay = OverlayImage() overlay = OverlayImage()