firmware/pyInterface.py at master · OpenAFM/firmware · 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
54
55
56
57
58
59
60
61
62
63
64
65
from	time	import	sleep
import	serial
import	matplotlib
matplotlib.use('TkAgg')
import	matplotlib.pyplot	as	plt
import	matplotlib.cm	as	cm
from	matplotlib	import	animation
import	re

figure=plt.figure()
ser	=	serial.Serial('COM10',	9600)	#	Establish	the	connection	on	a	specific	port
img=[]
ser.write('RDY;')
def	update():
	buffer_string	=	''
	print	"read	data"
	ser.write('RDY;')	#	Convert	the	decimal	number	to	ASCII	then	send	it	to	the	Arduino
	sleep(3)
		#buffer_string	=	buffer_string	+	ser.read(ser.inWaiting())
		#if	'RDY;'	in	buffer_string:
		#	#print	buffer_string
		#	buffer_string	=	''
		#elif	';'	in	buffer_string:
		#	#data=buffer_string
		#	print buffer_string#break
	data=ser.read(ser.inWaiting())
	print data
	data=data[4:]
	#	Read	the	newest	output	from	the	Arduino

	matchnum=re.compile("[0-9]")
	matchdata=re.compile(".*;")
	try:
		str=matchnum.search(data[0:5]).group()
	except:
		str=""

	if	not	str:
		print	"empty"
		return	update()
	if	str:
		data=matchdata.search(data).group()
		data=[int(x)	for	x	in	data.split(',')[:-1]	if	x]
		print	data
		return	data

def	animate(frame):
	plt.cla()
	data=update()
	img.append(data)
	plt.imshow(img,	cmap=cm.Greys_r)

#while	True:
#		print	'go'
#		ser.write('GO;')
#		sleep(.1)
#		data=	ser.readline()
#		print	'done'
#		print	data
#		if	data=='GO':
#			break;

anim	=	animation.FuncAnimation(figure,	animate,
								frames=50,	interval=50)
plt.show()