{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Example 1 - generate a rail from a manual specification\n",
    "From a map of tuples (cell_type, rotation)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "ename": "SystemError",
     "evalue": "Parent module '' not loaded, cannot perform relative import",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mSystemError\u001b[0m                               Traceback (most recent call last)",
      "\u001b[0;32m<ipython-input-2-b6a25a9cfbbb>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m\u001b[0m\n\u001b[0;32m----> 1\u001b[0;31m \u001b[0;32mfrom\u001b[0m \u001b[0;34m.\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mflatland\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0menvs\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mrail_generators\u001b[0m \u001b[0;32mimport\u001b[0m \u001b[0mrail_from_manual_specifications_generator\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m      2\u001b[0m \u001b[0;32mfrom\u001b[0m \u001b[0mflatland\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0menvs\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mobservations\u001b[0m \u001b[0;32mimport\u001b[0m \u001b[0mTreeObsForRailEnv\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m      3\u001b[0m \u001b[0;32mfrom\u001b[0m \u001b[0mflatland\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0menvs\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mrail_env\u001b[0m \u001b[0;32mimport\u001b[0m \u001b[0mRailEnv\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m      4\u001b[0m \u001b[0;32mfrom\u001b[0m \u001b[0mflatland\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mutils\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mrendertools\u001b[0m \u001b[0;32mimport\u001b[0m \u001b[0mRenderTool\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m      5\u001b[0m \u001b[0;32mfrom\u001b[0m \u001b[0mPIL\u001b[0m \u001b[0;32mimport\u001b[0m \u001b[0mImage\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;31mSystemError\u001b[0m: Parent module '' not loaded, cannot perform relative import"
     ]
    }
   ],
   "source": [
    "from flatland.envs.rail_generators import rail_from_manual_specifications_generator\n",
    "from flatland.envs.observations import TreeObsForRailEnv\n",
    "from flatland.envs.rail_env import RailEnv\n",
    "from flatland.utils.rendertools import RenderTool\n",
    "from PIL import Image"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "specs = [[(0, 0), (0, 0), (0, 0), (0, 0), (0, 0), (0, 0)],\n",
    "         [(0, 0), (0, 0), (0, 0), (0, 0), (7, 0), (0, 0)],\n",
    "         [(7, 270), (1, 90), (1, 90), (1, 90), (2, 90), (7, 90)],\n",
    "         [(0, 0), (0, 0), (0, 0), (0, 0), (0, 0), (0, 0)]]\n",
    "\n",
    "env = RailEnv(width=6,\n",
    "              height=4,\n",
    "              rail_generator=rail_from_manual_specifications_generator(specs),\n",
    "              number_of_agents=1,\n",
    "              obs_builder_object=TreeObsForRailEnv(max_depth=2))\n",
    "\n",
    "env.reset()\n",
    "\n",
    "env_renderer = RenderTool(env, gl=\"PILSVG\")\n",
    "env_renderer.render_env(show=False)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAs4AAAHgCAYAAAC1ouv3AAAMeUlEQVR4nO3dwXEbRxqA0cEWr6hSCpuC8/CG4BOdgBKgGYcYhZWHU/DJdxUmAOxhRS9BweRHEuQAPe9V6TAQWGrWT4gfuhrAZrfb7ScAAOBJ/1p6AQAAcAmEMwAABFcPL7bb7VLr4MTmeT64NttxmO24zHZcZjsusx3XsdnacQYAgEA4AwBAIJwBACAQzgAAEAhnAAAIhDMAAATCGQAAAuEMAACBcAYAgEA4AwBAIJwBACAQzgAAEAhnAAAIhDMAAATCGQAAAuEMAACBcAYAgEA4AwBAIJwBACAQzgAAEAhnAAAIhDMAAATCGQAAAuEMAACBcAYAgEA4AwBAIJwBACAQzgAAEAhnAAAIhDMAAATCGQAAAuEMAACBcAYAgEA4AwBAIJwBACAQzgAAEAhnAAAIhDMAAATCGQAAAuEMAACBcAYAgEA4AwBAIJwBACAQzgAAEAhnAAAIhDMAAATCGQAAAuEMAACBcAYAgEA4AwBAIJwBACAQzgAAEAhnAAAIhDMAAATCGQAAAuEMAACBcAYAgEA4AwBAIJwBACAQzgAAEAhnAAAIhDMAAATCGQAAAuEMAACBcAYAgEA4AwBAIJwBACAQzgAAEAhnAAAIhDMAAATCGQAAAuEMAACBcAYAgEA4AwBAIJwBACAQzgAAEAhnAAAIhDMAAATCGQAAAuEMAACBcAYAgEA4AwBAIJwBACAQzgAAEAhnAAAIhDMAAATCGQAAAuEMAACBcAYAgEA4AwBAIJwBACAQzgAAEAhnAAAIhDMAAATCGQAAAuEMAACBcAYAgEA4AwBAIJwBACAQzgAAEAhnAAAIhDMAAATCGQAAAuEMAACBcAYAgEA4AwBAIJwBACAQzgAAEAhnAAAIhDMAAATCGQAAAuEMAACBcAYAgEA4AwBAsNntdvulFwEAAOfOjjMAAATCGQAAAuEMAADB1cOL7Xa71Do4sXmeD67NdhxmOy6zHZfZjstsx3VstnacAQAgEM4AABAIZwAACIQzAAAEwhkAAALhDAAAgXAGAIBAOAMAQCCcAQAgEM4AABAIZwAACIQzAAAEwhkAAALhDAAAgXAGAIBAOAMAQCCcAQAgEM4AABAIZwAACIQzAAAEwhkAAALhDAAAgXAGAIBAOAMAQCCcAQAgEM4AABAIZwAACIQzAAAEwhkAAALhDAAAgXAGAIBAOAMAQCCcAQAgEM4AABAIZwAACIQzAAAEwhkAAALhDAAAgXAGAIBAOAMAQCCcAQAgEM4AABAIZwAACIQzAAAEwhkAAALhDAAAgXAGAIBAOAMAQCCcAQAgEM4AABAIZwAACIQzAAAEwhkAAALhDAAAgXAGAIDgaukFAABjuL29/eG2m5ubBVYC78OOMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAACCq4cX8zwvtQ7emdmOy2zHZbbjWtNs1/S9TtP6vt81mefZjjMAABTCGQA4ic+fPy+9BHhXwhkAAILNbrfb319st9tpmqbp293dYgvidT5dXx9cPz5jZbaXy2zHZbbjWutsf//69eD6Pz//vNBK3k+dLZfv2GztOAMAQPBDOI/27HctytzM9jKZ7bjMdlxmOy5zWzc7zgAAEFw9d4fHZ3k4H2991mu258tsx2W241r7bG9vb6fpp5/+vr65uVlwNadll5l7dpwBACAQzgAAEAhnAAAIhDMAcFIjnW+Gh4QzAPAmt7e3Sy8BPoRwBgCAQDgDAEAgnAEAIBDOAMCrPT7f7IWBjEw4AwBAIJwBACAQzgAAEAhnAOBVnG9mbYQzAAAEwhkAeDGfFsgaCWcAAAiEMwDwIsd2m51vZg2EMwAABMIZAMjsNrNmwhkASLwgkLUTzgDAq9ltZk2EMwDwLEc0QDgDAEAinAGAJ9lthv8RzgDAP/KCQPg/4QwAvIjdZtZKOAMARzmiAYeEMwAABMIZAPiB3Wb40dXSCwAAzosXBA5ks1l6Bc/b75deQWbHGQB4lt1mEM4AwAOOaAzkEnabp+ly1jkJZwDgO0c04GnOOAPAyj0VzHabB3JOZ4kvaJf5ITvOALBidpmhs+MMACtUgtluMxwSzgCwIs8Fs1iGf+aoBgCshGiGt7HjDACDE8xwGsIZAAYlmOG0HNUAgAGJZjg9O85wTu7f1/LLl+O3T9O0PfZ1T9z/rN63E3h3ghnej3AGgAEIZnh/jmoAwIUTzfAx7DgDwIUSzPCxhDMAXBCf+AfLEc4AcKZKJD8mmuH9CGcAOBOvCeV7ghnen3AGgAX8/vXrwfWff/2Vv1YkwzJWFc5veSZ/jv79xx8H18/9p/vS+1+Sm99+W3oJJ/X4F+ovb7j/aD/3l87jdlwvnu0b/i2P6491qY/Dx0+vTvVzs+Ynbt6ODgAAAuEMAACBcAYAgGBVZ5wZ1+0gZ5xHO6sNACPZ7Ha7/f3Fdrudvt3dHdzh0/X1hy+K5qlZzfN88HdmeyE2m2mapunbly8HN3/69dcnv+zJ++/3E+fD43ZcZjuul872bHz/nfK3c/p9cM5r++7YbB3VAACAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QznZL8//l6W97fv99O82x38ee7+AMBpCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAILhaegEAAHyAzWbpFVw8O84AABAIZwCAEV3Kp8deyjon4QwAAIkzzgAAo7qg3dxLYMcZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAILh67g7f7u4+Yh0swGzHZbbjMttxmS2cPzvOAAAQ/BDOn66vl1gHb1TmZraXyWzHZbbjMttxmdu62XEGAIBgs9vt9vcX2+12ybVwQvM8H1yb7TjMdlxmOy6zHZfZjuvYbO04AwBAcLDjDAAAHGfHGQAAAuEMAACBcAYAgODgkwO9EnQcXuU7LrMdl9mOy2zHZbbj8q4aAADwSsIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAAAC4QwAAIFwBgCAQDgDAEAgnAEAIBDOAAAQCGcAAAiEMwAABMIZAACCzW632y+9CAAAOHd2nAEAIBDOAAAQCGcAAAiuHl5st9ul1sGJzfN8cG224zDbcZntuMx2XGY7rmOz/S9WE9qvdTmG7wAAAABJRU5ErkJggg==\n",
      "text/plain": [
       "<PIL.Image.Image image mode=RGBA size=718x480 at 0x14DD8FD52E8>"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "Image.fromarray(env_renderer.gl.get_image())"
   ]
  }
 ],
 "metadata": {
  "hide_input": false,
  "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.5.2"
  },
  "latex_envs": {
   "LaTeX_envs_menu_present": true,
   "autoclose": false,
   "autocomplete": true,
   "bibliofile": "biblio.bib",
   "cite_by": "apalike",
   "current_citInitial": 1,
   "eqLabelWithNumbers": true,
   "eqNumInitial": 1,
   "hotkeys": {
    "equation": "Ctrl-E",
    "itemize": "Ctrl-I"
   },
   "labels_anchors": false,
   "latex_user_defs": false,
   "report_style_numbering": false,
   "user_envs_cfg": false
  },
  "toc": {
   "base_numbering": 1,
   "nav_menu": {},
   "number_sections": true,
   "sideBar": true,
   "skip_h1_title": false,
   "title_cell": "Table of Contents",
   "title_sidebar": "Contents",
   "toc_cell": false,
   "toc_position": {},
   "toc_section_display": true,
   "toc_window_display": false
  },
  "varInspector": {
   "cols": {
    "lenName": 16,
    "lenType": 16,
    "lenVar": 40
   },
   "kernels_config": {
    "python": {
     "delete_cmd_postfix": "",
     "delete_cmd_prefix": "del ",
     "library": "var_list.py",
     "varRefreshCmd": "print(var_dic_list())"
    },
    "r": {
     "delete_cmd_postfix": ") ",
     "delete_cmd_prefix": "rm(",
     "library": "var_list.r",
     "varRefreshCmd": "cat(var_dic_list()) "
    }
   },
   "types_to_exclude": [
    "module",
    "function",
    "builtin_function_or_method",
    "instance",
    "_Feature"
   ],
   "window_display": false
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}