parseData.py

import serial
import time
import binascii
import csv

order = 'big'
class AvionicsData:
 def __init__(self):
 self.imu = [0] * 9
 self.bar = [0] * 2
 self.gps = [0] * 4
 self.oxi = -1
 self.cmb = -1
 self.phs = -1
 self.upperVnt = -1
 self.injValve = -1
 self.lowerVnt = -1

 def __str__(self):
 phases = ["NA", "PRELAUNCH", "ARM", "BURN", "COAST", "DROGUE_DESCENT", "MAIN_DESCENT", "POSTLAUNCH", 
 "ABORT_COMMAND_RECEIVED", "ABORT_COMMUNICATION_ERROR", "ABORT_OXIDIZER_PRESSURE", "ABORT_UNSPECIFIED_REASON"]
 valveStatus = ["NA","Closed", "Open"]

 string="IMU - ACCEL:\t\t"+ str(self.imu[0:3])+" mg"+"\n"
 string+="IMU - GYRO:\t\t"+ str(self.imu[3:6])+" mdps"+"\n"
 string+="BAR - PRESS:\t\t"+ str(self.bar[0]/100) +" mbar"+"\n"
 string+="BAR - TEMP:\t\t"+ str(self.bar[1]/100)+" degrees C"+"\n"
 string+="OXI - P:\t\t"+ str(self.oxi/1000)+" psi"+"\n"
 string+="CMB - P:\t\t"+ str(self.cmb/1000)+" psi"+"\n"
 string+="PHS - PHASE:\t\t"+ str(phases[self.phs+1])+"\n"
 string+="Inj Valve:\t\t" + str(valveStatus[self.injValve+1])+"\n"
 string+="Lower Vent:\t\t" + str(valveStatus[self.lowerVnt+1])+"\n"

 with open('fligthData.csv', 'a') as csvfile:
 	spamwriter = csv.writer(csvfile, delimiter=',', quoting=csv.QUOTE_MINIMAL)
 	spamwriter.writerow([str(self.imu[0]), str(self.imu[1]), str(self.imu[2]), str(self.imu[3]), str(self.imu[4]), str(self.imu[5]), str(self.bar[0]), str(self.bar[1])])

 return string

def twos_complement(hexstr,bits):
 value = int(hexstr,16)
 if value & (1 << (bits-1)):
 value -= 1 << bits
 return value

def disconnect(ser):
 ser.close()
 time.sleep(1)
 return None

def readSerial(ser,data):
 line = ser.read(256).hex()
 i = 0
 print(line)
 while(i<len(line)):
 # IMU Data
 if((line[i:i+8]=='31313131') and (len(line)-i>=81)):
 if(line[i+80:i+82]=='00'):
 print(line[i:i+82])
 data.imu[0] = twos_complement(line[i+8:i+16], 32)
 data.imu[1] = twos_complement(line[i+16:i+24], 32)
 data.imu[2] = twos_complement(line[i+24:i+32], 32)
 data.imu[3] = twos_complement(line[i+32:i+40], 32)
 data.imu[4] = twos_complement(line[i+40:i+48], 32)
 data.imu[5] = twos_complement(line[i+48:i+56], 32)
 i+=81
 else: i+=1

 # Barometer Data
 elif((line[i:i+8])=='32323232' and len(line)-i>=25):
 if(line[i+24:i+26]=='00'):
 print(line[i:i+26])
 data.bar[0] = twos_complement(line[i+8:i+16], 32)
 data.bar[1] = twos_complement(line[i+16:i+24], 32)
 i+=25
 else:
 i+=1

 #GPS Data
 elif((line[i:i+8]=='33333333') and (len(line)-i>=41)):
 if(line[i+40:i+42]=='00'):
 print(line[i:i+42])
 data.gps[0] = twos_complement(line[i+8:i+16], 32)
 data.gps[1] = twos_complement(line[i+16:i+24], 32)
 data.gps[2] = twos_complement(line[i+24:i+32], 32)
 data.gps[3] = twos_complement(line[i+32:i+40], 32) 
 i+=41
 else:
 i+=1

 #Oxidizer Tank Pressure
 elif((line[i:i+8]=='34343434') and (len(line)-i>=17)):
 if(line[i+16:i+18]=='00'):
 data.oxi = twos_complement(line[i+8:i+16], 32)
 i+=17
 else: i+=1

 #Combustion Chamber Pressure
 elif((line[i:i+8]=='35353535') and (len(line)-i>=17)):
 if(line[i+16:i+18]=='00'):
 data.cmb = twos_complement(line[i+8:i+16], 32)
 i+=17
 else: i+=1

 #Flight Phase
 elif((line[i:i+8]=='36363636') and (len(line)-i>=13)):
 if(line[i+10:i+12]=='00'):
 data.phs = int(line[i+8:i+10], 16)
 i+=12
 else: i+=1

 #Injection Valve Status
 elif((line[i:i+8]=='38383838') and (len(line)-i>=13)):
 if(line[i+10:i+12]=='00'):
 data.injValve = int(line[i+8:i+10], 16)
 i+=12
 else: i+=1

 #Lower Vent Valve
 elif((line[i:i+8]=='39393939') and (len(line)-i>=13)):
 if(line[i+10:i+12]=='00'):
 data.lowerVnt = int(line[i+8:i+10], 16)
 i+=12
 else: i+=1

 #No packet detected
 else: i+=1
 print(data)


if __name__ == "__main__":
 ser = None
 data = AvionicsData()
 while(True):
 port = input('Enter a Serial Port to connect to:') #Linux: /dev/ttyUSBx, Windows: COMx
 ser = serial.Serial(port, 9600, timeout=0)

 while(ser!=None):
 time.sleep(1)
 readSerial(ser, data)