diff --git a/starbot/painter/DynamicPicGenerator.py b/starbot/painter/DynamicPicGenerator.py index 8c61304..6d60f7e 100644 --- a/starbot/painter/DynamicPicGenerator.py +++ b/starbot/painter/DynamicPicGenerator.py @@ -32,7 +32,7 @@ class DynamicPicGenerator: 动态图片的 Base64 字符串 """ width = 740 - height = 10000 + height = 100000 text_margin = 25 img_margin = 10 generator = PicGenerator(width, height) diff --git a/starbot/painter/LiveReportGenerator.py b/starbot/painter/LiveReportGenerator.py index d738eb7..0bfde3a 100644 --- a/starbot/painter/LiveReportGenerator.py +++ b/starbot/painter/LiveReportGenerator.py @@ -9,13 +9,14 @@ from typing import Union, Tuple, List, Dict, Any import jieba import numpy as np -from PIL import Image, ImageDraw +from PIL import Image from matplotlib import pyplot as plt from mpl_toolkits import axisartist from scipy.interpolate import make_interp_spline from wordcloud import WordCloud from .PicGenerator import Color, PicGenerator +from .RankingGenerator import RankingGenerator from ..core.model import LiveReport from ..utils import config from ..utils.utils import split_list, limit_str_length, mask_round, timestamp_format @@ -41,7 +42,7 @@ class LiveReportGenerator: 直播报告图片的 Base64 字符串 """ width = 1000 - height = 10000 + height = 100000 top_blank = 75 margin = 50 @@ -209,7 +210,9 @@ class LiveReportGenerator: if counts: pic.draw_section(f"弹幕排行 (Top {len(counts)})") - ranking_img = cls.__get_ranking(pic, faces, unames, counts, pic.width - (margin * 2)) + ranking_img = RankingGenerator.get_ranking( + pic.row_space, faces, unames, counts, pic.width - (margin * 2) + ) pic.draw_img_alpha(pic.auto_size_img_by_limit(ranking_img, logo_limit)) # 盲盒数量排行 @@ -221,7 +224,9 @@ class LiveReportGenerator: if counts: pic.draw_section(f"盲盒数量排行 (Top {len(counts)})") - ranking_img = cls.__get_ranking(pic, faces, unames, counts, pic.width - (margin * 2)) + ranking_img = RankingGenerator.get_ranking( + pic.row_space, faces, unames, counts, pic.width - (margin * 2) + ) pic.draw_img_alpha(pic.auto_size_img_by_limit(ranking_img, logo_limit)) # 盲盒盈亏排行 @@ -233,7 +238,9 @@ class LiveReportGenerator: if counts: pic.draw_section(f"盲盒盈亏排行 (Top {len(counts)})") - ranking_img = cls.__get_double_ranking(pic, faces, unames, counts, pic.width - (margin * 2)) + ranking_img = RankingGenerator.get_double_ranking( + pic.row_space, faces, unames, counts, pic.width - (margin * 2) + ) pic.draw_img_alpha(pic.auto_size_img_by_limit(ranking_img, logo_limit)) # 礼物排行 @@ -245,7 +252,9 @@ class LiveReportGenerator: if counts: pic.draw_section(f"礼物排行 (Top {len(counts)})") - ranking_img = cls.__get_ranking(pic, faces, unames, counts, pic.width - (margin * 2)) + ranking_img = RankingGenerator.get_ranking( + pic.row_space, faces, unames, counts, pic.width - (margin * 2) + ) pic.draw_img_alpha(pic.auto_size_img_by_limit(ranking_img, logo_limit)) # SC（醒目留言）排行 @@ -257,7 +266,9 @@ class LiveReportGenerator: if counts: pic.draw_section(f"SC(醒目留言)排行 (Top {len(counts)})") - ranking_img = cls.__get_ranking(pic, faces, unames, counts, pic.width - (margin * 2)) + ranking_img = RankingGenerator.get_ranking( + pic.row_space, faces, unames, counts, pic.width - (margin * 2) + ) pic.draw_img_alpha(pic.auto_size_img_by_limit(ranking_img, logo_limit)) # 开通大航海观众列表 @@ -410,169 +421,6 @@ class LiveReportGenerator: return logo - @classmethod - def __get_rank_bar_pic(cls, - width: int, - height: int, - start_color: Union[Color, Tuple[int, int, int]] = Color.DEEPBLUE, - end_color: Union[Color, Tuple[int, int, int]] = Color.LIGHTBLUE, - reverse: bool = False) -> Image: - """ - 生成排行榜中排行条图片 - - Args: - width: 排行条长度 - height: 排行条宽度 - start_color: 排行条渐变起始颜色。默认：深蓝色 (57, 119, 230) - end_color: 排行条渐变终止颜色。默认：浅蓝色 (55, 187, 248) - reverse: 是否生成反向排行条，用于双向排行榜的负数排行条。默认：False - """ - if isinstance(start_color, Color): - start_color = start_color.value - if isinstance(end_color, Color): - end_color = end_color.value - if reverse: - start_color, end_color = end_color, start_color - - r_step = (end_color[0] - start_color[0]) / width - g_step = (end_color[1] - start_color[1]) / width - b_step = (end_color[2] - start_color[2]) / width - - now_color = [start_color[0], start_color[1], start_color[2]] - - bar = Image.new("RGBA", (width, 1)) - draw = ImageDraw.Draw(bar) - - for i in range(width): - draw.point((i, 0), (int(now_color[0]), int(now_color[1]), int(now_color[2]))) - now_color[0] += r_step - now_color[1] += g_step - now_color[2] += b_step - - bar = bar.resize((width, height)) - - mask = Image.new("L", (width, height), 255) - mask_draw = ImageDraw.Draw(mask) - if not reverse: - mask_draw.polygon(((width - height, height), (width, 0), (width, height)), 0) - else: - mask_draw.polygon(((0, 0), (0, height), (height, height)), 0) - bar.putalpha(mask) - - return bar - - @classmethod - def __get_ranking(cls, - pic: PicGenerator, - faces: List[Image.Image], - unames: List[str], - counts: Union[List[int], List[float]], - width: int, - start_color: Union[Color, Tuple[int, int, int]] = Color.DEEPBLUE, - end_color: Union[Color, Tuple[int, int, int]] = Color.LIGHTBLUE) -> Image: - """ - 绘制排行榜 - - Args: - pic: 绘图器实例 - faces: 头像图片列表，按照数量列表降序排序 - unames: 昵称列表，按照数量列表降序排序 - counts: 数量列表，降序排序 - width: 排行榜图片宽度 - start_color: 排行条渐变起始颜色。默认：深蓝色 (57, 119, 230) - end_color: 排行条渐变终止颜色。默认：浅蓝色 (55, 187, 248) - """ - count = len(counts) - if count == 0 or len(faces) != len(unames) or len(unames) != len(counts): - raise ValueError - - face_size = 100 - offset = 10 - bar_height = 30 - - bar_x = face_size - offset - top_bar_width = width - face_size + offset - top_count = counts[0] - - chart = PicGenerator(width, (face_size * count) + (pic.row_space * (count - 1))) - chart.set_row_space(pic.row_space) - for i in range(count): - bar_width = int(counts[i] / top_count * top_bar_width) - if bar_width != 0: - bar = cls.__get_rank_bar_pic(bar_width, bar_height, start_color, end_color) - chart.draw_img_alpha(bar, (bar_x, chart.y + int((face_size - bar_height) / 2))) - chart.draw_tip(unames[i], Color.BLACK, (bar_x + (offset * 2), chart.y)) - count_pos = (max(chart.x + bar_width, bar_x + (offset * 3) + chart.get_tip_length(unames[i])), chart.y) - chart.draw_tip(str(counts[i]), xy=count_pos) - chart.draw_img_alpha(mask_round(faces[i].resize((face_size, face_size)).convert("RGBA"))) - - return chart.img - - @classmethod - def __get_double_ranking(cls, - pic: PicGenerator, - faces: List[Image.Image], - unames: List[str], - counts: Union[List[int], List[float]], - width: int, - start_color: Union[Color, Tuple[int, int, int]] = Color.DEEPRED, - end_color: Union[Color, Tuple[int, int, int]] = Color.LIGHTRED, - reverse_start_color: Union[Color, Tuple[int, int, int]] = Color.DEEPGREEN, - reverse_end_color: Union[Color, Tuple[int, int, int]] = Color.LIGHTGREEN) -> Image: - """ - 绘制双向排行榜 - - Args: - pic: 绘图器实例 - faces: 头像图片列表，按照数量列表降序排序 - unames: 昵称列表，按照数量列表降序排序 - counts: 数量列表，降序排序 - width: 排行榜图片宽度 - start_color: 正向排行条渐变起始颜色，数量为正时使用。默认：深红色 (57, 119, 230) - end_color: 正向排行条渐变终止颜色，数量为正时使用。默认：浅红色 (55, 187, 248) - reverse_start_color: 反向排行条渐变起始颜色，数量为负时使用。默认：深绿色 (57, 119, 230) - reverse_end_color: 反向排行条渐变终止颜色，数量为负时使用。默认：浅绿色 (55, 187, 248) - """ - count = len(counts) - if count == 0 or len(faces) != len(unames) or len(unames) != len(counts): - raise ValueError - - face_size = 100 - offset = 10 - bar_height = 30 - - face_x = int((width - face_size) / 2) - bar_x = face_x + face_size - offset - reverse_bar_x = face_x + offset - top_bar_width = (width - face_size) / 2 + offset - top_count = max(max(counts), abs(min(counts))) - - chart = PicGenerator(width, (face_size * count) + (pic.row_space * (count - 1))) - chart.set_row_space(pic.row_space) - for i in range(count): - bar_width = int(abs(counts[i]) / top_count * top_bar_width) - if counts[i] > 0: - bar = cls.__get_rank_bar_pic(bar_width, bar_height, start_color, end_color) - chart.draw_img_alpha(bar, (bar_x, chart.y + int((face_size - bar_height) / 2))) - elif counts[i] < 0: - bar = cls.__get_rank_bar_pic(bar_width, bar_height, reverse_start_color, reverse_end_color, True) - chart.draw_img_alpha(bar, (reverse_bar_x - bar_width, chart.y + int((face_size - bar_height) / 2))) - if counts[i] >= 0: - chart.draw_tip(unames[i], Color.BLACK, (bar_x + (offset * 2), chart.y)) - count_pos = (max(face_x + bar_width, bar_x + (offset * 3) + chart.get_tip_length(unames[i])), chart.y) - chart.draw_tip(str(counts[i]), xy=count_pos) - else: - uname_length = chart.get_tip_length(unames[i]) - count_length = chart.get_tip_length(str(counts[i])) - chart.draw_tip(unames[i], Color.BLACK, (reverse_bar_x - (offset * 2) - uname_length, chart.y)) - count_pos = (min(face_x + face_size - bar_width - count_length, - reverse_bar_x - (offset * 3) - uname_length - count_length), chart.y) - chart.draw_tip(str(counts[i]), xy=count_pos) - chart.set_pos(x=face_x).draw_img_alpha(mask_round(faces[i].resize((face_size, face_size)).convert("RGBA"))) - chart.set_pos(x=0) - - return chart.img - @classmethod def __get_guard_line_pic(cls, pic: PicGenerator, diff --git a/starbot/painter/RankingGenerator.py b/starbot/painter/RankingGenerator.py new file mode 100644 index 0000000..72b2cb8 --- /dev/null +++ b/starbot/painter/RankingGenerator.py @@ -0,0 +1,175 @@ +from typing import Union, Tuple, List + +from PIL import Image, ImageDraw + +from .PicGenerator import Color, PicGenerator +from ..utils.utils import mask_round + + +class RankingGenerator: + """ + 排行榜图片生成器 + """ + + @classmethod + def __get_rank_bar_pic(cls, + width: int, + height: int, + start_color: Union[Color, Tuple[int, int, int]] = Color.DEEPBLUE, + end_color: Union[Color, Tuple[int, int, int]] = Color.LIGHTBLUE, + reverse: bool = False) -> Image: + """ + 生成排行榜中排行条图片 + + Args: + width: 排行条长度 + height: 排行条宽度 + start_color: 排行条渐变起始颜色。默认：深蓝色 (57, 119, 230) + end_color: 排行条渐变终止颜色。默认：浅蓝色 (55, 187, 248) + reverse: 是否生成反向排行条，用于双向排行榜的负数排行条。默认：False + """ + if isinstance(start_color, Color): + start_color = start_color.value + if isinstance(end_color, Color): + end_color = end_color.value + if reverse: + start_color, end_color = end_color, start_color + + r_step = (end_color[0] - start_color[0]) / width + g_step = (end_color[1] - start_color[1]) / width + b_step = (end_color[2] - start_color[2]) / width + + now_color = [start_color[0], start_color[1], start_color[2]] + + bar = Image.new("RGBA", (width, 1)) + draw = ImageDraw.Draw(bar) + + for i in range(width): + draw.point((i, 0), (int(now_color[0]), int(now_color[1]), int(now_color[2]))) + now_color[0] += r_step + now_color[1] += g_step + now_color[2] += b_step + + bar = bar.resize((width, height)) + + mask = Image.new("L", (width, height), 255) + mask_draw = ImageDraw.Draw(mask) + if not reverse: + mask_draw.polygon(((width - height, height), (width, 0), (width, height)), 0) + else: + mask_draw.polygon(((0, 0), (0, height), (height, height)), 0) + bar.putalpha(mask) + + return bar + + @classmethod + def get_ranking(cls, + row_space: int, + faces: List[Image.Image], + unames: List[str], + counts: Union[List[int], List[float]], + width: int, + start_color: Union[Color, Tuple[int, int, int]] = Color.DEEPBLUE, + end_color: Union[Color, Tuple[int, int, int]] = Color.LIGHTBLUE) -> Image: + """ + 绘制排行榜 + + Args: + row_space: 行间距 + faces: 头像图片列表，按照数量列表降序排序 + unames: 昵称列表，按照数量列表降序排序 + counts: 数量列表，降序排序 + width: 排行榜图片宽度 + start_color: 排行条渐变起始颜色。默认：深蓝色 (57, 119, 230) + end_color: 排行条渐变终止颜色。默认：浅蓝色 (55, 187, 248) + """ + count = len(counts) + if count == 0 or len(faces) != len(unames) or len(unames) != len(counts): + raise ValueError + + face_size = 100 + offset = 10 + bar_height = 30 + + bar_x = face_size - offset + top_bar_width = width - face_size + offset + top_count = counts[0] + + chart = PicGenerator(width, (face_size * count) + (row_space * (count - 1))) + chart.set_row_space(row_space) + for i in range(count): + bar_width = int(counts[i] / top_count * top_bar_width) + if bar_width != 0: + bar = cls.__get_rank_bar_pic(bar_width, bar_height, start_color, end_color) + chart.draw_img_alpha(bar, (bar_x, chart.y + int((face_size - bar_height) / 2))) + chart.draw_tip(unames[i], Color.BLACK, (bar_x + (offset * 2), chart.y)) + count_pos = (max(chart.x + bar_width, bar_x + (offset * 3) + chart.get_tip_length(unames[i])), chart.y) + chart.draw_tip(str(counts[i]), xy=count_pos) + chart.draw_img_alpha(mask_round(faces[i].resize((face_size, face_size)).convert("RGBA"))) + + return chart.img + + @classmethod + def get_double_ranking(cls, + row_space: int, + faces: List[Image.Image], + unames: List[str], + counts: Union[List[int], List[float]], + width: int, + start_color: Union[Color, Tuple[int, int, int]] = Color.DEEPRED, + end_color: Union[Color, Tuple[int, int, int]] = Color.LIGHTRED, + reverse_start_color: Union[Color, Tuple[int, int, int]] = Color.DEEPGREEN, + reverse_end_color: Union[Color, Tuple[int, int, int]] = Color.LIGHTGREEN) -> Image: + """ + 绘制双向排行榜 + + Args: + row_space: 行间距 + faces: 头像图片列表，按照数量列表降序排序 + unames: 昵称列表，按照数量列表降序排序 + counts: 数量列表，降序排序 + width: 排行榜图片宽度 + start_color: 正向排行条渐变起始颜色，数量为正时使用。默认：深红色 (57, 119, 230) + end_color: 正向排行条渐变终止颜色，数量为正时使用。默认：浅红色 (55, 187, 248) + reverse_start_color: 反向排行条渐变起始颜色，数量为负时使用。默认：深绿色 (57, 119, 230) + reverse_end_color: 反向排行条渐变终止颜色，数量为负时使用。默认：浅绿色 (55, 187, 248) + """ + count = len(counts) + if count == 0 or len(faces) != len(unames) or len(unames) != len(counts): + raise ValueError + + face_size = 100 + offset = 10 + bar_height = 30 + + face_x = int((width - face_size) / 2) + bar_x = face_x + face_size - offset + reverse_bar_x = face_x + offset + top_bar_width = (width - face_size) / 2 + offset + top_count = max(max(counts), abs(min(counts))) + + chart = PicGenerator(width, (face_size * count) + (row_space * (count - 1))) + chart.set_row_space(row_space) + for i in range(count): + bar_width = int(abs(counts[i]) / top_count * top_bar_width) + if counts[i] > 0: + bar = cls.__get_rank_bar_pic(bar_width, bar_height, start_color, end_color) + chart.draw_img_alpha(bar, (bar_x, chart.y + int((face_size - bar_height) / 2))) + elif counts[i] < 0: + bar = cls.__get_rank_bar_pic(bar_width, bar_height, reverse_start_color, reverse_end_color, True) + chart.draw_img_alpha(bar, (reverse_bar_x - bar_width, chart.y + int((face_size - bar_height) / 2))) + if counts[i] >= 0: + chart.draw_tip(unames[i], Color.BLACK, (bar_x + (offset * 2), chart.y)) + count_pos = (max(face_x + bar_width, bar_x + (offset * 3) + chart.get_tip_length(unames[i])), chart.y) + chart.draw_tip(str(counts[i]), xy=count_pos) + else: + uname_length = chart.get_tip_length(unames[i]) + count_length = chart.get_tip_length(str(counts[i])) + chart.draw_tip(unames[i], Color.BLACK, (reverse_bar_x - (offset * 2) - uname_length, chart.y)) + count_pos = (min(face_x + face_size - bar_width - count_length, + reverse_bar_x - (offset * 3) - uname_length - count_length), chart.y) + chart.draw_tip(str(counts[i]), xy=count_pos) + chart.set_pos(x=face_x).draw_img_alpha(mask_round(faces[i].resize((face_size, face_size)).convert("RGBA"))) + chart.set_pos(x=0) + + return chart.img