ISSUE-2-感觉方形头像不如圆形的好看

.gitignore

@@ -1,3 +1,4 @@
 media
 assets
+assets.bak
 __pycache__

scene.py

@@ -1,6 +1,6 @@
 from manim import *
 from pathlib import Path
-from fetch_data import fetch_char_data
+from scripts.fetch_data import fetch_char_data
 
 class ArkVis(Scene):
     def construct(self):

fetch_data.py → scripts/fetch_data.py

@@ -87,4 +87,7 @@ def fetch_char_data() -> List[CHAR_DATA]:
                 elif kv['key'] == 'max_hp':
                     char_info['maxHp'] += kv['value']
 
-    return [(c['atk'], c['def'], c['maxHp'], f"assets/images/{c['name']}.png") for c in char_table.values()]
+    return [(c['atk'], c['def'], c['maxHp'], f"assets/images/{c['name']}.png") for c in char_table.values()]
+
+if __name__ == "__main__":
+    fetch_char_data()

scripts/make_avatar_circle.py

@@ -0,0 +1,31 @@
+from PIL import Image, ImageDraw, ImageChops
+import logging
+
+def crop_to_circle(image_path, output_path):
+    # 打开图片并转换为RGBA模式
+    img = Image.open(image_path).convert("RGBA")
+
+    # 创建一个相同尺寸的黑色
+    mask = Image.new("L", img.size, 0)
+
+    # 绘制白色圆形蒙版
+    draw = ImageDraw.Draw(mask)
+    draw.ellipse([(0, 0), img.size], fill=255, width=0)
+
+    # 把圆形内部，原图中的透明部分正确的附加到蒙版上去
+    mask = ImageChops.multiply(mask, img.split()[-1])
+
+    # 将蒙版应用到图片上
+    img.putalpha(mask)
+
+    # 将图像裁剪成圆形并保存
+    img.save(output_path, "PNG")
+
+def make_avatar_circle():
+    from fetch_data import IMAGE_DIR
+    for avatar_path in IMAGE_DIR.glob("*.png"):
+        logging.debug(f"Cropping {avatar_path.as_posix()} to circle format.")
+        crop_to_circle(avatar_path.as_posix(), avatar_path.as_posix())
+
+if __name__ == '__main__':
+    make_avatar_circle()