Initial commit

main.py

@@ -0,0 +1,215 @@
+"""
+TO DO:
+    - Splash screen
+    - Play again?
+"""
+
+import random
+import time
+import pgzrun
+from pgzhelper import *
+
+WIDTH = 800
+HEIGHT = 600
+
+# globals
+score = 0
+health = 19
+health_holdoff = 0
+powerup_time = 0
+game_over_played = False
+
+lasers = []
+laser_holdoff = 0
+laser_hit_idx = 0
+
+music.play('music1')
+
+
+# local helper functions
+def rotate(actor):
+    if actor.angle == 360:
+        actor.angle = 0
+    else:
+        actor.angle += 1
+
+
+def relocate(actor, scale=1):
+    # Avoid ship
+    x = ship.x
+    y = ship.y
+    while (x == ship.x) and (y == ship.y):
+        x = random.randint(20, 780)
+        y = random.randint(20, 580)
+    actor.x = x
+    actor.y = y
+    actor.scale = scale
+    actor.direction = random.randint(0, 359)
+
+
+background = Actor('spacebg')
+
+health_bar = Actor(f'health{health}')
+health_bar.scale = 0.8
+health_bar.x = 140
+health_bar.y = 30
+
+ship = Actor('playership')
+ship.images = [ship.image, 'playership_damage']
+ship.fps = 10
+ship.y = 550
+ship.x = 400
+
+meteors = [Actor('meteorbrown1'), Actor('meteorbrown2'), Actor('meteorbrown3'),
+           Actor('meteorbrown4'), Actor('meteorgrey1'), Actor('meteorgrey2'),
+           Actor('meteorgrey3'), Actor('meteorgrey4')]
+
+for meteor in meteors:
+    relocate(meteor, round(random.uniform(0.5, 1.5), 1))
+
+explosions = []
+powerups = []
+
+
+def on_mouse_move(pos, rel, buttons):
+    ship.x = pos[0]
+    ship.y = pos[1]
+
+
+def on_mouse_down(pos, button):
+    global lasers, laser_holdoff
+    if button == 1:
+        if laser_holdoff == 0:
+            sounds.laser.play()
+            laser = Actor('laser')
+            laser.angle = 90
+            laser.x = ship.x
+            laser.y = ship.y - 30
+            lasers.append(laser)
+            laser_holdoff = 1
+        else:
+            laser_holdoff -= 1
+
+def update():
+    global score, health, health_holdoff, laser_hit_idx, powerup_time
+
+    health_bar.image = f'health{health}'
+    health_bar.scale = 0.8
+    health_bar.x = 140
+    health_bar.y = 30
+
+    # meteors
+    for meteor in meteors:
+        rotate(meteor)
+        meteor.move_in_direction(round(random.uniform(0.5, 1.5), 1))  # meteor speed
+
+        # meteor collisions
+        if meteor.collidelist(lasers) != -1:
+            try:
+                del lasers[laser_hit_idx]
+            except IndexError:
+                pass
+            finally:
+                explotion = Actor('explosion_1')
+                explotion.images = [explotion.image, 'explosion_2', 'explosion_3',
+                                    'explosion_4', 'explosion_5', 'explosion_6', 'explosion_7',
+                                    'explosion_8', 'explosion_9', 'explosion_10']
+                explotion.x = meteor.x
+                explotion.y = meteor.y
+                explotion.scale = 0.5
+                explotion.fps = 10
+                explosions.append(explotion)
+                sounds.explosion.play()
+                score += 1
+                relocate(meteor, round(random.uniform(0.5, 1.5), 1))
+
+        if (meteor.x > 800) or (meteor.y > 600) or (meteor.x < 0) or (meteor.y < 0):
+            relocate(meteor, round(random.uniform(0.5, 1.5), 1))
+
+    for explotion in explosions:
+        if explotion.animate() >= 9:
+            explosions.remove(explotion)
+
+    # ship collides with meteors
+    if ship.collidelist(meteors) != -1:
+        ship.animate()
+        if health_holdoff == 0:
+            if health > 0:
+                health -= 1
+                sounds.loose.play()
+                health_holdoff = 50
+        else:
+            health_holdoff -= 1
+    else:
+        ship.image = 'playership'
+
+    # laser collisions
+    for idx, laser in enumerate(lasers):
+        if laser.y > 595:
+            lasers.remove(laser)
+        elif laser.collidelist(meteors) != -1:
+            laser_hit_idx = idx
+        else:
+            laser.y -= 5
+
+    # Power Up
+    if health < 19 and powerup_time == 0:
+        # TO DO: fix random time - use a counter with random values: powerup_time
+        powerup = Actor('powerup')
+        relocate(powerup)
+        powerups.append(powerup)
+        powerup_time = random.randint(100, 300)
+
+    if powerup_time > 0:
+        powerup_time -= 1
+
+    for powerup in powerups:
+        powerup.move_in_direction(round(random.uniform(0.5, 1.5), 1))  # powerup speed
+        # detect collision with ship
+        if powerup.colliderect(ship):
+            powerups.remove(powerup)
+            sounds.powerup.set_volume(4)
+            sounds.powerup.play()
+            health += 1
+            if health > 19:
+                health = 19
+        # remove if outside the screen
+        elif (powerup.x > 800) or (powerup.y > 600) or (powerup.x < 0) or (powerup.y < 0):
+            powerups.remove(powerup)
+
+def draw():
+    global game_over_played
+    screen.fill((0, 0, 0))
+    background.draw()
+    health_bar.draw()
+    if health > 0:
+        screen.draw.text(f'Score: {score}', (330, 18), color=(255, 255, 255), fontsize=40)
+
+        for meteor in meteors:
+            meteor.draw()
+
+        for explotion in explosions:
+            explotion.draw()
+
+        for powerup in powerups:
+            powerup.draw()
+
+        ship.draw()
+
+        for laser in lasers:
+            laser.draw()
+    else:
+        powerups.clear()
+        meteors.clear()
+        explosions.clear()
+        lasers.clear()
+        screen.draw.text(f'Score: {score}', (250, 300), color=(245, 243, 127), fontsize=100)
+        screen.draw.text('Game Over', (220, 450), color=(255, 255, 255), fontsize=100)
+        music.stop()
+        if not game_over_played:
+            sounds.game_over.play()
+            game_over_played = True
+
+
+# Main loop
+pgzrun.go()