tic_tac_toe.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {
    "collapsed": true,
    "deletable": true,
    "editable": true
   },
   "outputs": [],
   "source": [
    "from ipywidgets import widgets, HBox, VBox, Layout\n",
    "from IPython.display import display\n",
    "from functools import partial\n",
    "import numpy as np\n",
    "#import tictactoe\n",
    "import tictactoe_minimax_helper as minimax_helper"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# tic tac toe game using minimax algorithm"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "deletable": true,
    "editable": true
   },
   "source": [
    "References:\n",
    "http://neverstopbuilding.com/minimax"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {
    "collapsed": false,
    "deletable": true,
    "editable": true
   },
   "outputs": [],
   "source": [
    "class General_functions(object):\n",
    "    def __init__(self, matrix, actual_turn):\n",
    "        self.N = 3\n",
    "        self.button_list = None\n",
    "        self.text_box = None\n",
    "        self.matrix = matrix\n",
    "        self.game_finished = False\n",
    "        self.actual_turn = actual_turn\n",
    "        \n",
    "    def display_matrix(self):\n",
    "        N = self.N\n",
    "        childs = []\n",
    "        for i in range(N):\n",
    "            for j in range(N):\n",
    "                if self.matrix[i,j]==1:\n",
    "                    self.button_list[i*N + j].description = 'o'\n",
    "                if self.matrix[i,j]==-1:\n",
    "                    self.button_list[i*N + j].description = 'x'\n",
    "\n",
    "    def on_button_clicked(self, index, button):\n",
    "        N = self.N  \n",
    "\n",
    "        if self.game_finished:\n",
    "            return\n",
    "\n",
    "        y = index%N\n",
    "        x = int(index/N)\n",
    "        if self.matrix[x,y]!=0:\n",
    "            self.text_box.value = 'No se puede ahi!'\n",
    "            return\n",
    "        button.description = self.actual_turn[0]\n",
    "\n",
    "        if self.actual_turn == 'o':\n",
    "            self.matrix[x,y] = 1\n",
    "            self.game_finished, status = minimax_helper.game_over(self.matrix)\n",
    "            if self.game_finished:\n",
    "                if (status!=0):\n",
    "                    self.text_box.value = 'o wins'\n",
    "                else: \n",
    "                    self.text_box.value = 'draw'\n",
    "            else:\n",
    "                self.actual_turn = 'x'\n",
    "                self.text_box.value = 'Juega '+self.actual_turn\n",
    "        else:\n",
    "            self.matrix[x,y] = -1\n",
    "            self.game_finished, status = minimax_helper.game_over(self.matrix)\n",
    "            if self.game_finished:\n",
    "                if (status!=0):\n",
    "                    self.text_box.value = 'x wins'\n",
    "                else: \n",
    "                    self.text_box.value = 'draw'\n",
    "            else:\n",
    "                self.actual_turn = 'o'\n",
    "                self.text_box.value = 'Juega '+self.actual_turn\n",
    "        self.computer_play()\n",
    "        \n",
    "    def draw_board(self):\n",
    "        self.text_box = widgets.Text(value = 'Juega '+self.actual_turn, layout=Layout(width='129px', height='40px'))\n",
    "        self.button_list = []\n",
    "        for i in range(9):\n",
    "            button = widgets.Button(description='',\n",
    "            disabled=False,\n",
    "            button_style='', # 'success', 'info', 'warning', 'danger' or ''\n",
    "            tooltip='Click me',\n",
    "            icon='',\n",
    "            layout=Layout(width='40px', height='40px'))\n",
    "            self.button_list.append(button)\n",
    "            button.on_click(partial(self.on_button_clicked, i))\n",
    "        tic_tac_toe_board = VBox([HBox([self.button_list[0],self.button_list[1],self.button_list[2]]),\n",
    "                HBox([self.button_list[3],self.button_list[4],self.button_list[5]]),\n",
    "                HBox([self.button_list[6],self.button_list[7],self.button_list[8]])])\n",
    "        display(VBox([self.text_box, tic_tac_toe_board]))\n",
    "        return\n",
    "\n",
    "    def computer_play(self):\n",
    "\n",
    "        if self.game_finished:\n",
    "            return\n",
    "        \n",
    "        if self.actual_turn=='x':\n",
    "            turn = -1\n",
    "            next_turn = 'o'\n",
    "        if self.actual_turn=='o':\n",
    "            turn = 1\n",
    "            next_turn = 'x'\n",
    "        self.matrix = self.get_best_play(turn)\n",
    "        self.display_matrix()\n",
    "        self.actual_turn = next_turn\n",
    "        self.text_box.value = 'Juega '+self.actual_turn\n",
    "        self.game_finished, status = minimax_helper.game_over(self.matrix)\n",
    "        if self.game_finished:\n",
    "            if (status!=0):\n",
    "                self.text_box.value = 'computer wins'\n",
    "            else: \n",
    "                self.text_box.value = 'draw'\n",
    "\n",
    "    def get_best_play(self, turn):\n",
    "        # 1000 is an infinite value compared with the highest cost of 10 that we can get\n",
    "\n",
    "        choice, points, nodes_visited = minimax_helper.minimax(self.matrix, turn)\n",
    "        print('points:',points)\n",
    "        print('nodes_visited:',nodes_visited)\n",
    "        return choice"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {
    "collapsed": false,
    "deletable": true,
    "editable": true
   },
   "outputs": [],
   "source": [
    "def start_game(computer_starts = True, user_icon='x', start_mode = 'center'):\n",
    "    matrix = np.zeros((3,3))\n",
    "    \n",
    "    if user_icon=='x':\n",
    "        computer_icon_representation = 1\n",
    "    else:\n",
    "        computer_icon_representation = -1\n",
    "\n",
    "    GF = General_functions(matrix, user_icon)\n",
    "    GF.draw_board()\n",
    "\n",
    "    if computer_starts:\n",
    "        if start_mode == 'center':\n",
    "            matrix[1,1] = computer_icon_representation\n",
    "        elif start_mode == 'minimax':\n",
    "            GF.computer_play()\n",
    "        elif start_mode == 'random':\n",
    "            x = np.random.randint(3)\n",
    "            y = np.random.randint(3)\n",
    "            matrix[x,y] = computer_icon_representation\n",
    "\n",
    "    GF.display_matrix()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "524a1ed3446c4915ae69e4b58dfe986e"
      }
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "points: 0\n",
      "nodes_visited: 831\n",
      "points: 0\n",
      "nodes_visited: 104\n",
      "points: 0\n",
      "nodes_visited: 11\n",
      "points: 9\n",
      "nodes_visited: 1\n"
     ]
    }
   ],
   "source": [
    "# start_mode:\n",
    "# 'minimax': Uses minimax to select the first move\n",
    "# 'center': Starts on the center\n",
    "# 'random': Starts on a random position\n",
    "# user_icon:\n",
    "#  'x': user is x\n",
    "#  'o': user is o\n",
    "# computer_starts: True or False\n",
    "\n",
    "start_game(computer_starts = True, user_icon = 'x', start_mode = 'random')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": true,
    "deletable": true,
    "editable": true
   },
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "anaconda-cloud": {},
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.6.0"
  },
  "toc": {
   "colors": {
    "hover_highlight": "#DAA520",
    "navigate_num": "#000000",
    "navigate_text": "#333333",
    "running_highlight": "#FF0000",
    "selected_highlight": "#FFD700",
    "sidebar_border": "#EEEEEE",
    "wrapper_background": "#FFFFFF"
   },
   "moveMenuLeft": true,
   "nav_menu": {
    "height": "30px",
    "width": "252px"
   },
   "navigate_menu": true,
   "number_sections": true,
   "sideBar": true,
   "threshold": 4,
   "toc_cell": false,
   "toc_section_display": "block",
   "toc_window_display": false,
   "widenNotebook": false
  },
  "widgets": {
   "state": {
    "0688572cdfda44de9cc3994ffe0faf33": {
     "views": [
      {
       "cell_index": 3
      }
     ]
    }
   },
   "version": "1.2.0"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 1
}