optical_gui.py

#!/usr/bin/env python
# coding: utf-8

# In[ ]:


from pylab import *
from tkinter import *
import math
import serial
import tkinter as tk
import pyfirmata
import time
import threading 
import pandas as pd
from pyfirmata import Arduino 
from os import system
from timeit import default_timer as timer
from datetime import timedelta
root = Tk()
root.title("DBS Trigger Control")
root.geometry("450x500")
e = Entry(root)
n = 0

recording_number = []
parameter = []
notes = []



o_duration = 0
o_delay = 0
pulse_number = 0
number_of_trains = 1;run_x_times = number_of_trains
time_between_pulses = 0
time_between_trains = 0
pulse_number = 0


# In[ ]:


entry = tk.Entry(root)
variable=StringVar()


# In[ ]:


path = '/Users/granthughes/Desktop/eo_stim/data_frames/1_gui_runs_csv/12_09_21_test/'


# In[ ]:


#Arduino port designation
board = Arduino('/dev/cu.usbmodem14101')


# In[ ]:


def do_train():
    
    # Update vars
    recording_number.append(recording_number_window.get())
    parameter.append(parameter_window.get())
    notes.append(notes_window.get())

    o_duration = float(o_duration_entry.get())
    o_delay = float(o_delay_entry.get())
    train_number = float(run_x_times_entry.get())
    pulse_number = float(pulse_num_entry.get())
    time_between_pulses = float(time_between_pulses_entry.get())
    time_between_trains = float(time_between_trains_entry.get())
   
  
    

    all_data = {
        'run': [recording_number[-1]]*int(train_number),
        'parameter': [parameter[-1]]*int(train_number),
        'notes': [notes[-1]]*int(train_number),
        

        'o_duration': [o_duration] *int(train_number),
        'o_delay': [o_delay]* int(train_number),
        'number_of_trains': [train_number]*int(train_number),
        'pulse_number': [pulse_number]*int(train_number),
        'time_between_pulses': [time_between_pulses]* int(train_number),
        'time_between_trains': [time_between_trains]*int(train_number),
        
    }



    for train in range(int(number_of_trains)):
            for pulse in range(int(pulse_number)):
                print("train "+str(train+1)+"  pulse "+str(pulse+1))
                 #******start single pulse sequence

                board.digital[1].write(1)
                board.digital[2].write(1)
                time.sleep(o_duration)
                board.digital[1].write(0)
                board.digital[2].write(0)

                time.sleep(o_delay) #Add input in gooey for this time

                board.digital[3].write(1)
                board.digital[4].write(1)
                time.sleep(o_duration)
                board.digital[3].write(0)
                board.digital[4].write(0)

                time.sleep(o_delay)

                board.digital[5].write(1)
                board.digital[6].write(1)
                time.sleep(o_duration)
                board.digital[5].write(0)
                board.digital[6].write(0)

                time.sleep(o_delay)

                board.digital[5].write(1)
                board.digital[6].write(1)
                time.sleep(o_duration)
                board.digital[5].write(0)
                board.digital[6].write(0)

                time.sleep(o_delay)

                board.digital[7].write(1)
                board.digital[8].write(1)
                time.sleep(o_duration)
                board.digital[7].write(0)
                board.digital[8].write(0)

                train_counter = train+1
                Current_Loop_Number.config(text="Train Number Completed: " + str(train_counter))
                update_counter()
                time.sleep(Time_between_pulses)
                #********end single pulse sequence
            time.sleep(time_between_trains)

    return all_data_df


# In[ ]:


# # GUI window Control

#Tkinter Windows

app = Frame(root)
app.grid()

startx = Button(app, text='Run Train', command=do_train,fg='green')

## text_windows
recording_number_label = Label(app, text='run_number')
recording_number_window = Entry(app, text='')

parameter_label = Label(app, text='parameter')
parameter_window = Entry(app, text='')

notes_label = Label(app, text='Notes')
notes_window = Entry(app, text='')

### Int_windows
o_duration_label = Label(app, text= "Optical Pulse Duration (sec)")
o_duration_entry = Entry(app, text="Optical Duration Seconds")
o_duration_entry.insert(0,o_duration)

o_delay_label = Label(app, text= "o_delay")
o_delay_entry = Entry(app, text="o_delay")
o_delay_entry.insert(0,o_delay)

pulse_num_label = Label(app, text="(pulse_number) Number_of_pulses_per_train", fg='#9900ff')
pulse_num_entry = Entry(app, text="")
pulse_num_entry.insert(0,pulse_number)

time_between_trains_label = Label(app, text="Time_between_trains (sec)",fg='#9900ff')
time_between_trains_entry = Entry(app, text="")
time_between_trains_entry.insert(0,time_between_trains)

run_x_times_label = Label(app, text="Number_of_trains",fg='#9900ff')
run_x_times_entry = Entry(app, text= "")
run_x_times_entry.insert(0,run_x_times)

time_between_pulses_label = Label(app, text="Time_between_pulses (sec)")
time_between_pulses_entry = Entry(app, text="")
time_between_pulses_entry.insert(0,time_between_pulses)


Current_Loop_Number = Label(app, text = "Train Number Completed: " + str(n),fg='#9900ff')


e1 = tk.Entry(root)

# Gui Layout

startx.grid(row=1, column=0)

## text_windows
recording_number_label.grid(row=2, column=0)
recording_number_window.grid(row=3, column=0)

parameter_label.grid(row=4, column=0)
parameter_window.grid(row=5, column=0)

notes_label.grid(row=6, column=0)
notes_window.grid(row=7, column=0)

## int_windows

o_duration_label.grid(row=1, column=1)
o_duration_entry.grid(row=2, column=1)

o_delay_label.grid(row=3, column=1)
o_delay_entry.grid(row=4, column=1)

run_x_times_label.grid(row=5, column=1)
run_x_times_entry.grid(row=6, column=1)

pulse_num_label.grid(row=7, column=1) 
pulse_num_entry.grid(row=8, column=1) 

time_between_pulses_label.grid(row=9, column=1)
time_between_pulses_entry.grid(row=10, column=1)

time_between_trains_label.grid(row=11, column=1) 
time_between_trains_entry.grid(row=12, column=1) 



root.mainloop()


# In[ ]:





# In[ ]:





# In[ ]:





# In[ ]:





# In[ ]:





# In[ ]:





# In[ ]:





# In[ ]:





# In[ ]: