python实现代码雨

使用python实现代码雨的效果
效果如图：
python 复制代码
import random
import pygame
from pygame.locals import *
from sys import exit
import numpy as np
import math

# 设置窗口大小
PANEL_width = 800  # 窗口宽度
PANEL_highly = 600  # 窗口高度
FONT_PX = 40  # 字体大小

# 初始化pygame
pygame.init()

# 创建一个窗口
winSur = pygame.display.set_mode((PANEL_width, PANEL_highly), 32)

font = pygame.font.SysFont("SimHei", FONT_PX)

# 创建透明背景
bg_suface = pygame.Surface((PANEL_width, PANEL_highly), flags=pygame.SRCALPHA)
pygame.Surface.convert(bg_suface)
bg_suface.fill(pygame.Color(0, 0, 0, 20))

# 设置窗口背景
winSur.fill((0, 0, 0))

# LOVE字符组合
letter = ['L', 'O', 'V', 'E', 'LO', 'VE', 'LV', 'EO', 'LOVE', 'LVE', 'OVE', 'LOV', 'OV', 'EL', 'VL', 'LE', 'EV']

# 创建字符列表（LOVE组合）
texts = [font.render(str(letter[i]), True, (255, 105, 180)) for i in range(len(letter))]  # 粉色字体

# 按屏幕的宽度计算可以放多少列
column = int(PANEL_width / FONT_PX)
drops = [0 for i in range(column)]  # 初始化每列的字符位置

# 爱心曲线方程
def heart_curve(t, scale=1):
    x = 16 * np.sin(t) ** 3
    y = 13 * np.cos(t) - 5 * np.cos(2 * t) - 2 * np.cos(3 * t) - np.cos(4 * t)
    # 使用scale值来模拟跳动的效果
    return scale * x, scale * y

# 计算爱心的坐标
t = np.linspace(0, 2 * np.pi, 1000)

# 动态显示的"我爱你"字，使用更大的字体
love_text = font.render("爱你", True, (255, 105, 180))
love_text2 = font.render("LOVE", True, (255, 105, 180))

# 循环标志
show_first_text = True

# 主循环
while True:
    # 获取事件
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            exit()
        elif event.type == pygame.KEYDOWN:
            chang = pygame.key.get_pressed()
            if chang[32]:  # 检测是否按下空格退出
                exit()

    # 每帧暂停20毫秒，让动态更快
    pygame.time.delay(20)

    # 重新绘制背景
    winSur.blit(bg_suface, (0, 0))

    # 循环绘制字符（代码雨效果）
    for i in range(len(drops)):
        text = random.choice(texts)

        # 绘制字符
        winSur.blit(text, (i * FONT_PX, drops[i] * FONT_PX))

        # 更新字符下落位置
        drops[i] += 1
        if drops[i] * 10 > PANEL_highly or random.random() > 0.9:  # 加速字符下落
            drops[i] = 0

    # 计算心脏跳动的缩放系数
    scale = 1 + 0.3 * math.sin(pygame.time.get_ticks() / 1000)  # 使用sin函数控制跳动幅度

    # 绘制动态爱心
    x_heart, y_heart = heart_curve(t, scale)

    for i in range(len(x_heart)):
        # 根据爱心的坐标绘制"LOVE"字母
        char = "LOVE"[i % 4]  # 循环使用"LOVE"中的字符
        text = font.render(char, True, (255, 105, 180))  # 粉色字体
        # 计算绘制位置，放大爱心图形
        winSur.blit(text, (int(x_heart[i] * 15 + PANEL_width // 2), int(y_heart[i] * -15 + PANEL_highly // 3)))  # 放大爱心

    # 获取游戏时间并判断是否切换汉字组
    current_time = pygame.time.get_ticks()
    if current_time % 8000 < 4000:  # 控制显示"我爱你"4秒
        show_first_text = True
    else:
        show_first_text = False

    # 在爱心的中心位置动态显示"我爱你"或"永远爱你"
    love_x = PANEL_width // 2 - love_text.get_width() // 2
    love_y = PANEL_highly // 3 - love_text.get_height() // 2
    scale_text = 1 + 0.3 * math.sin(pygame.time.get_ticks() / 1000)  # "我爱你"跳动效果
    love_text_scaled = pygame.transform.scale(love_text, (int(love_text.get_width() * scale_text), int(love_text.get_height() * scale_text)))  # 缩放文字
    winSur.blit(love_text_scaled, (love_x, love_y))

    if not show_first_text:
        love_x2 = PANEL_width // 2 - love_text2.get_width() // 2
        love_y2 = PANEL_highly // 2 - love_text2.get_height() // 2
        scale_text2 = 1 + 0.3 * math.sin(pygame.time.get_ticks() / 1000 + math.pi)  # 给第二行文字不同的缩放值
        love_text2_scaled = pygame.transform.scale(love_text2, (int(love_text2.get_width() * scale_text2), int(love_text2.get_height() * scale_text2)))  # 缩放文字
        winSur.blit(love_text2_scaled, (love_x2, love_y2))

    # 更新显示
    pygame.display.flip()