Default default action.

[naja.git] / naja / gameboard.py

diff --git a/naja/gameboard.py b/naja/gameboard.py
index 5ec61507c7e31a5b10b7c28780b427414e97c462..980a046c4c4f68e690219a2a216f818df018ce9e 100644 (file)
--- a/naja/gameboard.py
+++ b/naja/gameboard.py
@@ -1,6 +1,8 @@
 from random import choice
 
-from naja.constants import BITS, DIRECTION_BITS, CONDITION_BITS
+from naja.constants import(
+    BITS, DIRECTION_BITS, CONDITION_BITS, PLAYER_DEFAULTS,
+    MOVE, ACT)
 from naja.player import Player
 from naja import actions
 
@@ -15,13 +17,17 @@ class GameBoard(object):
         self.wins_required = state['wins_required']
         self.health = state['health']
         self.wins = state['wins']
-        self.locations = self.import_locations(state['locations'])
+        self.locations = [item.copy() for item in state['locations']]
         self.player = player
         self.board_locations = board_locations
+        self.player_mode = MOVE
 
     @classmethod
-    def new_game(cls, initial_bits, initial_pos, max_health, wins_required,
-                 locations_definition):
+    def new_game(cls, locations_definition,
+                 initial_bits=PLAYER_DEFAULTS.INITIAL_BITS,
+                 initial_pos=PLAYER_DEFAULTS.INITIAL_POS,
+                 max_health=PLAYER_DEFAULTS.MAX_HEALTH,
+                 wins_required=PLAYER_DEFAULTS.WINS_REQUIRED):
         state = {
             'max_health': max_health,
             'health': max_health,
@@ -30,7 +36,8 @@ class GameBoard(object):
             'locations': locations_definition,
         }
         player = Player(initial_bits, initial_pos)
-        board_locations = cls.generate_board(locations_definition)
+        board_locations = cls.import_board_locations(
+            cls.generate_board(locations_definition))
         return cls(state, player, board_locations)
 
     @classmethod
@@ -47,14 +54,11 @@ class GameBoard(object):
             'health': self.health,
             'wins_required': self.wins_required,
             'wins': self.wins,
-            'locations': self.export_locations(),
+            'locations': [item.copy() for item in self.locations],
             'player': self.player.export(),
             'board_locations': self.export_board_locations(),
         }
 
-    def export_locations(self):
-        return [location.export() for location in self.locations]
-
     @classmethod
     def import_locations(cls, locations_definition):
         return [
@@ -64,7 +68,7 @@ class GameBoard(object):
     def export_board_locations(self):
         return dict(
             (position, location.export())
-            for position, location in self.board_locations)
+            for position, location in self.board_locations.iteritems())
 
     @classmethod
     def import_board_locations(cls, board_locations_definition):
@@ -74,13 +78,27 @@ class GameBoard(object):
 
     @classmethod
     def generate_board(cls, locations_definition):
-        # TODO: Choose some locations.
-        return {}
+        board_locations = {}
+        for x in range(5):
+            for y in range(5):
+                board_location = LocationCard.new_location(
+                    choice(locations_definition).copy())
+                board_locations[(x, y)] = board_location.export()
+        return board_locations
 
     def lose_health(self):
         self.health -= 1
         # TODO: Check win/lose
 
+    def change_mode(self):
+        """Advance to the next mode"""
+        if self.player_mode == MOVE:
+            self.player_mode = ACT
+        elif self.player_mode == ACT:
+            self.player_mode = MOVE
+        else:
+            raise RuntimeError("Illegal player mode %s" % self.player_mode)
+
 
 class LocationCard(object):
     """
@@ -90,10 +108,10 @@ class LocationCard(object):
     def __init__(self, bitwise_operand, location_actions):
         self.bitwise_operand = bitwise_operand
         self.actions = location_actions
+        self.check_actions()
 
     @classmethod
     def import_location(cls, state):
-        # TODO: Import real locations.
         location_actions = [
             cls.build_action(definition) for definition in state['actions']]
         return cls(state['bitwise_operand'], location_actions)
@@ -118,6 +136,19 @@ class LocationCard(object):
             'actions': [action.export() for action in self.actions],
         }
 
+    def check_actions(self):
+        if not self.actions:
+            print "Warning: Location has no actions."
+            self.insert_default_default_action()
+        if self.actions[0].required_bits:
+            self.insert_default_default_action()
+
+    def insert_default_default_action(self):
+        self.actions.insert(0, self.build_action({
+            'action_class': 'DoNothing',
+            'required_bits': [],
+        }))
+
     @staticmethod
     def generate_bitwise_operand():
         """