# -*- coding: utf-8 -*-
from GridEyeKit import GridEYEKit
import time
import numpy as np
import os
import matplotlib.pyplot as plt
import threading
OUTFILE = "dump.csv"
csv = open(OUTFILE, "a")
print("Connecting to Grideye\n")
try:
g = GridEYEKit()
g_status_connect = g.connect()
if not g_status_connect:
g.close()
print("could not connect to grideye...\n")
os._exit(1)
print("Connected\n")
except:
print("could not connect to grideye...\n")
g.close()
os._exit(1)
data = g.get_temperatures()
def plot():
global data
im = plt.imshow(data, vmin=20, vmax=30)
class GridEYE_Viewer():
def __init__(self, root):
""" Initialize Window """
self.tkroot = root
# Close serial connection and close window
self.tkroot.protocol('WM_DELETE_WINDOW', self.exitwindow)
""" Initialize variables for color interpolation """
self.HUEstart = 0.5 # initial color for min temp (0.5 = blue)
self.HUEend = 1 # initial color for max temp (1 = red)
self.HUEspan = self.HUEend - self.HUEstart
""" Grid Eye related variables"""
self. MULTIPLIER = 0.25 # temp output multiplier
""" Initialize Loop bool"""
self.START = False
"""Initialize frame tor temperature array (tarr)"""
self.frameTarr = tk.Frame(master=self.tkroot, bg='white')
self.frameTarr.place(x=5, y=5, width=400, height=400)
"""Initialize pixels tor temperature array (tarr)"""
self.tarrpixels = []
for i in range(8):
# frameTarr.rowconfigure(i,weight=1) # self alignment
# frameTarr.columnconfigure(i,weight=1) # self alignment
for j in range(8):
pix = tk.Label(master=self.frameTarr, bg='gray', text='11')
spacerx = 1
spacery = 1
pixwidth = 40
pixheight = 40
pix.place(x=spacerx + j * (spacerx + pixwidth), y=spacery +
i * (pixheight + spacery), width=pixwidth, height=pixheight)
print
# attache all pixels to tarrpixel list
self.tarrpixels.append(pix)
"""Initialize frame tor Elements"""
self.frameElements = tk.Frame(master=self.tkroot, bg='white')
self.frameElements.place(x=410, y=5, width=100, height=400)
"""Initialize controll buttons"""
self.buttonStart = tk.Button(master=self.frameElements, text='start', bg='white',
command=self.start_update)
self.buttonStart.pack()
self.buttonStop = tk.Button(master=self.frameElements, text='stop', bg='white',
command=self.stop_update)
self.buttonStop.pack()
"""Initialize temperature adjustment"""
self.lableTEMPMAX = tk.Label(
master=self.frameElements, text='Max Temp (red)')
self.lableTEMPMAX.pack()
self.MAXTEMP = tk.Scale(
self.frameElements, from_=-20, to=120, resolution=0.25)
self.MAXTEMP.set(31)
self.MAXTEMP.pack()
self.lableMINTEMP = tk.Label(
master=self.frameElements, text='Min Temp (blue)')
self.lableMINTEMP.pack()
self.MINTEMP = tk.Scale(
self.frameElements, from_=-20, to=120, resolution=0.25)
self.MINTEMP.set(27)
self.MINTEMP.pack()
self.kit = GridEYEKit()
def exitwindow(self):
""" if windwow is clsoed, serial connection has to be closed!"""
self.kit.close()
self.tkroot.destroy()
def stop_update(self):
""" stop button action - stops infinite loop """
self.START = False
self.update_tarrpixels()
def start_update(self):
if self.kit.connect():
""" start button action -start serial connection and start pixel update loop"""
self.START = True
""" CAUTION: Wrong com port error is not handled"""
self.update_tarrpixels()
else:
messagebox.showerror(
"Not connected", "Could not find Grid-EYE Eval Kit - please install driver and connect")
def get_tarr(self):
""" unnecessary function - only converts numpy array to tuple object"""
tarr = []
for temp in self.kit.get_temperatures(): # only fue to use of old rutines
for temp2 in temp:
tarr.append(temp2)
return tarr
def update_tarrpixels(self):
""" Loop for updating pixels with values from funcion "get_tarr" - recursive function with exit variable"""
if self.START == True:
tarr = self.get_tarr() # Get temerature array
i = 0 # counter for tarr
if len(tarr) == len(self.tarrpixels): # check if problem with readout
for tarrpix in self.tarrpixels:
tarrpix.config(text=tarr[i]) # Update Pixel text
if tarr[i] < self.MINTEMP.get(): # For colors, set borders to min/max temp
normtemp = 0
# For colors, set borders to min/max temp
elif tarr[i] > self.MAXTEMP.get():
normtemp = 1
else:
TempSpan = self.MAXTEMP.get() - self.MINTEMP.get()
if TempSpan <= 0: # avoid division by 0 and negative values
TempSpan = 1
# Normalize temperature 0...1
normtemp = (float(tarr[i]) -
self.MINTEMP.get()) / TempSpan
# Convert to HSV colors (only hue used)
h = normtemp * self.HUEspan + self.HUEstart
if h > 1:
print(h)
print(normtemp)
print(self.HUEspan)
print(self.HUEstart)
# convert to RGB colors
bgrgb = tuple(
int(255 * j) for j in colorsys.hsv_to_rgb(h, 1, 1))
# Convert to Hex String
tarrpix.config(bg=('#%02x%02x%02x' % bgrgb))
i += 1 # incement tarr counter
else:
print("Error - temperarure array lenth wrong")
# recoursive function call all 10 ms (get_tarr will need about 100 ms to respond)
self.frameTarr.after(10, self.update_tarrpixels)
class GridEYE_Viewer():
def __init__(self, root):
""" Initialize Window """
self.tkroot = root
self.tkroot.protocol(
'WM_DELETE_WINDOW',
self.exitwindow) # Close serial connection and close window
""" Initialize variables for color interpolation """
self.HUEstart = 0.5 # initial color for min temp (0.5 = blue)
self.HUEend = 1 # initial color for max temp (1 = red)
self.HUEspan = self.HUEend - self.HUEstart
""" Grid Eye related variables"""
self.MULTIPLIER = 0.25 # temp output multiplier
""" Initialize Loop bool"""
self.START = False
"""Initialize frame tor temperature array (tarr)"""
self.frameTarr = tk.Frame(master=self.tkroot, bg='white')
self.frameTarr.place(x=5, y=5, width=400, height=400)
"""Initialize pixels tor temperature array (tarr)"""
self.tarrpixels = []
for i in range(8):
# frameTarr.rowconfigure(i,weight=1) # self alignment
# frameTarr.columnconfigure(i,weight=1) # self alignment
for j in range(8):
pix = tk.Label(master=self.frameTarr, bg='gray', text='11')
spacerx = 1
spacery = 1
pixwidth = 40
pixheight = 40
pix.place(x=spacerx + j * (spacerx + pixwidth),
y=spacery + i * (pixheight + spacery),
width=pixwidth,
height=pixheight)
print(self.tarrpixels.append(
pix)) # attache all pixels to tarrpixel list
"""Initialize frame tor Elements"""
self.frameElements = tk.Frame(master=self.tkroot, bg='white')
self.frameElements.place(x=410, y=5, width=100, height=400)
"""Initialize controll buttons"""
self.buttonStart = tk.Button(master=self.frameElements,
text='start',
bg='white',
command=self.start_update)
self.buttonStart.pack()
self.buttonStop = tk.Button(master=self.frameElements,
text='stop',
bg='white',
command=self.stop_update)
self.buttonStop.pack()
self.buttonSave = tk.Button(master=self.frameElements,
text='Save',
bg='white',
command=self.save_to_file)
self.buttonSave.pack()
self.labelTop = tk.Label(master=self.frameElements, text="Gest:")
self.labelTop.pack()
self.combobox = ttk.Combobox(master=self.frameElements,
values=[
"gest0", "gest1", "gest2", "gest3",
"gest4", "gest5", "gest6", "gest7",
"gest8"
])
self.combobox.pack()
self.combobox.current(0)
"""Initialize temperature adjustment"""
self.lableTEMPMAX = tk.Label(master=self.frameElements,
text='Max Temp (red)')
self.lableTEMPMAX.pack()
self.MAXTEMP = tk.Scale(self.frameElements,
from_=-20,
to=120,
resolution=0.25)
self.MAXTEMP.set(31)
self.MAXTEMP.pack()
self.lableMINTEMP = tk.Label(master=self.frameElements,
text='Min Temp (blue)')
self.lableMINTEMP.pack()
self.MINTEMP = tk.Scale(self.frameElements,
from_=-20,
to=120,
resolution=0.25)
self.MINTEMP.set(27)
self.MINTEMP.pack()
self.kit = GridEYEKit()
def exitwindow(self):
""" if windwow is clsoed, serial connection has to be closed!"""
self.kit.close()
self.tkroot.destroy()
def save_to_file(self):
"""Nasza funkcja do zapisu"""
tarr = self.get_tarr()
#print(tarr)
global index
global first_time
global counting
#print("AAAAAAAA", self.combobox.current(), self.combobox.get()) ## wartość comboboxa
counting = True
file_index = open("index.txt", "w+")
print("index: ", index)
#print(str(index))
file_saved = open("saved.txt", "a")
file_saved.write(str(index) + "_" + str(self.combobox.current()))
file_saved.write("\n")
file_saved.close()
first_time = time.time()
global last_time
last_time = first_time
file_index.write(str(index))
file_index.close()
def stop_update(self):
""" stop button action - stops infinite loop """
self.START = False
self.update_tarrpixels()
def get_weights(self):
""" nasz kod do pobierania wag"""
global Num_of_Middle
global Num_of_Outputs
global Middle_Weights
global Output_Weights
for m in range(Num_of_Middle):
text = "wages" + str(m) + ".txt"
source = open(text, "r")
vect = np.array([], dtype=np.float)
for x in range(256):
text = source.readline()
var = float(text)
vect = np.append(vect, var)
source.close()
if Middle_Weights.size == 0:
Middle_Weights = vect
else:
Middle_Weights = np.vstack((Middle_Weights, vect))
for m in range(Num_of_Outputs):
text = "output_wages" + str(m) + ".txt"
source = open(text, "r")
vect = np.array([], dtype=np.float)
for x in range(Num_of_Middle):
text = source.readline()
var = float(text)
vect = np.append(vect, var)
source.close()
if Output_Weights.size == 0:
Output_Weights = vect
else:
Output_Weights = np.vstack((Output_Weights, vect))
print("Załadowano wagi!")
def start_update(self):
if self.kit.connect():
""" start button action -start serial connection and start pixel update loop"""
self.START = True
self.get_weights()
global index
file_index = open("index.txt", "r")
index = int(file_index.read())
index += 1
print(index)
file_index.close()
""" CAUTION: Wrong com port error is not handled"""
self.update_tarrpixels()
else:
#tkMessageBox.showerror("Not connected",
messagebox.showerror(
"Not connected",
"Could not find Grid-EYE Eval Kit - please install driver and connect"
)
def get_tarr(self):
""" unnecessary function - only converts numpy array to tuple object"""
tarr = []
for temp in self.kit.get_temperatures(
): # only fue to use of old rutines
for temp2 in temp:
tarr.append(temp2)
return tarr
def process_tarr(self):
" Nasza własna funkcja"
tarr_copy = self.get_tarr()
tarr_proc = []
tarr = tarr_copy
maxx = max(tarr)
minn = min(tarr)
diff = maxx - minn
mid = (maxx + minn) / 2
for x in range(len(tarr)):
temp = (tarr[x] - mid) / diff
tarr_proc.append(temp)
tarr_copy.clear()
#input("Press Enter to continue...")
return tarr_copy
def write_samples(self):
global samples
global index
idx = str(index)
file_example = open(idx + "_" + str(self.combobox.current()) + ".txt",
"a")
for x in range(256):
file_example.write(str(samples[x]))
file_example.write("\n")
file_example.close()
samples.clear()
index += 1
def transfer(self): ### przepuszczanie przez sieć
global Input
global Middle_Weights
global Output_Weights
global Num_of_Middle
global Num_of_Outputs
global Output
Between = np.zeros(Num_of_Middle)
for k in range(Num_of_Middle):
temp = 0.0
for ix in range(256):
temp += Middle_Weights[k][ix] * Input[ix]
Between[k] = 1.0 / (1.0 + np.exp((-1) * temp))
for k in range(Num_of_Outputs):
temp = 0.0
for ix in range(Num_of_Middle):
temp += Output_Weights[k][ix] * Between[ix]
Output[k] = 1.0 / (1.0 + np.exp((-1) * temp)) ## zapisane wyjscie
def update_tarrpixels(self):
""" Loop for updating pixels with values from funcion "get_tarr" - recursive function with exit variable"""
if self.START == True:
tarr = self.get_tarr() # Get temerature array
#tarr = self.process_tarr()
global counting
global transfering
if counting: ### kod odpowiedzialny za zapisywanie przykladow do nauki
global last_time
timex = time.time()
diff = timex - last_time
global iteration
global samples
if (diff > 0.25) & (iteration > 0):
tarr_proc = []
maxx = max(tarr)
minn = min(tarr)
diff = maxx - minn
mid = (maxx + minn) / 2
for x in range(len(tarr)):
temp = (tarr[x] - mid) / diff
tarr_proc.append(temp)
samples += tarr_proc # dolozenie
tarr_proc.clear()
last_time = timex
iteration += 1
print("Iteration: ", iteration)
if iteration > 3:
iteration = 0
counting = False
self.write_samples()
elif (diff > 1.0) & (iteration == 0):
tarr_proc = []
maxx = max(tarr)
minn = min(tarr)
diff = maxx - minn
mid = (maxx + minn) / 2
for x in range(len(tarr)):
temp = (tarr[x] - mid) / diff
tarr_proc.append(temp)
iteration += 1
print("Iteration: ", iteration)
last_time = timex
samples += tarr_proc # dolozenie
tarr_proc.clear()
elif iteration == 0:
print("Time to start: ", round(1.0 - diff, 2), "s")
elif transfering: ### kod odpowiedzialny za sprawdzanie w czasie rzeczywistym
global last_time_check
timex = time.time()
diff = timex - last_time_check
if diff > 0.25:
global Input
temp = np.array([])
maxx = max(tarr)
minn = min(tarr)
diff = maxx - minn
mid = (maxx + minn) / 2
for x in range(64):
tempxx = (tarr[x] - mid) / diff
temp = np.append(temp, 2 * tempxx)
if Input.size == 0:
Input = temp
else:
Input = np.hstack((Input, temp))
last_time_check = time.time()
if Input.size == 320:
Input = Input[64:320]
self.transfer()
global Output
maxxx = max(Output)
summ = sum(Output)
for m in range(Num_of_Outputs):
if maxxx == Output[m]:
if ((maxxx > 0.80) & ((summ - maxxx) < 0.5)):
if m == 0:
print("Gest0: Pięść")
elif m == 1:
print("Gest1: Otwarta dłoń")
elif m == 2:
print(
"Gest2: Otwarta dłoń zbliżająca się"
)
elif m == 3:
print("Gest3: jeden palec")
elif m == 4:
print("Gest4: dwa palce")
elif m == 5:
print("Gest5: Dłoń przesuwana w lewo")
elif m == 6:
print("Gest6: Dłoń przesuwana w prawo")
#elif m == 7:
# print("Gest7: NaN")
i = 0
if len(tarr) == len(
self.tarrpixels): # check if problem with readout
for tarrpix in self.tarrpixels:
tarrpix.config(text=tarr[i]) # Update Pixel text
if tarr[i] < self.MINTEMP.get(
): # For colors, set borders to min/max temp
normtemp = 0
elif tarr[i] > self.MAXTEMP.get(
): # For colors, set borders to min/max temp
normtemp = 1
else:
TempSpan = self.MAXTEMP.get() - self.MINTEMP.get()
if TempSpan <= 0: # avoid division by 0 and negative values
TempSpan = 1
normtemp = (float(tarr[i]) - self.MINTEMP.get()
) / TempSpan # Normalize temperature 0...1
h = normtemp * self.HUEspan + self.HUEstart # Convert to HSV colors (only hue used)
if h > 1:
print(h)
print(normtemp)
print(self.HUEspan)
print(self.HUEstart)
bgrgb = tuple(255 * j for j in colorsys.hsv_to_rgb(
h, 1, 1)) # convert to RGB colors
lixt = list(bgrgb)
for x in range(len(lixt)):
tempx = lixt[x]
lixt[x] = int(tempx)
bgrgb = tuple(lixt)
tarrpix.config(bg=('#%02x%02x%02x' %
bgrgb)) # Convert to Hex String
#tarrpix.config(bg="ivory")
i += 1 # incement tarr counter
else:
print("Error - temperarure array lenth wrong")
self.frameTarr.after(
10, self.update_tarrpixels
) # recoursive function call all 10 ms (get_tarr will need about 100 ms to respond)