Update rl_ball_sym & add Heatseeker

Cargo.toml

@@ -1,16 +1,14 @@
 [package]
 name = "rl_ball_sym_dll"
-version = "0.1.1"
+version = "0.1.2"
 edition = "2021"
 
 [lib]
 name = "rl_ball_sym"
 crate-type = ["cdylib", "staticlib"]
 
-[dependencies.rl_ball_sym]
-version = "4.0.0"
-default-features = false
-features = ["compression", "standard", "hoops", "dropshot", "throwback"]
+[dependencies]
+rl_ball_sym = { version = "4.1.0", features = ["compression"] }
 
 [profile.release]
 codegen-units=1

src/lib.rs

@@ -1,11 +1,15 @@
 use rl_ball_sym::{Ball, Game, Vec3A};
-use std::sync::RwLock;
+use std::sync::{
+    atomic::{AtomicBool, Ordering},
+    RwLock,
+};
 
 const TPS: usize = 120;
 const DT: f32 = 1.0 / TPS as f32;
 
 static GAME: RwLock<Option<Game>> = RwLock::new(None);
 static BALL: RwLock<Ball> = RwLock::new(Ball::const_default());
+static HEATSEEKER: AtomicBool = AtomicBool::new(false);
 
 fn set_game_and_ball((game, ball): (Game, Ball)) {
     let mut game_lock = GAME.write().unwrap();
@@ -15,24 +19,34 @@ fn set_game_and_ball((game, ball): (Game, Ball)) {
     *ball_lock = ball;
 }
 
+#[no_mangle]
+pub extern "C" fn load_heatseeker() {
+    set_game_and_ball(rl_ball_sym::load_standard_heatseeker());
+    HEATSEEKER.store(true, Ordering::Relaxed);
+}
+
 #[no_mangle]
 pub extern "C" fn load_standard() {
     set_game_and_ball(rl_ball_sym::load_standard());
+    HEATSEEKER.store(false, Ordering::Relaxed);
 }
 
 #[no_mangle]
 pub extern "C" fn load_dropshot() {
     set_game_and_ball(rl_ball_sym::load_dropshot());
+    HEATSEEKER.store(false, Ordering::Relaxed);
 }
 
 #[no_mangle]
 pub extern "C" fn load_hoops() {
     set_game_and_ball(rl_ball_sym::load_hoops());
+    HEATSEEKER.store(false, Ordering::Relaxed);
 }
 
 #[no_mangle]
 pub extern "C" fn load_standard_throwback() {
     set_game_and_ball(rl_ball_sym::load_standard_throwback());
+    HEATSEEKER.store(false, Ordering::Relaxed);
 }
 
 #[repr(C)]
@@ -82,6 +96,12 @@ impl From<Ball> for BallSlice {
     }
 }
 
+#[no_mangle]
+pub extern "C" fn set_heatseeker_target(blue_goal: u8) {
+    let mut ball = *BALL.write().unwrap();
+    ball.set_heatseeker_target(blue_goal == 1);
+}
+
 #[no_mangle]
 pub extern "C" fn step(current_ball: BallSlice) -> BallSlice {
     let game_lock = GAME.read().unwrap();
@@ -98,6 +118,10 @@ pub extern "C" fn step(current_ball: BallSlice) -> BallSlice {
         current_ball.angular_velocity.into(),
     );
 
-    ball.step(game, DT);
+    if HEATSEEKER.load(Ordering::Relaxed) {
+        ball.step_heatseeker(game, DT);
+    } else {
+        ball.step(game, DT);
+    }
     ball.into()
 }