stick_vector: ensure we do not divide by zero

BossHobby · Sep 3, 2024 · db844c3 · db844c3
1 parent b2ff500
commit db844c3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/flight/input.c b/src/flight/input.c
@@ -23,9 +23,9 @@ vec3_t input_stick_vector(float rx_input[]) {
       .yaw = fastcos(roll) * fastcos(pitch),
   };
 
-  const float length = (stickvector.roll * stickvector.roll + stickvector.pitch * stickvector.pitch);
-  if (length > 0) {
-    const float mag = 1.0f / sqrtf(length / (1 - stickvector.yaw * stickvector.yaw));
+  if (stickvector.yaw < 1.0f) {
+    const float length = (stickvector.roll * stickvector.roll + stickvector.pitch * stickvector.pitch);
+    const float mag = 1.0f / sqrtf(length / (1.0f - stickvector.yaw * stickvector.yaw));
     stickvector.roll *= mag;
     stickvector.pitch *= mag;
   } else {