绘制烟花的代码
时间: 2023-07-02 14:07:03 浏览: 115
以下是一个基于 Python 和 Pygame 的简单烟花绘制代码,供参考:
```python
import pygame
import random
# 初始化 Pygame
pygame.init()
# 设置屏幕大小和标题
screen = pygame.display.set_mode((800, 600))
pygame.display.set_caption("Fireworks")
# 定义颜色
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
RED = (255, 0, 0)
GREEN = (0, 255, 0)
BLUE = (0, 0, 255)
# 定义烟花类
class Firework(pygame.sprite.Sprite):
def __init__(self, x, y, color):
super().__init__()
self.color = color
self.image = pygame.Surface((5, 5))
self.image.fill(self.color)
self.rect = self.image.get_rect()
self.rect.x = x
self.rect.y = y
self.vel_x = random.randint(-10, 10)
self.vel_y = random.randint(-20, -10)
self.gravity = 0.5
self.timer = 60
def update(self):
self.rect.x += self.vel_x
self.rect.y += self.vel_y
self.vel_y += self.gravity
self.timer -= 1
if self.timer == 0:
sparks = []
for i in range(50):
spark = Spark(self.rect.x, self.rect.y, self.color)
sparks.append(spark)
all_sprites.add(sparks)
fireworks.remove(self)
all_sprites.remove(self)
del self
# 定义火花类
class Spark(pygame.sprite.Sprite):
def __init__(self, x, y, color):
super().__init__()
self.color = color
self.image = pygame.Surface((2, 2))
self.image.fill(self.color)
self.rect = self.image.get_rect()
self.rect.x = x
self.rect.y = y
self.vel_x = random.randint(-10, 10)
self.vel_y = random.randint(-10, 10)
self.gravity = 0.5
self.timer = 60
def update(self):
self.rect.x += self.vel_x
self.rect.y += self.vel_y
self.vel_y += self.gravity
self.timer -= 1
if self.timer == 0:
all_sprites.remove(self)
del self
# 创建精灵组
fireworks = pygame.sprite.Group()
all_sprites = pygame.sprite.Group()
# 游戏主循环
running = True
while running:
# 处理事件
for event in pygame.event.get():
if event.type == pygame.QUIT:
running = False
elif event.type == pygame.MOUSEBUTTONDOWN:
if event.button == 1:
x, y = event.pos
color = (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255))
firework = Firework(x, y, color)
fireworks.add(firework)
all_sprites.add(firework)
# 更新精灵组
all_sprites.update()
# 绘制屏幕
screen.fill(BLACK)
all_sprites.draw(screen)
pygame.display.flip()
# 退出 Pygame
pygame.quit()
```
这段代码实现了在 Pygame 窗口中绘制烟花效果。在代码中,烟花和火花都被定义为 Pygame 中的 Sprite 对象,并使用 Pygame 的精灵组进行管理。当鼠标左键按下时,会创建一个新的烟花对象,并将其添加到精灵组中。烟花会随机产生多个火花,并在一定时间后消失。