# -*- 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；當(dāng)status>100時(shí)，煙花的生命期終止
        self.nParticle=random.randint(20,30)  #粒子數(shù)量
        self.center=[random.randint(0,width-1),random.randint(0,height-1)]   #煙花隨機(jī)中心坐標(biāo)
        self.oneParticle=[]    #原始粒子坐標(biāo)（100%狀態(tài)時(shí)）
        self.rotTheta=random.uniform(0,2*math.pi)  #橢圓平面旋轉(zhuǎn)角

        #橢圓參數(shù)方程：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)  #產(chǎn)生一個(gè) [-1/16,1/16) 的隨機(jī)數(shù)
            x,y=self.ellipsePara[0]*math.cos(theta*i+t), self.ellipsePara[1]*math.sin(theta*i+t)    #橢圓參數(shù)方程
            xx,yy=x*math.cos(self.rotTheta)-y*math.sin(self.rotTheta),  y*math.cos(self.rotTheta)+x*math.sin(self.rotTheta)     #平面旋轉(zhuǎn)方程
            self.oneParticle.append([xx,yy])
        
        self.curParticle=self.oneParticle[0:]     #當(dāng)前粒子坐標(biāo)
        self.thread=threading.Thread(target=self.extend)   #建立線程對(duì)象
        

    def extend(self):         #粒子群狀態(tài)變化函數(shù)線程
        for i in range(100):
            self.status+=1    #更新狀態(tài)標(biāo)識(shí)
            self.curParticle=[[one[0]*self.status/100, one[1]*self.status/100] for one in self.oneParticle]   #更新粒子群坐標(biāo)
            time.sleep(self.speed/50)
    
    def explode(self):
        self.thread.setDaemon(True)    #把現(xiàn)程設(shè)為守護(hù)線程
        self.thread.start()          #啟動(dòng)線程
            

    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:]    #當(dāng)粒子壽命到70%時(shí)，顏色開始線性衰減
    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):   #遞歸生成煙花對(duì)象
    if n>maxFireworks or len(Fireworks)>maxFireworks:
        pass
    elif n==1:
        cl='#{0:0>6}'.format(hex(int(random.randint(0,16777215)))[2:])   # 產(chǎn)生一個(gè)0~16777215（0xFFFFFF）的隨機(jī)數(shù)，作為隨機(jī)顏色
        a=firework(cl,random.uniform(1.5,3.5),width,height)
        Fireworks.append( {'particle':a,'points':[]} )   #建立粒子顯示列表，‘particle'為一個(gè)煙花對(duì)象，‘points'為每一個(gè)粒子顯示時(shí)的對(duì)象變量集
        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:                #根據(jù)每個(gè)煙花對(duì)象，計(jì)算其中每個(gè)粒子的顯示對(duì)象
        oneP=p['particle']
        if oneP.status==100:        #狀態(tài)標(biāo)識(shí)為100，說(shuō)明煙花壽命結(jié)束
            Fireworks.remove(p)     #移出當(dāng)前煙花
            appendFirework()           #新增一個(gè)煙花
            continue
        else:
            li=[[int(cp[0]*2)+oneP.center[0],int(cp[1]*2)+oneP.center[1]] for cp in oneP.curParticle]       #把中心為原點(diǎn)的橢圓平移到隨機(jī)圓心坐標(biāo)上
            color=colorChange(oneP)   #根據(jù)煙花當(dāng)前狀態(tài)計(jì)算當(dāng)前顏色
            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))  #繪制煙花每個(gè)粒子

    root.after(50, show,c)  #回調(diào)，每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()