Source code for arcade.emitter

"""
Emitter - Invisible object that determines when Particles are emitted, actually emits them, and manages them
over their lifetime
"""

import arcade
from arcade.particle import Particle
from typing import Callable, cast
from arcade.utils import _Vec2
from arcade.arcade_types import Point, Vector


##########
[docs]class EmitController: """Base class for how a client configure the rate at which an Emitter emits Particles Subclasses allow the client to control the rate and duration of emitting""" def how_many(self, delta_time: float, current_particle_count: int) -> int: raise NotImplementedError("EmitterRate.how_many must be implemented") def is_complete(self) -> bool: raise NotImplementedError("EmitterRate.is_complete must be implemented")
[docs]class EmitBurst(EmitController): """Used to configure an Emitter to emit particles in one burst""" def __init__(self, count: int): self._is_complete = False self._count = count def how_many(self, delta_time: float, current_particle_count: int) -> int: if not self._is_complete: self._is_complete = True return self._count return 0 def is_complete(self) -> bool: return True
[docs]class EmitMaintainCount(EmitController): """Used to configure an Emitter so it emits particles so that the given count is always maintained""" def __init__(self, particle_count: int): self._target_count = particle_count def how_many(self, delta_time: float, current_particle_count: int) -> int: return self._target_count - current_particle_count def is_complete(self) -> bool: return False
[docs]class EmitInterval(EmitController): """Base class used to configure an Emitter to have a constant rate of emitting. Will emit indefinitely.""" def __init__(self, emit_interval: float): self._emit_interval = emit_interval self._carryover_time = 0.0 def how_many(self, delta_time: float, current_particle_count: int) -> int: self._carryover_time += delta_time emit_count = 0 while self._carryover_time >= self._emit_interval: self._carryover_time -= self._emit_interval emit_count += 1 return emit_count def is_complete(self) -> bool: return False
[docs]class EmitterIntervalWithCount(EmitInterval): """Configure an Emitter to emit particles with given interval, ending after emitting given number of particles""" def __init__(self, emit_interval: float, particle_count: int): super().__init__(emit_interval) self._count_remaining = particle_count def how_many(self, delta_time: float, current_particle_count: int) -> int: proposed_count = super().how_many(delta_time, current_particle_count) actual_count = min(proposed_count, self._count_remaining) self._count_remaining -= actual_count return actual_count def is_complete(self) -> bool: return self._count_remaining <= 0
[docs]class EmitterIntervalWithTime(EmitInterval): """Configure an Emitter to emit particles with given interval, ending after given number of seconds""" def __init__(self, emit_interval: float, lifetime: float): super().__init__(emit_interval) self._lifetime = lifetime def how_many(self, delta_time: float, current_particle_count: int) -> int: if self._lifetime <= 0.0: return 0 self._lifetime -= delta_time return super().how_many(delta_time, current_particle_count) def is_complete(self) -> bool: return self._lifetime <= 0
# Emitter
[docs]class Emitter: """Emits and manages Particles over their lifetime. The foundational class in a particle system.""" def __init__( self, center_xy: Point, emit_controller: EmitController, particle_factory: Callable[["Emitter"], Particle], change_xy: Vector = (0.0, 0.0), emit_done_cb: Callable[["Emitter"], None] = None, reap_cb: Callable[[], None] = None ): # Note Self-reference with type annotations: # https://www.python.org/dev/peps/pep-0484/#the-problem-of-forward-declarations self.change_x = change_xy[0] self.change_y = change_xy[1] self.center_x = center_xy[0] self.center_y = center_xy[1] self.angle = 0.0 self.change_angle = 0.0 self.rate_factory = emit_controller self.particle_factory = particle_factory self._emit_done_cb = emit_done_cb self._reap_cb = reap_cb self._particles: arcade.SpriteList = arcade.SpriteList(use_spatial_hash=False) def _emit(self): """Emit one particle, its initial position and velocity are relative to the position and angle of the emitter""" p = self.particle_factory(self) p.center_x += self.center_x p.center_y += self.center_y # given the velocity, rotate it by emitter's current angle vel = _Vec2(p.change_x, p.change_y).rotated(self.angle) p.change_x = vel.x p.change_y = vel.y self._particles.append(p) def get_count(self): return len(self._particles)
[docs] def get_pos(self) -> Point: """Get position of emitter""" # TODO: should this be a property so a method call isn't needed? return self.center_x, self.center_y
def update(self): # update emitter self.center_x += self.change_x self.center_y += self.change_y self.angle += self.change_angle # update particles emit_count = self.rate_factory.how_many(1 / 60, len(self._particles)) for _ in range(emit_count): self._emit() self._particles.update() particles_to_reap = [p for p in self._particles if cast(Particle, p).can_reap()] for dead_particle in particles_to_reap: dead_particle.kill() def draw(self): self._particles.draw()
[docs] def can_reap(self) -> bool: """Determine if Emitter can be deleted""" is_emit_complete = self.rate_factory.is_complete() can_reap = is_emit_complete and len(self._particles) <= 0 if is_emit_complete and self._emit_done_cb: self._emit_done_cb(self) self._emit_done_cb = None if can_reap and self._reap_cb: self._reap_cb() self._reap_cb = None return can_reap