Add launch function

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "rlviser-py"
-version = "0.6.8"
+version = "0.6.9"
 edition = "2021"
 description = "Python implementation that manages a UDP connection to RLViser"
 license = "MIT"
@@ -16,7 +16,7 @@ name = "rlviser_py"
 crate-type = ["cdylib"]
 
 [dependencies]
-pyo3 = "0.21.0"
+pyo3 = { version = "0.22.1", features = ["py-clone"] }
 
 [profile.release]
 lto = true

pytest.py

@@ -3,15 +3,7 @@
 import rlviser_py as vis
 import RocketSim as rs
 
-if __name__ == "__main__":
-    game_mode = rs.GameMode.HOOPS
-
-    # Create example arena
-    arena = rs.Arena(game_mode)
-
-    # Set boost pad locations
-    vis.set_boost_pad_locations([pad.get_pos().as_tuple() for pad in arena.get_boost_pads()])
-
+def run(arena, game_mode):
     # Setup example arena
     car = arena.add_car(rs.Team.BLUE)
     car.set_state(rs.CarState(pos=rs.Vec(z=17), vel=rs.Vec(x=50), boost=100))
@@ -40,6 +32,24 @@
         time.sleep(max(0, start_time + steps / arena.tick_rate - time.time()))
         steps += 1
 
+if __name__ == "__main__":
+    game_mode = rs.GameMode.HOOPS
+
+    # Create example arena
+    arena = rs.Arena(game_mode)
+
+    # Set boost pad locations
+    vis.set_boost_pad_locations([pad.get_pos().as_tuple() for pad in arena.get_boost_pads()])
+    run(arena, game_mode)
+
     # Tell RLViser to exit
+    print("Closing...")
+    vis.quit()
+    time.sleep(1)
+
+    print("Re-opening then running...")
+    vis.launch()
+    run(arena, game_mode)
+
     print("Exiting...")
     vis.quit()

rlviser_py.pyi

@@ -69,5 +69,10 @@ The items are (car_id, team, car_config, car_state) respectively
 def render(tick_count: int, tick_rate: float, game_mode: GameMode, boost_pad_states: Sequence[bool], ball: BallState, cars: Sequence[CarData]) -> ...:
     pass
 
+def launch():
+    """
+    Opens RLViser. Useful if `quit()` was called and you want to open it again.
+    """
+
 def quit() -> ...:
     pass

src/lib.rs

@@ -34,6 +34,7 @@ pynamedmodule! {
         report_game_speed,
         report_game_paused,
         render,
+        launch,
         quit
     ],
     vars: [
@@ -135,6 +136,11 @@ fn report_game_paused(paused: bool) {
 
 type Car = (u32, u8, CarConfig, CarState);
 
+#[pyfunction]
+fn launch() {
+    socket::launch().unwrap();
+}
+
 #[pyfunction]
 fn render(tick_count: u64, tick_rate: f32, game_mode: u8, boost_pad_states: Vec<bool>, ball: BallState, cars: Vec<Car>) {
     let game_state = GameState {

src/socket.rs

@@ -195,6 +195,14 @@ pub fn report_game_paused(paused: bool) -> io::Result<()> {
     Ok(())
 }
 
+pub fn launch() -> io::Result<()> {
+    if let Err(e) = Command::new(RLVISER_PATH).spawn() {
+        println!("Failed to launch RLViser ({RLVISER_PATH}): {e}");
+    }
+
+    Ok(())
+}
+
 pub fn quit() -> io::Result<()> {
     if let Some(socket_handler) = SOCKET.get() {
         socket_handler.send_quit()?;