Spectral_Visualization/Spectral_traid_graph_2d.py at master · kaabir/Spectral_Visualization · GitHub

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
import serial
import numpy   as np
import matplotlib.pyplot as plt
from drawnow import *
from serial import Serial
from scipy.interpolate import interp1d

spectreData = serial.Serial("COM23", 115200)

loop=True

spectreReadings = [0.0,0.0,0.0,0.0,0.0,0.0]
x = np.linspace(450.00, 650.00, num=6, endpoint=True)
defaultYLimit = 860
xnew = np.linspace(450.00, 650.00, num=6, endpoint=True)
fig = plt.figure()
plt.ion()

def handle_close(evt):
	spectreData.reset_input_buffer()
	plt.ioff()
	plt.close()
	global loop
	loop=False
	#print("close event")
	#exit()
	#sys.exit()

def doPlot():
	fig.canvas.mpl_connect('close_event', handle_close) # figured out finally how to close the "groundhog" window
	plt.ylim(0,defaultYLimit)
	plt.title('Spectral Response')
	plt.grid(True)
	plt.ylabel('Intensity Count')
	plt.xlabel('Wavelength in nm')
	#plt.plot(spectreReadings, 'ro-', label='Spectral readings')
	plt.plot(x, spectreReadings, 'o', xnew, f(xnew), '-')
	plt.legend(loc='upper left')

while (loop):
	while (spectreData.inWaiting()== 0):
		pass
	spectreString = spectreData.readline().decode('utf8')
	spectreList = spectreString.split(",")
	if(len(spectreList)==6):
		for num, value in enumerate(spectreList,start=0):
			spectreReadings[num]=float(value)

		defaultYLimit=max(spectreReadings)*1.3 # thus, the cubic spline should remain inside the plotarea
		f=interp1d(x,spectreReadings,kind='cubic')

		drawnow(doPlot)
print(".")