A few better types in arcade.future and arcade.particles.

Adding type parameters to some `SpriteList`s. One allows to get rid
of a cast.

Author: sjrd

arcade/future/input/input_manager_example.py

@@ -26,7 +26,7 @@ class Player(arcade.Sprite):
     def __init__(
         self,
         texture,
-        walls: arcade.SpriteList,
+        walls: arcade.SpriteList[arcade.Sprite],
         input_manager_template: InputManager,
         controller: pyglet.input.Controller | None = None,
         center_x: float = 0.0,
@@ -76,11 +76,11 @@ def __init__(
         }
 
         self.players: list[Player | None] = []
-        self.player_list = arcade.SpriteList()
+        self.player_list: arcade.SpriteList[Player] = arcade.SpriteList()
         self.device_labels_batch = pyglet.graphics.Batch()
         self.player_device_labels: list[arcade.Text | None] = []
 
-        self.wall_list = arcade.SpriteList(use_spatial_hash=True)
+        self.wall_list: arcade.SpriteList[arcade.Sprite] = arcade.SpriteList(use_spatial_hash=True)
 
         for x in range(0, self.width + 64, 64):
             wall = arcade.Sprite(":resources:images/tiles/grassMid.png", scale=0.5)

arcade/future/light/light_demo.py

@@ -18,7 +18,7 @@ def __init__(self, width, height, title):
         super().__init__(width, height, title)
         self.background = arcade.load_texture(":resources:images/backgrounds/abstract_1.jpg")
 
-        self.torch_list = arcade.SpriteList()
+        self.torch_list: arcade.SpriteList[arcade.Sprite] = arcade.SpriteList()
         self.torch_list.extend(
             [
                 arcade.Sprite(

arcade/particles/emitter.py

@@ -5,7 +5,7 @@
 
 from __future__ import annotations
 
-from typing import Callable, cast
+from typing import Callable
 
 import arcade
 from arcade import Vec2
@@ -151,7 +151,7 @@ def __init__(
         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)
+        self._particles: arcade.SpriteList[Particle] = arcade.SpriteList(use_spatial_hash=False)
 
     def _emit(self):
         """
@@ -189,7 +189,7 @@ def update(self, delta_time: float = 1 / 60):
         for _ in range(emit_count):
             self._emit()
         self._particles.update(delta_time)
-        particles_to_reap = [p for p in self._particles if cast(Particle, p).can_reap()]
+        particles_to_reap = [p for p in self._particles if p.can_reap()]
         for dead_particle in particles_to_reap:
             dead_particle.kill()