- Added demo mode that just shows, the AI playing against random player.

AngelVI13 · May 6, 2019 · 833fa79 · 833fa79
1 parent 4d71d19
commit 833fa79
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/gui/constants.py b/gui/constants.py
@@ -21,6 +21,7 @@
 class GameType(Enum):
     SINGLE_PLAYER = auto()
     MULTI_PLAYER = auto()
+    DEMO_MODE = auto()
 
 
 BORDER_THICKNESS = 2

diff --git a/gui/gui_app.py b/gui/gui_app.py
@@ -99,12 +99,8 @@ def click_random_cell(self):
     def get_best_engine_move(self):
         return uct_multi(self.board, itermax=1000, verbose=False)
 
-    def do_nothing(self):
-        pass  # todo remove this later
-
     def get_game_input(self, game_type, mouse_pos):
         if game_type == GameType.SINGLE_PLAYER:
-            # self.click_random_cell() used for debugginh
             if self.board.playerJustMoved == PLAYER_O:  # X's turn  todo allow user to select side to play
                 if mouse_pos is not None:
                     self.click_cell_under_mouse(mouse_pos)
@@ -121,6 +117,18 @@ def get_game_input(self, game_type, mouse_pos):
             if mouse_pos is not None:
                 self.click_cell_under_mouse(mouse_pos)
 
+        elif game_type == GameType.DEMO_MODE:
+            if self.board.playerJustMoved == PLAYER_O:
+                self.click_random_cell()
+            else:
+                board, move = self.get_best_engine_move()
+                for cell in self.allowed_cells:
+                    if cell.board_idx == board and cell.cell_idx == move:
+                        self.subcell_clicked(cell)
+                        break
+                else:
+                    raise Exception('Wrong engine move (move not in allowed moves) ({}{})'.format(board, move))
+
     def game_loop(self, game_type):
         pygame.event.clear(pygame.MOUSEBUTTONUP)  # clear all mouse clicks
         self.reset_game()
@@ -176,7 +184,7 @@ def menu_loop(self):
                         action=partial(self.game_loop, GameType.SINGLE_PLAYER))
             self.button("Two Player", **MENU_BUTTON_PROPERTIES[1],
                         action=partial(self.game_loop, GameType.MULTI_PLAYER))
-            self.button("Settings", **MENU_BUTTON_PROPERTIES[2], action=self.do_nothing)
+            self.button("Demo", **MENU_BUTTON_PROPERTIES[2], action=partial(self.game_loop, GameType.DEMO_MODE))
             self.button("Quit", **MENU_BUTTON_PROPERTIES[3], action=self.quit_game)
 
             pygame.display.update()