You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
98 lines
3.2 KiB
98 lines
3.2 KiB
2 months ago
|
from PIL import Image, ImageDraw, ImageFont
|
||
|
|
import random
|
||
|
|
import imageio
|
||
|
|
import os
|
||
|
|
import re
|
||
|
|
|
||
|
|
# 参数设置
|
||
|
|
WIDTH = 800 # 图片宽度
|
||
|
|
HEIGHT = 600 # 图片高度
|
||
|
|
FONT_SIZE = 20 # 字体大小
|
||
|
|
COL_WIDTH = FONT_SIZE
|
||
|
|
NUM_COLS = WIDTH // COL_WIDTH
|
||
|
|
MAX_TRAIL_LENGTH = 15 # 代码雨最大长度
|
||
|
|
FRAMES_PER_SUBTITLE = 10 # 每条字幕的帧数
|
||
|
|
OUTPUT_GIF = 'code_rain.gif' # 输出 GIF 文件名
|
||
|
|
|
||
|
|
def read_subtitles(filename):
|
||
|
|
"""读取字幕文件"""
|
||
|
|
with open(filename, 'r', encoding='utf-8') as f:
|
||
|
|
subtitles = f.readlines()
|
||
|
|
subtitles = [line.strip() for line in subtitles if line.strip()]
|
||
|
|
return subtitles
|
||
|
|
|
||
|
|
def extract_characters(subtitles):
|
||
|
|
"""从字幕中提取所有出现的字符"""
|
||
|
|
chars_set = set()
|
||
|
|
for line in subtitles:
|
||
|
|
chars_set.update(line)
|
||
|
|
return list(chars_set)
|
||
|
|
|
||
|
|
def main():
|
||
|
|
# 读取字幕
|
||
|
|
subtitles = read_subtitles(r'提取.txt')
|
||
|
|
|
||
|
|
# 提取字符集
|
||
|
|
CHARS = extract_characters(subtitles)
|
||
|
|
# 添加额外的字符（可选）
|
||
|
|
CHARS.extend(list('ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789@#$%^&*()+-'))
|
||
|
|
|
||
|
|
# 加载字体（请确保字体文件存在）
|
||
|
|
font_path = r'msyh.ttc' # 替换为您的字体文件完整路径，确保支持中文
|
||
|
|
|
||
|
|
# 检查字体文件是否存在
|
||
|
|
if not os.path.exists(font_path):
|
||
|
|
print(f"未找到字体文件 '{font_path}'。请提供正确的字体路径。")
|
||
|
|
return
|
||
|
|
|
||
|
|
try:
|
||
|
|
font = ImageFont.truetype(font_path, FONT_SIZE)
|
||
|
|
except OSError:
|
||
|
|
print(f"无法打开字体文件 '{font_path}'。请检查文件是否损坏或权限设置。")
|
||
|
|
return
|
||
|
|
|
||
|
|
# 初始化列的位置
|
||
|
|
column_positions = [random.randint(-HEIGHT, 0) for _ in range(NUM_COLS)]
|
||
|
|
|
||
|
|
frames = []
|
||
|
|
|
||
|
|
for subtitle in subtitles:
|
||
|
|
for _ in range(FRAMES_PER_SUBTITLE):
|
||
|
|
# 创建新图像
|
||
|
|
img = Image.new('RGB', (WIDTH, HEIGHT), color='black')
|
||
|
|
draw = ImageDraw.Draw(img)
|
||
|
|
|
||
|
|
# 绘制代码雨
|
||
|
|
for i in range(NUM_COLS):
|
||
|
|
x = i * COL_WIDTH
|
||
|
|
y = column_positions[i]
|
||
|
|
|
||
|
|
for j in range(MAX_TRAIL_LENGTH):
|
||
|
|
char = random.choice(CHARS)
|
||
|
|
y_pos = y - j * FONT_SIZE
|
||
|
|
if y_pos < 0 or y_pos > HEIGHT:
|
||
|
|
continue
|
||
|
|
green_value = int(255 / MAX_TRAIL_LENGTH * (MAX_TRAIL_LENGTH - j))
|
||
|
|
draw.text((x, y_pos), char, font=font, fill=(0, green_value, 0))
|
||
|
|
|
||
|
|
# 更新列的位置
|
||
|
|
column_positions[i] = (y + FONT_SIZE) % (HEIGHT + FONT_SIZE * MAX_TRAIL_LENGTH)
|
||
|
|
|
||
|
|
# 绘制字幕
|
||
|
|
bbox = draw.textbbox((0, 0), subtitle, font=font)
|
||
|
|
text_width = bbox[2] - bbox[0]
|
||
|
|
text_height = bbox[3] - bbox[1]
|
||
|
|
text_x = (WIDTH - text_width) // 2
|
||
|
|
text_y = HEIGHT - text_height - 10 # 距离底部 10 像素
|
||
|
|
draw.text((text_x, text_y), subtitle, font=font, fill=(255, 255, 255))
|
||
|
|
|
||
|
|
# 添加帧
|
||
|
|
frames.append(img)
|
||
|
|
|
||
|
|
# 保存为 GIF
|
||
|
|
frames[0].save(OUTPUT_GIF, save_all=True, append_images=frames[1:], optimize=False, duration=100, loop=0)
|
||
|
|
print(f"GIF 已保存为 '{OUTPUT_GIF}'")
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|