X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=koperkapel%2Fweapons.py;h=7dcb41ed154250bcc009f0f5ce8fdf80261e2a56;hb=c1691479e387f6676f03dbf6e431345eea60dd65;hp=d2213f6144a4e8f9b98ebf8e0e5611807b6a8e6a;hpb=b83ddd7e5bf2ea9c59f6fb2d78ebddc9dc845046;p=koperkapel.git

diff --git a/koperkapel/weapons.py b/koperkapel/weapons.py
index d2213f6..7dcb41e 100644
--- a/koperkapel/weapons.py
+++ b/koperkapel/weapons.py
@@ -1,5 +1,7 @@
 """ Tools for dealing with weapons. """
 
+from pygame.surface import Surface
+from pygame.draw import circle
 from pgzero.loaders import images, sounds
 from .actors.animsurf import AnimatedSurfActor
 from .util import safepath
@@ -18,6 +20,13 @@ class Weapon:
         if sound:
             self.sound = sounds.load(sound)
 
+    def play_sound(self):
+        if self.sound:
+            self.sound.play()
+
+    def assemble_bullet(self):
+        return BulletActor(self)
+
 
 WEAPONS = [
     Weapon("spit", damage=1, bullet_range=2, can_tape=False,
@@ -30,17 +39,33 @@ WEAPONS = [
 WEAPON_LOOKUP = {w.name: w for w in WEAPONS}
 
 
+def weapon_by_name(weapon_name):
+    return WEAPON_LOOKUP[weapon_name]
+
+
 class WeaponActor(AnimatedSurfActor):
     def __init__(self, weapon, **kw):
         super().__init__(**kw)
         self.weapon = weapon
 
 
+class BulletActor(AnimatedSurfActor):
+    def __init__(self, weapon, **kw):
+        radius = min(weapon.damage * 5, 32)
+        surf = Surface((radius * 2, radius * 2))
+        surf.convert_alpha()
+        surf.fill((255, 255, 255, 0))
+        circle(surf, (255, 0, 255, 32), (radius, radius), radius)
+        frames = [surf]
+        super().__init__(frames=frames, **kw)
+        self.weapon = weapon
+
+
 class WeaponFactory:
 
     def assemble_frame(self, suffix, weapon, tape):
         surf = images.load(safepath("weapons/%s%s")
-                            % (weapon.image_name, suffix))
+                           % (weapon.image_name, suffix))
         frame = surf.copy()
         if tape:
             tape_surf = images.load(safepath("weapons/tape"))
@@ -48,7 +73,7 @@ class WeaponFactory:
         return frame
 
     def assemble(self, weapon_name, tape=False):
-        weapon = WEAPON_LOOKUP[weapon_name]
+        weapon = weapon_by_name(weapon_name)
         tape = tape and weapon.can_tape
         frames = [
             self.assemble_frame(suffix, weapon, tape)