Initial CommitHEADmain

1 files changed, 88 insertions, 0 deletions

diff --git a/shooter.py b/shooter.py
new file mode 100644
index 0000000..94878e3
--- /dev/null
+++ b/shooter.py
@@ -0,0 +1,88 @@
+from dataclasses import dataclass
+from typing import List
+
+from game import DamageableSprite, Game, GameSprite
+from sprite import Sprite
+
+
+class Lazer(GameSprite):
+    """Lazer object that is shot by a shooter"""
+
+    def __init__(self, game: Game, velocity=-4, color="white"):
+        """Initialise the lazer
+
+        :param game: The game which this belongs to
+        :type game: Game
+        :param velocity: Velocity to move the lazer at
+        :param color: name of the colour of the lazer
+        """
+        self.velocity = velocity
+        self.game = game
+        super().__init__(game, game.texture_factory.get_image(
+            f"lazer:{color}"))
+
+    def tick(self):
+        """Update this object"""
+        self.move(0, self.velocity)
+        if self.y + self.h > self.game.h or self.y < 0:
+            self.destroy()
+
+
+@dataclass
+class ShooterAttributes:
+    """Attributes for a shooter object"""
+
+    lazer_color: str = "white"
+    cooldown: int = 40
+    velocity: int = 1
+    hp: int = 3
+
+
+class Shooter(DamageableSprite):
+    """A game object that is able to shoot lazers"""
+
+    def __init__(self, game: Game,
+                 image_name: str, attributes: ShooterAttributes):
+        """Initialise the shooter
+
+        :param game: The game which this belongs to
+        :type game: Game
+        :param image_name: The name of the image to use for this sprite
+        :type image_name: str
+        :param attributes: The attributes to use for this object
+        :type attributes: ShooterAttributes
+        """
+        super().__init__(game, image_name, hp=attributes.hp)
+        self.lazers: List[Lazer] = []
+        self.attributes = attributes
+        self.last_shot = self.game.alpha
+
+    def shoot(self):
+        """Soot a lazer if possible"""
+        next_shot = self.last_shot + self.attributes.cooldown
+
+        if not self.destroyed \
+                and self.game.alpha > next_shot:
+            self.last_shot = self.game.alpha
+
+            lazer = Lazer(self.game,
+                          velocity=self.attributes.velocity,
+                          color=self.attributes.lazer_color)
+            lazer.set_pos((self.x + self.w//2 - 1, self.y +
+                          self.h//2 - 1))
+            lazer.show()
+            self.lazers.append(lazer)
+
+    def tick(self):
+        """Update this object"""
+        super().tick()
+        for lazer in self.lazers:
+            lazer.tick()
+
+        self.lazers = Sprite.remove_destroyed(self.lazers)
+
+    def destroy(self):
+        """Remove all the associated objects"""
+        super().destroy()
+        for lazer in self.lazers:
+            self.game.sprites.append(lazer)