m

Author: ityard

yeke/README.md

@@ -20,6 +20,7 @@
 + [py-snow](https://github.com/JustDoPython/python-examples/tree/master/yeke/py-snow) ：用 Python 实现带音乐的雪花飘落雪景图
 + [py-discern](https://github.com/JustDoPython/python-examples/tree/master/yeke/py-discern) ：十行 Python 代码就提取了韦小宝的身份证信息
 + [py-flag](https://github.com/JustDoPython/python-examples/tree/master/yeke/py-flag) ：用 Python 给微信头像加个圣诞老人（平安果）小图标
++ [py-firework](https://github.com/JustDoPython/python-examples/tree/master/yeke/py-firework) ：快到春节了，用 Python 实现一场烟花秀
 
 ---
 

yeke/py-firework/__init__.py

@@ -0,0 +1,130 @@
+import tkinter as tk
+from time import time, sleep
+from PIL import Image, ImageTk
+from math import sin, cos, radians
+from random import choice, uniform, randint
+
+colors = ['red', 'blue', 'yellow', 'white', 'green', 'orange',
+          'purple', 'seagreen', 'indigo', 'cornflowerblue']
+
+# 烟花类
+class fireworks:
+    def __init__(self, cv, idx, total, explosion_speed, x=0., y=0.,
+                 vx=0., vy=0., size=2., color='red', lifespan=2, **kwargs):
+        self.id = idx
+        # 烟花绽放 x 轴
+        self.x = x
+        # 烟花绽放 x 轴
+        self.y = y
+        self.initial_speed = explosion_speed
+        # 外放 x 轴速度
+        self.vx = vx
+        # 外放 y 轴速度
+        self.vy = vy
+        # 绽放的粒子数
+        self.total = total
+        # 已停留时间
+        self.age = 0
+        # 颜色
+        self.color = color
+        # 画布
+        self.cv = cv
+        self.cid = self.cv.create_oval(x - size, y - size, x + size, y + size,
+        fill=self.color)
+        self.lifespan = lifespan
+
+    # 更新数据
+    def update(self, dt):
+        self.age += dt
+        # 粒子膨胀
+        if self.alive() and self.expand():
+            move_x = cos(radians(self.id * 360 / self.total)) * self.initial_speed
+            move_y = sin(radians(self.id * 360 / self.total)) * self.initial_speed
+            self.cv.move(self.cid, move_x, move_y)
+            self.vx = move_x / (float(dt) * 1000)
+        # 膨胀到最大下落
+        elif self.alive():
+            move_x = cos(radians(self.id * 360 / self.total))
+            self.cv.move(self.cid, self.vx + move_x, self.vy + 0.5 * dt)
+            self.vy += 0.5 * dt
+        # 过期移除
+        elif self.cid is not None:
+            cv.delete(self.cid)
+            self.cid = None
+
+    # 定义膨胀效果的时间帧
+    def expand(self):
+        return self.age <= 1.5
+
+    # 检查粒子是否仍在生命周期内
+    def alive(self):
+        return self.age <= self.lifespan
+
+# 燃放烟花
+def ignite(cv):
+    t = time()
+    # 烟花列表
+    explode_points = []
+    wait_time = randint(10, 100)
+    # 爆炸的个数
+    numb_explode = randint(6, 10)
+    for point in range(numb_explode):
+        # 爆炸粒子列表
+        objects = []
+        # 爆炸 x 轴
+        x_cordi = randint(50, 550)
+        # 爆炸 y 轴
+        y_cordi = randint(50, 150)
+        speed = uniform(0.5, 1.5)
+        size = uniform(0.5, 3)
+        color = choice(colors)
+        # 爆炸的绽放速度
+        explosion_speed = uniform(0.2, 1)
+        # 爆炸的粒子数半径
+        total_particles = randint(10, 50)
+        for i in range(1, total_particles):
+            r = fireworks(cv, idx=i, total=total_particles,
+                          explosion_speed=explosion_speed, x=x_cordi, y=y_cordi,
+                     vx=speed, vy=speed, color=color, size=size,
+                     lifespan=uniform(0.6, 1.75))
+            # 添加进粒子列表里
+            objects.append(r)
+        # 把粒子列表添加到烟花列表
+        explode_points.append(objects)
+    total_time = .0
+    # 在 1.8 秒时间帧内保持更新
+    while total_time < 1.8:
+        # 让画面暂停 0.01s
+        sleep(0.01)
+        # 刷新时间
+        tnew = time()
+        t, dt = tnew, tnew - t
+        # 遍历烟花列表
+        for point in explode_points:
+            # 遍历烟花里的粒子列表
+            for item in point:
+                # 更新时间
+                item.update(dt)
+        # 刷新页面
+        cv.update()
+        total_time += dt
+    root.after(wait_time, ignite, cv)
+
+def close(*ignore):
+    global root
+    root.quit()
+
+if __name__ == "__main__":
+    root = tk.Tk()
+    # 绘制一个画布
+    cv = tk.Canvas(root, height=457, width=690)
+    # 背景图
+    image = Image.open("bg.jpeg")
+    photo = ImageTk.PhotoImage(image)
+    # 在画板上绘制一张图片
+    cv.create_image(0, 0, image=photo, anchor='nw')
+    cv.pack()
+    root.protocol(close)
+    root.after(100, ignite, cv)
+    # # 生成窗口
+    root.mainloop()