gif_maker.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pickle\n",
    "import matplotlib.pyplot as plt\n",
    "from game import Game\n",
    "%matplotlib inline\n",
    "from matplotlib.animation import FuncAnimation\n",
    "import numpy as np\n",
    "import moviepy.editor as mp"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "version = 'ohe_2_steps_move_penalty_log_reward_512_3_layers_random'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [],
   "source": [
    "with open('./data/best_game_history_%s.pkl' % version, 'rb') as f:\n",
    "    game_history = pickle.load(f)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [],
   "source": [
    "game = Game(4)\n",
    "game_history = game_history[1015:1025]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [],
   "source": [
    "def action_label(a):\n",
    "    a_dict = {-1: '-', 0: '^', 1: 'V', 2: '<', 3: '>'}\n",
    "    return a_dict[a]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAALkAAAC0CAYAAADIM42CAAAABHNCSVQICAgIfAhkiAAAAAlwSFlzAAALEgAACxIB0t1+/AAAADh0RVh0U29mdHdhcmUAbWF0cGxvdGxpYiB2ZXJzaW9uMy4xLjEsIGh0dHA6Ly9tYXRwbG90bGliLm9yZy8QZhcZAAAUHElEQVR4nO3deVxVdf7H8de5Fy77joKsiiiakhhoGuqouVaQOZWZlqVpjY5L26hlto4zjdPir2maaiytX5k15bSMmkuWaEpBiPuYUpqCigqyw13O/HEnlwwMgXMv3/k8/9Jzwe/ncR5vjt97uee+NV3XEUJlJlcPIERLk5AL5UnIhfIk5EJ5EnKhPI/GfHFYaKgeFxvdUrM02rbtO0/out7m5x4LC/TVYyOCjR6pQfn7i+qdN9zXW48L9jd6pHrlFZ2sd1aAMC9PPdbfy8iRGpR/qrLeeRsV8rjYaD5b+UHzTNUMQmM6H6zvsdiIYNY/e7eR41xUeOZj9c4bF+zPpknXGTlOg/yeWlrvrACx/l6sGd7dqHEuKmJZdr3zynZFKE9CLpQnIRfKk5AL5UnIhfIk5EJ5EnKhPAm5UJ6EXChPQi6UJyEXypOQC+U16g1aTXG4sIipM3/HseJiTCYTE24dwz13TTBq+Uvy0odb+P8136Bp0DU+ghdmXo+3xdPVY51RY7Mz7I1V1Noc2B0ORnVtz7xfpXDnio3kFZ3Ew2wiLSqcF67pi6fZ9dez03U27vuqgL2l1WgaPHdlAr3CAwD4654iHt92iN2jryDMq3nPsWEh9zCbeXL+HHokd6O8ooLBI0czcEA6XTonGjVCoxSdLOPVj7PZ/OI0fLw8mfT0u6zI2snYq3u6erQzvMwmVo4fjr/FE6vdwZClqxjWMZoxyQm8Nqo/AHes2MiSbfuYnNrFxdPCvNyDDGoXzOJ+namzO6i2OwA4UlnLF0dPE+NraZF1DfvxjoxoS4/kbgAE+PvTuVNHio4eM2r5S2JzOKips2Kz26mqtRIZGuDqkc6jaRr+//2fxepwYHU40DQYkRiDpmlomkZadDhHyqpcPCmUW21sKS5nXILzLd8Ws4kgi/MaOz/vIPNTYtG0llnbsCv5uQ79cJjtO3eT2rOHK5b/RdqFBTJt1FWkTHoOb4snA3t2ZFBP9/tfx+5wkL74EwpOlTMlrQu9os/eN2C1O1i2o4CFw3q7cEKngxW1hHl5MDO7gF0lVVwe6sdTqfFkHS0j0sdCtxC/Flvb8I1aRWUlE6ZMZ8FjDxEY4D53wvxUaUU1q7L3kvvqLHYuuZ+qmjre3ZDv6rEuYDaZ2Do5k30zbyK38AS7jpeceWzWqq2kx0WQHhfhwgmdbA6dHSWVTEiMYP3IZHw9TPx5xxGe332E2ckxLbq2oSG3Wq1MmDKdG2/IIOOa4UYu3WhfbCsgPiKE8CA/PD3MXNe3K1/v/cHVY9Ur2NtC//gI1h44AsCCjds4UVXD00N7uXgypyhfC1G+FlLDnRe2jNhQtpdUcqiilsGrd5D2UR6FVXUMXb2T49V1zbq2YSHXdZ0ZDzxE58SOTJsy0ahlL1lMmyBy/n2Yqto6dF1nY/53dI6t95ZHlyiurKG0xhmIaquNDd8VkRQexJK8fawrKGTJDQMwtdRGt5Ha+liI8vVif1k1AFnHyrg8xI/do1PJyexJTmZPonwtrB3RnbY+zfsE1LA9efbXuSx//0Mu65LEgGGZADwy+z6GXj3QqBEaJTUphoz0yxg862U8zCaSE9px+/BUV491nqMVVUz5aDN2Xceh6/y6a3tGdool8PdvEBfkx6AlKwG4PimeuQNc//xnQWo8U7ccoM7uIN7fm0V9EgxZ17CQ9+mdxqnD+4xarlnMuXUQc24d5Oox6pUcEcqWyRkXHC97+HYXTHNx3UP8Grz5OSezZV6edf1vCIRoYRJyoTwJuVCehFwoT0IulCchF8qTkAvlSciF8iTkQnkScqE8CblQnoRcKE9CLpSnNaaRWdO0YqDBmg2DxdfXE+OGs0LrmrfeWaF1zduokAvRGsl2RShPQi6UJyEXypOQC+VJyIXyGlc77uelx4b6ttQsjZZ/uFSd2vEQHz0u2n0+hi5vV3HDteMhFj0+ysfIkRqUt7useWrHY0N9WX/v1c0zVTMIv/99dWrHowPY+MHNRo7ToICkFxt8DTw+yocv3u5n1DgXFZiyUmrHxf8uCblQnoRcKE9CLpQnIRfKk5AL5UnIhfIk5EJ5EnKhPAm5UJ6EXCivxZsmZryTw5o9Rwn392LTg0PPHH81az9/33wAD5PG0K7teCwjuaVHaTR3bGT+zdz1rP78IG3CfPjqk7EAPPz0ZlZt+B6Lp5kOcYG89IerCQ70wmq1M23eBvJ3F2Oz6YwdlcQDdxtXCXP4aDV3z8vn2MlaTJrGHb+OZeq4Dix4aR9LP/iB8BBnN9D86UkM798WgJ37ypj51E7KK2yYTPD5W+l4e5mbNEeLh/yWXvFM6teRactyzhzL2n+cVbsK2fjAELw8zBSX17T0GI3mro3M40Z35e7xlzNl9rozxwanx/L4/X3x8DDxyMIveeblXJ588CpWrD5AXZ2d7I/HUlVtpde1y7jp2k7ExwQaMquHWeP393clpWsQ5ZU2BozdxOA+4QBMG9+BGRPO7wyy2RxMfjifV57qQXJSICdL6/D0aPpmo8W3K1d1bEPIT+qkl3xZwMzBSXh5OH9C2wR4t/QYl8QdG5n79YoiJMjrvGNX94vD479h6JUSSeHRCgA0DSqrbdhsDqpr7Hh6mgjwb5lq758T2cablK5BAAT4eZCU4E/h8fovaOu3nKBbpwCSk5w/hGHBFszmprfXuWRPfqC4gi0FJxm26DMyXvyCbw6dcsUYDTq3kbnbhGcI9PN2y0bmn3rz/T0MHRAPwKjhHfHz8SCx3+tcNmgpMyb2JDTYNReUg0eq2L63jLRk53v8X3nnIH1vymLqo9spKbMCsP9gJZoGo37zFf1v2cTzrx9olrVdEnKbQ+d0dR2fzhjE4xnJ3PVmNu720RitpZH5XAtfysHDrDEmszMAOduPYzZpfJt1BzvX38YLr23jux9OGz5XRZWN2x74hj8+eBmB/p7cdXM8+Z8MZPPyfkSGe/HwM3sAsNt1tuaVsHhBCp++3pePNxzj8+wTTV7fJSGPCvLh2uRoNE3jirhQTJrGycrmbeFtqtbWyPzWir2s+vx7Fv95KNp/C2rf+2QfQ/rH4+lppk2YL32uiCRvx3FD57JaHYy//xtuviaKzKsjAWgb5oXZrGEyaUwYHUvuzlIAoiK8SU8NJSzEgq+PmWH92pC/p6zJM7gk5CO7R5G133my9xeXU2dzEOZn3F7xl2gNjcw/WrvxIM+9+g3LX7oWX5+zr/7EtAvgi+zD6LpOZZWVr/OP0TkhxLC5dF1n2uM7SOrgz29vO/sk82jx2X35x58do2ui87nO1Ve1Yde35VRV27HZHGzOPUVSgn+T52jxV1cmv5nN5gMnOFVZS/ITK5k9vCvjerdnxvIc+i1ci6fZxF/Gpp25+rgLd21kvvO+NWR9dYSTJTUkDVjCQ9N78+wrudTWObj+zg8B6NUjkkVPDGTKuO78Zu5n9L5uGboO40d3oXuXcMNm3bqthHc+OUK3TgGk35wFOF8u/MfqQnb8uwxN04iL8mHRPGeBbUigJ9Nu68DAcZvRNBjWry0jBrRt8hyN+pi4lNgQ3c3u8czVdT3t5x5L6RSlu+E9nvXOe0X3trqb3eNZ76wAV3QL0t3sHs9655XfeArlSciF8iTkQnkScqE8CblQnoRcKE9CLpQnIRfKk5AL5UnIhfIk5EJ5EnKhPAm5UJ40MhurNc0rjcxCtBayXRHKk5AL5UnIhfIk5EJ5EnKhvMY1MocG67Ex7VpqlkbL37G3/oZjX289LrjpH2fQnPKKTtbfIO3lqcf6e/3cQy6Rf6qywUbm0CAPPTbSfT5GZPu+6mZqZI5px/qPlzTLUM0hvH2f+huOg/3ZNOk6I8e5KL+nltbfIO3vxZrh3Y0cp0ERy7IbfA08NtLCqpc7GTXORUUP2i6NzOJ/l4RcKE9CLpQnIRfKk5AL5UnIhfIk5EJ5EnKhPAm5UJ6EXChPQi6UJyEXymvxzqCfstvtDMm4k8jINix77Rmjl29Qjc3OsDdWUWtzYHc4GNW1PfN+lcKdKzaSV3QSD7OJtKhwXrimL55m97g+nK6zcd9XBewtrUbT4LkrE+gV7iya+uueIh7fdojdo68gzMs1delX3rIHf18zJpOzoXnVy534+PNSnl1yjG8P1fKvlxLpkeQLwMaccha8chSrTcfTQ2PePe3od0UrKMb6qZdfX06nxPaUV1QavfRFeZlNrBw/HH+LJ1a7gyFLVzGsYzRjkhN4bVR/AO5YsZEl2/YxObWLi6d1mpd7kEHtglncrzN1dgfVdgcARypr+eLoaWJ8Xf922PeeSyA06GzUunTw5tUn4pnz7JHzvi40yIMlC9oTGe7J3u9qGPe7AnLfu6zJ6xt6OSosOs7az75k/C2ZRi77i2mahr/FecWzOhxYHQ40DUYkxqBpGpqmkRYdzpGyKhdP6lRutbGluJxxCc63UVvMJoIszjDNzzvI/JRY3KxUD4BO8d4kxl3YDN29kw+R4c7zn9Tei5o6ndo6R5PXM/RK/vATz/Ho3N9S4YZX8R/ZHQ7SF39CwalypqR1oVf02ffhW+0Olu0oYOGw3i6c8KyDFbWEeXkwM7uAXSVVXB7qx1Op8WQdLSPSx0K3ED9Xj4imwdgHC9DQGJ8RyviMsF/0ff/aeJruiT54WZp+HTYs5J+u30R4WAgpyV3YtCXXqGUbzWwysXVyJqU1dYx9bwO7jpfQra2z4HXWqq2kx0WQHhfh4imdbA6dHSWVLEhtT2q4Pw/nfs+fdxxhS3EZ7w50j+3UP19IJDLckxMlNm55oIDEOC/69Gh4n/3v72pY8MpR3v5Th2aZwbDtylc521m9Loue6aOYMv0RNn2Zwz2zHjVq+UYL9rbQPz6CtQec+8YFG7dxoqqGp4f2cvFkZ0X5WojytZAa7gxNRmwo20sqOVRRy+DVO0j7KI/CqjqGrt7J8WrX1Lr/uP0ID/FgZP9Atu2tbvDrC4vrmDT/exbNiaV9dPPcDmhYyB+ZPZUdWz8mb/M/eeWFJ+l3VRp/e/5xo5b/RYorayitcYah2mpjw3dFJIUHsSRvH+sKCllywwBMbrTJbetjIcrXi/1lzuBkHSvj8hA/do9OJSezJzmZPYnytbB2RHfa+hj/BLSq2kFFlf3Mn7/IqSCpw4V78R+drrBz+5zvmXtXO3olN99Wy/BXV9zZ0Yoqpny0Gbuu49B1ft21PSM7xRL4+zeIC/Jj0JKVAFyfFM/cAT1cPK3TgtR4pm45QJ3dQby/N4v6JFz8mwxSXGJl0iPOWy/tdp1RQ4IZ1DuAVVmnmfd/hZw6beP2ud/TraM3by9M4PUVJ/i+sJbn3zzG828eA2DZwgTCQ5oWU5eEvF/fVPr1dX1P/U8lR4SyZXLGBcfLHr7dBdP8Mt1D/Bq8ATons6eB05wvPsqLdYs7X3B8ZP8gRvYPuuD4rNsimHVb8z/fcY/faAjRgiTkQnkScqE8CblQnoRcKE9CLpQnIRfKk5AL5UnIhfIk5EJ5EnKhPAm5UJ6EXChPaseN1ZrmldpxIVoL2a4I5UnIhfIk5EJ5EnKhPAm5UF4ja8dD9bjY6JaapdG2bd9Zf423m1WkQ8M16a3p3AKEBpn0mAizkSM1aMe3tuapHY+LjeazlR80z1TNIDSmc/013m5WkQ4XqUlvRecWICbCzL9eCDZqnIuKG3FCasfF/y4JuVCehFwoT0IulCchF8qTkAvlSciF8iTkQnkScqE8CblQnoRcKE9CLpRnWMgPFxaRedNtXDlwBH0HX8Pf/r7UqKWbxG63M+ia2xk78X5Xj9Igdzm/DzxbTs8xJxlyd8mZY6XlDm6de5oBE09x69zTlJY7C2hXfFbDsHtKGHZPCTfcW8ruAtt5/5bdrjNyWgl3zD/dpJkMC7mH2cyT8+eQ/flq1nz0LouXvsXeffuNWv6S/ViT7u7c5fzeNNSbN546vw/oxeXVpKd4svG1UNJTPPnru862uthIM+8uDGLN30KYcasvcxZVnPd9r/2zhsTYptdaGRbyyIi29EjuBkCAvz+dO3Wk6Ogxo5a/JO5ek34udzm/VyZ7Ehxwfg3k2i113DjEWW144xBv1nzprJFMu8yT4ABnBHt28aDoxNmK8aJiO+u/ruOWEU3v8nTJnvzQD4fZvnM3qT3doyawPj/WpLtTd+cv4W7n90Spg4gwZ9QiwkycOO244GuWf1rDoDTPM39/7OVKHprkh6kZTr3hIa+orGTClOkseOwhAgMarp92pXNr0luT1nJ+z/Vlfh3LP61l7iRnQe267DrCg01c3ql5GjgN7fG0Wq1MmDKdG2/IIOOa4UYu3Wg/1qSv2/AltbV1lFdUcs+sR92uRfpc7np+w4NNHDvpvJofO+kgPOjstXVPgY3fPV/BG08GERLoPJ6zy8rarXVs+OoUtVad8iqdmU+Xs2h2wCWtb1jIdV1nxgMP0TmxI9OmTDRq2Uv2yOypPDJ7KgCbtuTy4qtvu3XA3fn8Du1j4R/rapg2xpd/rKthaF9nBfqR43amPFnG8w8GkBBz9n7RORP9mDPReVXfkl/Hy+9XX3LAwcDtSvbXuSx//0OyNm9lwLBMBgzLZO36z41aXnnucn5/+4cyRt1bSsFhO73Hn+Kd1TVMHeNDVp6VARNPkZVnZdoYHwAWvVVFSbnOvL9UMGJqCddOL22RmQy7kvfpncapw/uMWq5ZuWtN+rnc5fz+ZW7gzx5/548X1oz/6d4A/nRvw/9e3x4W+vawNGkm+Y2nUJ6EXChPQi6UJyEXypOQC+VJyIXyJORCeRJyoTwJuVCehFwoT0IulCchF8qTRmZjtaZ5pZFZiNZCtitCeRJyoTwJuVCehFwoT0IulCchF8qTkAvlSciF8iTkQnn/ASA0I7pc17p7AAAAAElFTkSuQmCC\n",
      "text/plain": [
       "<Figure size 216x216 with 16 Axes>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "cell_colors = {\n",
    "    0: '#FFFFFF',\n",
    "    2: '#EEE4DA',\n",
    "    4: '#ECE0C8',\n",
    "    8: '#ECB280',\n",
    "    16:'#EC8D53',\n",
    "    32:'#F57C5F',\n",
    "    64:'#E95937',\n",
    "    128:'#F3D96B',\n",
    "    256:'#F2D04A',\n",
    "    512:'#E5BF2E',\n",
    "    1024:'#E2B814',\n",
    "    2048:'#EBC502',\n",
    "    4096:'#00A2D8',\n",
    "    8192:'#9ED682',\n",
    "    16384:'#9ED682',\n",
    "    32768:'#9ED682',\n",
    "    65536:'#9ED682',\n",
    "    131072:'#9ED682',\n",
    "}\n",
    "\n",
    "board = game_history[0]['new_board']\n",
    "\n",
    "ncols = 4\n",
    "nrows = 4\n",
    "\n",
    "# create the plots\n",
    "fig = plt.figure(figsize=(3,3))\n",
    "# plt.suptitle('Best game')\n",
    "axes = [ fig.add_subplot(nrows, ncols, r * ncols + c) for r in range(0, nrows) for c in range(1, ncols+1) ]\n",
    "\n",
    "# add some data\n",
    "v = np.reshape(board, -1)\n",
    "for i, ax in enumerate(axes):\n",
    "    ax.text(0.5, 0.5, str(int(v[i])) if int(v[i]) > 0 else ' ', horizontalalignment='center', verticalalignment='center')\n",
    "    ax.set_facecolor(cell_colors[int(v[i])])\n",
    "\n",
    "# remove the x and y ticks\n",
    "for ax in axes:\n",
    "    ax.set_xticks([])\n",
    "    ax.set_yticks([])\n",
    "i = 0\n",
    "# if i < len(game_history) - 1:\n",
    "#     plt.suptitle('S: %d, Action: %s, Reward: %d' % (i, action_label(game_history[i+1]['action']), game_history[i+1]['reward']))\n",
    "# else:\n",
    "#     plt.suptitle('Final state')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [],
   "source": [
    "def update(i):\n",
    "#     plt.clf()\n",
    "    board = game_history[i]['new_board']\n",
    "    v = np.reshape(board, -1)\n",
    "    for i, ax in enumerate(axes):\n",
    "        ax.cla()\n",
    "        ax.text(0.5, 0.5, str(int(v[i])) if int(v[i]) > 0 else ' ', horizontalalignment='center', verticalalignment='center')\n",
    "        ax.set_facecolor(cell_colors[int(v[i])])\n",
    "        \n",
    "    for ax in axes:\n",
    "        ax.set_xticks([])\n",
    "        ax.set_yticks([])\n",
    "\n",
    "    return axes"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [],
   "source": [
    "anim = FuncAnimation(fig, update, frames=np.arange(1, len(game_history)), interval=1000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "MovieWriter imagemagick unavailable; trying to use <class 'matplotlib.animation.PillowWriter'> instead.\n"
     ]
    }
   ],
   "source": [
    "anim.save('./anim/%s_short.gif' % version, dpi=160, writer='imagemagick')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Moviepy - Building video ./anim/ohe_2_steps_move_penalty_log_reward_512_3_layers_random_full.mp4.\n",
      "Moviepy - Writing video ./anim/ohe_2_steps_move_penalty_log_reward_512_3_layers_random_full.mp4\n",
      "\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "                                                                                                                       \r"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Moviepy - Done !\n",
      "Moviepy - video ready ./anim/ohe_2_steps_move_penalty_log_reward_512_3_layers_random_full.mp4\n"
     ]
    }
   ],
   "source": [
    "clip = mp.VideoFileClip(\"./anim/%s_short.gif\" % version)\n",
    "clip.write_videofile(\"./anim/%s_full.mp4\" % version)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "1024\n"
     ]
    }
   ],
   "source": [
    "for i, g in enumerate(game_history):\n",
    "    if np.max(g['new_board']) == 2048.:\n",
    "        print(i)\n",
    "        break"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [],
   "source": [
    "game_history = game_history[1015:1025]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "drlnd",
   "language": "python",
   "name": "drlnd"
  },
  "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.9"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}