# -*- coding: utf-8 -*- |
import math, random, time |
import threading |
import tkinter as tk |
import re |
# import uuid |
Fireworks = [] |
maxFireworks = 16 |
height, width = 600 , 600 |
class firework( object ): |
def __init__( self , color, speed, width, height): |
# uid=uuid.uuid1() |
self .radius = random.randint( 1 , 4 ) # 粒子半径像素 |
self .color = color # 粒子颜色 |
self .speed = speed # speed是1.5-3.5秒 |
self .status = 0 # 在烟花未爆炸的情况下,status=0;爆炸后,status>=1;当status>100时,烟花的生命期终止 |
self .nParticle = random.randint( 20 , 30 ) # 粒子数量 |
self .center = [random.randint( 0 , width - 1 ), random.randint( 0 , height - 1 )] # 烟花随机中心坐标 |
self .oneParticle = [] # 原始粒子坐标(100%状态时) |
self .rotTheta = random.uniform( 0 , 2 * math.pi) # 椭圆平面旋转角 |
# 椭圆参数方程:x=a*cos(theta),y=b*sin(theta) |
# ellipsePara=[a,b] |
self .ellipsePara = [random.randint( 30 , 40 ), random.randint( 20 , 30 )] |
theta = 2 * math.pi / self .nParticle |
for i in range ( self .nParticle): |
t = random.uniform( - 1.0 / 16 , 1.0 / 16 ) # 产生一个 [-1/16,1/16) 的随机数 |
x, y = self .ellipsePara[ 0 ] * math.cos(theta * i + t), self .ellipsePara[ 1 ] * math.sin( |
theta * i + t) # 椭圆参数方程 |
xx, yy = x * math.cos( self .rotTheta) - y * math.sin( self .rotTheta), y * math.cos( |
self .rotTheta) + x * math.sin( self .rotTheta) # 平面旋转方程 |
self .oneParticle.append([xx, yy]) |
self .curParticle = self .oneParticle[ 0 :] # 当前粒子坐标 |
self .thread = threading.Thread(target = self .extend) # 建立线程对象 |
def extend( self ): # 粒子群状态变化函数线程 |
for i in range ( 100 ): |
self .status + = 1 # 更新状态标识 |
self .curParticle = [[one[ 0 ] * self .status / 100 , one[ 1 ] * self .status / 100 ] for one in |
self .oneParticle] # 更新粒子群坐标 |
time.sleep( self .speed / 50 ) |
def explode( self ): |
self .thread.setDaemon( True ) # 把现程设为守护线程 |
self .thread.start() # 启动线程 |
def __repr__( self ): |
return ( 'color:{color}\n' |
'speed:{speed}\n' |
'number of particle: {np}\n' |
'center:[{cx} , {cy}]\n' |
'ellipse:a={ea} , b={eb}\n' |
'particle:\n{p}\n' |
). format (color = self .color, speed = self .speed, np = self .nParticle, cx = self .center[ 0 ], cy = self .center[ 1 ], |
p = str ( self .oneParticle), ea = self .ellipsePara[ 0 ], eb = self .ellipsePara[ 1 ]) |
def colorChange(fire): |
rgb = re.findall(r '(.{2})' , fire.color[ 1 :]) |
cs = fire.status |
f = lambda x, c: hex ( int ( int (x, 16 ) * ( 100 - c) / 30 ))[ 2 :] # 当粒子寿命到70%时,颜色开始线性衰减 |
if cs > 70 : |
ccr, ccg, ccb = f(rgb[ 0 ], cs), f(rgb[ 1 ], cs), f(rgb[ 2 ], cs) |
else : |
ccr, ccg, ccb = rgb[ 0 ], rgb[ 1 ], rgb[ 2 ] |
return '#{0:0>2}{1:0>2}{2:0>2}' . format (ccr, ccg, ccb) |
def appendFirework(n = 1 ): # 递归生成烟花对象 |
if n > maxFireworks or len (Fireworks) > maxFireworks: |
pass |
elif n = = 1 : |
cl = '#{0:0>6}' . format ( hex ( int (random.randint( 0 , 16777215 )))[ 2 :]) # 产生一个 0 ~ 16777215 ( 0xFFFFFF )的随机数,作为随机颜色 |
a = firework(cl, random.uniform( 0.3 , 4.5 ), width, height) |
Fireworks.append({ 'particle' : a, 'points' : []}) # 建立粒子显示列表,‘particle’为一个烟花对象,‘points’为每一个粒子显示时的对象变量集 |
a.explode() |
else : |
appendFirework() |
appendFirework(n - 1 ) |
def show(c): |
for p in Fireworks: # 每次刷新显示,先把已有的所以粒子全部删除 |
for pp in p[ 'points' ]: |
c.delete(pp) |
for p in Fireworks: # 根据每个烟花对象,计算其中每个粒子的显示对象 |
oneP = p[ 'particle' ] |
if oneP.status = = 100 : # 状态标识为100,说明烟花寿命结束 |
Fireworks.remove(p) # 移出当前烟花 |
appendFirework() # 新增一个烟花 |
continue |
else : |
li = [[ int (cp[ 0 ] * 2 ) + oneP.center[ 0 ], int (cp[ 1 ] * 2 ) + oneP.center[ 1 ]] for cp in |
oneP.curParticle] # 把中心为原点的椭圆平移到随机圆心坐标上 |
color = colorChange(oneP) # 根据烟花当前状态计算当前颜色 |
for pp in li: |
p[ 'points' ].append( |
c.create_oval(pp[ 0 ] - oneP.radius, pp[ 1 ] - oneP.radius, pp[ 0 ] + oneP.radius, pp[ 1 ] + oneP.radius, |
fill = color)) # 绘制烟花每个粒子 |
root.after( 50 , show, c) # 回调,每50ms刷新一次 |
if __name__ = = '__main__' : |
appendFirework(maxFireworks) |
root = tk.Tk() |
cv = tk.Canvas(root, height = height, width = width) |
cv.create_rectangle( 0 , 0 , width, height, fill = "black" ) |
cv.pack() |
root.after( 50 , show, cv) |
root.mainloop() |