update to allow mode selector to show current mode

bluerobotics · Nov 19, 2023 · 5eee77d · 5eee77d
1 parent b73b03a
commit 5eee77d
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/stores/mainVehicle.ts b/src/stores/mainVehicle.ts
@@ -254,6 +254,15 @@ export const useMainVehicleStore = defineStore('main-vehicle', () => {
     return (vehicle as ArduPilot) || undefined
   }
 
+  function getStringFromValue(map: Map<string, number>, value: number): string | undefined {
+    for (const [key, val] of map.entries()) {
+      if (val === value) {
+        return key;
+      }
+    }
+    return undefined; 
+  }
+
   VehicleFactory.onVehicles.once((vehicles: WeakRef<Vehicle.Abstract>[]) => {
     mainVehicle.value = getAutoPilot(vehicles)
     modes.value = mainVehicle.value.modesAvailable()
@@ -303,6 +312,9 @@ export const useMainVehicleStore = defineStore('main-vehicle', () => {
       firmwareType.value = heartbeat.autopilot.type
       vehicleType.value = heartbeat.mavtype.type
       lastHeartbeat.value = new Date()
+      if (modes.value !== undefined) {
+        mode.value = getStringFromValue(modes.value, heartbeat.custom_mode);
+      }
     })
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     getAutoPilot(vehicles).onMode.add((vehicleMode: any) => {