Added arrays to simplify variables.

parent 4434d220
......@@ -8,6 +8,11 @@ import datetime
import tkinter.font as font
global i
global protocolNumber
h = 3
w = 6
global protocolParam
protocolParam = [[0 for x in range(w)] for y in range(h)]
global n_items_basal
global height_items_basal
global height_items_basal_bi
......@@ -181,6 +186,7 @@ def select_protocol(canvas, protocol):
for l in list:
if(l!=prot_text1 and l!=prot_text2 and l!=prot_text3 and l!=protocol_list1 and l!=protocol_list2 and l!=protocol_list3 and l!=button_prot_1 and l!=button_prot_2 and l!=button_prot_3):
l.destroy()
if (protocol=="Basal unipolar"):
prot_image = Image.open("protocolo_basal.PNG")
prot_photo = ImageTk.PhotoImage(prot_image.resize((420,110)))
......@@ -465,8 +471,14 @@ def select_protocol(canvas, protocol):
#Função para interface de seleção dos parâmetros de protocolo Basal unipolar
def select_par_basal(canvas, T, IPI, ITI, dur):
if(canvas==canvas1):
protocolNumber = 0
elif(canvas==canvas2):
protocolNumber = 1
elif(canvas==canvas3):
protocolNumber = 2
try:
T_f = int(T)
T_f = int(T)
IPI_f = int(IPI)
ITI_f = int(ITI)
dur_f = int(dur)
......@@ -478,39 +490,25 @@ def select_par_basal(canvas, T, IPI, ITI, dur):
n_train_text = Label(canvas, text="Nº de trens do protocolo:")
n_train_text.place(relx=0, rely=0.5+(i*height_items_basal), relwidth=0.5, relheight=height_items_basal)
n_train_text_var = Label(canvas, text=n_train_calc)
n_train_text_var.place(relx=0.5, rely=0.5+(i*height_items_basal), relwidth=0.5, relheight=height_items_basal)
if (canvas==canvas1):
global T_1
global IPI_1
global ITI_1
global dur_1
T_f += 256
IPI_f += 256
ITI_f += 256
dur_f += 256
T_1 = T_f.to_bytes(2,'little')
IPI_1 = IPI_f.to_bytes(2,'little')
ITI_1 = ITI_f.to_bytes(2,'little')
dur_1 = dur_f.to_bytes(2,'little')
print(T_1)
print(IPI_1)
print(ITI_1)
print(dur_1)
if (canvas==canvas2):
global T_f_2
global IPI_f_2
global ITI_f_2
global dur_f_2
if (canvas==canvas3):
global T_f_3
global IPI_f_3
global ITI_f_3
global dur_f_3
n_train_text_var.place(relx=0.5, rely=0.5+(i*height_items_basal), relwidth=0.5, relheight=height_items_basal)
T_f += 256
IPI_f += 256
ITI_f += 256
dur_f += 256
protocolParam[protocolNumber][0] = T_f.to_bytes(2,'little')
protocolParam[protocolNumber][1] = IPI_f.to_bytes(2,'little')
protocolParam[protocolNumber][2] = ITI_f.to_bytes(2,'little')
protocolParam[protocolNumber][3] = int(257).to_bytes(2,'little')
protocolParam[protocolNumber][4] = int(257).to_bytes(2,'little')
protocolParam[protocolNumber][5] = int(257).to_bytes(2,'little')
print(protocolParam[protocolNumber][0])
print(protocolParam[protocolNumber][1])
print(protocolParam[protocolNumber][2])
print(protocolParam[protocolNumber][3])
print(protocolParam[protocolNumber][4])
print(protocolParam[protocolNumber][5])
writeToLog("Parâmetros definidos.")
except:
writeToLog("Algum erro ocorreu. Tente preencher todos os parâmetros.")
......@@ -770,18 +768,11 @@ def detect_arduino():
#Função de envio de dados para Arduino
def send_protocol():
try:
arduino.write(b'<')
print(prot1)
arduino.write(b'<')
if(prot1 == "Basal Unipolar"):
arduino.write(b'B')
arduino.write(b'U')
arduino.write(T_1)
arduino.write(IPI_1)
arduino.write(ITI_1)
arduino.write(dur_1)
arduino.write(b'0')
arduino.write(b'0')
arduino.write(b'0')
arduino.write(b'0')
elif(prot1== "Basal Bipolar"):
arduino.write(b'B')
arduino.write(b'B')
......@@ -793,14 +784,94 @@ def send_protocol():
arduino.write(b'B')
elif(prot1 == "PBS"):
arduino.write(b'P')
arduino.write(b'B')
else:
writeToLog("Problema no envio. Defina o protocolo 1.")
arduino.write(protocolParam[0][0])
print(protocolParam[0][0])
arduino.write(protocolParam[0][1])
print(protocolParam[0][1])
arduino.write(protocolParam[0][2])
print(protocolParam[0][2])
arduino.write(protocolParam[0][3])
print(protocolParam[0][3])
arduino.write(protocolParam[0][4])
print(protocolParam[0][4])
arduino.write(protocolParam[0][5])
print(protocolParam[0][5])
arduino.write(b'>')
receb = recvFromArduino()
if(receb == "SR"):
writeToLog("Protocolo 1 enviado.")
else:
writeToLog("Problema no envio do protocolo 1.")
arduino.write(b'<')
if(prot2 == "Basal Unipolar"):
arduino.write(b'B')
arduino.write(b'U')
elif(prot2== "Basal Bipolar"):
arduino.write(b'B')
arduino.write(b'B')
elif(prot2 == "TBS Unipolar"):
arduino.write(b'T')
arduino.write(b'U')
elif(prot2 == "TBS Bipolar"):
arduino.write(b'T')
arduino.write(b'B')
elif(prot2 == "PBS"):
arduino.write(b'P')
arduino.write(b'B')
else:
writeToLog("Problema no envio. Defina o protocolo 2.")
arduino.write(protocolParam[1][0])
arduino.write(protocolParam[1][1])
arduino.write(protocolParam[1][2])
arduino.write(protocolParam[1][3])
arduino.write(protocolParam[1][4])
arduino.write(protocolParam[1][5])
arduino.write(b'>')
receb = recvFromArduino()
if(receb == "SR"):
writeToLog("Protocolo 2 enviado.")
else:
writeToLog("Problema no envio do protocolo 2.")
arduino.write(b'<')
if(prot3 == "Basal Unipolar"):
arduino.write(b'B')
arduino.write(b'U')
elif(prot3== "Basal Bipolar"):
arduino.write(b'B')
arduino.write(b'B')
elif(prot3 == "TBS Unipolar"):
arduino.write(b'T')
arduino.write(b'U')
elif(prot3 == "TBS Bipolar"):
arduino.write(b'T')
arduino.write(b'B')
elif(prot3 == "PBS"):
arduino.write(b'P')
arduino.write(b'B')
else:
writeToLog("Problema no envio. Defina os protocolos desejados.")
arduino.write(b'>')
writeToLog("Problema no envio. Defina o protocolo 3.")
arduino.write(protocolParam[2][0])
arduino.write(protocolParam[2][1])
arduino.write(protocolParam[2][2])
arduino.write(protocolParam[2][3])
arduino.write(protocolParam[2][4])
arduino.write(protocolParam[2][5])
arduino.write(b'>')
receb = recvFromArduino()
print(receb)
if(receb=="SR"):
writeToLog("Protocolos enviados e iniciados com sucesso.")
if(receb == "SR"):
writeToLog("Protocolo 3 enviado.")
else:
writeToLog("Problema no envio do protocolo 3.")
writeToLog("Protocolos enviados e iniciados com sucesso.")
except:
writeToLog("Não foi possível enviar. Problema na conexão com Arduino.")
......@@ -813,7 +884,7 @@ def interrupt_protocol():
arduino.write(b'>')
receb = recvFromArduino()
print(receb)
if(receb == "IR"):
if(receb == "CR"):
writeToLog("Protocolos interrompidos com sucesso.")
except:
writeToLog("Não foi possível enviar. Problema na conexão com Arduino.")
......@@ -827,7 +898,7 @@ def jump_protocol():
arduino.write(b'>')
receb = recvFromArduino()
print(receb)
if(receb == "AR"):
if(receb == "CR"):
writeToLog("Protocolo avançado com sucesso.")
except:
writeToLog("Não foi possível enviar. Problema na conexão com Arduino.")
......@@ -855,11 +926,9 @@ def heartBeat():
if(receb != "HB"):
writeToLog("Checagem de conexão falhou. Reconectar Arduino.")
arduino.close()
del arduino
except:
writeToLog("Checagem de conexão falhou. Reconectar Arduino.")
arduino.close()
del arduino
root.after(2000,heartBeat)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment