Merge pull request #32 from ll7/31-create-a-method-to-record-the-simu…

…lation

31 create a method to record the simulation

.devcontainer/postCreate.sh

@@ -10,7 +10,7 @@ pip install -e . || { echo "Failed to install the current directory as a package
 pip install -e ./fast-pysf || { echo "Failed to install ./fast-pysf as a package"; exit 1; }
 
 # Set the display environment variable for GUI applications based on the host OS
-if [[ $HOST_OS == *"Windows"* ]]; then
+if [ $HOST_OS == *"Windows"* ]; then
     # We are in Windows
     echo "export DISPLAY=host.docker.internal:0.0" >> ~/.bashrc || { echo "Failed to set DISPLAY environment variable for Windows"; exit 1; }
 else

.gitignore

@@ -1,18 +1,19 @@
-__pycache__/
-.scannerwork
-.pytest_cache
 *.c
+*.egg-info
 *.o
 *.so
-profile.json
+*training*.zip
+.pytest_cache
+.scannerwork
+__pycache__/
 build
-*.egg-info
 file.log
-profiles
-model
 images
+logs
+model
+profile.json
+profiles
 pysf_tests
 pysocialforce
-logs
-*training*.zip
+recordings/*.pkl
 wandb

examples/demo_offensive.py

@@ -9,7 +9,7 @@ def training():
     env_config = EnvSettings(
         sim_config=SimulationSettings(difficulty=0, ped_density_by_difficulty=[0.02]),
         robot_config=BicycleDriveSettings(radius=0.5, max_accel=3.0, allow_backwards=True))
-    env = RobotEnv(env_config, debug=True)
+    env = RobotEnv(env_config, debug=True, recording_enabled=True)
     model = PPO.load("./model/run_043", env=env)
 
     obs = env.reset()

recordings/README.md

@@ -0,0 +1,4 @@
+# Recordings
+
+The folder where recordings are stored by default.
+However, recordings are in the `.gitignore` file, so they are not uploaded to the repository.