import pygame
import random
# 初始化pygame
pygame.init()
# 设置屏幕大小
width, height = 800, 600
screen = pygame.display.set_mode((width, height))
# 定义一些工具函数
def random_color():
return random.randint(100, 255), random.randint(100, 255), random.randint(100, 255)
def draw_tail(screen, positions, color):
if len(positions) > 1:
for i in range(len(positions) - 1):
pygame.draw.line(screen, color, positions[i], positions[i+1], 2)
class Particle:
def __init__(self, x, y, color):
self.x = x
self.y = y
self.color = color
self.radius = random.randint(4, 7)
self.vel_x = random.uniform(-1, 1) * random.randint(3, 7)
self.vel_y = random.uniform(-1, 1) * random.randint(3, 7)
self.lifetime = random.randint(50, 100)
self.positions = []
def update(self):
self.x += self.vel_x
self.y += self.vel_y
self.lifetime -= 1
self.vel_y += 0.15 # 模拟重力
self.positions.append((self.x, self.y))
if len(self.positions) > 5:
self.positions.pop(0)
def draw(self):
pygame.draw.circle(screen, self.color, (int(self.x), int(self.y)), max(1, self.radius * self.lifetime / 60))
draw_tail(screen, self.positions, self.color)
class Firework:
def __init__(self):
self.x = random.randint(0, width)
self.y = height
self.color = random_color()
self.vel_y = random.randint(-18, -8)
self.exploded = False
self.particles = []
self.positions = []
def explode(self):
for _ in range(100): # 烟花爆炸产生的粒子数
self.particles.append(Particle(self.x, self.y, random_color()))
def update(self):
if not self.exploded:
self.y += self.vel_y
self.vel_y += 0.1 # 模拟重力
self.positions.append((self.x, self.y))
if len(self.positions) > 10:
self.positions.pop(0)
if self.vel_y >= 0:
self.exploded = True
self.explode()
else:
for particle in self.particles:
particle.update()
def draw(self):
if not self.exploded:
pygame.draw.circle(screen, self.color, (self.x, int(self.y)), 5)
draw_tail(screen, self.positions, self.color)
else:
for particle in self.particles:
particle.draw()
fireworks = []
running = True
while running:
screen.fill((0, 0, 0))
for event in pygame.event.get():
if event.type == pygame.QUIT:
running = False
if random.random() < 0.03: # 控制烟花生成的频率
fireworks.append(Firework())
for firework in fireworks[:]:
firework.update()
firework.draw()
if firework.exploded and all(p.lifetime <= 0 for p in firework.particles):
fireworks.remove(firework)
pygame.display.flip()
pygame.time.delay(30)
pygame.quit()
