fireworks-跨年烟花代码

import pygame import random import os import math import time # 初始化信息 WIDTH = 1152 HEIGHT = 720 FPS = 60 img_path = os.path.join(os.path.dirname(__file__), "img") audio_path = os.path.join(os.path.dirname(__file__), "audio") def randv(): # 生成随机初速度 v = random.random() * 22 while (v < 15): v = random.random() * 22 return v def rand_color(): # 随机生成烟花颜色 color_list = [(244, 214, 215), (55, 20, 88), (151, 68, 114), (230, 190, 146), (244, 252, 255), (230, 197, 246), (181, 180, 222), (191, 85, 177), (255, 199, 209), (200, 50, 66), (223, 219, 216), (158, 167, 164), (173, 254, 255), (185, 219, 149), (72, 141, 235), (252, 117, 249), (232, 169, 180), (155, 157, 170), (182, 98, 130), (248, 215, 20), (136, 214, 253), (221, 0, 27)] return color_list[random.randint(0, len(color_list) - 1)] def add_fireworks(list): # 添加新的烟花 if (random.random() < 3 / 60): list.append(Fireworks()) class Item(pygame.sprite.Sprite): # 烟花颗粒 def __init__(self, vy, x, color): pygame.sprite.Sprite.__init__(self) self.image = pygame.Surface((15, 15)) self.image.fill((0, 0, 0)) self.image.set_colorkey((0, 0, 0)) if color == (-1, -1, -1): # 纯色烟花或者混色烟花 color = rand_color() self.color = color self.radius = randv() / 5 # 烟花颗粒的半径 pygame.draw.circle(self.image, self.color, (8, 8), self.radius) # 形成圆形烟花颗粒 self.rect = self.image.get_rect() self.rect.center = (x, HEIGHT - 10) # 初始位置 self.vx = 0 self.vy = vy self.is_explode = False # 记录是否已经爆炸过 self.count = 0 def update(self, radius): self.move() if self.rect.y > HEIGHT - 20 and self.is_explode: self.kill() if self.vy < 1 and not self.is_explode: self.explode(radius) if self.is_explode: # 爆炸后一段时间删除颗粒 self.count += 1 if self.count > 55: self.kill() def move(self): # vx为水平方向速度,vy为竖直方向速度 t = 1 / 60 g = 9.8 self.rect.x += self.vx * t * 25 self.rect.y -= (2 * self.vy - g * t) * t / 2 * 25 self.vy = self.vy - g * t def explode(self, radius): # 爆炸函数，改变速度大小和方向 self.is_explode = True angle = random.randint(0, 359) * math.pi / 180 self.vx += math.cos(angle) * radius self.vy -= math.sin(angle) * radius class Fireworks(): # 烟花列表 def __init__(self): self.list = pygame.sprite.Group() self.vy = randv() self.x = WIDTH * random.random() self.num = random.randint(30, 40) # 单个烟花随机烟花颗粒数量 color = rand_color() if random.randint(1, 2) == 1: # 纯色烟花或者混色烟花 color = (-1, -1, -1) self.list.add(Item(self.vy, self.x, color) for i in range(self.num)) self.start_time = time.time() # 记录开始时间，一定时间后删除 def draw(self, screen): self.list.draw(screen) def update(self): # 更新后返回是否删除烟花 radius = random.random() * 10 while (radius < 8): radius = random.random() * 10 self.list.update(radius) if (time.time() - self.start_time > 5): # 定时删除 return True else: return False def main(): pygame.init() pygame.mixer.init() screen = pygame.display.set_mode((WIDTH, HEIGHT)) # 主画布大小 pygame.display.set_caption("Fireworks--烟花特效") pygame.display.set_icon(pygame.image.load(os.path.join((img_path), "fireworks.png")).convert()) # 设置图标 bgm = pygame.mixer.music.load(os.path.join(audio_path, "bgm.mp3")) # 背景音乐 pygame.mixer.music.play(-1) # 循环播放背景音乐 clock = pygame.time.Clock() fireworks_list = [Fireworks() for i in range(5)] # 初始烟花数目 running = True # 是否循环 count = 2 while running: # 循环主程序 clock.tick(FPS) for event in pygame.event.get(): if event.type == pygame.QUIT: running = False if count == 2: screen.blit(pygame.image.load(os.path.join(img_path, "bgp.jpg")), (0, 0)) # 设置背景图片 count = 0 count += 1 add_fireworks(fireworks_list) dels = [] # 需要删除的烟花的索引 for index, item in enumerate(fireworks_list): if item.update(): dels.append(index) item.draw(screen) pygame.display.flip() for i in dels: # 删除烟花 del fireworks_list[i] pygame.QUIT exit() if __name__ == "__main__": main()