烟花表白效果

[python]代码库

import pygame
import math
from random import randint
from math import sin, cos, radians, tan
import time
isQuit = False
def writeContent(x,y):
    color = (randint(0,255),randint(0,255),randint(0,255))
    pressKeySurf = BASICFONT.render("I love you.", True, color)
    pressKeyRect = pressKeySurf.get_rect()
    pressKeyRect.topleft = (x, y)
    screen.blit(pressKeySurf, pressKeyRect)
    
def writeBigContent():
    color = (randint(0,255),randint(0,255),randint(0,255))
    pressKeySurf = BASICFONT_B.render("I love you.", True, color)
    pressKeyRect = pressKeySurf.get_rect()
    pressKeyRect.topleft = (220, 220)
    screen.blit(pressKeySurf, pressKeyRect)

class Point:
    def __init__(self, idx, pos, screen, color) -> None:
        self.rad = radians(idx*6)
        self.x = pos[0]
        self.y = pos[1]
        self.t = 0
        self.screen = screen
        self.color = color
        self.v0 = 2  # 初速度
        self.limit = 51

    def get_pos(self) -> list:
        self.t += 1
        self.x += self.v0 * cos(self.rad)
        self.y -= self.v0 * sin(self.rad) - 0.08 * self.t
        return [self.x, self.y]

    def draw(self) -> None:
        if self.t >= 31:
            self.limit = randint(35, 50)
        if self.limit < self.t:
            return
        pygame.draw.circle(self.screen, self.color, self.get_pos(), 2)


class Fireworks:
    def __init__(self, screen) -> None:
        self.screen = screen
        self.pos = [-1, -1]
        # 一次只能发射一个烟花,免得出问题,刷新图层盖住就不好了
        #self.can_fire = True

    def run(self, event) -> None:
        if event.type == pygame.MOUSEBUTTONDOWN:
#             if not self.can_fire:
#                 return
            self.pos = pygame.mouse.get_pos()
            self.fire_to()

    def fire_to(self) -> None:
        # 一个烟花放映过程中不允许另一个进来,不然会混乱
        #self.can_fire = False
        color = (randint(0, 255), randint(0, 255), randint(0, 255))
        try:
            k = (600 - self.pos[1]) / (self.pos[0] - 400)
        except ZeroDivisionError:
            # x==400!
            # 指针指向屏幕最中间,除零错
            pass
        tmp_x, tmp_y = 400, 600
        # 分类讨论(可能有更好的算法,但这个最好理解)
        if self.pos[0] > 400:
            dX = 0.5
            dY = - dX * k
        elif self.pos[0] == 400:
            dX = 0
            dY = -0.5
        else:
            dX = -0.5
            dY = - dX * k
        v0 = 1  # 目标速度,根据dX与dY求解dX'和dY'
        # 斜方向速度
        dV = math.sqrt(dX**2 + dY**2)
        # 相似成比例求解
        dX *= v0 / dV
        dY *= v0 / dV
        while round(tmp_y) != self.pos[1]:
            tmp_x += dX
            tmp_y += dY
            self.screen.fill((0, 0, 0))
            #pygame.draw.circle(self.screen, color, [tmp_x, tmp_y], 2)
            writeContent(tmp_x-35,tmp_y-5)
            pygame.display.update()

        # 到达位置,炸开!
        self.bomb(color)
        #self.can_fire = True

    def bomb(self, color) -> None:
        self.screen.fill((0, 0, 0))
        pygame.display.update()
        ps = []
        for to in range(5):
            for i in range(60):
                ps.append(Point(i, self.pos, self.screen, color))
        for j in range(100):
            time.sleep(0.01)
            self.screen.fill((0, 0, 0))
            for point in ps:
                point.draw()
#                 print(".")  # 经测试,还是太快了
            pygame.display.update()
        self.screen.fill((0, 0, 0))


g_fireworks = None  # 烟花主类

def init():
    global g_fireworks,BASICFONT,BASICFONT_B,screen
    pygame.init()
    screen = pygame.display.set_mode([800, 600])
    pygame.mouse.set_visible(1)
    pygame.display.set_caption("烟花")
    BASICFONT = pygame.font.Font('freesansbold.ttf', 18)
    BASICFONT_B = pygame.font.Font('freesansbold.ttf', 85)
    g_fireworks = Fireworks(screen)

clock = pygame.time.Clock()

def loop():
    global isQuit
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            writeBigContent()
            isQuit = True
        else:
            g_fireworks.run(event)
    clock.tick(24)
    pygame.display.update()
    
init()
while not isQuit:
    loop()
time.sleep(2)
exit()