remove save images to tmp-images in test-collision.ipynb

8455735a · hagrid67 · 4fea2b28 · 8455735a
Commit 8455735a authored 4 years ago by hagrid67
--- a/notebooks/test-collision.ipynb
+++ b/notebooks/test-collision.ipynb
@@ -126,8 +126,8 @@
    "    aImg = oRT.get_image()\n",
    "    pilImg = PIL.Image.fromarray(aImg)\n",
    "    oCan.put_image_data(aImg)\n",
-    "    with open(f\"tmp-images/img-{iStep:03d}.png\", \"wb\") as fImg:\n",
-    "        pilImg.save(fImg)\n",
+    "    #with open(f\"tmp-images/img-{iStep:03d}.png\", \"wb\") as fImg:\n",
+    "    #    pilImg.save(fImg)\n",
    "    \n",
    "    time.sleep(0.05)"
   ]
@@ -143,24 +143,13 @@
  },
  {
   "cell_type": "code",
-   "execution_count": 6,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "oEC = ju.EnvCanvas(env)\n",
-    "env.reset(regenerate_rail=False)\n",
-    "oEC.oRT.render_env()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "a485406ea061479db50fb90c43833bc2",
+       "model_id": "47677058c61e4c71bf23d56e80bc88d3",
       "version_major": 2,
       "version_minor": 0
      },
@@ -173,15 +162,20 @@
    }
   ],
   "source": [
+    "oEC = ju.EnvCanvas(env)\n",
+    "env.reset(regenerate_rail=False)\n",
    "oEC.show()"
   ]
  },
  {
   "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 41,
   "metadata": {},
   "outputs": [],
-   "source": []
+   "source": [
+    "oEC.step()\n",
+    "oEC.render()"
+   ]
  }
 ],
 "metadata": {

 %% Cell type:markdown id: tags:

 # Test Collisions
 A visual test of a "rear-shunt" collision, to ensure that the agent does not get marked as collided permananently.

 %% Cell type:code id: tags:

 ``` python
 %load_ext autoreload
 %autoreload 2
 from IPython.core import display
 display.display(display.HTML("<style>.container { width:95% !important; }</style>"))
 ```

 %% Output


 %% Cell type:code id: tags:

 ``` python
 import PIL
 from IPython import display
 from ipycanvas import canvas
 import time
 ```

 %% Cell type:code id: tags:

 ``` python
 from flatland.envs import malfunction_generators as malgen
 from flatland.envs.agent_utils import EnvAgent
 #from flatland.envs import sparse_rail_gen as spgen
 from flatland.envs import rail_generators as rail_gen
 from flatland.envs import agent_chains as ac
 from flatland.envs.rail_env import RailEnv, RailEnvActions
 from flatland.envs.persistence import RailEnvPersister
 from flatland.utils.rendertools import RenderTool
 from flatland.utils import env_edit_utils as eeu
 from flatland.utils import jupyter_utils as ju
 ```

 %% Cell type:code id: tags:

 ``` python
 nAg=2
 bUCF=False
 bReverseStart=False
 env, envModel = eeu.makeTestEnv(sName="merging_spurs", nAg=nAg, bUCF=bUCF)
 oRT = RenderTool(env, show_debug=True)
 oRT.render_env(show_rowcols=True,  show_inactive_agents=True, show_observations=False)
 oCan = canvas.Canvas(size=(600,300))
 oCan.put_image_data(oRT.get_image())
 print(f"UCF: {bUCF}")
 display.display(oCan)


 iPauseStep=10
 iPauseLen = 5

 for iStep in range(25):

    if bReverseStart:
        iAgentStart = max((nAg - 2 - iStep*2), 0)
    else:
        iAgentStart = 0
    dActions = { i:int(RailEnvActions.MOVE_FORWARD) for i in range(iAgentStart, len(env.agents)) }

    if (iStep >= iPauseStep) and (iStep < iPauseStep + iPauseLen):
        dActions[0] = RailEnvActions.STOP_MOVING

    #print(dActions)

    env.step(dActions)

    oRT.render_env(show_rowcols=True,  show_inactive_agents=True, show_observations=False)

    aImg = oRT.get_image()
    pilImg = PIL.Image.fromarray(aImg)
    oCan.put_image_data(aImg)
-    with open(f"tmp-images/img-{iStep:03d}.png", "wb") as fImg:
-        pilImg.save(fImg)
+    #with open(f"tmp-images/img-{iStep:03d}.png", "wb") as fImg:
+    #    pilImg.save(fImg)

    time.sleep(0.05)
 ```

 %% Output

    UCF: False


 %% Cell type:code id: tags:

 ``` python
 #!ffmpeg -i ./tmp-images/img-%03d.png -vcodec mpeg4 -filter:v "setpts=8.0*PTS" -y movie_nucf_stop.mp4
 ```

 %% Cell type:code id: tags:

 ``` python
 oEC = ju.EnvCanvas(env)
 env.reset(regenerate_rail=False)
-oEC.oRT.render_env()
-```
-
-%% Cell type:code id: tags:
-
-``` python
 oEC.show()
 ```

 %% Output


 %% Cell type:code id: tags:

 ``` python
+oEC.step()
+oEC.render()
 ```