fluffy/OutGaugeInterpreter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added option toggle to write data to csv file

Browse source
This commit is contained in:
Seth Samuel 2024-11-12 00:35:02 +13:00
parent ec71dd1094
commit eb76c18efc
1 changed files with 29 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

44
OutGaugeInterpreter.py
View file

@ -13,6 +13,7 @@ UDP_IP = "127.0.0.1"
BEAMNG_UDP_PORT = 4444 BEAMNG_UDP_PORT = 4444
FORZA_UDP_PORT= 4843 FORZA_UDP_PORT= 4843
SERIAL_PORT = "" SERIAL_PORT = ""
firstRun= True #used to do the headers for csv files
class GameType(Enum): class GameType(Enum):
NONE=0, NONE=0,
@ -25,6 +26,19 @@ gameSelected = False
portToConnect= 0 portToConnect= 0
gameType= GameType.NONE gameType= GameType.NONE
carData = 0 carData = 0
csvOut = False
#Functions <-------------------------------------------------------------------------------------------
def csvWriteOut(firstRun,carData,csvFile):
if firstRun == True:
out=""
for im,nm in carData.items():
out = out + im + ","
csvFile.write(out + "\n")
outString=""
for itm,num in carData.items():
outString = outString + str(num) +","
csvFile.write(outString+"\n")
#code runs from here<----------------------------------------------------------------------------------------------------------- #code runs from here<-----------------------------------------------------------------------------------------------------------
@ -40,7 +54,7 @@ else:
#select game #select game
while gameSelected == False: while gameSelected == False:
gameNo = input("1:BEAMNG\n2:FORZA\n\n\n9:SET SERIAL PORT ("+SERIAL_PORT+")\n") gameNo = input("1:BEAMNG\n2:FORZA\n\n7:Toggle CSV out ("+str(csvOut)+")\n9:SET SERIAL PORT ("+SERIAL_PORT+")\n")
if gameNo == "1": if gameNo == "1":
portToConnect = BEAMNG_UDP_PORT portToConnect = BEAMNG_UDP_PORT
gameSelected = True gameSelected = True
@ -51,6 +65,11 @@ while gameSelected == False:
gameType=GameType.FORZA gameType=GameType.FORZA
gameSelected=True gameSelected=True
print("Forza Selected") print("Forza Selected")
elif gameNo == "7":
if csvOut == True:
csvOut=False
elif csvOut ==False:
csvOut =True
elif gameNo == "9": elif gameNo == "9":
SERIAL_PORT = input("set serial port: ") SERIAL_PORT = input("set serial port: ")
else: else:
@ -74,12 +93,11 @@ while connectedArduino == False:
print("please check connection to arduino and verify the correct serial port") print("please check connection to arduino and verify the correct serial port")
time.sleep(1) time.sleep(1)
csvOut = False if csvOut == True:
csvFile = open('dataOut.csv',"a") csvFile = open(str(int(time.time())+'.csv',"a"))
##csvWriter = csv.writer(csvFile)
firstRun= True
#print(struct.calcsize(FORZA_METHODS.FORZA_DATA_FORMAT))
print("ready:\n") print("ready:\n")
while True: while True:
data, addr = sock.recvfrom(1024) data, addr = sock.recvfrom(1024)
@ -87,22 +105,16 @@ while True:
unpackedData = struct.unpack(BEAMNG_METHODS.BEAMNG_DATA_FORMAT,data) unpackedData = struct.unpack(BEAMNG_METHODS.BEAMNG_DATA_FORMAT,data)
carData = BEAMNG_METHODS.unpackData(unpackedData) carData = BEAMNG_METHODS.unpackData(unpackedData)
kmh=carData["speed"]*3.6 kmh=carData["speed"]*3.6
if csvOut == True:
csvWriteOut(firstRun,carData,csvFile)
firstRun=False
elif gameType == GameType.FORZA: elif gameType == GameType.FORZA:
unpackedData = struct.unpack(FORZA_METHODS.FORZA_DATA_FORMAT,data) unpackedData = struct.unpack(FORZA_METHODS.FORZA_DATA_FORMAT,data)
carData = FORZA_METHODS.unpackData(unpackedData) carData = FORZA_METHODS.unpackData(unpackedData)
#carData = FORZA_METHODS.get_data(data)
kmh = carData["Speed"]*3.6 kmh = carData["Speed"]*3.6
if csvOut == True: if csvOut == True:
if firstRun == True: csvWriteOut(firstRun,carData,csvFile)
out=""
for im,nm in carData.items():
out = out + im + ","
csvFile.write(out + "\n")
firstRun=False firstRun=False
outString=""
for itm,num in carData.items():
outString = outString + str(num) +","
csvFile.write(outString+"\n")
toPi.write(str(kmh).encode()+":".encode()) toPi.write(str(kmh).encode()+":".encode())