用pygame为大家燃放新年烟花

参考文献：新年到~用Python安排一场烟花秀 (qq.com) 结尾附上源码

具体请看原文 本篇只介绍博主复刻后的相关笔记（学习心得） 大部分都在代码注释之中

原理介绍：

在介绍代码之前，先介绍下 Pygame 绘制烟花的基本原理，烟花从发射到绽放一共分为三个阶段：

1，发射阶段：在这一阶段烟花的形状是线性向上，通过设定一组大小不同、颜色不同的点来模拟“向上发射” 的运动运动，运动过程中 5个点被赋予不同大小的加速度，随着时间推移，后面的点会赶上前面的点，最终所有点会汇聚在一起，处于 绽放准备阶段；

2，烟花绽放：烟花绽放这个阶段，是由一个点分散多个点向不同方向发散，并且每个点的移动轨迹可需要被记录，目的是为了追踪整个绽放轨迹。

3，烟花湮灭，此阶段负责描绘绽放后烟花的效果，绽放后的烟花，而在每一时刻点的下降速度和亮度（代码中也叫透明度）是不一样的，因此在代码里，将烟花绽放后将每个点赋予两个属性：分别为重力向量和生命周期，来模拟烟花在不同时期时不同的展现效果， 

代码：

代码部分将烟花封装为三个类：

Firework : 烟花整体；

Particle：烟花粒子（包含轨迹）

Trail：烟花轨迹，本质上是一个点 。

三个类之间的关系为：一个Firework 由多个 Particle 构成，而一个 Particle 由多个 Trail 构成

首先设置全局变量，例如重力向量，窗口大小，Trail 的颜色列表（多为灰色或白色）以及不同状态下 Trail 之间间隔

vector = pygame.math.Vector2
gravity = vector(0, 0.3) # 重力设置

# 根据自己所用的图片大小进行调整展示框的长宽
DISPLAY_WIDTH = 1100
DISPLAY_HEIGHT = 700

trail_colours = [(45, 45, 45), (60, 60, 60), (75, 75, 75), (125, 125, 125), (150, 150, 150)]
dynamic_offset = 1
static_offset = 3

创建 Trail 类，定义 show 方法绘制轨迹 、get_pos  实时获取轨迹坐标

class Trail:
    def __init__(self, n, size, dynamic):

        self.pos_in_line = n
        self.pos = vector(-10, -10)
        self.dynamic = dynamic

        if self.dynamic:
            self.colour = trail_colours[n]
            self.size = int(size - n / 2)
        else:
            self.colour = (255, 255, 200)
            self.size = size - 2
            if self.size < 0:
                self.size = 0

    def get_pos(self, x, y):
        self.pos = vector(x, y)

    def show(self, win):
        pygame.draw.circle(win, self.colour, (int(self.pos.x), int(self.pos.y)), self.size)

然后编写Firework、Particle的相关代码，最后，写一个 main 方法来对 pygame 环境进行初始化，例如背景图片，文字，设置页面刷新间隔，程序中设置的每 60ms 刷新一次，再在其中加入背景音乐。

pygame.display.set_caption("Happy New Year") # 设置标题
    background = pygame.image.load("fire/1.PNG") # 设置背景
    myfont = pygame.font.Font("fire/abg.ttf",80) # 通过ttf字体文件获得字体对象
    testsurface = myfont.render("新年快乐!!!",False,(237, 90, 101))

    pygame.mixer.music.load("fire/SNH.mp3") # 加载背景音乐文件
    pygame.mixer.music.play(-1,8.0) # 播放背景音乐，-1表示循环无限次，第二个参数是播放起点（单位s）

    # pygame.image.load("") # 图片展示
    win = pygame.display.set_mode((DISPLAY_WIDTH, DISPLAY_HEIGHT))
    # win.blit(background)
    clock = pygame.time.Clock()

    fireworks = [Firework() for i in range(2)]  # create the first fireworks
    running = True

    # 程序主循环
    while running:
        clock.tick(60) # 每60ms刷新一次

        win.fill((20, 20, 30))  # draw background
        win.blit(background,(0,0))
        win.blit(testsurface,(200,30))

        if randint(0, 20) == 1:  # create new firework
            fireworks.append(Firework())

        update(win, fireworks)

相关细节介绍 

 好看颜色的RGB：中国色 － 中国传统颜色  字体下载网站：http://www.ddooo.com/zt/tffzt.htm

 最后祝大家新年快乐！平安健康，事事顺心。

 附上源码：Collection: Python写的一些小工具 - Gitee.com