ground-control/dashboard.py at master · aesirkth/ground-control · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
import sys
import tkinter as tk
from tkinter import E, N, S, W

from gui import (GPSWidget, LiveTimeGraphAcc, LiveTimeGraphAirSpeed,
                 LiveTimeGraphAltitude, LiveTimeGraphGyro, LaunchpadWidget,
                 RocketStatus, TelemetryWidget)
from utils import (DummySerialWrapper, Gateway, LaunchpadControl,
                   SerialWrapper, Sigmundr)


class MainApplication(tk.Frame):
    """ TKinter frame holding some useful widgets to control the Launch Pad Station

    Parameters
    ----------
    parent : Tkinter TK() instance
        TK() instance to hold the widgets
    telemetry : Gateway instance
        Gateway instance correctly set for the Telemetry Gateway
    lps : Gateway instance
        Gateway instance correctly set for the LPS gateway

    """

    def __init__(self, parent, telemetry, lps, *args, **kwargs):
        tk.Frame.__init__(self, parent, *args, **kwargs)
        self.parent = parent
        self.telemetry = telemetry
        self.lps = lps

        self.left_column = tk.Frame(self)
        self.left_column.grid(row=1, column=0, sticky=W+N)

        self.lps_widget = LaunchpadWidget(
            self.left_column, self.lps, bd=2, relief="ridge")
        self.lps_widget.grid(row=0, column=1, sticky=W+E+N+S)

        self.tm_widget = TelemetryWidget(
            self.left_column, self.telemetry, bd=2, relief="ridge")
        self.tm_widget.grid(row=1, column=1, sticky=W+E+N+S)

        self.rocket_status = RocketStatus(
            self.left_column, self.telemetry, bd=2, relief="ridge")
        self.rocket_status.grid(row=2, column=1, sticky=W+E+N+S)

        self.middle_column = tk.Frame(self)
        self.middle_column.grid(row=1, column=1, sticky=W+N)

        self.speed_graph = LiveTimeGraphAirSpeed(self.middle_column, self.telemetry)
        self.speed_graph.grid(
            row=1, column=2, padx=5, pady=5)

        self.gyro_graph = LiveTimeGraphAltitude(self.middle_column, self.telemetry)
        self.gyro_graph.grid(
            row=1, column=3, padx=5, pady=5)

        self.acceleration_graph = LiveTimeGraphAcc(self.middle_column, self.telemetry)
        self.acceleration_graph.grid(
            row=2, column=2, padx=5, pady=5)

        self.alt_graph = LiveTimeGraphGyro(self.middle_column, self.telemetry)
        self.alt_graph.grid(
            row=2, column=3, padx=5, pady=5)

        self.gps = GPSWidget(self, self.telemetry, bd=2, relief="ridge")
        self.gps.grid(row=1, column=4, sticky=N)


if __name__ == "__main__":
    # Get the first argument given
    if len(sys.argv) >= 2:
        if sys.argv[1] == "rfd":
            # Use this with a RFD900 modem
            serial_telemetry = SerialWrapper(115200, "Telemetry", rfd900=True)
        elif sys.argv[1] == "dummy":
            # Use this to simulate a telemetry data flow
            serial_telemetry = DummySerialWrapper('Dummy')
        elif sys.argv[1] == "file":
            # Use this to feed previously recorded data into the dashboard
            if len(sys.argv) >= 3:
                filepath = sys.argv[2]
            else:
                filepath = "./data/2019-12-04T11-15-39_Telemetry.log"
            dummy_sensors = Sigmundr()
            serial_telemetry = SerialWrapper(115200, "Telemetry", filepath=filepath, sensors=dummy_sensors)

        else:
            serial_telemetry = SerialWrapper(115200, "Telemetry", rfd900=True)
    else:
        serial_telemetry = SerialWrapper(115200, "Telemetry", rfd900=True)

    rocket_sensors = Sigmundr()

    telemetry = Gateway(serial_telemetry, rocket_sensors, "./data")

    serial_lps = SerialWrapper(115200, "LPS", bonjour="LAUNCHPADCONTROLLER")
    lps_sensors = LaunchpadControl()
    lps = Gateway(serial_lps, lps_sensors, "./data")

    root = tk.Tk()
    root.title("Sigmundr Dashboard")

    MainApplication(root, telemetry, lps).pack(
        side="top", fill="both", expand=True)

    root.mainloop()