Simulator: add reset hotkey (#29995)

add sim reset
This commit is contained in:
Justin Newberry
2023-09-20 17:28:41 -07:00
committed by GitHub
parent f69a45faba
commit 0857f4bb85
5 changed files with 16 additions and 2 deletions
+3
View File
@@ -110,6 +110,9 @@ class CarlaWorld(World):
def tick(self):
self.world.tick()
def reset(self):
pass
class CarlaBridge(SimulatorBridge):
TICKS_PER_FRAME = 5
+2
View File
@@ -127,6 +127,8 @@ class SimulatorBridge(ABC):
self.simulator_state.cruise_button = CruiseButtons.MAIN
elif m[0] == "ignition":
self.simulator_state.ignition = not self.simulator_state.ignition
elif m[0] == "reset":
self.world.reset()
elif m[0] == "quit":
break
+5 -2
View File
@@ -81,8 +81,11 @@ class MetaDriveWorld(World):
obs, _, terminated, _, info = self.env.step(self.vc)
if terminated:
self.env.reset()
self.reset_time = time.monotonic()
self.reset()
def reset(self):
self.env.reset()
self.reset_time = time.monotonic()
def close(self):
pass