asteroid.py asteroid_spawner.py spaceship.py nav_game.py

asteroid.py

@@ -2,3 +2,26 @@
 # ------------
 # By MWC Contributors
 # This module defines an asteroid agent class.
+
+class Asteroid:
+    character = 'O'
+    color="orange" # color added
+
+    def __init__(self, position):
+        self.position = position
+
+    def play_turn(self, game):
+        width, height = game.board_size
+        if game.turn_number % 2 == 0: 
+            x, y = self.position
+            if y == height - 1: 
+                game.remove_agent(self)
+            else:
+                ship = game.get_agent_by_name('ship')
+                new_position = (x, y + 1)
+                if new_position == ship.position: # Watch Out!
+                    ship.character="*" # changed to a collision indicator
+                    ship.color="red" # changed to a collision indicator
+                    game.end()
+                else:
+                    self.position = new_position

asteroid_spawner.py

@@ -2,3 +2,19 @@
 # -------------------
 # By MWC Contributors
 # This module defines an AsteroidSpawner agent class.
+
+from random import randint
+from asteroid import Asteroid
+
+class AsteroidSpawner:
+    display = False
+
+    def play_turn(self, game):
+        width, height = game.board_size
+        game.state['score'] += 1
+        if self.should_spawn_asteroid(game.turn_number):
+            asteroid = Asteroid((randint(0, width - 1), 0))
+            game.add_agent(asteroid)
+
+    def should_spawn_asteroid(self, turn_number):
+        return randint(0, 1000) < turn_number

nav_game.py

@@ -2,3 +2,13 @@
 # ------------
 # By MWC Contributors
 # This class implements a simple game where a spaceship avoids asteroids.
+
+from retro.game import Game
+from spaceship import Spaceship
+from asteroid_spawner import AsteroidSpawner
+
+board_size = (25, 25)
+ship = Spaceship(board_size)
+spawner = AsteroidSpawner()
+game = Game([ship, spawner], {"score": 0}, board_size=board_size)
+game.play()

spaceship.py

@@ -2,3 +2,26 @@
 # ------------
 # By MWC Contributors
 # This module defines a spaceship agent class.
+
+class Spaceship:
+    name = "ship"
+    character = '^'
+    color="blue"
+
+    def __init__(self, board_size):
+        board_width, board_height = board_size
+        self.position = (board_width // 2, board_height - 1)
+
+    def handle_keystroke(self, keystroke, game):
+        x, y = self.position
+        if keystroke.name in ("KEY_LEFT", "KEY_RIGHT"):
+            if keystroke.name == "KEY_LEFT": 
+                new_position = (x - 1, y)
+            else: 
+                new_position = (x + 1, y)
+            if game.on_board(new_position):
+                if game.is_empty(new_position):
+                    self.position = new_position
+                else:
+                    
+                    game.end()

test.py

@@ -0,0 +1,7 @@
+from retro.game import Game
+from retro.agent import ArrowKeyAgent
+
+agent = ArrowKeyAgent()
+state = {}
+game = Game([agent], state)
+game.play()