Lesson6.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "Lesson6.ipynb",
      "provenance": [],
      "toc_visible": true,
      "authorship_tag": "ABX9TyOYkFSoHXpuWw8y1pkD1xOx",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/WarwickAI/natural-selection-sim/blob/main/Lesson6.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "E4LECHrVUrMc"
      },
      "source": [
        "<p align=\"center\">\r\n",
        "    <img src=\"data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAARgAAAEYCAYAAACHjumMAAAVd0lEQVR4nOzdf5AkZ13H8b4jyA+BOcBeCm6O2cNCZA6zFbDKUdBeLCiVEqYoZoIIZC7+A7MntZmARn7ZF8QfVDlunTVX+AOdOxBkOsgeVSogQguJc4UWtRfpM+YImb1JhHR+2DkTiT9IrTXsXnkuu3vzfJ/+zjNs3q+q71XdH9PP091Pf7ZnuvvpK9bW1jwA0LDXdQcA7F4EDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1BAwANQQMADUEDAA1UxUw8/Pzn9qzZ8+apK6//vp3uO7/dk6fPn2l6frMzMzc5rrfO5mZmUml++r06dNlk7buuuuuZ0rb2hgbbdP1i6KoJm3vwIEDXzFtb9daW1ubmup2u68bdUlYWZZlBdfrsFXV6/UTknWKouglrvu+VYVheIN0PwVBEJu2t7CwsGQxLkZ1YTAYzJi2Wy6XE2mby8vLr3S9n6ahnHdgcxWLxfPSndrpdJqu+7+50jT1petTqVQ+77r/W6zP6Kz3bot99KZJbT/bsRFF0auk7dVqtZ7rfTUN5bwDW+zUunSnlkqlW133f3MtLS29zebAWFlZeb7rdbi0ut3uNdJ18X0/zbLsCpP2wjAM8wiYUql0u2R9S6XSbcI2H0mS5Hmu95frct6Brcr3/TukA6nf71dc9//SKpfLt9ocGM1m872u12HT+twiXZd2u70oGAuDPALGW//K+RrT9kdnPtL2Wq1W2/X+cl3OO7BVhWH4dulOHX1vdt3/i5UkSTmHv7x3ul6PizUYDEoW6zEwbW95efln8goX6dgYDAajr2gXpG2Otpnr/eaynHdgq8qyrOB53v3Sndrtdl/veh1GFQRBnMeBMTrQXK/LqBqNxp9I1yEMQ+PfQIIg+GyeAbMxNhqm/bD5mtZoND7oer+5LOcd2K5arZb4LCYIgk+77n8eZy8Xq1arnXC9PsPhcI/neY9K1yFNU9+kvX6/P5d3uHjrZzE3S9bf9/1U2OYjrvedy3Lege0qTdNnFAqF+6QDaXSAu+x/Xj9OjqpQKGSj7eFyfRYWFt4n7X+1Wj1p2l69Xo80AsYTfm1ZXFz8gLQ9yVnTbinnHdipms1mR7pTXV/i9X1fevVhy2q32y1X62J7qbjX6/2USXt5nv1tVc1m85jpNkiS5KC0vXK5/BXXx5Kr2jP6Z1qdOXPm4FVXXXWn9PNxHP/g/Py8+PNSN91008uvvvrqL+S5TN/3v3Hvvffuz3OZ4zp69OjCjTfeeFzy2XK5fObs2bNXmXzm8OHDx0+ePLkgaW8cozPCc+fOPXNmZsZo8B8+fLi7uro6K2nz+PHjrz906NC9ks9+T3OdcJerSqXycYu/VB1Hfe5r/OXt9Xo1F+tjc0er5LYBz/Me1jyD8dYvIb/D9dh+LJTzDlyukiR5vnQQjf5SZVn2tEn2N47jH9I6KBqNRnfS2z+KIvHjG6VS6Uum7eX529VOtXHT31Q+WrKbynkHxqnRgSUdSGEYXjfhvop/NxqnTK/G2FSWZVdYXD1Z63a7v2jS3mAwePYkzl4uGRvvdD22d3s578A41ev1ftriL9W/TLKvNgfkODXJu0M7nc4bpf0sFotDQXviu2aFY+Os67G928t5B8atUqkkvmU8iqL6JPrY7XZfM4ED475Jndrb3CjY6XR+aZL72KKfU3FT5m4t5x0Yt2z+uhWLxYk8BFmr1U5O6KBQD8x+vy++VOz7vvFjAe12+82TDhdPeKZFjV/OOzBubTw+8Ih0IGnfbp+m6Q94nvfQJA6KIAj+Snt7B0Hw19L+hWEYmrZn84Brt9u9xuZKV7fb/TnX43u3lvMOmFSr1bpROoi05+dotVq/Mcm/vP1+/7la62J7o5vpD9G9Xq9m0d7daZrubbfbr53WsfFYLucdMKnRwC0UCpl0IGk9PpBl2RN93//mJANmYWHh/Vrb2eaq3cLCwu+ZthcEwSlpe2EY3nBxOTa/4SRJ8kLX43s31lTfybuVw4cPf+DkyZO/KvlsuVz+/NmzZ1+Rd59OnDjRuPbaa0+Yfq5arX7s1KlTb9y3b9+/Xbhw4emmnx8MBs+ZnZ39punndrK6uvqsgwcP3ul53pNNP1soFB48d+7cc2dmZh4yaK908ODBVeOOet4oUFZXV1cPXvy/dD9462Pj78+ePfsy088dPXo0lLTnbdyxPDs7e7/0898TXCecadmevmucxQRB8LeSvvR6veqaxTNXYRgaT+B0ubL5MV1y57TN82ZhGP7y5uV5nvef0uXFcfxi0/5XKpW/kbY3jVO85l3OOyApmydtG43GH+fZlyRJRHfu+r5/T5ZlT1pbv/s3kCyjVCr9Y97btlQq3W5xgBqFd5qmj7f4yvvoYDDYt3mZNhOSS+6UtvmDVyqVbnN9LGmX8w5IajgcFqU71fO8bw+Hwz159aXVarUl/Wi1Wu+6dDmjwJEsJ0mSF+S1LlEUie/jkcwW12q13iNtr1arfXS75Vrc7PhwmqZPNV2PIAg+LV2POI5/zPXxpFnOOyCtIAg+Id2pCwsL78ujDxuXziV9+Faapv/vGanFxUXRqzkWFxd/M69tanmp12jOk9G2s7nreXTWt92yG43GsUmtx9r6DZZvkLbn4vmySZbzDkjL8tJmLs/0SH+vqNVqf755WdJTbd/3vz4F2/NBwbZ7k7S9YrG44+MfNl9bpHM624Tl6Izc9fGkVc47YFM2kzrl8aCb9Fb6brf76q2WJ30DQRRFr81hXT5lsS2vN21vbm7utLS9KIpeNcby/05z+Zur0WiI79Gq1+u/7/pY0irnHbCpKIpEP45661M5rGZZtlfadhzHhyTt+r5/xw7LFK2P7ZsULH+oNH4swGYKiHHfDLm8vPzz0jZ83z9vuk6DweCA53n/JW1zt759wHkHbMvmvUM2EzhVKpWbJG222+2Fy6yP6HeQOI5/wmJdxA81NptN4+kwisXicBL7rFgsfm2SY0P6g//GdtyV71By3gHb6na7DelOlT7TI30/UKFQuCdN0x3PmsIwFF1Zkf5YGMfxldLt53netweDgdEjC91u95XS9kZnFlmWff+4bdm8JTQIgs+absuNuYv/Rzg27nd9LGmU8w7YVpqmj7d5h1K/3/9h0zabzWZL0latVrvsPThJkrxQePClku1Xq9U+KN12krlparVaT9qeZOLzUShZjA3j571snqhvtVpvd3085V3OO5BHNRqND0l3ar1ej0zbK5VKX5W0tby8/NJxlh8EwWcEy7+QJMmzTdbD8tmuh0yvxCVJsl+6nwqFwr9KxkYYhu+UthkEwSdM25P+jubt0mk8nXcgj5rk4wOTuJzc7/crpsuv1WofN91uNk+Aj858TNur1+t/IG2v2WyKHu60feWK5NGSYrH4ZWl7nU7nWtfHU57lvAN5VRAE4mdCGo3G8XHbaTQaoh/yRp8zWZ9SqfR1k+UvLy8bvXto4yZB8VWPOI5fZtKezXutPcurLPV6XfzSuEaj8Yem7dlc3Zybm8v98Q+X5bwDeVWSJM+zGMCPjHO6n6bpFYVCQTQtw8rKyqzJ+iwtLY19E98oXE231+Li4q9Lt5fka6XN11jJBFab9tszCoWC6He60VfIJElmTNucm5v7gnR9O53O61wfT3mV8w7kWTY/II7zg2W73X6bZNmVSuUvTdfF5NS+3+9fabp86bNPG+1dZbguey1+iL8wHA6fYjs2bJ7abrVav2vans2d0cVi0fg+nGkt5x3Is5aXl+elO3V0FjMYDPbvtHzf90UTGkVR9OOS9anX6x+73LLHvfHs0rK5VCy5fGtzK0Fez+qsrKyIboy8WIPB4Jkm7WVZ9hTf9++StjfaR66PpzzKeQfyLpubq8IwPLLdcnu93isky7S5y7bf77/ocsuXTABuc6YnuQGtXC7fIm0vjuMX5TU2KpXK5yzGhvEl8na7LbqdwdtF03g670DeFUXRq6U7dafb3qXTSC4uLv6KzfrMzc39w3bLLhQK95kuL0mSF0i3T6FQeCDLMqOpLuI4frHF/ljJc2xM+pGIURUKBfF9OIPB4IDr48m2nHdAo2ymHWi322/dckMJ79BMksRqcu6dnthuNpvGvw1UKhXxZF3NZnNJ0J74Pd2dTuctCmPjL6T96fV6xj++NpvNoxbb+6jrY8m2nHdAo8IwFP0Y663fr/JdN3SFYfhrkmUFQfBJ23UZDAbP8jzv0W3+whldul1ZWTko3S6S9uI4fqnFflC56azb7YqniQiCYNtJrnbYfwek7RUKhbtcH0u25bwDGmU7mdHS0tLhPJYVx/GhPNanWq1+128mkkf8bd4WUK/XjW+sC4Lgw9L2wjB8k9b4KBQKq9J+SW68q1arJyy2g9UletflvANaZfMDW7lc/urF5UgvN5bL5dN5rctWfTCdpGg4HD7Z8zzxK18E7T1H2pbv+/dojo0wDN8t7VulUvmMYP+Jp47wPO+BNE2f7Pp4kpbzDmiW9KY4b/3s4zvvyalWq5+UfH5paSm3W76zLPu+0Ve3i8uWXLod/SWUbgtJe6MzHml7YRi+V3NcDAYD8TNR3vpNk8YPyE5yOtJpKucd0Kxms/nb0p3abDY7NpOL5zEl56V16XNDcRz/qMlnN15rKz6gTCemTpJEfPaise222Z42c7cYv57F5vEB2wnFXJbzDmiWzfMvhUIhq9frors/6/V6rq9GWfu/kPgPyVevMAzfbjG4vyxoT3y2VK/XPzSJsZEkybM9z7sgHRuSECyVSuLJ0aIoUn23ulY574B2lctl8Vyz0tJ6b3StVvtIv9+vmH7O933xzYfC9u6WtjfJCbBrtdrHpf1stVrvMW2v0+m8VdpesVj8J9fHkqScd0C7bC5LSmpubu5LWuti+mKzNctZ3UqlkvHZS6fTeYu0vWq1emKSY2N5eVn8tcXzvIdNL6MPBoOne553n7TNnV7VMq3lvAOTKJsf2Exrmn6Qy7LsSb7vG037sGldrjFs72k2B9DKykoul/UNx8bnpf0Nw/Bdpu2NPiNtTzrFq8ty3oFJlM0PbKaVpmlub420Lcsneoem7dl8BahUKp9zsY36/f5LpH3e6qbMceqx9A6l7xwMjwWzs7O3nj9//krNNqrV6odPnTrV0GzDxPz8fPzFL35xXvLZTqdzzZEjRz5i8pnZ2dmbz58//zJhe0f2799/+5kzZ57ied6+jXqS53lPGOfz9Xr9pkOHDv2zpG3L7fT6I0eORCafue66695/7Nixd0vaG61nFEVXSz7rhOuEm1TZ/HUdt2xeg5J32TzY5/v+1ybZXh5VqVRusRgb4le/Ss70bC/jS+4mdlXOOzCpGgwG+2xOTcc4KEWz+muVdGpPT3h7eqPR+COXAeNZTKuZZdnjbMZGt9t9g2mb9Xpd/NBpo9Ewvg/HVTnvwCSr1Wr9ltbgnrZXTtgcMEmSPMt4IHnef7sOmGazeUy6vdrt9qK0XckkXP1+X3yP1rT9MdtxXLjuwCQrTdOnWrymY9uS3nilVWEY3mBxsBg/AW5zY13O++EBi7Gx1+YhSNM5l9fWHwZdlrbX7Xbf7HqcjVPOOzDpWlxcFM/PsV3lNa1jHuXobQFbTifhomyePpa+VXNjDBjfgRzH8U9K2yuVSmdcj7VxynkHJl2278nZqnq93rzr9bpYO01QdbmqVCp90/aazaZ47h2leiBN08c7GBv3D4fDJ5i2aXOPVhRFRq+qcVHOO+Ci6vW6eJ6SzTVtD6KVSiXRxOQbA9bobQG27WmVzc2O1Wr1o9J2JWdPYRiK/yBM29jbqpx3wEXZPCW9uSTvS9aqdrv9CxaD1fiUu91uiw8OzSqXyzdLt6F0cveLlabpM0zasz2j7vV6r3E97nYq5x1wVdVqVfyg28UqFAqZ6/W4tCY954jv+99wHSbb1crKykEX2zEMw+tM21tYWHi/tL0gCD7setztVM474Kp6vd7P2g5iybwgiutTtViXbwnaEz+GMIlqNpu/I92W7Xb75dJ2fd+/3bS94XD4OJt1nebHB5x3wGXZ/n4wTU+3BkEgmnnPW/+re4OgPfFDgpMo6WtGLhkb4odEoygyfldVo9H4U2l7GvMP5VXOO+CybN44OE0/sG28mnViB2Icxz/iOkDGKUlw5jE2fN+/I8uyJ5q0ZzM5mrf+lfD5rsfhVuW8Ay4rTdOnep73oPCv1FS9oFz69sSd3ma5Xdm862jC9e9WB4fnPSBtW/JcWqVS+TNpe9P0df3Sesw8TQ1g8va67gCA3YuAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaCGgAGghoABoIaAAaDmfwMAAP//3sSN9oP5CnYAAAAASUVORK5CYII=\" alt=\"WAI Logo\"/>\r\n",
        "</p>\r\n",
        "\r\n",
        "# Lesson 6 - The grand finale!\r\n",
        "Over the past few weeks, we learned about a number of concepts.\r\n",
        "In Lesson 5, we have also written an Entity class which will be used in\r\n",
        "the simulation to handle our entities.\r\n",
        "\r\n",
        "In this lesson, we will write some of the remaining functions so that we have a fully working simulation."
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "2YM52xhwUjdA"
      },
      "source": [
        "import sys\r\n",
        "import random\r\n",
        "import uuid\r\n",
        "\r\n",
        "from IPython.display import clear_output\r\n",
        "\r\n",
        "\r\n",
        "class Color:\r\n",
        "    RESET = '\\033[0m'\r\n",
        "    RED = '\\033[91m'\r\n",
        "    GREEN = '\\033[92m'\r\n",
        "    BLUE = '\\033[94m'\r\n",
        "    CYAN = '\\033[36m'\r\n",
        "    YELLOW = '\\033[33m'\r\n",
        "    WHITE = '\\033[37m'\r\n",
        "\r\n",
        "\r\n",
        "class Simulation:\r\n",
        "    max_x = 20\r\n",
        "    max_y = 20\r\n",
        "\r\n",
        "    def __init__(self, max_x, max_y, seed):\r\n",
        "        self.seed = seed\r\n",
        "        self.max_x = max_x\r\n",
        "        self.max_y = max_y\r\n",
        "\r\n",
        "        random.seed(seed)\r\n",
        "\r\n",
        "    def do_cycles(self, num_moves, display=True):\r\n",
        "        '''\r\n",
        "        Exercise 1\r\n",
        "        ==========================================\r\n",
        "        Complete this method so that it \"ticks\"\r\n",
        "        each entity if it is active using the\r\n",
        "        Entity.tick() method.\r\n",
        "        Additionally, if `display` is true,\r\n",
        "        then you should also clear the output\r\n",
        "        using the `clear_output` function,\r\n",
        "        print the map of the simulation\r\n",
        "        using the `display` function defined in this\r\n",
        "        class, and ask for user input using `input()`.\r\n",
        "\r\n",
        "        This method should repeat the process repeated\r\n",
        "        above `num_moves` times.\r\n",
        "        '''\r\n",
        "        pass\r\n",
        "\r\n",
        "    def display_cells_info(self):\r\n",
        "        total_sense_distance = 0\r\n",
        "        num_alive_cells = 0\r\n",
        "\r\n",
        "        for entity in Entity.entities:\r\n",
        "            if type(entity) is Cell:\r\n",
        "                if entity.is_active:\r\n",
        "                    '''\r\n",
        "                    Exercise 2\r\n",
        "                    ==========================================\r\n",
        "                    Complete this section so that the information of a cell\r\n",
        "                    is printed, the number of alive cells is incremented\r\n",
        "                    and the total sense distance is updated.\r\n",
        "                    '''\r\n",
        "\r\n",
        "        avg_sense_distance = total_sense_distance / num_alive_cells\r\n",
        "\r\n",
        "        print()\r\n",
        "        print(f\"{Color.CYAN}Displaying Cell information:{Color.RESET}\")\r\n",
        "        print()\r\n",
        "\r\n",
        "        print(\"There are currently \" + Color.CYAN +\r\n",
        "              str(num_alive_cells) + Color.RESET + \" living Cells\")\r\n",
        "        print(\"The average sense distance is \" + Color.CYAN +\r\n",
        "              str(avg_sense_distance) + Color.RESET)\r\n",
        "\r\n",
        "    def build_entity_location_dict(self):\r\n",
        "        entities = {}\r\n",
        "\r\n",
        "        for entity in Entity.entities:\r\n",
        "            if entity.is_active:\r\n",
        "                entities[entity.get_location()] = entity\r\n",
        "        \r\n",
        "        return entities\r\n",
        "\r\n",
        "    def display(self):\r\n",
        "        entity_location = self.build_entity_location_dict()\r\n",
        "        map_text = \"\"\r\n",
        "\r\n",
        "        '''\r\n",
        "        Exercise 3\r\n",
        "        ==========================================\r\n",
        "        Complete this function so that it iterates over all\r\n",
        "        valid coordinates, checks if an entity exists\r\n",
        "        in each coordinate and adds the entity symbol.\r\n",
        "        If no entity exists, add the string \" _\".\r\n",
        "        '''\r\n",
        "\r\n",
        "        for y in range(1, self.max_y + 1):\r\n",
        "            map_text += \"\\n\"\r\n",
        "            for x in range(1, self.max_x + 1):\r\n",
        "                # Complete the code here\r\n",
        "                pass\r\n",
        "\r\n",
        "        print(map_text + Color.RESET + \"\\n\")\r\n",
        "\r\n",
        "\r\n",
        "class Entity:\r\n",
        "    entities = []\r\n",
        "\r\n",
        "    def __init__(self, x, y):\r\n",
        "        self.entities.append(self)\r\n",
        "\r\n",
        "        self.id = uuid.uuid4()\r\n",
        "        self.is_active = True\r\n",
        "        self.x = x\r\n",
        "        self.y = y\r\n",
        "\r\n",
        "    def die(self):\r\n",
        "        self.is_active = False\r\n",
        "\r\n",
        "    def get_location(self):\r\n",
        "        return (self.x, self.y)\r\n",
        "\r\n",
        "    def translate_by(self, dx, dy):\r\n",
        "        self.x += dx\r\n",
        "        self.y += dy\r\n",
        "\r\n",
        "    def manh_distance_to(self, entity):\r\n",
        "        target_x, target_y = entity.get_location()\r\n",
        "\r\n",
        "        return abs(self.x - target_x) + abs(self.y - target_y)\r\n",
        "\r\n",
        "    def get_closest_entity(self, of_type, max_distance=200):\r\n",
        "        min_entity = None\r\n",
        "        min_distance = 9999999\r\n",
        "\r\n",
        "        for entity in self.entities:\r\n",
        "            if type(entity) is of_type and entity.is_active:\r\n",
        "                distance = self.manh_distance_to(entity)\r\n",
        "\r\n",
        "                if distance < min_distance and distance <= max_distance:\r\n",
        "                    min_distance = distance\r\n",
        "                    min_entity = entity\r\n",
        "        \r\n",
        "        return min_entity, min_distance        \r\n",
        "\r\n",
        "    def move_close_to(self, entity):\r\n",
        "        target_x, target_y = entity.get_location()\r\n",
        "\r\n",
        "        if target_x > self.x:\r\n",
        "            self.translate_by(1, 0)\r\n",
        "        elif target_x < self.x:\r\n",
        "            self.translate_by(-1, 0)\r\n",
        "        elif target_y > self.y:\r\n",
        "            self.translate_by(0, 1)\r\n",
        "        else:\r\n",
        "            self.translate_by(0, -1)\r\n",
        "\r\n",
        "    def tick(self):\r\n",
        "        pass\r\n",
        "\r\n",
        "    def get_symbol(self):\r\n",
        "        pass\r\n",
        "\r\n",
        "\r\n",
        "class Food(Entity):\r\n",
        "    def __init__(self, x, y):\r\n",
        "        super().__init__(x, y)\r\n",
        "\r\n",
        "        self.energy = 10\r\n",
        "\r\n",
        "    def get_energy(self):\r\n",
        "        return self.energy\r\n",
        "\r\n",
        "    def get_symbol(self):\r\n",
        "        return Color.GREEN + \" o\"\r\n",
        "\r\n",
        "\r\n",
        "class Cell(Entity):\r\n",
        "    def __init__(self, generation, x, y, energy, sense_distance):\r\n",
        "        super().__init__(x, y)\r\n",
        "\r\n",
        "        self.generation = generation\r\n",
        "        self.energy = energy\r\n",
        "        self.sense_distance = sense_distance\r\n",
        "        self.total_moves = 0\r\n",
        "\r\n",
        "        '''\r\n",
        "        Exercise 4\r\n",
        "        ==========================================\r\n",
        "        Modify this method so that the cell mutates\r\n",
        "        with 50% chance.\r\n",
        "        '''\r\n",
        "\r\n",
        "    def get_symbol(self):\r\n",
        "        return Color.RED + \" @\"\r\n",
        "\r\n",
        "    def display_info(self):\r\n",
        "        info = (\"Cell ID: \" + str(self.id))\r\n",
        "        info += (\" | Energy: \" + str(self.energy))\r\n",
        "        info += (\" | XY: \" + \"(\" + str(self.x) + \", \" + str(self.y) + \")\")\r\n",
        "        info += (\" | Sense Distance: \" + str(self.sense_distance))\r\n",
        "        info += (\" | Moves: \" + str(self.total_moves))\r\n",
        "        info += (\" | Status: \" + \"Alive\" if self.is_active else \"Dead\")\r\n",
        "\r\n",
        "        return info\r\n",
        "\r\n",
        "    def move_random(self):\r\n",
        "        dx, dy = random.choice([(0, 1), (0, -1), (1, 0), (-1, 0)])\r\n",
        "\r\n",
        "        while not (0 < self.x + dx <= Simulation.max_x and 0 < self.y + dy <= Simulation.max_y):\r\n",
        "            dx, dy = random.choice([(0, 1), (0, -1), (1, 0), (-1, 0)])\r\n",
        "\r\n",
        "        self.translate_by(dx, dy)\r\n",
        "\r\n",
        "    def tick(self, directed=False):\r\n",
        "        nearest_food, distance = self.get_closest_entity(Food, max_distance=self.sense_distance)\r\n",
        "\r\n",
        "        '''\r\n",
        "        Exercise 5\r\n",
        "        ==========================================\r\n",
        "        This method implements the behaviour of the cell.\r\n",
        "\r\n",
        "        Complete it so that:\r\n",
        "\r\n",
        "        1.) If no piece of food is found nearby `nearest_food`,\r\n",
        "            the Cell moves at random using `move_random`.\r\n",
        "            Otherwise, move closer to the piece of food,\r\n",
        "            and if the distance to the piece of food is 1,\r\n",
        "            the cell should also eat it.\r\n",
        "        2.) After that, the total number of moves of the cell should be increased,\r\n",
        "            and its energy reduced by 1.\r\n",
        "        3.) If the energy of the cell is 0 or less, then the cell should die.\r\n",
        "            Otherwise, if it's greater than or equal to 100, it should reproduce.\r\n",
        "        4.) Finally, if the total moves of a cell is greater than 100, the cell should also die.\r\n",
        "        '''\r\n",
        "\r\n",
        "    def eat(self, food):\r\n",
        "        self.energy += food.get_energy()\r\n",
        "        food.die()\r\n",
        "\r\n",
        "    def mutate(self):\r\n",
        "        change = random.choice([-1, 1])\r\n",
        "        if self.sense_distance + change >= 1:\r\n",
        "            self.sense_distance += change\r\n",
        "\r\n",
        "    def reproduce(self):\r\n",
        "        if self.y < Simulation.max_y:\r\n",
        "            spawn_y = self.y + 1\r\n",
        "            spawn_x = self.x\r\n",
        "        elif self.x < Simulation.max_x:\r\n",
        "            spawn_x = self.x + 1\r\n",
        "            spawn_y = self.y\r\n",
        "        else:\r\n",
        "            spawn_x = self.x - 1\r\n",
        "            spawn_y = self.y - 1\r\n",
        "\r\n",
        "        '''\r\n",
        "        Exercise 6\r\n",
        "        ==========================================\r\n",
        "        This method is responsible for the cell reproduction.\r\n",
        "        \r\n",
        "        Complete it so that a new Cell is spawned,\r\n",
        "        with its generation increased by 1, coordinates\r\n",
        "        spawn_x and spawn_y, 50 units of energy,\r\n",
        "        and the same sense distance as the current cell.\r\n",
        "\r\n",
        "        Additionally, 50 units of energy should be removed from the current cell.\r\n",
        "        '''\r\n",
        "\r\n",
        "\r\n",
        "def add_food(num_food, verbose=False):\r\n",
        "    for i in range(1, num_food + 1):\r\n",
        "        x, y = random.randint(1, Simulation.max_x), random.randint(1, Simulation.max_y)\r\n",
        "        Food(x, y)\r\n",
        "\r\n",
        "    if verbose:\r\n",
        "        print(\"\\nAdding \" + str(num_food) + \" food\")\r\n",
        "\r\n",
        "def create_cells(num_cells, verbose=False):\r\n",
        "    for i in range(1, num_cells + 1):\r\n",
        "        x, y = random.randint(1, Simulation.max_x), random.randint(1, Simulation.max_y)\r\n",
        "        Cell(0, x, y, 30, 3)\r\n",
        "\r\n",
        "    if verbose:\r\n",
        "        print(\"Adding \" + str(num_cells) + \" cells\")\r\n",
        "\r\n",
        "\r\n",
        "if __name__ == '__main__':\r\n",
        "    simulation = Simulation(20, 20, seed=\"WAI\")\r\n",
        "    add_food(100)\r\n",
        "    create_cells(5)\r\n",
        "\r\n",
        "    # Begin menu display\r\n",
        "    exit = False\r\n",
        "    while not exit:\r\n",
        "        print(\"\\n\" + Color.BLUE + \"▄\" * 50 + \"\\n\" + Color.RESET)\r\n",
        "\r\n",
        "        print(f\"{Color.CYAN}Cell Natural Selection Simulation{Color.RESET}\\n\")\r\n",
        "\r\n",
        "        print(\"1) Add a Cell\")\r\n",
        "        print(\"2) Add Food\")\r\n",
        "        print(\"3) See Cell Info\")\r\n",
        "        print(\"4) View Map\")\r\n",
        "        print(\"x) Begin Evolution Cycle\")\r\n",
        "        print(\"\\n\" + Color.BLUE + \"▀\" * 50 + \"\\n\" + Color.RESET)\r\n",
        "\r\n",
        "        val = input(\"Press a key: \")\r\n",
        "        if val == \"1\":\r\n",
        "            create_cells(1, verbose=True)\r\n",
        "        elif val == \"2\":\r\n",
        "            add_food(10, verbose=True)\r\n",
        "        elif val == \"3\":\r\n",
        "            simulation.display_cells_info()\r\n",
        "        elif val == \"4\":\r\n",
        "            simulation.display()\r\n",
        "        elif val == \"x\":\r\n",
        "            print(\"Beginning evolution cycle\")\r\n",
        "            simulation.do_cycles(100)\r\n",
        "        elif val == \"q\":\r\n",
        "            print(\"Exiting\")\r\n",
        "            exit = True\r\n",
        "\r\n",
        "        input()"
      ],
      "execution_count": null,
      "outputs": []
    }
  ]
}