altos: Make balloon code run again

This fixes the balloon code so that it works with recent altos bits

Signed-off-by: Keith Packard <keithp@keithp.com>

1 files changed, 10 insertions, 3 deletions

diff --git a/src/core/ao_balloon.c b/src/core/ao_balloon.c
index 12752d1f..904a9c08 100644
--- a/src/core/ao_balloon.c
+++ b/src/core/ao_balloon.c
@@ -31,6 +31,13 @@
 #error Please define HAS_USB
 #endif
 
+#if HAS_SENSOR_ERRORS
+/* Any sensor can set this to mark the flight computer as 'broken' */
+__xdata uint8_t			ao_sensor_errors;
+#endif
+
+__pdata uint16_t		ao_motor_number;	/* number of motors burned so far */
+
 /* Main flight thread. */
 
 __pdata enum ao_flight_state	ao_flight_state;	/* current flight state */
@@ -67,7 +74,8 @@ ao_flight(void)
 				/* Disable the USB controller in flight mode
 				 * to save power
 				 */
-				ao_usb_disable();
+				if (!ao_usb_running)
+					ao_usb_disable();
 #endif
 
 				/* Disable packet mode in pad state */
@@ -112,9 +120,8 @@ ao_flight(void)
 				ao_wakeup(DATA_TO_XDATA(&ao_flight_state));
 			}
 			break;
-		case ao_flight_drogue:
+		default:
 			break;
-			
 		}
 	}
 }