用python画一个烟花_用python做一个烟花show

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

importtkinter as tkfrom PIL importImage, ImageTkfrom time importtime, sleepfrom random importchoice, uniform, randintfrom math importsin, cos, radians#模拟重力

GRAVITY = 0.05

#颜色选项(随机或者按顺序)

colors = ['red', 'blue', 'yellow', 'white', 'green', 'orange', 'purple', 'seagreen', 'indigo', 'cornflowerblue']'''particles 类

粒子在空中随机生成随机，变成一个圈、下坠、消失

属性:

- id: 粒子的id

- x, y: 粒子的坐标

- vx, vy: 在坐标的变化速度

- total: 总数

- age: 粒子存在的时长

- color: 颜色

- cv: 画布

- lifespan: 最高存在时长'''

classParticle: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

self.x=x

self.y=y

self.initial_speed=explosion_speed

self.vx=vx

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=lifespandefupdate(self, dt):

self.age+=dt#粒子范围扩大

if self.alive() andself.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)#以自由落体坠落

elifself.alive():

move_x= cos(radians(self.id * 360 /self.total))#we technically don't need to update x, y because move will do the job

self.cv.move(self.cid, self.vx + move_x, self.vy + GRAVITY *dt)

self.vy+= GRAVITY *dt#移除超过最高时长的粒子

elif self.cid is notNone:

cv.delete(self.cid)

self.cid=None#扩大的时间

defexpand (self):return self.age <= 1.2

#粒子是否在最高存在时长内

defalive(self):return self.age <=self.lifespan'''循环调用保持不停'''

defsimulate(cv):

t=time()

explode_points=[]

wait_time= randint(10, 100)

numb_explode= randint(6, 10)#创建一个所有粒子同时扩大的二维列表

for point inrange(numb_explode):

objects=[]

x_cordi= randint(50, 550)

y_cordi= randint(50, 150)

speed= uniform(0.5, 1.5)

size= uniform(1, 3)

color=choice(colors)

explosion_speed= uniform(0.2, 1)

total_particles= randint(10, 50)for i in range(1, total_particles):

r= Particle(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.8s内一直扩大

while total_time < 1.8:

sleep(0.01)

tnew=time()

t, dt= tnew, tnew -tfor point inexplode_points:for item inpoint:

item.update(dt)

cv.update()

total_time+=dt#循环调用

root.after(wait_time, simulate, cv)def close(*ignore):"""退出程序、关闭窗口"""

globalroot

root.quit()if __name__ == '__main__':

root=tk.Tk()

cv= tk.Canvas(root, height=400, width=600)#绘制一个黑色背景

cv.create_rectangle(0, 0, 600, 400, fill='black')#cv = tk.Canvas(root, height=400, width=600)

## 选一个好看的背景会让效果更惊艳！

#image = Image.open("./image.jpg")

#photo = ImageTk.PhotoImage(image)

# #cv.create_image(0, 0, image=photo, anchor='nw')

cv.pack()

root.protocol("WM_DELETE_WINDOW", close)

root.after(100, simulate, cv)

root.mainloop()