基于Python实现烟花效果的示例代码

python烟花代码

如下

# -*- coding: utf-8 -*-

import math, random,time
import threading
import tkinter as tk
import re
#import uuid

Fireworks=[]
maxFireworks=8
height,width=600,600

class firework(object):
   def __init__(self,color,speed,width,height):
       #uid=uuid.uuid1()
       self.radius=random.randint(2,4)  #粒子半径为2~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}
'  
               'speed:{speed}
'
               'number of particle: {np}
'
               'center:[{cx} , {cy}]
'
               'ellipse:a={ea} , b={eb}
'
               'particle:
{p}
'
               ).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(1.5,3.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()

图片展示

来源：https://blog.csdn.net/m0_54850467/article/details/125241220