74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
|
import socket
|
||
|
|
import select
|
||
|
|
from simpleLog import log
|
||
|
|
|
||
|
|
def main(host, port, outFile, event, exitAll):
|
||
|
|
log("Headset: Going to talk to "+str(host)+":"+str(port))
|
||
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||
|
|
sock.setblocking(False)
|
||
|
|
sock.sendto("START".encode(), (host, port))
|
||
|
|
f = open(outFile, "w")
|
||
|
|
f.close()
|
||
|
|
f = open(outFile, "a")
|
||
|
|
i = 0
|
||
|
|
|
||
|
|
while True:
|
||
|
|
if exitAll.is_set():
|
||
|
|
log("headset exiting prematurely.")
|
||
|
|
exit(0)
|
||
|
|
break
|
||
|
|
ready = select.select([sock], [], [], 3)
|
||
|
|
if (ready[0]):
|
||
|
|
# log("Data is ready!")
|
||
|
|
i += 1
|
||
|
|
data, addr = sock.recvfrom(512)
|
||
|
|
data = data.decode("utf-8")
|
||
|
|
sock.sendto("ACK".encode(), (host, port))
|
||
|
|
if i == 1:
|
||
|
|
log("Headset: Writing header and 0's to set baseline")
|
||
|
|
f.write("index,strength,attention,meditation,delta,theta,low_alpha,high_alpha,low_beta,high_beta,low_gamma,high_gamma\n")
|
||
|
|
f.write("0,0,0,0,0,0,0,0,0,0,0,0\n")
|
||
|
|
f.write(str(i)+","+data+"\n")
|
||
|
|
f.flush()
|
||
|
|
log("Headset: Writing packet and unblocking threads")
|
||
|
|
event.set() # Unblocks all other threads
|
||
|
|
elif i != 0:
|
||
|
|
if exitAll.is_set():
|
||
|
|
log("headset exiting prematurely.")
|
||
|
|
exit(0)
|
||
|
|
break
|
||
|
|
log("Waiting for data timed out! Re-sending START packet.")
|
||
|
|
sock.sendto("START".encode(), (host, port))
|
||
|
|
|
||
|
|
# strength, attention, meditation, delta, theta, low_alpha, high_alpha, low_beta, high_beta, low_gamma, high_gamma = data.split(",") #signal strength, attention, meditation, delta, theta, low alpha, high alpha, low beta, high beta, low gamma, high gamma
|
||
|
|
'''
|
||
|
|
strength = brain[0] # 0 good 200 bad
|
||
|
|
attention = brain[1] # proprietary "attention"
|
||
|
|
meditation = brain[2] # proprietary "meditation"
|
||
|
|
delta = brain[3] # sleep
|
||
|
|
theta = brain[4] # relaxed, meditative
|
||
|
|
low_alpha = brain[5] # eyes closed, relaxed
|
||
|
|
high_alpha = brain[6] # "wakeful relaxation, less depression and anxiety"
|
||
|
|
low_beta = brain[7] # alert, focused
|
||
|
|
high_beta = brain[8] # "significant stress, anxiety, paranoia"
|
||
|
|
low_gamma = brain[9] # multi-sensory processing
|
||
|
|
high_gamma = brain[10] # "happier, more receptive"
|
||
|
|
'''
|
||
|
|
'''
|
||
|
|
log("\n"*50)
|
||
|
|
if int(strength) != 0:
|
||
|
|
log("Signal strength is not ideal; "+strength)
|
||
|
|
log("Strength: "+strength)
|
||
|
|
log("Attention: " + attention)
|
||
|
|
log("Meditation: " + meditation)
|
||
|
|
log("Delta: " + delta)
|
||
|
|
log("Theta: " + theta)
|
||
|
|
log("Low Alpha: " + low_alpha)
|
||
|
|
log("High Alpha: " + high_alpha)
|
||
|
|
log("Low Beta: " + low_beta)
|
||
|
|
log("High Beta: " + high_beta)
|
||
|
|
log("Low Gamma: " + low_gamma)
|
||
|
|
log("High Gamma: " + high_gamma)
|
||
|
|
'''
|
||
|
|
|