78 lines
2.8 KiB
Python
78 lines
2.8 KiB
Python
import math
|
|
import random
|
|
|
|
particlePresets = {
|
|
"jump": {"count": 4, "color": (230, 230, 230), "life": 16, "spread": 1.2, "gravity": 0.1, "up": -0.5},
|
|
"dash": {"count": 6, "color": (120, 200, 255), "life": 14, "spread": 1.6, "gravity": 0.0, "up": 0.0},
|
|
"stomp": {"count": 6, "color": (200, 120, 60), "life": 18, "spread": 2.0, "gravity": 0.2, "up": -1.0},
|
|
"pickup": {"count": 5, "color": (247, 213, 29), "life": 20, "spread": 1.4, "gravity": -0.05, "up": -1.2},
|
|
"hurt": {"count": 8, "color": (220, 40, 40), "life": 22, "spread": 2.2, "gravity": 0.15, "up": -1.5},
|
|
"spark": {"count": 4, "color": (255, 220, 120), "life": 10, "spread": 2.4, "gravity": 0.0, "up": 0.0},
|
|
"alert": {"count": 3, "color": (255, 80, 80), "life": 16, "spread": 0.6, "gravity": -0.1, "up": -1.4},
|
|
"dust": {"count": 3, "color": (200, 190, 160), "life": 12, "spread": 1.0, "gravity": 0.05, "up": -0.2},
|
|
}
|
|
|
|
|
|
class Particle:
|
|
__slots__ = ("x", "y", "vx", "vy", "life", "maxLife", "color", "gravity", "size")
|
|
|
|
def __init__(self, x, y, vx, vy, life, color, gravity, size):
|
|
self.x = x
|
|
self.y = y
|
|
self.vx = vx
|
|
self.vy = vy
|
|
self.life = life
|
|
self.maxLife = life
|
|
self.color = color
|
|
self.gravity = gravity
|
|
self.size = size
|
|
|
|
def update(self):
|
|
self.vy += self.gravity
|
|
self.x += self.vx
|
|
self.y += self.vy
|
|
self.life -= 1
|
|
|
|
@property
|
|
def alive(self):
|
|
return self.life > 0
|
|
|
|
@property
|
|
def fade(self):
|
|
return max(0.0, self.life / self.maxLife)
|
|
|
|
|
|
class ParticleSystem:
|
|
def __init__(self, rng=None, cap=512):
|
|
self.particles = []
|
|
self.rng = rng if rng is not None else random.Random(1234)
|
|
self.cap = cap
|
|
|
|
def emit(self, preset, x, y, count=None):
|
|
config = particlePresets.get(preset, particlePresets["dust"])
|
|
emitCount = count if count is not None else config["count"]
|
|
for _ in range(emitCount):
|
|
if len(self.particles) >= self.cap:
|
|
break
|
|
angle = self.rng.uniform(0, math.tau)
|
|
magnitude = self.rng.uniform(0.2, config["spread"])
|
|
vx = math.cos(angle) * magnitude
|
|
vy = math.sin(angle) * magnitude + config["up"]
|
|
life = int(config["life"] * self.rng.uniform(0.7, 1.1))
|
|
size = self.rng.randint(1, 2)
|
|
self.particles.append(Particle(x, y, vx, vy, life, config["color"], config["gravity"], size))
|
|
|
|
def update(self):
|
|
survivors = []
|
|
for particle in self.particles:
|
|
particle.update()
|
|
if particle.alive:
|
|
survivors.append(particle)
|
|
self.particles = survivors
|
|
|
|
def clear(self):
|
|
self.particles = []
|
|
|
|
def __len__(self):
|
|
return len(self.particles)
|