-
Notifications
You must be signed in to change notification settings - Fork 1
/
control_serial.py
59 lines (48 loc) · 1.8 KB
/
control_serial.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#!/usr/bin/python
import time
import sys
import serial
import array
import ff_sim as ffs
def runner(per_strip, strip, my_flies, port, t):
#update state of all ffs
my_flies1 = ffs.update_state(my_flies, t, strip, per_strip)
#fill buffer
r_val = []
g_val = []
for i in xrange(strip):
for j in xrange(per_strip):
r_val.append(my_flies1[i][j].brightnessR)
g_val.append(my_flies1[i][j].brightnessG)
#pass signal
for i in r_val:
port.write(chr(i))
for j in g_val:
port.write(chr(j))
return my_flies1
#Expects to run with 2 arguments:
#num of strips and num of ff's per strip
if __name__ == '__main__':
#Grab input arguments
ff_num = int(raw_input("How many fireflies are on a strip?: "))
strip_num = int(raw_input("How many strips?: "))
stim_w = float(raw_input("Stimulus Period?: "))
A_min = float(raw_input("Minimum Reset Strength A?: "))
A_max = float(raw_input("Maximum Reset Strength A?: "))
w_min = float(raw_input("Minimum Period: "))
w_max = float(raw_input("Maximum Period?: "))
update_time = float(raw_input("What time step would you like to use? (takes about .003s to pass message): "))
t_up = float(raw_input("Blink up time?: "))
t_down = float(raw_input("Blink down time?: "))
#Create FF Agents
FFs = ffs.make_ff_array(ff_num, strip_num, A_max, A_min, stim_w, w_max, w_min, update_time, t_up, t_down)
#Set up serial port
port = serial.Serial(port="/dev/ttyAMA0", baudrate=115200)
port.open()
port.flushOutput()
#May eventually need a reset so that memory doesn't overflow
#Start Running
print "Press Ctrl-C to stop at any time"
while True:
FFs = runner(ff_num, strip_num, FFs, port, update_time)
time.sleep(update_time)