teleplot.py

import sys,os
#import curses
import time
from mqtt_client.subscriber import Subscriber
from threading import Thread
import json
import matplotlib.pyplot as plt
import matplotlib.animation as animation
from matplotlib import style

#plotting params
params = {
    'axes.titlesize' : 'xx-small'
}
plt.rcParams.update(params) 

# global variables for plt.(Show)
mag_compass_reading = 0
int_compass_reading = 0
time_reading = 0
lat_reading = 0
lon_reading = 0
speed_reading = 0
gps_heading_reading = 0
jet1_temp = 0
jet2_temp = 0
compartment_temp = 0
gps_distance = 0

jet1_current = 0 #starboard
jet2_current = 0 #port
pack_voltage = 0

#base for log files
_LOG_BASE = "log"

#globals for plotting
style.use ('dark_background')
fig, axs = plt.subplots(4,1)
fig.subplots_adjust(hspace = 0.5)
#fig2, axs2 = plt.subplots(1,1)


x2 = []
y2 = []
x1 = []
y1 = []
y3 = []
y4 = []
y5 = []
x6 = []
y6 = []

timeArray = []
ntime = 0
stime = 0
def animate(i):
    global time_reading
    global jet1_current
    global mag_compass_reading
    global ntime
    global stime
    global pack_voltage
    global lon_reading
    global lat_reading

    stime = stime + 0.1
    ntime = ntime + 0.1
    y2.append(jet1_current)
    x1.append(ntime)
    x2.append(stime)
    y1.append(mag_compass_reading)
    y3.append(jet2_current)
    y4.append(speed_reading)
    y5.append(pack_voltage)
    x6.append(lon_reading)
    y6.append(lat_reading)
    if len(x1) > 100:
        x1.pop(0)
    if len(y1) > 100:
        y1.pop(0)
    if len(y2) > 100:
        y2.pop(0)
    if len(y3) > 100:
        y3.pop(0)
    if len(y4) > 100:
        y4.pop(0) 
    #if len(y5) > 100:
    #    y5.pop(0) 
    
    
    axs[0].clear()
    axs[0].grid(which='major', linestyle=':', linewidth='0.5', color= 'white')
    axs[0].set_ylim(0,360)
    axs[0].set_title('Mag Compass Reading', fontsize = 12)
    axs[0].plot(x1,y1)
    axs[1].clear()
    axs[1].grid(which='major', linestyle=':', linewidth='0.5', color= 'white')
    axs[1].set_ylim(0,100)
    axs[1].set_title('Jet Currents', fontsize = 12)
    axs[1].plot(x1,y2)
    axs[1].plot(x1,y3)
    axs[2].clear()
    axs[2].grid(which='major', linestyle=':', linewidth='0.5', color= 'white')
    axs[2].set_ylim(0,15)
    axs[2].set_title('Speed (kn)', fontsize = 12)
    axs[2].plot(x1,y4)
    axs[3].clear()
    axs[3].grid(which='major', linestyle=':', linewidth='0.5', color= 'white')
    axs[3].set_ylim(0,20)
    axs[3].set_title('Pack Voltage', fontsize = 12)
    axs[3].plot(x2,y5)
    # axs2.clear()
    # axs2.grid(which='major', linestyle=':', linewidth='0.5', color= 'white')
    # axs2.set_title('Position', fontsize = 12)
    # axs2.scatter(x6,y6)


    

    #graph_data = #open(f"../logs/{_LOG_BASE}.txt", "r").read

# def on_log_received(client, userdata, message):
#     global _LOG_BASE
#     log_title = message.payload.decode("utf-8")
#     time = datetime.today()
#     log_time = (
#         f"{time.year}-{time.month}-{time.day}-{time.hour}:{time.minute}:{time.second}"
#         )
#     _LOG_BASE = log_title + "_" + log_time
#     print(_LOG_BASE)    

def on_compass_received(client, userdata, message):
    global mag_compass_reading
    obj = json.loads(message.payload.decode('utf-8'))
    mag_compass_reading = obj['compass']

def on_internal_compass_received(client, userdata, message):
    global int_compass_reading
    obj = json.loads(message.payload.decode('utf-8'))
    int_compass_reading = obj['heading']

def on_gps_received(client, userdata, message):
    # create global variables for UI
    global time_reading
    global lat_reading
    global lon_reading
    global speed_reading
    global gps_heading_reading
    global gps_distance
    
    obj = json.loads(message.payload.decode('utf-8'))

    # parse json into global variablesspeed_reading,
    time_reading = obj["time"]
    lat_reading = obj['latitude']
    lon_reading = obj['longitude']
    speed_reading = obj["speed"]
    gps_heading_reading = obj["course"]
    gps_distance = obj['distance']

def on_adc_received(client, userdata, message):
    global jet1_current
    global jet2_current
    global pack_voltage

    obj = json.loads(message.payload.decode('utf-8'))
    jet1_current = obj["jet1_amps"]
    jet2_current = obj["jet2_amps"]
    pack_voltage = obj['pack_voltage']
def on_temp_received(client, userdata, message):
    global jet1_temp
    global jet2_temp
    global compartment_temp

    obj = json.loads(message.payload.decode('utf-8'))
    jet1_temp = obj["jet1_temp"]
    jet2_temp = obj["jet2_temp"]
    compartment_temp = obj["compartment_temp"]

# def draw(stdscr):
#     # Make stdscr.getch non-blocking
#     stdscr.nodelay(True)
#     stdscr.clear()
#     width = 4
#     count = 0
#     direction = 1

#     begin_x = 20
#     begin_y = 7
#     height = 5
#     width = 40

#     second_column_width = 25

#     start_x_title = int((width // 2) - (len("RWM TELEMETRY") // 2) - len("RWM TELEMETRY") % 2)
#     start_x_partition = int((width // 2) - (len("--------------- NAV ---------------") // 2) - len("--------------- NAV ---------------") % 2)
#     curses.init_pair(1, curses.COLOR_RED, curses.COLOR_BLACK)
#     curses.init_pair(2, curses.COLOR_GREEN, curses.COLOR_BLACK)

    # define static text for UI
   

    # while True:
    #     c = stdscr.getch()

    #     curses.flushinp()
    #     stdscr.clear()

    #     # --- HEADER ---
    #     stdscr.addstr(0,start_x_title, "RWM TELEMETRY")
        
    
    #     # --- NAV VALUES ---
    #     stdscr.addstr(1,start_x_partition, "--------------- NAV ---------------")
        
    #     stdscr.addstr(2,0,"Mag Compass: ")
    #     stdscr.addstr(2,second_column_width,str(mag_compass_reading))

    #     stdscr.addstr(3,0,"Int. Mag Compass: ")
    #     stdscr.addstr(3,second_column_width,str(round(int_compass_reading,2)))

    #     stdscr.addstr(4,0,"GPS Compass: ")
    #     stdscr.addstr(4,second_column_width,str(gps_heading_reading))

    #     stdscr.addstr(5,0,"GPS Time: ")
    #     stdscr.addstr(5,second_column_width,str(time_reading))

    #     stdscr.addstr(6,0,"Latitude: ")
    #     stdscr.addstr(6,second_column_width,str(round(lat_reading,6)))

    #     stdscr.addstr(7,0,"Longitude: ")
    #     stdscr.addstr(7,second_column_width,str(round(lon_reading,6)))

    #     stdscr.addstr(8,0,"GPS Speed(kn): ")
    #     stdscr.addstr(8, second_column_width, str(round(speed_reading,2)))

    #     stdscr.addstr(9,0,"GPS distance (NM): ")
    #     stdscr.addstr(9, second_column_width, str(gps_distance))

        

    #     # --- JET VALUES ---

    #     stdscr.addstr(10,start_x_partition, "--------------- JET ---------------")
    #     stdscr.addstr(11,0,"Starboard Jet Current : ")

    #     jet1_amps = ((jet1_current - 2.47) / 0.013)
    #     jet2_amps = ((jet2_current- 2.47) / 0.013)
        
    #     if(jet1_current < 0):
    #         stdscr.addstr(11,second_column_width,str(round(jet1_amps,2)), curses.color_pair(1))
    #     else:
    #         stdscr.addstr(11,second_column_width,str(round(jet1_amps,2)), curses.color_pair(2))

    #     stdscr.addstr(12,0,"Port Jet Current: ")

    #     if(jet2_current < 0):
    #         stdscr.addstr(12,second_column_width,str(round(jet2_amps,2)), curses.color_pair(1))
    #     else:
    #         stdscr.addstr(12,second_column_width,str(round(jet2_amps,2)), curses.color_pair(2))

    #     stdscr.addstr(13,0,"JET Current Delta: ")

    #     jet_delta = jet1_amps - jet2_amps

    #     if(jet_delta < -3 or jet_delta > 3):
    #         stdscr.addstr(13,second_column_width,str(round(jet_delta,2)), curses.color_pair(1))
    #     else:
    #         stdscr.addstr(13,second_column_width,str(round(jet_delta,2)), curses.color_pair(2))

    #     stdscr.addstr(14,0,"Starboard Jet Temp c: ")
    #     stdscr.addstr(14,second_column_width,str(jet1_temp))

    #     stdscr.addstr(15,0,"Port Jet Temp c: ")
    #     stdscr.addstr(15,second_column_width,str(jet2_temp))

    #     stdscr.addstr(16,0,"Compartment Temp c: ")
    #     stdscr.addstr(16,second_column_width,str(compartment_temp))
     
        

    #     time.sleep(0.01)

# ==================
# -- MAIN METHOD -- 
# ==================
if __name__ == '__main__':

    try:
        default_subscriptions = {
            "/status/compass": on_compass_received,
            "/status/gps" : on_gps_received,
            "/status/adc" : on_adc_received,
            "/status/internal_compass" : on_internal_compass_received,
            "/status/temp" : on_temp_received
            #"/command/logging" : on_log_received
        }
        subber = Subscriber(client_id="teleplot_live", broker_ip="192.168.1.170", default_subscriptions=default_subscriptions)
        thread = Thread(target=subber.listen)
        thread.start()
        while True:
            
            ani = animation.FuncAnimation(fig, animate, interval=100)
            #ani2 = animation.FuncAnimation(fig2, animate, interval=100)
            plt.show()
            time.sleep(0.1)
        #plt.show()
        #curses.wrapper(draw)
    except KeyboardInterrupt:
        exit
        #curses.endwin()