帮我写一份粒子光线生长树的代码

好的，以下是一个简单的粒子光线生长树的代码示例：

import pygame
import random

# 初始化 Pygame
pygame.init()

# 定义一些常量
WIDTH = 800
HEIGHT = 600
FPS = 60
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)

# 创建窗口
screen = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("Particle Tree")

# 定义一个粒子类
class Particle(pygame.sprite.Sprite):
    def __init__(self, x, y):
        super().__init__()
        self.image = pygame.Surface((2, 2))
        self.image.fill(WHITE)
        self.rect = self.image.get_rect()
        self.rect.x = x
        self.rect.y = y
        self.vx = random.randint(-2, 2)
        self.vy = random.randint(-2, 2)
        self.age = 0

    def update(self):
        self.age += 1
        if self.age > 50:
            self.kill()
        self.rect.x += self.vx
        self.rect.y += self.vy

# 定义一个树类
class Tree(pygame.sprite.Sprite):
    def __init__(self, x, y):
        super().__init__()
        self.image = pygame.Surface((2, 2))
        self.image.fill(WHITE)
        self.rect = self.image.get_rect()
        self.rect.x = x
        self.rect.y = y
        self.angle = 0
        self.branches = []

    def update(self):
        self.angle += random.randint(-10, 10)
        if len(self.branches) < 50 and random.random() < 0.1:
            branch = Tree(self.rect.x, self.rect.y)
            self.branches.append(branch)
            all_sprites.add(branch)

        for branch in self.branches:
            branch.update()
            pygame.draw.line(screen, WHITE, (self.rect.x, self.rect.y), (branch.rect.x, branch.rect.y), 2)

        if random.random() < 0.5:
            particle = Particle(self.rect.x, self.rect.y)
            all_sprites.add(particle)

# 创建精灵组
all_sprites = pygame.sprite.Group()

# 创建一棵树
tree = Tree(WIDTH // 2, HEIGHT)
all_sprites.add(tree)

# 游戏循环
running = True
clock = pygame.time.Clock()
while running:
    # 处理事件
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False

    # 更新精灵组
    all_sprites.update()

    # 绘制背景
    screen.fill(BLACK)

    # 绘制精灵组
    all_sprites.draw(screen)

    # 更新屏幕
    pygame.display.flip()

    # 控制帧率
    clock.tick(FPS)

# 退出 Pygame
pygame.quit()

这个示例代码使用 Pygame 库创建了一个粒子光线生长树，树的生长过程是随机的，每个分支的长度和角度都是随机的，同时会发射一些粒子。你可以根据自己的需要修改代码中的参数和细节。