Add gps map function to tantalus profile

UBC-Rocket · Feb 3, 2025 · f1c827c · f1c827c
1 parent deb534a
commit f1c827c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -137,6 +137,9 @@ dmypy.json
 # PyCharm
 .idea/
 
+# Direwolf
+direwolf/
+
 logs/
 mapbox_cache/
 *.csv

diff --git a/profiles/rockets/tantalus.py b/profiles/rockets/tantalus.py
@@ -8,6 +8,7 @@
     update_state,
     update_aprs,
 )
+from ..mpl_funcs import receive_map
 from ..rocket_profile import RocketProfile, FlightPoint
 from connections.serial.serial_connection import SerialConnection
 from connections.debug.debug_connection import DebugConnection
@@ -47,7 +48,7 @@ def labels(self):
         return [
             Label(DeviceType.TANTALUS_STAGE_1_FLARE, "Altitude", update_altitude),
             Label(DeviceType.TANTALUS_STAGE_1_FLARE, "MaxAltitude", update_max_altitude, "Max Altitude"),
-            Label(DeviceType.TANTALUS_STAGE_1_FLARE, "GPS", update_gps),
+            Label(DeviceType.TANTALUS_STAGE_1_FLARE, "GPS", update_gps, map_fn=receive_map),
             Label(DeviceType.TANTALUS_STAGE_1_FLARE, "State", update_state),
             Label(DeviceType.TANTALUS_STAGE_1_FLARE, "Pressure", update_pressure),
             Label(DeviceType.TANTALUS_STAGE_1_FLARE, "Acceleration", update_acceleration),

diff --git a/start.py b/start.py
@@ -11,6 +11,7 @@
 from profiles.rockets.tantalus import TantalusProfile
 from util.self_test import SelfTest
 from util.detail import IS_PYINSTALLER, LOGGER
+from setup import get_launch_script_path
 
 if hasattr(QtCore.Qt, 'AA_EnableHighDpiScaling'):
     QtWidgets.QApplication.setAttribute(QtCore.Qt.AA_EnableHighDpiScaling, True)